Sync copies of pizza

python/examples/pizza/gnu.py

@@ -3,7 +3,7 @@
 #
 # Copyright (2005) Sandia Corporation.  Under the terms of Contract
 # DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
-# certain rights in this software.  This software is distributed under 
+# certain rights in this software.  This software is distributed under
 # the GNU General Public License.
 
 # for python3 compatibility
@@ -16,7 +16,7 @@ oneline = "Create plots via GnuPlot plotting program"
 docstr = """
 g = gnu()		       start up GnuPlot
 g.stop()		       shut down GnuPlot process
-    
+
 g.plot(a)                      plot vector A against linear index
 g.plot(a,b)	 	       plot B against A
 g.plot(a,b,c,d,...)	       plot B against A, D against C, etc
@@ -35,14 +35,14 @@ g("plot 'file.dat' using 2:3 with lines")      execute string in GnuPlot
 g.enter()	   	                enter GnuPlot shell
 gnuplot> plot sin(x) with lines         type commands directly to GnuPlot
 gnuplot> exit, quit	      	        exit GnuPlot shell
-    
+
 g.export("data",range(100),a,...)       create file with columns of numbers
 
   all vectors must be of equal length
   could plot from file with GnuPlot command: plot 'data' using 1:2 with lines
 
 g.select(N)  	               figure N becomes the current plot
-  
+
   subsequent commands apply to this plot
 
 g.hide(N)  	               delete window for figure N
@@ -98,7 +98,7 @@ except ImportError: PIZZA_GNUTERM = "x11"
 # Class definition
 
 class gnu:
-  
+
   # --------------------------------------------------------------------
 
   def __init__(self):
@@ -106,7 +106,7 @@ class gnu:
     self.file = "tmp.gnu"
     self.figures = []
     self.select(1)
-              
+
   # --------------------------------------------------------------------
 
   def stop(self):
@@ -118,7 +118,7 @@ class gnu:
   def __call__(self,command):
     self.GNUPLOT.write(command + '\n')
     self.GNUPLOT.flush()
-    
+
   # --------------------------------------------------------------------
 
   def enter(self):
@@ -159,7 +159,7 @@ class gnu:
         if i: partial_vecs.append(vec[:i])
         else: partial_vecs.append([0])
       self.plot(*partial_vecs)
-      
+
       if n < 10:     newfile = file + "000" + str(n)
       elif n < 100:  newfile = file + "00" + str(n)
       elif n < 1000: newfile = file + "0" + str(n)
@@ -167,7 +167,7 @@ class gnu:
 
       self.save(newfile)
       n += 1
-  
+
   # --------------------------------------------------------------------
   # write list of equal-length vectors to filename
 
@@ -208,7 +208,7 @@ class gnu:
   # do not continue until plot file is written out
   #   else script could go forward and change data file
   #   use tmp.done as semaphore to indicate plot is finished
-  
+
   def save(self,file):
     self.__call__("set terminal postscript enhanced solid lw 2 color portrait")
     cmd = "set output '%s.eps'" % file
@@ -219,7 +219,7 @@ class gnu:
     while not os.path.exists("tmp.done"): continue
     self.__call__("set output")
     self.select(self.current)
-  
+
   # --------------------------------------------------------------------
   # restore default attributes by creating a new fig object
 
@@ -228,7 +228,7 @@ class gnu:
     fig.ncurves = self.figures[self.current-1].ncurves
     self.figures[self.current-1] = fig
     self.draw()
-  
+
   # --------------------------------------------------------------------
 
   def aspect(self,value):
@@ -252,12 +252,12 @@ class gnu:
     else:
       self.figures[self.current-1].ylimit = (values[0],values[1])
     self.draw()
-    	    
+
   # --------------------------------------------------------------------
 
   def label(self,x,y,text):
     self.figures[self.current-1].labels.append((x,y,text))
-    self.figures[self.current-1].nlabels += 1  	    
+    self.figures[self.current-1].nlabels += 1
     self.draw()
 
   # --------------------------------------------------------------------
@@ -266,7 +266,7 @@ class gnu:
     self.figures[self.current-1].nlabel = 0
     self.figures[self.current-1].labels = []
     self.draw()
-      
+
   # --------------------------------------------------------------------
 
   def title(self,*strings):
@@ -283,13 +283,13 @@ class gnu:
   def xtitle(self,label):
     self.figures[self.current-1].xtitle = label
     self.draw()
-    
+
   # --------------------------------------------------------------------
 
   def ytitle(self,label):
     self.figures[self.current-1].ytitle = label
     self.draw()
-    
+
   # --------------------------------------------------------------------
 
   def xlog(self):
@@ -298,7 +298,7 @@ class gnu:
     else:
       self.figures[self.current-1].xlog = 1
     self.draw()
- 
+
   # --------------------------------------------------------------------
 
   def ylog(self):
@@ -307,7 +307,7 @@ class gnu:
     else:
       self.figures[self.current-1].ylog = 1
     self.draw()
-  
+
   # --------------------------------------------------------------------
 
   def curve(self,num,color):
@@ -323,10 +323,10 @@ class gnu:
   def draw(self):
     fig = self.figures[self.current-1]
     if not fig.ncurves: return
-    
+
     cmd = 'set size ratio ' + str(1.0/float(fig.aspect))
     self.__call__(cmd)
-      
+
     cmd = 'set title ' + '"' + fig.title + '"'
     self.__call__(cmd)
     cmd = 'set xlabel ' + '"' + fig.xtitle + '"'
@@ -338,11 +338,11 @@ class gnu:
     else: self.__call__("unset logscale x")
     if fig.ylog: self.__call__("set logscale y")
     else: self.__call__("unset logscale y")
-    if fig.xlimit: 
+    if fig.xlimit:
       cmd = 'set xr [' + str(fig.xlimit[0]) + ':' + str(fig.xlimit[1]) + ']'
       self.__call__(cmd)
     else: self.__call__("set xr [*:*]")
-    if fig.ylimit: 
+    if fig.ylimit:
       cmd = 'set yr [' + str(fig.ylimit[0]) + ':' + str(fig.ylimit[1]) + ']'
       self.__call__(cmd)
     else: self.__call__("set yr [*:*]")
@@ -372,7 +372,7 @@ class figure:
 
   def __init__(self):
     self.ncurves = 0
-    self.colors  = []    
+    self.colors  = []
     self.title   = ""
     self.xtitle  = ""
     self.ytitle  = ""