201 lines
5.6 KiB
Python
201 lines
5.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
光路可视化Blender脚本
|
|
读取miao_light_path_tsingtao.json文件并使用Blender绘制3D光路模型
|
|
|
|
使用方法:
|
|
blender --background --python light_path_simple.py
|
|
|
|
输出文件:
|
|
- light_path_simple.blend - Blender工程文件
|
|
- light_path_simple.glb - Web友好格式
|
|
- light_path_simple_render.png - 渲染图像
|
|
"""
|
|
|
|
import bpy
|
|
import json
|
|
import os
|
|
import numpy as np
|
|
|
|
# 清空场景
|
|
bpy.ops.wm.read_factory_settings(use_empty=True)
|
|
|
|
def load_light_paths(filename):
|
|
"""加载光路数据"""
|
|
try:
|
|
with open(filename, 'r', encoding='utf-8') as f:
|
|
data = json.load(f)
|
|
print(f"成功加载光路文件: {filename}")
|
|
print(f"包含 {len(data)} 条光路")
|
|
return data
|
|
except Exception as e:
|
|
print(f"加载光路文件失败: {e}")
|
|
return []
|
|
|
|
def create_light_path_material(color=(1.0, 0.2, 0.2, 1.0)):
|
|
"""创建光路材质"""
|
|
mat = bpy.data.materials.new(name="light_path")
|
|
mat.use_nodes = True
|
|
nodes = mat.node_tree.nodes
|
|
links = mat.node_tree.links
|
|
|
|
# 清除默认节点
|
|
nodes.clear()
|
|
|
|
# 创建发光节点
|
|
emission = nodes.new(type='ShaderNodeEmission')
|
|
emission.inputs['Color'].default_value = color
|
|
emission.inputs['Strength'].default_value = 1.0
|
|
|
|
# 创建输出节点
|
|
output = nodes.new(type='ShaderNodeOutputMaterial')
|
|
|
|
# 连接节点
|
|
links.new(emission.outputs['Emission'], output.inputs['Surface'])
|
|
|
|
return mat
|
|
|
|
def create_light_path_curve(points, name="light_path"):
|
|
"""创建光路曲线"""
|
|
# 创建曲线数据
|
|
curve_data = bpy.data.curves.new(name, type='CURVE')
|
|
curve_data.dimensions = '3D'
|
|
curve_data.resolution_u = 12
|
|
curve_data.bevel_depth = 0.02
|
|
curve_data.bevel_resolution = 4
|
|
|
|
# 创建样条线
|
|
spline = curve_data.splines.new('POLY')
|
|
spline.points.add(len(points) - 1)
|
|
|
|
# 设置点坐标
|
|
for i, point in enumerate(points):
|
|
spline.points[i].co = (point[0], point[1], point[2], 1.0)
|
|
|
|
# 创建曲线对象
|
|
curve_obj = bpy.data.objects.new(name, curve_data)
|
|
bpy.context.collection.objects.link(curve_obj)
|
|
|
|
return curve_obj
|
|
|
|
def create_light_paths(light_paths):
|
|
"""创建所有光路"""
|
|
objects = []
|
|
|
|
# 颜色方案
|
|
colors = [
|
|
(1.0, 0.2, 0.2, 1.0), # 红色
|
|
(0.2, 1.0, 0.2, 1.0), # 绿色
|
|
(0.2, 0.2, 1.0, 1.0), # 蓝色
|
|
(1.0, 1.0, 0.2, 1.0), # 黄色
|
|
(1.0, 0.2, 1.0, 1.0), # 紫色
|
|
]
|
|
|
|
for i, path in enumerate(light_paths[len(light_paths):]):
|
|
if len(path) < 2:
|
|
continue
|
|
|
|
# 选择颜色
|
|
color = colors[i % len(colors)]
|
|
|
|
# 创建材质
|
|
material = create_light_path_material(color)
|
|
|
|
# 创建光路对象
|
|
obj = create_light_path_curve(path, f"light_path_{i}")
|
|
|
|
# 应用材质
|
|
obj.data.materials.append(material)
|
|
|
|
objects.append(obj)
|
|
|
|
print(f"创建光路 {i+1}: {len(path)} 个点")
|
|
|
|
return objects
|
|
|
|
def setup_scene():
|
|
"""设置场景"""
|
|
# 添加相机
|
|
bpy.ops.object.camera_add(location=(10, -10, 8))
|
|
camera = bpy.context.active_object
|
|
camera.rotation_euler = (1.0, 0, 0.785)
|
|
bpy.context.scene.camera = camera
|
|
|
|
# 添加灯光
|
|
bpy.ops.object.light_add(type='SUN', location=(5, 5, 10))
|
|
sun = bpy.context.active_object
|
|
sun.data.energy = 2.0
|
|
|
|
# 设置世界背景
|
|
world = bpy.context.scene.world
|
|
if world is None:
|
|
world = bpy.data.worlds.new("World")
|
|
bpy.context.scene.world = world
|
|
|
|
world.use_nodes = True
|
|
bg_node = world.node_tree.nodes['Background']
|
|
bg_node.inputs['Color'].default_value = (0.05, 0.05, 0.1, 1.0)
|
|
bg_node.inputs['Strength'].default_value = 0.5
|
|
|
|
def save_files():
|
|
"""保存文件"""
|
|
try:
|
|
# 保存Blender文件
|
|
blend_path = os.path.abspath("./light_path_simple.blend")
|
|
bpy.ops.wm.save_as_mainfile(filepath=blend_path)
|
|
print(f"Blender文件已保存为: {blend_path}")
|
|
|
|
# 导出GLB文件
|
|
glb_path = os.path.abspath("./light_path_simple.glb")
|
|
bpy.ops.export_scene.gltf(
|
|
filepath=glb_path,
|
|
export_materials='EXPORT'
|
|
)
|
|
print(f"GLB文件已导出为: {glb_path}")
|
|
|
|
# 渲染图像
|
|
scene = bpy.context.scene
|
|
scene.render.resolution_x = 1280
|
|
scene.render.resolution_y = 720
|
|
scene.render.image_settings.file_format = 'PNG'
|
|
|
|
render_path = os.path.abspath("./light_path_simple_render.png")
|
|
scene.render.filepath = render_path
|
|
bpy.ops.render.render(write_still=True)
|
|
print(f"渲染图像已保存为: {render_path}")
|
|
|
|
return True
|
|
except Exception as e:
|
|
print(f"保存文件时出错: {e}")
|
|
return False
|
|
|
|
def main():
|
|
"""主函数"""
|
|
print("🎨 光学系统光路可视化 - 简化版")
|
|
print("=" * 40)
|
|
|
|
# 加载光路数据
|
|
light_paths = load_light_paths("miao_light_path_tsingtao.json")
|
|
|
|
if not light_paths:
|
|
print("❌ 无法加载光路数据")
|
|
return
|
|
|
|
print(f"处理前10条光路...")
|
|
|
|
# 创建光路对象
|
|
light_path_objects = create_light_paths(light_paths)
|
|
|
|
# 设置场景
|
|
setup_scene()
|
|
|
|
# 保存文件
|
|
if save_files():
|
|
print("\n🎉 任务完成！")
|
|
print(f"成功创建了 {len(light_path_objects)} 条光路")
|
|
else:
|
|
print("\n❌ 保存文件失败")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|