FFTConvolver.h

Go to the documentation of this file.

/*******************************************************************************
* Copyright 2015-2016 Juan Francisco Crespo Galán
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*   http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
******************************************************************************/
 
#pragma once
 
#include "IReader.h"
#include "ISound.h"
#include "util/FFTPlan.h"
 
#include <memory>
#include <vector>
 
AUD_NAMESPACE_BEGIN
class AUD_API FFTConvolver
{
private:
    std::shared_ptr<FFTPlan> m_plan;
 
    int m_N;
 
    int m_M;
 
    int m_L;
 
    int m_realBufLen;
 
    std::complex<sample_t>* m_inBuffer;
 
    sample_t* m_shiftBuffer;
 
    float* m_tail;
 
    std::shared_ptr<std::vector<std::complex<sample_t>>> m_irBuffer;
 
    int m_tailPos;
 
    // delete copy constructor and operator=
    FFTConvolver(const FFTConvolver&) = delete;
    FFTConvolver& operator=(const FFTConvolver&) = delete;
 
public:
    FFTConvolver(std::shared_ptr<std::vector<std::complex<sample_t>>> ir, std::shared_ptr<FFTPlan> plan);
    virtual ~FFTConvolver();
 
    void getNext(const sample_t* inBuffer, sample_t* outBuffer, int& length);
 
    void getNext(const sample_t* inBuffer, sample_t* outBuffer, int& length, fftwf_complex* transformedData);
 
    void getNext(const fftwf_complex* inBuffer, sample_t* outBuffer, int& length);
    
    void getTail(int& length, bool& eos, sample_t* buffer);
 
    void clear();
 
    void IFFT_FDL(const fftwf_complex* inBuffer, sample_t* outBuffer, int& length);
 
    void getNextFDL(const std::complex<sample_t>* inBuffer, std::complex<sample_t>* accBuffer);
 
    void getNextFDL(const sample_t* inBuffer, std::complex<sample_t>* accBuffer, int& length, fftwf_complex* transformedData);
 
    void setImpulseResponse(std::shared_ptr<std::vector<std::complex<sample_t>>> ir);
 
    std::shared_ptr<std::vector<std::complex<sample_t>>> getImpulseResponse();
};
 
AUD_NAMESPACE_END