from machine import Pin, PWM
import time

buzzer = PWM(Pin(13))
buzzer2 = PWM(Pin(12))

red = Pin(18, Pin.OUT)
amber = Pin(19, Pin.OUT)
green = Pin(20, Pin.OUT)

led_array = [red, amber, green]

#for i in range(3):
#	led = Pin(18 + i, Pin.OUT)
#	led_array.append(led)
#	print(led_array[i].value())

# Note collection:
B0 = 31
C1 = 33
CS1 = 35
D1 = 37
DS1 = 39
E1 = 41
F1 = 44
FS1 = 46
G1 = 49
GS1 = 52
A1 = 55
AS1 = 58
B1 = 62
C2 = 65
CS2 = 69
D2 = 73
DS2 = 78
E2 = 82
F2 = 87
FS2 = 93
G2 = 98
GS2 = 104
A2 = 110
AS2 = 117
B2 = 123
C3 = 131
CS3 = 139
D3 = 147
DS3 = 156
E3 = 165
F3 = 175
FS3 = 185
G3 = 196
GS3 = 208
A3 = 220
AS3 = 233
B3 = 247
C4 = 262
CS4 = 277
D4 = 294
DS4 = 311
E4 = 330
F4 = 349
FS4 = 370
G4 = 392
GS4 = 415
A4 = 440
AS4 = 466
B4 = 494
C5 = 523
CS5 = 554
D5 = 587
DS5 = 622
E5 = 659
F5 = 698
FS5 = 740
G5 = 784
GS5 = 831
A5 = 880
AS5 = 932
B5 = 988
C6 = 1047
CS6 = 1109
D6 = 1175
DS6 = 1245
E6 = 1319
F6 = 1397
FS6 = 1480
G6 = 1568
GS6 = 1661
A6 = 1760
AS6 = 1865
B6 = 1976
C7 = 2093
CS7 = 2217
D7 = 2349
DS7 = 2489
E7 = 2637
F7 = 2794
FS7 = 2960
G7 = 3136
GS7 = 3322
A7 = 3520
AS7 = 3729
B7 = 3951
C8 = 4186
CS8 = 4435
D8 = 4699
DS8 = 4978

vol = 500

def play(note, volume, delay1, delay2, leds):

	buzzer2.duty_u16(volume)
	buzzer2.freq(note)
	if leds == "red":
		led_array[0].value(1)
	elif leds == "amber":
		led_array[1].value(1)
	elif leds == "green":
		led_array[2].value(1)
	elif leds == "all":
		for i in range(3):
			led_array[i].value(1)
	else:
		for i in range(3):
			led_array[i].value(0)
	time.sleep(delay1)
	buzzer2.duty_u16(0)
	for i in range(3):
	    led_array[i].value(0)
	time.sleep(delay2)

def songLoops(section):
	if section == 1:
		play(G5, vol, 0.6, 0.075, "amber")
		play(G5, vol, 0.6, 0.075, "amber")
		play(FS5, vol, 0.6, 0.075, "amber")
		play(G5, vol, 0.6, 0.075, "amber")
		play(G5, vol, 0.6, 0.075, "amber")
		play(G5, vol, 0.3, 0.075, "amber")
		play(FS5, vol, 0.6, 0.075, "amber")
		play(G5, vol, 0.6, 0.075, "amber")
		play(FS5, vol, 0.3, 0.075, "amber")
	elif section == 2:
		play(G4, vol, 0.7, 0.1, "amber")
		play(D4, vol, 0.6, 0.2, "red")
		play(G4, vol, 0.7, 0.1, "amber")
		play(D4, vol, 0.6, 0.2, "red")
		play(G4, vol, 0.4, 0.1, "amber")
		play(A4, vol, 0.4, 0.1, "green")
		play(G4, vol, 0.4, 0.4, "amber")
		play(FS4, vol, 0.4, 0.4, "amber")
		play(G4, vol, 0.4, 0.4, "amber")
		play(FS4, vol, 0.4, 0.4, "amber")
		play(C5, vol, 0.4, 0.1, "amber")
		play(B4, vol, 0.4, 0.1, "amber")

		play(G4, vol, 0.7, 0.1, "amber")
		play(D4, vol, 0.7, 0.1, "red")
		play(B3, vol, 0.6, 0.2, "red")
		play(G4, vol, 0.7, 0.1, "amber")
		play(C4, vol, 0.4, 0.1, "amber")
		play(B3, vol, 0.5, 0, "amber")

		play(A4, vol, 0.7, 0.1, "amber")
		play(B4, vol, 0.6, 0.1, "amber")
		play(A4, vol, 0.7, 0.1, "amber")
		play(B4, vol, 0.6, 0.1, "amber")
		play(A4, vol, 0.25, 0, "amber")
		play(B4, vol, 0.25, 0, "amber")
		play(D5, vol, 0.25, 0, "amber")
		play(B4, vol, 0.2, 0.06, "amber")

songLoops(2)

# Duty to 0 to turn the buzzer off
buzzer2.duty_u16(0)