blob: d4926523998fa3518d7509fa05fe9396a86f8dd0 [file] [log] [blame]
Damien39cf3282013-10-05 23:18:38 +01001#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"
Damiend99b0522013-12-21 18:17:45 +00009#include "mpconfig.h"
Damien George55baff42014-01-21 21:40:13 +000010#include "qstr.h"
Damien39cf3282013-10-05 23:18:38 +010011#include "lexer.h"
Damien39cf3282013-10-05 23:18:38 +010012#include "parse.h"
Damien39cf3282013-10-05 23:18:38 +010013#include "scope.h"
Damiend99b0522013-12-21 18:17:45 +000014#include "runtime0.h"
Damien39cf3282013-10-05 23:18:38 +010015#include "emit.h"
16
17struct _emit_t {
Damien39cf3282013-10-05 23:18:38 +010018 scope_t *scope;
19};
20
Damien George35e2a4e2014-02-05 00:51:47 +000021emit_t *emit_pass1_new(void) {
Damien39cf3282013-10-05 23:18:38 +010022 emit_t *emit = m_new(emit_t, 1);
Damien39cf3282013-10-05 23:18:38 +010023 return emit;
24}
25
26void emit_pass1_free(emit_t *emit) {
Damien732407f2013-12-29 19:33:23 +000027 m_del_obj(emit_t, emit);
Damien39cf3282013-10-05 23:18:38 +010028}
29
Paul Sokolovsky520e2f52014-02-12 18:31:30 +020030STATIC void emit_pass1_dummy(emit_t *emit) {
Damien39cf3282013-10-05 23:18:38 +010031}
32
Paul Sokolovsky520e2f52014-02-12 18:31:30 +020033STATIC void emit_pass1_start_pass(emit_t *emit, pass_kind_t pass, scope_t *scope) {
Damien39cf3282013-10-05 23:18:38 +010034 assert(pass == PASS_1);
35 emit->scope = scope;
36}
37
Paul Sokolovsky520e2f52014-02-12 18:31:30 +020038STATIC void emit_pass1_end_pass(emit_t *emit) {
Damien39cf3282013-10-05 23:18:38 +010039}
40
Paul Sokolovsky520e2f52014-02-12 18:31:30 +020041STATIC void emit_pass1_load_id(emit_t *emit, qstr qstr) {
Damien39cf3282013-10-05 23:18:38 +010042 // name adding/lookup
43 bool added;
44 id_info_t *id = scope_find_or_add_id(emit->scope, qstr, &added);
45 if (added) {
Damien George35e2a4e2014-02-05 00:51:47 +000046#if MICROPY_EMIT_CPYTHON
47 if (qstr == MP_QSTR_super && emit->scope->kind == SCOPE_FUNCTION) {
Damien39cf3282013-10-05 23:18:38 +010048 // special case, super is a global, and also counts as use of __class__
49 id->kind = ID_INFO_KIND_GLOBAL_EXPLICIT;
Damien George35e2a4e2014-02-05 00:51:47 +000050 id_info_t *id2 = scope_find_local_in_parent(emit->scope, MP_QSTR___class__);
Damien39cf3282013-10-05 23:18:38 +010051 if (id2 != NULL) {
Damien George35e2a4e2014-02-05 00:51:47 +000052 id2 = scope_find_or_add_id(emit->scope, MP_QSTR___class__, &added);
Damien39cf3282013-10-05 23:18:38 +010053 if (added) {
54 id2->kind = ID_INFO_KIND_FREE;
Damien George35e2a4e2014-02-05 00:51:47 +000055 scope_close_over_in_parents(emit->scope, MP_QSTR___class__);
Damien39cf3282013-10-05 23:18:38 +010056 }
57 }
Damien George35e2a4e2014-02-05 00:51:47 +000058 } else
59#endif
60 {
Damien39cf3282013-10-05 23:18:38 +010061 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 Sokolovsky520e2f52014-02-12 18:31:30 +020072STATIC id_info_t *get_id_for_modification(scope_t *scope, qstr qstr) {
Damien39cf3282013-10-05 23:18:38 +010073 // 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 Sokolovsky520e2f52014-02-12 18:31:30 +020092STATIC void emit_pass1_store_id(emit_t *emit, qstr qstr) {
Damien39cf3282013-10-05 23:18:38 +010093 get_id_for_modification(emit->scope, qstr);
94}
95
Paul Sokolovsky520e2f52014-02-12 18:31:30 +020096STATIC void emit_pass1_delete_id(emit_t *emit, qstr qstr) {
Damien39cf3282013-10-05 23:18:38 +010097 get_id_for_modification(emit->scope, qstr);
98}
99
100const 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 George08335002014-01-18 23:24:36 +0000107 (void*)emit_pass1_dummy,
Damien39cf3282013-10-05 23:18:38 +0100108
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,
Damien39cf3282013-10-05 23:18:38 +0100189};