OPAL (Object Oriented Parallel Accelerator Library) 2024.2
OPAL
FFT< RCTransform, Dim, T > Class Template Reference

#include <FFT.h>

Inheritance diagram for FFT< RCTransform, Dim, T >:
Collaboration diagram for FFT< RCTransform, Dim, T >:

Public Types

typedef FieldLayout< DimLayout_t
typedef BareField< T, DimRealField_t
typedef LField< T, DimRealLField_t
typedef std::complex< T > Complex_t
typedef BareField< Complex_t, DimComplexField_t
typedef LField< Complex_t, DimComplexLField_t
typedef FFTBase< Dim, T >::Domain_t Domain_t
enum  
enum  FFT_e
typedef T Precision_t
typedef NDIndex< DimDomain_t
typedef FFTPACK< T > InternalFFT_t

Public Member Functions

 FFT (const Domain_t &rdomain, const Domain_t &cdomain, const bool transformTheseDims[Dim], const bool &compressTemps=false)
 FFT (const Domain_t &rdomain, const Domain_t &cdomain, const bool &compressTemps=false, int serialAxes=1)
 ~FFT (void)
void transform (int direction, RealField_t &f, ComplexField_t &g, const bool &constInput=false)
void transform (const char *directionName, RealField_t &f, ComplexField_t &g, const bool &constInput=false)
void transform (int direction, ComplexField_t &f, RealField_t &g, const bool &constInput=false)
void transform (const char *directionName, ComplexField_t &f, RealField_t &g, const bool &constInput=false)
void write (std::ostream &out) const
void setDirectionName (int direction, const char *directionName)
void setNormFact (Precision_t nf)
int transVnodes () const

Protected Member Functions

int getDirection (const char *directionName) const
 translate direction name string into dimension number
bool transformDim (unsigned d) const
 query whether this dimension is to be transformed
unsigned numTransformDims (void) const
 query number of transform dimensions
unsigned activeDimension (unsigned d) const
 get dimension number from list of transformed dimensions
InternalFFT_tgetEngine (void)
 access the internal FFT Engine
Precision_tgetNormFact (void)
 get the FFT normalization factor
const Domain_tgetDomain (void) const
 get our domain
bool checkDomain (const Domain_t &dom1, const Domain_t &dom2) const
 compare indexes of two domains
bool compressTemps (void) const
 do we compress temps?

Static Protected Attributes

static GuardCellSizes< DimnullGC
 null GuardCellSizes object for checking BareField arguments to transform

Private Member Functions

void setup (void)

Private Attributes

Layout_t ** tempLayouts_m
Layout_ttempRLayout_m
ComplexField_t ** tempFields_m
RealField_ttempRField_m
Domain_t complexDomain_m
int serialAxes_m
std::map< const char *, int > directions_m
 Stores user-defined names for FFT directions:
FFT_e transformType_m
 Indicates which type of transform we do.
bool transformDims_m [Dim]
 Indicates which dimensions are transformed.
unsigned nTransformDims_m
 Stores the number of dims to be transformed.
unsigned * activeDims_m
 Stores the numbers of these dims (0,1,2).
InternalFFT_t FFTEngine_m
 Internal FFT object for performing serial FFTs.
Precision_t normFact_m
 Normalization factor:
Domain_t Domain_m
 Domain of the input field, mainly used to check axis sizes and ordering, former const Domain_t& Domain_m;.
bool compressTempFields_m
 Switch to turn on/off compression of intermediate Fields (tempFields) as algorithm is finished with them.

Detailed Description

template<size_t Dim, class T>
class FFT< RCTransform, Dim, T >

real-to-complex FFT class

Definition at line 395 of file FFT.h.

Member Typedef Documentation

◆ Complex_t

template<size_t Dim, class T>
typedef std::complex<T> FFT< RCTransform, Dim, T >::Complex_t

Definition at line 405 of file FFT.h.

◆ ComplexField_t

template<size_t Dim, class T>
typedef BareField<Complex_t,Dim> FFT< RCTransform, Dim, T >::ComplexField_t

