Use dictionary for grid names in titles

MPAS-Dev · Mar 5, 2025 · c3c60a4 · c3c60a4
1 parent ce6b026
commit c3c60a4
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/mpas_analysis/shared/plot/plot_climatology_map_subtask.py b/mpas_analysis/shared/plot/plot_climatology_map_subtask.py
@@ -11,7 +11,6 @@
 
 import xarray as xr
 import numpy as np
-from string import capwords
 
 from mpas_analysis.shared import AnalysisTask
 
@@ -25,6 +24,8 @@
 from mpas_analysis.shared.climatology.comparison_descriptors import \
     get_comparison_descriptor
 
+from mpas_analysis.shared.projection import comparison_grid_titles
+
 
 class PlotClimatologyMapSubtask(AnalysisTask):
     """
@@ -656,8 +657,8 @@ def _plot_projection(self, remappedModelClimatology,
             extend=self.extend)
 
         if self.prependComparisonGrid:
-            upperGridName = capwords(comparisonGridName.replace('_', ' '))
-            galleryGroup = f'{upperGridName} {self.galleryGroup}'
+            gridTitle = comparison_grid_titles[comparisonGridName]
+            galleryGroup = f'{gridTitle} {self.galleryGroup}'
         else:
             galleryGroup = self.galleryGroup
 

diff --git a/mpas_analysis/shared/projection/__init__.py b/mpas_analysis/shared/projection/__init__.py
@@ -29,6 +29,19 @@
     'fris': 'Fris',
 }
 
+comparison_grid_titles = {
+    'latlon': 'Global',
+    'antarctic': 'Antarctic',
+    'arctic': 'Arctic',
+    'antarctic_extended': 'Antarctic',
+    'arctic_extended': 'Arctic',
+    'north_atlantic': 'North Atlantic',
+    'north_pacific': 'North Pacific',
+    'subpolar_north_atlantic': 'Subpolar North Atlantic',
+    'fris': 'Filchner-Ronne',
+}
+
+
 comparison_grid_file_suffixes = {
     'latlon': 'latlon',
     'antarctic': 'Antarctic_stereo',