From 0ceca17a9565b382719d60a402702398ec95659f Mon Sep 17 00:00:00 2001 From: Grzegorz Kowal Date: Mon, 18 Feb 2019 11:27:47 -0300 Subject: [PATCH] PYTHON: Add refinement 'level' to be read as dataset. Signed-off-by: Grzegorz Kowal --- python/amun.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/python/amun.py b/python/amun.py index 10510e6..a37f061 100644 --- a/python/amun.py +++ b/python/amun.py @@ -236,6 +236,7 @@ def amun_dataset(fname, vname, shrink = 1, progress = False): # add derived variables if possible # + variables.append('level') if 'velx' in variables and 'vely' in variables and 'velz' in variables: variables.append('velo') variables.append('divv') @@ -311,7 +312,11 @@ def amun_dataset(fname, vname, shrink = 1, progress = False): dz = amun_coordinate(lname, 'dz') f = h5.File(lname, 'r') g = f['variables'] - if vname == 'velo': + if vname == 'level': + dataset = np.zeros(g[variables[0]].shape) + for l in range(dblocks): + dataset[:,:,:,l] = levels[l] + elif vname == 'velo': dataset = np.sqrt(g['velx'][:,:,:,:]**2 \ + g['vely'][:,:,:,:]**2 \ + g['velz'][:,:,:,:]**2)