Definition at line 406 of file FFT.h.

◆ ComplexLField_t

template<size_t Dim, class T>
typedef LField<Complex_t,Dim> FFT< RCTransform, Dim, T >::ComplexLField_t

Definition at line 407 of file FFT.h.

◆ Domain_t [1/2]

template<size_t Dim, class T>
typedef FFTBase<Dim,T>::Domain_t FFT< RCTransform, Dim, T >::Domain_t

Definition at line 408 of file FFT.h.

◆ Domain_t [2/2]

typedef NDIndex<Dim> FFTBase< Dim, T >::Domain_t

Definition at line 60 of file FFTBase.h.

◆ InternalFFT_t

typedef FFTPACK<T> FFTBase< Dim, T >::InternalFFT_t

Definition at line 66 of file FFTBase.h.

◆ Layout_t

template<size_t Dim, class T>
typedef FieldLayout<Dim> FFT< RCTransform, Dim, T >::Layout_t

Definition at line 402 of file FFT.h.

◆ Precision_t

typedef T FFTBase< Dim, T >::Precision_t

Definition at line 59 of file FFTBase.h.

◆ RealField_t

template<size_t Dim, class T>
typedef BareField<T,Dim> FFT< RCTransform, Dim, T >::RealField_t

Definition at line 403 of file FFT.h.

◆ RealLField_t

template<size_t Dim, class T>
typedef LField<T,Dim> FFT< RCTransform, Dim, T >::RealLField_t

Definition at line 404 of file FFT.h.

Member Enumeration Documentation

◆ anonymous enum

anonymous enum

Definition at line 58 of file FFTBase.h.

◆ FFT_e

Definition at line 63 of file FFTBase.h.

Constructor & Destructor Documentation

◆ FFT() [1/2]

template<size_t Dim, class T>
FFT< RCTransform, Dim, T >::FFT ( const Domain_t & rdomain,
const Domain_t & cdomain,
const bool transformTheseDims[Dim],
const bool & compressTemps = false )

Create a new FFT object with the given domains for input/output Fields Specify which dimensions to transform along. Optional argument compress indicates whether or not to compress temporary Fields in between uses.

References compressTemps(), and Dim.

Referenced by FFT(), setup(), and ~FFT().

Here is the call graph for this function:

◆ FFT() [2/2]

template<size_t Dim, class T>
FFT< RCTransform, Dim, T >::FFT ( const Domain_t & rdomain,
const Domain_t & cdomain,
const bool & compressTemps = false,
int serialAxes = 1 )

Same as above, but transform all dims:

References compressTemps(), and FFT().

Here is the call graph for this function:

◆ ~FFT()

template<size_t Dim, class T>
FFT< RCTransform, Dim, T >::~FFT ( void )

Definition at line 922 of file FFT.hpp.

References FFT(), numTransformDims(), tempFields_m, tempLayouts_m, tempRField_m, and tempRLayout_m.

Here is the call graph for this function:

Member Function Documentation

◆ activeDimension()

unsigned FFTBase< Dim, T >::activeDimension ( unsigned d) const
inlineprotected

get dimension number from list of transformed dimensions

get dimension number from list of transformed dimensions

Parameters
d
Returns

Definition at line 151 of file FFTBase.h.

Referenced by setup().

◆ checkDomain()

bool FFTBase< Dim, T >::checkDomain ( const Domain_t & dom1,
const Domain_t & dom2 ) const
inlineprotected

compare indexes of two domains

helper function for comparing domains

Parameters
Dim
dom1
Dim
dom2
Returns

Definition at line 163 of file FFTBase.h.

◆ compressTemps()

bool FFTBase< Dim, T >::compressTemps ( void ) const
inlineprotected

do we compress temps?

Definition at line 166 of file FFTBase.h.

Referenced by FFT(), and FFT().

◆ getDirection()

int FFTBase< Dim, T >::getDirection ( const char * directionName) const
inlineprotected

translate direction name string into dimension number

Translate direction name string into dimension number

Parameters
directionName
Returns

Definition at line 142 of file FFTBase.h.

