Convolver.h

Go to the documentation of this file.

#ifndef INCLUDE_CONVOLUTION_H
#define INCLUDE_CONVOLUTION_H

#include "CSL_Core.h"
#include "Spectral.h"           // fftw includes and types


#define CHECK_PTR(ptr)                      \
    if (!ptr)    throw MemoryError("Can't allocate buffer")

#define initvec( name, size, type )             \
    if ((name = (type *) calloc(size, sizeof(type))) == NULL)  \
        throw MemoryError("Can't allocate buffer");

#define cmac(in1, in2, out)                 \
    out[0] += in1[0] * in2[0] - in1[1] * in2[1];        \
    out[1] += in1[0] * in2[1] + in1[1] * in2[0];

#define cbinc(buf, size)    if (++buf > size-1)     buf = 0;
#define cbdec(buf, size)    if (--buf < 0)          buf = size - 1;

#define cbarb(buf, size, amt)                   \
    buf += amt;                             \
    if (buf >= size) buf -= size;                   \
    else if (buf < 0) buf += size;

namespace csl {



class Convolver : public Effect {

public:                                 
    Convolver() : Effect() {};
    Convolver(char *IRfilename);            
    Convolver(unsigned len, Buffer & impulseResp);
    Convolver(unsigned len, char * IRfilename);
    Convolver(Buffer & inbuf);
    ~Convolver();
    void initialize(Buffer & buf) throw (CException);           
    void setFilters(fftwf_complex ** filterFFTs);               
    void setInputf(fftwf_complex * inFFT);                  
    CSL_FFTW_sample * mSampleBuffer;                    

    void nextBuffer(Buffer &outputBuffer, unsigned outBufNum) throw (CException);

protected:  
    CSL_FFTW_cmplx ** mFilterFFTs;                          
    CSL_FFTW_cmplx ** mInputFFTs;                           
    CSL_FFTW_cmplx * mSpectrumBuffer;                       
    CSL_FFTW_plan mForwardPlan, mInversePlan;               
    unsigned mFFTSize, mWindowCount, mNumBufs;      
    bool mMyBuffers;                                        
    bool mMyInput;                                      
    
    void initialize (const char *filename) throw (CException);  
    
                                                        // fast complex ptr mult-accum utility, might be in-lined
//  void complex_multiply_accumulate(fftwf_complex* left, fftwf_complex* right, fftwf_complex * output);

    inline void complex_multiply_accumulate(CSL_FFTW_cmplx * left, CSL_FFTW_cmplx * right, CSL_FFTW_cmplx * output) {
        CSL_FFTW_cmplx * loleft = left;                     // local copies of args
        CSL_FFTW_cmplx * loright = right;
        CSL_FFTW_cmplx * looutput = output;
        float re, im;                                       // temps
        float leftRe, rightRe, leftIm, rightIm;
        unsigned frame = mFFTSize;
    //  printf("MAC: %x - %x - %x (%u)\n", left, right, output, frame);
        while (frame--) {                                   // loop over frames
            leftRe = (*loleft)[0];                          // cache vector values
            rightRe = (*loright)[0];
            leftIm = (*loleft)[1]; 
            rightIm = (*loright)[1];
            re = (leftRe * rightRe) - (leftIm * rightIm);       // complex multiply
            im = (leftRe * rightIm) + (leftIm * rightRe);
            (*looutput)[0] += re;                           // sum into work fft
            (*looutput)[1] += im;
            loleft++;                                       // increment register pointers
            loright++;
            looutput++;
        }
    }

};

}

#endif

---