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

PythonModels/components/orifice.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
 
+from collections.abc import Mapping
 from math import sqrt
 
 from PythonModels.core.base import AlgebraicComponent
+from PythonModels.core.equations import EquationResidual
 from PythonModels.core.ports import PortState
 
 
@@ -13,8 +15,12 @@ class Orifice(AlgebraicComponent):
         super().__init__(name=name)
         self.opening = opening
         self.K = K
-        self.port_a = PortState()
-        self.port_b = PortState()
+        self.port_a = self.register_port(
+            PortState.pneumatic("port_a", nominal_role="inlet")
+        )
+        self.port_b = self.register_port(
+            PortState.pneumatic("port_b", nominal_role="outlet")
+        )
 
     @property
     def K_eff(self) -> float:
@@ -26,3 +32,37 @@ class Orifice(AlgebraicComponent):
             return 0.0
         return self.K_eff * sqrt(abs(dp)) * (1.0 if dp > 0.0 else -1.0)
 
+    def pressure_flow_equation_residuals(self) -> tuple[EquationResidual, ...]:
+        return (
+            EquationResidual(
+                id=f"{self.name}:mass_flow_balance",
+                owner="component",
+                owner_id=self.name,
+                relation="sumToZero",
+                variables=(
+                    f"{self.name}.port_a.m_flow",
+                    f"{self.name}.port_b.m_flow",
+                ),
+                role="flow",
+                value=self.port_a.m_flow + self.port_b.m_flow,
+            ),
+            EquationResidual(
+                id=f"{self.name}:pressure_flow_relation",
+                owner="component",
+                owner_id=self.name,
+                relation="constitutive",
+                variables=(
+                    f"{self.name}.port_a.p",
+                    f"{self.name}.port_b.p",
+                    f"{self.name}.port_a.m_flow",
+                ),
+                role="flow",
+                value=self.port_a.m_flow
+                - self.mass_flow(self.port_a.p, self.port_b.p),
+            ),
+        )
+
+    def update_stream_outflows(self, connected_h: Mapping[str, float]) -> None:
+        self.port_a.h_outflow = connected_h["port_b"]
+        self.port_b.h_outflow = connected_h["port_a"]
+