New Feature: Support plotting native grid data in lat_lon plots

auxiliary_tools/LCF_mixed-phase_partition_diagram.ipynb

@@ -220,7 +220,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:e3sm_unified_1.6.0_nompi] *",
+   "display_name": "Python [conda env:e3sm_unified_1.6.0_nompi]",
    "language": "python",
    "name": "conda-env-e3sm_unified_1.6.0_nompi-py"
   },

auxiliary_tools/plot_native_grid_data.py

@@ -0,0 +1,35 @@
+import uxarray as ux
+import matplotlib.pyplot as pl
+import cartopy.crs as ccrs
+import cartopy.feature as cfeature
+
+base_path = "/Users/zhang40/Documents/ACME_simulations/E3SM_v2/native_grid_data/"
+grid_info = "ne30pg2"
+grid_path = base_path + f"{grid_info}.nc"
+data_path = base_path + f"PRECC.{grid_info}.nc"
+
+uxds = ux.open_dataset(grid_path, data_path)
+
+pc = uxds["PRECT"].to_polycollection()
+#pc = uxds["PRECT"].to_polycollection(periodic_elements="split")
+
+# disables grid lines
+pc.set_antialiased(False)
+
+pc.set_cmap("plasma")
+
+fig, ax = plt.subplots(
+    1,
+    1,
+    figsize=(10, 5),
+    facecolor="w",
+    constrained_layout=True,
+    subplot_kw=dict(projection=ccrs.PlateCarree()),
+)
+
+ax.add_feature(cfeature.COASTLINE)
+ax.add_feature(cfeature.BORDERS)
+
+ax.add_collection(pc)
+ax.set_global()
+plt.title("PolyCollection Plot with Projection & Features")

conda-env/ci.yml

@@ -24,6 +24,7 @@ dependencies:
   - numpy >=2.0.0,<3.0.0
   - pywavelets
   - scipy
+  - uxarray >=2023.3.0
   - xarray >=2024.3.0
   - xcdat >=0.9.1,<1.0.0
   - xesmf >=0.8.7

conda-env/dev.yml

@@ -22,6 +22,7 @@ dependencies:
   - numpy >=2.0.0,<3.0.0
   - pywavelets
   - scipy
+  - uxarray >=2023.3.0,<2025.6.0
   - xcdat >=0.9.1,<1.0.0
   - xesmf >=0.8.7
   - xskillscore >=0.0.20

e3sm_diags/derivations/default_regions_xr.py

@@ -18,6 +18,8 @@
     "20N50N": {"lat": (20.0, 50)},
     "50N90N": {"lat": (50.0, 90)},
     "60S90N": {"lat": (-60.0, 90)},
+    "45S45N-120E60W": {"lat": (-45.0, 45), "lon": (120, 300)},
+    "30S30N-150E90W": {"lat": (-30.0, 30), "lon": (150, 270)},
     "60S60N": {"lat": (-60.0, 60)},
     "75S75N": {"lat": (-75.0, 75)},
     "ocean": {"value": 0.65},

e3sm_diags/derivations/derivations.py

@@ -624,6 +624,18 @@
         ("hfss",): rename,
         ("surf_sens_flux",): rename,  # EAMxx
     },
