gui.py

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()