#define BLYNK_TEMPLATE_ID "TMPL6r-sgG6Jr"
#define BLYNK_TEMPLATE_NAME "ENERGY METER"
#define BLYNK_AUTH_TOKEN "d4TkH2lr6kHvIgNEfmsxpFPIoqymWpw9"

#include <BlynkSimpleEsp32.h>
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include "EmonLib.h"  // Include Emon Library

// Include library for temperature sensor
#include <OneWire.h>
#include <DallasTemperature.h>

#define SCREEN_WIDTH 128 // OLED width,  in pixels
#define SCREEN_HEIGHT 64 // OLED height, in pixels

// create an OLED display object connected to I2C
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire);

const char* ssid = "Wokwi-GUEST";
const char* password = "";

const int voltagePin = 35;  // Analog input pin for voltage measurement
const int currentPin = 34;  // Analog input pin for current measurement
const int ledPin = 4;       // Digital output pin for LED

// Add temperature sensor pins and objects
#define ONE_WIRE_BUS 32 // Pin connected to the temperature sensor
OneWire oneWire(ONE_WIRE_BUS);
DallasTemperature tempSensor(&oneWire);

EnergyMonitor currentSensor;

float voltage, current, power, temperature, costOfElectricity;

// Electricity rate per kWh
float electricityRate = 0.8; // Example rate of RM0.80 per kWh

BlynkTimer timer;

void setup() {
  Serial.begin(115200);
  
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }
  delay(500);
  display.clearDisplay();
  display.setTextColor(WHITE);

  Blynk.begin(BLYNK_AUTH_TOKEN, ssid, password);

  pinMode(voltagePin, INPUT);
  pinMode(currentPin, INPUT);
  pinMode(ledPin, OUTPUT); // Set LED pin as output

  // Initialize temperature sensor
  tempSensor.begin();

  timer.setInterval(1000L, sendDataToBlynk); // Send data to Blynk every 1 second
}

void sendDataToBlynk() {
  // Read raw analog values
  int rawVoltage = analogRead(voltagePin);
  int rawCurrent = analogRead(currentPin);

  // Convert raw values to physical quantities
  voltage = (rawVoltage / 4095.0) * 220;  // Assuming 220V reference

  float currentVoltage = (rawCurrent / 4095.0) * 220; // Assuming 220V reference
  const float sensorSensitivity = 0.1; // Sensor sensitivity in V/A (adjust according to datasheet)
  current = currentVoltage / sensorSensitivity; // Calculate current in Amperes
  
  // Calculate power using P = VI
  power = voltage * current;
  
  // Calculate cost of electricity
  costOfElectricity = (power / 1000) * electricityRate; // Convert power to kWh and multiply by rate

  // Read temperature from sensor
  tempSensor.requestTemperatures();
  temperature = tempSensor.getTempCByIndex(0); // Assuming only one sensor connected

  Serial.print("Voltage: ");
  Serial.print(voltage);
  Serial.print(" V, Current: ");
  Serial.print(current);
  Serial.print(" A, Power: ");
  Serial.print(power);
  Serial.print(" W, Temperature: ");
  Serial.print(temperature);
  Serial.print(" °C, Cost: RM");
  Serial.println(costOfElectricity, 2);
  
  Blynk.virtualWrite(V0, voltage); // Send voltage value to Blynk's virtual pin V0
  Blynk.virtualWrite(V1, current); // Send current value to Blynk's virtual pin V1
  Blynk.virtualWrite(V2, power);   // Send power value to Blynk's virtual pin V2
  Blynk.virtualWrite(V3, temperature); // Send temperature value to Blynk's virtual pin V3
  Blynk.virtualWrite(V4, costOfElectricity); // Send cost of electricity to Blynk's virtual pin V4

  // Light up LED if temperature is above 40°C
  if (temperature > 40) {
    digitalWrite(ledPin, HIGH); // Turn on LED
  } else {
    digitalWrite(ledPin, LOW); // Turn off LED
  }
}

void loop() {
  Blynk.run();
  timer.run();

  // Display values on the OLED
  display.setTextSize(1);
  display.setTextColor(WHITE);
  display.setCursor(0, 0);
  display.print("Voltage: ");
  display.println(voltage, 2);
  display.println(" V");

  display.setCursor(0, 16);
  display.print("Current: ");
  display.println(current, 2);
  display.println("A");

  display.setCursor(0, 32);
  display.print("Temperature: ");
  display.println(temperature, 2);
  display.println("°C");


  display.setCursor(0, 48);
  display.print("Cost: RM");
  display.println(costOfElectricity, 2);

  display.display();
  delay(1000); // Add a short delay to ensure the OLED has time to refresh
  display.clearDisplay();
}