[Matplotlib-users] Problem: Can't combine mouse button events and pick events in a scatter plot

[joh [via matplotlib]]

Hello everybody, 


I am trying to combine mouse button events with a pick event in scatter
plot. Different functionalities take place, based on the mouse button
pressed each time. I want each functionality to be triggered with a
combination of mouse buttons events and/or key press events. When the left
mouse button is pressed matplotlib's Lasso widget is called and with the
included points functionality 1 takes place. When Shift+LMB are pressed a
Lasso is drawn and functionality 2 takes place with the included points.
When Alt+LMB are pressed a Lasso is drawn and with the included points
functionality 3 takes place. Last, but not least, when I press the RMB a
pick event is triggered and the index of the selected point in the scatter
plot is given. 

Since I added the pick event, the aforementioned functionalities work
correctly until a pick event is triggered. When it is triggered it seems
that the canvas gets locked and I can not use any other functionality. I
don't get the index of the selected points and I do not get any errors. 


----------------Code------------------------- 

import logging 
import matplotlib 
from matplotlib.widgets import Lasso 
from matplotlib.colors import colorConverter 
from matplotlib.collections import RegularPolyCollection 
from matplotlib import path 
import numpy as np 
import matplotlib.pyplot as plt 
from numpy.random import rand 

logger = logging.getLogger() 
logger.setLevel(logging.DEBUG) 


class Datum(object): 
      colorin = colorConverter.to_rgba('red') 
      colorShift = colorConverter.to_rgba('cyan') 
      colorCtrl = colorConverter.to_rgba('pink') 
      colorout = colorConverter.to_rgba('blue') 

      def __init__(self, x, y, include=False): 
         self.x = x 
         self.y = y 
         if include: 
            self.color = self.colorin 
         else: 
            self.color = self.colorout 


class LassoManager(object): 
      def __init__(self, ax, data): 
         self.axes = ax 
         self.canvas = ax.figure.canvas 
         self.data = data 

         self.Nxy = len(data) 

         facecolors = [d.color for d in data] 
         self.xys = [(d.x, d.y) for d in data] 
         fig = ax.figure 
         self.collection = RegularPolyCollection(fig.dpi, 6,
sizes=(100,),facecolors=facecolors, offsets = self.xys, transOffset =
ax.transData) 

         ax.add_collection(self.collection) 

         self.pick=self.canvas.mpl_connect('pick_event', self.onpick) 
         self.cid = self.canvas.mpl_connect('button_press_event',
self.onpress) 
         self.keyPress = self.canvas.mpl_connect('key_press_event',
self.onKeyPress) 
         self.keyRelease = self.canvas.mpl_connect('key_release_event',
self.onKeyRelease) 
         self.lasso = None 
         self.shiftKey = False 
         self.ctrlKey = False 

      def callback(self, verts): 
          logging.debug('in LassoManager.callback(). Shift: %s, Ctrl: %s' %
(self.shiftKey, self.ctrlKey)) 
          facecolors = self.collection.get_facecolors() 
          p = path.Path(verts) 
          ind = p.contains_points(self.xys) 
          for i in range(len(self.xys)): 
              if ind[i]: 
                 if self.shiftKey: 
                    facecolors[i] = Datum.colorShift 
                 elif self.ctrlKey: 
                    facecolors[i] = Datum.colorCtrl 
                 else: 
                    facecolors[i] = Datum.colorin 
                    print self.xys[i] 
              else: 
                   facecolors[i] = Datum.colorout 

          self.canvas.draw_idle() 
          self.canvas.widgetlock.release(self.lasso) 
          del self.lasso 

      def onpress(self, event): 
           if self.canvas.widgetlock.locked(): return 
           if event.inaxes is None:    return 
           self.lasso = Lasso(event.inaxes, (event.xdata, event.ydata),
self.callback) 
           # acquire a lock on the widget drawing 
           self.canvas.widgetlock(self.lasso) 

      def onKeyPress(self, event): 
          logging.debug('in LassoManager.onKeyPress(). Event received: %s
(key: %s)' % (event, event.key)) 
          if event.key == 'alt': 
             self.ctrlKey = True 
          if event.key == 'shift': 
             self.shiftKey = True 

      def onKeyRelease(self, event): 
          logging.debug('in LassoManager.onKeyRelease(). Event received: %s
(key: %s)' % (event, event.key)) 
          if event.key == 'alt': 
             self.ctrlKey = False 
          if event.key == 'shift': 
             self.shiftKey = False 


      def onpick(self,event): 

          if event.mouseevent.button == 3: 

             index = event.ind 
             print('onpick3 scatter:', index) 

      if __name__ == '__main__': 

         data = [Datum(*xy) for xy in rand(100, 2)] 

         ax = plt.axes(xlim=(0,1), ylim=(0,1), autoscale_on=False) 
         lman = LassoManager(ax, data) 

         plt.show() 

------------------End of Code----------------------------- 

Any suggestions on what might be causing this malfunction? What am I doing
wrong?




______________________________________
If you reply to this email, your message will be added to the discussion below:
http://matplotlib.1069221.n5.nabble.com/Problem-Can-t-combine-mouse-button-events-and-pick-events-in-a-scatter-plot-tp46112.html
This email was sent by joh (via Nabble)
To receive all replies by email, subscribe to this discussion: http://matplotlib.1069221.n5.nabble.com/template/NamlServlet.jtp?macro=subscribe_by_code&node=46112&code=bWF0cGxvdGxpYi11c2Vyc0BweXRob24ub3JnfDQ2MTEyfC0xNjQ0NjkzNDk1
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/matplotlib-users/attachments/20150909/329619f2/attachment.html>