26 #ifndef LIBRARIES_RECONSTRUCTION_SHIFT_CORR_ESTIMATOR_H_ 27 #define LIBRARIES_RECONSTRUCTION_SHIFT_CORR_ESTIMATOR_H_ 55 bool includingBatchFT,
bool includingSingleFT,
56 bool allowDataOverwrite)
override;
66 std::complex<T> *othersF,
77 std::complex<T> *inOut,
78 const std::complex<T> *ref,
80 bool center)
override;
84 std::complex<T> *inOut,
85 const std::complex<T> *ref,
89 sComputeCorrelations2DOneToN<true>(hw, inOut, ref, dims);
91 sComputeCorrelations2DOneToN<false>(hw, inOut, ref, dims);
98 std::complex<T> *inOut,
99 const std::complex<T> *ref,
109 std::complex<T> *m_single_FD;
110 std::complex<T> *m_batch_FD;
118 void init2DOneToN()
override;
119 void check()
override;
120 void setDefault()
override;
virtual ~ShiftCorrEstimator()
void load2DReferenceOneToN(const std::complex< T > *ref) override
void init2D(const std::vector< HW *> &hw, AlignType type, const FFTSettings< T > &dims, size_t maxShift, bool includingBatchFT, bool includingSingleFT, bool allowDataOverwrite) override
void computeCorrelations2DOneToN(std::complex< T > *inOut, bool center) override
HW & getHW() const override
static std::vector< Point2D< float > > computeShifts2DOneToN(const CPU &cpu, std::complex< T > *othersF, T *othersS, std::complex< T > *ref, const FFTSettings< T > &settings, void *plan, size_t maxShift)
void computeShift2DOneToN(T *others) override
static void sComputeCorrelations2DOneToN(const HW &hw, std::complex< T > *inOut, const std::complex< T > *ref, const Dimensions &dims, bool center)