extrude2DMeshApp.C

Go to the documentation of this file.

/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  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 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
    extrude2DMesh
 
Group
    grpMeshGenerationUtilities
 
Description
    Create a 3D mesh by extruding a 2D mesh with specified thickness.
    For the 2D mesh, all faces are 2 points only, no front and back faces.
 
Note
    Not sure about the walking of the faces to create the front and back faces.
 
\*---------------------------------------------------------------------------*/
 
#include "argList.H"
#include "Time.H"
#include "polyMesh.H"
#include "extrude2DMesh.H"
#include "extrudeModel.H"
#include "polyTopoChange.H"
#include "MeshedSurface.H"
#include "edgeCollapser.H"
#include "patchToPoly2DMesh.H"
#include "globalIndex.H"
#include "topoSet.H"
#include "processorMeshes.H"
 
using namespace Foam;
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 
enum ExtrudeMode
{
    POLYMESH2D,
    MESHEDSURFACE
};
 
static const Enum<ExtrudeMode> ExtrudeModeNames
{
    { ExtrudeMode::POLYMESH2D, "polyMesh2D" },
    { ExtrudeMode::MESHEDSURFACE, "MeshedSurface" },
};
 
 
//pointField moveInitialPoints
//(
//    primitiveFacePatch& fMesh,
//    const extrudeModel& model
//)
//{
//    pointField layer0Points(fMesh.nPoints());
//    pointField layer1Points(fMesh.nPoints());
//    pointField displacement(fMesh.nPoints());
 
//    forAll(layer0Points, pointi)
//    {
//        const labelList& meshPoints = fMesh.meshPoints();
//        label meshPointi = meshPoints[pointi];
 
//        layer0Points[meshPointi] = model
//        (
//            fMesh.points()[meshPointi],
//            fMesh.pointNormals()[pointi],
//            0
//        );
 
//        layer1Points[meshPointi] = model
//        (
//            fMesh.points()[meshPointi],
//            fMesh.pointNormals()[pointi],
//            1
//        );
 
//        displacement[pointi] =
//            layer1Points[meshPointi]
//          - layer0Points[meshPointi];
//    }
 
//    fMesh.movePoints(layer0Points);
 
//    return displacement;
//}
 
 
 
