import matplotlib.pyplot as plt
import math


def hamming(wz):
  coded = [0] * 7
  coded[0] = wz[0] #x7
  coded[1] = wz[1] #x6
  coded[2] = wz[2] #x5
  coded[4] = wz[3] #x3
  coded[3] = coded[2]^coded[1]^coded[0] #x4
  coded[5] = coded[4]^coded[1]^coded[0] #x2
  coded[6] = coded[4]^coded[2]^coded[0] #x1
  return coded


def corr():
    flaga = False
    corrected[0] = received[4] ^ received[2] ^ received[0] #x1'
    corrected[0] = received[4] ^ received[1] ^ received[0] #x2'
    corrected[0] = received[2] ^ received[1] ^ received[0] #x4'

    err[0] = coded[6] ^ corrected[0] #|x1
    err[0] = coded[5] ^ corrected[1] #|x2
    err[0] = coded[3] ^ corrected[2] #|x4

    err_count = err[0] + (err[1] * 2) + (err[2] * 4)
    if err_count == 0:
        print("Nie wystepuja bledy")
    else:
        corrected[0] = received[4] ^ received[2] ^ received[0] #x1'
        corrected[0] = received[4] ^ received[1] ^ received[0] #x2'
        corrected[0] = received[2] ^ received[1] ^ received[0] #x4'

        err[0] = coded[6] ^ corrected[0] #|x1
        err[0] = coded[5] ^ corrected[1] #|x2
        err[0] = coded[3] ^ corrected[2] #|x4

        err_count = err[0] + (err[1] * 2) + (err[2] * 4)
        if err_count == 0:
            print("Nie wystepuja bledy")

        else:
            print("Wystepuja bledy")


if __name__ == "__main__":
    wz = [1, 0, 1, 1]
    print("Bity: ", wz)

    coded = hamming(wz)
    print("Zakodowane: ",coded)