All_At_Once.ino

/* colors:
 * 1 = red
 * 2 = green
 * 3 = blue
 * 4 = off
 */
int rightColor = 2;  // can be customized
int runs = 10;       // total amount of times to display light array

////////////////////////////////////////////////////////////////

int correct_leds = 0;
int tpast = 0;  // for check_button()

const int tot_run = runs;
int num_correct = 0;
int num_incorrect = 0;
int total_incorrect = 0;
int total_correct = 0;

const int num_buttons = 15;
const int redPins[] = {10, 6, 2, 17, 21, 29, 22, 37, 30, 45, 38, 53, 46, 69, 65};
const int greenPins[] = {11, 7, 3, 16, 20, 27, 24, 35, 32, 43, 40, 51, 48, 68, 64};
const int bluePins[] = {12, 8, 4, 15, 19, 25, 26, 33, 34, 41, 42, 49, 50, 67, 63};
const int buttons[] = {13, 9, 5, 14, 18, 23, 28, 31, 36, 39, 44, 47, 52, 66, 62};

int buttonStates[num_buttons];                     // array for storing states of buttons
int ledColor[num_buttons];                     // array for storing colors of LEDs

unsigned long initial_time = 0;          // time led turns on
unsigned long final_time = 0;            // time button pressed
unsigned long elapsed_time = 0;          // reaction time
unsigned long total_rxn_time = 0;

void setup() {
  for (int i = 0; i < num_buttons; i++) {
    pinMode(buttons[i], INPUT);
    pinMode(redPins[i], OUTPUT);
    pinMode(greenPins[i], OUTPUT);
    pinMode(bluePins[i], OUTPUT);
    buttonStates[i]=HIGH;
    setColor(i, 4);
  }
  Serial.begin(9600);
  randomSeed(analogRead(A3));
  rule(rightColor);
  colorIndicator(rightColor);
}

void loop() {
  if (runs > 0) {
    runs --;
    delay(random(500)+50);                    //delays LED light up
    correct_leds = 1;                            // every run has at least one correct led
//    total_correct += num_correct;
//    total_incorrect += num_incorrect;
//    num_correct = 0; num_incorrect = 0;
    
    //sets random LED to be the correct color
    int number = random(0, num_buttons);
    setColor(number, rightColor);
    ledColor[number] = rightColor;
    initial_time = millis();
    
    //sets the other LEDs to random colors
    for (int i = 0; i < num_buttons; i++){
      if (i == number)
        continue;
      int color = random(1, 4); 
      if (color == rightColor)
        correct_leds++;
      setColor(i, color);
      ledColor[i] = color;
    }
    tpast = millis();        // for check_button()
    
    while (correct_leds > 0) {  //runs until all correct leds are turned off
      correct_leds = check_button();
    }
    
    time_elapsed();
    //accuracy();
    reset();
  }
  else {
    while (end_func() == false);
    return;
  }
}

int check_button(){
  int tnow = millis();
  int time_diff = tnow - tpast;                   // how long until last call of function
  
  for (int i = 0; i < num_buttons; i ++){                   // reads and records button states
    buttonStates[i] = digitalRead(buttons[i]);
  }
  
  for (int i = 0; i < num_buttons; i ++){
    if ((buttonStates[i] == LOW) && (ledColor[i] == rightColor) && time_diff > 70){  // if button pressed was correct button
      digitalWrite(redPins[i], HIGH);
      digitalWrite(greenPins[i], HIGH);
      digitalWrite(bluePins[i], HIGH);
      ledColor[i] = 4;     // turn color to off-state so led will not be counted twice
      correct_leds--;
      //num_correct ++;
      total_correct++;
      buttonStates[i] == HIGH;
    }
    if ((buttonStates[i] == LOW) && !(ledColor[i] == rightColor) && !(ledColor[i] == 4) && time_diff > 200){  // if button pressed was incorrect
      delay(300);
      //num_incorrect++;
      total_incorrect++;
      buttonStates[i] == HIGH;
    }
  }
  int tpast = tnow;       // prepares for next call
  return correct_leds;
}

