wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-16 12:07:26 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

bar and tooltips

Browse Source
This commit is contained in:
Yousef Rafat 2026-05-14 16:04:27 +03:00
parent e94cad71ca
commit 5fde70e5f4
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
comfy_extras/nodes_trellis2.py
View File

@ -4,6 +4,7 @@ from comfy.ldm.trellis2.vae import SparseTensor
import comfy.model_management import comfy.model_management
from PIL import Image from PIL import Image
import numpy as np import numpy as np
import comfy.utils
import logging import logging
import torch import torch
import scipy import scipy
@ -1330,11 +1331,19 @@ class PostProcessMesh(IO.ComfyNode):
return IO.Schema( return IO.Schema(
node_id="PostProcessMesh", node_id="PostProcessMesh",
category="latent/3d", category="latent/3d",
description=(
"Applies a sequence of mesh post-processing operations including optional hole filling"
" and mesh simplification to a target face count."
),
inputs=[ inputs=[
IO.Mesh.Input("mesh"), IO.Mesh.Input("mesh"),
IO.Int.Input("target_face_count", default=1_000_000, min=0, max=50_000_000, IO.Int.Input("target_face_count", default=1_000_000, min=0, max=50_000_000,
tooltip="Target maximum number of faces after mesh simplification. Set to 0 to disable simplification."), tooltip="Target maximum number of faces after mesh simplification. Set to 0 to disable simplification."),
IO.Float.Input("fill_holes_perimeter", default=0.03, min=0.0, step=0.0001) IO.Float.Input("fill_holes_perimeter", default=0.03, min=0.0, step=0.0001,
tooltip=(
"Maximum hole perimeter threshold for filling holes in the mesh. "
"Smaller values only fill tiny holes, larger values fill larger gaps. "
"Set to 0 to disable hole filling."))
], ],
outputs=[ outputs=[
IO.Mesh.Output("output_mesh"), IO.Mesh.Output("output_mesh"),
@ -1349,11 +1358,14 @@ class PostProcessMesh(IO.ComfyNode):
def process_single(v, f, c): def process_single(v, f, c):
if fill_holes_perimeter > 0: if fill_holes_perimeter > 0:
v, f = fill_holes_fn(v, f, max_perimeter=fill_holes_perimeter) v, f = fill_holes_fn(v, f, max_perimeter=fill_holes_perimeter)
bar.update(1)
if target_face_count > 0 and f.shape[0] > target_face_count: if target_face_count > 0 and f.shape[0] > target_face_count:
v, f, c = simplify_fn(v, f, colors=c, target=target_face_count) v, f, c = simplify_fn(v, f, colors=c, target=target_face_count)
bar.update(1)
v, f = make_double_sided(v, f) v, f = make_double_sided(v, f)
bar.update(1)
return v, f, c return v, f, c
# Check if batch is Jagged (List) or Uniform (3D Tensor) # Check if batch is Jagged (List) or Uniform (3D Tensor)
@ -1363,6 +1375,7 @@ class PostProcessMesh(IO.ComfyNode):
if is_list or is_batched_tensor: if is_list or is_batched_tensor:
out_v, out_f, out_c = [], [],[] out_v, out_f, out_c = [], [],[]
bsz = len(mesh.vertices) if is_list else mesh.vertices.shape[0] bsz = len(mesh.vertices) if is_list else mesh.vertices.shape[0]
bar = comfy.utils.ProgressBar(3 * bsz)
for i in range(bsz): for i in range(bsz):
v_i = mesh.vertices[i] v_i = mesh.vertices[i]
@ -1373,7 +1386,7 @@ class PostProcessMesh(IO.ComfyNode):
if hasattr(mesh, 'colors') and mesh.colors is not None: if hasattr(mesh, 'colors') and mesh.colors is not None:
c_i = mesh.colors[i] if (isinstance(mesh.colors, list) or mesh.colors.ndim == 3) else mesh.colors c_i = mesh.colors[i] if (isinstance(mesh.colors, list) or mesh.colors.ndim == 3) else mesh.colors
v_i, f_i, c_i = process_single(v_i, f_i, c_i) v_i, f_i, c_i = process_single(v_i, f_i, c_i, bar)
out_v.append(v_i) out_v.append(v_i)
out_f.append(f_i) out_f.append(f_i)