mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-12-22 09:31:10 -05:00
160 lines
4.1 KiB
C++
160 lines
4.1 KiB
C++
/* iir.c
|
|
|
|
This file is part of a program that implements a Software-Defined Radio.
|
|
|
|
Copyright (C) 2014, 2022, 2023 Warren Pratt, NR0V
|
|
Copyright (C) 2024 Edouard Griffiths, F4EXB Adapted to SDRangel
|
|
|
|
This program is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License
|
|
as published by the Free Software Foundation; either version 2
|
|
of the License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
The author can be reached by email at
|
|
|
|
warren@wpratt.com
|
|
|
|
*/
|
|
|
|
#include "comm.hpp"
|
|
#include "bqbp.hpp"
|
|
|
|
namespace WDSP {
|
|
|
|
/********************************************************************************************************
|
|
* *
|
|
* Complex Bi-Quad Band-Pass *
|
|
* *
|
|
********************************************************************************************************/
|
|
|
|
void BQBP::calc()
|
|
{
|
|
double f0;
|
|
double w0;
|
|
double bw;
|
|
double q;
|
|
double sn;
|
|
double cs;
|
|
double c;
|
|
double den;
|
|
|
|
bw = f_high - f_low;
|
|
f0 = (f_high + f_low) / 2.0;
|
|
q = f0 / bw;
|
|
w0 = TWOPI * f0 / rate;
|
|
sn = sin(w0);
|
|
cs = cos(w0);
|
|
c = sn / (2.0 * q);
|
|
den = 1.0 + c;
|
|
a0 = +c / den;
|
|
a1 = 0.0;
|
|
a2 = -c / den;
|
|
b1 = 2.0 * cs / den;
|
|
b2 = (c - 1.0) / den;
|
|
flush();
|
|
}
|
|
|
|
BQBP::BQBP(
|
|
int _run,
|
|
int _size,
|
|
float* _in,
|
|
float* _out,
|
|
double _rate,
|
|
double _f_low,
|
|
double _f_high,
|
|
double _gain,
|
|
int _nstages
|
|
) :
|
|
run(_run),
|
|
size(_size),
|
|
in(_in),
|
|
out(_out),
|
|
rate(_rate),
|
|
f_low(_f_low),
|
|
f_high(_f_high),
|
|
gain(_gain),
|
|
nstages(_nstages)
|
|
{
|
|
x0.resize(nstages * 2); // (float*)malloc0(nstages * sizeof(complex));
|
|
x1.resize(nstages * 2); // (float*)malloc0(nstages * sizeof(complex));
|
|
x2.resize(nstages * 2); // (float*)malloc0(nstages * sizeof(complex));
|
|
y0.resize(nstages * 2); // (float*)malloc0(nstages * sizeof(complex));
|
|
y1.resize(nstages * 2); // (float*)malloc0(nstages * sizeof(complex));
|
|
y2.resize(nstages * 2); // (float*)malloc0(nstages * sizeof(complex));
|
|
calc();
|
|
}
|
|
|
|
void BQBP::flush()
|
|
{
|
|
for (int i = 0; i < nstages; i++)
|
|
{
|
|
x1[2 * i + 0] = x2[2 * i + 0] = y1[2 * i + 0] = y2[2 * i + 0] = 0.0;
|
|
x1[2 * i + 1] = x2[2 * i + 1] = y1[2 * i + 1] = y2[2 * i + 1] = 0.0;
|
|
}
|
|
}
|
|
|
|
void BQBP::execute()
|
|
{
|
|
if (run)
|
|
{
|
|
for (int i = 0; i < size; i++)
|
|
{
|
|
for (int j = 0; j < 2; j++)
|
|
{
|
|
x0[j] = gain * in[2 * i + j];
|
|
|
|
for (int n = 0; n < nstages; n++)
|
|
{
|
|
if (n > 0)
|
|
x0[2 * n + j] = y0[2 * (n - 1) + j];
|
|
|
|
y0[2 * n + j] = a0 * x0[2 * n + j]
|
|
+ a1 * x1[2 * n + j]
|
|
+ a2 * x2[2 * n + j]
|
|
+ b1 * y1[2 * n + j]
|
|
+ b2 * y2[2 * n + j];
|
|
y2[2 * n + j] = y1[2 * n + j];
|
|
y1[2 * n + j] = y0[2 * n + j];
|
|
x2[2 * n + j] = x1[2 * n + j];
|
|
x1[2 * n + j] = x0[2 * n + j];
|
|
}
|
|
|
|
out[2 * i + j] = (float) y0[2 * (nstages - 1) + j];
|
|
}
|
|
}
|
|
}
|
|
else if (out != in)
|
|
{
|
|
std::copy(in, in + size * 2, out);
|
|
}
|
|
}
|
|
|
|
void BQBP::setBuffers(float* _in, float* _out)
|
|
{
|
|
in = _in;
|
|
out = _out;
|
|
}
|
|
|
|
void BQBP::setSamplerate( int _rate)
|
|
{
|
|
rate = _rate;
|
|
calc();
|
|
}
|
|
|
|
void BQBP::setSize(int _size)
|
|
{
|
|
size = _size;
|
|
flush();
|
|
}
|
|
|
|
} // namespace WDSP
|