wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-06 22:51:18 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Node cleanup

Browse Source
This commit is contained in:
kijai 2026-06-30 01:55:53 +03:00
parent 42ac23f6f6
commit 56183a82ec
1 changed files with 17 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
comfy_extras/nodes_mesh_postprocess.py
View File

@ -358,7 +358,7 @@ def _nearest_voxel_sample_gpu(positions, voxel_coords_np, color_np, resolution):
P = torch.from_numpy(np.ascontiguousarray(positions)).to(dev).float() P = torch.from_numpy(np.ascontiguousarray(positions)).to(dev).float()
VC = torch.from_numpy(np.ascontiguousarray(voxel_coords_np)).to(dev).long() VC = torch.from_numpy(np.ascontiguousarray(voxel_coords_np)).to(dev).long()
col = torch.from_numpy(np.ascontiguousarray(color_np)).to(dev).float() col = torch.from_numpy(np.ascontiguousarray(color_np)).to(dev).float()
M, K, C = VC.shape[0], P.shape[0], col.shape[1] M, K = VC.shape[0], P.shape[0]
key = (VC[:, 0] * R + VC[:, 1]) * R + VC[:, 2] key = (VC[:, 0] * R + VC[:, 1]) * R + VC[:, 2]
skey, order = key.sort() skey, order = key.sort()
@ -2222,14 +2222,12 @@ class RemeshMesh(IO.ComfyNode):
# sub-widgets show per sign_mode (DynamicCombo). # sub-widgets show per sign_mode (DynamicCombo).
sign_mode_options = [ sign_mode_options = [
IO.DynamicCombo.Option(key="udf", inputs=[ IO.DynamicCombo.Option(key="udf", inputs=[
IO.Boolean.Input("qef", default=False, IO.Boolean.Input("qef", default=False, advanced=True,
tooltip="Experimental: QEF dual-vertex placement for sharper edges; may " tooltip="QEF dual-vertex placement for sharper edges."),
"misbehave near the UDF double shell."), IO.Boolean.Input("drop_inverted_components", default=False, advanced=True,
IO.Boolean.Input("drop_inverted_components", default=True,
tooltip="Drop inward-normal (negative-volume) closed components — the UDF inner shell."), tooltip="Drop inward-normal (negative-volume) closed components — the UDF inner shell."),
IO.Boolean.Input("drop_enclosed_components", default=True, IO.Boolean.Input("drop_enclosed_components", default=False, advanced=True,
tooltip="Drop components inside the largest's bbox that fail a point-in-mesh " tooltip="Drop components inside the largest's bbox that fail a point-in-mesh raycast. Disable for legitimate nested parts."),
"raycast. Disable for legitimate nested parts."),
]), ]),
IO.DynamicCombo.Option(key="sdf", inputs=[ IO.DynamicCombo.Option(key="sdf", inputs=[
IO.Boolean.Input("qef", default=True, IO.Boolean.Input("qef", default=True,
@ -2250,34 +2248,31 @@ class RemeshMesh(IO.ComfyNode):
), ),
inputs=[ inputs=[
IO.Mesh.Input("mesh"), IO.Mesh.Input("mesh"),
IO.Int.Input("target_faces", default=0, min=0, max=50_000_000, IO.Int.Input("resolution", default=512, min=32, max=1024,
tooltip="0 = use 'resolution'. >0 = auto-pick resolution to roughly hit this " tooltip="Voxel grid resolution (output density). 256 ~ 100k faces, 512 ~ 1M. "
"count (±30-50%); overshoot then DecimateMesh to exact."), "For an exact face count, follow with DecimateMesh."),
IO.Int.Input("resolution", default=256, min=32, max=1024,
tooltip="Voxel grid resolution (when target_faces=0). 256 ~ 100k faces, 512 ~ 1M."),
IO.DynamicCombo.Input("sign_mode", options=sign_mode_options, display_name="sign_mode", IO.DynamicCombo.Input("sign_mode", options=sign_mode_options, display_name="sign_mode",
tooltip="udf: robust to messy/non-manifold input. sdf: clean single " tooltip="udf: robust to messy/non-manifold input. sdf: clean single "
"surface with QEF sharp-feature recovery, but needs consistent winding."), "surface with QEF sharp-feature recovery, but needs consistent winding."),
IO.Float.Input("band", default=1.0, min=0.5, max=4.0, step=0.1, IO.Float.Input("band", default=1.0, min=0.5, max=4.0, step=0.1, advanced=True,
tooltip="Narrow-band width in voxel units. In UDF mode also offsets the surface."), tooltip="Narrow-band width in voxel units. In UDF mode also offsets the surface."),
IO.Float.Input("project_back", default=0.0, min=0.0, max=1.0, step=0.05, IO.Float.Input("project_back", default=0.0, min=0.0, max=1.0, step=0.05, advanced=True,
tooltip="Lerp verts toward the original surface (0 = pure DC, 1 = snapped)."), tooltip="Lerp verts toward the original surface (0 = pure DC, 1 = snapped)."),
IO.Boolean.Input("fix_poles", default=False, IO.Boolean.Input("fix_poles", default=False, advanced=True,
tooltip="Collapse valence-3 vertex pairs (DC T-junction artifact)."), tooltip="Collapse valence-3 vertex pairs (DC T-junction artifact)."),
IO.Int.Input("smooth_iters", default=0, min=0, max=20, IO.Int.Input("smooth_iters", default=0, min=0, max=20,
tooltip="Taubin smoothing iters (0 = off). 2-3 cleans DC stairstepping; higher rounds off QEF edges."), tooltip="Taubin smoothing iters (0 = off). 2-3 cleans DC stairstepping; higher rounds off QEF edges."),
IO.Float.Input("drop_small_components", default=0.01, min=0.0, max=0.5, step=0.005, IO.Float.Input("drop_small_components", default=0.01, min=0.0, max=0.5, step=0.005, advanced=True,
tooltip="Drop components below this fraction of the largest's face count. 0 disables."), tooltip="Drop components below this fraction of the largest's face count. 0 disables."),
IO.Int.Input("precluster_max_verts", default=0, min=0, max=50_000_000, IO.Int.Input("precluster_max_verts", default=8_000_000, min=0, max=50_000_000, advanced=True,
tooltip="If input exceeds this (>0), cluster-decimate first so field queries don't " tooltip="Cap input vertex count before the field queries, inputs above this are cluster-decimated to it first. Prevents OOM on huge meshes."),
"OOM. 0 = off; 1-2M for very large meshes."),
], ],
outputs=[IO.Mesh.Output("mesh")], outputs=[IO.Mesh.Output("mesh")],
hidden=[IO.Hidden.unique_id], hidden=[IO.Hidden.unique_id],
) )
@classmethod @classmethod
def execute(cls, mesh, target_faces, resolution, sign_mode, band, def execute(cls, mesh, resolution, sign_mode, band,
project_back, fix_poles, smooth_iters, project_back, fix_poles, smooth_iters,
drop_small_components, precluster_max_verts): drop_small_components, precluster_max_verts):
mode = sign_mode.get("sign_mode", "udf") mode = sign_mode.get("sign_mode", "udf")
@ -2310,7 +2305,7 @@ class RemeshMesh(IO.ComfyNode):
rv, rf, rc = remesh_narrow_band_dc( rv, rf, rc = remesh_narrow_band_dc(
vv, ff, vv, ff,
resolution=int(resolution), target_faces=int(target_faces), resolution=int(resolution),
band=float(band), project_back=float(project_back), band=float(band), project_back=float(project_back),
qef=qef, sign_mode=mode, qef=qef, sign_mode=mode,
manifold=manifold, fix_poles=bool(fix_poles), manifold=manifold, fix_poles=bool(fix_poles),