import pygame
from colors import *
from fonts import *
from knobs import Knobs
from lozenge_button import LozengeButton

class GuiMain:

    def __init__(self):
        # pygame setup
        pygame.init()
        self.screen = pygame.display.set_mode((800, 480))
        #self.screen = pygame.display.set_mode((0,0), pygame.FULLSCREEN)
        self.screenH = self.screen.get_height()
        self.screenW = self.screen.get_width()

        fx_mode_labels = [
            'Strings',
            'Beat',
            'L-Loop',
            'R-Loop',
            'Jack',
            'Athmo'
        ]

        # Create and position the fx_mode buttons
        self.fx_mode_buttons = []
        
        for i in range(len(fx_mode_labels)):
            x = self.screenW * 2.5 / 19
            y = i * self.screenH / 7.5 + 45
            r_x = self.screenW * 0.35 / 19
            r_y = (self.screenH-20) / 16
            self.fx_mode_buttons.append(LozengeButton(self, fx_mode_labels[i], (x,y), r_x, r_y))

        self.fx_mode_buttons[0].focused = True

        self.knobs = Knobs(self)

        self.clock = pygame.time.Clock()
        self.running = True
        self.run()

    def set_fx_mode(self, mode):
        for b in self.fx_mode_buttons:
            if b.name != mode: b.focused = False

    def run(self):
        while self.running:
            # poll for events
            # pygame.QUIT event means the user clicked X to close your window
            for event in pygame.event.get():
                if event.type == pygame.QUIT:
                    self.running = False

            # fill the screen with a color to wipe away anything from last frame
            self.screen.fill("black")

            # RENDER YOUR GAME HERE

            self.knobs.display()

            for button in self.fx_mode_buttons:
                button.display()
            
            for i in range(1,4):
                pygame.draw.line(self.screen, color_primary_dark,
                                (self.fx_mode_buttons[i].radius_x*0.5 + 60, self.fx_mode_buttons[i].y),
                                (self.fx_mode_buttons[i].x - self.fx_mode_buttons[i].radius_x * 1.5,
                                 self.fx_mode_buttons[i].y))
                pygame.draw.circle(self.screen, color_primary,
                                   (40, self.fx_mode_buttons[i].y), 20)
            pygame.draw.rect(self.screen, color_primary,
                             ((20, self.fx_mode_buttons[0].y-20),(40, 40)))

            pygame.draw.polygon(self.screen, "white", [(0, self.screenH/2), (self.screenW*3 / 19, self.screenH), (0, self.screenH)])
            # flip() the display to put your work on screen
            pygame.display.flip()

            self.clock.tick(60)  # limits FPS to 60
            
            # exit when esc is pressed
            for event in pygame.event.get():
               if event.type == pygame.KEYDOWN:
                   if event.key == pygame.K_ESCAPE:
                       pygame.quit()
                       return

        pygame.quit()

GuiMain()