OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

add mpi serialization for PolymerConfig

Browse Source
This commit is contained in:
Arne Morten Kvarving 2020-03-05 11:42:24 +01:00
parent 390c7b21d4
commit 5d0e9164f3
3 changed files with 40 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
opm/simulators/utils/ParallelRestart.cpp
View File

@ -31,6 +31,7 @@
#include <opm/parser/eclipse/EclipseState/InitConfig/Equil.hpp> #include <opm/parser/eclipse/EclipseState/InitConfig/Equil.hpp>
#include <opm/parser/eclipse/EclipseState/InitConfig/FoamConfig.hpp> #include <opm/parser/eclipse/EclipseState/InitConfig/FoamConfig.hpp>
#include <opm/parser/eclipse/EclipseState/InitConfig/InitConfig.hpp> #include <opm/parser/eclipse/EclipseState/InitConfig/InitConfig.hpp>
#include <opm/parser/eclipse/EclipseState/InitConfig/PolymerConfig.hpp>
#include <opm/parser/eclipse/EclipseState/IOConfig/IOConfig.hpp> #include <opm/parser/eclipse/EclipseState/IOConfig/IOConfig.hpp>
#include <opm/parser/eclipse/EclipseState/IOConfig/RestartConfig.hpp> #include <opm/parser/eclipse/EclipseState/IOConfig/RestartConfig.hpp>
#include <opm/parser/eclipse/EclipseState/Edit/EDITNNC.hpp> #include <opm/parser/eclipse/EclipseState/Edit/EDITNNC.hpp>
@ -686,10 +687,16 @@ std::size_t packSize(const FoamConfig& data, Dune::MPIHelper::MPICommunicator co
packSize(data.getMobilityModel(), comm); packSize(data.getMobilityModel(), comm);
} }
std::size_t packSize(const PolymerConfig& data, Dune::MPIHelper::MPICommunicator comm)
{
return packSize(data.shrate(), comm);
}
std::size_t packSize(const InitConfig& data, Dune::MPIHelper::MPICommunicator comm) std::size_t packSize(const InitConfig& data, Dune::MPIHelper::MPICommunicator comm)
{ {
return packSize(data.getEquil(), comm) + return packSize(data.getEquil(), comm) +
packSize(data.getFoamConfig(), comm) + packSize(data.getFoamConfig(), comm) +
packSize(data.getPolymerConfig(), comm) +
packSize(data.filleps(), comm) + packSize(data.filleps(), comm) +
packSize(data.hasGravity(), comm) + packSize(data.hasGravity(), comm) +
packSize(data.restartRequested(), comm) + packSize(data.restartRequested(), comm) +
@ -2297,11 +2304,18 @@ void pack(const FoamConfig& data, std::vector<char>& buffer, int& position,
pack(data.getMobilityModel(), buffer, position, comm); pack(data.getMobilityModel(), buffer, position, comm);
} }
void pack(const PolymerConfig& data, std::vector<char>& buffer, int& position,
Dune::MPIHelper::MPICommunicator comm)
{
pack(data.shrate(), buffer, position, comm);
}
void pack(const InitConfig& data, std::vector<char>& buffer, int& position, void pack(const InitConfig& data, std::vector<char>& buffer, int& position,
Dune::MPIHelper::MPICommunicator comm) Dune::MPIHelper::MPICommunicator comm)
{ {
pack(data.getEquil(), buffer, position, comm); pack(data.getEquil(), buffer, position, comm);
pack(data.getFoamConfig(), buffer, position, comm); pack(data.getFoamConfig(), buffer, position, comm);
pack(data.getPolymerConfig(), buffer, position, comm);
pack(data.filleps(), buffer, position, comm); pack(data.filleps(), buffer, position, comm);
pack(data.hasGravity(), buffer, position, comm); pack(data.hasGravity(), buffer, position, comm);
pack(data.restartRequested(), buffer, position, comm); pack(data.restartRequested(), buffer, position, comm);
@ -4145,22 +4159,32 @@ void unpack(FoamConfig& data, std::vector<char>& buffer, int& position,
data = FoamConfig(records, transport_phase, mobility_model); data = FoamConfig(records, transport_phase, mobility_model);
} }
void unpack(PolymerConfig& data, std::vector<char>& buffer, int& position,
Dune::MPIHelper::MPICommunicator comm)
{
bool shrate;
unpack(shrate, buffer, position, comm);
data = PolymerConfig(shrate);
}
void unpack(InitConfig& data, std::vector<char>& buffer, int& position, void unpack(InitConfig& data, std::vector<char>& buffer, int& position,
Dune::MPIHelper::MPICommunicator comm) Dune::MPIHelper::MPICommunicator comm)
{ {
Equil equil; Equil equil;
FoamConfig foam; FoamConfig foam;
PolymerConfig polymer;
bool filleps, hasGravity, restartRequested; bool filleps, hasGravity, restartRequested;
int restartStep; int restartStep;
std::string restartRootName; std::string restartRootName;
unpack(equil, buffer, position, comm); unpack(equil, buffer, position, comm);
unpack(foam, buffer, position, comm); unpack(foam, buffer, position, comm);
unpack(polymer, buffer, position, comm);
unpack(filleps, buffer, position, comm); unpack(filleps, buffer, position, comm);
unpack(hasGravity, buffer, position, comm); unpack(hasGravity, buffer, position, comm);
unpack(restartRequested, buffer, position, comm); unpack(restartRequested, buffer, position, comm);
unpack(restartStep, buffer, position, comm); unpack(restartStep, buffer, position, comm);
unpack(restartRootName, buffer, position, comm); unpack(restartRootName, buffer, position, comm);
data = InitConfig(equil, foam, filleps, hasGravity, data = InitConfig(equil, foam, polymer, filleps, hasGravity,
restartRequested, restartStep, restartRootName); restartRequested, restartStep, restartRootName);
} }

2
opm/simulators/utils/ParallelRestart.hpp
View File

@ -112,6 +112,7 @@ class PlymwinjTable;
class PolyInjTable; class PolyInjTable;
class PVCDORecord; class PVCDORecord;
class PvcdoTable; class PvcdoTable;
class PolymerConfig;
class PvtgTable; class PvtgTable;
class PvtoTable; class PvtoTable;
class PVTWRecord; class PVTWRecord;
@ -564,6 +565,7 @@ ADD_PACK_PROTOTYPES(OilVaporizationProperties)
ADD_PACK_PROTOTYPES(Phases) ADD_PACK_PROTOTYPES(Phases)
ADD_PACK_PROTOTYPES(PlymwinjTable) ADD_PACK_PROTOTYPES(PlymwinjTable)
ADD_PACK_PROTOTYPES(PolyInjTable) ADD_PACK_PROTOTYPES(PolyInjTable)
ADD_PACK_PROTOTYPES(PolymerConfig)
ADD_PACK_PROTOTYPES(PVCDORecord) ADD_PACK_PROTOTYPES(PVCDORecord)
ADD_PACK_PROTOTYPES(PvcdoTable) ADD_PACK_PROTOTYPES(PvcdoTable)
ADD_PACK_PROTOTYPES(PvtgTable) ADD_PACK_PROTOTYPES(PvtgTable)

15
tests/test_ParallelRestart.cpp
View File

@ -38,6 +38,7 @@
#include <opm/parser/eclipse/EclipseState/InitConfig/Equil.hpp> #include <opm/parser/eclipse/EclipseState/InitConfig/Equil.hpp>
#include <opm/parser/eclipse/EclipseState/InitConfig/FoamConfig.hpp> #include <opm/parser/eclipse/EclipseState/InitConfig/FoamConfig.hpp>
#include <opm/parser/eclipse/EclipseState/InitConfig/InitConfig.hpp> #include <opm/parser/eclipse/EclipseState/InitConfig/InitConfig.hpp>
#include <opm/parser/eclipse/EclipseState/InitConfig/PolymerConfig.hpp>
#include <opm/parser/eclipse/EclipseState/IOConfig/IOConfig.hpp> #include <opm/parser/eclipse/EclipseState/IOConfig/IOConfig.hpp>
#include <opm/parser/eclipse/EclipseState/IOConfig/RestartConfig.hpp> #include <opm/parser/eclipse/EclipseState/IOConfig/RestartConfig.hpp>
#include <opm/parser/eclipse/EclipseState/Schedule/Action/ActionAST.hpp> #include <opm/parser/eclipse/EclipseState/Schedule/Action/ActionAST.hpp>
@ -806,11 +807,21 @@ BOOST_AUTO_TEST_CASE(FoamConfig)
} }
BOOST_AUTO_TEST_CASE(PolymerConfig)
{
#if HAVE_MPI
Opm::PolymerConfig val1(true);
auto val2 = PackUnpack(val1);
DO_CHECKS(PolymerConfig)
#endif
}
BOOST_AUTO_TEST_CASE(InitConfig) BOOST_AUTO_TEST_CASE(InitConfig)
{ {
#if HAVE_MPI #if HAVE_MPI
Opm::InitConfig val1(Opm::Equil({getEquilRecord(), getEquilRecord()}), Opm::InitConfig val1(Opm::Equil({getEquilRecord(), getEquilRecord()}),
getFoamConfig(), getFoamConfig(), Opm::PolymerConfig(true),
true, true, true, 20, "test1"); true, true, true, 20, "test1");
auto val2 = PackUnpack(val1); auto val2 = PackUnpack(val1);
DO_CHECKS(InitConfig) DO_CHECKS(InitConfig)
@ -2275,7 +2286,7 @@ BOOST_AUTO_TEST_CASE(EclipseConfig)
Opm::IOConfig io(true, false, true, false, false, true, "test1", true, Opm::IOConfig io(true, false, true, false, false, true, "test1", true,
"test2", true, "test3", false); "test2", true, "test3", false);
Opm::InitConfig init(Opm::Equil({getEquilRecord(), getEquilRecord()}), Opm::InitConfig init(Opm::Equil({getEquilRecord(), getEquilRecord()}),
getFoamConfig(), getFoamConfig(), Opm::PolymerConfig(true),
true, true, true, 20, "test1"); true, true, true, 20, "test1");
Opm::EclipseConfig val1{init, io}; Opm::EclipseConfig val1{init, io};