Changeset 977 for trunk/UTIL

Timestamp:

Jun 4, 2013, 4:41:53 PM (12 years ago)

Author:

aslmd

Message:

UTIL PYTHON planetoplot_v2. Added rainbow spahetti plot based on an idea by Alizee. Simply set a colormap for a 1D plot and you get it. Also added in pp_reload (simple rainbow function in ppplot). Added also xmin and xmax capabilitiy. Fixed various bugs related to dummy axis, logarithmic axes. Removed time_counter from set_ppclass.txt it is useless in most cases.

Location:

trunk/UTIL/PYTHON/planetoplot_v2

Files:

• examples/les_psfc.py (modified) (1 diff)
• examples/meso_profile.py (modified) (3 diffs)
• examples/vertpress1d.py (added)
• pp.py (modified) (1 diff)
• pp_reload.py (modified) (2 diffs)
• ppclass.py (modified) (5 diffs)
• ppplot.py (modified) (12 diffs)
• set_ppclass.txt (modified) (1 diff)

trunk/UTIL/PYTHON/planetoplot_v2/examples/les_psfc.py

@@ -2 +2 @@
 from ppclass import pp
 
-les = pp(file="/home/aymeric/Big_Data/psfc_f18.nc",var="PSFC",y=["50,250"],x=["50","100"])
+les = pp(file="/home/aymeric/Big_Data/LES_dd/psfc_f18.nc",var="PSFC",y=["50,250"],x=["50","100"])
 les.verbose = True
 les.getplot()

trunk/UTIL/PYTHON/planetoplot_v2/examples/meso_profile.py

@@ -11 +11 @@
 m.y = -5. # setting a fixed y value
 m.z = None # leaving z as a free dimension
-m.t = [6.,12.,18.,24.] # setting 4 fixed t values
+m.t = [6.,9.,12.,15.,18.,21.,24.] # setting 4 fixed t values
 
 # define settings
@@ -17 +17 @@
 #m.verbose = True # making the programe verbose
 #m.out = "pdf" # output format
+m.colorb = "spectral" # color cycle according to a color map
 
 # get data and make plot with default settings
@@ -48 +49 @@
 S.p[0].xlabel="Potential temperature (K)"
 S.filename = "meso_profile"
+S.colorb = None # come back to default color cycle
 
 # make the plot

trunk/UTIL/PYTHON/planetoplot_v2/pp.py

@@ -74 +74 @@
 parser.add_option('--xcoeff',action='append',dest='xcoeff',type="float",default=None,help="[1D] multiply x axis")
 parser.add_option('--ycoeff',action='append',dest='ycoeff',type="float",default=None,help="[1D] multiply y axis")
+parser.add_option('--xmin',action='append',dest='xmin',type="float",default=None,help="[1D] min bound x axis")
+parser.add_option('--ymin',action='append',dest='ymin',type="float",default=None,help="[1D] min bound y axis")
+parser.add_option('--xmax',action='append',dest='xmax',type="float",default=None,help="[1D] max bound x axis")
+parser.add_option('--ymax',action='append',dest='ymax',type="float",default=None,help="[1D] max bound y axis")
 # -- 2D plot
 parser.add_option('-C','--colorb',action='append',dest='colorb',type="string",default=None,help="[2D] colormap: http://micropore.files.wordpress.com/2010/06/colormaps.png")

trunk/UTIL/PYTHON/planetoplot_v2/pp_reload.py

@@ -7 +7 @@
 from optparse import OptionParser ### TBR by argparse
 from ppclass import pp
+from ppplot import rainbow
 # -----------------------------------------------------------------------
 parser = OptionParser()
@@ -34 +35 @@
         plot.proj = opt.proj
     if opt.colorb is not None:
+      yeah.colorb = opt.colorb
       for plot in yeah.p:
         plot.colorb = opt.colorb

trunk/UTIL/PYTHON/planetoplot_v2/ppclass.py

@@ -784 +784 @@
                     if self.title is not None: plobj.title = self.title
                     if self.units is not None: plobj.units = self.units
+                    if self.colorb is not None: plobj.colorb = self.colorb
                     # -- 1D specific
                     if dp == 1:
@@ -795 +796 @@
                         if self.vmin is not None: plobj.vmin = self.vmin
                         if self.vmax is not None: plobj.vmax = self.vmax
-                        if self.colorb is not None: plobj.colorb = self.colorb
                     # finally append plot object
                     self.p.append(plobj)
@@ -909 +909 @@
                     self.fig.add_subplot(1,1,1,axisbg=pl.axisbg) # define one subplot (still needed for user-defined font sizes)
                     sav = pl.xlabel,pl.ylabel,pl.xcoeff,pl.ycoeff,pl.title,pl.swaplab # save titles and labels
+                    # possibility to color lines according to a color map
+                    # ... made so that all plots span the whole color map automatically.
+                    if self.colorb is not None: 
+                        if self.verbose: print "**** OK. We make a rainbow spaghetti plot with color map ",self.colorb
+                        ppplot.rainbow(cb=self.colorb,num=self.howmanyplots)
                 else: 
                     pl.invert = False ; pl.lstyle = None # don't invert again axis
@@ -932 +937 @@
         # once completed show the plot (cannot show intermediate plotin)
         # ... added a fix (customplot=True) for the label problem in basemap
-        print "**** PPCLASS. Done step: makeplot"
+        if not self.quiet: print "**** PPCLASS. Done step: makeplot"
         if (self.n == self.howmanyplots):
             ppplot.save(mode=self.out,filename=self.filename,folder=self.folder,custom=self.customplot,includedate=self.includedate,res=self.res)
