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

Timestamp:

Oct 6, 2016, 3:04:05 PM (8 years ago)

Author:

lfita

Message:

Adding dxdyfmt', colorbarvalues' and `close' in 'draw_2D_shad_cont'

File:

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

trunk/tools/drawing_tools.py

@@ -1850 +1850 @@
         elif u == 'ms-2': lu='$ms^{-2}$'
         elif u == 'minutes': lu='$minute$'
+        elif u == 'meters MSL': lu='$m$'
+        elif u == 'none': lu='-'
         elif u == 'Pa/s': lu='$Pas^{-1}$'
         elif u == 'Pas-1': lu='$Pas^{-1}$'
@@ -2709 +2711 @@
     rangegrph = np.zeros((2), dtype=np.float)
 # Changing limits of the colors
-    if type(vrng[0]) != type(np.float(1.)) and type(vrng[0]) != type(np.float64(1.)):
+    if vrng[0][0:1] == 'S':
         if vrng[0] == 'Srange':
             rangegrph[0] = np.min(varsV)
@@ -2765 +2767 @@
         print '    ' + fname + ': modified shadow min,max:',rangegrph
     else:
-        rangegrph[0] = vrng[0]
-        rangegrph[1] = vrng[1]
+        rangegrph[0] = np.float(vrng[0])
+        rangegrph[1] = np.float(vrng[1])
 
     return rangegrph
@@ -4443 +4445 @@
     return
 
-def plot_2D_shadow_contour(varsv,varcv,vnames,dimxv,dimyv,dimxu,dimyu,dimn,          \
-  colorbar,ckind,clabfmt,vs,vc,uts,vtit,kfig,reva,mapv):
+def plot_2D_shadow_contour(varsv, varcv, vnames, dimxv, dimyv, dimxu, dimyu, xaxv,   \
+  yaxv, dimn, colorbar, cbarfmt, cbaror, ckind, clabfmt, vs, vc, uts, vtit, kfig,    \
+  reva, mapv, ifclose):
     """ Adding labels and other staff to the graph
       varsv= 2D values to plot with shading
@@ -4451 +4454 @@
       dim[x/y]v = values at the axes of x and y
       dim[x/y]u = units at the axes of x and y
+      xaxv= list with the x-axis paramteres [style, format, number and orientation]
+      yaxv= list with the y-axis paramteres [style, format, number and orientation]
       dimn= dimension names to plot
       colorbar= name of the color bar to use
+      cbarfmt= format of the numbers in the colorbar
+      cbaror= orientation of the colorbar
       ckind= contour kind
         'cmap': as it gets from colorbar
@@ -4477 +4484 @@
           * 'h', high
           * 'f', full
+      ifclose= boolean value whether figure should be close (finish) or not
     """
 ##    import matplotlib as mpl
@@ -4488 +4496 @@
         quit()
 
+    if len(varsv.shape) != 2:
+        print errormsg
+        print '  ' + fname + ': wrong variable shape:',varsv.shape,'is has to be 2D!!'
+        quit(-1)
+    if len(varcv.shape) != 2:
+        print errormsg
+        print '  ' + fname + ': wrong variable contour:',varcv.shape,'is has to be 2D!!'
+        quit(-1)
+
+    # Axis ticks
     # Usually axis > x must be the lon, thus...
     dimxv0 = dimxv.copy()
     dimyv0 = dimyv.copy()
-    dimxt0 = np.array(pretty_int(dimxv.min(),dimxv.max(),10))
-    dimyt0 = np.array(pretty_int(dimyv.min(),dimyv.max(),10))
+
+    dxn = dimxv.min()
+    dxx = dimxv.max()
+    dyn = dimyv.min()
+    dyx = dimyv.max()
+
+    if xaxv[0] == 'pretty':
+        dimxt0 = np.array(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(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('{:.1f}'.format(dimxt0[i]))
+    for i in range(len(dimxt0)): dimxl0.append('{:{style}}'.format(dimxt0[i], style=xaxv[1]))
     dimyl0 = []
-    for i in range(len(dimyt0)): dimyl0.append('{:.1f}'.format(dimyt0[i]))
+    for i in range(len(dimyt0)): dimyl0.append('{:{style}}'.format(dimyt0[i], style=yaxv[1]))
+
     dimxT0 = variables_values(dimn[0])[0] + ' (' + units_lunits(dimxu) + ')'
     dimyT0 = variables_values(dimn[1])[0] + ' (' + units_lunits(dimyu) + ')'
@@ -4592 +4628 @@
 
     else:
-        x,y = gen.lonlat2D(dimxv,dimyv)
+        # No following data values
+        #x = dimxv
+        #y = dimyv
+        x = (dimxv-np.min(dimxv))/(np.max(dimxv) - np.min(dimxv))
+        y = (dimyv-np.min(dimyv))/(np.max(dimyv) - np.min(dimyv))
 
     plt.rc('text', usetex=True)
 
     plt.pcolormesh(x, y, varsv, cmap=plt.get_cmap(colorbar), vmin=vs[0], vmax=vs[1])
-    cbar = plt.colorbar()
+    if cbaror == 'horizontal':
+        cbar = plt.colorbar(format=cbarfmt,orientation=cbaror)
+        # 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=cbarfmt,orientation=cbaror)
 
 # contour
@@ -4638 +4687 @@
         plt.ylabel(dimyT)
 
-        plt.xticks(dimxt, dimxl)
-        plt.yticks(dimyt, dimyl)
+        plt.xticks(dimxt, dimxl, rotation=xaxv[3])
+        plt.yticks(dimyt, dimyl, rotation=yaxv[3])
 
 # set the limits of the plot to the limits of the data
@@ -4646 +4695 @@
 
 # units labels
-    cbar.set_label(vnames[0].replace('_','\_') + ' (' + units_lunits(uts[0]) + ')')
-    plt.annotate(vnames[1].replace('_','\_') +' (' + units_lunits(uts[1]) + ') [' +  \
+    cbar.set_label(gen.latex_text(vnames[0]) + ' (' + units_lunits(uts[0]) + ')')
+    plt.annotate(gen.latex_text(vnames[1]) +' (' + units_lunits(uts[1]) + ') [' +  \
       mincntS + ', ' + maxcntS + ']', xy=(0.55,0.04), xycoords='figure fraction',    \
       color=coln)
 
     figname = '2Dfields_shadow-contour'
-    graphtit = vtit.replace('_','\_').replace('&','\&')
+    graphtit = gen.latex_text(vtit)
 
     plt.title(graphtit)
     
-    output_kind(kfig, figname, True)
+    output_kind(kfig, figname, ifclose)
 
     return