Oops: add emitpass1.c for new factorised emit code.

1 files changed, 196 insertions, 0 deletions

diff --git a/py/emitpass1.c b/py/emitpass1.c
new file mode 100644
index 0000000000..661a622bbc
--- /dev/null
+++ b/py/emitpass1.c
@@ -0,0 +1,196 @@
+#include <unistd.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <string.h>
+#include <assert.h>
+
+#include "misc.h"
+#include "lexer.h"
+#include "machine.h"
+#include "parse.h"
+#include "compile.h"
+#include "scope.h"
+#include "runtime.h"
+#include "emit.h"
+
+struct _emit_t {
+    qstr qstr___class__;
+    scope_t *scope;
+};
+
+emit_t *emit_pass1_new(qstr qstr___class__) {
+    emit_t *emit = m_new(emit_t, 1);
+    emit->qstr___class__ = qstr___class__;
+    return emit;
+}
+
+void emit_pass1_free(emit_t *emit) {
+    m_free(emit);
+}
+
+static void emit_pass1_dummy(emit_t *emit) {
+}
+
+static void emit_pass1_start_pass(emit_t *emit, pass_kind_t pass, scope_t *scope) {
+    assert(pass == PASS_1);
+    emit->scope = scope;
+}
+
+static void emit_pass1_end_pass(emit_t *emit) {
+}
+
+static void emit_pass1_load_id(emit_t *emit, qstr qstr) {
+    // name adding/lookup
+    bool added;
+    id_info_t *id = scope_find_or_add_id(emit->scope, qstr, &added);
+    if (added) {
+        if (strcmp(qstr_str(qstr), "AssertionError") == 0) {
+            id->kind = ID_INFO_KIND_GLOBAL_EXPLICIT;
+            // TODO how much of a hack is this?
+        } else if (strcmp(qstr_str(qstr), "super") == 0 && emit->scope->kind == SCOPE_FUNCTION) {
+            // special case, super is a global, and also counts as use of __class__
+            id->kind = ID_INFO_KIND_GLOBAL_EXPLICIT;
+            id_info_t *id2 = scope_find_local_in_parent(emit->scope, emit->qstr___class__);
+            if (id2 != NULL) {
+                id2 = scope_find_or_add_id(emit->scope, emit->qstr___class__, &added);
+                if (added) {
+                    id2->kind = ID_INFO_KIND_FREE;
+                    scope_close_over_in_parents(emit->scope, emit->qstr___class__);
+                }
+            }
+        } else {
+            id_info_t *id2 = scope_find_local_in_parent(emit->scope, qstr);
+            if (id2 != NULL && (id2->kind == ID_INFO_KIND_LOCAL || id2->kind == ID_INFO_KIND_CELL || id2->kind == ID_INFO_KIND_FREE)) {
+                id->kind = ID_INFO_KIND_FREE;
+                scope_close_over_in_parents(emit->scope, qstr);
+            } else {
+                id->kind = ID_INFO_KIND_GLOBAL_IMPLICIT;
+            }
+        }
+    }
+}
+
+static id_info_t *get_id_for_modification(scope_t *scope, qstr qstr) {
+    // name adding/lookup
+    bool added;
+    id_info_t *id = scope_find_or_add_id(scope, qstr, &added);
+    if (added) {
+        if (scope->kind == SCOPE_MODULE || scope->kind == SCOPE_CLASS) {
+            id->kind = ID_INFO_KIND_GLOBAL_IMPLICIT;
+        } else {
+            id->kind = ID_INFO_KIND_LOCAL;
+        }
+    } else if (scope->kind >= SCOPE_FUNCTION && scope->kind <= SCOPE_GEN_EXPR && id->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) {
+        // rebind as a local variable
+        id->kind = ID_INFO_KIND_LOCAL;
+    }
+
+    assert(id != NULL); // TODO can this ever fail?
+
+    return id;
+}
+
+static void emit_pass1_store_id(emit_t *emit, qstr qstr) {
+    get_id_for_modification(emit->scope, qstr);
+}
+
+static void emit_pass1_delete_id(emit_t *emit, qstr qstr) {
+    get_id_for_modification(emit->scope, qstr);
+}
+
+const emit_method_table_t emit_pass1_method_table = {
+    (void*)emit_pass1_dummy,
+    emit_pass1_start_pass,
+    emit_pass1_end_pass,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+
+    emit_pass1_load_id,
+    emit_pass1_store_id,
+    emit_pass1_delete_id,
+
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+    (void*)emit_pass1_dummy,
+};