Merge pull request #59 from chiranthsiddappa/trellis-plots

Trellis plots

Author: chiranthsiddappa

requirements.txt

@@ -1,3 +1,3 @@
 numpy>=1.20.0
-matplotlib
+matplotlib>=3.0.0
 scipy>=1.1.0

src/sk_dsp_comm/fec_conv.py

@@ -735,17 +735,17 @@ def trellis_plot(self,fsize=(6,4)):
         plt.axis([-0.01, 1.01, -(self.Nstates-1)-0.05, 0.05])
         for m in range(self.Nstates):
             if branches_from.input1[m] == 0:
-                plt.plot([0, 1],[-branches_from.states1[m], -m],'b')
-                plt.plot([0, 1],[-branches_from.states1[m], -m],'r.')
+                plt.plot([0, 1],[-branches_from.states1[m], [-m]],'b')
+                plt.plot([0, 1],[-branches_from.states1[m], [-m]],'r.')
             if branches_from.input2[m] == 0:
-                plt.plot([0, 1],[-branches_from.states2[m], -m],'b')
-                plt.plot([0, 1],[-branches_from.states2[m], -m],'r.')
+                plt.plot([0, 1],[-branches_from.states2[m], [-m]],'b')
+                plt.plot([0, 1],[-branches_from.states2[m], [-m]],'r.')
             if branches_from.input1[m] == 1:
-                plt.plot([0, 1],[-branches_from.states1[m], -m],'g')
-                plt.plot([0, 1],[-branches_from.states1[m], -m],'r.')
+                plt.plot([0, 1],[-branches_from.states1[m], [-m]],'g')
+                plt.plot([0, 1],[-branches_from.states1[m], [-m]],'r.')
             if branches_from.input2[m] == 1:
-                plt.plot([0, 1],[-branches_from.states2[m], -m],'g')
-                plt.plot([0, 1],[-branches_from.states2[m], -m],'r.')
+                plt.plot([0, 1],[-branches_from.states2[m], [-m]],'g')
+                plt.plot([0, 1],[-branches_from.states2[m], [-m]],'r.')
         #plt.grid()
         plt.xlabel('One Symbol Transition')
         plt.ylabel('-State Index')