Changeset 1937 in lmdz_wrf for trunk/tools/drawing_tools.py

Timestamp:

Jul 17, 2018, 3:02:28 AM (7 years ago)

Author:

lfita

Message:

Adding:

• `draw_multi_SkewT': creation of a SkewT-logP diagram with multiple lines using matplotlib's API example

File:

• trunk/tools/drawing_tools.py (modified) (3 diffs)

trunk/tools/drawing_tools.py

@@ -1808 +1808 @@
     """ Function to determine the values for the legend
       legstring= char separated list of [locleg][char][fontsize]: 
-          [locleg]: location of the legend (0, autmoatic)
+          [locleg]: 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',
@@ -11497 +11497 @@
 def plot_SkewT(ta, tda, pres, taxtrm, presxtrm, figtitle, kfig, close):
     """ Function to plot a SkewT-logP diagram using matplotlib's API example  
-      ta: air temperature [K]
-      tda: dew point air temperature [K]
+      ta: air temperature [C]
+      tda: dew point air temperature [C]
       pres: pessures for ta and tda [Pa]
       figtitle: title of figure
@@ -11581 +11581 @@
 
     return
+
+def plot_multi_SkewT(tvals, press, taxtrm, presxtrm, kindgS, kindgv, legv, figtitle, 
+  kfig, close):
+    """ Function to plot a SkewT-logP diagram with multiple data using matplotlib's 
+      API example 
+      tvals: dictionary with multiple air temperatures [C]
+      press: dictionary with multiple pessures for ta and tda [Pa]
+      [kindgS]: kind of graphic to produce. To select from:
+          'multilines': multiple lines
+      [kindgv]: dictionary with values for the given [kindgS]
+        * For 'multilines':
+          'labels' = list of labels for each line
+          'collines' = list of colors ('#RRGGBB' values) for each line
+          'kindmarkers' = list of marker types for each line
+          'kindlines' = list of line types (matplolib values) of lines
+          'widthlines' = list of line widths
+      [legv] = [[locleg], [fontsize]] list with the values for the legend
+          [locleg]: 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'
+          [fontsize]: font size for the legend ('auto' for 12)
+      figtitle: title of figure
+      kfig: kind of graphical output (jpg, pdf, png)
+      close: whether figure should be closed or not
+    """
+    import skewt
+
+    fname = 'plot_SkewT'
+    
+    # Create a new figure. The dimensions here give a good aspect ratio
+    fig = plt.figure(figsize=(6.5875, 6.2125))
+    ax = fig.add_subplot(111, projection='skewx')
+
+    plt.grid(True)
+
+    # Quantity of dryadiabats
+    xmax = taxtrm[1]
+    xmin = taxtrm[0]
+    print 'len:', np.arange(xmin,xmax,10.)
+    if len(np.arange(xmin,xmax,10.)) < 10: xfrqtk = 10.
+    else: xfrqtk = 20.
+
+    # Including dry adiabats
+    ddt = int((xmax+xfrqtk-xmin)/xfrqtk)
+    for itt in range(ddt):
+        dryt = thetapotct(xmin+itt*xfrqtk, pmin=10000.)
+        ax.semilogy(dryt[:,1], dryt[:,0]/100., color='#FFCCCC', linewidth=0.75)
+
+    # Including moist- adiabats
+    #for itt in range(ddt):
+    #    moist = ethetapotct(xmin+itt*xfrqtk, pmin=10000.)
+    #    if xmin+itt*xfrqtk == 10.: print moist[:,1], moist[:,0]/100.
+    #    ax.semilogy(moist[:,1], moist[:,0]/100., color='#CCFFCC', linewidth=0.75)   
+
+    # An example of a slanted line at constant X
+    l = ax.axvline(0, color='C0')
+
+    Nlines = len(kindgv['labels'])
+
+    for iline in range(Nlines):
+        ta = tvals[iline]
+        pres = press[iline]
+
+        # line characteristics
+        lline = kindgv['labels'][iline]
+        cline = kindgv['collines'][iline]
+        kmark = kindgv['kindmarkers'][iline]
+        kline = kindgv['kindlines'][iline]
+        wline = kindgv['widthlines'][iline]
+
+        # Checking for variable consistency
+        if len(ta.shape) != 1:
+            print errormsg
+            print '  ' + fname + ": values of 'T[" + str(iline) + "]' must have " +  \
+              "1D-rank!!"
+            print '    passed shape:', ta.shape
+            quit(-1)
+        if len(pres.shape) != 1:
+            print errormsg
+            print '  ' +fname+ ": values of 'pres[" + str(iline) + "]' must have " + \
+              "1D-rank!!"
+            print '    passed shape:', pres.shape
+            quit(-1)
+
+        # Plot the data using normal plotting functions, in this case using
+        # log scaling in Y, as dictated by the typical meteorological plot
+        tamask = ta.mask
+
+        if lline != 'None':
+            ax.semilogy(ta.compressed(), pres[~tamask], kline, color=cline,          \
+              linewidth=wline, marker=kmark, label=lline)
+        else:
+            ax.semilogy(ta.compressed(), pres[~tamask], kline, color=cline,          \
+              linewidth=wline, marker=kmark)
+
+    # Disables the log-formatting that comes with semilogy
+    ax.yaxis.set_major_formatter(ScalarFormatter())
+    ax.yaxis.set_minor_formatter(NullFormatter())
+    ax.set_yticks(np.linspace(100, 1000, 10))
+    ax.set_ylim(presxtrm[1], presxtrm[0])
+
+    ax.xaxis.set_major_locator(MultipleLocator(10))
+    ax.set_xlim(taxtrm[0], taxtrm[1])
+
+    # Indices
+    #plt.annotate('CAPE: ' + str(stvals['CAPE']), xy=(0.98,0.97),                         \
+    #  xycoords='axes fraction', ha="right", backgroundcolor='#DDDDDD')
+    #plt.annotate('CI: '+ str(stvals['CI']), xy=(0.98,0.93), xycoords='axes fraction',    \
+    #  ha='right', backgroundcolor='#DDDDDD')
+
+    plt.legend(loc=legv[0], prop={'size':legv[1]})
+
+    ax.set_title(gen.latex_text(figtitle))
+
+    figname = 'multi_SkewT'
+    output_kind(kfig, figname, close)
+
+    return