from machine import Pin, time_pulse_us
import time

# Ultrasonic
trig = Pin(3, Pin.OUT)
echo = Pin(2, Pin.IN)

# LED
led_merah = Pin(6, Pin.OUT)
led_kuning = Pin(7, Pin.OUT)
led_hijau = Pin(8, Pin.OUT)

def baca_jarak():
    trig.low()
    time.sleep_us(2)
    trig.high()
    time.sleep_us(10)
    trig.low()

    durasi = time_pulse_us(echo, 1, 30000)
    jarak = (durasi * 0.0343) / 2
    return jarak

while True:
    jarak = baca_jarak()
    print("Jarak:", jarak, "cm")

    # Reset LED
    led_merah.off()
    led_kuning.off()
    led_hijau.off()

    if jarak < 10:
        led_merah.on()
    elif jarak < 30:
        led_kuning.on()
    elif jarak < 60:
        led_hijau.on()

    time.sleep(0.5)