#include <Robojax_AllegroACS_Current_Sensor.h>
#include <DallasTemperature.h>
#include <OneWire.h>
#include <TM1637Display.h>
#include <Adafruit_NeoPixel.h>
#include <Wire.h>
#define SECOND_ADDR 9
#ifdef __AVR__
  #include <avr/power.h>
#endif


//DECLARING CONSTANT VARIABLES AND PINS
const float Vratio = 5.128;   //ratio of voltage reduction based on experiment
const float VCC = 5.04;       // value of VCC of the board based on measurements
const int CurrentPin = A1;    // pin where current sensor is conncected
const int VoltPin = A0;        // pin where voltage sensor is connected
const int TempPin = 2;       // pin where temperature sensor is connected
const int MODEL = 0;          // value of the current sensor model, see below
const int CLKI = 3;          // clock for current (I) readout LED display 
const int DIOI = 4;          // DIO for current (I) readout LED display)
const int CLKV = 5;          // clock for voltage (V) readout LED display
const int DIOV = 6;          // DIO for voltage (V) readout LED display 
const int LEDPIN = 7;        // LED pin for Adafruit
const int NUMPIXELS = 8;     // Number of pixels on the Adafruit
int Voltbits;                // Bit value of the voltage reading from the volt sensor for I2C
int Currbits;                // Bit value for the current reading from the current sensor for I2C
int Tempbits;                // Bit value for the temperature reading from the temp sensor for I2C

// SET UP THE CURRENT SENSOR
  //'robojax' function uses format robojax(MODEL, INPUT-VARIABLE) where the model I am using is model ACS712ELECT-5B 
  //which corresponds to a value of 0 according to https://robojax.com/learn/arduino/?vid=robojax_Alegro_ACS712_curren_sensor
    Robojax_AllegroACS_Current_Sensor robojax(MODEL,CurrentPin);

//SET UP THE TEMPERATURE SENSOR
  //set the pin (TempPin) that the OneWire will communicate through
    OneWire oneWire(TempPin);
  //pass oneWire reference to Dallas Temperature
    DallasTemperature sensors(&oneWire);

//SET UP LED DISPLAYS
    TM1637Display displayI = TM1637Display(CLKI, DIOI);
    TM1637Display displayV = TM1637Display(CLKV, DIOV);

//SET UP LED BAR GRAPH
    Adafruit_NeoPixel pixels(NUMPIXELS, LEDPIN, NEO_GRB + NEO_KHZ800); 

//Function to perform when an I2C request is made from ESP32
void requestEvent(){

    Wire.write(Voltbits);
    Wire.write(Currbits);
    Wire.write(Tempbits);

    Serial.println("Request Received");
   }

void setup() {

  //Start serial monitor
  Serial.begin(115200);

  //Start sensors library for temperature sensor
  sensors.begin(); 

  //Clear LED voltage and current displays
  displayV.clear();
  displayI.clear();


  //Setup LED bar graph
  #if defined(__AVR_ATtiny85__) && (F_CPU == 16000000)
  clock_prescale_set(clock_div_1);
  #endif
  pixels.begin();
  pixels.clear();
  pixels.show();
  pixels.setBrightness(20);

  //SET UP I2C PROTOCOL
    Wire.begin(SECOND_ADDR);

  //SEND BIT VALUES FOR VOLTAGE CURRENT AND TEMPERATURE TO ESP32 VIA I2C when requested
   Wire.onRequest(requestEvent);
 

  //Set pin modes
  pinMode(CurrentPin,INPUT);
  pinMode(VoltPin, INPUT);
  pinMode(TempPin, INPUT);
   
}//end of void setup()

