add mpi serialization for WellSegments

This commit is contained in:
Arne Morten Kvarving
2019-12-11 11:44:58 +01:00
parent 5bc506cd3b
commit c31d209295
3 changed files with 79 additions and 0 deletions

View File

@@ -1143,6 +1143,21 @@ std::size_t packSize(const UnitSystem& data,
packSize(data.use_count(), comm);
}
std::size_t packSize(const WellSegments& data,
Dune::MPIHelper::MPICommunicator comm)
{
return packSize(data.wellName(), comm) +
packSize(data.depthTopSegment(), comm) +
packSize(data.lengthTopSegment(), comm) +
packSize(data.volumeTopSegment(), comm) +
packSize(data.lengthDepthType(), comm) +
packSize(data.compPressureDrop(), comm) +
packSize(data.multiPhaseModel(), comm) +
packSize(data.segments(), comm) +
packSize(data.segmentNumberIndex(), comm);
}
////// pack routines
template<class T>
@@ -2300,6 +2315,21 @@ void pack(const UnitSystem& data,
pack(data.use_count(), buffer, position, comm);
}
void pack(const WellSegments& data,
std::vector<char>& buffer, int& position,
Dune::MPIHelper::MPICommunicator comm)
{
pack(data.wellName(), buffer, position, comm);
pack(data.depthTopSegment(), buffer, position, comm);
pack(data.lengthTopSegment(), buffer, position, comm);
pack(data.volumeTopSegment(), buffer, position, comm);
pack(data.lengthDepthType(), buffer, position, comm);
pack(data.compPressureDrop(), buffer, position, comm);
pack(data.multiPhaseModel(), buffer, position, comm);
pack(data.segments(), buffer, position, comm);
pack(data.segmentNumberIndex(), buffer, position, comm);
}
/// unpack routines
template<class T>
@@ -3916,6 +3946,32 @@ void unpack(UnitSystem& data,
data = UnitSystem(name, type, dimensions, use_count);
}
void unpack(WellSegments& data,
std::vector<char>& buffer, int& position,
Dune::MPIHelper::MPICommunicator comm)
{
std::string wellName;
double depthTopSegment, lengthTopSegment, volumeTopSegment;
WellSegments::CompPressureDrop compPressureDrop;
WellSegments::LengthDepth lengthDepthType;
WellSegments::MultiPhaseModel multiPhaseModel;
std::vector<Segment> segments;
std::map<int,int> segmentNumberIndex;
unpack(wellName, buffer, position, comm);
unpack(depthTopSegment, buffer, position, comm);
unpack(lengthTopSegment, buffer, position, comm);
unpack(volumeTopSegment, buffer, position, comm);
unpack(lengthDepthType, buffer, position, comm);
unpack(compPressureDrop, buffer, position, comm);
unpack(multiPhaseModel, buffer, position, comm);
unpack(segments, buffer, position, comm);
unpack(segmentNumberIndex, buffer, position, comm);
data = WellSegments(wellName, depthTopSegment, lengthTopSegment,
volumeTopSegment, lengthDepthType, compPressureDrop,
multiPhaseModel, segments, segmentNumberIndex);
}
#define INSTANTIATE_PACK_VECTOR(T) \
template std::size_t packSize(const std::vector<T>& data, \
Dune::MPIHelper::MPICommunicator comm); \

View File

@@ -125,6 +125,7 @@ class WellEconProductionLimits;
class WellFoamProperties;
class WellPolymerProperties;
class WellSegmentDims;
class WellSegments;
class WellTracerProperties;
namespace Mpi
@@ -612,6 +613,7 @@ ADD_PACK_PROTOTYPES(WellEconProductionLimits)
ADD_PACK_PROTOTYPES(WellFoamProperties)
ADD_PACK_PROTOTYPES(WellPolymerProperties)
ADD_PACK_PROTOTYPES(WellSegmentDims)
ADD_PACK_PROTOTYPES(WellSegments)
ADD_PACK_PROTOTYPES(WellTestConfig)
ADD_PACK_PROTOTYPES(WellTestConfig::WTESTWell)
ADD_PACK_PROTOTYPES(WellTracerProperties)