close all;
clear;
clc;

for i=1:3
L = 38400;
chip_data = chip_gen(L);

M = input('Input modem type (4 or 16 or 64) ? ');
num_Tx = input('Input the number of transmit antennas ? ');

[D, L, M, bit_n] = dec_gen(chip_data, M);
dd = scatterplot(0);

snr = [0:2:30];
for idx=1:length(snr)
    SNR = snr(idx);
    
[Tx_1, Tx_2, Tx, scale] = mimo_en(D,M, num_Tx, L, bit_n);

[Rx_1, Rx_2, Rx, h_1, h_2] = channel(Tx_1, Tx_2, L, bit_n, SNR);



[z, R, r_1, r_2] = mimo_de(Rx_1, Rx_2, M, L, bit_n, h_1, h_2, scale);

dd = scatterplot(R, 1, 0, 'b.', dd);
drawnow;

[num_t, rt_t, num_r, rt_r]= result(D, z);

    num_SER(idx,i) = num_t;
    num_BER(idx,i) = num_r;
    rt_SER(idx,i) = rt_t;
    rt_BER(idx,i) = rt_r;
    
end
end

semilogy(snr,rt_SER(:,1),'-ro')
grid on;
hold on;
semilogy(snr,rt_BER(:,1),'-b*')
grid on;
hold on;

semilogy(snr,rt_SER(:,2),'--g+')
grid on;
hold on;
semilogy(snr,rt_BER(:,2),'--cx')
grid on;
hold on;

semilogy(snr,rt_SER(:,3),'-.md')
grid on;
hold on;
semilogy(snr,rt_BER(:,3),'-.ks')
grid on;
hold on;
title('QPSK/16QAM/64QAM 2_1');
xlabel('SNR(dB)');
ylabel('SER & BER');
legend('QPSK_SER','QPSK_BER', '16QAM_SER', '16QAM_BER', '64QAM_SER', '64QAM_BER');