Changeset 2198 in lmdz_wrf

Timestamp:

Oct 19, 2018, 3:06:26 PM (6 years ago)

Author:

lfita

Message:

 

File:

• trunk/tools/nc_var_tools.py (modified) (4 diffs)

trunk/tools/nc_var_tools.py

@@ -26199 +26199 @@
             quit(-1)
 
-
     ovar1 = onc.variables[varn1]
     ovar2 = onc.variables[varn2]
     v1u = ovar1.units
     v2u = ovar2.units
+    dimnv1 = list(ovar1.dimensions)
+    dimnv2 = list(ovar2.dimensions)
 
     dvar1 = (maxvar1 - minvar1+slicevar1)/slicevar1
@@ -26255 +26256 @@
 
     # variable dimensions
+    for dn in dimnv1:
+        if not gen.searchInlist(onewnc.dimensions,dn): add_dims(onc, onewnc, [dn])
+
     newvar = onewnc.createVariable('slice_'+varn1, 'f', ('slice_'+varn1))
     newvar[:] = slcvalsc1[:]
@@ -26265 +26269 @@
       varn1, v1u)
 
+    slcdims = ['slice_'+varn1] + dimnv1
+    slcshape = [Nslices1] + list(ovar1.shape)
+    slcv = np.zeros(tuple(slcshape), dtype=int)
+    for islc in range(Nslices1):
+        vvslc = slcvar1[islc,]
+        slcv[islc,] = np.where(vvslc, 1, 0)
+    newvar = onewnc.createVariable(varn1+'sliced', 'i', tuple(slcdims))
+    newvar[:] = slcv[:]
+    basicvardef(newvar, varn1+'sliced', 'sliced variable ' + varn1, v1u)
+
+    for dn in dimnv2:
+        if not gen.searchInlist(onewnc.dimensions,dn): add_dims(onc, onewnc, [dn])
+
     newvar = onewnc.createVariable('slice_'+varn2, 'f', ('slice_'+varn2))
     newvar[:] = slcvalsc2[:]
@@ -26274 +26291 @@
     basicvardef(newvar, 'slice_'+varn2+'_bnds', 'boundaries of slices for variable '+\
       varn2, v2u)
+
+    slcdims = ['slice_'+varn2] + dimnv2
+    slcshape = [Nslices2] + list(ovar2.shape)
+    slcv = np.zeros(tuple(slcshape), dtype=int)
+    for islc in range(Nslices2):
+        vvslc = slcvar2[islc,]
+        slcv[islc,] = np.where(vvslc, 1, 0)
+    newvar = onewnc.createVariable(varn2+'sliced', 'i', tuple(slcdims))
+    newvar[:] = slcv[:]
+    basicvardef(newvar, varn2+'sliced', 'sliced variable ' + varn2, v2u)
 
     for varn in varns: