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

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

const int buttonPin = 2; // Pin del botón
const int sensorPin = A0; // Pin del sensor analógico
const int dhtPin = 3; // Pin del DHT22
#define DHTTYPE DHT22 // Tipo de sensor DHT
DHT dht(dhtPin, DHTTYPE); // Inicialización del sensor DHT

int buttonState = 0; // Estado del botón
int lastButtonState = 0; // Último estado del botón
int counter = 0; // Contador

void setup() {
  pinMode(buttonPin, INPUT);
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);

  dht.begin(); // Inicialización del sensor DHT
}

void loop() {
  buttonState = digitalRead(buttonPin);

  if (buttonState == HIGH && lastButtonState == LOW) {
    // El botón ha sido pulsado
    counter++;
    if (counter > 3) {
      counter = 1; // Reseteamos a 1 si supera el 3
    }
    delay(200); // Anti-rebote
  }
  lastButtonState = buttonState;

  display.clearDisplay();
  display.setCursor(0, 0);

  float sensorValue = analogRead(sensorPin);
  display.print("Counter: ");
  display.println(counter);

  switch (counter) {
    case 1:
      display.print("Sensor Value 1: ");
      display.print(sensorValue/204.8);
      display.print("V");
      break;
    case 2: {
      float temperature = dht.readTemperature();
      float humidity = dht.readHumidity();

      if (isnan(temperature) || isnan(humidity)) {
        display.println("Error reading DHT22!");
      } else {
        display.print("Temp: ");
        display.print(temperature);
        display.println(" °C");

        display.print("Humidity: ");
        display.print(humidity);
        display.println(" %");
      }
      break;
    }
    case 3:
      display.print("Sensor Value 3: ");
      display.println(sensorValue);
      break;
  }
  
  display.display();
  delay(100);
}