Changeset 2329 in lmdz_wrf

Timestamp:

Feb 11, 2019, 6:23:19 PM (6 years ago)

Author:

lfita

Message:

Adding value of the dimension to `ovar_reducedims'

File:

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

trunk/tools/nc_var_tools.py

@@ -27699 +27699 @@
 #area_weighted('yes:min,max,mean,stddev',fvals,'ct_values')
 
-def ovar_reducedims(ovar, dimns):
+def ovar_reducedims(ovar, dimns, dimvs=[0]):
     """ Function to reduce a variable removing some of its dimensions
       ovar: Object variable to reduce
       dimns: list of dimensions to remove from the variable
+      dimvs: list values of the dimensions to use
     """
     fname = 'ovar_reducedims'
  
     dimv = ovar.dimensions
+
+    Ndims = len(dimns)
+    if len(dimvs) == 1 and Ndims > 1:
+        oldimvs = dimvs + []
+        dimvs = []
+        for iid in range(Ndims): dimvs.append(oldimvs[0])
     
     slicevar = []
@@ -27713 +27720 @@
     iid = 0
     for idn in dimv:
-        if gen.searchInlist(dimns,idn): slicevar.append(0)
+        if gen.searchInlist(dimns,idn): slicevar.append(dimvs[idn])
         else:
             ldim = ovar.shape[iid]
@@ -29387 +29394 @@
 #compute_slices_stats_areaweighted(values, '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', 'T2')
 
+def except_fillValue(values, ncfile, variable):
+    """ Function to fill a file with fill_Value except a given value of a variable
+      values= [varname]:[value]:[rmdims]
+        [varname]: name of the variable to use its values for the filling
+        [value]: value of the variable from which grid points will not be filled by 
+            fill_value
+          'exact',[value]: just a single value to use to avoid filling
+          'range',[ivalue],[evalue]: avoid filling points with values within the 
+            range [ivaule],[evalue]
+        [rmdims]: ',' separated list of [dimn]@[dimv] name of dimensions and value to
+          remove from [varname] to find the value
+          [dimn]: name of the dimension from [varname]
+          [dimv]: value of [dimn] to use
+      ncfile= netCDF file to fill
+      variable= name of the variables to use
+    """
+    fname = 'except_fillValue'
+
+    availvalue = ["'exact',[value]", "'range',[ivalue],[evalue]"]
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print except_fillValue.__doc__
+        quit()
+
+    expectargs = '[varname]:[value]:[rmdims]'
+    gen.check_arguments(fname, values, expectargs, ':')
+
+    varname = values.split(':')[0]
+    value = values.split(':')[1]
+    rmdims = values.split(':')[2]
+
+    if value[0:5] == 'exact':
+        varvalue = value.split(',')[1]
+    elif value[0:5] == 'range':
+        ivarvalue = value.split(',')[1]
+        evarvalue = value.split(',')[2]
+    else:
+        print errormsg
+        print '  ' + fname + ": value for variable '" + value +  "' not ready !!"
+        print '    available ones:', availvalue
+        quit(-1)
+
+    onc = NetCDFFile(ncfile, 'r')
+    if not onc.variables.has_key(varname):
+        print errormsg
+        print '  ' + fname + ": file '" + + "' does not have variable '" + varname + \
+          "' !!"
+        varns = list(onc.variables.keys())
+        varns.sort()
+        print '    available ones:', varns
+        quit(-1)
+
+    ovar = onc.variables[varname]
+    
+
 #quit()