Changeset 202 for trunk

Timestamp:

01/18/05 17:15:27 (20 years ago)

Author:

mar637

Message:

• removed internal self.frame1 to fix resizing bug.
• removed 'white' from default colors.
• added set_limits to set a "zoom" window
• added save function to save plotter from the command line

File:

• trunk/python/asaplot.py (modified) (8 diffs)

trunk/python/asaplot.py

@@ -28 +28 @@
     """
 
-    def __init__(self, rows=1, cols=0, title='', size=(8,6), buffering=False):
+    def __init__(self, rows=1, cols=0, title='', size=(7,5), buffering=False):
         """
         Create a new instance of the ASAPlot plotting class.
@@ -42 +42 @@
         
         self.window.protocol("WM_DELETE_WINDOW", dest_callback) 
-        self.frame1 = Tk.Frame(self.window, relief=Tk.RIDGE, borderwidth=4)
-        self.frame1.pack(fill=Tk.BOTH)
-
-        self.figure = Figure(figsize=size, facecolor='#ddddee')
-        self.canvas = FigureCanvasTkAgg(self.figure, master=self.frame1)
+        #self.frame1 = Tk.Frame(self.window, relief=Tk.RIDGE, borderwidth=4)
+        #self.frame1.pack(fill=Tk.BOTH)
+
+        self.figure = Figure(figsize=size,facecolor='#ddddee')
+        self.canvas = FigureCanvasTkAgg(self.figure, master=self.window)
         self.canvas.get_tk_widget().pack(side=Tk.TOP, fill=Tk.BOTH, expand=1)
 
@@ -64 +64 @@
 
         # Set matplotlib default colour sequence.
-        self.colours = [1, 'b', 'g', 'r', 'c', 'm', 'y', 'k', 'w']
+        self.colours = [1, 'b', 'g', 'r', 'c', 'm', 'y', 'k']
         self.attributes = {}
         self.loc = 1
@@ -490 +490 @@
         if rows < 1:
             rows = 1
-
+        nel = 1
         if cols == 0:
+            nel = rows
             i = int(sqrt(rows))
             if i*i < rows: i += 1
@@ -498 +499 @@
             if i*(i-1) >= rows: i -= 1
             rows = i
-
+            
         if 0 <= n < rows*cols:
             i = len(self.subplots)
@@ -510 +511 @@
         else:
             self.subplots = []
-            for i in range(0,rows*cols):
+            for i in range(0,nel):
                 self.subplots.append({})
                 self.subplots[i]['axes']  = self.figure.add_subplot(rows,
@@ -536 +537 @@
         if not self.buffering:
             if self.loc:
-                lines  = []
-                labels = []
-                i = 0
-                for line in self.lines:
-                    i += 1
-                    if line is not None:
-                        lines.append(line[0])
-                        lbl = line[0].get_label()
-                        if lbl == '':
-                            lbl = str(i)
-                        labels.append(lbl)
-
-                if len(lines):
-                    self.axes.legend(tuple(lines), tuple(labels), self.loc)
-                else:
-                    self.axes.legend((' '))
+                for j in range(len(self.subplots)):
+                    lines  = []
+                    labels = []
+                    i = 0
+                    for line in self.subplots[j]['lines']:
+                        i += 1
+                        if line is not None:
+                            lines.append(line[0])
+                            lbl = line[0].get_label()
+                            if lbl == '':
+                                lbl = str(i)
+                            labels.append(lbl)
+
+                    if len(lines):
+                        self.subplots[j]['axes'].legend(tuple(lines), tuple(labels), self.loc)
+                    else:
+                        self.subplots[j]['axes'].legend((' '))
 
             self.window.wm_deiconify()
@@ -595 +597 @@
         """
         self.window.wm_withdraw()
+
+    def set_limits(self,xlim=None,ylim=None):
+        for s in self.subplots:
+            self.axes  = s['axes']
+            self.lines = s['lines']
+            if xlim is not None:
+                self.axes.set_xlim(xlim)
+            if ylim is not None:
+                self.axes.set_ylim(ylim)
+        return
+
+    def save(self, fname=None):
+        if fname is None:
+            from datetime import datetime
+            dstr = datetime.now().strftime('%Y%m%d_%H%M%S')
+            fname = 'asap'+dstr+'.png'
+            
+        d = ['png','.ps','eps']
+        if fname[-3:].lower() in d:
+            try:
+                self.canvas.print_figure(fname)
+            except IOError, msg:
+                print 'Failed to save %s: Error msg was\n\n%s' % (fname, err)
+                return
+        else:
+            print "Invalid image type. Valid types are:"
+            print d