Convolver.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 "FFTConvolver.h"
#include "util/ThreadPool.h"
#include "util/FFTPlan.h"
 
#include <memory>
#include <vector>
#include <mutex>
#include <future>
#include <atomic>
#include <deque>
 
AUD_NAMESPACE_BEGIN
class AUD_API Convolver
{
private:
    int m_N;
 
    int m_M;
 
    int m_L;
 
    std::shared_ptr<std::vector<std::shared_ptr<std::vector<std::complex<sample_t>>>>> m_irBuffers;
 
    std::vector<fftwf_complex*> m_threadAccBuffers;
 
    std::vector<std::unique_ptr<FFTConvolver>> m_fftConvolvers;
 
    int m_numThreads;
 
    std::shared_ptr<ThreadPool> m_threadPool;
 
    std::vector<std::future<bool>> m_futures;
 
    std::mutex m_sumMutex;
 
    std::atomic_bool m_resetFlag;
 
    fftwf_complex* m_accBuffer;
 
    std::deque<fftwf_complex*> m_delayLine;
 
    int m_irLength;
 
    int m_tailCounter;
 
    bool m_eos;
 
    // delete copy constructor and operator=
    Convolver(const Convolver&) = delete;
    Convolver& operator=(const Convolver&) = delete;
 
public:
 
    Convolver(std::shared_ptr<std::vector<std::shared_ptr<std::vector<std::complex<sample_t>>>>> ir, int irLength, std::shared_ptr<ThreadPool> threadPool, std::shared_ptr<FFTPlan> plan);
 
    virtual ~Convolver();
 
    void getNext(sample_t* inBuffer, sample_t* outBuffer, int& length, bool& eos);
 
    void reset();
 
    std::shared_ptr<std::vector<std::shared_ptr<std::vector<std::complex<sample_t>>>>> getImpulseResponse();
 
    void setImpulseResponse(std::shared_ptr<std::vector<std::shared_ptr<std::vector<std::complex<sample_t>>>>> ir);
 
private:
 
    bool threadFunction(int id);
};
 
AUD_NAMESPACE_END