Imported Upstream version 3.0
[debian/gnuradio] / gnuradio-core / src / lib / general / gr_simple_squelch_cc.cc
1 /* -*- c++ -*- */
2 /*
3  * Copyright 2005 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., 51 Franklin Street,
20  * Boston, MA 02110-1301, USA.
21  */
22
23 #ifdef HAVE_CONFIG_H
24 #include "config.h"
25 #endif
26 #include <gr_simple_squelch_cc.h>
27 #include <gr_io_signature.h>
28 #include <cmath>
29
30 gr_simple_squelch_cc_sptr
31 gr_make_simple_squelch_cc(double threshold_db, double alpha)
32 {
33   return gr_simple_squelch_cc_sptr(new gr_simple_squelch_cc(threshold_db, alpha));
34 }
35
36 gr_simple_squelch_cc::gr_simple_squelch_cc (double threshold_db, double alpha)
37   : gr_sync_block ("simple_squelch_cc",
38                    gr_make_io_signature(1, 1, sizeof(gr_complex)),
39                    gr_make_io_signature(1, 1, sizeof(gr_complex))),
40     d_iir(alpha), d_unmuted(false)
41 {
42   set_threshold (threshold_db);
43 }
44
45 gr_simple_squelch_cc::~gr_simple_squelch_cc()
46 {
47 }
48
49
50 int
51 gr_simple_squelch_cc::work(int noutput_items,
52                            gr_vector_const_void_star &input_items,
53                            gr_vector_void_star &output_items)
54 {
55   const gr_complex *in = (const gr_complex *) input_items[0];
56   gr_complex *out = (gr_complex *) output_items[0];
57
58   for (int i = 0; i < noutput_items; i++){
59     double mag_sqrd = in[i].real()*in[i].real() + in[i].imag()*in[i].imag();
60     double f = d_iir.filter(mag_sqrd);
61     if (f >= d_threshold)
62       out[i] = in[i];
63     else
64       out[i] = 0;
65   }
66
67   d_unmuted = d_iir.prev_output() >= d_threshold;
68   return noutput_items;
69 }
70
71 void
72 gr_simple_squelch_cc::set_threshold(double decibels)
73 {
74   // convert to absolute threshold (mag squared)
75   d_threshold = std::pow(10.0, decibels/10);
76 }
77
78 double
79 gr_simple_squelch_cc::threshold() const
80 {
81   return 10 * log10(d_threshold);
82 }
83
84 void
85 gr_simple_squelch_cc::set_alpha(double alpha)
86 {
87   d_iir.set_taps(alpha);
88 }
89
90 std::vector<float>
91 gr_simple_squelch_cc::squelch_range() const
92 {
93   std::vector<float> r(3);
94   r[0] = -50.0;                         // min  FIXME
95   r[1] = +50.0;                         // max  FIXME
96   r[2] = (r[1] - r[0]) / 100;           // step size
97
98   return r;
99 }