← Back to branch summary

~ubuntu-branches/ubuntu/precise/gnuradio/precise

« back to all changes in this revision

Viewing changes to gnuradio-core/src/python/gnuradio/gr/qa_ofdm_insert_preamble.py

  • Committer: Bazaar Package Importer
  • Author(s): Kamal Mostafa
  • Date: 2010-03-13 07:46:01 UTC
  • mfrom: (2.1.2 sid)
  • Revision ID: james.westby@ubuntu.com-20100313074601-zjsa893a87bozyh7
Tags: 3.2.2.dfsg-1ubuntu1
https://launchpad.net/bugs/260406
* Fix build for Ubuntu lucid (LP: #260406)
  - add binary package dep for libusrp0, libusrp2-0: adduser
  - debian/rules clean: remove pre-built Qt moc files

Show diffs side-by-side

added added

removed removed

Lines of Context:
gnuradio-core/src/python/gnuradio/gr/qa_ofdm_insert_preamble.py
 
1
#!/usr/bin/env python
 
2
#
 
3
# Copyright 2007 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 3, 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
from pprint import pprint
 
25
 
 
26
class testing (gr_unittest.TestCase):
 
27
 
 
28
    def setUp (self):
 
29
        self.tb = gr.top_block ()
 
30
 
 
31
    def tearDown (self):
 
32
        self.tb = None
 
33
 
 
34
    def helper(self, v0, v1, fft_length, preamble):
 
35
        tb = self.tb
 
36
        src0 = gr.vector_source_c(v0)
 
37
        src1 = gr.vector_source_b(v1)
 
38
        
 
39
        s2v = gr.stream_to_vector(gr.sizeof_gr_complex, fft_length)
 
40
 
 
41
        # print "len(v) = %d" % (len(v))
 
42
 
 
43
        op = gr.ofdm_insert_preamble(fft_length, preamble)
 
44
 
 
45
        v2s = gr.vector_to_stream(gr.sizeof_gr_complex, fft_length)
 
46
        dst0 = gr.vector_sink_c()
 
47
        dst1 = gr.vector_sink_b()
 
48
 
 
49
        tb.connect(src0, s2v, (op, 0))
 
50
        tb.connect(src1, (op, 1))
 
51
        tb.connect((op, 0), v2s, dst0)
 
52
        tb.connect((op, 1), dst1)
 
53
 
 
54
        tb.run()
 
55
        r0 = dst0.data()
 
56
        r0v = []
 
57
        for i in range(len(r0)//fft_length):
 
58
            r0v.append(r0[i*fft_length:(i+1)*fft_length])
 
59
            
 
60
        r1 = dst1.data()
 
61
        self.assertEqual(len(r0v), len(r1))
 
62
        return (r1, r0v)
 
63
        
 
64
    def check_match(self, actual, expected_list):
 
65
        lst = []
 
66
        map(lambda x: lst.append(x), expected_list)
 
67
        self.assertEqual(actual, lst)
 
68
 
 
69
 
 
70
    # ----------------------------------------------------------------
 
71
 
 
72
    def test_000(self):
 
73
        # no preamble, 1 symbol payloads
 
74
 
 
75
        preamble = ()
 
76
        fft_length = 8
 
77
        npayloads = 8
 
78
        v = []
 
79
        p = []
 
80
        for i in range(npayloads):
 
81
            t = fft_length*[(i + i*1j)]
 
82
            p.append(tuple(t))
 
83
            v += t
 
84
 
 
85
        p = tuple(p)
 
86
            
 
87
        r = self.helper(v, npayloads*[1], fft_length, preamble)
 
88
        # pprint(r)
 
89
 
 
90
        self.assertEqual(r[0], tuple(npayloads*[1]))
 
91
        self.check_match(r[1], (p[0],p[1],p[2],p[3],p[4],p[5],p[6],p[7]))
 
92
            
 
93
 
 
94
    def test_001(self):
 
95
        # 1 symbol preamble, 1 symbol payloads
 
96
        preamble = ((100, 101, 102, 103, 104, 105, 106, 107),)
 
97
        p0 = preamble[0]
 
98
        fft_length = 8
 
99
        npayloads = 8
 
100
        v = []
 
101
        p = []
 
102
        for i in range(npayloads):
 
103
            t = fft_length*[(i + i*1j)]
 
104
            p.append(tuple(t))
 
105
            v += t
 
106
 
 
107
        
 
108
        r = self.helper(v, npayloads*[1], fft_length, preamble)
 
109
 
 
110
        self.assertEqual(r[0], tuple(npayloads*[1, 0]))
 
111
        self.check_match(r[1], (p0, p[0],
 
112
                                p0, p[1],
 
113
                                p0, p[2],
 
114
                                p0, p[3],
 
115
                                p0, p[4],
 
116
                                p0, p[5],
 
117
                                p0, p[6],
 
118
                                p0, p[7]))
 
119
 
 
120
    def test_002(self):
 
121
        # 2 symbol preamble, 1 symbol payloads
 
122
        preamble = ((100, 101, 102, 103, 104, 105, 106, 107),
 
123
                    (200, 201, 202, 203, 204, 205, 206, 207))
 
124
        p0 = preamble[0]
 
125
        p1 = preamble[1]
 
126
 
 
127
        fft_length = 8
 
128
        npayloads = 8
 
129
        v = []
 
130
        p = []
 
131
        for i in range(npayloads):
 
132
            t = fft_length*[(i + i*1j)]
 
133
            p.append(tuple(t))
 
134
            v += t
 
135
        
 
136
        r = self.helper(v, npayloads*[1], fft_length, preamble)
 
137
 
 
138
        self.assertEqual(r[0], tuple(npayloads*[1, 0, 0]))
 
139
        self.check_match(r[1], (p0, p1, p[0],
 
140
                                p0, p1, p[1],
 
141
                                p0, p1, p[2],
 
142
                                p0, p1, p[3],
 
143
                                p0, p1, p[4],
 
144
                                p0, p1, p[5],
 
145
                                p0, p1, p[6],
 
146
                                p0, p1, p[7]))
 
147
 
 
148
 
 
149
    def xtest_003_preamble(self):
 
150
        # 2 symbol preamble, 2 symbol payloads
 
151
        preamble = ((100, 101, 102, 103, 104, 105, 106, 107),
 
152
                    (200, 201, 202, 203, 204, 205, 206, 207))
 
153
        p0 = preamble[0]
 
154
        p1 = preamble[1]
 
155
 
 
156
        fft_length = 8
 
157
        npayloads = 8
 
158
        v = []
 
159
        p = []
 
160
        for i in range(npayloads * 2):
 
161
            t = fft_length*[(i + i*1j)]
 
162
            p.append(tuple(t))
 
163
            v += t
 
164
        
 
165
        r = self.helper(v, npayloads*[1, 0], fft_length, preamble)
 
166
 
 
167
        self.assertEqual(r[0], tuple(npayloads*[1, 0, 0, 0]))
 
168
        self.check_match(r[1], (p0, p1, p[0],  p[1],
 
169
                                p0, p1, p[2],  p[3],
 
170
                                p0, p1, p[4],  p[5],
 
171
                                p0, p1, p[6],  p[7],
 
172
                                p0, p1, p[8],  p[9],
 
173
                                p0, p1, p[10], p[11],
 
174
                                p0, p1, p[12], p[13],
 
175
                                p0, p1, p[14], p[15]))
 
176
 
 
177
 
 
178
if __name__ == '__main__':
 
179
    gr_unittest.main ()