36 lines
957 B
Python
36 lines
957 B
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Literal
|
|
|
|
from app.simulation.core.ports import VariableRole
|
|
|
|
|
|
EquationOwner = Literal["connection", "component"]
|
|
EquationRelation = Literal["equal", "sumToZero", "constitutive", "state"]
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class EquationDefinition:
|
|
"""One declarative equation in the compiled model interface."""
|
|
|
|
id: str
|
|
owner: EquationOwner
|
|
owner_id: str
|
|
relation: EquationRelation
|
|
variables: tuple[str, ...]
|
|
role: VariableRole | None = None
|
|
|
|
def as_definition_dict(self) -> dict[str, object]:
|
|
return {
|
|
"id": self.id,
|
|
"owner": self.owner,
|
|
"ownerId": self.owner_id,
|
|
"relation": self.relation,
|
|
"variables": list(self.variables),
|
|
"role": self.role,
|
|
}
|
|
|
|
def as_interface_dict(self) -> dict[str, object]:
|
|
return self.as_definition_dict()
|