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
from classes.simple_vector import Vector2
class Percentage:
def __init__(self, unit_size, scale, color1, color2, border_color1, border_color2, number):
self.size = unit_size * scale
self.center = [self.size // 2, self.size // 2]
self.color1 = color1
self.color2 = color2
self.border_color1 = border_color1
self.border_color2 = border_color2
self.number = number
self.type = type
self.canvas = pygame.Surface([self.size, self.size - 1], flags=pygame.SRCALPHA)
self.canvas.fill((0, 0, 0, 0))
self.draw_circles()
def get_canvas(self):
return self.canvas
def update_values(self, number):
self.number = number
self.canvas.fill((0, 0, 0, 0))
self.draw_circles()
def draw_circles(self):
r = self.size // 2 - self.size // 10
m = (self.size - r * 2) // 2
angles = (self.number * 3.6, (100 - self.number) * 3.6)
#angle_start = [int(angles[0]), 0] #int(round((self.number * angle) / 2.0))
#angle_start = [int(round(angles[0] / 2.0 - angles[0])), int(round(angles[0] / 2.0)) + int(angles[0])]
angle_start = [int(round(angles[0] / 2.0 - angles[0])), int(round(angles[0] / 2.0 - angles[0]))+int(angles[0])]
angle_start_float = [angles[0] / 2.0 - angles[0], angles[0] / 2.0]
#angle_start = [270, 270+int(angles[0])]
for i in range(2):
if i == 0:
col = self.color1
bd_col = self.border_color1
offset = self.size // 40
else:
col = self.color2
bd_col = self.border_color2
offset = self.size // 40
# 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], angle_start[i] + int(round(angles[i])), 1):
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[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, col, p)
if self.number not in [0, 100]:
pygame.draw.polygon(self.canvas, bd_col, p, 2)
else:
pygame.draw.polygon(self.canvas, bd_col, p[1:-1], 2)
|
