src/OpenFOAM/meshes/primitiveMesh/primitiveMeshCells.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 1991-2008 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 "primitiveMesh.H"
  28
  29 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  30
  31 namespace Foam
  32 {
  33
  34 // * * * * * * * * * * * * * Private Member Functions  * * * * * * * * * * * //
  35
  36 void primitiveMesh::calcCells
  37 (
  38     cellList& cellFaceAddr,
  39     const unallocLabelList& own,
  40     const unallocLabelList& nei,
  41     const label inNCells
  42 )
  43 {
  44     label nCells = inNCells;
  45
  46     if (nCells == -1)
  47     {
  48         nCells = -1;
  49
  50         forAll(own, faceI)
  51         {
  52             nCells = max(nCells, own[faceI]);
  53         }
  54         nCells++;
  55     }
  56
  57     // 1. Count number of faces per cell
  58
  59     labelList ncf(nCells, 0);
  60
  61     forAll (own, faceI)
  62     {
  63         ncf[own[faceI]]++;
  64     }
  65
  66     forAll (nei, faceI)
  67     {
  68         if (nei[faceI] >= 0)
  69         {
  70             ncf[nei[faceI]]++;
  71         }
  72     }
  73
  74     // Create the storage
  75     cellFaceAddr.setSize(ncf.size());
  76
  77
  78     // 2. Size and fill cellFaceAddr
  79
  80     forAll (cellFaceAddr, cellI)
  81     {
  82         cellFaceAddr[cellI].setSize(ncf[cellI]);
  83     }
  84     ncf = 0;
  85
  86     forAll (own, faceI)
  87     {
  88         label cellI = own[faceI];
  89
  90         cellFaceAddr[cellI][ncf[cellI]++] = faceI;
  91     }
  92
  93     forAll (nei, faceI)
  94     {
  95         label cellI = nei[faceI];
  96
  97         if (cellI >= 0)
  98         {
  99             cellFaceAddr[cellI][ncf[cellI]++] = faceI;
 100         }
 101     }
 102 }
 103
 104
 105 void primitiveMesh::calcCells() const
 106 {
 107     // Loop through faceCells and mark up neighbours
 108
 109     if (debug)
 110     {
 111         Pout<< "primitiveMesh::calcCells() : calculating cells"
 112             << endl;
 113     }
 114
 115     // It is an error to attempt to recalculate cells
 116     // if the pointer is already set
 117     if (cfPtr_)
 118     {
 119         FatalErrorIn("primitiveMesh::calcCells() const")
 120             << "cells already calculated"
 121             << abort(FatalError);
 122     }
 123     else
 124     {
 125         // Create the storage
 126         cfPtr_ = new cellList(nCells());
 127         cellList& cellFaceAddr = *cfPtr_;
 128
 129         calcCells
 130         (
 131             cellFaceAddr,
 132             faceOwner(),
 133             faceNeighbour(),
 134             nCells()
 135         );
 136     }
 137 }
 138
 139
 140 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 141
 142 const cellList& primitiveMesh::cells() const
 143 {
 144     if (!cfPtr_)
 145     {
 146         calcCells();
 147     }
 148
 149     return *cfPtr_;
 150 }
 151
 152
 153 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 154
 155 } // End namespace Foam
 156
 157 // ************************************************************************* //