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

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

MPU6050 mpu;

const int ledPin = 9;         
const int buzzerPin = 8;      
const int threshold = 1000;   
bool isWarningActive = false;
unsigned long lastBlinkTime = 0;
bool blinkState = false;

void setup() {
  Serial.begin(9600);
  Wire.begin();
  mpu.initialize();

  pinMode(ledPin, OUTPUT);
  pinMode(buzzerPin, OUTPUT);

  if (mpu.testConnection()) {
    Serial.println("MPU6050 connected successfully");
  } else {
    Serial.println("MPU6050 connection failed");
  }

  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("OLED FAILED"));
    for (;;);
  }
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
}

void loop() {
  int16_t ax, ay, az;
  mpu.getAcceleration(&ax, &ay, &az);


  long accelerometerValue = abs(ax) + abs(ay) + abs(az);


  if (accelerometerValue > threshold) {
    isWarningActive = true;
    digitalWrite(ledPin, HIGH);
    digitalWrite(buzzerPin, HIGH);
    displayWarning();
  } else {
    isWarningActive = false;
    digitalWrite(ledPin, LOW);
    digitalWrite(buzzerPin, LOW);
    display.clearDisplay();
    display.display();
  }

  if (isWarningActive) {
    unsigned long currentMillis = millis();
    if (currentMillis - lastBlinkTime >= 500) {
      lastBlinkTime = currentMillis;
      blinkState = !blinkState;
      displayWarning();
    }
  }

  delay(100); 
}

void displayWarning() {
  display.clearDisplay();
  display.setCursor(0, 10);
  display.print("PERINGATAN!");
  display.setCursor(0, 20);
  display.print("GEMPA");

  if (blinkState) {
    display.setCursor(0, 40);
    display.print("BERLINDUNG");
  }

  display.display();
}