print("WATER STORAGE TUB LEVEL SENSOR")

# Import necessary module/libraries
import hcsr04  #ultrasonic library
import ssd1306 #oled library
from utime import sleep
from machine import SoftI2C,Pin, PWM #Pulse Width Mode

#Create an object called ultrasonic_sensor using OOP (Object Oriented Programming)
#OOP format is ---> library name.class name(.....)
ultrasonic_sensor = hcsr04.HCSR04(trigger_pin=13, echo_pin=27, echo_timeout_us=500*2*30) #echotimeout kiraan jarak 400cm

# Declare pin on OLED Display
i2c_oled = SoftI2C(scl=Pin(22), sda=Pin(21))
oled_width = 128 #based on spec
oled_height = 64 #based on spec

oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c_oled)

#Declare buzzer connection
buzzer = PWM(Pin(23),Pin.OUT)

#Declare LED pin
led_red = Pin(25,Pin.OUT) #Built-in LED is connected to Pin 12
led_yellow = Pin(33,Pin.OUT)
led_green = Pin(32,Pin.OUT)

#Declare push button is connected to pin 4
#push_button = Pin(4, Pin.IN)

while True:

    #Let's test our ultrasonic sensor
    level_in_cm = ultrasonic_sensor.distance_cm() #use when u need decimal point number
    level_in_mm = ultrasonic_sensor.distance_mm() #use when u want whole number

    #Display the water level values on Serial Monitor
    print('The water level in the water tank is :',level_in_cm,'cm')
    print('The water level in the water tank is :',level_in_mm,'mm')
    print('---------------------------------------------')
    print('\n')


    if level_in_cm < 50: # HIGH WATER LEVEL 
        for i in range (5):

            
            #Display water level value on OLED screen
            oled.fill(0) #To make screen becoming black
            oled.text('Water Level', 0, 0, 1) #1st 0 is X-axis, 2nd 0 is Y-axis, 1 for white,0 for black)
            oled.text('almost FULL!', 0, 10, 1)
            oled.text(str(level_in_cm), 0, 30, 1)  
            oled.text('cm',70,30,1)
            oled.text(str(level_in_mm), 0, 40, 1)
            oled.text('mm',40,40,1)

            oled.show()

            buzzer.init(freq=800,duty=272) #ubah value bunyi berlainan
            sleep(0.3)
            buzzer.init(freq=1,duty=0)
            sleep(0.3)

            led_green.on()
            sleep(0.2)
            led_green.off()
            sleep(0.1)
            led_red.off()
            led_yellow.off()


    elif 50 <= level_in_cm < 300: # NORMAL WATER LEVEL 
        
            buzzer.init(freq=1,duty=0)

            #Display water level value on OLED screen
            oled.fill(0) #To make screen becoming black
            oled.text('Water Level', 0, 0, 1) #1st 0 is X-axis, 2nd 0 is Y-axis, 1 for white,0 for black)
            oled.text('in range!', 0, 10, 1)
            oled.text(str(level_in_cm), 0, 30, 1)  
            oled.text('cm',70,30,1)
            oled.text(str(level_in_mm), 0, 40, 1)
            oled.text('mm',40,40,1)

            oled.show()

            led_yellow.on()
            sleep(1)
            led_red.off()
            
            
    else: # LOW WATER LEVEL
        for i in range (5):

            #Display water level value on OLED screen
            oled.fill(0) #To make screen becoming black
            oled.text('Water Level', 0, 0, 1) #1st 0 is X-axis, 2nd 0 is Y-axis, 1 for white,0 for black)
            oled.text('is LOW!', 0, 10, 1)
            oled.text(str(level_in_cm), 0, 30, 1)  
            oled.text('cm',70,30,1)
            oled.text(str(level_in_mm), 0, 40, 1)
            oled.text('mm',40,40,1)

            oled.show()

            buzzer.init(freq=1200,duty=272) 
            sleep(0.2)
            buzzer.init(freq=1,duty=0)
            sleep(0.2)

            led_red.on()
            sleep(0.2)
            led_red.off()
            sleep(0.1)
            led_green.off()
            led_yellow.off()

    sleep(3)