cfft2d_calc_block.hpp

/* fft_calc_block.hpp */
/* $Id$ */
#ifndef __cfft2d_calc_block_hpp
#define __cfft2d_calc_block_hpp

//#ifdef U_CUDA_ENABLED

#ifdef GE_BUILD
#include <calc_block.hpp>
#else
#include <dcf/calc_block.hpp>
#endif

namespace dcf
{
  class CFFT2DCalcBlock : public CalcBlock
  {
  public:
    
    CFFT2DCalcBlock(const dcf::Manager &manager, int direction = 1, bool normalize = true);
    
    virtual ~CFFT2DCalcBlock();
    
    virtual void start();
      
    void setData(size_t dim_x, size_t dim_y, float *re_data, float *im_data);
    
    void setDirection(int dir) {m_data.dir = dir;}
    
    int getDirection() const {return m_data.dir;}
    
    bool isNormalize() const {return m_data.normalize;}
    
    void setNormalize(bool normalize) {m_data.normalize = normalize;}
    
    void calcModule(size_t dim_x, size_t dim_y, float *re, float *im, float *mod);
    
  protected:
    
    struct
    {
      size_t dim_x, dim_y;      
      bool new_plan;            
      float *re;                
      float *im;                
      int dir;                  
      bool normalize;           
    } m_data;
            
    virtual void run();
    
    bool cuda_fft();
    
    dcf::error_t select_device();
    
    dcf::error_t free_device();
    
    void cuda_set_data(size_t dim_x, size_t dim_y, float *re_data, float *im_data);
  
  private:
    
  };
};

//#endif /* U_CUDA_ENABLED */

#endif /* cfft2d_calc_block.hpp */