kopia lustrzana https://github.com/f4exb/sdrangel
				
				
				
			
		
			
				
	
	
		
			157 wiersze
		
	
	
		
			3.7 KiB
		
	
	
	
		
			C++
		
	
	
			
		
		
	
	
			157 wiersze
		
	
	
		
			3.7 KiB
		
	
	
	
		
			C++
		
	
	
/*  dbqbp.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 "dbqbp.hpp"
 | 
						|
 | 
						|
namespace WDSP {
 | 
						|
 | 
						|
/********************************************************************************************************
 | 
						|
*                                                                                                       *
 | 
						|
*                                     Double Bi-Quad Band-Pass                                          *
 | 
						|
*                                                                                                       *
 | 
						|
********************************************************************************************************/
 | 
						|
 | 
						|
void DBQBP::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();
 | 
						|
}
 | 
						|
 | 
						|
DBQBP::DBQBP(
 | 
						|
    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); // (float*)malloc0(nstages * sizeof(float));
 | 
						|
    x1.resize(nstages); // (float*)malloc0(nstages * sizeof(float));
 | 
						|
    x2.resize(nstages); // (float*)malloc0(nstages * sizeof(float));
 | 
						|
    y0.resize(nstages); // (float*)malloc0(nstages * sizeof(float));
 | 
						|
    y1.resize(nstages); // (float*)malloc0(nstages * sizeof(float));
 | 
						|
    y2.resize(nstages); // (float*)malloc0(nstages * sizeof(float));
 | 
						|
    calc();
 | 
						|
}
 | 
						|
 | 
						|
void DBQBP::flush()
 | 
						|
{
 | 
						|
    for (int i = 0; i < nstages; i++)
 | 
						|
    {
 | 
						|
        x1[i] = x2[i] = y1[i] = y2[i] = 0.0;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
void DBQBP::execute()
 | 
						|
{
 | 
						|
    if (run)
 | 
						|
    {
 | 
						|
 | 
						|
        for (int i = 0; i < size; i++)
 | 
						|
        {
 | 
						|
            x0[0] = gain * in[i];
 | 
						|
 | 
						|
            for (int n = 0; n < nstages; n++)
 | 
						|
            {
 | 
						|
                if (n > 0)
 | 
						|
                    x0[n] = y0[n - 1];
 | 
						|
 | 
						|
                y0[n] = a0 * x0[n]
 | 
						|
                    + a1 * x1[n]
 | 
						|
                    + a2 * x2[n]
 | 
						|
                    + b1 * y1[n]
 | 
						|
                    + b2 * y2[n];
 | 
						|
                y2[n] = y1[n];
 | 
						|
                y1[n] = y0[n];
 | 
						|
                x2[n] = x1[n];
 | 
						|
                x1[n] = x0[n];
 | 
						|
            }
 | 
						|
 | 
						|
            out[i] = (float) y0[nstages - 1];
 | 
						|
        }
 | 
						|
    }
 | 
						|
    else if (out != in)
 | 
						|
    {
 | 
						|
        std::copy(in, in + size, out);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
void DBQBP::setBuffers(float* _in, float* _out)
 | 
						|
{
 | 
						|
    in = _in;
 | 
						|
    out = _out;
 | 
						|
}
 | 
						|
 | 
						|
void DBQBP::setSamplerate(int _rate)
 | 
						|
{
 | 
						|
    rate = _rate;
 | 
						|
    calc();
 | 
						|
}
 | 
						|
 | 
						|
void DBQBP::setSize(int _size)
 | 
						|
{
 | 
						|
    size = _size;
 | 
						|
    flush();
 | 
						|
}
 | 
						|
 | 
						|
} // namespace WDSP
 |