2024-07-13 01:20:26 +00:00
|
|
|
/* 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
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
/********************************************************************************************************
|
|
|
|
* *
|
|
|
|
* Partitioned Overlap-Save Filter Kernel *
|
|
|
|
* *
|
|
|
|
********************************************************************************************************/
|
|
|
|
|
|
|
|
#ifndef wdsp_fircore_h
|
|
|
|
#define wdsp_fircore_h
|
|
|
|
|
2024-08-06 04:26:53 +00:00
|
|
|
#include <array>
|
|
|
|
#include <vector>
|
|
|
|
|
2024-07-13 01:20:26 +00:00
|
|
|
#include "fftw3.h"
|
|
|
|
#include "export.h"
|
|
|
|
|
|
|
|
namespace WDSP {
|
|
|
|
|
|
|
|
class WDSP_API FIRCORE
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
int size; // input/output buffer size, power of two
|
|
|
|
float* in; // input buffer
|
|
|
|
float* out; // output buffer, can be same as input
|
|
|
|
int nc; // number of filter coefficients, power of two, >= size
|
2024-08-06 04:26:53 +00:00
|
|
|
std::vector<float> impulse; // impulse response of filter
|
|
|
|
std::vector<float> imp;
|
2024-07-13 01:20:26 +00:00
|
|
|
int nfor; // number of buffers in delay line
|
2024-08-06 04:26:53 +00:00
|
|
|
std::vector<float> fftin; // fft input buffer
|
|
|
|
std::array<std::vector<std::vector<float>>, 2> fmask; // frequency domain masks
|
|
|
|
std::vector<std::vector<float>> fftout; // fftout delay line
|
|
|
|
std::vector<float> accum; // frequency domain accumulator
|
2024-07-13 01:20:26 +00:00
|
|
|
int buffidx; // fft out buffer index
|
|
|
|
int idxmask; // mask for index computations
|
2024-08-06 04:26:53 +00:00
|
|
|
std::vector<float> maskgen; // input for mask generation FFT
|
|
|
|
std::vector<fftwf_plan> pcfor; // array of forward FFT plans
|
2024-07-13 01:20:26 +00:00
|
|
|
fftwf_plan crev; // reverse fft plan
|
2024-08-06 04:26:53 +00:00
|
|
|
std::array<std::vector<fftwf_plan>, 2> maskplan; // plans for frequency domain masks
|
2024-07-13 01:20:26 +00:00
|
|
|
int cset;
|
|
|
|
int mp;
|
|
|
|
int masks_ready;
|
|
|
|
|
2024-08-06 04:26:53 +00:00
|
|
|
FIRCORE(
|
|
|
|
int size,
|
|
|
|
float* in,
|
|
|
|
float* out,
|
|
|
|
int mp,
|
2024-08-10 10:21:04 +00:00
|
|
|
const std::vector<float>& impulse
|
2024-08-06 04:26:53 +00:00
|
|
|
);
|
|
|
|
FIRCORE(const FIRCORE&) = delete;
|
|
|
|
FIRCORE& operator=(const FIRCORE& other) = delete;
|
|
|
|
~FIRCORE();
|
|
|
|
|
|
|
|
void flush();
|
|
|
|
void execute();
|
|
|
|
void setBuffers(float* in, float* out);
|
|
|
|
void setSize(int size);
|
2024-08-10 10:21:04 +00:00
|
|
|
void setImpulse(const std::vector<float>& impulse, int update);
|
|
|
|
void setNc(const std::vector<float>& impulse);
|
2024-08-06 04:26:53 +00:00
|
|
|
void setMp(int mp);
|
|
|
|
void setUpdate();
|
2024-07-13 01:20:26 +00:00
|
|
|
|
|
|
|
private:
|
2024-08-06 04:26:53 +00:00
|
|
|
void plan();
|
|
|
|
void calc(int flip);
|
|
|
|
void deplan();
|
2024-07-13 01:20:26 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace WDSP
|
|
|
|
|
|
|
|
#endif
|