Changeset 2194 in lmdz_wrf

Timestamp:

Oct 18, 2018, 3:52:29 PM (6 years ago)

Author:

lfita

Message:

Working version !!

File:

• trunk/tools/nc_var_tools.py (modified) (1 diff)

trunk/tools/nc_var_tools.py

@@ -26234 +26234 @@
         slcvar2[islc,] = ma.masked_inside(var2v, slices2[islc], slices2[islc+1]).mask
 
-    # Looking for coincident dimensions
-    for dn1 in ovar1.diensions:
-        for dn2 in ovar1.diensions:
-            print 'Hola'
+    if variable == 'all':
+        varns = onc.variables.keys()
+    else:
+        varns = gen.str_list(variable, ',')
+
+    for varn in varns:
+        ovar = onc.variables[varn]
+        varv = ovar[:]
+        vdimns = list(ovar.dimensions)
+        vshape = list(ovar.shape)
+
+        # mask1 in var slice
+        NOTcoinc1 = list(set(ovar1.dimensions) - set(vdimns))
+        if len(NOTcoinc1) > 0:
+            notcoincdim1 = {}
+            for dn in NOTcoinc1: notcoincdim1[dn] = 0
+        else:
+            notcoincdim1 = None
+        mask1varslice = []
+        for islc in range(Nslices1):
+            varmask1 = gen.mat_dimreshape(list(ovar1.dimensions), slcvar1[islc,],    \
+              vdimns, vshape, notcoincdim1)
+            mask1varslice.append(varmask1)
+
+        # mask2 in var slice
+        NOTcoinc2 = list(set(ovar2.dimensions) - set(vdimns))
+        if len(NOTcoinc2) > 0:
+            notcoincdim2 = {}
+            for dn in NOTcoinc2: notcoincdim2[dn] = 0
+        else:
+            notcoincdim2 = None
+        mask2varslice = []
+        for islc in range(Nslices2):
+            varmask2 = gen.mat_dimreshape(list(ovar2.dimensions), slcvar2[islc,],    \
+              vdimns, vshape, notcoincdim2)
+            mask2varslice.append(varmask2)
+
+        newvarshape = [Nslices2, Nslices1] + vshape
+        newvar2masked = np.zeros(tuple(newvarshape), dtype=ovar.dtype)
+
+        for islc2 in range(Nslices2):
+            newmask2 = mask2varslice[islc2]
+            for islc1 in range(Nslices1):
+                newmask1 = mask1varslice[islc1]
+                newmask = newmask2+newmask1
+
+                print '   Lluis shapes: newvar2masked',  newvar2masked[islc2,islc1,].shape, 'varv:', varv[newmask].shape
+                newvar2masked[islc2,islc1,] = ma.masked_array(varv, mask=~newmask)
+
+        onewnc = NetCDFFile(fname + '.nc', 'w')
+        # dimensions
+        newdim = onewnc.createDimension('lon', varv.shape[2])
+        newdim = onewnc.createDimension('lat', varv.shape[1])
+        newdim = onewnc.createDimension('time', None)
+        newdim = onewnc.createDimension('slice_'+varn1, Nslices1)
+        newdim = onewnc.createDimension('slice_'+varn2, Nslices2)
+
+        # variables
+        newvar = onewnc.createVariable(varn, 'f', ('slice_'+varn2, 'slice_'+varn1, 'time', 'lat', 'lon'))
+        newvar[:] = newvar2masked[:]
+
+        onewnc.sync()
+        onewnc.close()
+
+        quit()
 
     return
 
-#values='south_north,XLAT,-63.,19.,2.,all,HGT,250.,7000.,500.'
-
-#compute_slice2Dstats(values, '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', 'T2')
+values='south_north,XLAT,-63.,19.,2.,all,HGT,250.,7000.,500.'
+
+compute_slice2Dstats(values, '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', 'T2')
 
 #quit()