+    "TGCLDLWP": OrderedDict(
+        [
+            (
+                ("TGCLDLWP",),
+                lambda x: convert_units(x, target_units="g/m^2"),
+            ),
+            (
+                ("LiqWaterPath",),
+                lambda x: convert_units(x, target_units="g/m^2"),
+            ),  # EAMxx
+        ]
+    ),
     "TGCLDLWP_OCN": OrderedDict(
         [
             (

e3sm_diags/driver/default_diags/lat_lon_native_model_vs_model.cfg

@@ -0,0 +1,119 @@
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["PRECT"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+regions = ["global", "60S60N", "30S30N-150E90W"]
+test_colormap = "WhiteBlueGreenYellowRed.rgb"
+reference_colormap = "WhiteBlueGreenYellowRed.rgb"
+diff_colormap = "BrBG"
+contour_levels = [0.5, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 13, 14, 15, 16]
+diff_levels = [-2.5, -2, -1.5, -1, -0.5, -0.25, 0.25, 0.5, 1, 1.5, 2, 2.5]
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["PRECC"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+regions = ["global", "60S60N", "30S30N-150E90W"]
+test_colormap = "WhiteBlueGreenYellowRed.rgb"
+reference_colormap = "WhiteBlueGreenYellowRed.rgb"
+diff_colormap = "BrBG"
+contour_levels = [0.5, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 13, 14, 15, 16]
+diff_levels = [-2.5, -2, -1.5, -1, -0.5, -0.25, 0.25, 0.5, 1, 1.5, 2, 2.5]
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["TGCLDLWP"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+regions = ["global", "60S60N", "30S30N-150E90W"]
+#test_colormap = "Blues"
+#reference_colormap = "Blues"
+diff_colormap = "RdBu"
+#contour_levels = [10, 25, 50, 75, 100, 125, 150, 175, 200,225, 250]
+diff_levels = [-35, -30, -25, -20, -15, -10, -5, 5, 10, 15, 20, 25, 30, 35]
+
+
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["TREFHT"]
+regions = ["global"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [-35, -30, -25, -20, -15, -10, -5, 0, 5, 10, 15, 20, 25, 30, 35, 40]
+diff_levels = [-5, -4, -3, -2, -1, -0.5, -0.2, 0.2, 0.5, 1, 2, 3, 4, 5]
+
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["SWCFSRF"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [-170, -150, -135, -120, -105, -90, -75, -60, -45, -30, -15, 0, 15, 30, 45]
+diff_levels = [-30, -25, -20, -15, -10, -5, -2, 2, 5, 10, 15, 20, 25, 30]
+
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["LWCFSRF"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [0, 10, 20, 30, 40, 50, 60, 70, 80]
+diff_levels = [-30, -25, -20, -15, -10, -5, -2, 2, 5, 10, 15, 20, 25, 30]
+
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["LHFLX"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [0,5, 15, 30, 60, 90, 120, 150, 180, 210, 240, 270, 300]
+diff_levels = [-75, -50, -25, -10, -5, -2, 2, 5, 10, 25, 50, 75]
+
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["SHFLX"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [-100, -75, -50, -25, -10, 0, 10, 25, 50, 75, 100, 125, 150]
+diff_levels = [-75, -50, -25, -10, -5, -2, 2, 5, 10, 25, 50, 75]
+
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["NET_FLUX_SRF"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [-200, -160, -120, -80, -40, 0, 40, 80, 120, 160, 200]
+diff_levels = [-75, -50, -25, -10, -5, -2, 2, 5, 10, 25, 50, 75]
+
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["TMQ"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60]
+diff_levels = [-12, -9, -6, -4, -3, -2, -1, 1, 2, 3, 4, 6, 9, 12]
+
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["QREFHT"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [0.2, 0.5, 1, 2.5, 5, 7.5, 10, 12.5, 15, 17.5]
+diff_levels = [-5, -4, -3, -2, -1, -0.25, 0.25, 1, 2, 3, 4, 5]
+
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_model"
+variables = ["U10"]
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+test_colormap = "PiYG_r"
+reference_colormap = "PiYG_r"
+contour_levels = [2, 4, 6, 8, 10, 12, 14, 16, 18, 20]
+diff_levels = [-8, -6, -5, -4, -3, -2, -1, 1, 2, 3, 4, 5, 6, 8]

e3sm_diags/driver/default_diags/lat_lon_native_model_vs_obs.cfg

@@ -0,0 +1,16 @@
+[#]
+sets = ["lat_lon_native"]
+case_id = "model_vs_obs"
+variables = ["PRECT"]
+seasons = ["ANN", "DJF", "MAM", "JJA", "SON"]
+regions = ["global"]
+test_colormap = "WhiteBlueGreenYellowRed.rgb"
+reference_colormap = "WhiteBlueGreenYellowRed.rgb"
+diff_colormap = "BrBG"
+contour_levels = [0.5, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 13, 14, 15, 16]
+diff_levels = [-2.5, -2, -1.5, -1, -0.5, -0.25, 0.25, 0.5, 1, 1.5, 2, 2.5]
+
+# Native grid settings
+grid_file = ""
+split_periodic_elements = True
+antialiased = False