import time
import board
import digitalio

# -----------------------------
# Keypad setup (4x4)
# -----------------------------
rows = [board.GP13, board.GP12, board.GP11, board.GP10]   # outputs
cols = [board.GP9, board.GP8, board.GP7, board.GP6]       # inputs

row_pins = []
col_pins = []

for r in rows:
    pin = digitalio.DigitalInOut(r)
    pin.direction = digitalio.Direction.OUTPUT
    pin.value = True
    row_pins.append(pin)

for c in cols:
    pin = digitalio.DigitalInOut(c)
    pin.direction = digitalio.Direction.INPUT
    pin.pull = digitalio.Pull.UP
    col_pins.append(pin)

# Key map (row-column mapping)
keys = [
    ['1', '2', '3', 'A'],
    ['4', '5', '6', 'B'],
    ['7', '8', '9', 'C'],
    ['*', '0', '#', 'D']
]

# -----------------------------
# 7-Segment setup
# -----------------------------
seg_pins = [
    board.GP16, # a
    board.GP17, # b
    board.GP18, # c
    board.GP19, # d
    board.GP20, # e
    board.GP21, # f
    board.GP22  # g
]

seg_outputs = []
for p in seg_pins:
    pin = digitalio.DigitalInOut(p)
    pin.direction = digitalio.Direction.OUTPUT
    seg_outputs.append(pin)

# -----------------------------
# 7-Segment Character Map
# -----------------------------
segments = {
    '0': [1,1,1,1,1,1,0],
    '1': [0,1,1,0,0,0,0],
    '2': [1,1,0,1,1,0,1],
    '3': [1,1,1,1,0,0,1],
    '4': [0,1,1,0,0,1,1],
    '5': [1,0,1,1,0,1,1],
    '6': [1,0,1,1,1,1,1],
    '7': [1,1,1,0,0,0,0],
    '8': [1,1,1,1,1,1,1],
    '9': [1,1,1,1,0,1,1],
    'A': [1,1,1,0,1,1,1],   # A
    'B': [1,1,1,1,1,1,1],   # B
    'C': [0,0,0,1,1,0,1],   # C (lower c)
    'D': [0,1,1,1,1,0,1],   # D
    '-': [0,0,0,0,0,0,1],   # dash for '*'
    'h': [0,1,1,0,1,1,1],   # h for '#'
    ' ': [0,0,0,0,0,0,0]    # blank
}

# -----------------------------
# Keypad scanning function
# -----------------------------
def scan_keypad():
    for i, r in enumerate(row_pins):
        r.value = False
        for j, c in enumerate(col_pins):
            if not c.value:  # active LOW
                r.value = True
                return keys[i][j]
        r.value = True
    return None

# -----------------------------
# Display function
# -----------------------------
def display(char):
    pattern = segments.get(char, segments[' '])
    for i in range(7):
        seg_outputs[i].value = pattern[i]

# -----------------------------
# Main loop
# -----------------------------
last_key = None

while True:
    key = scan_keypad()
    if key and key != last_key:
        print("Key:", key)

        if key.isdigit():
            display(key)
        elif key in ['A', 'B', 'C', 'D']:
            display(key)
        elif key == '*':
            display('-')
        elif key == '#':
            display('h')

        last_key = key

    if not key:
        last_key = None  # reset when released

    time.sleep(0.05)