Changeset 700 in lmdz_wrf for trunk

Timestamp:

Mar 14, 2016, 6:30:50 PM (9 years ago)

Author:

lfita

Message:

Adding '_FillVal' to `field_stats'

File:

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

trunk/tools/nc_var_tools.py

@@ -9852 +9852 @@
 #        newval = float32(val)
     elif vtype == type(np.int(1)):
-        newval = np.int(val)
+        typeinf = np.iinfo(np.int)
+        if (abs(np.float64(val)) > typeinf.max):
+            newval = abs(np.float64(val))/np.float64(val) * np.int(typeinf.max)
     elif vtype == type(np.int16(1)):
-        newval = np.int16(val)
+        typeinf = np.iinfo(np.int16)
+        if (abs(np.float64(val)) > typeinf.max):
+            newval = abs(np.float64(val))/np.float64(val) * np.int16(typeinf.max)
     elif vtype == type(np.int32(1)):
-        newval = np.int32(val)
+        typeinf = np.iinfo(np.int32)
+        if (abs(np.float64(val)) > typeinf.max):
+            newval = abs(np.float64(val))/np.float64(val) * np.int32(typeinf.max)
     elif vtype == type(np.int64(1)):
         newval = np.int64(val)
     elif vtype == type(np.float(1)):
         newval = np.float(val)
-    elif vtype == type(np.float16(1)):
-        newval = np.float16(val)
+#    elif vtype == type(np.float16(1)):
+#        newval = np.float16(val)
     elif vtype == type(np.float32(1)):
         newval = np.float32(val)
@@ -13609 +13615 @@
     """ Function to retrieve statistics from a field
     field_stats(values, ncfile, varn)
-      [values]= kind of statistics 
-        'full': all statistics given variable
+      [values]= [stats],[fillVals]
+        [stats]: kind of statistics 
+          'full': all statistics given variable
+        [fillVals]: ':' list of _fillValues ('None' for any)
       [ncfile]= name of the netCDF file to use
       [varn]= variable name to use ('all' for all variables)
     """
-
+    import numpy.ma as ma
     fname='field_stats'
 
@@ -13621 +13629 @@
         print field_stats.__doc__
         quit()
+
+
+    arguments = '[stats],[fillVals]'
+    check_arguments(fname,values,arguments,',')
+
+    stats=values.split(',')[0]
+    fillVals=values.split(',')[1]
+
+    if fillVals == 'None':
+        fillV = None
+    else:
+        if fillVals.find(':') != -1:
+            fillV = fillVals.split(':')
+            NfillV = len(fillV)
+        else:
+            fillV = [fillVals]
+            NfillV = 1
 
     ncobj = NetCDFFile(ncfile, 'r')
@@ -13639 +13664 @@
         objfield = ncobj.variables[vn]
         field = objfield[:]
+        dtype = objfield.dtype
+        print 'Lluis dtype:',dtype,'fillVals:',fillV
+        if fillVals is not None:
+            for ifV in range(NfillV):
+                print 'fV:', fillV[ifV]
+                fillval = retype(fillV[ifV], dtype)
+                field = ma.masked_equal(field, fillval)
 
         print '   ' + vn + '... .. .'
-        if values == 'full':
+        if stats == 'full':
             minv = np.min(field)
             maxv = np.max(field)