FreeFOAM The Cross-Platform CFD Toolkit
cylinderToCell.C
Go to the documentation of this file.
1 /*---------------------------------------------------------------------------*\
2  ========= |
3  \\ / F ield | OpenFOAM: The Open Source CFD Toolbox
4  \\ / O peration |
5  \\ / A nd | Copyright (C) 1991-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 <OpenFOAM/polyMesh.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 
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 
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
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 
122 {}
123 
124 
125 // * * * * * * * * * * * * * * * Member Functions * * * * * * * * * * * * * //
126 
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 // ************************ vim: set sw=4 sts=4 et: ************************ //