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

from lab7_2 import samples_per_bit,am_transmit,am_receive

if __name__ == '__main__':
    message_size = 32

    # a random binary message
    message = numpy.random.randint(2,size=message_size)

    # run transmitter
    fc = 125e3  # carrier frequency
    xmit_out = am_transmit(message,samples_per_bit,fc)

    # plot spectrum of transmitted message
    xmit_out.spectrum(title='spectrum of received samples',
                      npoints=256000)

    ##################################################
    ## ADDED FOR TASK #4: a transmission delay
    ##################################################
    delayed_xmit_out = xmit_out.delay(nsamples=4)

    # run receiver
    received = am_receive(delayed_xmit_out,fc,samples_per_bit)

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

    # display the plots
    w.show()