xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2025-12-15 07:44:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

workaround to not always save the mesh for preview

Browse Source
This commit is contained in:
kijai 2025-02-04 22:29:44 +02:00
parent 0e18f865eb
commit 2846bbf5cc
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nodes.py
View File

@ -1385,6 +1385,9 @@ class Hy3DExportMesh:
"filename_prefix": ("STRING", {"default": "3D/Hy3D"}), "filename_prefix": ("STRING", {"default": "3D/Hy3D"}),
"file_format": (["glb", "obj", "ply", "stl", "3mf", "dae"],), "file_format": (["glb", "obj", "ply", "stl", "3mf", "dae"],),
}, },
"optional": {
"save_file": ("BOOLEAN", {"default": True}),
},
} }
RETURN_TYPES = ("STRING",) RETURN_TYPES = ("STRING",)
@ -1392,13 +1395,17 @@ class Hy3DExportMesh:
FUNCTION = "process" FUNCTION = "process"
CATEGORY = "Hunyuan3DWrapper" CATEGORY = "Hunyuan3DWrapper"
def process(self, mesh, filename_prefix, file_format): def process(self, mesh, filename_prefix, file_format, save_file=True):
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}_.{file_format}') 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)
if save_file:
mesh.export(output_glb_path, file_type=file_format) mesh.export(output_glb_path, file_type=file_format)
relative_path = Path(subfolder) / f'{filename}_{counter:05}_.{file_format}' relative_path = Path(subfolder) / f'{filename}_{counter:05}_.{file_format}'
else:
temp_file = Path(full_output_folder, f'hy3dtemp_.{file_format}')
mesh.export(temp_file, file_type=file_format)
relative_path = Path(subfolder) / f'hy3dtemp_.{file_format}'
return (str(relative_path), ) return (str(relative_path), )