add mapping function with hard mapping to Modem class

veeresht · Apr 5, 2016 · 81d037c · 81d037c
1 parent 4488c19
commit 81d037c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/commpy/modulation.py b/commpy/modulation.py
@@ -93,6 +93,14 @@ def demodulate(self, input_symbols, demod_type, noise_var = 0):
 
         return demod_bits
 
+    def mapping(self, input_symbols, map_type, noise_var = 0):
+        if map_type == 'hard':
+            index_list = map(lambda i: self.constellation[argmin(abs(input_symbols[i] - self.constellation))], \
+                             xrange(0, len(input_symbols)))
+        else:
+            pass
+        return index_list
+
 
 class PSKModem(Modem):
     """ Creates a Phase Shift Keying (PSK) Modem object. """