// Define pin numbers for lights, buzzer, and button
const int greenLight = 2;
const int redLight1 = 3;
const int redLight2 = 4;
const int buzzer = 10;
const int button = 1;

// Change to 0.5 for a slower version of the song, 1.25 for a faster version
const float songSpeed = 1.0;

// Defining frequency of each music note
#define NOTE_B0 31
#define NOTE_C1 33
#define NOTE_CS1 35
#define NOTE_D1 37
#define NOTE_DS1 39
#define NOTE_E1 41
#define NOTE_F1 44
#define NOTE_FS1 46
#define NOTE_G1 49
#define NOTE_GS1 52
#define NOTE_A1 55
#define NOTE_AS1 58
#define NOTE_B1 62
#define NOTE_C2 65
#define NOTE_CS2 69
#define NOTE_D2 73
#define NOTE_DS2 78
#define NOTE_E2 82
#define NOTE_F2 87
#define NOTE_FS2 93
#define NOTE_G2 98
#define NOTE_GS2 104
#define NOTE_A2 110
#define NOTE_AS2 117
#define NOTE_B2 123
#define NOTE_C3 131
#define NOTE_CS3 139
#define NOTE_D3 147
#define NOTE_DS3 156
#define NOTE_E3 165
#define NOTE_F3 175
#define NOTE_FS3 185
#define NOTE_G3 196
#define NOTE_GS3 208
#define NOTE_A3 220
#define NOTE_AS3 233
#define NOTE_B3 247
#define NOTE_C4 262
#define NOTE_CS4 277
#define NOTE_D4 294
#define NOTE_DS4 311
#define NOTE_E4 330
#define NOTE_F4 349
#define NOTE_FS4 370
#define NOTE_G4 392
#define NOTE_GS4 415
#define NOTE_A4 440
#define NOTE_AS4 466
#define NOTE_B4 494
#define NOTE_C5 523
#define NOTE_CS5 554
#define NOTE_D5 587
#define NOTE_DS5 622
#define NOTE_E5 659
#define NOTE_F5 698
#define NOTE_FS5 740
#define NOTE_G5 784
#define NOTE_GS5 831
#define NOTE_A5 880
#define NOTE_AS5 932
#define NOTE_B5 988
#define NOTE_C6 1047
#define NOTE_CS6 1109
#define NOTE_D6 1175
#define NOTE_DS6 1245
#define NOTE_E6 1319
#define NOTE_F6 1397
#define NOTE_FS6 1480
#define NOTE_G6 1568
#define NOTE_GS6 1661
#define NOTE_A6 1760
#define NOTE_AS6 1865
#define NOTE_B6 1976
#define NOTE_C7 2093
#define NOTE_CS7 2217
#define NOTE_D7 2349
#define NOTE_DS7 2489
#define NOTE_E7 2637
#define NOTE_F7 2794
#define NOTE_FS7 2960
#define NOTE_G7 3136
#define NOTE_GS7 3322
#define NOTE_A7 3520
#define NOTE_AS7 3729
#define NOTE_B7 3951
#define NOTE_C8 4186
#define NOTE_CS8 4435
#define NOTE_D8 4699
#define NOTE_DS8 4978

// Pac-Man theme song notes and durations
int notes[] = {
  NOTE_B4, NOTE_B5, NOTE_FS5, NOTE_DS5, 
  NOTE_B5, NOTE_FS5, NOTE_DS5, NOTE_C5, 
  NOTE_B5, NOTE_FS5, NOTE_DS5, NOTE_B4, 
  NOTE_CS5, NOTE_D5, NOTE_CS5, NOTE_B4, 
  NOTE_CS5, NOTE_D5, NOTE_CS5, NOTE_B4, 
  NOTE_B4, NOTE_B4, NOTE_C5, NOTE_D5, 
  NOTE_G4, NOTE_F4, NOTE_DS4, NOTE_B4, 
  NOTE_CS5, NOTE_D5, NOTE_CS5, NOTE_B4, 
  NOTE_CS5, NOTE_D5, NOTE_CS5, NOTE_B4
};

int durations[] = {
  150, 150, 150, 150, 
  150, 150, 150, 150, 
  150, 150, 150, 150, 
  150, 150, 150, 150, 
  150, 150, 150, 150, 
  150, 150, 150, 150, 
  150, 150, 150, 150, 
  150, 150, 150, 150, 
  150, 150, 150, 150
};

void setup() {
  pinMode(greenLight, OUTPUT);
  pinMode(redLight1, OUTPUT);
  pinMode(redLight2, OUTPUT);
  pinMode(buzzer, OUTPUT);
  pinMode(button, INPUT);

  // Turn on red lights initially
  digitalWrite(redLight1, HIGH);
  digitalWrite(redLight2, HIGH);
  digitalWrite(greenLight, LOW);
}

void loop() {
  if (digitalRead(button) == HIGH) {
    // Turn off red lights and turn on green light
    digitalWrite(redLight1, LOW);
    digitalWrite(redLight2, LOW);
    digitalWrite(greenLight, HIGH);

    // Play the song
    playSong();

    // Turn off green light and turn on red lights after the song
    digitalWrite(greenLight, LOW);
    digitalWrite(redLight1, HIGH);
    digitalWrite(redLight2, HIGH);
  }
}

void playSong() {
  const int totalNotes = sizeof(notes) / sizeof(int);

  for (int i = 0; i < totalNotes; i++) {
    const int currentNote = notes[i];
    float wait = durations[i] / songSpeed;

    // Play tone if currentNote is not 0 frequency, otherwise pause (noTone)
    if (currentNote != 0) {
      tone(buzzer, currentNote, wait); // tone(pin, frequency, duration)
    } else {
      noTone(buzzer);
    }

    // delay is used to wait for tone to finish playing before moving to next loop
    delay(wait * 1.3); // Adding a bit of delay to separate notes
  }
  noTone(buzzer); // Ensure the buzzer is turned off after the song is played
}