traxys/tree-sitter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(lib): don't always clear the tree stack

Browse source 
Only do so if the parser is not resuming balancing
This commit is contained in:
Amaan Qureshi 2025-01-21 00:07:25 -05:00
parent 48059b72a8
commit c8353a52af
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
lib/src/parser.c
View file

@ -1850,10 +1850,14 @@ static unsigned ts_parser__condense_stack(TSParser *self) {
static bool ts_parser__balance_subtree(TSParser *self) {
Subtree finished_tree = self->finished_tree;
array_clear(&self->tree_pool.tree_stack);
if (!self->canceled_balancing && ts_subtree_child_count(finished_tree) > 0 && finished_tree.ptr->ref_count == 1) {
array_push(&self->tree_pool.tree_stack, ts_subtree_to_mut_unsafe(finished_tree));
// If we haven't canceled balancing in progress before, then we want to clear the tree stack and
// push the initial finished tree onto it. Otherwise, if we're resuming balancing after a
// cancellation, we don't want to clear the tree stack.
if (!self->canceled_balancing) {
array_clear(&self->tree_pool.tree_stack);
if (ts_subtree_child_count(finished_tree) > 0 && finished_tree.ptr->ref_count == 1) {
array_push(&self->tree_pool.tree_stack, ts_subtree_to_mut_unsafe(finished_tree));
}
}
while (self->tree_pool.tree_stack.size > 0) {