mirror of
https://github.com/f4exb/sdrangel.git
synced 2025-10-30 20:40:20 -04:00
201 lines
8.1 KiB
C++
201 lines
8.1 KiB
C++
|
/* firmin.h
|
||
|
|
|
||
|
|
This file is part of a program that implements a Software-Defined Radio.
|
||
|
|
|
||
|
|
Copyright (C) 2016 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
|
||
|
|
|
||
|
|
*/
|
||
|
|
|
||
|
|
/********************************************************************************************************
|
||
|
|
* *
|
||
|
|
* Time-Domain FIR *
|
||
|
|
* *
|
||
|
|
********************************************************************************************************/
|
||
|
|
|
||
|
|
#ifndef wdsp_firmin_h
|
||
|
|
#define wdsp_firmin_h
|
||
|
|
|
||
|
|
#include "fftw3.h"
|
||
|
|
#include "export.h"
|
||
|
|
|
||
|
|
namespace WDSP {
|
||
|
|
|
||
|
|
class WDSP_API FIRMIN
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
int run; // run control
|
||
|
|
int position; // position at which to execute
|
||
|
|
int size; // input/output buffer size, power of two
|
||
|
|
double* in; // input buffer
|
||
|
|
double* out; // output buffer, can be same as input
|
||
|
|
int nc; // number of filter coefficients, power of two
|
||
|
|
double f_low; // low cutoff frequency
|
||
|
|
double f_high; // high cutoff frequency
|
||
|
|
double* ring; // internal complex ring buffer
|
||
|
|
double* h; // complex filter coefficients
|
||
|
|
int rsize; // ring size, number of complex samples, power of two
|
||
|
|
int mask; // mask to update indexes
|
||
|
|
int idx; // ring input/output index
|
||
|
|
double samplerate; // sample rate
|
||
|
|
int wintype; // filter window type
|
||
|
|
double gain; // filter gain
|
||
|
|
|
||
|
|
static FIRMIN* create_firmin (int run, int position, int size, double* in, double* out,
|
||
|
|
int nc, double f_low, double f_high, int samplerate, int wintype, double gain);
|
||
|
|
static void destroy_firmin (FIRMIN *a);
|
||
|
|
static void flush_firmin (FIRMIN *a);
|
||
|
|
static void xfirmin (FIRMIN *a, int pos);
|
||
|
|
static void setBuffers_firmin (FIRMIN *a, double* in, double* out);
|
||
|
|
static void setSamplerate_firmin (FIRMIN *a, int rate);
|
||
|
|
static void setSize_firmin (FIRMIN *a, int size);
|
||
|
|
static void setFreqs_firmin (FIRMIN *a, double f_low, double f_high);
|
||
|
|
|
||
|
|
private:
|
||
|
|
static void calc_firmin (FIRMIN *a);
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace WDSP
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
/********************************************************************************************************
|
||
|
|
* *
|
||
|
|
* Standalone Partitioned Overlap-Save Bandpass *
|
||
|
|
* *
|
||
|
|
********************************************************************************************************/
|
||
|
|
|
||
|
|
#ifndef wdsp_firopt_h
|
||
|
|
#define wdsp_firopt_h
|
||
|
|
|
||
|
|
#include "fftw3.h"
|
||
|
|
#include "export.h"
|
||
|
|
|
||
|
|
namespace WDSP {
|
||
|
|
|
||
|
|
class WDSP_API FIROPT
|
||
|
|
{
|
||
|
|
int run; // run control
|
||
|
|
int position; // position at which to execute
|
||
|
|
int size; // input/output buffer size, power of two
|
||
|
|
double* in; // input buffer
|
||
|
|
double* out; // output buffer, can be same as input
|
||
|
|
int nc; // number of filter coefficients, power of two, >= size
|
||
|
|
double f_low; // low cutoff frequency
|
||
|
|
double f_high; // high cutoff frequency
|
||
|
|
double samplerate; // sample rate
|
||
|
|
int wintype; // filter window type
|
||
|
|
double gain; // filter gain
|
||
|
|
int nfor; // number of buffers in delay line
|
||
|
|
double* fftin; // fft input buffer
|
||
|
|
double** fmask; // frequency domain masks
|
||
|
|
double** fftout; // fftout delay line
|
||
|
|
double* accum; // frequency domain accumulator
|
||
|
|
int buffidx; // fft out buffer index
|
||
|
|
int idxmask; // mask for index computations
|
||
|
|
double* maskgen; // input for mask generation FFT
|
||
|
|
fftw_plan* pcfor; // array of forward FFT plans
|
||
|
|
fftw_plan crev; // reverse fft plan
|
||
|
|
fftw_plan* maskplan; // plans for frequency domain masks
|
||
|
|
|
||
|
|
static FIROPT* create_firopt (int run, int position, int size, double* in, double* out,
|
||
|
|
int nc, double f_low, double f_high, int samplerate, int wintype, double gain);
|
||
|
|
static void xfiropt (FIROPT *a, int pos);
|
||
|
|
static void destroy_firopt (FIROPT *a);
|
||
|
|
static void flush_firopt (FIROPT *a);
|
||
|
|
static void setBuffers_firopt (FIROPT *a, double* in, double* out);
|
||
|
|
static void setSamplerate_firopt (FIROPT *a, int rate);
|
||
|
|
static void setSize_firopt (FIROPT *a, int size);
|
||
|
|
static void setFreqs_firopt (FIROPT *a, double f_low, double f_high);
|
||
|
|
|
||
|
|
private:
|
||
|
|
static void plan_firopt (FIROPT *a);
|
||
|
|
static void calc_firopt (FIROPT *a);
|
||
|
|
static void deplan_firopt (FIROPT *a);
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace WDSP
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
/********************************************************************************************************
|
||
|
|
* *
|
||
|
|
* Partitioned Overlap-Save Filter Kernel *
|
||
|
|
* *
|
||
|
|
********************************************************************************************************/
|
||
|
|
|
||
|
|
#ifndef wdsp_fircore_h
|
||
|
|
#define wdsp_fircore_h
|
||
|
|
|
||
|
|
#include <QRecursiveMutex>
|
||
|
|
#include "export.h"
|
||
|
|
|
||
|
|
namespace WDSP {
|
||
|
|
|
||
|
|
class WDSP_API FIRCORE
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
int size; // input/output buffer size, power of two
|
||
|
|
double* in; // input buffer
|
||
|
|
double* out; // output buffer, can be same as input
|
||
|
|
int nc; // number of filter coefficients, power of two, >= size
|
||
|
|
double* impulse; // impulse response of filter
|
||
|
|
double* imp;
|
||
|
|
int nfor; // number of buffers in delay line
|
||
|
|
double* fftin; // fft input buffer
|
||
|
|
double*** fmask; // frequency domain masks
|
||
|
|
double** fftout; // fftout delay line
|
||
|
|
double* accum; // frequency domain accumulator
|
||
|
|
int buffidx; // fft out buffer index
|
||
|
|
int idxmask; // mask for index computations
|
||
|
|
double* maskgen; // input for mask generation FFT
|
||
|
|
fftw_plan* pcfor; // array of forward FFT plans
|
||
|
|
fftw_plan crev; // reverse fft plan
|
||
|
|
fftw_plan** maskplan; // plans for frequency domain masks
|
||
|
|
QRecursiveMutex update;
|
||
|
|
int cset;
|
||
|
|
int mp;
|
||
|
|
int masks_ready;
|
||
|
|
|
||
|
|
static FIRCORE* create_fircore (int size, double* in, double* out,
|
||
|
|
int nc, int mp, double* impulse);
|
||
|
|
static void xfircore (FIRCORE *a);
|
||
|
|
static void destroy_fircore (FIRCORE *a);
|
||
|
|
static void flush_fircore (FIRCORE *a);
|
||
|
|
static void setBuffers_fircore (FIRCORE *a, double* in, double* out);
|
||
|
|
static void setSize_fircore (FIRCORE *a, int size);
|
||
|
|
static void setImpulse_fircore (FIRCORE *a, double* impulse, int update);
|
||
|
|
static void setNc_fircore (FIRCORE *a, int nc, double* impulse);
|
||
|
|
static void setMp_fircore (FIRCORE *a, int mp);
|
||
|
|
static void setUpdate_fircore (FIRCORE *a);
|
||
|
|
|
||
|
|
private:
|
||
|
|
static void plan_fircore (FIRCORE *a);
|
||
|
|
static void calc_fircore (FIRCORE *a, int flip);
|
||
|
|
static void deplan_fircore (FIRCORE *a);
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace WDSP
|
||
|
|
|
||
|
|
#endif
|