git.gag.com Git - debian/gnuradio/blob - gr-error-correcting-codes/src/python/qa_test_coding_2.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., 59 Temple Place - Suite 330,
  20 # Boston, MA 02111-1307, 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         enc_chan_in = 2
  49         src_out_chan = src.output_signature().max_streams()
  50         dst = audio.sink (sample_rate, str(enc_chan_in))
  51         dst_in_chan = dst.input_signature().max_streams()
  52         audio_el_size = dst.input_signature().sizeof_stream_item(0)
  53         frame_size = 10
  54         enc_code_in_chan = enc_chan_in
  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         ss_enc = ecc.streams_encode_convolutional (frame_size,
  60                                                    enc_code_in_chan,
  61                                                    enc_code_out_chan,
  62                                                    code_generators,
  63                                                    do_termination)
  64
  65         for i in range (src_out_chan):
  66             p2up = gr.packed_to_unpacked_bb (1, 1)
  67             self.connect ((src, i), p2up, (ss_enc, i))
  68
  69         tau_bits = 0  # 0 -> decode all first
  70         ss_dec = error-correcting-codes.dec_blk_conv_soft_full (audio_el_size,
  71                                              frame_size,
  72                                              do_termination,
  73                                              do_mux_outputs,
  74                                              enc_code_in_chan,
  75                                              enc_code_out_chan,
  76                                              code_generators)
  77         up2bf0 = gr.chunks_to_symbols_bf ([1.0, -1.0])
  78         self.connect ((ss_enc, 0), p2up0, up2bf0, (ss_dec, 0))
  79         if enc_dec_chan > 1:
  80             p2up1 = gr.packed_to_unpacked_bb (1, 1)
  81             up2bf1 = gr.chunks_to_symbols_bf ([1.0, -1.0])
  82             self.connect ((ss_enc, 1), p2up1, up2bf1, (ss_dec, 1))
  83
  84         for i in range (dst_in_chan):
  85             self.connect ((ss_dec, i), (dst, i))
  86
  87 if __name__ == '__main__':
  88     try:
  89         my_graph().run()
  90     except KeyboardInterrupt:
  91         pass