Creation of OpenFOAM-dev repository 15/04/2008

applications/test/router/Gather/GatherBase.C

@@ -0,0 +1,121 @@
+/*---------------------------------------------------------------------------*\
+  =========                 |
+  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
+   \\    /   O peration     |
+    \\  /    A nd           | Copyright (C) 1991-2007 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
+
+// ************************************************************************* //