git.gag.com Git - debian/gnuradio/blob - gnuradio-core/src/python/gnuradio/blks2impl/pfb_channelizer.py

   1 #!/usr/bin/env python
   2 #
   3 # Copyright 2009,2010 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, optfir
  24
  25 class pfb_channelizer_ccf(gr.hier_block2):
  26     '''
  27     Make a Polyphase Filter channelizer (complex in, complex out, floating-point taps)
  28
  29     This simplifies the interface by allowing a single input stream to connect to this block.
  30     It will then output a stream for each channel.
  31     '''
  32     def __init__(self, numchans, taps=None, oversample_rate=1, atten=100):
  33         gr.hier_block2.__init__(self, "pfb_channelizer_ccf",
  34                                 gr.io_signature(1, 1, gr.sizeof_gr_complex), # Input signature
  35                                 gr.io_signature(numchans, numchans, gr.sizeof_gr_complex)) # Output signature
  36
  37         self._numchans = numchans
  38         self._oversample_rate = oversample_rate
  39
  40         if taps is not None:
  41             self._taps = taps
  42         else:
  43             # Create a filter that covers the full bandwidth of the input signal
  44             bw = 0.4
  45             tb = 0.2
  46             ripple = 0.1
  47             made = False
  48             while not made:
  49                 try:
  50                     self._taps = optfir.low_pass(1, self._numchans, bw, bw+tb, ripple, atten)
  51                     made = True
  52                 except RuntimeError:
  53                     ripple += 0.01
  54                     made = False
  55                     print("Warning: set ripple to %.4f dB. If this is a problem, adjust the attenuation or create your own filter taps." % (ripple))
  56
  57                     # Build in an exit strategy; if we've come this far, it ain't working.
  58                     if(ripple >= 1.0):
  59                         raise RuntimeError("optfir could not generate an appropriate filter.")
  60
  61         self.s2ss = gr.stream_to_streams(gr.sizeof_gr_complex, self._numchans)
  62         self.pfb = gr.pfb_channelizer_ccf(self._numchans, self._taps,
  63                                           self._oversample_rate)
  64         self.v2s = gr.vector_to_streams(gr.sizeof_gr_complex, self._numchans)
  65
  66         self.connect(self, self.s2ss)
  67
  68         for i in xrange(self._numchans):
  69             self.connect((self.s2ss,i), (self.pfb,i))
  70
  71         # Get independent streams from the filterbank and send them out
  72         self.connect(self.pfb, self.v2s)
  73
  74         for i in xrange(self._numchans):
  75             self.connect((self.v2s,i), (self,i))
  76
  77
  78
  79