from percettrone import Percettrone
from stampe_video import disegna_funzione, stampa_risultati_multilayer

x = [(0,0),(0,1),(1,0),(1,1)] # Combinazioni
#output = (0,1,1,0) # XOR Logico
#output = (0,0,0,1) # AND Logico
output = (0,1,1,1) # OR Logico
corrette = 0

pin_est_1 = Percettrone(bias=0)
pin_est_2 = Percettrone(bias=-1.5)
pinout = Percettrone(bias=0)
""" pin_est_1 = Percettrone(w1=0.3, w2=4, bias=-0.5)
pin_est_2 = Percettrone(w1=0.2, w2=4, bias=-1.5)
pinout = Percettrone(w1=0.3, w2=2, bias=-1) """

for i in range(1,100000): #Epoche

    if corrette == 4:
        print(f"Epoche necessarie: {i-1}")
        stampa_risultati_multilayer(pin_est_1, pin_est_2, pinout)
        break

    corrette = 0;
    print(f"\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\tEPOCA {i}")

    for j in range(0,4): #Combinazioni
        y_est_1 = pin_est_1.funzione_gradino(x[j][0], x[j][1])
        y_est_2 = pin_est_2.funzione_gradino(x[j][0], x[j][1])
        yout = pinout.funzione_gradino(y_est_1, y_est_2)

        errore = output[j] - yout

        print("\n")
        disegna_funzione(pin_est_1, y_est_1, x[j][0], x[j][1], False)
        disegna_funzione(pinout, yout, y_est_1, y_est_2, True, errore)
        disegna_funzione(pin_est_2, y_est_2, x[j][0], x[j][1], False)

        if errore != 0:
            pin_est_1.correggi_pesi(x[j][0], x[j][1], errore)
            pin_est_2.correggi_pesi(x[j][0], x[j][1], errore)
            pinout.correggi_pesi(y_est_1, y_est_2, errore)
        else:
            corrette += 1