from machine import Pin, PWM
import utime
import neopixel

# Define the frequencies for each note (in Hz)
NOTE_C4 = 261
NOTE_D4 = 294
NOTE_E4 = 329
NOTE_F4 = 349
NOTE_G4 = 392
NOTE_A4 = 440
NOTE_B4 = 494
NOTE_C5 = 523

# Define the durations for each note (in milliseconds)
DURATION_QUARTER = 250
DURATION_HALF = 500
DURATION_WHOLE = 1000

# Define the pin for sound output
SOUND_PIN = 15
# Define the pin for NeoPixel
NEOPIXEL_PIN = 13
NUM_PIXELS = 256

def play_tune(frequency, duration):
  """
  Play a single note.
  """
  pwm.freq(frequency)
  pwm.duty_u16(3000)  # 50% duty cycle
  utime.sleep_ms(duration)
  pwm.duty_u16(0)  # Turn off the sound
  utime.sleep_ms(50)   # Add a small delay to distinguish between consecutive notes

def play_tunes(tunes, durations):
  """
  Play a sequence of notes with specified durations.
  """
  for tune, duration in zip(tunes, durations):
    play_tune(tune, duration)

def display_reducing_square():
  """
  Display a reducing square animation on NeoPixel.
  """
  for size in range(16, 1, -8):
    rainbow_square(size)
    pixels.write() # 更新 LED 顯示
    utime.sleep(1)

def rainbow_square(size):
  """
  Fill NeoPixel with a rainbow-colored square of specified size.
  """
  for i in range(size):
    color = wheel((i * 256 // size) & 255)
    # Fill top row
    for col in range(i, size):
      pixels[i * 16 + col] = color
    # Fill bottom row
    for col in range(i, size):
      pixels[(size - 1) * 16 + col] = color
    # Fill left column
    for row in range(i + 1, size - 1):
      pixels[row * 16 + i] = color
    # Fill right column
    for row in range(i + 1, size - 1):
      pixels[row * 16 + size - 1] = color

def wheel(wheel_pos):
  """
  Generate a color based on a position in a color wheel.
  """
  wheel_pos = 255 - wheel_pos
  if wheel_pos < 85:
    return (255 - wheel_pos * 3, 0, wheel_pos * 3)
  if wheel_pos < 170:
    wheel_pos -= 85
    return (0, wheel_pos * 3, 255 - wheel_pos * 3)
  wheel_pos -= 170
  return (wheel_pos * 3, 255 - wheel_pos * 3, 0)

# Define musical tunes and their durations
tunes_list = [
  ([NOTE_C4, NOTE_D4, NOTE_E4, NOTE_F4, NOTE_G4, NOTE_A4, NOTE_B4, NOTE_C5], [DURATION_QUARTER] * 8),
  ([NOTE_E4, NOTE_D4, NOTE_C4, NOTE_D4, NOTE_E4, NOTE_E4, NOTE_E4, NOTE_D4, NOTE_D4, NOTE_D4, NOTE_E4, NOTE_E4, NOTE_E4, NOTE_E4, NOTE_D4, NOTE_D4, NOTE_C4, NOTE_D4, NOTE_E4, NOTE_E4, NOTE_E4, NOTE_E4, NOTE_D4, NOTE_D4, NOTE_E4, NOTE_D4, NOTE_C4], [DURATION_QUARTER] * 14 + [DURATION_HALF] * 2),
  ([NOTE_C4, NOTE_E4, NOTE_G4, NOTE_A4, NOTE_G4, NOTE_E4, NOTE_C4], [DURATION_HALF] * 7),
  ([NOTE_E4, NOTE_G4, NOTE_F4, NOTE_C4, NOTE_D4, NOTE_A4, NOTE_B4, NOTE_E4], [DURATION_QUARTER] * 8 + [DURATION_WHOLE]),
  ([NOTE_C4, NOTE_D4, NOTE_E4, NOTE_F4, NOTE_G4, NOTE_A4, NOTE_B4, NOTE_C5], [DURATION_HALF] * 8),
  ([NOTE_E4, NOTE_A4, NOTE_C4, NOTE_B4, NOTE_G4, NOTE_D4, NOTE_E4, NOTE_F4] + [NOTE_C4, NOTE_D4, NOTE_E4, NOTE_F4, NOTE_G4, NOTE_A4, NOTE_B4, NOTE_C5], [DURATION_QUARTER] * 8 + [DURATION_QUARTER] * 7 + [DURATION_WHOLE])
]

# Initialize NeoPixel strip
pixels = neopixel.NeoPixel(Pin(NEOPIXEL_PIN), NUM_PIXELS)

# Initialize PWM for sound
pwm = PWM(Pin(SOUND_PIN))

# Play tunes and display animation
for tunes, durations in tunes_list:
   play_tunes(tunes, durations)
   utime.sleep_ms(1000)  # Pause between tunes

# Start NeoPixel animation
while True:
  display_reducing_square()