Changeset 1937 in lmdz_wrf for trunk

Timestamp:

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

Author:

lfita

Message:

Adding:

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

Location:

trunk/tools

Files:

• drawing.py (modified) (9 diffs)
• drawing_tools.py (modified) (3 diffs)

trunk/tools/drawing.py

@@ -17 +17 @@
 import sys
 from cStringIO import StringIO
+import numpy.ma as ma
 
 ## e.g. # drawing.py -f /media/data2/etudes/WRF_LMDZ/WL_HyMeX/IIphase/medic950116/wlmdza/wrfout/wrfout_d01_1995-01-13_00:00:00 -o create_movie -S 'draw_2D_shad#Time@WRFTimes@10@95@191@1#tas:East_West|-1,North_South|-1,Time|2:longitude:latitude:Summer:270.,300.:tas|at|t=0:pdf:None:None' -v T2
@@ -58 +59 @@
 # movie_2D_shad -f '../PY/wrfout_d01_1995-01-01_00:00:00' -S 'tas:west_east|-1,south_north|-1,Time|-1:XLONG:XLAT:auto:rainbow,auto,auto:Srange,Srange:png:None:cyl,l:Time,Times:WRFdate,$%d^{%H}$:15:mp4' -v T2
 ## e.g. # drawing.py -o draw_SkewT -f UWyoming_snd_87576.nc -S 'time|0:auto:auto:Sounding!at!Ezeiza!airport!on!3rd!July!1998:png:yes' -v ta,tda,pres
+## e.g. # drawing.py -o draw_multi_SkewT -f '/home/lluis/estudios/ChemGBsAs/tests/199807/obs/snd/UWyoming_snd_87576.nc:pres|-1,time|0:ta,pres;/home/lluis/estudios/ChemGBsAs/tests/199807/obs/snd/UWyoming_snd_87576.nc:pres|-1,time|0:tda,pres' -S 'auto:auto:multilines!ta,tda!#6464FF,#FF6464!-!,!2:0,auto:Sounding!at!Ezeiza!airport!on!3rd!July!1998:png:yes'
 
 #######
@@ -82 +84 @@
 # draw_lines_time: Function to draw different lines at the same time from different files with times
 # draw_multi_2D_shad: plotting multiple 2D fields with same projection with shading and sharing colorbar
+# draw_multi_SkewT: creation of a SkewT-logP diagram with multiple lines using matplotlib's API example
 # draw_Neighbourghood_evol: Function to draw the temporal evolution of a neighbourghood around a point
 # draw_points: Function to plot a series of points
@@ -114 +117 @@
   'draw_2D_shad_line_time', 'draw_bar', 'draw_bar_line', 'draw_bar_line_time',       \
   'draw_bar_time', 'draw_barbs', 'draw_basins', 'draw_cycle', 'draw_ensembles_time', \
-  'draw_2lines', 'draw_2lines_time', 'draw_multi_2D_shad', 'draw_lines',             \
-  'draw_lines_time',                                                                 \
+  'draw_2lines', 'draw_2lines_time', 'draw_multi_2D_shad', 'draw_multi_SkewT',       \
+  'draw_lines', 'draw_lines_time',                                                   \
   'draw_Neighbourghood_evol',                                                        \
   'draw_points', 'draw_points_lonlat',                                               \
@@ -8913 +8916 @@
         print errormsg
         print '  ' +fname+ ": requires 3 name of variables as '[tavarn],[tdavarn]" + \
-          ",[tavarn]' !!"
+          ",[presvarn]' !!"
         print "    variables provided: '" + varname + "' "
         quit(-1)
@@ -8936 +8939 @@
             print ' ' + fname + ": file '" + ncfile + "' does not have variable '" + \
               varn + "' !!"
-            print '    available ones:', onc.variables-keys()
+            print '    available ones:', onc.variables.keys()
             quit(-1)
 
@@ -8966 +8969 @@
 #varn='ta,tda,pres'
 #draw_SkewT(filen, values, varn)
