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

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

// Inisialisasi DHT
#define DHTPIN 3      // Pin data sensor DHT
#define DHTTYPE DHT11 // Ubah ke DHT22 jika menggunakan sensor DHT22
DHT dht(DHTPIN, DHTTYPE);

// Pin sensor ultrasonik
#define TRIG_PIN 9
#define ECHO_PIN 10

// Pin tombol
#define BUTTON_PIN 2

// Variabel untuk penyimpanan data
float suhu = 0.0;
long duration;
float distance;

// Variabel untuk pergantian layar
int currentScreen = 0;
unsigned long lastDebounceTime = 0;
unsigned long debounceDelay = 10;

void setup() {
  // Inisialisasi OLED
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) { // Alamat I2C bisa berbeda, sesuaikan jika perlu
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }
  display.display();
  delay(2000); // Tampilkan layar awal selama 2 detik
  display.clearDisplay();

  // Inisialisasi DHT
  dht.begin();

  // Inisialisasi pin ultrasonik
  pinMode(TRIG_PIN, OUTPUT);
  pinMode(ECHO_PIN, INPUT);

  // Inisialisasi tombol dengan pull-up internal
  pinMode(BUTTON_PIN, INPUT_PULLUP);

  // Tampilkan pesan selamat datang
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(7, 0);
  display.print(">_<");
  display.display();
  delay(2000);
  display.clearDisplay();
}

void loop() {
  // Baca data dari sensor suhu
  suhu = dht.readTemperature();

  // Baca data dari sensor ultrasonik
  digitalWrite(TRIG_PIN, LOW);
  delayMicroseconds(2);
  digitalWrite(TRIG_PIN, HIGH);
  delayMicroseconds(10);
  digitalWrite(TRIG_PIN, LOW);
  duration = pulseIn(ECHO_PIN, HIGH);
  distance = duration * 0.034 / 2;

  // Deteksi tombol ditekan untuk pergantian layar
  if (digitalRead(BUTTON_PIN) == LOW) {
    unsigned long currentTime = millis();
    if (currentTime - lastDebounceTime > debounceDelay) {
      currentScreen = (currentScreen + 1) % 2; // Ganti layar
      lastDebounceTime = currentTime;
    }
  }

  // Tampilkan data sesuai layar yang dipilih
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(SSD1306_WHITE);
  if (currentScreen == 0) {
    display.setCursor(0, 0);
    display.print("Suhu:");
    display.setCursor(0, 30);
    display.print(suhu);
    display.print(" C");
  } else if (currentScreen == 1) {
    display.setCursor(0, 0);
    display.print("Jarak:");
    display.setCursor(0, 30);
    display.print(distance);
    display.print(" cm");
  }
  display.display();

  delay(500); // Tunggu sebentar sebelum mengulangi loop
}