From 290c539726692ca76c7beee93f18318353bb9732 Mon Sep 17 00:00:00 2001 From: Alan Bidart Date: Wed, 3 May 2023 16:43:21 -0400 Subject: [PATCH] update p_outbits so that length of non_sys_stream is a function of puncture matrix --- commpy/channelcoding/convcode.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/commpy/channelcoding/convcode.py b/commpy/channelcoding/convcode.py index d1dcf16..651913f 100644 --- a/commpy/channelcoding/convcode.py +++ b/commpy/channelcoding/convcode.py @@ -521,7 +521,8 @@ def conv_encode(message_bits, trellis, termination = 'term', puncture_matrix=Non outbits = np.zeros(number_outbits, 'int') if puncture_matrix is not None: - p_outbits = np.zeros(number_outbits, 'int') + number_punctured_bits = int(number_outbits * puncture_matrix.sum() / puncture_matrix.size) + p_outbits = np.zeros(number_punctured_bits, 'int') else: p_outbits = np.zeros(int(number_outbits* puncture_matrix[0:].sum()/np.size(puncture_matrix, 1)), 'int')