int main(int argc, char *argv[])
{
    argList::addNote
    (
        "Create a 3D mesh from a 2D mesh by extruding with specified thickness"
    );
 
    argList::addArgument("surfaceFormat");
 
    #include "addOverwriteOption.H"
 
    argList::noFunctionObjects();  // Never use function objects
 
    #include "setRootCase.H"
 
    Info<< "Create time\n" << endl;
 
    Time runTimeExtruded
    (
        Time::controlDictName,
        args.rootPath(),
        args.caseName()
    );
 
    // For safety
    runTimeExtruded.functionObjects().off();
 
    const ExtrudeMode surfaceFormat = ExtrudeModeNames[args[1]];
    const bool overwrite = args.found("overwrite");
 
    Info<< "Extruding from " << ExtrudeModeNames[surfaceFormat]
        << " at time " << runTimeExtruded.timeName() << endl;
 
    IOdictionary extrude2DMeshDict
    (
        IOobject
        (
            "extrude2DMeshDict",
            runTimeExtruded.system(),
            runTimeExtruded,
            IOobject::MUST_READ,
            IOobject::NO_WRITE,
            IOobject::NO_REGISTER
        )
    );
 
    // Point generator
    autoPtr<extrudeModel> model(extrudeModel::New(extrude2DMeshDict));
 
    autoPtr<MeshedSurface<face>> fMesh;
 
    autoPtr<polyMesh> mesh;
 
    autoPtr<polyTopoChange> meshMod;
 
    labelListList extrudeEdgePatches;
 
    if (surfaceFormat == MESHEDSURFACE)
    {
        fMesh.reset(new MeshedSurface<face>("MeshedSurface.obj"));
 
        EdgeMap<label> edgeRegionMap;
        wordList patchNames(1, "default");
        labelList patchSizes(1, fMesh().nEdges() - fMesh().nInternalEdges());
 
        const edgeList& edges = fMesh().edges();
        forAll(edges, edgeI)
        {
            if (!fMesh().isInternalEdge(edgeI))
            {
                edgeRegionMap.insert(edges[edgeI], 0);
            }
        }
 
        patchToPoly2DMesh poly2DMesh
        (
            fMesh(),
            patchNames,
            patchSizes,
            edgeRegionMap
        );
 
        poly2DMesh.createMesh();
 
        mesh = autoPtr<polyMesh>::New
        (
            IOobject
            (
                polyMesh::defaultRegion,
                runTimeExtruded.constant(),
                runTimeExtruded,
                IOobject::NO_READ,
                IOobject::NO_WRITE,
                IOobject::NO_REGISTER
            ),
            std::move(poly2DMesh.points()),
            std::move(poly2DMesh.faces()),
            std::move(poly2DMesh.owner()),
            std::move(poly2DMesh.neighbour())
        );
 
        Info<< "Constructing patches." << endl;
        List<polyPatch*> patches(poly2DMesh.patchNames().size());
 
        forAll(patches, patchi)
        {
            patches[patchi] = new polyPatch
            (
                poly2DMesh.patchNames()[patchi],
                poly2DMesh.patchSizes()[patchi],
                poly2DMesh.patchStarts()[patchi],
                patchi,
                mesh().boundaryMesh(),
                polyPatch::typeName
            );
        }
 
        mesh().addPatches(patches);
    }
    else if (surfaceFormat == POLYMESH2D)
    {
        mesh = autoPtr<polyMesh>::New
        (
            IOobject
            (
                polyMesh::defaultRegion,
                runTimeExtruded.timeName(),
                runTimeExtruded,
                IOobject::MUST_READ
            )
        );
    }
 
    // Engine to extrude mesh
    extrude2DMesh extruder(mesh(), extrude2DMeshDict, model());
 
    extruder.addFrontBackPatches();
 
    meshMod.reset(new polyTopoChange(mesh().boundaryMesh().size()));
 
    extruder.setRefinement(meshMod());
 
    // Create a mesh from topo changes.
    autoPtr<mapPolyMesh> morphMap = meshMod().changeMesh(mesh(), false);
 
    mesh().updateMesh(morphMap());
 
    {
        edgeCollapser collapser(mesh());
 
        const edgeList& edges = mesh().edges();
        const pointField& points = mesh().points();
 
        const boundBox& bb = mesh().bounds();
        const scalar mergeDim = 1e-4 * bb.minDim();
 
        bitSet collapseEdge(mesh().nEdges());
        Map<point> collapsePointToLocation(mesh().nPoints());
 
        forAll(edges, edgeI)
        {
            const edge& e = edges[edgeI];
 
            scalar d = e.mag(points);
 
            if (d < mergeDim)
            {
                Info<< "Merging edge " << e << " since length " << d
                    << " << " << mergeDim << nl;
 
                collapseEdge.set(edgeI);
                collapsePointToLocation.set(e[1], points[e[0]]);
            }
        }
 
        List<pointEdgeCollapse> allPointInfo;
        const globalIndex globalPoints(mesh().nPoints());
        labelList pointPriority(mesh().nPoints(), Zero);
 
        collapser.consistentCollapse
        (
            globalPoints,
            pointPriority,
            collapsePointToLocation,
            collapseEdge,
            allPointInfo
        );
 
        polyTopoChange meshModCollapse(mesh());
 
        collapser.setRefinement(allPointInfo, meshModCollapse);
 
        // Create a mesh from topo changes.
        autoPtr<mapPolyMesh> morphMap
            = meshModCollapse.changeMesh(mesh(), false);
 
        mesh().updateMesh(morphMap());
    }
 
    if (!overwrite)
    {
        ++runTimeExtruded;
    }
    else
    {
        mesh().setInstance("constant");
    }
 
    // Take over refinement levels and write to new time directory.
    Info<< "\nWriting extruded mesh to time = " << runTimeExtruded.timeName()
        << nl << endl;
 
    mesh().write();
    topoSet::removeFiles(mesh());
    processorMeshes::removeFiles(mesh());
 
    Info<< "End\n" << endl;
 
    return 0;
}
 
 
// ************************************************************************* //