Changeset 518 for trunk

Timestamp:

Feb 8, 2012, 10:38:45 AM (13 years ago)

Author:

aslmd

Message:

UTIL/PYTHON: added the possibility to use -c onebar for multiplots as one would easily guess why

Location:

trunk/UTIL/PYTHON

Files:

• README.PP (modified) (1 diff)
• myplot.py (modified) (2 diffs)
• myscript.py (modified) (2 diffs)
• planetoplot.py (modified) (6 diffs)

trunk/UTIL/PYTHON/README.PP

@@ -118 +118 @@
 ***********************************************************************************
 pp.py -f wrfout_d01_9999-09-09_09:00:00 -v W,tpot --lat 60 --time 15 -i 4 -l 30,130,100 --div 50
+***********************************************************************************
+
+***********************************************************************************
+EXAMPLE : Plotting surface temperature maps from TES
+***********************************************************************************
+pp.py -f TES/TES.MappedClimatology.nadir.MY25.nc -v Tsurf_day -p npstere --blat 60 --trans 0.5 -b vishires --time 50
+***********************************************************************************
+pp.py -f /d5/emlmd/TES/TES.MappedClimatology.nadir.MY25.nc,/d5/emlmd/TES/TES.MappedClimatology.nadir.MY26.nc   -p npstere -v Tsurf_day --time 90,105 --time 105,120 --time 120,135 --time 135,150 --blat 70 --div 30 -m 190 -M 270 -b vishires --trans 0.7 
+***********************************************************************************
+pp.py -f /d5/emlmd/TES/TES.MappedClimatology.nadir.MY25.nc,/d5/emlmd/TES/TES.MappedClimatology.nadir.MY26.nc -p npstere -v Tsurf_day --time 110,130 --blat 70 --div 30 -m 190 -M 270 -b vishires --trans 0.7 -c onebar --title "Surface temperature (K)"
 ***********************************************************************************
 

trunk/UTIL/PYTHON/myplot.py

@@ -770 +770 @@
 
                     }
+    if "TSURF" in whichvar: whichvar = "TSURF"
     if whichvar not in fmtvar:
         whichvar = "def"
@@ -816 +817 @@
 #spectral BrBG RdBu_r
     #print "predefined colorbars"
+    if "TSURF" in whichone: whichone = "TSURF"
     if whichone not in whichcolorb:
         whichone = "def"

trunk/UTIL/PYTHON/myscript.py

@@ -23 +23 @@
 
     ### VERTICAL INTERPOLATION
-    parser.add_option('-l', '--level',  action='store',dest='lvl',       type="string",  default="0",   help='level / start,stop,step (-i 2: p,mb)(-i 3,4: z,km) [0]')
+    parser.add_option('-l', '--level',  action='store',dest='lvl',       type="string",  default="0",   help='level / start,stop,step (-i 2: p,Pa)(-i 3,4: z,km) [0]')
     parser.add_option('-i', '--interp', action='store',dest='itp',       type="int",     default=None,  help='interpolation (2: p, 3: z-amr, 4:z-als)')
     parser.add_option('--intas',        action='store',dest='intas',     type="string",  default=None,  help='specify "mcs" or "tes" for gcm P interpolation grid')
@@ -29 +29 @@
 
     ### GENERIC GRAPHICS SETTINGS
-    parser.add_option('-c', '--color',  action='store',dest='clb',       type="string",  default="def", help='change colormap (nobar: no colorbar)')
+    parser.add_option('-c', '--color',  action='store',dest='clb',       type="string",  default="def", help='change colormap (also: nobar,onebar)')
     parser.add_option('--div',          action='store',dest='ndiv',      type="int",     default=10,    help='number of divisions in colorbar [10]')
     parser.add_option('--title',        action='store',dest='zetitle',   type="string",  default="fill",help='customize the whole title')

trunk/UTIL/PYTHON/planetoplot.py

@@ -76 +76 @@
     from mymath import deg,max,min,mean,get_tsat,writeascii,fig2data,fig2img
     import matplotlib as mpl
-    from matplotlib.pyplot import contour,contourf, subplot, figure, rcParams, savefig, colorbar, pcolor, show, plot, clabel, title, close, legend, xlabel, axis, ylabel
+    from matplotlib.pyplot import contour,contourf, subplot, figure, rcParams, savefig, colorbar, pcolor, show, plot, clabel, title, close, legend, xlabel, axis, ylabel, subplots_adjust, axes
     from matplotlib.cm import get_cmap
     #from mpl_toolkits.basemap import cm
@@ -372 +372 @@
        ### General plot settings
        changesubplot = (numplot > 1) and (len(what_I_plot.shape) != 1)  ## default for 1D plots: superimposed. to be reworked for better flexibility.
-       if changesubplot: subplot(subv,subh,nplot)
+       if changesubplot: subplot(subv,subh,nplot) #; subplots_adjust(wspace=0,hspace=0)
        ####################################################################
        if error:
@@ -391 +391 @@
                else:                                                   zevmin, zevmax = calculate_bounds(what_I_plot,vmin=vmin,vmax=vmax)
                if (fileref is not None) and (index_f == numplot-1):    colorb = "RdBu_r"
-               if colorb in ["def","nobar"]:                           palette = get_cmap(name=defcolorb(fvar.upper()))
+               if colorb in ["def","nobar","onebar"]:                  palette = get_cmap(name=defcolorb(fvar.upper()))
                else:                                                   palette = get_cmap(name=colorb)
                #palette = cm.GMT_split
@@ -469 +469 @@
                             elif mapmode == 0:     pcolor( x, y, what_I_plot_frame, cmap = palette, vmin=zevmin, vmax=zevmax, alpha=trans)
 
-                        if colorb != 'nobar':        
+                        if colorb not in ['nobar','onebar']:        
                             if (fileref is not None) and (index_f == numplot-1):   daformat = "%.3f" 
                             elif mult != 1:                                        daformat = "%.1f"
@@ -477 +477 @@
                             zecb = colorbar( fraction=0.05,pad=0.03,format=daformat,orientation=zeorientation,\
                                       ticks=np.linspace(zevmin,zevmax,num=min([ticks/2+1,21])),extend='neither',spacing='proportional' ) 
-                            if zeorientation == "horizontal":
-                                zecb.ax.set_xlabel(zetitle)
-                                zetitle=""
+                            if zeorientation == "horizontal" and zetitle != "fill": zecb.ax.set_xlabel(zetitle) ; zetitle=""
                         if winds:
                             if typefile in ['mesoapi','meso']:
@@ -578 +576 @@
 #       if indexvert is not None:     plottitle = plottitle + " vert: " + str(min(vert[indexvert])) +" "+ str(max(vert[indexvert]))
 #       if indextime is not None:     plottitle = plottitle + " time: " + str(min(time[indextime])) +" "+ str(max(time[indextime]))
-       title( plottitle )
+       if colorb != "onebar": title( plottitle )
        if nplot >= numplot: error = True
        nplot += 1
 
-  
-
-
-
-
+    if colorb == "onebar":
+        cax = axes([0.1, 0.2, 0.8, 0.03]) # a ameliorer
+        zecb = colorbar(cax=cax, orientation="horizontal", format=fmtvar(fvar.upper()),\
+                 ticks=np.linspace(zevmin,zevmax,num=min([ticks/2+1,21])),extend='neither',spacing='proportional')
+        if zetitle != "fill": zecb.ax.set_xlabel(zetitle) ; zetitle=""