forked from Rowland/EG
362 lines
13 KiB
Python
362 lines
13 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
FBX模型导入测试 - 演示新的缩放处理选项
|
||
|
||
修复内容：
|
||
1. 默认保持模型原有缩放结构
|
||
2. 提供可选的单位转换功能
|
||
3. 智能缩放标准化（处理子节点大缩放值）
|
||
4. 避免缩放层级混乱问题
|
||
|
||
使用说明：
|
||
1. 运行脚本启动3D编辑器
|
||
2. 通过文件菜单或拖拽导入FBX模型
|
||
3. 观察模型的缩放层级结构
|
||
4. 按U键切换单位转换模式
|
||
5. 按N键切换缩放标准化模式
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
|
||
# 添加主目录到Python路径
|
||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
|
||
|
||
from main import MyWorld
|
||
from ui.main_window import setup_main_window
|
||
from PyQt5.QtCore import Qt
|
||
|
||
|
||
def setup_fbx_import_demo():
|
||
"""设置FBX导入演示"""
|
||
|
||
# 创建世界对象
|
||
world = MyWorld()
|
||
|
||
# 使用新的UI模块创建主窗口
|
||
app, main_window = setup_main_window(world)
|
||
|
||
# 设置窗口标题
|
||
main_window.setWindowTitle("FBX导入测试 - 缩放层级修复")
|
||
|
||
# 设置焦点策略
|
||
main_window.setFocusPolicy(Qt.StrongFocus)
|
||
main_window.setFocus()
|
||
|
||
# 导入选项
|
||
unit_conversion_enabled = False
|
||
scale_normalization_enabled = True # 默认开启缩放标准化
|
||
|
||
def keyPressEvent(event):
|
||
nonlocal unit_conversion_enabled, scale_normalization_enabled
|
||
key = event.key()
|
||
|
||
if key == Qt.Key_U: # U键切换单位转换模式
|
||
unit_conversion_enabled = not unit_conversion_enabled
|
||
status = "开启" if unit_conversion_enabled else "关闭"
|
||
print(f"\n=== 单位转换模式已{status} ===")
|
||
print("下次导入FBX文件时将使用此设置")
|
||
event.accept()
|
||
return
|
||
elif key == Qt.Key_N: # N键切换缩放标准化模式
|
||
scale_normalization_enabled = not scale_normalization_enabled
|
||
status = "开启" if scale_normalization_enabled else "关闭"
|
||
print(f"\n=== 缩放标准化模式已{status} ===")
|
||
print("下次导入FBX文件时将使用此设置")
|
||
event.accept()
|
||
return
|
||
elif key == Qt.Key_I: # I键显示导入信息
|
||
print_import_info(world)
|
||
event.accept()
|
||
return
|
||
elif key == Qt.Key_R: # R键切换射线显示
|
||
state = world.toggleRayDisplay()
|
||
status = "开启" if state else "关闭"
|
||
print(f"\n=== 射线显示已{status} ===")
|
||
event.accept()
|
||
return
|
||
elif key == Qt.Key_S: # S键显示当前设置
|
||
print_current_settings(unit_conversion_enabled, scale_normalization_enabled)
|
||
event.accept()
|
||
return
|
||
|
||
# 调用原始键盘事件处理
|
||
if hasattr(main_window, '_original_keyPressEvent'):
|
||
main_window._original_keyPressEvent(event)
|
||
else:
|
||
event.ignore()
|
||
|
||
# 覆盖importModel方法以使用当前的导入设置
|
||
original_import = world.scene_manager.importModel
|
||
|
||
def enhanced_import(filepath):
|
||
"""增强的导入方法，使用当前导入设置"""
|
||
print(f"\n" + "="*60)
|
||
print(f"导入模型: {os.path.basename(filepath)}")
|
||
print(f"单位转换: {'开启' if unit_conversion_enabled else '关闭'}")
|
||
print(f"缩放标准化: {'开启' if scale_normalization_enabled else '关闭'}")
|
||
print("="*60)
|
||
|
||
result = original_import(
|
||
filepath,
|
||
apply_unit_conversion=unit_conversion_enabled,
|
||
normalize_scales=scale_normalization_enabled
|
||
)
|
||
|
||
if result:
|
||
print("\n导入后的模型结构:")
|
||
print_model_structure(result, max_depth=3, world=world) # 限制显示深度
|
||
|
||
return result
|
||
|
||
world.scene_manager.importModel = enhanced_import
|
||
|
||
# 保存原始键盘事件处理器
|
||
if hasattr(main_window, 'keyPressEvent'):
|
||
main_window._original_keyPressEvent = main_window.keyPressEvent
|
||
main_window.keyPressEvent = keyPressEvent
|
||
|
||
# 添加自定义菜单
|
||
add_custom_menus(main_window, world, unit_conversion_enabled, scale_normalization_enabled)
|
||
|
||
# 输出使用说明
|
||
print_usage_instructions()
|
||
|
||
return app, main_window, world
|
||
|
||
|
||
def add_custom_menus(main_window, world, unit_conversion_enabled, scale_normalization_enabled):
|
||
"""添加自定义菜单选项"""
|
||
# 添加FBX测试菜单
|
||
fbx_menu = main_window.menuBar().addMenu('FBX测试')
|
||
|
||
# 切换单位转换
|
||
toggle_unit_action = fbx_menu.addAction('切换单位转换 (U)')
|
||
toggle_unit_action.triggered.connect(lambda: toggle_unit_conversion())
|
||
|
||
# 切换缩放标准化
|
||
toggle_scale_action = fbx_menu.addAction('切换缩放标准化 (N)')
|
||
toggle_scale_action.triggered.connect(lambda: toggle_scale_normalization())
|
||
|
||
fbx_menu.addSeparator()
|
||
|
||
# 显示当前设置
|
||
settings_action = fbx_menu.addAction('显示当前设置 (S)')
|
||
settings_action.triggered.connect(lambda: print_current_settings(unit_conversion_enabled, scale_normalization_enabled))
|
||
|
||
# 显示导入信息
|
||
info_action = fbx_menu.addAction('显示导入信息 (I)')
|
||
info_action.triggered.connect(lambda: print_import_info(world))
|
||
|
||
# 射线显示切换
|
||
ray_action = fbx_menu.addAction('切换射线显示 (R)')
|
||
ray_action.triggered.connect(lambda: world.toggleRayDisplay())
|
||
|
||
fbx_menu.addSeparator()
|
||
|
||
# 手动标准化当前模型
|
||
normalize_action = fbx_menu.addAction('手动标准化当前模型缩放')
|
||
normalize_action.triggered.connect(lambda: manual_normalize_current_models(world))
|
||
|
||
# 重置所有模型缩放
|
||
reset_action = fbx_menu.addAction('重置所有模型缩放为1.0')
|
||
reset_action.triggered.connect(lambda: reset_all_model_scales(world))
|
||
|
||
def toggle_unit_conversion():
|
||
nonlocal unit_conversion_enabled
|
||
unit_conversion_enabled = not unit_conversion_enabled
|
||
status = "开启" if unit_conversion_enabled else "关闭"
|
||
print(f"\n=== 单位转换模式已{status} ===")
|
||
|
||
def toggle_scale_normalization():
|
||
nonlocal scale_normalization_enabled
|
||
scale_normalization_enabled = not scale_normalization_enabled
|
||
status = "开启" if scale_normalization_enabled else "关闭"
|
||
print(f"\n=== 缩放标准化模式已{status} ===")
|
||
|
||
|
||
def print_model_structure(model, depth=0, max_depth=5, world=None):
|
||
"""打印模型的层级结构、缩放和位置信息"""
|
||
if depth > max_depth:
|
||
return
|
||
|
||
indent = " " * depth
|
||
scale = model.getScale()
|
||
local_pos = model.getPos()
|
||
|
||
# 如果有world引用，显示世界位置
|
||
if world:
|
||
world_pos = model.getPos(world.render)
|
||
pos_info = f"本地{local_pos} / 世界{world_pos}"
|
||
else:
|
||
pos_info = f"{local_pos}"
|
||
|
||
# 计算最大缩放分量
|
||
max_scale = max(abs(scale.x), abs(scale.y), abs(scale.z))
|
||
scale_status = "🔴大" if max_scale > 10 else "🟢正常"
|
||
|
||
print(f"{indent}📦 {model.getName()}")
|
||
print(f"{indent} 缩放: {scale} {scale_status}")
|
||
print(f"{indent} 位置: {pos_info}")
|
||
print(f"{indent} 类型: {model.node().__class__.__name__}")
|
||
|
||
# 递归打印重要子节点（限制数量避免输出过多）
|
||
child_count = model.getNumChildren()
|
||
max_children_to_show = 3 if depth == 0 else 2
|
||
|
||
for i in range(min(child_count, max_children_to_show)):
|
||
child = model.getChild(i)
|
||
print_model_structure(child, depth + 1, max_depth, world)
|
||
|
||
if child_count > max_children_to_show:
|
||
print(f"{indent} ... 还有 {child_count - max_children_to_show} 个子节点")
|
||
|
||
|
||
def print_import_info(world):
|
||
"""打印当前导入的模型信息"""
|
||
print("\n" + "="*60)
|
||
print("🔍 当前场景中的模型信息")
|
||
print("="*60)
|
||
|
||
if not world.models:
|
||
print("❌ 场景中没有模型")
|
||
return
|
||
|
||
for i, model in enumerate(world.models):
|
||
print(f"\n📦 模型 {i+1}: {model.getName()}")
|
||
print(f" 文件: {model.getTag('file') if model.hasTag('file') else '未知'}")
|
||
print(f" 单位转换: {'✅是' if model.hasTag('unit_conversion_applied') else '❌否'}")
|
||
print(f" 缩放标准化: {'✅是' if model.hasTag('scale_normalization_applied') else '❌否'}")
|
||
print(f" 根节点位置: {model.getPos()}")
|
||
print(f" 根节点缩放: {model.getScale()}")
|
||
print(f" 子节点数量: {model.getNumChildren()}")
|
||
|
||
# 检查子节点的缩放情况
|
||
large_scale_children = 0
|
||
if model.getNumChildren() > 0:
|
||
print(" 📋 子节点缩放概况:")
|
||
for j in range(min(5, model.getNumChildren())): # 只检查前5个子节点
|
||
child = model.getChild(j)
|
||
child_scale = child.getScale()
|
||
max_scale = max(abs(child_scale.x), abs(child_scale.y), abs(child_scale.z))
|
||
|
||
if max_scale > 10:
|
||
large_scale_children += 1
|
||
status = "🔴大缩放"
|
||
else:
|
||
status = "🟢正常"
|
||
|
||
print(f" {child.getName()}: {child_scale} {status}")
|
||
|
||
if model.getNumChildren() > 5:
|
||
print(f" ... 还有 {model.getNumChildren() - 5} 个子节点")
|
||
|
||
if large_scale_children > 0:
|
||
print(f" ⚠️ 发现 {large_scale_children} 个子节点有大缩放值")
|
||
|
||
print("="*60)
|
||
|
||
|
||
def print_current_settings(unit_conversion_enabled, scale_normalization_enabled):
|
||
"""显示当前导入设置"""
|
||
print("\n" + "="*40)
|
||
print("⚙️ 当前FBX导入设置")
|
||
print("="*40)
|
||
print(f"单位转换 (U键): {'🟢开启' if unit_conversion_enabled else '🔴关闭'}")
|
||
print(f"缩放标准化 (N键): {'🟢开启' if scale_normalization_enabled else '🔴关闭'}")
|
||
print("\n说明:")
|
||
print("• 单位转换: 将FBX的厘米单位转换为米")
|
||
print("• 缩放标准化: 自动处理子节点的大缩放值(如100)")
|
||
print("="*40)
|
||
|
||
|
||
def manual_normalize_current_models(world):
|
||
"""手动标准化当前所有模型的缩放"""
|
||
print("\n🔧 手动标准化所有模型缩放...")
|
||
|
||
if not world.models:
|
||
print("❌ 没有模型需要处理")
|
||
return
|
||
|
||
for model in world.models:
|
||
print(f"\n处理模型: {model.getName()}")
|
||
world.scene_manager._normalizeModelScales(model)
|
||
|
||
print("✅ 手动标准化完成")
|
||
|
||
|
||
def reset_all_model_scales(world):
|
||
"""重置所有模型的缩放为1.0（调试用）"""
|
||
print("\n🔄 重置所有模型缩放为1.0...")
|
||
|
||
count = 0
|
||
for model in world.models:
|
||
# 递归重置所有节点
|
||
reset_node_scale_recursive(model)
|
||
count += 1
|
||
|
||
print(f"✅ 已重置 {count} 个模型的所有节点缩放")
|
||
|
||
|
||
def reset_node_scale_recursive(node, depth=0):
|
||
"""递归重置节点缩放"""
|
||
indent = " " * depth
|
||
node.setScale(1.0, 1.0, 1.0)
|
||
print(f"{indent}重置 {node.getName()}")
|
||
|
||
for i in range(node.getNumChildren()):
|
||
child = node.getChild(i)
|
||
reset_node_scale_recursive(child, depth + 1)
|
||
|
||
|
||
def print_usage_instructions():
|
||
"""打印使用说明"""
|
||
print("\n" + "="*60)
|
||
print("🚀 FBX导入测试启动完成！")
|
||
print("="*60)
|
||
print("🎯 主要改进：")
|
||
print("✅ 智能缩放标准化 - 自动处理子节点大缩放值")
|
||
print("✅ 保持模型原有缩放结构")
|
||
print("✅ 避免根节点0.01 + 子节点100的复杂层级")
|
||
print("✅ 缩放时保持世界位置不变 - 修复位置偏移问题")
|
||
print("✅ 提供灵活的导入选项")
|
||
print("")
|
||
print("⌨️ 键盘快捷键：")
|
||
print("• U键 - 切换单位转换模式")
|
||
print("• N键 - 切换缩放标准化模式")
|
||
print("• S键 - 显示当前设置")
|
||
print("• I键 - 显示模型信息")
|
||
print("• R键 - 切换射线显示")
|
||
print("")
|
||
print("📁 导入方式：")
|
||
print("• 拖拽FBX文件到3D场景")
|
||
print("• 使用菜单 [文件] -> [导入模型]")
|
||
print("• 使用菜单 [FBX测试] 查看更多选项")
|
||
print("")
|
||
print("🎛️ 缩放处理模式：")
|
||
print("• 关闭单位转换 + 开启缩放标准化（推荐）")
|
||
print(" → 保持FBX结构，但标准化大缩放值")
|
||
print("• 开启单位转换 + 关闭缩放标准化")
|
||
print(" → 传统方式，应用0.01根缩放")
|
||
print("• 两者都关闭")
|
||
print(" → 完全保持原始FBX结构")
|
||
print("")
|
||
print("📊 当前设置：")
|
||
print("• 单位转换: 🔴关闭")
|
||
print("• 缩放标准化: 🟢开启 (推荐)")
|
||
print("="*60)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
app, main_window, world = setup_fbx_import_demo()
|
||
|
||
# 启动应用程序
|
||
sys.exit(app.exec_())
|
||
|
||
except Exception as e:
|
||
print(f"❌ 启动失败: {str(e)}")
|
||
import traceback
|
||
traceback.print_exc() |