Changeset 1763 in lmdz_wrf

Timestamp:

Jan 24, 2018, 4:06:22 PM (8 years ago)

Author:

lfita

Message:

Adgin variable slicing and improving in-line help

File:

• trunk/tools/drawing.py (modified) (7 diffs)

trunk/tools/drawing.py

@@ -2339 +2339 @@
     """ Function to draw different lines at the same time from different files with times
     draw_lines_time(ncfilens, values, varname):
-      ncfilens= [filen] ',' separated list of netCDF files
+      ncfilens= [filen]%[dimval] ',' separated list of netCDF files and the 'slice' 
+        along the dimensions of the file
+          [dimval]: ';' list of [dimn]|[dimval] to get the values for [varn]
+            [dimn]: name of the dimension
+            [dimval]: value of the dimension variable a given value is required:
+              * [integer]: which value of the dimension
+              * -1: all along the dimension
+              * -9: last value of the dimension
+              * [beg]@[end]@[inc] slice from [beg] to [end] every [inc]
+              * NOTE, no dim name all the dimension size
       values= [dimvname];[valuesaxis];[dimtit];[dimxyfmt];[leglabels];[vtit];[title];[rangevals];[timevals];
         [legvals];[graphk];[collines];[points];[linewidths];[pointsizes];[pointfreq];[period];[close]
-        [dimvname]: ',' list of names of the variables with he values of the common dimension ('WRFtime' for WRF Times variable)
+        [dimvname]: ',' list of names of the variables with the values of the common dimension for all lines ('WRFtime' for WRF Times variable)
         [valuesaxis]: which axis will be used for the values ('x', or 'y')
         [dimtit]: title for the common dimension ('|' for spaces)
@@ -2366 +2375 @@
            [timen]; name of the time variable
            [units]; units string according to CF conventions ([tunits] since 
-             [YYYY]-[MM]-[DD] [[HH]:[MI]:[SS]], '!' for spaces)
+             [YYYY]-[MM]-[DD] [[HH]:[MI]:[SS]], '!' for spaces) this is used to 
+             re-compute the time values to be the same reference for all the lines
            [kind]; kind of output
              'Nval': according to a given number of values as 'Nval',[Nval]
@@ -2380 +2390 @@
             9: 'upper center', 10: 'center'
           [fontsize]: font size for the legend (auto for 12)
-        [graphk]: kind of the graphic
-        [lines]: ',' list of type of lines, None for automatic, single value all the same
-        [collines]: ',' list of colors for the lines, None for automatic, single 
-          value all the same
-        [points]: '@' list of type of points for the lines, None for automatic, single
-          value all the same
+        [graphk]: kind of the output of the graphic ('png', 'pdf', 'eps', ...)
+        [lines]: ',' list of type of lines (matplotlib syntax), 'None' for automatic,
+          providing a single value will be used the same for all the lines
+        [collines]: ',' list of colors for the lines (matplotlib syntax), 'None' for 
+          automatic, single value all the same
+        [points]: '@' list of type of points (matplotlib syntax ',' no points) for 
+          the lines, 'None' for automatic, single value all the same
         [linewidths]: ',' list of widths for the lines, None for automatic, single
           value all the same
         [pointsizes]: ',' list of widths for the lines, None for automatic, single
           value all the same
-        [pointfreq]: frequency of point plotting, 'all' for all time steps
-        [period]: which period to plot
-          '-1': all period
-          [beg],[end]: beginning and end of the period in reference time-units of first file
+        [pointfreq]: frequency of point plotting (every given number of values), 
+          'all' to plot all values using all time steps
+        [period]: which period of the data to plot:
+          '-1': all period of data
+          [beg],[end]: beginning and end of the period in reference time-units of 
+            first file
         [close]: Whether figure should be finished or not
-      varnames= ',' list of variable names to plot (assuming only 1 variable per file)
+      varnames= ',' list of variable names to plot (assuming only 1 variable per file
+        consecutively)
       values= 'time;y;time ([DD]${[HH]}$);32x32;$wss^{*}$;wss Taylor's turbulence term;time|hours!since!1949-12-01_00:00:00;exct,12,h|%d$^{%H}$;2;pdf'
     """
@@ -2411 +2425 @@
     drw.check_arguments(fname,values,expectargs,';')
 
-    ncfiles = ncfilens.split(',')
+    ncfiledims = ncfilens.split(',')
     dimvname0 = values.split(';')[0]
     valuesaxis = values.split(';')[1]
@@ -2531 +2545 @@
     maxtval = -1.e20
 
-    for ifile in ncfiles:
-        filen = ifile.split('@')[0]
-
-        print '    filen:',filen
+    for ifile in ncfiledims:
+
+        filen = ifile.split('%')[0]
+        dims = ifile.split('%')[1]
+
+        print '    filen:', filen, 'section:', dims
 
         if not os.path.isfile(filen):
@@ -2589 +2605 @@
             if objfile.variables.has_key(var):
                 foundvar = True
-                vvobj = objfile.variables[var]
+                vvobj0 = objfile.variables[var]
+
+                # Slicing variables
+                dictslice = {}
+                for dnv in dimvalues.split(';'):
+                    dimn = dnv.split('|')[0]
+                    dimv = dnv.split('|')[1]
+                    if dimv.find(',') != -1:
+                        dictslice[dimn] = list(np.array(dimv.split('@'), dtype=int))
+                    else:
+                        dictslice[dimn] = int(dimv)
+
+                slicev, ddv = ncvar.SliceVarDict(vvobj0, dictslice)
+                vvobj = timeobj[tuple(slicet)]
+
                 if len(vvobj.shape) != 1:
                     print errormsg
@@ -2671 +2701 @@
 
         if ifn == 0:
-            varunits = drw.units_lunits(vvobj.units)
+            varunits = drw.units_lunits(vvobj0.units)
+        else:
+            if varunits != drw.units_lunits(vvobj0.units):
+                print errormsg
+                print '  ' + fname + ': wrong units:', vvobj0.units, " of " +        \
+                  "variable '" + var +  "' with respect first variable '", varunits, \
+                  "' !!"
+                quit(-1)
 
         objfile.close()