#include <Servo.h>
#include <LiquidCrystal_I2C.h>
#include <HX711.h>

const int trigPin1 = 9; // Ultrasonic sensor untuk mendeteksi orang
const int echoPin1 = 10;
const int trigPin2 = 11; // Ultrasonic sensor untuk mendeteksi ketinggian sampah
const int echoPin2 = 12;
const int pirPin = 2;
const int ledMerah = 3, ledKuning = 4, ledHijau = 5;
const int buzzerPin = 6;

Servo servoMotor;
LiquidCrystal_I2C lcd(0x27, 16, 2); // Pastikan alamat I2C sesuai
HX711 scale;

const int trashHeight = 100; // Asumsi tinggi tempat sampah dalam cm

void setup() {
  Serial.begin(9600); // Untuk debugging

  // Inisialisasi pin
  pinMode(trigPin1, OUTPUT);
  pinMode(echoPin1, INPUT);
  pinMode(trigPin2, OUTPUT);
  pinMode(echoPin2, INPUT);
  pinMode(pirPin, INPUT);
  pinMode(ledMerah, OUTPUT);
  pinMode(ledKuning, OUTPUT);
  pinMode(ledHijau, OUTPUT);
  pinMode(buzzerPin, OUTPUT);
  
  // Inisialisasi servo
  servoMotor.attach(7);

  // Inisialisasi LCD dan pengecekan
  if (!lcd.begin(16, 2)) { // Coba inisialisasi LCD
    Serial.println("LCD gagal diinisialisasi");
    while (1);
  }
  lcd.backlight();
  lcd.clear();
  lcd.print("Tes Komponen");
  delay(2000);
  
  // Inisialisasi load cell
  scale.begin(A1, A0);
  scale.set_scale(2280.f); 
  scale.tare();

  Serial.println("Setup selesai");
}

void loop() {
  // Cek apakah PIR mendeteksi orang
  if (digitalRead(pirPin) == HIGH) {
    long distance1 = getUltrasonicDistance(trigPin1, echoPin1);
    Serial.print("Jarak orang: ");
    Serial.println(distance1);
    
    if (distance1 < 50) {  // Jika seseorang terdeteksi dalam jarak 50 cm
      servoMotor.write(90); // Buka tutup
      delay(5000);          // Tunggu 5 detik sebelum menutup kembali
      servoMotor.write(0);  // Tutup kembali
    }
  }

  // Pengukuran ketinggian sampah
  long distance2 = getUltrasonicDistance(trigPin2, echoPin2);
  int trashLevel = trashHeight - distance2;
  Serial.print("Tinggi sampah: ");
  Serial.println(trashLevel);

  // Menentukan status sampah dan indikator
  lcd.clear();
  if (trashLevel > 75) {
    lcd.print("Tempat Sampah Penuh");
    digitalWrite(ledMerah, HIGH);
    digitalWrite(ledKuning, LOW);
    digitalWrite(ledHijau, LOW);
    digitalWrite(buzzerPin, HIGH); // Aktifkan buzzer
  } else if (trashLevel > 50) {
    lcd.print("Tempat Sampah Terisi");
    digitalWrite(ledMerah, LOW);
    digitalWrite(ledKuning, HIGH);
    digitalWrite(ledHijau, LOW);
    digitalWrite(buzzerPin, LOW); // Nonaktifkan buzzer
  } else {
    lcd.print("Tempat Sampah Kosong");
    digitalWrite(ledMerah, LOW);
    digitalWrite(ledKuning, LOW);
    digitalWrite(ledHijau, HIGH);
    digitalWrite(buzzerPin, LOW); // Nonaktifkan buzzer
  }

  // Tampilkan berat sampah saat penuh
  if (trashLevel > 75) {
    float weight = scale.get_units(10); 
    lcd.setCursor(0, 1);
    lcd.print("Berat: ");
    lcd.print(weight, 1);
    lcd.print(" kg");
    Serial.print("Berat sampah: ");
    Serial.println(weight);
  }

  delay(1000); // Waktu tunda untuk memperbarui data
}

// Fungsi untuk mendapatkan jarak menggunakan sensor ultrasonik
long getUltrasonicDistance(int trigPin, int echoPin) {
  digitalWrite(trigPin, LOW);
  delayMicroseconds(2);
  digitalWrite(trigPin, HIGH);
  delayMicroseconds(10);
  digitalWrite(trigPin, LOW);
  long duration = pulseIn(echoPin, HIGH);
  long distance = duration * 0.034 / 2; // Menghitung jarak dalam cm
  Serial.print("Jarak ultrasonik: ");
  Serial.println(distance);
  return distance;
}