Changeset 2265 in lmdz_wrf

Timestamp:

Dec 26, 2018, 4:07:49 PM (6 years ago)

Author:

lfita

Message:

Starting to introduce 'area_weighted': Function to provide an area weighted statistics within two different values

of matrices assuming regular lon/lat projections without taking into account
Earth's curvature

File:

• trunk/tools/nc_var_tools.py (modified) (1 diff)

trunk/tools/nc_var_tools.py

@@ -27177 +27177 @@
     return
 
+def area_weighted(values, ncfiles, variable):
+    """ Function to provide an area weighted statistics within two different values 
+        of matrices assuming regular lon/lat projections without taking into account 
+        Earth's curvature
+      [values]= [strict]:[stats] ':' separated list of values to refine the function
+        [strict]: give an error if the area of the polygon from the 'reference' file
+          is not fully covered with data from the 'get' file
+        [stats]: ',' separated list of space-statistics within each `ref' grid to 
+            provide for each grid cell from the 'get' data
+          'mean': weghted mean (always provided)
+          'min': minimum value and is associated weight
+          'max': maximum value and is associated weight
+          'mean2': weghted quadratic mean
+          'stddev': standard deviation
+      [ncfiles]= ',' separated list of [netcdfref]:[dimvalsref],[netcdfget]:[dimvalsget]
+        [netcdfref]: netCDF file name to be used as reference or destination        
+        [netcdfget]: netCDF file name to get the values to be space-weighted into 
+          destination
+        [dimvalsref]/[dimvalsget]: ';' separated list of values indicating necessary
+            data from each file
+          [dimvals]= [dimxn];[varxn];[slicedim];[dimyn];[varyn];[slicedim]
+            [dimxn]: name of the dimension along the x-axis
+            [varxn]: name of the variable with the values of dimxn
+            [slicedim]: series of values to define the slicing of the variable
+              * [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
+      [variable]= ',' separated list of variables from 'get' file to area-weight into
+        'ref' ('all, for all that variables with dimensions dimxn and dimyn)
+    """
+    fname = 'area_weighted' 
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print area_weighted.__doc__
+        quit()
+
+    statsavail = ['mean', 'min', 'max', 'mean2', 'stddev']
+
+    expectargs = '[strict]:[stats]'
+    gen.check_arguments(fname, values, expectargs, ':')
+
+    strict = gen.Str_bool(values.split(':')[[0])
+    stats = values.split(':')[[0].split(',')
+
+    for stn in stats:
+        if not gen.searchInlist(statsavail, stn):
+            print errormsg
+            print '  ' + fname + ": statistics '" + stn + "' not ready !!"
+            print '    available ones:', statsavail
+            quit(-1)
+
+    files = ncfiles.split(',')
+    if len(files) != 2:
+        print errormsg
+        print '  ' + fname + ": wrong value for [ncfiles]: '" + ncfiles + '" !!"
+        print "    information for 2 files must be provided as '[netcdfref]:" +      \
+          "[dimvalsref],[netcdfget]:[dimvalsget]'"
+        quit(-1)
+
+    iif = 0
+    for fileinf in files:
+        expectargs = '[netcdfname]:[dimvals]'
+        gen.check_arguments(fname, fileinf, expectargs, ':')
+
+        if iif == 0:
+            netcdfrefn = fileinf.split(':')[0]
+            dimvalsref = fileinf.split(':')[1].split(';')
+
+            if not os.path.isfile(netcdfrefn):
+                print errormsg
+                print '  ' + fname + ": reference file '" + netcdfrefn +             \
+                  "' does not exist !!"
+                quit(-1)
+
+            if len(dimvalsref) != 6:
+                print errormsg
+                print '  ' + fname + ": wrong number of dimension-values !!"
+                print '    6 values are expected and you provided', len(dimvalsref)
+                print "    values must be '[dimxn];[varxn];[slicedim];[dimyn];" +    \
+                  "[varyn];[slicedim]'"
+                quit(-1)
+
+            oncref = NetCDFFile(netcdfrefn, 'r')
+
+            refdimxn = dimvalsref[0]
+            refvarxn = dimvalsref[0]
+            slicedimx = dimvalsref[0]
+            dimxn = dimvalsref[0]
+
+
+            
+
+        elif:
+            netcdfgetn = fileinf.split(':')[0]
+            dimvalsget = fileinf.split(':')[1].split(';')
+        iif = iif + 1
+
+    # Getting values
+    
+            
+
+    return
 
 #quit()