Initial commit.

1 files changed, 171 insertions, 0 deletions

diff --git a/py/emitcommon.c b/py/emitcommon.c
new file mode 100644
index 000000000..1fd8697c3
--- /dev/null
+++ b/py/emitcommon.c
@@ -0,0 +1,171 @@
+#include <unistd.h>
+#include <stdio.h>
+#include <stdint.h>
+#include <string.h>
+#include <assert.h>
+
+#include "misc.h"
+#include "lexer.h"
+#include "machine.h"
+#include "parse.h"
+#include "scope.h"
+#include "runtime.h"
+#include "emit.h"
+
+#define EMIT(fun, arg...) (emit_##fun(emit, ##arg))
+
+void emit_common_declare_global(pass_kind_t pass, scope_t *scope, qstr qstr) {
+    if (pass == PASS_1) {
+        if (scope->kind == SCOPE_MODULE) {
+            printf("SyntaxError?: can't declare global in outer code\n");
+            return;
+        }
+        bool added;
+        id_info_t *id_info = scope_find_or_add_id(scope, qstr, &added);
+        if (!added) {
+            printf("SyntaxError?: identifier already declared something\n");
+            return;
+        }
+        id_info->kind = ID_INFO_KIND_GLOBAL_EXPLICIT;
+
+        // if the id exists in the global scope, set its kind to EXPLICIT_GLOBAL
+        id_info = scope_find_global(scope, qstr);
+        if (id_info != NULL) {
+            id_info->kind = ID_INFO_KIND_GLOBAL_EXPLICIT;
+        }
+    }
+}
+
+void emit_common_declare_nonlocal(pass_kind_t pass, scope_t *scope, qstr qstr) {
+    if (pass == PASS_1) {
+        if (scope->kind == SCOPE_MODULE) {
+            printf("SyntaxError?: can't declare nonlocal in outer code\n");
+            return;
+        }
+        bool added;
+        id_info_t *id_info = scope_find_or_add_id(scope, qstr, &added);
+        if (!added) {
+            printf("SyntaxError?: identifier already declared something\n");
+            return;
+        }
+        id_info_t *id_info2 = scope_find_local_in_parent(scope, qstr);
+        if (id_info2 == NULL || !(id_info2->kind == ID_INFO_KIND_LOCAL || id_info2->kind == ID_INFO_KIND_CELL || id_info2->kind == ID_INFO_KIND_FREE)) {
+            printf("SyntaxError: no binding for nonlocal '%s' found\n", qstr_str(qstr));
+            return;
+        }
+        id_info->kind = ID_INFO_KIND_FREE;
+        scope_close_over_in_parents(scope, qstr);
+    }
+}
+
+void emit_common_load_id(pass_kind_t pass, scope_t *scope, qstr qstr___class__, emitter_t *emit, qstr qstr) {
+    id_info_t *id_info = NULL;
+    if (pass == PASS_1) {
+        // name adding/lookup
+        bool added;
+        id_info = scope_find_or_add_id(scope, qstr, &added);
+        if (added) {
+            if (strcmp(qstr_str(qstr), "AssertionError") == 0) {
+                id_info->kind = ID_INFO_KIND_GLOBAL_EXPLICIT;
+                // TODO how much of a hack is this?
+            } else if (strcmp(qstr_str(qstr), "super") == 0 && scope->kind == SCOPE_FUNCTION) {
+                // special case, super is a global, and also counts as use of __class__
+                id_info->kind = ID_INFO_KIND_GLOBAL_EXPLICIT;
+                id_info_t *id_info2 = scope_find_local_in_parent(scope, qstr___class__);
+                if (id_info2 != NULL) {
+                    id_info2 = scope_find_or_add_id(scope, qstr___class__, &added);
+                    if (added) {
+                        id_info2->kind = ID_INFO_KIND_FREE;
+                        scope_close_over_in_parents(scope, qstr___class__);
+                    }
+                }
+            } else {
+                id_info_t *id_info2 = scope_find_local_in_parent(scope, qstr);
+                if (id_info2 != NULL && (id_info2->kind == ID_INFO_KIND_LOCAL || id_info2->kind == ID_INFO_KIND_CELL || id_info2->kind == ID_INFO_KIND_FREE)) {
+                    id_info->kind = ID_INFO_KIND_FREE;
+                    scope_close_over_in_parents(scope, qstr);
+                } else {
+                    id_info->kind = ID_INFO_KIND_GLOBAL_IMPLICIT;
+                }
+            }
+        }
+    } else {
+        id_info = scope_find(scope, qstr);
+    }
+
+    assert(id_info != NULL); // TODO can this ever fail?
+
+    // call the emit backend with the correct code
+    if (id_info == NULL || id_info->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) {
+        EMIT(load_name, qstr);
+    } else if (id_info->kind == ID_INFO_KIND_GLOBAL_EXPLICIT) {
+        EMIT(load_global, qstr);
+    } else if (id_info->kind == ID_INFO_KIND_LOCAL) {
+        EMIT(load_fast, qstr, id_info->local_num);
+    } else if (id_info->kind == ID_INFO_KIND_CELL || id_info->kind == ID_INFO_KIND_FREE) {
+        EMIT(load_deref, qstr);
+    } else {
+        assert(0);
+    }
+}
+
+static id_info_t *get_id_for_modification(pass_kind_t pass, scope_t *scope, emitter_t *emit, qstr qstr) {
+    id_info_t *id_info = NULL;
+    if (pass == PASS_1) {
+        // name adding/lookup
+        bool added;
+        id_info = scope_find_or_add_id(scope, qstr, &added);
+        if (added) {
+            if (scope->kind == SCOPE_MODULE || scope->kind == SCOPE_CLASS) {
+                id_info->kind = ID_INFO_KIND_GLOBAL_IMPLICIT;
+            } else {
+                id_info->kind = ID_INFO_KIND_LOCAL;
+            }
+        } else if (scope->kind >= SCOPE_FUNCTION && scope->kind <= SCOPE_GEN_EXPR && id_info->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) {
+            // rebind as a local variable
+            id_info->kind = ID_INFO_KIND_LOCAL;
+        }
+    } else {
+        id_info = scope_find(scope, qstr);
+    }
+
+    assert(id_info != NULL); // TODO can this ever fail?
+
+    return id_info;
+}
+
+void emit_common_store_id(pass_kind_t pass, scope_t *scope, emitter_t *emit, qstr qstr) {
+    // create/get the id info
+    id_info_t *id = get_id_for_modification(pass, scope, emit, qstr);
+
+    // call the emit backend with the correct code
+    if (id == NULL || id->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) {
+        EMIT(store_name, qstr);
+    } else if (id->kind == ID_INFO_KIND_GLOBAL_EXPLICIT) {
+        EMIT(store_global, qstr);
+    } else if (id->kind == ID_INFO_KIND_LOCAL) {
+        EMIT(store_fast, qstr, id->local_num);
+    } else if (id->kind == ID_INFO_KIND_CELL || id->kind == ID_INFO_KIND_FREE) {
+        EMIT(store_deref, qstr);
+    } else {
+        assert(0);
+    }
+}
+
+void emit_common_delete_id(pass_kind_t pass, scope_t *scope, emitter_t *emit, qstr qstr) {
+    // create/get the id info
+    id_info_t *id = get_id_for_modification(pass, scope, emit, qstr);
+
+    // call the emit backend with the correct code
+    if (id == NULL || id->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) {
+        EMIT(delete_name, qstr);
+    } else if (id->kind == ID_INFO_KIND_GLOBAL_EXPLICIT) {
+        EMIT(delete_global, qstr);
+    } else if (id->kind == ID_INFO_KIND_LOCAL) {
+        EMIT(delete_fast, qstr, id->local_num);
+    } else if (id->kind == ID_INFO_KIND_CELL || id->kind == ID_INFO_KIND_FREE) {
+        EMIT(delete_deref, qstr);
+    } else {
+        assert(0);
+    }
+}