149 lines
5.6 KiB
Python
149 lines
5.6 KiB
Python
from tkinter import Frame, ttk
|
|
from PIL import ImageDraw
|
|
|
|
from .base import View
|
|
from .button_interface import ButtonInterface
|
|
from .draw_utils import draw_long_press
|
|
|
|
class ButtonsManager(View):
|
|
|
|
def __init__(self, parent, im_size, center, curr_view):
|
|
self.current_view: ButtonInterface = curr_view
|
|
self.long_press_threshold = 1000 # milliseconds
|
|
self.left_press_start = None
|
|
self.right_press_start = None
|
|
self.both_press_start = None
|
|
self.left_press_job = None
|
|
self.right_press_job = None
|
|
self.both_press_job = None
|
|
|
|
super().__init__(parent, im_size, center)
|
|
|
|
def init_ui(self, parent):
|
|
self.ui = Frame(parent)
|
|
self.left_button = ttk.Button(self.ui, text="Left")
|
|
# Bind mouse events for press detection
|
|
self.left_button.bind("<Button-1>", self._left_button_press_start)
|
|
self.left_button.bind("<ButtonRelease-1>", self._left_button_press_end)
|
|
self.left_button.pack(side="left")
|
|
|
|
self.right_button = ttk.Button(self.ui, text="Right")
|
|
# Bind mouse events for press detection
|
|
self.right_button.bind("<Button-1>", self._right_button_press_start)
|
|
self.right_button.bind("<ButtonRelease-1>", self._right_button_press_end)
|
|
self.right_button.pack(side="right")
|
|
|
|
self.both_buttons = ttk.Button(self.ui, text="Both")
|
|
# Bind mouse events for press detection
|
|
self.both_buttons.bind("<Button-1>", self._both_buttons_press_start)
|
|
self.both_buttons.bind("<ButtonRelease-1>", self._both_buttons_press_end)
|
|
self.both_buttons.pack()
|
|
|
|
self.ui.pack()
|
|
|
|
def update_weight(self, weight: float) -> None:
|
|
im = self.bkg_im.copy()
|
|
draw = ImageDraw.Draw(im)
|
|
|
|
has_buttons = self.current_view.has_button()
|
|
# Draw left button
|
|
if has_buttons[0]:
|
|
draw.circle((10, 10), 2, fill='black')
|
|
self.current_view.render_left_press(draw, 20, 10)
|
|
|
|
if has_buttons[1]:
|
|
y = self.size[1] - 10
|
|
draw_long_press(draw, (10, y))
|
|
self.current_view.render_left_long_press(draw, 24, y)
|
|
|
|
# Draw right button
|
|
if has_buttons[2]:
|
|
draw.circle((self.size[0] - 10, 10), 2, fill='black')
|
|
self.current_view.render_right_press(draw, self.size[0] - 20, 4)
|
|
|
|
if has_buttons[3]:
|
|
y = self.size[1] - 10
|
|
draw_long_press(draw, (self.size[0] - 10, y))
|
|
self.current_view.render_right_long_press(draw, self.size[0] - 24, y)
|
|
|
|
return im
|
|
|
|
############ BUTTON ACTIONS ###########
|
|
def left_button_press(self):
|
|
self.current_view.left_press()
|
|
|
|
def left_button_long_press(self):
|
|
self.current_view.left_long_press()
|
|
|
|
def right_button_press(self):
|
|
self.current_view.right_press()
|
|
|
|
def right_button_long_press(self):
|
|
self.current_view.right_long_press()
|
|
|
|
def both_buttons_long_press(self):
|
|
self.current_view.both_long_press()
|
|
|
|
############ BUTTON PRESS HANDLERS ###########
|
|
def _left_button_press_start(self, event):
|
|
"""Handle left button press start"""
|
|
self.left_press_start = self.ui.after_idle(lambda: None) # Get current time reference
|
|
# Schedule long press detection
|
|
self.left_press_job = self.ui.after(self.long_press_threshold, self._left_long_press_detected)
|
|
|
|
def _left_button_press_end(self, event):
|
|
"""Handle left button press end"""
|
|
if self.left_press_job:
|
|
self.ui.after_cancel(self.left_press_job)
|
|
self.left_press_job = None
|
|
# If we get here, it was a short press
|
|
self.left_button_press()
|
|
self.left_press_start = None
|
|
|
|
def _left_long_press_detected(self):
|
|
"""Called when long press threshold is reached for left button"""
|
|
self.left_press_job = None
|
|
self.left_button_long_press()
|
|
|
|
|
|
|
|
def _right_button_press_start(self, event):
|
|
"""Handle right button press start"""
|
|
self.right_press_start = self.ui.after_idle(lambda: None) # Get current time reference
|
|
# Schedule long press detection
|
|
self.right_press_job = self.ui.after(self.long_press_threshold, self._right_long_press_detected)
|
|
|
|
def _right_button_press_end(self, event):
|
|
"""Handle right button press end"""
|
|
if self.right_press_job:
|
|
self.ui.after_cancel(self.right_press_job)
|
|
self.right_press_job = None
|
|
# If we get here, it was a short press
|
|
self.right_button_press()
|
|
self.right_press_start = None
|
|
|
|
def _right_long_press_detected(self):
|
|
"""Called when long press threshold is reached for right button"""
|
|
self.right_press_job = None
|
|
self.right_button_long_press()
|
|
|
|
|
|
|
|
def _both_buttons_press_start(self, event):
|
|
"""Handle both buttons press start"""
|
|
self.both_press_start = self.ui.after_idle(lambda: None) # Get current time reference
|
|
# Schedule long press detection
|
|
self.both_press_job = self.ui.after(self.long_press_threshold, self._both_buttons_long_press_detected)
|
|
|
|
def _both_buttons_press_end(self, event):
|
|
"""Handle both buttons press end"""
|
|
if self.both_press_job:
|
|
self.ui.after_cancel(self.both_press_job)
|
|
self.both_press_job = None
|
|
# If we get here, it was a short press (no action defined)
|
|
self.both_press_start = None
|
|
|
|
def _both_buttons_long_press_detected(self):
|
|
"""Called when long press threshold is reached for both buttons"""
|
|
self.both_press_job = None
|
|
self.both_buttons_long_press() |