Changeset 882 in lmdz_wrf for trunk/tools

Timestamp:

Jun 18, 2016, 7:16:25 PM (9 years ago)

Author:

lfita

Message:

Adding `SliceVarDict?': Function to slice a given variable throughout a dictionary

File:

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

trunk/tools/nc_var_tools.py

@@ -91 +91 @@
 # setvar_asciivalues: Function to set de values of a variable with an ASCII file (common Fortran-like format)
 # SliceVar: Function to slice a given variable throughout a given list of dimensions
+# SliceVarDict: Function to slice a given variable throughout a dictionary
 # slice_variable: Function to return a slice of a given variable according to values to its dimensions
 # sorttimesmat: Function to sort the time values of a given file
@@ -8891 +8892 @@
             * -1: all along the dimension
             * [beg]:[end] slice from [beg] to [end]
+            NOTE: that dimension of the variables which not appear al values will be used
+        [dimsoper]; [dimname1]:[dimname2]:[...[dimnameN]] names of the dimensions along 
+          which operation has to be done
+        [opkind]; operation to perform along the dimensions with a range: max, mean, 
+          mean2, min, sum
+        [dimvn]; [varname1]:[varname2]:[...[varnameM]] variables with the values of the 
+          dimensions
+      ncfile= netCDF file
+      varn= ',' list of variable names ('all' for all)
+    """
+    fname = 'file_oper_alongdims'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print file_oper_alongdims.__doc__
+        quit()
+
+    dimvals = values.split(',')[0]
+    dimsoper = values.split(',')[1]
+    operkind = values.split(',')[2]
+    dimvn = values.split(',')[3]
+
+    dimsoperS = gen.numVector_String(dimsoper.split(':'),', ')
+    objnc = NetCDFFile(ncfile, 'r')
+
+    if varn.find(',') != -1:
+        varns = varn.split(',')
+    else:
+        if varn == 'all':
+            varns = objnc.variables
+        else:
+            varns = [varn]
+   
+    ofile = 'file_oper_alongdims_' + operkind + '.nc'
+
+    if dimsoper.find(':') != -1:
+        dimstouse = dimsoper.split(':')
+    else:
+        dimstouse = [dimsoper]
+
+# Getting slice dimensions
+    dimslice = []
+    dimvalslice = []
+    for dv in dimvals:
+        dnv = dv.slice('|')[0]
+        dvv = dv.slice('|')[1]
+        dimslice.append(dnv)
+        dimvalslice.append(dvv)
+
+
+# Creation of output file
+##
+    objnewnc = NetCDFFile(ofile, 'w')
+
+    print '  ' + fname + ": '" + operkind + "' with:",dimsoper,'_______'
+    for vn in varns:
+        print "'" + vn + "' ... .. ."
+        if not objnc.variables.has_key(vn):
+            print errormsg
+            print '  ' + fname + ': netCDF file "' + ncfile +                        \
+              '" does not have variable "' + vn + '" !!'
+            quit(-1)
+
+        # Do we have to compute this variable?
+        ov = objnc.variables[vn]
+        tocompute = False
+        for vd in ov.dimensions:
+            if searchInlist(dimsoper, vd): tocompute = True
+
+        if tocopmute:
+        
+
+
+
+    return
+
+def file_oper_alongdims_old(values, ncfile, varn):
+    """ Function to operate a file along different dimensions of a variable
+    file_oper_alongdims(values, ncfile, varn)
+      values= [dimvals],[dimsoper],[opkind],[dimvn]
+        [dimvals]; [dimname1]:[val1]|[dimdname2]:[val2]|[...[dimnameN]:[valN]]
+          [value]; 
+            * [integer]: which value of the dimension
+            * -1: all along the dimension
+            * [beg]:[end] slice from [beg] to [end]
         [dimsoper]; [dimname1]:[dimname2]:[...[dimnameN]] names of the dimensions along 
           which operation has to be done
@@ -12761 +12847 @@
         if gen.searchInlist(dloop,dim):
             slicevals.append(dloopindex[idim])
+        else:
+            slicevals.append(slice(0,varshape[idim]))
+        idim = idim + 1
+
+    return slicevals
+
+def SliceVarDict(ovar,slicedcit):
+    """ Function to slice a given variable throughout a dictionary
+      ovar= variable object to slice
+      slicedict= dictionary with the values to slice
+        slicedict[dimn] = [value]
+          [value] = -1, all the values
+          [value] = int, a single value
+          [value] = [beg, end, frq], from a beginning to an end with a given frequency
+    """
+    fname = 'SliceVarDict'
+
+    varshape = ovar.shape
+
+    slicevals = []
+    idim = 0
+    for dim in ovar.dimensions:
+        if gen.searchInlist(slicedict.keys(),dim):
+            dictv = slicedict[dim]
+            if type(dictv) == type([1]):
+                slicevals.append( slice(dictv[0], dictv[1], dictv[2]) )
+            elif type(dictv) == type(int(1)):
+                if dictv == -1:
+                    slicevals.append(slice(0,varshape[idim]))
+                else
+                    slicevals.append(dictv)
         else:
             slicevals.append(slice(0,varshape[idim]))