@@ -1133 +1138 @@
             except:
                 try: self.p[iii].wscale = opt.wscale[0]
+                except: pass
+            ###
+            try: self.p[iii].xmin = opt.xmin[iii]
+            except:
+                try: self.p[iii].xmin = opt.xmin[0]
+                except: pass
+            ###
+            try: self.p[iii].ymin = opt.ymin[iii]
+            except:
+                try: self.p[iii].ymin = opt.ymin[0]
+                except: pass
+            ###
+            try: self.p[iii].xmax = opt.xmax[iii]
+            except:
+                try: self.p[iii].xmax = opt.xmax[0]
+                except: pass
+            ###
+            try: self.p[iii].ymax = opt.ymax[iii]
+            except:
+                try: self.p[iii].ymax = opt.ymax[0]
                 except: pass
 

trunk/UTIL/PYTHON/planetoplot_v2/ppplot.py

@@ -180 +180 @@
 # useful functions #
 ####################
+
+# a function to change color lines according to a color map (idea by A. Pottier)
+# ... two optional arguments: color maps + a number telling how much intervals are needed
+def rainbow(cb="jet",num=8):
+    ax = mpl.gca()
+    pal = mpl.cm.get_cmap(name=cb)
+    ax._get_lines.set_color_cycle([pal(i) for i in np.linspace(0,0.9,num)])
 
 # a function to define subplot
@@ -299 +306 @@
                  ycoeff=1.,\
                  fmt=None,\
+                 colorb="jet",\
                  units="",\
                  title=""):
@@ -318 +326 @@
         self.fmt = fmt
         self.units = units
+        self.colorb = colorb
         ## other useful arguments
         ## ... not used here in ppplot but need to be attached to plot object
@@ -382 +391 @@
         self.marker = marker
         self.label = label
+        self.xmin = None
+        self.ymin = None
+        self.xmax = None
+        self.ymax = None
 
     # define_from_var
@@ -405 +418 @@
         mpl.grid(color='grey')
         if self.lstyle == "": self.lstyle = " " # to allow for no line at all with ""
+        # set dummy x axis if not defined
+        if self.absc is None: 
+            self.absc = np.array(range(self.field.shape[0]))
+            print "!! WARNING !! dummy coordinates on x axis"
         # swapping if requested
         if self.swap:  x = self.field ; y = self.absc
@@ -427 +444 @@
         # ... this must be after plot so that axis bounds are well-defined
         # ... also inverting must be after making the thing logarithmic
-        if self.logx: mpl.semilogx()
-        if self.logy: mpl.semilogy()
+        if self.logx: mpl.xscale("log") # not mpl.semilogx() because excludes log on y
+        if self.logy: mpl.yscale("log") # not mpl.semilogy() because excludes log on x
         if self.invert: ax = mpl.gca() ; ax.set_ylim(ax.get_ylim()[::-1])
         # add a label for line(s)
@@ -446 +463 @@
         if xmin < x1: x1 = xmin
         if xmax > x2: x2 = xmax
+        if self.xmin is not None: x1 = self.xmin
+        if self.xmax is not None: x2 = self.xmax
         ax.set_xbound(lower=x1,upper=x2)
         # y-axis
@@ -453 +472 @@
         if ymin < y1: y1 = ymin
         if ymax > y2: y2 = ymax
+        if self.ymin is not None: y1 = self.ymin
+        if self.ymax is not None: y2 = self.ymax
         ax.set_ybound(lower=y1,upper=y2)
         ## set with .div the number of ticks. (is it better than automatic?)
@@ -478 +499 @@
                  proj="cyl",\
                  back=None,\
-                 colorb="jet",\
                  trans=1.0,\
                  addvecx=None,\
@@ -498 +518 @@
         self.proj = proj
         self.back = back
-        self.colorb = colorb
         self.trans = trans
         self.addvecx = addvecx
@@ -527 +546 @@
         # get what is done in the parent class...
         plot.make(self)
-        if self.fmt is None: self.fmt = "%.2e"
+        if self.fmt is None: self.fmt = "%.1e"
         # ... then add specific stuff
         ############################################################################################
         ### PRE-SETTINGS
         ############################################################################################
-        # transposing if necessary
-        shape = self.field.shape
-        if shape[0] != shape[1]:
-         if len(self.absc) == shape[0] and len(self.ordi) == shape[1]:
-            print "!! WARNING !! Transposing axes"
-            self.field = np.transpose(self.field)
         # set dummy xy axis if not defined
         if self.absc is None: 
@@ -547 +560 @@
             self.mapmode = False
             print "!! WARNING !! dummy coordinates on y axis"
+        # transposing if necessary
+        shape = self.field.shape
+        if shape[0] != shape[1]:
+         if len(self.absc) == shape[0] and len(self.ordi) == shape[1]:
+            print "!! WARNING !! Transposing axes"
+            self.field = np.transpose(self.field)
         # bound field
         zevmin, zevmax = calculate_bounds(self.field,vmin=self.vmin,vmax=self.vmax)

trunk/UTIL/PYTHON/planetoplot_v2/set_ppclass.txt

@@ -13 +13 @@
   # t-axis: possible names to search for. add in the line with a separating ;
   #
-Times;time;Time;time_counter;Ls
+Times;time;Time;Ls
   # area: possible names to search for. add in the line with a separating ;
   #