added a line vertex selector widget

jdh2358 · jdh2358 · commit 99f3a369ef64 · 2008-05-26T17:02:46.000Z
svn path=/branches/v0_91_maint/; revision=5271
diff --git a/CODING_GUIDE b/CODING_GUIDE
@@ -9,12 +9,6 @@ and standards.  Please edit and extend this document.
 svn co https://matplotlib.svn.sourceforge.net/svnroot/matplotlib/trunk matplotlib --username=youruser --password=yourpass
 
 
-# checking out the main src
-svn co https://matplotlib.svn.sourceforge.net/svnroot/matplotlib/trunk/matplotlib matplotlib --username=youruser --password=yourpass
-
-# branch checkouts, eg the transforms branch
-svn co https://matplotlib.svn.sourceforge.net/svnroot/matplotlib/branches/transforms transbranch
-
 == Committing changes ==
 
 When committing changes to matplotlib, there are a few things to bear
diff --git a/lib/matplotlib/lines.py b/lib/matplotlib/lines.py
@@ -569,6 +569,10 @@ def get_markerfacecolor(self):
 
     def get_markersize(self): return self._markersize
 
+    def get_data(self, orig=True):
+        'return the xdata, ydata; if orig is True, return the original data'
+        return self.get_xdata(orig=orig), self.get_ydata(orig=orig)
+
     def get_xdata(self, orig=True):
         """
         return the xdata; if orig is true return the original data,
@@ -1460,6 +1464,57 @@ def is_dashed(self):
         'return True if line is dashstyle'
         return self._linestyle in ('--', '-.', ':')
 
+class VertexSelector:
+    """
+    manage the callbacks to maintain a list of selected vertices for
+    matplotlib.lines.Lin2D. Derived classes should override
+    process_selected to do something with the picks
+    """
+    def __init__(self, line):
+        """
+        Initialize the class with a matplotlib.lines.Line2D instance.
+        The line should already be added to some matplotlib.axes.Axes
+        instance and should have the picker property set.
+        """
+        if not hasattr(line, 'axes'):
+            raise RuntimeError('You must first add the line to the Axes')
+
+        if line.get_picker() is None:
+            raise RuntimeError('You must first set the picker property of the line')
+
+        self.axes = line.axes
+        self.line = line
+        self.canvas = self.axes.figure.canvas
+        self.cid = self.canvas.mpl_connect('pick_event', self.onpick)
+
+        self.ind = set()
+
+
+    def process_selected(self, ind, xs, ys):
+        """
+        Default do nothing implementation of the process_selected method.
+
+        ind are the indices of the selected vertices.  xs and ys are
+        the coordinates of the selected vertices.
+        """
+        pass
+
+    def onpick(self, event):
+        'when the line is picked, update the set of selected indicies'
+        if event.artist is not self.line: return
+
+        for i in event.ind:
+            if i in self.ind:
+                self.ind.remove(i)
+            else:
+                self.ind.add(i)
+
+
+        ind = list(self.ind)
+        ind.sort()
+        ind = npy.array(ind)
+        xdata, ydata = self.line.get_data()
+        self.process_selected(ind, xdata[ind], ydata[ind])
 
 lineStyles = Line2D._lineStyles
 lineMarkers = Line2D._markers
