Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Tree: 97cf67d699
Fetching contributors…

Cannot retrieve contributors at this time

189 lines (162 sloc) 5.276 kB
/*---------------------------------------------------------------------------*\
========= |
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
\\ / O peration |
\\ / A nd | Copyright (C) 2011 OpenFOAM Foundation
\\/ M anipulation |
-------------------------------------------------------------------------------
License
This file is part of OpenFOAM.
OpenFOAM 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.
OpenFOAM 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 OpenFOAM. If not, see <http://www.gnu.org/licenses/>.
Description
Gather data from all processors onto single processor according to some
communication schedule (usually linear-to-master or tree-to-master).
The gathered data will be a single value constructed from the values
on individual processors using a user-specified operator.
\*---------------------------------------------------------------------------*/
#include "UOPstream.H"
#include "OPstream.H"
#include "UIPstream.H"
#include "IPstream.H"
#include "contiguous.H"
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
namespace Foam
{
// * * * * * * * * * * * * * * * Member Functions * * * * * * * * * * * * * //
template <class T, class BinaryOp>
void Pstream::gather
(
const List<UPstream::commsStruct>& comms,
T& Value,
const BinaryOp& bop
)
{
if (UPstream::parRun())
{
// Get my communication order
const commsStruct& myComm = comms[UPstream::myProcNo()];
// Receive from my downstairs neighbours
forAll(myComm.below(), belowI)
{
T value;
if (contiguous<T>())
{
UIPstream::read
(
UPstream::scheduled,
myComm.below()[belowI],
reinterpret_cast<char*>(&value),
sizeof(T)
);
}
else
{
IPstream fromBelow(UPstream::scheduled, myComm.below()[belowI]);
fromBelow >> value;
}
Value = bop(Value, value);
}
// Send up Value
if (myComm.above() != -1)
{
if (contiguous<T>())
{
UOPstream::write
(
UPstream::scheduled,
myComm.above(),
reinterpret_cast<const char*>(&Value),
sizeof(T)
);
}
else
{
OPstream toAbove(UPstream::scheduled, myComm.above());
toAbove << Value;
}
}
}
}
template <class T, class BinaryOp>
void Pstream::gather(T& Value, const BinaryOp& bop)
{
if (UPstream::nProcs() < UPstream::nProcsSimpleSum)
{
gather(UPstream::linearCommunication(), Value, bop);
}
else
{
gather(UPstream::treeCommunication(), Value, bop);
}
}
template <class T>
void Pstream::scatter(const List<UPstream::commsStruct>& comms, T& Value)
{
if (UPstream::parRun())
{
// Get my communication order
const commsStruct& myComm = comms[UPstream::myProcNo()];
// Reveive from up
if (myComm.above() != -1)
{
if (contiguous<T>())
{
UIPstream::read
(
UPstream::scheduled,
myComm.above(),
reinterpret_cast<char*>(&Value),
sizeof(T)
);
}
else
{
IPstream fromAbove(UPstream::scheduled, myComm.above());
fromAbove >> Value;
}
}
// Send to my downstairs neighbours
forAll(myComm.below(), belowI)
{
if (contiguous<T>())
{
UOPstream::write
(
UPstream::scheduled,
myComm.below()[belowI],
reinterpret_cast<const char*>(&Value),
sizeof(T)
);
}
else
{
OPstream toBelow(UPstream::scheduled,myComm.below()[belowI]);
toBelow << Value;
}
}
}
}
template <class T>
void Pstream::scatter(T& Value)
{
if (UPstream::nProcs() < UPstream::nProcsSimpleSum)
{
scatter(UPstream::linearCommunication(), Value);
}
else
{
scatter(UPstream::treeCommunication(), Value);
}
}
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
} // End namespace Foam
// ************************************************************************* //
Jump to Line
Something went wrong with that request. Please try again.