Diffusion µSpectre (bugfix/T2542) Browse

T2542

fftwmpi_engine.hh
View Options

			/**
			* @file fftwmpi_engine.hh
			*
			* @author Lars Pastewka <lars.pastewka@imtek.uni-freiburg.de>
			*
			* @date 06 Mar 2017
			*
			* @brief FFT engine using MPI-parallel FFTW
			*
			* Copyright © 2017 Till Junge
			*
			* µSpectre 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, or (at
			* your option) any later version.
			*
			* µSpectre 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
			* General Public License for more details.
			*
			* You should have received a copy of the GNU Lesser General Public License
			* along with µSpectre; see the file COPYING. If not, write to the
			* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
			* * Boston, MA 02111-1307, USA.
			*
			* Additional permission under GNU GPL version 3 section 7
			*
			* If you modify this Program, or any covered work, by linking or combining it
			* with proprietary FFT implementations or numerical libraries, containing parts
			* covered by the terms of those libraries' licenses, the licensors of this
			* Program grant you additional permission to convey the resulting work.
			*/

			#ifndef SRC_FFT_FFTWMPI_ENGINE_HH_
			#define SRC_FFT_FFTWMPI_ENGINE_HH_

			#include "fft/fft_engine_base.hh"

			#include <fftw3-mpi.h>

			namespace muSpectre {

			/**
			* implements the `muSpectre::FFTEngineBase` interface using the
			* FFTW library
			*/
			template <Dim_t Dim>
			class FFTWMPIEngine : public FFTEngineBase<Dim> {
			public:
			using Parent = FFTEngineBase<Dim>; //!< base class
			using Ccoord = typename Parent::Ccoord; //!< cell coordinates type
			//! field for Fourier transform of second-order tensor
			using Workspace_t = typename Parent::Workspace_t;
			//! real-valued second-order tensor
			using Field_t = typename Parent::Field_t;
			//! Default constructor
			FFTWMPIEngine() = delete;

			//! Constructor with system resolutions
			FFTWMPIEngine(Ccoord resolutions, Dim_t nb_components,
			Communicator comm = Communicator());

			//! Copy constructor
			FFTWMPIEngine(const FFTWMPIEngine & other) = delete;

			//! Move constructor
			FFTWMPIEngine(FFTWMPIEngine && other) = default;

			//! Destructor
			virtual ~FFTWMPIEngine() noexcept;

			//! Copy assignment operator
			FFTWMPIEngine & operator=(const FFTWMPIEngine & other) = delete;

			//! Move assignment operator
			FFTWMPIEngine & operator=(FFTWMPIEngine && other) = default;

			// compute the plan, etc
			void initialise(FFT_PlanFlags plan_flags) override;

			//! forward transform
			Workspace_t & fft(Field_t & field) override;

			//! inverse transform
			void ifft(Field_t & field) const override;

			protected:
			static int
			nb_engines; //!< number of times this engine has been instatiated
			fftw_plan plan_fft{}; //!< holds the plan for forward fourier transform
			fftw_plan plan_ifft{}; //!< holds the plan for inverse fourier transform
			ptrdiff_t
			workspace_size{}; //!< size of workspace buffer returned by planner
			Real * real_workspace{}; //!< temporary real workspace that is correctly
			//!< padded
			bool initialised{false}; //!< to prevent double initialisation
			};

			} // namespace muSpectre

			#endif // SRC_FFT_FFTWMPI_ENGINE_HH_

muSpectre/src/fft/fftwmpi_engine.hh676022f5dfa6bugfix/T2542

fftwmpi_engine.hhView Options

muSpectre/src/fft/fftwmpi_engine.hh
676022f5dfa6bugfix/T2542

fftwmpi_engine.hh
View Options