gmio/src/gmio_support/occ_libstl.cpp

/****************************************************************************
** GeomIO Library
** Copyright FougSys (2 Mar. 2015)
** contact@fougsys.fr
**
** This software is a reusable library whose purpose is to provide complete
** I/O support for various CAD file formats (eg. STL)
**
** This software is governed by the CeCILL-B license under French law and
** abiding by the rules of distribution of free software.  You can  use,
** modify and/ or redistribute the software under the terms of the CeCILL-B
** license as circulated by CEA, CNRS and INRIA at the following URL
** "http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html".
****************************************************************************/

#include <gmio_support/occ_libstl.h>

#include <cstring>
#include <StlMesh_Mesh.hxx>
#include <StlMesh_MeshTriangle.hxx>
#include <StlMesh_SequenceOfMeshTriangle.hxx>
#include <TColgp_SequenceOfXYZ.hxx>

namespace internal {

/* Common */

static StlMesh_Mesh* occMeshPtr(const Handle_StlMesh_Mesh& mesh)
{
    return mesh.operator->();
}

static void occmesh_add_triangle(void* cookie,
                                 uint32_t tri_id,
                                 const gmio_stl_triangle_t* tri)
{
    StlMesh_Mesh* mesh = static_cast<StlMesh_Mesh*>(cookie);
    if (tri_id == 0)
        mesh->AddDomain();
    mesh->AddTriangle(mesh->AddOnlyNewVertex(tri->v1.x, tri->v1.y, tri->v1.z),
                      mesh->AddOnlyNewVertex(tri->v2.x, tri->v2.y, tri->v2.z),
                      mesh->AddOnlyNewVertex(tri->v3.x, tri->v3.y, tri->v3.z),
                      tri->normal.x, tri->normal.y, tri->normal.z);
}

static void occmesh_get_triangle(
        const void* cookie, uint32_t tri_id, gmio_stl_triangle_t* triangle)
{
    const gmio_OccStlMeshDomain* meshCookie =
            static_cast<const gmio_OccStlMeshDomain*>(cookie);
    const StlMesh_SequenceOfMeshTriangle& occTriangles =
            meshCookie->mesh()->Triangles(meshCookie->domainId());
    const Handle_StlMesh_MeshTriangle& occTri =
            occTriangles.Value(tri_id + 1);
    Standard_Integer v1;
    Standard_Integer v2;
    Standard_Integer v3;
    Standard_Real xN;
    Standard_Real yN;
    Standard_Real zN;
    occTri->GetVertexAndOrientation(v1, v2, v3, xN, yN, zN);
    triangle->normal.x = float(xN);
    triangle->normal.y = float(yN);
    triangle->normal.z = float(zN);

    const TColgp_SequenceOfXYZ& vertices =
            meshCookie->mesh()->Vertices(meshCookie->domainId());
    const gp_XYZ& coordsV1 = vertices.Value(v1);
    const gp_XYZ& coordsV2 = vertices.Value(v2);
    const gp_XYZ& coordsV3 = vertices.Value(v3);
    triangle->v1.x = float(coordsV1.X());
    triangle->v2.x = float(coordsV2.X());
    triangle->v3.x = float(coordsV3.X());

    triangle->v1.y = float(coordsV1.Y());
    triangle->v2.y = float(coordsV2.Y());
    triangle->v3.y = float(coordsV3.Y());

    triangle->v1.z = float(coordsV1.Z());
    triangle->v2.z = float(coordsV2.Z());
    triangle->v3.z = float(coordsV3.Z());
}

} // namespace internal

void gmio_stl_set_occmesh(
        gmio_stl_mesh_t *mesh, const gmio_OccStlMeshDomain &meshCookie)
{
    std::memset(mesh, 0, sizeof(gmio_stl_mesh_t));
    mesh->cookie = &meshCookie;
    mesh->triangle_count = meshCookie.mesh()->NbTriangles(meshCookie.domainId());
    mesh->get_triangle_func = internal::occmesh_get_triangle;
}

gmio_stl_mesh gmio_stl_occmesh(const gmio_OccStlMeshDomain &meshCookie)
{
    gmio_stl_mesh occmesh = { 0 };
    gmio_stl_set_occmesh(&occmesh, meshCookie);
    return occmesh;
}

void gmio_stl_set_occmesh_creator(
        gmio_stl_mesh_creator_t *creator, const Handle_StlMesh_Mesh &mesh)
{
    std::memset(creator, 0, sizeof(gmio_stl_mesh_creator_t));
    creator->cookie = internal::occMeshPtr(mesh);
    creator->add_triangle_func = internal::occmesh_add_triangle;
}

gmio_stl_mesh_creator gmio_stl_occmesh_creator(const Handle_StlMesh_Mesh &mesh)
{
    gmio_stl_mesh_creator occ_creator = { 0 };
    gmio_stl_set_occmesh_creator(&occ_creator, mesh);
    return occ_creator;
}

gmio_OccStlMeshDomain::gmio_OccStlMeshDomain(
        const Handle_StlMesh_Mesh &stlMesh, int domId)
    : m_mesh(stlMesh),
      m_domainId(domId)
{
}

const Handle_StlMesh_Mesh &gmio_OccStlMeshDomain::mesh() const
{
    return m_mesh;
}

void gmio_OccStlMeshDomain::setMesh(const Handle_StlMesh_Mesh &stlMesh)
{
    m_mesh = stlMesh;
}

int gmio_OccStlMeshDomain::domainId() const
{
    return m_domainId;
}

void gmio_OccStlMeshDomain::setDomainId(int domId)
{
    m_domainId = domId;
}