blob: a3e15062b3e7e148b25dfc6738d92cd22c27ce5f [file] [log] [blame]
// -*- C++ -*-
// Copyright (C) 2005, 2006, 2009, 2010 Free Software Foundation, Inc.
//
// This file is part of the GNU ISO C++ Library. This library is free
// software; you can redistribute it and/or modify it under the terms
// of the GNU General Public License as published by the Free Software
// Foundation; either version 3, 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
// General Public License for more details.
// Under Section 7 of GPL version 3, you are granted additional
// permissions described in the GCC Runtime Library Exception, version
// 3.1, as published by the Free Software Foundation.
// You should have received a copy of the GNU General Public License and
// a copy of the GCC Runtime Library Exception along with this program;
// see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
// <http://www.gnu.org/licenses/>.
// Copyright (C) 2004 Ami Tavory and Vladimir Dreizin, IBM-HRL.
// Permission to use, copy, modify, sell, and distribute this software
// is hereby granted without fee, provided that the above copyright
// notice appears in all copies, and that both that copyright notice
// and this permission notice appear in supporting documentation. None
// of the above authors, nor IBM Haifa Research Laboratories, make any
// representation about the suitability of this software for any
// purpose. It is provided "as is" without express or implied
// warranty.
/**
* @file point_iterators.hpp
* Contains an implementation class for bin_search_tree_.
*/
#ifndef PB_DS_PAT_TRIE_FIND_ITERATORS_HPP
#define PB_DS_PAT_TRIE_FIND_ITERATORS_HPP
#include <debug/debug.h>
namespace __gnu_pbds
{
namespace detail
{
#define PB_DS_CONST_IT_C_DEC \
pat_trie_const_it_< \
Type_Traits, \
Node, \
Leaf, \
Head, \
Internal_Node, \
Is_Forward_Iterator, \
Allocator>
#define PB_DS_CONST_ODIR_IT_C_DEC \
pat_trie_const_it_< \
Type_Traits, \
Node, \
Leaf, \
Head, \
Internal_Node, \
!Is_Forward_Iterator, \
Allocator>
#define PB_DS_IT_C_DEC \
pat_trie_it_< \
Type_Traits, \
Node, \
Leaf, \
Head, \
Internal_Node, \
Is_Forward_Iterator, \
Allocator>
#define PB_DS_ODIR_IT_C_DEC \
pat_trie_it_< \
Type_Traits, \
Node, \
Leaf, \
Head, \
Internal_Node, \
!Is_Forward_Iterator, \
Allocator>
// Const iterator.
template<typename Type_Traits,
class Node,
class Leaf,
class Head,
class Internal_Node,
bool Is_Forward_Iterator,
class Allocator>
class pat_trie_const_it_
{
private:
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<
Head>::other::pointer
head_pointer;
typedef
typename Allocator::template rebind<
Internal_Node>::other::pointer
internal_node_pointer;
public:
typedef std::bidirectional_iterator_tag iterator_category;
typedef typename Allocator::difference_type difference_type;
typedef typename Type_Traits::value_type value_type;
typedef typename Type_Traits::pointer pointer;
typedef typename Type_Traits::const_pointer const_pointer;
typedef typename Type_Traits::reference reference;
typedef typename Type_Traits::const_reference const_reference;
public:
inline
pat_trie_const_it_(node_pointer p_nd = 0) : m_p_nd(p_nd)
{ }
inline
pat_trie_const_it_(const PB_DS_CONST_ODIR_IT_C_DEC& other)
: m_p_nd(other.m_p_nd)
{ }
inline
PB_DS_CONST_IT_C_DEC&
operator=(const PB_DS_CONST_IT_C_DEC& other)
{
m_p_nd = other.m_p_nd;
return *this;
}
inline
PB_DS_CONST_IT_C_DEC&
operator=(const PB_DS_CONST_ODIR_IT_C_DEC& other)
{
m_p_nd = other.m_p_nd;
return *this;
}
inline const_pointer
operator->() const
{
_GLIBCXX_DEBUG_ASSERT(m_p_nd->m_type == pat_trie_leaf_node_type);
return &static_cast<leaf_pointer>(m_p_nd)->value();
}
inline const_reference
operator*() const
{
_GLIBCXX_DEBUG_ASSERT(m_p_nd->m_type == pat_trie_leaf_node_type);
return static_cast<leaf_pointer>(m_p_nd)->value();
}
inline bool
operator==(const PB_DS_CONST_IT_C_DEC& other) const
{ return (m_p_nd == other.m_p_nd); }
inline bool
operator==(const PB_DS_CONST_ODIR_IT_C_DEC& other) const
{ return (m_p_nd == other.m_p_nd); }
inline bool
operator!=(const PB_DS_CONST_IT_C_DEC& other) const
{ return (m_p_nd != other.m_p_nd); }
inline bool
operator!=(const PB_DS_CONST_ODIR_IT_C_DEC& other) const
{ return (m_p_nd != other.m_p_nd); }
inline PB_DS_CONST_IT_C_DEC&
operator++()
{
inc(integral_constant<int,Is_Forward_Iterator>());
return *this;
}
inline PB_DS_CONST_IT_C_DEC
operator++(int)
{
PB_DS_CONST_IT_C_DEC ret_it(m_p_nd);
operator++();
return ret_it;
}
inline PB_DS_CONST_IT_C_DEC&
operator--()
{
dec(integral_constant<int,Is_Forward_Iterator>());
return *this;
}
inline PB_DS_CONST_IT_C_DEC
operator--(int)
{
PB_DS_CONST_IT_C_DEC ret_it(m_p_nd);
operator--();
return ret_it;
}
protected:
inline void
inc(false_type)
{ dec(true_type()); }
void
inc(true_type)
{
if (m_p_nd->m_type == pat_trie_head_node_type)
{
m_p_nd = static_cast<head_pointer>(m_p_nd)->m_p_min;
return;
}
node_pointer p_y = m_p_nd->m_p_parent;
while (p_y->m_type != pat_trie_head_node_type &&
get_larger_sibling(m_p_nd) == 0)
{
m_p_nd = p_y;
p_y = p_y->m_p_parent;
}
if (p_y->m_type == pat_trie_head_node_type)
{
m_p_nd = p_y;
return;
}
m_p_nd = leftmost_descendant(get_larger_sibling(m_p_nd));
}
inline void
dec(false_type)
{ inc(true_type()); }
void
dec(true_type)
{
if (m_p_nd->m_type == pat_trie_head_node_type)
{
m_p_nd = static_cast<head_pointer>(m_p_nd)->m_p_max;
return;
}
node_pointer p_y = m_p_nd->m_p_parent;
while (p_y->m_type != pat_trie_head_node_type &&
get_smaller_sibling(m_p_nd) == 0)
{
m_p_nd = p_y;
p_y = p_y->m_p_parent;
}
if (p_y->m_type == pat_trie_head_node_type)
{
m_p_nd = p_y;
return;
}
m_p_nd = rightmost_descendant(get_smaller_sibling(m_p_nd));
}
inline static node_pointer
get_larger_sibling(node_pointer p_nd)
{
internal_node_pointer p_parent =
static_cast<internal_node_pointer>(p_nd->m_p_parent);
typename Internal_Node::iterator it = p_parent->begin();
while (*it != p_nd)
++it;
typename Internal_Node::iterator next_it = it;
++next_it;
return ((next_it == p_parent->end())? 0 :* next_it);
}
inline static node_pointer
get_smaller_sibling(node_pointer p_nd)
{
internal_node_pointer p_parent =
static_cast<internal_node_pointer>(p_nd->m_p_parent);
typename Internal_Node::iterator it = p_parent->begin();
if (*it == p_nd)
return (0);
typename Internal_Node::iterator prev_it;
do
{
prev_it = it;
++it;
if (*it == p_nd)
return (*prev_it);
}
while (true);
_GLIBCXX_DEBUG_ASSERT(false);
return (0);
}
inline static leaf_pointer
leftmost_descendant(node_pointer p_nd)
{
if (p_nd->m_type == pat_trie_leaf_node_type)
return static_cast<leaf_pointer>(p_nd);
return static_cast<internal_node_pointer>(p_nd)->leftmost_descendant();
}
inline static leaf_pointer
rightmost_descendant(node_pointer p_nd)
{
if (p_nd->m_type == pat_trie_leaf_node_type)
return static_cast<leaf_pointer>(p_nd);
return static_cast<internal_node_pointer>(p_nd)->rightmost_descendant();
}
public:
node_pointer m_p_nd;
};
// Iterator.
template<typename Type_Traits,
class Node,
class Leaf,
class Head,
class Internal_Node,
bool Is_Forward_Iterator,
class Allocator>
class pat_trie_it_ :
public PB_DS_CONST_IT_C_DEC
{
private:
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<
Head>::other::pointer
head_pointer;
typedef
typename Allocator::template rebind<
Internal_Node>::other::pointer
internal_node_pointer;
public:
typedef typename Type_Traits::value_type value_type;
typedef typename Type_Traits::const_pointer const_pointer;
typedef typename Type_Traits::pointer pointer;
typedef typename Type_Traits::const_reference const_reference;
typedef typename Type_Traits::reference reference;
inline
pat_trie_it_(node_pointer p_nd = 0) : PB_DS_CONST_IT_C_DEC((node_pointer)p_nd)
{ }
inline
pat_trie_it_(const PB_DS_ODIR_IT_C_DEC& other) : PB_DS_CONST_IT_C_DEC(other.m_p_nd)
{ }
inline
PB_DS_IT_C_DEC&
operator=(const PB_DS_IT_C_DEC& other)
{
base_it_type::m_p_nd = other.m_p_nd;
return *this;
}
inline
PB_DS_IT_C_DEC&
operator=(const PB_DS_ODIR_IT_C_DEC& other)
{
base_it_type::m_p_nd = other.m_p_nd;
return *this;
}
inline pointer
operator->() const
{
_GLIBCXX_DEBUG_ASSERT(base_it_type::m_p_nd->m_type == pat_trie_leaf_node_type);
return &static_cast<leaf_pointer>(base_it_type::m_p_nd)->value();
}
inline reference
operator*() const
{
_GLIBCXX_DEBUG_ASSERT(base_it_type::m_p_nd->m_type == pat_trie_leaf_node_type);
return static_cast<leaf_pointer>(base_it_type::m_p_nd)->value();
}
inline PB_DS_IT_C_DEC&
operator++()
{
PB_DS_CONST_IT_C_DEC::
operator++();
return *this;
}
inline PB_DS_IT_C_DEC
operator++(int)
{
PB_DS_IT_C_DEC ret_it(base_it_type::m_p_nd);
operator++();
return ret_it;
}
inline PB_DS_IT_C_DEC&
operator--()
{
PB_DS_CONST_IT_C_DEC::operator--();
return *this;
}
inline PB_DS_IT_C_DEC
operator--(int)
{
PB_DS_IT_C_DEC ret_it(base_it_type::m_p_nd);
operator--();
return ret_it;
}
protected:
typedef PB_DS_CONST_IT_C_DEC base_it_type;
friend class PB_DS_CLASS_C_DEC;
};
#undef PB_DS_CONST_IT_C_DEC
#undef PB_DS_CONST_ODIR_IT_C_DEC
#undef PB_DS_IT_C_DEC
#undef PB_DS_ODIR_IT_C_DEC
} // namespace detail
} // namespace __gnu_pbds
#endif