"""
 Press and hold the button for a second to start 
    playing "we wish you a merry christmas"
    and again to stop. for now the snowflake animation pauses
    since wokwi doesn't like multithreading
"""

import time
import _thread

time.sleep(0.1) # Wait for USB to become ready

from neopixel import Neopixel
import random

from machine import PWM
from utime import sleep

from melodies import * 
from notes import *    


num_leds = 10
pin = 0
state_machine = 0
strip = Neopixel(num_leds, state_machine, pin, mode="RGB")
strip.brightness(50)

led_structure = [
    [0, 1, 2],  # Row 1: D1, D2, D3
    [7, 3],     # Row 2: D8, D4
    [6, 5, 4],  # Row 3: D7, D6, D5
    [9, 8]      # Row 4: D10, D9
]
wait = 0.5

from machine import Pin
import time

button = Pin(14, Pin.IN, Pin.PULL_UP)

snowflakes = [0] * num_leds

song = False



# https://github.com/twisst/Music-for-Raspberry-Pi-Pico/blob/main/play.py
buzzer = PWM(Pin(15))

track = 23
volume = 600


def playtone(frequency):
    buzzer.duty_u16(volume)
    buzzer.freq(frequency)

def be_quiet():
    buzzer.duty_u16(0)

def duration(tempo, t):
    
    wholenote = (60000 / tempo) * 4

    if t > 0:
      noteDuration = wholenote // t
    elif (t < 0):
      noteDuration = wholenote // abs(t)
      noteDuration *= 1.5
    
    return noteDuration

def playsong(mysong):

    try:
        
        print(mysong[0]) 
        tempo = mysong[1] 

        for thisNote in range(2, len(mysong), 2):
            
            noteduration = duration(tempo, int(mysong[thisNote+1]))
            
            if (mysong[thisNote] == "REST"):
                be_quiet()
            else:
                playtone(notes[mysong[thisNote]])
            
            sleep(noteduration*0.9/1000) 
            be_quiet()
            sleep(noteduration*0.1/1000) 

            if button.value() == 0:
                print("escaping")
                song = not song
                while button.value() == 0: time.sleep(0.01)
                return
    except:
        be_quiet()



while True:
    if button.value() == 0:
        while button.value() == 0: time.sleep(0.01)
        print("pressed")
        song = not song
    
    color = (255, 255, 255)
    for led in led_structure[0]:
        if random.random() < 0.5:
            snowflakes[led] = color
        else:
            snowflakes[led] = (0, 0, 0)

    # this bit is courtesy copilot
    for row in range(len(led_structure) - 1, 0, -1):
        for i in range(len(led_structure[row])):
            snowflakes[led_structure[row][i]] = snowflakes[led_structure[row-1][i % len(led_structure[row-1])]]
    
    print(snowflakes)

    for i in range(len(snowflakes)):
        if snowflakes[i] != 0:
            strip.set_pixel(i, snowflakes[i])
        else:
            strip.set_pixel(i, (0, 0, 0))

    strip.show()
    time.sleep(wait)

    if song:
        playsong(melody[track])