import json
import panda3d.core as p3d
from typing import List

class TreeInfo:
    def __init__(self, root: p3d.NodePath):
        self.name = root.getName()
        self.root = root
        self.type = self.classify_np(root)
        self.childs: List[TreeInfo] = []
        self._build_children()
        
    def classify_np(self,np: p3d.NodePath) -> str:
        node = np.node()

        if isinstance(node, p3d.GeomNode):
            return "mesh(GeomNode)"
        if isinstance(node, p3d.CollisionNode):
            return "collision(CollisionNode)"
        if isinstance(node, p3d.Camera):
            return "camera"
        if isinstance(node, p3d.Light):
            return "light"
        if isinstance(node, p3d.TextNode):
            return "text"
        
        # 不同节点可能是模型根、LOD根等，但核心是：是否包含 Geom/Collision
        if not np.find("**/+GeomNode").isEmpty():
            return "model(has GeomNode children)"
        if not np.find("**/+CollisionNode").isEmpty():
            return "collision_parent(has CollisionNode children)"

    def _build_children(self):
        if not self.root or self.root.isEmpty():
            return
        for child in self.root.getChildren():
            child_info = TreeInfo(child)  # 递归构建子树
            self.childs.append(child_info)

    def to_dict(self):
        return {
            "name": self.name,
            "type": self.type or "",
            "children": [child.to_dict() for child in self.childs],
        }
        
    def save_tree_json(self,path: str):
        """保存树的 name/type 结构为 JSON。"""
        data = self.to_dict()
        with open(path, "w", encoding="utf-8") as f:
            json.dump(data, f, ensure_ascii=False, indent=2)