src/meshTools/sets/pointSources/cellToPoint/cellToPoint.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 "cellToPoint.H"
  30 #include "polyMesh.H"
  31 #include "cellSet.H"
  32
  33 #include "addToRunTimeSelectionTable.H"
  34
  35 // * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
  36
  37 namespace Foam
  38 {
  39
  40 defineTypeNameAndDebug(cellToPoint, 0);
  41
  42 addToRunTimeSelectionTable(topoSetSource, cellToPoint, word);
  43
  44 addToRunTimeSelectionTable(topoSetSource, cellToPoint, istream);
  45
  46 }
  47
  48
  49 Foam::topoSetSource::addToUsageTable Foam::cellToPoint::usage_
  50 (
  51     cellToPoint::typeName,
  52     "\n    Usage: cellToPoint <cellSet> all\n\n"
  53     "    Select all points of cells in the cellSet\n\n"
  54 );
  55
  56 template<>
  57 const char* Foam::NamedEnum<Foam::cellToPoint::cellAction, 1>::names[] =
  58 {
  59     "all"
  60 };
  61
  62 const Foam::NamedEnum<Foam::cellToPoint::cellAction, 1>
  63     Foam::cellToPoint::cellActionNames_;
  64
  65
  66 // * * * * * * * * * * * * * Private Member Functions  * * * * * * * * * * * //
  67
  68 void Foam::cellToPoint::combine(topoSet& set, const bool add) const
  69 {
  70     // Load the set
  71     cellSet loadedSet(mesh_, setName_);
  72
  73     // Add all point from cells in loadedSet
  74     for
  75     (
  76         cellSet::const_iterator iter = loadedSet.begin();
  77         iter != loadedSet.end();
  78         ++iter
  79     )
  80     {
  81         label cellI = iter.key();
  82
  83         const labelList& cFaces = mesh_.cells()[cellI];
  84
  85         forAll(cFaces, cFaceI)
  86         {
  87             const face& f = mesh_.faces()[cFaces[cFaceI]];
  88
  89             forAll(f, fp)
  90             {
  91                 addOrDelete(set, f[fp], add);
  92             }
  93         }
  94     }
  95 }
  96
  97
  98 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  99
 100 // Construct from components
 101 Foam::cellToPoint::cellToPoint
 102 (
 103     const polyMesh& mesh,
 104     const word& setName,
 105     const cellAction option
 106 )
 107 :
 108     topoSetSource(mesh),
 109     setName_(setName),
 110     option_(option)
 111 {}
 112
 113
 114 // Construct from dictionary
 115 Foam::cellToPoint::cellToPoint
 116 (
 117     const polyMesh& mesh,
 118     const dictionary& dict
 119 )
 120 :
 121     topoSetSource(mesh),
 122     setName_(dict.lookup("set")),
 123     option_(cellActionNames_.read(dict.lookup("option")))
 124 {}
 125
 126
 127 // Construct from Istream
 128 Foam::cellToPoint::cellToPoint
 129 (
 130     const polyMesh& mesh,
 131     Istream& is
 132 )
 133 :
 134     topoSetSource(mesh),
 135     setName_(checkIs(is)),
 136     option_(cellActionNames_.read(checkIs(is)))
 137 {}
 138
 139
 140 // * * * * * * * * * * * * * * * * Destructor  * * * * * * * * * * * * * * * //
 141
 142 Foam::cellToPoint::~cellToPoint()
 143 {}
 144
 145
 146 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 147
 148 void Foam::cellToPoint::applyToSet
 149 (
 150     const topoSetSource::setAction action,
 151     topoSet& set
 152 ) const
 153 {
 154     if ((action == topoSetSource::NEW) || (action == topoSetSource::ADD))
 155     {
 156         Pout<< "    Adding from " << setName_ << " ..." << endl;
 157
 158         combine(set, true);
 159     }
 160     else if (action == topoSetSource::DELETE)
 161     {
 162         Pout<< "    Removing from " << setName_ << " ..." << endl;
 163
 164         combine(set, false);
 165     }
 166 }
 167
 168
 169 // ************************************************************************* //