From cc89806f98ac0e681b27a4efa3393cc6b37752a6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nils=20Br=C3=BCggemann?= <nils.brueggemann@mpimet.mpg.de>
Date: Mon, 10 Mar 2025 15:38:04 +0100
Subject: [PATCH] config_ckdtree_r2b9_oce_r0004.py: Also deriving npz files for
section ckdtree files.
---
.../config_ckdtree_r2b9_oce_r0004.py | 67 ++++++++++++++++---
1 file changed, 56 insertions(+), 11 deletions(-)
diff --git a/config_ckdtree/config_ckdtree_r2b9_oce_r0004.py b/config_ckdtree/config_ckdtree_r2b9_oce_r0004.py
index 7734ba2..3561ae2 100644
--- a/config_ckdtree/config_ckdtree_r2b9_oce_r0004.py
+++ b/config_ckdtree/config_ckdtree_r2b9_oce_r0004.py
@@ -1,10 +1,7 @@
import numpy as np
from netCDF4 import Dataset
import sys, os
-from importlib import reload
-
import pyicon as pyic
-reload(pyic)
ts = pyic.timing([0], 'start')
@@ -13,14 +10,14 @@ tgname = f'r2b9_oce_{rev}'
gname = f'OceanOnly_IcosSymmetric_4932m_rotatedZ37d_modified_srtm30_1min'
path_tgrid = f'/pool/data/ICON/oes/input/{rev}/{gname}/'
fname_tgrid = f'{gname}.nc'
-path_ckdtree = f'/mnt/lustre01/work/mh0033/m300602/icon/grids/{tgname}/ckdtree/'
+path_ckdtree = f'/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_1.0',
+ 'global_0.3',
+ 'global_0.1',
'global_0.02',
]
@@ -31,10 +28,9 @@ all_secs = [
'170W_300pts',
]
-#all_grids = []
-all_secs = []
-
-gnames = [gname]
+load_cgrid=True,
+load_egrid=True,
+load_vgrid=True,
if not os.path.exists(path_rgrid):
os.makedirs(path_rgrid)
@@ -44,6 +40,8 @@ fpath = f'{path_ckdtree}/../{tgname}_tgrid.nc'
if not os.path.exists(fpath):
os.symlink(path_tgrid+fname_tgrid, fpath)
+gnames = [gname]
+print(gnames)
for gname in gnames:
ts = pyic.timing(ts, gname)
print(gname)
@@ -58,6 +56,10 @@ for gname in gnames:
sname = sname,
gname = gname,
tgname = tgname,
+ load_cgrid=load_cgrid,
+ load_egrid=load_egrid,
+ load_vgrid=load_vgrid,
+ save_as='both',
)
sname = 'global_0.3'
@@ -69,6 +71,10 @@ for gname in gnames:
sname = sname,
gname = gname,
tgname = tgname,
+ load_cgrid=load_cgrid,
+ load_egrid=load_egrid,
+ load_vgrid=load_vgrid,
+ save_as='both',
)
sname = 'global_0.1'
@@ -80,6 +86,10 @@ for gname in gnames:
sname = sname,
gname = gname,
tgname = tgname,
+ load_cgrid=load_cgrid,
+ load_egrid=load_egrid,
+ load_vgrid=load_vgrid,
+ save_as='both',
)
sname = 'global_0.02'
@@ -91,6 +101,25 @@ for gname in gnames:
sname = sname,
gname = gname,
tgname = tgname,
+ load_cgrid=load_cgrid,
+ load_egrid=load_egrid,
+ load_vgrid=load_vgrid,
+ save_as='both',
+ )
+
+ sname = 'global_0.01'
+ if sname in all_grids:
+ pyic.ckdtree_hgrid(lon_reg=[-180.,180.], lat_reg=[-90.,90.], res=0.01,
+ 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,
+ save_as='both',
)
@@ -104,6 +133,10 @@ for gname in gnames:
sname = sname,
gname = gname,
tgname = tgname,
+ load_cgrid=load_cgrid,
+ load_egrid=load_egrid,
+ load_vgrid=load_vgrid,
+ save_as='both',
)
sname = '170W_200pts'
@@ -115,6 +148,10 @@ for gname in gnames:
sname = sname,
gname = gname,
tgname = tgname,
+ load_cgrid=load_cgrid,
+ load_egrid=load_egrid,
+ load_vgrid=load_vgrid,
+ save_as='both',
)
sname = '30W_300pts'
@@ -126,6 +163,10 @@ for gname in gnames:
sname = sname,
gname = gname,
tgname = tgname,
+ load_cgrid=load_cgrid,
+ load_egrid=load_egrid,
+ load_vgrid=load_vgrid,
+ save_as='both',
)
sname = '170W_300pts'
@@ -137,6 +178,10 @@ for gname in gnames:
sname = sname,
gname = gname,
tgname = tgname,
+ load_cgrid=load_cgrid,
+ load_egrid=load_egrid,
+ load_vgrid=load_vgrid,
+ save_as='both',
)
print('make_ckdtree.py: All done!')
--
GitLab