01-callable.py

from collections import deque
import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np
plt.ion()

# disable the built in key bindings
for k in ['keymap.all_axes',
          'keymap.back',
          'keymap.forward',
          'keymap.fullscreen',
          'keymap.grid',
          'keymap.home',
          'keymap.pan',
          'keymap.save',
          'keymap.xscale',
          'keymap.yscale',
          'keymap.zoom']:
    mpl.rcParams[k] = []


class EventCollector:
    def __init__(self, maxlen=12):
        self.event_deque = deque([], maxlen=maxlen)

    def __call__(self, event):
        print('called {} at ({}, {})'.format(event.name,
                                             event.xdata,
                                             event.ydata))
        self.event_deque.append(event)


fig, ax = plt.subplots()

th = np.linspace(0, 2*np.pi, 64)
ln, = ax.plot(th, np.sin(th), 'o-', picker=5)

ec = EventCollector()
cid = fig.canvas.mpl_connect('key_press_event', ec)


# EXERCISE
# - Try connecting the same object to events calls
# - add helper function to manage event buffer