Damn, it works, the lexer actually works, that's amazing, need to test the bad cases but at least i'm sure it can detect and process correct math expressions

This commit is contained in:
2026-03-10 07:08:12 -06:00
parent 0de6cf5024
commit 73451fcca9
3 changed files with 94 additions and 0 deletions

View File

@@ -72,5 +72,6 @@ size_t ASTNodeArray_len(ASTNodeArray *arr);
LexerErr tokenize(const char* input, ASTNodeArray *out); LexerErr tokenize(const char* input, ASTNodeArray *out);
LexerErr tokenize_number(const char* input, size_t *offset, ASTNode *out); LexerErr tokenize_number(const char* input, size_t *offset, ASTNode *out);
LexerErr string_to_integer(const char buf[], int64_t *number); LexerErr string_to_integer(const char buf[], int64_t *number);
bool isoperator(int c);
#endif // !LEXER_H #endif // !LEXER_H

View File

@@ -112,7 +112,48 @@ size_t ASTNodeArray_len(ASTNodeArray *arr) {
return arr->len; return arr->len;
} }
LexerErr tokenize(const char *input, ASTNodeArray *out) {
size_t offset = 0;
LexerState state = WAIT_FOR_NUMBER;
ASTNodeArray arr = ASTNodeArray_init(0); // 0 defaults to 64
while (input[offset] != '\n' && input[offset] != '\0') {
int current = input[offset];
if (isdigit(current) && state == WAIT_FOR_NUMBER) {
ASTNode new_node;
LexerErr result = tokenize_number(input, &offset, &new_node);
if (result != LEXER_OK) {
ASTNodeArray_free(&arr);
return result;
}
ASTNodeArray_push(&arr, new_node);
state = WAIT_FOR_OPERATOR;
} else if (isoperator(current) && state == WAIT_FOR_OPERATOR) {
ASTNode new_node = {
.type = NODE_BINARY_OP,
.data.binary.op = current,
.data.binary.right = NULL,
.data.binary.left = NULL,
};
ASTNodeArray_push(&arr, new_node);
state = WAIT_FOR_NUMBER;
} else if (isspace(current)) {
// Nothing...
} else {
ASTNodeArray_free(&arr);
return LEXER_NOT_RECOGNIZED_SYMBOL;
}
offset++;
}
*out = arr;
return LEXER_OK;
}
// CURRENTLY, it only supports ints, not clear how floating // CURRENTLY, it only supports ints, not clear how floating
// point is implemented but i'll figure it out // point is implemented but i'll figure it out
@@ -165,3 +206,15 @@ LexerErr string_to_integer(const char *buf, int64_t *number) {
*number = count; *number = count;
return LEXER_OK; return LEXER_OK;
} }
bool isoperator(int c) {
switch (c) {
case '+':
case '-':
case '/':
case '*':
return true;
default:
return false;
}
}

View File

@@ -6,6 +6,45 @@
#include <setjmp.h> #include <setjmp.h>
#include <cmocka.h> #include <cmocka.h>
static void test_tokenize_normal_expresion(void **state) {
(void) state;
char expr[256] = "2 + 3 / 66 * 789";
ASTNodeArray tokens;
ASTNode node;
assert_int_equal(tokenize(expr, &tokens), LEXER_OK);
assert_int_equal(tokens.len, 7);
ASTNodeArray_get(&tokens, 0, &node);
assert_int_equal(node.type, NODE_INTEGER);
assert_int_equal(node.data.integer, 2);
ASTNodeArray_get(&tokens, 1, &node);
assert_int_equal(node.type, NODE_BINARY_OP);
assert_int_equal(node.data.binary.op, '+');
ASTNodeArray_get(&tokens, 2, &node);
assert_int_equal(node.type, NODE_INTEGER);
assert_int_equal(node.data.integer, 3);
ASTNodeArray_get(&tokens, 3, &node);
assert_int_equal(node.type, NODE_BINARY_OP);
assert_int_equal(node.data.binary.op, '/');
ASTNodeArray_get(&tokens, 4, &node);
assert_int_equal(node.type, NODE_INTEGER);
assert_int_equal(node.data.integer, 66);
ASTNodeArray_get(&tokens, 5, &node);
assert_int_equal(node.type, NODE_BINARY_OP);
assert_int_equal(node.data.binary.op, '*');
ASTNodeArray_get(&tokens, 6, &node);
assert_int_equal(node.type, NODE_INTEGER);
assert_int_equal(node.data.integer, 789);
}
static void test_string_to_number_normal(void **state) { static void test_string_to_number_normal(void **state) {
(void) state; (void) state;
@@ -36,6 +75,7 @@ int main(void) {
const struct CMUnitTest tests[] = { const struct CMUnitTest tests[] = {
cmocka_unit_test(test_string_to_number_normal), cmocka_unit_test(test_string_to_number_normal),
cmocka_unit_test(test_string_to_number_overflow), cmocka_unit_test(test_string_to_number_overflow),
cmocka_unit_test(test_tokenize_normal_expresion),
}; };
return cmocka_run_group_tests(tests, NULL, NULL); return cmocka_run_group_tests(tests, NULL, NULL);