config_ckdtree/config_ckdtree_r2b11_oce.py...

config_ckdtree/config_ckdtree_r2b11_oce.py config_ckdtree/run_config_ckdtree_r2b11_oce.py: Added script and slurm script for R2B11 ckdtree derivation.

config_ckdtree/config_ckdtree_r2b11_oce.py

+import numpy as np
+from netCDF4 import Dataset
+import sys, os
+import matplotlib
+if len(sys.argv)==2:
+  matplotlib.use('Agg')
+import pyicon as pyic
+
+ts = pyic.timing([0], 'start')
+
+#rev           = f'r0004'
+tgname        = f'r2b11_oce'
+gname         = f'OceanOnly_Global_IcosSymmetric_1233m_rotatedZ37d_srtm15_2018'
+path_tgrid    = f'/pool/data/ICON/oes/grids/OceanOnly/'
+fname_tgrid   = f'{gname}.nc'
+path_ckdtree  = f'/mnt/lustre01/work/mh0033/m300602/icon/grids/{tgname}/ckdtree/'
+path_rgrid    = path_ckdtree + 'rectgrids/' 
+path_sections = path_ckdtree + 'sections/' 
+
+all_grids = [
+#  'global_1.0',
+#  'global_0.3',
+#  'global_0.1',
+  'global_0.02',
+            ]
+
+all_secs = [
+#  '30W_200pts',
+#  '170W_200pts',
+#  '30W_300pts',
+#  '170W_300pts',
+            ]
+
+load_cgrid=True,
+load_egrid=False,
+load_vgrid=False,
+
+gnames = [gname]
+
+if not os.path.exists(path_rgrid): 
+  os.makedirs(path_rgrid)
+if not os.path.exists(path_sections): 
+  os.makedirs(path_sections)
+fpath = f'{path_ckdtree}/../{tgname}_tgrid.nc'
+if not os.path.exists(fpath):
+  os.symlink(path_tgrid+fname_tgrid, fpath)
+
+for gname in gnames:
+  ts = pyic.timing(ts, gname)
+  print(gname)
+
+  # --- grids
+  sname = 'global_1.0'
+  if sname in all_grids:
+    pyic.ckdtree_hgrid(lon_reg=[-180.,180.], lat_reg=[-90.,90.], res=1.0,
+                      fname_tgrid  = fname_tgrid,
+                      path_tgrid   = path_tgrid,
+                      path_ckdtree = path_rgrid,
+                      sname = sname,
+                      gname = gname,
+                      tgname = tgname,
+                      load_cgrid=load_cgrid,
+                      load_egrid=load_egrid,
+                      load_vgrid=load_vgrid,
+                      )
+  
+  sname = 'global_0.3'
+  if sname in all_grids:
+    pyic.ckdtree_hgrid(lon_reg=[-180.,180.], lat_reg=[-90.,90.], res=0.3,
+                      fname_tgrid  = fname_tgrid,
+                      path_tgrid   = path_tgrid,
+                      path_ckdtree = path_rgrid,
+                      sname = sname,
+                      gname = gname,
+                      tgname = tgname,
+                      )
+  
+  sname = 'global_0.1'
+  if sname in all_grids:
+    pyic.ckdtree_hgrid(lon_reg=[-180.,180.], lat_reg=[-90.,90.], res=0.1,
+                      fname_tgrid  = fname_tgrid,
+                      path_tgrid   = path_tgrid,
+                      path_ckdtree = path_rgrid,
+                      sname = sname,
+                      gname = gname,
+                      tgname = tgname,
+                      )
+
+  sname = 'global_0.02'
+  if sname in all_grids:
+    pyic.ckdtree_hgrid(lon_reg=[-180.,180.], lat_reg=[-90.,90.], res=0.02,
+                      fname_tgrid  = fname_tgrid,
+                      path_tgrid   = path_tgrid,
+                      path_ckdtree = path_rgrid,
+                      sname = sname,
+                      gname = gname,
+                      tgname = tgname,
+                      n_jobs = 10,
+                      )
+  
+
+  # --- sections
+  sname = '30W_200pts'
+  if sname in all_secs:
+    dckdtree, ickdtree, lon_sec, lat_sec, dist_sec = pyic.ckdtree_section(p1=[-30,-80], p2=[-30,80], npoints=200,
+                      fname_tgrid  = fname_tgrid,
+                      path_tgrid   = path_tgrid,
+                      path_ckdtree = path_sections,
+                      sname = sname,
+                      gname = gname,
+                      tgname = tgname,
+                      )
+    
+  sname = '170W_200pts'
+  if sname in all_secs:
+    dckdtree, ickdtree, lon_sec, lat_sec, dist_sec = pyic.ckdtree_section(p1=[-170,-80], p2=[-170,80], npoints=200,
+                      fname_tgrid  = fname_tgrid,
+                      path_tgrid   = path_tgrid,
+                      path_ckdtree = path_sections,
+                      sname = sname,
+                      gname = gname,
+                      tgname = tgname,
+                      )
+
+  sname = '30W_300pts'
+  if sname in all_secs:
+    dckdtree, ickdtree, lon_sec, lat_sec, dist_sec = pyic.ckdtree_section(p1=[-30,-80], p2=[-30,80], npoints=300,
+                      fname_tgrid  = fname_tgrid,
+                      path_tgrid   = path_tgrid,
+                      path_ckdtree = path_sections,
+                      sname = sname,
+                      gname = gname,
+                      tgname = tgname,
+                      )
+    
+  sname = '170W_300pts'
+  if sname in all_secs:
+    dckdtree, ickdtree, lon_sec, lat_sec, dist_sec = pyic.ckdtree_section(p1=[-170,-80], p2=[-170,80], npoints=300,
+                      fname_tgrid  = fname_tgrid,
+                      path_tgrid   = path_tgrid,
+                      path_ckdtree = path_sections,
+                      sname = sname,
+                      gname = gname,
+                      tgname = tgname,
+                      )
+
+print('make_ckdtree.py: All done!')
+ts = pyic.timing(ts, 'All done!')

config_ckdtree/run_config_ckdtree_r2b11_oce.py

+#! /bin/bash
+#SBATCH --job-name=pysmt
+#SBATCH --time=08:00:00
+#SBATCH --output=log.o-%j.out
+#SBATCH --error=log.o-%j.out
+#SBATCH --ntasks=1
+#SBATCH --partition=gpu
+#####SBATCH --account=bm0371
+#####SBATCH --account=mh0033
+#SBATCH --account=bm1102
+
+#module load python/2.7.12
+module list
+
+#source /home/mpim/m300602/bin/myactcondenv.sh
+source /home/mpim/m300602/pyicon/tools/conda_act_mistral_pyicon_env.sh
+which python
+
+startdate=`date +%Y-%m-%d\ %H:%M:%S`
+python -u config_ckdtree_r2b11_oce.py --slurm
+enddate=`date +%Y-%m-%d\ %H:%M:%S`
+echo "Started at ${startdate}"
+echo "Ended at   ${enddate}"
+