User:Walter/favoriteslayout.py
Jump to navigation
Jump to search
# Copyright (C) 2008 One Laptop Per Child # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA import logging import math import hashlib import gobject import gtk import hippo from sugar.graphics import style from sugar import activity from view.home.grid import Grid _logger = logging.getLogger('FavoritesLayout') _CELL_SIZE = 4 _BASE_SCALE = 1000 class FavoritesLayout(gobject.GObject, hippo.CanvasLayout): __gtype_name__ = 'FavoritesLayout' def __init__(self): gobject.GObject.__init__(self) self.box = None self.fixed_positions = {} def do_set_box(self, box): self.box = box def do_get_height_request(self, for_width): return 0, gtk.gdk.screen_height() - style.GRID_CELL_SIZE def do_get_width_request(self): return 0, gtk.gdk.screen_width() def compare_activities(self, icon_a, icon_b): return 0 def append(self, icon, locked=False): self.box.insert_sorted(icon, 0, self.compare_activities) if hasattr(icon, 'fixed_position'): relative_x, relative_y = icon.fixed_position if relative_x >= 0 and relative_y >= 0: min_width_, width = self.box.get_width_request() min_height_, height = self.box.get_height_request(width) self.fixed_positions[icon] = \ (int(relative_x * _BASE_SCALE / float(width)), int(relative_y * _BASE_SCALE / float(height))) def remove(self, icon): if icon in self.fixed_positions: del self.fixed_positions[icon] self.box.remove(icon) def move_icon(self, icon, x, y, locked=False): if icon not in self.box.get_children(): raise ValueError('Child not in box.') if hasattr(icon, 'get_bundle_id') and hasattr(icon, 'get_version'): min_width_, width = self.box.get_width_request() min_height_, height = self.box.get_height_request(width) registry = activity.get_registry() registry.set_activity_position( icon.get_bundle_id(), icon.get_version(), x * width / float(_BASE_SCALE), y * height / float(_BASE_SCALE)) self.fixed_positions[icon] = (x, y) def do_allocate(self, x, y, width, height, req_width, req_height, origin_changed): raise NotImplementedError() def allow_dnd(self): return False class RandomLayout(FavoritesLayout): __gtype_name__ = 'RandomLayout' def __init__(self): FavoritesLayout.__init__(self) min_width_, width = self.do_get_width_request() min_height_, height = self.do_get_height_request(width) self._grid = Grid(width / _CELL_SIZE, height / _CELL_SIZE) self._grid.connect('child-changed', self.__grid_child_changed_cb) def __grid_child_changed_cb(self, grid, child): child.emit_request_changed() def append(self, icon, locked=False): FavoritesLayout.append(self, icon, locked) min_width_, child_width = icon.get_width_request() min_height_, child_height = icon.get_height_request(child_width) min_width_, width = self.box.get_width_request() min_height_, height = self.box.get_height_request(width) if icon in self.fixed_positions: x, y = self.fixed_positions[icon] x = min(x, width - child_width) y = min(y, height - child_height) elif hasattr(icon, 'get_bundle_id'): name_hash = hashlib.md5(icon.get_bundle_id()) x = int(name_hash.hexdigest()[:5], 16) % (width - child_width) y = int(name_hash.hexdigest()[-5:], 16) % (height - child_height) else: x = None y = None if x is None or y is None: self._grid.add(icon, child_width / _CELL_SIZE, child_height / _CELL_SIZE) else: self._grid.add(icon, child_width / _CELL_SIZE, child_height / _CELL_SIZE, x / _CELL_SIZE, y / _CELL_SIZE) def remove(self, icon): self._grid.remove(icon) FavoritesLayout.remove(self, icon) def move_icon(self, icon, x, y, locked=False): self._grid.move(icon, x / _CELL_SIZE, y / _CELL_SIZE, locked) FavoritesLayout.move_icon(self, icon, x, y, locked) def do_allocate(self, x, y, width, height, req_width, req_height, origin_changed): for child in self.box.get_layout_children(): # We need to always get requests to not confuse hippo min_w_, child_width = child.get_width_request() min_h_, child_height = child.get_height_request(child_width) rect = self._grid.get_child_rect(child.item) child.allocate(rect.x * _CELL_SIZE, rect.y * _CELL_SIZE, child_width, child_height, origin_changed) def allow_dnd(self): return True _MINIMUM_RADIUS = style.XLARGE_ICON_SIZE / 2 + style.DEFAULT_SPACING + \ style.STANDARD_ICON_SIZE * 2 _MAXIMUM_RADIUS = (gtk.gdk.screen_height() - style.GRID_CELL_SIZE) / 2 - \ style.STANDARD_ICON_SIZE - style.DEFAULT_SPACING class RingLayout(FavoritesLayout): __gtype_name__ = 'RingLayout' def __init__(self): FavoritesLayout.__init__(self) self._locked_children = {} def append(self, icon, locked=False): FavoritesLayout.append(self, icon, locked) if locked: child = self.box.find_box_child(icon) self._locked_children[child] = (0, 0) def remove(self, icon): child = self.box.find_box_child(icon) if child in self._locked_children: del self._locked_children[child] FavoritesLayout.remove(self, icon) def move_icon(self, icon, x, y, locked=False): FavoritesLayout.move_icon(self, icon, x, y, locked) if locked: child = self.box.find_box_child(icon) self._locked_children[child] = (x, y) def _calculate_radius_and_icon_size(self, children_count): angle = 2 * math.pi / children_count # what's the radius required without downscaling? distance = style.STANDARD_ICON_SIZE + style.DEFAULT_SPACING icon_size = style.STANDARD_ICON_SIZE if children_count == 1: radius = 0 else: radius = math.sqrt(distance ** 2 / (math.sin(angle) ** 2 + (math.cos(angle) - 1) ** 2)) if radius < _MINIMUM_RADIUS: # we can upscale, if we want icon_size += style.STANDARD_ICON_SIZE * \ (0.5 * (_MINIMUM_RADIUS - radius) / _MINIMUM_RADIUS) radius = _MINIMUM_RADIUS elif radius > _MAXIMUM_RADIUS: radius = _MAXIMUM_RADIUS # need to downscale. what's the icon size required? distance = math.sqrt((radius * math.sin(angle)) ** 2 + \ (radius * (math.cos(angle) - 1)) ** 2) icon_size = distance - style.DEFAULT_SPACING return radius, icon_size def _calculate_position(self, radius, icon_size, index, children_count): width, height = self.box.get_allocation() # go around 2 times, e.g., 4 pi angle = index * (4 * math.pi / children_count) - math.pi / 2 x = radius * math.cos(angle) + (width - icon_size) / 2 y = radius * math.sin(angle) + (height - icon_size - style.GRID_CELL_SIZE) / 2 return x, y def _get_children_in_ring(self): children_in_ring = [child for child in self.box.get_layout_children() \ if child not in self._locked_children] return children_in_ring def _update_icon_sizes(self): children_in_ring = self._get_children_in_ring() radius_, icon_size = \ self._calculate_radius_and_icon_size(len(children_in_ring)) for child in children_in_ring: child.item.props.size = icon_size def do_allocate(self, x, y, width, height, req_width, req_height, origin_changed): children_in_ring = self._get_children_in_ring() if children_in_ring: radius, icon_size = \ self._calculate_radius_and_icon_size(len(children_in_ring)) for n in range(len(children_in_ring)): child = children_in_ring[n] x, y = self._calculate_position(radius, icon_size, n, len(children_in_ring)) # We need to always get requests to not confuse hippo min_w_, child_width = child.get_width_request() min_h_, child_height = child.get_height_request(child_width) child.allocate(int(x), int(y), child_width, child_height, origin_changed) # decrease the radius slightly with each icon radius -= 6; for child in self._locked_children.keys(): x, y = self._locked_children[child] # We need to always get requests to not confuse hippo min_w_, child_width = child.get_width_request() min_h_, child_height = child.get_height_request(child_width) child.allocate(int(x), int(y), child_width, child_height, origin_changed) def compare_activities(self, icon_a, icon_b): if hasattr(icon_a, 'installation_time') and \ hasattr(icon_b, 'installation_time'): return icon_b.installation_time - icon_a.installation_time else: return 0