Updated FSF address in all files. Fixes ticket:51
[debian/gnuradio] / gr-error-correcting-codes / examples / qa_test_encoder_convolutional_1.py
1 #!/usr/bin/env python
2 #
3 # Copyright 2004,2005 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, audio
24 from gnuradio.eng_option import eng_option
25 from optparse import OptionParser
26 # must be imported from local directory so that make check
27 # can run before installation
28 import ecc
29
30 class my_graph(gr.flow_graph):
31
32     def __init__(self):
33         gr.flow_graph.__init__(self)
34
35         parser = OptionParser(option_class=eng_option)
36         parser.add_option("-O", "--audio-output", type="string", default="",
37                           help="pcm output device name.  E.g., hw:0,0 or /dev/dsp")
38         parser.add_option("-r", "--sample-rate", type="eng_float", default=48000,
39                           help="set sample rate to RATE (48000)")
40         (options, args) = parser.parse_args ()
41         if len(args) != 0:
42             parser.print_help()
43             raise SystemExit, 1
44
45         sample_rate = int(options.sample_rate)
46         audio_option = options.audio_output
47         src = audio.source (sample_rate, audio_option)
48         src_out_chan = src.output_signature().max_streams()
49         dst = audio.sink (sample_rate, str(src_out_chan))
50         dst_in_chan = dst.input_signature().max_streams()
51         audio_el_size = src.output_signature().sizeof_stream_item(1)
52         frame_size = 10
53         enc_code_in_chan = src_out_chan
54         code_generators = [05, 06] #, 03, 04] # , 0, 07]
55         enc_code_out_chan = len (code_generators) / enc_code_in_chan
56         do_termination = 1
57
58         ss_enc = ecc.streams_encode_convolutional (frame_size,
59                                                    enc_code_in_chan,
60                                                    enc_code_out_chan,
61                                                    code_generators,
62                                                    do_termination)
63         for i in range (enc_code_in_chan):
64             self.connect ((src, i), (ss_enc, i))
65
66         for i in range (enc_code_out_chan):
67             ns = gr.null_sink (1)
68             self.connect ((ss_enc, i), ns)
69
70 if __name__ == '__main__':
71     try:
72         my_graph().run()
73     except KeyboardInterrupt:
74         pass