chore: add maximum_rate to CompressorModelResult (#285)

test: update tests with maximum rate chore: update snapshot
equinor · Dec 20, 2023 · 2cb09e2 · 2cb09e2
1 parent 28885f2
commit 2cb09e2
Show file tree

Hide file tree

Showing 7 changed files with 45,602 additions and 24 deletions.
diff --git a/src/libecalc/core/graph_result.py b/src/libecalc/core/graph_result.py
@@ -757,6 +757,17 @@ def get_asset_result(self) -> libecalc.dto.result.EcalcModelResult:
                             rate_type=RateType.STREAM_DAY,
                             regularity=regularity.for_timesteps(model.timesteps).values,
                         )
+                        maximum_rate = TimeSeriesRate(
+                            timesteps=model.timesteps,
+                            values=model.max_standard_rate[
+                                0
+                            ]  # WORKAROUND: We only pick max rate for first stream for now
+                            if model.max_standard_rate is not None
+                            else [math.nan] * len(model.timesteps),
+                            unit=Unit.STANDARD_CUBIC_METER_PER_DAY,
+                            rate_type=RateType.STREAM_DAY,
+                            regularity=regularity.for_timesteps(model.timesteps).values,
+                        )
                     else:
                         rate = TimeSeriesRate(
                             timesteps=model.timesteps,
@@ -765,6 +776,15 @@ def get_asset_result(self) -> libecalc.dto.result.EcalcModelResult:
                             rate_type=RateType.STREAM_DAY,
                             regularity=regularity.for_timesteps(model.timesteps).values,
                         )
+                        maximum_rate = TimeSeriesRate(
+                            timesteps=model.timesteps,
+                            values=model.max_standard_rate
+                            if model.max_standard_rate is not None
+                            else [math.nan] * len(model.timesteps),
+                            unit=Unit.STANDARD_CUBIC_METER_PER_DAY,
+                            rate_type=RateType.STREAM_DAY,
+                            regularity=regularity.for_timesteps(model.timesteps).values,
+                        )
 
                     models.extend(
                         [
@@ -780,6 +800,7 @@ def get_asset_result(self) -> libecalc.dto.result.EcalcModelResult:
                                 requested_inlet_pressure=requested_inlet_pressure,
                                 requested_outlet_pressure=requested_outlet_pressure,
                                 rate=rate,
+                                maximum_rate=maximum_rate,
                                 stage_results=model_stage_results,
                                 failure_status=model.failure_status,
                                 timesteps=model.timesteps,

diff --git a/src/libecalc/dto/result/results.py b/src/libecalc/dto/result/results.py
@@ -214,6 +214,7 @@ class CompressorModelResult(ConsumerModelResultBase):
     requested_inlet_pressure: TimeSeriesFloat
     requested_outlet_pressure: TimeSeriesFloat
     rate: TimeSeriesRate
+    maximum_rate: TimeSeriesRate
     stage_results: List[CompressorModelStageResult]
     turbine_result: Optional[TurbineModelResult] = None
     energy_usage_unit: Unit

diff --git a/src/libecalc/examples/advanced/model.yaml b/src/libecalc/examples/advanced/model.yaml
@@ -68,6 +68,7 @@ MODELS:
   - NAME: simplified_compressor_train_model
     TYPE: SIMPLIFIED_VARIABLE_SPEED_COMPRESSOR_TRAIN
     FLUID_MODEL: medium_fluid
+    CALCULATE_MAX_RATE: True
     COMPRESSOR_TRAIN:
       STAGES:
         - INLET_TEMPERATURE: 30