void loop() {

//GET AND PRINT CURRENT
  // Read current from the sensor
  float Current = robojax.getCurrent();

  //Map the current value to a value that can be sent via I2C
    Currbits = map(Current,-5,5,1,255);

  Serial.print("Current = ");
  Serial.print(Current,3); //print the current to 3 decimal places
  Serial.print(" A    ");

//GET AND PRINT VOLTAGE
  //Read the voltage from the volt sensor
  int Volts = analogRead(VoltPin);

   //Map the voltage value to 
    Voltbits = map(Volts,0,1023,1,255);

  //Map the voltage to a value that can be sent via I2C
  Voltbits = map(Volts,0,1023,1,255);

  //calculate voltage and convert from bit reading to voltage value in volts
  float Voltage = VCC*Volts*Vratio/1023; 
  
  Serial.print("Voltage = ");
  Serial.print(Voltage);
  Serial.print(" V    ");


//GET AND PRINT TEMPERATURE
  sensors.requestTemperatures();
  
  float tempC = sensors.getTempCByIndex(0);

  //Map the temperature to a value that can be sent via I2C
    Tempbits = map(tempC,0,125,1,255);

    Serial.print("Temperature = ");
    Serial.print(tempC);
    Serial.print(" degC   ");


//PRINT VOLTAGE AND CURRENT ON LED DISPLAYS
  int voltdisplay = Voltage*100;
  displayV.setBrightness(5); 
  displayV.showNumberDecEx(voltdisplay, 0b01000000, false, 4, 0);// Prints voltage on LED

  int currdisplay = Current *10000;
  displayI.setBrightness(5);
  displayI.showNumberDecEx(currdisplay, 0b01000000, false, 4, 1);// Prints voltage on LED
  
//DISPLAY TEMPERATURE ON LED BARGRAPH
  
  int led = 0;
  led = map(tempC, 20, 125, 0, 7);
  Serial.print("LED is :  ");
  Serial.println(led);


if(led == 0){
  pixels.clear();
  pixels.show();
  pixels.setPixelColor(0,pixels.Color(0,0,255));
  pixels.show();
  }

else if(led == 1){
  pixels.clear();
  pixels.show();
  pixels.setPixelColor(0,pixels.Color(0,0,255));
  pixels.setPixelColor(1,pixels.Color(0,0,128));
  pixels.show();
  }


else if(led == 2){
  pixels.clear();
  pixels.show();
  pixels.setPixelColor(0,pixels.Color(0,0,255));
  pixels.setPixelColor(1,pixels.Color(0,0,128));
  pixels.setPixelColor(2,pixels.Color(135,206,235));
  pixels.show();
  }

else if(led == 3){
  pixels.clear();
  pixels.show();
  pixels.setPixelColor(0,pixels.Color(0,0,255));
  pixels.setPixelColor(1,pixels.Color(0,0,128));
  pixels.setPixelColor(2,pixels.Color(135,206,235));
  pixels.setPixelColor(3,pixels.Color(255,255,0));
  pixels.show();
  }

else if(led == 4){
  pixels.clear();
  pixels.show();
  pixels.setPixelColor(0,pixels.Color(0,0,255));
  pixels.setPixelColor(1,pixels.Color(0,0,128));
  pixels.setPixelColor(2,pixels.Color(135,206,235));
  pixels.setPixelColor(3,pixels.Color(255,255,0));
  pixels.setPixelColor(4,pixels.Color(255,140,0));
  pixels.show();
  }


else if(led == 5){
  pixels.clear();
  pixels.show();
  pixels.setPixelColor(0,pixels.Color(0,0,255));
  pixels.setPixelColor(1,pixels.Color(0,0,128));
  pixels.setPixelColor(2,pixels.Color(135,206,235));
  pixels.setPixelColor(3,pixels.Color(255,255,0));
  pixels.setPixelColor(4,pixels.Color(255,140,0));
  pixels.setPixelColor(5,pixels.Color(255,69,0));
  pixels.show();
  }

else if(led == 6){
  pixels.clear();
  pixels.show();
  pixels.setPixelColor(0,pixels.Color(0,0,255));
  pixels.setPixelColor(1,pixels.Color(0,0,128));
  pixels.setPixelColor(2,pixels.Color(135,206,235));
  pixels.setPixelColor(3,pixels.Color(255,255,0));
  pixels.setPixelColor(4,pixels.Color(255,140,0));
  pixels.setPixelColor(5,pixels.Color(255,69,0));
  pixels.setPixelColor(6,pixels.Color(255,30,0));
  pixels.show();
  }
  
else if(led == 7){
  pixels.clear();
  pixels.show();
  pixels.setPixelColor(0,pixels.Color(0,0,255));
  pixels.setPixelColor(1,pixels.Color(0,0,128));
  pixels.setPixelColor(2,pixels.Color(135,206,235));
  pixels.setPixelColor(3,pixels.Color(255,255,0));
  pixels.setPixelColor(4,pixels.Color(255,140,0));
  pixels.setPixelColor(5,pixels.Color(255,69,0));
  pixels.setPixelColor(6,pixels.Color(255,30,0));
  pixels.setPixelColor(7,pixels.Color(255,0,0));
  pixels.show();
  }

  else {
    pixels.clear();
    pixels.show();
  }
 delay(500);

 
}//end of void loop()