#include <Wire.h>
#include "rgb_lcd.h"
#include "Air_Quality_Sensor.h"

// Seeed Air Quality op poort A0
// RGB LCD op poort I2C

// Initialiseer het LCD en de sensor
rgb_lcd lcd;
AirQualitySensor sensor(A0);

void setup() {
    Serial.begin(9600);
    
    // Initialiseer LCD (16 kolommen, 2 rijen)
    lcd.begin(16, 2);
    lcd.print("Opwarmen...");

    // Initialiseer de sensor (opwarmtijd is nodig)
    if (sensor.init()) {
        Serial.println("Sensor klaar.");
    } else {
        Serial.println("Sensor ERROR!");
    }
    
    delay(2000); // Korte pauze voor het opstarten
}

void loop() {
    int quality = sensor.slope();
    int val = sensor.getValue();

    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Waarde: ");
    lcd.print(val);

    lcd.setCursor(0, 1);

    if (quality == AirQualitySensor::FORCE_SIGNAL || quality == AirQualitySensor::HIGH_POLLUTION) {
        lcd.setRGB(255, 0, 0); // Rood
        lcd.print("Zware vervuiling");
    } else if (quality == AirQualitySensor::LOW_POLLUTION) {
        lcd.setRGB(255, 150, 0); // Geel/Oranje
        lcd.print("Lage vervuiling");
    } else if (quality == AirQualitySensor::FRESH_AIR) {
        lcd.setRGB(0, 255, 0); // Groen
        lcd.print("Frisse lucht");
    }

    delay(1000); // Update elke seconde
}