applications/utilities/surface/surfaceSmooth/surfaceSmooth.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 Description
  25     Example of a simple laplacian smoother
  26
  27 \*---------------------------------------------------------------------------*/
  28
  29 #include "argList.H"
  30 #include "OFstream.H"
  31 #include "boundBox.H"
  32
  33 #include "MeshedSurfaces.H"
  34
  35 using namespace Foam;
  36
  37
  38 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  39 // Main program:
  40
  41 int main(int argc, char *argv[])
  42 {
  43     argList::noParallel();
  44     argList::validOptions.clear();
  45     argList::validArgs.append("surfaceFile");
  46     argList::validArgs.append("underrelax factor (0..1)");
  47     argList::validArgs.append("iterations");
  48     argList::validArgs.append("output surfaceFile");
  49     argList args(argc, argv);
  50
  51     const fileName surfFileName = args[1];
  52     const scalar relax = args.argRead<scalar>(2);
  53     const label  iters = args.argRead<label>(3);
  54     const fileName outFileName = args[4];
  55
  56     if (relax <= 0 || relax > 1)
  57     {
  58         FatalErrorIn(args.executable()) << "Illegal relaxation factor "
  59             << relax << endl
  60             << "0: no change   1: move vertices to average of neighbours"
  61             << exit(FatalError);
  62     }
  63
  64     Info<< "Relax:" << relax << nl
  65         << "Iters:" << iters << nl
  66         << "Reading surface from " << surfFileName << " ..." << endl;
  67
  68     meshedSurface surf1(surfFileName);
  69
  70     Info<< "Faces        : " << surf1.size() << nl
  71         << "Vertices     : " << surf1.nPoints() << nl
  72         << "Bounding Box : " << boundBox(surf1.localPoints()) << endl;
  73
  74     pointField newPoints(surf1.localPoints());
  75
  76     const labelListList& pointEdges = surf1.pointEdges();
  77
  78     for (label iter = 0; iter < iters; iter++)
  79     {
  80         forAll(pointEdges, vertI)
  81         {
  82             vector avgPos(vector::zero);
  83
  84             const labelList& myEdges = pointEdges[vertI];
  85
  86             forAll(myEdges, myEdgeI)
  87             {
  88                 const edge& e = surf1.edges()[myEdges[myEdgeI]];
  89
  90                 label otherVertI = e.otherVertex(vertI);
  91
  92                 avgPos += surf1.localPoints()[otherVertI];
  93             }
  94             avgPos /= myEdges.size();
  95
  96             newPoints[vertI] = (1-relax)*newPoints[vertI] + relax*avgPos;
  97         }
  98     }
  99
 100     Info<< "Writing surface to " << outFileName << " ..." << endl;
 101
 102     meshedSurface
 103     (
 104         xferMove(newPoints),
 105         xferCopy(surf1.localFaces()),
 106         xferCopy(surf1.surfZones())
 107     ).write(outFileName);
 108
 109     Info<< "End\n" << endl;
 110
 111     return 0;
 112 }
 113
 114
 115 // ************************************************************************* //