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

Tail-call-optimize recursive functions

Browse source 
Refs https://github.com/maxbrunsfeld/node-tree-sitter-compiler/pull/7
This commit is contained in:
Max Brunsfeld 2016-02-04 12:59:44 -08:00
parent c96c4a08e6
commit f6f02182c1
1 changed files with 22 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

26
src/runtime/tree.c
View file

@ -62,13 +62,21 @@ TSTree *ts_tree_make_copy(TSTree *self) {
}
void ts_tree_assign_parents(TSTree *self) {
TSLength offset = ts_length_zero();
TSLength offset;
recur:
offset = ts_length_zero();
for (size_t i = 0; i < self->child_count; i++) {
TSTree *child = self->children[i];
if (child->context.parent != self) {
child->context.parent = self;
child->context.index = i;
child->context.offset = offset;
if (i == self->child_count - 1) {
self = child;
goto recur;
}
ts_tree_assign_parents(child);
}
offset = ts_length_add(offset, ts_tree_total_size(child));
@ -135,13 +143,23 @@ void ts_tree_retain(TSTree *self) {
void ts_tree_release(TSTree *self) {
if (!self)
return;
recur:
assert(self->ref_count > 0);
self->ref_count--;
if (self->ref_count == 0) {
for (size_t i = 0; i < self->child_count; i++)
ts_tree_release(self->children[i]);
if (self->child_count > 0)
if (self->child_count > 0) {
for (size_t i = 0; i < self->child_count - 1; i++)
ts_tree_release(self->children[i]);
TSTree *last_child = self->children[self->child_count - 1];
ts_free(self->children);
ts_free(self);
self = last_child;
goto recur;
}
ts_free(self);
}
}