Changeset 865 for trunk/UTIL/PYTHON/planetoplot.py

Timestamp:

Jan 24, 2013, 1:05:12 PM (12 years ago)

Author:

aslmd

Message:

UTIL PYTHON. Improved redope (also available now with contours). Added --add (same as --mult for an addition). Added colormaps. Tried a different way to put wind vector reference (above colorbar).

File:

• trunk/UTIL/PYTHON/planetoplot.py (modified) (7 diffs)

trunk/UTIL/PYTHON/planetoplot.py

@@ -38 +38 @@
            ndiv=10,\
            mult=1.,\
+           add=0.,\
            zetitle=["fill"],\
            slon=None,\
@@ -96 +97 @@
                                   subplots_adjust, axes, clabel
     from matplotlib.cm import get_cmap
-    #from mpl_toolkits.basemap import cm
+    from mpl_toolkits.basemap import cm
     import numpy as np
     from numpy.core.defchararray import find
@@ -107 +108 @@
     import os
     from scipy import ndimage
+
 
 #########################
@@ -126 +128 @@
     if clb is None:            clb = ["def"]*len(var)
     elif len(clb) < len(var):  clb = [clb[0]]*len(var) ; print "WARNING: less color than vars! setting all to 1st value."
+    if redope is not None and winds: winds=False ; print "WARNING: no winds with redope. setting winds to False."
 ### we set option trycol i.e. the user wants to try a set of colorbars
     if trycol: clb = ["Greys","Blues","YlOrRd","jet","spectral","hot","RdBu","RdYlBu","Paired"] ; zetitle = clb ; var = [var[0]]*9
@@ -501 +504 @@
            what_I_plot, error = reducefield( all_var[index_f], d4=indextime, d1=indexlon, d2=indexlat, d3=indexvert, \
                                              yint=yintegral, alt=vert, anomaly=anomaly, redope=redope, mesharea=area, unidim=is1d)
+           if add != 0.:      what_I_plot = what_I_plot + add
            if mult != 2718.:  what_I_plot = what_I_plot*mult 
            else:              what_I_plot = np.log10(what_I_plot) ; print "log plot"
-                  
+
        if var2:      ### what is contoured.
            what_I_plot_contour, error = reducefield( all_var2[index_f], d4=indextime, d1=indexlon, d2=indexlat , d3=indexvert, \
-                                                     yint=yintegral, alt=vert )
+                                                     yint=yintegral, alt=vert, redope=redope )
        if winds:     ### what is plot as vectors.
            vecx, error = reducefield( all_windu[index_f], d4=indextime, d3=indexvert, yint=yintegral, alt=vert)
@@ -569 +573 @@
                #if (fileref is not None) and (index_f == numplot-1):    colorb = "RdBu_r"
                if colorb in ["def","nobar","onebar"]:                  palette = get_cmap(name=defcolorb(fvar.upper()))
+               elif colorb == "relief":                                palette = cm.GMT_relief
+               elif colorb == "haxby":                                 palette = cm.GMT_haxby
                else:                                                   palette = get_cmap(name=colorb)
                #palette = cm.GMT_split
@@ -794 +800 @@
                             
                     elif which == "contour":
+                        rcParams['contour.negative_linestyle'] = 'solid' # no dashed line for negative values
                         zevminc, zevmaxc = calculate_bounds(what_I_plot_frame, vmin=min(what_I_plot_frame), vmax=max(what_I_plot_frame))
                         zelevels = np.linspace(zevminc,zevmaxc,ticks/2) #20)
                         ### another dirty specific stuff in the wall
-                        if var2 == 'HGT':        zelevels = np.arange(-10000.,30000.,500.) #1000.)
+                        if var2 == 'HGT':        zelevels = np.arange(-10000.,30000.,250.) #1000.)
                         elif var2 == 'tpot':     zelevels = np.arange(270,370,5)
                         elif var2 == 'tk':       zelevels = np.arange(150,250,5)