Changeset 1277 in lmdz_wrf for trunk

Timestamp:

Nov 8, 2016, 12:15:59 AM (8 years ago)

Author:

lfita

Message:

Improving `draw_Neighbourghood_evol'

Location:

trunk/tools

Files:

• drawing.py (modified) (12 diffs)
• drawing_tools.py (modified) (4 diffs)

trunk/tools/drawing.py

@@ -29 +29 @@
 ## e.g. # $ drawing.py -f WRF/current/hurs_wrfout_tturb_xmean_last.nc,WRF/micro1/hurs_wrfout_tturb_xmean_last.nc,WRF/micro2/hurs_wrfout_tturb_xmean_last.nc,WRF_LMDZ/AR40/hurs_wrfout_tturb_xmean_last.nc,WRF_LMDZ/NPv31/hurs_wrfout_tturb_xmean_last.nc,LMDZ/AR40/hurs_reproj-histins_tturb_xmean_last.nc,LMDZ/NPv31/hurs_reproj-histins_tturb_xmean_last.nc -o draw_lines -S 'lat:x:lat ($degrees\ North$):auto:wcurr,wmp1,wmp2,wlmdza,wlmdzb,lmdza,lmdzb:hurs:all model-experiments meridional hurs$_{[tturb\ xmean\ last]}$:0|auto:None:-:,:2.:2.:all:Nlines_hurs_wrfout_tturb-xmean-last:png:True' -v hursturbmean
 ## e.g. # drawing.py -o draw_lines_time -f wrfout_d01_2001-11-11_00:00:00_west_east_B20-E20-I1_south_north_B20-E20-I1.nc,wrfout_d01_2001-11-11_00:00:00_west_east_B25-E25-I1_south_north_B25-E25-I1.nc,wrfout_d01_2001-11-11_00:00:00_west_east_B35-E35-I1_south_north_B35-E35-I1.nc -S 'WRFtime;y;time ([DD]${[HH]}$);auto;we=20$\times$sn=20,we=25$\times$sn=25,we=35$\times$sn=35;tas;tas|evolution|at|3|different|grid|points;None;time|hours!since!1949-12-01_00:00:00|exct,3,h|%d$^{%H}$;0|12;pdf;-;r,g,b;.;2.;2.;all;-1;True' -v T2
+## e.g. # drawing.py -o draw_Neighbourghood_evol -S 'vas:Time|-1|WRFtime,south_north|44|XLAT,west_east|88|XLONG:south_north,west_east:5:auto:time|($[DD]^{[HH]}$),time|($[DD]^{[HH]}$):exct,1,h|exct,3,h:$%d^{%H}$,$%d^{%H}$:5|pts|neighbourghood|temporal|evolution|on|2001|Nov.|at|x=88,|y=44:0.0,20.:rainbow,auto,auto:auto:png:vas_Neigh_evol:True' -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -v V10
 
 
@@ -2666 +2667 @@
       filen= netCDF file name
       values= [gvarname]:[dimsval]:[neigdims]:[Nneig]:[Ncol]:[timetits]:[tkinds]:
-       [timefmts]:[gtitle]:[shadxtrms]:[cbar]:[gkind]:[ofile]
+       [timefmts]:[gtitle]:[shadxtrms]:[colobarvals]:[neighlinevals]:[gkind]:[ofile]:[close]
+        [gvarname]: ':' list of names of the variables in the plot
         [dimsval]: [dimn1]|[val1]|[dimv1],...,[dimnN]|[valN]|[dimvN] dimension names, values to get 
           (-1, for all; no name/value pair given full length) and variable with values of the dimension
+          'WRFtime' for WRF times
           NOTE: when dimsval[X,Y] == neigdims[X,Y], valX,valY --> valX,valY-Nneig/2, valX,valY+Nneig/2
         [neigdims]: [dimnX],[dimnY] dimensions mnames along which the neigbourghood should be defined
@@ -2683 +2686 @@
         [timefmts]: [tfmtX],[tfmtY] format of the time labels
         [gtitle]: title of the graphic ('|' for spaces)
-        [shadxtrms]: Extremes for the shading
-        [cbar]: colorbar to use
+        [shadxtrms]: minimum and maximum value for the shading or:
+          'Srange': for full range
+          'Saroundmean@val': for mean-xtrm,mean+xtrm where xtrm = np.min(mean-min@val,max@val-mean)
+          'Saroundminmax@val': for min*val,max*val
+          'Saroundpercentile@val': for median-xtrm,median+xtrm where xtrm = np.min(median-percentile_(val),
+            percentile_(100-val)-median)
+          'Smean@val': for -xtrm,xtrm where xtrm = np.min(mean-min*@val,max*@val-mean)
+          'Smedian@val': for -xtrm,xtrm where xtrm = np.min(median-min@val,max@val-median)
+          'Spercentile@val': for -xtrm,xtrm where xtrm = np.min(median-percentile_(val),
+             percentile_(100-val)-median)
+        [colorbarvals]=[colbarn],[fmtcolorbar],[orientation] characteristics of the colormap and colorbar
+          [colorbarn]: name of the color bar
+          [fmtcolorbar]: format of the numbers in the color bar 'C'-like ('auto' for %6g)
+          [orientation]: orientation of the colorbar ('vertical' (default, by 'auto'), 'horizontal')
+        [neighlinevals]=[linecol],[linestyle],[linewidth] characterisitcs of the lines to mark the limits of the neighborhood
+          ('auto' for: ['#646464', '-', 2.])
+          [linecol]: color of the line
+          [linestyle]: style of the line
+          [linewidth]: width of the line
         [gkind]: kind of graphical output
         [ofile]: True/False whether the netcdf with data should be created or not
+        [close]: Whether figure should be finished or not
       variable= name of the variable
       values = 'q:Time|-1|Times,bottom_top|6|ZNU,south_north|3|XLAT,west_east|26|XLONG:south_north,west_east:5:auto:time|($[DD]^{[HH]}$),time|($[DD]^{[HH]}$):exct,2,h|exct,1,d:$%d^{%H}$,$%d^{%H}$:5|pts|neighbourghood|temporal|evolution:0.0,0.004:BuPu:pdf:True'
@@ -2699 +2720 @@
 
     expectargs = '[gvarname]:[dimsval]:[neigdims]:[Nneig]:[Ncol]:' +                 \
-      '[timetits]:[tkinds]:[timefmts]:[gtitle]:[shadxtrms]:[cbar]:[gkind]:[ofile]'
+      '[timetits]:[tkinds]:[timefmts]:[gtitle]:[shadxtrms]:[colorbarvals]:' +        \
+      '[neighlinevals]:[gkind]:[ofile]:[close]'
  
     drw.check_arguments(fname,values,expectargs,':')
@@ -2713 +2735 @@
     gtitle = values.split(':')[8].replace('|',' ')
     shadxtrms = values.split(':')[9].split(',')
-    cbar = values.split(':')[10]
-    gkind = values.split(':')[11]
-    ofile = values.split(':')[12]
+    colorbarvals = values.split(':')[10]
+    neighlinevals = values.split(':')[11]
+    gkind = values.split(':')[12]
+    ofile = values.split(':')[13]
+    close = gen.Str_Bool(values.split(':')[14])
 
     if Ncol0 != 'auto': 
@@ -2727 +2751 @@
     if np.mod(Nneig,2) == 0:
         print errormsg
-        print '  ' + fname + ": an odd value for 'Nneig':",Nneig,'is required !!!'
+        print '  ' + fname + ": an odd value for 'Nneig':", Nneig, 'is required !!!'
         quit(-1)
 
@@ -2756 +2780 @@
             slicevar.append(slice(0, dimsize+1))
             newdimn.append(dimn)
-            newdimsvar[dimn] = dimseize
+            newdimsvar[dimn] = dimsize
 
         for dimslicen in dimvslice.keys():
@@ -2795 +2819 @@
             dimtime = dim1
 
+    # Number of columns and rows
     if Ncol == 'auto':
-        dimtsqx = int(np.sqrt(dimt)) + 1
-        dimtsqy = int(np.sqrt(dimt)) + 1
+        dimtsqx = int(np.sqrt(dimt))
     else:
         dimtsqx = int(Ncol)
-        dimtsqy = dimt/dimtsqx + 1
+
+    dimtsqy = dimt/dimtsqx + 1
+    print '  ' + fname + '; plotting ', dimtsqx, 'x', dimtsqy, 'time-windows of:',   \
+      Nneig, 'x', Nneig, 'grid-points'
 
     neighbourghood = np.ones((dimtsqy*Nneig,dimtsqx*Nneig), dtype=np.float)*fillValue
@@ -2821 +2848 @@
 
 # Time values at the X/Y axes
-    if ncobj.variables[dimvvalues[dimtime]].dtype == '|S1':
-        print '    ' + fname + ': WRF time variable!'
+    if dimvvalues[dimtime] == 'WRFtime':
+        print '    ' + fname + ": WRF time variable!: 'Times'"
         refdate = '19491201000000'
         tunitsval = 'hours'
         dimtvalues = np.zeros((dimt), dtype=np.float)
-        tvals = ncobj.variables[dimvvalues[dimtime]]
+        tvals = ncobj.variables['Times']
         yrref=refdate[0:4]
         monref=refdate[4:6]
@@ -2868 +2895 @@
         newnc.close()
         print fname + ": Successfull generation of file '" + ofilen + "' !!"
+# Colorbar values
+    colbarn, fmtcolbar, colbaror = drw.colorbar_vals(colorbarvals,',')
+    colormapv = [colbarn, fmtcolbar, colbaror]
+
+# Neighborhood line values
+    if neighlinevals == 'auto':
+        neiglinev = ['#646464', '-', 2.]
+    else:
+        neiglinev = neighlinevals.split(',')
 
 # Time ticks
@@ -2881 +2917 @@
 
     drw.plot_Neighbourghood_evol(neighbourghood, dimxv, dimyv, gvarname, timetits,   \
-      timepos, timelabels, cbar, Nneig, shadxtrms, vunits, gtitle, gkind, True)
+      timepos, timelabels, colormapv, neiglinev, Nneig, shadxtrms, vunits, gtitle,   \
+      gkind, close)
 
 def draw_points(filen, values):
@@ -5053 +5090 @@
     varn=opts.varname
     oper=opts.operation
+
+    if opts.operation is None:
+        print errormsg
+        print '  No operation provided !!'
+        print "  an operation must be provided as '-o [operationname]' "
+        quit(-1)
 
     if opts.ncfile is not None and not os.path.isfile(opts.ncfile) and                   \

trunk/tools/drawing_tools.py

@@ -5560 +5560 @@
     return
 
-def plot_Neighbourghood_evol(varsv, dxv, dyv, vnames, ttits, tpos, tlabels, colorbar, \
-  Nng, vs, uts, gtit, kfig, ifclose):
+def plot_Neighbourghood_evol(varsv, dxv, dyv, vnames, ttits, tpos, tlabels, cbarv,   \
+  neighv, Nng, vs, uts, gtit, kfig, ifclose):
     """ Plotting neighbourghood evolution
       varsv= 2D values to plot with shading
@@ -5569 +5569 @@
       tpos= positions of the time ticks
       tlabels= labels of the time ticks
-      colorbar= name of the color bar to use
+      cbarv= list with the parameters of the color bar [colorbar, cbarfmt, cbaror]
+        colorbar: name of the color bar to use
+        cbarfmt: format of the numbers in the colorbar
+        cbaror: orientation of the colorbar
+      neighv = list with the parameters for the lines of the limits of the neighborhood [color, style, width]
       Nng= Number of grid points of the full side of the box (odd value)
       vs= minmum and maximum values to plot in shadow or:
@@ -5671 +5675 @@
 
 #    plt.pcolormesh(dxv, dyv, varsv, cmap=plt.get_cmap(colorbar), vmin=vsend[0], vmax=vsend[1])
-    plt.pcolormesh(varsvmask, cmap=plt.get_cmap(colorbar), vmin=vsend[0], vmax=vsend[1])
-    cbar = plt.colorbar()
-
-    newtposx = (tpos[0][:] - np.min(dxv)) * len(dxv) * Nng / (np.max(dxv) - np.min(dxv))
-    newtposy = (tpos[1][:] - np.min(dyv)) * len(dyv) * Nng / (np.max(dyv) - np.min(dyv))
-
-    plt.xticks(newtposx, tlabels[0])
-    plt.yticks(newtposy, tlabels[1])
+    plt.pcolormesh(varsvmask, cmap=plt.get_cmap(cbarv[0]), vmin=vsend[0], vmax=vsend[1])
+    if cbarv[2] == 'horizontal':
+        cbar = plt.colorbar(format=cbarv[1],orientation=cbarv[2])
+        # From: http://stackoverflow.com/questions/32050030/rotation-of-colorbar-tick-labels-in-matplotlib
+        ticklabels= cbar.ax.get_xticklabels()
+        Nticks = len(ticklabels)
+        ticklabs = []
+        for itick in range(Nticks): ticklabs.append(ticklabels[itick].get_text())
+        cbar.ax.set_xticklabels(ticklabs,rotation=90)
+    else:
+        cbar = plt.colorbar(format=cbarv[1],orientation=cbarv[2])
+
+    newtposx = (tpos[0][:] - np.min(dxv)) * (len(dxv)-1) * Nng / (np.max(dxv) -      \
+      np.min(dxv))
+    newtposy = (tpos[1][:] - np.min(dyv)) * (len(dyv)-1) * Nng / (np.max(dyv) -      \
+      np.min(dyv))
+
+    plt.xlim(np.min(newtposx), np.max(newtposx))
+    plt.ylim(np.min(newtposy), np.max(newtposy)+Nng)
+
+    # Plotting neighborhood boundaries
+    Nng2 = Nng/2.
+    for iy in range(len(dyv)):
+        plt.plot([np.min(newtposx),np.max(newtposx)], [newtposy[iy],newtposy[iy]],   \
+          linestyle=neighv[1], color=neighv[0], linewidth=neighv[2])
+    for ix in range(len(dxv)):
+        plt.plot([newtposx[ix],newtposx[ix]],[np.min(newtposy),np.max(newtposy)+Nng],\
+          linestyle=neighv[1], color=neighv[0], linewidth=neighv[2])
+    # Plotting center
+    for iy in range(len(dyv)):
+        for ix in range(len(dxv)):
+            plt.plot([newtposx[ix]+Nng2,newtposx[ix]+Nng2], [newtposy[iy]+Nng2-0.125,\
+              newtposy[iy]+Nng2+0.125], linestyle=neighv[1], color=neighv[0],        \
+              linewidth=neighv[2])
+            plt.plot([newtposx[ix]+Nng2-0.125,newtposx[ix]+Nng2+0.125],              \
+              [newtposy[iy]+Nng2,newtposy[iy]+Nng2], linestyle=neighv[1],            \
+              color=neighv[0], linewidth=neighv[2])
+            
+    plt.xticks(newtposx+Nng/2., tlabels[0])
+    plt.yticks(newtposy+Nng/2., tlabels[1])
     plt.xlabel(ttits[0])
     plt.ylabel(ttits[1])
@@ -5688 +5724 @@
 
 # units labels
-    cbar.set_label(vnames.replace('_','\_') + ' (' + units_lunits(uts) + ')')
+    cbar.set_label(gen.latex_text(vnames) + ' (' + units_lunits(uts) + ')')
 
     figname = 'Neighbourghood_evol'
-    graphtit = gtit.replace('_','\_').replace('&','\&')
+    graphtit = gen.latex_text(gtit)
 
     plt.title(graphtit, position=(0.5,1.05))