from machine import Pin
import time

myled1 = Pin(0, Pin.OUT)  
myled2 = Pin(1, Pin.OUT) 
myled3 = Pin(2, Pin.OUT)  
myled4 = Pin(3, Pin.OUT)  
myled5 = Pin(4, Pin.OUT)  
myled6 = Pin(5, Pin.OUT)  
myled7 = Pin(6, Pin.OUT)
myled8 = Pin(7, Pin.OUT)  

rows = [Pin(18, Pin.OUT), Pin(15, Pin.OUT), Pin(17, Pin.OUT)]
columns = [Pin(9, Pin.IN), Pin(10, Pin.IN), Pin(11, Pin.IN), Pin(13, Pin.IN)]

def clear_all_leds():
    myled1.value(0)
    myled2.value(0)
    myled3.value(0)
    myled4.value(0)
    myled5.value(0)
    myled6.value(0)
    myled7.value(0)
    myled8.value(0)

while True:
    # Postavi red 1 aktivnim
    rows[0].value(1)
    rows[1].value(0)
    rows[2].value(0)
    
    if columns[0].value() == 1: myled1.value(1)
    if columns[1].value() == 1: myled2.value(1)
    if columns[2].value() == 1: myled3.value(1)
    
    # Postavi red 4 aktivnim
    rows[0].value(0)
    rows[1].value(0)
    rows[2].value(1)
    
    if columns[0].value() == 1: myled4.value(1)
    if columns[1].value() == 1: myled5.value(1)
    if columns[2].value() == 1: myled6.value(1)
    
    # Postavi red 2 aktivnim
    rows[0].value(0)
    rows[1].value(1)
    rows[2].value(0)
    
    if columns[0].value() == 1: myled7.value(1)
    if columns[1].value() == 1: myled8.value(1)
    
    # Ako je pritisnuta bilo koja kolona
    if columns[3].value() == 1:
        clear_all_leds()
    
    time.sleep_us(100000)