src/meshTools/sets/cellSources/cylinderAnnulusToCell/cylinderAnnulusToCell.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 2011-2011 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 "cylinderAnnulusToCell.H"
  27 #include "polyMesh.H"
  28 #include "addToRunTimeSelectionTable.H"
  29
  30 // * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
  31
  32 namespace Foam
  33 {
  34     defineTypeNameAndDebug(cylinderAnnulusToCell, 0);
  35     addToRunTimeSelectionTable(topoSetSource, cylinderAnnulusToCell, word);
  36     addToRunTimeSelectionTable(topoSetSource, cylinderAnnulusToCell, istream);
  37 }
  38
  39
  40 Foam::topoSetSource::addToUsageTable Foam::cylinderAnnulusToCell::usage_
  41 (
  42     cylinderAnnulusToCell::typeName,
  43     "\n    Usage: cylinderAnnulusToCell (p1X p1Y p1Z) (p2X p2Y p2Z)"
  44     " outerRadius innerRadius\n\n"
  45     "    Select all cells with cell centre within bounding cylinder annulus\n\n"
  46 );
  47
  48
  49 // * * * * * * * * * * * * * Private Member Functions  * * * * * * * * * * * //
  50
  51 void Foam::cylinderAnnulusToCell::combine(topoSet& set, const bool add) const
  52 {
  53     const vector axis = p2_ - p1_;
  54     const scalar orad2 = sqr(outerRadius_);
  55     const scalar irad2 = sqr(innerRadius_);
  56     const scalar magAxis2 = magSqr(axis);
  57
  58     const pointField& ctrs = mesh_.cellCentres();
  59
  60     forAll(ctrs, cellI)
  61     {
  62         vector d = ctrs[cellI] - p1_;
  63         scalar magD = d & axis;
  64
  65         if ((magD > 0) && (magD < magAxis2))
  66         {
  67             scalar d2 = (d & d) - sqr(magD)/magAxis2;
  68             if ((d2 < orad2) && (d2 > irad2))
  69             {
  70                 addOrDelete(set, cellI, add);
  71             }
  72         }
  73     }
  74 }
  75
  76
  77 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  78
  79 Foam::cylinderAnnulusToCell::cylinderAnnulusToCell
  80 (
  81     const polyMesh& mesh,
  82     const vector& p1,
  83     const vector& p2,
  84     const scalar outerRadius,
  85     const scalar innerRadius
  86 )
  87 :
  88     topoSetSource(mesh),
  89     p1_(p1),
  90     p2_(p2),
  91     outerRadius_(outerRadius),
  92     innerRadius_(innerRadius)
  93 {}
  94
  95
  96 Foam::cylinderAnnulusToCell::cylinderAnnulusToCell
  97 (
  98     const polyMesh& mesh,
  99     const dictionary& dict
 100 )
 101 :
 102     topoSetSource(mesh),
 103     p1_(dict.lookup("p1")),
 104     p2_(dict.lookup("p2")),
 105     outerRadius_(readScalar(dict.lookup("outerRadius"))),
 106     innerRadius_(readScalar(dict.lookup("innerRadius")))
 107 {}
 108
 109
 110 // Construct from Istream
 111 Foam::cylinderAnnulusToCell::cylinderAnnulusToCell
 112 (
 113     const polyMesh& mesh,
 114     Istream& is
 115 )
 116 :
 117     topoSetSource(mesh),
 118     p1_(checkIs(is)),
 119     p2_(checkIs(is)),
 120     outerRadius_(readScalar(checkIs(is))),
 121     innerRadius_(readScalar(checkIs(is)))
 122 {}
 123
 124
 125 // * * * * * * * * * * * * * * * * Destructor  * * * * * * * * * * * * * * * //
 126
 127 Foam::cylinderAnnulusToCell::~cylinderAnnulusToCell()
 128 {}
 129
 130
 131 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 132
 133 void Foam::cylinderAnnulusToCell::applyToSet
 134 (
 135     const topoSetSource::setAction action,
 136     topoSet& set
 137 ) const
 138 {
 139     if ((action == topoSetSource::NEW) || (action == topoSetSource::ADD))
 140     {
 141         Info<< "    Adding cells with centre within cylinder annulus,"
 142             << " with p1 = "
 143             << p1_ << ", p2 = " << p2_ << " and outer radius = " << outerRadius_
 144         << " and inner radius = " << innerRadius_
 145         << endl;
 146
 147         combine(set, true);
 148     }
 149     else if (action == topoSetSource::DELETE)
 150     {
 151         Info<< "    Removing cells with centre within cylinder, with p1 = "
 152             << p1_ << ", p2 = " << p2_ << " and outer radius = " << outerRadius_
 153         << " and inner radius " << innerRadius_
 154         << endl;
 155
 156         combine(set, false);
 157     }
 158 }
 159
 160
 161 // ************************************************************************* //