src/foam/containers/Lists/ListOps/ListOps.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | foam-extend: Open Source CFD
   4    \\    /   O peration     | Version:     3.2
   5     \\  /    A nd           | Web:         http://www.foam-extend.org
   6      \\/     M anipulation  | For copyright notice see file Copyright
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of foam-extend.
  10
  11     foam-extend 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 3 of the License, or (at your
  14     option) any later version.
  15
  16     foam-extend is distributed in the hope that it will be useful, but
  17     WITHOUT ANY WARRANTY; without even the implied warranty of
  18     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19     General Public License for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with foam-extend.  If not, see <http://www.gnu.org/licenses/>.
  23
  24 \*---------------------------------------------------------------------------*/
  25
  26 #include "ListOps.H"
  27
  28 // * * * * * * * * * * * * * * * Global Functions  * * * * * * * * * * * * * //
  29
  30 Foam::labelList Foam::invert
  31 (
  32     const label len,
  33     const UList<label>& map
  34 )
  35 {
  36     labelList inverse(len, -1);
  37
  38     forAll(map, i)
  39     {
  40         label newPos = map[i];
  41
  42         if (newPos >= 0)
  43         {
  44             if (inverse[newPos] >= 0)
  45             {
  46                 FatalErrorIn("invert(const label, const UList<label>&)")
  47                     << "Map is not one-to-one. At index " << i
  48                     << " element " << newPos << " has already occurred before"
  49                     << nl << "Please use invertOneToMany instead"
  50                     << abort(FatalError);
  51             }
  52
  53             inverse[newPos] = i;
  54         }
  55     }
  56     return inverse;
  57 }
  58
  59
  60 Foam::labelListList Foam::invertOneToMany
  61 (
  62     const label len,
  63     const UList<label>& map
  64 )
  65 {
  66     labelList nElems(len, 0);
  67
  68     forAll(map, i)
  69     {
  70         if (map[i] >= 0)
  71         {
  72             nElems[map[i]]++;
  73         }
  74     }
  75
  76     labelListList inverse(len);
  77
  78     forAll(nElems, i)
  79     {
  80         inverse[i].setSize(nElems[i]);
  81         nElems[i] = 0;
  82     }
  83
  84     forAll(map, i)
  85     {
  86         label newI = map[i];
  87
  88         if (newI >= 0)
  89         {
  90             inverse[newI][nElems[newI]++] = i;
  91         }
  92     }
  93
  94     return inverse;
  95 }
  96
  97
  98 Foam::labelList Foam::identity(const label len)
  99 {
 100     labelList map(len);
 101
 102     forAll(map, i)
 103     {
 104         map[i] = i;
 105     }
 106     return map;
 107 }
 108
 109
 110 // ************************************************************************* //