// Define pin numbers for lights, buzzer, and buttons
const int greenLight = 2;
const int redLight = 3;
const int buzzer = 10;
const int button1 = 4;
const int button2 = 5;
const int button3 = 6;

// 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

// Example tunes for the buttons
int tune1[] = {NOTE_C4, NOTE_D4, NOTE_E4, NOTE_F4};
int duration1[] = {250, 250, 250, 250};

int tune2[] = {NOTE_G4, NOTE_A4, NOTE_B4, NOTE_C5};
int duration2[] = {250, 250, 250, 250};

int tune3[] = {NOTE_E5, NOTE_D5, NOTE_C5, NOTE_B4};
int duration3[] = {250, 250, 250, 250};

// To hold the sequence of button presses
int sequence[3] = {0, 0, 0};
int sequenceIndex = 0;

void setup() {
  pinMode(greenLight, OUTPUT);
  pinMode(redLight, OUTPUT);
  pinMode(buzzer, OUTPUT);
  pinMode(button1, INPUT);
  pinMode(button2, INPUT);
  pinMode(button3, INPUT);

  digitalWrite(redLight, HIGH);
  digitalWrite(greenLight, LOW);
}

void loop() {
  if (digitalRead(button1) == HIGH) {
    playTune(tune1, duration1, sizeof(tune1) / sizeof(int));
    delay(300); // debounce delay
  } else if (digitalRead(button2) == HIGH) {
    playTune(tune2, duration2, sizeof(tune2) / sizeof(int));
    delay(300); // debounce delay
  } else if (digitalRead(button3) == HIGH) {
    playTune(tune3, duration3, sizeof(tune3) / sizeof(int));
    delay(300); // debounce delay
  }

  if (sequenceIndex == 3) {
    playSequence();
    sequenceIndex = 0; // reset the sequence index
  }

  if (sequenceIndex == 0) {
    digitalWrite(greenLight, LOW);
    digitalWrite(redLight, HIGH);
  }
}

void playSequence() {
  for (int i = 0; i < sequenceIndex; i++) {
    if (sequence[i] == 1) {
      playTune(tune1, duration1, sizeof(tune1) / sizeof(int));
    } else if (sequence[i] == 2) {
      playTune(tune2, duration2, sizeof(tune2) / sizeof(int));
    } else if (sequence[i] == 3) {
      playTune(tune3, duration3, sizeof(tune3) / sizeof(int));
    }
  }
}

void playTune(int* tune, int* duration, int length) {
  digitalWrite(greenLight, HIGH);
  digitalWrite(redLight, LOW);

  for (int i = 0; i < length; i++) {
    tone(buzzer, tune[i], duration[i]);
    delay(duration[i] * 1.3);
  }

  noTone(buzzer);
  digitalWrite(greenLight, LOW);
  digitalWrite(redLight, HIGH);
}