增加可选性能埋点并完成物性效率评估

app/simulation/components/amesim/media/mediums.py

@@ -12,6 +12,7 @@ from app.simulation.core.medium import (
     ThermodynamicProperties,
 )
 from app.simulation.core.peng_robinson import HELIUM_PR, PengRobinsonFluid
+from app.simulation.performance import profile_property, record_property_iterations
 
 
 @dataclass(frozen=True)
@@ -69,6 +70,7 @@ class AmesimHeliumPengRobinsonMedium(IdealGasMedium):
         del T
         return self.cv
 
+    @profile_property("density")
     def density(self, p: float, T: float) -> float:
         return self.fluid.density(p, T)
 
@@ -134,6 +136,7 @@ class AmesimHeliumPengRobinsonMedium(IdealGasMedium):
         )
         return factor, exponent
 
+    @profile_property("isentropic_density_pressure_factor")
     def isentropic_density_pressure_factor(
         self,
         p: float,
@@ -166,12 +169,14 @@ class AmesimHeliumPengRobinsonMedium(IdealGasMedium):
             raise ValueError("Volume must stay positive.")
         return self.fluid.pressure_from_density(T, m / V)
 
+    @profile_property("specific_internal_energy")
     def specific_internal_energy(self, T: float) -> float:
         return self.R_gas * (
             (self.nasa_cp_over_R - 1.0) * T
             + self.nasa_enthalpy_constant_K
         )
 
+    @profile_property("specific_internal_energy_at_pressure")
     def specific_internal_energy_at_pressure(self, p: float, T: float) -> float:
         density = self.density(p, T)
         return (
@@ -179,12 +184,14 @@ class AmesimHeliumPengRobinsonMedium(IdealGasMedium):
             + self.fluid.residual_specific_internal_energy_at_density(T, density)
         )
 
+    @profile_property("specific_enthalpy")
     def specific_enthalpy(self, T: float) -> float:
         return self.R_gas * (
             self.nasa_cp_over_R * T
             + self.nasa_enthalpy_constant_K
         )
 
+    @profile_property("specific_enthalpy_at_pressure")
     def specific_enthalpy_at_pressure(self, p: float, T: float) -> float:
         return self.specific_enthalpy(T) + self.fluid.residual_specific_enthalpy(p, T)
 
@@ -198,6 +205,7 @@ class AmesimHeliumPengRobinsonMedium(IdealGasMedium):
             h / self.R_gas - self.nasa_enthalpy_constant_K
         ) / self.nasa_cp_over_R
 
+    @profile_property("temperature_from_pressure_enthalpy", track_cache=True)
     @lru_cache(maxsize=8192)
     def temperature_from_pressure_enthalpy(self, p: float, h: float) -> float:
         temperature = max(self.temperature_from_enthalpy(h), 2.2)
@@ -211,8 +219,18 @@ class AmesimHeliumPengRobinsonMedium(IdealGasMedium):
                 temperature,
                 1.0,
             ):
+                record_property_iterations(
+                    "temperature_from_pressure_enthalpy",
+                    _iteration + 1,
+                    True,
+                )
                 return next_temperature
             temperature = next_temperature
+        record_property_iterations(
+            "temperature_from_pressure_enthalpy",
+            16,
+            False,
+        )
         return temperature
 
     def temperature_from_mass_internal_energy(self, m: float, U: float) -> float:
@@ -222,6 +240,7 @@ class AmesimHeliumPengRobinsonMedium(IdealGasMedium):
             )
         return self.temperature_from_internal_energy(U / m)
 
+    @profile_property("properties_from_mU", track_cache=True)
     @lru_cache(maxsize=8192)
     def properties_from_mU(
         self,
@@ -249,6 +268,7 @@ class AmesimHeliumPengRobinsonMedium(IdealGasMedium):
             self.temperature_from_internal_energy(target_internal_energy),
             2.2,
         )
+        converged = False
         for _iteration in range(16):
             residual_internal_energy = (
                 self.fluid.residual_specific_internal_energy_at_density(
@@ -267,8 +287,14 @@ class AmesimHeliumPengRobinsonMedium(IdealGasMedium):
                 1.0,
             ):
                 temperature = next_temperature
+                converged = True
                 break
             temperature = next_temperature
+        record_property_iterations(
+            "properties_from_mU",
+            _iteration + 1,
+            converged,
+        )
         pressure = self.fluid.pressure_from_density(temperature, density)
         return ThermodynamicProperties(
             p=pressure,