git.gag.com Git - debian/gnuradio/blob - gnuradio-core/src/lib/general/gr_wvps_ff.cc

   1 /* -*- c++ -*- */
   2 /*
   3  * Copyright 2004 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 #ifdef HAVE_CONFIG_H
  24 #include "config.h"
  25 #endif
  26
  27 #include <gr_wvps_ff.h>
  28 #include <gr_io_signature.h>
  29 #include <string.h>
  30
  31 static int
  32 ceil_log2(int k)
  33 {
  34   int m = 0;
  35   for (int n = k-1; n > 0; n >>= 1) m++;
  36   return m;
  37 }
  38
  39 gr_wvps_ff_sptr
  40 gr_make_wvps_ff(int ilen)
  41 {
  42   return gr_wvps_ff_sptr(new gr_wvps_ff(ilen));
  43 }
  44
  45 gr_wvps_ff::gr_wvps_ff(int ilen)
  46   : gr_sync_block("wvps_ff",
  47                   gr_make_io_signature(1, 1, sizeof(float) * ilen),
  48                   gr_make_io_signature(1, 1, sizeof(float) * ceil_log2(ilen))),
  49     d_ilen(ilen), d_olen(ceil_log2(ilen))
  50 {
  51 }
  52
  53 // input vector assumed to be output from gsl wavelet computation
  54
  55 int
  56 gr_wvps_ff::work(int noutput_items,
  57                  gr_vector_const_void_star &input_items,
  58                  gr_vector_void_star &output_items)
  59 {
  60   const float *in  = (const float *) input_items[0];
  61   float       *out = (float *) output_items[0];
  62
  63   for (int count = 0; count < noutput_items; count++) {
  64
  65     // any power?
  66
  67     if (in[0] == 0.0) {
  68       for (int i = 0; i < d_olen; i++)
  69         out[i] = 0.0;
  70
  71     } else {
  72
  73       // get power normalization from 0-th wavelet coefficient
  74
  75       float scl = 1.0/(in[0]*in[0]);
  76       int k = 1;
  77
  78       // sum powers over sequences of bins,
  79       // sequence lengths in increasing powers of 2
  80
  81       for (int e = 0; e < d_olen; e++) {
  82         int m = 01<<e;
  83         float sum = 0.0;
  84
  85         for (int l = 0; l < m; l++)
  86           sum += (in[k+l]*in[k+l]);
  87
  88         out[e] = scl*sum;
  89         k += m;
  90       }
  91     }
  92
  93     in  += d_ilen;
  94     out += d_olen;
  95   }
  96
  97   return noutput_items;
  98 }