Changeset 531 in lmdz_wrf for trunk/tools

Timestamp:

Jun 25, 2015, 11:46:49 AM (10 years ago)

Author:

lfita

Message:

Adding in `drawing_lines_time', multiple dim-vars and vars
Adding in `drawing_lines_time', providing the name of the colors to use

Location:

trunk/tools

Files:

• drawing.py (modified) (6 diffs)
• drawing_tools.py (modified) (5 diffs)

trunk/tools/drawing.py

@@ -1640 +1640 @@
     return
 
-def draw_lines_time(ncfilens, values, varname):
+def draw_lines_time(ncfilens, values, varname0):
     """ 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
-      values= [dimvname];[valuesaxis];[dimtit];[leglabels];[vtit];[title];[timevals];[locleg];[graphk]
-        [dimvname]: name of the variable with he values of the common dimension
+      values= [dimvname];[valuesaxis];[dimtit];[leglabels];[vtit];[title];[timevals];[locleg];
+        [graphk];[collines]
+        [dimvname]: ',' list of names of the variables with he values of the common dimension
         [valuesaxis]: which axis will be used for the values ('x', or 'y')
         [dimtit]: title for the common dimension
@@ -1667 +1668 @@
           9: 'upper center', 10: 'center'
         [graphk]: kind of the graphic
-      varname= variable to plot
+        [collines]: ',' list of colors for the lines or None for automatic
+      varname0= ',' list of variable names to plot (assuming only 1 variable per file)
       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'
     """
@@ -1679 +1681 @@
 
     expectargs = '[dimvname];[valuesaxis];[dimtit];[leglabels];[vtit];[title];'
-    expectargs = expectargs + '[timevals];[locleg];[graphk]'
+    expectargs = expectargs + '[timevals];[locleg];[graphk];[collines]'
     drw.check_arguments(fname,len(expectargs.split(';')),values,';',expectargs)
 
     ncfiles = ncfilens.split(',')
-    dimvname = values.split(';')[0]
+    dimvname0 = values.split(';')[0]
     valuesaxis = values.split(';')[1]
     dimtit = values.split(';')[2]
@@ -1692 +1694 @@
     locleg = int(values.split(';')[7])
     graphk = values.split(';')[8]
+    collines0 = values.split(';')[9]
 
     Nfiles = len(ncfiles)
+
+# Multiple variable-dimension names?
+    if dimvname0.find(',') != -1:
+        dimvname = dimvname0.split(',')
+    else:
+        dimvname = [dimvname0]
+
+# Multiple variables?
+    if varname0.find(',') != -1:
+        varname = varname0.split(',')
+    else:
+        varname = [varname0]
+
+# Multiple color names?
+    if collines0.find(',') != -1:
+        collines = collines0.split(',')
+    else:
+        collines = None
 
 # Getting values
@@ -1719 +1740 @@
         objfile = NetCDFFile(filen, 'r')
 
-        if not objfile.variables.has_key(dimvname):
+        founddvar = False
+        for dvar in dimvname:
+            if objfile.variables.has_key(dvar):
+                founddvar = True
+                vdobj = objfile.variables[dvar]
+                if len(vdobj.shape) != 1:
+                    print errormsg
+                    print '  ' + fname + ': wrong shape:',vdobj.shape," of " +       \
+                      "variable '" + dvar +  "' !!"
+                    quit(-1)
+                break
+        if not founddvar:
             print errormsg
             print '  ' + fname + ": netCDF file '" + filen +                         \
-              "' does not have variable '" + dimvname + "' !!"
+            "' has any variable '", dimvname, "' !!"
             quit(-1)
 
-        if not objfile.variables.has_key(varname):
+        foundvar = False
+        for var in varname:
+            if objfile.variables.has_key(var):
+                foundvar = True
+                vvobj = objfile.variables[var]
+                if len(vvobj.shape) != 1:
+                    print errormsg
+                    print '  ' + fname + ': wrong shape:',vvobj.shape," of " +       \
+                      "variable '" + var +  "' !!"
+                    quit(-1)
+
+                break
+        if not foundvar:
             print errormsg
             print '  ' + fname + ": netCDF file '" + filen +                         \
