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