Number of elements in conv_encode is declared using if-else condition to avoid extra zeros in the output of encoded bits without puncturing

Author: prsudheer

commpy/channelcoding/convcode.py

@@ -8,9 +8,8 @@
 import matplotlib.patches as mpatches
 import matplotlib.pyplot as plt
 import numpy as np
-from matplotlib.collections import PatchCollection
-
 from commpy.utilities import dec2bitarray, bitarray2dec, hamming_dist, euclid_dist
+from matplotlib.collections import PatchCollection
 
 __all__ = ['Trellis', 'conv_encode', 'viterbi_decode']
 
@@ -381,7 +380,10 @@ def conv_encode(message_bits, trellis, termination = 'term', puncture_matrix=Non
             number_outbits = int(number_inbits/rate)
 
     outbits = np.zeros(number_outbits, 'int')
-    p_outbits = np.zeros(int(number_outbits*
+    if puncture_matrix is not None:
+        p_outbits = np.zeros(number_outbits)
+    else:
+        p_outbits = np.zeros(int(number_outbits*
             puncture_matrix[0:].sum()/np.size(puncture_matrix, 1)), 'int')
     next_state_table = trellis.next_state_table
     output_table = trellis.output_table