UIElems.py

'''
This file defines some classes used to create UI elements (mostly buttons)
'''

#Container to hold all the elements used
class UIContainer():
    def __init__(self, x, y, elems = None):
        self.x = x
        self.y = y
        self.elems = [] if elems == None else elems
        self.container = None
    
    #Add an element to the container
    def add(self, elem):
        elem.container = self
        self.elems.append(elem)
    
    #Call onClick() of any buttons clicked
    def onClick(self, app, event):
        for elem in self.elems:
            if elem.clickInBounds(event): 
                if isinstance(elem, UIContainer):
                    elem.onClick(app, event)
                else:
                    elem.onClick(app)
    
    def draw(self, app, canvas):
        for elem in self.elems:
            elem.draw(app, canvas)
    
    def positionOffset(self):
        if self.container == None:
            return self.x, self.y
        else:
            offX, offY = self.container.positionOffset()
            return self.x + offX, self.y + offY

    #This is just so you can put a container inside a container
    def clickInBounds(self, event): return True

#Button class
class UIButton():
    def __init__(self, x, y, width, height, label = "", container=None, color = 'grey'):
        self.x, self.y = x, y
        self.width, self.height = width, height
        self.container = container
        self.label = label
        self.color = color
    
    def draw(self, app, canvas):
        containerX, containerY = self.container.positionOffset()
        #position is relative to container
        bX = self.x + containerX
        bY = self.y + containerY

        canvas.create_rectangle(bX, bY, bX + self.width, bY + self.height,
            fill=self.color, width=0)

        canvas.create_text((2*bX + self.width)/2, (2*bY + self.height)/2,
            font="Arial 12", text=self.label)
    
    #Return whether a click was on the button
    def clickInBounds(self, event):
        containerX, containerY = self.container.positionOffset()

        bXMin = self.x + containerX
        bXMax = bXMin + self.width

        bYMin = self.y + containerY
        bYMax = bYMin + self.height

        return (bXMin <= event.x <= bXMax and
                bYMin <= event.y <= bYMax)

    #Subclasses should make this actually do something
    def onClick(self, app):
        print(f"Button {self.label} clicked.")

class UICheckbox(UIButton):
    def __init__(self, x, y, width, height, checked=False):
        super().__init__(x,y,width,height)
        self.checked = checked
    
    def onClick(self, app):
        if self.checked:
            self.checked = False
            self.container.checked = False
            self.onUnCheck(app)
        else:
            self.checked = True
            self.container.checked = True
            self.onCheck(app)

    def draw(self, app, canvas):
        containerX, containerY = self.container.positionOffset()
        #position is relative to container
        bX = self.x + containerX
        bY = self.y + containerY

        if self.checked:
            cx, cy = bX + self.width/2, bY + self.height/2
            r = min(self.width, self.height)/2
            canvas.create_oval(cx-r, cy-r, cx+r, cy+r, fill="red", width=0)

        canvas.create_rectangle(bX, bY, bX + self.width, bY + self.height,
            width="2")

    #Subclasses should make these functions do stuff
    def onCheck(self, app):
        pass

    def onUnCheck(self, app):
        pass

    

class UILabeledCheckbox(UIContainer):
    def __init__(self, x, y, label, checked=False, boxSize = 14, margin = 7):
        super().__init__(x,y)
        self.checked = checked
        self.boxSize = boxSize
        self.margin = margin
        self.label=label

        self.checkButton = UICheckbox(0, 0, boxSize, boxSize, checked=checked)
        self.add(self.checkButton)

    
    def draw(self, app, canvas):
        super().draw(app, canvas)
        posX, posY = self.positionOffset()
        #draw the label
        canvas.create_text(posX+self.margin+self.boxSize, posY-self.margin+3, 
            text=self.label, font="Helvetica 14", anchor = 'nw')