import pygame
from knob import *
from colors import *

# pygame setup
pygame.init()
screenW = 561
screenH = 325
screen = pygame.display.set_mode((screenW, screenH))

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

# Create and position the knobs
knobs_radius = screenW / 25
knobs_spacing = knobs_radius * 2
knobs_x = (screenW - (knobs_radius + knobs_spacing) * 2) / 2
x = knobs_x
knobs_y = (screenH - (knobs_radius + knobs_spacing) * 3) / 2
y = knobs_y
knobs = []
for i in range(8):
    knobs.append(Knob(screen, color_primary, knobs_radius, (x, y)))
    y += knobs_radius + knobs_spacing
    if i == 3 or i == 6:
        x += knobs_radius + knobs_spacing
        y = knobs_y

pygame.font.init() # you have to call this at the start, 
                   # if you want to use this module.
font_helvetica = pygame.font.SysFont('Helvetica', 30)
font_helvetica16 = pygame.font.SysFont('Helvetica', 15)

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

def display_label(index):
    fx_label = font_helvetica.render('REVERB WET', False, color_primary_light)
    fx_value = font_helvetica.render('[ 86% ]', False, color_primary_light)

    label_x = screenW - fx_label.get_width() - 20
    label_y = (screenH - fx_label.get_height() - fx_value.get_height()) / 2
    screen.blit(fx_label, (label_x, label_y))
    screen.blit(fx_value, (label_x, screenH / 2))
    pygame.draw.line(screen, color_primary_dark, (label_x - 10, label_y), (label_x - 10, screenH - label_y), 2)
    
    line_points = []
    line_points.append(knobs[index].get_position())

    if (index == 0 or index == 1 or index == 4 or index == 5):
        x = knobs[index].get_position()[0] + knobs_radius * 1.5
        y = knobs[index].get_position()[1] + knobs_radius * 1.5
        line_points.append((x, y))
    elif (index == 2 or index == 3 or index == 6):
        x = knobs[index].get_position()[0] + knobs_radius * 1.5
        y = knobs[index].get_position()[1] - knobs_radius * 1.5
        line_points.append((x, y))
    elif (index == 7):
        x = knobs[index].get_position()[0] - knobs_radius * 1.5
        y = knobs[index].get_position()[1] + knobs_radius * 1.5
        line_points.append((x, y))

    if (index == 0 or index == 3):
        line_points.append((knobs_x + knobs_radius * 1.5, screenH/2))
    elif (index == 4 or index == 6 or index == 7):
        line_points.append((knobs[4].get_position()[0] + knobs_radius * 1.5, screenH/2))

    line_points.append((label_x - 10, screenH/2))

    for i in range(len(line_points)-1):
        pygame.draw.line(screen, color_primary_dark, line_points[i], line_points[i+1], 2)
    
    knobs[index].set_value(0.86)
    knobs[index].set_color(color_primary_light)

def draw_lozi(x, y, color, outlined = False):
    rX = screenW * 0.35 / 19
    rY = (screenH-20) / 16
    w = 2 if outlined else 0
    pygame.draw.polygon(screen, color, [(x-rX, y), (x, y-rY), (x+rX, y), (x, y+rY)], w)

while 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:
            running = False

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

    # RENDER YOUR GAME HERE

    display_label(4)

    for i in range(8):
        knobs[i].display()

    pygame.draw.rect(screen, color_primary_dark, ((10, 10), (screenW * 1.5 / 19, screenH - 18)))
    fx_mode = 1
    for i in range(6):
        x = screenW * 2.5 / 19
        y = i * screenH / 7.3 + 35
        if i == fx_mode:
            draw_lozi(x, y, color_primary)
            fx_mode_label = font_helvetica16.render(fx_modes[i], False, color_primary)
            screen.blit(fx_mode_label, (x+20, y-8))
        else:
            draw_lozi(x, y, color_primary, True)
            fx_mode_label = font_helvetica16.render(str(fx_modes[i])[0:1], False, color_primary)
            screen.blit(fx_mode_label, (x-4, y-7))
    """
    for i in range(7):
        pygame.draw.rect(screen, color_primary, ((screenW * 1.5 / 19 + 10, i * (screenH-20)/7 + 10), (screenW * 1.5 / 19 - 10 , (screenH-20)/7 - 10)))
    """

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

    clock.tick(60)  # limits FPS to 60

pygame.quit()