mirror of
https://github.com/OPM/opm-simulators.git
synced 2024-12-26 17:20:59 -06:00
136 lines
3.8 KiB
C++
136 lines
3.8 KiB
C++
/*
|
|
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/>.
|
|
*/
|
|
#include <config.h>
|
|
#include "MPIPacker.hpp"
|
|
|
|
#include <bitset>
|
|
#include <cstdint>
|
|
#include <ctime>
|
|
#include <string>
|
|
#include <type_traits>
|
|
|
|
#include <opm/common/utility/TimeService.hpp>
|
|
|
|
namespace Opm {
|
|
namespace Mpi {
|
|
namespace detail {
|
|
|
|
template<std::size_t Size>
|
|
std::size_t Packing<false,std::bitset<Size>>::
|
|
packSize(const std::bitset<Size>& data,
|
|
Parallel::MPIComm comm)
|
|
{
|
|
return Packing<true,unsigned long long>::packSize(data.to_ullong(), comm);
|
|
}
|
|
|
|
template<std::size_t Size>
|
|
void Packing<false,std::bitset<Size>>::
|
|
pack(const std::bitset<Size>& data,
|
|
std::vector<char>& buffer,
|
|
int& position,
|
|
Parallel::MPIComm comm)
|
|
{
|
|
Packing<true,unsigned long long>::pack(data.to_ullong(), buffer, position, comm);
|
|
}
|
|
|
|
template<std::size_t Size>
|
|
void Packing<false,std::bitset<Size>>::
|
|
unpack(std::bitset<Size>& data,
|
|
std::vector<char>& buffer,
|
|
int& position,
|
|
Parallel::MPIComm comm)
|
|
{
|
|
unsigned long long d;
|
|
Packing<true,unsigned long long>::unpack(d, buffer, position, comm);
|
|
data = std::bitset<Size>(d);
|
|
}
|
|
|
|
std::size_t Packing<false,std::string>::
|
|
packSize(const std::string& data, Parallel::MPIComm comm)
|
|
{
|
|
int size;
|
|
MPI_Pack_size(1, Dune::MPITraits<std::size_t>::getType(), comm, &size);
|
|
int totalSize = size;
|
|
MPI_Pack_size(strlen(data.c_str()), MPI_CHAR, comm, &size);
|
|
return totalSize + size;
|
|
}
|
|
|
|
void Packing<false,std::string>::
|
|
pack(const std::string& data,
|
|
std::vector<char>& buffer,
|
|
int& position,
|
|
Parallel::MPIComm comm)
|
|
{
|
|
std::size_t length = strlen(data.c_str());
|
|
MPI_Pack(&length, 1, Dune::MPITraits<std::size_t>::getType(), buffer.data(),
|
|
buffer.size(), &position, comm);
|
|
MPI_Pack(data.c_str(), length, MPI_CHAR, buffer.data(), buffer.size(),
|
|
&position, comm);
|
|
}
|
|
|
|
void Packing<false,std::string>::
|
|
unpack(std::string& data,
|
|
std::vector<char>& buffer,
|
|
int& position,
|
|
Opm::Parallel::MPIComm comm)
|
|
{
|
|
std::size_t length = 0;
|
|
MPI_Unpack(buffer.data(), buffer.size(), &position, &length, 1,
|
|
Dune::MPITraits<std::size_t>::getType(), comm);
|
|
std::vector<char> cStr(length+1, '\0');
|
|
MPI_Unpack(buffer.data(), buffer.size(), &position, cStr.data(), length,
|
|
MPI_CHAR, comm);
|
|
data.clear();
|
|
data.append(cStr.data());
|
|
}
|
|
|
|
std::size_t Packing<false,time_point>::
|
|
packSize(const time_point&, Opm::Parallel::MPIComm comm)
|
|
{
|
|
return Packing<true,std::time_t>::packSize(std::time_t(), comm);
|
|
}
|
|
|
|
void Packing<false,time_point>::
|
|
pack(const time_point& data,
|
|
std::vector<char>& buffer,
|
|
int& position,
|
|
Parallel::MPIComm comm)
|
|
{
|
|
Packing<true,std::time_t>::pack(TimeService::to_time_t(data),
|
|
buffer, position, comm);
|
|
}
|
|
|
|
void Packing<false,time_point>::
|
|
unpack(time_point& data,
|
|
std::vector<char>& buffer,
|
|
int& position,
|
|
Parallel::MPIComm comm)
|
|
{
|
|
std::time_t res;
|
|
Packing<true,std::time_t>::unpack(res, buffer, position, comm);
|
|
data = TimeService::from_time_t(res);
|
|
}
|
|
|
|
template struct Packing<false,std::bitset<4>>;
|
|
template struct Packing<false,std::bitset<10>>;
|
|
|
|
} // end namespace detail
|
|
} // end namespace Mpi
|
|
} // end namespace Opm
|