122 lines
3.0 KiB
C
122 lines
3.0 KiB
C
/*---------------------------------------------------------------------------*\
|
|
========= |
|
|
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
|
|
\\ / O peration |
|
|
\\ / A nd | Copyright (C) 1991-2009 OpenCFD Ltd.
|
|
\\/ 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 2 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, write to the Free Software Foundation,
|
|
Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
Description
|
|
|
|
\*---------------------------------------------------------------------------*/
|
|
|
|
#include "GatherBase.H"
|
|
#include "IPstream.H"
|
|
#include "OPstream.H"
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
namespace Foam
|
|
{
|
|
|
|
// * * * * * * * * * * * * * * * Member Functions * * * * * * * * * * * * * //
|
|
|
|
template <class Type>
|
|
Type GatherBase::flatten(const List<Type> lst)
|
|
{
|
|
label sum = 0;
|
|
|
|
forAll(lst, lstI)
|
|
{
|
|
sum += lst[lstI].size();
|
|
}
|
|
|
|
Type result(sum);
|
|
|
|
label index = 0;
|
|
|
|
forAll(lst, lstI)
|
|
{
|
|
const Type& sub = lst[lstI];
|
|
|
|
forAll(sub, subI)
|
|
{
|
|
result[index++] = sub[subI];
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
template <class DataType, class IndexType, class AddOp>
|
|
IndexType GatherBase::offset
|
|
(
|
|
const List<DataType>& values,
|
|
const List<IndexType>& indices,
|
|
AddOp aop
|
|
)
|
|
{
|
|
if (values.size() != indices.size())
|
|
{
|
|
FatalErrorIn
|
|
(
|
|
"GatherBase::offset(const List<DataType>&, "
|
|
"const List<IndexType>&, AddOp)"
|
|
) << "Input data and indices lists not equal size." << endl
|
|
<< "data size:" << values.size()
|
|
<< " indices:" << indices.size()
|
|
<< abort(FatalError);
|
|
}
|
|
|
|
|
|
label sum = 0;
|
|
|
|
forAll(indices, lstI)
|
|
{
|
|
sum += indices[lstI].size();
|
|
}
|
|
|
|
IndexType result(sum);
|
|
|
|
label index = 0;
|
|
|
|
label offset = 0;
|
|
|
|
forAll(indices, lstI)
|
|
{
|
|
const IndexType& sub = indices[lstI];
|
|
|
|
forAll(sub, subI)
|
|
{
|
|
result[index++] = aop(sub[subI], offset);
|
|
}
|
|
offset += values[lstI].size();
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
} // End namespace Foam
|
|
|
|
// ************************************************************************* //
|