add trellis support

2026-02-10 00:19:10 -05:00
parent 668d66b70e
commit 63bc583059
12 changed files with 510 additions and 1 deletions
@@ -0,0 +1,106 @@
+"""
+Runware 3D Inference Settings Node
+Main settings for 3D inference: textureSize, decimationTarget, remesh, resolution,
+and sub-configs sparseStructure, shapeSlat, texSlat from connected nodes.
+"""
+
+from typing import Dict, Any
+
+
+class Runware3DInferenceSettings:
+    """Runware 3D Inference Settings Node"""
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {},
+            "optional": {
+                "useTextureSize": ("BOOLEAN", {
+                    "tooltip": "Enable to include textureSize in settings",
+                    "default": False,
+                }),
+                "textureSize": ("INT", {
+                    "tooltip": "Texture size for 3D output (1024-4096, step 1024). Only used when enabled.",
+                    "default": 2048,
+                    "min": 1024,
+                    "max": 4096,
+                    "step": 1024,
+                }),
+                "useDecimationTarget": ("BOOLEAN", {
+                    "tooltip": "Enable to include decimationTarget in settings",
+                    "default": False,
+                }),
+                "decimationTarget": ("INT", {
+                    "tooltip": "Decimation target for mesh (100000-1000000). Only used when enabled.",
+                    "default": 500000,
+                    "min": 100000,
+                    "max": 1000000,
+                    "step": 10000,
+                }),
+                "useRemesh": ("BOOLEAN", {
+                    "tooltip": "Enable to include remesh in settings",
+                    "default": False,
+                }),
+                "remesh": ("BOOLEAN", {
+                    "tooltip": "Whether to remesh the output. Only used when enabled.",
+                    "default": True,
+                }),
+                "useResolution": ("BOOLEAN", {
+                    "tooltip": "Enable to include resolution in settings",
+                    "default": False,
+                }),
+                "resolution": ([512, 1024, 1536], {
+                    "tooltip": "Resolution for 3D generation (512, 1024, or 1536). Only used when enabled.",
+                    "default": 1024,
+                }),
+                "sparseStructure": ("RUNWARE3DINFERENCESETTINGSLAT", {
+                    "tooltip": "Connect Runware 3D Inference Settings Sparse Structure node.",
+                }),
+                "shapeSlat": ("RUNWARE3DINFERENCESETTINGSLAT", {
+                    "tooltip": "Connect Runware 3D Inference Settings Shape Slat node.",
+                }),
+                "texSlat": ("RUNWARE3DINFERENCESETTINGSLAT", {
+                    "tooltip": "Connect Runware 3D Inference Settings Tex Slat node.",
+                }),
+            }
+        }
+
+    RETURN_TYPES = ("RUNWARE3DINFERENCESETTINGS",)
+    RETURN_NAMES = ("Settings",)
+    FUNCTION = "create"
+    CATEGORY = "Runware"
+    DESCRIPTION = "Configure Runware 3D Inference settings: texture, decimation, remesh, resolution, and lat configs."
+
+    def create(self, **kwargs) -> tuple[Dict[str, Any]]:
+        settings: Dict[str, Any] = {}
+
+        if kwargs.get("useTextureSize", False):
+            settings["textureSize"] = int(kwargs.get("textureSize", 2048))
+        if kwargs.get("useDecimationTarget", False):
+            settings["decimationTarget"] = int(kwargs.get("decimationTarget", 500000))
+        if kwargs.get("useRemesh", False):
+            settings["remesh"] = bool(kwargs.get("remesh", True))
+        if kwargs.get("useResolution", False):
+            settings["resolution"] = int(kwargs.get("resolution", 1024))
+
+        sparse = kwargs.get("sparseStructure", None)
+        if sparse is not None and isinstance(sparse, dict) and len(sparse) > 0:
+            settings["sparseStructure"] = sparse
+
+        shape = kwargs.get("shapeSlat", None)
+        if shape is not None and isinstance(shape, dict) and len(shape) > 0:
+            settings["shapeSlat"] = shape
+
+        tex = kwargs.get("texSlat", None)
+        if tex is not None and isinstance(tex, dict) and len(tex) > 0:
+            settings["texSlat"] = tex
+
+        return (settings,)
+
+
+NODE_CLASS_MAPPINGS = {
+    "Runware3DInferenceSettings": Runware3DInferenceSettings,
+}
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "Runware3DInferenceSettings": "Runware 3D Inference Settings",
+}