FMSToSurface.C

Go to the documentation of this file.

/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  F ield         | cfMesh: A library for mesh generation
   \\    /   O peration     |
    \\  /    A nd           | Author: Franjo Juretic (franjo.juretic@c-fields.com)
     \\/     M anipulation  | Copyright (C) Creative Fields, Ltd.
-------------------------------------------------------------------------------
License
    This file is part of cfMesh.
 
    cfMesh 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.
 
    cfMesh 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 cfMesh.  If not, see <http://www.gnu.org/licenses/>.
 
Description
    Creates surface patches from surface subsets
 
\*---------------------------------------------------------------------------*/
 
#include "argList.H"
#include "triSurf.H"
#include "triSurfaceCopyParts.H"
#include "demandDrivenData.H"
#include "OFstream.H"
 
using namespace Foam;
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 
void exportFeatureEdges
(
    const triSurf& origSurf,
    const fileName& edgeFileName
)
{
    OFstream file(edgeFileName);
 
    const pointField& points = origSurf.points();
    labelList newPointLabel(points.size(), -1);
    label nPoints(0);
 
    const edgeLongList& featureEdges = origSurf.featureEdges();
    forAll(featureEdges, feI)
    {
        const edge& e = featureEdges[feI];
 
        if( newPointLabel[e[0]] == -1 )
            newPointLabel[e[0]] = nPoints++;
        if( newPointLabel[e[1]] == -1 )
            newPointLabel[e[1]] = nPoints++;
    }
 
    pointField pCopy(nPoints);
    forAll(newPointLabel, pI)
    {
        if( newPointLabel[pI] < 0 )
            continue;
 
        pCopy[newPointLabel[pI]] = points[pI];
    }
 
    //- write the header
    file << "# vtk DataFile Version 3.0\n";
    file << "vtk output\n";
    file << "ASCII\n";
    file << "DATASET POLYDATA\n";
 
    //- write points
    file << "POINTS " << pCopy.size() << " float\n";
    forAll(pCopy, pI)
    {
        const point& p = pCopy[pI];
        file << p.x() << ' ' << p.y() << ' ' << p.z() << '\n';
    }
 
    file << "\nLINES " << featureEdges.size()
         << ' ' << 3*featureEdges.size() << nl;
    forAll(featureEdges, edgeI)
    {
        const edge& e = featureEdges[edgeI];
        file << "2 " << newPointLabel[e[0]]
             << token::SPACE << newPointLabel[e[1]] << nl;
    }
    file << nl;
 
    if( !file )
        FatalErrorIn
        (
            "void exportFeatureEdges(const triSurf&, const fileName&)"
        ) << "Writting of feature edges failed!" << exit(FatalError);
}
 
int main(int argc, char *argv[])
{
    argList::noParallel();
    argList::validArgs.clear();
 
    argList::validArgs.append("input surface file");
    argList::validArgs.append("output surface file");
    argList::validOptions.insert("exportSubsets", "");
    argList::validOptions.insert("exportFeatureEdges", "");
    argList args(argc, argv);
 
    fileName inFileName(args.args()[1]);
    fileName outFileName(args.args()[2]);
 
    fileName outFileNoExt = outFileName.lessExt();
    fileName outExtension = outFileName.ext();
 
    Info << "Out file no ext " << outFileNoExt << endl;
    Info << "Extension " << outExtension << endl;
 
    //- read the inout surface
    triSurf origSurf(inFileName);
 
    //- write the surface in the requated format
    origSurf.writeSurface(outFileName);
 
    //- export surface subsets as separate surface meshes
    if( args.options().found("exportSubsets") )
    {
        DynList<label> subsetIDs;
        origSurf.facetSubsetIndices(subsetIDs);
 
        triSurfaceCopyParts copyParts(origSurf);
 
        forAll(subsetIDs, subsetI)
        {
            //- get the name of the subset
            triSurf copySurf;
            wordList subsetName(1);
            subsetName[0] = origSurf.facetSubsetName(subsetIDs[subsetI]);
 
            //- create a surface mesh corresponding to the subset
            copyParts.copySurface(subsetName, copySurf);
 
            //- write the mesh on disk
            fileName fName = outFileNoExt+"_facetSubset_"+subsetName[0];
            fName += '.'+outExtension;
 
            copySurf.writeSurface(fName);
        }
    }
 
    if( args.options().found("exportFeatureEdges") )
    {
        fileName fName = outFileNoExt+"_featureEdges";
        fName += ".vtk";
        exportFeatureEdges(origSurf, fName);
    }
 
    Info << "End\n" << endl;
    return 0;
}
 
// ************************************************************************* //