Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 1 | #include <unistd.h> |
| 2 | #include <stdlib.h> |
| 3 | #include <stdint.h> |
| 4 | #include <stdio.h> |
| 5 | #include <string.h> |
| 6 | #include <assert.h> |
| 7 | |
| 8 | #include "misc.h" |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 9 | #include "mpconfig.h" |
Damien George | 55baff4 | 2014-01-21 21:40:13 +0000 | [diff] [blame] | 10 | #include "qstr.h" |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 11 | #include "lexer.h" |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 12 | #include "parse.h" |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 13 | #include "scope.h" |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 14 | #include "runtime0.h" |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 15 | #include "emit.h" |
| 16 | |
| 17 | struct _emit_t { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 18 | scope_t *scope; |
| 19 | }; |
| 20 | |
Damien George | 35e2a4e | 2014-02-05 00:51:47 +0000 | [diff] [blame] | 21 | emit_t *emit_pass1_new(void) { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 22 | emit_t *emit = m_new(emit_t, 1); |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 23 | return emit; |
| 24 | } |
| 25 | |
| 26 | void emit_pass1_free(emit_t *emit) { |
Damien | 732407f | 2013-12-29 19:33:23 +0000 | [diff] [blame] | 27 | m_del_obj(emit_t, emit); |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 28 | } |
| 29 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 30 | STATIC void emit_pass1_dummy(emit_t *emit) { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 31 | } |
| 32 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 33 | STATIC void emit_pass1_start_pass(emit_t *emit, pass_kind_t pass, scope_t *scope) { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 34 | assert(pass == PASS_1); |
| 35 | emit->scope = scope; |
| 36 | } |
| 37 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 38 | STATIC void emit_pass1_end_pass(emit_t *emit) { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 39 | } |
| 40 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 41 | STATIC void emit_pass1_load_id(emit_t *emit, qstr qstr) { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 42 | // name adding/lookup |
| 43 | bool added; |
| 44 | id_info_t *id = scope_find_or_add_id(emit->scope, qstr, &added); |
| 45 | if (added) { |
Damien George | 35e2a4e | 2014-02-05 00:51:47 +0000 | [diff] [blame] | 46 | #if MICROPY_EMIT_CPYTHON |
| 47 | if (qstr == MP_QSTR_super && emit->scope->kind == SCOPE_FUNCTION) { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 48 | // special case, super is a global, and also counts as use of __class__ |
| 49 | id->kind = ID_INFO_KIND_GLOBAL_EXPLICIT; |
Damien George | 35e2a4e | 2014-02-05 00:51:47 +0000 | [diff] [blame] | 50 | id_info_t *id2 = scope_find_local_in_parent(emit->scope, MP_QSTR___class__); |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 51 | if (id2 != NULL) { |
Damien George | 35e2a4e | 2014-02-05 00:51:47 +0000 | [diff] [blame] | 52 | id2 = scope_find_or_add_id(emit->scope, MP_QSTR___class__, &added); |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 53 | if (added) { |
| 54 | id2->kind = ID_INFO_KIND_FREE; |
Damien George | 35e2a4e | 2014-02-05 00:51:47 +0000 | [diff] [blame] | 55 | scope_close_over_in_parents(emit->scope, MP_QSTR___class__); |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 56 | } |
| 57 | } |
Damien George | 35e2a4e | 2014-02-05 00:51:47 +0000 | [diff] [blame] | 58 | } else |
| 59 | #endif |
| 60 | { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 61 | id_info_t *id2 = scope_find_local_in_parent(emit->scope, qstr); |
| 62 | if (id2 != NULL && (id2->kind == ID_INFO_KIND_LOCAL || id2->kind == ID_INFO_KIND_CELL || id2->kind == ID_INFO_KIND_FREE)) { |
| 63 | id->kind = ID_INFO_KIND_FREE; |
| 64 | scope_close_over_in_parents(emit->scope, qstr); |
| 65 | } else { |
| 66 | id->kind = ID_INFO_KIND_GLOBAL_IMPLICIT; |
| 67 | } |
| 68 | } |
| 69 | } |
| 70 | } |
| 71 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 72 | STATIC id_info_t *get_id_for_modification(scope_t *scope, qstr qstr) { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 73 | // name adding/lookup |
| 74 | bool added; |
| 75 | id_info_t *id = scope_find_or_add_id(scope, qstr, &added); |
| 76 | if (added) { |
| 77 | if (scope->kind == SCOPE_MODULE || scope->kind == SCOPE_CLASS) { |
| 78 | id->kind = ID_INFO_KIND_GLOBAL_IMPLICIT; |
| 79 | } else { |
| 80 | id->kind = ID_INFO_KIND_LOCAL; |
| 81 | } |
| 82 | } else if (scope->kind >= SCOPE_FUNCTION && scope->kind <= SCOPE_GEN_EXPR && id->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) { |
| 83 | // rebind as a local variable |
| 84 | id->kind = ID_INFO_KIND_LOCAL; |
| 85 | } |
| 86 | |
| 87 | assert(id != NULL); // TODO can this ever fail? |
| 88 | |
| 89 | return id; |
| 90 | } |
| 91 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 92 | STATIC void emit_pass1_store_id(emit_t *emit, qstr qstr) { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 93 | get_id_for_modification(emit->scope, qstr); |
| 94 | } |
| 95 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 96 | STATIC void emit_pass1_delete_id(emit_t *emit, qstr qstr) { |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 97 | get_id_for_modification(emit->scope, qstr); |
| 98 | } |
| 99 | |
| 100 | const emit_method_table_t emit_pass1_method_table = { |
| 101 | (void*)emit_pass1_dummy, |
| 102 | emit_pass1_start_pass, |
| 103 | emit_pass1_end_pass, |
| 104 | (void*)emit_pass1_dummy, |
| 105 | (void*)emit_pass1_dummy, |
| 106 | (void*)emit_pass1_dummy, |
Damien George | 0833500 | 2014-01-18 23:24:36 +0000 | [diff] [blame] | 107 | (void*)emit_pass1_dummy, |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 108 | |
| 109 | emit_pass1_load_id, |
| 110 | emit_pass1_store_id, |
| 111 | emit_pass1_delete_id, |
| 112 | |
| 113 | (void*)emit_pass1_dummy, |
| 114 | (void*)emit_pass1_dummy, |
| 115 | (void*)emit_pass1_dummy, |
| 116 | (void*)emit_pass1_dummy, |
| 117 | (void*)emit_pass1_dummy, |
| 118 | (void*)emit_pass1_dummy, |
| 119 | (void*)emit_pass1_dummy, |
| 120 | (void*)emit_pass1_dummy, |
| 121 | (void*)emit_pass1_dummy, |
| 122 | (void*)emit_pass1_dummy, |
| 123 | (void*)emit_pass1_dummy, |
| 124 | (void*)emit_pass1_dummy, |
| 125 | (void*)emit_pass1_dummy, |
| 126 | (void*)emit_pass1_dummy, |
| 127 | (void*)emit_pass1_dummy, |
| 128 | (void*)emit_pass1_dummy, |
| 129 | (void*)emit_pass1_dummy, |
| 130 | (void*)emit_pass1_dummy, |
| 131 | (void*)emit_pass1_dummy, |
| 132 | (void*)emit_pass1_dummy, |
| 133 | (void*)emit_pass1_dummy, |
| 134 | (void*)emit_pass1_dummy, |
| 135 | (void*)emit_pass1_dummy, |
| 136 | (void*)emit_pass1_dummy, |
| 137 | (void*)emit_pass1_dummy, |
| 138 | (void*)emit_pass1_dummy, |
| 139 | (void*)emit_pass1_dummy, |
| 140 | (void*)emit_pass1_dummy, |
| 141 | (void*)emit_pass1_dummy, |
| 142 | (void*)emit_pass1_dummy, |
| 143 | (void*)emit_pass1_dummy, |
| 144 | (void*)emit_pass1_dummy, |
| 145 | (void*)emit_pass1_dummy, |
| 146 | (void*)emit_pass1_dummy, |
| 147 | (void*)emit_pass1_dummy, |
| 148 | (void*)emit_pass1_dummy, |
| 149 | (void*)emit_pass1_dummy, |
| 150 | (void*)emit_pass1_dummy, |
| 151 | (void*)emit_pass1_dummy, |
| 152 | (void*)emit_pass1_dummy, |
| 153 | (void*)emit_pass1_dummy, |
| 154 | (void*)emit_pass1_dummy, |
| 155 | (void*)emit_pass1_dummy, |
| 156 | (void*)emit_pass1_dummy, |
| 157 | (void*)emit_pass1_dummy, |
| 158 | (void*)emit_pass1_dummy, |
| 159 | (void*)emit_pass1_dummy, |
| 160 | (void*)emit_pass1_dummy, |
| 161 | (void*)emit_pass1_dummy, |
| 162 | (void*)emit_pass1_dummy, |
| 163 | (void*)emit_pass1_dummy, |
| 164 | (void*)emit_pass1_dummy, |
| 165 | (void*)emit_pass1_dummy, |
| 166 | (void*)emit_pass1_dummy, |
| 167 | (void*)emit_pass1_dummy, |
| 168 | (void*)emit_pass1_dummy, |
| 169 | (void*)emit_pass1_dummy, |
| 170 | (void*)emit_pass1_dummy, |
| 171 | (void*)emit_pass1_dummy, |
| 172 | (void*)emit_pass1_dummy, |
| 173 | (void*)emit_pass1_dummy, |
| 174 | (void*)emit_pass1_dummy, |
| 175 | (void*)emit_pass1_dummy, |
| 176 | (void*)emit_pass1_dummy, |
| 177 | (void*)emit_pass1_dummy, |
| 178 | (void*)emit_pass1_dummy, |
| 179 | (void*)emit_pass1_dummy, |
| 180 | (void*)emit_pass1_dummy, |
| 181 | (void*)emit_pass1_dummy, |
| 182 | (void*)emit_pass1_dummy, |
| 183 | (void*)emit_pass1_dummy, |
| 184 | (void*)emit_pass1_dummy, |
| 185 | (void*)emit_pass1_dummy, |
| 186 | (void*)emit_pass1_dummy, |
| 187 | (void*)emit_pass1_dummy, |
| 188 | (void*)emit_pass1_dummy, |
Damien | 39cf328 | 2013-10-05 23:18:38 +0100 | [diff] [blame] | 189 | }; |