applications/utilities/mesh/manipulation/cellSet/cellSet.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     Selects a cell set through a dictionary.
  27
  28 \*---------------------------------------------------------------------------*/
  29
  30 #include "argList.H"
  31 #include "Time.H"
  32 #include "polyMesh.H"
  33 #include "topoSetSource.H"
  34 #include "cellSet.H"
  35
  36 using namespace Foam;
  37
  38 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  39
  40 // Copy set
  41 void backup
  42 (
  43     const polyMesh& mesh,
  44     const word& fromName,
  45     const topoSet& fromSet,
  46     const word& toName
  47 )
  48 {
  49     Info<< "Backing up " << fromName << " into " << toName << endl;
  50
  51     topoSet backupSet(mesh, toName, fromSet);
  52
  53     backupSet.write();
  54 }
  55
  56
  57 // Read and copy set
  58 void backup
  59 (
  60     const polyMesh& mesh,
  61     const word& fromName,
  62     const word& toName
  63 )
  64 {
  65     topoSet fromSet(mesh, fromName, IOobject::READ_IF_PRESENT);
  66
  67     backup(mesh, fromName, fromSet, toName);
  68 }
  69
  70
  71 // Main program:
  72
  73 int main(int argc, char *argv[])
  74 {
  75 #   include "setRootCase.H"
  76 #   include "createTime.H"
  77 #   include "createPolyMesh.H"
  78
  79     Info<< "Reading cellSetDict\n" << endl;
  80
  81     IOdictionary cellSetDict
  82     (
  83         IOobject
  84         (
  85             "cellSetDict",
  86             runTime.system(),
  87             mesh,
  88             IOobject::MUST_READ,
  89             IOobject::NO_WRITE
  90         )
  91     );
  92
  93
  94     word setName(cellSetDict.lookup("name"));
  95
  96     word actionName(cellSetDict.lookup("action"));
  97
  98     topoSetSource::setAction action = topoSetSource::toAction(actionName);
  99
 100
 101     // Create topoSetSources
 102     PtrList<topoSetSource> topoSetSources
 103     (
 104         cellSetDict.lookup("topoSetSources"),
 105         topoSetSource::iNew(mesh)
 106     );
 107
 108
 109     // Load set to work
 110     autoPtr<topoSet> currentSetPtr(NULL);
 111     IOobject::readOption r;
 112
 113     if ((action == topoSetSource::NEW) || (action == topoSetSource::CLEAR))
 114     {
 115         r = IOobject::NO_READ;
 116
 117         backup(mesh, setName, setName + "_old");
 118
 119         currentSetPtr.reset
 120         (
 121             new cellSet
 122             (
 123                 mesh,
 124                 setName,
 125                 mesh.nCells()/10+1  // Reasonable size estimate.
 126             )
 127         );
 128     }
 129     else
 130     {
 131         r = IOobject::MUST_READ;
 132
 133         currentSetPtr.reset
 134         (
 135             new cellSet
 136             (
 137                 mesh,
 138                 setName,
 139                 r
 140             )
 141         );
 142     }
 143
 144     topoSet& currentSet = currentSetPtr();
 145
 146     Info<< "Set:" << currentSet.name()
 147         << "  Size:" << currentSet.size()
 148         << "  Action:" << actionName
 149         << endl;
 150
 151     if ((r == IOobject::MUST_READ) && (action != topoSetSource::LIST))
 152     {
 153         // currentSet has been read so can make copy.
 154         backup(mesh, setName, currentSet, setName + "_old");
 155     }
 156
 157     if (action == topoSetSource::CLEAR)
 158     {
 159         // Already handled above by not reading
 160     }
 161     else if (action == topoSetSource::INVERT)
 162     {
 163         currentSet.invert(currentSet.maxSize(mesh));
 164     }
 165     else if (action == topoSetSource::LIST)
 166     {
 167         currentSet.writeDebug(Info, mesh, 100);
 168         Info<< endl;
 169     }
 170     else if (action == topoSetSource::SUBSET)
 171     {
 172         // Apply topoSetSources to it to handle new/add/delete
 173         forAll(topoSetSources, topoSetSourceI)
 174         {
 175             // Backup current set.
 176             topoSet oldSet(mesh, currentSet.name() + "_old2", currentSet);
 177
 178             currentSet.clear();
 179
 180             topoSetSources[topoSetSourceI].applyToSet
 181             (
 182                 topoSetSource::NEW,
 183                 currentSet
 184             );
 185
 186             // Combine new value of currentSet with old one.
 187             currentSet.subset(oldSet);
 188         }
 189     }
 190     else
 191     {
 192         // Apply topoSetSources to it to handle new/add/delete
 193         forAll(topoSetSources, topoSetSourceI)
 194         {
 195             topoSetSources[topoSetSourceI].applyToSet(action, currentSet);
 196         }
 197     }
 198
 199
 200     if (action != topoSetSource::LIST)
 201     {
 202         // Set has changed.
 203
 204         // Sync across coupled patches.
 205         currentSet.sync(mesh);
 206
 207         Info<< "Writing " << currentSet.name()
 208             << " (size " << currentSet.size() << ") to "
 209             << currentSet.instance()/currentSet.local()
 210                /currentSet.name()
 211             << endl << endl;
 212
 213         currentSet.write();
 214     }
 215
 216     Info << nl << "End" << endl;
 217
 218     return 0;
 219 }
 220
 221
 222 // ************************************************************************* //