MPIPacker.hpp

Go to the documentation of this file.

/*
  Copyright 2019 Equinor AS.
 
  This file is part of the Open Porous Media project (OPM).
 
  OPM is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.
 
  OPM 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 General Public License
  along with OPM.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef MPI_PACKER_HPP
#define MPI_PACKER_HPP
 
#include <opm/common/utility/TimeService.hpp>
#include <opm/simulators/utils/ParallelCommunication.hpp>
 
#include <dune/common/parallel/mpitraits.hh>
 
#include <bitset>
#include <cstddef>
#include <limits>
#include <string>
 
 
namespace Opm::Mpi {
 
namespace detail {
 
std::size_t mpi_buffer_size(const std::size_t bufsize, const std::size_t position);
 
template <bool pod, class T>
struct Packing
{
    static std::size_t packSize(const T&, Parallel::MPIComm);
    static void pack(const T&, std::vector<char>&, std::size_t&, Parallel::MPIComm);
    static void unpack(T&, const std::vector<char>&, std::size_t&, Parallel::MPIComm);
};
 
template<class T>
struct Packing<true,T>
{
    static std::size_t packSize(const T& data, Parallel::MPIComm comm)
    {
        return packSize(&data, 1, comm);
    }
 
    static std::size_t packSize(const T*, std::size_t n, Parallel::MPIComm comm)
    {
        // For now we do not handle the situation where a a single call to packSize/pack/unpack
        // is likely to require an MPI_Pack_size value larger than intmax
        if (n*sizeof(T) > std::numeric_limits<int>::max())
            throw std::invalid_argument("packSize will be larger than max integer - this is not supported.");
        int size = 0;
        MPI_Pack_size(n, Dune::MPITraits<T>::getType(), comm, &size);
        return static_cast<std::size_t>(size);
    }
 
    static void pack(const T& data,
                     std::vector<char>& buffer,
                     std::size_t& position,
                     Parallel::MPIComm comm)
    {
        pack(&data, 1, buffer, position, comm);
    }
 
    static void pack(const T* data,
                     std::size_t n,
                     std::vector<char>& buffer,
                     std::size_t& position,
                     Parallel::MPIComm comm)
    {
        int int_position = 0;
        MPI_Pack(data, n, Dune::MPITraits<T>::getType(), buffer.data()+position,
                 mpi_buffer_size(buffer.size(), position), &int_position, comm);
        position += int_position;
    }
 
    static void unpack(T& data,
                       const std::vector<char>& buffer,
                       std::size_t& position,
                       Parallel::MPIComm comm)
    {
        unpack(&data, 1, buffer, position, comm);
    }
 
    static void unpack(T* data,
                       std::size_t n,
                       const std::vector<char>& buffer,
                       std::size_t& position,
                       Parallel::MPIComm comm)
    {
        int int_position = 0;
        MPI_Unpack(buffer.data()+position, mpi_buffer_size(buffer.size(), position), &int_position, data, n,
                   Dune::MPITraits<T>::getType(), comm);
        position += int_position;
    }
};
 
template<class T>
struct Packing<false,T>
{
    static std::size_t packSize(const T&, Parallel::MPIComm)
    {
        static_assert(!std::is_same_v<T,T>, "Packing not supported for type");
        return 0;
    }
 
    static void pack(const T&, std::vector<char>&, std::size_t&,
                     Parallel::MPIComm)
    {
      static_assert(!std::is_same_v<T,T>, "Packing not supported for type");
    }
 
    static void unpack(T&, const std::vector<char>&, std::size_t&,
                       Parallel::MPIComm)
    {
        static_assert(!std::is_same_v<T,T>, "Packing not supported for type");
    }
};
 
template <std::size_t Size>
struct Packing<false,std::bitset<Size>>
{
    static std::size_t packSize(const std::bitset<Size>&, Opm::Parallel::MPIComm);
    static void pack(const std::bitset<Size>&, std::vector<char>&, std::size_t&, Opm::Parallel::MPIComm);
    static void unpack(std::bitset<Size>&, const std::vector<char>&,
                       std::size_t&, Opm::Parallel::MPIComm);
};
 
#define ADD_PACK_SPECIALIZATION(T) \
    template<> \
    struct Packing<false,T> \
    { \
        static std::size_t packSize(const T&, Parallel::MPIComm); \
        static void pack(const T&, std::vector<char>&, std::size_t&, Parallel::MPIComm); \
        static void unpack(T&, const std::vector<char>&, std::size_t&, Parallel::MPIComm); \
    };
 
ADD_PACK_SPECIALIZATION(std::string)
ADD_PACK_SPECIALIZATION(time_point)
 
#undef ADD_PACK_SPECIALIZATION
 
}
 
struct Packer
{
    Packer(Parallel::Communication comm)
        : m_comm(comm)
    {}
 
    template<class T>
    std::size_t packSize(const T& data) const
    {
        return detail::Packing<std::is_pod_v<T>,T>::packSize(data, m_comm);
    }
 
    template<class T>
    std::size_t packSize(const T* data, std::size_t n) const
    {
        static_assert(std::is_pod_v<T>, "Array packing not supported for non-pod data");
        return detail::Packing<true,T>::packSize(data, n, m_comm);
    }
 
    template<class T>
    void pack(const T& data,
              std::vector<char>& buffer,
              std::size_t& position) const
    {
        detail::Packing<std::is_pod_v<T>,T>::pack(data, buffer, position, m_comm);
    }
 
    template<class T>
    void pack(const T* data,
              std::size_t n,
              std::vector<char>& buffer,
              std::size_t& position) const
    {
        static_assert(std::is_pod_v<T>, "Array packing not supported for non-pod data");
        detail::Packing<true,T>::pack(data, n, buffer, position, m_comm);
    }
 
    template<class T>
    void unpack(T& data,
                const std::vector<char>& buffer,
                std::size_t& position) const
    {
        detail::Packing<std::is_pod_v<T>,T>::unpack(data, buffer, position, m_comm);
    }
 
    template<class T>
    void unpack(T* data,
                std::size_t n,
                const std::vector<char>& buffer,
                std::size_t& position) const
    {
        static_assert(std::is_pod_v<T>, "Array packing not supported for non-pod data");
        detail::Packing<true,T>::unpack(data, n, buffer, position, m_comm);
    }
 
private:
    Parallel::Communication m_comm; 
};
 
} // end namespace Opm::Mpi
 
#endif // MPI_PACKER_HPP