openg2g.grid¶

@dataclass(frozen=True)
class PhaseVoltages:
    """Per-phase voltage magnitudes in per-unit.

    Phases missing from the bus have NaN for that field.

    Attributes:
        a: Phase A voltage magnitude (pu).
        b: Phase B voltage magnitude (pu).
        c: Phase C voltage magnitude (pu).
    """

    a: float
    b: float
    c: float

@dataclass(frozen=True)
class BusVoltages:
    """Per-bus, per-phase voltage map.

    Access: voltages["671"].a -> Vpu for bus 671, phase A.
    Buses missing a phase have NaN for that field.
    """

    _data: dict[str, PhaseVoltages]

    def __getitem__(self, bus: str) -> PhaseVoltages:
        return self._data[bus]

    def buses(self) -> list[str]:
        """Return the list of bus names."""
        return list(self._data.keys())

    def __contains__(self, bus: str) -> bool:
        return bus in self._data

@dataclass(frozen=True)
class GridState:
    """State emitted by the grid simulator each timestep.

    Attributes:
        time_s: Simulation time in seconds.
        voltages: Per-bus, per-phase voltage magnitudes.
        tap_positions: Current regulator tap positions, or `None` if
            no regulator is present.
    """

    time_s: float
    voltages: BusVoltages
    tap_positions: TapPosition | None = None

class GridBackend(Generic[GridStateT], ABC):
    """Interface for grid simulation backends."""

    _INIT_SENTINEL = object()

    def __init__(self) -> None:
        self._state: GridStateT | None = None
        self._history: list[GridStateT] = []
        self._grid_base_init = GridBackend._INIT_SENTINEL

    def _check_base_init(self) -> None:
        if getattr(self, "_grid_base_init", None) is not GridBackend._INIT_SENTINEL:
            raise TypeError(f"{type(self).__name__}.__init__ must call super().__init__().")

    @property
    @abstractmethod
    def dt_s(self) -> Fraction:
        """Native timestep as a Fraction (seconds)."""

    @final
    @property
    def state(self) -> GridStateT:
        """Latest emitted state.

        Raises:
            RuntimeError: If accessed before the first `step()` call.
        """
        self._check_base_init()
        if self._state is None:
            raise RuntimeError(f"{type(self).__name__}.state accessed before first step().")
        return self._state

    @final
    def history(self, n: int | None = None) -> list[GridStateT]:
        """Return emitted state history (all, or latest `n`)."""
        self._check_base_init()
        if n is None:
            return list(self._history)
        if n <= 0:
            return []
        return list(self._history[-int(n) :])

    @final
    def do_step(
        self,
        clock: SimulationClock,
        power_samples_w: dict[DatacenterBackend, list[ThreePhase]],
        events: EventEmitter,
    ) -> GridStateT:
        """Call `step`, record the state, and return it.

        Called by the coordinator. Subclasses should not override this.
        """
        self._check_base_init()
        state = self.step(clock, power_samples_w, events)
        self._state = state
        self._history.append(state)
        return state

    @abstractmethod
    def step(
        self,
        clock: SimulationClock,
        power_samples_w: dict[DatacenterBackend, list[ThreePhase]],
        events: EventEmitter,
    ) -> GridStateT:
        """Advance one native timestep and return state for this step."""

    @abstractmethod
    def apply_control(self, command: GridCommand, events: EventEmitter) -> None:
        """Apply one control command."""

    @abstractmethod
    def voltages_vector(self) -> np.ndarray:
        """Return voltage magnitudes in `v_index` order."""

    @abstractmethod
    def estimate_sensitivity(
        self, perturbation_kw: float = 100.0, dc: DatacenterBackend | None = None
    ) -> tuple[np.ndarray, np.ndarray]:
        """Estimate voltage sensitivity matrix (H = dv/dp) and return `(H, v0)`."""

    @property
    @abstractmethod
    def v_index(self) -> list[tuple[str, int]]:
        """Fixed (bus, phase) ordering used by [`voltages_vector`][..voltages_vector]."""

    @final
    def do_reset(self) -> None:
        """Clear history and call `reset`.

        Called by the coordinator. Subclasses should not override this.
        """
        self._check_base_init()
        self._state = None
        self._history.clear()
        self.reset()

    @abstractmethod
    def reset(self) -> None:
        """Reset simulation state to initial conditions.

        Called by the coordinator (via `do_reset`) before each
        [`start`][..start]. Must clear all simulation state: counters,
        cached values. Configuration (dt_s, case files, tap schedules)
        is not affected. History is cleared automatically by
        `do_reset`.

        Abstract so every implementation explicitly enumerates its state.
        A forgotten field is a bug -- not clearing it silently corrupts
        the second run.
        """

    def start(self) -> None:
        """Acquire per-run resources (solver circuits, connections).

        Called after [`reset`][..reset], before the simulation loop.
        Override for backends that need resource acquisition (e.g.,
        [`OpenDSSGrid`][openg2g.grid.opendss.OpenDSSGrid] compiles its
        DSS circuit here). No-op by default because most offline
        components have no resources to acquire.
        """

    def stop(self) -> None:
        """Release per-run resources. Simulation state is preserved.

        Called after the simulation loop in LIFO order. Override for
        backends that acquired resources in [`start`][..start]. No-op
        by default.
        """

class GridCommand:
    """Base for commands targeting the grid backend.

    Subclass this for each concrete grid command kind.
    The coordinator routes commands to backends based on this type hierarchy.
    """

    def __init__(self) -> None:
        if type(self) is GridCommand:
            raise TypeError("GridCommand cannot be instantiated directly; subclass it.")

