node_iterators.hpp [plain text]
#ifndef PB_DS_PAT_TRIE_NODE_ITERATORS_HPP
#define PB_DS_PAT_TRIE_NODE_ITERATORS_HPP
#include <debug/debug.h>
namespace pb_ds
{
namespace detail
{
#define PB_DS_PAT_TRIE_CONST_NODE_ITERATOR_C_DEC \
pat_trie_const_node_it_< \
Node, \
Leaf, \
Head, \
Internal_Node, \
Const_Iterator, \
Iterator, \
E_Access_Traits, \
Allocator>
#define PB_DS_PAT_TRIE_NODE_ITERATOR_C_DEC \
pat_trie_node_it_< \
Node, \
Leaf, \
Head, \
Internal_Node, \
Const_Iterator, \
Iterator, \
E_Access_Traits, \
Allocator>
template<typename Node,
class Leaf,
class Head,
class Internal_Node,
class Const_Iterator,
class Iterator,
class E_Access_Traits,
class Allocator>
class pat_trie_const_node_it_
{
protected:
typedef
typename Allocator::template rebind<
Node>::other::pointer
node_pointer;
typedef
typename Allocator::template rebind<
Leaf>::other::const_pointer
const_leaf_pointer;
typedef
typename Allocator::template rebind<
Leaf>::other::pointer
leaf_pointer;
typedef
typename Allocator::template rebind<
Internal_Node>::other::pointer
internal_node_pointer;
typedef
typename Allocator::template rebind<
Internal_Node>::other::const_pointer
const_internal_node_pointer;
typedef
typename Allocator::template rebind<
E_Access_Traits>::other::const_pointer
const_e_access_traits_pointer;
private:
inline typename E_Access_Traits::const_iterator
pref_begin() const
{
if (m_p_nd->m_type == pat_trie_leaf_node_type)
return (m_p_traits->begin(
m_p_traits->extract_key(
static_cast<const_leaf_pointer>(m_p_nd)->value())));
_GLIBCXX_DEBUG_ASSERT(m_p_nd->m_type == pat_trie_internal_node_type);
return (static_cast<const_internal_node_pointer>(m_p_nd)->pref_b_it());
}
inline typename E_Access_Traits::const_iterator
pref_end() const
{
if (m_p_nd->m_type == pat_trie_leaf_node_type)
return (m_p_traits->end(
m_p_traits->extract_key(
static_cast<const_leaf_pointer>(m_p_nd)->value())));
_GLIBCXX_DEBUG_ASSERT(m_p_nd->m_type == pat_trie_internal_node_type);
return (static_cast<const_internal_node_pointer>(m_p_nd)->pref_e_it());
}
public:
typedef typename Allocator::size_type size_type;
typedef trivial_iterator_tag iterator_category;
typedef trivial_iterator_difference_type difference_type;
typedef Const_Iterator value_type;
typedef value_type reference;
typedef value_type const_reference;
typedef E_Access_Traits e_access_traits;
typedef typename e_access_traits::const_iterator const_e_iterator;
typedef typename Node::metadata_type metadata_type;
typedef
typename Allocator::template rebind<
metadata_type>::other::const_reference
const_metadata_reference;
inline
pat_trie_const_node_it_(node_pointer p_nd = NULL,
const_e_access_traits_pointer p_traits = NULL)
: m_p_nd(const_cast<node_pointer>(p_nd)), m_p_traits(p_traits)
{ }
inline std::pair<const_e_iterator, const_e_iterator>
valid_prefix() const
{ return std::make_pair(pref_begin(), pref_end()); }
inline const_reference
operator*() const
{
_GLIBCXX_DEBUG_ASSERT(num_children() == 0);
return Const_Iterator(m_p_nd);
}
inline const_metadata_reference
get_metadata() const
{ return m_p_nd->get_metadata(); }
inline size_type
num_children() const
{
if (m_p_nd->m_type == pat_trie_leaf_node_type)
return 0;
_GLIBCXX_DEBUG_ASSERT(m_p_nd->m_type == pat_trie_internal_node_type);
return std::distance(static_cast<internal_node_pointer>(m_p_nd)->begin(), static_cast<internal_node_pointer>(m_p_nd)->end());
}
PB_DS_PAT_TRIE_CONST_NODE_ITERATOR_C_DEC
get_child(size_type i) const
{
_GLIBCXX_DEBUG_ASSERT(m_p_nd->m_type == pat_trie_internal_node_type);
typename Internal_Node::iterator it =
static_cast<internal_node_pointer>(m_p_nd)->begin();
std::advance(it, i);
return PB_DS_PAT_TRIE_CONST_NODE_ITERATOR_C_DEC(*it, m_p_traits);
}
inline bool
operator==(const PB_DS_PAT_TRIE_CONST_NODE_ITERATOR_C_DEC& other) const
{ return (m_p_nd == other.m_p_nd); }
inline bool
operator!=(const PB_DS_PAT_TRIE_CONST_NODE_ITERATOR_C_DEC& other) const
{ return m_p_nd != other.m_p_nd; }
private:
friend class PB_DS_CLASS_C_DEC;
public:
node_pointer m_p_nd;
const_e_access_traits_pointer m_p_traits;
};
template<typename Node,
class Leaf,
class Head,
class Internal_Node,
class Const_Iterator,
class Iterator,
class E_Access_Traits,
class Allocator>
class pat_trie_node_it_ :
public PB_DS_PAT_TRIE_CONST_NODE_ITERATOR_C_DEC
{
private:
typedef
typename Allocator::template rebind<
Node>::other::pointer
node_pointer;
typedef Iterator iterator;
typedef PB_DS_PAT_TRIE_CONST_NODE_ITERATOR_C_DEC base_type;
typedef
typename base_type::const_e_access_traits_pointer
const_e_access_traits_pointer;
typedef typename base_type::internal_node_pointer internal_node_pointer;
public:
typedef
typename PB_DS_PAT_TRIE_CONST_NODE_ITERATOR_C_DEC::size_type
size_type;
typedef Iterator value_type;
typedef value_type reference;
typedef value_type const_reference;
inline
pat_trie_node_it_(node_pointer p_nd = NULL, const_e_access_traits_pointer p_traits = NULL) : base_type(p_nd, p_traits)
{ }
inline reference
operator*() const
{
_GLIBCXX_DEBUG_ASSERT(base_type::num_children() == 0);
return Iterator(base_type::m_p_nd);
}
PB_DS_PAT_TRIE_NODE_ITERATOR_C_DEC
get_child(size_type i) const
{
_GLIBCXX_DEBUG_ASSERT(base_type::m_p_nd->m_type == pat_trie_internal_node_type);
typename Internal_Node::iterator it =
static_cast<internal_node_pointer>(base_type::m_p_nd)->begin();
std::advance(it, i);
return PB_DS_PAT_TRIE_NODE_ITERATOR_C_DEC(*it, base_type::m_p_traits);
}
private:
friend class PB_DS_CLASS_C_DEC;
};
#undef PB_DS_PAT_TRIE_CONST_NODE_ITERATOR_C_DEC
#undef PB_DS_PAT_TRIE_NODE_ITERATOR_C_DEC
} }
#endif