TriangleSetTopologyContainer.h

Go to the documentation of this file.

/******************************************************************************
*       SOFA, Simulation Open-Framework Architecture, version 1.0 beta 3      *
*                (c) 2006-2008 MGH, INRIA, USTL, UJF, CNRS                    *
*                                                                             *
* This library is free software; you can redistribute it and/or modify it     *
* under the terms of the GNU Lesser General Public License as published by    *
* the Free Software Foundation; either version 2.1 of the License, or (at     *
* your option) any later version.                                             *
*                                                                             *
* This library 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 Lesser General Public License *
* for more details.                                                           *
*                                                                             *
* You should have received a copy of the GNU Lesser General Public License    *
* along with this library; if not, write to the Free Software Foundation,     *
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.          *
*******************************************************************************
*                               SOFA :: Modules                               *
*                                                                             *
* Authors: The SOFA Team and external contributors (see Authors.txt)          *
*                                                                             *
* Contact information: contact@sofa-framework.org                             *
******************************************************************************/
#ifndef SOFA_COMPONENT_TOPOLOGY_TRIANGLESETTOPOLOGYCONTAINER_H
#define SOFA_COMPONENT_TOPOLOGY_TRIANGLESETTOPOLOGYCONTAINER_H

#include <sofa/component/topology/EdgeSetTopologyContainer.h>  

namespace sofa
{

namespace component
{

namespace topology
{
    class TriangleSetTopologyModifier;

    using core::componentmodel::topology::BaseMeshTopology;

    typedef BaseMeshTopology::PointID           PointID;
    typedef BaseMeshTopology::EdgeID            EdgeID;
    typedef BaseMeshTopology::TriangleID        TriangleID;
    typedef BaseMeshTopology::Edge              Edge;
    typedef BaseMeshTopology::Triangle          Triangle;
    typedef BaseMeshTopology::SeqTriangles      SeqTriangles;
    typedef BaseMeshTopology::TriangleEdges     TriangleEdges;
    typedef BaseMeshTopology::VertexTriangles   VertexTriangles;
    typedef BaseMeshTopology::EdgeTriangles     EdgeTriangles;

    class TriangleSetTopologyContainer : public EdgeSetTopologyContainer 
    {
        friend class TriangleSetTopologyModifier;

    public:

        TriangleSetTopologyContainer();

        TriangleSetTopologyContainer(const sofa::helper::vector< Triangle > &triangles );

        virtual ~TriangleSetTopologyContainer() {}

        virtual void clear();
        virtual void addTriangle( int a, int b, int c );

        virtual void init();


        const SeqTriangles& getTriangles()
        {
            return getTriangleArray();
        }

        virtual int getTriangleIndex(PointID v1, PointID v2, PointID v3);

        const TriangleEdges& getEdgeTriangleShell(TriangleID i)
        {
            return getTriangleEdge(i);
        }

        virtual const VertexTriangles& getTriangleVertexShell(PointID i);


        virtual const EdgeTriangles& getTriangleEdgeShell(EdgeID i) ;

        virtual int getVertexIndexInTriangle(const Triangle &t, PointID vertexIndex) const;

        virtual int getEdgeIndexInTriangle(const TriangleEdges &t, EdgeID edgeIndex) const;


        virtual bool checkTopology() const;

        const sofa::helper::vector<Triangle> &getTriangleArray();

        const sofa::helper::vector< sofa::helper::vector<unsigned int> > &getTriangleVertexShellArray();

        const sofa::helper::vector< TriangleEdges > &getTriangleEdgeArray() ;

        const sofa::helper::vector< sofa::helper::vector<unsigned int> > &getTriangleEdgeShellArray() ;

        unsigned int getNumberOfTriangles() const;


        const TriangleEdges &getTriangleEdge(const unsigned int i) ;

        inline friend std::ostream& operator<< (std::ostream& out, const TriangleSetTopologyContainer& t)
        {
            out << t.m_triangle.size() << " " << t.m_triangle << " "
              << t.m_triangleEdge.size() << " " << t.m_triangleEdge << " "
              << t.m_triangleVertexShell.size();
            for (unsigned int i=0; i<t.m_triangleVertexShell.size(); i++)
            {
              out << " " << t.m_triangleVertexShell[i].size();
              out << " " <<t.m_triangleVertexShell[i] ;
            }
            out  << " " << t.m_triangleEdgeShell.size();
            for (unsigned int i=0; i<t.m_triangleEdgeShell.size(); i++)
            {
              out  << " " << t.m_triangleEdgeShell[i].size();
              out  << " " << t.m_triangleEdgeShell[i];
            }

            return out;
        }

        inline friend std::istream& operator>>(std::istream& in, TriangleSetTopologyContainer& t)
        {
            unsigned int s;
            in >> s;
            for (unsigned int i=0;i<s;i++) 
            {
              Triangle T; in >> T;
              t.m_triangle.push_back(T);
            }
            in >> s;
            for (unsigned int i=0;i<s;i++) 
            {
              TriangleEdges T; in >> T;
              t.m_triangleEdge.push_back(T);
            }

            unsigned int sub;
            in >> s;
            for (unsigned int i=0;i<s;i++)
            {
              in >> sub;
              sofa::helper::vector< unsigned int > v;
              for (unsigned int j=0;j<sub;j++)
            {
              unsigned int value;
              in >> value;
              v.push_back(value);
            }
              t.m_triangleVertexShell.push_back(v);
            }

            in >> s;
            for (unsigned int i=0;i<s;i++)
            {
              in >> sub;
              sofa::helper::vector< unsigned int > v;
              for (unsigned int j=0;j<sub;j++)
            {
              unsigned int value;
              in >> value;
              v.push_back(value);
            }
              t.m_triangleEdgeShell.push_back(v);
            }  

                return in;
        }

    protected:
        virtual void createTriangleSetArray();

        virtual void createEdgeSetArray();

        bool hasTriangles() const;

        bool hasTriangleEdges() const;

        bool hasTriangleVertexShell() const;

        bool hasTriangleEdgeShell() const;

        void clearTriangles();

        void clearTriangleEdges();

        void clearTriangleVertexShell();

        void clearTriangleEdgeShell();

    private:
        void createTriangleEdgeArray();
        void createTriangleVertexShellArray();

        void createTriangleEdgeShellArray();

        sofa::helper::vector< unsigned int > &getTriangleVertexShellForModification(const unsigned int vertexIndex); 
        sofa::helper::vector< unsigned int > &getTriangleEdgeShellForModification(const unsigned int edgeIndex); 

    protected:
        sofa::helper::vector<Triangle> m_triangle;
        sofa::helper::vector<TriangleEdges> m_triangleEdge;
        sofa::helper::vector< sofa::helper::vector< unsigned int > > m_triangleVertexShell;
        sofa::helper::vector< sofa::helper::vector< unsigned int > > m_triangleEdgeShell;       

        virtual void loadFromMeshLoader(sofa::component::MeshLoader* loader);
    };

} // namespace topology

} // namespace component

} // namespace sofa

#endif