# -------------------------------------------------- #
# This file is autogenerated by pioasm; do not edit! #
# -------------------------------------------------- #

from machine import Pin
from utime import sleep
import rp2
from rp2 import PIO
from machine import Pin
from time import sleep
# ------ #
# sevseg #
# ------ #
pins = [
    Pin(28, Pin.OUT),  # A
    Pin(27, Pin.OUT),  # B
    Pin(26, Pin.OUT),  # C
    Pin(22, Pin.OUT),  # D
    Pin(21, Pin.OUT),  # E
    Pin(20, Pin.OUT),  # F
    Pin(19, Pin.OUT),  # G
    Pin(0, Pin.OUT)   # DP (not connected)
]

# Common anode 7-segment display digit patterns
digits = [
    [0, 0, 0, 0, 0, 0, 1, 1], # 0
    [1, 0, 0, 1, 1, 1, 1, 1], # 1
    [0, 0, 1, 0, 0, 1, 0, 1], # 2 
    [0, 0, 0, 0, 1, 1, 0, 1], # 3
    [1, 0, 0, 1, 1, 0, 0, 1], # 4
    [0, 1, 0, 0, 1, 0, 0, 1], # 5
    [0, 1, 0, 0, 0, 0, 0, 1], # 6
    [0, 0, 0, 1, 1, 1, 1, 1], # 7
    [0, 0, 0, 0, 0, 0, 0, 1], # 8
    [0, 0, 0, 1, 1, 0, 0, 1], # 9
    [0, 0, 0, 1, 0, 0, 0, 1], # a
    [1, 1, 0, 0, 0, 0, 0, 1], # b
    [0, 1, 1, 0, 0, 0, 1, 1], # C
    [1, 0, 0, 0, 0, 1, 0, 1], # d
]

@rp2.asm_pio(set_init=[PIO.IN_HIGH]*4)
def keypad():
    wrap_target()
    set(y, 0)                             # 0
    label("1")
    mov(isr, null)                        # 1
    set(pindirs, 1)                       # 2
    in_(pins, 4)                          # 3
    set(pindirs, 2)                       # 4
    in_(pins, 4)                          # 5
    set(pindirs, 4)                       # 6
    in_(pins, 4)                          # 7
    set(pindirs, 8)                       # 8
    in_(pins, 4)                          # 9
    mov(x, isr)                           # 10
    jmp(x_not_y, "13")                    # 11
    jmp("1")                              # 12
    label("13")
    push(block)                           # 13
    irq(0)
    mov(y, x)                             # 14
    jmp("1")                              # 15
    wrap()

def reset():
    """Turns off all segments on the 7-segment display."""
    for pin in pins:
        pin.value(1)

reset()

for i in range(10, 14):
  Pin(i, Pin.IN, Pin.PULL_DOWN)

key_names = "*7410852#963DCBA"

def oninput(machine):
  keys = machine.get()
  while machine.rx_fifo():
      keys = machine.get()
  pressed = []
  for i in range(len(key_names)):
      if (keys==key_names[i]):
        for j in range(len(pins) - 1):
            pins[j].value(digits[i][j])
            sleep(0.5)
      pressed.append(key_names[i])
  print("Keys changed! pressed keys:", pressed)
  
sm = rp2.StateMachine(0, keypad, freq=2000, in_base=Pin(10, Pin.IN, Pin.PULL_DOWN), set_base=Pin(6))
sm.active(1)
sm.irq(oninput)

print("Please press any key on the keypad, or press Ctrl+C to enter REPL.")
while True:
  sleep(0.1)