MHO_DelayRate.hh

Go to the documentation of this file.

#ifndef MHO_DelayRate_HH__
#define MHO_DelayRate_HH__
 
#include <cmath>
#include <complex>
 
#include "MHO_ContainerDefinitions.hh"
#include "MHO_TableContainer.hh"
 
#include "MHO_BinaryOperator.hh"
#include "MHO_CyclicRotator.hh"
#include "MHO_EndZeroPadder.hh"
#include "MHO_MultidimensionalFastFourierTransform.hh"
#include "MHO_SubSample.hh"
 
#ifdef HOPS_USE_FFTW3
    #include "MHO_MultidimensionalFastFourierTransformFFTW.hh"
#endif
 
namespace hops
{
 
class MHO_DelayRate: public MHO_BinaryOperator< visibility_type, weight_type, sbd_type >
{
    public:
        MHO_DelayRate();
        virtual ~MHO_DelayRate();
 
        void SetReferenceFrequency(double ref_freq) { fRefFreq = ref_freq; };
 
        int GetDelayRateSearchSpaceSize() const { return fDRSPSize; }
 
        unsigned int CalculateSearchSpaceSize(unsigned int input_size);
 
    protected:
        using XArgType1 = visibility_type;
        using XArgType2 = weight_type;
        using XArgType3 = sbd_type;
 
        virtual bool InitializeImpl(const XArgType1* in1, const XArgType2* in2, XArgType3* out) override;
        virtual bool ExecuteImpl(const XArgType1* in1, const XArgType2* in2, XArgType3* out) override;
 
    private:
        std::size_t fInDims[VIS_NDIM];
        std::size_t fOutDims[VIS_NDIM];
 
        void ApplyDataWeights(const XArgType2* in2, XArgType3* out);
        void ConditionallyResizeOutput(const std::size_t* dims, std::size_t size, XArgType3* out);
 
        void ApplyInterpolation(const XArgType1* in1, XArgType3* out);
 
#ifdef HOPS_USE_FFTW3
        using FFT_ENGINE_TYPE = MHO_MultidimensionalFastFourierTransformFFTW< visibility_type >;
#else
        using FFT_ENGINE_TYPE = MHO_MultidimensionalFastFourierTransform< visibility_type >;
#endif
 
        MHO_SubSample< sbd_type > fSubSampler;
        MHO_CyclicRotator< sbd_type > fCyclicRotator;
 
        MHO_EndZeroPadder< visibility_type > fZeroPadder;
        FFT_ENGINE_TYPE fFFTEngine;
 
        int fDRSPSize;
        double fRefFreq;
 
        bool fInitialized;
};
 
} // namespace hops
 
#endif