Skip to content

Commit 39cf328

Browse files
committed
Oops: add emitpass1.c for new factorised emit code.
1 parent 826005c commit 39cf328

2 files changed

Lines changed: 197 additions & 0 deletions

File tree

py/.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
11
*.o
22
py
33
*.py
4+
.*.swp

py/emitpass1.c

Lines changed: 196 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,196 @@
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"
9+
#include "lexer.h"
10+
#include "machine.h"
11+
#include "parse.h"
12+
#include "compile.h"
13+
#include "scope.h"
14+
#include "runtime.h"
15+
#include "emit.h"
16+
17+
struct _emit_t {
18+
qstr qstr___class__;
19+
scope_t *scope;
20+
};
21+
22+
emit_t *emit_pass1_new(qstr qstr___class__) {
23+
emit_t *emit = m_new(emit_t, 1);
24+
emit->qstr___class__ = qstr___class__;
25+
return emit;
26+
}
27+
28+
void emit_pass1_free(emit_t *emit) {
29+
m_free(emit);
30+
}
31+
32+
static void emit_pass1_dummy(emit_t *emit) {
33+
}
34+
35+
static void emit_pass1_start_pass(emit_t *emit, pass_kind_t pass, scope_t *scope) {
36+
assert(pass == PASS_1);
37+
emit->scope = scope;
38+
}
39+
40+
static void emit_pass1_end_pass(emit_t *emit) {
41+
}
42+
43+
static void emit_pass1_load_id(emit_t *emit, qstr qstr) {
44+
// name adding/lookup
45+
bool added;
46+
id_info_t *id = scope_find_or_add_id(emit->scope, qstr, &added);
47+
if (added) {
48+
if (strcmp(qstr_str(qstr), "AssertionError") == 0) {
49+
id->kind = ID_INFO_KIND_GLOBAL_EXPLICIT;
50+
// TODO how much of a hack is this?
51+
} else if (strcmp(qstr_str(qstr), "super") == 0 && emit->scope->kind == SCOPE_FUNCTION) {
52+
// special case, super is a global, and also counts as use of __class__
53+
id->kind = ID_INFO_KIND_GLOBAL_EXPLICIT;
54+
id_info_t *id2 = scope_find_local_in_parent(emit->scope, emit->qstr___class__);
55+
if (id2 != NULL) {
56+
id2 = scope_find_or_add_id(emit->scope, emit->qstr___class__, &added);
57+
if (added) {
58+
id2->kind = ID_INFO_KIND_FREE;
59+
scope_close_over_in_parents(emit->scope, emit->qstr___class__);
60+
}
61+
}
62+
} else {
63+
id_info_t *id2 = scope_find_local_in_parent(emit->scope, qstr);
64+
if (id2 != NULL && (id2->kind == ID_INFO_KIND_LOCAL || id2->kind == ID_INFO_KIND_CELL || id2->kind == ID_INFO_KIND_FREE)) {
65+
id->kind = ID_INFO_KIND_FREE;
66+
scope_close_over_in_parents(emit->scope, qstr);
67+
} else {
68+
id->kind = ID_INFO_KIND_GLOBAL_IMPLICIT;
69+
}
70+
}
71+
}
72+
}
73+
74+
static id_info_t *get_id_for_modification(scope_t *scope, qstr qstr) {
75+
// name adding/lookup
76+
bool added;
77+
id_info_t *id = scope_find_or_add_id(scope, qstr, &added);
78+
if (added) {
79+
if (scope->kind == SCOPE_MODULE || scope->kind == SCOPE_CLASS) {
80+
id->kind = ID_INFO_KIND_GLOBAL_IMPLICIT;
81+
} else {
82+
id->kind = ID_INFO_KIND_LOCAL;
83+
}
84+
} else if (scope->kind >= SCOPE_FUNCTION && scope->kind <= SCOPE_GEN_EXPR && id->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) {
85+
// rebind as a local variable
86+
id->kind = ID_INFO_KIND_LOCAL;
87+
}
88+
89+
assert(id != NULL); // TODO can this ever fail?
90+
91+
return id;
92+
}
93+
94+
static void emit_pass1_store_id(emit_t *emit, qstr qstr) {
95+
get_id_for_modification(emit->scope, qstr);
96+
}
97+
98+
static void emit_pass1_delete_id(emit_t *emit, qstr qstr) {
99+
get_id_for_modification(emit->scope, qstr);
100+
}
101+
102+
const emit_method_table_t emit_pass1_method_table = {
103+
(void*)emit_pass1_dummy,
104+
emit_pass1_start_pass,
105+
emit_pass1_end_pass,
106+
(void*)emit_pass1_dummy,
107+
(void*)emit_pass1_dummy,
108+
(void*)emit_pass1_dummy,
109+
110+
emit_pass1_load_id,
111+
emit_pass1_store_id,
112+
emit_pass1_delete_id,
113+
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,
189+
(void*)emit_pass1_dummy,
190+
(void*)emit_pass1_dummy,
191+
(void*)emit_pass1_dummy,
192+
(void*)emit_pass1_dummy,
193+
(void*)emit_pass1_dummy,
194+
(void*)emit_pass1_dummy,
195+
(void*)emit_pass1_dummy,
196+
};

0 commit comments

Comments
 (0)