|
@@ -0,0 +1,1781 @@
|
|
|
+#include <ctype.h>
|
|
|
+#include <stdio.h>
|
|
|
+#include <stdarg.h>
|
|
|
+#include <stdlib.h>
|
|
|
+
|
|
|
+size_t GID = 0;
|
|
|
+
|
|
|
+typedef struct {
|
|
|
+ void **data;
|
|
|
+
|
|
|
+ size_t length;
|
|
|
+} list_t;
|
|
|
+
|
|
|
+list_t *list_new(void) {
|
|
|
+ list_t *list = malloc(sizeof(list_t));
|
|
|
+ list->data = NULL;
|
|
|
+ list->length = 0;
|
|
|
+
|
|
|
+ return list;
|
|
|
+}
|
|
|
+
|
|
|
+void list_push(list_t *l, void *v) {
|
|
|
+ size_t i = l->length++;
|
|
|
+
|
|
|
+ l->data = realloc(l->data, l->length * sizeof(void *));
|
|
|
+
|
|
|
+ l->data[i] = v;
|
|
|
+}
|
|
|
+
|
|
|
+void *list_pop(list_t *l) {
|
|
|
+ if (!l->length)
|
|
|
+ return NULL;
|
|
|
+
|
|
|
+ return l->data[--l->length];
|
|
|
+}
|
|
|
+
|
|
|
+void *list_index(list_t *l, ssize_t index) {
|
|
|
+ if (!l->length)
|
|
|
+ return NULL;
|
|
|
+
|
|
|
+ if (index < 0)
|
|
|
+ index += ((ssize_t)l->length);
|
|
|
+
|
|
|
+ if (index < 0 || index >= l->length)
|
|
|
+ return NULL;
|
|
|
+
|
|
|
+ return l->data[index];
|
|
|
+}
|
|
|
+
|
|
|
+void list_set(list_t *l, ssize_t index, void *v) {
|
|
|
+ if (!l->length)
|
|
|
+ return;
|
|
|
+
|
|
|
+ if (index < 0)
|
|
|
+ index += ((ssize_t)l->length);
|
|
|
+
|
|
|
+ if (index < 0 || index >= l->length)
|
|
|
+ return;
|
|
|
+
|
|
|
+ l->data[index] = v;
|
|
|
+}
|
|
|
+
|
|
|
+typedef struct {
|
|
|
+ size_t *data;
|
|
|
+ size_t length;
|
|
|
+} stack_t;
|
|
|
+
|
|
|
+stack_t *stack_new(void) {
|
|
|
+ stack_t *stack = malloc(sizeof(list_t));
|
|
|
+ stack->data = NULL;
|
|
|
+ stack->length = 0;
|
|
|
+
|
|
|
+ return stack;
|
|
|
+}
|
|
|
+
|
|
|
+void stack_push(stack_t *l, size_t v) {
|
|
|
+ size_t i = l->length++;
|
|
|
+
|
|
|
+ l->data = realloc(l->data, l->length * sizeof(size_t));
|
|
|
+ l->data[i] = v;
|
|
|
+}
|
|
|
+
|
|
|
+size_t stack_pop(stack_t *l) {
|
|
|
+ if (!l->length)
|
|
|
+ return 0;
|
|
|
+
|
|
|
+ return l->data[--l->length];
|
|
|
+}
|
|
|
+
|
|
|
+struct entry_t {
|
|
|
+ char *key;
|
|
|
+ void *value;
|
|
|
+};
|
|
|
+
|
|
|
+struct table_t {
|
|
|
+ struct entry_t *entries;
|
|
|
+
|
|
|
+ size_t used;
|
|
|
+ size_t capacity;
|
|
|
+};
|
|
|
+
|
|
|
+typedef struct entry_t entry_t;
|
|
|
+typedef struct table_t table_t;
|
|
|
+
|
|
|
+table_t *table_new() {
|
|
|
+ table_t *table = malloc(sizeof(table_t));
|
|
|
+ table->used = 0;
|
|
|
+
|
|
|
+ table->capacity = 32;
|
|
|
+ table->entries = calloc(table->capacity, sizeof(entry_t));
|
|
|
+
|
|
|
+ return table;
|
|
|
+}
|
|
|
+
|
|
|
+unsigned long ht_hash(const char* key) {
|
|
|
+ unsigned long hash = 5381;
|
|
|
+ int c;
|
|
|
+
|
|
|
+ while ((c = *key++))
|
|
|
+ hash = ((hash << 5) + hash) + c;
|
|
|
+
|
|
|
+ return hash;
|
|
|
+}
|
|
|
+
|
|
|
+void *table_get(table_t *table, char *key) {
|
|
|
+ if (!table->used)
|
|
|
+ return NULL;
|
|
|
+
|
|
|
+ unsigned long hash = ht_hash(key);
|
|
|
+ size_t index = hash % table->capacity;
|
|
|
+
|
|
|
+ size_t i = index;
|
|
|
+
|
|
|
+ while (table->entries[i].key) {
|
|
|
+ if (strcmp(table->entries[i].key, key) == 0)
|
|
|
+ return table->entries[i].value;
|
|
|
+
|
|
|
+ i++;
|
|
|
+
|
|
|
+ if (i >= table->capacity)
|
|
|
+ i = 0;
|
|
|
+
|
|
|
+ if (i == index)
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ return NULL;
|
|
|
+}
|
|
|
+
|
|
|
+static void table_entry_set(entry_t *entries, char *key, void *value, size_t capacity, size_t *used) {
|
|
|
+ unsigned long hash = ht_hash(key);
|
|
|
+ size_t index = hash % capacity;
|
|
|
+
|
|
|
+ size_t i = index;
|
|
|
+
|
|
|
+ while (entries[i].key) {
|
|
|
+ if (strcmp(entries[i].key, key) == 0) {
|
|
|
+ entries[i].value = value;
|
|
|
+
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ i++;
|
|
|
+
|
|
|
+ if (i >= capacity)
|
|
|
+ i = 0;
|
|
|
+
|
|
|
+ if (i == index)
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (used)
|
|
|
+ (*used)++;
|
|
|
+
|
|
|
+ entries[i].key = key;
|
|
|
+ entries[i].value = value;
|
|
|
+}
|
|
|
+
|
|
|
+table_t *table_set(table_t *table, char *key, void *value) {
|
|
|
+ if (table->used >= table->capacity) {
|
|
|
+ size_t capacity = table->capacity + 32;
|
|
|
+ entry_t *entries = calloc(capacity, sizeof(entry_t));
|
|
|
+
|
|
|
+ for (size_t i = 0; i < table->capacity; i++) {
|
|
|
+ entry_t entry = table->entries[i];
|
|
|
+
|
|
|
+ if (entry.key)
|
|
|
+ table_entry_set(entries, entry.key, entry.value, capacity, NULL);
|
|
|
+ }
|
|
|
+
|
|
|
+ table->entries = entries;
|
|
|
+ table->capacity = capacity;
|
|
|
+ }
|
|
|
+
|
|
|
+ table_entry_set(table->entries, key, value, table->capacity, &table->used);
|
|
|
+
|
|
|
+ return table;
|
|
|
+}
|
|
|
+
|
|
|
+#define table_iterate(table, code) \
|
|
|
+ { \
|
|
|
+ if (table->used) { \
|
|
|
+ size_t i = 0; \
|
|
|
+ while (i < (table)->capacity) { \
|
|
|
+ entry_t entry = (table)->entries[i]; \
|
|
|
+ if (entry.key) { \
|
|
|
+ code; \
|
|
|
+ } \
|
|
|
+ i++; \
|
|
|
+ } \
|
|
|
+ } \
|
|
|
+ }
|
|
|
+
|
|
|
+typedef struct {
|
|
|
+ char *str;
|
|
|
+
|
|
|
+ size_t size;
|
|
|
+} buffer_t;
|
|
|
+
|
|
|
+buffer_t *buffer_new(void) {
|
|
|
+ buffer_t *buf = malloc(sizeof(buffer_t));
|
|
|
+
|
|
|
+ buf->str = NULL;
|
|
|
+ buf->size = 0;
|
|
|
+
|
|
|
+ return buf;
|
|
|
+}
|
|
|
+
|
|
|
+void buffer_append(buffer_t *buf, char c) {
|
|
|
+ buf->size++;
|
|
|
+
|
|
|
+ void *p = malloc(sizeof(char) * buf->size);
|
|
|
+ if (buf->str)
|
|
|
+ memcpy(p, buf->str, buf->size - 1);
|
|
|
+
|
|
|
+ buf->str = p;
|
|
|
+
|
|
|
+ buf->str[buf->size - 1] = c;
|
|
|
+}
|
|
|
+
|
|
|
+char *buffer_read(buffer_t *buf) {
|
|
|
+ if (buf->size == 0 || buf->str[buf->size - 1])
|
|
|
+ buffer_append(buf, 0);
|
|
|
+
|
|
|
+ return buf->str;
|
|
|
+}
|
|
|
+
|
|
|
+void buffer_appends(buffer_t *buf, char *s) {
|
|
|
+ for (size_t i = 0; i < strlen(s); i++)
|
|
|
+ buffer_append(buf, s[i]);
|
|
|
+}
|
|
|
+
|
|
|
+void buffer_appendb(buffer_t *dst, buffer_t *src) {
|
|
|
+ for (size_t i = 0; i < src->size; i++)
|
|
|
+ buffer_append(dst, src->str[i]);
|
|
|
+}
|
|
|
+
|
|
|
+void buffer_fmt(buffer_t *buf, const char *fmt, ...) {
|
|
|
+ va_list args;
|
|
|
+ va_start(args, fmt);
|
|
|
+
|
|
|
+ size_t size = vsnprintf(NULL, 0, fmt, args);
|
|
|
+ char *str = malloc(sizeof(char) * (size + 1));
|
|
|
+
|
|
|
+ vsnprintf(str, size + 1, fmt, args);
|
|
|
+
|
|
|
+ va_end(args);
|
|
|
+
|
|
|
+ buffer_appends(buf, str);
|
|
|
+}
|
|
|
+
|
|
|
+typedef struct {
|
|
|
+ enum {
|
|
|
+ T_EOF,
|
|
|
+
|
|
|
+ T_NUMBER,
|
|
|
+ T_STRING,
|
|
|
+ T_NAME,
|
|
|
+
|
|
|
+ T_VAR,
|
|
|
+ T_IF,
|
|
|
+ T_ELSE,
|
|
|
+ T_ELIF,
|
|
|
+ T_FOR,
|
|
|
+ T_BREAK,
|
|
|
+ T_CONTINUE,
|
|
|
+ T_PASS,
|
|
|
+ T_FUNC,
|
|
|
+ T_USE,
|
|
|
+ T_RETURN,
|
|
|
+ T_DEFER,
|
|
|
+
|
|
|
+ T_LPAR,
|
|
|
+ T_RPAR,
|
|
|
+ T_LSB,
|
|
|
+ T_RSB,
|
|
|
+ T_LCB,
|
|
|
+ T_RCB,
|
|
|
+
|
|
|
+ T_EQUALS,
|
|
|
+ T_PLUSASSIGN,
|
|
|
+ T_BARBAR,
|
|
|
+ T_ANDAND,
|
|
|
+ T_STARSTAR,
|
|
|
+ T_PLUSPLUS,
|
|
|
+ T_MINUSMINUS,
|
|
|
+
|
|
|
+ T_PLUS,
|
|
|
+ T_MINUS,
|
|
|
+ T_QM,
|
|
|
+ T_COLON,
|
|
|
+ T_BAR,
|
|
|
+ T_AND,
|
|
|
+ T_RAISE,
|
|
|
+ T_LT,
|
|
|
+ T_LTLT,
|
|
|
+ T_GT,
|
|
|
+ T_GTGT,
|
|
|
+ T_STAR,
|
|
|
+ T_SLASH,
|
|
|
+ T_COMMA,
|
|
|
+ T_DOT,
|
|
|
+
|
|
|
+ T_ASSIGN,
|
|
|
+ T_SEMI
|
|
|
+ } tag;
|
|
|
+
|
|
|
+ char *text;
|
|
|
+
|
|
|
+ size_t fi;
|
|
|
+ size_t pos;
|
|
|
+} token_t;
|
|
|
+
|
|
|
+token_t *token(int tag, char *text) {
|
|
|
+ token_t *tok = malloc(sizeof(token_t));
|
|
|
+ tok->tag = tag;
|
|
|
+ tok->text = text;
|
|
|
+
|
|
|
+ return tok;
|
|
|
+}
|
|
|
+
|
|
|
+#define TK(tk) (token(T_##tk, NULL))
|
|
|
+#define WS() while (source[*pos] == ' ' || source[*pos] == '\t' || source[*pos] == '\n' || source[*pos] == '\r') { (*pos)++; }
|
|
|
+
|
|
|
+void consume_ignored(char *source, size_t *pos) {
|
|
|
+ WS();
|
|
|
+
|
|
|
+ while (source[*pos] == '#') {
|
|
|
+ (*pos)++;
|
|
|
+
|
|
|
+ for (;;) {
|
|
|
+ if (!source[*pos])
|
|
|
+ break;
|
|
|
+
|
|
|
+ if (source[*pos] == '\n') {
|
|
|
+ (*pos)++;
|
|
|
+
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ (*pos)++;
|
|
|
+ }
|
|
|
+
|
|
|
+ WS();
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+list_t *FILES;
|
|
|
+
|
|
|
+void traverse(char *source, size_t pos, size_t *line, size_t *col) {
|
|
|
+ *line = 1;
|
|
|
+ *col = 1;
|
|
|
+
|
|
|
+ for (size_t i = 0; i < pos; i++) {
|
|
|
+ if (source[i] == '\n') {
|
|
|
+ (*line)++;
|
|
|
+ (*col) = 1;
|
|
|
+ } else (*col)++;
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+void format_error(char *filename, char *source, size_t pos, char *fmt, ...) {
|
|
|
+ size_t line, col;
|
|
|
+ traverse(source, pos, &line, &col);
|
|
|
+
|
|
|
+ va_list args;
|
|
|
+ va_start(args, fmt);
|
|
|
+
|
|
|
+ fprintf(stderr, "%s (%zu:%zu): ", filename, line, col);
|
|
|
+ vfprintf(stderr, fmt, args);
|
|
|
+ fputc('\n', stderr);
|
|
|
+
|
|
|
+ va_end(args);
|
|
|
+}
|
|
|
+
|
|
|
+#define GETFNAME(fi) ((char *)((list_t *)list_index(FILES, fi))->data[0])
|
|
|
+#define GETSRC(fi) ((char *)((list_t *)list_index(FILES, fi))->data[1])
|
|
|
+
|
|
|
+#define LEX_ERROR(fmt, ...) { format_error(GETFNAME(-1), source, *pos, fmt, ##__VA_ARGS__); exit(1); }
|
|
|
+
|
|
|
+token_t *next_token(char *source, size_t *pos) {
|
|
|
+ if (!source[*pos])
|
|
|
+ return token(T_EOF, NULL);
|
|
|
+
|
|
|
+ if (source[*pos] == '"' || source[*pos] == '\'') {
|
|
|
+ char term = source[(*pos)++];
|
|
|
+
|
|
|
+ buffer_t *text = buffer_new();
|
|
|
+
|
|
|
+ while (source[*pos] != term) {
|
|
|
+ if (!source[*pos])
|
|
|
+ LEX_ERROR("unterminated string literal");
|
|
|
+
|
|
|
+ char c = source[(*pos)++];
|
|
|
+
|
|
|
+ buffer_append(text, c);
|
|
|
+ }
|
|
|
+
|
|
|
+ (*pos)++;
|
|
|
+
|
|
|
+ return token(T_STRING, buffer_read(text));
|
|
|
+ } else if (isdigit(source[*pos])) {
|
|
|
+ buffer_t *number = buffer_new();
|
|
|
+ int dot = 0;
|
|
|
+ int sub = 0;
|
|
|
+
|
|
|
+ do {
|
|
|
+ buffer_append(number, source[(*pos)++]);
|
|
|
+
|
|
|
+ if (!dot && source[*pos] == '.') {
|
|
|
+ buffer_append(number, source[(*pos)++]);
|
|
|
+
|
|
|
+ if (!isdigit(source[*pos]))
|
|
|
+ LEX_ERROR("illegal number literal (missing part after floating point)");
|
|
|
+
|
|
|
+ dot = 1;
|
|
|
+ } else if (!sub && source[*pos] == '.') {
|
|
|
+ (*pos)++;
|
|
|
+
|
|
|
+ sub = 1;
|
|
|
+ } else if (sub) sub = 0;
|
|
|
+ } while (isdigit(source[*pos]));
|
|
|
+
|
|
|
+ return token(T_NUMBER, buffer_read(number));
|
|
|
+ } else if (isalpha(source[*pos]) || source[*pos] == '_') {
|
|
|
+ buffer_t *text = buffer_new();
|
|
|
+
|
|
|
+ do {
|
|
|
+ buffer_append(text, source[(*pos)++]);
|
|
|
+ } while (isalpha(source[*pos]) || source[*pos] == '_' || isdigit(source[*pos]));
|
|
|
+
|
|
|
+ char *name = buffer_read(text);
|
|
|
+
|
|
|
+ if (strcmp(name, "var") == 0)
|
|
|
+ return TK(VAR);
|
|
|
+ else if (strcmp(name, "if") == 0)
|
|
|
+ return TK(IF);
|
|
|
+ else if (strcmp(name, "else") == 0)
|
|
|
+ return TK(ELSE);
|
|
|
+ else if (strcmp(name, "elif") == 0)
|
|
|
+ return TK(ELIF);
|
|
|
+ else if (strcmp(name, "for") == 0)
|
|
|
+ return TK(FOR);
|
|
|
+ else if (strcmp(name, "break") == 0)
|
|
|
+ return TK(BREAK);
|
|
|
+ else if (strcmp(name, "continue") == 0)
|
|
|
+ return TK(CONTINUE);
|
|
|
+ else if (strcmp(name, "func") == 0)
|
|
|
+ return TK(FUNC);
|
|
|
+ else if (strcmp(name, "use") == 0)
|
|
|
+ return TK(USE);
|
|
|
+ else if (strcmp(name, "return") == 0)
|
|
|
+ return TK(RETURN);
|
|
|
+ else if (strcmp(name, "defer") == 0)
|
|
|
+ return TK(DEFER);
|
|
|
+ else if (strcmp(name, "pass") == 0)
|
|
|
+ return TK(PASS);
|
|
|
+
|
|
|
+ return token(T_NAME, name);
|
|
|
+ } else if (strncmp(&source[*pos], "==", 2) == 0 && ++(*pos) && ++(*pos))
|
|
|
+ return TK(EQUALS);
|
|
|
+ else if (strncmp(&source[*pos], "+=", 2) == 0 && ++(*pos) && ++(*pos))
|
|
|
+ return TK(PLUSASSIGN);
|
|
|
+ else if (strncmp(&source[*pos], "||", 2) == 0 && ++(*pos) && ++(*pos))
|
|
|
+ return TK(BARBAR);
|
|
|
+ else if (strncmp(&source[*pos], "&&", 2) == 0 && ++(*pos) && ++(*pos))
|
|
|
+ return TK(ANDAND);
|
|
|
+ else if (strncmp(&source[*pos], "++", 2) == 0 && ++(*pos) && ++(*pos))
|
|
|
+ return TK(PLUSPLUS);
|
|
|
+ else if (strncmp(&source[*pos], "--", 2) == 0 && ++(*pos) && ++(*pos))
|
|
|
+ return TK(MINUSMINUS);
|
|
|
+ else if (source[*pos] == '(' && ++(*pos))
|
|
|
+ return TK(LPAR);
|
|
|
+ else if (source[*pos] == ')' && ++(*pos))
|
|
|
+ return TK(RPAR);
|
|
|
+ else if (source[*pos] == '[' && ++(*pos))
|
|
|
+ return TK(LSB);
|
|
|
+ else if (source[*pos] == ']' && ++(*pos))
|
|
|
+ return TK(RSB);
|
|
|
+ else if (source[*pos] == '{' && ++(*pos))
|
|
|
+ return TK(LCB);
|
|
|
+ else if (source[*pos] == '}' && ++(*pos))
|
|
|
+ return TK(RCB);
|
|
|
+ else if (source[*pos] == '+' && ++(*pos))
|
|
|
+ return TK(PLUS);
|
|
|
+ else if (source[*pos] == '-' && ++(*pos))
|
|
|
+ return TK(MINUS);
|
|
|
+ else if (source[*pos] == '*' && ++(*pos))
|
|
|
+ return TK(STAR);
|
|
|
+ else if (source[*pos] == '/' && ++(*pos))
|
|
|
+ return TK(SLASH);
|
|
|
+ else if (source[*pos] == '?' && ++(*pos))
|
|
|
+ return TK(QM);
|
|
|
+ else if (source[*pos] == ':' && ++(*pos))
|
|
|
+ return TK(COLON);
|
|
|
+ else if (source[*pos] == '=' && ++(*pos))
|
|
|
+ return TK(ASSIGN);
|
|
|
+ else if (source[*pos] == ';' && ++(*pos))
|
|
|
+ return TK(SEMI);
|
|
|
+ else if (source[*pos] == ',' && ++(*pos))
|
|
|
+ return TK(COMMA);
|
|
|
+ else if (source[*pos] == '.' && ++(*pos))
|
|
|
+ return TK(DOT);
|
|
|
+
|
|
|
+ LEX_ERROR("unexpected input")
|
|
|
+}
|
|
|
+
|
|
|
+#define NEXT()
|
|
|
+
|
|
|
+list_t *tokenize(char *source) {
|
|
|
+ size_t pos = 0;
|
|
|
+ list_t *toks = list_new();
|
|
|
+
|
|
|
+ do {
|
|
|
+ consume_ignored(source, &pos);
|
|
|
+
|
|
|
+ size_t tok_pos = pos;
|
|
|
+
|
|
|
+ token_t *tok = next_token(source, &pos);
|
|
|
+ tok->fi = FILES->length-1;
|
|
|
+ tok->pos = tok_pos;
|
|
|
+
|
|
|
+ if (tok->tag == T_EOF)
|
|
|
+ break;
|
|
|
+
|
|
|
+ list_push(toks, tok);
|
|
|
+ } while (1);
|
|
|
+
|
|
|
+ return toks;
|
|
|
+}
|
|
|
+
|
|
|
+struct _node_t {
|
|
|
+ enum {
|
|
|
+ N_PROGRAM,
|
|
|
+ N_EXPRSTMT,
|
|
|
+
|
|
|
+ N_BLOCK,
|
|
|
+
|
|
|
+ N_NEGATE,
|
|
|
+
|
|
|
+ N_LITERAL,
|
|
|
+ N_LIST,
|
|
|
+ N_TUPLE,
|
|
|
+ N_NILTUPLE,
|
|
|
+ N_TABLE,
|
|
|
+
|
|
|
+ N_CALL,
|
|
|
+ N_MEMBER,
|
|
|
+ N_INDEX,
|
|
|
+
|
|
|
+ N_ADD,
|
|
|
+ N_SUB,
|
|
|
+ N_MUL,
|
|
|
+ N_DIV,
|
|
|
+
|
|
|
+ N_ASSIGN,
|
|
|
+ N_ASSIGN_ADD,
|
|
|
+
|
|
|
+ N_INC,
|
|
|
+ N_DEC,
|
|
|
+
|
|
|
+ N_VAR,
|
|
|
+ N_IF,
|
|
|
+ N_FOR,
|
|
|
+ N_BREAK,
|
|
|
+ N_CONTINUE,
|
|
|
+ N_FUNCDEF,
|
|
|
+ N_RETURN,
|
|
|
+ N_DEFER,
|
|
|
+ N_PASS,
|
|
|
+
|
|
|
+ N_IFEXPR,
|
|
|
+ N_FUNCEXPR,
|
|
|
+
|
|
|
+ N_LOGOR,
|
|
|
+ N_LOGAND,
|
|
|
+ } tag;
|
|
|
+
|
|
|
+ struct _node_t *a;
|
|
|
+ struct _node_t *b;
|
|
|
+ struct _node_t *c;
|
|
|
+ struct _node_t *d;
|
|
|
+
|
|
|
+ list_t *l;
|
|
|
+
|
|
|
+ table_t *h;
|
|
|
+ table_t *h2;
|
|
|
+
|
|
|
+ token_t *t;
|
|
|
+
|
|
|
+ size_t fi;
|
|
|
+ size_t pos;
|
|
|
+};
|
|
|
+typedef struct _node_t node_t;
|
|
|
+
|
|
|
+node_t *node_pos(node_t *node, size_t fi, size_t pos) {
|
|
|
+ node->fi = fi;
|
|
|
+ node->pos = pos;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *nodet(int tag, token_t *t) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->t = t;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODET(n, a) (node_pos(nodet(N_##n, (a)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *nodel(int tag, list_t *l) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->l = l;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODEL(n, a) (node_pos(nodel(N_##n, (a)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *nodeh(int tag, table_t *h) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->h = h;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODEH(n, a) (node_pos(nodeh(N_##n, (a)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *node0(int tag) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODE0(n) (node_pos(node0(N_##n), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *node1(int tag, node_t *a) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->a = a;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODE1(n, a) (node_pos(node1(N_##n, (a)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *node1l(int tag, node_t *a, list_t *l) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->a = a;
|
|
|
+ node->l = l;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODE1l(n, a, l) (node_pos(node1l(N_##n, (a), (l)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *node1t(int tag, node_t *a, token_t *t) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->a = a;
|
|
|
+ node->t = t;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODE1t(n, a, t) (node_pos(node1t(N_##n, (a), (t)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *node2(int tag, node_t *a, node_t *b) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->a = a;
|
|
|
+ node->b = b;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODE2(n, a, b) (node_pos(node2(N_##n, (a), (b)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *node3(int tag, node_t *a, node_t *b, node_t *c) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->a = a;
|
|
|
+ node->b = b;
|
|
|
+ node->c = c;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODE3(n, a, b, c) (node_pos(node3(N_##n, (a), (b), (c)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *node4(int tag, node_t *a, node_t *b, node_t *c, node_t *d) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->a = a;
|
|
|
+ node->b = b;
|
|
|
+ node->c = c;
|
|
|
+ node->d = d;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODE4(n, a, b, c, d) (node_pos(node4(N_##n, (a), (b), (c), (d)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+node_t *nodef(int tag, token_t *name, table_t *params, table_t *captured, node_t *body) {
|
|
|
+ node_t *node = malloc(sizeof(node_t));
|
|
|
+
|
|
|
+ node->tag = tag;
|
|
|
+ node->t = name;
|
|
|
+ node->h = params;
|
|
|
+ node->h2 = captured;
|
|
|
+ node->a = body;
|
|
|
+
|
|
|
+ return node;
|
|
|
+}
|
|
|
+
|
|
|
+#define NODEF(n, a, b, c, d) (node_pos(nodef(N_##n, (a), (b), (c), (d)), ((token_t *)tokens->data[(*pos)-1])->fi, ((token_t *)tokens->data[(*pos)-1])->pos))
|
|
|
+
|
|
|
+#define AT(tk) (*pos < tokens->length && ((token_t *)tokens->data[*pos])->tag == T_##tk)
|
|
|
+#define MATCH(tk) (AT(tk) && ++(*pos))
|
|
|
+#define PARSE_ERROR(fmt, ...) { format_error(GETFNAME(((token_t *)tokens->data[(*pos)>0?(*pos)-1:(*pos)])->fi), GETSRC(((token_t *)tokens->data[(*pos)>0?(*pos)-1:(*pos)])->fi), ((token_t *)tokens->data[(*pos)>0?(*pos)-1:(*pos)])->pos, fmt, ##__VA_ARGS__); exit(1); }
|
|
|
+#define EXPECT(tk, s) { if (!MATCH(tk)) PARSE_ERROR("expected %s", (s)); }
|
|
|
+
|
|
|
+node_t *parse_expr(list_t *tokens, size_t *pos);
|
|
|
+
|
|
|
+list_t *parse_sequence(list_t *tokens, size_t *pos, int term) {
|
|
|
+ list_t *seq = list_new();
|
|
|
+
|
|
|
+ do {
|
|
|
+ if (term != -1 && *pos < tokens->length && ((token_t *)tokens->data[*pos])->tag == term)
|
|
|
+ break;
|
|
|
+
|
|
|
+ list_push(seq, parse_expr(tokens, pos));
|
|
|
+ } while (MATCH(COMMA));
|
|
|
+
|
|
|
+ return seq;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_func(list_t *tokens, size_t *pos, int is_expr);
|
|
|
+
|
|
|
+node_t *parse_primary(list_t *tokens, size_t *pos) {
|
|
|
+ if (MATCH(FUNC))
|
|
|
+ return parse_func(tokens, pos, 1);
|
|
|
+ else if (MATCH(LPAR)) {
|
|
|
+ if (MATCH(RPAR))
|
|
|
+ return NODE0(NILTUPLE);
|
|
|
+
|
|
|
+ node_t *a = parse_expr(tokens, pos);
|
|
|
+
|
|
|
+ if (MATCH(COMMA)) {
|
|
|
+ list_t *l = list_new();
|
|
|
+ list_push(l, a);
|
|
|
+
|
|
|
+ if (!AT(RPAR))
|
|
|
+ do {
|
|
|
+ node_t *n = parse_expr(tokens, pos);
|
|
|
+
|
|
|
+ list_push(l, n);
|
|
|
+ } while (MATCH(COMMA));
|
|
|
+
|
|
|
+ a = NODEL(TUPLE, l);
|
|
|
+ }
|
|
|
+
|
|
|
+ EXPECT(RPAR, ")");
|
|
|
+
|
|
|
+ return a;
|
|
|
+ } else if (MATCH(LSB)) {
|
|
|
+ list_t *a = parse_sequence(tokens, pos, T_RSB);
|
|
|
+
|
|
|
+ EXPECT(RSB, "]");
|
|
|
+
|
|
|
+ return NODEL(LIST, a);
|
|
|
+ } else if (MATCH(LCB)) {
|
|
|
+ table_t *table = table_new();
|
|
|
+
|
|
|
+ do {
|
|
|
+ if (AT(RCB))
|
|
|
+ break;
|
|
|
+
|
|
|
+ if (!AT(NAME) && !AT(STRING))
|
|
|
+ PARSE_ERROR("expected identifier or string");
|
|
|
+
|
|
|
+ char *key = ((token_t *)tokens->data[(*pos)++])->text;
|
|
|
+
|
|
|
+ EXPECT(COLON, ":");
|
|
|
+
|
|
|
+ node_t *val = parse_expr(tokens, pos);
|
|
|
+
|
|
|
+ table_set(table, key, val);
|
|
|
+ } while (MATCH(COMMA));
|
|
|
+
|
|
|
+ EXPECT(RCB, "}");
|
|
|
+
|
|
|
+ return NODEH(TABLE, table);
|
|
|
+ } else if (MATCH(NUMBER) || MATCH(STRING) || MATCH(NAME))
|
|
|
+ return NODET(LITERAL, tokens->data[(*pos)-1]);
|
|
|
+
|
|
|
+ PARSE_ERROR("expected expression");
|
|
|
+
|
|
|
+ return NULL;
|
|
|
+}
|
|
|
+
|
|
|
+size_t get_lineno(token_t *tok) {
|
|
|
+ size_t line, col;
|
|
|
+ traverse(GETSRC(tok->fi), tok->pos, &line, &col);
|
|
|
+
|
|
|
+ return line;
|
|
|
+}
|
|
|
+
|
|
|
+#define CLIFF (get_lineno(((token_t *)tokens->data[(*pos)>0?(*pos)-1:(*pos)])) != get_lineno(((token_t *)tokens->data[(*pos)>=tokens->length?tokens->length-1:(*pos)])))
|
|
|
+
|
|
|
+node_t *parse_call(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_primary(tokens, pos);
|
|
|
+
|
|
|
+ do {
|
|
|
+ if (!CLIFF && MATCH(LPAR)) {
|
|
|
+ list_t *b = NULL;
|
|
|
+
|
|
|
+ if (!AT(RPAR))
|
|
|
+ b = parse_sequence(tokens, pos, -1);
|
|
|
+
|
|
|
+ EXPECT(RPAR, ")");
|
|
|
+
|
|
|
+ a = NODE1l(CALL, a, b);
|
|
|
+
|
|
|
+ continue;
|
|
|
+ } else if (!CLIFF && MATCH(LSB)) {
|
|
|
+ node_t *b = parse_expr(tokens, pos);
|
|
|
+
|
|
|
+ EXPECT(RSB, "]");
|
|
|
+
|
|
|
+ a = NODE2(INDEX, a, b);
|
|
|
+
|
|
|
+ continue;
|
|
|
+ } else if (!CLIFF && MATCH(DOT)) {
|
|
|
+ if (!AT(NAME))
|
|
|
+ PARSE_ERROR("expected identifier after .");
|
|
|
+
|
|
|
+ a = NODE1t(MEMBER, a, tokens->data[(*pos)++]);
|
|
|
+
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+
|
|
|
+ break;
|
|
|
+ } while (1);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_postfix(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_call(tokens, pos);
|
|
|
+
|
|
|
+ if (CLIFF)
|
|
|
+ return a;
|
|
|
+
|
|
|
+ if (MATCH(PLUSPLUS))
|
|
|
+ return NODE1(INC, a);
|
|
|
+ else if (MATCH(MINUSMINUS))
|
|
|
+ return NODE1(DEC, a);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_unary(list_t *tokens, size_t *pos) {
|
|
|
+ if (MATCH(MINUS)) {
|
|
|
+ node_t *a = parse_unary(tokens, pos);
|
|
|
+
|
|
|
+ return NODE1(NEGATE, a);
|
|
|
+ }
|
|
|
+
|
|
|
+ return parse_postfix(tokens, pos);
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_pow(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_unary(tokens, pos);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_mul(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_pow(tokens, pos);
|
|
|
+
|
|
|
+ do {
|
|
|
+ if (MATCH(STAR)) {
|
|
|
+ node_t *b = parse_pow(tokens, pos);
|
|
|
+
|
|
|
+ a = NODE2(MUL, a, b);
|
|
|
+
|
|
|
+ continue;
|
|
|
+ } else if (MATCH(SLASH)) {
|
|
|
+ node_t *b = parse_pow(tokens, pos);
|
|
|
+
|
|
|
+ a = NODE2(DIV, a, b);
|
|
|
+
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+
|
|
|
+ break;
|
|
|
+ } while (1);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_add(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_mul(tokens, pos);
|
|
|
+
|
|
|
+ do {
|
|
|
+ if (MATCH(PLUS)) {
|
|
|
+ node_t *b = parse_mul(tokens, pos);
|
|
|
+
|
|
|
+ a = NODE2(ADD, a, b);
|
|
|
+
|
|
|
+ continue;
|
|
|
+ } else if (MATCH(MINUS)) {
|
|
|
+ node_t *b = parse_mul(tokens, pos);
|
|
|
+
|
|
|
+ a = NODE2(SUB, a, b);
|
|
|
+
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+
|
|
|
+ break;
|
|
|
+ } while (1);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_shift(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_add(tokens, pos);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_relation(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_shift(tokens, pos);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_equality(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_relation(tokens, pos);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_bitand(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_equality(tokens, pos);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_bitxor(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_bitand(tokens, pos);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_bitor(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_bitxor(tokens, pos);
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_logand(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_bitor(tokens, pos);
|
|
|
+
|
|
|
+ if (MATCH(ANDAND)) {
|
|
|
+ node_t *b = parse_logand(tokens, pos);
|
|
|
+
|
|
|
+ return NODE2(LOGAND, a, b);
|
|
|
+ }
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_logor(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_logand(tokens, pos);
|
|
|
+
|
|
|
+ if (MATCH(BARBAR)) {
|
|
|
+ node_t *b = parse_logor(tokens, pos);
|
|
|
+
|
|
|
+ return NODE2(LOGOR, a, b);
|
|
|
+ }
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_assignment(list_t *tokens, size_t *pos);
|
|
|
+
|
|
|
+node_t *parse_conditional(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_logor(tokens, pos);
|
|
|
+
|
|
|
+ if (MATCH(QM)) {
|
|
|
+ node_t *b = parse_assignment(tokens, pos);
|
|
|
+
|
|
|
+ EXPECT(COLON, ":");
|
|
|
+
|
|
|
+ node_t *c = parse_assignment(tokens, pos);
|
|
|
+
|
|
|
+ return NODE3(IFEXPR, a, b, c);
|
|
|
+ }
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_assignment(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_conditional(tokens, pos);
|
|
|
+
|
|
|
+ if (MATCH(ASSIGN)) {
|
|
|
+ node_t *b = parse_assignment(tokens, pos);
|
|
|
+
|
|
|
+ return NODE2(ASSIGN, a, b);
|
|
|
+ } else if (MATCH(PLUSASSIGN)) {
|
|
|
+ node_t *b = parse_assignment(tokens, pos);
|
|
|
+
|
|
|
+ return NODE2(ASSIGN_ADD, a, b);
|
|
|
+ }
|
|
|
+
|
|
|
+ return a;
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_expr(list_t *tokens, size_t *pos) {
|
|
|
+ return parse_assignment(tokens, pos);
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_stmt(list_t *tokens, size_t *pos);
|
|
|
+
|
|
|
+node_t *parse_block(list_t *tokens, size_t *pos) {
|
|
|
+ EXPECT(LCB, "{");
|
|
|
+
|
|
|
+ list_t *stmts = list_new();
|
|
|
+ while (!AT(EOF) && !AT(RCB))
|
|
|
+ list_push(stmts, parse_stmt(tokens, pos));
|
|
|
+
|
|
|
+ EXPECT(RCB, "}");
|
|
|
+
|
|
|
+ return NODEL(PROGRAM, stmts);
|
|
|
+}
|
|
|
+
|
|
|
+#define BLOCK() (MATCH(COLON)?parse_stmt(tokens, pos):parse_block(tokens, pos))
|
|
|
+
|
|
|
+node_t *parse_if(list_t *tokens, size_t *pos) {
|
|
|
+ node_t *a = parse_expr(tokens, pos);
|
|
|
+ node_t *b = BLOCK();
|
|
|
+ node_t *c = NULL;
|
|
|
+
|
|
|
+ if (MATCH(ELSE))
|
|
|
+ c = BLOCK();
|
|
|
+ else if (MATCH(ELIF))
|
|
|
+ c = parse_if(tokens, pos);
|
|
|
+
|
|
|
+ return NODE3(IF, a, b, c);
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_var(list_t *tokens, size_t *pos) {
|
|
|
+ table_t *h = table_new();
|
|
|
+
|
|
|
+ do {
|
|
|
+ if(!AT(NAME))
|
|
|
+ PARSE_ERROR("expected identifier");
|
|
|
+
|
|
|
+ char *k = ((token_t *)tokens->data[(*pos)++])->text;
|
|
|
+ node_t *v = NULL;
|
|
|
+
|
|
|
+ if (MATCH(ASSIGN))
|
|
|
+ v = parse_expr(tokens, pos);
|
|
|
+
|
|
|
+ table_set(h, k, v);
|
|
|
+ } while (MATCH(COMMA));
|
|
|
+
|
|
|
+ return NODEH(VAR, h);
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_func(list_t *tokens, size_t *pos, int is_expr) {
|
|
|
+ token_t *name = NULL;
|
|
|
+
|
|
|
+ if (!is_expr) {
|
|
|
+ if(!AT(NAME))
|
|
|
+ PARSE_ERROR("expected identifier");
|
|
|
+
|
|
|
+ name = tokens->data[(*pos)++];
|
|
|
+ }
|
|
|
+
|
|
|
+ EXPECT(LPAR, "(");
|
|
|
+
|
|
|
+ table_t *params = NULL;
|
|
|
+
|
|
|
+ if (!AT(RPAR)) {
|
|
|
+ int flag = 0;
|
|
|
+
|
|
|
+ params = table_new();
|
|
|
+
|
|
|
+ size_t argc = 0;
|
|
|
+
|
|
|
+ do {
|
|
|
+ if(!AT(NAME))
|
|
|
+ PARSE_ERROR("expected identifier");
|
|
|
+
|
|
|
+ char *l = ((token_t *)tokens->data[(*pos)++])->text;
|
|
|
+ node_t *r = NULL;
|
|
|
+
|
|
|
+ if (!flag && AT(ASSIGN))
|
|
|
+ flag = 1;
|
|
|
+
|
|
|
+ if (flag) {
|
|
|
+ EXPECT(ASSIGN, "=");
|
|
|
+
|
|
|
+ r = parse_expr(tokens, pos);
|
|
|
+ }
|
|
|
+
|
|
|
+ list_t *pair = list_new();
|
|
|
+
|
|
|
+ size_t *argcp = malloc(sizeof(size_t));
|
|
|
+ memcpy(argcp, &argc, sizeof(size_t));
|
|
|
+
|
|
|
+ argc++;
|
|
|
+
|
|
|
+ list_push(pair, argcp);
|
|
|
+ list_push(pair, r);
|
|
|
+
|
|
|
+ table_set(params, l, pair);
|
|
|
+ } while (MATCH(COMMA));
|
|
|
+ }
|
|
|
+
|
|
|
+ EXPECT(RPAR, ")");
|
|
|
+
|
|
|
+ table_t *captured = NULL;
|
|
|
+
|
|
|
+ if (MATCH(USE)) {
|
|
|
+ EXPECT(RPAR, "(");
|
|
|
+
|
|
|
+ captured = table_new();
|
|
|
+
|
|
|
+ do {
|
|
|
+ if(!AT(NAME))
|
|
|
+ PARSE_ERROR("expected identifier");
|
|
|
+
|
|
|
+ token_t *name = tokens->data[(*pos)++];
|
|
|
+
|
|
|
+ table_set(captured, name->text, NODET(LITERAL, name));
|
|
|
+ } while (MATCH(COMMA));
|
|
|
+
|
|
|
+ EXPECT(RPAR, ")");
|
|
|
+ }
|
|
|
+
|
|
|
+ node_t *body = BLOCK();
|
|
|
+
|
|
|
+ if (is_expr)
|
|
|
+ return NODEF(FUNCEXPR, NULL, params, captured, body);
|
|
|
+
|
|
|
+ return NODEF(FUNCDEF, name, params, captured, body);
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_stmt(list_t *tokens, size_t *pos) {
|
|
|
+ if (MATCH(LCB)) {
|
|
|
+ list_t *stmts = list_new();
|
|
|
+
|
|
|
+ while (!AT(EOF) && !AT(RCB)) {
|
|
|
+ node_t *n = parse_stmt(tokens, pos);
|
|
|
+
|
|
|
+ MATCH(SEMI);
|
|
|
+
|
|
|
+ list_push(stmts, n);
|
|
|
+ }
|
|
|
+
|
|
|
+ EXPECT(RCB, "}");
|
|
|
+
|
|
|
+ return NODEL(BLOCK, stmts);
|
|
|
+ } else if (MATCH(VAR))
|
|
|
+ return parse_var(tokens, pos);
|
|
|
+ else if (MATCH(IF))
|
|
|
+ return parse_if(tokens, pos);
|
|
|
+ else if (MATCH(FOR)) {
|
|
|
+ node_t *a = NULL;
|
|
|
+ node_t *b = NULL;
|
|
|
+ node_t *c = NULL;
|
|
|
+
|
|
|
+ if (!AT(LCB) && !AT(COLON)) {
|
|
|
+ if (MATCH(VAR)) {
|
|
|
+ a = parse_var(tokens, pos);
|
|
|
+ EXPECT(SEMI, ";");
|
|
|
+ b = parse_expr(tokens, pos);
|
|
|
+ EXPECT(SEMI, ";");
|
|
|
+ c = parse_expr(tokens, pos);
|
|
|
+ } else a = parse_expr(tokens, pos);
|
|
|
+ }
|
|
|
+
|
|
|
+ node_t *d = BLOCK();
|
|
|
+
|
|
|
+ return NODE4(FOR, a, b, c, d);
|
|
|
+ } else if (MATCH(BREAK)) return NODE0(BREAK);
|
|
|
+ else if (MATCH(CONTINUE)) return NODE0(CONTINUE);
|
|
|
+ else if (MATCH(FUNC))
|
|
|
+ return parse_func(tokens, pos, 0);
|
|
|
+ else if (MATCH(RETURN)) {
|
|
|
+ node_t *a = NULL;
|
|
|
+
|
|
|
+ if (!AT(RCB) && !AT(EOF) && !CLIFF)
|
|
|
+ a = parse_expr(tokens, pos);
|
|
|
+
|
|
|
+ return NODE1(RETURN, a);
|
|
|
+ } else if (MATCH(DEFER)) {
|
|
|
+ node_t *a;
|
|
|
+
|
|
|
+ if (AT(LCB) || AT(COLON))
|
|
|
+ a = BLOCK();
|
|
|
+ else a = parse_stmt(tokens, pos);
|
|
|
+
|
|
|
+ return NODE1(DEFER, a);
|
|
|
+ } else if (MATCH(PASS)) return NODE0(PASS);
|
|
|
+
|
|
|
+ node_t *n = parse_expr(tokens, pos);
|
|
|
+
|
|
|
+ return NODE1(EXPRSTMT, n);
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse_program(list_t *tokens, size_t *pos) {
|
|
|
+ list_t *stmts = list_new();
|
|
|
+
|
|
|
+ while (*pos < tokens->length) {
|
|
|
+ node_t *n = parse_stmt(tokens, pos);
|
|
|
+
|
|
|
+ MATCH(SEMI);
|
|
|
+
|
|
|
+ list_push(stmts, n);
|
|
|
+ }
|
|
|
+
|
|
|
+ return NODEL(PROGRAM, stmts);
|
|
|
+}
|
|
|
+
|
|
|
+node_t *parse(char *source) {
|
|
|
+ size_t pos = 0;
|
|
|
+
|
|
|
+ return parse_program(tokenize(source), &pos);
|
|
|
+}
|
|
|
+
|
|
|
+#define NEWGID() size_t gid = GID++
|
|
|
+
|
|
|
+#define EMIT(fmt, ...) buffer_fmt(buf, (fmt), ##__VA_ARGS__);
|
|
|
+#define BINOP(s) { EMIT("qi_" s "(state, "); compile_node(gbuf, buf, ctx, lstk, node->a); EMIT(", "); compile_node(gbuf, buf, ctx, lstk, node->b); EMIT(")"); }
|
|
|
+#define UNOP(s) { EMIT("qi_" s "(state, "); compile_node(gbuf, buf, ctx, lstk, node->a); EMIT(")"); }
|
|
|
+#define ASSIGN(lhs, rhs) {\
|
|
|
+ if ((lhs)->tag == N_LITERAL && (lhs)->t->tag == T_NAME) {\
|
|
|
+ EMIT("qi_set(state, false, \"%s\", ", (lhs)->t->text);\
|
|
|
+ rhs;\
|
|
|
+ EMIT(")");\
|
|
|
+ } else if ((lhs)->tag == N_INDEX) {\
|
|
|
+ EMIT("qi_index_set(state, false, ");\
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, (lhs)->a);\
|
|
|
+ EMIT(", ");\
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, (lhs)->b);\
|
|
|
+ EMIT(", ");\
|
|
|
+ rhs;\
|
|
|
+ EMIT(")");\
|
|
|
+ } else if ((lhs)->tag == N_MEMBER) {\
|
|
|
+ EMIT("qi_index_set(state, false, ");\
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, (lhs)->a);\
|
|
|
+ EMIT(", qi_make_string(state, \"%s\"), ", (lhs)->t->text);\
|
|
|
+ rhs;\
|
|
|
+ EMIT(")");\
|
|
|
+ } else COMPILE_ERROR("illegal assignment left-hand side");\
|
|
|
+}
|
|
|
+#define COMPASSIGN(lhs, s, rhs) {\
|
|
|
+ ASSIGN(node->a, {\
|
|
|
+ EMIT("qi_%s(state, ", s);\
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, (lhs));\
|
|
|
+ EMIT(", ");\
|
|
|
+ rhs;\
|
|
|
+ EMIT(")");\
|
|
|
+ });\
|
|
|
+}
|
|
|
+#define COMPILE_ERROR(fmt, ...) { format_error(GETFNAME(node->fi), GETSRC(node->fi), node->pos, fmt, ##__VA_ARGS__); exit(1); }
|
|
|
+
|
|
|
+void compile_node(buffer_t *gbuf, buffer_t *buf, list_t *ctx, stack_t *lstk, node_t *node);
|
|
|
+
|
|
|
+void compile_list(buffer_t *gbuf, buffer_t *buf, list_t *ctx, stack_t *lstk, list_t *seq) {
|
|
|
+ if (!seq || seq->length < 1) {
|
|
|
+ EMIT("NULL");
|
|
|
+
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ buffer_t *tbuf = buffer_new();
|
|
|
+
|
|
|
+ NEWGID();
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "qi_list_t *__list%d(qi_state_t *state) {\n", gid);
|
|
|
+ buffer_fmt(tbuf, "qi_list_t *list = qi_list_make();\n");
|
|
|
+
|
|
|
+ for (size_t i = 0; i < seq->length; i++) {
|
|
|
+ buffer_fmt(tbuf, "qi_list_push(list, ");
|
|
|
+
|
|
|
+ compile_node(gbuf, tbuf, ctx, lstk, seq->data[i]);
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, ");\n");
|
|
|
+ }
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "return list;\n");
|
|
|
+ buffer_fmt(tbuf, "}\n");
|
|
|
+
|
|
|
+ buffer_appendb(gbuf, tbuf);
|
|
|
+
|
|
|
+ EMIT("__list%d(state)", gid);
|
|
|
+}
|
|
|
+
|
|
|
+void compile_table(buffer_t *gbuf, buffer_t *buf, list_t *ctx, stack_t *lstk, table_t *table) {
|
|
|
+ if (!table || table->used < 1) {
|
|
|
+ EMIT("NULL");
|
|
|
+
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ buffer_t *tbuf = buffer_new();
|
|
|
+
|
|
|
+ NEWGID();
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "qi_table_t *__table%d(qi_state_t *state) {\n", gid);
|
|
|
+ buffer_fmt(tbuf, "qi_table_t *table = qi_table_make();\n");
|
|
|
+
|
|
|
+ table_iterate(table, {
|
|
|
+ buffer_fmt(tbuf, "qi_table_set(table, \"%s\", ", entry.key);
|
|
|
+
|
|
|
+ compile_node(gbuf, tbuf, ctx, lstk, entry.value);
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, ");\n");
|
|
|
+ });
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "return table;\n");
|
|
|
+ buffer_fmt(tbuf, "}\n");
|
|
|
+
|
|
|
+ buffer_appendb(gbuf, tbuf);
|
|
|
+
|
|
|
+ EMIT("__table%d(state)", gid);
|
|
|
+}
|
|
|
+
|
|
|
+#define CTXPUSH(s) list_push(ctx, (s))
|
|
|
+#define CTXPOP() list_pop(ctx)
|
|
|
+
|
|
|
+int in_context(list_t *ctx, char *s) {
|
|
|
+ if (!ctx->length)
|
|
|
+ return 0;
|
|
|
+
|
|
|
+ for (ssize_t i = ctx->length - 1; i >= 0; i--) {
|
|
|
+ if (strcmp(ctx->data[i], "gap") == 0)
|
|
|
+ break;
|
|
|
+ else if (strcmp(ctx->data[i], s) == 0)
|
|
|
+ return 1;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+#define INCTX(s) (in_context(ctx, (s)))
|
|
|
+
|
|
|
+#define LPUSH(i) stack_push(lstk, (i))
|
|
|
+#define LPOP() stack_pop(lstk)
|
|
|
+#define LID (lstk->data[lstk->length-1])
|
|
|
+
|
|
|
+void compile_func(buffer_t *gbuf, buffer_t *buf, list_t *ctx, stack_t *lstk, node_t *node) {
|
|
|
+ NEWGID();
|
|
|
+
|
|
|
+ buffer_t *tbuf = buffer_new();
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "qi_value_t *__func%d(qi_state_t *state, qi_size_t pargc, qi_list_t *pargs) {\n", gid);
|
|
|
+
|
|
|
+ CTXPUSH("gap");
|
|
|
+ CTXPUSH("func");
|
|
|
+
|
|
|
+ size_t optargc = 0;
|
|
|
+
|
|
|
+ if (node->h) {
|
|
|
+ table_iterate(node->h, {
|
|
|
+ list_t *pair = entry.value;
|
|
|
+
|
|
|
+ size_t argc = *(size_t *)pair->data[0];
|
|
|
+
|
|
|
+ if (pair->data[1]) {
|
|
|
+ optargc++;
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "qi_set(state, false, false, \"%s\", pargc >= %d? qi_list_index(pargs, %d): ", entry.key, argc+1, argc);
|
|
|
+ compile_node(gbuf, tbuf, ctx, lstk, pair->data[1]);
|
|
|
+ buffer_fmt(tbuf, ");\n");
|
|
|
+ } else
|
|
|
+ buffer_fmt(tbuf, "qi_set(state, false, false, \"%s\", qi_list_index(pargs, %d));\n", entry.key, argc);
|
|
|
+
|
|
|
+ argc++;
|
|
|
+ });
|
|
|
+ }
|
|
|
+
|
|
|
+ compile_node(gbuf, tbuf, ctx, lstk, node->a);
|
|
|
+
|
|
|
+ CTXPOP();
|
|
|
+ CTXPOP();
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "return state->nil;\n");
|
|
|
+ buffer_fmt(tbuf, "}\n");
|
|
|
+
|
|
|
+ buffer_appendb(gbuf, tbuf);
|
|
|
+
|
|
|
+ tbuf = buffer_new();
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "qi_make_function(state, \"%s\", %d, __func%d, ", node->t? node->t->text: "<anon>", !node->h? 0: (node->h->used - optargc), gid);
|
|
|
+ compile_table(gbuf, tbuf, ctx, lstk, node->h2);
|
|
|
+ buffer_fmt(tbuf, ")");
|
|
|
+
|
|
|
+ if (node->tag == N_FUNCEXPR) {
|
|
|
+ buffer_appendb(buf, tbuf);
|
|
|
+
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ EMIT("qi_set(state, false, false, \"%s\", ", node->t->text);
|
|
|
+ buffer_appendb(buf, tbuf);
|
|
|
+ EMIT(");");
|
|
|
+}
|
|
|
+
|
|
|
+void compile_block(buffer_t *gbuf, buffer_t *buf, list_t *ctx, stack_t *lstk, list_t *block) {
|
|
|
+ for (size_t i = 0; i < block->length; i++) {
|
|
|
+ node_t *n = block->data[i];
|
|
|
+
|
|
|
+ if (n->tag == N_FUNCDEF) {
|
|
|
+ compile_func(gbuf, buf, ctx, lstk, n);
|
|
|
+
|
|
|
+ EMIT("\n");
|
|
|
+ } else if (n->tag == N_VAR) {
|
|
|
+ table_iterate(n->h, {
|
|
|
+ EMIT("qi_decl(state, \"%s\", ", entry.key);
|
|
|
+
|
|
|
+ if (entry.value)
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, entry.value);
|
|
|
+ else EMIT("state->nil");
|
|
|
+
|
|
|
+ EMIT(");\n");
|
|
|
+ });
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ for (size_t i = 0; i < block->length; i++) {
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, block->data[i]);
|
|
|
+
|
|
|
+ EMIT("\n");
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+void compile_node(buffer_t *gbuf, buffer_t *buf, list_t *ctx, stack_t *lstk, node_t *node) {
|
|
|
+ switch (node->tag) {
|
|
|
+ case N_PROGRAM:
|
|
|
+ compile_block(gbuf, buf, ctx, lstk, node->l);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_EXPRSTMT:
|
|
|
+ EMIT("(void)(");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->a);
|
|
|
+ EMIT(");");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_BLOCK:
|
|
|
+ EMIT("qi_new_scope(state);\n");
|
|
|
+ compile_block(gbuf, buf, ctx, lstk, node->l);
|
|
|
+ EMIT("qi_old_scope(state);");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_LITERAL:
|
|
|
+ switch (node->t->tag) {
|
|
|
+ case T_NUMBER:
|
|
|
+ EMIT("qi_make_number(state, %s)", node->t->text);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case T_STRING:
|
|
|
+ if (!*(node->t->text)) {
|
|
|
+ EMIT("state->empty_string");
|
|
|
+ } else {
|
|
|
+ EMIT("qi_make_string(state, \"%s\")", node->t->text);
|
|
|
+ }
|
|
|
+ break;
|
|
|
+
|
|
|
+ case T_NAME:
|
|
|
+ EMIT("qi_get(state, \"%s\")", node->t->text);
|
|
|
+ break;
|
|
|
+
|
|
|
+ default:
|
|
|
+ COMPILE_ERROR("not yet implemented");
|
|
|
+ }
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_LIST:
|
|
|
+ EMIT("qi_make_list(state, ");
|
|
|
+ compile_list(gbuf, buf, ctx, lstk, node->l);
|
|
|
+ EMIT(")");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_TUPLE:
|
|
|
+ EMIT("qi_make_tuple(state, ");
|
|
|
+ compile_list(gbuf, buf, ctx, lstk, node->l);
|
|
|
+ EMIT(")");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_NILTUPLE: EMIT("state->empty_tuple"); break;
|
|
|
+
|
|
|
+ case N_TABLE:
|
|
|
+ EMIT("qi_make_table(state, ");
|
|
|
+ compile_table(gbuf, buf, ctx, lstk, node->h);
|
|
|
+ EMIT(")");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_CALL:
|
|
|
+ EMIT("qi_call(state, ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->a);
|
|
|
+ EMIT(", ");
|
|
|
+ compile_list(gbuf, buf, ctx, lstk, node->l);
|
|
|
+ EMIT(")");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_MEMBER:
|
|
|
+ EMIT("qi_index(state, ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->a);
|
|
|
+ EMIT(", qi_make_string(state, \"%s\"))", node->t->text);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_INDEX:
|
|
|
+ EMIT("qi_index(state, ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->a);
|
|
|
+ EMIT(", ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->b);
|
|
|
+ EMIT(")");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_ASSIGN: ASSIGN(node->a, compile_node(gbuf, buf, ctx, lstk, node->b)); break;
|
|
|
+
|
|
|
+ case N_ASSIGN_ADD: COMPASSIGN(node->a, "add", compile_node(gbuf, buf, ctx, lstk, node->b)); break;
|
|
|
+
|
|
|
+ case N_INC:
|
|
|
+ COMPASSIGN(node->a, "add", EMIT("state->one"));
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_DEC:
|
|
|
+ COMPASSIGN(node->a, "sub", EMIT("state->one"));
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_VAR: break;
|
|
|
+
|
|
|
+ case N_IF:
|
|
|
+ EMIT("if (_qi_truthy(state, ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->a);
|
|
|
+ EMIT(") {\n");
|
|
|
+ EMIT("qi_new_subscope(state);\n");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->b);
|
|
|
+ EMIT("qi_old_scope(state);\n");
|
|
|
+ if (node->c) {
|
|
|
+ EMIT("} else {\n");
|
|
|
+ EMIT("qi_new_subscope(state);\n");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->c);
|
|
|
+ EMIT("qi_old_scope(state);\n");
|
|
|
+ }
|
|
|
+ EMIT("}");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_FOR: {
|
|
|
+ NEWGID();
|
|
|
+
|
|
|
+ EMIT("qi_new_subscope(state);\n");
|
|
|
+
|
|
|
+ if (!node->a) {
|
|
|
+ EMIT("for (;;) {\n");
|
|
|
+ } else if (node->a && !node->b) {
|
|
|
+ EMIT("while (_qi_truthy(state, ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->a);
|
|
|
+ EMIT(")) {\n");
|
|
|
+ } else {
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->a);
|
|
|
+ EMIT("while (_qi_truthy(state, ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->b);
|
|
|
+ EMIT(")) {\n");
|
|
|
+ }
|
|
|
+
|
|
|
+ LPUSH(gid);
|
|
|
+ CTXPUSH("for");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->d);
|
|
|
+ CTXPOP();
|
|
|
+ LPOP();
|
|
|
+
|
|
|
+ EMIT("__continue%d:;\n", gid);
|
|
|
+
|
|
|
+ if (node->c)
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->c);
|
|
|
+
|
|
|
+ EMIT("}\n");
|
|
|
+
|
|
|
+ EMIT("__break%d:;\n", gid);
|
|
|
+
|
|
|
+ EMIT("qi_old_scope(state);\n");
|
|
|
+ } break;
|
|
|
+
|
|
|
+ case N_BREAK:
|
|
|
+ if (!INCTX("for"))
|
|
|
+ COMPILE_ERROR("break outside of a loop");
|
|
|
+
|
|
|
+ EMIT("goto __break%d;", LID);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_CONTINUE:
|
|
|
+ if (!INCTX("for"))
|
|
|
+ COMPILE_ERROR("continue outside of a loop");
|
|
|
+
|
|
|
+ EMIT("goto __continue%d;", LID);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_DEFER: {
|
|
|
+ NEWGID();
|
|
|
+
|
|
|
+ buffer_t *tbuf = buffer_new();
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "void __defer%d(qi_state_t *state) {\n", gid);
|
|
|
+
|
|
|
+ CTXPUSH("gap");
|
|
|
+ compile_node(gbuf, tbuf, ctx, lstk, node->a);
|
|
|
+ CTXPOP();
|
|
|
+
|
|
|
+ buffer_fmt(tbuf, "\n");
|
|
|
+ buffer_fmt(tbuf, "}\n");
|
|
|
+
|
|
|
+ buffer_appendb(gbuf, tbuf);
|
|
|
+
|
|
|
+ EMIT("qi_add_defer(state, -1, __defer%d);", gid);
|
|
|
+ } break;
|
|
|
+
|
|
|
+ case N_RETURN:
|
|
|
+ if (!INCTX("func"))
|
|
|
+ COMPILE_ERROR("return outside of a function");
|
|
|
+
|
|
|
+ EMIT("return ");
|
|
|
+
|
|
|
+ if (node->a)
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->a);
|
|
|
+
|
|
|
+ EMIT(";");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_FUNCDEF: break;
|
|
|
+ case N_PASS: break;
|
|
|
+
|
|
|
+ case N_IFEXPR:
|
|
|
+ EMIT("(_qi_truthy(state, ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->a);
|
|
|
+ EMIT(")? ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->b);
|
|
|
+ EMIT(": ");
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, node->c);
|
|
|
+ EMIT(")");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_FUNCEXPR:
|
|
|
+ compile_func(gbuf, buf, ctx, lstk, node);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_ADD:
|
|
|
+ BINOP("add");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_SUB:
|
|
|
+ BINOP("sub");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_MUL:
|
|
|
+ BINOP("mul");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_DIV:
|
|
|
+ BINOP("div");
|
|
|
+ break;
|
|
|
+
|
|
|
+ case N_NEGATE:
|
|
|
+ UNOP("negate");
|
|
|
+ break;
|
|
|
+
|
|
|
+ default:
|
|
|
+ COMPILE_ERROR("not yet implemented");
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+char *compile(char *source) {
|
|
|
+ node_t *n = parse(source);
|
|
|
+
|
|
|
+ list_t *ctx = list_new();
|
|
|
+ stack_t *lstk = stack_new();
|
|
|
+
|
|
|
+ buffer_t *gbuf = buffer_new();
|
|
|
+
|
|
|
+ buffer_appends(gbuf, "#include <qirt.h>\n");
|
|
|
+
|
|
|
+ buffer_t *buf = buffer_new();
|
|
|
+
|
|
|
+ compile_node(gbuf, buf, ctx, lstk, n);
|
|
|
+
|
|
|
+ buffer_t *rbuf = buffer_new();
|
|
|
+
|
|
|
+ buffer_appendb(rbuf, gbuf);
|
|
|
+
|
|
|
+ buffer_appends(rbuf, "int main(int argc, char **argv) {\n");
|
|
|
+ buffer_appends(rbuf, "qi_state_t *state;\n");
|
|
|
+ buffer_appends(rbuf, "qi_state_init(&state);\n");
|
|
|
+ buffer_appendb(rbuf, buf);
|
|
|
+ buffer_appends(rbuf, "qi_old_scope(state);\n");
|
|
|
+ buffer_appends(rbuf, "qi_finalize();\n");
|
|
|
+ buffer_appends(rbuf, "return 0;\n");
|
|
|
+ buffer_appends(rbuf, "}\n");
|
|
|
+
|
|
|
+ return buffer_read(rbuf);
|
|
|
+}
|
|
|
+
|
|
|
+char *compile_file(char *filename, FILE *fd) {
|
|
|
+ buffer_t *buf = buffer_new();
|
|
|
+
|
|
|
+ for (;;) {
|
|
|
+ char line[512];
|
|
|
+
|
|
|
+ if (!fgets(line, sizeof(line), fd))
|
|
|
+ break;
|
|
|
+
|
|
|
+ buffer_appends(buf, line);
|
|
|
+ }
|
|
|
+
|
|
|
+ char *source = buffer_read(buf);
|
|
|
+
|
|
|
+ list_t *pair = list_new();
|
|
|
+ list_push(pair, filename);
|
|
|
+ list_push(pair, source);
|
|
|
+
|
|
|
+ list_push(FILES, pair);
|
|
|
+
|
|
|
+ char *out = compile(source);
|
|
|
+
|
|
|
+ list_pop(FILES);
|
|
|
+
|
|
|
+ return out;
|
|
|
+}
|
|
|
+
|
|
|
+int main(int argc, char **argv) {
|
|
|
+ FILES = list_new();
|
|
|
+
|
|
|
+ char *out = compile_file("<stdin>", stdin);
|
|
|
+
|
|
|
+ fwrite(out, sizeof(char), strlen(out), stdout);
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|