46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
|
|
# Import all available modules
|
|
from .modules.image_loader import coco_loader
|
|
from .modules.load_exr import load_exr
|
|
from .modules.saver import saver
|
|
from .modules.load_exr_layer_by_name import load_exr_layer_by_name, shamble_cryptomatte
|
|
from .modules.colorspace import colorspace
|
|
from .modules.znormalize import znormalize
|
|
|
|
# Initialize node mappings
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
# Explicitly set the web directory path relative to this file
|
|
import os
|
|
NODE_DIR = os.path.dirname(os.path.realpath(__file__))
|
|
WEB_DIRECTORY = os.path.join(NODE_DIR, "js")
|
|
|
|
# Add all available node classes
|
|
NODE_CLASS_MAPPINGS.update({
|
|
"ImageLoader": coco_loader,
|
|
"LoadExr": load_exr,
|
|
"SaverNode": saver,
|
|
"LoadExrLayerByName": load_exr_layer_by_name,
|
|
"CryptomatteLayer": shamble_cryptomatte,
|
|
"ColorspaceNode": colorspace,
|
|
"ZNormalizeNode": znormalize
|
|
})
|
|
|
|
# Add display names for better UI presentation
|
|
NODE_DISPLAY_NAME_MAPPINGS.update({
|
|
"ImageLoader": "Image Loader",
|
|
"LoadExr": "Load EXR",
|
|
"SaverNode": "Save Image",
|
|
"LoadExrLayerByName": "Load EXR Layer by Name",
|
|
"CryptomatteLayer": "Cryptomatte Layer",
|
|
"ColorspaceNode": "Colorspace",
|
|
"ZNormalizeNode": "Z Normalize"
|
|
})
|
|
|
|
# Expose what ComfyUI needs
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
]
|