103 lines
3 KiB
Python
103 lines
3 KiB
Python
|
from gui.GUIMasterData import GUIMasterData
|
||
|
|
from gui.GUIMasterGUI import GUIMasterGUI
|
||
|
|
import tkinter as tk
|
||
|
|
from tkinter import ttk
|
||
|
|
|
||
|
|
|
||
|
|
class GUIOutputModeData(GUIMasterData):
|
||
|
|
|
||
|
|
enable_cam: bool = True
|
||
|
|
enable_yolo: bool = True
|
||
|
|
enable_contour: bool = True
|
||
|
|
enable_percept: bool = True
|
||
|
|
|
||
|
|
|
||
|
|
class GUIOutputModeGUI(GUIMasterGUI):
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
tk_root: tk.Tk | tk.ttk.Labelframe | tk.ttk.Frame,
|
||
|
|
name: str = "Output Filter",
|
||
|
|
row_id: int = 0,
|
||
|
|
column_id: int = 3,
|
||
|
|
data_class=None,
|
||
|
|
) -> None:
|
||
|
|
super().__init__(
|
||
|
|
tk_root,
|
||
|
|
name=name,
|
||
|
|
row_id=row_id,
|
||
|
|
column_id=column_id,
|
||
|
|
data_class=data_class,
|
||
|
|
)
|
||
|
|
|
||
|
|
width_element: int = 10
|
||
|
|
width_label: int = 20
|
||
|
|
width_button_extra: int = 5
|
||
|
|
|
||
|
|
# option0 ->
|
||
|
|
self.option0_var = tk.IntVar()
|
||
|
|
self.option0_var.set(int(self.data.enable_cam))
|
||
|
|
self.option0 = ttk.Checkbutton(
|
||
|
|
self.frame,
|
||
|
|
text="CAM",
|
||
|
|
onvalue=1,
|
||
|
|
offvalue=0,
|
||
|
|
variable=self.option0_var,
|
||
|
|
command=self.selection_changed,
|
||
|
|
width=(width_label + width_element + width_button_extra),
|
||
|
|
)
|
||
|
|
self.option0.pack(anchor=tk.W)
|
||
|
|
# <- option0
|
||
|
|
|
||
|
|
# option1 ->
|
||
|
|
self.option1_var = tk.IntVar()
|
||
|
|
self.option1_var.set(int(self.data.enable_yolo))
|
||
|
|
self.option1 = ttk.Checkbutton(
|
||
|
|
self.frame,
|
||
|
|
text="YOLO",
|
||
|
|
onvalue=1,
|
||
|
|
offvalue=0,
|
||
|
|
variable=self.option1_var,
|
||
|
|
command=self.selection_changed,
|
||
|
|
width=(width_label + width_element + width_button_extra),
|
||
|
|
)
|
||
|
|
self.option1.pack(anchor=tk.W)
|
||
|
|
# <- option1
|
||
|
|
|
||
|
|
# option2 ->bool(self.option0_var.get())
|
||
|
|
self.option2_var = tk.IntVar()
|
||
|
|
self.option2_var.set(int(self.data.enable_contour))
|
||
|
|
self.option2 = ttk.Checkbutton(
|
||
|
|
self.frame,
|
||
|
|
text="Contour",
|
||
|
|
onvalue=1,
|
||
|
|
offvalue=0,
|
||
|
|
variable=self.option2_var,
|
||
|
|
command=self.selection_changed,
|
||
|
|
width=(width_label + width_element + width_button_extra),
|
||
|
|
)
|
||
|
|
self.option2.pack(anchor=tk.W)
|
||
|
|
# <- option2
|
||
|
|
|
||
|
|
# option3 ->
|
||
|
|
self.option3_var = tk.IntVar()
|
||
|
|
self.option3_var.set(int(self.data.enable_percept))
|
||
|
|
self.option3 = ttk.Checkbutton(
|
||
|
|
self.frame,
|
||
|
|
text="Percept",
|
||
|
|
onvalue=1,
|
||
|
|
offvalue=0,
|
||
|
|
variable=self.option3_var,
|
||
|
|
command=self.selection_changed,
|
||
|
|
width=(width_label + width_element + width_button_extra),
|
||
|
|
)
|
||
|
|
self.option3.pack(anchor=tk.W)
|
||
|
|
# <- option3
|
||
|
|
|
||
|
|
def selection_changed(self):
|
||
|
|
self.data.enable_cam = bool(self.option0_var.get())
|
||
|
|
self.data.enable_yolo = bool(self.option1_var.get())
|
||
|
|
self.data.enable_contour = bool(self.option2_var.get())
|
||
|
|
self.data.enable_percept = bool(self.option3_var.get())
|
||
|
|
|
||
|
|
self.data.data_changed = True
|