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

Precompute trees' visible children

Browse source
This commit is contained in:
Max Brunsfeld 2014-07-18 13:08:14 -07:00
parent b3385f20c8
commit 0e11bf7271
6 changed files with 92 additions and 71 deletions
Download patch file Download diff file Expand all files Collapse all files

62
src/runtime/node.c
View file

@ -2,19 +2,25 @@
#include "tree_sitter/parser.h"
#include "runtime/tree.h"
TSNode * ts_node_make(TSTree *tree, TSNode *parent, size_t position, TSParserConfig *config) {
TSNode * ts_node_make(const TSTree *tree, TSNode *parent, size_t start_position, TSParserConfig *config) {
if (parent) ts_node_retain(parent);
TSNode *result = malloc(sizeof(TSNode));
*result = (TSNode) {
.ref_count = 1,
.parent = parent,
.content = tree,
.position = position,
.start_position = start_position,
.config = config
};
return result;
}
TSNode * ts_node_make_root(const TSTree *tree, TSParserConfig *config) {
while (ts_tree_is_wrapper(tree))
tree = tree->children[0];
return ts_node_make(tree, NULL, 0, config);
}
void ts_node_retain(TSNode *node) {
node->ref_count++;
}
@ -28,7 +34,7 @@ void ts_node_release(TSNode *node) {
}
size_t ts_node_pos(const TSNode *node) {
return node->position;
return node->start_position + node->content->offset;
}
size_t ts_node_size(const TSNode *node) {
@ -39,55 +45,17 @@ const char * ts_node_name(const TSNode *node) {
return node->config->symbol_names[node->content->symbol];
}
TSTree * ts_tree_child(const TSTree *tree, size_t goal_index, size_t *offset, size_t *children_seen) {
*offset = 0;
*children_seen = 0;
size_t child_count;
TSTree **children = ts_tree_children(tree, &child_count);
if (!children) {
*offset = tree->offset + tree->size;
}
for (size_t i = 0; i < child_count; i++) {
size_t delta_index = 0, delta_offset = 0;
TSTree *child = children[i];
TSTree *result = NULL;
if (ts_tree_is_visible(child)) {
delta_offset = child->offset;
if (*children_seen == goal_index) {
result = child;
} else {
delta_index = 1;
delta_offset += child->size;
}
} else {
result = ts_tree_child(child, (goal_index - *children_seen), &delta_offset, &delta_index);
}
*offset += delta_offset;
*children_seen += delta_index;
if (result) {
return result;
}
}
return NULL;
}
const char * ts_node_string(const TSNode *node) {
return ts_tree_string(node->content, node->config->symbol_names);
}
TSNode * ts_node_child(TSNode *parent, size_t goal_index) {
size_t offset, index;
TSTree *child = ts_tree_child(parent->content, goal_index, &offset, &index);
if (child)
return ts_node_make(child, parent, offset, parent->config);
else
TSNode * ts_node_child(TSNode *parent, size_t index) {
size_t child_count;
TSChildWithPosition *children = ts_tree_visible_children(parent->content, &child_count);
if (child_count <= index)
return NULL;
size_t position = parent->start_position + children[index].position;
return ts_node_make(children[index].tree, parent, position, parent->config);
}
TSNode * ts_node_leaf_at_pos(TSNode *parent, size_t child_index) {