fix: seed PNVO pipe series pressure

app/simulation/solvers/algebraic.py

@@ -7,7 +7,10 @@ from math import expm1, isfinite, log, sqrt
 import os
 
 from app.simulation.components.amesim.boundary.sources import AmesimPnpl01
-from app.simulation.components.amesim.flow.orifices import AmesimPnor001
+from app.simulation.components.amesim.flow.orifices import (
+    AmesimPnor001,
+    AmesimPnvo001FixedOpening,
+)
 from app.simulation.components.amesim.flow.pipes import (
     AmesimPnl00r,
     AmesimPnl0001,
@@ -179,9 +182,11 @@ class AlgebraicEquationSubset:
 
 
 @dataclass(frozen=True)
-class PnorPnl0001SeriesBinding:
-    orifice: AmesimPnor001
-    orifice_port: str
+class ResistancePnl0001SeriesBinding:
+    resistance: AmesimPnor001 | AmesimPnvo001FixedOpening
+    resistance_port: str
+    resistance_other_port: str
+    positive_flow_port: str
     pipe: AmesimPnl0001
     pipe_port: str
 
@@ -315,8 +320,8 @@ class PressureFlowSolver:
             variable: self._build_effort_equality_groups(variable)
             for variable in ("p", "x", "v")
         }
-        self._pnor_pnl0001_series_plan = (
-            self._build_pnor_pnl0001_series_plan()
+        self._resistance_pnl0001_series_plan = (
+            self._build_resistance_pnl0001_series_plan()
         )
         self._closed_resistance_pressure_plan = (
             self._build_closed_resistance_pressure_plan()
@@ -498,7 +503,7 @@ class PressureFlowSolver:
             return failed("activeSetCausalizationRequired")
         if self._closed_resistance_pressure_plan:
             return failed("specialClosedResistancePressureSeed")
-        if self._pnor_pnl0001_series_plan:
+        if self._resistance_pnl0001_series_plan:
             return failed("specialSeriesPressureSeed")
         if len(self._unknowns_by_id) != len(self.unknowns):
             return failed("duplicateAlgebraicUnknown")
@@ -1895,68 +1900,84 @@ class PressureFlowSolver:
             component.get_port(binding.port_name).p = pressure
             binding.neighbor.get_port(binding.neighbor_port).p = pressure
 
-    def _build_pnor_pnl0001_series_plan(
+    def _build_resistance_pnl0001_series_plan(
         self,
-    ) -> tuple[PnorPnl0001SeriesBinding, ...]:
-        bindings: list[PnorPnl0001SeriesBinding] = []
+    ) -> tuple[ResistancePnl0001SeriesBinding, ...]:
+        bindings: list[ResistancePnl0001SeriesBinding] = []
+        resistance_types = (AmesimPnor001, AmesimPnvo001FixedOpening)
         for connection in self.network.connections:
             first_endpoint, second_endpoint = connection.endpoints
             first = self.network.components[first_endpoint.component]
             second = self.network.components[second_endpoint.component]
-            if isinstance(first, AmesimPnor001) and isinstance(second, AmesimPnl0001):
-                orifice, orifice_port = first, first_endpoint.port
+            if isinstance(first, resistance_types) and isinstance(
+                second, AmesimPnl0001
+            ):
+                resistance, resistance_port = first, first_endpoint.port
                 pipe, pipe_port = second, second_endpoint.port
-            elif isinstance(second, AmesimPnor001) and isinstance(first, AmesimPnl0001):
-                orifice, orifice_port = second, second_endpoint.port
+            elif isinstance(second, resistance_types) and isinstance(
+                first, AmesimPnl0001
+            ):
+                resistance, resistance_port = second, second_endpoint.port
                 pipe, pipe_port = first, first_endpoint.port
             else:
                 continue
             if isinstance(pipe, AmesimPnl0002) or pipe_port != "port_1":
                 continue
+            if isinstance(resistance, AmesimPnor001):
+                positive_flow_port, negative_flow_port = "port_1", "port_2"
+            else:
+                positive_flow_port, negative_flow_port = "port_2", "port_3"
+            if resistance_port == positive_flow_port:
+                resistance_other_port = negative_flow_port
+            elif resistance_port == negative_flow_port:
+                resistance_other_port = positive_flow_port
+            else:
+                continue
             bindings.append(
-                PnorPnl0001SeriesBinding(
-                    orifice=orifice,
-                    orifice_port=orifice_port,
+                ResistancePnl0001SeriesBinding(
+                    resistance=resistance,
+                    resistance_port=resistance_port,
+                    resistance_other_port=resistance_other_port,
+                    positive_flow_port=positive_flow_port,
                     pipe=pipe,
                     pipe_port=pipe_port,
                 )
             )
         return tuple(bindings)
 
-    def _seed_pnor_pnl0001_series_pressures(self) -> None:
-        """Causalize the pressure between a PNOR001 and PNL0001 R port."""
+    def _seed_resistance_pnl0001_series_pressures(self) -> None:
+        """Causalize pressure between an orifice/valve and a PNL0001 R port."""
 
         from scipy.optimize import brentq
 
-        for binding in self._pnor_pnl0001_series_plan:
-            orifice = binding.orifice
-            orifice_port = binding.orifice_port
+        for binding in self._resistance_pnl0001_series_plan:
+            resistance = binding.resistance
+            resistance_port = binding.resistance_port
             pipe = binding.pipe
             pipe_port = binding.pipe_port
 
-            orifice_other = "port_2" if orifice_port == "port_1" else "port_1"
-            pressure_a = orifice.get_port(orifice_other).p
+            pressure_a = resistance.get_port(binding.resistance_other_port).p
             pressure_b = pipe.properties().p
             lower = min(pressure_a, pressure_b)
             upper = max(pressure_a, pressure_b)
 
             def mismatch(intermediate_pressure: float) -> float:
-                if orifice_port == "port_2":
-                    orifice_flow_into_connection = -orifice.mass_flow(
-                        pressure_a,
+                if resistance_port == binding.positive_flow_port:
+                    resistance_flow_into_connection = resistance.mass_flow(
                         intermediate_pressure,
+                        pressure_a,
                     )
                 else:
-                    orifice_flow_into_connection = orifice.mass_flow(
-                        intermediate_pressure,
+                    resistance_flow_into_connection = -resistance.mass_flow(
                         pressure_a,
+                        intermediate_pressure,
                     )
                 pipe_flow_into_connection = pipe.mass_flow(
                     intermediate_pressure,
                     pressure_b,
                     pipe.properties().T,
                 )
-                return orifice_flow_into_connection + pipe_flow_into_connection
+                return resistance_flow_into_connection + pipe_flow_into_connection
 
             lower_value = mismatch(lower)
             upper_value = mismatch(upper)
@@ -1977,7 +1998,7 @@ class PressureFlowSolver:
                         maxiter=32,
                     )
                 )
-            orifice.get_port(orifice_port).p = pressure
+            resistance.get_port(resistance_port).p = pressure
             pipe.get_port(pipe_port).p = pressure
 
     def _equation_scale_plan(self, equation) -> EquationScalePlan:
@@ -2083,7 +2104,7 @@ class PressureFlowSolver:
         else:
             self._seed_equal_efforts(effort_variables)
         self._seed_closed_resistance_pressures()
-        self._seed_pnor_pnl0001_series_pressures()
+        self._seed_resistance_pnl0001_series_pressures()
         seeded_flow_ids = self._solve_explicit_flow_unknowns()
         contact_bindings = self._seed_unilateral_contacts()
         if contact_bindings: