from machine import Pin, I2C
import sh1106
import time

led= Pin("LED", Pin.OUT)
led.on()

i2c = I2C(1, scl=Pin(27), sda=Pin(26), freq=400000)
oled = sh1106.SH1106_I2C(128, 64, i2c, Pin(16),0x3c , rotate=180)
oled_width = 128
oled_height = 64

screen1_row1 = "Screen 1, row 1"
screen1_row2 = "Screen 1, row 2"
screen1_row3 = "Screen 1, row 3"

screen2_row1 = "Screen 2, row 1"
screen2_row2 = "Screen 2, row 2"

screen3_row1 = "Screen 3, row 1"

screen1 = [[0, 0 , screen1_row1], [0, 16, screen1_row2], [0, 32, screen1_row3]]
screen2 = [[0, 0 , screen2_row1], [0, 16, screen2_row2]]
screen3 = [[0, 40 , screen3_row1]]

# Scroll in screen horizontally from left to right
def scroll_h_in(screen):
  for i in range (0, oled_width+1, 4):
    for line in screen:
      oled.text(line[2], -oled_width+i, line[1])
    oled.show()
    if i!= oled_width:
      oled.fill(0)

# Scroll out screen horizontally from left to right
def scroll_h_out(speed):
  for i in range ((oled_width+1)/speed):
    for j in range (oled_height):
      oled.pixel(i, j, 0)
    oled.scroll(speed,0)
    oled.show()

# Continuous horizontal scroll
def scroll_h_in_out(screen):
  for i in range (0, (oled_width+1)*2, 1):
    for line in screen:
      oled.text(line[2], -oled_width+i, line[1])
    oled.show()
    if i!= oled_width:
      oled.fill(0)

# Scroll in screen vertically
def scroll_v_in(screen):
  for i in range (0, (oled_height+1), 1):
    for line in screen:
      oled.text(line[2], line[0], -oled_height+i+line[1])
    oled.show()
    if i!= oled_height:
      oled.fill(0)

# Scroll out screen vertically
def scroll_v_out(speed):
  for i in range ((oled_height+1)/speed):
    for j in range (oled_width):
      oled.pixel(j, i, 0)
    oled.scroll(0,speed)
    oled.show()

# Continous vertical scroll
def scroll_v_in_out(screen):
  for i in range (0, (oled_height*2+1), 1):
    for line in screen:
      oled.text(line[2], line[0], -oled_height+i+line[1])
    oled.show()
    if i!= oled_height:
      oled.fill(0)


while True:

  # Scroll in, stop, scroll out (horizontal)
  scroll_h_in(screen1)
  time.sleep(2)
  scroll_h_out(4)

  scroll_h_in(screen2)
  time.sleep(2)
  scroll_h_out(4)

  scroll_h_in(screen3)
  time.sleep(2)
  scroll_h_out(4)

  # Continuous horizontal scroll
  scroll_h_in_out(screen1)
  scroll_h_in_out(screen2)
  scroll_h_in_out(screen3)

  # Scroll in, stop, scroll out (vertical)
  scroll_v_in(screen1)
  time.sleep(2)
  scroll_v_out(4)

  scroll_v_in(screen2)
  time.sleep(2)
  scroll_v_out(4)

  scroll_v_in(screen3)
  time.sleep(2)
  scroll_v_out(4)

  # Continuous verticall scroll
  scroll_v_in_out(screen1)
  scroll_v_in_out(screen2)
  scroll_v_in_out(screen3)