@dataclass(frozen=True)
class SetTaps(GridCommand):
    """Set regulator tap positions.

    Attributes:
        tap_position: Per-phase tap ratios. Phases set to `None` are
            unchanged.
    """

    tap_position: TapPosition

@dataclass(frozen=True)
class SetStoragePower(GridCommand):
    """Set energy storage real/reactive power.

    Positive real power discharges storage into the grid; negative real power
    charges storage from the grid.

    Attributes:
        storage: Attached storage resource to command.
        power_kw: Real-power setpoint in kW.
        reactive_power_kvar: Reactive-power setpoint in kvar. Positive values
            inject reactive power.
    """

    storage: EnergyStorage
    power_kw: float
    reactive_power_kvar: float = 0.0

@dataclass(frozen=True)
class TapPosition:
    """Regulator tap position as a mapping of regulator names to tap ratios.

    All regulators are stored in a single `regulators` dict.  For
    convenience, per-phase keyword arguments `a`, `b`, `c` are
    accepted and stored under those keys:

    ```python
    # These are equivalent:
    TapPosition(a=1.075, b=1.05, c=1.075)
    TapPosition(regulators={"a": 1.075, "b": 1.05, "c": 1.075})
    ```

    Named regulators for multi-bank systems:

    ```python
    TapPosition(regulators={"creg1a": 1.075, "creg1b": 1.05, "creg2a": 1.0})
    ```

    Attributes:
        regulators: Mapping of regulator name to tap ratio (pu).
    """

    regulators: dict[str, float] = field(default_factory=dict)

    def __init__(
        self,
        *,
        a: float | None = None,
        b: float | None = None,
        c: float | None = None,
        regulators: dict[str, float] | None = None,
    ) -> None:
        merged = dict(regulators) if regulators else {}
        for key, val in zip(_PHASE_KEYS, (a, b, c), strict=True):
            if val is not None:
                merged[key] = val
        if not merged:
            raise ValueError("TapPosition requires at least one regulator tap value.")
        object.__setattr__(self, "regulators", merged)

    @property
    def a(self) -> float | None:
        """Phase A tap ratio, or `None` if not set."""
        return self.regulators.get("a")

    @property
    def b(self) -> float | None:
        """Phase B tap ratio, or `None` if not set."""
        return self.regulators.get("b")

    @property
    def c(self) -> float | None:
        """Phase C tap ratio, or `None` if not set."""
        return self.regulators.get("c")

    def at(self, t: float) -> TapSchedule:
        """Schedule this position at time `t` seconds."""
        return TapSchedule(((t, self),))

class TapSchedule:
    """Ordered sequence of scheduled tap positions.

    Build using [`TapPosition.at`][..TapPosition.at] and the `|` operator:

    ```python
    TAP_STEP = 0.00625  # standard 5/8% tap step
    schedule = (
        TapPosition(a=1.0 + 14 * TAP_STEP, b=1.0 + 6 * TAP_STEP, c=1.0 + 15 * TAP_STEP).at(t=0)
        | TapPosition(a=1.0 + 16 * TAP_STEP).at(t=25 * 60)
    )
    ```

    Raises:
        ValueError: If two entries share the same timestamp.
    """

    __slots__ = ("_entries",)

    def __init__(self, entries: tuple[tuple[float, TapPosition], ...]) -> None:
        self._entries = tuple(sorted(entries, key=lambda e: e[0]))
        times = [t for t, _ in self._entries]
        if len(times) != len(set(times)):
            seen: set[float] = set()
            dupes = sorted({t for t in times if t in seen or seen.add(t)})
            raise ValueError(f"TapSchedule has duplicate timestamps: {dupes}")

    def __or__(self, other: TapSchedule) -> TapSchedule:
        return TapSchedule(self._entries + other._entries)

    def __iter__(self) -> Iterator[tuple[float, TapPosition]]:
        return iter(self._entries)

    def __len__(self) -> int:
        return len(self._entries)

    def __bool__(self) -> bool:
        return bool(self._entries)

    def __repr__(self) -> str:
        parts: list[str] = []
        for t, p in self._entries:
            fields = []
            if p.a is not None:
                fields.append(f"a={p.a}")
            if p.b is not None:
                fields.append(f"b={p.b}")
            if p.c is not None:
                fields.append(f"c={p.c}")
            # Show non-phase regulators
            for name, val in p.regulators.items():
                if name not in _PHASE_KEYS:
                    fields.append(f"{name}={val}")
            parts.append(f"TapPosition({', '.join(fields)}).at(t={t})")
        return " | ".join(parts)

class Generator(ABC):
    """Abstract power source attached to a grid bus.

    Subclass this to define how real power output varies over time.
    The grid calls [`power_kw`][..power_kw] at each simulation timestep.
    """

    @abstractmethod
    def power_kw(self, t: float) -> float:
        """Return generator output in kW at simulation time *t* (seconds)."""

class ConstantGenerator(Generator):
    """Fixed power output at all times.

    Args:
        peak_kw: Constant output in kW.
    """

    def __init__(self, peak_kw: float) -> None:
        self._peak_kw = float(peak_kw)

    def power_kw(self, t: float) -> float:
        return self._peak_kw

class CSVProfileGenerator(Generator):
    """Power output interpolated from a CSV time series.

    The CSV file must have two columns: time (seconds) and power (kW).
    The first row is treated as a header and skipped.

    Args:
        csv_path: Path to the CSV file.
    """

    def __init__(self, csv_path: Path) -> None:
        data = np.loadtxt(csv_path, delimiter=",", skiprows=1)
        self._time = data[:, 0]
        self._power = data[:, 1]

    def power_kw(self, t: float) -> float:
        return float(np.interp(t, self._time, self._power))

