84 lines
2.3 KiB
C
84 lines
2.3 KiB
C
|
#ifndef TREE_SITTER_WASM_H_
|
||
|
|
#define TREE_SITTER_WASM_H_
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
extern "C" {
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include "./array.h"
|
||
|
|
#include "./wasm/wasm.h"
|
||
|
|
#include "./lexer.h"
|
||
|
|
|
||
|
|
typedef struct {
|
||
|
|
uint32_t id;
|
||
|
|
wasm_instance_t *instance;
|
||
|
|
uint32_t main_lex_fn_index;
|
||
|
|
uint32_t keyword_lex_fn_index;
|
||
|
|
uint32_t external_scan_index;
|
||
|
|
} LanguageInstance;
|
||
|
|
|
||
|
|
struct TSWasmStore {
|
||
|
|
wasm_store_t *store;
|
||
|
|
wasm_table_t *function_table;
|
||
|
|
wasm_memory_t *memory;
|
||
|
|
TSLexer *current_lexer;
|
||
|
|
LanguageInstance *current_instance;
|
||
|
|
Array(LanguageInstance) language_instances;
|
||
|
|
};
|
||
|
|
|
||
|
|
struct TSWasmLanguage {
|
||
|
|
uint32_t id;
|
||
|
|
volatile uint32_t ref_count;
|
||
|
|
wasm_shared_module_t *module;
|
||
|
|
TSLanguage *language;
|
||
|
|
const char *name;
|
||
|
|
char *symbol_name_buffer;
|
||
|
|
char *field_name_buffer;
|
||
|
|
};
|
||
|
|
|
||
|
|
void ts_wasm_language_retain(const TSWasmLanguage *);
|
||
|
|
bool ts_wasm_store_start(TSWasmStore *self, TSLexer *lexer, const TSWasmLanguage *language);
|
||
|
|
void ts_wasm_store_stop(TSWasmStore *self);
|
||
|
|
|
||
|
|
static inline bool ts_wasm_store_run_main_lex_function(TSWasmStore *self, TSStateId state) {
|
||
|
|
uint32_t function_index = self->current_instance->main_lex_fn_index;
|
||
|
|
wasm_ref_t *lex_ref = wasm_table_get(self->function_table, function_index);
|
||
|
|
wasm_func_t *lex_func = wasm_ref_as_func(lex_ref);
|
||
|
|
wasm_val_t args_list[2] = {
|
||
|
|
WASM_I32_VAL(state),
|
||
|
|
WASM_I32_VAL(state),
|
||
|
|
};
|
||
|
|
wasm_val_t results_list[1] = {
|
||
|
|
WASM_I32_VAL(0),
|
||
|
|
};
|
||
|
|
wasm_val_vec_t args = WASM_ARRAY_VEC(args_list);
|
||
|
|
wasm_val_vec_t results = WASM_ARRAY_VEC(results_list);
|
||
|
|
wasm_func_call(lex_func, &args, &results);
|
||
|
|
assert(results_list[0].kind == WASM_I32);
|
||
|
|
return results_list[0].of.i32;
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline bool ts_wasm_store_run_keyword_lex_function(TSWasmStore *self, TSStateId state) {
|
||
|
|
uint32_t function_index = self->current_instance->keyword_lex_fn_index;
|
||
|
|
wasm_ref_t *lex_ref = wasm_table_get(self->function_table, function_index);
|
||
|
|
wasm_func_t *lex_func = wasm_ref_as_func(lex_ref);
|
||
|
|
wasm_val_t args_list[2] = {
|
||
|
|
WASM_I32_VAL(state),
|
||
|
|
WASM_I32_VAL(state),
|
||
|
|
};
|
||
|
|
wasm_val_t results_list[1] = {
|
||
|
|
WASM_I32_VAL(0),
|
||
|
|
};
|
||
|
|
wasm_val_vec_t args = WASM_ARRAY_VEC(args_list);
|
||
|
|
wasm_val_vec_t results = WASM_ARRAY_VEC(results_list);
|
||
|
|
wasm_func_call(lex_func, &args, &results);
|
||
|
|
assert(results_list[0].kind == WASM_I32);
|
||
|
|
return results_list[0].of.i32;
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#endif // TREE_SITTER_WASM_H_
|