// Based on https://wokwi.com/projects/372153001566564353

// Removed: playSequence(), gameOver(), checkUserSequence(), playLevelUpSound()
// Simplified the code to function as a basic counter
// Serves as a starting point for implementing my project

// Copyright (C) 2023, Uri Shaked. Released under the MIT License.

#include "pitches.h"

/* Define pin numbers for LEDs, buttons and speaker: */
const uint8_t buttonPins[] = {0, 1, 2, 3};
const uint8_t ledPins[] = {8, 7, 6, 5};
#define SPEAKER_PIN 10

// These are connected to 74HC595 shift register (used to show game score):
const int LATCH_PIN = 18;  // 74HC595 pin 12
const int DATA_PIN = 19;   // 74HC595 pin 14
const int CLOCK_PIN = 9;  // 74HC595 pin 11

#define MAX_GAME_LENGTH 100

const int gameTones[] = { NOTE_G3, NOTE_C4, NOTE_E4, NOTE_G5};

/* Global variables - store the game state */
uint8_t gameSequence[MAX_GAME_LENGTH] = {0};
uint8_t gameIndex = 0;

/**
	Set up the Arduino board and initialize Serial communication
*/
void setup() {
	Serial.begin(9600);
	for (byte i = 0; i < 4; i++) {
	pinMode(ledPins[i], OUTPUT);
	pinMode(buttonPins[i], INPUT_PULLUP);
	}
	pinMode(SPEAKER_PIN, OUTPUT);
	pinMode(LATCH_PIN, OUTPUT);
	pinMode(CLOCK_PIN, OUTPUT);
	pinMode(DATA_PIN, OUTPUT);
}

/* Digit table for the 7-segment display */
const uint8_t digitTable[] = {
	0b11000000,
	0b11111001,
	0b10100100,
	0b10110000,
	0b10011001,
	0b10010010,
	0b10000010,
	0b11111000,
	0b10000000,
	0b10010000,
};
const uint8_t DASH = 0b10111111;

void sendScore(uint8_t high, uint8_t low) {
	digitalWrite(LATCH_PIN, LOW);
	shiftOut(DATA_PIN, CLOCK_PIN, MSBFIRST, low);
	shiftOut(DATA_PIN, CLOCK_PIN, MSBFIRST, high);
	digitalWrite(LATCH_PIN, HIGH);
}

void displayScore() {
	int high = gameIndex % 100 / 10;
	int low = gameIndex % 10;
	sendScore(high ? digitTable[high] : 0xff, digitTable[low]);
}


void lightLedAndPlayTone(byte ledIndex) {
	digitalWrite(ledPins[ledIndex], HIGH);
	tone(SPEAKER_PIN, gameTones[ledIndex]);
	delay(300);
	digitalWrite(ledPins[ledIndex], LOW);
	noTone(SPEAKER_PIN);
}


/**
	Waits until the user presses one of the buttons,
	then lights up corresponding LED, plays sound and increases variable gameIndex
*/
void read_buttons() {
	for (byte i = 0; i < 4; i++) {  // Loop through all buttons
	if (digitalRead(buttonPins[i]) == LOW) {  // If any button is pressed
		lightLedAndPlayTone(i); // different functions are not yet defined for different buttons
		gameIndex += 1;  // Increase the score
		Serial.println("Score!");
		delay(200);  // Debounce delay to avoid multiple counts from one press
		return;  // Exit the function after the first button press
	}
	}
}
	
	
	/**
		The main game loop, made this much simpler
	*/
	void loop() {
		displayScore();
		read_buttons();
	}