mirror of https://github.com/f4exb/sdrangel.git
148 lines
2.6 KiB
C++
148 lines
2.6 KiB
C++
/*
|
|
* agc.cpp
|
|
*
|
|
* Created on: Sep 7, 2015
|
|
* Author: f4exb
|
|
*/
|
|
|
|
#include "dsp/agc.h"
|
|
|
|
|
|
AGC::AGC(int historySize, Real R) :
|
|
m_u0(1.0),
|
|
m_R(R),
|
|
m_moving_average(historySize, m_R),
|
|
m_historySize(historySize),
|
|
m_count(0)
|
|
{}
|
|
|
|
AGC::~AGC()
|
|
{}
|
|
|
|
void AGC::resize(int historySize, Real R)
|
|
{
|
|
m_R = R;
|
|
m_moving_average.resize(historySize, R);
|
|
m_historySize = historySize;
|
|
m_count = 0;
|
|
}
|
|
|
|
Real AGC::getValue()
|
|
{
|
|
return m_u0;
|
|
}
|
|
|
|
Real AGC::getAverage()
|
|
{
|
|
return m_moving_average.average();
|
|
}
|
|
|
|
//MagSquaredAGC::MagSquaredAGC() :
|
|
// AGC(),
|
|
// m_magsq(0.0)
|
|
//{}
|
|
|
|
MagSquaredAGC::MagSquaredAGC(int historySize, double R, double threshold) :
|
|
AGC(historySize, R),
|
|
m_magsq(0.0),
|
|
m_threshold(threshold)
|
|
{}
|
|
|
|
MagSquaredAGC::~MagSquaredAGC()
|
|
{}
|
|
|
|
void MagSquaredAGC::feed(Complex& ci)
|
|
{
|
|
m_magsq = ci.real()*ci.real() + ci.imag()*ci.imag();
|
|
m_moving_average.feed(m_magsq);
|
|
m_u0 = m_R / m_moving_average.average();
|
|
ci *= m_u0;
|
|
}
|
|
|
|
double MagSquaredAGC::feedAndGetValue(const Complex& ci)
|
|
{
|
|
m_magsq = ci.real()*ci.real() + ci.imag()*ci.imag();
|
|
m_moving_average.feed(m_magsq);
|
|
m_u0 = m_R / m_moving_average.average();
|
|
return m_magsq > m_threshold ? m_u0 : 1.0;
|
|
}
|
|
|
|
//MagAGC::MagAGC() :
|
|
// AGC(),
|
|
// m_magsq(0.0)
|
|
//{}
|
|
|
|
MagAGC::MagAGC(int historySize, double R, double threshold) :
|
|
AGC(historySize, R),
|
|
m_magsq(0.0),
|
|
m_threshold(threshold)
|
|
{}
|
|
|
|
MagAGC::~MagAGC()
|
|
{}
|
|
|
|
void MagAGC::feed(Complex& ci)
|
|
{
|
|
m_magsq = ci.real()*ci.real() + ci.imag()*ci.imag();
|
|
m_moving_average.feed(m_magsq);
|
|
m_u0 = m_R / sqrt(m_moving_average.average());
|
|
ci *= m_u0;
|
|
}
|
|
|
|
double MagAGC::feedAndGetValue(const Complex& ci)
|
|
{
|
|
m_magsq = ci.real()*ci.real() + ci.imag()*ci.imag();
|
|
m_moving_average.feed(m_magsq);
|
|
m_u0 = m_R / sqrt(m_moving_average.average());
|
|
return m_magsq > m_threshold ? m_u0 : 1.0;
|
|
}
|
|
|
|
//AlphaAGC::AlphaAGC() :
|
|
// AGC(),
|
|
// m_alpha(0.5),
|
|
// m_magsq(0.0),
|
|
// m_squelchOpen(true)
|
|
//{}
|
|
|
|
AlphaAGC::AlphaAGC(int historySize, Real R) :
|
|
AGC(historySize, R),
|
|
m_alpha(0.5),
|
|
m_magsq(0.0),
|
|
m_squelchOpen(true)
|
|
{}
|
|
|
|
|
|
AlphaAGC::AlphaAGC(int historySize, Real R, Real alpha) :
|
|
AGC(historySize, R),
|
|
m_alpha(alpha),
|
|
m_magsq(0.0),
|
|
m_squelchOpen(true)
|
|
{}
|
|
|
|
AlphaAGC::~AlphaAGC()
|
|
{}
|
|
|
|
void AlphaAGC::resize(int historySize, Real R, Real alpha)
|
|
{
|
|
m_R = R;
|
|
m_alpha = alpha;
|
|
m_squelchOpen = true;
|
|
m_moving_average.resize(historySize, R);
|
|
}
|
|
|
|
void AlphaAGC::feed(Complex& ci)
|
|
{
|
|
m_magsq = ci.real()*ci.real() + ci.imag()*ci.imag();
|
|
|
|
if (m_squelchOpen && (m_magsq))
|
|
{
|
|
m_moving_average.feed(m_moving_average.average() - m_alpha*(m_moving_average.average() - m_magsq));
|
|
}
|
|
else
|
|
{
|
|
//m_squelchOpen = true;
|
|
m_moving_average.feed(m_magsq);
|
|
}
|
|
ci *= m_u0;
|
|
}
|