63 lines
1.3 KiB
Python
63 lines
1.3 KiB
Python
|
import math
|
||
|
|
||
|
import cairo
|
||
|
|
||
|
class SnowflakeLayout:
|
||
|
_BASE_RADIUS = 65
|
||
|
_CHILDREN_FACTOR = 1
|
||
|
_FLAKE_DISTANCE = 6
|
||
|
_BORDER = 20
|
||
|
|
||
|
def __init__(self):
|
||
|
self._root = None
|
||
|
self._children = []
|
||
|
|
||
|
def set_root(self, icon):
|
||
|
self._root = icon
|
||
|
|
||
|
def add_child(self, icon):
|
||
|
self._children.append(icon)
|
||
|
self._layout()
|
||
|
|
||
|
def remove_child(self, icon):
|
||
|
self._children.remove(icon)
|
||
|
self._layout()
|
||
|
|
||
|
def _layout_root(self):
|
||
|
[width, height] = self._root.get_size_request()
|
||
|
|
||
|
matrix = cairo.Matrix(1, 0, 0, 1, 0, 0)
|
||
|
matrix.translate(self._cx, self._cy)
|
||
|
self._root.set_transform(matrix)
|
||
|
|
||
|
def _layout_child(self, child, index):
|
||
|
r = self._r
|
||
|
if (len(self._children) > 10):
|
||
|
r += SnowflakeLayout._FLAKE_DISTANCE * (index % 3)
|
||
|
|
||
|
angle = 2 * math.pi / len(self._children) * index
|
||
|
|
||
|
x = self._cx + math.cos(angle) * r
|
||
|
y = self._cy + math.sin(angle) * r
|
||
|
|
||
|
matrix = cairo.Matrix(1, 0, 0, 1, 0, 0)
|
||
|
matrix.translate(x, y)
|
||
|
child.set_transform(matrix)
|
||
|
|
||
|
def _layout(self):
|
||
|
self._r = SnowflakeLayout._BASE_RADIUS + \
|
||
|
SnowflakeLayout._CHILDREN_FACTOR * len(self._children)
|
||
|
|
||
|
c = self._r + SnowflakeLayout._BORDER + \
|
||
|
SnowflakeLayout._FLAKE_DISTANCE * 2
|
||
|
|
||
|
self._cx = c
|
||
|
self._cy = c
|
||
|
|
||
|
self._layout_root()
|
||
|
|
||
|
index = 0
|
||
|
for child in self._children:
|
||
|
self._layout_child(child, index)
|
||
|
index += 1
|