zhyang-dev/openfoam
1
0
Fork 0
mirror of https://develop.openfoam.com/Development/openfoam.git synced 2025-11-28 03:28:01 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

ENH: globalIndex: inplace gather

Browse Source
This commit is contained in:
mattijs
2013-04-03 16:30:08 +01:00
parent ceb4f3a645
commit 5c517879ed
2 changed files with 94 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/OpenFOAM/meshes/polyMesh/globalMeshData/globalIndex.H
View File

@ -162,6 +162,17 @@ public:
const int tag = UPstream::msgType()
) const;
//- Inplace collect data in processor order on master
// (== procIDs[0]). Needs offsets only on master.
template<class Type>
void gather
(
const label comm,
const labelList& procIDs,
List<Type>& fld,
const int tag = UPstream::msgType()
) const;
//- Distribute data in processor order. Requires fld to be sized!
template<class Type>
void scatter

83
src/OpenFOAM/meshes/polyMesh/globalMeshData/globalIndexTemplates.C
View File

@ -109,6 +109,89 @@ void Foam::globalIndex::gather
}
template<class Type>
void Foam::globalIndex::gather
(
const label comm,
const labelList& procIDs,
List<Type>& fld,
const int tag
) const
{
if (Pstream::myProcNo(comm) == procIDs[0])
{
List<Type> allFld(size());
// Assign my local data
SubList<Type>(allFld, fld.size(), 0).assign(fld);
for (label i = 1; i < procIDs.size(); i++)
{
SubList<Type> procSlot
(
allFld,
offsets_[i+1]-offsets_[i],
offsets_[i]
);
if (contiguous<Type>())
{
IPstream::read
(
Pstream::scheduled,
procIDs[i],
reinterpret_cast<char*>(procSlot.begin()),
procSlot.byteSize(),
tag,
comm
);
}
else
{
IPstream fromSlave
(
Pstream::scheduled,
procIDs[i],
0,
tag,
comm
);
fromSlave >> procSlot;
}
}
fld.transfer(allFld);
}
else
{
if (contiguous<Type>())
{
OPstream::write
(
Pstream::scheduled,
procIDs[0],
reinterpret_cast<const char*>(fld.begin()),
fld.byteSize(),
tag,
comm
);
}
else
{
OPstream toMaster
(
Pstream::scheduled,
procIDs[0],
0,
tag,
comm
);
toMaster << fld;
}
}
}
template<class Type>
void Foam::globalIndex::scatter
(