~ubuntu-branches/ubuntu/trusty/gnuradio/trusty

« back to all changes in this revision

Viewing changes to gnuradio-core/src/lib/general/gr_probe_mpsk_snr_c.cc

  • Committer: Bazaar Package Importer
  • Author(s): Kamal Mostafa
  • Date: 2010-03-13 07:46:01 UTC
  • mfrom: (2.1.2 sid)
  • Revision ID: james.westby@ubuntu.com-20100313074601-zjsa893a87bozyh7
Tags: 3.2.2.dfsg-1ubuntu1
* Fix build for Ubuntu lucid (LP: #260406)
  - add binary package dep for libusrp0, libusrp2-0: adduser
  - debian/rules clean: remove pre-built Qt moc files

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
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_probe_mpsk_snr_c.h>
 
28
#include <gr_io_signature.h>
 
29
 
 
30
gr_probe_mpsk_snr_c_sptr
 
31
gr_make_probe_mpsk_snr_c(double alpha)
 
32
{
 
33
  return gr_probe_mpsk_snr_c_sptr(new gr_probe_mpsk_snr_c(alpha));
 
34
}
 
35
 
 
36
gr_probe_mpsk_snr_c::gr_probe_mpsk_snr_c(double alpha)
 
37
  : gr_sync_block ("probe_mpsk_snr_c",
 
38
                   gr_make_io_signature(1, 1, sizeof(gr_complex)),
 
39
                   gr_make_io_signature(0, 0, 0)),
 
40
    d_signal_mean(0.0),
 
41
    d_noise_variance(0.0)
 
42
{
 
43
  set_alpha(alpha);
 
44
}
 
45
 
 
46
gr_probe_mpsk_snr_c::~gr_probe_mpsk_snr_c()
 
47
{
 
48
}
 
49
 
 
50
int
 
51
gr_probe_mpsk_snr_c::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
 
 
57
  for (int i = 0; i < noutput_items; i++){
 
58
    // Update of signal mean estimate
 
59
    double mag = abs(in[i]);
 
60
    d_signal_mean = d_alpha*abs(in[i]) + d_beta*d_signal_mean;
 
61
 
 
62
    // Update noise variance estimate
 
63
    double noise = mag-d_signal_mean;
 
64
    double var = noise*noise;
 
65
    d_noise_variance = d_alpha*var + d_beta*d_noise_variance;
 
66
  }
 
67
 
 
68
  return noutput_items;
 
69
}
 
70
 
 
71
double
 
72
gr_probe_mpsk_snr_c::snr() const
 
73
{
 
74
  if (d_noise_variance == 0.0)
 
75
    return 0.0;
 
76
  else
 
77
    return 10*log10(d_signal_mean*d_signal_mean/d_noise_variance);
 
78
}
 
79
 
 
80
void
 
81
gr_probe_mpsk_snr_c::set_alpha(double alpha)
 
82
{
 
83
  d_alpha = alpha;
 
84
  d_beta = 1.0-alpha;
 
85
}