diff --git a/sigproc.py b/sigproc.py index 4194d7c..97c8a3e 100644 --- a/sigproc.py +++ b/sigproc.py @@ -47,10 +47,9 @@ class QAM(object): yield s def decode(self, symbols): - keys = np.array(self._dec.keys()) for s in symbols: - index = np.argmin(np.abs(s - keys)) - yield self._dec[ keys[index] ] + index = np.argmin(np.abs(s - self.points)) + yield self._dec[ self.points[index] ] modulator = QAM(bits_per_symbol=2, radii=[1.0])