openfoam/applications/utilities/surface/surfaceAdd/surfaceAdd.C
Mark Olesen b060378dca ENH: improve consistency of fileName handling windows/non-windows (#2057)
- wrap command-line retrieval of fileName with an implicit validate.

  Instead of this:
      fileName input(args[1]);
      fileName other(args["someopt"]);

  Now use this:
      auto input = args.get<fileName>(1);
      auto other = args.get<fileName>("someopt");

  which adds a fileName::validate on the inputs

  Because of how it is implemented, it will automatically also apply
  to argList getOrDefault<fileName>, readIfPresent<fileName> etc.

- adjust fileName::validate and clean to handle backslash conversion.
  This makes it easier to ensure that path names arising from MS-Windows
  are consistently handled internally.

- dictionarySearch: now check for initial '/' directly instead of
  relying on fileName isAbsolute(), which now does more things

BREAKING: remove fileName::clean() const method

- relying on const/non-const to control the behaviour (inplace change
  or return a copy) is too fragile and the const version was
  almost never used.

  Replace:
      fileName sanitized = constPath.clean();

  With:
      fileName sanitized(constPath);
      sanitized.clean());

STYLE: test empty() instead of comparing with fileName::null
2021-04-19 16:33:42 +00:00

307 lines
8.7 KiB
C

