2019-05-07 01:05:16 +02:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <logger.hh>
|
|
|
|
|
|
|
|
#include <complex>
|
2019-08-08 14:47:07 +02:00
|
|
|
#include <map>
|
2019-05-07 01:05:16 +02:00
|
|
|
|
|
|
|
#if defined(USE_MPI)
|
|
|
|
#include <mpi.h>
|
|
|
|
#include <fftw3-mpi.h>
|
|
|
|
#else
|
|
|
|
#include <fftw3.h>
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef USE_SINGLEPRECISION
|
|
|
|
using real_t = float;
|
|
|
|
using complex_t = fftwf_complex;
|
|
|
|
#define FFTW_PREFIX fftwf
|
|
|
|
#else
|
|
|
|
using real_t = double;
|
|
|
|
using complex_t = fftw_complex;
|
|
|
|
#define FFTW_PREFIX fftw
|
|
|
|
#endif
|
|
|
|
|
2019-08-02 19:07:45 +02:00
|
|
|
enum class fluid_component { density, vx, vy, vz, dx, dy, dz };
|
|
|
|
enum class cosmo_species { dm, baryon, neutrino };
|
2019-08-08 14:47:07 +02:00
|
|
|
extern std::map<cosmo_species,std::string> cosmo_species_name;
|
2019-08-02 19:07:45 +02:00
|
|
|
|
2019-05-07 01:05:16 +02:00
|
|
|
using ccomplex_t = std::complex<real_t>;
|
|
|
|
|
|
|
|
#define FFTW_GEN_NAME_PRIM(a, b) a##_##b
|
|
|
|
#define FFTW_GEN_NAME(a, b) FFTW_GEN_NAME_PRIM(a, b)
|
|
|
|
#define FFTW_API(x) FFTW_GEN_NAME(FFTW_PREFIX, x)
|
|
|
|
|
|
|
|
using fftw_plan_t = FFTW_GEN_NAME(FFTW_PREFIX, plan);
|
|
|
|
|
|
|
|
#if defined(FFTW_MODE_PATIENT)
|
|
|
|
#define FFTW_RUNMODE FFTW_PATIENT
|
|
|
|
#elif defined(FFTW_MODE_MEASURE)
|
|
|
|
#define FFTW_RUNMODE FFTW_MEASURE
|
|
|
|
#else
|
|
|
|
#define FFTW_RUNMODE FFTW_ESTIMATE
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#if defined(USE_MPI)
|
|
|
|
inline double get_wtime()
|
|
|
|
{
|
|
|
|
return MPI_Wtime();
|
|
|
|
}
|
2019-05-12 17:39:15 +02:00
|
|
|
|
|
|
|
inline int MPI_Get_rank( void ){
|
|
|
|
int rank, ret;
|
|
|
|
ret = MPI_Comm_rank(MPI_COMM_WORLD, &rank);
|
|
|
|
assert( ret==MPI_SUCCESS );
|
|
|
|
return rank;
|
|
|
|
}
|
|
|
|
|
|
|
|
inline int MPI_Get_size( void ){
|
|
|
|
int size, ret;
|
|
|
|
ret = MPI_Comm_size(MPI_COMM_WORLD, &size);
|
|
|
|
assert( ret==MPI_SUCCESS );
|
|
|
|
return size;
|
|
|
|
}
|
2019-05-13 09:56:58 +02:00
|
|
|
|
|
|
|
template<typename T>
|
|
|
|
MPI_Datatype GetMPIDatatype( void )
|
|
|
|
{
|
|
|
|
if( typeid(T) == typeid(std::complex<float>) )
|
|
|
|
return MPI_COMPLEX;
|
|
|
|
|
|
|
|
if( typeid(T) == typeid(std::complex<double>) )
|
|
|
|
return MPI_DOUBLE_COMPLEX;
|
|
|
|
|
|
|
|
if( typeid(T) == typeid(int) )
|
|
|
|
return MPI_INT;
|
|
|
|
|
|
|
|
if( typeid(T) == typeid(unsigned) )
|
|
|
|
return MPI_UNSIGNED;
|
|
|
|
|
|
|
|
if( typeid(T) == typeid(float) )
|
|
|
|
return MPI_FLOAT;
|
|
|
|
|
|
|
|
if( typeid(T) == typeid(double) )
|
|
|
|
return MPI_DOUBLE;
|
|
|
|
|
|
|
|
if( typeid(T) == typeid(char) )
|
|
|
|
return MPI_CHAR;
|
|
|
|
|
|
|
|
abort();
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-05-07 01:05:16 +02:00
|
|
|
#else
|
|
|
|
#if defined(_OPENMP)
|
|
|
|
#include <omp.h>
|
|
|
|
inline double get_wtime()
|
|
|
|
{
|
|
|
|
return omp_get_wtime();
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
#include <ctime>
|
|
|
|
inline double get_wtime()
|
|
|
|
{
|
|
|
|
return std::clock() / double(CLOCKS_PER_SEC);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
|
2019-08-05 13:22:59 +02:00
|
|
|
inline void multitask_sync_barrier( void )
|
|
|
|
{
|
|
|
|
#if defined(USE_MPI)
|
|
|
|
MPI_Barrier( MPI_COMM_WORLD );
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2019-05-13 09:56:58 +02:00
|
|
|
|
2019-05-12 21:12:07 +02:00
|
|
|
|
2019-05-07 01:05:16 +02:00
|
|
|
namespace CONFIG
|
|
|
|
{
|
|
|
|
extern int MPI_thread_support;
|
|
|
|
extern int MPI_task_rank;
|
|
|
|
extern int MPI_task_size;
|
|
|
|
extern bool MPI_ok;
|
|
|
|
extern bool MPI_threads_ok;
|
|
|
|
extern bool FFTW_threads_ok;
|
2019-05-22 22:25:27 +02:00
|
|
|
extern int num_threads;
|
2019-05-14 16:33:18 +02:00
|
|
|
} // namespace CONFIG
|