Changeset 194 in lmdz_wrf for trunk

Timestamp:

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

Author:

lfita

Message:

Adding 'draw_line_time': script to draw lines with times

Location:

trunk/tools

Files:

• drawing.py (modified) (4 diffs)
• drawing_tools.py (modified) (1 diff)

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)

trunk/tools/drawing_tools.py

@@ -4459 +4459 @@
 
     return
+
+def plot_lines_time(vardv, varvv, vaxis, dtit, linesn, vtit, vunit, tpos, tlabs,     \
+  gtit, kfig):
+    """ Function to plot a collection of lines with a time axis
+      vardv= list of set of dimension values
+      varvv= list of set of values
+      vaxis= which axis will be used for the time values ('x', or 'y')
+      dtit= title for the common dimension
+      linesn= names for the legend
+      vtit= title for the vaxis
+      vunit= units of the vaxis
+      tpos= positions of the time ticks
+      tlabs= labels of the time ticks
+      gtit= main title
+      kfig= kind of figure
+      plot_lines([np.arange(10)], [np.sin(np.arange(10)*np.pi/2.5)], 'y', 'time (s)',      \
+  ['2.5'], 'sin', '-', 'sinus frequency dependency', 'pdf')
+    """
+    fname = 'plot_lines'
+
+    if vardv == 'h':
+        print fname + '_____________________________________________________________'
+        print plot_lines.__doc__
+        quit()
+
+# Canging line kinds every 7 lines (end of standard colors)
+    linekinds=['.-','x-','o-']
+
+    Ntraj = len(vardv)
+
+    N7lines = 0
+
+    plt.rc('text', usetex=True)
+
+    if vaxis == 'x':
+        for il in range(Ntraj):
+            plt.plot(varvv[il], vardv[il], linekinds[N7lines], label= linesn[il])
+            if il == 6: N7lines = N7lines + 1
+
+        plt.xlabel(vtit + ' (' + vunit + ')')
+        plt.ylabel(dtit)
+        plt.xlim(np.min(varvv[:]),np.max(varvv[:]))
+        plt.ylim(np.min(vardv[:]),np.max(vardv[:]))
+        plt.yticks(tpos, tlabs)
+    else:
+        for il in range(Ntraj):
+            plt.plot(vardv[il], varvv[il], linekinds[N7lines], label= linesn[il])
+            if il == 6: N7lines = N7lines + 1
+
+        plt.xlabel(dtit)
+        plt.ylabel(vtit + ' (' + vunit + ')')
+        plt.xlim(np.min(vardv[:]),np.max(vardv[:]))
+        plt.ylim(np.min(varvv[:]),np.max(varvv[:]))
+        plt.xticks(tpos, tlabs)
+
+    figname = 'lines_time'
+    graphtit = gtit
+
+    plt.title(graphtit)
+    plt.legend()
+    
+    output_kind(kfig, figname, True)
+
+    return