applications/test/router/Gather/GatherBase.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 1991-2010 OpenCFD Ltd.
   6      \\/     M anipulation  |
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of OpenFOAM.
  10
  11     OpenFOAM is free software: you can redistribute it and/or modify it
  12     under the terms of the GNU General Public License as published by
  13     the Free Software Foundation, either version 3 of the License, or
  14     (at your option) any later version.
  15
  16     OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
  17     ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  18     FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  19     for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with OpenFOAM.  If not, see <http://www.gnu.org/licenses/>.
  23
  24 Description
  25
  26 \*---------------------------------------------------------------------------*/
  27
  28 #include "GatherBase.H"
  29 #include "IPstream.H"
  30 #include "OPstream.H"
  31
  32 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  33
  34 namespace Foam
  35 {
  36
  37 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
  38
  39 template <class Type>
  40 Type GatherBase::flatten(const List<Type> lst)
  41 {
  42     label sum = 0;
  43
  44     forAll(lst, lstI)
  45     {
  46         sum += lst[lstI].size();
  47     }
  48
  49     Type result(sum);
  50
  51     label index = 0;
  52
  53     forAll(lst, lstI)
  54     {
  55         const Type& sub = lst[lstI];
  56
  57         forAll(sub, subI)
  58         {
  59             result[index++] = sub[subI];
  60         }
  61     }
  62
  63     return result;
  64 }
  65
  66
  67 template <class DataType, class IndexType, class AddOp>
  68 IndexType GatherBase::offset
  69 (
  70     const List<DataType>& values,
  71     const List<IndexType>& indices,
  72     AddOp aop
  73 )
  74 {
  75     if (values.size() != indices.size())
  76     {
  77         FatalErrorIn
  78         (
  79             "GatherBase::offset(const List<DataType>&, "
  80             "const List<IndexType>&, AddOp)"
  81         )   << "Input data and indices lists not equal size." << endl
  82             << "data size:" << values.size()
  83             << "  indices:" << indices.size()
  84             << abort(FatalError);
  85     }
  86
  87
  88     label sum = 0;
  89
  90     forAll(indices, lstI)
  91     {
  92         sum += indices[lstI].size();
  93     }
  94
  95     IndexType result(sum);
  96
  97     label index = 0;
  98
  99     label offset = 0;
 100
 101     forAll(indices, lstI)
 102     {
 103         const IndexType& sub = indices[lstI];
 104
 105         forAll(sub, subI)
 106         {
 107             result[index++] = aop(sub[subI], offset);
 108         }
 109         offset += values[lstI].size();
 110     }
 111
 112     return result;
 113 }
 114
 115
 116 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 117
 118 } // End namespace Foam
 119
 120 // ************************************************************************* //