实现test_mql PNL0001管路动态

PythonModels/systems/test_mql.py

@@ -3859,6 +3859,7 @@ class TestMqlSystem:
     def __init__(self, archive_path: Path | None = None) -> None:
         self.archive_path = archive_path or Path(__file__).resolve().parents[2] / AMESIM_ARCHIVE_RELATIVE_PATH
         self.network = SimulationNetwork(name=MODEL_NAME)
+        self.pnl0001_assembly = self._build_pnl0001_assembly()
         self.pneumatic_assembly = self._build_pneumatic_assembly()
         pneumatic_components = self._pneumatic_components_by_alias()
         for spec in COMPONENT_SPECS:
@@ -3881,6 +3882,13 @@ class TestMqlSystem:
 
         return build_test_mql_pneumatic_assembly()
 
+    def _build_pnl0001_assembly(self):
+        from PythonModels.systems.test_mql_pneumatic_lines import (
+            build_test_mql_pnl0001_assembly,
+        )
+
+        return build_test_mql_pnl0001_assembly(self.archive_path)
+
     def _pneumatic_components_by_alias(self) -> dict[str, Component]:
         return {
             **self.pneumatic_assembly.fixed_chambers,
@@ -3893,6 +3901,10 @@ class TestMqlSystem:
     def typed_pneumatic_component_count(self) -> int:
         return len(self._pneumatic_components_by_alias())
 
+    @property
+    def typed_pnl0001_line_count(self) -> int:
+        return len(self.pnl0001_assembly.lines)
+
     def pneumatic_state_vector(self) -> list[float]:
         return self.network.initial_state_vector()
 
@@ -4071,6 +4083,84 @@ class TestMqlSystem:
             t_eval=t_eval,
         )
 
+    def pnl0001_chamber_segment_closure_from_spec(
+        self,
+        spec,
+        *,
+        inlet_node_pressure_pa: float,
+        outlet_pressure_pa: float,
+        inlet_node_temperature_k: float = 293.15,
+        outlet_temperature_k: float = 293.15,
+    ):
+        """Insert the topology-derived inlet PNL0001 into a chamber segment."""
+        from PythonModels.components.amesim_pneumatic import (
+            AmesimPneumaticOrifice,
+            AmesimPneumaticVolume,
+        )
+        from PythonModels.systems.test_mql_closure import (
+            TestMqlPneumaticBoundaryCondition,
+            TestMqlPnl0001ChamberSegmentClosure,
+            TestMqlPnl0001ChamberSegmentComponents,
+        )
+
+        inlet_line = self.pnl0001_assembly.lines[spec.inlet_line_alias]
+        volume = self.network.components[spec.volume_alias]
+        inlet_orifice = self.network.components[spec.inlet_orifice_alias]
+        outlet_orifice = self.network.components[spec.outlet_orifice_alias]
+        if not isinstance(volume, AmesimPneumaticVolume):
+            raise TypeError(f"{spec.volume_alias} is not an AMESim pneumatic volume")
+        if not isinstance(inlet_orifice, AmesimPneumaticOrifice):
+            raise TypeError(
+                f"{spec.inlet_orifice_alias} is not an AMESim pneumatic orifice"
+            )
+        if not isinstance(outlet_orifice, AmesimPneumaticOrifice):
+            raise TypeError(
+                f"{spec.outlet_orifice_alias} is not an AMESim pneumatic orifice"
+            )
+        return TestMqlPnl0001ChamberSegmentClosure(
+            components=TestMqlPnl0001ChamberSegmentComponents(
+                inlet_line=inlet_line,
+                volume=volume,
+                inlet_orifice=inlet_orifice,
+                outlet_orifice=outlet_orifice,
+                spec=spec,
+            ),
+            inlet_node=TestMqlPneumaticBoundaryCondition(
+                pressure_pa=inlet_node_pressure_pa,
+                temperature_k=inlet_node_temperature_k,
+            ),
+            outlet_boundary=TestMqlPneumaticBoundaryCondition(
+                pressure_pa=outlet_pressure_pa,
+                temperature_k=outlet_temperature_k,
+            ),
+        )
+
+    def simulate_pnl0001_chamber_segment_from_spec(
+        self,
+        spec,
+        *,
+        inlet_node_pressure_pa: float,
+        outlet_pressure_pa: float,
+        inlet_node_temperature_k: float = 293.15,
+        outlet_temperature_k: float = 293.15,
+        config: SolveIVPConfig | None = None,
+        t_eval: list[float] | None = None,
+    ):
+        closure = self.pnl0001_chamber_segment_closure_from_spec(
+            spec,
+            inlet_node_pressure_pa=inlet_node_pressure_pa,
+            outlet_pressure_pa=outlet_pressure_pa,
+            inlet_node_temperature_k=inlet_node_temperature_k,
+            outlet_temperature_k=outlet_temperature_k,
+        )
+        run_config = config or SolveIVPConfig(t_stop=1.0e-4, max_step=1.0e-5)
+        return integrate_ode(
+            rhs=lambda t, state: closure.rhs(state),
+            initial_state=closure.initial_state_vector(),
+            config=run_config,
+            t_eval=t_eval,
+        )
+
     def pneumatic_branch_closure_from_spec(self, spec):
         return self.pneumatic_branch_closure(
             name=spec.name,