109 lines
4.2 KiB
Python
109 lines
4.2 KiB
Python
"""
|
|
Custom node registry that aggregates the standalone node modules in this package.
|
|
|
|
Intentionally excludes the legacy `ImageUtils.py` bundle; only the extracted
|
|
modules are imported and registered here.
|
|
"""
|
|
|
|
from .auto_adjust import (
|
|
NODE_CLASS_MAPPINGS as AUTO_ADJUST_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as AUTO_ADJUST_DISPLAY_MAP,
|
|
)
|
|
from .outpaint_config import (
|
|
NODE_CLASS_MAPPINGS as OUTPAINT_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as OUTPAINT_DISPLAY_MAP,
|
|
)
|
|
from .multiprompt import (
|
|
NODE_CLASS_MAPPINGS as MULTIPROMPT_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as MULTIPROMPT_DISPLAY_MAP,
|
|
)
|
|
from .stitch_image_by_mask import (
|
|
NODE_CLASS_MAPPINGS as STITCH_BY_MASK_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as STITCH_BY_MASK_DISPLAY_MAP,
|
|
)
|
|
from .crop_utils import (
|
|
NODE_CLASS_MAPPINGS as CROP_UTILS_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as CROP_UTILS_DISPLAY_MAP,
|
|
)
|
|
from .gpu_cache_clear import (
|
|
NODE_CLASS_MAPPINGS as GPU_CACHE_CLEAR_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as GPU_CACHE_CLEAR_DISPLAY_MAP,
|
|
)
|
|
from .filename_append_suffix import (
|
|
NODE_CLASS_MAPPINGS as FILENAME_APPEND_SUFFIX_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as FILENAME_APPEND_SUFFIX_DISPLAY_MAP,
|
|
)
|
|
from .auto_color_match import (
|
|
NODE_CLASS_MAPPINGS as AUTO_COLOR_MATCH_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as AUTO_COLOR_MATCH_DISPLAY_MAP,
|
|
)
|
|
from .image_megapixel_selector import (
|
|
NODE_CLASS_MAPPINGS as IMAGE_MPX_SELECTOR_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as IMAGE_MPX_SELECTOR_DISPLAY_MAP,
|
|
)
|
|
from .flux_resolution_prepare import (
|
|
NODE_CLASS_MAPPINGS as FLUX_PREP_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as FLUX_PREP_DISPLAY_MAP,
|
|
)
|
|
from .workflow_config import (
|
|
NODE_CLASS_MAPPINGS as WORKFLOW_CONFIG_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as WORKFLOW_CONFIG_DISPLAY_MAP,
|
|
)
|
|
from .smart_crop import (
|
|
NODE_CLASS_MAPPINGS as SMART_CROP_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as SMART_CROP_DISPLAY_MAP,
|
|
)
|
|
from .paired_image_loader import (
|
|
NODE_CLASS_MAPPINGS as PAIRED_LOADER_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as PAIRED_LOADER_DISPLAY_MAP,
|
|
)
|
|
from .load_image_combined import LoadImageCombined
|
|
from .comparison_gate import (
|
|
NODE_CLASS_MAPPINGS as COMPARISON_GATE_CLASS_MAP,
|
|
NODE_DISPLAY_NAME_MAPPINGS as COMPARISON_GATE_DISPLAY_MAP,
|
|
)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
# Merge the mappings exposed by the extracted modules.
|
|
NODE_CLASS_MAPPINGS.update(AUTO_ADJUST_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(OUTPAINT_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(MULTIPROMPT_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(STITCH_BY_MASK_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(CROP_UTILS_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(GPU_CACHE_CLEAR_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(FILENAME_APPEND_SUFFIX_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(AUTO_COLOR_MATCH_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(IMAGE_MPX_SELECTOR_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(FLUX_PREP_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(WORKFLOW_CONFIG_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(SMART_CROP_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(PAIRED_LOADER_CLASS_MAP)
|
|
NODE_CLASS_MAPPINGS.update(COMPARISON_GATE_CLASS_MAP)
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(AUTO_ADJUST_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(OUTPAINT_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(MULTIPROMPT_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(STITCH_BY_MASK_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CROP_UTILS_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(GPU_CACHE_CLEAR_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(FILENAME_APPEND_SUFFIX_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(AUTO_COLOR_MATCH_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(IMAGE_MPX_SELECTOR_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(FLUX_PREP_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(WORKFLOW_CONFIG_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(SMART_CROP_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(PAIRED_LOADER_DISPLAY_MAP)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(COMPARISON_GATE_DISPLAY_MAP)
|
|
|
|
# Register the load-image node which does not expose pre-built mappings.
|
|
NODE_CLASS_MAPPINGS["LoadImageCombined"] = LoadImageCombined
|
|
NODE_DISPLAY_NAME_MAPPINGS["LoadImageCombined"] = "Load Image (Combined)"
|
|
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
]
|