# template tile for Lab #7, Task #2

import numpy
import waveforms as w

from lab7_1 import samples_per_bit,am_transmit,am_receive

if __name__ == '__main__':
    message_size = 32

    # two random binary messages
    message1 = numpy.random.randint(2,size=message_size)
    message2 = numpy.random.randint(2,size=message_size)

    # run transmitters
    fc1 = 125e3  # carrier frequency
    xmit1 = am_transmit(message1,samples_per_bit,fc1)

    fc2 = 175e3  # carrier frequency
    xmit2 = am_transmit(message2,samples_per_bit,fc2)

    # combine results on shared channel, plot spectrum
    rf = xmit1 + xmit2
    rf.spectrum(title='spectrum of shared channel',
                npoints=256000)

    # run receiver for channel 1
    receive1 = am_receive(rf,fc1,samples_per_bit)

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

    # display the plots
    w.show()