mirror of https://github.com/gnss-sdr/gnss-sdr
144 lines
6.0 KiB
C++
144 lines
6.0 KiB
C++
/*!
|
|
* \file notch_lite_cc.cc
|
|
* \brief Implements a multi state notch filter algorithm
|
|
* \author Antonio Ramos (antonio.ramosdet(at)gmail.com)
|
|
*
|
|
* -----------------------------------------------------------------------------
|
|
*
|
|
* Copyright (C) 2010-2019 (see AUTHORS file for a list of contributors)
|
|
*
|
|
* GNSS-SDR is a software defined Global Navigation
|
|
* Satellite Systems receiver
|
|
*
|
|
* This file is part of GNSS-SDR.
|
|
*
|
|
* SPDX-License-Identifier: GPL-3.0-or-later
|
|
*
|
|
* -----------------------------------------------------------------------------
|
|
*/
|
|
|
|
#include "notch_lite_cc.h"
|
|
#include "gnss_sdr_make_unique.h"
|
|
#include <boost/math/distributions/chi_squared.hpp>
|
|
#include <gnuradio/io_signature.h>
|
|
#include <volk/volk.h>
|
|
#include <algorithm>
|
|
#include <cmath>
|
|
#include <cstring>
|
|
|
|
|
|
notch_lite_sptr make_notch_filter_lite(float p_c_factor, float pfa, int32_t length, int32_t n_segments_est, int32_t n_segments_reset, int32_t n_segments_coeff)
|
|
{
|
|
return notch_lite_sptr(new NotchLite(p_c_factor, pfa, length, n_segments_est, n_segments_reset, n_segments_coeff));
|
|
}
|
|
|
|
|
|
NotchLite::NotchLite(float p_c_factor,
|
|
float pfa,
|
|
int32_t length,
|
|
int32_t n_segments_est,
|
|
int32_t n_segments_reset,
|
|
int32_t n_segments_coeff) : gr::block("NotchLite",
|
|
gr::io_signature::make(1, 1, sizeof(gr_complex)),
|
|
gr::io_signature::make(1, 1, sizeof(gr_complex)))
|
|
{
|
|
const int32_t alignment_multiple = volk_get_alignment() / sizeof(gr_complex);
|
|
set_alignment(std::max(1, alignment_multiple));
|
|
set_history(2);
|
|
p_c_factor_ = gr_complex(p_c_factor, 0.0);
|
|
n_segments_est_ = n_segments_est;
|
|
n_segments_reset_ = n_segments_reset;
|
|
n_segments_coeff_reset_ = n_segments_coeff;
|
|
n_segments_coeff_ = 0;
|
|
length_ = length;
|
|
pfa_ = pfa;
|
|
n_segments_ = 0;
|
|
n_deg_fred_ = 2 * length_;
|
|
noise_pow_est_ = 0.0;
|
|
filter_state_ = false;
|
|
z_0_ = gr_complex(0.0, 0.0);
|
|
last_out_ = gr_complex(0.0, 0.0);
|
|
boost::math::chi_squared_distribution<float> my_dist_(n_deg_fred_);
|
|
thres_ = boost::math::quantile(boost::math::complement(my_dist_, pfa_));
|
|
c_samples1_ = gr_complex(0.0, 0.0);
|
|
c_samples2_ = gr_complex(0.0, 0.0);
|
|
angle1_ = 0.0;
|
|
angle2_ = 0.0;
|
|
power_spect_ = volk_gnsssdr::vector<float>(length_);
|
|
#if GNURADIO_FFT_USES_TEMPLATES
|
|
d_fft_ = std::make_unique<gr::fft::fft_complex_fwd>(length_);
|
|
#else
|
|
d_fft_ = std::make_unique<gr::fft::fft_complex>(length_, true);
|
|
#endif
|
|
}
|
|
|
|
|
|
int NotchLite::general_work(int noutput_items, gr_vector_int &ninput_items __attribute__((unused)),
|
|
gr_vector_const_void_star &input_items, gr_vector_void_star &output_items)
|
|
{
|
|
int32_t index_out = 0;
|
|
float sig2dB = 0.0;
|
|
float sig2lin = 0.0;
|
|
lv_32fc_t dot_prod_;
|
|
const auto *in = reinterpret_cast<const gr_complex *>(input_items[0]);
|
|
auto *out = reinterpret_cast<gr_complex *>(output_items[0]);
|
|
in++;
|
|
while ((index_out + length_) < noutput_items)
|
|
{
|
|
if ((n_segments_ < n_segments_est_) && (filter_state_ == false))
|
|
{
|
|
memcpy(d_fft_->get_inbuf(), in, sizeof(gr_complex) * length_);
|
|
d_fft_->execute();
|
|
volk_32fc_s32f_power_spectrum_32f(power_spect_.data(), d_fft_->get_outbuf(), 1.0, length_);
|
|
volk_32f_s32f_calc_spectral_noise_floor_32f(&sig2dB, power_spect_.data(), 15.0, length_);
|
|
sig2lin = std::pow(10.0F, (sig2dB / 10.0F)) / static_cast<float>(n_deg_fred_);
|
|
noise_pow_est_ = (static_cast<float>(n_segments_) * noise_pow_est_ + sig2lin) / static_cast<float>(n_segments_ + 1);
|
|
memcpy(out, in, sizeof(gr_complex) * length_);
|
|
}
|
|
else
|
|
{
|
|
volk_32fc_x2_conjugate_dot_prod_32fc(&dot_prod_, in, in, length_);
|
|
if ((lv_creal(dot_prod_) / noise_pow_est_) > thres_)
|
|
{
|
|
if (filter_state_ == false)
|
|
{
|
|
filter_state_ = true;
|
|
last_out_ = gr_complex(0, 0);
|
|
n_segments_coeff_ = 0;
|
|
}
|
|
if (n_segments_coeff_ == 0)
|
|
{
|
|
volk_32fc_x2_multiply_conjugate_32fc(&c_samples1_, (in + 1), in, 1);
|
|
volk_32fc_s32f_atan2_32f(&angle1_, &c_samples1_, static_cast<float>(1.0), 1);
|
|
volk_32fc_x2_multiply_conjugate_32fc(&c_samples2_, (in + length_ - 1), (in + length_ - 2), 1);
|
|
volk_32fc_s32f_atan2_32f(&angle2_, &c_samples2_, static_cast<float>(1.0), 1);
|
|
float angle_ = (angle1_ + angle2_) / 2.0F;
|
|
z_0_ = std::exp(gr_complex(0, 1) * angle_);
|
|
}
|
|
for (int32_t aux = 0; aux < length_; aux++)
|
|
{
|
|
*(out + aux) = *(in + aux) - z_0_ * (*(in + aux - 1)) + p_c_factor_ * z_0_ * last_out_;
|
|
last_out_ = *(out + aux);
|
|
}
|
|
n_segments_coeff_++;
|
|
n_segments_coeff_ = n_segments_coeff_ % n_segments_coeff_reset_;
|
|
}
|
|
else
|
|
{
|
|
if (n_segments_ > n_segments_reset_)
|
|
{
|
|
n_segments_ = 0;
|
|
}
|
|
filter_state_ = false;
|
|
memcpy(out, in, sizeof(gr_complex) * length_);
|
|
}
|
|
}
|
|
index_out += length_;
|
|
n_segments_++;
|
|
in += length_;
|
|
out += length_;
|
|
}
|
|
consume_each(index_out);
|
|
return index_out;
|
|
}
|