Changeset 1077 in lmdz_wrf for trunk

Timestamp:

Sep 1, 2016, 6:47:35 PM (9 years ago)

Author:

lfita

Message:

Adding `draw_2lines_time:' Function to plot two time-lines in different axes (x/x2 or y/y2)

Location:

trunk/tools

Files:

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

trunk/tools/drawing.py

@@ -33 +33 @@
 ## e.g. # drawing.py -o draw_subbasin -f Caceres_subbasin.nc -S 'Caceres:None:cyl,l:2,True:Caceres:pdf:0:Caceres_subbasin'
 ## e.g. # drawing.py -o draw_2lines -f /home/lluis/etudes/WRF_LMDZ/WaquaL_highres/tests/model_graphics/WRF/current/wss_wrfout_tvar_xmean.nc,/home/lluis/etudes/WRF_LMDZ/WaquaL_highres/tests/model_graphics/WRF/current/tas_wrfout_tvar_xmean.nc -v wssvarmean,tasvarmean -S 'lat:0.,20.:0.,4.:-90.,90.:y:wss,tas:red,blue:2.,2.:-:2.,2.:,:wss!tas!mean!meridional!tvar:lon:0:wss_tas_wrfout_tvar_xmean:pdf'
+## e.g. # drawing.py -o draw_2lines_time -f /home/lluis/etudes/WRF_LMDZ/WaquaL_highres/tests/model_graphics/WRF/current/wss_wrfout_xvar_ymean.nc,/home/lluis/etudes/WRF_LMDZ/WaquaL_highres/tests/model_graphics/WRF/current/tas_wrfout_xvar_ymean.nc -v wssvarmean,tasvarmean -S 'time:0.,20.:0.,4.:exct,1,d;%d:x:wss,tas:red,blue:2.,2.:-:2.,2.:,:wss!tas!mean!longitudinal!xvar:time!($[DD]$):0:wss_tas_wrfout_xvar_ymean:pdf'
 
 main = 'drawing.py'
@@ -43 +44 @@
   'draw_2D_shad_cont', 'draw_2D_shad_cont_time', 'draw_2D_shad_line',                \
   'draw_2D_shad_line_time', 'draw_barbs', 'draw_basins',                             \
-  'draw_2lines', 'draw_lines', 'draw_lines_time', 'draw_Neighbourghood_evol',        \
+  'draw_2lines', 'draw_2lines_time', 'draw_lines', 'draw_lines_time',                \
+  'draw_Neighbourghood_evol',                                                        \
   'draw_points', 'draw_points_lonlat',                                               \
   'draw_ptZvals', 'draw_river_desc', 'draw_subbasin', 'draw_timeSeries',             \
@@ -4286 +4288 @@
       linekinds, psizes, pointkinds, graphtitle, labelaxis, lloc, figname, figkind)
 
+
+def draw_2lines_time(ncfiles, values, varnames):
+    """ Function to plot two time-lines in different axes (x/x2 or y/y2)
+      values= [timevardim]:[varangeA]:[varangeB]:[varangeaxis]:[axisvals]:[figvarns]:[colors]:
+       [widths]:[styles]:[sizemarks]:[marks]:[graphtitle]:[labelaxis]:[lloc]:[figname]:[figkind]
+        [timevardim]: name of the common variable-dimension time
+        [varangeA]: ',' separated list of range (min,max) for A values ('None', automatic; 'Extrs' from values extremes)
+        [varangeB]: ',' separated list of range (min,max) for B values ('None', automatic; 'Extrs' from values extremes)
+        [timeaxisfmt]=[tkind];[tfmt]: format of the ticks for the time axis:
+           [kind]: kind of time to appear in the graph
+             '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
+        [timeaxis]: which is the time axis in the plot ('x' or 'y')
+        [figvarns]: ',' separated list of names of the variables in the  plot
+        [colors]: ',' list with color names of the lines for the variables ('None', automatic)
+        [widths]: ',' list with widths of the lines for the variables ('None', automatic)
+        [styles]: ',' list with the styles of the lines ('None', automatic)
+        [sizemarks]: ',' list with the size of the markers of the lines ('None', automatic)
+        [marks]: ',' list with the markers of the lines ('None', automatic)
+        [graphtitle]: title of the figure ('!' for spaces)
+        [labelaxis]: label in the figure of the common axis ('!' for spaces)
+        [lloc]: location of the legend (0, automatic)
+          1: 'upper right', 2: 'upper left', 3: 'lower left', 4: 'lower right',
+          5: 'right', 6: 'center left', 7: 'center right', 8: 'lower center',
+          9: 'upper center', 10: 'center'      kfig= kind of figure
+        [figname]: name of the figure
+        [figkind]: kind of figure
+      ncfiles= ',' separated list of files to use
+      varnames=  ',' separated list of variables names in the files to plot
+    """
+    fname = 'draw_2lines_time'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print draw_2lines_time.__doc__
+        quit()
+
+    expectargs = '[timevardim]:[varangeA]:[varangeB]:[timeaxisfmt]:[timeaxis]:' +    \
+     '[figvarns]:[colors]:[widths]:[styles]:[sizemarks]:[marks]:[graphtitle]:' +     \
+     '[labelaxis]:[lloc]:[figname]:[figkind]'
+ 
+    drw.check_arguments(fname,values,expectargs,':')
+
+    timevardim = values.split(':')[0]
+    varangeA0 = values.split(':')[1]
+    varangeB0 = values.split(':')[2]
+    timeaxisfmt = values.split(':')[3]
+    timeaxis = values.split(':')[4]
+    figvarns = values.split(':')[5].split(',')
+    colors = gen.str_list(values.split(':')[6],',')
+    widths = gen.str_list_k(values.split(':')[7],',','np.float')
+    styles = gen.str_list(values.split(':')[8],',')
+    sizemarks = gen.str_list_k(values.split(':')[9],',','np.float')
+    marks = gen.str_list(values.split(':')[10],',')
+    graphtitle = values.split(':')[11].replace('!',' ')
+    labelaxis = values.split(':')[12].replace('!',' ')
+    lloc = np.int(values.split(':')[13])
+    figname = values.split(':')[14]
+    figkind = values.split(':')[15]
+
+    files = ncfiles.split(',')
+    invarns = varnames.split(',')
+
+    varunits = []
+
+    # Values line A
+    if not os.path.isfile(files[0]):
+        print errormsg
+        print '  ' + fname + ": file '" + files[0] + "' does not exist !!"
+        quit(-1)
+
+    oncA = NetCDFFile(files[0], 'r')
+
+    if not gen.searchInlist(oncA.variables.keys(), invarns[0]):
+        print errormsg
+        print '  ' + fname + ": A file '" + files[0] + "' does not have variable '" +\
+          invarns[0] + "' !!"
+        quit(-1)
+    if not gen.searchInlist(oncA.variables.keys(), timevardim):
+        print errormsg
+        print '  ' + fname + ": A file '" + files[0] + "' does not have time " +     \
+          "variable '" + timevardim + "' !!"
+        quit(-1)
+
+    objvA = oncA.variables[invarns[0]]
+    varvalsA = objvA[:]
+    varangeA = np.zeros((2),dtype=np.float)
+    objtA = oncA.variables[timevardim]
+    timevalsA = objtA[:]
+    trangeA = [np.min(timevalsA), np.max(timevalsA)]
+    tunitsA = objtA.getncattr('units')
+
+    if gen.searchInlist(objvA.ncattrs(), 'units'):
+        varunits.append(drw.units_lunits(objvA.getncattr('units')))
+    else:
+        valsA = gen.variables_values(invarns[0])
+        varunits.append(drw.units_lunits(valsA[5]))
+    if varangeA0 == 'None':
+        varangeA = [valsA[2], valsA[3]]
+    elif varangeA0 == 'Extrs':
+        varangeA = [np.min(varvalsA), np.max(varvalsA)]
+    else:
+        for iv in range(2): varangeA[iv] = np.float(varangeA0.split(',')[iv])
+
+    oncA.close()
+
+    # Values line B
+    if not os.path.isfile(files[1]):
+        print errormsg
+        print '  ' + fname + ": file '" + files[1] + "' does not exist !!"
+        quit(-1)
+
+    oncB = NetCDFFile(files[1], 'r')
+
+    if not gen.searchInlist(oncB.variables.keys(), invarns[1]):
+        print errormsg
+        print '  ' + fname + ": B file '" + files[1] + "' does not have variable '" +\
+          invarns[1] + "' !!"
+        quit(-1)
+    if not gen.searchInlist(oncB.variables.keys(), timevardim):
+        print errormsg
+        print '  ' + fname + ": B file '" + files[1] + "' does not have time " +     \
+          "variable '" + timevardim + "' !!"
+        quit(-1)
+
+    objvB = oncB.variables[invarns[1]]
+    varvalsB = objvB[:]
+    varangeB = np.zeros((2),dtype=np.float)
+    objtB = oncB.variables[timevardim]
+    timevalsB = objtB[:]
+    tunitsB = objtB.getncattr('units')
+
+    if gen.searchInlist(objvB.ncattrs(), 'units'):
+        varunits.append(drw.units_lunits(objvB.getncattr('units')))
+    else:
+        valsB = gen.variables_values(invarns[1])
+        varunits.append(drw.units_lunits(valsB[5]))
+    if varangeB0 == 'None':
+        varangeB = [valsB[2], valsB[3]]
+    elif varangeB0 == 'Extrs':
+        varangeA = [np.min(varvalsA), np.max(varvalsA)]
+    else:
+        for iv in range(2): varangeB[iv] = np.float(varangeB0.split(',')[iv])
+    
+    oncB.close()
+
+    # Time axis taking time units in line A as reference
+    varvalsaxisB = drw.coincident_CFtimes(timevalsB, tunitsA, tunitsB)
+    trangeB = [np.min(varvalsaxisB), np.max(varvalsaxisB)]
+
+    varangeaxis = [np.min([trangeA[0],trangeB[0]]), np.max([trangeA[1],trangeB[1]])]
+
+    timevals = np.arange(varangeaxis[0],varangeaxis[1])
+    tkind = timeaxisfmt.split(';')[0]
+    tformat = timeaxisfmt.split(';')[1]
+    tpos, tlabels = drw.CFtimes_plot(timevals, tunitsA, tkind, tformat)
+
+    # Lines characteristics
+    colvalues, linekinds, pointkinds, lwidths, psizes = drw.ColorsLinesPointsStyles( \
+      2, colors, styles, marks, widths, sizemarks, 'None')
+
+    drw.plot_2lines_time(varvalsA, varvalsB, timevalsA, varvalsaxisB, varangeA,      \
+      varangeB, tpos, tlabels, timeaxis, figvarns, varunits, colvalues, lwidths,     \
+      linekinds, psizes, pointkinds, graphtitle, labelaxis, lloc, figname, figkind)
+
 #quit()
 
@@ -4324 +4495 @@
 # Not checking file operation
 Notcheckingfile = ['draw_2D_shad_cont', 'draw_2D_shad_cont_time',                    \
-  'draw_2D_shad_line', 'draw_2D_shad_line_time', 'draw_2lines', 'draw_lines',        \
+  'draw_2D_shad_line', 'draw_2D_shad_line_time', 'draw_2lines', 'draw_2lines_time',  \
+  'draw_lines',                                                                      \
   'draw_lines_time', 'draw_points', 'draw_topo_geogrid_boxes', 'draw_trajectories',  \
   'draw_vals_trajectories', 'variable_values']
@@ -4362 +4534 @@
 elif oper == 'draw_2lines':
     draw_2lines(opts.ncfile, opts.values, opts.varname)
+elif oper == 'draw_2lines_time':
+    draw_2lines_time(opts.ncfile, opts.values, opts.varname)
 elif oper == 'draw_lines':
     draw_lines(opts.ncfile, opts.values, opts.varname)

trunk/tools/drawing_tools.py

@@ -68 +68 @@
 # plot_2D_shadow_line:
 # plot_2lines: Function to plot two lines in different axes (x/x2 or y/y2)
+# plot_2lines_time: Function to plot two time-lines in different axes (x/x2 or y/y2)
 # plot_lines: Function to plot a collection of lines
 # plot_ZQradii: Function to plot following radial averages only at exact grid poins
@@ -7543 +7544 @@
 
     return
+
+
+def plot_2lines_time(valsA, valsB, valsaxisA, valsaxisB, rangeA, rangeB, valstaxis,  \
+  labelstaxis, taxis, varns, varus, cols, wdths, styls, szmks, marks, gtitle, axisn, \
+  gloc, fign, figk):
+    """ Function to plot two time-lines in different axes (x/x2 or y/y2)
+      valsA= values to be plotted on axis x or y
+      valsB= values to be plotted on axis x2 or y2
+      valsaxisA= values at the common axis for valsA
+      valsaxisB= values at the common axis for valsB
+      rangeA= range of values for valsA
+      rangeB= range of values for valsB
+      valstaxis= values for the tick in the time-axis
+      labelstaxis= labels for the tick in the time-axis
+      taxis= which is the axis for the time values ('x' or 'y')
+      varns= names of the variables in the  plot
+      varus= units of the variables
+      cols= list with color names of the lines for the variables
+      wdths= list with widths of the lines for the variables
+      styls= list with the styles of the lines
+      szmks= list with the size of the markers of the lines
+      marks= list with the markers of the lines
+      gtitle= title of the figure
+      axisn= label in the figure of the time axis
+      gloc= location of the legend
+      fign= name of the figure
+      figk= kind of figure
+    """
+    fname = 'plot_2lines_time'
+
+    if taxis == 'x':
+        titX = axisn
+        titX2 = axisn
+        titY = varns[0] + ' (' + varus[0] + ')'
+        titY2 = varns[1] + ' (' + varus[1] + ')'
+        xmin = np.min(valstaxis)
+        xmax = np.max(valstaxis)
+        ymin = rangeA[0]
+        ymax = rangeA[1]
+        ymin2 = rangeB[0]
+        ymax2 = rangeB[1]
+    else:
+        titX = varns[0] + ' (' + varus[0] + ')'
+        titX2 = varns[1] + ' (' + varus[1] + ')'
+        titY = axisn
+        titY2 = axisn
+        xmin = rangeA[0]
+        xmax = rangeA[1]
+        xmin2 = rangeB[0]
+        xmax2 = rangeB[1]
+        ymin = np.min(valstaxis)
+        ymax = np.max(valstaxis)
+
+# From: http://matplotlib.org/examples/api/two_scales.html
+    fig, ax1 = plt.subplots()
+
+    # Plotting lines
+    il=0
+    if taxis == 'x':
+        lA, = ax1.plot(valsaxisA, valsA, styls[il], marker=marks[il],                \
+          linewidth=wdths[il], markersize=szmks[il], label=varns[il], color=cols[il])
+        ax1.set_xlabel(titX, color='black')
+        ax1.set_ylabel(titY, color=cols[il])
+        ax1.set_xlim(xmin,xmax)
+        ax1.set_ylim(ymin,ymax)
+        ax1.set_xticks(valstaxis)
+        ax1.set_xticklabels(labelstaxis)
+
+        ax2 = ax1.twinx()
+        il = il + 1
+        lB, = ax2.plot(valsaxisB, valsB, styls[il], marker=marks[il],                \
+          linewidth=wdths[il], markersize=szmks[il], label=varns[il], color=cols[il])
+        ax2.set_ylabel(titY2, color=cols[il])
+        ax2.set_ylim(ymin2,ymax2)
+        ax2.set_xlim(xmin,xmax)
+        ax2.set_xticks(valstaxis)
+        ax2.set_xticklabels(labelstaxis)
+        titleloc = (0.5,1.)
+    else:
+        lA, = ax1.plot(valsA, valsaxisA, styls[il], marker=marks[il],                \
+          linewidth=wdths[il], markersize=szmks[il], label=varns[il], color=cols[il])
+        ax1.set_xlabel(titX, color=cols[il])
+        ax1.set_ylabel(titY, color='black')
+        ax1.set_xlim(xmin,xmax)
+        ax1.set_ylim(ymin,ymax)
+        ax1.set_yticks(valstaxis)
+        ax1.set_yticklabels(labelstaxis)
+
+        ax2 = ax1.twiny()
+        il = il + 1
+        lB, = ax2.plot(valsB, valsaxisB, styls[il], marker=marks[il],                \
+          linewidth=wdths[il], markersize=szmks[il], label=varns[il], color=cols[il])
+        ax2.set_xlabel(titX2, color=cols[il])
+        ax2.set_xlim(xmin2,xmax2)
+        ax2.set_ylim(ymin,ymax)
+        ax2.set_yticks(valstaxis)
+        ax2.set_yticklabels(labelstaxis)
+        titleloc = (0.5,1.075)
+        
+#    graphtit = gtitle.replace('_','\_').replace('&','\&')
+    graphtit = gtitle
+
+    plt.title(graphtit,position=titleloc)
+    # plt.legend([lA, lB], loc=gloc)
+    
+    output_kind(figk, fign, True)
+
+    return