#include <Wire.h>
#include <LiquidCrystal_I2C.h>
#include <MQUnifiedsensor.h>

#define Board ("ESP-32")
#define Pin (4)
#define Threshold (23)
#define Type ("MQ-2")
#define Voltage_Resolution (5.0)
#define ADC_Bit_Resolution (12)
#define RatioMQ2CleanAir (9.83)

// MQ2 Library Instance Setup
MQUnifiedsensor MQ2(Board, Voltage_Resolution, ADC_Bit_Resolution, Pin, Type);

// Display Setup
LiquidCrystal_I2C lcd(0x27, 16, 2);

// Variables
int gas_value = 0;

void setup() {
  // LCD Init
  lcd.init();
  lcd.begin(16, 2);
  lcd.backlight();
  lcd.clear();
  lcd.print("Gas Sensor");
  delay(1000);

  // Serial port configuration
  Serial.begin(9600);

  // Set Parameters to detect PPM concentration for LPG
  MQ2.setRegressionMethod(1);
  MQ2.setA(574.25);
  MQ2.setB(-2.222);

  // MQ2 init
  MQ2.init();

  // Calibrate Sensor
  Serial.print("Calibrating Sensor, please wait.");
  float calcR0 = 0;
  for (int i = 1; i <= 10; i++) {
    MQ2.update();
    calcR0 += MQ2.calibrate(RatioMQ2CleanAir);
    Serial.print(".");
  }
  MQ2.setR0(calcR0/10);
  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("MQ2 = ");
  lcd.print(calcR0);
  delay(500);

  // Exception handling
  // Connection Issue Exception
  if(isinf(calcR0)) { Serial.println("Warning: R0 value is infinite. Please check your wiring."); while(1);}
  if(calcR0 == 0) { Serial.println("Warning: R0 value is zero. Please check your wiring."); while(1);}

  // Show MQ2 Debug Log

  // Threshold Pin Setup
  pinMode(Threshold, INPUT);
}

void loop() {
  // Update reading
  MQ2.update();
  MQ2.readSensor();
  MQ2.serialDebug();
  // Read Values
  gas_value = analogRead(Pin);
  gas_value = map(gas_value, 0, 4095, 0, 100);
  Serial.println(gas_value);

  // Show value in display
  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("MQ2 = ");
  lcd.print(gas_value);
  lcd.print("%");

  // Show Threshold Indicator
  lcd.setCursor(0, 1);
  if(digitalRead(Threshold) == HIGH) {
    lcd.print("Unsafe");
  } else {
    lcd.print("Safe");
  }
  
  delay(500); 
}