◆ getDomain()

const Domain_t & FFTBase< Dim, T >::getDomain ( void ) const
inlineprotected

get our domain

Definition at line 160 of file FFTBase.h.

Referenced by setup().

◆ getEngine()

InternalFFT_t & FFTBase< Dim, T >::getEngine ( void )
inlineprotected

access the internal FFT Engine

Definition at line 154 of file FFTBase.h.

◆ getNormFact()

Precision_t & FFTBase< Dim, T >::getNormFact ( void )
inlineprotected

get the FFT normalization factor

Definition at line 157 of file FFTBase.h.

◆ numTransformDims()

unsigned FFTBase< Dim, T >::numTransformDims ( void ) const
inlineprotected

query number of transform dimensions

Definition at line 148 of file FFTBase.h.

Referenced by setup(), and ~FFT().

◆ setDirectionName()

void FFTBase< Dim, T >::setDirectionName ( int direction,
const char * directionName )
inline

Allow the user to name the transform directions, for code clarity.

Parameters
direction
directionName

Allow the user to name the transform directions, for code clarity. Typical values might be "x_to_k", "k_to_x", "t_to_omega", "omega_to_t"

Definition at line 110 of file FFTBase.h.

◆ setNormFact()

void FFTBase< Dim, T >::setNormFact ( Precision_t nf)
inline

Set the FFT normalization factor (to something other than the default)

Parameters
nf

Definition at line 117 of file FFTBase.h.

◆ setup()

template<size_t Dim, class T>
void FFT< RCTransform, Dim, T >::setup ( void )
private

complex-to-real FFT on GPU: pass pointer to GPU memory where complex field is stored, do the inverse FFT and transfer real field back to host memory setup performs all the initializations necessary after the transform directions have been specified.

Definition at line 785 of file FFT.hpp.

References activeDimension(), complexDomain_m, Dim, FFT(), getDomain(), numTransformDims(), PARALLEL, PAssert_GT, PAssert_LT, PInsist, SERIAL, serialAxes_m, tempFields_m, tempLayouts_m, tempRField_m, tempRLayout_m, and transVnodes().

Referenced by transform().

Here is the call graph for this function:

◆ transform() [1/4]

template<size_t Dim, class T>
void FFT< RCTransform, Dim, T >::transform ( const char * directionName,
ComplexField_t & f,
RealField_t & g,
const bool & constInput = false )

References setup().

Here is the call graph for this function:

◆ transform() [2/4]

template<size_t Dim, class T>
void FFT< RCTransform, Dim, T >::transform ( const char * directionName,
RealField_t & f,
ComplexField_t & g,
const bool & constInput = false )

◆ transform() [3/4]

template<size_t Dim, class T>
void FFT< RCTransform, Dim, T >::transform ( int direction,
ComplexField_t & f,
RealField_t & g,
const bool & constInput = false )

real-to-complex FFT on GPU: transfer the real field to GPU execute FFT return the pointer to memory on GPU where complex results are stored complex-to-real FFT Same as above, but with input and output field types reversed.

◆ transform() [4/4]

template<size_t Dim, class T>
void FFT< RCTransform, Dim, T >::transform ( int direction,
RealField_t & f,
ComplexField_t & g,
const bool & constInput = false )

real-to-complex FFT: specify +1 or -1 to indicate forward or inverse transform, or specify the user-defined name string for the direction. Supply a second BareField to store the output. optional argument constInput indicates whether or not to treat the input Field argument f as const. If not, we can use it as a temporary in order to avoid an additional data transpose.

◆ transformDim()

bool FFTBase< Dim, T >::transformDim ( unsigned d) const
inlineprotected

query whether this dimension is to be transformed

query whether this dimension is to be transformed

Parameters
d
Returns

Definition at line 145 of file FFTBase.h.

◆ transVnodes()

int FFTBase< Dim, T >::transVnodes ( ) const
inline

Utility to determine the number of vnodes to use in temporary transpose fields; this is either -1, or a limited number set on the command line

Returns

Definition at line 125 of file FFTBase.h.

