datatype.hpp

Go to the documentation of this file.

// Copyright (C) 2010-2013 von Karman Institute for Fluid Dynamics, Belgium
//
// This software is distributed under the terms of the
// GNU Lesser General Public License version 3 (LGPLv3).
// See doc/lgpl.txt and doc/gpl.txt for the license text.

#ifndef cf3_common_PE_datatype_hpp
#define cf3_common_PE_datatype_hpp


#include <mpi.h>

#include <boost/type_traits/is_pod.hpp>

#include "common/PE/types.hpp"
// #include "common/PE/debug.hpp" // for debugging mpi



namespace cf3 {
  namespace common {
    namespace PE {


namespace detail {
template <typename T> inline Datatype get_mpi_datatype_impl(const T& ) { return nullptr; };
} // namespace detail


template <typename T> inline Datatype get_mpi_datatype(const T& ref_of_type)
{
  if (detail::get_mpi_datatype_impl(ref_of_type)!=nullptr) return detail::get_mpi_datatype_impl<T>(ref_of_type);
  static Datatype type(nullptr);
  if (type==nullptr){
    //PEProcessSortedExecute(-1,CFinfo << "Registering type of size " << sizeof(T) << CFendl;);
    //if (!boost::is_pod<T>::value) throw NotSupported(FromHere(),"Non-POD (plain old datatype) is not supported by parallel environment communications.");
    MPI_CHECK_RESULT(MPI_Type_contiguous,(sizeof(T), MPI_BYTE, &type));
    MPI_CHECK_RESULT(MPI_Type_commit,(&type));
  }
  return type;
}

template <typename T> inline Datatype get_mpi_datatype()
{
  static Datatype type(nullptr);
  if (type==nullptr){
    //PEProcessSortedExecute(-1,CFinfo << "Registering type of size " << sizeof(T) << CFendl;);
    //if (!boost::is_pod<T>::value) throw NotSupported(FromHere(),"Non-POD (plain old datatype) is not supported by parallel environment communications.");
    MPI_CHECK_RESULT(MPI_Type_contiguous,(sizeof(T), MPI_BYTE, &type));
    MPI_CHECK_RESULT(MPI_Type_commit,(&type));
  }
  return type;
}



namespace detail {
  template<> inline Datatype get_mpi_datatype_impl <char>               (const char& )               { return MPI_CHAR; };
  template<> inline Datatype get_mpi_datatype_impl <unsigned char>      (const unsigned char& )      { return MPI_UNSIGNED_CHAR; };
  template<> inline Datatype get_mpi_datatype_impl <short>              (const short& )              { return MPI_SHORT; };
  template<> inline Datatype get_mpi_datatype_impl <unsigned short>     (const unsigned short& )     { return MPI_UNSIGNED_SHORT; };
  template<> inline Datatype get_mpi_datatype_impl <int>                (const int& )                { return MPI_INT; };
  template<> inline Datatype get_mpi_datatype_impl <unsigned int>       (const unsigned int& )       { return MPI_UNSIGNED; }; // there is no MPI_UNSIGNED_INT
  template<> inline Datatype get_mpi_datatype_impl <long>               (const long& )               { return MPI_LONG; };
  template<> inline Datatype get_mpi_datatype_impl <unsigned long>      (const unsigned long& )      { return MPI_UNSIGNED_LONG; };
  template<> inline Datatype get_mpi_datatype_impl <long long>          (const long long& )          { return MPI_LONG_LONG; };
  template<> inline Datatype get_mpi_datatype_impl <unsigned long long> (const unsigned long long& ) { return MPI_UNSIGNED_LONG_LONG; };
  template<> inline Datatype get_mpi_datatype_impl <float>              (const float&)               { return MPI_FLOAT; };
  template<> inline Datatype get_mpi_datatype_impl <double>             (const double& )             { return MPI_DOUBLE; };
  template<> inline Datatype get_mpi_datatype_impl <long double>        (const long double& )        { return MPI_LONG_DOUBLE; };
} // namespace detail




    } // namespace PE
  } // namespace common
} // namespace cf3


#endif // cf3_common_PE_datatype_hpp