]>
Commit | Line | Data |
---|---|---|
36fed23c | 1 | // -*- C++ -*- |
2 | ||
f1717362 | 3 | // Copyright (C) 2005-2016 Free Software Foundation, Inc. |
36fed23c | 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 | |
6bc9506f | 8 | // Foundation; either version 3, or (at your option) any later |
36fed23c | 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 | ||
6bc9506f | 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. | |
36fed23c | 19 | |
6bc9506f | 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/>. | |
36fed23c | 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 | /** | |
4f4a327e | 37 | * @file rb_tree_map_/split_join_fn_imps.hpp |
36fed23c | 38 | * Contains an implementation for rb_tree_. |
39 | */ | |
40 | ||
41 | PB_DS_CLASS_T_DEC | |
42 | inline void | |
43 | PB_DS_CLASS_C_DEC:: | |
44 | join(PB_DS_CLASS_C_DEC& other) | |
45 | { | |
2548203e | 46 | PB_DS_ASSERT_VALID((*this)) |
47 | PB_DS_ASSERT_VALID(other) | |
ea18d4a3 | 48 | if (base_type::join_prep(other) == false) |
2b31bec4 | 49 | { |
2548203e | 50 | PB_DS_ASSERT_VALID((*this)) |
51 | PB_DS_ASSERT_VALID(other) | |
2b31bec4 | 52 | return; |
53 | } | |
36fed23c | 54 | |
55 | const node_pointer p_x = other.split_min(); | |
36fed23c | 56 | join_imp(p_x, other.m_p_head->m_p_parent); |
ea18d4a3 | 57 | base_type::join_finish(other); |
2548203e | 58 | PB_DS_ASSERT_VALID((*this)) |
59 | PB_DS_ASSERT_VALID(other) | |
2b31bec4 | 60 | } |
36fed23c | 61 | |
62 | PB_DS_CLASS_T_DEC | |
63 | void | |
64 | PB_DS_CLASS_C_DEC:: | |
65 | join_imp(node_pointer p_x, node_pointer p_r) | |
66 | { | |
e4bb1925 | 67 | _GLIBCXX_DEBUG_ASSERT(p_x != 0); |
68 | if (p_r != 0) | |
36fed23c | 69 | p_r->m_red = false; |
70 | ||
ea18d4a3 | 71 | const size_type h = black_height(base_type::m_p_head->m_p_parent); |
36fed23c | 72 | const size_type other_h = black_height(p_r); |
36fed23c | 73 | node_pointer p_x_l; |
74 | node_pointer p_x_r; | |
36fed23c | 75 | std::pair<node_pointer, node_pointer> join_pos; |
36fed23c | 76 | const bool right_join = h >= other_h; |
36fed23c | 77 | if (right_join) |
78 | { | |
ea18d4a3 | 79 | join_pos = find_join_pos_right(base_type::m_p_head->m_p_parent, |
2b31bec4 | 80 | h, other_h); |
36fed23c | 81 | p_x_l = join_pos.first; |
82 | p_x_r = p_r; | |
83 | } | |
84 | else | |
85 | { | |
ea18d4a3 | 86 | p_x_l = base_type::m_p_head->m_p_parent; |
87 | base_type::m_p_head->m_p_parent = p_r; | |
e4bb1925 | 88 | if (p_r != 0) |
ea18d4a3 | 89 | p_r->m_p_parent = base_type::m_p_head; |
36fed23c | 90 | |
ea18d4a3 | 91 | join_pos = find_join_pos_left(base_type::m_p_head->m_p_parent, |
2b31bec4 | 92 | h, other_h); |
36fed23c | 93 | p_x_r = join_pos.first; |
94 | } | |
95 | ||
96 | node_pointer p_parent = join_pos.second; | |
ea18d4a3 | 97 | if (p_parent == base_type::m_p_head) |
36fed23c | 98 | { |
ea18d4a3 | 99 | base_type::m_p_head->m_p_parent = p_x; |
100 | p_x->m_p_parent = base_type::m_p_head; | |
36fed23c | 101 | } |
102 | else | |
103 | { | |
104 | p_x->m_p_parent = p_parent; | |
36fed23c | 105 | if (right_join) |
106 | p_x->m_p_parent->m_p_right = p_x; | |
107 | else | |
108 | p_x->m_p_parent->m_p_left = p_x; | |
109 | } | |
110 | ||
111 | p_x->m_p_left = p_x_l; | |
e4bb1925 | 112 | if (p_x_l != 0) |
36fed23c | 113 | p_x_l->m_p_parent = p_x; |
114 | ||
115 | p_x->m_p_right = p_x_r; | |
e4bb1925 | 116 | if (p_x_r != 0) |
36fed23c | 117 | p_x_r->m_p_parent = p_x; |
118 | ||
119 | p_x->m_red = true; | |
120 | ||
ea18d4a3 | 121 | base_type::initialize_min_max(); |
2548203e | 122 | PB_DS_STRUCT_ONLY_ASSERT_VALID((*this)) |
ea18d4a3 | 123 | base_type::update_to_top(p_x, (node_update* )this); |
36fed23c | 124 | insert_fixup(p_x); |
2548203e | 125 | PB_DS_STRUCT_ONLY_ASSERT_VALID((*this)) |
36fed23c | 126 | } |
127 | ||
128 | PB_DS_CLASS_T_DEC | |
129 | inline typename PB_DS_CLASS_C_DEC::node_pointer | |
130 | PB_DS_CLASS_C_DEC:: | |
131 | split_min() | |
132 | { | |
ea18d4a3 | 133 | node_pointer p_min = base_type::m_p_head->m_p_left; |
36fed23c | 134 | |
2b31bec4 | 135 | #ifdef _GLIBCXX_DEBUG |
ea18d4a3 | 136 | const node_pointer p_head = base_type::m_p_head; |
2b31bec4 | 137 | _GLIBCXX_DEBUG_ASSERT(p_min != p_head); |
138 | #endif | |
36fed23c | 139 | |
140 | remove_node(p_min); | |
ea18d4a3 | 141 | return p_min; |
36fed23c | 142 | } |
143 | ||
144 | PB_DS_CLASS_T_DEC | |
145 | std::pair< | |
146 | typename PB_DS_CLASS_C_DEC::node_pointer, | |
147 | typename PB_DS_CLASS_C_DEC::node_pointer> | |
148 | PB_DS_CLASS_C_DEC:: | |
149 | find_join_pos_right(node_pointer p_l, size_type h_l, size_type h_r) | |
150 | { | |
2b31bec4 | 151 | _GLIBCXX_DEBUG_ASSERT(h_l >= h_r); |
36fed23c | 152 | |
e4bb1925 | 153 | if (base_type::m_p_head->m_p_parent == 0) |
154 | return (std::make_pair((node_pointer)0, base_type::m_p_head)); | |
36fed23c | 155 | |
ea18d4a3 | 156 | node_pointer p_l_parent = base_type::m_p_head; |
36fed23c | 157 | while (h_l > h_r) |
158 | { | |
159 | if (p_l->m_red == false) | |
160 | { | |
2b31bec4 | 161 | _GLIBCXX_DEBUG_ASSERT(h_l > 0); |
36fed23c | 162 | --h_l; |
163 | } | |
164 | ||
165 | p_l_parent = p_l; | |
36fed23c | 166 | p_l = p_l->m_p_right; |
167 | } | |
168 | ||
169 | if (!is_effectively_black(p_l)) | |
170 | { | |
171 | p_l_parent = p_l; | |
36fed23c | 172 | p_l = p_l->m_p_right; |
173 | } | |
174 | ||
2b31bec4 | 175 | _GLIBCXX_DEBUG_ASSERT(is_effectively_black(p_l)); |
176 | _GLIBCXX_DEBUG_ASSERT(black_height(p_l) == h_r); | |
e4bb1925 | 177 | _GLIBCXX_DEBUG_ASSERT(p_l == 0 || p_l->m_p_parent == p_l_parent); |
2b31bec4 | 178 | return std::make_pair(p_l, p_l_parent); |
36fed23c | 179 | } |
180 | ||
181 | PB_DS_CLASS_T_DEC | |
182 | std::pair< | |
183 | typename PB_DS_CLASS_C_DEC::node_pointer, | |
184 | typename PB_DS_CLASS_C_DEC::node_pointer> | |
185 | PB_DS_CLASS_C_DEC:: | |
186 | find_join_pos_left(node_pointer p_r, size_type h_l, size_type h_r) | |
187 | { | |
2b31bec4 | 188 | _GLIBCXX_DEBUG_ASSERT(h_r > h_l); |
e4bb1925 | 189 | if (base_type::m_p_head->m_p_parent == 0) |
190 | return (std::make_pair((node_pointer)0, | |
ea18d4a3 | 191 | base_type::m_p_head)); |
192 | node_pointer p_r_parent = base_type::m_p_head; | |
36fed23c | 193 | while (h_r > h_l) |
194 | { | |
195 | if (p_r->m_red == false) | |
196 | { | |
2b31bec4 | 197 | _GLIBCXX_DEBUG_ASSERT(h_r > 0); |
36fed23c | 198 | --h_r; |
199 | } | |
200 | ||
201 | p_r_parent = p_r; | |
36fed23c | 202 | p_r = p_r->m_p_left; |
203 | } | |
204 | ||
205 | if (!is_effectively_black(p_r)) | |
206 | { | |
207 | p_r_parent = p_r; | |
36fed23c | 208 | p_r = p_r->m_p_left; |
209 | } | |
210 | ||
2b31bec4 | 211 | _GLIBCXX_DEBUG_ASSERT(is_effectively_black(p_r)); |
212 | _GLIBCXX_DEBUG_ASSERT(black_height(p_r) == h_l); | |
e4bb1925 | 213 | _GLIBCXX_DEBUG_ASSERT(p_r == 0 || p_r->m_p_parent == p_r_parent); |
2b31bec4 | 214 | return std::make_pair(p_r, p_r_parent); |
36fed23c | 215 | } |
216 | ||
217 | PB_DS_CLASS_T_DEC | |
218 | inline typename PB_DS_CLASS_C_DEC::size_type | |
219 | PB_DS_CLASS_C_DEC:: | |
220 | black_height(node_pointer p_nd) | |
221 | { | |
222 | size_type h = 1; | |
e4bb1925 | 223 | while (p_nd != 0) |
36fed23c | 224 | { |
225 | if (p_nd->m_red == false) | |
226 | ++h; | |
36fed23c | 227 | p_nd = p_nd->m_p_left; |
228 | } | |
2b31bec4 | 229 | return h; |
36fed23c | 230 | } |
231 | ||
232 | PB_DS_CLASS_T_DEC | |
233 | void | |
234 | PB_DS_CLASS_C_DEC:: | |
4f4a327e | 235 | split(key_const_reference r_key, PB_DS_CLASS_C_DEC& other) |
36fed23c | 236 | { |
2548203e | 237 | PB_DS_ASSERT_VALID((*this)) |
238 | PB_DS_ASSERT_VALID(other) | |
36fed23c | 239 | |
2548203e | 240 | if (base_type::split_prep(r_key, other) == false) |
241 | { | |
242 | PB_DS_ASSERT_VALID((*this)) | |
243 | PB_DS_ASSERT_VALID(other) | |
244 | return; | |
245 | } | |
36fed23c | 246 | |
2548203e | 247 | PB_DS_STRUCT_ONLY_ASSERT_VALID((*this)) |
248 | PB_DS_STRUCT_ONLY_ASSERT_VALID(other) | |
8411500a | 249 | node_pointer p_nd = this->upper_bound(r_key).m_p_nd; |
36fed23c | 250 | do |
251 | { | |
252 | node_pointer p_next_nd = p_nd->m_p_parent; | |
2b31bec4 | 253 | if (Cmp_Fn::operator()(r_key, PB_DS_V2F(p_nd->m_value))) |
36fed23c | 254 | split_at_node(p_nd, other); |
255 | ||
2548203e | 256 | PB_DS_STRUCT_ONLY_ASSERT_VALID((*this)) |
257 | PB_DS_STRUCT_ONLY_ASSERT_VALID(other) | |
2b31bec4 | 258 | p_nd = p_next_nd; |
36fed23c | 259 | } |
ea18d4a3 | 260 | while (p_nd != base_type::m_p_head); |
36fed23c | 261 | |
ea18d4a3 | 262 | base_type::split_finish(other); |
2548203e | 263 | PB_DS_ASSERT_VALID((*this)) |
2b31bec4 | 264 | } |
36fed23c | 265 | |
266 | PB_DS_CLASS_T_DEC | |
267 | void | |
268 | PB_DS_CLASS_C_DEC:: | |
269 | split_at_node(node_pointer p_nd, PB_DS_CLASS_C_DEC& other) | |
270 | { | |
e4bb1925 | 271 | _GLIBCXX_DEBUG_ASSERT(p_nd != 0); |
36fed23c | 272 | |
273 | node_pointer p_l = p_nd->m_p_left; | |
274 | node_pointer p_r = p_nd->m_p_right; | |
36fed23c | 275 | node_pointer p_parent = p_nd->m_p_parent; |
ea18d4a3 | 276 | if (p_parent == base_type::m_p_head) |
36fed23c | 277 | { |
ea18d4a3 | 278 | base_type::m_p_head->m_p_parent = p_l; |
e4bb1925 | 279 | if (p_l != 0) |
36fed23c | 280 | { |
ea18d4a3 | 281 | p_l->m_p_parent = base_type::m_p_head; |
36fed23c | 282 | p_l->m_red = false; |
283 | } | |
284 | } | |
285 | else | |
286 | { | |
287 | if (p_parent->m_p_left == p_nd) | |
288 | p_parent->m_p_left = p_l; | |
289 | else | |
290 | p_parent->m_p_right = p_l; | |
291 | ||
e4bb1925 | 292 | if (p_l != 0) |
36fed23c | 293 | p_l->m_p_parent = p_parent; |
294 | ||
8411500a | 295 | this->update_to_top(p_parent, (node_update* )this); |
36fed23c | 296 | |
297 | if (!p_nd->m_red) | |
298 | remove_fixup(p_l, p_parent); | |
299 | } | |
300 | ||
ea18d4a3 | 301 | base_type::initialize_min_max(); |
36fed23c | 302 | other.join_imp(p_nd, p_r); |
2548203e | 303 | PB_DS_STRUCT_ONLY_ASSERT_VALID((*this)) |
304 | PB_DS_STRUCT_ONLY_ASSERT_VALID(other) | |
36fed23c | 305 | } |
306 |