Changeset 991 for trunk/UTIL/PYTHON

Timestamp:

Jun 14, 2013, 10:10:43 AM (12 years ago)

Author:

aslmd

Message:

UTIL PYTHON planetoplot_v2. added --modx to produce a modulo on x labelling (e.g. for local time plots or Ls plots). thanks Tanguy for the request.

Location:

trunk/UTIL/PYTHON/planetoplot_v2

Files:

• pp.py (modified) (1 diff)
• ppclass.py (modified) (5 diffs)
• ppplot.py (modified) (5 diffs)

trunk/UTIL/PYTHON/planetoplot_v2/pp.py

@@ -83 +83 @@
 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")
+parser.add_option('--modx',action='append',dest='modx',type="float",default=None,help="[1D] change xticks with a modulo")
 # -- 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/ppclass.py

@@ -162 +162 @@
                       units=None,\
                       savtxt=False,\
+                      modx=None,\
                       title=None):
         self.request = None
@@ -201 +202 @@
         self.changetime = changetime
         self.savtxt = savtxt
+        self.modx = modx
         ## here are user-defined plot settings 
         ## -- if not None, valid on all plots in the pp() objects
@@ -265 +267 @@
             self.changetime = other.changetime
             self.savtxt = other.savtxt
+            self.modx = other.modx
         else:
             print "!! ERROR !! argument must be a pp object." ; exit()
@@ -797 +800 @@
                     if self.units is not None: plobj.units = self.units
                     if self.colorb is not None: plobj.colorb = self.colorb
+                    if self.modx is not None: plobj.modx = self.modx
                     # -- 1D specific
                     if dp == 1:
@@ -1178 +1182 @@
                 try: self.p[iii].ymax = opt.ymax[0]
                 except: pass
-
+            ###
+            try: self.p[iii].modx = opt.modx[iii]
+            except:
+                try: self.p[iii].modx = opt.modx[0]
+                except: pass
 
 

trunk/UTIL/PYTHON/planetoplot_v2/ppplot.py

@@ -248 +248 @@
     return what_I_plot
 
+# a function to change labels with modulo
+# ---------------------------------------
+def labelmodulo(ax,mod):
+    mpl.draw()
+    strtab = []
+    for tick in ax.get_xaxis().get_ticklabels():
+        num = float(tick.get_text())
+        strtab.append(num % mod)
+    ax.get_xaxis().set_ticklabels(strtab)
+    return ax
+
 # a function to output an ascii file
 # ----------------------------------
@@ -265 +276 @@
         else:
             print "!! WARNING !! Not printing the file, 2D fields not supported yet."
-#    mydata = np.transpose([absc,np.transpose(field)]) #tab
-#    for line in mydata:
-#        zeline = str(line)
-#        zeline = zeline.replace('[','')
-#        zeline = zeline.replace(']','')
-#        myfile.write(zeline + '\n')
     return
 
@@ -333 +338 @@
                  colorb="jet",\
                  units="",\
+                 modx=None,\
                  title=""):
         ## what could be defined by the user
@@ -352 +358 @@
         self.units = units
         self.colorb = colorb
+        self.modx = modx
         ## other useful arguments
         ## ... not used here in ppplot but need to be attached to plot object
@@ -505 +512 @@
         else:
             print "!! WARNING. in logx mode, ticks are set automatically."
+        ## specific modulo labels
+        if self.modx is not None:
+            ax = labelmodulo(ax,self.modx)
 
 ################################