67 lines
1.6 KiB
Python
67 lines
1.6 KiB
Python
import math
|
|
|
|
import goocanvas
|
|
|
|
class PieceItem(goocanvas.Path):
|
|
def __init__(self, angle_start, angle_end, **kwargs):
|
|
goocanvas.Path.__init__(self, **kwargs)
|
|
self._angle_start = angle_start
|
|
self._angle_end = angle_end
|
|
|
|
self.set_property('fill-color', '#e8e8e8')
|
|
self.set_property('stroke-color', '#d8d8d8')
|
|
self.set_property('line-width', 4)
|
|
|
|
def construct(self):
|
|
r = self.get_parent().get_radius()
|
|
|
|
data = 'M0,0 '
|
|
|
|
dx = r * math.cos(self._angle_start)
|
|
dy = - r * math.sin(self._angle_start)
|
|
|
|
data += 'l%f,%f ' % (dx, dy)
|
|
|
|
dx = r * math.cos(self._angle_end)
|
|
dy = - r * math.sin(self._angle_end)
|
|
|
|
data += 'A%f,%f 0 0,0 %f,%f ' % (r, r, dx, dy)
|
|
|
|
data += 'z'
|
|
|
|
self.set_property('data', data)
|
|
|
|
class DonutItem(goocanvas.Group):
|
|
def __init__(self, radius, **kwargs):
|
|
goocanvas.Group.__init__(self, **kwargs)
|
|
self._radius = radius
|
|
self._angle_start = 0
|
|
|
|
bg = goocanvas.Ellipse(radius_x=radius, radius_y=radius,
|
|
fill_color='#c2c3c5', line_width=0)
|
|
self.add_child(bg)
|
|
|
|
fg_radius = radius / 2
|
|
fg = goocanvas.Ellipse(radius_x=fg_radius, radius_y=fg_radius,
|
|
fill_color='#d8d8d8', line_width=0)
|
|
self.add_child(fg)
|
|
|
|
def add_piece(self, perc):
|
|
angle_end = self._angle_start + perc * 2 * math.pi / 100
|
|
piece_item = PieceItem(self._angle_start, angle_end)
|
|
self._angle_start = angle_end
|
|
|
|
# FIXME can't override set_parent on the
|
|
# PieceItem and there is no signal.
|
|
self.add_child(piece_item, 1)
|
|
piece_item.construct()
|
|
|
|
return piece_item
|
|
|
|
def remove_piece(self, piece_item):
|
|
index = self.find(piece_item)
|
|
self.remove_child(index)
|
|
|
|
def get_radius(self):
|
|
return self._radius
|