ConvolveWithConst.h

Go to the documentation of this file.

#ifndef CONVOLVEWITHCONST_H_
#define CONVOLVEWITHCONST_H_
 
#include <complex>
#include <vector>
#include "TVirtualFFT.h"
#include <string>
#include <map>// we have no unordered_map.
/**
 * @brief basically, convolve with fft.
 *  convolves 1-d.
 *  if size of 2 inputs are fixed, it may yield optimal speed.
 *  currently not thread safe.
 * 
 */
//struct convParams;
//struct fft2;
class ConvolveWithConst{
public:
    enum TransformOptimOpt{opt_ES,opt_M,opt_P,opt_EX};
    enum TransformLengthOpt{opt_AsShortAsPsb, opt_Nearst2Pow};
    typedef std::complex<double> Complex;
private:
    
    struct convParams;
    struct fft2;
    TVirtualFFT * m_fft, * m_ifft;
    int m_length;
    int m_BLength;
    bool m_initialized;
    //int m_outLength;
    mutable std::vector<double> A_fft, B_fft, AB_fft,B_tmp;// *out;
    std::vector<double> zeros;
    TransformOptimOpt m_optOptim;
    TransformLengthOpt m_optLength;
    //static const char options1[4][10];
    //static const char options2[4][10];
 
    
 
    static std::map<convParams,fft2> m_map;
    std::map<convParams,fft2>::iterator m_ref;
 
public:
    ConvolveWithConst();// default constructor, uninitialized
    void init(const double* ConstArray,const int ConstArrayLength, const int Blength, TransformOptimOpt opt=opt_EX,TransformLengthOpt optL=opt_AsShortAsPsb);
 
    ConvolveWithConst(const double* ConstArray,const int ConstArrayLength, const int Blength, TransformOptimOpt opt=opt_EX,TransformLengthOpt optL=opt_AsShortAsPsb);
    //ConvolveWithConst(const int convolveLength ,TransformOptimOpt opt=opt_EX,TransformLengthOpt optL=opt_AsShortAsPsb);
    //ConvolveWithConst():m_fft(nullptr),m_ifft(nullptr){};
    ~ConvolveWithConst();
    void convolve(double * output , const double *  B,const int leftIndex,const int sizeofOut,double factor=1) const;
    //static void destoryAll();// clear all ffts. after this all ConvolveWithConst instances become unusable...not sure where i should place this. global var is evil)
                                // so we did a ref count, removing the need of destoryAll.
    int getLength() const{return m_length;};
    //TVirtualFFT * getFFT(){return m_fft;}
    //TVirtualFFT * getIFFT(){return m_ifft;}
    //void MultiplyAndAdd(Complex* outHalfPlus1, const Complex* inHalfPlus1, double factor) const;
    void MultiplyAndAdd(Complex* outHalfPlus1, const Complex* inHalfPlus1, double factor=1) const;
    void FFT(Complex* outHalfPlus1, const double * inNormalLength, int LengthOutAsDouble, int LengthIn) const;
    void IFFT(double* outNormalLength, const Complex * inHalfPlus1, int lengthOut, int lengthInAsDouble) const;
 
};
 
 
 
#endif