diff options
Diffstat (limited to 'gr-wavelet/lib/wvps_ff_impl.cc')
-rw-r--r-- | gr-wavelet/lib/wvps_ff_impl.cc | 106 |
1 files changed, 53 insertions, 53 deletions
diff --git a/gr-wavelet/lib/wvps_ff_impl.cc b/gr-wavelet/lib/wvps_ff_impl.cc index 9512ce45f1..d428a8a8a1 100644 --- a/gr-wavelet/lib/wvps_ff_impl.cc +++ b/gr-wavelet/lib/wvps_ff_impl.cc @@ -29,75 +29,75 @@ #include <string.h> namespace gr { - namespace wavelet { - - static int - ceil_log2(int k) - { - int m = 0; - for (int n = k-1; n > 0; n >>= 1) m++; - return m; - } +namespace wavelet { - wvps_ff::sptr wvps_ff::make(int ilen) - { - return gnuradio::get_initial_sptr(new wvps_ff_impl(ilen)); - } +static int ceil_log2(int k) +{ + int m = 0; + for (int n = k - 1; n > 0; n >>= 1) + m++; + return m; +} - wvps_ff_impl::wvps_ff_impl(int ilen) - : sync_block("wvps_ff", - io_signature::make(1, 1, sizeof(float) * ilen), - io_signature::make(1, 1, sizeof(float) * ceil_log2(ilen))), - d_ilen(ilen), d_olen(ceil_log2(ilen)) - { - } +wvps_ff::sptr wvps_ff::make(int ilen) +{ + return gnuradio::get_initial_sptr(new wvps_ff_impl(ilen)); +} - // input vector assumed to be output from gsl wavelet computation +wvps_ff_impl::wvps_ff_impl(int ilen) + : sync_block("wvps_ff", + io_signature::make(1, 1, sizeof(float) * ilen), + io_signature::make(1, 1, sizeof(float) * ceil_log2(ilen))), + d_ilen(ilen), + d_olen(ceil_log2(ilen)) +{ +} - int - wvps_ff_impl::work(int noutput_items, - gr_vector_const_void_star &input_items, - gr_vector_void_star &output_items) - { - const float *in = (const float *) input_items[0]; - float *out = (float *) output_items[0]; +// input vector assumed to be output from gsl wavelet computation - for (int count = 0; count < noutput_items; count++) { +int wvps_ff_impl::work(int noutput_items, + gr_vector_const_void_star& input_items, + gr_vector_void_star& output_items) +{ + const float* in = (const float*)input_items[0]; + float* out = (float*)output_items[0]; - // any power? + for (int count = 0; count < noutput_items; count++) { - if (in[0] == 0.0) { - for (int i = 0; i < d_olen; i++) - out[i] = 0.0; + // any power? - } else { + if (in[0] == 0.0) { + for (int i = 0; i < d_olen; i++) + out[i] = 0.0; - // get power normalization from 0-th wavelet coefficient + } else { - float scl = 1.0/(in[0]*in[0]); - int k = 1; + // get power normalization from 0-th wavelet coefficient - // sum powers over sequences of bins, - // sequence lengths in increasing powers of 2 + float scl = 1.0 / (in[0] * in[0]); + int k = 1; - for (int e = 0; e < d_olen; e++) { - int m = 01<<e; - float sum = 0.0; + // sum powers over sequences of bins, + // sequence lengths in increasing powers of 2 - for (int l = 0; l < m; l++) - sum += (in[k+l]*in[k+l]); + for (int e = 0; e < d_olen; e++) { + int m = 01 << e; + float sum = 0.0; - out[e] = scl*sum; - k += m; - } - } + for (int l = 0; l < m; l++) + sum += (in[k + l] * in[k + l]); - in += d_ilen; - out += d_olen; - } + out[e] = scl * sum; + k += m; + } + } - return noutput_items; + in += d_ilen; + out += d_olen; } - } /* namespace wavelet */ + return noutput_items; +} + +} /* namespace wavelet */ } /* namespace gr */ |