My twist on the Twister game

/*
 *  Code by Tal Ofer  - talofer99@hotmail.com
 *  
 *  
 */
#include <FastLED.h>
#include <MPR121.h>
#include <Wire.h>
#include "serial_mp3.h"

#define NUM_LEDS 4
#define DATA_PIN 3
#define NUM_BUTTONS 12

// Define the array of leds
CRGB leds[NUM_LEDS];

// Define the array of leds
#define COLOR_BLUE 0
#define COLOR_GREEN 1
#define COLOR_YELLOW 2
#define COLOR_RED 3

// Define system state
#define SYS_STATE_START 0
#define SYS_STATE_WIN 1
#define SYS_STATE_FAIL 2



CRGB colour[] = {CRGB::Blue, CRGB::Green,  CRGB::Yellow, CRGB::Red};
byte buttonColor[NUM_BUTTONS] = {COLOR_BLUE, COLOR_BLUE, COLOR_BLUE, COLOR_GREEN, COLOR_GREEN, COLOR_GREEN, COLOR_YELLOW, COLOR_YELLOW, COLOR_YELLOW, COLOR_RED, COLOR_RED, COLOR_RED};

byte currentColor[NUM_LEDS];
byte currentColorIndex = 0;
byte systemState;

void setup()
{
  Serial.begin(115200);
  Serial.println("System started");
  randomSeed(analogRead(A0) + analogRead(A1) + analogRead(A2) + analogRead(A3));
  Wire.begin();

  // 0x5C is the MPR121 I2C address on the Bare Touch Board
  if (!MPR121.begin(0x5A)) {
    Serial.println("error setting up MPR121");
    switch (MPR121.getError()) {
      case NO_ERROR:
        Serial.println("no error");
        break;
      case ADDRESS_UNKNOWN:
        Serial.println("incorrect address");
        break;
      case READBACK_FAIL:
        Serial.println("readback failure");
        break;
      case OVERCURRENT_FLAG:
        Serial.println("overcurrent on REXT pin");
        break;
      case OUT_OF_RANGE:
        Serial.println("electrode out of range");
        break;
      case NOT_INITED:
        Serial.println("not initialised");
        break;
      default:
        Serial.println("unknown error");
        break;
    }
    while (1);
  }

  // pin 4 is the MPR121 interrupt on the Bare Touch Board
  MPR121.setInterruptPin(2);

  // this is the touch threshold - setting it low makes it more like a proximity trigger
  // default value is 40 for touch
  MPR121.setTouchThreshold(40);

  // this is the release threshold - must ALWAYS be smaller than the touch threshold
  // default value is 20 for touch
  MPR121.setReleaseThreshold(20);

  // initial data update
  MPR121.updateTouchData();

  // start fast led
  FastLED.addLeds<NEOPIXEL, DATA_PIN>(leds, NUM_LEDS);

  serialmp3_init();
  delay(200);
  serialmp3_set_vol(15);
  delay(200);


  setSystemState(SYS_STATE_START);
}

void loop()
{
  mp3_SerialProcess();


  if (MPR121.touchStatusChanged() && systemState == 0) {
    MPR121.updateTouchData();
    for (int i = 0; i < NUM_BUTTONS; i++) {
      if (MPR121.isNewTouch(i)) {
        
        if (buttonColor[i] == currentColor[currentColorIndex]) {
          addColor();
          break;
        }
        else
        {
          setSystemState(SYS_STATE_FAIL);
          break;
        } //end if

      } else if (MPR121.isNewRelease(i)) {
        
        for (byte z = 0; z < currentColorIndex; z++) {
          if (currentColor[z] == buttonColor[i]) {
            setSystemState(SYS_STATE_FAIL);
            break;
          } //end if
        } //end for
        
      } //end if
    } //end for
  } //end if
}

void addColor() {
  currentColorIndex++;
  if (currentColorIndex < NUM_LEDS)
  {
    currentColor[currentColorIndex] = getRandomColor(currentColor[currentColorIndex - 1]);
    Serial.println(currentColor[currentColorIndex]);
    leds[currentColorIndex] = colour[currentColor[currentColorIndex]];
    FastLED.show();
  }
  else
  {
    setSystemState(SYS_STATE_WIN);
  }

}


void setSystemState(byte newState) {
  //set newState
  systemState = newState;
  switch (newState) {
    case SYS_STATE_START:
      currentColorIndex = 0;
      currentColor[currentColorIndex] = getRandomColor(99);
      Serial.println(currentColor[currentColorIndex]);
      leds[currentColorIndex] = colour[currentColor[currentColorIndex]];
      for (byte i = 1; i < NUM_LEDS; i++)
        leds[i] = CRGB::Black;
      FastLED.show();
      break;
    case SYS_STATE_WIN:
      serialmp3_play(5);
      blinkSuccess();
      setSystemState(0);
      Serial.println("system state  - success");
      break;
    case SYS_STATE_FAIL:
      serialmp3_play(6);
      blinkFail();
      setSystemState(0);
      Serial.println("system state  - fail");
      break;

  } //end switch
}

byte getRandomColor(byte avoidColor) {
  byte newColor;
  do {
    newColor = random(0, 4);
  } while (avoidColor == newColor);
  serialmp3_play(newColor + 1); // play color name
  return newColor;
}



void blinkSuccess() {
  for (byte z = 0; z < 3; z++) {
    for (byte i = 0; i < NUM_LEDS; i++)
      leds[i] = CRGB::Green;
    FastLED.show();
    delay(500);
    for (byte i = 0; i < NUM_LEDS; i++)
      leds[i] = CRGB::Black;
    FastLED.show();
    delay(500);
  }
}

void blinkFail() {
  for (byte z = 0; z < 3; z++) {
    for (byte i = 0; i < NUM_LEDS; i++)
      leds[i] = CRGB::Red;
    FastLED.show();
    delay(500);
    for (byte i = 0; i < NUM_LEDS; i++)
      leds[i] = CRGB::Black;
    FastLED.show();
    delay(500);
  }
}