58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
from .controls.value_controls import FloatControl, IntControl, StringControl
|
|
from .controls.sequence_controls import FloatSequence, IntSequence, StringSequence
|
|
from .controls.utility_controls import FPSMonitor
|
|
from .controls.detector_controls import IntDetectionControl,FloatDetectionControl,StringDetectionControl
|
|
from .detectors.detector_nodes import MotionDetectorNode, RegionOfInterest, BrightnessDetectorNode
|
|
from .quick_shape_mask import QuickShapeMask
|
|
from .tiny_vae import TAESDVaeEncode, TAESDVaeDecode
|
|
from .misc_nodes import DTypeConverter
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FloatControl": FloatControl,
|
|
"IntControl": IntControl,
|
|
"StringControl": StringControl,
|
|
|
|
|
|
"FloatSequence": FloatSequence,
|
|
"IntSequence": IntSequence,
|
|
"StringSequence": StringSequence,
|
|
|
|
"FPSMonitor": FPSMonitor,
|
|
"QuickShapeMask": QuickShapeMask,
|
|
|
|
"MotionDetectorNode": MotionDetectorNode,
|
|
"BrightnessDetectorNode": BrightnessDetectorNode,
|
|
"RegionOfInterest": RegionOfInterest,
|
|
"IntDetectionControl": IntDetectionControl,
|
|
"FloatDetectionControl": FloatDetectionControl,
|
|
"StringDetectionControl": StringDetectionControl,
|
|
|
|
|
|
"TAESDVaeEncode": TAESDVaeEncode,
|
|
"TAESDVaeDecode": TAESDVaeDecode,
|
|
"DTypeConverter": DTypeConverter
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FloatControl": "Float Control (RyanOnTheInside)",
|
|
"IntControl": "Int Control (RyanOnTheInside)",
|
|
"StringControl": "String Control (RyanOnTheInside)",
|
|
"FloatSequence": "Float Sequence (RyanOnTheInside)",
|
|
"IntSequence": "Int Sequence (RyanOnTheInside)",
|
|
"StringSequence": "String Sequence (RyanOnTheInside)",
|
|
"FPSMonitor": "FPS Monitor (RyanOnTheInside)",
|
|
"MotionDetectorNode": "Motion Detector (RyanOnTheInside)",
|
|
"BrightnessDetectorNode": "Brightness Detector (RyanOnTheInside)",
|
|
"RegionOfInterest": "Region of Interest (RyanOnTheInside)",
|
|
"IntDetectionControl": "Detector Int Control (RyanOnTheInside)",
|
|
"FloatDetectionControl": "Detector Float Control (RyanOnTheInside)",
|
|
"StringDetectionControl": "Detector String Control (RyanOnTheInside)",
|
|
"QuickShapeMask": "Quick Shape Mask (RyanOnTheInside)",
|
|
"TAESDVaeEncode": "TAESD VAE Encode (RyanOnTheInside)",
|
|
"TAESDVaeDecode": "TAESD VAE Decode (RyanOnTheInside)",
|
|
"DTypeConverter": "DType Converter (RyanOnTheInside)"
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web/js"
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |