RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-27 12:07:45 +00:00
Code Issues Activity Actions

LibWeb: Add TreeNode<T>::next_in_pre_order(T* stay_within) variant

Browse source 
This is a scoped variant of the pre-order traversal helper that aborts
when attempting to leave the `stay_within` node.
This commit is contained in:
Andreas Kling 2022-01-17 11:38:14 +01:00
parent 1ce6c49f5e
commit 9358f108c4
1 changed files with 20 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

20
Userland/Libraries/LibWeb/TreeNode.h
View file

@ -140,11 +140,31 @@ public:
return node; return node;
} }
T* next_in_pre_order(T const* stay_within)
{
if (first_child())
return first_child();
T* node = static_cast<T*>(this);
T* next = nullptr;
while (!(next = node->next_sibling())) {
node = node->parent();
if (!node || node == stay_within)
return nullptr;
}
return next;
}
T const* next_in_pre_order() const T const* next_in_pre_order() const
{ {
return const_cast<TreeNode*>(this)->next_in_pre_order(); return const_cast<TreeNode*>(this)->next_in_pre_order();
} }
T const* next_in_pre_order(T const* stay_within) const
{
return const_cast<TreeNode*>(this)->next_in_pre_order(stay_within);
}
T* previous_in_pre_order() T* previous_in_pre_order()
{ {
if (auto* node = previous_sibling()) { if (auto* node = previous_sibling()) {