-
Notifications
You must be signed in to change notification settings - Fork 0
/
gui.py
151 lines (136 loc) · 5.19 KB
/
gui.py
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
141
142
143
144
145
146
147
148
149
150
151
import matplotlib
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from matplotlib.figure import Figure
import tkinter as tk
from tkinter import ttk
import cv2
import PIL.Image, PIL.ImageTk
buttons = []
checkboxes = []
checkbox_variables = []
radio_button_groups = []
radio_button_variables = []
messages = []
message_variables = []
click_command = False
class App(tk.Tk):
def __init__(self, *args, **kwargs):
tk.Tk.__init__(self, *args, **kwargs)
global window
window = self
for button in buttons:
row, column, text, press, release = button
b = ttk.Button(self, text = text)
b.bind("<ButtonPress-1>",
(lambda press: (lambda event: press()))(press))
b.bind("<ButtonRelease-1>",
(lambda release: (lambda event: release()))(release))
b.grid(row = row, column = column)
for checkbox in checkboxes:
row, column, text, command = checkbox
variable = tk.IntVar()
checkbox_variables.append(variable)
ttk.Checkbutton(self,
text = text,
variable = variable,
command = command).grid(row = row, column = column)
for radio_button_group in radio_button_groups:
radio_buttons, command = radio_button_group
variable = tk.IntVar()
radio_button_variables.append(variable)
for radio_button in radio_buttons:
row, column, text, value = radio_button
ttk.Radiobutton(
self,
text = text,
variable = variable,
value = value,
command = command).grid(row = row, column = column)
for message in messages:
row, column, columnspan = message
variable = tk.StringVar()
message_variables.append(variable)
m = tk.Label(self)
m.config(font = ("Verdana", 14), textvariable = variable)
m.grid(row = row, column = column, columnspan = columnspan)
global canvas
if use_matplotlib:
canvas = FigureCanvasTkAgg(figure, self)
if fixed_size:
axes.set_xlim(0, 1)
axes.set_ylim(0, 1)
canvas.show()
if click_command:
canvas.mpl_connect('button_press_event',
lambda event: click_command(event.xdata,
event.ydata))
canvas.get_tk_widget().grid(row = button_rows,
columnspan = button_columns)
else:
canvas = tk.Canvas(
self, width = window_width, height = window_height)
canvas.grid(row = button_rows, columnspan = button_columns)
self.bind("<Control-c>", lambda event: done())
self.bind("<Escape>", lambda event: done())
def show_image(self, bgr_image):
rgb_image = cv2.cvtColor(bgr_image, cv2.COLOR_BGR2RGB)
self.photo = PIL.ImageTk.PhotoImage(
image = PIL.Image.fromarray(rgb_image))
canvas.create_image(0, 0, image = self.photo, anchor = tk.NW)
def nothing(): return
def done(): exit()
def add_button(row, column, text, press, release):
buttons.append([row, column, text, press, release])
def add_checkbox(row, column, text, command):
checkboxes.append([row, column, text, command])
i = len(checkboxes)-1
return lambda: checkbox_variables[i].get()==1
def add_radio_button_group(buttons, command):
radio_button_groups.append([buttons, command])
i = len(radio_button_groups)-1
return lambda: radio_button_variables[i].get()
def add_message(row, column, columnspan):
messages.append([row, column, columnspan])
i = len(messages)-1
return lambda text: message_variables[i].set(text)
def add_click(command):
global click_command
click_command = command
def get_axes(): return axes
def redraw():
if fixed_size:
axes.set_xlim(0, 1)
axes.set_ylim(0, 1)
canvas.show()
def get_window(): return window
def start_fixed_size_matplotlib(width, height, rows, columns):
global use_matplotlib
use_matplotlib = True
global fixed_size, figure, axes, button_rows, button_columns
fixed_size = True
button_rows = rows
button_columns = columns
global axes
figure = Figure(figsize = (width, height), dpi = 100)
axes = figure.add_subplot(111)
App().mainloop()
def start_variable_size_matplotlib(width, height, rows, columns):
global use_matplotlib
use_matplotlib = True
global fixed_size, figure, axes, button_rows, button_columns
fixed_size = False
button_rows = rows
button_columns = columns
global axes
figure = Figure(figsize = (width, height), dpi = 100)
axes = figure.add_subplot(111)
App().mainloop()
def start_video(width, height, rows, columns):
global use_matplotlib
use_matplotlib = False
global window_width, window_height, button_rows, button_columns
window_width = width
window_height = height
button_rows = rows
button_columns = columns
App().mainloop()