接入PNL0001事件窗口RHS诊断

PythonModels/scripts/run_test_mql_full_state_comparison.py

@@ -6,7 +6,7 @@ from datetime import UTC, datetime
 from math import nextafter
 from pathlib import Path
 
-from PythonModels.core.solver import SolveIVPConfig
+from PythonModels.core.solver import SolveIVPConfig, integrate_ode
 from PythonModels.reporting.amesim_results import AmesimResults, load_test_mql_amesim_results
 from PythonModels.reporting.test_mql_comparison import (
     TestMqlComparisonResult,
@@ -22,7 +22,11 @@ from PythonModels.reporting.test_mql_output_validation import (
     compare_validated_test_mql_output,
     validate_test_mql_output,
 )
-from PythonModels.systems.test_mql import TestMqlSimulationResult, TestMqlSystem
+from PythonModels.systems.test_mql import (
+    TestMqlPnl0001LineRhsDiagnostic,
+    TestMqlSimulationResult,
+    TestMqlSystem,
+)
 
 
 DEFAULT_FULL_STATE_COMPARISON_DATA_PATHS = (
@@ -248,6 +252,54 @@ class TestMqlPnvoEventBoundaryDiagnostic:
         )
 
 
+@dataclass(frozen=True)
+class TestMqlPnvoEventWindowSegmentDiagnostic:
+    t_start: float
+    t_stop: float
+    method: str
+    rtol: float
+    atol: float
+    max_step: float
+    rhs_evaluations: int
+    success: bool
+    message: str
+
+
+@dataclass(frozen=True)
+class TestMqlPnvoEventWindowSampleDiagnostic:
+    time_s: float
+    data_paths: tuple[str, ...]
+    python_values_by_data_path: dict[str, float]
+    amesim_values_by_data_path: dict[str, float]
+    pnl0001_rhs_diagnostics: tuple[TestMqlPnl0001LineRhsDiagnostic, ...] = field(
+        default_factory=tuple
+    )
+
+    def abs_error(self, data_path: str) -> float:
+        return abs(
+            self.python_values_by_data_path[data_path]
+            - self.amesim_values_by_data_path[data_path]
+        )
+
+
+@dataclass(frozen=True)
+class TestMqlPnvoEventWindowDiagnostic:
+    orifice_alias: str
+    event_time_s: float
+    final_time_s: float
+    data_paths: tuple[str, ...]
+    segment_diagnostics: tuple[TestMqlPnvoEventWindowSegmentDiagnostic, ...]
+    sample_diagnostics: tuple[TestMqlPnvoEventWindowSampleDiagnostic, ...]
+    python_values_by_data_path: dict[str, float]
+    amesim_values_by_data_path: dict[str, float]
+
+    def abs_error(self, data_path: str) -> float:
+        return abs(
+            self.python_values_by_data_path[data_path]
+            - self.amesim_values_by_data_path[data_path]
+        )
+
+
 @dataclass(frozen=True)
 class TestMqlFullStateComparisonPathConfig:
     archive_path: Path = field(
@@ -457,6 +509,257 @@ def format_test_mql_pnvo_event_boundary_summary(
     return "\n".join(lines) + "\n"
 
 
+def run_test_mql_pnvo_event_window_diagnostic(
+    config: TestMqlFullStateComparisonScriptConfig | None = None,
+    *,
+    orifice_alias: str = "pn_morifice_1",
+) -> TestMqlPnvoEventWindowDiagnostic:
+    config = config or TestMqlFullStateComparisonScriptConfig()
+    system = TestMqlSystem(archive_path=config.paths.archive_path)
+    control = system.pneumatic_assembly.variable_orifice_controls[orifice_alias]
+    event_time_s = control.step.step_time_s
+    final_time_s = 0.05
+    spec = system.discover_pneumatic_branch_topology().chamber_segment_specs[0]
+    closure_kwargs = {
+        "inlet_node_pressure_pa": config.execution.inlet_node_pressure_pa,
+        "resistance_boundary_pressure_pa": (
+            config.execution.resistance_boundary_pressure_pa
+        ),
+        "inlet_node_temperature_k": config.execution.inlet_node_temperature_k,
+        "resistance_boundary_temperature_k": (
+            config.execution.resistance_boundary_temperature_k
+        ),
+    }
+    closure = system.full_state_closure_from_spec(spec, **closure_kwargs)
+    state_vector = closure.initial_state_vector()
+    solver_template = config.execution.solver
+    segments = (
+        SolveIVPConfig(
+            t_start=solver_template.t_start,
+            t_stop=nextafter(event_time_s, 0.0),
+            method=solver_template.method,
+            rtol=solver_template.rtol,
+            atol=solver_template.atol,
+            max_step=solver_template.max_step,
+        ),
+        SolveIVPConfig(
+            t_start=event_time_s,
+            t_stop=0.040001,
+            method="Radau",
+            rtol=solver_template.rtol,
+            atol=solver_template.atol,
+            max_step=1.0e-7,
+            first_step=1.0e-10,
+        ),
+        SolveIVPConfig(
+            t_start=0.040001,
+            t_stop=0.04001,
+            method="Radau",
+            rtol=solver_template.rtol,
+            atol=solver_template.atol,
+            max_step=1.0e-6,
+        ),
+        SolveIVPConfig(
+            t_start=0.04001,
+            t_stop=0.0401,
+            method="Radau",
+            rtol=solver_template.rtol,
+            atol=solver_template.atol,
+            max_step=1.0e-5,
+        ),
+        SolveIVPConfig(
+            t_start=0.0401,
+            t_stop=0.041,
+            method="Radau",
+            rtol=solver_template.rtol,
+            atol=solver_template.atol,
+            max_step=1.0e-5,
+        ),
+        SolveIVPConfig(
+            t_start=0.041,
+            t_stop=0.048,
+            method="BDF",
+            rtol=solver_template.rtol,
+            atol=solver_template.atol,
+            max_step=1.0e-5,
+        ),
+        SolveIVPConfig(
+            t_start=0.048,
+            t_stop=final_time_s,
+            method="BDF",
+            rtol=1.0e-5,
+            atol=1.0e-8,
+            max_step=1.0e-5,
+        ),
+    )
+    data_paths = (
+        "press@pn_c1_8",
+        "temp@pn_c1_8",
+        "vol@pn_c1_8",
+        "mgas1@pn_c1_8",
+        "p2@pneumatic_69",
+        "t2@pneumatic_69",
+        "mgas@pneumatic_69",
+        "re@pneumatic_69",
+        "v@pneumatic_69",
+        "ff@pneumatic_69",
+        "dm1@pneumatic_69",
+        f"xv@{orifice_alias}",
+        f"dm2@{orifice_alias}",
+    )
+    sample_times = (event_time_s, 0.041, 0.042, 0.045, 0.048, final_time_s)
+    state_vector_by_sample_time: dict[float, list[float]] = {}
+    segment_diagnostics: list[TestMqlPnvoEventWindowSegmentDiagnostic] = []
+    for segment in segments:
+        rhs_evaluations = 0
+        segment_sample_times = tuple(
+            sample_time
+            for sample_time in sample_times
+            if segment.t_start <= sample_time <= segment.t_stop
+        )
+        t_eval = tuple(
+            sorted({segment.t_start, segment.t_stop, *segment_sample_times})
+        )
+
+        def rhs(time_s, values):
+            nonlocal rhs_evaluations
+            rhs_evaluations += 1
+            return closure.rhs_at(time_s, values)
+
+        solution = integrate_ode(
+            rhs=rhs,
+            initial_state=state_vector,
+            config=segment,
+            t_eval=list(t_eval),
+        )
+        segment_diagnostics.append(
+            TestMqlPnvoEventWindowSegmentDiagnostic(
+                t_start=segment.t_start,
+                t_stop=segment.t_stop,
+                method=segment.method,
+                rtol=segment.rtol,
+                atol=segment.atol,
+                max_step=segment.max_step,
+                rhs_evaluations=rhs_evaluations,
+                success=bool(solution.success),
+                message=str(solution.message),
+            )
+        )
+        for sample_time in segment_sample_times:
+            if sample_time in solution.t:
+                sample_index = list(solution.t).index(sample_time)
+                state_vector_by_sample_time[sample_time] = [
+                    row[sample_index] for row in solution.y
+                ]
+        state_vector = [row[-1] for row in solution.y]
+        if not solution.success:
+            break
+    amesim_results = load_test_mql_amesim_results(config.paths.archive_path)
+    sample_diagnostics = []
+    for sample_time in sample_times:
+        if sample_time not in state_vector_by_sample_time:
+            continue
+        python_sample_values = closure.data_path_values(
+            time_s=sample_time,
+            state_vector=state_vector_by_sample_time[sample_time],
+            data_paths=data_paths,
+        )
+        amesim_sample_values = {
+            data_path: interpolate_series_value(
+                amesim_results.times,
+                amesim_results.series(data_path),
+                sample_time,
+            )
+            for data_path in data_paths
+        }
+        pnl0001_rhs_diagnostics = (
+            closure.pnl0001_line_rhs_diagnostic(
+                line_alias="pneumatic_69",
+                state_vector=state_vector_by_sample_time[sample_time],
+                time_s=sample_time,
+            ),
+        )
+        sample_diagnostics.append(
+            TestMqlPnvoEventWindowSampleDiagnostic(
+                time_s=sample_time,
+                data_paths=data_paths,
+                python_values_by_data_path=python_sample_values,
+                amesim_values_by_data_path=amesim_sample_values,
+                pnl0001_rhs_diagnostics=pnl0001_rhs_diagnostics,
+            )
+        )
+    python_values = closure.data_path_values(
+        time_s=final_time_s,
+        state_vector=state_vector,
+        data_paths=data_paths,
+    )
+    amesim_values = {
+        data_path: interpolate_series_value(
+            amesim_results.times,
+            amesim_results.series(data_path),
+            final_time_s,
+        )
+        for data_path in data_paths
+    }
+    return TestMqlPnvoEventWindowDiagnostic(
+        orifice_alias=orifice_alias,
+        event_time_s=event_time_s,
+        final_time_s=final_time_s,
+        data_paths=data_paths,
+        segment_diagnostics=tuple(segment_diagnostics),
+        sample_diagnostics=tuple(sample_diagnostics),
+        python_values_by_data_path=python_values,
+        amesim_values_by_data_path=amesim_values,
+    )
+
+
+def format_test_mql_pnvo_event_window_summary(
+    diagnostic: TestMqlPnvoEventWindowDiagnostic,
+) -> str:
+    lines = [
+        "Model: test_mql",
+        f"Mode: PNVO event window diagnostic ({diagnostic.orifice_alias})",
+        f"Event time: {diagnostic.event_time_s}",
+        f"Final time: {diagnostic.final_time_s}",
+        "Segments:",
+    ]
+    for segment in diagnostic.segment_diagnostics:
+        lines.append(
+            f"  - {segment.t_start} -> {segment.t_stop}: "
+            f"method={segment.method}, rtol={segment.rtol}, "
+            f"atol={segment.atol}, max_step={segment.max_step}, "
+            f"rhs={segment.rhs_evaluations}, success={segment.success}"
+        )
+    if diagnostic.sample_diagnostics:
+        lines.append("Sample comparisons:")
+        for sample in diagnostic.sample_diagnostics:
+            lines.append(f"  t={sample.time_s}")
+            for data_path in sample.data_paths:
+                lines.append(
+                    f"    - {data_path}: "
+                    f"python={sample.python_values_by_data_path[data_path]}, "
+                    f"amesim={sample.amesim_values_by_data_path[data_path]}, "
+                    f"abs_error={sample.abs_error(data_path)}"
+                )
+            for rhs in sample.pnl0001_rhs_diagnostics:
+                lines.append(
+                    f"    - rhs@{rhs.line_alias}: "
+                    f"chamber_to_line={rhs.chamber_to_line_flow_kg_s}, "
+                    f"node_to_line={rhs.node_to_line_flow_kg_s}, "
+                    f"dm_dt={rhs.mass_derivative_kg_s}, "
+                    f"dU_dt={rhs.energy_derivative_w}"
+                )
+    lines.append("Final comparison:")
+    for data_path in diagnostic.data_paths:
+        lines.append(
+            f"  - {data_path}: "
+            f"python={diagnostic.python_values_by_data_path[data_path]}, "
+            f"amesim={diagnostic.amesim_values_by_data_path[data_path]}, "
+            f"abs_error={diagnostic.abs_error(data_path)}"
+        )
+    return "\n".join(lines) + "\n"
+
+
 def format_test_mql_full_state_comparison_summary(
     run: TestMqlFullStateComparisonRun,
 ) -> str:
@@ -609,11 +912,20 @@ def main() -> None:
         action="store_true",
         help="compare the STEP0 left-state/right-opening boundary at t=0.04 s",
     )
+    parser.add_argument(
+        "--pnvo-event-window",
+        action="store_true",
+        help="run the segmented PNVO opening window through the t=0.05 s save point",
+    )
     args = parser.parse_args()
     if args.pnvo_event_boundary:
         diagnostic = run_test_mql_pnvo_event_boundary_diagnostic()
         print(format_test_mql_pnvo_event_boundary_summary(diagnostic), end="")
         return
+    if args.pnvo_event_window:
+        diagnostic = run_test_mql_pnvo_event_window_diagnostic()
+        print(format_test_mql_pnvo_event_window_summary(diagnostic), end="")
+        return
 
     run, output_dir = run_test_mql_full_state_comparison()
     print(format_test_mql_full_state_comparison_summary(run), end="")