/*  gen.h

This file is part of a program that implements a Software-Defined Radio.

Copyright (C) 2013 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_gen_h
#define wdsp_gen_h

#include "export.h"

namespace WDSP {

class TXA;

class WDSP_API GEN
{
public:
    int run;                    // run
    int size;                   // number of samples per buffer
    float* in;                 // input buffer (retained in case I want to mix in a generated signal)
    float* out;                // output buffer
    float rate;                // sample rate
    int mode;
    struct _tone
    {
        float mag;
        float freq;
        float phs;
        float delta;
        float cosdelta;
        float sindelta;
    } tone;
    struct _tt
    {
        float mag1;
        float mag2;
        float f1;
        float f2;
        float phs1;
        float phs2;
        float delta1;
        float delta2;
        float cosdelta1;
        float cosdelta2;
        float sindelta1;
        float sindelta2;
    } tt;
    struct _noise
    {
        float mag;
    } noise;
    struct _sweep
    {
        float mag;
        float f1;
        float f2;
        float sweeprate;
        float phs;
        float dphs;
        float d2phs;
        float dphsmax;
    } sweep;
    struct _saw
    {
        float mag;
        float f;
        float period;
        float delta;
        float t;
    } saw;
    struct _tri
    {
        float mag;
        float f;
        float period;
        float half;
        float delta;
        float t;
        float t1;
    } tri;
    struct _pulse
    {
        float mag;
        float pf;
        float pdutycycle;
        float ptranstime;
        float* ctrans;
        int pcount;
        int pnon;
        int pntrans;
        int pnoff;
        float pperiod;
        float tf;
        float tphs;
        float tdelta;
        float tcosdelta;
        float tsindelta;
        int state;
    } pulse;

    GEN(
        int run,
        int size,
        float* in,
        float* out,
        int rate,
        int mode
    );
    ~GEN();

    void flush();
    void execute();
    void setBuffers(float* in, float* out);
    void setSamplerate(int rate);
    void setSize(int size);
    // Public Properties
    void SetPreRun(int run);
    void SetPreMode(int mode);
    void SetPreToneMag(float mag);
    void SetPreToneFreq(float freq);
    void SetPreNoiseMag(float mag);
    void SetPreSweepMag(float mag);
    void SetPreSweepFreq(float freq1, float freq2);
    void SetPreSweepRate(float rate);
    // TXA Properties
    static void SetPreGenRun (TXA& txa, int run);
    static void SetPreGenMode (TXA& txa, int mode);
    static void SetPreGenToneMag (TXA& txa, float mag);
    static void SetPreGenToneFreq (TXA& txa, float freq);
    static void SetPreGenNoiseMag (TXA& txa, float mag);
    static void SetPreGenSweepMag (TXA& txa, float mag);
    static void SetPreGenSweepFreq (TXA& txal, float freq1, float freq2);
    static void SetPreGenSweepRate (TXA& txa, float rate);
    static void SetPreGenSawtoothMag (TXA& txa, float mag);
    static void SetPreGenSawtoothFreq (TXA& txa, float freq);
    static void SetPreGenTriangleMag (TXA& txa, float mag);
    static void SetPreGenTriangleFreq (TXA& txa, float freq);
    static void SetPreGenPulseMag (TXA& txa, float mag);
    static void SetPreGenPulseFreq (TXA& txa, float freq);
    static void SetPreGenPulseDutyCycle (TXA& txa, float dc);
    static void SetPreGenPulseToneFreq (TXA& txa, float freq);
    static void SetPreGenPulseTransition (TXA& txa, float transtime);
    // 'PostGen', gen1
    static void SetPostGenRun (TXA& txa, int run);
    static void SetPostGenMode (TXA& txa, int mode);
    static void SetPostGenToneMag (TXA& txa, float mag);
    static void SetPostGenToneFreq (TXA& txa, float freq);
    static void SetPostGenTTMag (TXA& txa, float mag1, float mag2);
    static void SetgenRun (TXA& txa, int run);
    static void SetPostGenTTFreq (TXA& txa, float freq1, float freq2);
    static void SetPostGenSweepMag (TXA& txa, float mag);
    static void SetPostGenSweepFreq (TXA& txa, float freq1, float freq2);
    static void SetPostGenSweepRate (TXA& txa, float rate);

private:
    void calc_tone();
    void calc_tt();
    void calc_sweep();
    void calc_sawtooth();
    void calc_triangle();
    void calc_pulse();
    void calc();
    void decalc();
};

} // namespace WDSP

#endif