Convolutional-Encoding-and-Decoding/communication_link.py at main · MoEssamKhattab/Convolutional-Encoding-and-Decoding · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
from conv.encoder import encode
from BPSK.bpsk_transmitter import bpsk_transmitter
from channel_demod_channel_decode import channel_demod_channel_decode


s = '101101' * 200
K = 3
generator_polynomials = [[1, 1, 1], [1, 1, 0], [1, 0, 1]]
fc = 0.1
Ab = 1
Tb = 100
n = 1000
snr_start = -32
snr_end = -10
snr_step = 1

def communication_link(s,generator_polynomials,K,fc,Ab,Tb,n,snr_start,snr_end,snr_step):
    """
    Communication Link (Source Encoding -> Channel Encoding -> Modulation -> AWGN Channel -> Demodulation -> Channel Decoding)
    :param s: original bit sequence
    :param generator_polynomials: generator polynomials
    :param K: constraint length
    :param fc: carrier frequency
    :param Ab: amplitude of carrier
    :param Tb: bit duration
    :param n: number of samples in one bit duration
    :param snr_start: start point of SNR
    :param snr_end: end point of SNR
    :param snr_step: step of SNR
    :return: restored bit sequence channel encoded signal
    """
    channel_encoded_sequence = encode(s, generator_polynomials, 3)
    modulated_signal = bpsk_transmitter(channel_encoded_sequence,fc,Ab,Tb,n)
    modulated_signal_without_conv = bpsk_transmitter(s,fc,Ab,Tb,n)
    decoded_signal = channel_demod_channel_decode(s,modulated_signal_without_conv,modulated_signal,generator_polynomials,K,snr_start,snr_end,snr_step,fc,Tb,n)

    return decoded_signal


decoded_signal = communication_link(s,generator_polynomials,K,fc,Ab,Tb,n,snr_start,snr_end,snr_step)