Add ESGF links and more simulations to v1 data

.gitignore

@@ -116,6 +116,7 @@ conda-build/
 
 *~
 utils/out.txt
+utils/out_*.txt
 utils/simulation_table.txt
 utils/reproduction_table.txt
 utils/test_reproduction_scripts.o*

utils/generate_html.bash

@@ -1,11 +1,12 @@
-pr_num=59
+pr_num=60
+try_num=9
 
 # Chrysalis
 #destination_dir=/lcrc/group/e3sm/public_html/diagnostic_output/$USER/data_docs_${pr_num}
 #web_page="https://web.lcrc.anl.gov/public/e3sm/diagnostic_output/$USER/data_docs_${pr_num}/html/"
 # Perlmutter
-destination_dir=/global/cfs/cdirs/e3sm/www/$USER/data_docs_${pr_num}
-web_page="https://portal.nersc.gov/cfs/e3sm/$USER/data_docs_${pr_num}/html/"
+destination_dir=/global/cfs/cdirs/e3sm/www/$USER/data_docs_${pr_num}_try${try_num}
+web_page="https://portal.nersc.gov/cfs/e3sm/$USER/data_docs_${pr_num}_try${try_num}/html/"
 
 python generate_tables.py
 if [ $? != 0 ]; then

utils/generate_tables.py

@@ -4,15 +4,24 @@
 import requests
 from collections import OrderedDict
 from typing import Dict, List, Tuple
+import urllib.parse
 
 # Functions to compute fields for simulations ###########################################
 def get_data_size_and_hpss(hpss_path: str) -> Tuple[str, str]:
         """Get the data size in TB"""
-        output = "out.txt"
+        is_symlink: bool = check_if_symlink(hpss_path)
+        output = "out_du.txt"
         if os.path.exists(output):
             os.remove(output)
         try:
-            os.system(f'(hsi "du {hpss_path}") 2>&1 | tee {output}')
+            if is_symlink:
+                # The `/*` expands symlinks on HSI!
+                # This will actually work fine even if it's not a symlink,
+                # but we needed to check for symlinks anyway to note "(symlink)" by the HPSS path,
+                # so we might as well handle the cases separately here.
+                os.system(f'(hsi "du {hpss_path}/*") 2>&1 | tee {output}')
+            else:
+                os.system(f'(hsi "du {hpss_path}") 2>&1 | tee {output}')
         except Exception as e:
             print(f"hsi failed: {e}")
             return ("", "")
@@ -31,37 +40,85 @@ def get_data_size_and_hpss(hpss_path: str) -> Tuple[str, str]:
         data_size = int(num_bytes)/1e12
         if data_size > 0:
             data_size = f"{data_size:.0f}"
-            hpss = hpss_path
+            if is_symlink:
+                hpss = f"(symlink) {hpss_path}"
+            else:
+                hpss = hpss_path
         else:
             data_size = ""
             hpss = ""
         return (data_size, hpss)
 
-def get_esgf(source_id: str, model_version: str, experiment: str, ensemble_num: str, link_type: str, node: str) -> str:
+def check_if_symlink(hpss_path: str) -> bool:
+    output: str = "out_symlink_check.txt"
+    if os.path.exists(output):
+        os.remove(output)
+    try:
+        os.system(f'(hsi "ls {hpss_path}") 2>&1 | tee {output}')
+    except Exception as e:
+        print(f"hsi failed: {e}")
+        return False
+    with open(output, "r") as f:
+        for line in f:
+            # Symlinks on HSI/HPSS end in `@`
+            match_object = re.search(f"{os.path.basename(hpss_path)}@", line)
+            if match_object:
+                return True
+    return False
+
+
+def get_esgf(model_version: str, resolution: str, simulation_name: str, experiment: str, ensemble_num: str, link_type: str, node: str) -> str:
     esgf: str
     if link_type == "none":
         esgf = ""
