Changeset 194 in lmdz_wrf for trunk/tools/drawing.py

Timestamp:

Dec 1, 2014, 11:50:52 AM (10 years ago)

Author:

lfita

Message:

Adding 'draw_line_time': script to draw lines with times

File:

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

trunk/tools/drawing.py

@@ -31 +31 @@
   'draw_2D_shad_line_time', 'draw_timeSeries', 'draw_topo_geogrid',                  \
   'draw_topo_geogrid_boxes', 'draw_trajectories', 'draw_vals_trajectories',          \
-  'draw_lines', 'draw_Neighbourghood_evol', 'list_graphics', 'variable_values']
+  'draw_lines', 'draw_lines_time', 'draw_Neighbourghood_evol', 'list_graphics',      \
+  'variable_values']
 
 def draw_2D_shad(ncfile, values, varn):
@@ -1475 +1476 @@
 def draw_lines(ncfilens, values, varname):
     """ Function to draw different lines at the same time from different files
-    draw_trajectories(ncfilens, values, varname):
+    draw_lines(ncfilens, values, varname):
       ncfilens= [filen] ',' separated list of netCDF files
       values= [dimvname]:[valuesaxis]:[dimtit]:[leglabels]:[vtit]:[title]:[graphk]
@@ -1569 +1570 @@
     drw.plot_lines(dimvalues, varvalues, valuesaxis, dimtit, leglabels.split(','),   \
       vartit, varunits, title, graphk)
+
+    return
+
+def draw_lines_time(ncfilens, values, varname):
+    """ 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];[graphk]
+        [dimvname]: name of the variable 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
+        [leglabels]: ',' separated list of names for the legend
+        [vartit]: name of the variable in the graph
+        [title]: title of the plot ('|' for spaces)
+        [timevals]: [timen]|[units]|[kind]|[tfmt] time labels characteristics
+           [timen]; name of the time variable
+           [units]; units string according to CF conventions ([tunits] since 
+             [YYYY]-[MM]-[DD] [[HH]:[MI]:[SS]], '!' for spaces)
+           [kind]; kind of output
+             'Nval': according to a given number of values as 'Nval',[Nval]
+             'exct': according to an exact time unit as 'exct',[tunit]; 
+               tunit= [Nunits],[tu]; [tu]= 'c': centuries, 'y': year, 'm': month,
+                'w': week, 'd': day, 'h': hour, 'i': minute, 's': second, 
+                'l': milisecond
+           [tfmt]; desired format
+        [graphk]: kind of the graphic
+      varname= variable to plot
+      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}$;pdf'
+    """
+
+    fname = 'draw_lines_time'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print draw_lines.__doc__
+        quit()
+
+    expectargs = '[dimvname];[valuesaxis];[dimtit];[leglabels];[vtit];[title];'
+    expectargs = expectargs + '[timevals];[graphk]'
+    drw.check_arguments(fname,len(expectargs.split(';')),values,';',expectargs)
+
+    ncfiles = ncfilens.split(',')
+    dimvname = values.split(';')[0]
+    valuesaxis = values.split(';')[1]
+    dimtit = values.split(';')[2]
+    leglabels = values.split(';')[3].replace('_','\_')
+    vartit = values.split(';')[4]
+    title = values.split(';')[5].replace('|',' ')
+    timevals = values.split(';')[6]
+    graphk = values.split(';')[7]
+
+    Nfiles = len(ncfiles)
+
+# Getting values
+##
+    varvalues = []
+    dimvalues = []
+    timvalues = []
+    timvals0 = timvalues
+
+    print '  ' + fname
+    ifn = 0
+    for ifile in ncfiles:
+        filen = ifile.split('@')[0]
+
+        print '    filen:',filen
+
+        if not os.path.isfile(filen):
+            print errormsg
+            print '  ' + fname + ": netCDF file '" + filen + "' does not exist !!"
+            quit(-1)
+
+        objfile = NetCDFFile(filen, 'r')
+
+        if not objfile.variables.has_key(dimvname):
+            print errormsg
+            print '  ' + fname + ": netCDF file '" + filen +                         \
+              "' does not have variable '" + dimvname + "' !!"
+            quit(-1)
+
+        if not objfile.variables.has_key(varname):
+            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 +  "' !!"
+            quit(-1)
+
+        varvalues.append(vvobj[:])
+        dimvalues.append(vdobj[:])
+
+        timvalues = timvalues + list(vdobj[:])
+
+        if ifn == 0:
+            varunits = vvobj.units
+
+        objfile.close()
+
+        ifn = ifn + 1
+
+# Times
+    timename = timevals.split('|')[0]
+    timeunit = timevals.split('|')[1].replace('!',' ')
+    timekind = timevals.split('|')[2]
+    timefmt = timevals.split('|')[3]
+
+    tvals = list(set(timvalues))
+    tvals.sort()
+
+    timepos, timelabels = drw.CFtimes_plot(tvals, timeunit, timekind, timefmt)
+
+    drw.plot_lines_time(dimvalues, varvalues, valuesaxis, dimtit, leglabels.split(','),   \
+      vartit, varunits, timepos, timelabels, title, graphk)
 
     return
@@ -2275 +2401 @@
 elif oper == 'draw_lines':
     draw_lines(opts.ncfile, opts.values, opts.varname)
+elif oper == 'draw_lines_time':
+    draw_lines_time(opts.ncfile, opts.values, opts.varname)
 elif oper == 'draw_timeSeries':
     draw_timeSeries(opts.ncfile, opts.values, opts.varname)