#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <DHT.h>
#include <OneWire.h>
#include <DallasTemperature.h>

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

// DHT22 settings
#define DHTPIN 5      // Pin connected to the DHT22
#define DHTTYPE DHT22 // DHT22 type
DHT dht(DHTPIN, DHTTYPE);

// DS18B20 settings
#define ONE_WIRE_BUS 5 // Pin connected to the DS18B20
OneWire oneWire(ONE_WIRE_BUS);
DallasTemperature sensors(&oneWire);

DeviceAddress sensorAddress;

void setup() {
  Serial.begin(9600);
  
  // Initialize OLED display
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for(;;); // Don't proceed, loop forever
  }
  display.clearDisplay();
  
  // Initialize sensors
  dht.begin();
  sensors.begin();
  
  // Search for DS18B20 sensor
  if (sensors.getAddress(sensorAddress, 0)) {
    Serial.println(F("DS18B20 sensor found!"));
  } else {
    Serial.println(F("No DS18B20 sensor found."));
  }

  delay(2000); // Wait for sensor initialization
}


void printAddress(DeviceAddress deviceAddress) {
  for (uint8_t i = 0; i < 8; i++) {
    if (deviceAddress[i] < 16) display.print("0");
    display.print(deviceAddress[i], HEX);
  }
}

void loop() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  
  // Check for DHT22 sensor
  float humidity = dht.readHumidity();
  float temperatureDHT = dht.readTemperature();
  
  if (!isnan(humidity) && !isnan(temperatureDHT)) {
    display.setCursor(0, 0);
    display.println(F("DHT22 Sensor:"));
    display.print(F("Temp: "));
    display.print(temperatureDHT);
    display.println(F(" C"));
    display.print(F("Humidity: "));
    display.print(humidity);
    display.println(F(" %"));
  } else {
    // If no valid DHT22 data, check DS18B20
    sensors.requestTemperatures(); // Send command to get temperatures
    float temperatureDS = sensors.getTempCByIndex(0);
    
    if (temperatureDS != DEVICE_DISCONNECTED_C) {
      display.setCursor(0, 0);
      display.println(F("DS18B20 Sensor:"));
      display.print(F("Temp: "));
      display.print(temperatureDS);
      display.println(F(" C"));
      
      // Print DS18B20 device ID
      display.print(F("ID: "));
      printAddress(sensorAddress);
    } else {
      display.setCursor(0, 0);
      display.println(F("No sensors detected"));
    }
  }

  display.display();
  delay(2000); // Update every 2 seconds
}