Add other export file formats

This commit is contained in:
kijai 2025-02-02 13:15:14 +02:00
parent 2758f91a89
commit 205be20e08

View File

@ -1327,6 +1327,7 @@ class Hy3DExportMesh:
"required": { "required": {
"mesh": ("HY3DMESH",), "mesh": ("HY3DMESH",),
"filename_prefix": ("STRING", {"default": "3D/Hy3D"}), "filename_prefix": ("STRING", {"default": "3D/Hy3D"}),
"file_format": (["glb", "obj", "ply", "stl", "3mf", "dae"],),
}, },
} }
@ -1335,13 +1336,13 @@ class Hy3DExportMesh:
FUNCTION = "process" FUNCTION = "process"
CATEGORY = "Hunyuan3DWrapper" CATEGORY = "Hunyuan3DWrapper"
def process(self, mesh, filename_prefix): def process(self, mesh, filename_prefix, file_format):
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, folder_paths.get_output_directory()) full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, folder_paths.get_output_directory())
output_glb_path = Path(full_output_folder, f'{filename}_{counter:05}_.glb') output_glb_path = Path(full_output_folder, f'{filename}_{counter:05}_.{file_format}')
output_glb_path.parent.mkdir(exist_ok=True) output_glb_path.parent.mkdir(exist_ok=True)
mesh.export(output_glb_path) mesh.export(output_glb_path, file_type=file_format)
relative_path = Path(subfolder) / f'{filename}_{counter:05}_.glb' relative_path = Path(subfolder) / f'{filename}_{counter:05}_.{file_format}'
return (str(relative_path), ) return (str(relative_path), )