#include <TFT_eSPI.h>
#include <SPI.h>


// Wymaga konfiguracjie:
// C:\Users\siiee\Documents\Arduino\libraries\TFT_eSPI

TFT_eSPI tft = TFT_eSPI();   // domyślna konfiguracja z User_Setup.h

//2DO:
// obsługa przycisków zmieniających menu i LED
//160x128

#define Version "0.0.1"
#include <SPI.h>
#include <Wire.h>
#include <Adafruit_GFX.h>
#include "AS726X.h"
// #define TFT_CS        5
// #define TFT_RST       4
// #define TFT_DC        2

#define Btn1 33             //on board: A1   menu
#define Btn2 34            //on board: A2   gain
#define Btn3 35            //on board: A3   ???

#define WHITE 0xFFFF
#define BLACK 0x0000
#define GRAY 0x8410
#define RED 0xF800
#define ORANGE 0xFA60
#define YELLOW 0xFFE0
#define LIME 0x07FF
#define GREEN 0x07E0
#define CYAN 0x07FF
#define AQUA 0x04FF
#define BLUE 0x001F
#define MAGENTA 0xF81F
#define PINK 0xF8FF
//#define ST7735 NEW 0xD500FF




AS726X sensor;

int RawRed = 0;
int RawOrange = 0;
int RawYellow = 0;
int RawGreen = 0;
int RawBlue = 0;
int RawViolet = 0;
float CalcRed  = 0.0;
float CalcOrange = 0.0;
float CalcYellow = 0.0;
float CalcGreen = 0.0;
float CalcBlue = 0.0;
float CalcViolet = 0.0;
int HeightRed = 0;
int HeightOrange = 0;
int HeightYellow = 0;
int HeightGreen = 0;
int HeightBlue = 0;
int HeightViolet = 0;

byte gain = 3; //  values: 0,1,2,3
float gainValue = 64;
int menu_number = 0; //  values:0 menu 1,2,3 bez led, 4,5,6 z LED
int LedPower = 0;
int GraphHeight = 128;
int GrapWidth = 25;
float ReadValues[6] = {0, 0, 0, 0, 0, 0};
float maxValue = 0.0;
float minValue = 0.0;
float test_vle = 1.0;
char FloatInChar[20];
char ChValue[20];







