import goocanvas import gobject from sugar.canvas.IconItem import IconItem from sugar.canvas.Grid import Grid class _MenuShell: def __init__(self): self._menu_controller = None def set_active(self, controller): if self._menu_controller: self._menu_controller.popdown() self._menu_controller = controller 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(IconItem, goocanvas.Item): _menu_shell = _MenuShell() def __init__(self, grid, **kwargs): IconItem.__init__(self, **kwargs) self._grid = grid self._menu = None self._hover_menu = False self._popdown_on_leave = False self._popdown_sid = 0 self._menu_strategy = _MenuStrategy() def popdown(self): if self._menu: self._menu.destroy() self._menu = None def set_menu_strategy(self, strategy): self._menu_strategy = strategy def _popup(self, x1, y1, x2, y2): self.popdown() MenuIcon._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() MenuIcon._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 do_create_view(self, canvas, parent_view): view = IconItem.do_create_view(self, canvas, parent_view) view.connect('enter-notify-event', self._enter_notify_event_cb, canvas) view.connect('leave-notify-event', self._leave_notify_event_cb) return view 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 _enter_notify_event_cb(self, view, target, event, canvas): self._stop_popdown_timeout() [x1, y1] = canvas.convert_to_pixels(view.get_bounds().x1, view.get_bounds().y1) [x2, y2] = canvas.convert_to_pixels(view.get_bounds().x2, view.get_bounds().y2) [window_x, window_y] = canvas.window.get_origin() x1 += window_x y1 += window_y x2 += window_x y2 += window_y self._popup(int(x1), int(y1), int(x2), int(y2)) def _popdown_timeout_cb(self): self._popdown_sid = 0 if not self._hover_menu: self.popdown() else: self._popdown_on_leave = True return False def _leave_notify_event_cb(self, view, target, event): self._start_popdown_timeout()