Changeset 1283 in lmdz_wrf

Timestamp:

Nov 8, 2016, 4:54:42 PM (8 years ago)

Author:

lfita

Message:

Improving `draw_points' with dimxyfmt, slice capabilities and colobarvalues

Location:

trunk/tools

Files:

• drawing.py (modified) (13 diffs)
• drawing_tools.py (modified) (6 diffs)

trunk/tools/drawing.py

@@ -30 +30 @@
 ## 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
-
+## e.g. # drawing.py -o draw_points -S 'SuperStorm/tslist.dat,#,3,2,1:SuperStorm|sfc|stations:auto:cyl,i:labelled,10,r:auto:None:0:png:stations_loc:True' -f 'geo_em.d02.nc,XLONG_M,XLAT_M,HGT_M,Time|0@west_east|30;180;1@south_north|175;255;1,height,0.,1500.,terrain,auto,auto,m'
 
 ## e.g. # drawing.py -o draw_trajectories -f 'WRF/control/trajectory.dat@-1@/home/lluis/etudes/domains/WL_HyMeX_HighRes_C/geo_em.d03.nc#XLONG_M#XLAT_M,WRF_LMDZ/wlmdza/trajectory.dat@-1@/home/lluis/etudes/domains/WL_HyMeX_C/geo_em.d01.nc#XLONG_M#XLAT_M,WRF_LMDZ/wlmdzb/trajectory.dat@-1@/home/lluis/etudes/domains/WL_HyMeX_C/geo_em.d01.nc#XLONG_M#XLAT_M,WRF_LMDZ/wlmdzb_ii/trajectory.dat@-1@/home/lluis/etudes/domains/WL_HyMeX_C/geo_em.d01.nc#XLONG_M#XLAT_M' -S '$WRF_{CRM}$,$LMDZ_{AR4.0}$,$LMDZ_{NPv3.1}$,$LMDZ_{NPv3.1b}$|None|medicane trajectories|pdf|cyl,i' -v obs/trajectory.dat,satellite,-1
@@ -2699 +2699 @@
        [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
+        [dimsval]: [dimn1]|[val1]|[dimv1],...,[dimnN]|[valN]|[dimvN] dimension names, values to get:
+          * [integer]: which value of the dimension
+          * -1: all along the dimension
+          * NOTE, no dim name all the dimension size
           'WRFtime' for WRF times
           NOTE: when dimsval[X,Y] == neigdims[X,Y], valX,valY --> valX,valY-Nneig/2, valX,valY+Nneig/2
@@ -2951 +2953 @@
 
 def draw_points(filen, values):
-    """ Function to plot a series of points
-      [values]= [ptasciifile]:[gtit]:[mapvalues]:[kindfigure]:[pointcolor]:[pointlabels]:
-        [legvals]:[figureko]:[figuren]
+    """ Function to plot a series of points read from an ASCII file with lon, lat, label
+      draw_points(filen, values)
+      [values]= [ptasciifile]:[gtit]:[dimxyfmt]:[mapvalues]:[kindfigure]:[pointcolor]:[pointlabels]:
+        [legvals]:[figureko]:[figuren]:[close]
         [ptasciifile]:[file],[comchar],[collon],[collat],[lab]
           [file]: column ASCII file with the location of the points
@@ -2962 +2965 @@
             the values from the [pointlabels] variable
         [gtit]: title of the figure ('|' for spaces)
+        [dimxyfmt]: [dxs],[dxf],[Ndx],[ordx],[dys],[dyf],[Ndy],[ordx]: format of the values at each axis (or 'auto')
+          [dxs]: style of x-axis ('auto' for 'pretty')
+            'Nfix', values computed at even 'Ndx'
+            'Vfix', values computed at even 'Ndx' increments
+            'pretty', values computed following aprox. 'Ndx' at 'pretty' intervals (2.,2.5,4,5,10)
+          [dxf]: format of the labels at the x-axis ('auto' for '%5g')
+          [Ndx]: Number of ticks at the x-axis ('auto' for 5)
+          [ordx]: angle of orientation of ticks at the x-axis ('auto' for horizontal)
+          [dys]: style of y-axis ('auto' for 'pretty')
+          [dyf]: format of the labels at the y-axis ('auto' for '%5g')
+          [Ndy]: Number of ticks at the y-axis ('auto' for 5)
+          [ordy]: angle of orientation of ticks at the y-axis ('auto' for horizontal)
         [mapvalues]: drawing coastaline ([proj],[res]) or None
           [proj]: projection
@@ -2976 +2991 @@
           'labelled',[txtsize],[txtcol]: points with the names and size, color of text
         [pointcolor]: color for the points ('auto' for "red")
-        [pointlabels]: ',' of labels [only used if [collab]='None'] ('None' for no labels)
+        [pointlabels]: ',' of labels [only used if [collab]='None'] ('None' for no labels, '!' for spaces)
         [legvals]=[locleg]|[fontsize]: 
           [locleg]: location of the legend (0, autmoatic)
@@ -2985 +3000 @@
         [figureko]: kind of the output file (pdf, png, ...)
         [figuren]: name of the figure
-      [filen]= [ncfile],[lonvarn],[latvarn][,[varn],[dimvals],[vargn],[min],[max],[cbar],[varu]]
+        [close]: Whether figure should be finished or not
+      [filen]= [ncfile],[lonvarn],[latvarn][,[varn],[dimvals],[vargn],[min],[max],[colbarn],[fmtcolorbar],[orientation],[varu]]
         [ncfile]: netCDF to use to geolocalize the points
         [lonvarn]: name of the variable with the longitudes
         [latvarn]: name of the variable with the latitudes
-        [varn]: optional variable to add staff into the graph
-        [dimval]: '@' list of [dimn]|[dimval] to get the values for [varn]
-          [dimn]: name of the dimension
-          [dimval]: value of the dimension 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
-        [vargn]: name of the variable in the graph
-        [min]: minimum value for the extra variable
-        [max]: maximum value for the extra variable
-        [cbar]: color bar
-        [varu]: units of the variable
+        Optional values:
+          [varn]: optional variable to add staff into the graph
+          [dimval]: '@' list of [dimn]|[dimval] to get the values for [varn]
+            [dimn]: name of the dimension
+            [dimval]: value of the dimension 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
+          [vargn]: name of the variable in the graph
+          [min]: minimum value for the extra variable
+          [max]: maximum value for the extra variable
+          [colbarn]: 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')
+          [varu]: units of the variable
     """
     fname = 'draw_points'
@@ -3011 +3030 @@
         quit()
 
-    expectargs = '[ptasciifile]:[gtit]:[mapvalues]:[kindfigure]:[pointcolor]:' +     \
-      '[pointlabels]:[locleg]:[figurek]:[figuren]'
+    expectargs = '[ptasciifile]:[gtit]:[dimxyfmt]:[mapvalues]:[kindfigure]:' +       \
+      '[pointcolor]:[pointlabels]:[locleg]:[figurek]:[figuren]:[close]'
  
     drw.check_arguments(fname,values,expectargs,':')
@@ -3018 +3037 @@
     ptasciifile = values.split(':')[0]
     gtit = values.split(':')[1]
-    mapvalues = values.split(':')[2]
-    kindfigure = values.split(':')[3]
-    pointcolor = values.split(':')[4]
-    pointlabels = values.split(':')[5]
-    locleg = int(values.split(':')[6])
-    figureko = values.split(':')[7]
-    figuren = values.split(':')[8]
+    dimxyfmt = values.split(':')[2]
+    mapvalues = values.split(':')[3]
+    kindfigure = values.split(':')[4]
+    pointcolor = values.split(':')[5]
+    pointlabels = values.split(':')[6].replace('!',' ')
+    locleg = int(values.split(':')[7])
+    figureko = values.split(':')[8]
+    figuren = values.split(':')[9]
+    close = gen.Str_Bool(values.split(':')[10])
 
 # Getting station locations
@@ -3074 +3095 @@
     varn = None
     varextrav = None
-    if Nvals == 10:
+    if Nvals == 12:
         varn = filev[3]
         dimvals = filev[4]
         varextrav = [filev[5], np.float(filev[6]), np.float(filev[7]), filev[8],     \
-          filev[9]]
+          filev[9], filev[10], filev[11]]
+        if filev[9] == 'auto': varextrav[4] = '%6g'
+        if filev[10] == 'auto': varextrav[5] = 'vertical'
     
     if not os.path.isfile(ncfile):
@@ -3087 +3110 @@
     onc = NetCDFFile(ncfile, 'r')
     
-    lonv, latv = drw.lonlat2D(onc.variables[lonvarn], onc.variables[latvarn])
+    # Slicing lon,lats
+    dd = {}
+    for dn in dimvals.split('@'):
+        ddn = dn.split('|')[0]
+        ddv = dn.split('|')[1]
+        dd[ddn] = ddv
+
+    objlon = onc.variables[lonvarn]
+    vard = objlon.dimensions
+    slicevar = []
+    for dv in vard:
+        found= False
+        for dn in dd.keys():
+            if dn == dv:
+                if dd[dn].find(';') == -1:
+                    if dd[dn] == '-1':
+                        slicevar.append(slice(0,len(onc.dimensions[dv])))
+                    elif dd[dn] == '-9':
+                        slicevar.append(len(onc.dimensions[dv]))
+                    else:
+                        slicevar.append(int(dd[dn]))
+                else:
+                    islc = int(dd[dn].split(';')[0])
+                    eslc = int(dd[dn].split(';')[1])
+                    tslc = int(dd[dn].split(';')[2])
+                    slicevar.append(slice(islc,eslc,tslc))
+                found = True
+                break
+        if not found:
+            slicevar.append(slice(0,len(onc.dimensions[dv])))
+    lonvals = np.squeeze(objlon[tuple(slicevar)])
+
+    objlat = onc.variables[latvarn]
+    vard = objlat.dimensions
+    slicevar = []
+    for dv in vard:
+        found= False
+        for dn in dd.keys():
+            if dn == dv:
+                if dd[dn].find(';') == -1:
+                    if dd[dn] == '-1':
+                        slicevar.append(slice(0,len(onc.dimensions[dv])))
+                    elif dd[dn] == '-9':
+                        slicevar.append(len(onc.dimensions[dv]))
+                    else:
+                        slicevar.append(int(dd[dn]))
+                else:
+                    islc = int(dd[dn].split(';')[0])
+                    eslc = int(dd[dn].split(';')[1])
+                    tslc = int(dd[dn].split(';')[2])
+                    slicevar.append(slice(islc,eslc,tslc))
+                found = True
+                break
+        if not found:
+            slicevar.append(slice(0,len(onc.dimensions[dv])))
+    latvals = np.squeeze(objlat[tuple(slicevar)])
+
+    lonv, latv = drw.lonlat2D(lonvals, latvals)
 
     if varn is not None:
         objextra = onc.variables[varn]
         vard = objextra.dimensions
-        dd = {}
-        for dn in dimvals.split('@'):
-            ddn = dn.split('|')[0]
-            ddv = dn.split('|')[1]
-            dd[ddn] = ddv
         slicevar = []
         for dv in vard:
@@ -3102 +3177 @@
             for dn in dd.keys():
                 if dn == dv:
-                    slicevar.append(int(dd[dn]))
+                    if dd[dn].find(';') == -1:
+                        if dd[dn] == '-1':
+                            slicevar.append(slice(0,len(onc.dimensions[dv])))
+                        elif dd[dn] == '-9':
+                            slicevar.append(len(onc.dimensions[dv]))
+                        else:
+                            slicevar.append(int(dd[dn]))
+                    else:
+                        islc = int(dd[dn].split(';')[0])
+                        eslc = int(dd[dn].split(';')[1])
+                        tslc = int(dd[dn].split(';')[2])
+                        slicevar.append(slice(islc,eslc,tslc))
                     found = True
                     break
@@ -3110 +3196 @@
         varextra = np.squeeze(objextra[tuple(slicevar)])
 
+    # Axis values
+    xstyl, xaxf, Nxax, xaxor, ystyl, yaxf, Nyax, yaxor = drw.format_axes(dimxyfmt,',')
+    xaxis = [xstyl, xaxf, Nxax, xaxor]
+    yaxis = [ystyl, yaxf, Nyax, yaxor]
+
     if mapvalues == 'None':
         mapV = None
@@ -3115 +3206 @@
         mapV = mapvalues
 
-    drw.plot_points(xptval, yptval, lonv, latv, varextra, varextrav, gtit, mapV,     \
-      kindfigure, pointcolor, ptlabels, locleg, figureko, figuren)
+    drw.plot_points(xptval, yptval, lonv, latv, varextra, varextrav, gtit, xaxis,    \
+      yaxis, mapV, kindfigure, pointcolor, ptlabels, locleg, figureko, figuren, close)
 
     onc.close()

trunk/tools/drawing_tools.py

@@ -3021 +3021 @@
 #quit()
 
-def plot_points(xval, yval, vlon, vlat, extravals, extrapar, vtit, mapv, figk, color,\
-  labels, lloc, kfig, figname):
+def plot_points(xval, yval, vlon, vlat, extravals, extrapar, vtit, xaxv, yaxv, mapv, \
+  figk, color, labels, lloc, kfig, figname, ifclose):
     """ plotting points
       [x/yval]: x,y values to plot
@@ -3030 +3030 @@
       extrapar= [varname, min, max, cbar, varunits] of the extra variable
       vtit= title of the graph ('|' for spaces)
+      xaxv= list with the x-axis paramteres [style, format, number and orientation]
+      yaxv= list with the y-axis paramteres [style, format, number and orientation]
       mapv= map characteristics: [proj],[res]
         see full documentation: http://matplotlib.org/basemap/
@@ -3049 +3051 @@
       kfig= kind of figure (jpg, pdf, png)
       figname= name of the figure
-
+      ifclose= boolean value whether figure should be close (finish) or not
     """
     fname = 'plot_points'
+
+    # Axis ticks
+    # Usually axis > x must be the lon, thus...
+    dimxv0 = np.array(xval)
+    dimyv0 = np.array(yval)
+
+    dxn = dimxv0.min()
+    dxx = dimxv0.max()
+    dyn = dimyv0.min()
+    dyx = dimyv0.max()
+
+    if xaxv[0] == 'pretty':
+        dimxt0 = np.array(gen.pretty_int(dxn,dxx,xaxv[2]))
+    elif xaxv[0] == 'Nfix':
+        dimxt0 = np.arange(dxn,dxx,(dxx-dxn)/(1.*xaxv[2]))
+    elif xaxv[0] == 'Vfix':
+        dimxt0 = np.arange(0,dxx,xaxv[2])
+    if yaxv[0] == 'pretty':
+        dimyt0 = np.array(gen.pretty_int(dyn,dyx,yaxv[2]))
+    elif yaxv[0] == 'Nfix':
+        dimyt0 = np.arange(dyn,dyx,(dyx-dyn)/(1.*yaxv[2]))
+    elif yaxv[0] == 'Vfix':
+        dimyt0 = np.arange(0,dyx,yaxv[2])
+
+    dimxl0 = []
+    for i in range(len(dimxt0)): dimxl0.append('{:{style}}'.format(dimxt0[i], style=xaxv[1]))
+    dimyl0 = []
+    for i in range(len(dimyt0)): dimyl0.append('{:{style}}'.format(dimyt0[i], style=yaxv[1]))
+
+    dimxT0 = 'x'
+    dimyT0 = 'y'
+
+    if mapv is not None:
+        pixkind = 'data'
+    else:
+        # No following data values
+        dimxt0 = np.arange(len(dimxt0),dtype=np.float)/(len(dimxt0))
+        dimyt0 = np.arange(len(dimyt0),dtype=np.float)/(len(dimyt0))
+        pixkind = 'fixpixel'
+
+# No transformation
+#    if reva is not None:
+#        extravals, xval, yval, dimxt, dimyt, dimxl, dimyl, dimxT, dimyT =            \
+#          transform(extravals, reva, dxv=dimxv0, dyv=dimyv0, dxt=dimxt0, dyt=dimyt0, \
+#          dxl=dimxl0, dyl=dimyl0, dxtit=dimxT0, dytit=dimyT0)
+#    else:
+#        dimxv = dimxv0
+#        dimyv = dimyv0
+#        dimxt = dimxt0
+#        dimyt = dimyt0
+#        dimxl = dimxl0
+#        dimyl = dimyl0
+#        dimxT = dimxT0
+#        dimyT = dimyT0
+
+    dimxv = dimxv0
+    dimyv = dimyv0
+    dimxt = dimxt0
+    dimyt = dimyt0
+    dimxl = dimxl0
+    dimyl = dimyl0
+    dimxT = dimxT0
+    dimyT = dimyT0
+
 # Canging line kinds every 7 pts (end of standard colors)
     ptkinds=['.','x','o','*','+','8','>','D','h','p','s']
@@ -3128 +3194 @@
         plt.pcolormesh(x, y, extravals, cmap=plt.get_cmap(extrapar[3]),              \
           vmin=extrapar[1], vmax=extrapar[2])
-        cbar = plt.colorbar()
-        cbar.set_label(extrapar[0].replace('_','\_') +'('+ units_lunits(extrapar[4])+\
+        if extrapar[5] == 'horizontal':
+            cbar = plt.colorbar(format=extrapar[4],orientation=extrapar[5])
+            # 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=extrapar[4],orientation=extrapar[5])
+
+        cbar.set_label(gen.latex_text(extrapar[0]) +'('+ units_lunits(extrapar[6]) + \
           ')')
 
@@ -3139 +3215 @@
 
         if figk[0:8] == 'labelled':
+            if len(figk.split(',')) != 3:
+                print errormsg
+                print '  ' + fname + ": 'labelled' requires two additional values !!"
+                print "    'labelled',[txtsize],[txtcolor]"
+                quit(-1)
             txtsize=int(figk.split(',')[1])
             txtcol=figk.split(',')[2]
@@ -3150 +3231 @@
         plt.plot(xval, yval, '.', color=ptcol)
 
-    graphtit = vtit.replace('_','\_').replace('&','\&')
-
-    plt.title(graphtit.replace('|', ' '))
+    graphtit = gen.latex_text(vtit.replace('|',' '))
+
+    plt.title(graphtit)
     
-    output_kind(kfig, figname, True)
+    output_kind(kfig, figname, ifclose)
 
     return