-              "' does not have variable '" + varname + "' !!"
-            quit(-1)
-
-        vvobj = objfile.variables[varname]
-        if len(vvobj.shape) != 1:
-            print errormsg
-            print '  ' + fname + ': wrong shape:',vvobj.shape," of variable '" +     \
-              varname +  "' !!"
-            quit(-1)
-
-        vdobj = objfile.variables[dimvname]
-        if len(vdobj.shape) != 1:
-            print errormsg
-            print '  ' + fname + ': wrong shape:',vdobj.shape," of variable '" +     \
-              dimvname +  "' !!"
+              "' has any variable '", varname, "' !!"
             quit(-1)
 
@@ -1773 +1803 @@
 
     drw.plot_lines_time(dimvalues, varvalues, valuesaxis, dimtit, leglabels.split(','),   \
-      vartit, varunits, timepos, timelabels, title, locleg, graphk)
+      vartit, varunits, timepos, timelabels, title, locleg, graphk, collines)
 
     return

trunk/tools/drawing_tools.py

@@ -1406 +1406 @@
 
 # Units which does not change
-    same = ['1', 'category', 'day', 'degrees East', 'degrees Nord', 'degrees North', \
-      'g', 'gpm', 'hour', 'hPa', 'K', 'Km', 'kg', 'km', 'm', 'minute', 'mm', 'month', 'Pa', \
-      's', 'second', 'um', 'year', '-']
+    same = ['1', 'category', 'day', 'deg', 'degrees East', 'degrees Nord',           \
+      'degrees North', 'g', 'gpm', 'hour', 'hPa', 'K', 'Km', 'kg', 'km', 'm',        \
+      'minute', 'mm', 'month', 'Pa', 's', 'second', 'um', 'year', '-']
 
     if searchInlist(same,u):
@@ -5086 +5086 @@
 
 def plot_lines_time(vardv, varvv, vaxis, dtit, linesn, vtit, vunit, tpos, tlabs,     \
-  gtit, gloc, kfig):
+  gtit, gloc, kfig, coll):
     """ Function to plot a collection of lines with a time axis
       vardv= list of set of dimension values
@@ -5103 +5103 @@
         9: 'upper center', 10: 'center'
       kfig= kind of figure
+      coll= ',' list of colors for the lines or None for automatic
       plot_lines([np.arange(10)], [np.sin(np.arange(10)*np.pi/2.5)], 'y', 'time (s)',      \
   ['2.5'], 'sin', '-', 'sinus frequency dependency', 'pdf')
@@ -5126 +5127 @@
     if vaxis == 'x':
         for il in range(Ntraj):
-            plt.plot(varvv[il], vardv[il], linekinds[N7lines], label= linesn[il])
+            if coll is None:
+                plt.plot(varvv[il], vardv[il], linekinds[N7lines], label= linesn[il])
+            else:
+                plt.plot(varvv[il], vardv[il], linekinds[N7lines], label= linesn[il],\
+                  color=coll[il])
+
             varTvv = varTvv + list(varvv[il])
             varTdv = varTdv + list(vardv[il])
@@ -5138 +5144 @@
     else:
         for il in range(Ntraj):
-            plt.plot(vardv[il], varvv[il], linekinds[N7lines], label= linesn[il])
+            if coll is None:
+                plt.plot(vardv[il], varvv[il], linekinds[N7lines], label= linesn[il])
+            else:
+                plt.plot(vardv[il], varvv[il], linekinds[N7lines], label= linesn[il],\
+                  color=coll[il])
+
             varTvv = varTvv + list(varvv[il])
             varTdv = varTdv + list(vardv[il])