#include <Adafruit_NeoPixel.h>
#define PIN 6  //Pin to which the DIN for the first pod is connected

const int numReadings = 5;
int reading; // the readings from the analog input
int index = 0; // the index of the current reading
int total = 0; // the running total
int average = 0; // the average
int podTrigger = 0;

char sensorPins[6] = {A5,A4,A3,A2,A1,A0};  //Sensor pins
int sensorValues[6] = {0,0,0,0,0,0};
int calibratedsensorValues[6] = {0,0,0,0,0,0};
char podColours[6][3] = {{0,0,0},{0,0,0},{0,0,0},{0,0,0},{0,0,0},{0,0,0}};
char podLeds[6][6] = {{0,1,2,3,4,5},{6,7,8,9,10,11},{12,13,14,15,16,17},{18,19,20,21,22,23},{24,25,26,27,28,29},{30,31,32,33,34,35}}; //Array in which the seperate led's are defined with regard to the pod they are part of
int podTriggered[6] = {0,0,0,0,0,0};  //Array for storing which pods have cups on them

Adafruit_NeoPixel strip = Adafruit_NeoPixel(36, PIN, NEO_GRB + NEO_KHZ800);


void setup() {
  strip.begin();  //Initialises the led's
  strip.show(); 
  Serial.begin(9600);   //For debugging purposes
}

void loop() {
  updateSensors();
  updatePodColours();
  
///////// Default neopixel functions to be used in futhur expansion //////
/* colorWipe(strip.Color(255, 0, 0), 50); // Red   
  colorWipe(strip.Color(0, 255, 0), 50); // Green
  colorWipe(strip.Color(0, 0, 255), 50); // Blue
  // Send a theater pixel chase in...
  theaterChase(strip.Color(127, 127, 127), 50); // White
  theaterChase(strip.Color(127, 0, 0), 50); // Red
  theaterChase(strip.Color(0, 0, 127), 50); // Blue

  rainbow(20);
  rainbowCycle(20);
  theaterChaseRainbow(50);
*/  
}

//////////////Read Sensors and POD update ////////////
void updateSensors(){   // Reads sensor values and stores themin the array to be used later
   for( int i = 0; i<6;i++){ 
       average = 0;
       total =0;
        for ( int k = 0; k<numReadings;k++){ //Used to determine an average reading from the sensor 
          reading = analogRead(sensorPins[i]);
          total = total + reading;
          delay(5);     
        }
        average = total / numReadings;
        sensorValues[i] = average; //Stored the average value in the sensor value array         
   }
  Serial.println(sensorValues[5]); //Debugging purposes  show the sensor value of pod 5
}

void updatePodColours(){
  podTrigger = 0;
   for( int i = 0; i<6;i++){
     if(sensorValues[i] >974){  // Sensor threshold for the different pods cahmphered 968    square 974   use the serial debugging to tune the threshold to our spesific setup
      if(podTriggered[i] == 0){  // Checks if the pod has been triggered previously or not
         podTriggered[i] = 1;
         podTrigger++; 
         podColours[i][0] = 000; //Sets the colours for the triggered pod
         podColours[i][1] = 000;
         podColours[i][2] = 255;
         for(int j = 0; j<6 ; j++){  //Updates and displays the pods colours
            strip.setPixelColor(podLeds[i][j], strip.Color(podColours[i][0],podColours[i][1],podColours[i][2]));  
            strip.show();
          }
       }
     }
     else{    
        podTriggered[i] = 0;
        podColours[i][0] = 255; 
        podColours[i][1] = 000;
        podColours[i][2] = 000;
        for(int j = 0; j<6 ; j++){
          strip.setPixelColor(podLeds[i][j], strip.Color(podColours[i][0],podColours[i][1],podColours[i][2]));  
          strip.show();
      }
     }
   }    
}

////////////// Extra default neopixel POD FUNCTIONS for use in furthur expansion //////////////////

// Fill the dots one after the other with a color
void colorWipe(uint32_t c, uint8_t wait) {
  for(uint16_t i=0; i<strip.numPixels(); i++) {
    strip.setPixelColor(i, c);
    strip.show();
    delay(wait);
  }
}

void rainbow(uint8_t wait) {
  uint16_t i, j;

  for(j=0; j<256; j++) {
    for(i=0; i<strip.numPixels(); i++) {
      strip.setPixelColor(i, Wheel((i+j) & 255));
    }
    strip.show();
    delay(wait);
  }
}

// Slightly different, this makes the rainbow equally distributed throughout
void rainbowCycle(uint8_t wait) {
  uint16_t i, j;

  for(j=0; j<256*5; j++) { // 5 cycles of all colors on wheel
    for(i=0; i< strip.numPixels(); i++) {
      strip.setPixelColor(i, Wheel(((i * 256 / strip.numPixels()) + j) & 255));
    }
    strip.show();
    delay(wait);
  }
}

//Theatre-style crawling lights.
void theaterChase(uint32_t c, uint8_t wait) {
  for (int j=0; j<10; j++) {  //do 10 cycles of chasing
    for (int q=0; q < 3; q++) {
      for (int i=0; i < strip.numPixels(); i=i+3) {
        strip.setPixelColor(i+q, c);    //turn every third pixel on
      }
      strip.show();

      delay(wait);

      for (int i=0; i < strip.numPixels(); i=i+3) {
        strip.setPixelColor(i+q, 0);        //turn every third pixel off
      }
    }
  }
}

//Theatre-style crawling lights with rainbow effect
void theaterChaseRainbow(uint8_t wait) {
  for (int j=0; j < 256; j++) {     // cycle all 256 colors in the wheel
    for (int q=0; q < 3; q++) {
      for (int i=0; i < strip.numPixels(); i=i+3) {
        strip.setPixelColor(i+q, Wheel( (i+j) % 255));    //turn every third pixel on
      }
      strip.show();

      delay(wait);

      for (int i=0; i < strip.numPixels(); i=i+3) {
        strip.setPixelColor(i+q, 0);        //turn every third pixel off
      }
    }
  }
}

// Input a value 0 to 255 to get a color value.
// The colours are a transition r - g - b - back to r.
uint32_t Wheel(byte WheelPos) {
  WheelPos = 255 - WheelPos;
  if(WheelPos < 85) {
    return strip.Color(255 - WheelPos * 3, 0, WheelPos * 3);
  }
  if(WheelPos < 170) {
    WheelPos -= 85;
    return strip.Color(0, WheelPos * 3, 255 - WheelPos * 3);
  }
  WheelPos -= 170;
  return strip.Color(WheelPos * 3, 255 - WheelPos * 3, 0);
}