src/OpenFOAM/containers/Lists/ListListOps/ListListOps.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright held by original author
   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 \*---------------------------------------------------------------------------*/
  26
  27 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  28
  29 namespace Foam
  30 {
  31
  32 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
  33
  34 template <class AccessType, class T, class AccessOp>
  35 AccessType ListListOps::combine(const List<T>& lst, AccessOp aop)
  36 {
  37     label sum = 0;
  38
  39     forAll(lst, lstI)
  40     {
  41         sum += aop(lst[lstI]).size();
  42     }
  43
  44     AccessType result(sum);
  45
  46     label globalElemI = 0;
  47
  48     forAll(lst, lstI)
  49     {
  50         const T& sub = lst[lstI];
  51
  52         forAll(aop(sub), elemI)
  53         {
  54             result[globalElemI++] = aop(sub)[elemI];
  55         }
  56     }
  57     return result;
  58 }
  59
  60
  61 template <class T, class AccessOp>
  62 labelList ListListOps::subSizes(const List<T>& lst, AccessOp aop)
  63 {
  64     labelList sizes(lst.size());
  65
  66     forAll(lst, lstI)
  67     {
  68         sizes[lstI] = aop(lst[lstI]).size();
  69     }
  70     return sizes;
  71 }
  72
  73
  74 template <class AccessType, class T, class AccessOp, class OffsetOp>
  75 AccessType ListListOps::combineOffset
  76 (
  77     const List<T>& lst,
  78     const labelList& sizes,
  79     AccessOp aop,
  80     OffsetOp oop
  81 )
  82 {
  83     label sum = 0;
  84
  85     forAll(lst, lstI)
  86     {
  87         sum += aop(lst[lstI]).size();
  88     }
  89
  90     AccessType result(sum);
  91
  92     label globalElemI = 0;
  93
  94     label offset = 0;
  95
  96     forAll(lst, lstI)
  97     {
  98         const T& sub = lst[lstI];
  99
 100         forAll(aop(sub), elemI)
 101         {
 102             result[globalElemI++] = oop(aop(sub)[elemI], offset);
 103         }
 104
 105         offset += sizes[lstI];
 106     }
 107     return result;
 108 }
 109
 110 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 111
 112 } // End namespace Foam
 113
 114
 115 // ************************************************************************* //