io_stl.cpp

Go to the documentation of this file.

/*
 * Copyright (c) 2012-2017, Association Scientifique pour la Geologie et ses
 * Applications (ASGA). All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Neither the name of ASGA nor the
 *       names of its contributors may be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL ASGA BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 *     http://www.ring-team.org
 *
 *     RING Project
 *     Ecole Nationale Superieure de Geologie - GeoRessources
 *     2 Rue du Doyen Marcel Roubault - TSA 70605
 *     54518 VANDOEUVRE-LES-NANCY
 *     FRANCE
 */

namespace {

    void save_header( const GeoModel3D& geomodel, std::ostream& out )
    {
        out << "solid " << geomodel.name() << EOL;
    }

    void save_footer( const GeoModel3D& geomodel, std::ostream& out )
    {
        out << "endsolid" << geomodel.name() << EOL;
    }

    void save_normal(
        const GeoModel3D& geomodel,
        index_t triangle_id,
        std::ostream& out )
    {
        out << "facet normal " << geomodel.mesh.polygons.normal( triangle_id )
            << EOL;
    }

    void begin_triangle( std::ostream& out )
    {
        out << "outer loop" << EOL;
    }

    void end_triangle( std::ostream& out )
    {
        out << "endloop" << EOL;
    }

    void save_triangle_vertex(
        const GeoModel3D& geomodel,
        index_t triangle_id,
        index_t local_vertex_id,
        std::ostream& out )
    {
        out << "vertex "
            << geomodel.mesh.vertices.vertex(
                geomodel.mesh.polygons.vertex(
                    ElementLocalVertex( triangle_id, local_vertex_id ) ) )
            << EOL;
    }

    void save_triangle(
        const GeoModel3D& geomodel,
        index_t triangle_id,
        std::ostream& out )
    {
        save_normal( geomodel, triangle_id, out );
        begin_triangle( out );
        for( auto vertex : range( 3 ) ) {
            save_triangle_vertex( geomodel, triangle_id, vertex, out );
        }
        end_triangle( out );
    }

    void save_triangles( const GeoModel3D& geomodel, std::ostream& out )
    {
        for( auto triangle : range( geomodel.mesh.polygons.nb_triangle() ) ) {
            save_triangle( geomodel, triangle, out );
        }
    }

    void check_stl_validity( const GeoModel3D& geomodel )
    {
        if( geomodel.mesh.polygons.nb() != geomodel.mesh.polygons.nb_triangle() ) {
            throw RINGMeshException( "I/O",
                "Geological model save in STL format support only triangles" );
        }
    }

    class STLIOHandler final: public GeoModelIOHandler< 3 > {
    public:
        void load( const std::string& filename, GeoModel3D& geomodel ) final
        {
            ringmesh_unused( filename );
            ringmesh_unused( geomodel );
            throw RINGMeshException( "I/O",
                "Geological model loading of a from STL mesh not yet implemented" );
        }

        void save(
            const GeoModel3D& geomodel,
            const std::string& filename ) final
        {
            check_stl_validity( geomodel );
            std::ofstream out( filename.c_str() );
            out.precision( 17 );
            save_header( geomodel, out );
            save_triangles( geomodel, out );
            save_footer( geomodel, out );
            out << std::flush;
        }
    };
}