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=15):
        self.event_deque = deque([], maxlen=maxlen)

    def __call__(self, event):
        print(f'called {event.name} at ')
        print(f' screen: ({event.x}, {event.y})')
        if event.inaxes:
            print(f'   data: ({event.xdata:.3g}, {event.ydata:.3g})')
        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 (10 minutes)
# - Try connecting the same object to other events
# - add helper method to manage event buffer
#   - print out message of key press events