Files
beyondprompting-ComfyUI-Bey…/node_mappings.py
T
2025-12-02 18:26:26 -03:00

75 lines
2.8 KiB
Python

try:
from .nodes.Compositor4 import Compositor4
from .nodes.CompositorConfig4 import CompositorConfig4
from .nodes.Compositor4TransformsOut import Compositor4TransformsOut
from .nodes.Compositor4MasksOutput import Compositor4MasksOutput
except ImportError:
print("\033[34mBeyond Nodes: \033[92mFailed loading Compositor nodes\033[0m")
try:
from .nodes.Logic import Compare, Int, Float, Bool, String, IfExecute, DebugPrint, AlwaysEqualProxy
except ImportError:
print("\033[34mBeyond Nodes: \033[92mFailed loading Logic nodes\033[0m")
try:
from .nodes.Masking import Mask_Rectangular_Area, ImageMaskScaleAs
except ImportError:
print("\033[34mBeyond Nodes: \033[92mFailed loading Masking nodes\033[0m")
try:
from .nodes.Switch import TwoWaySwitch, CSwitchBooleanAny, CSwitchFromAny, ImageMaskSwitch
except ImportError:
print("\033[34mBeyond Nodes: \033[92mFailed loading Switch nodes\033[0m")
NODE_CLASS_MAPPINGS = {
### Compositor Nodes
"Compositor4-beyond_nodes": Compositor4,
"CompositorConfig4-beyond_nodes": CompositorConfig4,
"Compositor4TransformsOut-beyond_nodes": Compositor4TransformsOut,
"Compositor4MasksOutput-beyond_nodes": Compositor4MasksOutput,
### Logic Nodes
"Compare-beyond_nodes": Compare,
"Int-beyond_nodes": Int,
"Float-beyond_nodes": Float,
"Bool-beyond_nodes": Bool,
"String-beyond_nodes": String,
"If ANY return A else B-beyond_nodes": IfExecute,
"DebugPrint-beyond_nodes": DebugPrint,
### Masking Nodes
"MaskRectArea-beyond_nodes": Mask_Rectangular_Area,
### Switch Nodes
"TwoWaySwitch-beyond_nodes": TwoWaySwitch,
"BooleanSwitch-beyond_nodes": CSwitchBooleanAny,
"ImageMaskSwitch-beyond_nodes": ImageMaskSwitch,
"ImageMaskScaleAs-beyond_nodes": ImageMaskScaleAs,
}
NODE_DISPLAY_NAME_MAPPINGS = {
### Compositor Nodes
"Compositor4-beyond_nodes": "🦾 Compositor V4 ",
"CompositorConfig4-beyond_nodes": "🦾 Compositor Config V4 🦾",
"Compositor4TransformsOut-beyond_nodes": "🦾 Compositor Transforms Output V4 🦾",
"Compositor4MasksOutput-beyond_nodes": "🦾 Compositor Masks Output V4 🦾",
### Logic Nodes
"Compare-beyond_nodes": "Compare 🦾",
"Int-beyond_nodes": "Int 🦾",
"Float-beyond_nodes": "Float 🦾",
"Bool-beyond_nodes": "Bool 🦾",
"String-beyond_nodes": "String 🦾",
"If ANY return A else B-beyond_nodes": "If ANY return A else B 🦾",
"DebugPrint-beyond_nodes": "DebugPrint 🦾",
### Masking Nodes
"MaskRectArea-beyond_nodes": "Mask Rectangular Area 🦾",
### Switch Nodes
"TwoWaySwitch-beyond_nodes": "Two Way Switch 🦾",
"BooleanSwitch-beyond_nodes": "Boolean Switch Any Type 🦾",
"ImageMaskSwitch-beyond_nodes":"Image and Mask Switch 🦾",
"ImageMaskScaleAs-beyond_nodes": "Scale an Image/Mask as a Ref 🦾",
}