/*---------------------------------------------------------------------------*\
========= |
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
\\ / O peration |
\\ / A nd | www.openfoam.com
\\/ M anipulation |
-------------------------------------------------------------------------------
Copyright (C) 2011-2016 OpenFOAM Foundation
Copyright (C) 2016-2021 OpenCFD Ltd.
-------------------------------------------------------------------------------
License
This file is part of OpenFOAM.
OpenFOAM is free software: you can redistribute it and/or modify it
under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
for more details.
You should have received a copy of the GNU General Public License
along with OpenFOAM. If not, see <http://www.gnu.org/licenses/>.
Application
surfaceAdd
Group
grpSurfaceUtilities
Description
Add two surfaces. Does geometric merge on points.
Does not check for overlapping/intersecting triangles.
Keeps patches separate by renumbering.
\*---------------------------------------------------------------------------*/
#include "argList.H"
#include "fileName.H"
#include "triSurface.H"
#include "Fstream.H"
#include "triFace.H"
#include "triFaceList.H"
using namespace Foam;
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
int main(int argc, char *argv[])
{
argList::addNote
(
"Add two surfaces via a geometric merge on points."
" Does not check for overlapping/intersecting triangles."
);
argList::noParallel();
argList::addArgument("surface1", "The input surface file 1");
argList::addArgument("surface2", "The input surface file 2");
argList::addArgument("output", "The output surface file");
argList::addOption
(
"points",
"file",
"Provide additional points"
);
argList::addBoolOption
(
"mergeRegions",
"Combine regions from both surfaces"
);
argList::addOption
(
"scale",
"factor",
"Geometry scaling factor on input surfaces"
);
argList args(argc, argv);
const auto inFileName1 = args.get<fileName>(1);
const auto inFileName2 = args.get<fileName>(2);
const auto outFileName = args.get<fileName>(3);
const bool addPoint = args.found("points");
const bool mergeRegions = args.found("mergeRegions");
const scalar scaleFactor = args.getOrDefault<scalar>("scale", -1);
if (addPoint)
{
Info<< "Reading a surface and adding points from a file"
<< "; merging the points and writing the surface to another file"
<< nl << endl;
Info<< "Surface : " << inFileName1<< nl
<< "Points : " << args.get<fileName>("points") << nl
<< "Writing : " << outFileName << nl << endl;
}
else
{
Info<< "Reading two surfaces"
<< "; merging points and writing the surface to another file"
<< nl << endl;
if (mergeRegions)
{
Info<< "Regions from the two files will get merged" << nl
<< "Do not use this option if you want to keep the regions"
<< " separate" << nl << endl;
}
else
{
Info<< "Regions from the two files will not get merged" << nl
<< "Regions from " << inFileName2 << " will get offset so"
<< " as not to overlap with the regions in " << inFileName1
<< nl << endl;
}
Info<< "Surface1 : " << inFileName1<< nl
<< "Surface2 : " << inFileName2<< nl
<< "Writing : " << outFileName << nl << endl;
}
if (scaleFactor > 0)
{
Info<< "Scaling : " << scaleFactor << nl;
}
const triSurface surface1(inFileName1, scaleFactor);
Info<< "Surface1:" << endl;
surface1.writeStats(Info);
Info<< endl;
const pointField& points1 = surface1.points();
// Final surface
triSurface combinedSurf;
if (addPoint)
{
IFstream pointsFile(args.get<fileName>("points"));
const pointField extraPoints(pointsFile);
Info<< "Additional Points:" << extraPoints.size() << endl;
vectorField pointsAll(points1);
label pointi = pointsAll.size();
pointsAll.setSize(pointsAll.size() + extraPoints.size());
for (const auto& pt : extraPoints)
{
pointsAll[pointi++] = pt;
}
combinedSurf = triSurface(surface1, surface1.patches(), pointsAll);
}
else
{
const triSurface surface2(inFileName2, scaleFactor);
Info<< "Surface2:" << endl;
surface2.writeStats(Info);
Info<< endl;
// Make new storage
List<labelledTri> facesAll(surface1.size() + surface2.size());
const pointField& points2 = surface2.points();
vectorField pointsAll(points1.size() + points2.size());
label pointi = 0;
// Copy points1 into pointsAll
for (const auto& pt : points1)
{
pointsAll[pointi++] = pt;
}
// Add surface2 points
for (const auto& pt : points2)
{
pointsAll[pointi++] = pt;
}
label trianglei = 0;
// Determine map for both regions
label nNewPatches = 0;
labelList patch1Map(surface1.patches().size());
labelList patch2Map(surface2.patches().size());
if (mergeRegions)
{
HashTable<label> nameToPatch;
forAll(surface1.patches(), i)
{
const word& name = surface1.patches()[i].name();
// Lookup or insert
const label combinedi = nameToPatch(name, nameToPatch.size());
patch1Map[i] = combinedi;
}
// Determine map for surface2 regions
forAll(surface2.patches(), i)
{
const word& name = surface2.patches()[i].name();
// Lookup or insert
const label combinedi = nameToPatch(name, nameToPatch.size());
patch2Map[i] = combinedi;
}
nNewPatches = nameToPatch.size();
}
else
{
Info<< "Surface " << inFileName1
<< " has " << surface1.patches().size()
<< " regions"
<< nl
<< "All region numbers in " << inFileName2 << " will be offset"
<< " by this amount" << nl << endl;
patch1Map = identity(surface1.patches().size());
patch2Map = identity(surface2.patches().size(), patch1Map.size());
nNewPatches = surface1.patches().size()+surface2.patches().size();
}
// Copy triangles1 into trianglesAll
for (const labelledTri& tri : surface1)
{
labelledTri& destTri = facesAll[trianglei++];
destTri.triFace::operator=(tri);
destTri.region() = patch1Map[tri.region()];
}
// Add (renumbered) surface2 triangles
for (const labelledTri& tri : surface2)
{
labelledTri& destTri = facesAll[trianglei++];
destTri[0] = tri[0] + points1.size();
destTri[1] = tri[1] + points1.size();
destTri[2] = tri[2] + points1.size();
destTri.region() = patch2Map[tri.region()];
}
geometricSurfacePatchList newPatches(nNewPatches);
forAll(surface1.patches(), patchi)
{
newPatches[patch1Map[patchi]] = surface1.patches()[patchi];
}
forAll(surface2.patches(), patchi)
{
newPatches[patch2Map[patchi]] = surface2.patches()[patchi];
}
Info<< "New patches:" << nl;
forAll(newPatches, patchi)
{
Info<< " " << patchi << '\t' << newPatches[patchi].name() << nl;
}
Info<< endl;
// Construct new surface mesh
combinedSurf = triSurface(facesAll, newPatches, pointsAll);
}
// Merge all common points and do some checks
combinedSurf.cleanup(true);
Info<< "Merged surface:" << endl;
combinedSurf.writeStats(Info);
Info<< endl;
Info<< "Writing : " << outFileName << endl;
// If merging regions also sort
combinedSurf.write(outFileName, mergeRegions);
Info<< "End\n" << endl;
return 0;
}
// ************************************************************************* //