void time_elapsed() {
  final_time = millis();
  for (int i = 0; i < num_buttons; i++) {  // turns off all lights
    digitalWrite(redPins[i], HIGH);
    digitalWrite(greenPins[i], HIGH);
    digitalWrite(bluePins[i], HIGH);
  }
  elapsed_time = final_time - initial_time;
  total_rxn_time += elapsed_time;
  Serial.print("Elapsed time = ");
  Serial.print(float(elapsed_time)/1000);
  Serial.println(" s");
}

//void accuracy(){
//  Serial.print("   Total hits = ");
//  Serial.println(num_correct+num_incorrect);
//  Serial.print("   Correct hits = ");
//  Serial.println(num_correct);
//  Serial.print("   Incorrect hits = ");
//  Serial.println(num_incorrect);
//  Serial.println("");
//}

void reset() {
  for (int i = 0; i < num_buttons; i++) {
    buttonStates[i]=HIGH;
  }
}

bool end_func() {
  stopSign();
  Serial.println();
  Serial.println("Results:");
  Serial.print("   Average reaction time = "); Serial.print(float(float(total_rxn_time)/float(tot_run))/1000); Serial.println(" s");
  Serial.print("   Total hits = "); Serial.println(total_incorrect + total_correct);
  Serial.print("   Accuracy = "); Serial.print((float(total_correct)/float(total_incorrect + total_correct))*100); Serial.println("%");
//  Serial.print("      Correct = "); Serial.println(total_correct);
//  Serial.print("      Incorrect = "); Serial.println(total_incorrect);
  
  while(digitalRead(buttons[14]) == HIGH);
  runs = tot_run; total_rxn_time = 0; total_incorrect = 0; total_correct = 0;
  Serial.println(""); Serial.println("Resetting..."); Serial.println("");
  delay(1000);
  rightColor = random(1, 4); rule(rightColor);
  delay(1000);
  colorIndicator(rightColor);
  return true;
}

void colorIndicator(int color) {
  setColor(0, rightColor); setColor(1, rightColor); setColor(2, rightColor); setColor(3, rightColor); setColor(4, rightColor); setColor(5, rightColor); setColor(6, rightColor); setColor(7, rightColor);
  delay(1000);
  setColor(0, 4); setColor(2, 4); setColor(3, 4); setColor(4, 4); setColor(5, 4); setColor(6, 4); setColor(7, 4); setColor(1, 4);
  setColor(8, rightColor); setColor(9, rightColor); setColor(10, rightColor); setColor(11, rightColor); setColor(12, rightColor); setColor(13, rightColor);
  delay(1000);
  setColor(8, 4); setColor(9, 4); setColor(10, 4); setColor(11, 4); setColor(12, 4); setColor(13, 4);
  setColor(14, rightColor);
  delay(1000);
  setColor(14, 4);
  delay(1000);
}

void rule(int color) {
  Serial.print("Only hit ");
  switch(color) {
    case 1: Serial.print("red\n"); break;
    case 2: Serial.println("green");  break;
    case 3: Serial.println("blue");  break;
  }
  Serial.println("");
}

void stopSign() {
  delay(500);
  for (int i = 8; i<num_buttons; i++) {
    setColor(i, 1);
  }
  delay(500);
  for (int i = 8; i<num_buttons; i++) {
    setColor(i, 4);
  }
  delay(500);
  for (int i = 8; i<num_buttons; i++) {
    setColor(i, 1);
  }
  delay(500);
  for (int i = 8; i<num_buttons; i++) {
    setColor(i, 4);
  }
  delay(500);
  for (int i = 8; i<num_buttons; i++) {
    setColor(i, 1);
  }
  delay(500);
  for (int i = 8; i<num_buttons; i++) {
    setColor(i, 4);
  }
}

void setColor(int led, int color) {
  if (color == 1) {
    digitalWrite(redPins[led], LOW);
    digitalWrite(greenPins[led], HIGH);
    digitalWrite(bluePins[led], HIGH);
  }
  else if (color == 2) {
    digitalWrite(redPins[led], HIGH);
    digitalWrite(greenPins[led], LOW);
    digitalWrite(bluePins[led], HIGH);
  }
  else if (color == 3) {
    digitalWrite(redPins[led], HIGH);
    digitalWrite(greenPins[led], HIGH);
    digitalWrite(bluePins[led], LOW);
  }
  else if (color == 4) {
    digitalWrite(redPins[led], HIGH);
    digitalWrite(greenPins[led], HIGH);
    digitalWrite(bluePins[led], HIGH);
  }
}