import bpy
import os

# 输入 .fbx 文件的文件夹路径
fbx_folder = "/public/home/wangshuo/gap/assembly/data/ldr2fbx"  # 替换为你实际的 .fbx 文件夹路径
obj_folder = "/public/home/wangshuo/gap/assembly/data/ldr2fbx/output_obj"  # 替换为你希望保存 .obj 文件的文件夹路径

# 获取文件夹中所有的 .fbx 文件
fbx_files = [f for f in os.listdir(fbx_folder) if f.endswith('.fbx')]

# 确保 OBJ 输出目录存在
if not os.path.exists(obj_folder):
    os.makedirs(obj_folder)

# 导入 LDR 文件并导出为 OBJ
for fbx_file in fbx_files:
    fbx_path = os.path.join(fbx_folder, fbx_file)
    
    # 获取文件名前缀，用作 .obj 文件名
    obj_name = os.path.splitext(fbx_file)[0]
    obj_path = os.path.join(obj_folder, obj_name + ".obj")

    # 清空当前场景
    bpy.ops.object.select_all(action='SELECT')
    bpy.ops.object.delete()

    # 导入 .fbx 文件
    bpy.ops.import_scene.fbx(filepath=fbx_path)

    # 检查导出路径和文件类型
    if bpy.context.scene.objects:
        print(f"Exporting to {obj_path}...")
        bpy.ops.export_scene.obj(filepath=obj_path)
    else:
        print(f"Error: No objects to export in {fbx_file}")
    
    print(f"Converted {fbx_file} to {obj_name}.obj")