菜单

菜单示例

输出：

  1. 100 371 91 29
  2. 100 342 91 29
  3. 100 313 91 29
  4. 100 284 91 29
  5. 100 255 91 29
  1. import numpy as np
  2. import matplotlib.colors as colors
  3. import matplotlib.patches as patches
  4. import matplotlib.mathtext as mathtext
  5. import matplotlib.pyplot as plt
  6. import matplotlib.artist as artist
  7. import matplotlib.image as image
  10. class ItemProperties(object):
  11.     def __init__(self, fontsize=14, labelcolor='black', bgcolor='yellow',
  12.                  alpha=1.0):
  13.         self.fontsize = fontsize
  14.         self.labelcolor = labelcolor
  15.         self.bgcolor = bgcolor
  16.         self.alpha = alpha
  18.         self.labelcolor_rgb = colors.to_rgba(labelcolor)[:3]
  19.         self.bgcolor_rgb = colors.to_rgba(bgcolor)[:3]
  22. class MenuItem(artist.Artist):
  23.     parser = mathtext.MathTextParser("Bitmap")
  24.     padx = 5
  25.     pady = 5
  27.     def __init__(self, fig, labelstr, props=None, hoverprops=None,
  28.                  on_select=None):
  29.         artist.Artist.__init__(self)
  31.         self.set_figure(fig)
  32.         self.labelstr = labelstr
  34.         if props is None:
  35.             props = ItemProperties()
  37.         if hoverprops is None:
  38.             hoverprops = ItemProperties()
  40.         self.props = props
  41.         self.hoverprops = hoverprops
  43.         self.on_select = on_select
  45.         x, self.depth = self.parser.to_mask(
  46.             labelstr, fontsize=props.fontsize, dpi=fig.dpi)
  48.         if props.fontsize != hoverprops.fontsize:
  49.             raise NotImplementedError(
  50.                 'support for different font sizes not implemented')
  52.         self.labelwidth = x.shape[1]
  53.         self.labelheight = x.shape[0]
  55.         self.labelArray = np.zeros((x.shape[0], x.shape[1], 4))
  56.         self.labelArray[:, :, -1] = x/255.
  58.         self.label = image.FigureImage(fig, origin='upper')
  59.         self.label.set_array(self.labelArray)
  61.         # we'll update these later
  62.         self.rect = patches.Rectangle((0, 0), 1, 1)
  64.         self.set_hover_props(False)
  66.         fig.canvas.mpl_connect('button_release_event', self.check_select)
  68.     def check_select(self, event):
  69.         over, junk = self.rect.contains(event)
  70.         if not over:
  71.             return
  73.         if self.on_select is not None:
  74.             self.on_select(self)
  76.     def set_extent(self, x, y, w, h):
  77.         print(x, y, w, h)
  78.         self.rect.set_x(x)
  79.         self.rect.set_y(y)
  80.         self.rect.set_width(w)
  81.         self.rect.set_height(h)
  83.         self.label.ox = x + self.padx
  84.         self.label.oy = y - self.depth + self.pady/2.
  86.         self.hover = False
  88.     def draw(self, renderer):
  89.         self.rect.draw(renderer)
  90.         self.label.draw(renderer)
  92.     def set_hover_props(self, b):
  93.         if b:
  94.             props = self.hoverprops
  95.         else:
  96.             props = self.props
  98.         r, g, b = props.labelcolor_rgb
  99.         self.labelArray[:, :, 0] = r
  100.         self.labelArray[:, :, 1] = g
  101.         self.labelArray[:, :, 2] = b
  102.         self.label.set_array(self.labelArray)
  103.         self.rect.set(facecolor=props.bgcolor, alpha=props.alpha)
  105.     def set_hover(self, event):
  106.         'check the hover status of event and return true if status is changed'
  107.         b, junk = self.rect.contains(event)
  109.         changed = (b != self.hover)
  111.         if changed:
  112.             self.set_hover_props(b)
  114.         self.hover = b
  115.         return changed
  118. class Menu(object):
  119.     def __init__(self, fig, menuitems):
  120.         self.figure = fig
  121.         fig.suppressComposite = True
  123.         self.menuitems = menuitems
  124.         self.numitems = len(menuitems)
  126.         maxw = max(item.labelwidth for item in menuitems)
  127.         maxh = max(item.labelheight for item in menuitems)
  129.         totalh = self.numitems*maxh + (self.numitems + 1)*2*MenuItem.pady
  131.         x0 = 100
  132.         y0 = 400
  134.         width = maxw + 2*MenuItem.padx
  135.         height = maxh + MenuItem.pady
  137.         for item in menuitems:
  138.             left = x0
  139.             bottom = y0 - maxh - MenuItem.pady
  141.             item.set_extent(left, bottom, width, height)
  143.             fig.artists.append(item)
  144.             y0 -= maxh + MenuItem.pady
  146.         fig.canvas.mpl_connect('motion_notify_event', self.on_move)
  148.     def on_move(self, event):
  149.         draw = False
  150.         for item in self.menuitems:
  151.             draw = item.set_hover(event)
  152.             if draw:
  153.                 self.figure.canvas.draw()
  154.                 break
  157. fig = plt.figure()
  158. fig.subplots_adjust(left=0.3)
  159. props = ItemProperties(labelcolor='black', bgcolor='yellow',
  160.                        fontsize=15, alpha=0.2)
  161. hoverprops = ItemProperties(labelcolor='white', bgcolor='blue',
  162.                             fontsize=15, alpha=0.2)
  164. menuitems = []
  165. for label in ('open', 'close', 'save', 'save as', 'quit'):
  166.     def on_select(item):
  167.         print('you selected %s' % item.labelstr)
  168.     item = MenuItem(fig, label, props=props, hoverprops=hoverprops,
  169.                     on_select=on_select)
  170.     menuitems.append(item)
  172. menu = Menu(fig, menuitems)
  173. plt.show()

下载这个示例