-#ifndef __CPEXTENSIONS__DATASTRUCTURES__GRAPH__H__
-#define __CPEXTENSIONS__DATASTRUCTURES__GRAPH__H__
+#ifndef __cpExtensions__DataStructures__Graph__h__
+#define __cpExtensions__DataStructures__Graph__h__
+#include <cpExtensions/Config.h>
#include <map>
#include <set>
#include <vector>
{
/** \brief A generic graph with templated index types.
*
- * @param V Vertex type.
- * @param C Cost type.
- * @param I Index type (it should be a strict weak ordering type).
+ * @param _TVertex Vertex type.
+ * @param _TCost Cost type.
+ * @param _TIndex Index type (it should be a strict weak ordering type).
*/
- template< class V, class C, class I = unsigned long >
+ template< class _TVertex, class _TCost, class _TIndex = unsigned long, class _TIndexCompare = std::less< _TIndex > >
class Graph
: public itk::LightObject
{
typedef itk::SmartPointer< Self > Pointer;
typedef itk::SmartPointer< const Self > ConstPointer;
- typedef V TVertex;
- typedef C TCost;
- typedef I TIndex;
+ typedef _TVertex TVertex;
+ typedef _TCost TCost;
+ typedef _TIndex TIndex;
+ typedef _TIndexCompare TIndexCompare;
// Base types
- typedef std::map< TIndex, TVertex > TVertices;
- typedef std::vector< TCost > TEdges;
- typedef std::map< TIndex, TEdges > TMatrixRow;
- typedef std::map< TIndex, TMatrixRow > TMatrix;
+ typedef std::map< TIndex, TVertex, TIndexCompare > TVertices;
+ typedef std::vector< TCost > TEdges;
+ typedef std::map< TIndex, TEdges, TIndexCompare > TMatrixRow;
+ typedef std::map< TIndex, TMatrixRow, TIndexCompare > TMatrix;
public:
itkNewMacro( Self );
/*! \brief Vertex manipulation methods.
* Names are self-explanatory.
*/
- inline bool HasVertexIndex( const I& i ) const
+ inline bool HasVertexIndex( const TIndex& i ) const
{ return( this->m_Vertices.find( i ) != this->m_Vertices.end( ) ); }
- inline void SetVertex( const I& index, V& vertex )
+ inline void SetVertex( const TIndex& index, TVertex& vertex )
{ this->m_Vertices[ index ] = vertex; }
- inline V& GetVertex( const I& index )
+ inline TVertex& GetVertex( const TIndex& index )
{ return( this->m_Vertices[ index ] ); }
- inline const V& GetVertex( const I& index ) const
+ inline const TVertex& GetVertex( const TIndex& index ) const
{ return( this->m_Vertices[ index ] ); }
- bool RenameVertex( const I& old_index, const I& new_index );
- void RemoveVertex( const I& index );
+ bool RenameVertex( const TIndex& old_index, const TIndex& new_index );
+ void RemoveVertex( const TIndex& index );
/*! \brief Edge manipulation methods.
* Names are self-explanatory.
*/
- inline void AddEdge( const I& orig, const I& dest, const C& cost )
+ inline void AddEdge( const TIndex& orig, const TIndex& dest, const TCost& cost )
{ this->m_Matrix[ orig ][ dest ].push_back( cost ); }
- inline TEdges& GetEdges( const I& orig, const I& dest )
- { return( this->m_Matrix[ orig ][ dest ] ); }
- inline const TEdges& GetEdges( const I& orig, const I& dest ) const
- { return( this->m_Matrix[ orig ][ dest ] ); }
- bool HasEdge( const I& orig, const I& dest ) const;
- void RemoveEdge( const I& orig, const I& dest, const C& cost );
- void RemoveEdges( const I& orig, const I& dest );
+ TEdges& GetEdges( const TIndex& orig, const TIndex& dest );
+ const TEdges& GetEdges( const TIndex& orig, const TIndex& dest ) const;
+ bool HasEdge( const TIndex& orig, const TIndex& dest ) const;
+ void RemoveEdge( const TIndex& orig, const TIndex& dest, const TCost& cost );
+ void RemoveEdges( const TIndex& orig, const TIndex& dest );
/*! \brief Returns graph's sinks.
*
*
* @return Sinks ordered by their index.
*/
- std::set< I > GetSinks( ) const;
+ std::set< TIndex, TIndexCompare > GetSinks( ) const;
protected:
Graph( );
} // ecapseman
#ifndef ITK_MANUAL_INSTANTIATION
-#include <cpExtensions/DataStructures/Graph.hxx>
+# include <cpExtensions/DataStructures/Graph.hxx>
#endif // ITK_MANUAL_INSTANTIATION
-#endif // __CPEXTENSIONS__DATASTRUCTURES__GRAPH__H__
+#endif // __cpExtensions__DataStructures__Graph__h__
// eof - $RCSfile$