class SyntheticPV(Generator):
    """Synthetic PV profile with cloud dips, trends, and fluctuation.

    Each `site_idx` produces a visually distinct curve. These are
    synthetic "lorem ipsum" profiles for demonstration, not based on
    real irradiance data.

    Args:
        peak_kw: Peak PV output in kW.
        site_idx: Site index for distinct per-site profiles (0 to 2).
    """

    _T = 3600  # assumed total simulation duration for trend shaping

    def __init__(self, peak_kw: float, site_idx: int = 0) -> None:
        self._peak_kw = float(peak_kw)
        self._site_idx = int(site_idx)

    def power_kw(self, t: float) -> float:
        T = self._T
        idx = self._site_idx
        if idx == 0:
            trend = 0.85 - 0.30 * (t / T)
            cloud = 1.0
            cloud -= 0.55 * smooth_bump(t, 600, 120)
            cloud -= 0.40 * smooth_bump(t, 2100, 180)
            fluct = irregular_fluct(t, seed=0.3)
            return max(0.0, self._peak_kw * trend * max(cloud, 0.05) * fluct)
        elif idx == 1:
            ramp = 0.55 + 0.40 * smooth_bump(t, 1200, 900)
            cloud = 1.0
            cloud -= 0.60 * smooth_bump(t, 1680, 240)
            cloud -= 0.25 * smooth_bump(t, 2400, 150)
            fluct = irregular_fluct(t, seed=2.1)
            return max(0.0, self._peak_kw * ramp * max(cloud, 0.05) * fluct)
        elif idx == 2:
            ramp = 0.30 + 0.65 * min(1.0, t / 900.0)
            cloud = 1.0
            cloud -= 0.70 * smooth_bump(t, 2700, 300)
            cloud -= 0.30 * smooth_bump(t, 1200, 100)
            fluct = irregular_fluct(t, seed=2.0 + idx * 3.7)
            return max(0.0, self._peak_kw * ramp * max(cloud, 0.05) * fluct)
        raise ValueError(f"Unsupported site_idx {idx} for SyntheticPV; valid range is 0-2.")

class ExternalLoad(ABC):
    """Abstract time-varying load attached to a grid bus.

    Subclass this to define how real power consumption varies over time.
    The grid calls [`power_kw`][..power_kw] at each simulation timestep.
    """

    @abstractmethod
    def power_kw(self, t: float) -> float:
        """Return load consumption in kW at simulation time *t* (seconds)."""

class ConstantLoad(ExternalLoad):
    """Fixed power consumption at all times.

    Args:
        peak_kw: Constant consumption in kW.
    """

    def __init__(self, peak_kw: float) -> None:
        self._peak_kw = float(peak_kw)

    def power_kw(self, t: float) -> float:
        return self._peak_kw

class CSVProfileLoad(ExternalLoad):
    """Power consumption interpolated from a CSV time series.

    The CSV file must have two columns: time (seconds) and power (kW).
    The first row is treated as a header and skipped.

    Args:
        csv_path: Path to the CSV file.
    """

    def __init__(self, csv_path: Path) -> None:
        data = np.loadtxt(csv_path, delimiter=",", skiprows=1)
        self._time = data[:, 0]
        self._power = data[:, 1]

    def power_kw(self, t: float) -> float:
        return float(np.interp(t, self._time, self._power))

class SyntheticLoad(ExternalLoad):
    """Synthetic load profile with bumps and fluctuation.

    Each `site_idx` produces a visually distinct curve with different
    diurnal patterns. These are synthetic profiles for demonstration.

    Args:
        peak_kw: Peak load consumption in kW.
        site_idx: Site index for distinct per-site profiles (0 to 4).
    """

    def __init__(self, peak_kw: float, site_idx: int = 0) -> None:
        self._peak_kw = float(peak_kw)
        self._site_idx = int(site_idx)

    def power_kw(self, t: float) -> float:
        idx = self._site_idx
        fluct_period = 130.0 + idx * 37
        fluct = 1.0 + 0.06 * math.sin(2 * math.pi * t / fluct_period + idx * 1.4)

        if idx == 0:
            base = 0.15 + 0.85 * smooth_bump(t, 2280, 1400)
            surge = 0.20 * smooth_bump(t, 2280, 180)
            return max(0.0, self._peak_kw * (base + surge) * fluct)
        elif idx == 1:
            base = 0.10
            base += 0.50 * smooth_bump(t, 1500, 600)
            base += 0.80 * smooth_bump(t, 2880, 500)
            return max(0.0, self._peak_kw * base * fluct)
        elif idx == 2:
            base = 0.80 - 0.55 * smooth_bump(t, 1800, 1200)
            surge = 0.70 * smooth_bump(t, 2520, 400)
            return max(0.0, self._peak_kw * (base + surge) * fluct)
        elif idx == 3:
            base = 0.10 + 0.90 * smooth_bump(t, 3120, 800)
            return max(0.0, self._peak_kw * base * fluct)
        elif idx == 4:
            base = 0.10
            base += 0.60 * smooth_bump(t, 1080, 300)
            base += 0.75 * smooth_bump(t, 2100, 350)
            base += 0.90 * smooth_bump(t, 3300, 300)
            return max(0.0, self._peak_kw * base * fluct)
        raise ValueError(f"Invalid site_idx {idx} for SyntheticLoad; must be 0 to 4.")

