BackendFFT.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "BackendBase.h"
#include "Parameters.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CTensor;
 
        template <typename T>
        class FL_EXPORT CBackendFFT : public CBackendBase<T>
        {
        public:
            CBackendFFT();
            CBackendFFT(const CBackendFFT<T>& bf);
            virtual ~CBackendFFT();
 
            virtual const CResult SetFFTParams();
 
            virtual bool CheckFFTSize(const std::vector<int64_t>& vctShape) const;
            virtual T GetNormValue(ENormFFT eNormFFT, bool bFFT, int64_t i64NormBase) const;
            virtual const CResult FFT(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult, const std::vector<int64_t>* pVctResultShape, T tNorm);
            virtual const CResult IFFT(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult, const std::vector<int64_t>* pVctResultShape, T tNorm);
            virtual const CResult ShiftFourier(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult, bool bInverse);
            virtual int64_t GetRequiredTemporaryMemory(const std::vector<int64_t>& vctShape) const;
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CBackendFFT<T>, *this);
 
        protected:
            virtual bool CheckFirstTwo(int64_t i64Size) const;
            virtual void CalculateReverseOrder(int64_t* pI64Order, int64_t i64Length, bool bFirstTwo);
            virtual void CalibrationInterval(T* pTRealVal, T* pTImaginVal, int64_t i64Length, bool bInverse);
            virtual int64_t GetClosestPowerTwo(int64_t BaseNum);
            virtual void MakeGBufferWidth(T* pTSource, T* pTRFTG, T* pTRealValWidth, T* pTImaginValWidth, int64_t i64OprWidth, int64_t i64DFTWidth, bool bSrcReal);
            virtual void MakeHBuffer(T* pTRFTH, T* pTRealValWidth, T* pTImaginValWidth, int64_t i64OprWidth, int64_t i64DFTWidth);
            virtual void SetRFTTwo(T* pTSource, T* pTRFT, T* pTRFTEnd, int64_t* pI64OrderWidth, bool bSrcReal, T tNorm);
            virtual void SetRFTFour(T* pTSource, T* pTRFT, T* pTRFTEnd, int64_t* pI64OrderWidth, bool bSrcReal, T tNorm);
            virtual void SetRFTFourI(T* pTSource, T* pTRFT, T* pTRFTEnd, int64_t* pI64OrderWidth, bool bSrcReal, T tNorm);
            virtual void RFTCalc(T* pRFTBuffer, T* pTBufEnd, T* pTRealValWidth, T* pTImaginValWidth, int64_t i64CumulN, int64_t i64CycleLength);
            virtual void RFTCalcI(T* pRFTBuffer, T* pTBufEnd, T* pTRealValWidth, T* pTImaginValWidth, int64_t i64CumulN, int64_t i64CycleLength);
            virtual void MakeGBufferHeight(T* pTSource, T* pTGBuffer, int64_t* pI64OrderHeight, T* pTRealValHeight, T* pTImaginValHeight, int64_t i64BatchRowCount, int64_t i64OprHeight, int64_t i64OprWidth, int64_t i64DFTHeight);
            virtual const CResult FFTHeightLoop(T* pTResultMemory, T* pTRealValHeight, T* pTImaginValHeight, int64_t i64BatchRowCount, int64_t i64DFTHeight, int64_t i64OprWidth, bool bHeightFirstTwo);
            virtual const CResult IFFTHeightLoop(T* pTResultMemory, T* pTRealValHeight, T* pTImaginValHeight, int64_t i64BatchRowCount, int64_t i64DFTHeight, int64_t i64OprWidth, bool bHeightFirstTwo);
 
            virtual const CResult FFTAssignNorm(T* pTSourceMemory, T* pTResultMemory, int64_t* pI64OrderHeight, int64_t i64BatchRowCount, int64_t i64DFTHeight, int64_t i64DFTWidth, bool bOprReal, bool bOrderHeight, T tNorm);
            virtual const CResult FFTPowerTwo(T* pTSourceMemory, T* pTRFTBuffer, T* pTResultMemory, int64_t* pI64OrderWidth, int64_t* pI64OrderHeight, T* pTRealValWidth, T* pTImaginValWidth, int64_t i64BatchRowCount, int64_t i64DFTHeight, int64_t i64DFTWidth, bool bWidthFirstTwo, bool bOprReal, bool bOrderHeight, T tNorm);
            virtual const CResult IFFTPowerTwo(T* pTSourceMemory, T* pTRFTBuffer, T* pTResultMemory, int64_t* pI64OrderWidth, int64_t* pI64OrderHeight, T* pTRealValWidth, T* pTImaginValWidth, int64_t i64BatchRowCount, int64_t i64DFTHeight, int64_t i64DFTWidth, bool bWidthFirstTwo, bool bOprReal, bool bOrderHeight, T tNorm);
            virtual const CResult FFTChirpWidth(T* pTSourceMemory, T* pTRFTBuffer, T* pTHBuffer, T* pTResultMemory, int64_t* pI64OrderWidth, int64_t* pI64OrderHeight, T* pTRealValWidth, T* pTImaginValWidth, int64_t i64BatchRowCount, int64_t i64OprHeight, int64_t i64OprWidth, int64_t i64DFTWidth, bool bWidthFirstTwo, bool bOprReal, bool bOrderHeight, T tNorm);
            virtual const CResult IFFTChirpWidth(T* pTSourceMemory, T* pTRFTBuffer, T* pTHBuffer, T* pTResultMemory, int64_t* pI64OrderWidth, int64_t* pI64OrderHeight, T* pTRealValWidth, T* pTImaginValWidth, int64_t i64BatchRowCount, int64_t i64OprHeight, int64_t i64OprWidth, int64_t i64DFTWidth, bool bWidthFirstTwo, bool bOprReal, bool bOrderHeight, T tNorm);
            virtual const CResult FFTChirpHeight(T* pTResultMemory, int64_t* pI64OrderHeight, T* pTRealValHeight, T* pTImaginValHeight, T* pTHeightGBuffer, T* pTHeightTBuffer, T* pTHBuffer, int64_t i64BatchChannel, int64_t i64OprHeight, int64_t i64OprWidth, int64_t i64DFTHeight, bool bHeightFirstTwo);
            virtual const CResult IFFTChirpHeight(T* pTResultMemory, int64_t* pI64OrderHeight, T* pTRealValHeight, T* pTImaginValHeight, T* pTHeightGBuffer, T* pTHeightTBuffer, T* pTHBuffer, int64_t i64BatchChannel, int64_t i64OprHeight, int64_t i64OprWidth, int64_t i64DFTHeight, bool bHeightFirstTwo);
        };
    }
}
 
#endif