Do not use virtual methods in FFT class

[hercules2020/kcf.git] / src / fft_cufft.h

diff --git a/src/fft_cufft.h b/src/fft_cufft.h
index bbc64258920f1f506be2fbfc3657c7cb27251dd8..4241c0664f7cbe86bf4a836e305a92250110b34f 100644 (file)
--- a/src/fft_cufft.h
+++ b/src/fft_cufft.h
@@ -1,13 +1,12 @@
 #ifndef FFT_CUDA_H
 #define FFT_CUDA_H
 
-
 #include <cufft.h>
 #include <cuda_runtime.h>
 #include <cublas_v2.h>
 
 #include "fft.h"
-#include "cuda/cuda_error_check.cuh"
+#include "cuda_error_check.hpp"
 #include "pragmas.h"
 
 struct ThreadCtx;
@@ -16,16 +15,23 @@ class cuFFT : public Fft
 {
 public:
     cuFFT();
-    void init(unsigned width, unsigned height, unsigned num_of_feats, unsigned num_of_scales) override;
-    void set_window(const MatDynMem &window) override;
-    void forward(const MatDynMem &real_input, ComplexMat &complex_result) override;
-    void forward_window(MatDynMem &patch_feats_in, ComplexMat &complex_result, MatDynMem &tmp) override;
-    void inverse(ComplexMat &complex_input, MatDynMem &real_result) override;
-    ~cuFFT() override;
+    void init(unsigned width, unsigned height, unsigned num_of_feats, unsigned num_of_scales);
+    void set_window(const MatDynMem &window);
+    void forward(const MatScales &real_input, ComplexMat &complex_result);
+    void forward_window(MatScaleFeats &patch_feats_in, ComplexMat &complex_result, MatScaleFeats &tmp);
+    void inverse(ComplexMat &complex_input, MatScales &real_result);
+    ~cuFFT();
+
+protected:
+    cufftHandle create_plan_fwd(uint howmany) const;
+    cufftHandle create_plan_inv(uint howmany) const;
 
 private:
     cv::Mat m_window;
-    cufftHandle plan_f, plan_fw, plan_i_features, plan_i_1ch;
+    cufftHandle plan_f, plan_fw, plan_i_1ch;
+#ifdef BIG_BATCH
+    cufftHandle plan_f_all_scales, plan_fw_all_scales, plan_i_all_scales;
+#endif
     cublasHandle_t cublas;
 };