Tweak graph title/legend labels. mark WIP/TODO to add GEOPHIRES profile data to graph

Author: softwareengineerprogrammer

src/geophires_docs/__init__.py

@@ -1,10 +1,15 @@
 from __future__ import annotations
 
+import json
 import os
 from pathlib import Path
 from typing import Any
 
+from pint.facets.plain import PlainQuantity
+
 from geophires_x_client import GeophiresInputParameters
+from geophires_x_client import GeophiresXClient
+from geophires_x_client import GeophiresXResult
 
 
 def _get_file_path(file_name) -> Path:
@@ -75,3 +80,20 @@ def _get_input_parameters_dict(  # TODO consolidate with FervoProjectCape5TestCa
         # TODO preserve newlines
 
     return ret
+
+
+def _get_full_profile(input_and_result: tuple[GeophiresInputParameters, GeophiresXResult], profile_key: str):
+    input_params: GeophiresInputParameters = input_and_result[0]
+    result = GeophiresXClient().get_geophires_result(input_params)
+
+    with open(result.json_output_file_path, encoding='utf-8') as f:
+        full_result_obj = json.load(f)
+
+    net_gen_obj = full_result_obj[profile_key]
+    net_gen_obj_unit = net_gen_obj['CurrentUnits'].replace('CELSIUS', 'degC')
+    profile = [PlainQuantity(it, net_gen_obj_unit) for it in net_gen_obj['value']]
+    return profile
+
+
+def _get_full_production_temperature_profile(input_and_result: tuple[GeophiresInputParameters, GeophiresXResult]):
+    return _get_full_profile(input_and_result, 'Produced Temperature')

src/geophires_docs/generate_fervo_project_cape_5_graphs.py

@@ -1,21 +1,20 @@
 from __future__ import annotations
 
-import json
 from math import ceil
 from pathlib import Path
 from typing import Any
 
 import numpy as np
 from matplotlib import pyplot as plt
-from pint.facets.plain import PlainQuantity
 
 from geophires_docs import _FPC5_INPUT_FILE_PATH
 from geophires_docs import _FPC5_RESULT_FILE_PATH
 from geophires_docs import _PROJECT_ROOT
+from geophires_docs import _get_full_production_temperature_profile
+from geophires_docs import _get_full_profile
 from geophires_docs import _get_input_parameters_dict
 from geophires_docs import _get_logger
 from geophires_x_client import GeophiresInputParameters
-from geophires_x_client import GeophiresXClient
 from geophires_x_client import GeophiresXResult
 from geophires_x_client import ImmutableGeophiresInputParameters
 
@@ -32,27 +31,10 @@ def _get_full_total_electricity_generation_profile(input_and_result: tuple[Geoph
     return _get_full_profile(input_and_result, 'Total Electricity Production')
 
 
-def _get_full_production_temperature_profile(input_and_result: tuple[GeophiresInputParameters, GeophiresXResult]):
-    return _get_full_profile(input_and_result, 'Produced Temperature')
-
-
 def _get_full_thermal_drawdown_profile(input_and_result: tuple[GeophiresInputParameters, GeophiresXResult]):
     return _get_full_profile(input_and_result, 'Thermal Drawdown')
 
 
-def _get_full_profile(input_and_result: tuple[GeophiresInputParameters, GeophiresXResult], profile_key: str):
-    input_params: GeophiresInputParameters = input_and_result[0]
-    result = GeophiresXClient().get_geophires_result(input_params)
-
-    with open(result.json_output_file_path, encoding='utf-8') as f:
-        full_result_obj = json.load(f)
-
-    net_gen_obj = full_result_obj[profile_key]
-    net_gen_obj_unit = net_gen_obj['CurrentUnits'].replace('CELSIUS', 'degC')
-    profile = [PlainQuantity(it, net_gen_obj_unit) for it in net_gen_obj['value']]
-    return profile
-
-
 def _get_redrilling_event_indexes(
     input_and_result: tuple[GeophiresInputParameters, GeophiresXResult], threshold_degc: float = 0.5
 ) -> list[float]:

src/geophires_docs/generate_fervo_project_red_2026_docs.py

@@ -227,12 +227,13 @@ def _get_steady_state_mask(df_prod: pd.DataFrame, steady_state_start_years: floa
     return is_steady
 
 
-def _regenerate_graph_from_csv(
+def _generate_production_temperature_graph_from_fervo_graph_data_csv_and_project_red_geophires_result_data(
     production_csv_path: Path,
     model_csv_path: Path,
     steady_state_csv_path: Path,
     output_path: Path,
     steady_state_start_years: float = _STEADY_STATE_START_YEARS,
+    # TODO/WIP pass GEOPHIRES production profile data
 ) -> None:
     df_prod = pd.read_csv(production_csv_path)
     df_model = pd.read_csv(model_csv_path)
@@ -253,7 +254,10 @@ def _regenerate_graph_from_csv(
         s=22,
         linewidths=1.0,
         alpha=0.85,
-        label=f'Measured flowing temperature (Included), n={len(df_included)}',
+        label='Measured '
+        # f'Flowing '
+        'Temperature (Thermal Conditioning and Steady State)',
+        # f', n={len(df_included)}',
     )
 
     if not df_excluded.empty:
@@ -265,7 +269,10 @@ def _regenerate_graph_from_csv(
             s=22,
             linewidths=1.0,
             alpha=0.5,
-            label=f'Measured flowing temperature (Excluded), n={len(df_excluded)}',
+            label='Measured '
+            # f'Flowing '
+            'Temperature (Excluded Operational Periods)',
+            # f', n={len(df_excluded)}',
         )
 
     ax.plot(
@@ -274,12 +281,18 @@ def _regenerate_graph_from_csv(
         color='black',
         linestyle='--',
         linewidth=1.5,
-        label='Modeled output',
+        label='Fervo-Modeled Temperature',
     )
 
     ax.set_xlabel('Time (Years)', fontsize=12)
-    ax.set_ylabel('Flowing Temperature (°C)', fontsize=12)
-    ax.set_title('Fervo Project Red: Measured vs. Modeled Flowing Temperature (Regenerated)', fontsize=13)
+    ax.set_ylabel('Temperature (°C)', fontsize=12)
+    ax.set_title(
+        'Project Red Temperature: Measured vs. Fervo-Modeled'
+        # ' vs. GEOPHIRES-Modeled'  # WIP/TODO
+        '',
+        fontsize=13,
+    )
+
     ax.set_xlim(0.0, 2.0)
     ax.set_ylim(0.0, 200.0)
     ax.grid(True, linestyle='--', alpha=0.5)
@@ -330,5 +343,7 @@ def _regenerate_graph_from_csv(
         df_steady_state.to_csv(steady_state_csv_path, index=False)
         _log.info(f'Wrote variance analysis CSV:  {steady_state_csv_path}')
 
-    _regenerate_graph_from_csv(production_csv_path, model_csv_path, steady_state_csv_path, regenerated_graph_path)
+    _generate_production_temperature_graph_from_fervo_graph_data_csv_and_project_red_geophires_result_data(
+        production_csv_path, model_csv_path, steady_state_csv_path, regenerated_graph_path
+    )
     _log.info(f'Wrote regenerated graph:      {regenerated_graph_path}')