Upload code/cube3d/render/render_bricks_safe.py with huggingface_hub

code/cube3d/render/render_bricks_safe.py

@@ -305,7 +305,7 @@ def _manual_import_importldraw(module_dir: Path):
 try:
     print("尝试导入ImportLDraw...")
     import ImportLDraw
-    from ImportLDraw.loadldraw.loadldraw import FileSystem
+    from ImportLDraw.loadldraw.loadldraw import Options, Configure, loadFromFile, FileSystem
     print("ImportLDraw导入成功")
     print(f"ImportLDraw文件位置: {{ImportLDraw.__file__}}")
 except ImportError as import_error:
@@ -314,7 +314,7 @@ except ImportError as import_error:
     try:
         print("尝试手动加载 ImportLDraw...")
         ImportLDraw = _manual_import_importldraw(IMPORT_LDRAW_MODULE_DIR)
-        from ImportLDraw.loadldraw.loadldraw import FileSystem
+        from ImportLDraw.loadldraw.loadldraw import Options, Configure, loadFromFile, FileSystem
         print("✅ 手动加载 ImportLDraw 成功")
     except Exception as manual_error:
         print(f"❌ 手动加载 ImportLDraw 失败: {{manual_error}}")
@@ -340,32 +340,51 @@ try:
             for i, line in enumerate(content.split('\\n')[:5]):
                 print(f"  {{i+1}}: {{line}}")
 
-    print("清理场景（为导入做准备）...")
+    # 设置 ImportLDraw 选项（demo.zip 原始实现）
+    print("设置 ImportLDraw Options...")
+    Options.ldrawDirectory = "{ldraw_path}"
+    Options.instructionsLook = {instructions_look_bool}
+    Options.useLogoStuds = True
+    Options.useUnofficialParts = True
+    Options.gaps = True
+    Options.studLogoDirectory = os.path.join(str(plugin_path), 'studs')
+    Options.LSynthDirectory = os.path.join(str(plugin_path), 'lsynth')
+    Options.verbose = 1
+    Options.overwriteExistingMaterials = True
+    Options.overwriteExistingMeshes = True
+    Options.scale = 0.01
+    Options.createInstances = True
+    Options.removeDoubles = True
+    Options.positionObjectOnGroundAtOrigin = True
+    Options.flattenHierarchy = False
+    Options.edgeSplit = True
+    Options.addBevelModifier = True
+    Options.bevelWidth = 0.5
+    Options.addEnvironmentTexture = True
+    Options.scriptDirectory = os.path.join(str(plugin_path), 'loadldraw')
+    Options.addWorldEnvironmentTexture = True
+    Options.addGroundPlane = True
+    Options.setRenderSettings = True
+    Options.removeDefaultObjects = True
+    Options.positionCamera = {reposition_camera_bool}
+    Options.cameraBorderPercent = 0.05
+    print("✅ Options 设置完成")
+
+    print("执行 Configure()...")
+    Configure()
+    print("✅ Configure 执行成功")
+
+    print("清理场景...")
     bpy.ops.object.select_all(action='DESELECT')
     bpy.ops.object.select_by_type(type='MESH')
     bpy.ops.object.delete()
     print("场景清理完成")
 
-    # 注册 ImportLDraw operator（官方推荐方式）
-    print("注册 ImportLDraw operator...")
-    ImportLDraw.register()
-    print("✅ ImportLDraw operator 已注册")
-
-    print("使用 Blender operator 导入 LDR 文件...")
-    ldr_file_path = FileSystem.locate("{input_file}")
-    print(f"LDR 文件路径: {{ldr_file_path}}")
-
-    # 使用 Blender operator 导入（Blender 自动提供正确的 context）
-    try:
-        bpy.ops.import_scene.importldraw(
-            filepath=ldr_file_path,
-            ldrawPath="{ldraw_path}"
-        )
-        print("✅ LDR 文件导入成功")
-    except Exception as import_error:
-        print(f"❌ LDR 导入失败: {{import_error}}")
-        traceback.print_exc()
-        sys.exit(1)
+    print("尝试加载 LDR 文件（使用 loadFromFile）...")
+    ldr_file = FileSystem.locate("{input_file}")
+    print(f"找到 LDR 文件: {{ldr_file}}")
+    loadFromFile(None, ldr_file)
+    print("✅ LDR 文件加载成功")
     
     # 检查加载的对象
     mesh_objects = [obj for obj in bpy.data.objects if obj.type == 'MESH']