Modular-multiplication-circle/mod_mult_square.py at main · oonap0oo/Modular-multiplication-circle · 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
# Modular multiplication square
import tkinter as tk
from tkinter import simpledialog
from PIL import Image, ImageDraw, ImageFont
from math import *


# this function draws a actual image
# output is done both on the tkinter canvas for real time viewing
# and on a PIL image to be able to export to a PNG image file later
def draw_image(slider_value):
    # calculate coordinates of points on square and store in lists
    x_list = []; y_list = []
    side = int(m.get() / 4)
    for n in range(side):
        x_list.append( x_center - radius  + int(2 * radius * n / side) )
        y_list.append( y_center - radius )
    for n in range(side):
        x_list.append( x_center + radius )
        y_list.append( y_center - radius  + int(2 * radius * n / side))
    for n in range(side):
        x_list.append( x_center + radius  - int(2 * radius * n / side) )
        y_list.append( y_center + radius )
    for n in range(side):
        x_list.append( x_center - radius )
        y_list.append( y_center + radius  - int(2 * radius * n / side) )

    # clear screen tkinter
    canvas1.delete("all")
    # clear image PIL
    draw.rectangle([(0,0),(screen_width,screen_height)],fill = "black", outline = None)
    # draw square on tkinter canvas
    canvas1.create_rectangle(x_center - radius, y_center - radius,
                        x_center + radius, y_center + radius,
                        outline = line_color, fill = "")
    # draw square on PIL image
    draw.rectangle([(x_center - radius, y_center - radius),
                  (x_center + radius, y_center + radius)],
                 fill = None, outline = "white")
    # draw lines
    m_actual = len(x_list)
    for n in range(m_actual):
        value = (n * p.get()) % m_actual
        # draw line on tkinter canvas
        canvas1.create_line(x_list[value], y_list[value], x_list[n], y_list[n], fill = line_color)
        # draw line on PIL image
        draw.line([(x_list[value], y_list[value]), (x_list[n], y_list[n])], fill = "white")
    # generate text to add
    txt = f"Modular multiplication square, m = {m_actual}, p = {p.get()}"
    # text on tkinter canvas
    canvas1.create_text(30, 5, text = txt,
                        anchor = "nw", font = ("TkFixedFont",10,"normal"), fill = "white")
    # text on PIL image
    draw.text((30,5), txt, font = font_PIL, align = "left", fill = "white")

# export image to PNG file
def export_png():
    m_actual = int(m.get() / 4) * 4
    file_name = f"{default_file_name}_{m_actual}_{p.get()}.png"
    file_name = simpledialog.askstring("Export image as PNG", "File name:", parent = root1,
                                           initialvalue = file_name)
    if file_name is not None:
        img.save(file_name, "PNG")

# event handler for mouse click
def click_handler(event):
    if event.num == 1: # left mouse button
        pass
    elif event.num == 3: # right mouse button
        export_png()


# parameters
m_max = 400 # modulus
screen_width = 1000; screen_height = 1000
radius = 450
line_color = "white"
default_file_name = "mod_mult_square"

# center of square
x_center = screen_width // 2; y_center = screen_height // 2

# font to use for PIL image
font_PIL = ImageFont.truetype(r"FreeMono.ttf", 20)

# make tkinter root and canvas objects
root1 = tk.Tk()
root1.resizable(False, False)
root1.title("Modular multiplication square using Python and tkinter, right click to save")
m = tk.IntVar()
m.set(m_max)
p = tk.IntVar()
p.set(351)
slider_m = tk.Scale(root1, variable = m, label = "m", command = draw_image,
                   from_ = 2, to = m_max, orient = tk.VERTICAL, length = screen_height - 100)
slider_m.pack(side = tk.LEFT)
slider_p = tk.Scale(root1, variable = p, label = "p", command = draw_image,
                   from_ = 2, to = m.get(), orient = tk.VERTICAL, length = screen_height - 100)
slider_p.pack(side = tk.LEFT)
canvas1 = tk.Canvas(root1, background = "black",
                    height = screen_height, width = screen_width)
canvas1.pack()

# create a new PIL Image object
img = Image.new("RGB",(screen_width, screen_height),
                "black")
draw = ImageDraw.Draw(img)

# font to use for PIL image
font_PIL = ImageFont.truetype(r"FreeMono.ttf", 23)

# define event handler for mouse click
root1.bind("<Button>", click_handler)

# first image drawing, subsequent images are drawn when the sliders are moved
draw_image(0)

# tkinter main loop
root1.mainloop()