src/meshTools/sets/cellSources/cylinderToCell/cylinderToCell.C

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