Changeset 1071 in lmdz_wrf for trunk

Timestamp:

Aug 31, 2016, 6:12:37 PM (9 years ago)

Author:

lfita

Message:

Getting on `draw_2lines'

Location:

trunk/tools

Files:

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

trunk/tools/drawing.py

@@ -4115 +4115 @@
 
     return
+
+def draw_2lines(values,ncfiles,varnames):
+    """ Fucntion to plot two lines in different axes (x/x2 or y/y2)
+      values= [varnA]:[varnB]:[commonvardim]:[varangeA]:[varangeB]:[varangeaxis]:[axisvals]:[figvarns]:[colors]:
+       [widths]:[styles]:[sizemarks]:[marks]:[graphtitle]:[labelaxis]:[lloc]:[figname]:[figkind]
+        [varnA]: name of the variable A in fileA
+        [varnB]: name of the variable B in fileB
+        [commonvardim]: name of the common variable-dimension
+        [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)
+        [varangeaxis]: ',' separated list of range (min,max) for common axis values ('None', automatic; 'Extrs' from 
+          values extremes)
+        [axisvals]: which is the axis to plot the values ('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
+    """
+    function = 'draw_2lines'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print draw_2lines.__doc__
+        quit()
+
+    expectargs = '[varnA]:[varnB]:[commonvardim]:[varangeA]:[varangeB]:' +           \
+     '[varangeaxis]:[axisvals]:[figvarns]:[colors]:[widths]:[styles]:[sizemarks]:' + \
+     '[marks]:[graphtitle]:[labelaxis]:[lloc]:[figname]:[figkind]'
+ 
+    drw.check_arguments(fname,values,expectargs,':')
+
+    varns = [values.split(':')[0], values.split(':')[1]]
+    commonvardim = values.split(':')[2]
+    varangeA0 = values.split(':')[3]
+    varangeB0 = values.split(':')[4]
+    varangeaxis0 = values.split(':')[5]
+    axisvals = values.split(':')[6]
+    figvarns = values.split(':')[7].split(',')
+    colors = gen.str_list(values.split(':')[8],',')
+    widths = gen.str_list(values.split(':')[9],',')
+    styles = gen.str_list(values.split(':')[10],',')
+    sizemarks = gen.str_list(values.split(':')[11],',')
+    marks = gen.str_list(values.split(':')[12],',')
+    graphtitle = values.split(':')[13].replace('!',' ')
+    labelaxis = values.split(':')[14].replace('!',' ')
+    lloc = np.int(values.split(':')[15])
+    figname = values.split(':')[16]
+    figkind = values.split(':')[17]
+
+    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.searchInfile(oncA.variables.kesy(), invarns[0]):
+        print errormsg
+        print '  ' + fname + ": A file '" + files[0] + "' does not have variable '" +\
+          invarns[0] + "' !!"
+        quit(-1)
+    objvA = oncA.variables[invarns[0]]
+    varvalsA = objvA[:]
+    varangeA = []
+
+    if gen.searchInfile(objvA.ncattrs(), 'units'):
+        varunits.append(objva.getncattr('units'))
+    else:
+        valsA = gen.variables_values(invarns[0])
+        varunits.append(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[iv])
+
+    if not gen.searchInfile(oncA.variables.kesy(), commvardim):
+        print errormsg
+        print '  ' + fname + ": A file '" + files[0] + "' does not have common " +   \
+          "dimvar '" + commonvardim + "' !!"
+        quit(-1)
+    objvd = oncA.variables[commonvardim]
+    varvalsaxisA = objvd[:]    
+
+    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.searchInfile(oncB.variables.keys(), invarns[1]):
+        print errormsg
+        print '  ' + fname + ": B file '" + files[1] + "' does not have variable '" +\
+          invarns[1] + "' !!"
+        quit(-1)
+    objvB = oncB.variables[invarns[1]]
+    varvalsB = objvB[:]
+    varangeB = []
+
+    if gen.searchInfile(objvB.ncattrs(), 'units'):
+        varunits.append(objvB.getncattr('units'))
+    else:
+        valsB = gen.variables_values(invarns[1])
+        varunits.append(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[iv])
+
+    if not gen.searchInfile(oncB.variables.keys(), commvardim):
+        print errormsg
+        print '  ' + fname + ": B file '" + files[1] + "' does not have common " +   \
+          "dimvar '" + commonvardim + "' !!"
+        quit(-1)
+    objvd = oncB.variables[commonvardim]
+    varvalsaxisB = objvd[:]
+
+    # Range of the axis
+    varangeaxis = []
+    if gen.searchInfile(objvd.ncattrs(), 'units'):
+        dimvarunits.append(objvd.getncattr('units'))
+    else:
+        valsVD = gen.variables_values(commonvardim)
+        varunits.append(valsVD[5])
+        if varangeaxis0 == 'None':
+            varangeaxis = [valsVD[2], valsVD[3]]
+        elif varangeaxis0 == 'Extrs':
+            varangeaxis[0] = np.min([np.min(varvalsaxisA), np.min(varvalsaxisB)])
+            varangeaxis[1] = np.max([np.max(varvalsaxisA), np.max(varvalsaxisB)])
+        else:
+            for iv in range(2): varangeaxis0[iv] = np.float(varangeaxis[iv])
+    
+    oncB.close()
+
+    plot_2lines(varvalsA, varvalsB, varvalsaxisA, varvalsaxisB, varangeA, varrangeB, \
+      varangeaxis, axisvals, varns, varunits, colors, widths, styles, sizemarks,     \
+      marks, graphtitle, labelaxis, lloc, figname, figkind):
 
 #quit()

trunk/tools/drawing_tools.py

@@ -7443 +7443 @@
     return
 
+def plot_2lines(valsA, valsB, valsaxisA, valsaxisB, rangeA, rangeB, rangeaxis,       \
+  axisvals, varns, varus, cols, wdths, styls, szmks, marks, gtitle, axisn, gloc,     \
+  fign, figk):
+    """ Fucntion to plot two 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
+      rangeaxis= range of values for the common axis
+      axisvals= which is the axis to plot the 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 common axis
+      gloc= location of the legend
+      fign= name of the figure
+      figk= kind of figure
+    """
+    function = 'plot_2lines'
+
+    if axisvals == 'x':
+        titX = axisn
+        titX2 = axisn
+        titY = varns[0] + ' ' + varus[0]
+        titY2 = varns[1] + ' (' + varus[1] + ')'
+        xmin = rangeaxis[0]
+        xmax = rangeaxis[1]
+        xmin2 = rangeaxis[0]
+        xmax2 = rangeaxis[1]
+        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 = rangeaxis[0]
+        ymax = rangeaxis[1]
+        ymin2 = rangeaxis[0]
+        ymax2 = rangeaxis[1]
+
+# From: http://matplotlib.org/examples/api/two_scales.html
+    fig, ax1 = plt.subplots()
+
+    # line A
+    if axisvals == 'x':
+        il=0
+        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)
+
+        ax2 = ax1.twiny()
+        il = il + 1
+        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)
+    else:
+        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)
+        ax2 = ax1.twinx()
+        il = il + 1
+        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)
+        
+    graphtit = gtitle.replace('_','\_').replace('&','\&')
+
+    plt.set_title(graphtit)
+    plt.legend(loc=gloc)
+    
+    output_kind(figk, fign, True)
+
+    return