from hcsr04 import HCSR04
from time import sleep
from machine import Pin, SoftI2C
import ssd1306
import math



# ESP32
sensor = HCSR04(trigger_pin=5, echo_pin=18, echo_timeout_us=10000)

# ESP8266
#sensor = HCSR04(trigger_pin=12, echo_pin=14, echo_timeout_us=10000)
i2c = SoftI2C(scl=Pin(22), sda=Pin(21))

# ESP8266 Pin assignment
#i2c = SoftI2C(scl=Pin(5), sda=Pin(4))

oled_width = 128
oled_height = 64
oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)
# oled.text("sdfsdf", 0, 0)

while True:
    oled.fill(0)
    distance = round(sensor.distance_cm())
    dis2 = str(distance)
    print('Distance:', distance, 'cm')
    oled.text("Water Tank: " + dis2 + "%", 0, 20)

    oled.show()
    sleep(1)