From ee80201f80264f6f2093d57da4a0b7ea5aff30c2 Mon Sep 17 00:00:00 2001 From: Roman Zeyde Date: Wed, 16 Jul 2014 21:11:32 +0300 Subject: [PATCH] add error handler to sigproc.modulator.decode() --- sigproc.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/sigproc.py b/sigproc.py index 1d2c499..bce0d44 100644 --- a/sigproc.py +++ b/sigproc.py @@ -53,10 +53,12 @@ class QAM(object): s = self._enc[bits_tuple] yield s - def decode(self, symbols): + def decode(self, symbols, error_handler=None): for s in symbols: index = np.argmin(np.abs(s - self.symbols)) S = self.symbols[index] + if error_handler: + error_handler(received=s, decoded=S) yield self._dec[S] modulator = QAM(common.symbols)