Changeset 865

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).

Location:

trunk/UTIL/PYTHON

Files:

• myplot.py (modified) (6 diffs)
• myscript.py (modified) (1 diff)
• planetoplot.py (modified) (7 diffs)
• pp.py (modified) (1 diff)

trunk/UTIL/PYTHON/myplot.py

@@ -255 +255 @@
     csmooth = 12 ## a fair amount of grid points (too high results in high computation time)
     if redope is not None:
-       if   redope == "mint":     input = min(input,axis=0) ; d1 = None
-       elif redope == "maxt":     input = max(input,axis=0) ; d1 = None
-       elif redope == "edge_y1":  input = input[:,:,0,:]    ; d2 = None
+    #   if   redope == "mint":     input = min(input,axis=0) ; d1 = None
+    #   elif redope == "maxt":     input = max(input,axis=0) ; d1 = None
+       if redope == "edge_y1":  input = input[:,:,0,:]    ; d2 = None
        elif redope == "edge_y2":  input = input[:,:,-1,:]   ; d2 = None
        elif redope == "edge_x1":  input = input[:,:,:,0]    ; d1 = None
        elif redope == "edge_x2":  input = input[:,:,:,-1]   ; d1 = None
-       else:                      errormess("not supported. but try lines in reducefield beforehand.")
+    #   else:                      errormess("not supported. but try lines in reducefield beforehand.")
        #elif redope == "minz":     input = min(input,axis=1) ; d2 = None
        #elif redope == "maxz":     input = max(input,axis=1) ; d2 = None
@@ -430 +430 @@
         elif d3 is not None:        output = reduce_zaxis(input[:,d3,:,:],ax=1,yint=yint,vert=alt,indice=d3)
         elif d4 is not None:        output = mean(input[d4,:,:,:],axis=0)
+
+    if redope is not None:
+       if   redope == "mint":     output = min(output,axis=0)
+       elif redope == "maxt":     output = max(output,axis=0) 
+
     dimension2 = np.array(output).ndim
     shape2 = np.array(output).shape
@@ -826 +831 @@
     #posx = np.min(x) - np.std(x) / 10.
     #posy = np.min(y) - np.std(y) / 10.
-    posx = np.min(x) 
-    posy = np.min(y) - 4.*np.std(y) / 10.
+    #posx = np.min(x) 
+    #posy = np.min(y) - 4.*np.std(y) / 10.
     u = smooth(u,csmooth)
     v = smooth(v,csmooth)
@@ -839 +844 @@
     if color in ['white','yellow']:     kcolor='black'
     else:                               kcolor=color
-    if key: p = plt.quiverkey(q,posx,posy,scale,\
-                   str(int(scale)),coordinates='data',color=kcolor,labelpos='S',labelsep = 0.03)
+    if key: p = plt.quiverkey(q,-0.06,0.98,scale,\
+                   str(int(scale)),color=kcolor,labelpos='S',labelsep = 0.03)
     return 
 
@@ -1158 +1163 @@
                 "mola":         "black",\
                 "gist_heat":    "white",\
+                "hot":          "white",\
                 "gist_rainbow": "black",\
                 "spectral":     "black",\
@@ -1240 +1246 @@
         "Sirenum_Crater_small":  [[-36.,-26.],[-168.,-156.]],\
         "Rupes":                 [[ 72., 90.],[-120.,- 20.]],\
+        "Rupes2":                [[ 80., 85.],[- 90.,- 60.]],\
         "Xanadu":                [[-40., 20.],[  40., 120.]],\
         "Hyperboreae":           [[ 80., 87.],[- 70.,- 10.]],\

trunk/UTIL/PYTHON/myscript.py

@@ -16 +16 @@
     parser.add_option('-a', '--anomaly',action='store_true',dest='anomaly',              default=False, help='compute and plot relative anomaly in %')
     parser.add_option('--mult',         action='store',dest='mult',      type="float",   default=1.,    help='multiplicative factor to plotted field (2718=log)')
+    parser.add_option('--add',          action='store',dest='add',       type="float",   default=0.,    help='additive factor to plotted field')
     parser.add_option('-m', '--min',    action='append',dest='vmin',     type="float",   default=None,  help='bounding minimum value [min]')    
     parser.add_option('-M', '--max',    action='append',dest='vmax',     type="float",   default=None,  help='bounding maximum value [max]') 

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)

trunk/UTIL/PYTHON/pp.py

@@ -170 +170 @@
                 hole=opt.hole,save=opt.save,\
                 anomaly=opt.anomaly,var2=opt.var2,ndiv=opt.ndiv,\
-                mult=opt.mult,zetitle=separatenames(opt.zetitle),\
+                mult=opt.mult,add=opt.add,zetitle=separatenames(opt.zetitle),\
                 slon=zeslon,slat=zeslat,svert=zesvert,stime=zestime,\
                 outputname=opt.out,resolution=opt.res,\