src/OpenFOAM/containers/Lists/ListOps/ListOps.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 \*---------------------------------------------------------------------------*/
  26
  27 #include "ListOps.H"
  28
  29 // * * * * * * * * * * * * * * * Global Functions  * * * * * * * * * * * * * //
  30
  31 Foam::labelList Foam::invert
  32 (
  33     const label len,
  34     const UList<label>& map
  35 )
  36 {
  37     labelList inverse(len, -1);
  38
  39     forAll(map, i)
  40     {
  41         label newPos = map[i];
  42
  43         if (newPos >= 0)
  44         {
  45             if (inverse[newPos] >= 0)
  46             {
  47                 FatalErrorIn("invert(const label, const UList<label>&)")
  48                     << "Map is not one-to-one. At index " << i
  49                     << " element " << newPos << " has already occurred before"
  50                     << nl << "Please use invertOneToMany instead"
  51                     << abort(FatalError);
  52             }
  53
  54             inverse[newPos] = i;
  55         }
  56     }
  57     return inverse;
  58 }
  59
  60
  61 Foam::labelListList Foam::invertOneToMany
  62 (
  63     const label len,
  64     const UList<label>& map
  65 )
  66 {
  67     labelList nElems(len, 0);
  68
  69     forAll(map, i)
  70     {
  71         if (map[i] >= 0)
  72         {
  73             nElems[map[i]]++;
  74         }
  75     }
  76
  77     labelListList inverse(len);
  78
  79     forAll(nElems, i)
  80     {
  81         inverse[i].setSize(nElems[i]);
  82         nElems[i] = 0;
  83     }
  84
  85     forAll(map, i)
  86     {
  87         label newI = map[i];
  88
  89         if (newI >= 0)
  90         {
  91             inverse[newI][nElems[newI]++] = i;
  92         }
  93     }
  94
  95     return inverse;
  96 }
  97
  98
  99 Foam::labelList Foam::identity(const label len)
 100 {
 101     labelList map(len);
 102
 103     forAll(map, i)
 104     {
 105         map[i] = i;
 106     }
 107     return map;
 108 }
 109
 110
 111 // ************************************************************************* //