/*  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

#include "fftw3.h"
#include "export.h"

namespace WDSP {

class TXA;

class WDSP_API CFCOMP
{
public:
    int run;
    int position;
    int bsize;
    double* in;
    double* out;
    int fsize;
    int ovrlp;
    int incr;
    double* window;
    int iasize;
    double* inaccum;
    double* forfftin;
    double* forfftout;
    int msize;
    double* cmask;
    double* mask;
    int mask_ready;
    double* cfc_gain;
    double* revfftin;
    double* revfftout;
    double** save;
    int oasize;
    double* outaccum;
    double rate;
    int wintype;
    double pregain;
    double postgain;
    int nsamps;
    int iainidx;
    int iaoutidx;
    int init_oainidx;
    int oainidx;
    int oaoutidx;
    int saveidx;
    fftw_plan Rfor;
    fftw_plan Rrev;

    int comp_method;
    int nfreqs;
    double* F;
    double* G;
    double* E;
    double* fp;
    double* gp;
    double* ep;
    double* comp;
    double precomp;
    double precomplin;
    double* peq;
    int peq_run;
    double prepeq;
    double prepeqlin;
    double winfudge;

    double gain;
    double mtau;
    double mmult;
    // display stuff
    double dtau;
    double dmult;
    double* delta;
    double* delta_copy;
    double* cfc_gain_copy;

    static CFCOMP* create_cfcomp (
        int run,
        int position,
        int peq_run,
        int size,
        double* in,
        double* out,
        int fsize,
        int ovrlp,
        int rate,
        int wintype,
        int comp_method,
        int nfreqs,
        double precomp,
        double prepeq,
        double* F,
        double* G,
        double* E,
        double mtau,
        double dtau
    );
    static void destroy_cfcomp (CFCOMP *a);
    static void flush_cfcomp (CFCOMP *a);
    static void xcfcomp (CFCOMP *a, int pos);
    static void setBuffers_cfcomp (CFCOMP *a, double* in, double* out);
    static void setSamplerate_cfcomp (CFCOMP *a, int rate);
    static void setSize_cfcomp (CFCOMP *a, int size);
    // TXA Properties
    static void SetCFCOMPRun (TXA& txa, int run);
    static void SetCFCOMPPosition (TXA& txa, int pos);
    static void SetCFCOMPprofile (TXA& txa, int nfreqs, double* F, double* G, double *E);
    static void SetCFCOMPPrecomp (TXA& txa, double precomp);
    static void SetCFCOMPPeqRun (TXA& txa, int run);
    static void SetCFCOMPPrePeq (TXA& txa, double prepeq);
    static void GetCFCOMPDisplayCompression (TXA& txa, double* comp_values, int* ready);

private:
    static void calc_cfcwindow (CFCOMP *a);
    static int fCOMPcompare (const void *a, const void *b);
    static void calc_comp (CFCOMP *a);
    static void calc_cfcomp(CFCOMP *a);
    static void decalc_cfcomp(CFCOMP *a);
    static void calc_mask (CFCOMP *a);
};

} // namespace WDSP

#endif