增加环绕侦察场景适配

tools/test_validate/modules/json_visualizer.py

@@ -0,0 +1,101 @@
+import os
+import json
+import logging
+import time
+from .visualizer import generate_visualization
+
+# 配置日志
+logging.basicConfig(level=logging.INFO, format='%(message)s')
+
+def run_json_visualization():
+    print("\n🚀 进入 JSON 文件可视化模式")
+    print("说明: 读取包含行为树(Pytree)结构的 JSON 文件并生成 PNG 图片。")
+    print("      支持的 JSON 结构: 直接是 Pytree 根节点对象，或包含 'root'/'data' 字段的对象。")
+    
+    while True:
+        try:
+            file_path = input("\n请输入 JSON 文件路径 (输入 'q' 返回): ").strip()
+            
+            if file_path.lower() in ['q', 'exit', 'quit']:
+                break
+                
+            if not file_path:
+                continue
+                
+            # 处理相对路径
+            if not os.path.isabs(file_path):
+                # 尝试相对于当前工作目录
+                if not os.path.exists(file_path):
+                    # 尝试相对于 tools/test_validate
+                    base_dir = os.path.dirname(os.path.dirname(__file__))
+                    alt_path = os.path.join(base_dir, file_path)
+                    if os.path.exists(alt_path):
+                        file_path = alt_path
+            
+            if not os.path.exists(file_path):
+                print(f"❌ 文件不存在: {file_path}")
+                continue
+                
+            if not os.path.isfile(file_path):
+                print(f"❌ 路径不是一个文件: {file_path}")
+                continue
+                
+            print(f"📖 正在读取: {file_path}")
+            try:
+                with open(file_path, 'r', encoding='utf-8') as f:
+                    data = json.load(f)
+            except json.JSONDecodeError as e:
+                print(f"❌ JSON 解析失败: {e}")
+                continue
+            except Exception as e:
+                print(f"❌ 读取文件失败: {e}")
+                continue
+                
+            # 尝试提取 root 节点
+            root_node = None
+            
+            # 情况 1: 顶层就是 root 节点 (包含 name 和 type)
+            if isinstance(data, dict) and 'name' in data and 'type' in data:
+                root_node = data
+                print("ℹ️  识别到顶层即为 Pytree 根节点")
+            
+            # 情况 2: 包含 root 字段
+            elif isinstance(data, dict) and 'root' in data:
+                root_node = data['root']
+                print("ℹ️  从 'root' 字段提取节点")
+                
+            # 情况 3: 包含 data 字段，data 里有 root
+            elif isinstance(data, dict) and 'data' in data and isinstance(data['data'], dict) and 'root' in data['data']:
+                root_node = data['data']['root']
+                print("ℹ️  从 'data.root' 字段提取节点")
+
+            # 情况 4: 包含 data 字段，data 本身就是 root
+            elif isinstance(data, dict) and 'data' in data and isinstance(data['data'], dict) and 'name' in data['data']:
+                root_node = data['data']
+                print("ℹ️  从 'data' 字段提取节点")
+                
+            if not root_node:
+                print("❌ 无法识别 Pytree 结构。JSON 对象必须包含 'name' 和 'type' 字段，或在 'root'/'data' 字段中包含它们。")
+                continue
+                
+            # 生成输出路径
+            dir_name = os.path.dirname(file_path)
+            base_name = os.path.splitext(os.path.basename(file_path))[0]
+            output_path = os.path.join(dir_name, f"{base_name}.png")
+            
+            print("🎨 正在生成可视化...")
+            if generate_visualization(root_node, output_path):
+                print(f"✅ 可视化图已生成: {output_path}")
+                # 尝试打开图片 (仅限支持的系统)
+                # if os.name == 'posix':
+                #    os.system(f"xdg-open '{output_path}'")
+            else:
+                print("❌ 生成失败，请检查 graphviz 是否安装正确")
+                
+        except KeyboardInterrupt:
+            print("\n已取消")
+            break
+        except Exception as e:
+            print(f"❌ 发生未预期的错误: {e}")
+            
+    print("\n已返回主菜单")