carma_fft.h

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//  This file is part of COMPASS <https://anr-compass.github.io/compass/>
//
//  Copyright (C) 2011-2023 COMPASS Team <https://github.com/ANR-COMPASS>
//  All rights reserved.
 
// -----------------------------------------------------------------------------
 
 
 
#ifndef _CARMA_FFT_H_
#define _CARMA_FFT_H_
 
#include <carma_obj.h>
#include <cufft.h>
 
template <class T_in, class T_out>
class CarmaFFT {
 protected:
  CarmaObj<T_in> *d_input;    
  CarmaObj<T_out> *d_output;  
  cufftHandle plan;            
  cufftType type_plan;             
  int inplace;  
 
 public:
  CarmaFFT(long *dims_data, int inplace);
  ~CarmaFFT();
 
  int host2device(T_in *data);
  int device2host(T_out *data);
  int compute(int dir);
  int compute(T_in *input, T_out *output, int dir);
};
 
typedef CarmaFFT<cuFloatComplex, cuFloatComplex> caFFT_C2C;
typedef CarmaFFT<cufftReal, cuFloatComplex> caFFT_R2C;
typedef CarmaFFT<cuFloatComplex, cufftReal> caFFT_C2R;
 
typedef CarmaFFT<cuDoubleComplex, cuDoubleComplex> caFFT_Z2Z;
typedef CarmaFFT<cufftDoubleReal, cuDoubleComplex> caFFT_D2Z;
typedef CarmaFFT<cuDoubleComplex, cufftDoubleReal> caFFT_Z2D;
 
#endif  // _CARMA_FFT_H_