kopia lustrzana https://github.com/f4exb/sdrangel
98 wiersze
3.9 KiB
C++
98 wiersze
3.9 KiB
C++
///////////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (C) 2023 Edouard Griffiths, F4EXB. //
|
|
// //
|
|
// This is the code from ft8mon: https://github.com/rtmrtmrtmrtm/ft8mon //
|
|
// written by Robert Morris, AB1HL //
|
|
// reformatted and adapted to Qt and SDRangel context //
|
|
// //
|
|
// 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 as version 3 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 V3 for more details. //
|
|
// //
|
|
// You should have received a copy of the GNU General Public License //
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>. //
|
|
///////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef FFT_H
|
|
#define FFT_H
|
|
|
|
#include <mutex>
|
|
#include <vector>
|
|
#include <complex>
|
|
#include <fftw3.h>
|
|
|
|
#include "export.h"
|
|
|
|
namespace FT8
|
|
{
|
|
class FT8_API FFTEngine
|
|
{
|
|
public:
|
|
// a cached fftw plan, for both of:
|
|
// fftwf_plan_dft_r2c_1d(n, m_in, m_out, FFTW_ESTIMATE);
|
|
// fftwf_plan_dft_c2r_1d(n, m_in, m_out, FFTW_ESTIMATE);
|
|
class Plan
|
|
{
|
|
public:
|
|
int n_;
|
|
int type_;
|
|
|
|
//
|
|
// real -> complex
|
|
//
|
|
fftwf_complex *c_; // (n_ / 2) + 1 of these
|
|
float *r_; // n_ of these
|
|
fftwf_plan fwd_; // forward plan
|
|
fftwf_plan rev_; // reverse plan
|
|
|
|
//
|
|
// complex -> complex
|
|
//
|
|
fftwf_complex *cc1_; // n
|
|
fftwf_complex *cc2_; // n
|
|
fftwf_plan cfwd_; // forward plan
|
|
fftwf_plan crev_; // reverse plan
|
|
|
|
// how much CPU time spent in FFTs that use this plan.
|
|
#if TIMING
|
|
double time_;
|
|
#endif
|
|
const char *why_;
|
|
int uses_;
|
|
}; // Plan
|
|
|
|
FFTEngine() : nplans(0)
|
|
{}
|
|
|
|
Plan *get_plan(int n, const char *why);
|
|
|
|
std::vector<std::complex<float>> one_fft(const std::vector<float> &samples, int i0, int block, const char *why, Plan *p);
|
|
std::vector<float> one_ifft(const std::vector<std::complex<float>> &bins, const char *why);
|
|
typedef std::vector<std::vector<std::complex<float>>> ffts_t;
|
|
ffts_t ffts(const std::vector<float> &samples, int i0, int block, const char *why);
|
|
std::vector<std::complex<float>> one_fft_c(const std::vector<float> &samples, int i0, int block, const char *why);
|
|
std::vector<std::complex<float>> one_fft_cc(const std::vector<std::complex<float>> &samples, int i0, int block, const char *why);
|
|
std::vector<std::complex<float>> one_ifft_cc(const std::vector<std::complex<float>> &bins, const char *why);
|
|
std::vector<std::complex<float>> analytic(const std::vector<float> &x, const char *why);
|
|
std::vector<float> hilbert_shift(const std::vector<float> &x, float hz0, float hz1, int rate);
|
|
|
|
private:
|
|
void fft_stats();
|
|
std::mutex plansmu;
|
|
std::mutex plansmu2;
|
|
Plan *plans[1000];
|
|
int nplans;
|
|
// MEASURE=0, ESTIMATE=64, PATIENT=32
|
|
static const int fftw_type = FFTW_ESTIMATE;
|
|
}; // FFTEngine
|
|
|
|
} // namespace FT8
|
|
|
|
#endif
|