src/OpenFOAM/meshes/primitiveMesh/primitiveMeshPointCells.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 2004-2010 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
  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 "primitiveMesh.H"
  27 #include "cell.H"
  28
  29 // * * * * * * * * * * * * * Private Member Functions  * * * * * * * * * * * //
  30
  31 void Foam::primitiveMesh::calcPointCells() const
  32 {
  33     // Loop through cells and mark up points
  34
  35     if (debug)
  36     {
  37         Pout<< "primitiveMesh::calcPointCells() : "
  38             << "calculating pointCells"
  39             << endl;
  40
  41         if (debug == -1)
  42         {
  43             // For checking calls:abort so we can quickly hunt down
  44             // origin of call
  45             FatalErrorIn("primitiveMesh::calcPointCells()")
  46                 << abort(FatalError);
  47         }
  48     }
  49
  50     // It is an error to attempt to recalculate pointCells
  51     // if the pointer is already set
  52     if (pcPtr_)
  53     {
  54         FatalErrorIn("primitiveMesh::calcPointCells() const")
  55             << "pointCells already calculated"
  56             << abort(FatalError);
  57     }
  58     else
  59     {
  60         const cellList& cf = cells();
  61
  62         // Count number of cells per point
  63
  64         labelList npc(nPoints(), 0);
  65
  66         forAll(cf, cellI)
  67         {
  68             const labelList curPoints = cf[cellI].labels(faces());
  69
  70             forAll(curPoints, pointI)
  71             {
  72                 label ptI = curPoints[pointI];
  73
  74                 npc[ptI]++;
  75             }
  76         }
  77
  78
  79         // Size and fill cells per point
  80
  81         pcPtr_ = new labelListList(npc.size());
  82         labelListList& pointCellAddr = *pcPtr_;
  83
  84         forAll(pointCellAddr, pointI)
  85         {
  86             pointCellAddr[pointI].setSize(npc[pointI]);
  87         }
  88         npc = 0;
  89
  90
  91         forAll(cf, cellI)
  92         {
  93             const labelList curPoints = cf[cellI].labels(faces());
  94
  95             forAll(curPoints, pointI)
  96             {
  97                 label ptI = curPoints[pointI];
  98
  99                 pointCellAddr[ptI][npc[ptI]++] = cellI;
 100             }
 101         }
 102     }
 103 }
 104
 105
 106 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 107
 108 const Foam::labelListList& Foam::primitiveMesh::pointCells() const
 109 {
 110     if (!pcPtr_)
 111     {
 112         calcPointCells();
 113     }
 114
 115     return *pcPtr_;
 116 }
 117
 118
 119 const Foam::labelList& Foam::primitiveMesh::pointCells
 120 (
 121     const label pointI,
 122     DynamicList<label>& storage
 123 ) const
 124 {
 125     if (hasPointCells())
 126     {
 127         return pointCells()[pointI];
 128     }
 129     else
 130     {
 131         const labelList& own = faceOwner();
 132         const labelList& nei = faceNeighbour();
 133         const labelList& pFaces = pointFaces()[pointI];
 134
 135         storage.clear();
 136
 137         forAll(pFaces, i)
 138         {
 139             const label faceI = pFaces[i];
 140
 141             // Append owner
 142             storage.append(own[faceI]);
 143
 144             // Append neighbour
 145             if (faceI < nInternalFaces())
 146             {
 147                 storage.append(nei[faceI]);
 148             }
 149         }
 150
 151         // Filter duplicates
 152         if (storage.size() > 1)
 153         {
 154             sort(storage);
 155
 156             label n = 1;
 157             for (label i = 1; i < storage.size(); i++)
 158             {
 159                 if (storage[i-1] != storage[i])
 160                 {
 161                     storage[n++] = storage[i];
 162                 }
 163             }
 164
 165             // truncate addressed list
 166             storage.setSize(n);
 167         }
 168
 169         return storage;
 170     }
 171 }
 172
 173
 174 const Foam::labelList& Foam::primitiveMesh::pointCells(const label pointI) const
 175 {
 176     return pointCells(pointI, labels_);
 177 }
 178
 179
 180 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 181
 182 // ************************************************************************* //