"""
# for my keyboard, it has 61 buttons, each button only got 2 pins. when you press, it connects.

1.each button has two pin, one is at up side called pin1, another is at down side called pin2.
2.for row_x, all button pin1 is connected.
3.for column_x, all button pin2 is connected.
4.if row1 has 3v power, when you click a button, i can know what button you clicked by check the column input for each column. (if you pressed a button, that column will become 1, otherwise, 0.)
5.in mathmatics, column is x number, row is y number.
6.maybe I should put shift and ctrl in a splited row, so you can press it while hit other key.
"""
from machine import Pin
from time import sleep

column_pin_list = []
for column_pin_number in range(2,15+1):
    column_pin_list.append(Pin(column_pin_number, Pin.IN, Pin.PULL_DOWN))

row_pin_list = []
for row_pin_number in range(20, 25+1):
    row_pin_list.append(Pin(row_pin_number, Pin.OUT))

def get_a_raw_key_press(only_unpressed=True):
    for row_index, row_pin in enumerate(row_pin_list):
        # set other row to 0
        for temp_row_index, temp_row_pin in enumerate(row_pin_list):
            temp_row_pin.value(0)
        # set the target row to 1
        row_pin.value(1)
        for column_index, column_pin in enumerate(column_pin_list):
            if column_pin.value() == 1:
                # if pin is 1, it is pressed in this special case
                if only_unpressed == True:
                    while True:
                        if column_pin.value() == 0:
                            break
                return "{},{}".format(row_index, column_index)
        row_pin.value(0)
    return ""

key_table = { "0,0": 'Escape', "1,0": '`', "1,1": '1', "1,2": '2', "1,3": '3', "1,4": '4', "1,5": '5', "1,6": '6', "1,7": '7', "1,8": '8', "1,9": '9', "1,10": '0', "1,11": '-', "1,12": '=', "1,13": 'BackSpace', "2,0": 'Tab', "2,1": 'q', "2,2": 'w', "2,3": 'e', "2,4": 'r', "2,5": 't', "2,6": 'y', "2,7": 'u', "2,8": 'i', "2,9": 'o', "2,10": 'p', "2,11": '[', "2,12": ']', "2,13": '\\', "3,0": 'Caps_Lock', "3,1": 'a', "3,2": 's', "3,3": 'd', "3,4": 'f', "3,5": 'g', "3,6": 'h', "3,7": 'j', "3,8": 'k', "3,9": 'l', "3,10": ';', "3,11": "'", "3,12": "Return", "4,0": "Shift_L", "4,1": "z", "4,2": "x", "4,3": "c", "4,4": "v", "4,5": "b", "4,6": "n", "4,7": "m", "4,8": ",", "4,9": ".", "4,10": "/", "4,11": "Shift_R", "5,0": "Control_L", "5,1": "Super_L", "5,2": "Alt_L", "5,5": "space", "5,8": "Alt_R", "5,9": "Menu", "5,10": "Control_R", "5,11": "Left", "5,12": "Up", "5,13": "Right", }
key_table2 = { "1,0": '~', "1,1": '!', "1,2": '@', "1,3": '#', "1,4": '$', "1,5": '%', "1,6": '^', "1,7": '&', "1,8": '*', "1,9": '(', "1,10": ')', "1,11": '_', "1,12": '+', "2,1": 'Q', "2,2": 'W', "2,3": 'E', "2,4": 'R', "2,5": 'T', "2,6": 'Y', "2,7": 'U', "2,8": 'I', "2,9": 'O', "2,10": 'P', "2,11": '{', "2,12": '}', "2,13": '|', "3,1": 'A', "3,2": 'S', "3,3": 'D', "3,4": 'F', "3,5": 'G', "3,6": 'H', "3,7": 'J', "3,8": 'K', "3,9": 'L', "3,10": ':', "3,11": '"', "4,1": "Z", "4,2": "X", "4,3": "C", "4,4": "V", "4,5": "B", "4,6": "N", "4,7": "M", "4,8": "<", "4,9": ">", "4,10": "?", "5,12": "Down", }
shift_pin = Pin(26, Pin.PULL_DOWN)
use_which_key_table = 0
def get_key_press():
    global use_which_key_table
    raw_key = get_a_raw_key_press()
    result = key_table.get(raw_key)
    if result == "Caps_Lock":
        if use_which_key_table == 1:
            use_which_key_table = 0
        else:
            use_which_key_table = 1

    if shift_pin.value() == 1:
        use_which_key_table = 1
    else:
        use_which_key_table = 0

    if use_which_key_table == 0:
        temp_table = key_table
    else:
        temp_table = key_table2

    if raw_key in temp_table:
        result = temp_table[raw_key]
        return result
    return ""

def get_pressed_raw_character():
    value = get_key_press()
    if value == "BackSpace":
        value = "\b"
    if value == "Return":
        value = "\n"
    if value == "space":
        value = " "
    if value == "Tab":
        value = "    "
    if value != "":
        if (value != "") and ("Shift" not in value):
            print(value, end="")
            return value
    return ""

from machine import Pin
from soft_spi import Simple_Output_Soft_SPI
simple_output = Simple_Output_Soft_SPI(Pin(16), Pin(17), 200)

while True:
    try:
        value = get_pressed_raw_character()
        if value != "":
            #simple_output.write(value.encode("ascii"))
            simple_output.write(bytes([0x01, 0x02]) + value.encode("ascii") + bytes([0x04]) + bytes([0x00]*70))
        sleep(0.01)
    except Exception as e:
        print(e)