1
0
mirror of https://github.com/f4exb/sdrangel.git synced 2024-11-17 13:51:47 -05:00
sdrangel/sdrbase/dsp/agc.cpp

140 lines
2.1 KiB
C++

/*
* agc.cpp
*
* Created on: Sep 7, 2015
* Author: f4exb
*/
#include "dsp/agc.h"
AGC::AGC() :
m_u0(1.0),
m_R(1.0),
m_moving_average(),
m_historySize(0),
m_count(0)
{}
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, Real R) :
AGC(historySize, R),
m_magsq(0.0)
{}
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;
}
MagAGC::MagAGC() :
AGC(),
m_magsq(0.0)
{}
MagAGC::MagAGC(int historySize, Real R) :
AGC(historySize, R),
m_magsq(0.0)
{}
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;
}
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;
}