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