#include "DHT.h"
#include <Adafruit_MPU6050.h>
#include <Adafruit_Sensor.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <Wire.h>

#define DHTPIN 15
#define DHTTYPE DHT22

DHT dht(DHTPIN, DHTTYPE);

Adafruit_MPU6050 mpu;

Adafruit_SSD1306 display(128, 64, &Wire, -1);

void setup() {

  Serial.begin(115200);

  // Start I2C
  Wire.begin(21, 22);

  // Start OLED
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println("OLED not found!");
    while (1);
  }

  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println("System Starting...");
  display.display();

  // Start DHT22
  dht.begin();

  // Start MPU6050
  if (!mpu.begin()) {
    Serial.println("MPU6050 not found!");

    while (1) {
      delay(10);
    }
  }

  Serial.println("System started!");
}

void loop() {

  // Read DHT22
  float temperature = dht.readTemperature();
  float humidity = dht.readHumidity();

display.display();

  // Read MPU6050
  sensors_event_t acceleration;
  sensors_event_t gyro;
  sensors_event_t temp;

  mpu.getEvent(&acceleration, &gyro, &temp);

  // Calculate total acceleration
  float vibration = sqrt(
    acceleration.acceleration.x * acceleration.acceleration.x +
    acceleration.acceleration.y * acceleration.acceleration.y +
    acceleration.acceleration.z * acceleration.acceleration.z
  );

  // Calculate vibration change
  float vibrationChange = abs(vibration - 9.81);

  String vibrationStatus;

  if (vibrationChange < 2) {
    vibrationStatus = "LOW";
  }
  else if (vibrationChange < 5) {
    vibrationStatus = "MEDIUM";
  }
  else {
    vibrationStatus = "HIGH";
  }
  display.clearDisplay();
display.setCursor(0, 0);

display.print("Temperature: ");
display.print(temperature);
display.println(" C");

display.print("Humidity: ");
display.print(humidity);
display.println(" %");

display.print("Vibration: ");
display.println(vibrationStatus);

display.display();

  // Serial Monitor
  Serial.println("------------------------");

  Serial.print("Vibration: ");
  Serial.print(vibration);
  Serial.println(" m/s^2");

  Serial.print("Vibration Change: ");
  Serial.print(vibrationChange);
  Serial.println(" m/s^2");

  Serial.print("Vibration Status: ");
  Serial.println(vibrationStatus);

  Serial.print("Temperature: ");
  Serial.print(temperature);
  Serial.println(" °C");

  Serial.print("Humidity: ");
  Serial.print(humidity);
  Serial.println(" %");

  Serial.print("Acceleration X: ");
  Serial.print(acceleration.acceleration.x);
  Serial.println(" m/s^2");

  Serial.print("Acceleration Y: ");
  Serial.print(acceleration.acceleration.y);
  Serial.println(" m/s^2");

  Serial.print("Acceleration Z: ");
  Serial.print(acceleration.acceleration.z);
  Serial.println(" m/s^2");

  delay(2000);
}