Referenced by setup().

◆ write()

void FFTBase< Dim, T >::write ( std::ostream & out) const

I/O for FFT object

Parameters
out

Definition at line 102 of file FFTBase.hpp.

Member Data Documentation

◆ activeDims_m

unsigned* FFTBase< Dim, T >::activeDims_m
private

Stores the numbers of these dims (0,1,2).

Definition at line 176 of file FFTBase.h.

◆ complexDomain_m

template<size_t Dim, class T>
Domain_t FFT< RCTransform, Dim, T >::complexDomain_m
private

domain of the resulting complex fields const Domain_t& complexDomain_m;

Definition at line 490 of file FFT.h.

Referenced by setup().

◆ compressTempFields_m

bool FFTBase< Dim, T >::compressTempFields_m
private

Switch to turn on/off compression of intermediate Fields (tempFields) as algorithm is finished with them.

Definition at line 188 of file FFTBase.h.

◆ directions_m

std::map<const char*,int> FFTBase< Dim, T >::directions_m
private

Stores user-defined names for FFT directions:

Definition at line 171 of file FFTBase.h.

◆ Domain_m

Domain_t FFTBase< Dim, T >::Domain_m
private

Domain of the input field, mainly used to check axis sizes and ordering, former const Domain_t& Domain_m;.

Definition at line 185 of file FFTBase.h.

◆ FFTEngine_m

InternalFFT_t FFTBase< Dim, T >::FFTEngine_m
private

Internal FFT object for performing serial FFTs.

Definition at line 179 of file FFTBase.h.

◆ normFact_m

Precision_t FFTBase< Dim, T >::normFact_m
private

Normalization factor:

Definition at line 182 of file FFTBase.h.

◆ nTransformDims_m

unsigned FFTBase< Dim, T >::nTransformDims_m
private

Stores the number of dims to be transformed.

Definition at line 175 of file FFTBase.h.

◆ nullGC

GuardCellSizes<Dim> FFTBase< Dim, T >::nullGC
staticprotected

null GuardCellSizes object for checking BareField arguments to transform

! These members are used by the derived FFT classes

Definition at line 139 of file FFTBase.h.

◆ serialAxes_m

template<size_t Dim, class T>
int FFT< RCTransform, Dim, T >::serialAxes_m
private

number of axes to make serial

Definition at line 495 of file FFT.h.

Referenced by setup().

◆ tempFields_m

template<size_t Dim, class T>
ComplexField_t** FFT< RCTransform, Dim, T >::tempFields_m
private

The array of temporary fields, one for each transform direction These use the corresponding tempLayouts.

Definition at line 479 of file FFT.h.

Referenced by setup(), and ~FFT().

◆ tempLayouts_m

template<size_t Dim, class T>
Layout_t** FFT< RCTransform, Dim, T >::tempLayouts_m
private

How the temporary fields are laid out; these are computed from the input Field's domain. This will be allocated as an array of FieldLayouts with nTransformDims elements. Each is SERIAL along the zeroth dimension and the axes are permuted so that the transform direction is first

Definition at line 469 of file FFT.h.

Referenced by setup(), and ~FFT().

◆ tempRField_m

template<size_t Dim, class T>
RealField_t* FFT< RCTransform, Dim, T >::tempRField_m
private

We need one real internal Field in this case.

Definition at line 484 of file FFT.h.

Referenced by setup(), and ~FFT().

◆ tempRLayout_m

template<size_t Dim, class T>
Layout_t* FFT< RCTransform, Dim, T >::tempRLayout_m
private

extra layout for the one real Field needed

Definition at line 474 of file FFT.h.

Referenced by setup(), and ~FFT().

◆ transformDims_m

bool FFTBase< Dim, T >::transformDims_m[Dim]
private

Indicates which dimensions are transformed.

Definition at line 174 of file FFTBase.h.

◆ transformType_m

FFT_e FFTBase< Dim, T >::transformType_m
private

Indicates which type of transform we do.

Definition at line 173 of file FFTBase.h.


The documentation for this class was generated from the following files: