|
62 | 62 | NODE_DISPLAY_NAME_MAPPINGS as _PAINT_DISPLAY, |
63 | 63 | ) |
64 | 64 |
|
| 65 | +# ── NukeNodeMax suite (P0..F7) ──────────────────────────────────────── |
| 66 | +from .nodes.propainter_temporal_inpaint import ProPainterTemporalMEC |
| 67 | +from .nodes.propainter_flow_refine import FlowRefineMEC |
| 68 | +from .nodes.propainter_stitch_suite import ( |
| 69 | + NODE_CLASS_MAPPINGS as _PROPAINTER_STITCH_MAPPINGS, |
| 70 | + NODE_DISPLAY_NAME_MAPPINGS as _PROPAINTER_STITCH_DISPLAY, |
| 71 | +) |
| 72 | +from .nodes.video_stabilizer_mec import ( |
| 73 | + NODE_CLASS_MAPPINGS as _STABILIZER_MAPPINGS, |
| 74 | + NODE_DISPLAY_NAME_MAPPINGS as _STABILIZER_DISPLAY, |
| 75 | +) |
| 76 | +from .nodes.optical_flow import OpticalFlowMEC |
| 77 | +from .nodes.roto import VectorRotoMEC |
| 78 | +# NOTE: Deep* nodes (DeepFromImage / DeepMerge / DeepHoldout / DeepComposite) |
| 79 | +# now live exclusively in ComfyUI-NukeMaxNodes (May 2026 migration). The MEC |
| 80 | +# duplicates here have been deleted to avoid registry collisions and |
| 81 | +# divergent DEEP_IMAGE type semantics. |
| 82 | +from .nodes.shuffle import ShuffleMEC |
| 83 | +from .nodes.clipboard_tcl import ( |
| 84 | + TclSerializeMEC, TclParseMEC, |
| 85 | + register_routes as _register_tcl_routes, |
| 86 | +) |
| 87 | +from .nodes.insight import InsightStatusMEC, install as _install_insight_hook |
| 88 | +from .nodes.integrity_guard import ( |
| 89 | + IntegrityStatusMEC, |
| 90 | + register_routes as _register_integrity_routes, |
| 91 | + start_background_scan as _start_integrity_scan, |
| 92 | +) |
| 93 | + |
| 94 | +_NUKEMAX_MAPPINGS = { |
| 95 | + "ProPainterTemporalMEC": ProPainterTemporalMEC, |
| 96 | + "FlowRefineMEC": FlowRefineMEC, |
| 97 | + "OpticalFlowMEC": OpticalFlowMEC, |
| 98 | + "VectorRotoMEC": VectorRotoMEC, |
| 99 | + "ShuffleMEC": ShuffleMEC, |
| 100 | + "TclSerializeMEC": TclSerializeMEC, |
| 101 | + "TclParseMEC": TclParseMEC, |
| 102 | + "InsightStatusMEC": InsightStatusMEC, |
| 103 | + "IntegrityStatusMEC": IntegrityStatusMEC, |
| 104 | +} |
| 105 | +_NUKEMAX_DISPLAY = { |
| 106 | + "ProPainterTemporalMEC": "ProPainter Temporal Inpaint (MEC)", |
| 107 | + "FlowRefineMEC": "Optical Flow Refine (MEC)", |
| 108 | + "OpticalFlowMEC": "Optical Flow Re-Vector (MEC)", |
| 109 | + "VectorRotoMEC": "Vector Roto (MEC)", |
| 110 | + "ShuffleMEC": "Shuffle Channels (MEC)", |
| 111 | + "TclSerializeMEC": "TCL Serialize (MEC)", |
| 112 | + "TclParseMEC": "TCL Parse (MEC)", |
| 113 | + "InsightStatusMEC": "Insight Status (MEC)", |
| 114 | + "IntegrityStatusMEC": "Integrity Status (MEC)", |
| 115 | +} |
| 116 | + |
65 | 117 | # ── VFX nodes migrated to ComfyUI-NukeMaxNodes (Apr 2026) ───────────── |
66 | 118 | # (color_science, exr_io, render_pass, plate_tools, geometry_nodes, |
67 | 119 | # metadata_nodes, exr_metadata_reader, universal_reroute) |
|
147 | 199 | **_MEC_MAPPINGS, |
148 | 200 | **_MA_MAPPINGS, |
149 | 201 | **_PAINT_MAPPINGS, |
| 202 | + **_NUKEMAX_MAPPINGS, |
| 203 | + **_PROPAINTER_STITCH_MAPPINGS, |
| 204 | + **_STABILIZER_MAPPINGS, |
150 | 205 | } |
151 | 206 | NODE_DISPLAY_NAME_MAPPINGS = { |
152 | 207 | **_FOLDER_DISPLAY, |
153 | 208 | **_MEC_DISPLAY, |
154 | 209 | **_MA_DISPLAY, |
155 | 210 | **_PAINT_DISPLAY, |
| 211 | + **_NUKEMAX_DISPLAY, |
| 212 | + **_PROPAINTER_STITCH_DISPLAY, |
| 213 | + **_STABILIZER_DISPLAY, |
156 | 214 | } |
157 | 215 |
|
158 | 216 | WEB_DIRECTORY = "./js" |
|
168 | 226 | except Exception: |
169 | 227 | pass # Server not available (e.g. during import-only testing) |
170 | 228 |
|
| 229 | +# ── Register NukeNodeMax server-side hooks & routes ─────────────────── |
| 230 | +try: |
| 231 | + import server as _comfy_server # noqa: F811 |
| 232 | + _ps = _comfy_server.PromptServer.instance |
| 233 | + _register_tcl_routes(_ps) |
| 234 | + _register_integrity_routes(_ps) |
| 235 | + _start_integrity_scan() |
| 236 | + _install_insight_hook() |
| 237 | + print("[MEC] NukeNodeMax routes + hooks registered.") |
| 238 | +except Exception as _e: |
| 239 | + print(f"[MEC] NukeNodeMax server hooks deferred: {_e}") |
| 240 | + |
171 | 241 | print(f"[MEC] Loaded {len(_MEC_MAPPINGS)} MaskEditControl nodes.") |
0 commit comments