完善 XML 通用仿真与结果查看

PythonModels/core/base.py

@@ -1,11 +1,55 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
+from collections.abc import Mapping
+from typing import Any
+
+from PythonModels.core.equations import EquationResidual
+from PythonModels.core.ports import PortDefinition, PortState
 
 
 class Component(ABC):
     def __init__(self, name: str) -> None:
         self.name = name
+        self.model_type = self.__class__.__name__.lower()
+        self._ports: dict[str, PortState] = {}
+
+    @property
+    def ports(self) -> dict[str, PortState]:
+        return dict(self._ports)
+
+    @property
+    def port_definitions(self) -> tuple[PortDefinition, ...]:
+        return tuple(
+            port.definition
+            for port in self._ports.values()
+            if port.definition is not None
+        )
+
+    def register_port(self, port: PortState) -> PortState:
+        definition = port.definition
+        if definition is None:
+            raise ValueError(f"Component {self.name} cannot register an undefined port.")
+        if definition.name in self._ports:
+            raise ValueError(f"Duplicate port {self.name}.{definition.name}.")
+        self._ports[definition.name] = port
+        return port
+
+    def get_port(self, name: str) -> PortState:
+        try:
+            return self._ports[name]
+        except KeyError as exc:
+            raise ValueError(f"Component {self.name} has no port named {name}.") from exc
+
+    def pressure_flow_equation_residuals(self) -> tuple[EquationResidual, ...]:
+        """Return algebraic residuals after the network assigns port states."""
+
+        return ()
+
+    def update_stream_outflows(self, connected_h: Mapping[str, float]) -> None:
+        """Update connector outflow properties from current flow directions."""
+
+        return None
 
 
 class DynamicComponent(Component):
@@ -43,6 +87,15 @@ class DynamicComponent(Component):
     def set_state_vector(self, values: list[float]) -> None:
         raise NotImplementedError
 
+    def refresh_thermodynamic_ports(self) -> Any:
+        raise NotImplementedError
+
+    def state_derivative_from_ports(
+        self,
+        connected_h: Mapping[str, float],
+    ) -> list[float]:
+        raise NotImplementedError
+
 
 class AlgebraicComponent(Component):
     """Stateless element described by algebraic constraints only."""