void setup() {
  tft.init();
  tft.setRotation(1);
  tft.fillScreen(TFT_BLACK);

  tft.setTextColor(TFT_YELLOW, TFT_BLACK);
  tft.setTextSize(2);
  tft.drawString("Version:"+ String(Version), 0, 0);
  delay(500);
  Wire.begin();
  sensor.begin();
  pinMode(Btn1, INPUT);
  pinMode(Btn2, INPUT);
  tft.fillRect(0, 0, 240, 240, TFT_BLACK);

}
void drawControls() {
    // tft.fillRect(0, 0, 240, 240, TFT_BLACK);

  //  ////tft.setRotation(0);
  //    linia pozycyjna LED
  tft.drawPixel(127, 24, WHITE);
  if (LedPower == 1) {
    tft.drawLine(127, 26, 127, 31, WHITE);
  }
  //gain:
  tft.drawPixel(127, 0, WHITE);
  if (gain >= 1) {
    tft.drawPixel(127, 2, WHITE);
  }
  if (gain >= 2) {
    tft.drawPixel(127, 4, WHITE);
  }
  if (gain >= 3) {
    tft.drawPixel(127, 6, WHITE);
  }
  // menu
  if (menu_number >= 1  ) {
    tft.drawPixel(127, 10, WHITE);
  }
  if ((menu_number >= 2 && menu_number < 5 ) || menu_number >= 6 ) {
    tft.drawPixel(127, 12, WHITE);
  }
  if ((menu_number >= 3 && menu_number < 5)  || menu_number >= 7 ) {
    tft.drawPixel(127, 14, WHITE);
  }
  if ((menu_number >= 4 && menu_number < 5 ) || menu_number >= 8 ) {
    tft.drawPixel(127, 16, WHITE);
  }

}
void loop() {
  tft.setTextSize(1);
  tft.setTextColor(TFT_WHITE, TFT_BLACK);

// //  //TESTY
//  tft.setTextColor(TFT_WHITE);
//  tft.drawString("test",0, 0);
//  tft.fillRect(0, 64 , 25, 64, TFT_RED);
//  tft.fillRect(25, 64 , 25, 64, TFT_ORANGE);
//  tft.fillRect(50, 64 , 25, 64, TFT_YELLOW);
//  tft.fillRect(75, 64 , 25, 64, TFT_GREEN);
//  tft.fillRect(100, 64 , 25, 64, TFT_BLUE);
//  tft.fillRect(125, 64 , 25, 64, TFT_MAGENTA);
//  tft.fillRect(150, 64 , 25, 64, TFT_WHITE);




  //menu change
  if (digitalRead(Btn2) == HIGH) {
    tft.fillRect(0, 0, 240, 240, TFT_BLACK);
    if (menu_number < 8) {
      menu_number = menu_number + 1;
    } else {
      menu_number = 0;
    }
  }


  if (menu_number == 0 ) {//show menu
    tft.setTextSize(2);
    tft.drawString("GAIN:" + String(gainValue),0, 0);
    tft.drawString("LED: 5-8",0, 16);
    tft.drawString("0:Menu",0, 32);
    tft.drawString("1:Calc.Values",0, 48);
    tft.drawString("2:Raw",0, 64);
    tft.drawString("3:Rel. Graph",0, 80);
    tft.drawString("4:Calc. Graph",0, 96);
    drawControls();

    delay(100);  
  }
// show calculated values
  if (menu_number == 1 || menu_number == 5  ) {//show calculated values
    drawControls();
    tft.drawString("Calculated Values",0,0);
    tft.drawString("R:" + String(CalcRed),0, 20);
    tft.drawString("O:" + String(CalcOrange),0, 39);
    tft.drawString("Y:" + String(CalcYellow),0, 58);
    tft.drawString("G:" + String(CalcGreen),0, 77);
    tft.drawString("B:" + String(CalcBlue),0, 96);
    tft.drawString("V:" + String(CalcViolet),0, 115);
  }
  // //show raw values
  if (menu_number == 2 || menu_number == 6) {
    drawControls();
    tft.drawString("Raw Values",0,0);
    tft.drawString("R:" + String(RawRed),0, 20);
    tft.drawString("O:" + String(RawOrange),0, 39);
    tft.drawString("Y:" + String(RawYellow),0, 58);
    tft.drawString("G:" + String(RawGreen),0, 77);
    tft.drawString("B:" + String(RawBlue),0, 96);
    tft.drawString("V:" + String(RawViolet),0, 115);

  }
  //show relative calculated values graph
   if (menu_number == 3 || menu_number == 7 ){
    tft.setTextColor(TFT_WHITE, TFT_BLACK);

    maxValue = 0.0;
    minValue = 0.0;
      for (int i = 0; i < 5; i++) {
          maxValue = max(maxValue, ReadValues[i]);
      }
      HeightRed     = int((CalcRed     / maxValue) * GraphHeight);
      HeightOrange  = int((CalcOrange  / maxValue) * GraphHeight);
      HeightYellow  = int((CalcYellow  / maxValue) * GraphHeight);
      HeightGreen   = int((CalcGreen   / maxValue) * GraphHeight);
      HeightBlue    = int((CalcBlue    / maxValue) * GraphHeight);
      HeightViolet  = int((CalcViolet  / maxValue) * GraphHeight);
    for (int i = 0; i < 5; i++) {
          maxValue = min(maxValue, ReadValues[i]);
      }

      tft.fillRect(minValue, 0, 240, maxValue-minValue, TFT_BLACK);
      drawControls();
      tft.setTextColor(TFT_WHITE, TFT_BLACK);
      tft.drawString("Relvative Calculated Values",0,0);
       drawControls();
      tft.fillRect(0 * GrapWidth,     GraphHeight - HeightRed+8,    GrapWidth, HeightRed,     TFT_RED);
      tft.fillRect(1 * GrapWidth + 1, GraphHeight - HeightOrange+8, GrapWidth, HeightOrange,  TFT_ORANGE);
      tft.fillRect(2 * GrapWidth + 2, GraphHeight - HeightYellow+8, GrapWidth, HeightYellow,  TFT_YELLOW);
      tft.fillRect(3 * GrapWidth + 3, GraphHeight - HeightGreen+8,  GrapWidth, HeightGreen,   TFT_GREEN);
      tft.fillRect(4 * GrapWidth + 4, GraphHeight - HeightBlue+8,   GrapWidth, HeightBlue,    TFT_BLUE);
      tft.fillRect(5 * GrapWidth + 5, GraphHeight - HeightViolet+8, GrapWidth, HeightViolet,  TFT_MAGENTA);
      tft.setCursor(1 * GrapWidth -10, 0);

      

    tft.setTextColor(TFT_BLACK, TFT_RED);
    tft.drawString(" R:",1 * GrapWidth -20, 120);
    tft.setTextColor(TFT_BLACK, TFT_ORANGE);
    tft.drawString(" O:",2 * GrapWidth -19, 120);
    tft.setTextColor(TFT_BLACK, TFT_YELLOW);
    tft.drawString(" Y:",3 * GrapWidth -18, 120);
    tft.setTextColor(TFT_BLACK, TFT_GREEN);
    tft.drawString(" G:",4 * GrapWidth -17, 120);
    tft.setTextColor(TFT_BLACK, TFT_BLUE);
    tft.drawString(" B:",5 * GrapWidth -16, 120);
    tft.setTextColor(TFT_BLACK, TFT_MAGENTA);
    tft.drawString(" V:",6 * GrapWidth -15, 120);
     maxValue = 0.0;
     minValue = 0.0;
    }



    // //show relative raw data  graph
    if (menu_number == 4 || menu_number == 8 ){
    tft.setTextColor(TFT_WHITE, TFT_BLACK);
    maxValue = 0.0;
    minValue = 0.0;
      for (int i = 0; i < 5; i++) {
          maxValue = max(maxValue, ReadValues[i]);
      }
      for (int i = 0; i < 5; i++) {
          minValue = min(minValue, ReadValues[i]);
      }
      HeightRed     = int(((RawRed + abs(minValue))     / (maxValue + abs(minValue))) * GraphHeight);
      HeightOrange  = int(((RawOrange + abs(minValue))  / (maxValue + abs(minValue))) * GraphHeight);
      HeightYellow  = int(((RawYellow + abs(minValue))  / (maxValue + abs(minValue))) * GraphHeight);
      HeightGreen   = int(((RawGreen + abs(minValue))   / (maxValue + abs(minValue))) * GraphHeight);
      HeightBlue    = int(((RawBlue + abs(minValue))    / (maxValue + abs(minValue))) * GraphHeight);
      HeightViolet  = int(((RawViolet + abs(minValue))  / (maxValue + abs(minValue))) * GraphHeight);
      tft.fillRect(minValue, 0, 240, maxValue-minValue, TFT_BLACK);
      drawControls();
      tft.setTextColor(TFT_WHITE, TFT_BLACK);
      tft.drawString("Relvative Raw Values",0,0);

      tft.fillRect(0 * GrapWidth,     GraphHeight - HeightRed+8,    GrapWidth, HeightRed,     TFT_RED);
      tft.fillRect(1 * GrapWidth + 1, GraphHeight - HeightOrange+8, GrapWidth, HeightOrange,  TFT_ORANGE);
      tft.fillRect(2 * GrapWidth + 2, GraphHeight - HeightYellow+8, GrapWidth, HeightYellow,  TFT_YELLOW);
      tft.fillRect(3 * GrapWidth + 3, GraphHeight - HeightGreen+8,  GrapWidth, HeightGreen,   TFT_GREEN);
      tft.fillRect(4 * GrapWidth + 4, GraphHeight - HeightBlue+8,   GrapWidth, HeightBlue,    TFT_BLUE);
      tft.fillRect(5 * GrapWidth + 5, GraphHeight - HeightViolet+8, GrapWidth, HeightViolet,  TFT_MAGENTA);
    tft.setTextColor(TFT_BLACK, TFT_RED);
    tft.drawString(" R:",1 * GrapWidth -20, 120);
    tft.setTextColor(TFT_BLACK, TFT_ORANGE);
    tft.drawString(" O:",2 * GrapWidth -19, 120);
    tft.setTextColor(TFT_BLACK, TFT_YELLOW);
    tft.drawString(" Y:",3 * GrapWidth -18, 120);
    tft.setTextColor(TFT_BLACK, TFT_GREEN);
    tft.drawString(" G:",4 * GrapWidth -17, 120);
    tft.setTextColor(TFT_BLACK, TFT_BLUE);
    tft.drawString(" B:",5 * GrapWidth -16, 120);
    tft.setTextColor(TFT_BLACK, TFT_MAGENTA);
    tft.drawString(" V:",6 * GrapWidth -15, 120);
     maxValue = 0.0;
     minValue = 0.0;
    }
  delay(100);
  
  if (menu_number > 4 ) {
    LedPower = 1;
  } else {
    LedPower = 0;
  }

  if (LedPower == 1) {
    sensor.enableBulb();
    sensor.takeMeasurementsWithBulb();
  } else {
    sensor.disableBulb();
    sensor.takeMeasurements();
  }

  //gain change
  if (digitalRead(Btn1) == HIGH) {
    if (gain < 3) {
      gain = gain + 1;
      sensor.setGain(gain);

    } else {
      gain = 0;
      sensor.setGain(gain);
    }
  }

  if (gain == 0 ) {
    gainValue = 1;
  } else if (gain == 1) {
    gainValue = 3.7;
  } else if (gain == 2) {
    gainValue = 16;
  } else {
    gainValue = 64;
  }



  //data haverest
  RawRed    = sensor.getRed();
  RawOrange = sensor.getOrange();
  RawYellow = sensor.getYellow();
  RawGreen  = sensor.getGreen();
  RawBlue   = sensor.getBlue();
  RawViolet = sensor.getViolet();

  CalcRed    = sensor.getCalibratedRed();
  CalcOrange = sensor.getCalibratedOrange();
  CalcYellow = sensor.getCalibratedYellow();
  CalcGreen  = sensor.getCalibratedGreen();
  CalcBlue   = sensor.getCalibratedBlue();
  CalcViolet = sensor.getCalibratedViolet();

  ReadValues[0] = CalcRed;
  ReadValues[1] = CalcOrange;
  ReadValues[2] = CalcYellow;
  ReadValues[3] = CalcGreen;
  ReadValues[4] = CalcBlue;
  ReadValues[5] = CalcViolet;
}