"""
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
┃ Raspberry Pi Pico SSD1306 OLED Display (MicroPython)     ┃
┃                                                          ┃
┃ A program to display Raspberry Pi logo, text, and a      ┃
┃ simple timer animation on an SSD1306 OLED display        ┃
┃ connected to a Raspberry Pi Pico.                        ┃
┃                                                          ┃
┃ Copyright (c) 2023 Anderson Costa                        ┃
┃ GitHub: github.com/arcostasi                             ┃
┃ License: MIT                                             ┃
┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
"""

from machine import Pin, I2C
from ssd1306 import SSD1306_I2C
import framebuf, sys
import utime

pix_res_x = 128
pix_res_y = 64

button_Red = Pin(15, Pin.IN, Pin.PULL_UP)
button_Black = Pin(14, Pin.IN, Pin.PULL_UP)

def init_i2c(scl_pin, sda_pin):
   i2c_dev = I2C(1, scl=Pin(scl_pin), sda=Pin(sda_pin), freq=200000)
   i2c_addr = [hex(ii) for ii in i2c_dev.scan()]
   if not i2c_addr:

       print('No I2C Display Found')
       sys.exit()
   else:
       print("I2C Address      : {}".format(i2c_addr[0]))
       print("I2C Configuration: {}".format(i2c_dev))
   
   return i2c_dev

def display_logo(oled):
   buffer = bytearray(b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00|?\x00\x01\x86@\x80\x01\x01\x80\x80\x01\x11\x88\x80\x01\x05\xa0\x80\x00\x83\xc1\x00\x00C\xe3\x00\x00~\xfc\x00\x00L'\x00\x00\x9c\x11\x00\x00\xbf\xfd\x00\x00\xe1\x87\x00\x01\xc1\x83\x80\x02A\x82@\x02A\x82@\x02\xc1\xc2@\x02\xf6>\xc0\x01\xfc=\x80\x01\x18\x18\x80\x01\x88\x10\x80\x00\x8c!\x00\x00\x87\xf1\x00\x00\x7f\xf6\x00\x008\x1c\x00\x00\x0c \x00\x00\x03\xc0\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00")
   fb = framebuf.FrameBuffer(buffer, 32, 32, framebuf.MONO_HLSB)

   oled.fill(0)
   oled.blit(fb, 96, 0)
   oled.show()

def display_text(oled):
   oled.text("Raspberry Pi", 5, 5)
   oled.text("Pico", 5, 15)
   oled.show()

def display_anima(oled):
   counter = 0
   last_state_Red = 1
   last_state_Black = 1

   blink = True
   last_blink_time = utime.ticks_ms()
   blink_interval = 500

   
   while True:
       current_state_Red = button_Red.value()
       current_state_Black = button_Black.value()

       if last_state_Red == 1 and current_state_Red == 0:
           if counter < 10:
               counter += 1
            

       if last_state_Black == 1 and current_state_Black == 0:
           counter = 0

       last_state_Red = current_state_Red
       last_state_Black = current_state_Black
      
       current_time = utime.ticks_ms()
       if utime.ticks_diff(current_time, last_blink_time) > blink_interval:
            blink = not blink
            last_blink_time = current_time

       oled.fill_rect(0, 30, oled.width, 20, 0)

       if counter >= 10:
            if blink:
                 oled.text("MAX LOAD", 5, 30)
                 oled.text("REACHED", 5, 40)
       else:
           oled.text("Counter:", 5, 30)
           oled.text(str(counter), 5, 40)

       oled.show()
       utime.sleep_ms(50)

def main():
   i2c_dev = init_i2c(scl_pin=27, sda_pin=26)
   oled = SSD1306_I2C(pix_res_x, pix_res_y, i2c_dev)
   display_logo(oled)
   display_text(oled)
   display_anima(oled)

if __name__ == '__main__':
   main()