import hippo import gobject from sugar.graphics.canvasicon import CanvasIcon class _MenuStrategy: def get_menu_position(self, menu, grid_x1, grid_y1, grid_x2, grid_y2): grid_x = grid_x2 if grid_x + menu.get_width() > Grid.COLS: grid_x = grid_x1 - menu.get_width() + 1 grid_y = grid_y1 if grid_y < 0: grid_y = 0 if grid_y + menu.get_width() > Grid.ROWS: grid_y = Grid.ROWS - menu.get_width() return [grid_x, grid_y] class MenuIcon(CanvasIcon): def __init__(self, menu_shell, **kwargs): CanvasIcon.__init__(self, **kwargs) self._menu_shell = menu_shell self._grid = menu_shell.get_grid() self._menu = None self._hover_menu = False self._popdown_on_leave = False self._popdown_sid = 0 self._menu_strategy = _MenuStrategy() self.connect('motion-notify-event', self._motion_notify_event_cb) def popdown(self): if self._menu: self._menu.destroy() self._menu = None self._menu_shell.set_active(None) def set_menu_strategy(self, strategy): self._menu_strategy = strategy def _popup(self, x1, y1, x2, y2): self.popdown() self._menu_shell.set_active(None) grid = self._shell.get_grid() self._menu = self.create_menu() self._menu.connect('enter-notify-event', self._menu_enter_notify_event_cb) self._menu.connect('leave-notify-event', self._menu_leave_notify_event_cb) [grid_x1, grid_y1] = grid.convert_from_screen(x1, y1) [grid_x2, grid_y2] = grid.convert_from_screen(x2, y2) strategy = self._menu_strategy [grid_x, grid_y] = strategy.get_menu_position(self._menu, grid_x1, grid_y1, grid_x2, grid_y2) grid.set_constraints(self._menu, grid_x, grid_y, self._menu.get_width(), self._menu.get_height()) self._menu.show() self._menu_shell.set_active(self) def _menu_enter_notify_event_cb(self, widget, event): self._hover_menu = True def _menu_leave_notify_event_cb(self, widget, event): self._hover_menu = False if self._popdown_on_leave: self.popdown() def _start_popdown_timeout(self): self._stop_popdown_timeout() self._popdown_sid = gobject.timeout_add(1000, self._popdown_timeout_cb) def _stop_popdown_timeout(self): if self._popdown_sid > 0: gobject.source_remove(self._popdown_sid) self._popdown_sid = 0 def _motion_notify_event_cb(self, item, event): if event.detail == hippo.MOTION_DETAIL_ENTER: self._motion_notify_enter() elif event.detail == hippo.MOTION_DETAIL_LEAVE: self._motion_notify_leave() def _motion_notify_enter(self): self._stop_popdown_timeout() [x, y] = self.get_context().translate_to_widget(self) [width, height] = self.get_allocation() self._popup(x, y, width, height) def _motion_notify_leave(self): self._start_popdown_timeout() def _popdown_timeout_cb(self): self._popdown_sid = 0 if not self._hover_menu: self.popdown() else: self._popdown_on_leave = True return False