+
+def draw_multi_SkewT(ncfiles, values):
+    """ creation of a SkewT-logP diagram with multiple lines using matplotlib's API example
+      https://matplotlib.org/examples/api/skewt.html (skewt.py, included as external python)
+    draw_SkewT(ncfile, values, varn)
+      ncfiles = ';' list of [filen]:[dimvals]:[varn],[presvarn] file and variables to use 
+        from each file
+        [dimvals]: ',' list of [dimname]|[value] telling at which dimension of the 
+          variable a given value is required:
+            * [integer]: which value of the dimension
+            * -1: all along the dimension
+            * -9: last value of the dimension
+            * [beg]@[end]@[inc] slice from [beg] to [end] every [inc]
+            * NOTE, no dim name all the dimension size
+        [varn]: name of the variable to plot [C]
+        [presvarn]: name of the variable with the pressures of [varn] [Pa]
+      values = [taminv],[tamaxv]:[presminv],[presmaxv]:[kindgraph]:[legvals]:[figt]:
+        [kindfig]:[close]
+        [taminv],[tamaxv]: minimum and maximum of temperatures to plot [C] 
+          ('auto' for -50,50)
+        [presminv],[presmaxv]: minimum and maximum of pressures to plot [hPa]
+          ('auto' for 100.,1020.)
+        [kindgraph]: kind of graphic to produce. To select from:
+          'multilines'![labels]![collines]![kindlines]![kindmarkers]![widthlines]
+            [labels] = ',' separated list of labels for each file ('None' for no 
+              label)
+            [collines] = ',' separated list of colors ('#RRGGBB' values) of lines for
+              each file (1 value for all the lines the same)
+            [kindmarkers] = ';' separated list of marker types (matplolib values) for
+              each file (1 value for all the lines the same)
+            [kindlines] = ';' separated list of line types (matplolib values) of 
+              lines for each file (1 value for all the lines the same)
+            [widthlines] = ';' separated list of line widths (matplolib values) of 
+              lines for each file (1 value for all the lines the same)
+        [legvals]=[loclegend],[fonstsize] 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)
+        [figt]: title of figure ('!' for spaces)
+        [kindfig]: kind of graphical output (ps, png, pdf)
+        [close]: wether figure should be closed or not
+    """
+    fname = 'draw_SkewT'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print draw_SkewT.__doc__
+        quit()
+
+    availgraph = ['multilines']
+
+    expectargs = '[taminv],[tamaxv]:[presminv],[presmaxv]:[figt]:[kindgraph]:' +     \
+      '[legvals]:[kindfig]:[close]'
+ 
+    drw.check_arguments(fname,values,expectargs,':')
+
+    if values.split(':')[0] != 'auto':
+        tanx = gen.str_list_k(values.split(':')[1], ',', 'np.float')
+    else:
+        tanx = [-50.,50.]
+    if values.split(':')[1] != 'auto':
+        presnx = gen.str_list_k(values.split(':')[2], ',', 'np.float')
+    else:
+        presnx = [100.,1020.]
+    kindgraph = values.split(':')[2]
+    legv = drw.legend_values(values.split(':')[3], ',')
+    figt = values.split(':')[4].replace('!',' ')
+    kindfig = values.split(':')[5]
+    close = gen.Str_Bool(values.split(':')[6])
+
+    filevals = ncfiles.split(';')
+    Nfiles = len(filevals)
+
+    # Getting graphic parameters
+    kindgraphv = kindgraph.split('!')
+    kgraphS = kindgraphv[0]
+    kgraphv = {}
+    if kgraphS == 'multilines':
+        kgraphv['labels'] = kindgraphv[1].split(',')
+        kgraphv['collines'] = gen.str_list_rep(kindgraphv[2], ',', Nfiles)
+        kgraphv['kindlines'] = gen.str_list_rep(kindgraphv[3], ',', Nfiles)
+        kgraphv['kindmarkers'] = gen.str_list_rep(kindgraphv[4], ',', Nfiles)
+        kgraphv['widthlines'] = gen.str_list_rep(kindgraphv[5], ',', Nfiles)
+    else:
+        print errormsg
+        print '  ' + fname + ": graphic type '" + kindgraphv[0] + "' not ready !!"
+        print '    available ones:', availgraph
+        quit(-1)
+
+    vavals = {}
+    presvals = {}
+    ifile = 0
+    # Getting values to plot
+    for fileval in filevals:
+        ncfile = fileval.split(':')[0]
+        dimvals= fileval.split(':')[1].replace('|',':')
+        varname = fileval.split(':')[2]
+
+        # Variables
+        varns = varname.split(',')
+        if len(varns) != 2:
+            print errormsg
+            print '  ' +fname+ ": requires 2 name of variables as '[varn]," +      \
+              "[presvarn]' !!"
+            print "    variables provided: '" + varname + "' "
+            quit(-1)
+
+        onc = NetCDFFile(ncfile, 'r')
+        for varn in varns:
+            if not onc.variables.has_key(varn):
+                print errormsg
+                print ' ' + fname + ": file '" + ncfile + "' does not have " +       \
+                  "variable '" + varn + "' !!"
+                print '    available ones:', onc.variables.keys()
+                quit(-1)
+
+        ov = onc.variables[varns[0]]
+        opres = onc.variables[varns[1]]
+
+        # Getting values
+        vals, valdims = drw.slice_variable(ov, dimvals.replace(',','|'))
+        pvals, presdims = drw.slice_variable(opres, dimvals.replace(',','|'))
+
+        # Expected masked values
+        if type(vals) != type(gen.mamat):
+            vavals[ifile] = ma.masked_array(vals)
+        else:
+            vavals[ifile] = vals
+        if type(presvals) != type(gen.mamat):
+            presvals[ifile] = ma.masked_array(pvals)
+        else:
+            presvals[ifile] = pvals
+        ifile = ifile + 1
+
+    drw.plot_multi_SkewT(vavals, presvals, tanx, presnx, kgraphS, kgraphv, legv,     \
+      figt, kindfig, close)
+
+    onc.close()
+
+    return
+
+#filen='/home/lluis/estudios/ChemGBsAs/tests/199807/obs/snd/UWyoming_snd_87576.nc:pres|-1,time|0:ta,pres;' + \
+#  '/home/lluis/estudios/ChemGBsAs/tests/199807/obs/snd/UWyoming_snd_87576.nc:pres|-1,time|0:tda,pres'
+#values='auto:auto:multilines!ta,tda!#6464FF,#FF6464!-!,!2:0,auto:Sounding!at!Ezeiza!airport!on!3rd!July!1998:png:yes'
+#draw_multi_SkewT(filen, values)
 
 #quit()
@@ -8990 +9140 @@
   'draw_2D_shad_line', 'draw_2D_shad_line_time', 'draw_2lines', 'draw_2lines_time',  \
   'draw_bar', 'draw_bar_line', 'draw_bar_line_time', 'draw_bar_time',                \
-  'draw_cycle', 'draw_ensembles_time', 'draw_multi_2D_shad', 'draw_lines',           \
+  'draw_cycle', 'draw_ensembles_time', 'draw_multi_2D_shad', 'draw_multi_SkewT',     \
+  'draw_lines',                                                                      \
   'draw_lines_time', 'draw_points', 'draw_Taylor', 'draw_time_lag',                  \
   'draw_topo_geogrid_boxes',                                                         \
@@ -9080 +9231 @@
     elif oper == 'draw_multi_2D_shad':
         draw_multi_2D_shad(opts.ncfile, opts.values)
+    elif oper == 'draw_multi_SkewT':
+        draw_multi_SkewT(opts.ncfile, opts.values)
     elif oper == 'draw_points':
         draw_points(opts.ncfile, opts.values)

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