#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>

#define SCREEN_WIDTH 128  // OLED display width
#define SCREEN_HEIGHT 64  // OLED display height
#define OLED_RESET    -1  // Reset pin (or -1 if sharing Arduino reset pin)

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

const int analog = 5;
const int Green = 17;
const int Yellow = 18;
const int Red = 19;

void setup() {
  // put your setup code here, to run once:
  Serial.begin(115200);
  pinMode(Green, OUTPUT);
  pinMode(Yellow, OUTPUT);
  pinMode(Red, OUTPUT);

    // Initialize OLED
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) { // Address 0x3C or 0x3D
    Serial.println(F("SSD1306 allocation failed"));
    for(;;); // Don't proceed, loop forever
  }

  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0,0);
  display.print("System Ready");
  display.setCursor(0,16);
  display.print("to test POT");
  display.display();
  
  digitalWrite(Green, LOW);
  digitalWrite(Yellow, LOW);
  digitalWrite(Red, LOW);
  delay(100);
}

void loop() {
  // put your main code here, to run repeatedly:
 float value = analogRead(analog);
 float voltage = (value/8191)*3.3; //Dmax = single read mode (12 bits)
  Serial.print("Analog value: "); Serial.print(value);
  Serial.print("\tVoltage: "); Serial.print(voltage);
  Serial.println();
  delay(10);
  display.clearDisplay();
  display.setCursor(0,0);
  display.print("Voltage: ");
  display.print(voltage);
  display.print(" V");
  display.display();
  delay(10);

  if(voltage >=0.1 && voltage <=1.3){
    digitalWrite(Green, HIGH);
    digitalWrite(Yellow, LOW);
    digitalWrite(Red, LOW);
    delay(10);
    display.setCursor(0,16);
    display.print("GREEN TRIG   ");
    display.display();
    delay(10);
  }
  else if(voltage >1.3 && voltage <=2.3){
    digitalWrite(Yellow, HIGH);
    digitalWrite(Green, LOW);
    digitalWrite(Red, LOW);
    delay(10);
    display.setCursor(0,16);
    display.print("YELLOW TRIG   ");
    display.display();
    delay(10);
  }
  else if(voltage >2.3){
    digitalWrite(Red, HIGH);
    digitalWrite(Green, LOW);
    digitalWrite(Yellow, LOW);
    delay(10);
    display.setCursor(0,16);
    display.print("RED TRIG       ");
    display.display();
    delay(10);
  }
  else{
    digitalWrite(Green, LOW);
    digitalWrite(Yellow, LOW);
    digitalWrite(Red, LOW);
    delay(10);
    display.setCursor(0,16);
    display.print("ALL OFF       ");
    display.display();
    delay(10);
  }
  delay(10); // this speeds up the simulation
}