1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
# -*- coding: utf-8 -*-
import pygame
from math import pi, cos, sin, radians
from classes.simple_vector import Vector2
class Percentage:
def __init__(self, unit_size, scale, colors, b_colors, numbers):
self.size = unit_size * scale
self.center = [self.size // 2, self.size // 2]
self.numbers = numbers
self.type = type
self.canvas = pygame.Surface([self.size, self.size - 1], flags=pygame.SRCALPHA)
self.set_colors(colors, b_colors)
self.redraw()
def get_canvas(self):
return self.canvas
def set_colors(self, colors, b_colors):
self.colors = colors
self.b_colors = b_colors
def redraw(self):
self.canvas.fill((0, 0, 0, 0))
self.draw_slices()
def update_values(self, numbers):
self.numbers = numbers
self.canvas.fill((0, 0, 0, 0))
self.draw_slices()
def draw_slices(self):
r = self.size // 2 - self.size // 10
m = (self.size - r * 2) // 2
angles = []
angle_start = []
angle_start_float = []
for i in range(len(self.numbers)):
angles.append(self.numbers[i] * 3.6)
if i == 0:
angle_start.append(270)
angle_start_float.append(270)
else:
angle_start.append(int(angle_start_float[-1] + angles[i-1]))
angle_start_float.append(angle_start_float[-1] + angles[i-1])
for i in range(len(self.numbers)):
offset = 0
# Center and radius of pie chart
cx = m + r
cy = m + r
#create vector to push the pieces away from the centre towards middle of the arch
centre_vect = Vector2().from_points([cx, cy],
[cx + int(round(r * cos(((angles[i] - angles[i] / 2.0) + angle_start_float[i]) * pi / 180))),
cy + int(round(r * sin(((angles[i] - angles[i] / 2.0) + angle_start_float[i]) * pi / 180)))])
centre_vect.normalize()
#first point
p = [(cx + centre_vect[0] * offset, cy + centre_vect[1] * offset)]
# Get points on arc
for n in range(angle_start[i], int(round(angle_start_float[i] + angles[i])), 3):
x = cx + int(round(r * cos(n * pi / 180))) + centre_vect[0] * offset
y = cy + int(round(r * sin(n * pi / 180))) + centre_vect[1] * offset
p.append((x, y))
#final point on arc
n = int(round(angle_start_float[i] + angles[i]))
x = cx + int(round(r * cos(n * pi / 180))) + centre_vect[0] * offset
y = cy + int(round(r * sin(n * pi / 180))) + centre_vect[1] * offset
p.append((x, y))
#last point
p.append((cx + centre_vect[0] * offset, cy + centre_vect[1] * offset))
# Draw pie segment
if len(p) > 2 and angles[i] > 0:
pygame.draw.polygon(self.canvas, self.colors[i], p)
if self.numbers[i] not in [0, 100]:
pygame.draw.polygon(self.canvas, self.b_colors[i], p, 2)
else:
pygame.draw.polygon(self.canvas, self.b_colors[i], p[1:-1], 2)
|
