Changeset 793 in lmdz_wrf for trunk/tools

Timestamp:

Jun 1, 2016, 3:11:21 PM (9 years ago)

Author:

lfita

Message:

Adding single-point retrieval in `DataSetSection_multivars' (with int=0)

File:

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

trunk/tools/nc_var_tools.py

@@ -12396 +12396 @@
           [endi] = -1, maximum value
           [inti] = -1, all the values within the range
+                 = 0, single value. [begi] will be taken as the reference value
           NOTE: variables without section by the variables are taken allover their size
           NOTE2: if inti = 'str', it is assumed that the variable is a string-list of values
@@ -12425 +12426 @@
     endvs = np.zeros((Nvars), dtype=int)
     intvs = np.zeros((Nvars), dtype=int)
+    fbegvs = np.zeros((Nvars), dtype=np.float)
+    fendvs = np.zeros((Nvars), dtype=np.float)
+    fintvs = np.zeros((Nvars), dtype=np.float)
 
     for ivar in range(Nvars):
@@ -12460 +12464 @@
             endvs[ivar] = np.float(val.split(',')[2])
             intvs[ivar] = np.float(val.split(',')[3])
+            fbegvs[ivar] = np.float(val.split(',')[1])
+            fendvs[ivar] = np.float(val.split(',')[2])
+            fintvs[ivar] = np.float(val.split(',')[3])
             ofiletile = ofiletile + '_' + varns[ivar] + '_B' + str(begvs[ivar]) + '-E' + \
               str(endvs[ivar]) + '-I' + str(intvs[ivar])
@@ -12513 +12520 @@
             vardims = ovar.dimensions
 
-        if intvs[ivar] != -99.:
+        if intvs[ivar] == -99.:
+            ostrvar = nciobj.variables[varns[ivar]]
+            finalmask = np.ones((ostrvar.shape), dtype= bool)
+            finalmask[int(begvs[ivar]),:] = False
+        elif intvs[ivar] == 0.:
+            vmask = ma.masked_not_equal(vals, fbegvs[ivar])
+            finalmask = vmask.mask
+            if np.all(finalmask == True): 
+                print errormsg
+                print '  ' + fname + ": variable '" + varns[ivar] + "' does not " +  \
+                  'have value=', fbegvs[ivar]
+                quit(-1)
+        else:
             if endvs[ivar] == -1: endvs[ivar] = np.max(vals)
             maskinf = ma.masked_less(vals, begvs[ivar])
@@ -12524 +12543 @@
             finalmask = maskinf.mask + masksup.mask
             idn = 0
-        else:
-            ostrvar = nciobj.variables[varns[ivar]]
-            finalmask = np.ones((ostrvar.shape), dtype= bool)
-            finalmask[int(begvs[ivar]),:] = False
 
         slicedims = {}
@@ -12537 +12552 @@
             rightvals = np.where(finalmask, False, True)
 
-            if intvs[ivar] != -99.:
-                slicedims[dn]=[np.min(dinds[rightvals][0,:]), np.max(dinds[rightvals][0,:])+1, None]
-            else:
+            if intvs[ivar] == -99.:
                 if iidim == 0:
                     slicedims[dn]=[int(begvs[ivar]), int(begvs[ivar]), None]
                 else:
                     slicedims[dn]=[0, ddn, None]
+            elif intvs[ivar] == 0.:
+                if np.sum(~finalmask,axis=iidim) == 0:
+                    slicedims[dn]=[0, ddn, None]
+                elif np.sum(~finalmask,axis=iidim) == 1:
+                    ivals = gen.index_mat(finalmask, False)
+                    ival = ivals[iidim]
+                    slicedims[dn]=[ival, ival, None]
+                else:
+                    print errormsg
+                    print '  ' + fname + ': multiple', np.sum(~finalmask,axis=iidim), \
+                      'possible values!'
+                    print '    I do not know how to proceed'
+                    quit(-1)
+            else:
+                slicedims[dn]=[np.min(dinds[rightvals][0,:]), np.max(dinds[rightvals][0,:])+1, None]
 
             iidim = iidim + 1