Add support for Trellis 2

2026-05-29 23:12:40 +05:30
parent fec0623cc1
commit dde8589886
6 changed files with 190 additions and 7 deletions
@@ -0,0 +1,90 @@
+"""
+Runware 3D Inference Settings Mesh Cluster Node
+Outputs meshCluster config: thresholdConeHalfAngleRad, refineIterations,
+globalIterations, smoothStrength
+"""
+
+import math
+from typing import Dict, Any
+
+
+class Runware3DInferenceSettingsMeshCluster:
+    """Runware 3D Inference Settings Mesh Cluster Node"""
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {},
+            "optional": {
+                "useThresholdConeHalfAngleRad": ("BOOLEAN", {
+                    "tooltip": "Enable to include thresholdConeHalfAngleRad in meshCluster",
+                    "default": False,
+                }),
+                "thresholdConeHalfAngleRad": ("FLOAT", {
+                    "tooltip": "Cone half-angle threshold (radians) used during UV chart clustering. Only used when enabled.",
+                    "default": math.pi / 2,
+                    "min": 0.0,
+                    "max": math.pi,
+                    "step": 0.01,
+                }),
+                "useRefineIterations": ("BOOLEAN", {
+                    "tooltip": "Enable to include refineIterations in meshCluster",
+                    "default": False,
+                }),
+                "refineIterations": ("INT", {
+                    "tooltip": "Local refinement iterations for chart clustering. Only used when enabled.",
+                    "default": 0,
+                    "min": 0,
+                    "max": 64,
+                    "step": 1,
+                }),
+                "useGlobalIterations": ("BOOLEAN", {
+                    "tooltip": "Enable to include globalIterations in meshCluster",
+                    "default": False,
+                }),
+                "globalIterations": ("INT", {
+                    "tooltip": "Global iterations for chart clustering. Only used when enabled.",
+                    "default": 1,
+                    "min": 0,
+                    "max": 64,
+                    "step": 1,
+                }),
+                "useSmoothStrength": ("BOOLEAN", {
+                    "tooltip": "Enable to include smoothStrength in meshCluster",
+                    "default": False,
+                }),
+                "smoothStrength": ("INT", {
+                    "tooltip": "Smoothing strength applied during chart clustering. Only used when enabled.",
+                    "default": 1,
+                    "min": 0,
+                    "max": 10,
+                    "step": 1,
+                }),
+            }
+        }
+
+    RETURN_TYPES = ("RUNWARE3DINFERENCESETTINGSLAT",)
+    RETURN_NAMES = ("Mesh Cluster",)
+    FUNCTION = "create"
+    CATEGORY = "Runware"
+    DESCRIPTION = "Configure mesh cluster parameters for Runware 3D Inference settings."
+
+    def create(self, **kwargs) -> tuple[Dict[str, Any]]:
+        out: Dict[str, Any] = {}
+        if kwargs.get("useThresholdConeHalfAngleRad", False):
+            out["thresholdConeHalfAngleRad"] = float(kwargs.get("thresholdConeHalfAngleRad", math.pi / 2))
+        if kwargs.get("useRefineIterations", False):
+            out["refineIterations"] = int(kwargs.get("refineIterations", 0))
+        if kwargs.get("useGlobalIterations", False):
+            out["globalIterations"] = int(kwargs.get("globalIterations", 1))
+        if kwargs.get("useSmoothStrength", False):
+            out["smoothStrength"] = int(kwargs.get("smoothStrength", 1))
+        return (out,)
+
+
+NODE_CLASS_MAPPINGS = {
+    "Runware3DInferenceSettingsMeshCluster": Runware3DInferenceSettingsMeshCluster,
+}
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "Runware3DInferenceSettingsMeshCluster": "Runware 3D Inference Settings Mesh Cluster",
+}