EdgeSetTopologyContainer.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_EDGESETTOPOLOGYCONTAINER_H
#define SOFA_COMPONENT_TOPOLOGY_EDGESETTOPOLOGYCONTAINER_H

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

namespace sofa
{

namespace component
{

namespace topology
{   
    using core::componentmodel::topology::BaseMeshTopology;

    typedef BaseMeshTopology::PointID           PointID;
    typedef BaseMeshTopology::EdgeID            EdgeID;
    typedef BaseMeshTopology::Edge              Edge;
    typedef BaseMeshTopology::SeqEdges          SeqEdges;
    typedef BaseMeshTopology::VertexEdges       VertexEdges;

    class EdgeSetTopologyContainer : public PointSetTopologyContainer 
    {
        friend class EdgeSetTopologyModifier;

    public:        
        EdgeSetTopologyContainer();

        EdgeSetTopologyContainer(const sofa::helper::vector< Edge > &edges);

        virtual ~EdgeSetTopologyContainer() {}

        virtual void clear();
        virtual void addEdge( int a, int b );

        virtual void init();


        virtual const SeqEdges& getEdges()
        {
            return getEdgeArray();
        }

        virtual const VertexEdges& getEdgeVertexShell(PointID i);

        virtual int getEdgeIndex(PointID v1, PointID v2);


        virtual bool checkTopology() const;

        unsigned int getNumberOfEdges() const; 

        virtual const sofa::helper::vector<Edge> &getEdgeArray();


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


        virtual int getNumberConnectedComponents(sofa::helper::vector<unsigned int>& components); 

        inline friend std::ostream& operator<< (std::ostream& out, const EdgeSetTopologyContainer& t)
        {
            out << t.m_edge.size();
            for (unsigned int i=0; i<t.m_edge.size(); i++)
                out << " " << t.m_edge[i][0] << " " << t.m_edge[i][1] ;

            return out;
        }

        inline friend std::istream& operator>>(std::istream& in, EdgeSetTopologyContainer& t)
        {
            unsigned int s;
            in >> s;
            for (unsigned int i=0; i<s; i++)
            {
                Edge T; in >> T;
                t.m_edge.push_back(T);
            }
            return in;
        }

        virtual sofa::helper::vector< unsigned int > &getEdgeVertexShellForModification(const unsigned int i); 

        virtual void createEdgeSetArray();

        virtual void createEdgeVertexShellArray();

        bool hasEdges() const;

        bool hasEdgeVertexShell() const;

        void clearEdges();
        
        void clearEdgeVertexShell();

    protected:
        /*** The array that stores the set of edges in the edge set */
        sofa::helper::vector<Edge> m_edge;

        sofa::helper::vector< sofa::helper::vector< unsigned int > > m_edgeVertexShell;

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

} // namespace topology

} // namespace component

} // namespace sofa

#endif