Changeset 192 for trunk

Timestamp:

Jul 4, 2011, 4:03:07 PM (13 years ago)

Author:

aslmd

Message:

MESOSCALE: python graphics. possibility to change color mapping.

File:

• trunk/MESOSCALE/PLOT/PYTHON/scripts/winds.py (modified) (4 diffs)

trunk/MESOSCALE/PLOT/PYTHON/scripts/winds.py

@@ -19 +19 @@
            winds=True):
 
+    ####################################################################################################################
+    ### Colorbars http://www.scipy.org/Cookbook/Matplotlib/Show_colormaps?action=AttachFile&do=get&target=colormaps3.png
+
     #################################
     ### Load librairies and functions
@@ -24 +27 @@
     from myplot import getcoord2d,define_proj,makeplotpng,simplinterv,vectorfield,ptitle,latinterv,getproj,wrfinterv,dumpbdy
     from matplotlib.pyplot import contourf, subplot, figure, rcParams, savefig, colorbar
+    from matplotlib.cm import get_cmap
     import numpy as np
 
@@ -193 +197 @@
                if dimension == 2:                what_I_plot = field[:,:]
                elif dimension == 3:              what_I_plot = field[i,:,:]
-           contourf(x, y, what_I_plot, 30)
+           contourf( x, y, what_I_plot, 30, cmap = get_cmap(name=colorb) )
            if colorb:     colorbar(fraction=0.05,pad=0.1)
 
@@ -257 +261 @@
     parser.add_option('-n', action='store', dest='numplot',     type="int",     default=4,     help='number of plots (def=1)(<0: 1 plot of LT -*numplot*)')
     parser.add_option('-i', action='store', dest='interp',      type="int",     default=None,  help='interpolation method (2: press, 3: z-amr, 4:z-als)')
-    parser.add_option('-c', action='store', dest='colorb',      type="string",  default=None,  help='colorbar')
+    parser.add_option('-c', action='store', dest='colorb',      type="string",  default=None,  help='change colormap (and draw a colorbar)')
     parser.add_option('-x', action='store_false', dest='winds',                 default=True,  help='flag: no wind vectors')
     (opt,args) = parser.parse_args()