接入test_mql PNVO阶跃控制与流量诊断

PythonModels/systems/test_mql_pneumatic.py

@@ -18,12 +18,39 @@ DEFAULT_TEST_MQL_TEMPERATURE_K = 293.15
 DEFAULT_VARIABLE_CHAMBER_PRESSURE_BAR = 1.0
 
 
+@dataclass(frozen=True)
+class TestMqlStepSignalSpec:
+    alias: str
+    initial_output: float
+    final_output: float
+    step_time_s: float
+    transition_duration_s: float
+    transition_type: int
+
+    def output_at(self, time_s: float) -> float:
+        if self.transition_type != 1:
+            raise ValueError(
+                f"unsupported STEP0 transition type {self.transition_type} on {self.alias}"
+            )
+        return self.initial_output if time_s < self.step_time_s else self.final_output
+
+
+@dataclass(frozen=True)
+class TestMqlVariableOrificeControl:
+    orifice_alias: str
+    step: TestMqlStepSignalSpec
+
+    def opening_at(self, time_s: float) -> float:
+        return self.step.output_at(time_s)
+
+
 @dataclass(frozen=True)
 class TestMqlPneumaticAssembly:
     fixed_chambers: dict[str, AmesimPneumaticVolume]
     variable_chambers: dict[str, AmesimVariablePneumaticVolume]
     fixed_orifices: dict[str, AmesimPneumaticOrifice]
     variable_orifices: dict[str, AmesimPneumaticOrifice]
+    variable_orifice_controls: dict[str, TestMqlVariableOrificeControl]
     fixed_initial_absolute_pressure_pa: float
     variable_initial_absolute_pressure_pa: float
 
@@ -51,6 +78,14 @@ class TestMqlPneumaticAssembly:
     def component_count(self) -> int:
         return self.chamber_count + self.orifice_count
 
+    @property
+    def variable_orifice_control_count(self) -> int:
+        return len(self.variable_orifice_controls)
+
+    def set_variable_orifice_openings(self, time_s: float) -> None:
+        for alias, control in self.variable_orifice_controls.items():
+            self.variable_orifices[alias].opening = control.opening_at(time_s)
+
     @property
     def aliases(self) -> tuple[str, ...]:
         return tuple(
@@ -101,12 +136,13 @@ def build_test_mql_pneumatic_assembly(
         )
         for component in config.components_by_submodel("PNOR001")
     }
+    variable_orifice_controls = _build_variable_orifice_controls(config)
     variable_orifices = {
         component.alias: _build_orifice(
             component,
             area_parameter="area0",
             gas=gas,
-            opening=1.0,
+            opening=variable_orifice_controls[component.alias].opening_at(0.0),
         )
         for component in config.components_by_submodel("PNVO001")
     }
@@ -115,11 +151,58 @@ def build_test_mql_pneumatic_assembly(
         variable_chambers=variable_chambers,
         fixed_orifices=fixed_orifices,
         variable_orifices=variable_orifices,
+        variable_orifice_controls=variable_orifice_controls,
         fixed_initial_absolute_pressure_pa=fixed_initial_absolute_pressure_pa,
         variable_initial_absolute_pressure_pa=variable_initial_absolute_pressure_pa,
     )
 
 
+def _build_variable_orifice_controls(
+    config: TestMqlConfig,
+) -> dict[str, TestMqlVariableOrificeControl]:
+    from PythonModels.systems.test_mql import CONNECTION_SPECS
+
+    components_by_alias = {component.alias: component for component in config.components}
+    variable_orifice_aliases = {
+        component.alias for component in config.components_by_submodel("PNVO001")
+    }
+    controls: dict[str, TestMqlVariableOrificeControl] = {}
+    for connection in CONNECTION_SPECS:
+        if connection["submodel"] != "DIRECT":
+            continue
+        source_alias = str(connection["source_component"])
+        target_alias = str(connection["target_component"])
+        if target_alias in variable_orifice_aliases:
+            orifice_alias = target_alias
+            step_alias = source_alias
+        elif source_alias in variable_orifice_aliases:
+            orifice_alias = source_alias
+            step_alias = target_alias
+        else:
+            continue
+        step_component = components_by_alias.get(step_alias)
+        if step_component is None or step_component.submodel != "STEP0":
+            continue
+        controls[orifice_alias] = TestMqlVariableOrificeControl(
+            orifice_alias=orifice_alias,
+            step=TestMqlStepSignalSpec(
+                alias=step_alias,
+                initial_output=step_component.parameter_value("out0"),
+                final_output=step_component.parameter_value("out1"),
+                step_time_s=step_component.parameter_value("t0"),
+                transition_duration_s=step_component.parameter_value("td"),
+                transition_type=int(step_component.parameter_value("transitionType")),
+            ),
+        )
+    missing = variable_orifice_aliases - controls.keys()
+    if missing:
+        raise ValueError(
+            "missing STEP0 controls for PNVO001 components: "
+            + ", ".join(sorted(missing))
+        )
+    return controls
+
+
 def absolute_pressure_from_amesim_bar_parameter(pressure_bar: float) -> float:
     return pressure_bar * BAR_TO_PA