Changeset 2197 in lmdz_wrf

Timestamp:

Oct 18, 2018, 10:20:14 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `same_deltasign': Function to determine if a given series of 1D values share the same sign of increase/decrease between consecutive values

Location:

trunk/tools

Files:

• nc_var.py (modified) (4 diffs)
• nc_var_tools.py (modified) (2 diffs)

trunk/tools/nc_var.py

@@ -63 +63 @@
 ## e.g. # nc_var.py -o retrieve_stations -f wrfout_d01_1995-01-01_00:00:00 -S 'tmin_percentiles.nc:stname:None:stlon:stlat:None:nearest:west_east:XLONG:south_north:XLAT:HGT:Time:WRFtime' -v T2,QVAPOR
 ## e.g. # nc_var.py -o compute_slice2Dstats -S 'XLAT,-63.,19.,2.,HGT,250.,7000.,500.,Time|Times:west_east|XLONG:south_north|XLAT' -f wrfout_d01_1995-01-01_00:00:00 -v T2,Q2
+## e.g. # nc_var.py -o same_deltasign -f wrfout_d01_1995-01-01_00:00:00 -S 'Time:0|south_north:60|west_east:-1' -v T2,Q2,XLAT,XLONG
 
 from optparse import OptionParser
@@ -162 +163 @@
 # remapnn: Function to remap to the nearest neightbor a variable using projection from another file
 # retrieve_stations: Function to retrieve temporal values at given stations provided by a secondary netcdf
+# same_deltasign: Function to determine if a given series of 1D values share the same sign of 
+#   increase/decrease between consecutive values
 # seasmean: Function to compute the seasonal mean of a variable
 # sellonlatbox: Function to select a lotlan box from a data-set
@@ -225 +228 @@
   'Partialmap_EntiremapFor', 'Partialmap_EntiremapForExact',                         \
   'pinterp', 'reconstruct_matrix_from_vector', 'remapnn', 'retrieve_stations',       \
-  'rm_FillValue',                                                                    \
+  'rm_FillValue', 'same_deltasign',                                                  \
   'seasmean', 'sellonlatbox', 'sellonlatlevbox', 'selvar', 'setvar_asciivalues',     \
   'setvar_nc', 'sorttimesmat', 'spacemean', 'SpatialWeightedMean',                   \
@@ -482 +485 @@
 elif oper == 'rm_FillValue':
     ncvar.rm_FillValue(opts.values, opts.ncfile, opts.varname)
+elif oper == 'same_deltasign':
+    ncvar.same_deltasign(opts.values, opts.ncfile, opts.varname)
 elif oper == 'seasmean':
     ncvar.seasmean(timename, opts.ncfile, opts.varname)

trunk/tools/nc_var_tools.py

@@ -138 +138 @@
 # retrieve_stations: Function to retrieve temporal values at given stations provided by a secondary netcdf
 # rm_FillValue: Operation to remove the _FillValue from a variable inside a netCDF file
+# same_deltasign: Function to determine if a given series of 1D values share the same sign of 
+#   increase/decrease between consecutive values
 # seasmean: Function to compute the seasonal mean of a variable
 # sellonlatbox: Function to select a lotlan box from a data-set
@@ -26401 +26403 @@
 #compute_slice2Dstats(values, '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', 'T2,Q2')
 
-def same_delat_sign(values, ncfile, variable):
+def same_deltasign(values, ncfile, variable):
     """ Function to determine if a given series of 1D values share the same sign of 
          increase/decrease between consecutive values
-      values= [slicedims], values to the dimensions to perform a slice of variables
-        [slicedims]: '|' separated list of [dimname]:[valdim]
-        [valdim]: 
-          * [integer]: which value of the dimension
-          * -1: all along the dimension
-          * -9: last value of the dimension
-          * [beg]@[end]@[freq] slice from [beg] to [end] every [freq]
-          * NOTE, no dim name all the dimension size
+      values= [slicedims]
+        [slicedims]: '|' separated list of [dimname]:[valdim] with he values to the 
+          dimensions with which perform a slice of the variables
+          [valdim]: 
+            * [integer]: which value of the dimension
+            * -1: all along the dimension
+            * -9: last value of the dimension
+            * [beg]@[end]@[freq] slice from [beg] to [end] every [freq]
+            * NOTE, no dim name all the dimension size
       ncfile= netCDF file to use
       variable: ',' list of variables ('all' for all variables)
     """
-    fname = 'same_delat_sign'
+    fname = 'same_deltasign'
 
     if values == 'h':
         print fname + '_____________________________________________________________'
-        print same_delat_sign.__doc__
+        print same_deltasign.__doc__
         quit()
 
-    expectargs = 'varn1,minvar1,maxvar1,slcevar1,varn2,minvar2,maxvar2,slcevar2,' +  \
-      'dimvars'
+    expectargs = '[slicedims]'
     gen.check_arguments(fname, values, expectargs, ',')
 
-    varn1 = values.split(',')[0]
-    minvar1 = np.float(values.split(',')[1])
+    slicedims = values.split(',')[0]
+
+    if variable == 'all':
+        varns = onc.variables.keys()
+    else:
+        varns = gen.str_list(variable, ',')
+    
+    onc = NetCDFFile(ncfile, 'r')
+    for varn in varns:
+        if not onc.variables.has_key(varn):
+            print errormsg
+            print '  ' +fname+ ": file '" + ncfile + "' does noty have variable '" + \
+              varn + "' !!"
+            ivarns = onc.variables.keys()
+            ivarns.sort()
+            print '    available ones:', ivarns
+            quit(-1)
+
+        ovar = onc.variables[varn]
+        slcvar, slcdims = slice_variable(ovar, slicedims)
+
+        if len(slcvar.shape) != 1:
+            print errormsg
+            print '  ' + fname + ": resultant slice of variable '" + varn +         \
+              "' has not rank 1 !!"
+            print '    dimensions of the variable:', ovar.dimensions
+            print '    shape of the variable:', ovar.dimensions
+            print '    resultant slice shape:', slcvar.shape
+            print '      provided values for the slice _______'
+            dicslice = gen.stringS_dictvar(slicedims, '|', ':')
+            gen.printing_dictionary(dicslice)
+            quit(-1)
+
+        dim1 = slcvar.shape[0]
+        deltavar = slcvar[1:dim1] - slcvar[0:dim1-1]
+        absdeltav = np.abs(deltavar)
+        singdelta = deltavar/absdeltav
+
+        delta0 = singdelta[0]
+        if np.all(singdelta == delta0):
+            print '  *' + varn + ': -same-', delta0
+        else:
+            print '  *' + varn + ': -NOTsame- !!'
+
+    onc.close()
 
     return
 
+#values='Time:0|south_north:60|west_east:-1'
+#same_deltasign(values, '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', 'T2,Q2,XLAT,XLONG')
+
 #quit()