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) 2011 OpenFOAM Foundation
   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 \*---------------------------------------------------------------------------*/
  25
  26 #include "GatherBase.H"
  27
  28 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  29
  30 namespace Foam
  31 {
  32
  33 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
  34
  35 template <class Type>
  36 Type GatherBase::flatten(const List<Type> lst)
  37 {
  38     label sum = 0;
  39
  40     forAll(lst, lstI)
  41     {
  42         sum += lst[lstI].size();
  43     }
  44
  45     Type result(sum);
  46
  47     label index = 0;
  48
  49     forAll(lst, lstI)
  50     {
  51         const Type& sub = lst[lstI];
  52
  53         forAll(sub, subI)
  54         {
  55             result[index++] = sub[subI];
  56         }
  57     }
  58
  59     return result;
  60 }
  61
  62
  63 template <class DataType, class IndexType, class AddOp>
  64 IndexType GatherBase::offset
  65 (
  66     const List<DataType>& values,
  67     const List<IndexType>& indices,
  68     AddOp aop
  69 )
  70 {
  71     if (values.size() != indices.size())
  72     {
  73         FatalErrorIn
  74         (
  75             "GatherBase::offset(const List<DataType>&, "
  76             "const List<IndexType>&, AddOp)"
  77         )   << "Input data and indices lists not equal size." << endl
  78             << "data size:" << values.size()
  79             << "  indices:" << indices.size()
  80             << abort(FatalError);
  81     }
  82
  83
  84     label sum = 0;
  85
  86     forAll(indices, lstI)
  87     {
  88         sum += indices[lstI].size();
  89     }
  90
  91     IndexType result(sum);
  92
  93     label index = 0;
  94
  95     label offset = 0;
  96
  97     forAll(indices, lstI)
  98     {
  99         const IndexType& sub = indices[lstI];
 100
 101         forAll(sub, subI)
 102         {
 103             result[index++] = aop(sub[subI], offset);
 104         }
 105         offset += values[lstI].size();
 106     }
 107
 108     return result;
 109 }
 110
 111
 112 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 113
 114 } // End namespace Foam
 115
 116 // ************************************************************************* //