Updated FSF address in all files. Fixes ticket:51
[debian/gnuradio] / gr-error-correcting-codes / examples / qa_test_coding_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 = 1000
53         do_mux_outputs = 0
54         enc_code_in_chan = src_out_chan
55         code_generators = [05, 06] #, 03, 04] # , 0, 07]
56         enc_code_out_chan = len (code_generators) / enc_code_in_chan
57         do_termination = 1
58
59         if do_mux_outputs == 1:
60             enc_dec_chan = 1
61         else:
62             enc_dec_chan = enc_code_out_chan
63
64         ss_enc = ecc.streams_encode_convolutional (frame_size,
65                                                    enc_code_in_chan,
66                                                    enc_code_out_chan,
67                                                    code_generators,
68                                                    do_termination)
69 # for now
70         ss2s = gr.streams_to_stream (1, enc_dec_chan);
71         ns = gr.null_sink (1);
72 # end for now
73
74 #       ss_dec = error-correcting-codes.frames_to_streams (audio_el_size, 1, 1000)
75
76         for i in range (src_out_chan):
77             self.connect ((src, i), (ss_enc, i))
78 # for now
79         for i in range (enc_dec_chan):
80             self.connect ((ss_enc, i), (ss2s, i))
81         self.connect (ss2s, ns)
82 # end for now
83
84 #       for i in range (enc_dec_chan):
85 #           self.connect ((ss_enc, i), (ss_dec, i))
86
87 #       for i in range (dst_in_chan):
88 #           self.connect ((ss_dec, i), (dst, i))
89
90 if __name__ == '__main__':
91     try:
92         my_graph().run()
93     except KeyboardInterrupt:
94         pass