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

Timestamp:

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

Author:

lfita

Message:

Improving `draw_points' with dimxyfmt, slice capabilities and colobarvalues

File:

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

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