P4 fix: register MaskRefineMEC + UnifiedSegmentation in NODE_CLASS_MAPPINGS

- MaskRefineMEC class existed but mask_matting/__init__.py never imported it
- UnifiedSegmentation class existed but had no NODE_CLASS_MAPPINGS at module level
- Both are now properly registered; verified via /object_info (200) + visual sweep (6/6 PASS)
- Functional smoke tests (compute_integrity, roto_quality erosion, SeC fallback markers): 5/5 PASS

Author: Code2Collapse

__init__.py

@@ -72,6 +72,18 @@
     NODE_CLASS_MAPPINGS as _MASKMATTE_MAPPINGS,
     NODE_DISPLAY_NAME_MAPPINGS as _MASKMATTE_DISPLAY,
 )
+# Unified Segmentation (SAM2/SAM3/SeC, single node)
+try:
+    from .nodes.unified_segmentation import (
+        NODE_CLASS_MAPPINGS as _USEG_MAPPINGS,
+        NODE_DISPLAY_NAME_MAPPINGS as _USEG_DISPLAY,
+    )
+except Exception as _exc:  # pragma: no cover
+    import logging
+    logging.getLogger("MEC.UnifiedSeg").warning(
+        "UnifiedSegmentation unavailable: %s", _exc
+    )
+    _USEG_MAPPINGS, _USEG_DISPLAY = {}, {}
 
 # ── NukeNodeMax suite (P0..F7) ────────────────────────────────────────
 # ── ProPainter unified dispatcher (absorbs Temporal/Remove/Stitch/StitchRefine/FlowRefine) ──
@@ -168,6 +180,7 @@
     **_PAINT_MAPPINGS,
     **_FACE_FIXER_MAPPINGS,
     **_MASKMATTE_MAPPINGS,
+    **_USEG_MAPPINGS,
     **_NUKEMAX_MAPPINGS,
     **_STABILIZER_MAPPINGS,
 }
@@ -178,6 +191,7 @@
     **_PAINT_DISPLAY,
     **_FACE_FIXER_DISPLAY,
     **_MASKMATTE_DISPLAY,
+    **_USEG_DISPLAY,
     **_NUKEMAX_DISPLAY,
     **_STABILIZER_DISPLAY,
 }

nodes/mask_matting/__init__.py

@@ -44,10 +44,24 @@
     MaskTemporalMEC = None  # type: ignore
     _MT_MAPPINGS, _MT_DISPLAY = {}, {}
 
-NODE_CLASS_MAPPINGS = {**_MO_MAPPINGS, **_MT_MAPPINGS}
-NODE_DISPLAY_NAME_MAPPINGS = {**_MO_DISPLAY, **_MT_DISPLAY}
+try:
+    from .refine_node import (
+        MaskRefineMEC,
+        NODE_CLASS_MAPPINGS as _MR_MAPPINGS,
+        NODE_DISPLAY_NAME_MAPPINGS as _MR_DISPLAY,
+    )
+except Exception as _exc:  # pragma: no cover
+    import logging
+    logging.getLogger("MEC.MaskMatting").warning(
+        "MaskRefineMEC unavailable: %s", _exc
+    )
+    MaskRefineMEC = None  # type: ignore
+    _MR_MAPPINGS, _MR_DISPLAY = {}, {}
+
+NODE_CLASS_MAPPINGS = {**_MO_MAPPINGS, **_MT_MAPPINGS, **_MR_MAPPINGS}
+NODE_DISPLAY_NAME_MAPPINGS = {**_MO_DISPLAY, **_MT_DISPLAY, **_MR_DISPLAY}
 
 __all__ = [
-    "MaskOpsMEC", "MaskTemporalMEC",
+    "MaskOpsMEC", "MaskTemporalMEC", "MaskRefineMEC",
     "NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS",
 ]

nodes/unified_segmentation.py

@@ -737,3 +737,8 @@ def _sam_video(
 
         finally:
             shutil.rmtree(tmp, ignore_errors=True)
+
+
+NODE_CLASS_MAPPINGS = {"UnifiedSegmentation": UnifiedSegmentation}
+NODE_DISPLAY_NAME_MAPPINGS = {"UnifiedSegmentation": "Unified Segmentation (SAM2/SAM3/SeC)"}
+