123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730 |
- /*
- *
- *
- * Copyright (c) 1994
- * Hewlett-Packard Company
- *
- * Copyright (c) 1996,1997
- * Silicon Graphics Computer Systems, Inc.
- *
- * Copyright (c) 1997
- * Moscow Center for SPARC Technology
- *
- * Copyright (c) 1999
- * Boris Fomitchev
- *
- * This material is provided "as is", with absolutely no warranty expressed
- * or implied. Any use is at your own risk.
- *
- * Permission to use or copy this software for any purpose is hereby granted
- * without fee, provided the above notices are retained on all copies.
- * Permission to modify the code and to distribute modified code is granted,
- * provided the above notices are retained, and a notice that the code was
- * modified is included with the above copyright notice.
- *
- * Modified CRP 7/10/00 for improved conformance / efficiency on insert_unique /
- * insert_equal with valid hint -- efficiency is improved all around, and it is
- * should now be standard conforming for complexity on insert point immediately
- * after hint (amortized constant time).
- *
- */
- #ifndef _STLP_TREE_C
- #define _STLP_TREE_C
- #ifndef _STLP_INTERNAL_TREE_H
- # include <stl/_tree.h>
- #endif
- #if defined (_STLP_DEBUG)
- # define _Rb_tree _STLP_NON_DBG_NAME(Rb_tree)
- #endif
- // fbp: these defines are for outline methods definitions.
- // needed for definitions to be portable. Should not be used in method bodies.
- #if defined (_STLP_NESTED_TYPE_PARAM_BUG)
- # define __iterator__ _Rb_tree_iterator<_Value, _STLP_HEADER_TYPENAME _Traits::_NonConstTraits>
- # define __size_type__ size_t
- # define iterator __iterator__
- #else
- # define __iterator__ _STLP_TYPENAME_ON_RETURN_TYPE _Rb_tree<_Key, _Compare, _Value, _KeyOfValue, _Traits, _Alloc>::iterator
- # define __size_type__ _STLP_TYPENAME_ON_RETURN_TYPE _Rb_tree<_Key, _Compare, _Value, _KeyOfValue, _Traits, _Alloc>::size_type
- #endif
- _STLP_BEGIN_NAMESPACE
- _STLP_MOVE_TO_PRIV_NAMESPACE
- #if defined (_STLP_EXPOSE_GLOBALS_IMPLEMENTATION)
- template <class _Dummy> void _STLP_CALL
- _Rb_global<_Dummy>::_Rotate_left(_Rb_tree_node_base* __x,
- _Rb_tree_node_base*& __root) {
- _Rb_tree_node_base* __y = __x->_M_right;
- __x->_M_right = __y->_M_left;
- if (__y->_M_left != 0)
- __y->_M_left->_M_parent = __x;
- __y->_M_parent = __x->_M_parent;
- if (__x == __root)
- __root = __y;
- else if (__x == __x->_M_parent->_M_left)
- __x->_M_parent->_M_left = __y;
- else
- __x->_M_parent->_M_right = __y;
- __y->_M_left = __x;
- __x->_M_parent = __y;
- }
- template <class _Dummy> void _STLP_CALL
- _Rb_global<_Dummy>::_Rotate_right(_Rb_tree_node_base* __x,
- _Rb_tree_node_base*& __root) {
- _Rb_tree_node_base* __y = __x->_M_left;
- __x->_M_left = __y->_M_right;
- if (__y->_M_right != 0)
- __y->_M_right->_M_parent = __x;
- __y->_M_parent = __x->_M_parent;
- if (__x == __root)
- __root = __y;
- else if (__x == __x->_M_parent->_M_right)
- __x->_M_parent->_M_right = __y;
- else
- __x->_M_parent->_M_left = __y;
- __y->_M_right = __x;
- __x->_M_parent = __y;
- }
- template <class _Dummy> void _STLP_CALL
- _Rb_global<_Dummy>::_Rebalance(_Rb_tree_node_base* __x,
- _Rb_tree_node_base*& __root) {
- __x->_M_color = _S_rb_tree_red;
- while (__x != __root && __x->_M_parent->_M_color == _S_rb_tree_red) {
- if (__x->_M_parent == __x->_M_parent->_M_parent->_M_left) {
- _Rb_tree_node_base* __y = __x->_M_parent->_M_parent->_M_right;
- if (__y && __y->_M_color == _S_rb_tree_red) {
- __x->_M_parent->_M_color = _S_rb_tree_black;
- __y->_M_color = _S_rb_tree_black;
- __x->_M_parent->_M_parent->_M_color = _S_rb_tree_red;
- __x = __x->_M_parent->_M_parent;
- }
- else {
- if (__x == __x->_M_parent->_M_right) {
- __x = __x->_M_parent;
- _Rotate_left(__x, __root);
- }
- __x->_M_parent->_M_color = _S_rb_tree_black;
- __x->_M_parent->_M_parent->_M_color = _S_rb_tree_red;
- _Rotate_right(__x->_M_parent->_M_parent, __root);
- }
- }
- else {
- _Rb_tree_node_base* __y = __x->_M_parent->_M_parent->_M_left;
- if (__y && __y->_M_color == _S_rb_tree_red) {
- __x->_M_parent->_M_color = _S_rb_tree_black;
- __y->_M_color = _S_rb_tree_black;
- __x->_M_parent->_M_parent->_M_color = _S_rb_tree_red;
- __x = __x->_M_parent->_M_parent;
- }
- else {
- if (__x == __x->_M_parent->_M_left) {
- __x = __x->_M_parent;
- _Rotate_right(__x, __root);
- }
- __x->_M_parent->_M_color = _S_rb_tree_black;
- __x->_M_parent->_M_parent->_M_color = _S_rb_tree_red;
- _Rotate_left(__x->_M_parent->_M_parent, __root);
- }
- }
- }
- __root->_M_color = _S_rb_tree_black;
- }
- template <class _Dummy> _Rb_tree_node_base* _STLP_CALL
- _Rb_global<_Dummy>::_Rebalance_for_erase(_Rb_tree_node_base* __z,
- _Rb_tree_node_base*& __root,
- _Rb_tree_node_base*& __leftmost,
- _Rb_tree_node_base*& __rightmost) {
- _Rb_tree_node_base* __y = __z;
- _Rb_tree_node_base* __x;
- _Rb_tree_node_base* __x_parent;
- if (__y->_M_left == 0) // __z has at most one non-null child. y == z.
- __x = __y->_M_right; // __x might be null.
- else {
- if (__y->_M_right == 0) // __z has exactly one non-null child. y == z.
- __x = __y->_M_left; // __x is not null.
- else { // __z has two non-null children. Set __y to
- __y = _Rb_tree_node_base::_S_minimum(__y->_M_right); // __z's successor. __x might be null.
- __x = __y->_M_right;
- }
- }
- if (__y != __z) { // relink y in place of z. y is z's successor
- __z->_M_left->_M_parent = __y;
- __y->_M_left = __z->_M_left;
- if (__y != __z->_M_right) {
- __x_parent = __y->_M_parent;
- if (__x) __x->_M_parent = __y->_M_parent;
- __y->_M_parent->_M_left = __x; // __y must be a child of _M_left
- __y->_M_right = __z->_M_right;
- __z->_M_right->_M_parent = __y;
- }
- else
- __x_parent = __y;
- if (__root == __z)
- __root = __y;
- else if (__z->_M_parent->_M_left == __z)
- __z->_M_parent->_M_left = __y;
- else
- __z->_M_parent->_M_right = __y;
- __y->_M_parent = __z->_M_parent;
- _STLP_STD::swap(__y->_M_color, __z->_M_color);
- __y = __z;
- // __y now points to node to be actually deleted
- }
- else { // __y == __z
- __x_parent = __y->_M_parent;
- if (__x) __x->_M_parent = __y->_M_parent;
- if (__root == __z)
- __root = __x;
- else {
- if (__z->_M_parent->_M_left == __z)
- __z->_M_parent->_M_left = __x;
- else
- __z->_M_parent->_M_right = __x;
- }
- if (__leftmost == __z) {
- if (__z->_M_right == 0) // __z->_M_left must be null also
- __leftmost = __z->_M_parent;
- // makes __leftmost == _M_header if __z == __root
- else
- __leftmost = _Rb_tree_node_base::_S_minimum(__x);
- }
- if (__rightmost == __z) {
- if (__z->_M_left == 0) // __z->_M_right must be null also
- __rightmost = __z->_M_parent;
- // makes __rightmost == _M_header if __z == __root
- else // __x == __z->_M_left
- __rightmost = _Rb_tree_node_base::_S_maximum(__x);
- }
- }
- if (__y->_M_color != _S_rb_tree_red) {
- while (__x != __root && (__x == 0 || __x->_M_color == _S_rb_tree_black))
- if (__x == __x_parent->_M_left) {
- _Rb_tree_node_base* __w = __x_parent->_M_right;
- if (__w->_M_color == _S_rb_tree_red) {
- __w->_M_color = _S_rb_tree_black;
- __x_parent->_M_color = _S_rb_tree_red;
- _Rotate_left(__x_parent, __root);
- __w = __x_parent->_M_right;
- }
- if ((__w->_M_left == 0 ||
- __w->_M_left->_M_color == _S_rb_tree_black) && (__w->_M_right == 0 ||
- __w->_M_right->_M_color == _S_rb_tree_black)) {
- __w->_M_color = _S_rb_tree_red;
- __x = __x_parent;
- __x_parent = __x_parent->_M_parent;
- } else {
- if (__w->_M_right == 0 ||
- __w->_M_right->_M_color == _S_rb_tree_black) {
- if (__w->_M_left) __w->_M_left->_M_color = _S_rb_tree_black;
- __w->_M_color = _S_rb_tree_red;
- _Rotate_right(__w, __root);
- __w = __x_parent->_M_right;
- }
- __w->_M_color = __x_parent->_M_color;
- __x_parent->_M_color = _S_rb_tree_black;
- if (__w->_M_right) __w->_M_right->_M_color = _S_rb_tree_black;
- _Rotate_left(__x_parent, __root);
- break;
- }
- } else { // same as above, with _M_right <-> _M_left.
- _Rb_tree_node_base* __w = __x_parent->_M_left;
- if (__w->_M_color == _S_rb_tree_red) {
- __w->_M_color = _S_rb_tree_black;
- __x_parent->_M_color = _S_rb_tree_red;
- _Rotate_right(__x_parent, __root);
- __w = __x_parent->_M_left;
- }
- if ((__w->_M_right == 0 ||
- __w->_M_right->_M_color == _S_rb_tree_black) && (__w->_M_left == 0 ||
- __w->_M_left->_M_color == _S_rb_tree_black)) {
- __w->_M_color = _S_rb_tree_red;
- __x = __x_parent;
- __x_parent = __x_parent->_M_parent;
- } else {
- if (__w->_M_left == 0 ||
- __w->_M_left->_M_color == _S_rb_tree_black) {
- if (__w->_M_right) __w->_M_right->_M_color = _S_rb_tree_black;
- __w->_M_color = _S_rb_tree_red;
- _Rotate_left(__w, __root);
- __w = __x_parent->_M_left;
- }
- __w->_M_color = __x_parent->_M_color;
- __x_parent->_M_color = _S_rb_tree_black;
- if (__w->_M_left) __w->_M_left->_M_color = _S_rb_tree_black;
- _Rotate_right(__x_parent, __root);
- break;
- }
- }
- if (__x) __x->_M_color = _S_rb_tree_black;
- }
- return __y;
- }
- template <class _Dummy> _Rb_tree_node_base* _STLP_CALL
- _Rb_global<_Dummy>::_M_decrement(_Rb_tree_node_base* _M_node) {
- if (_M_node->_M_color == _S_rb_tree_red && _M_node->_M_parent->_M_parent == _M_node)
- _M_node = _M_node->_M_right;
- else if (_M_node->_M_left != 0) {
- _M_node = _Rb_tree_node_base::_S_maximum(_M_node->_M_left);
- }
- else {
- _Base_ptr __y = _M_node->_M_parent;
- while (_M_node == __y->_M_left) {
- _M_node = __y;
- __y = __y->_M_parent;
- }
- _M_node = __y;
- }
- return _M_node;
- }
- template <class _Dummy> _Rb_tree_node_base* _STLP_CALL
- _Rb_global<_Dummy>::_M_increment(_Rb_tree_node_base* _M_node) {
- if (_M_node->_M_right != 0) {
- _M_node = _Rb_tree_node_base::_S_minimum(_M_node->_M_right);
- }
- else {
- _Base_ptr __y = _M_node->_M_parent;
- while (_M_node == __y->_M_right) {
- _M_node = __y;
- __y = __y->_M_parent;
- }
- // check special case: This is necessary if _M_node is the
- // _M_head and the tree contains only a single node __y. In
- // that case parent, left and right all point to __y!
- if (_M_node->_M_right != __y)
- _M_node = __y;
- }
- return _M_node;
- }
- #endif /* _STLP_EXPOSE_GLOBALS_IMPLEMENTATION */
- template <class _Key, class _Compare,
- class _Value, class _KeyOfValue, class _Traits, class _Alloc>
- _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc>&
- _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc> ::operator=(
- const _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc>& __x) {
- if (this != &__x) {
- // Note that _Key may be a constant type.
- clear();
- _M_node_count = 0;
- _M_key_compare = __x._M_key_compare;
- if (__x._M_root() == 0) {
- _M_root() = 0;
- _M_leftmost() = &this->_M_header._M_data;
- _M_rightmost() = &this->_M_header._M_data;
- }
- else {
- _M_root() = _M_copy(__x._M_root(), &this->_M_header._M_data);
- _M_leftmost() = _S_minimum(_M_root());
- _M_rightmost() = _S_maximum(_M_root());
- _M_node_count = __x._M_node_count;
- }
- }
- return *this;
- }
- // CRP 7/10/00 inserted argument __on_right, which is another hint (meant to
- // act like __on_left and ignore a portion of the if conditions -- specify
- // __on_right != 0 to bypass comparison as false or __on_left != 0 to bypass
- // comparison as true)
- template <class _Key, class _Compare,
- class _Value, class _KeyOfValue, class _Traits, class _Alloc>
- __iterator__
- _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc> ::_M_insert(_Rb_tree_node_base * __parent,
- const _Value& __val,
- _Rb_tree_node_base * __on_left,
- _Rb_tree_node_base * __on_right) {
- // We do not create the node here as, depending on tests, we might call
- // _M_key_compare that can throw an exception.
- _Base_ptr __new_node;
- if ( __parent == &this->_M_header._M_data ) {
- __new_node = _M_create_node(__val);
- _S_left(__parent) = __new_node; // also makes _M_leftmost() = __new_node
- _M_root() = __new_node;
- _M_rightmost() = __new_node;
- }
- else if ( __on_right == 0 && // If __on_right != 0, the remainder fails to false
- ( __on_left != 0 || // If __on_left != 0, the remainder succeeds to true
- _M_key_compare( _KeyOfValue()(__val), _S_key(__parent) ) ) ) {
- __new_node = _M_create_node(__val);
- _S_left(__parent) = __new_node;
- if (__parent == _M_leftmost())
- _M_leftmost() = __new_node; // maintain _M_leftmost() pointing to min node
- }
- else {
- __new_node = _M_create_node(__val);
- _S_right(__parent) = __new_node;
- if (__parent == _M_rightmost())
- _M_rightmost() = __new_node; // maintain _M_rightmost() pointing to max node
- }
- _S_parent(__new_node) = __parent;
- _Rb_global_inst::_Rebalance(__new_node, this->_M_header._M_data._M_parent);
- ++_M_node_count;
- return iterator(__new_node);
- }
- template <class _Key, class _Compare,
- class _Value, class _KeyOfValue, class _Traits, class _Alloc>
- __iterator__
- _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc> ::insert_equal(const _Value& __val) {
- _Base_ptr __y = &this->_M_header._M_data;
- _Base_ptr __x = _M_root();
- while (__x != 0) {
- __y = __x;
- if (_M_key_compare(_KeyOfValue()(__val), _S_key(__x))) {
- __x = _S_left(__x);
- }
- else
- __x = _S_right(__x);
- }
- return _M_insert(__y, __val, __x);
- }
- template <class _Key, class _Compare,
- class _Value, class _KeyOfValue, class _Traits, class _Alloc>
- pair<__iterator__, bool>
- _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc> ::insert_unique(const _Value& __val) {
- _Base_ptr __y = &this->_M_header._M_data;
- _Base_ptr __x = _M_root();
- bool __comp = true;
- while (__x != 0) {
- __y = __x;
- __comp = _M_key_compare(_KeyOfValue()(__val), _S_key(__x));
- __x = __comp ? _S_left(__x) : _S_right(__x);
- }
- iterator __j = iterator(__y);
- if (__comp) {
- if (__j == begin())
- return pair<iterator,bool>(_M_insert(__y, __val, /* __x*/ __y), true);
- else
- --__j;
- }
- if (_M_key_compare(_S_key(__j._M_node), _KeyOfValue()(__val))) {
- return pair<iterator,bool>(_M_insert(__y, __val, __x), true);
- }
- return pair<iterator,bool>(__j, false);
- }
- // Modifications CRP 7/10/00 as noted to improve conformance and
- // efficiency.
- template <class _Key, class _Compare,
- class _Value, class _KeyOfValue, class _Traits, class _Alloc>
- __iterator__
- _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc> ::insert_unique(iterator __position,
- const _Value& __val) {
- if (__position._M_node == this->_M_header._M_data._M_left) { // begin()
- // if the container is empty, fall back on insert_unique.
- if (empty())
- return insert_unique(__val).first;
- if (_M_key_compare(_KeyOfValue()(__val), _S_key(__position._M_node))) {
- return _M_insert(__position._M_node, __val, __position._M_node);
- }
- // first argument just needs to be non-null
- else {
- bool __comp_pos_v = _M_key_compare( _S_key(__position._M_node), _KeyOfValue()(__val) );
- if (__comp_pos_v == false) // compare > and compare < both false so compare equal
- return __position;
- //Below __comp_pos_v == true
- // Standard-conformance - does the insertion point fall immediately AFTER
- // the hint?
- iterator __after = __position;
- ++__after;
- // Check for only one member -- in that case, __position points to itself,
- // and attempting to increment will cause an infinite loop.
- if (__after._M_node == &this->_M_header._M_data)
- // Check guarantees exactly one member, so comparison was already
- // performed and we know the result; skip repeating it in _M_insert
- // by specifying a non-zero fourth argument.
- return _M_insert(__position._M_node, __val, 0, __position._M_node);
- // All other cases:
- // Optimization to catch insert-equivalent -- save comparison results,
- // and we get this for free.
- if (_M_key_compare( _KeyOfValue()(__val), _S_key(__after._M_node) )) {
- if (_S_right(__position._M_node) == 0)
- return _M_insert(__position._M_node, __val, 0, __position._M_node);
- else
- return _M_insert(__after._M_node, __val, __after._M_node);
- }
- else {
- return insert_unique(__val).first;
- }
- }
- }
- else if (__position._M_node == &this->_M_header._M_data) { // end()
- if (_M_key_compare(_S_key(_M_rightmost()), _KeyOfValue()(__val))) {
- // pass along to _M_insert that it can skip comparing
- // v, Key ; since compare Key, v was true, compare v, Key must be false.
- return _M_insert(_M_rightmost(), __val, 0, __position._M_node); // Last argument only needs to be non-null
- }
- else
- return insert_unique(__val).first;
- }
- else {
- iterator __before = __position;
- --__before;
- bool __comp_v_pos = _M_key_compare(_KeyOfValue()(__val), _S_key(__position._M_node));
- if (__comp_v_pos
- && _M_key_compare( _S_key(__before._M_node), _KeyOfValue()(__val) )) {
- if (_S_right(__before._M_node) == 0)
- return _M_insert(__before._M_node, __val, 0, __before._M_node); // Last argument only needs to be non-null
- else
- return _M_insert(__position._M_node, __val, __position._M_node);
- // first argument just needs to be non-null
- }
- else {
- // Does the insertion point fall immediately AFTER the hint?
- iterator __after = __position;
- ++__after;
- // Optimization to catch equivalent cases and avoid unnecessary comparisons
- bool __comp_pos_v = !__comp_v_pos; // Stored this result earlier
- // If the earlier comparison was true, this comparison doesn't need to be
- // performed because it must be false. However, if the earlier comparison
- // was false, we need to perform this one because in the equal case, both will
- // be false.
- if (!__comp_v_pos) {
- __comp_pos_v = _M_key_compare(_S_key(__position._M_node), _KeyOfValue()(__val));
- }
- if ( (!__comp_v_pos) // comp_v_pos true implies comp_v_pos false
- && __comp_pos_v
- && (__after._M_node == &this->_M_header._M_data ||
- _M_key_compare( _KeyOfValue()(__val), _S_key(__after._M_node) ))) {
- if (_S_right(__position._M_node) == 0)
- return _M_insert(__position._M_node, __val, 0, __position._M_node);
- else
- return _M_insert(__after._M_node, __val, __after._M_node);
- } else {
- // Test for equivalent case
- if (__comp_v_pos == __comp_pos_v)
- return __position;
- else
- return insert_unique(__val).first;
- }
- }
- }
- }
- template <class _Key, class _Compare,
- class _Value, class _KeyOfValue, class _Traits, class _Alloc>
- __iterator__
- _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc> ::insert_equal(iterator __position,
- const _Value& __val) {
- if (__position._M_node == this->_M_header._M_data._M_left) { // begin()
- // Check for zero members
- if (size() <= 0)
- return insert_equal(__val);
- if (!_M_key_compare(_S_key(__position._M_node), _KeyOfValue()(__val)))
- return _M_insert(__position._M_node, __val, __position._M_node);
- else {
- // Check for only one member
- if (__position._M_node->_M_left == __position._M_node)
- // Unlike insert_unique, can't avoid doing a comparison here.
- return _M_insert(__position._M_node, __val);
- // All other cases:
- // Standard-conformance - does the insertion point fall immediately AFTER
- // the hint?
- iterator __after = __position;
- ++__after;
- // Already know that compare(pos, v) must be true!
- // Therefore, we want to know if compare(after, v) is false.
- // (i.e., we now pos < v, now we want to know if v <= after)
- // If not, invalid hint.
- if ( __after._M_node == &this->_M_header._M_data ||
- !_M_key_compare( _S_key(__after._M_node), _KeyOfValue()(__val) ) ) {
- if (_S_right(__position._M_node) == 0)
- return _M_insert(__position._M_node, __val, 0, __position._M_node);
- else
- return _M_insert(__after._M_node, __val, __after._M_node);
- }
- else { // Invalid hint
- return insert_equal(__val);
- }
- }
- }
- else if (__position._M_node == &this->_M_header._M_data) { // end()
- if (!_M_key_compare(_KeyOfValue()(__val), _S_key(_M_rightmost())))
- return _M_insert(_M_rightmost(), __val, 0, __position._M_node); // Last argument only needs to be non-null
- else {
- return insert_equal(__val);
- }
- }
- else {
- iterator __before = __position;
- --__before;
- // store the result of the comparison between pos and v so
- // that we don't have to do it again later. Note that this reverses the shortcut
- // on the if, possibly harming efficiency in comparisons; I think the harm will
- // be negligible, and to do what I want to do (save the result of a comparison so
- // that it can be re-used) there is no alternative. Test here is for before <= v <= pos.
- bool __comp_pos_v = _M_key_compare(_S_key(__position._M_node), _KeyOfValue()(__val));
- if (!__comp_pos_v &&
- !_M_key_compare(_KeyOfValue()(__val), _S_key(__before._M_node))) {
- if (_S_right(__before._M_node) == 0)
- return _M_insert(__before._M_node, __val, 0, __before._M_node); // Last argument only needs to be non-null
- else
- return _M_insert(__position._M_node, __val, __position._M_node);
- }
- else {
- // Does the insertion point fall immediately AFTER the hint?
- // Test for pos < v <= after
- iterator __after = __position;
- ++__after;
- if (__comp_pos_v &&
- ( __after._M_node == &this->_M_header._M_data ||
- !_M_key_compare( _S_key(__after._M_node), _KeyOfValue()(__val) ) ) ) {
- if (_S_right(__position._M_node) == 0)
- return _M_insert(__position._M_node, __val, 0, __position._M_node);
- else
- return _M_insert(__after._M_node, __val, __after._M_node);
- }
- else { // Invalid hint
- return insert_equal(__val);
- }
- }
- }
- }
- template <class _Key, class _Compare,
- class _Value, class _KeyOfValue, class _Traits, class _Alloc>
- _Rb_tree_node_base*
- _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc> ::_M_copy(_Rb_tree_node_base* __x,
- _Rb_tree_node_base* __p) {
- // structural copy. __x and __p must be non-null.
- _Base_ptr __top = _M_clone_node(__x);
- _S_parent(__top) = __p;
- _STLP_TRY {
- if (_S_right(__x))
- _S_right(__top) = _M_copy(_S_right(__x), __top);
- __p = __top;
- __x = _S_left(__x);
- while (__x != 0) {
- _Base_ptr __y = _M_clone_node(__x);
- _S_left(__p) = __y;
- _S_parent(__y) = __p;
- if (_S_right(__x))
- _S_right(__y) = _M_copy(_S_right(__x), __y);
- __p = __y;
- __x = _S_left(__x);
- }
- }
- _STLP_UNWIND(_M_erase(__top))
- return __top;
- }
- // this has to stay out-of-line : it's recursive
- template <class _Key, class _Compare,
- class _Value, class _KeyOfValue, class _Traits, class _Alloc>
- void
- _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc>::_M_erase(_Rb_tree_node_base *__x) {
- // erase without rebalancing
- while (__x != 0) {
- _M_erase(_S_right(__x));
- _Base_ptr __y = _S_left(__x);
- _STLP_STD::_Destroy(&_S_value(__x));
- this->_M_header.deallocate(__STATIC_CAST(_Link_type, __x),1);
- __x = __y;
- }
- }
- #if defined (_STLP_DEBUG)
- inline int
- __black_count(_Rb_tree_node_base* __node, _Rb_tree_node_base* __root) {
- if (__node == 0)
- return 0;
- else {
- int __bc = __node->_M_color == _S_rb_tree_black ? 1 : 0;
- if (__node == __root)
- return __bc;
- else
- return __bc + __black_count(__node->_M_parent, __root);
- }
- }
- template <class _Key, class _Compare,
- class _Value, class _KeyOfValue, class _Traits, class _Alloc>
- bool _Rb_tree<_Key,_Compare,_Value,_KeyOfValue,_Traits,_Alloc>::__rb_verify() const {
- if (_M_node_count == 0 || begin() == end())
- return ((_M_node_count == 0) &&
- (begin() == end()) &&
- (this->_M_header._M_data._M_left == &this->_M_header._M_data) &&
- (this->_M_header._M_data._M_right == &this->_M_header._M_data));
- int __len = __black_count(_M_leftmost(), _M_root());
- for (const_iterator __it = begin(); __it != end(); ++__it) {
- _Base_ptr __x = __it._M_node;
- _Base_ptr __L = _S_left(__x);
- _Base_ptr __R = _S_right(__x);
- if (__x->_M_color == _S_rb_tree_red)
- if ((__L && __L->_M_color == _S_rb_tree_red) ||
- (__R && __R->_M_color == _S_rb_tree_red))
- return false;
- if (__L && _M_key_compare(_S_key(__x), _S_key(__L)))
- return false;
- if (__R && _M_key_compare(_S_key(__R), _S_key(__x)))
- return false;
- if (!__L && !__R && __black_count(__x, _M_root()) != __len)
- return false;
- }
- if (_M_leftmost() != _Rb_tree_node_base::_S_minimum(_M_root()))
- return false;
- if (_M_rightmost() != _Rb_tree_node_base::_S_maximum(_M_root()))
- return false;
- return true;
- }
- #endif /* _STLP_DEBUG */
- _STLP_MOVE_TO_STD_NAMESPACE
- _STLP_END_NAMESPACE
- #undef _Rb_tree
- #undef __iterator__
- #undef iterator
- #undef __size_type__
- #endif /* _STLP_TREE_C */
- // Local Variables:
- // mode:C++
- // End:
|