summaryrefslogtreecommitdiff
path: root/gr-digital/python/psk2.py
diff options
context:
space:
mode:
Diffstat (limited to 'gr-digital/python/psk2.py')
-rw-r--r--gr-digital/python/psk2.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/gr-digital/python/psk2.py b/gr-digital/python/psk2.py
index 089af11dd0..6ab398e42c 100644
--- a/gr-digital/python/psk2.py
+++ b/gr-digital/python/psk2.py
@@ -27,8 +27,10 @@ from math import pi, log
from cmath import exp
from gnuradio import gr, modulation_utils2
-from gnuradio.digital.generic_mod_demod import generic_mod, generic_demod
-from gnuradio.digital.utils import mod_codes, gray_code
+import digital_swig
+#from gnuradio.digital.generic_mod_demod import generic_mod, generic_demod
+from utils import mod_codes, gray_code
+from generic_mod_demod import generic_mod, generic_demod
# Default number of points in constellation.
_def_constellation_points = 4
@@ -65,7 +67,7 @@ def psk_constellation(m=_def_constellation_points, mod_code=_def_mod_code):
if post_diff_code is not None:
inverse_post_diff_code = mod_codes.invert_code(post_diff_code)
points = [points[x] for x in inverse_post_diff_code]
- constellation = gr.constellation_psk(points, pre_diff_code, m)
+ constellation = digital_swig.constellation_psk(points, pre_diff_code, m)
return constellation
# /////////////////////////////////////////////////////////////////////////////