Clocks/AnalogClockGenerator.py at main · StreamController/Clocks · GitHub

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
import math
from PIL import Image, ImageDraw
import time
from datetime import datetime

class AnalogClockGenerator:
    def __init__(self, hour_hand_color: tuple = (255, 255, 255, 255),
                 minute_hand_color: tuple = (255, 255, 255, 255),
                 second_hand_color: tuple = (255, 255, 255, 255),
                 origin_color: tuple = (255, 255, 255, 255),
                 background_color: tuple = (0, 0, 0, 0),
                 hour_markings_width: int = 5,
                 hour_hand_width: int = 7,
                 minute_hand_width: int = 5,
                 second_hand_width: int = 3
                 ):

        self.hour_hand_color = hour_hand_color
        self.minute_hand_color = minute_hand_color
        self.second_hand_color = second_hand_color
        self.origin_color = origin_color
        self.background_color = background_color

        self.hour_markings_width = hour_markings_width
        self.hour_hand_width = hour_hand_width
        self.minute_hand_width = minute_hand_width
        self.second_hand_width = second_hand_width

    def get_current_clock(self) -> Image.Image:
        now = datetime.now()

        hour = now.hour
        minute = now.minute
        second = now.second

        return self.get_clock(hour, minute, second)

    def get_clock(self, hour: int, minute: int, second: int) -> Image.Image:
        canvas = Image.new("RGBA", (500, 500), color=self.background_color)
        draw = ImageDraw.Draw(canvas)

        center = canvas.size[0] / 2

        # Draw hour markings
        self.draw_hour_markings(draw, center, width=self.hour_markings_width)

        # Draw origin in the center
        radius = 15
        draw.ellipse((center - radius, center - radius, center + radius, center + radius), fill=self.origin_color)

        # Draw hour hand
        lenght = 150
        angle = hour * (360 / 12) + minute * (360 / 12) / 60
        self.draw_hand(draw, center, angle, lenght, self.hour_hand_color, self.hour_hand_width)

        # Draw minute hand
        lenght = 175
        angle = minute * (360 / 60) + second * (360 / 60) / 60
        self.draw_hand(draw, center, angle, lenght, self.minute_hand_color, self.minute_hand_width)

        # Draw second hand
        lenght = 200
        angle = second * (360 / 60)
        self.draw_hand(draw, center, angle, lenght, self.second_hand_color, self.second_hand_width)


        return canvas

    # -------------- #
    # Helper methods #
    # -------------- #

    def draw_hand(self, draw: ImageDraw.ImageDraw, center, angle, length, color: tuple = (255, 255, 255, 255), width: int = 1):
        x_end = center + length * math.sin(math.radians(angle))
        y_end = center - length * math.cos(math.radians(angle))
        draw.line((center, center, x_end, y_end), fill=color, width=width)

    def draw_hour_markings(self, draw: ImageDraw.ImageDraw, center, start_distance: int = 170, length: int = 50, color: tuple = (255, 255, 255, 255), width: int = 5):
        for i in range(0, 12):
            angle = i * (360 / 12)
            x_start = center + start_distance * math.sin(math.radians(angle))
            y_start = center + start_distance * math.cos(math.radians(angle))

            x_end = center + (start_distance + length) * math.sin(math.radians(angle))
            y_end = center + (start_distance + length) * math.cos(math.radians(angle))
            draw.line((x_start, y_start, x_end, y_end), fill=color, width=width)