-    elif node == "cels.anl":
-        esgf = f"`CMIP <https://esgf-node.{node}.gov/search/?project=CMIP6&activeFacets=%7B%22source_id%22%3A%22{source_id}%22%2C%22experiment_id%22%3A%22{experiment}%22%2C%22variant_label%22%3A%22r{ensemble_num}i1p1f1%22%7D>`_"
-    elif experiment and ensemble_num:
-        # See https://github.com/E3SM-Project/CMIP6-Metadata/pull/9#issuecomment-1246086256 for the table of ensemble numbers
-        # Note that `[1:]`` removes `v` from `model_version`
-        esgf_native: str = f"`Native <https://esgf-node.{node}.gov/search/e3sm/?model_version={model_version[1:]}_0&experiment={experiment}&ensemble_member=ens{ensemble_num}>`_"
-        if experiment == 'hist-all-xGHG-xaer':
-            experiment_id = 'hist-nat'
+    elif model_version == "v1":
+        v1_institution_id: str
+        variant_suffix: str
+        if simulation_name.startswith("LE_"):
+            v1_institution_id = "UCSB"
+            variant_suffix = "i2p2f1"
         else:
-            experiment_id = experiment
-        esgf_cmip: str = f"`CMIP <https://esgf-node.{node}.gov/search/cmip6/?source_id={source_id}&experiment_id={experiment_id}&variant_label=r{ensemble_num}i1p1f1>`_"
-        if link_type == "cmip":
-            esgf = esgf_cmip
-        elif link_type == "native":
-            esgf = esgf_native
-        elif link_type == "both":
-            esgf = esgf_cmip + ', ' + esgf_native
-        else:
-            raise ValueError(f"Invalid link_type={link_type}")
+            v1_institution_id = "E3SM-Project"
+            variant_suffix = "i1p1f1"
+        human_readable_active_facets: str = f'{{"institution_id":"{v1_institution_id}","source_id":"E3SM-1-0","experiment_id":"{experiment}","variant_label":"r{ensemble_num}{variant_suffix}"}}'
+        url_active_facets: str = urllib.parse.quote(human_readable_active_facets)
+        esgf = f"`CMIP <https://esgf-node.{node}.gov/search?project=CMIP6&activeFacets={url_active_facets}>`_"
     else:
-        esgf = ""
+        # v2, v2.1
+        # Determine source_id
+        if (len(model_version) == 4) and (model_version[2] == "."):
+            source_id = f"E3SM-{model_version[1]}-{model_version[3]}"
+        elif (len (model_version) == 2):
+            if resolution == "NARRM":
+                source_id = f"E3SM-{model_version[1]}-0-{resolution}"
+            else:
+                source_id = f"E3SM-{model_version[1]}-0"
+        else:
+            raise RuntimeError(f"Invalid model-version={model_version}")
+        # Determine esgf
+        if node == "cels.anl": # v2.1 only
+            human_readable_active_facets = f'{{"source_id":"{source_id}","experiment_id":"{experiment}","variant_label":"r{ensemble_num}i1p1f1"}}'
+            url_active_facets: str = urllib.parse.quote(human_readable_active_facets)
+            esgf = f"`CMIP <https://esgf-node.{node}.gov/search/?project=CMIP6&activeFacets={url_active_facets}>`_"
+        elif experiment and ensemble_num:
+            # See https://github.com/E3SM-Project/CMIP6-Metadata/pull/9#issuecomment-1246086256 for the table of ensemble numbers
+            # Note that `[1:]`` removes `v` from `model_version`
+            esgf_native: str = f"`Native <https://esgf-node.{node}.gov/search/e3sm/?model_version={model_version[1:]}_0&experiment={experiment}&ensemble_member=ens{ensemble_num}>`_"
+            if experiment == 'hist-all-xGHG-xaer':
+                experiment_id = 'hist-nat'
+            else:
+                experiment_id = experiment
+            esgf_cmip: str = f"`CMIP <https://esgf-node.{node}.gov/search/cmip6/?source_id={source_id}&experiment_id={experiment_id}&variant_label=r{ensemble_num}i1p1f1>`_"
+            if link_type == "cmip":
+                esgf = esgf_cmip
+            elif link_type == "native":
+                esgf = esgf_native
+            elif link_type == "both":
+                esgf = esgf_cmip + ', ' + esgf_native
+            else:
+                raise ValueError(f"Invalid link_type={link_type}")
+        else:
+            esgf = ""
     return esgf
 
 def get_run_script_original(model_version: str, simulation_name: str) -> str:
