30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Dict, Any
|
|
|
|
from citygrid import LocalProblem, SharedVariables
|
|
|
|
|
|
class EnergiBridge:
|
|
"""Lightweight bridge translating between adapters and the canonical IR.
|
|
|
|
This is intentionally small: it provides two helpers to map between a
|
|
simplistic adapter payload (local problem) and the canonical LocalProblem/SharedVariables
|
|
structures used by the solver.
|
|
"""
|
|
|
|
@staticmethod
|
|
def to_canonical(local_problem: Dict[str, Any]) -> LocalProblem:
|
|
return LocalProblem(
|
|
id=local_problem.get("id", "lp-unknown"),
|
|
domain=local_problem.get("domain", "unknown"),
|
|
assets=local_problem.get("assets", []),
|
|
objective=local_problem.get("objective", {}),
|
|
constraints=local_problem.get("constraints", {}),
|
|
solver_hint=local_problem.get("solver_hint"),
|
|
)
|
|
|
|
@staticmethod
|
|
def from_canonical(shared: SharedVariables) -> Dict[str, Any]:
|
|
return {"version": shared.version, "signals": shared.signals}
|