// written by kl980

// libraries used
#include "DFRobot_RGBLCD1602.h"
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>

///// PIN ASSIGNMENT ////////
const int redLED1 = 2;
const int blueLED1 = 3;
const int redLED2 = 4;
const int blueLED2 = 5;
const int redPot = A0;
const int bluePot = A1;
const int buzzer = 6;
const int button = 13;
const int joystick = A7;
const int segmentPins[7] = {7,8,9,10,11,12,A2};
const int scl = A5;
const int sda = A4;

/////// OLED SCREEN ///////////
const int oledHeight = 96;
const int oledWidth = 16;
Adafruit_SSD1306 oled(oledHeight, oledWidth, &Wire, -1);


////// LCD SCREEN //////////
DFRobot_RGBLCD1602 lcd(0x2D, 16, 2); // address, columns, rows


////// GAME DESIGN ////////
//custom pixel characters
uint8_t asteroidSymbol[8] = {
  B00000, // .....
  B00100, // ..*..
  B01110, // .***.
  B11111, // *****
  B11111, // *****
  B01110, // .***.
  B00100, // ..*..
  B00000  // .....
};

uint8_t shipSymbol[8] = {
  B00011, // ...**
  B00110, // ..**.
  B01100, // .**..
  B11111, // *****
  B11111, // *****
  B01100, // .**..
  B00110, // ..**.
  B00011  // ...**
};

// handles the movement of asteroid
int asteroidActive[5] = {0,0,0,0,0};
int asteroidRows[5] = {0,0,0,0,0};
int asteroidCols[5] = {0,0,0,0,0};
int asteroidRow;
unsigned long timeMove = 0;
int moveDelay = 200;
int joystickVal;

// LED colour game variables
bool newColour = 1;
bool match = 0;
int redRGB1;
int blueRGB1;
int redRGB2;
int blueRGB2;
long unsigned timeMatch = 0;
int matchDelay = 100;

// asteroid game variables
int lives = 3;
const int startLives = 3;
int shipRow = 0;
bool shoot = 0;
int spawnDelay = 500;
long unsigned timeSpawn = 0;
int newSpawn = 0;
int spawn;
int spawnLocation = 0;
const int shipCol = 13;
bool dead = 0;
int timer = 0;
int wins = 0;
bool complete = 0;
bool start = 0;

// time variables
long unsigned timeNow = 0;

// states of 7 segment display - 10 values correspond to displaying the numbers 0 - 9
int states0[10] = {0,1,0,0,1,0,0,0,0,0};
int states1[10] = {0,0,0,0,0,1,1,0,0,0};
int states2[10] = {0,0,1,0,0,0,0,0,0,0};
int states3[10] = {0,1,0,0,1,0,0,1,0,1};
int states4[10] = {0,1,0,1,1,1,0,1,0,1};
int states5[10] = {0,1,1,1,0,0,0,1,0,0};
int states6[10] = {1,1,0,0,0,0,0,1,0,0};


void setup() {
  // initiate lcd
  Wire.begin();
  lcd.init();
  lcd.customSymbol(0, asteroidSymbol);
  lcd.customSymbol(1, shipSymbol);

  // initiate oled
  oled.begin(SSD1306_SWITCHCAPVCC, 0x3c);
  oled.clearDisplay();
  oled.setTextSize(1);
  oled.setTextColor(SSD1306_WHITE);
  oled.println("Avoid Asteroids");
  oled.print("Restore core");
  oled.display();

  // set up pin modes
  pinMode(redLED1, OUTPUT);
  pinMode(blueLED1, OUTPUT);
  pinMode(redLED2, OUTPUT);
  pinMode(blueLED2, OUTPUT);
  pinMode(redPot, INPUT);
  pinMode(bluePot, INPUT);
  pinMode(buzzer, OUTPUT);
  pinMode(button, INPUT);
  for (int i = 0; i<=6; i++){
    pinMode(segmentPins[i], OUTPUT);
  }

  digitalWrite(buzzer, 0);

  //spawn an asteroid
  asteroidRow = random(0,2);
  asteroidActive[spawnLocation] = 1; // lcd will print asteroid position
  asteroidRows[spawnLocation] = asteroidRow; //set row value  - this will stay constant for that asteroid
  asteroidCols[spawnLocation] = 0; // set column location at beginning
  timeSpawn = millis();

  // set the starting screen for the lcd
  lcd.setCursor(0,0);
  lcd.write("Press the button");
  lcd.setCursor(0,1);
  lcd.write("to start");


}

