135 lines
3.6 KiB
Python
135 lines
3.6 KiB
Python
from .md import *
|
|
CATEGORY_TYPE = "🎈LAOGOU/Switch"
|
|
class LazySwitch2way:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"boolean": ("BOOLEAN", {"default": True}),
|
|
"ON_TRUE": (any, {"lazy": True}),
|
|
"on_true": (any, {"lazy": True}),
|
|
"ON_FALSE": (any, {"lazy": True}),
|
|
"on_false": (any, {"lazy": True}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = (any, any,)
|
|
RETURN_NAMES = ("OUTPUT", "output",)
|
|
FUNCTION = "switch"
|
|
CATEGORY = CATEGORY_TYPE
|
|
|
|
def check_lazy_status(self, boolean, ON_TRUE, on_true, ON_FALSE, on_false):
|
|
result = []
|
|
if boolean:
|
|
if ON_TRUE is None:
|
|
result.append("ON_TRUE")
|
|
if on_true is None:
|
|
result.append("on_true")
|
|
else:
|
|
if ON_FALSE is None:
|
|
result.append("ON_FALSE")
|
|
if on_false is None:
|
|
result.append("on_false")
|
|
return result if result else None
|
|
|
|
def switch(self, boolean, ON_TRUE, on_true, ON_FALSE, on_false):
|
|
if boolean:
|
|
return (ON_TRUE, on_true,)
|
|
else:
|
|
return (ON_FALSE, on_false,)
|
|
|
|
class LazySwitch1way:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"boolean": ("BOOLEAN", {"default": True}),
|
|
"ON_TRUE": (any, {"lazy": True}),
|
|
"ON_FALSE": (any, {"lazy": True}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = (any,)
|
|
RETURN_NAMES = ("OUTPUT",)
|
|
FUNCTION = "switch"
|
|
CATEGORY = CATEGORY_TYPE
|
|
|
|
def check_lazy_status(self, boolean, ON_TRUE, ON_FALSE):
|
|
result = []
|
|
if boolean:
|
|
if ON_TRUE is None:
|
|
result.append("ON_TRUE")
|
|
else:
|
|
if ON_FALSE is None:
|
|
result.append("ON_FALSE")
|
|
return result if result else None
|
|
|
|
def switch(self, boolean, ON_TRUE, ON_FALSE):
|
|
if boolean:
|
|
return (ON_TRUE,)
|
|
else:
|
|
return (ON_FALSE,)
|
|
|
|
class GroupSwitcher:
|
|
"""
|
|
组切换节点，带有布尔值控制
|
|
"""
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"boolean": ("BOOLEAN", {"default": True}), # 布尔值控制
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = () # 无输出端口
|
|
FUNCTION = "switch"
|
|
CATEGORY = CATEGORY_TYPE
|
|
|
|
def switch(self, boolean):
|
|
|
|
return ()
|
|
|
|
class MuterSwitcher:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"boolean": ("BOOLEAN", {"default": True}),
|
|
},
|
|
"optional": {
|
|
"ON_TRUE": (any,),
|
|
"on_true": (any,),
|
|
"ON_FALSE": (any,),
|
|
"on_false": (any,),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = (any, any,)
|
|
RETURN_NAMES = ("OUTPUT", "output",)
|
|
FUNCTION = "switch"
|
|
CATEGORY = CATEGORY_TYPE
|
|
|
|
def switch(self, ON_TRUE=None, on_true=None, ON_FALSE=None, on_false=None, boolean=True):
|
|
# 根据布尔值选择输出
|
|
if boolean:
|
|
return (ON_TRUE, on_true,)
|
|
else:
|
|
return (ON_FALSE, on_false,)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LazySwitch2way": LazySwitch2way,
|
|
"LazySwitch1way": LazySwitch1way,
|
|
"GroupSwitcher": GroupSwitcher,
|
|
"MuterSwitcher": MuterSwitcher,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LazySwitch2way": "🎈LazySwitch2way",
|
|
"LazySwitch1way": "🎈LazySwitch1way",
|
|
"GroupSwitcher": "🎈GroupSwitcher",
|
|
"MuterSwitcher": "🎈MuterSwitcher"
|
|
}
|
|
|
|
|