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


#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET    -1
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

#define SOIL_SENSOR_PIN 34  // Simulated Soil Moisture Sensor
#define TEMP_SENSOR_PIN 35  // Simulated Temperature Sensor
#define HUMIDITY_SENSOR_PIN 32  // Simulated Humidity Sensor
#define RELAY_PIN 23  // Water Pump Control
#define RED_LED 27  // Dry Condition Alert
#define GREEN_LED 26  // Normal Condition Indicator
#define BUZZER 25  // Alarm for Extreme Conditions

void setup() {
    Serial.begin(115200);
    
    pinMode(SOIL_SENSOR_PIN, INPUT);
    pinMode(TEMP_SENSOR_PIN, INPUT);
    pinMode(HUMIDITY_SENSOR_PIN, INPUT);
    pinMode(RELAY_PIN, OUTPUT);
    pinMode(RED_LED, OUTPUT);
    pinMode(GREEN_LED, OUTPUT);
    pinMode(BUZZER, OUTPUT);

    digitalWrite(RELAY_PIN, LOW);
    digitalWrite(RED_LED, LOW);
    digitalWrite(GREEN_LED, HIGH);
    digitalWrite(BUZZER, LOW);

    // Initialize OLED display
    if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
        Serial.println("SSD1306 allocation failed");
        for(;;);
    }

    display.clearDisplay();
    display.setTextSize(1);
    display.setTextColor(WHITE);
    display.setCursor(0, 0);
    display.println("Smart Agriculture System");
    display.display();
    delay(2000);
}

void loop() {
    int soilValue = analogRead(SOIL_SENSOR_PIN);
    int tempValue = analogRead(TEMP_SENSOR_PIN);
    int humidityValue = analogRead(HUMIDITY_SENSOR_PIN);
    
    float soilMoisture = map(soilValue, 0, 4095, 0, 100);  // 0 to 100%
    float temperature = map(tempValue, 0, 4095, 10, 50);  // 10°C to 50°C
    float humidity = map(humidityValue, 0, 4095, 20, 90);  // 20% to 90% RH

    Serial.print("Soil Moisture: "); Serial.print(soilMoisture); Serial.println(" %");
    Serial.print("Temperature: "); Serial.print(temperature); Serial.println(" C");
    Serial.print("Humidity: "); Serial.print(humidity); Serial.println(" %");

    // Update OLED Display
    display.clearDisplay();
    display.setCursor(0, 10);
    display.print("Soil Moisture: "); display.print(soilMoisture); display.println(" %");
    display.print("Temperature: "); display.print(temperature); display.println(" C");
    display.print("Humidity: "); display.print(humidity); display.println(" %");

    // Automation Logic
    if (soilMoisture < 30) {  // Dry Soil Condition
        digitalWrite(RED_LED, HIGH);
        digitalWrite(GREEN_LED, LOW);
        digitalWrite(BUZZER, HIGH);
        digitalWrite(RELAY_PIN, HIGH);  // Turn on water pump
        display.println("⚠ Irrigation ON! ⚠");
    } else {
        digitalWrite(RED_LED, LOW);
        digitalWrite(GREEN_LED, HIGH);
        digitalWrite(BUZZER, LOW);
        digitalWrite(RELAY_PIN, LOW);  // Turn off water pump
        display.println("🌱 Soil Moisture OK");
    }

    display.display();
    delay(2000);
}