171 lines
5.2 KiB
Python
171 lines
5.2 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
运行脚本 - 使用Blender生成3D光学系统模型
|
||
|
||
使用方法:
|
||
1. 确保已安装Blender
|
||
2. 运行此脚本: python run_blender_input.py -i jsonfile.json
|
||
|
||
输出文件:
|
||
- optical_system.obj - 3D模型文件
|
||
- optical_system.glb - Web友好格式
|
||
- optical_system.blend - Blender工程文件
|
||
- optical_system_render.png - 渲染图像
|
||
"""
|
||
|
||
import subprocess
|
||
import sys
|
||
import os
|
||
import platform
|
||
import argparse
|
||
|
||
def find_blender():
|
||
"""查找Blender可执行文件"""
|
||
system = platform.system()
|
||
|
||
common_paths = []
|
||
if system == "Windows":
|
||
common_paths = [
|
||
r"C:\Program Files\Blender Foundation\Blender 3.6\blender.exe",
|
||
r"C:\Program Files\Blender Foundation\Blender 4.0\blender.exe",
|
||
r"C:\Program Files (x86)\Blender Foundation\Blender 3.6\blender.exe",
|
||
]
|
||
elif system == "Darwin": # macOS
|
||
common_paths = [
|
||
"/Applications/Blender.app/Contents/MacOS/Blender",
|
||
"/usr/local/bin/blender",
|
||
]
|
||
elif system == "Linux":
|
||
common_paths = [
|
||
"/usr/bin/blender",
|
||
"/usr/local/bin/blender",
|
||
"/snap/bin/blender",
|
||
]
|
||
|
||
# 首先尝试从PATH中找到blender
|
||
try:
|
||
result = subprocess.run(["which", "blender"], capture_output=True, text=True)
|
||
if result.returncode == 0:
|
||
return result.stdout.strip()
|
||
except:
|
||
pass
|
||
|
||
# 尝试常见路径
|
||
for path in common_paths:
|
||
if os.path.exists(path):
|
||
return path
|
||
|
||
return None
|
||
|
||
def run_blender_script(json_file):
|
||
"""运行Blender脚本"""
|
||
blender_path = find_blender()
|
||
|
||
if not blender_path:
|
||
print("❌ 错误：未找到Blender安装")
|
||
print("\n请确保已安装Blender并且可以从命令行访问")
|
||
print("下载地址: https://www.blender.org/download/")
|
||
return False
|
||
|
||
print(f"✅ 找到Blender: {blender_path}")
|
||
|
||
script_path = os.path.abspath("toModel.py")
|
||
if not os.path.exists(script_path):
|
||
print(f"❌ 错误：未找到脚本文件 {script_path}")
|
||
return False
|
||
|
||
json_path = os.path.abspath(json_file)
|
||
if not os.path.exists(json_path):
|
||
print(f"❌ 错误：未找到JSON数据文件 {json_path}")
|
||
return False
|
||
|
||
print(f"📄 使用JSON文件: {json_path}")
|
||
print("🚀 开始运行Blender脚本...")
|
||
print("这可能需要几分钟时间，请耐心等待...")
|
||
|
||
try:
|
||
# 运行Blender脚本，传递JSON文件路径
|
||
cmd = [
|
||
blender_path,
|
||
"--background", # 后台运行
|
||
"--python", script_path,
|
||
"--", # 分隔符，后面的参数会传递给Python脚本
|
||
json_path
|
||
]
|
||
|
||
result = subprocess.run(cmd, capture_output=True, text=True, timeout=300)
|
||
|
||
if result.returncode == 0:
|
||
print("✅ Blender脚本执行成功！")
|
||
print("\n输出信息:")
|
||
print(result.stdout)
|
||
|
||
# 检查生成的文件
|
||
expected_files = [
|
||
"optical_system.obj",
|
||
"optical_system.glb",
|
||
"optical_system.blend",
|
||
"optical_system_render.png"
|
||
]
|
||
|
||
print("\n📁 生成的文件:")
|
||
for filename in expected_files:
|
||
if os.path.exists(filename):
|
||
size = os.path.getsize(filename)
|
||
print(f"✅ {filename} ({size:,} bytes)")
|
||
else:
|
||
print(f"❌ {filename} (未找到)")
|
||
|
||
return True
|
||
else:
|
||
print("❌ Blender脚本执行失败")
|
||
print("错误输出:")
|
||
print(result.stderr)
|
||
if result.stdout:
|
||
print("标准输出:")
|
||
print(result.stdout)
|
||
return False
|
||
|
||
except subprocess.TimeoutExpired:
|
||
print("❌ 脚本执行超时（超过5分钟）")
|
||
return False
|
||
except Exception as e:
|
||
print(f"❌ 执行时出错: {e}")
|
||
return False
|
||
|
||
def main():
|
||
# 设置命令行参数解析
|
||
parser = argparse.ArgumentParser(
|
||
description="使用Blender生成3D光学系统模型",
|
||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||
epilog="""
|
||
使用示例:
|
||
python run_blender_input.py -i data.json
|
||
python run_blender_input.py --input config.json
|
||
"""
|
||
)
|
||
|
||
parser.add_argument(
|
||
"-i", "--input",
|
||
required=True,
|
||
help="输入的JSON文件路径"
|
||
)
|
||
|
||
args = parser.parse_args()
|
||
|
||
print("🎨 Blender 3D光学系统建模脚本")
|
||
print("=" * 40)
|
||
|
||
if run_blender_script(args.input):
|
||
print("\n🎉 任务完成！")
|
||
print("\n您可以:")
|
||
print("1. 使用Blender打开 optical_system.blend 文件进行编辑")
|
||
print("2. 在3D软件中导入 optical_system.obj 文件")
|
||
print("3. 在Web浏览器中查看 optical_system.glb 文件")
|
||
print("4. 查看渲染图像 optical_system_render.png")
|
||
else:
|
||
print("\n❌ 任务失败，请检查错误信息")
|
||
sys.exit(1)
|
||
|
||
if __name__ == "__main__":
|
||
main() |