Imported Upstream version 3.2.2
[debian/gnuradio] / gnuradio-core / src / lib / general / gr_stretch_ff.cc
1 /* -*- c++ -*- */
2 /*
3  * Copyright 2008 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_stretch_ff.h>
28 #include <gr_io_signature.h>
29
30 gr_stretch_ff_sptr
31 gr_make_stretch_ff(float lo, size_t vlen)
32 {
33   return gr_stretch_ff_sptr(new gr_stretch_ff(lo, vlen));
34 }
35
36 gr_stretch_ff::gr_stretch_ff(float lo, size_t vlen)
37   : gr_sync_block("stretch_ff",
38                   gr_make_io_signature(1, 1, vlen * sizeof(float)),
39                   gr_make_io_signature(1, 1, vlen * sizeof(float))),
40     d_lo(lo), d_vlen(vlen)
41 {
42 }
43
44 int
45 gr_stretch_ff::work(int noutput_items,
46                     gr_vector_const_void_star &input_items,
47                     gr_vector_void_star &output_items)
48 {
49   const float *in  = (const float *) input_items[0];
50   float       *out = (float *) output_items[0];
51   
52   for (int count = 0; count < noutput_items; count++) {
53     float vmax = in[0] - d_lo;
54
55     for (unsigned int i = 1; i < d_vlen; i++) {
56       float vtmp = in[i] - d_lo;
57       if (vtmp > vmax)
58         vmax = vtmp;
59     }
60     
61     if (vmax != 0.0)
62       for (unsigned int i = 0; i < d_vlen; i++)
63         out[i] = d_lo * (1.0 - (in[i] - d_lo) / vmax);
64     else
65       for (unsigned int i = 0; i < d_vlen; i++)
66         out[i] = in[i];
67
68     in  += d_vlen;
69     out += d_vlen;
70   }
71   
72   return noutput_items;
73 }
74