src/meshTools/octree/octreeDataCell.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 Description
  26
  27 \*---------------------------------------------------------------------------*/
  28
  29 #include "octreeDataCell.H"
  30 #include "polyMesh.H"
  31 #include "primitiveMesh.H"
  32 #include "treeNode.H"
  33
  34 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  35
  36 // Construct from components
  37 Foam::octreeDataCell::octreeDataCell
  38 (
  39     const polyMesh& mesh,
  40     const labelList& cellLabels,
  41     const treeBoundBoxList& bbs
  42 )
  43 :
  44     mesh_(mesh),
  45     cellLabels_(cellLabels),
  46     bbs_(bbs)
  47 {}
  48
  49
  50 // Construct from mesh (assumes all cells)
  51 Foam::octreeDataCell::octreeDataCell
  52 (
  53     const polyMesh& mesh
  54 )
  55 :
  56     mesh_(mesh),
  57     cellLabels_(mesh_.nCells()),
  58     bbs_
  59     (
  60         mesh_.nCells(),
  61         treeBoundBox
  62         (
  63             vector(GREAT, GREAT, GREAT),
  64             vector(-GREAT, -GREAT, -GREAT)
  65         )
  66     )
  67 {
  68     // Set one-one indexing
  69     for (label i=0; i < mesh_.nCells(); i++)
  70     {
  71         cellLabels_[i] = i;
  72     }
  73
  74     const pointField& points = mesh_.points();
  75     const faceList& faces = mesh_.faces();
  76     const cellList& cells = mesh_.cells();
  77
  78     forAll(cells, celli)
  79     {
  80         const labelList& facesi = cells[celli];
  81
  82         forAll(facesi, facei)
  83         {
  84             const labelList& pointsi = faces[facesi[facei]];
  85
  86             forAll(pointsi, pointi)
  87             {
  88                 const point& p = points[pointsi[pointi]];
  89
  90                 bbs_[celli].min() = min(bbs_[celli].min(), p);
  91                 bbs_[celli].max() = max(bbs_[celli].max(), p);
  92             }
  93         }
  94     }
  95 }
  96
  97
  98 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
  99
 100 Foam::label Foam::octreeDataCell::getSampleType
 101 (
 102     octree<octreeDataCell>&,
 103     const point&
 104 ) const
 105 {
 106     return octree<octreeDataCell>::UNKNOWN;
 107 }
 108
 109
 110 bool Foam::octreeDataCell::overlaps
 111 (
 112     const label index,
 113     const treeBoundBox& cubeBb
 114 ) const
 115 {
 116     return cubeBb.intersects(bbs_[index]);
 117 }
 118
 119
 120 bool Foam::octreeDataCell::contains
 121 (
 122     const label index,
 123     const point& sample
 124 ) const
 125 {
 126     return mesh_.pointInCell(sample, cellLabels_[index]);
 127 }
 128
 129
 130 bool Foam::octreeDataCell::intersects
 131 (
 132     const label,
 133     const point&,
 134     const point&,
 135     point&
 136 ) const
 137 {
 138     //Hack: don't know what to do here.
 139
 140     notImplemented
 141     (
 142         "octreeDataCell::intersects(const label, const point&,"
 143         "const point&, point&)"
 144     );
 145
 146     return false;
 147 }
 148
 149
 150 bool Foam::octreeDataCell::findTightest
 151 (
 152     const label index,
 153     const point& sample,
 154     treeBoundBox& tightest
 155 ) const
 156 {
 157
 158     // get nearest and furthest away vertex
 159     point myNear, myFar;
 160     bbs_[index].calcExtremities(sample, myNear, myFar);
 161
 162     const point dist = myFar - sample;
 163     scalar myFarDist = mag(dist);
 164
 165     point tightestNear, tightestFar;
 166     tightest.calcExtremities(sample, tightestNear, tightestFar);
 167
 168     scalar tightestFarDist = mag(tightestFar - sample);
 169
 170     if (tightestFarDist < myFarDist)
 171     {
 172         // Keep current tightest.
 173         return false;
 174     }
 175     else
 176     {
 177         // Construct bb around sample and myFar
 178         const point dist2(fabs(dist.x()), fabs(dist.y()), fabs(dist.z()));
 179
 180         tightest.min() = sample - dist2;
 181         tightest.max() = sample + dist2;
 182
 183         return true;
 184     }
 185 }
 186
 187
 188 // Determine numerical value of sign of sample compared to shape at index
 189 Foam::scalar Foam::octreeDataCell::calcSign
 190 (
 191     const label,
 192     const point&,
 193     vector& n
 194 ) const
 195 {
 196     n = vector::zero;
 197
 198     return GREAT;
 199 }
 200
 201
 202 // Calculate nearest point on/in shapei
 203 Foam::scalar Foam::octreeDataCell::calcNearest
 204 (
 205     const label index,
 206     const point& sample,
 207     point& nearest
 208 ) const
 209 {
 210     nearest = mesh_.cellCentres()[cellLabels_[index]];
 211
 212     return mag(nearest - sample);
 213 }
 214
 215
 216 // Calculate nearest point on/in shapei
 217 Foam::scalar Foam::octreeDataCell::calcNearest
 218 (
 219     const label index,
 220     const linePointRef& ln,
 221     point& linePt,
 222     point& shapePt
 223 ) const
 224 {
 225     notImplemented
 226     (
 227         "octreeDataCell::calcNearest(const label, const linePointRef&"
 228         ", point& linePt, point&)"
 229     );
 230     return GREAT;
 231 }
 232
 233
 234 void Foam::octreeDataCell::write
 235 (
 236     Ostream& os,
 237     const label index
 238 ) const
 239 {
 240     os << cellLabels_[index] << " " << bbs_[index];
 241 }
 242
 243
 244 // ************************************************************************* //