class OpenDSSGrid(GridBackend[GridState]):
    """OpenDSS-based grid simulator for distribution-level voltage analysis.

    Uses OpenDSS as a power flow solver. The user's DSS case file defines
    the network topology. Datacenters, generators, loads, and storage systems
    are attached to specific buses via [`attach_dc`][..attach_dc],
    [`attach_generator`][..attach_generator], [`attach_load`][..attach_load],
    and [`attach_storage`][..attach_storage] before calling
    [`start`][..start].

    Bus voltages (kV) are looked up from the DSS model after compile --
    callers never need to specify `bus_kv`.

    Args:
        dss_case_dir: Path to the directory containing OpenDSS case files.
        dss_master_file: Name of the master DSS file relative to *dss_case_dir*.
        dt_s: Grid simulation timestep (seconds).
        source_pu: Override source voltage (pu). If None, uses the DSS default.
        dss_controls: Whether to let OpenDSS iterate its built-in control
            loops during each solve. Default False.
        initial_tap_position: Initial regulator tap position applied before
            the first solve.
        exclude_buses: Buses to exclude from voltage indexing.
    """

    def __init__(
        self,
        *,
        dss_case_dir: str | Path,
        dss_master_file: str,
        dt_s: Fraction = Fraction(1),
        source_pu: float | None = None,
        dss_controls: bool = False,
        initial_tap_position: TapPosition | None = None,
        exclude_buses: Sequence[str] = (),
    ) -> None:
        super().__init__()
        if dss is None:
            raise RuntimeError("OpenDSSDirect is required. Install with: pip install openg2g[opendss]")

        self._case_dir = str(Path(dss_case_dir).resolve())
        self._master = str(dss_master_file)
        self._dt_s = dt_s
        self._source_pu = source_pu
        self._dss_controls = bool(dss_controls)
        self._initial_tap_position = initial_tap_position
        self._exclude_buses = tuple(str(b) for b in exclude_buses)

        self._reg_map: dict[str, tuple[str, int]] | None = None
        self._phase_to_reg: dict[int, str | None] | None = None

        # Attachments (populated before start)
        self._dc_attachments: dict[DatacenterBackend, _DCAttachment] = {}
        self._gen_attachments: list[_GenAttachment] = []
        self._load_attachments: list[_LoadAttachment] = []
        self._storage_attachments: list[_StorageAttachment] = []

        # Simulation state (cleared by reset)
        self._prev_power: dict[DatacenterBackend, ThreePhase] = {}

        # DSS-derived data (populated by start)
        self._started = False
        self.all_buses: list[str] = []
        self.buses_with_phase: dict[int, list[str]] = {}
        self._v_index: list[tuple[str, int]] = []

    def attach_dc(
        self,
        dc: DatacenterBackend,
        *,
        bus: str,
        connection_type: str = "wye",
        power_factor: float = 0.95,
    ) -> None:
        """Attach a datacenter load to a grid bus.

        Args:
            dc: Datacenter backend whose power output will be injected at *bus*.
            bus: Bus name on the grid.
            connection_type: Wye or delta connection.
            power_factor: Power factor for reactive power computation.
        """
        if self._started:
            raise RuntimeError("Cannot attach after start().")
        if dc in self._dc_attachments:
            raise ValueError(f"Datacenter {dc.name!r} already attached.")
        if not _DSS_SAFE_NAME.match(dc.name):
            raise ValueError(
                f"Datacenter name {dc.name!r} contains characters unsafe for DSS commands. "
                "Use only letters, digits, underscores, and hyphens."
            )
        pf = max(min(float(power_factor), 0.999999), 1e-6)
        self._dc_attachments[dc] = _DCAttachment(
            bus=bus,
            connection_type=connection_type,
            power_factor=power_factor,
            tanphi=math.tan(math.acos(pf)),
        )

    def attach_generator(
        self,
        generator: Generator,
        *,
        bus: str,
        power_factor: float = 1.0,
    ) -> None:
        """Attach a generator (PV, wind, etc.) to a grid bus.

        Args:
            generator: Generator whose output will be injected at *bus*.
            bus: Bus name on the grid.
            power_factor: Power factor for reactive power computation.
        """
        if self._started:
            raise RuntimeError("Cannot attach after start().")
        pf = max(min(float(power_factor), 0.999999), 1e-6)
        self._gen_attachments.append(
            _GenAttachment(
                bus=bus,
                generator=generator,
                power_factor=power_factor,
                tanphi=math.tan(math.acos(pf)),
            )
        )

    def attach_load(
        self,
        load: ExternalLoad,
        *,
        bus: str,
        power_factor: float = 0.96,
    ) -> None:
        """Attach a time-varying external load to a grid bus.

        Args:
            load: Load whose consumption will be injected at *bus*.
            bus: Bus name on the grid.
            power_factor: Power factor for reactive power computation.
        """
        if self._started:
            raise RuntimeError("Cannot attach after start().")
        pf = max(min(float(power_factor), 0.999999), 1e-6)
        self._load_attachments.append(
            _LoadAttachment(
                bus=bus,
                load=load,
                power_factor=power_factor,
                tanphi=math.tan(math.acos(pf)),
            )
        )

    def attach_storage(
        self,
        storage: EnergyStorage,
        *,
        bus: str,
        connection_type: str = "wye",
    ) -> None:
        """Attach an energy storage system to a grid bus.

        Args:
            storage: Energy storage resource whose dispatch will be written to
                a native OpenDSS `Storage` element. Positive real power
                discharges into the grid; negative real power charges from it.
            bus: Bus name on the grid.
            connection_type: Wye or delta connection.
        """
        if self._started:
            raise RuntimeError("Cannot attach after start().")
        if not _DSS_SAFE_NAME.match(storage.name):
            raise ValueError(
                f"Storage name {storage.name!r} contains characters unsafe for DSS commands. "
                "Use only letters, digits, underscores, and hyphens."
            )
        for att in self._storage_attachments:
            if att.storage is storage:
                raise ValueError(f"Storage {storage.name!r} is already attached.")
            if att.storage.name.lower() == storage.name.lower():
                raise ValueError(
                    f"Storage name {storage.name!r} collides with an already attached storage. "
                    "DSS element names must be unique (case-insensitive)."
                )

        conn = str(connection_type).lower()
        if conn not in {"wye", "delta"}:
            raise ValueError("connection_type must be 'wye' or 'delta'.")

        self._storage_attachments.append(_StorageAttachment(bus=bus, storage=storage, connection_type=conn))

    @property
    def dt_s(self) -> Fraction:
        return self._dt_s

    @property
    def v_index(self) -> list[tuple[str, int]]:
        if not self._started:
            raise RuntimeError("OpenDSSGrid.v_index accessed before start().")
        return list(self._v_index)

    def dc_bus(self, dc: DatacenterBackend) -> str:
        """Return the bus name a datacenter is attached to."""
        return self._dc_attachments[dc].bus

    def step(
        self,
        clock: SimulationClock,
        power_samples_w: dict[DatacenterBackend, list[ThreePhase]],
        events: EventEmitter,
    ) -> GridState:
        """Advance one grid period and return the resulting grid state.

        Args:
            clock: Simulation clock.
            power_samples_w: Dict mapping datacenter objects to lists of
                three-phase power samples (watts) collected since the last
                grid step.
            events: Event emitter for grid events.
        """
        # 1. Set DC load powers
        for dc, att in self._dc_attachments.items():
            dc_samples = power_samples_w.get(dc, [])
            if not dc_samples:
                if dc not in self._prev_power:
                    raise RuntimeError(
                        f"OpenDSSGrid.step() called with no power samples for DC '{dc.name}' and no previous power."
                    )
                power = self._prev_power[dc]
            else:
                power = dc_samples[-1]

            self._prev_power[dc] = power

            kW_A = power.a / 1e3
            kW_B = power.b / 1e3
            kW_C = power.c / 1e3

            for name, kw in zip(att.load_names, (kW_A, kW_B, kW_C), strict=True):
                dss.Loads.Name(name)
                dss.Loads.kW(kw)
                dss.Loads.kvar(kw * att.tanphi)

        # 2. Set generator powers (negative loads = injection)
        for att in self._gen_attachments:
            kw = att.generator.power_kw(clock.time_s)
            kvar = kw * att.tanphi
            for name in att.load_names:
                dss.Loads.Name(name)
                dss.Loads.kW(-kw)
                dss.Loads.kvar(-kvar)

        # 3. Set external load powers
        for att in self._load_attachments:
            kw = att.load.power_kw(clock.time_s)
            kvar = kw * att.tanphi
            for name in att.load_names:
                dss.Loads.Name(name)
                dss.Loads.kW(kw)
                dss.Loads.kvar(kvar)

        # 4. Set storage dispatch. OpenDSS Storage uses signed kW:
        # positive = discharging, negative = charging.
        for att in self._storage_attachments:
            self._set_storage_dispatch(att, clock.time_s)

        self._solve()

        voltages = self._snapshot_bus_voltages()
        tap_positions = self._read_current_taps()
        self._finish_storage_timestep(clock.time_s)
        return GridState(time_s=clock.time_s, voltages=voltages, tap_positions=tap_positions)

    @functools.singledispatchmethod
    def apply_control(self, command: GridCommand, events: EventEmitter) -> None:
        """Apply a control command. Dispatches on command type."""
        raise TypeError(f"OpenDSSGrid does not support {type(command).__name__}")

    @apply_control.register
    def apply_control_set_taps(self, command: SetTaps, events: EventEmitter) -> None:
        tap_map = self._tap_position_to_reg_dict(command.tap_position)
        self._set_reg_taps(tap_map)
        events.emit(
            "grid.taps.updated",
            {"tap_position": command.tap_position},
        )

    @apply_control.register
    def apply_control_set_storage_power(self, command: SetStoragePower, events: EventEmitter) -> None:
        att = self._storage_attachment_for(command.storage)
        att.storage.set_power_kw(command.power_kw, command.reactive_power_kvar)
        events.emit(
            "grid.storage_power.updated",
            {
                "storage_name": att.storage.name,
                "power_kw": float(command.power_kw),
                "reactive_power_kvar": float(command.reactive_power_kvar),
            },
        )

    def reset(self) -> None:
        self._prev_power = {}
        self._reset_storage_resources()
        self._started = False

    def start(self) -> None:
        if not self._dc_attachments:
            raise RuntimeError("At least one datacenter must be attached before start().")
        self._reset_storage_resources()
        self._init_dss()
        self._v_index = self._build_v_index()
        self._build_vmag_indices()
        self._build_snapshot_indices()
        self._started = True
        dc_info = ", ".join(f"{dc.name}@{att.bus}" for dc, att in self._dc_attachments.items())
        n_gen = len(self._gen_attachments)
        n_load = len(self._load_attachments)
        n_storage = len(self._storage_attachments)
        logger.info(
            "OpenDSSGrid: case=%s, dc=[%s], %d gen, %d ext load, %d storage, dt=%s s, controls=%s, "
            "%d buses, %d bus-phases",
            self._master,
            dc_info,
            n_gen,
            n_load,
            n_storage,
            self._dt_s,
            self._dss_controls,
            len(self.all_buses),
            len(self._v_index),
        )

    def voltages_vector(self) -> np.ndarray:
        """Return voltage magnitudes (pu) in the fixed
        [`v_index`][openg2g.grid.base.GridBackend.v_index] ordering."""
        if not self._started:
            raise RuntimeError("OpenDSSGrid.voltages_vector() called before start().")
        vmag = dss.Circuit.AllBusMagPu()
        return vmag[self._v_index_to_vmag]

    def estimate_sensitivity(
        self,
        perturbation_kw: float = 100.0,
        dc: DatacenterBackend | None = None,
    ) -> tuple[np.ndarray, np.ndarray]:
        """Estimate voltage sensitivity matrix H = dv/dp (pu per kW).

        Uses finite differences on the 3 single-phase DC loads for a specific
        datacenter.

        Args:
            perturbation_kw: Perturbation size in kW.
            dc: Which datacenter's loads to perturb. Required when multiple
                DCs are attached; auto-selected when only one exists.

        Returns:
            Tuple of `(sensitivity, baseline_voltages)`.
                `sensitivity` has shape `(M, 3)` where M is the number
                of bus-phase pairs in `v_index`.
                `baseline_voltages` has shape `(M,)`.
        """
        perturbation_kw = float(perturbation_kw)
        if perturbation_kw <= 0:
            raise ValueError("perturbation_kw must be positive.")

        if dc is None:
            if len(self._dc_attachments) == 1:
                dc = next(iter(self._dc_attachments))
            else:
                raise ValueError("dc is required when multiple datacenters are attached.")

        att = self._dc_attachments[dc]
        load_names = att.load_names
        dq_kvar = perturbation_kw * att.tanphi

        dss.Solution.SolveNoControl()
        baseline_voltages = self.voltages_vector()

        p0 = np.zeros(3, dtype=float)
        q0 = np.zeros(3, dtype=float)
        for j, ld in enumerate(load_names):
            dss.Loads.Name(ld)
            p0[j] = float(dss.Loads.kW())
            q0[j] = float(dss.Loads.kvar())

        M = len(self._v_index)
        sensitivity = np.zeros((M, 3), dtype=float)

        for j, ld in enumerate(load_names):
            dss.Text.Command(f"Edit Load.{ld} kW={p0[j] + perturbation_kw:.6f} kvar={q0[j] + dq_kvar:.6f}")
            dss.Solution.SolveNoControl()

            sensitivity[:, j] = (self.voltages_vector() - baseline_voltages) / perturbation_kw

            dss.Text.Command(f"Edit Load.{ld} kW={p0[j]:.6f} kvar={q0[j]:.6f}")

        self._solve()

        return sensitivity, baseline_voltages

    def _init_dss(self) -> None:
        dss.Basic.ClearAll()
        master_path = str(Path(self._case_dir) / self._master)
        dss.Text.Command(f'Compile "{master_path}"')

        # Override source voltage if requested
        if self._source_pu is not None:
            dss.Text.Command(f"Edit Vsource.source pu={self._source_pu}")

        self._reg_map = self._cache_regcontrol_map()
        self._phase_to_reg = self._build_phase_to_reg_map(self._reg_map)

        # Helper to look up bus line-to-neutral kV from DSS model
        def _bus_kv_ln(bus: str) -> float:
            dss.Circuit.SetActiveBus(bus)
            return float(dss.Bus.kVBase())

        # Create DC load elements
        _DC_LOAD_NAMES = ("DataCenterA", "DataCenterB", "DataCenterC")
        for dc, att in self._dc_attachments.items():
            if len(self._dc_attachments) == 1:
                load_names = _DC_LOAD_NAMES
            else:
                load_names = (f"DC_{dc.name}_A", f"DC_{dc.name}_B", f"DC_{dc.name}_C")
            att.load_names = load_names
            kv_ln = _bus_kv_ln(att.bus)
            conn = att.connection_type
            if conn == "delta":
                load_kv = kv_ln * math.sqrt(3.0)
            else:
                load_kv = kv_ln
            for ph, nm in zip(_PHASES, load_names, strict=True):
                dss.Text.Command(
                    f"New Load.{nm} bus1={att.bus}.{ph} phases=1 conn={conn} kV={load_kv:.6f} kW=0 kvar=0 model=1"
                )

        # Create generator elements (negative loads)
        for i, att in enumerate(self._gen_attachments):
            load_names = (f"Gen_{i}_A", f"Gen_{i}_B", f"Gen_{i}_C")
            att.load_names = load_names
            kv_ln = _bus_kv_ln(att.bus)
            for ph, nm in zip(_PHASES, load_names, strict=True):
                dss.Text.Command(
                    f"New Load.{nm} bus1={att.bus}.{ph} phases=1 conn=wye kV={kv_ln:.6f} kW=0 kvar=0 model=1"
                )

        # Create external load elements
        for i, att in enumerate(self._load_attachments):
            load_names = (f"ExtLoad_{i}_A", f"ExtLoad_{i}_B", f"ExtLoad_{i}_C")
            att.load_names = load_names
            kv_ln = _bus_kv_ln(att.bus)
            for ph, nm in zip(_PHASES, load_names, strict=True):
                dss.Text.Command(
                    f"New Load.{nm} bus1={att.bus}.{ph} phases=1 conn=wye kV={kv_ln:.6f} kW=0 kvar=0 model=1"
                )

        # Create storage elements. OpenDSS expects line-to-line kV for
        # three-phase Storage elements.
        for att in self._storage_attachments:
            storage = att.storage
            att.element_name = storage.name
            self._require_three_phase_bus(att.bus, storage.name)
            kv_ll = _bus_kv_ln(att.bus) * math.sqrt(3.0)
            initial_kwh = storage.capacity_kwh * storage.initial_soc
            reserve_pct = storage.reserve_soc * 100.0
            charge_eff_pct = storage.charge_efficiency * 100.0
            discharge_eff_pct = storage.discharge_efficiency * 100.0
            dss.Text.Command(
                f"New Storage.{att.element_name} bus1={att.bus}.1.2.3 phases=3 conn={att.connection_type} "
                f"kV={kv_ll:.6f} kVA={storage.rated_apparent_power_kva:.6f} "
                f"kWRated={storage.rated_power_kw:.6f} kWhRated={storage.capacity_kwh:.6f} "
                f"kWhStored={initial_kwh:.6f} %Reserve={reserve_pct:.6f} "
                f"%EffCharge={charge_eff_pct:.6f} %EffDischarge={discharge_eff_pct:.6f} "
                f"%IdlingkW={storage.idle_loss_percent:.6f} DispMode=EXTERNAL State=IDLING kW=0 kvar=0"
            )

        dss.Text.Command("Reset")
        dss.Text.Command("Set Mode=Time")
        dss.Text.Command(f"Set Stepsize={float(self._dt_s)}s")
        if self._dss_controls:
            dss.Text.Command("Set ControlMode=Time")
        else:
            dss.Text.Command("Set ControlMode=Off")

        if self._initial_tap_position is not None:
            self._set_reg_taps(self._tap_position_to_reg_dict(self._initial_tap_position))

        self._solve()
        self._sync_storage_states(time_s=0.0)
        self._cache_node_map()
        self._cache_buses_with_phases()

    def _solve(self) -> None:
        """Run the OpenDSS power flow solver."""
        if self._dss_controls:
            dss.Solution.Solve()
        else:
            dss.Solution.SolveNoControl()

    def _set_storage_dispatch(self, att: _StorageAttachment, time_s: float) -> None:
        """Write a storage dispatch setpoint into OpenDSS."""
        storage = att.storage
        power_kw = float(storage.power_kw(time_s))
        reactive_power_kvar = float(storage.reactive_power_kvar(time_s))

        if abs(power_kw) > storage.rated_power_kw + 1e-9:
            raise ValueError(
                f"Storage {storage.name!r} requested {power_kw:.6g} kW, "
                f"exceeding rating {storage.rated_power_kw:.6g} kW."
            )
        apparent_power = math.hypot(power_kw, reactive_power_kvar)
        if apparent_power > storage.rated_apparent_power_kva + 1e-9:
            raise ValueError(
                f"Storage {storage.name!r} requested {apparent_power:.6g} kVA, "
                f"exceeding rating {storage.rated_apparent_power_kva:.6g} kVA."
            )

        dss.Storages.Name(att.element_name)
        if abs(power_kw) <= 1e-9:
            dss.Storages.State(0)  # IDLING
            power_kw = 0.0
        elif power_kw > 0.0:
            dss.Storages.State(1)  # DISCHARGING
        else:
            dss.Storages.State(-1)  # CHARGING

        dss.Properties.Value("kW", power_kw)
        dss.Properties.Value("kvar", reactive_power_kvar)

    def _storage_attachment_for(self, storage: EnergyStorage) -> _StorageAttachment:
        for att in self._storage_attachments:
            if att.storage is storage:
                return att
        attached = [att.storage.name for att in self._storage_attachments]
        raise ValueError(f"Storage {storage.name!r} is not attached to this grid. Attached: {attached}")

    @staticmethod
    def _require_three_phase_bus(bus: str, storage_name: str) -> None:
        """Raise if *bus* does not expose all three phases on the compiled circuit."""
        dss.Circuit.SetActiveBus(bus)
        nodes = tuple(int(n) for n in dss.Bus.Nodes() if int(n) in _PHASES)
        missing = tuple(_PHASE_NAME[ph] for ph in _PHASES if ph not in nodes)
        if missing:
            raise ValueError(
                f"Storage {storage_name!r} is attached to bus {bus!r}, which is missing "
                f"phase(s) {', '.join(missing)}. Storage requires a three-phase bus."
            )

    def _reset_storage_resources(self) -> None:
        for att in self._storage_attachments:
            att.storage.reset()

    def _finish_storage_timestep(self, time_s: float) -> None:
        """Advance OpenDSS native storage physics and sync Python state."""
        if not self._storage_attachments:
            return

        if not self._dss_controls:
            dss.Solution.Cleanup()
        self._sync_storage_states(time_s=time_s)

    def _sync_storage_states(self, time_s: float) -> None:
        """Read OpenDSS Storage state back into attached storage objects."""
        for att in self._storage_attachments:
            dss.Storages.Name(att.element_name)
            dss.Circuit.SetActiveElement(f"Storage.{att.element_name}")

            soc = float(dss.Storages.puSOC())
            capacity_kwh = float(att.storage.capacity_kwh)
            stored_kwh = capacity_kwh * soc
            dss_state_value = int(dss.Storages.State())
            dss_state = _STORAGE_STATE_NAME.get(dss_state_value, str(dss_state_value))

            powers = dss.CktElement.Powers()
            power_kw = -float(sum(powers[0::2]))
            reactive_power_kvar = -float(sum(powers[1::2]))

            state = StorageState(
                time_s=time_s,
                stored_kwh=stored_kwh,
                soc=soc,
                power_kw=power_kw,
                reactive_power_kvar=reactive_power_kvar,
                dss_state=dss_state,
            )
            att.storage.update_state(state)

    def _cache_buses_with_phases(self) -> None:
        """Populate `all_buses` and `buses_with_phase` from the compiled circuit."""
        self.all_buses = list(dss.Circuit.AllBusNames())
        self.buses_with_phase = {ph: [] for ph in _PHASES}
        for bus, phase in self._node_map:
            if phase in _PHASES:
                self.buses_with_phase[phase].append(bus)

    def _cache_node_map(self) -> None:
        """Cache the mapping from AllBusMagPu indices to (bus, phase) pairs."""
        self._node_map: list[tuple[str, int]] = []
        for name in dss.Circuit.AllNodeNames():
            parts = name.split(".")
            bus = parts[0]
            phase = int(parts[1]) if len(parts) > 1 else 0
            self._node_map.append((bus, phase))

    def _build_vmag_indices(self) -> None:
        """Pre-compute index arrays for fast voltage vector extraction."""
        node_idx = {(bus, ph): i for i, (bus, ph) in enumerate(self._node_map)}
        self._v_index_to_vmag = np.array(
            [node_idx[(bus, ph)] for bus, ph in self._v_index],
            dtype=int,
        )

    def _build_snapshot_indices(self) -> None:
        """Pre-compute index arrays for `_snapshot_bus_voltages`."""
        bus_to_idx = {bus: i for i, bus in enumerate(self.all_buses)}
        n_buses = len(self.all_buses)
        self._snap_indices = np.full((n_buses, 3), -1, dtype=int)
        for vmag_idx, (bus, phase) in enumerate(self._node_map):
            if 1 <= phase <= 3:
                bus_idx = bus_to_idx.get(bus)
                if bus_idx is not None:
                    self._snap_indices[bus_idx, phase - 1] = vmag_idx

    def _snapshot_bus_voltages(self) -> BusVoltages:
        """Snapshot all per-bus, per-phase voltage magnitudes into BusVoltages."""
        vmag = dss.Circuit.AllBusMagPu()
        vmag_ext = np.append(vmag, float("nan"))
        volts = vmag_ext[self._snap_indices]
        data = {
            bus: PhaseVoltages(a=float(volts[i, 0]), b=float(volts[i, 1]), c=float(volts[i, 2]))
            for i, bus in enumerate(self.all_buses)
        }
        return BusVoltages(_data=data)

    def _build_v_index(self) -> list[tuple[str, int]]:
        excl = {b.lower() for b in self._exclude_buses}
        v_index: list[tuple[str, int]] = []
        for ph in _PHASES:
            for b in self.buses_with_phase.get(ph, []):
                if str(b).lower() in excl:
                    continue
                v_index.append((str(b), int(ph)))
        return v_index

    @staticmethod
    def _cache_regcontrol_map() -> dict[str, tuple[str, int]]:
        """Enumerate RegControls and discover their transformer and winding."""
        reg_map: dict[str, tuple[str, int]] = {}
        for rc in dss.RegControls:
            rc_name = rc.Name().lower()
            xf = rc.Transformer()
            w = int(rc.Winding())
            reg_map[rc_name] = (xf, w)
        return reg_map

    @staticmethod
    def _build_phase_to_reg_map(reg_map: dict[str, tuple[str, int]]) -> dict[int, str | None]:
        """Build a mapping from phase (1/2/3) to RegControl name.

        When multiple RegControls share the same phase (multi-bank systems),
        the phase entry is set to None to prevent the `a`/`b`/`c` shorthand
        from silently targeting the wrong regulator.
        """
        phase_to_reg: dict[int, str | None] = {}
        for rc_name, (xf, _wdg) in reg_map.items():
            dss.Transformers.Name(xf)
            bus_names = list(dss.CktElement.BusNames())
            phase = 0
            for bus_str in bus_names:
                parts = str(bus_str).split(".")
                if len(parts) >= 2:
                    try:
                        phase = int(parts[1])
                    except ValueError:
                        continue
                    if phase in (1, 2, 3):
                        break
                    phase = 0

            if phase not in (1, 2, 3):
                logger.debug(
                    "RegControl '%s' (transformer=%s, buses=%s): cannot determine "
                    "phase from bus data; use regulator name in TapPosition.",
                    rc_name,
                    xf,
                    bus_names,
                )
                continue

            if phase in phase_to_reg:
                logger.debug(
                    "Multiple RegControls on phase %s: '%s' and '%s'. "
                    "Phase shorthand (a/b/c) disabled for this phase; use regulator names.",
                    _PHASE_NAME[phase],
                    phase_to_reg[phase],
                    rc_name,
                )
                phase_to_reg[phase] = None
            else:
                phase_to_reg[phase] = rc_name
        return phase_to_reg

    def _tap_position_to_reg_dict(self, pos: TapPosition) -> dict[str, float]:
        """Map tap position to OpenDSS RegControl names."""
        if self._phase_to_reg is None:
            raise RuntimeError("_phase_to_reg not initialized; call start() first")

        d: dict[str, float] = {}
        for reg_name, tap_val in pos.regulators.items():
            key = reg_name.lower()
            phase = _ATTR_TO_PHASE.get(key)
            if phase is not None and phase in self._phase_to_reg:
                rc_name = self._phase_to_reg[phase]
                if rc_name is None:
                    raise ValueError(
                        f"TapPosition uses phase shorthand '{key}' but multiple "
                        f"RegControls exist on phase {_PHASE_NAME[phase]}. "
                        f"Use explicit regulator names instead."
                    )
                d[rc_name] = tap_val
            else:
                d[key] = tap_val
        return d

    def _set_reg_taps(self, tap_map: dict[str, float]) -> None:
        """Write tap ratios to OpenDSS RegControl transformers."""
        if self._reg_map is None:
            self._reg_map = self._cache_regcontrol_map()

        tap_map_lc = {str(k).lower(): float(v) for k, v in tap_map.items()}

        for rc_key, (xfmr, wdg) in self._reg_map.items():
            if rc_key in tap_map_lc:
                tap_pu = tap_map_lc[rc_key]
                dss.Text.Command(f"Edit Transformer.{xfmr} Wdg={wdg} Tap={tap_pu:.6f}")

    def _read_current_taps(self) -> TapPosition:
        """Read current regulator tap positions from OpenDSS."""
        if self._reg_map is None:
            self._reg_map = self._cache_regcontrol_map()

        regulators: dict[str, float] = {}
        for rc_key, (xfmr, wdg) in self._reg_map.items():
            dss.Transformers.Name(xfmr)
            dss.Transformers.Wdg(wdg)
            regulators[rc_key] = float(dss.Transformers.Tap())

        return TapPosition(regulators=regulators)