#include <Adafruit_NeoPixel.h>
#ifdef __AVR__
  #include <avr/power.h>
#endif
// Which pin on the Arduino is connected to the NeoPixels?
// On a Trinket or Gemma we suggest changing this to 1
#define PIN 6
// How many NeoPixels are attached to the Arduino?
#define NUMPIXELS      48
// How much brightness for the LEDs (from 0-255)?
#define BRIGHTNESS     255
// How many different colors?
#define NUMCOLORS      7
// Parameter 1 = number of pixels in strip
// Parameter 2 = Arduino pin number (most are valid)
// Parameter 3 = pixel type flags, add together as needed:
//   NEO_KHZ800  800 KHz bitstream (most NeoPixel products w/WS2812 LEDs)
//   NEO_KHZ400  400 KHz (classic 'v1' (not v2) FLORA pixels, WS2811 drivers)
//   NEO_GRB     Pixels are wired for GRB bitstream (most NeoPixel products)
//   NEO_RGB     Pixels are wired for RGB bitstream (v1 FLORA pixels, not v2)
//   NEO_RGBW    Pixels are wired for RGBW bitstream (NeoPixel RGBW products)
Adafruit_NeoPixel strip = Adafruit_NeoPixel(NUMPIXELS, PIN, NEO_GRB + NEO_KHZ800);
Adafruit_NeoPixel pixels = Adafruit_NeoPixel(NUMPIXELS, PIN, NEO_GRB + NEO_KHZ800);
// IMPORTANT: To reduce NeoPixel burnout risk, add 1000 uF capacitor across
// pixel power leads, add 300 - 500 Ohm resistor on first pixel's data input
// and minimize distance between Arduino and first pixel.  Avoid connecting
// on a live circuit...if you must, connect GND first.
// 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);
}
  int delayval = 100; // delay between each light show for the pollution values
  int delayval1 = 500; // delay between show of white light and start of pollution values
  int delayval2 = 5000; // 
  int pollutionValues[NUMPIXELS]; // list of pollution values
  int thresholdValues[NUMCOLORS] = {0, 21, 43, 61, 81, 100}; // Thresholds for the categories of pollution values
  int redColors[NUMCOLORS] = {0, 0, 0, 255, 255, 255}; // Values on red scale for the pollution categories
  int greenColors[NUMCOLORS] = {0, 255, 255, 255, 53, 0}; // Values on green scale for the pollution categories
  int blueColors[NUMCOLORS] = {0, 255, 0, 0, 0, 0}; // Values on blue scale for the pollution categories
  
void setup() {
  // This is for Trinket 5V 16MHz, you can remove these three lines if you are not using a Trinket
  #if defined (__AVR_ATtiny85__)
    if (F_CPU == 16000000) clock_prescale_set(clock_div_1);
  #endif
  // End of trinket special code
  strip.begin();
  strip.show(); // Initialize all pixels to 'off'
}
void rainbowCycle(uint8_t wait) {
  uint16_t i, j;
  for(j=0; j<256*1; j++) { // 1 cycle 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);
  }
}
void loop() {
  //initialisation: show of pollution scale with arbitrary pollution values
  for (int i=0; i<NUMPIXELS; i++) {
    pollutionValues[i] = 2 * i;
  }
  for (int i=0; i<NUMPIXELS; i++) {
    // Iterate over the thresholds to find the right pollution category
    for (int j=0; j<NUMCOLORS; j++) {
      if (pollutionValues[i] < thresholdValues[j]) {
        pixels.setPixelColor(i, pixels.Color(redColors[j],greenColors[j],blueColors[j]));
        pixels.show();
        delay(delayval);
        break;
      }
    }
  }
  //rainbow animation
  rainbowCycle(6);
    
  for (int i=0; i<NUMPIXELS; i++) {
    pixels.setPixelColor(i, pixels.Color(80,80,80));
  }
  pixels.show();
  delay(delayval1);
  // Show of pollution values since beginning of the year
  int pollutionValuesBis[NUMPIXELS] = {43, 52, 91, 45, 36, 93, 62, 28, 26, 59, 21, 47, 28, 57, 33, 25, 36, 18, 30, 34, 27, 29, 41, 19, 29, 25, 25, 18, 30, 38, 44, 61, 37, 44, 45, 39, 34, 41, 40, 56, 42, 36, 35, 54, 81, 56, 38, 66};
  //int pollutionValuesLast[NUMPIXELS] = {30, 38, 44, 61, 37, 44, 45, 39, 34, 41, 40, 56, 42, 36, 35, 54, 81, 56, 38, 66};//
  for (int i=0; i<NUMPIXELS; i++) {
    pollutionValues[i] = pollutionValuesBis[i];
  }
  // [UPDATE LED]
  for (int i=0; i<28; i++) {
    // Iterate over the thresholds to find the right pollution category
    for (int j=0; j<NUMCOLORS; j++) {
      if (pollutionValues[i] < thresholdValues[j]) {
        pixels.setPixelColor(i, pixels.Color(redColors[j],greenColors[j],blueColors[j]));
        pixels.show();
        delay(delayval);
        break;
      }
    }
  }
  for (int i=28; i<29; i++) {
    for (int j=0; j<NUMCOLORS; j++) {
      if (pollutionValues[i] < thresholdValues[j]) {
        pixels.setPixelColor(i, pixels.Color(redColors[j],greenColors[j],blueColors[j]));
        pixels.show();
        delay(delayval);
        break;
      }
    }
  }
  // for later dates than our date: show of last year's pollution levels
  for (int i=29; i<NUMPIXELS; i++) {
    for (int j=0; j<NUMCOLORS; j++) {
      if (pollutionValues[i] < thresholdValues[j]) {
        pixels.setPixelColor(i, pixels.Color(redColors[j]/50,greenColors[j]/50,blueColors[j]/50));
        pixels.show();
        delay(delayval);
        break;
      }
    }
  }
  // blink of current week LED
  int var = 0;
  while(var < 15){
    for (int i=28; i<29; i++) {
      for (int j=0; j<NUMCOLORS; j++) {
        if (pollutionValues[i] < thresholdValues[j]) {
          pixels.setPixelColor(i, pixels.Color(redColors[j],greenColors[j],blueColors[j]));
          pixels.show();
          delay(delayval1);
          pixels.setPixelColor(i, pixels.Color(0,0,0));
          pixels.show();
          delay(delayval1);
          break;
        }
      }
    }
  var++;
  }
  // turn off all LEDs before restarting the loop
  for (int i=0; i<NUMPIXELS; i++) {
    pixels.setPixelColor(i, pixels.Color(0,0,0));
  }
  pixels.show();
  delay(delayval);
}