Homec4science
Diffusion tamaas (archive/legacy_mpi) Browse

tamaas/src/core/tamaas.hh
b7fa37a3b7a3archive/legacy_mpi

Last Change
Raw File

tamaas.hh
View Options

/**
* @mainpage Welcome to Tamaas !
*
* @section Introduction
* Tamaas is a spectral-boundary-element based contact library. It is made with
* love to be fast and friendly!
*
* @author Guillaume Anciaux <guillaume.anciaux@epfl.ch>
* @author Lucas Frérot <lucas.frerot@epfl.ch>
* @author Valentine Rey <valentine.rey@epfl.ch>
*
* @section LICENSE
*
* Copyright (©) 2016-2017 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 __TAMAAS_HH__
#define __TAMAAS_HH__
/* -------------------------------------------------------------------------- */
// Standard includes
#include <exception>
#include <iostream>
#include <memory>
#include <string>
/* -------------------------------------------------------------------------- */
// Special thrust includes
#include <thrust/complex.h>
#include <thrust/random.h>
#ifdef USE_CUDA
#include "unified_allocator.hh"
#else
#include "fftw_allocator.hh"
#endif
/// @section Namespace macros
#define __BEGIN_TAMAAS__ namespace tamaas {
#define __END_TAMAAS__ }
/// @section Cuda specific definitions
#define CUDA_LAMBDA __device__ __host__
#ifdef USE_CUDA
#define DEFAULT_ALLOCATOR UnifiedAllocator<T>
#else
#define DEFAULT_ALLOCATOR FFTWAllocator<T>
#endif
/* -------------------------------------------------------------------------- */
/// @section Standard definitions pulled from C++14
namespace std {
#if __cplusplus < 201402L
/// exchange
template <class T, class U = T>
T exchange(T& obj, U&& new_value) {
T old_value = move(obj);
obj = forward<U>(new_value);
return old_value;
}
/// make_unique
template <typename T, typename... Args>
unique_ptr<T> make_unique(Args&&... args) {
return unique_ptr<T>(new T(forward<Args>(args)...));
}
#endif
}
/* -------------------------------------------------------------------------- */
__BEGIN_TAMAAS__
/* -------------------------------------------------------------------------- */
/// @section Common types definitions
using Real = double; ///< default floating point type
using UInt = unsigned int; ///< default unsigned integer type
using Int = int; ///< default signed integer type
template <typename T>
using complex = thrust::complex<T>; ///< template complex wrapper
using Complex = complex<Real>; ///< default floating point complex type
static constexpr Real zero_threshold = 1e-14;
/// @section Defining random toolbox
using ::thrust::random::normal_distribution;
using random_engine = ::thrust::random::default_random_engine;
/* -------------------------------------------------------------------------- */
/// initialize tamaas (0 threads => let OMP_NUM_THREADS decide)
void initialize(UInt num_threads = 0);
/// cleanup tamaas
void finalize();
/* -------------------------------------------------------------------------- */
/// Generic exception class
class Exception : public std::exception {
public:
/// Constructor
Exception(const std::string& mesg) : msg(mesg) {}
virtual const char* what() const noexcept { return msg.c_str(); }
virtual ~Exception() = default;
private:
std::string msg; ///< message of exception
};
/* -------------------------------------------------------------------------- */
/// Enumeration of reduction operations
enum class operation { plus, times, min, max };
/* -------------------------------------------------------------------------- */
__END_TAMAAS__
/* -------------------------------------------------------------------------- */
/// @section Convenience macros
#define TAMAAS_EXCEPTION(mesg) \
{ \
std::stringstream sstr; \
sstr << __FILE__ << ":" << __LINE__ << ":FATAL: " << mesg << std::endl; \
std::cerr.flush(); \
throw ::tamaas::Exception(sstr.str()); \
}
#define SURFACE_FATAL(mesg) TAMAAS_EXCEPTION(mesg)
#if defined(TAMAAS_DEBUG)
#define TAMAAS_ASSERT(cond, reason) \
do { \
if (!(cond)) { \
TAMAAS_EXCEPTION(#cond " assert failed: " << reason); \
} \
} while (0)
#define TAMAAS_DEBUG_EXCEPTION(reason) TAMAAS_EXCEPTION(reason)
#else
#define TAMAAS_ASSERT(cond, reason)
#define TAMAAS_DEBUG_EXCEPTION(reason)
#endif
#define TAMAAS_ACCESSOR(var, type, name) \
type& get##name() { return var; } \
void set##name(const type& new_var) { var = new_var; }
#define TAMAAS_LOG(mesg) \
std::cout << __FILE__ << ":" << __LINE__ << ":(" << mpi_rank << ") " << mesg \
<< std::endl;
/* -------------------------------------------------------------------------- */
#endif // TAMAAS_HH
c4science · Help