Skip to content

Commit 10ed466

Browse files
committed
[libc++] Document how __tree is layed out and how we iterate through it
1 parent 0b3ee20 commit 10ed466

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

libcxx/include/__tree

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,27 @@
4949
_LIBCPP_PUSH_MACROS
5050
#include <__undef_macros>
5151

52+
// __tree is a red-black-tree implementation used for the associative containers (i.e. (multi)map/set). To allow for
53+
// constant time lookup, it stores
54+
// - (1) a pointer to the node with the smallest (i.e. leftmost) element, namely __begin_node_
55+
// - (2) the number of nodes in the tree, namely __size_
56+
//
57+
// A pointer to the root of the tree is stored in the __end_node_.
58+
// A tree looks like this in memory:
59+
//
60+
// __end_node_
61+
// |
62+
// root
63+
// / \
64+
// l1 r1
65+
// / \ / \
66+
// ... ... ... ...
67+
//
68+
// All nodes except __end_node_ have a __left_ and __right_ pointer as well as a __parent_ pointer.
69+
// __end_node_ only contains a __left_ pointer, which point to the root of the tree.
70+
// This layout allows for iteration through the tree without a need for special handling of the end node. See
71+
// __tree_next_iter and __tree_prev_iter for more details.
72+
5273
_LIBCPP_BEGIN_NAMESPACE_STD
5374

5475
template <class _Tp, class _Compare, class _Allocator>
@@ -183,6 +204,11 @@ _LIBCPP_HIDE_FROM_ABI _NodePtr __tree_next(_NodePtr __x) _NOEXCEPT {
183204
return __x->__parent_unsafe();
184205
}
185206

207+
// __tree_next_iter and __tree_prev_iter implement iteration through the tree. The order is as follows:
208+
// left sub-tree -> node -> right sub-tree. When the right-most node of a sub-tree is reached, we walk up the tree until
209+
// we find a node where we were in the left sub-tree. We are _always_ in a left sub-tree, since the __end_node_ points
210+
// to the actual root of the tree through a __left_ pointer. incrementing the end() pointer is UB, so we can assume that
211+
// never happens.
186212
template <class _EndNodePtr, class _NodePtr>
187213
inline _LIBCPP_HIDE_FROM_ABI _EndNodePtr __tree_next_iter(_NodePtr __x) _NOEXCEPT {
188214
_LIBCPP_ASSERT_INTERNAL(__x != nullptr, "node shouldn't be null");

0 commit comments

Comments
 (0)