make views combineable

frontend/views/combined_view.py

@@ -0,0 +1,69 @@
+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
+
+class CombinedView(tk.Frame):
+    def __init__(self, parent,
+                 tare_command=None, calibrate_command=None,
+                 **kwargs):
+        super().__init__(parent, **kwargs)
+        self.views = []
+        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)
+
+        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):
+        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 = []
+        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.logical_and(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)