Delete xyz_grid.py

xyz_grid.py

@@ -1,285 +0,0 @@
-"""
-lib_mega_freeu/global_state.py
-Runtime state, data structures, presets for ⚡ Mega FreeU.
-
-Sources:
-  sd-webui-freeu/lib_free_u/global_state.py  -- StageInfo layout, State, preset JSON, XYZ
-  WAS FreeU_Advanced/nodes.py                -- BLEND_MODE_NAMES, MSCALES
-  ComfyUI_FreeU_V2_Advanced/FreeU_B1B2.py   -- b_start/b_end, channel_threshold
-  ComfyUI_FreeU_V2_Advanced/FreeU_S1S2.py   -- s_start/s_end, adaptive cap
-  nrs_kohaku_enhanced_v3_5.py                -- hf_boost, gaussian standalone
-"""
-import dataclasses
-import json
-import math
-import pathlib
-import re
-import sys
-from typing import Any, Dict, List, Optional, Union
-
-# ─── Blending modes (WAS FreeU_Advanced/nodes.py blending_modes keys) ─────────
-BLEND_MODE_NAMES: List[str] = [
-    "lerp", "inject", "bislerp", "colorize",
-    "cosine interp", "cuberp", "hslerp", "stable_slerp", "linear dodge",
-]
-
-# ─── Multi-scale presets (WAS nodes.py mscales dict -- exact) ─────────────────
-MSCALES: Dict[str, Optional[list]] = {
-    "Default":                None,
-    "Low-Pass":               [(10, 1.0)],
-    "Pass-Through":           [(10, 1.0)],
-    "Gaussian-Blur":          [(10, 0.5)],
-    "Edge-Enhancement":       [(10, 2.0)],
-    "Sharpen":                [(10, 1.5)],
-    "Multi-Bandpass":         [[(5, 0.0),  (15, 1.0), (25, 0.0)]],
-    "Multi-Low-Pass":         [[(5, 1.0),  (10, 0.5), (15, 0.2)]],
-    "Multi-High-Pass":        [[(5, 0.0),  (10, 0.5), (15, 0.8)]],
-    "Multi-Pass-Through":     [[(5, 1.0),  (10, 1.0), (15, 1.0)]],
-    "Multi-Gaussian-Blur":    [[(5, 0.5),  (10, 0.8), (15, 0.2)]],
-    "Multi-Edge-Enhancement": [[(5, 1.2),  (10, 1.5), (15, 2.0)]],
-    "Multi-Sharpen":          [[(5, 1.5),  (10, 2.0), (15, 2.5)]],
-}
-
-ALL_VERSIONS: Dict[str, str] = {"Version 1": "1", "Version 2": "2"}
-REVERSED_VERSIONS: Dict[str, str] = {v: k for k, v in ALL_VERSIONS.items()}
-FFT_TYPES: List[str] = ["gaussian", "box"]
-STAGES_COUNT: int = 3
-
-_shorthand_re = re.compile(r"^([a-z]{1,3})(\d+)$")
-
-# ─── StageInfo ─────────────────────────────────────────────────────────────────
-@dataclasses.dataclass
-class StageInfo:
-    """
-    All per-stage parameters.
-    Fields 1-6: same order as sd-webui-freeu for PNG backwards compat.
-    New fields appended at end.
-    """
-    # sd-webui-freeu compat (DO NOT REORDER first 6)
-    backbone_factor:      float = 1.0
-    skip_factor:          float = 1.0
-    backbone_offset:      float = 0.0
-    backbone_width:       float = 0.5
-    skip_cutoff:          float = 0.0
-    skip_high_end_factor: float = 1.0
-    # WAS blending
-    backbone_blend_mode:  str   = "lerp"
-    backbone_blend:       float = 1.0
-    # ComfyUI V2 independent timestep ranges
-    b_start_ratio:  float = 0.0
-    b_end_ratio:    float = 1.0
-    s_start_ratio:  float = 0.0
-    s_end_ratio:    float = 1.0
-    # FFT
-    fft_type:         str   = "box"
-    fft_radius_ratio: float = 0.07
-    hf_boost:         float = 1.0
-    # Adaptive Cap (FreeU_S1S2)
-    enable_adaptive_cap: bool  = False
-    cap_threshold:       float = 0.35
-    cap_factor:          float = 0.6
-    adaptive_cap_mode:   str   = "adaptive"
-
-    def to_dict(self, include_default=False):
-        default = StageInfo()
-        d = dataclasses.asdict(self)
-        if not include_default:
-            d = {k: v for k, v in d.items() if v != getattr(default, k)}
-        return d
-
-    def copy(self):
-        return StageInfo(**dataclasses.asdict(self))
-
-STAGE_FIELD_NAMES = [f.name for f in dataclasses.fields(StageInfo)]
-STAGE_FIELD_COUNT = len(STAGE_FIELD_NAMES)
-
-# ─── State ─────────────────────────────────────────────────────────────────────
-@dataclasses.dataclass
-class State:
-    enable:                bool  = True
-    start_ratio:           Any   = 0.0
-    stop_ratio:            Any   = 1.0
-    transition_smoothness: float = 0.0
-    version:               str   = "1"
-    multiscale_mode:       str   = "Default"
-    multiscale_strength:   float = 1.0
-    override_scales:       str   = ""
-    channel_threshold:     int   = 96
-    stage_infos: List[Any] = dataclasses.field(
-        default_factory=lambda: [StageInfo() for _ in range(STAGES_COUNT)]
-    )
-    # Post-CFG Shift (WAS_PostCFGShift) — stored in presets & PNG
-    pcfg_enabled:   bool  = False
-    pcfg_steps:     int   = 20
-    pcfg_mode:      str   = "inject"
-    pcfg_blend:     float = 1.0
-    pcfg_b:         float = 1.1
-    pcfg_fourier:   bool  = False
-    pcfg_ms_mode:   str   = "Default"
-    pcfg_ms_str:    float = 1.0
-    pcfg_threshold: int   = 1
-    pcfg_s:         float = 0.5
-    pcfg_gain:      float = 1.0
-    verbose:        bool  = False
-
-    def __post_init__(self):
-        self.stage_infos = self._coerce_stages()
-        self.version = ALL_VERSIONS.get(self.version, self.version)
-
-    def _coerce_stages(self):
-        result, raw = [], list(self.stage_infos)
-        i = 0
-        while i < len(raw) and len(result) < STAGES_COUNT:
-            item = raw[i]
-            if isinstance(item, StageInfo):
-                result.append(item); i += 1
-            elif isinstance(item, dict):
-                known = {k: v for k, v in item.items() if k in STAGE_FIELD_NAMES}
-                result.append(StageInfo(**known)); i += 1
-            else:
-                chunk = raw[i:i+STAGE_FIELD_COUNT]
-                result.append(StageInfo(*chunk))
-                i += STAGE_FIELD_COUNT
-        while len(result) < STAGES_COUNT:
-            result.append(StageInfo())
-        return result
-
-    def to_dict(self):
-        d = dataclasses.asdict(self)
-        d["stage_infos"] = [si.to_dict() for si in self.stage_infos]
-        del d["enable"]
-        return d
-
-    def copy(self):
-        d = dataclasses.asdict(self)
-        d["stage_infos"] = [StageInfo(**s) for s in d["stage_infos"]]
-        return State(**d)
-
-    def update_attr(self, key, value):
-        if m := _shorthand_re.match(key):
-            char, idx = m.group(1), int(m.group(2))
-            if 0 <= idx < STAGES_COUNT:
-                si = self.stage_infos[idx]
-                _MAP = {
-                    "b":"backbone_factor","s":"skip_factor","o":"backbone_offset",
-                    "w":"backbone_width","t":"skip_cutoff","h":"skip_high_end_factor",
-                    "bm":"backbone_blend_mode","bb":"backbone_blend",
-                    "bs":"b_start_ratio","be":"b_end_ratio",
-                    "ss":"s_start_ratio","se":"s_end_ratio",
-                    "ft":"fft_type","fr":"fft_radius_ratio","hfb":"hf_boost",
-                    "cap":"enable_adaptive_cap","ct":"cap_threshold","cf":"cap_factor","acm":"adaptive_cap_mode",
-                }
-                if char in _MAP:
-                    setattr(si, _MAP[char], value); return
-        if hasattr(self, key):
-            setattr(self, key, value)
-
-# ─── Singletons ────────────────────────────────────────────────────────────────
-instance: State = State()
-xyz_attrs: Dict[str, Any] = {}
-current_sampling_step: int = 0
-
-# ─── Preset builders ───────────────────────────────────────────────────────────
-def _v1(*pairs):
-    infos = [StageInfo(backbone_factor=b, skip_factor=s) for b,s in pairs]
-    while len(infos) < STAGES_COUNT: infos.append(StageInfo())
-    return State(version="1", stage_infos=infos)
-
-def _v2g(pairs):
-    infos = []
-    for b, s, r, hfb, bs, be, ss, se in pairs:
-        infos.append(StageInfo(
-            backbone_factor=b, skip_factor=s,
-            fft_type="gaussian", fft_radius_ratio=r, hf_boost=hfb,
-            b_start_ratio=bs, b_end_ratio=be,
-            s_start_ratio=ss, s_end_ratio=se,
-        ))
-    while len(infos) < STAGES_COUNT: infos.append(StageInfo())
-    return State(version="2", stage_infos=infos)
-
-default_presets: Dict[str, State] = {
-    "SD1.4 Recommendations": _v1((1.2,0.9),(1.4,0.2),(1.0,1.0)),
-    "SD2.1 Recommendations": _v1((1.1,0.9),(1.2,0.2),(1.0,1.0)),
-    "SDXL Recommendations":  _v1((1.1,0.6),(1.2,0.4),(1.0,1.0)),
-    "SD1.5 V2 Gaussian": _v2g([
-        (1.2,0.9,0.07,1.0, 0.0,0.35, 0.35,1.0),
-        (1.4,0.2,0.07,1.0, 0.0,0.35, 0.35,1.0),
-    ]),
-    "SD1.5 V2 High Detail": _v2g([
-        (1.4,0.8,0.08,1.2, 0.0,0.35, 0.35,1.0),
-        (1.6,0.1,0.06,1.0, 0.0,0.35, 0.35,1.0),
-    ]),
-    "SDXL V2 Gaussian": _v2g([
-        (1.1,0.6,0.05,1.1, 0.0,0.35, 0.35,1.0),
-        (1.2,0.4,0.05,1.1, 0.0,0.35, 0.35,1.0),
-    ]),
-    "SD1.5 Adaptive Cap": State(version="2", stage_infos=[
-        StageInfo(backbone_factor=1.3,skip_factor=0.9,
-                  fft_type="gaussian",fft_radius_ratio=0.08,hf_boost=1.2,
-                  b_start_ratio=0.0,b_end_ratio=0.35,
-                  s_start_ratio=0.35,s_end_ratio=1.0,
-                  enable_adaptive_cap=True,cap_threshold=0.35,
-                  cap_factor=0.6,adaptive_cap_mode="adaptive"),
-        StageInfo(backbone_factor=1.4,skip_factor=0.2,
-                  fft_type="gaussian",fft_radius_ratio=0.06,hf_boost=1.0,
-                  b_start_ratio=0.0,b_end_ratio=0.35,
-                  s_start_ratio=0.35,s_end_ratio=1.0,
-                  enable_adaptive_cap=True,cap_threshold=0.70,
-                  cap_factor=0.6,adaptive_cap_mode="adaptive"),
-        StageInfo(),
-    ]),
-    "Independent B/S (SD1.5)": _v2g([
-        (1.2,0.9,0.07,1.0, 0.0,0.35, 0.35,1.0),
-        (1.4,0.2,0.06,1.0, 0.0,0.35, 0.35,1.0),
-    ]),
-}
-
-all_presets: Dict[str, State] = {}
-PRESETS_PATH = pathlib.Path(__file__).parent.parent / "presets.json"
-
-def reload_presets():
-    all_presets.clear()
-    all_presets.update(default_presets)
-    all_presets.update(_load_user_presets())
-
-def _load_user_presets():
-    if not PRESETS_PATH.exists(): return {}
-    try:
-        with open(PRESETS_PATH, encoding="utf-8") as f:
-            raw = json.load(f)
-    except Exception as e:
-        print(f"[MegaFreeU] preset load error: {e}", file=sys.stderr)
-        return {}
-    result = {}
-    _state_fields = {f.name for f in dataclasses.fields(State)}
-    for k, v in raw.items():
-        try:
-            # Filter unknown keys so future/old fields don't crash State(**v)
-            known = {fk: fv for fk, fv in v.items() if fk in _state_fields}
-            result[k] = State(**known)
-        except Exception as e:
-            print(f"[MegaFreeU] skipping preset {k!r}: {e}", file=sys.stderr)
-    return result
-
-def save_presets(custom=None):
-    if custom is None: custom = get_user_presets()
-    try:
-        PRESETS_PATH.parent.mkdir(parents=True, exist_ok=True)
-        with open(PRESETS_PATH, "w", encoding="utf-8") as f:
-            json.dump({k: v.to_dict() for k,v in custom.items()}, f, indent=4)
-    except Exception as e:
-        print(f"[MegaFreeU] preset save error: {e}", file=sys.stderr)
-
-def get_user_presets():
-    return {k: v for k,v in all_presets.items() if k not in default_presets}
-
-def apply_xyz():
-    global instance
-    if pk := xyz_attrs.get("preset"):
-        if p := all_presets.get(pk):
-            instance = p.copy()
-        elif pk != "UI Settings":
-            print(f"[MegaFreeU] XYZ preset '{pk}' not found", file=sys.stderr)
-    for k, v in xyz_attrs.items():
-        if k != "preset":
-            instance.update_attr(k, v)