@@ -114,16 +171,7 @@ def __init__(self, simulation_dict):
             hpss_path = f"/home/projects/e3sm/www/{self.group}/E3SM{self.model_version}/{self.resolution}/{self.simulation_name}"
         self.data_size, self.hpss = get_data_size_and_hpss(hpss_path)
 
-        if (len(self.model_version) == 4) and (self.model_version[2] == "."):
-            source_id = f"E3SM-{self.model_version[1]}-{self.model_version[3]}"
-        elif (len (self.model_version) == 2):
-            if self.resolution == "NARRM":
-                source_id = f"E3SM-{self.model_version[1]}-0-{self.resolution}"
-            else:
-                source_id = f"E3SM-{self.model_version[1]}-0"
-        else:
-            raise RuntimeError(f"Invalid model-version={self.model_version}")
-        self.esgf = get_esgf(source_id, self.model_version, self.experiment, self.ensemble_num, self.link_type, self.node)
+        self.esgf = get_esgf(self.model_version, self.resolution, self.simulation_name, self.experiment, self.ensemble_num, self.link_type, self.node)
 
         self.run_script_original = get_run_script_original(self.model_version, self.simulation_name)
         self.run_script_reproduction = get_run_script_reproduction(self.model_version, self.simulation_name)
@@ -286,7 +334,7 @@ def construct_pages(csv_file: str, model_version: str, group_name: str, include_
         resolutions,
         ["Simulation", "Data Size (TB)", "ESGF Links", "HPSS Path"],
         f"../docs/source/{model_version}/{group_name}/simulation_data/simulation_table.rst",
-        [85, 15, 400, 130]
+        [85, 15, 400, 140]
     )
     if include_reproduction_scripts:
         generate_table(

utils/make_symlinks.bash

@@ -0,0 +1,18 @@
+# This will be a problem if these simulations are ever removed from the publication archives!
+for i in $(seq 1 20); do
+    hsi ln -s /home/projects/e3sm/www/publication-archives/pub_archive_E3SM_1_0_LE_historical_ens$i /home/projects/e3sm/www/WaterCycle/E3SMv1/LR/LE_historical_ens$i
+done
+
+for i in $(seq 1 20); do
+    hsi ln -s /home/projects/e3sm/www/publication-archives/pub_archive_E3SM_1_0_LE_ssp370_ens$i /home/projects/e3sm/www/WaterCycle/E3SMv1/LR/LE_ssp370_ens$i
+done
+
+# Symlink last remaining large simulation
+# This will be a problem if ndk ever deletes the source!
+hsi ln -s /home/n/ndk/2019/theta.20190910.branch_noCNT.n825def.unc06.A_WCYCL1950S_CMIP6_HR.ne120_oRRS18v3_ICG /home/projects/e3sm/www/WaterCycle/E3SMv1/LR/theta.20190910.branch_noCNT.n825def.unc06.A_WCYCL1950S_CMIP6_HR.ne120_oRRS18v3_ICG
+
+# Note:
+# It seems impossible to do a recursive remove with HSI/on HPSS.
+# > rm -rf E3SM_1_0_LE_historical_ens1@ # Trying to remove mislabeled directory
+# Unknown option or missing argument: 'r' ignored
+# Unknown option or missing argument: 'f' ignored

utils/print_ensemble_rows.bash

@@ -0,0 +1,7 @@
+for i in $(seq 1 20); do
+    echo "v1, WaterCycle, LR, LargeEnsemble, LE_historical_ens$i, , , historical-large-ensemble, $i, none, ,"
+done
+
+for i in $(seq 1 20); do
+    echo "v1, WaterCycle, LR, LargeEnsemble, LE_ssp370_ens$i, , , ssp370-large-ensemble, $i, none, ,"
+done