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