From 3450c2c5701c9168ff77aa1b500aa68f75fbadeb Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Mon, 19 Jan 2015 16:15:40 +0200
Subject: [PATCH] equalizer: use constant symbols' prefix (for analog
 debugging)

---
 amodem/equalizer.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/amodem/equalizer.py b/amodem/equalizer.py
index 9d4bb46..da25e33 100644
--- a/amodem/equalizer.py
+++ b/amodem/equalizer.py
@@ -18,10 +18,13 @@ class Equalizer(object):
         self.Nfreq = config.Nfreq
         self.Nsym = config.Nsym
 
-    def train_symbols(self, length, seed=0):
+    def train_symbols(self, length, seed=0, constant_prefix=16):
         r = random.Random(seed)
         choose = lambda: [r.choice(_constellation) for j in range(self.Nfreq)]
-        return np.array([choose() for _ in range(length)])
+        symbols = np.array([choose() for _ in range(length)])
+        # Constant symbols (for analog debugging)
+        symbols[:constant_prefix, :] = 1
+        return symbols
 
     def modulator(self, symbols):
         gain = 1.0 / len(self.carriers)