void loop() {
  // if the game has been started but not completed
  if (complete == 0 && start == 1){
    lcd.setRGB(100, 100, 100);

    // set the RGB LED to a new colour
    if (newColour) {
      redRGB1 = random(0,5)*51;
      blueRGB1 = random(0,5)*51;
      analogWrite(redLED1, redRGB1);
      analogWrite(blueLED1, blueRGB1);
      newColour = 0;
    }

    // main loop
    timeNow = millis();

      // read potentiometer and set LEDs
      redRGB2 = analogRead(redPot)*5/4095*51;
      blueRGB2 = analogRead(bluePot)*5/4095*51;

      analogWrite(redLED2, redRGB2);
      analogWrite(blueLED2, blueRGB2);

      // check if leds match and add lives and sound buzzer
      if (redRGB1 == redRGB2 && blueRGB1 == blueRGB2 && match == 0){
        tone(buzzer, 500);
        match = 1;
        lives = lives + 1;
        if (lives == 10){
          lives = 9; // max of 9 lives
        }
        timeMatch = millis();
      }


      // read joystick and set poisition
      joystickVal = analogRead(joystick);
      if (joystickVal < 1800){ // right
        shipRow = 1;
      }
      if (joystickVal > 2000) { //left
        shipRow = 0;
      }


      // read button
      shoot = digitalRead(button);
      if (shoot) {
        for ( int i = 0; i <= 2; i++){
          if (shipRow == asteroidRows[i] && shipCol + 1 == asteroidCols[i]){
            // hit
            asteroidActive[i] = 0;
            lcd.setRGB(0, 255, 0);
            asteroidRows[i] = 0;
            asteroidCols[i] = 0;
          }
        }
      }

      // set 7 segment display to lives
      // would be better as a 3d array in future then can run as a loop
      digitalWrite(segmentPins[0], states0[lives]);
      digitalWrite(segmentPins[1], states1[lives]);
      digitalWrite(segmentPins[2], states2[lives]);
      digitalWrite(segmentPins[3], states3[lives]);
      digitalWrite(segmentPins[4], states4[lives]);
      digitalWrite(segmentPins[5], states5[lives]);
      digitalWrite(segmentPins[6], states6[lives]);

    // define asteroid movement
    // set arrays to store locations of each asteroid
    // no more than 5 at a time

    if (timeNow - timeMove > moveDelay){
      int asteroidRow = 0;
      // columns, rows(2)
      lcd.clear();

      // print the ship
      lcd.setCursor(shipCol, shipRow);
      lcd.write((unsigned char)1);

      for (int i = 0; i <= 4; i++){ // this needs to be a variable for max number of asteroids in future
        if (asteroidActive[i] == 1){ // if there is an active asteroid
          lcd.setCursor(asteroidCols[i],asteroidRows[i]);
          lcd.write((unsigned char)0);
          // increase column by 1 so next time it will move
          asteroidCols[i] = asteroidCols[i] + 1;
          if (asteroidCols[i] == 17 ){
            lives = lives - 1;
            // kill the asteroid, assign i to newSpawn
            asteroidActive[i] = 0;
            lcd.setRGB(255, 0, 0);
            if (lives == 0){
              dead = 1;
            }
          }
        }
      }
      timeMove = millis();
      timer += 1;
      Serial.println(timer);
    }

    // turn buzzer off and set new colour certain amount of time after match
    if ((timeNow - timeMatch > matchDelay) && match == 1){
      noTone(buzzer);
      newColour = 1;
      match = 0;
    }

    // check to spawn new asteroid
    // again clunky needs variable to handle
    if ((timeNow - timeSpawn > spawnDelay) && (asteroidActive[0] == 0 | asteroidActive[1] == 0 | asteroidActive[2] == 0) | asteroidActive[3] == 0 | asteroidActive[4] == 0){
      spawn = random(0,2);
      if (spawn) {
        if (asteroidActive[0] == 0) {
        spawnLocation = 0;
        }
        else if (asteroidActive[1] == 0) {
          spawnLocation = 1;
        }
        else if (asteroidActive[2] == 0) {
          spawnLocation = 2;
        }
        else if (asteroidActive[3] == 0) {
          spawnLocation = 3;
        }
        else if (asteroidActive[4] == 0) {
          spawnLocation = 4;
        }

        asteroidRow = random(0,2);
        asteroidActive[spawnLocation] = 1;
        asteroidRows[spawnLocation] = asteroidRow; //set row value for final
        asteroidCols[spawnLocation] = 0;

      }
      timeSpawn = millis();

    }

    // lives = 0; reset
    if (dead) {
      lives = startLives;
      lcd.setRGB(255, 0, 0);
      tone(buzzer, 100);
      delay(1000);
      noTone(buzzer);
      timer = 0;
      dead = 0;
    }

    // won a round
    if (timer > 200){
      lcd.setRGB(245, 224, 86);
      lives = startLives;
      moveDelay = moveDelay - 15;
      timer = 0;
      tone(buzzer, 261);
      delay(200);
      tone(buzzer, 293);
      delay(200);
      tone(buzzer, 329);
      delay(200);
      noTone(buzzer);
      delay(200);
      wins = wins + 1;

    }

  // won 3 times
    if (wins > 2){
      complete = 1;
    }
  }

  // game complete
  else if (complete) {
    lcd.setRGB(245,224,86);
    lcd.setCursor(0,0);
    lcd.write("Home Coordinates");
    lcd.setCursor(0,1);
    lcd.write("20,25,09");
    delay(10000);
    start = 0;
  }

  // start if button pressed
  else if (digitalRead(button)==1){
    start = 1;
  }
}