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

Add ts_parser_set_enabled API

Browse source
This commit is contained in:
Max Brunsfeld 2018-05-16 17:42:38 -07:00
parent 39c3de3fc8
commit 1fece241aa
3 changed files with 76 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

29
src/runtime/parser.c
View file

@ -60,6 +60,7 @@ struct TSParser {
FILE *dot_graph_file;
bool halt_on_error;
unsigned accept_count;
volatile bool enabled;
};
typedef struct {
@ -705,6 +706,15 @@ static void ts_parser__start(TSParser *self, TSInput input, const Subtree *previ
self->accept_count = 0;
}
static void ts_parser__stop(TSParser *self) {
ts_stack_clear(self->stack);
ts_parser__set_cached_token(self, 0, NULL, NULL);
if (self->finished_tree) {
ts_subtree_release(&self->tree_pool, self->finished_tree);
self->finished_tree = NULL;
}
}
static void ts_parser__accept(TSParser *self, StackVersion version, const Subtree *lookahead) {
assert(lookahead->symbol == ts_builtin_sym_end);
ts_stack_push(self->stack, version, lookahead, false, 1);
@ -1308,6 +1318,7 @@ TSParser *ts_parser_new() {
self->reusable_node = reusable_node_new();
self->dot_graph_file = NULL;
self->halt_on_error = false;
self->enabled = true;
ts_parser__set_cached_token(self, 0, NULL, NULL);
return self;
}
@ -1362,6 +1373,14 @@ void ts_parser_halt_on_error(TSParser *self, bool should_halt_on_error) {
self->halt_on_error = should_halt_on_error;
}
bool ts_parser_enabled(TSParser *self) {
return self->enabled;
}
void ts_parser_set_enabled(TSParser *self, bool enabled) {
self->enabled = enabled;
}
TSTree *ts_parser_parse(TSParser *self, const TSTree *old_tree, TSInput input) {
if (!self->language) return NULL;
ts_parser__start(self, input, old_tree ? old_tree->root : NULL);
@ -1372,6 +1391,11 @@ TSTree *ts_parser_parse(TSParser *self, const TSTree *old_tree, TSInput input) {
for (StackVersion version = 0;
version_count = ts_stack_version_count(self->stack), version < version_count;
version++) {
if (!self->enabled) {
ts_parser__stop(self);
return NULL;
}
bool allow_node_reuse = version_count == 1;
while (ts_stack_is_active(self->stack, version)) {
LOG("process version:%d, version_count:%u, state:%d, row:%u, col:%u",
@ -1404,10 +1428,13 @@ TSTree *ts_parser_parse(TSParser *self, const TSTree *old_tree, TSInput input) {
ts_parser__set_cached_token(self, 0, NULL, NULL);
ts_subtree_balance(self->finished_tree, &self->tree_pool, self->language);
TSTree *result = ts_tree_new(self->finished_tree, self->language);
LOG("done");
LOG_TREE();
self->finished_tree = NULL;
return ts_tree_new(self->finished_tree, self->language);
ts_parser__stop(self);
return result;
}
TSTree *ts_parser_parse_string(TSParser *self, const TSTree *old_tree,