clockabout/draw.py at master · zeroaltitude/clockabout · 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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
#!/usr/local/bin/python3

import math

from PIL import Image, ImageDraw


scale = 5
width = scale
limit_x = 36 * scale
limit_y = 36 * scale
abs_offset_x = 2 * scale
abs_offset_y = 2 * scale
initial_offset_x = 1 * scale
initial_offset_y = 1 * scale
center_point = (int(limit_x / 2) - scale, int(limit_y / 2) - scale)
points = [0 for _ in range(60)]


def get_coord_for_second(s):
    # simple: index into points with offset
    offset = 7
    seconds = 60
    return points[(s + offset) % seconds]


def increment(point):
    if point[0] == 0:
        p0 = point[0] + math.floor(scale/2) + 1
    else:
        p0 = point[0] - math.floor(scale/2) - 1
    if point[1] == 0:
        p1 = point[1] + math.floor(scale/2) + 1
    else:
        p1 = point[1] - math.floor(scale/2) - 1
    return (p0, p1)


def draw_face(draw):
    global points
    state = 0
    for i in range(15):
        pointq1 = (i * (2 * scale) + initial_offset_x + abs_offset_x, 0 + abs_offset_y)
        pointq2 = (limit_x - scale - abs_offset_x, i * (2 * scale) + initial_offset_y + abs_offset_y)
        pointq3 = (i * (2 * scale) + initial_offset_x + abs_offset_x, limit_y - scale - abs_offset_y)
        pointq4 = (0 + abs_offset_x, i * (2 * scale) + initial_offset_y + abs_offset_y)
        draw.line([pointq1, increment(pointq1)], fill="black", width=width)
        draw.line([pointq2, increment(pointq2)], fill="black", width=width)
        draw.line([pointq3, increment(pointq3)], fill="black", width=width)
        draw.line([pointq4, increment(pointq4)], fill="black", width=width)
        points[i] = pointq1
        points[i + 15] = pointq2
        points[44 - i] = pointq3
        points[59 - i] = pointq4

    # colorize the hour points
    for j in range(0, 60, 5):
        c = get_coord_for_second(j)
        draw.point(c, fill="blue")
        if c[0] == 0 + abs_offset_x:
            draw.line([(c[0], c[1]), (c[0] + scale + 2, c[1])], fill="blue", width=width)
        elif c[0] == limit_x - scale - abs_offset_x:
            draw.line([(c[0], c[1]), (c[0] - scale - 2, c[1])], fill="blue", width=width)
        elif c[1] == 0 + abs_offset_y:
            draw.line([(c[0], c[1]), (c[0], c[1] + scale + 2)], fill="blue", width=width)
        elif c[1] == limit_y - scale - abs_offset_y:
            draw.line([(c[0], c[1]), (c[0], c[1] - scale - 2)], fill="blue", width=width)


def draw_hand(draw, pos, length=0, color=None):
    dest_point = get_coord_for_second(pos)
    if center_point[0] == dest_point[0]:
        # infinite slope, simply alter y
        if dest_point[1] == 0 + abs_offset_y:
            dest_point = (dest_point[0], center_point[1] - length)
        else:
            dest_point = (dest_point[0], center_point[1] + length)
    elif center_point[1] == dest_point[1]:
        # zero slope, simply alter x
        if dest_point[0] == 0 + abs_offset_x:
            dest_point = (center_point[0] - length, dest_point[1])
        else:
            dest_point = (center_point[0] + length, dest_point[1])
    else:
        # slope
        slope = (dest_point[1] - center_point[1]) / (dest_point[0] - center_point[0])
        # solve for length: x**2 + y**2 = length
        # y = mx
        # (1 + m**2)x**2 = length**2
        xcoord = math.sqrt((length ** 2) / (1 + (slope ** 2)))
        ycoord = slope * xcoord
        xmathmod = math.ceil
        ymathmod = math.ceil
        if pos > 30:
            # 3rd and 4th quadrants: reverse the coord modifier
            xcoord = xcoord * -1.
            ycoord = ycoord * -1.
        if xcoord > 0 and ycoord > 0:
            xmathmod = math.floor
        elif xcoord < 0 and ycoord < 0:
            ymathmod = math.floor
        elif xcoord > 0 and ycoord < 0:
            pass
        elif xcoord < 0 and ycoord > 0:
            ymathmod = math.floor
        dest_point = (center_point[0] + xmathmod(xcoord), center_point[1] + ymathmod(ycoord))
    draw.line([center_point, dest_point], fill=color, width=width)


def draw_second_hand(draw, s):
    draw_hand(draw, s, length=17 * scale, color="#990000")


def draw_minute_hand(draw, m):
    draw_hand(draw, m, length=15 * scale, color="#12111d")


def draw_hour_hand(draw, h):
    draw_hand(draw, h * 5, length=8 * scale, color="#12111d")


def draw_clock(word, h, m, s):
    im = Image.new("RGBA", (limit_x - scale - 1, limit_y - scale - 1), "#cfccc7")
    # trans = [(255, 255, 255, 0) for _ in range(limit_y * limit_x)]
    # im.putdata(trans)
    draw = ImageDraw.Draw(im)
    draw_face(draw)
    draw_second_hand(draw, s)
    draw_minute_hand(draw, m)
    draw_hour_hand(draw, h)
    im.save("png/%s.png" % word, "PNG")


if __name__ == "__main__":
    draw_clock("test", 1, 9, 13)
    #draw_clock("test", 3, 30, 45)
    #draw_clock("test", 4, 25, 29)
    #draw_clock("test", 7, 40, 44)
    #draw_clock("test", 10, 55, 59)
    #draw_clock("test", 10, 37, 31)