import message
import mixer
import channel
import receiver
from plots import plot_before_after, plot_before_after_amqam, show

# message
orig1 = message.sin()
orig2 = message.square()
m1 = message.sin()
m2 = message.square()

# mixer
m = mixer.AMQAM(m1, m2)

# channel
m = channel.attenuation(m)
m = channel.fading(m)
m = channel.gauss_noise(m)

# receiver
m = receiver.unattenuate(m)
m1, m2 = receiver.demuxAMQAM(m)
m1 = receiver.lpf(m1)
m2 = receiver.lpf(m2)

plot_before_after_amqam(orig1, orig2, m1, m2)
show()
        original_message = msg()
        original_message_aud = msg(length=2)
        m = fn(original_message)
        m_aud = fn(original_message_aud)
        m_aud.write(filename='audio/'+message_name+'_'+function_name+'.wav')
        plot_both(m, filename='plots/'+message_name+'_'+function_name+'.png')

#am-qam
for ind1 in range(len(types)):
    msg1 = types[ind1]
    name1 = str(msg1).split()[1]
    orig1 = msg1()
    orig1aud = msg1(length=2)
    orig1aud.write(filename='audio/qam/'+name1+'_orig.wav')
    plot_both(orig1, filename='plots/qam/'+name1+'_orig.png')
    for msg2 in types[ind1:]:
        name2 = str(msg2).split()[1]
        orig2 = msg2()
        orig2aud = msg2(length=2)
        m = mixer.AMQAM(msg1(), msg2())
        maud = mixer.AMQAM(msg1(length=2), msg2(length=2))
        m1, m2 = receiver.demuxAMQAM(m)
        m1aud, m2aud = receiver.demuxAMQAM(maud)
        m1 = receiver.lpf(m1)
        m2 = receiver.lpf(m2)
        m1aud = receiver.lpf(m1aud)
        m2aud = receiver.lpf(m2aud)
        m1aud.write(filename='audio/qam/'+name1+'_'+name2+'_msg1.wav')
        m2aud.write(filename='audio/qam/'+name1+'_'+name2+'_msg2.wav')
        plot_before_after_amqam(orig1, orig2, m1, m2, filename='plots/qam/'+name1+'_'+name2+'.png')