Files
Runware-ComfyUI-Runware/modules/threeDInferenceSettingsDracoCompression.py
T

110 lines
4.5 KiB
Python

"""
Runware 3D Inference Settings Draco Compression Node
Outputs dracoCompression config: enabled, level, quantizationPosition,
quantizationNormal, quantizationTexCoord
"""
from typing import Dict, Any
class Runware3DInferenceSettingsDracoCompression:
"""Runware 3D Inference Settings Draco Compression Node"""
@classmethod
def INPUT_TYPES(cls):
return {
"required": {},
"optional": {
"useEnabled": ("BOOLEAN", {
"tooltip": "Enable to include enabled in dracoCompression",
"default": False,
}),
"enabled": ("BOOLEAN", {
"tooltip": (
"If true, applies KHR_draco_mesh_compression to the output GLB "
"(5-10x geometry size reduction with no triangle loss). "
"Only used when 'Use Enabled' is on."
),
"default": False,
"label_on": "true",
"label_off": "false",
}),
"useLevel": ("BOOLEAN", {
"tooltip": "Enable to include level in dracoCompression",
"default": False,
}),
"level": ("INT", {
"tooltip": "Draco compression level (0-10). Higher = better compression but slower encoding. Only used when enabled.",
"default": 7,
"min": 0,
"max": 10,
"step": 1,
}),
"useQuantizationPosition": ("BOOLEAN", {
"tooltip": "Enable to include quantizationPosition in dracoCompression",
"default": False,
}),
"quantizationPosition": ("INT", {
"tooltip": "Quantization bits for vertex positions (8-30). Higher = better quality, larger file. Only used when enabled.",
"default": 16,
"min": 8,
"max": 30,
"step": 1,
}),
"useQuantizationNormal": ("BOOLEAN", {
"tooltip": "Enable to include quantizationNormal in dracoCompression",
"default": False,
}),
"quantizationNormal": ("INT", {
"tooltip": "Quantization bits for vertex normals (8-30). Higher = smoother shading, larger file. Only used when enabled.",
"default": 14,
"min": 8,
"max": 30,
"step": 1,
}),
"useQuantizationTexCoord": ("BOOLEAN", {
"tooltip": "Enable to include quantizationTexCoord in dracoCompression",
"default": False,
}),
"quantizationTexCoord": ("INT", {
"tooltip": "Quantization bits for texture coordinates (8-30). Higher = sharper textures at UV seams. Only used when enabled.",
"default": 14,
"min": 8,
"max": 30,
"step": 1,
}),
}
}
RETURN_TYPES = ("RUNWARE3DINFERENCESETTINGSLAT",)
RETURN_NAMES = ("Draco Compression",)
FUNCTION = "create"
CATEGORY = "Runware"
DESCRIPTION = (
"Configure settings.dracoCompression for Runware 3D Inference "
"(enabled, level, quantizationPosition, quantizationNormal, quantizationTexCoord). "
"Connect to Runware 3D Inference Settings."
)
def create(self, **kwargs) -> tuple[Dict[str, Any]]:
out: Dict[str, Any] = {}
if kwargs.get("useEnabled", False):
out["enabled"] = bool(kwargs.get("enabled", False))
if kwargs.get("useLevel", False):
out["level"] = int(kwargs.get("level", 7))
if kwargs.get("useQuantizationPosition", False):
out["quantizationPosition"] = int(kwargs.get("quantizationPosition", 16))
if kwargs.get("useQuantizationNormal", False):
out["quantizationNormal"] = int(kwargs.get("quantizationNormal", 14))
if kwargs.get("useQuantizationTexCoord", False):
out["quantizationTexCoord"] = int(kwargs.get("quantizationTexCoord", 14))
return (out,)
NODE_CLASS_MAPPINGS = {
"Runware3DInferenceSettingsDracoCompression": Runware3DInferenceSettingsDracoCompression,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"Runware3DInferenceSettingsDracoCompression": "Runware 3D Inference Settings Draco Compression",
}