81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
import io
|
|
import tkinter as tk
|
|
from tkinter import Frame, Canvas, ttk, PhotoImage
|
|
|
|
from PIL import Image, ImageChops
|
|
|
|
from ..config import DISPLAY_TYPES
|
|
from . import NumberView, CircleView, TimerView
|
|
|
|
class CombinedView(tk.Frame):
|
|
def __init__(self, parent,
|
|
tare_command=None, calibrate_command=None,
|
|
**kwargs):
|
|
super().__init__(parent, **kwargs)
|
|
self.views = []
|
|
self.timer_view = None # Timer view is always active
|
|
self.tare_command = tare_command
|
|
self.calibrate_command = calibrate_command
|
|
|
|
self.actions = Frame(self)
|
|
self.actions.pack()
|
|
self.calibrate_button = ttk.Button(self.actions, text="Calibrate", command=calibrate_command)
|
|
self.calibrate_button.pack()
|
|
self.tare_button = ttk.Button(self.actions, text="Tare", command=tare_command)
|
|
self.tare_button.pack()
|
|
|
|
self.im_size = (168, 144)
|
|
self.center = (168 // 2, 144 // 2)
|
|
|
|
# Create timer view that's always active
|
|
self.timer_view = TimerView(self.actions, self.im_size, self.center)
|
|
|
|
self.canvas = Canvas(self, width=168, height=144, background='white',
|
|
highlightthickness=1, highlightbackground="black")
|
|
self.canvas.pack()
|
|
|
|
|
|
def update_views(self, selected_types: DISPLAY_TYPES):
|
|
# Clear only the selectable views, not the timer
|
|
for v in self.views:
|
|
if v.ui is not None:
|
|
v.ui.destroy()
|
|
self.views.clear()
|
|
|
|
if selected_types & DISPLAY_TYPES.NUMBER:
|
|
number_view = NumberView(self.actions, self.im_size, self.center)
|
|
self.views.append(number_view)
|
|
|
|
if selected_types & DISPLAY_TYPES.CIRCLE:
|
|
circle_view = CircleView(self.actions, self.im_size, self.center)
|
|
self.views.append(circle_view)
|
|
|
|
|
|
def refresh(self, weight: float):
|
|
ims = []
|
|
|
|
# Always include timer view
|
|
if self.timer_view:
|
|
timer_im = self.timer_view.update_weight(weight)
|
|
ims.append(timer_im)
|
|
|
|
# Add other selected views
|
|
for view in self.views:
|
|
im = view.update_weight(weight)
|
|
ims.append(im)
|
|
|
|
self.canvas.delete("all")
|
|
# Combine images by logical_and
|
|
if ims:
|
|
combined_im = ims[0]
|
|
for im in ims[1:]:
|
|
combined_im = ImageChops.invert(ImageChops.logical_xor(combined_im, im))
|
|
|
|
# Convert PIL image to bytes
|
|
buffer = io.BytesIO()
|
|
combined_im.save(buffer, format='PNG')
|
|
buffer.seek(0)
|
|
|
|
# Load into PhotoImage and display on canvas
|
|
self.photo = PhotoImage(data=buffer.getvalue())
|
|
self.canvas.create_image(0, 0, anchor="nw", image=self.photo) |