src/meshTools/sets/faceSources/normalToFace/normalToFace.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 Description
  26
  27 \*---------------------------------------------------------------------------*/
  28
  29 #include "normalToFace.H"
  30 #include "polyMesh.H"
  31 #include "faceSet.H"
  32
  33 #include "addToRunTimeSelectionTable.H"
  34
  35 // * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
  36
  37 namespace Foam
  38 {
  39
  40 defineTypeNameAndDebug(normalToFace, 0);
  41
  42 addToRunTimeSelectionTable(topoSetSource, normalToFace, word);
  43
  44 addToRunTimeSelectionTable(topoSetSource, normalToFace, istream);
  45
  46 }
  47
  48
  49 Foam::topoSetSource::addToUsageTable Foam::normalToFace::usage_
  50 (
  51     normalToFace::typeName,
  52     "\n    Usage: normalToFace (nx ny nz) <tol>\n\n"
  53     "    Select faces with normal aligned to unit vector (nx ny nz)\n"
  54     "    to within tol\n"
  55 );
  56
  57
  58 // * * * * * * * * * * * * * Private Member Functions  * * * * * * * * * * * //
  59
  60 void Foam::normalToFace::setNormal()
  61 {
  62     normal_ /= mag(normal_) + VSMALL;
  63
  64     Info<< "    normalToFace : Normalized vector to " << normal_ << endl;
  65
  66     if (tol_ < -1 || tol_ > 1)
  67     {
  68         FatalErrorIn
  69         (
  70             "normalToFace::normalToFace(const polyMesh&, const vector&"
  71             ", const scalar)"
  72         )   << "tolerance not within range -1..1 : " << tol_
  73             << exit(FatalError);
  74     }
  75 }
  76
  77
  78 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  79
  80 // Construct from components
  81 Foam::normalToFace::normalToFace
  82 (
  83     const polyMesh& mesh,
  84     const vector& normal,
  85     const scalar tol
  86 )
  87 :
  88     topoSetSource(mesh),
  89     normal_(normal),
  90     tol_(tol)
  91 {
  92     setNormal();
  93 }
  94
  95
  96 // Construct from dictionary
  97 Foam::normalToFace::normalToFace(const polyMesh& mesh, const dictionary& dict)
  98 :
  99     topoSetSource(mesh),
 100     normal_(dict.lookup("normal")),
 101     tol_(readScalar(dict.lookup("cos")))
 102 {
 103     setNormal();
 104 }
 105
 106
 107 // Construct from Istream
 108 Foam::normalToFace::normalToFace(const polyMesh& mesh, Istream& is)
 109 :
 110     topoSetSource(mesh),
 111     normal_(checkIs(is)),
 112     tol_(readScalar(checkIs(is)))
 113 {
 114     setNormal();
 115 }
 116
 117
 118 // * * * * * * * * * * * * * * * * Destructor  * * * * * * * * * * * * * * * //
 119
 120 Foam::normalToFace::~normalToFace()
 121 {}
 122
 123
 124 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 125
 126 void Foam::normalToFace::applyToSet
 127 (
 128     const topoSetSource::setAction action,
 129     topoSet& set
 130 ) const
 131 {
 132     if ((action == topoSetSource::NEW) || (action == topoSetSource::ADD))
 133     {
 134         Info<< "    Adding faces according to normal being aligned with "
 135             << normal_ << " (to within " << tol_ << ") ..." << endl;
 136
 137         forAll(mesh_.faceAreas(), faceI)
 138         {
 139             vector n = mesh_.faceAreas()[faceI];
 140             n /= mag(n) + VSMALL;
 141
 142             if (mag(1 - (n & normal_)) < tol_)
 143             {
 144                 set.insert(faceI);
 145             }
 146         }
 147     }
 148     else if (action == topoSetSource::DELETE)
 149     {
 150         Info<< "    Removing faces according to normal being aligned with "
 151             << normal_ << " (to within " << tol_ << ") ..." << endl;
 152
 153
 154         DynamicList<label> toBeRemoved(set.size()/10);
 155
 156         forAllIter(topoSet, set, iter)
 157         {
 158             label faceI = iter.key();
 159
 160             vector n = mesh_.faceAreas()[faceI];
 161             n /= mag(n) + VSMALL;
 162
 163             if (mag(1 - (n & normal_)) < tol_)
 164             {
 165                 toBeRemoved.append(faceI);
 166             }
 167         }
 168
 169         forAll(toBeRemoved, i)
 170         {
 171             set.erase(toBeRemoved[i]);
 172         }
 173     }
 174 }
 175
 176
 177 // ************************************************************************* //