107 lines
4.3 KiB
Python
107 lines
4.3 KiB
Python
"""
|
|
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",
|
|
}
|