aboutsummaryrefslogtreecommitdiff
path: root/parser.c
diff options
context:
space:
mode:
Diffstat (limited to 'parser.c')
-rw-r--r--parser.c735
1 files changed, 735 insertions, 0 deletions
diff --git a/parser.c b/parser.c
new file mode 100644
index 0000000..cee5e3b
--- /dev/null
+++ b/parser.c
@@ -0,0 +1,735 @@
+#include "parser.h"
+#include "macros.h"
+#include <ctype.h>
+#include <math.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+array_t *STACK;
+ht_t *WORD_TABLE;
+char *INBUF;
+parser_t *PARSER;
+
+array_t *init_array(size_t size) {
+ array_t *a = calloc(1, sizeof(array_t));
+ a->size = 0;
+ a->capacity = size;
+ a->items = calloc(a->capacity, sizeof(value_t *));
+ return a;
+}
+
+void array_append(array_t *a, value_t *v) {
+ if (a->size >= a->capacity - 2) {
+ a->capacity = a->capacity * 2;
+ a->items = realloc(a->items, a->capacity * sizeof(value_t *));
+ }
+ a->items[a->size] = v;
+ a->size++;
+}
+
+value_t *array_pop(array_t *a) {
+ if (a->size > 0) {
+ value_t *v = a->items[a->size - 1];
+ a->size--;
+ return v;
+ }
+ return NULL;
+}
+
+void array_extend(array_t *a, array_t *b) {
+ for (int i = 0; i < b->size; i++) {
+ array_append(a, b->items[i]);
+ }
+}
+
+void array_free(array_t *a) {
+ for (int i = 0; i < a->size; i++) {
+ value_free(a->items[i]);
+ }
+ free(a->items);
+ free(a);
+}
+
+array_t *array_copy(array_t *a) {
+ array_t *b = calloc(1, sizeof(array_t));
+ b->size = a->size;
+ b->capacity = a->capacity;
+ b->items = calloc(b->capacity, sizeof(value_t *));
+ for (int i = 0; i < a->size; i++) {
+ b->items[i] = value_copy(a->items[i]);
+ }
+ return b;
+}
+
+value_t *init_value(int type) {
+ value_t *v = calloc(1, sizeof(value_t));
+ v->type = type;
+ v->escaped = false;
+ return v;
+}
+
+value_t *value_copy(value_t *v) {
+ value_t *a = init_value(VINT);
+ a->type = v->type;
+ if (v->type == VINT || v->type == VFLOAT) {
+ a->int_float = v->int_float;
+ } else if (v->type == VSTR || v->type == VWORD) {
+ a->str_word = string_copy(v->str_word);
+ } else if (v->type == VQUOTE) {
+ a->quote = array_copy(v->quote);
+ }
+ return a;
+}
+
+void value_free(value_t *v) {
+ if (v->type == VSTR || v->type == VSTR || v->type == VWORD) {
+ string_free(v->str_word);
+ }
+ if (v->type == VQUOTE) {
+ array_free(v->quote);
+ }
+ free(v);
+}
+
+parser_t *init_parser(char *source) {
+ parser_t *p = calloc(1, sizeof(parser_t));
+ p->i = 0;
+ p->source = source;
+ p->c = source[0];
+ return p;
+}
+
+void parser_reset(parser_t *p, char *source) {
+ p->source = source;
+ p->i = 0;
+ p->c = source[0];
+}
+
+void parser_move(parser_t *p) {
+ if (p->i < strlen(p->source) && p->c != '\0') {
+ p->i++;
+ p->c = p->source[p->i];
+ }
+}
+
+void parser_skip_whitespace(parser_t *p) {
+ while (isspace(p->c)) {
+ parser_move(p);
+ }
+}
+
+value_t *parse_string(parser_t *p) {
+ value_t *retv = init_value(VSTR);
+ parser_move(p);
+ string_t *s = init_string(NULL);
+ while (p->c != '"' && p->c != '\0') {
+ string_append(s, p->c);
+ parser_move(p);
+ }
+ parser_move(p);
+ retv->str_word = s;
+ return retv;
+}
+
+value_t *parse_quote(parser_t *p) {
+ value_t *retv = init_value(VQUOTE);
+ retv->quote = init_array(10);
+ parser_move(p);
+ parser_skip_whitespace(p);
+ while (p->c != ']') {
+ array_append(retv->quote, parser_get_next(p));
+ parser_skip_whitespace(p);
+ }
+ parser_move(p);
+ return retv;
+}
+
+void parser_error(parser_t *p) { exit(1); }
+
+value_t *parse_word(parser_t *p) {
+ value_t *retv = init_value(VWORD);
+ string_t *s = init_string(NULL);
+ if (p->c == '\\') {
+ retv->escaped = true;
+ parser_move(p);
+ if (isspace(p->c) || p->c == '\0') {
+ parser_error(p);
+ }
+ }
+ while (!isspace(p->c) && p->c != '\0') {
+ string_append(s, p->c);
+ parser_move(p);
+ }
+ retv->str_word = s;
+ return retv;
+}
+
+value_t *parse_num(parser_t *p) {
+ value_t *retv;
+ string_t *s = init_string(NULL);
+ bool is_float = false;
+ while (isdigit(p->c) || p->c == '.') {
+ if (p->c == '.')
+ is_float = true;
+ string_append(s, p->c);
+ parser_move(p);
+ }
+ if (is_float)
+ retv = init_value(VFLOAT);
+ else
+ retv = init_value(VINT);
+ retv->int_float = atof(s->value);
+ string_free(s);
+ return retv;
+}
+
+value_t *parser_get_next(parser_t *p) {
+ parser_skip_whitespace(p);
+ if (isdigit(p->c)) {
+ return parse_num(p);
+ }
+ switch (p->c) {
+ case '"':
+ return parse_string(p);
+ case '[':
+ return parse_quote(p);
+ case '\0':
+ return NULL;
+ default:
+ return parse_word(p);
+ }
+}
+
+node_t *init_node(string_t *key, value_t *value) {
+ node_t *n = calloc(1, sizeof(node_t));
+ n->key = key;
+ n->value = value;
+ n->next = NULL;
+ return n;
+}
+
+void node_free(node_t *n) {
+ string_free(n->key);
+ value_free(n->value);
+ free(n);
+}
+
+sll_t *init_sll() {
+ sll_t *l = calloc(1, sizeof(sll_t));
+ l->size = 0;
+ l->head = NULL;
+ return l;
+}
+
+void sll_add(sll_t *l, string_t *s, value_t *v) {
+ if (l->head == NULL) {
+ node_t *n = init_node(s, v);
+ l->head = n;
+ l->size++;
+ return;
+ }
+ node_t *cur = l->head;
+ while (cur->next != NULL) {
+ if (strcmp(s->value, cur->value->str_word->value) == 0) {
+ value_free(cur->value);
+ string_free(s);
+ cur->value = v;
+ return;
+ }
+ cur = cur->next;
+ }
+ if (strcmp(s->value, cur->value->str_word->value) == 0) {
+ value_free(cur->value);
+ string_free(s);
+ cur->value = v;
+ return;
+ }
+ node_t *n = init_node(s, v);
+ cur->next = n;
+}
+
+value_t *sll_get(sll_t *l, string_t *k) {
+ if (l->head == NULL)
+ return NULL;
+ node_t *cur = l->head;
+ while (cur != NULL) {
+ if (strcmp(k->value, cur->key->value) == 0)
+ return cur->value;
+ cur = cur->next;
+ }
+ return NULL;
+}
+/* TODO */
+void sll_free(sll_t *l) {
+ node_t *cur = l->head;
+ node_t *tmp;
+ while (cur != NULL) {
+ tmp = cur;
+ cur = cur->next;
+ node_free(tmp);
+ }
+ free(l);
+}
+
+ht_t *init_ht(size_t size) {
+ ht_t *h = calloc(1, sizeof(ht_t));
+ h->size = size;
+ h->buckets = calloc(h->size, sizeof(sll_t *));
+ for (int i = 0; i < size; i++) {
+ h->buckets[i] = init_sll();
+ }
+ return h;
+}
+
+void ht_add(ht_t *h, string_t *key, value_t *v) {
+ sll_add(h->buckets[hash(h, key->value)], key, v);
+}
+
+value_t *ht_get(ht_t *h, string_t *key) {
+ return sll_get(h->buckets[hash(h, key->value)], key);
+}
+
+void ht_free(ht_t *h) {
+ for (int i = 0; i < h->size; i++) {
+ sll_free(h->buckets[i]);
+ }
+ free(h->buckets);
+ free(h);
+}
+
+/* DJB2 HASH FUNCTION */
+unsigned long hash(ht_t *h, char *key) {
+ unsigned long hash = 5381;
+ int c;
+
+ while (c = *key++)
+ hash = ((hash << 5) + hash) + c; /* hash * 33 + c */
+
+ return hash % h->size;
+}
+
+void print_value(value_t *v) {
+ switch (v->type) {
+ case VINT:
+ case VFLOAT:
+ printf("%Lf\n", v->int_float);
+ break;
+ case VSTR:
+ printf("%s\n", v->str_word->value);
+ break;
+ case VWORD:
+ printf("W: %s\n", v->str_word->value);
+ break;
+ case VQUOTE:
+ printf("Q:\n");
+ for (int i = 0; i < v->quote->size; i++) {
+ print_value(v->quote->items[i]);
+ }
+ break;
+ }
+}
+
+void eval_error() { exit(1); }
+
+bool eval_builtins(value_t *v) {
+ char *str = v->str_word->value;
+ value_t *v1;
+ value_t *v2;
+ value_t *v3;
+ value_t *retval;
+ if (strcmp(str, "func") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL) {
+ eval_error();
+ }
+ v1 = array_pop(STACK);
+ if (v1 == NULL) {
+ array_append(STACK, v2);
+ eval_error();
+ }
+ if (v1->type != VWORD)
+ eval_error();
+ ht_add(WORD_TABLE, string_copy(v1->str_word), v2);
+ value_free(v1);
+ } else if (strcmp(str, "+") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL) {
+ eval_error();
+ }
+ v1 = array_pop(STACK);
+ if (v1 == NULL) {
+ array_append(STACK, v2);
+ eval_error();
+ }
+ retval = init_value(VFLOAT);
+ if (v1->type == VINT && v2->type == VINT) {
+ retval->type = VINT;
+ }
+ retval->int_float = v1->int_float + v2->int_float;
+ array_append(STACK, retval);
+ value_free(v1);
+ value_free(v2);
+ } else if (strcmp(str, "-") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL) {
+ eval_error();
+ }
+ v1 = array_pop(STACK);
+ if (v1 == NULL) {
+ array_append(STACK, v2);
+ eval_error();
+ }
+ retval = init_value(VFLOAT);
+ if (v1->type == VINT && v2->type == VINT) {
+ retval->type = VINT;
+ }
+ retval->int_float = v1->int_float - v2->int_float;
+ array_append(STACK, retval);
+ value_free(v1);
+ value_free(v2);
+ } else if (strcmp(str, "/") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL) {
+ eval_error();
+ }
+ v1 = array_pop(STACK);
+ if (v1 == NULL) {
+ array_append(STACK, v2);
+ eval_error();
+ }
+ retval = init_value(VFLOAT);
+ if (v1->type == VINT && v2->type == VINT) {
+ retval->type = VINT;
+ }
+ retval->int_float = v1->int_float / v2->int_float;
+ array_append(STACK, retval);
+ value_free(v1);
+ value_free(v2);
+ } else if (strcmp(str, "*") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL) {
+ eval_error();
+ }
+ v1 = array_pop(STACK);
+ if (v1 == NULL) {
+ array_append(STACK, v2);
+ eval_error();
+ }
+ retval = init_value(VFLOAT);
+ if (v1->type == VINT && v2->type == VINT) {
+ retval->type = VINT;
+ }
+ retval->int_float = v1->int_float * v2->int_float;
+ array_append(STACK, retval);
+ value_free(v1);
+ value_free(v2);
+ } else if (strcmp(str, "pow") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL) {
+ eval_error();
+ }
+ v1 = array_pop(STACK);
+ if (v1 == NULL) {
+ array_append(STACK, v2);
+ eval_error();
+ }
+ retval = init_value(VFLOAT);
+ if (v1->type == VINT && v2->type == VINT) {
+ retval->type = VINT;
+ }
+ retval->int_float = powl(v1->int_float, v2->int_float);
+ array_append(STACK, retval);
+ value_free(v1);
+ value_free(v2);
+ } else if (strcmp(str, "eval") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL) {
+ eval_error();
+ }
+ if (v1->type == VQUOTE) {
+ for (int i = 0; i < v1->quote->size; i++) {
+ eval(value_copy(v1->quote->items[i]));
+ }
+ value_free(v1);
+ } else {
+ eval(v1);
+ }
+ } else if (strcmp(str, ".") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL) {
+ eval_error();
+ }
+ print_value(v1);
+ value_free(v1);
+ } else if (strcmp(str, "?") == 0) {
+ for (int i = 0; i < STACK->size; i++) {
+ print_value(STACK->items[i]);
+ }
+ } else if (strcmp(str, "sin") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+ retval = init_value(VFLOAT);
+ retval->int_float = sinhl(v1->int_float);
+ array_append(STACK, retval);
+ value_free(v1);
+ } else if (strcmp(str, "cos") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+ retval = init_value(VFLOAT);
+ retval->int_float = coshl(v1->int_float);
+ array_append(STACK, retval);
+ value_free(v1);
+ } else if (strcmp(str, "exp") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+ retval = init_value(VFLOAT);
+ retval->int_float = expl(v1->int_float);
+ array_append(STACK, retval);
+ value_free(v1);
+ } else if (strcmp(str, "floor") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+ retval = init_value(VFLOAT);
+ retval->int_float = floor(v1->int_float);
+ array_append(STACK, retval);
+ value_free(v1);
+ } else if (strcmp(str, "ceil") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+ retval = init_value(VFLOAT);
+ retval->int_float = ceil(v1->int_float);
+ array_append(STACK, retval);
+ value_free(v1);
+ } else if (strcmp(str, "ln") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+ retval = init_value(VFLOAT);
+ retval->int_float = logl(v1->int_float);
+ array_append(STACK, retval);
+ value_free(v1);
+ } else if (strcmp(str, "ssize") == 0) {
+ retval = init_value(VINT);
+ retval->int_float = STACK->size;
+ array_append(STACK, retval);
+ } else if (strcmp(str, "compose") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL)
+ eval_error();
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+
+ if (v2->type == VSTR && v1->type == VSTR) {
+ retval = init_value(VSTR);
+ string_concat(v1->str_word, v2->str_word);
+ retval->str_word = string_copy(v1->str_word);
+ value_free(v1);
+ value_free(v2);
+ } else if (v2->type == VWORD && v1->type == VWORD) {
+ retval = init_value(VWORD);
+ string_concat(v1->str_word, v2->str_word);
+ retval->str_word = string_copy(v1->str_word);
+ value_free(v1);
+ value_free(v2);
+ } else if (v2->type == VQUOTE && v1->type == VQUOTE) {
+ retval = v1;
+ array_extend(v1->quote, v2->quote);
+ free(v2->quote);
+ free(v2);
+ } else {
+ eval_error();
+ }
+ array_append(STACK, retval);
+ } else if (strcmp(str, "wtostr") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+
+ if (v1->type != VWORD)
+ eval_error();
+ v1->type = VSTR;
+ array_append(STACK, v1);
+ } else if (strcmp(str, "=") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL)
+ eval_error();
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+
+ retval = init_value(VINT);
+ if (v1->type == VSTR && v2->type == VSTR ||
+ v1->type == VWORD && v2->type == VWORD) {
+ retval->int_float = strcmp(v1->str_word->value, v2->str_word->value) == 0;
+ } else if ((v1->type == VINT || v1->type == VFLOAT) &&
+ (v2->type == VINT || v2->type == VFLOAT)) {
+ retval->int_float = v1->int_float == v2->int_float;
+ } else {
+ eval_error();
+ }
+ array_append(STACK, retval);
+ value_free(v1);
+ value_free(v2);
+ } else if (strcmp(str, "if") == 0) {
+ v3 = array_pop(STACK);
+ if (v3 == NULL)
+ eval_error();
+ v2 = array_pop(STACK);
+ if (v2 == NULL)
+ eval_error();
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+
+ if (v3->type != VINT)
+ eval_error();
+
+ if (v3->int_float) {
+ if (v1->type == VQUOTE) {
+ for (int i = 0; i < v1->quote->size; i++) {
+ eval(value_copy(v1->quote->items[i]));
+ }
+ value_free(v1);
+ } else {
+ eval(v1);
+ }
+ value_free(v2);
+ } else {
+ if (v2->type == VQUOTE) {
+ for (int i = 0; i < v2->quote->size; i++) {
+ eval(value_copy(v2->quote->items[i]));
+ }
+ value_free(v2);
+ } else {
+ eval(v2);
+ }
+ value_free(v1);
+ }
+ value_free(v3);
+ } else if (strcmp(str, "clear") == 0) {
+ for (int i = 0; i < STACK->size; i++) {
+ value_free(array_pop(STACK));
+ }
+ } else if (strcmp(str, "curry") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL)
+ eval_error();
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+
+ if (v2->type != VQUOTE)
+ eval_error();
+
+ array_append(v2->quote, v1);
+ array_append(STACK, v2);
+ } else if (strcmp(str, "len") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+
+ retval = init_value(VINT);
+ if (v1->type == VINT || v1->type == VFLOAT) {
+ retval->int_float = 1;
+ } else if (v1->type == VSTR || v1->type == VWORD) {
+ retval->int_float = strlen(v1->str_word->value);
+ } else if (v1->type == VQUOTE) {
+ retval->int_float = v1->quote->size;
+ }
+ array_append(STACK, v1);
+ array_append(STACK, retval);
+ } else if (strcmp(str, "dup") == 0) {
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+
+ retval = value_copy(v1);
+ array_append(STACK, v1);
+ array_append(STACK, retval);
+ } else if (strcmp(str, "exit") == 0) {
+ ht_free(WORD_TABLE);
+ array_free(STACK);
+ free(INBUF);
+ free(PARSER);
+ value_free(v);
+ exit(1);
+ } else if (strcmp(str, "vat") == 0) {
+ v2 = array_pop(STACK);
+ if (v2 == NULL)
+ eval_error();
+ v1 = array_pop(STACK);
+ if (v1 == NULL)
+ eval_error();
+
+ if (v1->type != VINT) {
+ eval_error();
+ }
+ if (v2->type == VINT || v2->type == VFLOAT) {
+ eval_error();
+ }
+
+ if (v2->type == VQUOTE) {
+ if (v2->quote->size <= v1->int_float) {
+ eval_error();
+ }
+ array_append(STACK, v2);
+ array_append(STACK, value_copy(v2->quote->items[(int)v1->int_float]));
+ value_free(v1);
+ } else {
+ eval_error();
+ }
+ } else {
+ return false;
+ }
+ value_free(v);
+ return true;
+}
+
+bool eval_ht(value_t *v) {
+ value_t *func = ht_get(WORD_TABLE, v->str_word);
+ if (func == NULL)
+ return false;
+ if (func->type == VQUOTE) {
+ for (int i = 0; i < func->quote->size; i++) {
+ eval(value_copy(func->quote->items[i]));
+ }
+ } else {
+ eval(value_copy(func));
+ }
+ value_free(v);
+ return true;
+}
+
+void eval(value_t *v) {
+ switch (v->type) {
+ case VINT:
+ case VFLOAT:
+ case VSTR:
+ case VQUOTE:
+ array_append(STACK, v);
+ printf("OK\n");
+ break;
+ case VWORD:
+ if (v->escaped) {
+ v->escaped = false;
+ array_append(STACK, v);
+ printf("OK\n");
+ } else {
+ if (!eval_builtins(v)) {
+ if (!eval_ht(v)) {
+ array_append(STACK, v);
+ printf("OK\n");
+ }
+ }
+ }
+ break;
+ }
+}
generated by cgit (git 2.34.1) at 2025-01-18 13:20:14 +0000