import matplotlib.pyplot as plt
from sk_dsp_comm.fec_conv import FECConv
from sk_dsp_comm import digitalcom as dc
import numpy as np
cc = FECConv()
x = np.random.randint(0,2,100)
state = '00'
y,state = cc.conv_encoder(x,state)
yn = dc.cpx_awgn(2*y-1,5,1) # SNR = 5 dB
yn = (yn.real+1)/2*7
z = cc.viterbi_decoder(yn)
cc.traceback_plot()
plt.show()