netcdf-mpi

fftransform_cufft.hh
View Options

			/**
			*
			* @author Lucas Frérot <lucas.frerot@epfl.ch>
			*
			* @section LICENSE
			*
			* Copyright (©) 2016 EPFL (Ecole Polytechnique Fédérale de
			* Lausanne) Laboratory (LSMS - Laboratoire de Simulation en Mécanique des
			* Solides)
			*
			* Tamaas is free software: you can redistribute it and/or modify it under the
			* terms of the GNU Lesser General Public License as published by the Free
			* Software Foundation, either version 3 of the License, or (at your option) any
			* later version.
			*
			* Tamaas is distributed in the hope that it will be useful, but WITHOUT ANY
			* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
			* A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
			* details.
			*
			* You should have received a copy of the GNU Lesser General Public License
			* along with Tamaas. If not, see <http://www.gnu.org/licenses/>.
			*
			*/

			/* -------------------------------------------------------------------------- */

			#ifndef FFTRANSFORM_CUFFT_HH
			#define FFTRANSFORM_CUFFT_HH

			#include "fftransform.hh"
			#include "grid.hh"
			#include <cufft.h>

			__BEGIN_TAMAAS__

			template <typename T, UInt dim>
			class FFTransformCUFFT : public FFTransform<T, 2> {
			public:
			/// Constructor
			FFTransformCUFFT(Grid<T, dim> & real,
			GridHermitian<T, dim> & spectral);

			/// Destructor
			virtual ~FFTransformCUFFT();

			public:
			/// Perform FFT
			void forwardTransform() override;

			/// Perform IFFT
			void backwardTransform() override;

			private:
			cufftHandle forward_plan;
			cufftHandle backward_plan;
			};

			__END_TAMAAS__

			#endif // FFTRANSFORM_CUFFT_HH