#include <Arduino.h>

// Pin definitions
const int NUM_PLAYERS = 2;
const int BUTTON_PINS[NUM_PLAYERS] = {5, 4};
const int LED_PINS[NUM_PLAYERS] = {18, 19};
const int BUZZER_PINS[NUM_PLAYERS] = {2, 3};
const int RESET_BUTTON_PIN = 10;

volatile bool buttonPressed[NUM_PLAYERS] = {false, false};
volatile unsigned long pressTime[NUM_PLAYERS] = {0, 0};

// Note frequency definitions (in Hertz)
// Each note corresponds to a specific frequency that creates that pitch
#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  // Middle C
#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  // Concert A
#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
#define REST 0  // REST means silence
boolean someoneBuzzed = false;

void IRAM_ATTR player1ISR() {
  if (!someoneBuzzed) {
    buttonPressed[0] = true;
    pressTime[0] = micros();
  }
}

void IRAM_ATTR player2ISR() {
  if (!someoneBuzzed) {
    buttonPressed[1] = true;
    pressTime[1] = micros();
  }
}

// Player 1 Victory Jingle (Ascending chromatic scale)
int player1_melody[] = {
    NOTE_C5, NOTE_CS5, NOTE_D5, NOTE_DS5, NOTE_E5, NOTE_F5, NOTE_FS5, NOTE_G5, NOTE_GS5, NOTE_A5, NOTE_AS5, NOTE_B5, NOTE_C6
};
int player1_durations[] = {
    75, 75, 75, 75, 75, 75, 75, 75, 75, 75, 75, 75, 200
};
int player1_notes = 13;

// Player 2 Victory Jingle (Ta-da! style)
int player2_melody[] = {
  NOTE_D4, NOTE_D4, NOTE_D4, NOTE_G4,
  REST, NOTE_G4, NOTE_A4, NOTE_B4, NOTE_G4
};
int player2_durations[] = {
  100, 100, 200, 400,
  100, 200, 200, 200, 600
};
int player2_notes = 9;

// Calculate total melody durations for LED synchronization
const unsigned long PLAYER1_MELODY_DURATION = 1100; // 12*75 + 200 = 1100ms
const unsigned long PLAYER2_MELODY_DURATION = 1300; // 100+100+200+400+100+200+200+200+600 = 1300ms

// Array to store melody durations for easy access
const unsigned long MELODY_DURATIONS[NUM_PLAYERS] = {PLAYER1_MELODY_DURATION, PLAYER2_MELODY_DURATION};

// Game state variables
unsigned long ledOnTimes[NUM_PLAYERS] = {0, 0};
boolean ledsOn[NUM_PLAYERS] = {false, false};
boolean jinglesPlayed[NUM_PLAYERS] = {false, false};
int lastButtonStates[NUM_PLAYERS] = {HIGH, HIGH};
int lastResetButtonState = HIGH;

int winner = -1;

void setup() {
  for (int i = 0; i < NUM_PLAYERS; i++) {
    pinMode(BUTTON_PINS[i], INPUT);
    pinMode(LED_PINS[i], OUTPUT);
    pinMode(BUZZER_PINS[i], OUTPUT);
  }
  pinMode(RESET_BUTTON_PIN, INPUT_PULLUP);
  attachInterrupt(digitalPinToInterrupt(BUTTON_PINS[0]), 
                  player1ISR, FALLING);
  attachInterrupt(digitalPinToInterrupt(BUTTON_PINS[1]), 
                  player2ISR, FALLING);
  Serial.begin(115200);
}

void playJingle(int player) {
  if (player == 0) {
    // Play Player 1's jingle
    for (int i = 0; i < player1_notes; i++) {
      if (player1_melody[i] != REST) {
        tone(BUZZER_PINS[player], player1_melody[i]);
      } else {
        noTone(BUZZER_PINS[player]);
      }
      delay(player1_durations[i]);
    }
  } else if (player == 1) {
    // Play Player 2's jingle
    for (int i = 0; i < player2_notes; i++) {
      if (player2_melody[i] != REST) {
        tone(BUZZER_PINS[player], player2_melody[i]);
      } else {
        noTone(BUZZER_PINS[player]);
      }
      delay(player2_durations[i]);
    }
  }
  noTone(BUZZER_PINS[player]);
}

void loop() {
  // Check reset button
  int resetButtonState = digitalRead(RESET_BUTTON_PIN);
  
  if (resetButtonState == LOW && lastResetButtonState == HIGH) {
    // Reset everything
    someoneBuzzed = false;
    winner = -1;
    
    for (int i = 0; i < NUM_PLAYERS; i++) {
      digitalWrite(LED_PINS[i], LOW);
      noTone(BUZZER_PINS[i]);
      ledsOn[i] = false;
      jinglesPlayed[i] = false;
      ledOnTimes[i] = 0;
    }
    
    Serial.println("RESET! Ready for next round.");
  }
  
  lastResetButtonState = resetButtonState;
  
  // Check player buttons
  for (int player = 0; player < NUM_PLAYERS; player++) {
    int buttonState = digitalRead(BUTTON_PINS[player]);
    
    // Detect button press
    if (buttonState == LOW && 
        lastButtonStates[player] == HIGH && 
        !someoneBuzzed) {
      
      // This player wins!
      digitalWrite(LED_PINS[player], HIGH);
      ledOnTimes[player] = millis();
      ledsOn[player] = true;
      someoneBuzzed = true;
      winner = player;
      
      Serial.print("BUZZ! Player ");
      Serial.print(player + 1);
      Serial.println(" wins!");
      
      // Play the victory jingle!
      playJingle(player);
      jinglesPlayed[player] = true;
    }
    
    // Turn off LED after melody duration instead of fixed 2 seconds
    if (ledsOn[player] && 
        (millis() - ledOnTimes[player] >= MELODY_DURATIONS[player])) {
      digitalWrite(LED_PINS[player], LOW);
      ledsOn[player] = false;
    }
    
    lastButtonStates[player] = buttonState;
  }
  
  delay(10);
}