1// -*- C++ -*-
2
3// Copyright (C) 2005-2013 Free Software Foundation, Inc.
4//
5// This file is part of the GNU ISO C++ Library.  This library is free
6// software; you can redistribute it and/or modify it under the terms
7// of the GNU General Public License as published by the Free Software
8// Foundation; either version 3, or (at your option) any later
9// version.
10
11// This library is distributed in the hope that it will be useful, but
12// WITHOUT ANY WARRANTY; without even the implied warranty of
13// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14// General Public License for more details.
15
16// Under Section 7 of GPL version 3, you are granted additional
17// permissions described in the GCC Runtime Library Exception, version
18// 3.1, as published by the Free Software Foundation.
19
20// You should have received a copy of the GNU General Public License and
21// a copy of the GCC Runtime Library Exception along with this program;
22// see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
23// <http://www.gnu.org/licenses/>.
24
25// Copyright (C) 2004 Ami Tavory and Vladimir Dreizin, IBM-HRL.
26
27// Permission to use, copy, modify, sell, and distribute this software
28// is hereby granted without fee, provided that the above copyright
29// notice appears in all copies, and that both that copyright notice
30// and this permission notice appear in supporting documentation. None
31// of the above authors, nor IBM Haifa Research Laboratories, make any
32// representation about the suitability of this software for any
33// purpose. It is provided "as is" without express or implied
34// warranty.
35
36/**
37 * @file bin_search_tree_/node_iterators.hpp
38 * Contains an implementation class for bin_search_tree_.
39 */
40
41#ifndef PB_DS_BIN_SEARCH_TREE_NODE_ITERATORS_HPP
42#define PB_DS_BIN_SEARCH_TREE_NODE_ITERATORS_HPP
43
44#include <ext/pb_ds/tag_and_trait.hpp>
45
46namespace __gnu_pbds
47{
48  namespace detail
49  {
50#define PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC \
51    bin_search_tree_const_node_it_<Node, Const_Iterator, Iterator, _Alloc>
52
53    /// Const node iterator.
54    template<typename Node,
55	     class Const_Iterator,
56	     class Iterator,
57	     typename _Alloc>
58    class bin_search_tree_const_node_it_
59    {
60    private:
61      typedef
62      typename _Alloc::template rebind<
63      Node>::other::pointer
64      node_pointer;
65
66    public:
67      /// Category.
68      typedef trivial_iterator_tag iterator_category;
69
70      /// Difference type.
71      typedef trivial_iterator_difference_type difference_type;
72
73      /// Iterator's value type.
74      typedef Const_Iterator value_type;
75
76      /// Iterator's reference type.
77      typedef Const_Iterator reference;
78
79      /// Iterator's __const reference type.
80      typedef Const_Iterator const_reference;
81
82      /// Metadata type.
83      typedef typename Node::metadata_type metadata_type;
84
85      /// Const metadata reference type.
86      typedef
87      typename _Alloc::template rebind<metadata_type>::other::const_reference
88      metadata_const_reference;
89
90
91      bin_search_tree_const_node_it_(const node_pointer p_nd = 0)
92      : m_p_nd(const_cast<node_pointer>(p_nd))
93      { }
94
95      /// Access.
96      const_reference
97      operator*() const
98      { return Const_Iterator(m_p_nd); }
99
100      /// Metadata access.
101      metadata_const_reference
102      get_metadata() const
103      { return m_p_nd->get_metadata(); }
104
105      /// Returns the __const node iterator associated with the left node.
106      PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC
107      get_l_child() const
108      { return PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC(m_p_nd->m_p_left); }
109
110      /// Returns the __const node iterator associated with the right node.
111      PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC
112      get_r_child() const
113      { return PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC(m_p_nd->m_p_right); }
114
115      /// Compares to a different iterator object.
116      bool
117      operator==(const PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC& other) const
118      { return m_p_nd == other.m_p_nd; }
119
120      /// Compares (negatively) to a different iterator object.
121      bool
122      operator!=(const PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC& other) const
123      { return m_p_nd != other.m_p_nd; }
124
125      node_pointer m_p_nd;
126    };
127
128#define PB_DS_TREE_NODE_ITERATOR_CLASS_C_DEC \
129    bin_search_tree_node_it_<Node, Const_Iterator, Iterator, _Alloc>
130
131    /// Node iterator.
132    template<typename Node,
133	     class Const_Iterator,
134	     class Iterator,
135	     typename _Alloc>
136    class bin_search_tree_node_it_
137    : public PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC
138    {
139    private:
140      typedef
141      typename _Alloc::template rebind<
142      Node>::other::pointer
143      node_pointer;
144
145    public:
146      /// Iterator's value type.
147      typedef Iterator value_type;
148
149      /// Iterator's reference type.
150      typedef Iterator reference;
151
152      /// Iterator's __const reference type.
153      typedef Iterator const_reference;
154
155      inline
156      bin_search_tree_node_it_(const node_pointer p_nd = 0)
157      : PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC(const_cast<node_pointer>(p_nd))
158      { }
159
160      /// Access.
161      Iterator
162      operator*() const
163      { return Iterator(PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC::m_p_nd); }
164
165      /// Returns the node iterator associated with the left node.
166      PB_DS_TREE_NODE_ITERATOR_CLASS_C_DEC
167      get_l_child() const
168      {
169	return PB_DS_TREE_NODE_ITERATOR_CLASS_C_DEC(
170						    PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC::m_p_nd->m_p_left);
171      }
172
173      /// Returns the node iterator associated with the right node.
174      PB_DS_TREE_NODE_ITERATOR_CLASS_C_DEC
175      get_r_child() const
176      {
177	return PB_DS_TREE_NODE_ITERATOR_CLASS_C_DEC(
178						    PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC::m_p_nd->m_p_right);
179      }
180
181    };
182
183#undef PB_DS_TREE_CONST_NODE_ITERATOR_CLASS_C_DEC
184#undef PB_DS_TREE_NODE_ITERATOR_CLASS_C_DEC
185
186  } // namespace detail
187} // namespace __gnu_pbds
188
189#endif // #ifndef PB_DS_BIN_SEARCH_TREE_NODE_ITERATORS_HPP
190