git.gag.com Git - debian/gnuradio/blob - gnuradio-core/src/python/gnuradio/gr/qa_correlate_access_code.py

   1 #!/usr/bin/env python
   2 #
   3 # Copyright 2006 Free Software Foundation, Inc.
   4 #
   5 # This file is part of GNU Radio
   6 #
   7 # GNU Radio is free software; you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation; either version 2, or (at your option)
  10 # any later version.
  11 #
  12 # GNU Radio is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU General Public License
  18 # along with GNU Radio; see the file COPYING.  If not, write to
  19 # the Free Software Foundation, Inc., 51 Franklin Street,
  20 # Boston, MA 02110-1301, USA.
  21 #
  22
  23 from gnuradio import gr, gr_unittest
  24 import math
  25
  26 default_access_code = '\xAC\xDD\xA4\xE2\xF2\x8C\x20\xFC'
  27
  28 def string_to_1_0_list(s):
  29     r = []
  30     for ch in s:
  31         x = ord(ch)
  32         for i in range(8):
  33             t = (x >> i) & 0x1
  34             r.append(t)
  35
  36     return r
  37
  38 def to_1_0_string(L):
  39     return ''.join(map(lambda x: chr(x + ord('0')), L))
  40
  41 class test_correlate_access_code(gr_unittest.TestCase):
  42
  43     def setUp(self):
  44         self.fg = gr.flow_graph()
  45
  46     def tearDown(self):
  47         self.fg = None
  48
  49     def test_001(self):
  50         pad = (0,) * 64
  51         #           0  0  0  1  0  0  0  1
  52         src_data = (1, 0, 1, 1, 1, 1, 0, 1, 1) + pad + (0,) * 7
  53         expected_result = pad + (1, 0, 1, 1, 3, 1, 0, 1, 1, 2) + (0,) * 6
  54         src = gr.vector_source_b (src_data)
  55         op = gr.correlate_access_code_bb("1011", 0)
  56         dst = gr.vector_sink_b ()
  57         self.fg.connect (src, op, dst)
  58         self.fg.run ()
  59         result_data = dst.data ()
  60         self.assertEqual (expected_result, result_data)
  61
  62
  63     def test_002(self):
  64         code = tuple(string_to_1_0_list(default_access_code))
  65         access_code = to_1_0_string(code)
  66         pad = (0,) * 64
  67         #print code
  68         #print access_code
  69         src_data = code + (1, 0, 1, 1) + pad
  70         expected_result = pad + code + (3, 0, 1, 1)
  71         src = gr.vector_source_b (src_data)
  72         op = gr.correlate_access_code_bb(access_code, 0)
  73         dst = gr.vector_sink_b ()
  74         self.fg.connect (src, op, dst)
  75         self.fg.run ()
  76         result_data = dst.data ()
  77         self.assertEqual (expected_result, result_data)
  78
  79
  80
  81 if __name__ == '__main__':
  82     gr_unittest.main ()
  83