# template tile for Lab #7, Task #3
import numpy
import waveforms as w

# copy over the body (non-test code) from your Task #2
# and put it here!

if __name__ == '__main__':
    n = 10  # no. of bits of ISI to test for
    # generate all possible n-bit sequences
    message = [0] * (n * 2**n)
    for i in xrange(2**n):
        for j in xrange(n):
            message[i*n + j] = 1 if (i & (1<<j)) else 0
    message = numpy.array(message,dtype=numpy.int)

    # run transmitter
    fc = 125e3
    xmit = am_transmit(message,samples_per_bit,fc)

    # run receiver
    receive = am_receive(xmit,fc,samples_per_bit)

    # report results
    print 'message: ',message
    print 'received:',receive
    if not numpy.array_equal(message,receive):
        print 'differences'
    else:
        print 'message received okay'

    # display the plots
    w.show()