Changeset 462 for sans/utils/bt5/bt5plot2/bt5plot2

Timestamp:

Dec 11, 2008 4:53:14 PM (14 years ago)

Author:

ajj

Message:

 

File:

• sans/utils/bt5/bt5plot2/bt5plot2 (modified) (7 diffs)

sans/utils/bt5/bt5plot2/bt5plot2

@@ -4 +4 @@
 import os
 import matplotlib
+import numpy
 matplotlib.use('GTK')
 
@@ -78 +79 @@
         
         self.canvas = FigureCanvasGTK(self.figure)
+        self.figure.canvas.mpl_connect('pick_event',self.handle_plot_click)
         self.canvas.show()
         
@@ -85 +87 @@
         self.metadataView = self.wTree.get_widget("tv_metadata")
         self.mdlist = gtk.ListStore(str,str)
-        
+       
         
         #self.filelistview.enable_model_drag_source( gtk.gdk.BUTTON1_MASK,
@@ -159 +161 @@
                                                                                          model[path][3][0],model[path][3][5], 'o')
         else:
-            model[path][4] = self.axis.plot(model[path][3][0],model[path][3][1], 'bo')
+            model[path][4] = self.axis.plot(model[path][3][0],model[path][3][1], 'bo', picker=5)
         
         self.rescale_and_redraw()
@@ -258 +260 @@
                 self.axis.lines.remove(line)
         
-        
-        
-        #self.canvas.draw()     
-        self.rescale_and_redraw()
+        if (len(self.axis.lines) > 0):
+                self.rescale_and_redraw()
+        else:
+                self.canvas.draw()      
+
         return
 
@@ -292 +295 @@
         if widget.get_active():
                 self.plottype = widget.get_name().split('_')[1]
-                print self.plottype
+                #print self.plottype
                 
         return
@@ -331 +334 @@
         return
 
+    def handle_plot_click(self,event):
+        if isinstance(event.artist, matplotlib.lines.Line2D):
+            pickedline = event.artist
+            xdata = pickedline.get_xdata()
+            ydata = pickedline.get_ydata()
+            ind = event.ind
+            print 'Plot Click: ',zip(numpy.take(xdata,ind), numpy.take(ydata,ind))
+
 app = appGui()
 gtk.main()