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

query: Use uint32_t for capture list IDs

Browse source
This commit is contained in:
Douglas Creager 2021-06-02 13:19:52 -04:00
parent cd96552448
commit 1f6eac555c
2 changed files with 10 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

16
lib/include/tree_sitter/api.h
View file

@ -801,16 +801,12 @@ void ts_query_cursor_exec(TSQueryCursor *, const TSQuery *, TSNode);
* Manage the maximum number of in-progress matches allowed by this query
* cursor.
*
* Query cursors have a maximum capacity for storing lists of in-progress
* captures. If this capacity is exceeded, then the earliest-starting match will
* silently be dropped to make room for further matches.
*
* By default, this limit is 65,536 pending matches, which is effectively
* unlimited for most queries and syntax trees. You can optionally set this to a
* lower number if you want to have (and check) a tighter bound on query
* complexity.
*
* If you update the match limit, it must be > 0 and <= 65536.
* Query cursors have an optional maximum capacity for storing lists of
* in-progress captures. If this capacity is exceeded, then the
* earliest-starting match will silently be dropped to make room for further
* matches. This maximum capacity is optional — by default, query cursors allow
* any number of pending matches, dynamically allocating new space for them as
* needed as the query is executed.
*/
bool ts_query_cursor_did_exceed_match_limit(const TSQueryCursor *);
uint32_t ts_query_cursor_match_limit(const TSQueryCursor *);

10
lib/src/query.c
View file

@ -152,10 +152,10 @@ typedef struct {
*/
typedef struct {
uint32_t id;
uint32_t capture_list_id;
uint16_t start_depth;
uint16_t step_index;
uint16_t pattern_index;
uint16_t capture_list_id;
uint16_t consumed_capture_count: 12;
bool seeking_immediate_match: 1;
bool has_in_progress_alternatives: 1;
@ -183,7 +183,7 @@ typedef struct {
// use. We reuse those existing-but-unused capture lists before trying to
// allocate any new ones. We use an invalid value (UINT32_MAX) for a capture
// list's length to indicate that it's not in use.
uint16_t free_capture_list_count;
uint32_t free_capture_list_count;
} CaptureListPool;
/*
@ -367,9 +367,7 @@ static CaptureListPool capture_list_pool_new(void) {
return (CaptureListPool) {
.list = array_new(),
.empty_list = array_new(),
// The maximum maxmimum is 64K, since we use `uint16_t` as our capture list
// index type.
.max_capture_list_count = 65536,
.max_capture_list_count = UINT32_MAX,
.free_capture_list_count = 0,
};
}
@ -2318,7 +2316,7 @@ uint32_t ts_query_cursor_match_limit(const TSQueryCursor *self) {
}
void ts_query_cursor_set_match_limit(TSQueryCursor *self, uint32_t limit) {
assert(limit > 0 && limit <= 65536);
assert(limit > 0);
self->capture_list_pool.max_capture_list_count = limit;
}