1
0
mirror of https://github.com/f4exb/sdrangel.git synced 2024-11-11 02:46:12 -05:00
sdrangel/wdsp/cfcomp.hpp

158 lines
3.6 KiB
C++
Raw Normal View History

2024-06-16 05:31:13 -04:00
/* cfcomp.h
This file is part of a program that implements a Software-Defined Radio.
Copyright (C) 2017, 2021 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
*/
#ifndef wdsp_cfcomp_h
#define wdsp_cfcomp_h
2024-08-05 14:05:59 -04:00
#include <vector>
2024-06-16 05:31:13 -04:00
#include "fftw3.h"
#include "export.h"
namespace WDSP {
class TXA;
class WDSP_API CFCOMP
{
public:
int run;
int position;
int bsize;
2024-06-24 21:50:48 -04:00
float* in;
float* out;
2024-06-16 05:31:13 -04:00
int fsize;
int ovrlp;
int incr;
2024-08-05 14:05:59 -04:00
std::vector<double> window;
2024-06-16 05:31:13 -04:00
int iasize;
2024-08-05 14:05:59 -04:00
std::vector<double> inaccum;
std::vector<float> forfftin;
std::vector<float> forfftout;
2024-06-16 05:31:13 -04:00
int msize;
2024-08-05 14:05:59 -04:00
std::vector<double> cmask;
std::vector<double> mask;
2024-06-16 05:31:13 -04:00
int mask_ready;
2024-08-05 14:05:59 -04:00
std::vector<double> cfc_gain;
std::vector<float> revfftin;
std::vector<float> revfftout;
std::vector<std::vector<double>> save;
2024-06-16 05:31:13 -04:00
int oasize;
2024-08-05 14:05:59 -04:00
std::vector<double> outaccum;
double rate;
2024-06-16 05:31:13 -04:00
int wintype;
2024-08-05 14:05:59 -04:00
double pregain;
double postgain;
2024-06-16 05:31:13 -04:00
int nsamps;
int iainidx;
int iaoutidx;
int init_oainidx;
int oainidx;
int oaoutidx;
int saveidx;
2024-06-24 21:50:48 -04:00
fftwf_plan Rfor;
fftwf_plan Rrev;
2024-06-16 05:31:13 -04:00
int comp_method;
int nfreqs;
2024-08-05 14:05:59 -04:00
std::vector<double> F;
std::vector<double> G;
std::vector<double> E;
std::vector<double> fp;
std::vector<double> gp;
std::vector<double> ep;
std::vector<double> comp;
double precomp;
double precomplin;
std::vector<double> peq;
2024-06-16 05:31:13 -04:00
int peq_run;
2024-08-05 14:05:59 -04:00
double prepeq;
double prepeqlin;
double winfudge;
2024-06-16 05:31:13 -04:00
2024-08-05 14:05:59 -04:00
double gain;
double mtau;
double mmult;
2024-06-16 05:31:13 -04:00
// display stuff
2024-08-05 14:05:59 -04:00
double dtau;
double dmult;
std::vector<double> delta;
std::vector<double> delta_copy;
std::vector<double> cfc_gain_copy;
2024-06-16 05:31:13 -04:00
2024-08-05 14:05:59 -04:00
CFCOMP(
2024-06-16 05:31:13 -04:00
int run,
int position,
int peq_run,
int size,
2024-06-24 21:50:48 -04:00
float* in,
float* out,
2024-06-16 05:31:13 -04:00
int fsize,
int ovrlp,
int rate,
int wintype,
int comp_method,
int nfreqs,
2024-08-05 14:05:59 -04:00
double precomp,
double prepeq,
const double* F,
const double* G,
const double* E,
double mtau,
double dtau
2024-06-16 05:31:13 -04:00
);
2024-08-05 14:05:59 -04:00
CFCOMP(const CFCOMP&) = delete;
CFCOMP& operator=(CFCOMP& other) = delete;
~CFCOMP();
void flush();
void execute(int pos);
void setBuffers(float* in, float* out);
void setSamplerate(int rate);
void setSize(int size);
2024-06-16 05:31:13 -04:00
// TXA Properties
2024-08-05 14:05:59 -04:00
void setRun(int run);
void setPosition(int pos);
void setProfile(int nfreqs, const double* F, const double* G, const double *E);
void setPrecomp(double precomp);
void setPeqRun(int run);
void setPrePeq(double prepeq);
void getDisplayCompression(double* comp_values, int* ready);
2024-06-16 05:31:13 -04:00
private:
2024-08-05 14:05:59 -04:00
void calc_cfcwindow();
2024-06-16 05:31:13 -04:00
static int fCOMPcompare (const void *a, const void *b);
2024-08-05 14:05:59 -04:00
void calc_comp();
void calc_cfcomp();
void decalc_cfcomp();
void calc_mask();
2024-06-16 05:31:13 -04:00
};
} // namespace WDSP
#endif