259 lines
6.3 KiB
C++
259 lines
6.3 KiB
C++
/*---------------------------------------------------------------------------*\
|
|
========= |
|
|
\\ / F ield | foam-extend: Open Source CFD
|
|
\\ / O peration | Version: 4.0
|
|
\\ / A nd | Web: http://www.foam-extend.org
|
|
\\/ M anipulation | For copyright notice see file Copyright
|
|
-------------------------------------------------------------------------------
|
|
License
|
|
This file is part of foam-extend.
|
|
|
|
foam-extend 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.
|
|
|
|
foam-extend 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 foam-extend. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
Description
|
|
Divides external faces into patches based on (user supplied) feature
|
|
angle.
|
|
|
|
\*---------------------------------------------------------------------------*/
|
|
|
|
#include "argList.H"
|
|
#include "polyMesh.H"
|
|
#include "foamTime.H"
|
|
#include "boundaryMesh.H"
|
|
#include "repatchPolyTopoChanger.H"
|
|
#include "mathematicalConstants.H"
|
|
#include "OFstream.H"
|
|
#include "ListOps.H"
|
|
|
|
using namespace Foam;
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
// Get all feature edges.
|
|
void collectFeatureEdges(const boundaryMesh& bMesh, labelList& markedEdges)
|
|
{
|
|
markedEdges.setSize(bMesh.mesh().nEdges());
|
|
|
|
label markedI = 0;
|
|
|
|
forAll(bMesh.featureSegments(), i)
|
|
{
|
|
const labelList& segment = bMesh.featureSegments()[i];
|
|
|
|
forAll(segment, j)
|
|
{
|
|
label featEdgeI = segment[j];
|
|
|
|
label meshEdgeI = bMesh.featureToEdge()[featEdgeI];
|
|
|
|
markedEdges[markedI++] = meshEdgeI;
|
|
}
|
|
}
|
|
markedEdges.setSize(markedI);
|
|
}
|
|
|
|
|
|
// Main program:
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
argList::noParallel();
|
|
argList::validArgs.append("feature angle[0-180]");
|
|
argList::validOptions.insert("overwrite", "");
|
|
|
|
# include "setRootCase.H"
|
|
# include "createTime.H"
|
|
runTime.functionObjects().off();
|
|
# include "createPolyMesh.H"
|
|
const word oldInstance = mesh.pointsInstance();
|
|
|
|
Info<< "Mesh read in = "
|
|
<< runTime.cpuTimeIncrement()
|
|
<< " s\n" << endl << endl;
|
|
|
|
|
|
//
|
|
// Use boundaryMesh to reuse all the featureEdge stuff in there.
|
|
//
|
|
|
|
boundaryMesh bMesh;
|
|
|
|
scalar featureAngle(readScalar(IStringStream(args.additionalArgs()[0])()));
|
|
bool overwrite = args.optionFound("overwrite");
|
|
|
|
scalar minCos = Foam::cos(featureAngle * mathematicalConstant::pi/180.0);
|
|
|
|
Info<< "Feature:" << featureAngle << endl
|
|
<< "minCos :" << minCos << endl
|
|
<< endl;
|
|
|
|
bMesh.read(mesh);
|
|
|
|
// Set feature angle (calculate feature edges)
|
|
bMesh.setFeatureEdges(minCos);
|
|
|
|
// Collect all feature edges as edge labels
|
|
labelList markedEdges;
|
|
|
|
collectFeatureEdges(bMesh, markedEdges);
|
|
|
|
|
|
|
|
// (new) patch ID for every face in mesh.
|
|
labelList patchIDs(bMesh.mesh().size(), -1);
|
|
|
|
//
|
|
// Fill patchIDs with values for every face by floodfilling without
|
|
// crossing feature edge.
|
|
//
|
|
|
|
// Current patch number.
|
|
label newPatchI = bMesh.patches().size();
|
|
|
|
label suffix = 0;
|
|
|
|
while (true)
|
|
{
|
|
// Find first unset face.
|
|
label unsetFaceI = findIndex(patchIDs, -1);
|
|
|
|
if (unsetFaceI == -1)
|
|
{
|
|
// All faces have patchID set. Exit.
|
|
break;
|
|
}
|
|
|
|
// Found unset face. Create patch for it.
|
|
word patchName;
|
|
do
|
|
{
|
|
patchName = "auto" + name(suffix++);
|
|
}
|
|
while (bMesh.findPatchID(patchName) != -1);
|
|
|
|
bMesh.addPatch(patchName);
|
|
|
|
bMesh.changePatchType(patchName, "patch");
|
|
|
|
|
|
// Fill visited with all faces reachable from unsetFaceI.
|
|
boolList visited(bMesh.mesh().size());
|
|
|
|
bMesh.markFaces(markedEdges, unsetFaceI, visited);
|
|
|
|
|
|
// Assign all visited faces to current patch
|
|
label nVisited = 0;
|
|
|
|
forAll(visited, faceI)
|
|
{
|
|
if (visited[faceI])
|
|
{
|
|
nVisited++;
|
|
|
|
patchIDs[faceI] = newPatchI;
|
|
}
|
|
}
|
|
|
|
Info<< "Assigned " << nVisited << " faces to patch " << patchName
|
|
<< endl << endl;
|
|
|
|
newPatchI++;
|
|
}
|
|
|
|
|
|
|
|
const PtrList<boundaryPatch>& patches = bMesh.patches();
|
|
|
|
// Create new list of patches with old ones first
|
|
List<polyPatch*> newPatchPtrList(patches.size());
|
|
|
|
newPatchI = 0;
|
|
|
|
// Copy old patches
|
|
forAll(mesh.boundaryMesh(), patchI)
|
|
{
|
|
const polyPatch& patch = mesh.boundaryMesh()[patchI];
|
|
|
|
newPatchPtrList[newPatchI] =
|
|
patch.clone
|
|
(
|
|
mesh.boundaryMesh(),
|
|
newPatchI,
|
|
patch.size(),
|
|
patch.start()
|
|
).ptr();
|
|
|
|
newPatchI++;
|
|
}
|
|
|
|
// Add new ones with empty size.
|
|
for (label patchI = newPatchI; patchI < patches.size(); patchI++)
|
|
{
|
|
const boundaryPatch& bp = patches[patchI];
|
|
|
|
newPatchPtrList[newPatchI] = polyPatch::New
|
|
(
|
|
polyPatch::typeName,
|
|
bp.name(),
|
|
0,
|
|
mesh.nFaces(),
|
|
newPatchI,
|
|
mesh.boundaryMesh()
|
|
).ptr();
|
|
|
|
newPatchI++;
|
|
}
|
|
|
|
if (!overwrite)
|
|
{
|
|
runTime++;
|
|
}
|
|
|
|
|
|
// Change patches
|
|
repatchPolyTopoChanger polyMeshRepatcher(mesh);
|
|
polyMeshRepatcher.changePatches(newPatchPtrList);
|
|
|
|
|
|
// Change face ordering
|
|
|
|
// Since bMesh read from mesh there is one to one mapping so we don't
|
|
// have to do the geometric stuff.
|
|
const labelList& meshFace = bMesh.meshFace();
|
|
|
|
forAll(patchIDs, faceI)
|
|
{
|
|
label meshFaceI = meshFace[faceI];
|
|
|
|
polyMeshRepatcher.changePatchID(meshFaceI, patchIDs[faceI]);
|
|
}
|
|
|
|
polyMeshRepatcher.repatch();
|
|
|
|
// Write resulting mesh
|
|
if (overwrite)
|
|
{
|
|
mesh.setInstance(oldInstance);
|
|
}
|
|
mesh.write();
|
|
|
|
|
|
Info<< "End\n" << endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
// ************************************************************************* //
|