/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  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;
}

// ************************************************************************* //