Add functions to separately register different types of operators.

This commit is contained in:
Danila Fedorin 2018-02-17 14:00:37 -08:00
parent a3ce8fbd9c
commit 8cd1d23120
4 changed files with 45 additions and 8 deletions

View File

@ -48,7 +48,26 @@ libab_result libab_init(libab* ab);
* @param func the function that describes the functionality of the operator. * @param func the function that describes the functionality of the operator.
* @return the result of the initialization. * @return the result of the initialization.
*/ */
libab_result libab_register_operator(libab* ab, const char* op, int precedence, libab_function_ptr func); libab_result libab_register_operator_infix(libab* ab, const char* op, int precedence, libab_function_ptr func);
/**
* Registers an operation with libabacus that appears
* before its operand.
* @param ab the libabacus instance to register the operator with.
* @param op the operator string to register.
* @param func the function that describes the functionality of the operator.
* @return the result of the registration.
*/
libab_result libab_register_operator_prefix(libab* ab, const char* op, libab_function_ptr func);
/**
* Registers an operation with libabacus that appears
* after its operand.
* @param ab the libabacus instance to register the operator with.
* @param op the operator string to register.
* @param func the function that describes the functionality of the operator.
* @return the result of the registration.
*/
libab_result libab_register_operator_postfix(libab* ab, const char* op, libab_function_ptr func);
/** /**
* Registers a function with libabacus. * Registers a function with libabacus.
* @param ab the libabacus instance used to keep state. * @param ab the libabacus instance used to keep state.

View File

@ -30,8 +30,8 @@ struct libab_table_s {
enum libab_table_entry_variant_e { enum libab_table_entry_variant_e {
ENTRY_VALUE, ENTRY_VALUE,
ENTRY_TYPE, ENTRY_TYPE,
ENTRY_OPERATOR, ENTRY_OP,
ENTRY_FUNCTION ENTRY_FUN
}; };
/** /**

View File

@ -1,5 +1,6 @@
#include "libabacus.h" #include "libabacus.h"
#include <stdlib.h> #include <stdlib.h>
#include "util.h"
libab_result libab_init(libab* ab) { libab_result libab_init(libab* ab) {
libab_table_init(&ab->table); libab_table_init(&ab->table);
@ -7,33 +8,50 @@ libab_result libab_init(libab* ab) {
return libab_lexer_init(&ab->lexer); return libab_lexer_init(&ab->lexer);
} }
libab_result libab_register_operator(libab* ab, const char* op, int precedence, libab_function_ptr func) { libab_result _register_operator(libab* ab, const char* op, int token_type, int precedence, libab_function_ptr func) {
libab_result result = LIBAB_SUCCESS; libab_result result = LIBAB_SUCCESS;
libab_table_entry* new_entry; libab_table_entry* new_entry;
if((new_entry = malloc(sizeof(*new_entry)))) { if((new_entry = malloc(sizeof(*new_entry)))) {
new_entry->variant = ENTRY_OPERATOR; new_entry->variant = ENTRY_OP;
new_entry->data_u.op.function = func; new_entry->data_u.op.function = func;
new_entry->data_u.op.precedence = precedence; new_entry->data_u.op.precedence = precedence;
} else { } else {
result = LIBAB_MALLOC; result = LIBAB_MALLOC;
} }
if(result == LIBAB_SUCCESS) {
result = libab_convert_lex_result(eval_config_add(&ab->lexer.config, op, token_type));
}
if(result == LIBAB_SUCCESS) { if(result == LIBAB_SUCCESS) {
result = libab_table_put(&ab->table, op, new_entry); result = libab_table_put(&ab->table, op, new_entry);
} }
if(result != LIBAB_SUCCESS) { if(result != LIBAB_SUCCESS) {
eval_config_remove(&ab->lexer.config, op, token_type);
free(new_entry); free(new_entry);
} }
return result; return result;
} }
libab_result libab_register_operator_infix(libab* ab, const char* op, int precedence, libab_function_ptr func) {
return _register_operator(ab, op, TOKEN_OP_INFIX, precedence, func);
}
libab_result libab_register_operator_prefix(libab* ab, const char* op, libab_function_ptr func) {
return _register_operator(ab, op, TOKEN_OP_PREFIX, 0, func);
}
libab_result libab_register_operator_postfix(libab* ab, const char* op, libab_function_ptr func) {
return _register_operator(ab, op, TOKEN_OP_POSTFIX, 0, func);
}
libab_result libab_register_function(libab* ab, const char* name, libab_function_ptr func) { libab_result libab_register_function(libab* ab, const char* name, libab_function_ptr func) {
libab_result result = LIBAB_SUCCESS; libab_result result = LIBAB_SUCCESS;
libab_table_entry* new_entry; libab_table_entry* new_entry;
if((new_entry = malloc(sizeof(*new_entry)))) { if((new_entry = malloc(sizeof(*new_entry)))) {
new_entry->variant = ENTRY_FUNCTION; new_entry->variant = ENTRY_FUN;
new_entry->data_u.function.function = func; new_entry->data_u.function.function = func;
} else { } else {
result = LIBAB_MALLOC; result = LIBAB_MALLOC;

View File

@ -17,7 +17,7 @@ libab_table_entry* table_search(libab_table* table, const char* string) {
libab_operator* libab_table_search_operator(libab_table* table, const char* string) { libab_operator* libab_table_search_operator(libab_table* table, const char* string) {
libab_table_entry* entry = table_search(table, string); libab_table_entry* entry = table_search(table, string);
libab_operator* to_return = NULL; libab_operator* to_return = NULL;
if(entry && entry->variant == ENTRY_OPERATOR) { if(entry && entry->variant == ENTRY_OP) {
to_return = &entry->data_u.op; to_return = &entry->data_u.op;
} }
return to_return; return to_return;
@ -25,7 +25,7 @@ libab_operator* libab_table_search_operator(libab_table* table, const char* stri
libab_function* libab_table_search_function(libab_table* table, const char* string) { libab_function* libab_table_search_function(libab_table* table, const char* string) {
libab_table_entry* entry = table_search(table, string); libab_table_entry* entry = table_search(table, string);
libab_function* to_return = NULL; libab_function* to_return = NULL;
if(entry && entry->variant == ENTRY_FUNCTION) { if(entry && entry->variant == ENTRY_FUN) {
to_return = &entry->data_u.function; to_return = &entry->data_u.function;
} }
return to_return; return to_return;