from percettrone import Percettrone

# rete neurale
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) """

def stampa_operazione(p, y, x1, x2, isFinal, errore = 0):
    if isFinal == 0:
        print(f"\tW1: {p.w1}")
        print(f"X1: {x1} -------->          ")
        print(f"\t\t( bias: {p.bias} ) -------> Y: {y}")
        print(f"X2: {x2} -------->          ")
        print(f"\tW2: {p.w2}")
    elif isFinal == 1:
        print(f"\t\t\t\tW1: {p.w1}")
        print(f"\t\t\tX1: {x1} -------->          ")
        print(f"\t\t\t\t\t( bias: {p.bias} ) -------> Y: {y}")
        print(f"\t\t\tX2: {x2} -------->          ")
        print(f"\t\t\t\tW2: {p.w2}")
    else:
        print(f"\t\t\t\t\t\tW1: {p.w1}")
        print(f"\t\t\t\t\tX1: {x1} -------->          ")
        print(f"\t\t\t\t\t\t\t( bias: {p.bias} ) -------> Y: {y} ----> errore: {errore}")
        print(f"\t\t\t\t\tX2: {x2} -------->          ")
        print(f"\t\t\t\t\t\tW2: {p.w2}")
def stampa_risultati_multilayer():
    print("Percettrone Esterno 1:")
    print(f"\t W1: {pin_est_1.w1}, W2: {pin_est_1.w2}, bias: {pin_est_1.bias}")
    print("Percettrone Esterno 2:")
    print(f"\t W1: {pin_est_2.w1}, W2: {pin_est_2.w2}, bias: {pin_est_2.bias}")
    #print("Percettrone Interno 1:")
    #print(f"\t W1: {pin_int_1.w1}, W2: {pin_int_1.w2}, bias: {pin_int_1.bias}")
    #print("Percettrone Interno 2:")
    #print(f"\t W1: {pin_int_2.w1}, W2: {pin_int_2.w2}, bias: {pin_int_2.bias}")
    print("Percettrone OUT:")
    print(f"\t W1: {pinout.w1}, W2: {pinout.w2}, bias: {pinout.bias}")

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

    if corrette == 4:
        print(f"Epoche necessarie: {i-1}")
        stampa_risultati_multilayer()
        break
    
    print(f"\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\tEPOCA {i}")
    corrette = 0;

    for j in range(0,4): #Combinazioni
        print("\n")
        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

        stampa_operazione(pin_est_1, y_est_1, x[j][0], x[j][1], 0)
        stampa_operazione(pinout, yout, y_est_1, y_est_2, 2, errore)
        stampa_operazione(pin_est_2, y_est_2, x[j][0], x[j][1], 0)

        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