Include renaming scheme for chemicals for OEM

C2SM · Feb 5, 2024 · c5e5457 · c5e5457
1 parent f4533bb
commit c5e5457
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/cases/icon-art-oem-test/config.yaml b/cases/icon-art-oem-test/config.yaml
@@ -41,6 +41,8 @@ chem:
     nameformat: '%Y%m%d_%H'
     suffix: .grb
     inc: 3
+    chem_in_to_chem_out_remap_tracers:
+        TRCH4_chemtr: CH4_BG
 
 icontools_runjobs:
     - icontools_remap_ic_runjob.cfg

diff --git a/jobs/prepare_art_oem.py b/jobs/prepare_art_oem.py
@@ -74,7 +74,10 @@ def main(cfg):
             ds_chem['PS'] = ds_chem['LNPS']
             ds_chem['PS'].attrs = ds_chem['LNPS'].attrs
             ds_chem['PS'].attrs["long_name"] = 'surface pressure'
-            ds_chem['TRCH4_chemtr'] = ds_chem['CH4_BG']
+            # Remapping chemical tracer names
+            if "chem_in_to_chem_out_remap_tracers" in cfg.chem:
+                for chem_in, chem_out in cfg.chem['chem_in_to_chem_out_remap_tracers'].items():
+                    ds_chem[chem_out] = ds_chem[chem_in]
             # merge:
             ds_merged = xr.merge([ds_meteo, ds_chem], compat="override")
             #ds_merged.attrs = ds.attrs