blob: 743c3211191bc7dbe9a840c7780bda909ee1123a [file] [log] [blame]
Damien13ed3a62013-10-08 09:05:10 +01001// Essentially normal Python has 1 type: Python objects
2// Viper has more than 1 type, and is just a more complicated (a superset of) Python.
3// If you declare everything in Viper as a Python object (ie omit type decls) then
4// it should in principle be exactly the same as Python native.
5// Having types means having more opcodes, like binary_op_nat_nat, binary_op_nat_obj etc.
6// In practice we won't have a VM but rather do this in asm which is actually very minimal.
7
8// Because it breaks strict Python equivalence it should be a completely separate
9// decorator. It breaks equivalence because overflow on integers wraps around.
10// It shouldn't break equivalence if you don't use the new types, but since the
11// type decls might be used in normal Python for other reasons, it's probably safest,
12// cleanest and clearest to make it a separate decorator.
13
14// Actually, it does break equivalence because integers default to native integers,
15// not Python objects.
16
17// for x in l[0:8]: can be compiled into a native loop if l has pointer type
18
xbeefe34222014-03-16 00:14:26 -070019#include <stdbool.h>
Damien13ed3a62013-10-08 09:05:10 +010020#include <stdint.h>
21#include <stdio.h>
22#include <string.h>
23#include <assert.h>
24
25#include "misc.h"
Damiend99b0522013-12-21 18:17:45 +000026#include "mpconfig.h"
Damien George55baff42014-01-21 21:40:13 +000027#include "qstr.h"
Damien13ed3a62013-10-08 09:05:10 +010028#include "lexer.h"
Damien13ed3a62013-10-08 09:05:10 +010029#include "parse.h"
30#include "scope.h"
Damiend99b0522013-12-21 18:17:45 +000031#include "runtime0.h"
Damien13ed3a62013-10-08 09:05:10 +010032#include "emit.h"
Damien George2326d522014-03-27 23:26:35 +000033#include "emitglue.h"
Damiend99b0522013-12-21 18:17:45 +000034#include "obj.h"
35#include "runtime.h"
Damien13ed3a62013-10-08 09:05:10 +010036
Damien13ed3a62013-10-08 09:05:10 +010037// wrapper around everything in this file
Damien Georgee67ed5d2014-01-04 13:55:24 +000038#if (MICROPY_EMIT_X64 && N_X64) || (MICROPY_EMIT_THUMB && N_THUMB)
Damien13ed3a62013-10-08 09:05:10 +010039
Damien3ef4abb2013-10-12 16:53:13 +010040#if N_X64
Damien13ed3a62013-10-08 09:05:10 +010041
42// x64 specific stuff
43
44#include "asmx64.h"
45
46#define REG_LOCAL_1 (REG_RBX)
47#define REG_LOCAL_NUM (1)
48
49#define EXPORT_FUN(name) emit_native_x64_##name
50
51#define REG_TEMP0 (REG_RAX)
52#define REG_TEMP1 (REG_RDI)
53#define REG_TEMP2 (REG_RSI)
54#define ASM_MOV_REG_TO_LOCAL(reg, local_num) asm_x64_mov_r64_to_local(emit->as, (reg), (local_num))
55#define ASM_MOV_IMM_TO_REG(imm, reg) asm_x64_mov_i64_to_r64_optimised(emit->as, (imm), (reg))
Damieneb19efb2013-10-10 22:06:54 +010056#define ASM_MOV_IMM_TO_LOCAL_USING(imm, local_num, reg_temp) do { asm_x64_mov_i64_to_r64_optimised(emit->as, (imm), (reg_temp)); asm_x64_mov_r64_to_local(emit->as, (reg_temp), (local_num)); } while (false)
Damien13ed3a62013-10-08 09:05:10 +010057#define ASM_MOV_LOCAL_TO_REG(local_num, reg) asm_x64_mov_local_to_r64(emit->as, (local_num), (reg))
58#define ASM_MOV_REG_TO_REG(reg_src, reg_dest) asm_x64_mov_r64_to_r64(emit->as, (reg_src), (reg_dest))
59#define ASM_MOV_LOCAL_ADDR_TO_REG(local_num, reg) asm_x64_mov_local_addr_to_r64(emit->as, (local_num), (reg))
60
Damien3ef4abb2013-10-12 16:53:13 +010061#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +010062
63// thumb specific stuff
64
65#include "asmthumb.h"
66
67#define REG_LOCAL_1 (REG_R4)
68#define REG_LOCAL_2 (REG_R5)
69#define REG_LOCAL_3 (REG_R6)
70#define REG_LOCAL_NUM (3)
71
72#define EXPORT_FUN(name) emit_native_thumb_##name
73
74#define REG_TEMP0 (REG_R0)
75#define REG_TEMP1 (REG_R1)
76#define REG_TEMP2 (REG_R2)
77#define ASM_MOV_REG_TO_LOCAL(reg, local_num) asm_thumb_mov_local_reg(emit->as, (local_num), (reg))
78#define ASM_MOV_IMM_TO_REG(imm, reg) asm_thumb_mov_reg_i32_optimised(emit->as, (reg), (imm))
Damieneb19efb2013-10-10 22:06:54 +010079#define ASM_MOV_IMM_TO_LOCAL_USING(imm, local_num, reg_temp) do { asm_thumb_mov_reg_i32_optimised(emit->as, (reg_temp), (imm)); asm_thumb_mov_local_reg(emit->as, (local_num), (reg_temp)); } while (false)
Damien13ed3a62013-10-08 09:05:10 +010080#define ASM_MOV_LOCAL_TO_REG(local_num, reg) asm_thumb_mov_reg_local(emit->as, (reg), (local_num))
81#define ASM_MOV_REG_TO_REG(reg_src, reg_dest) asm_thumb_mov_reg_reg(emit->as, (reg_dest), (reg_src))
82#define ASM_MOV_LOCAL_ADDR_TO_REG(local_num, reg) asm_thumb_mov_reg_local_addr(emit->as, (reg), (local_num))
83
84#endif
85
86typedef enum {
Damienff8ed772013-10-08 22:18:32 +010087 STACK_VALUE,
88 STACK_REG,
89 STACK_IMM,
90} stack_info_kind_t;
Damien13ed3a62013-10-08 09:05:10 +010091
92typedef enum {
93 VTYPE_UNBOUND,
94 VTYPE_PYOBJ,
95 VTYPE_BOOL,
96 VTYPE_INT,
97 VTYPE_PTR,
98 VTYPE_PTR_NONE,
99 VTYPE_BUILTIN_V_INT,
100} vtype_kind_t;
101
Damienff8ed772013-10-08 22:18:32 +0100102typedef struct _stack_info_t {
103 vtype_kind_t vtype;
104 stack_info_kind_t kind;
105 union {
106 int u_reg;
107 machine_int_t u_imm;
108 };
109} stack_info_t;
110
Damien13ed3a62013-10-08 09:05:10 +0100111struct _emit_t {
112 int pass;
113
114 bool do_viper_types;
Damienff8ed772013-10-08 22:18:32 +0100115
116 int local_vtype_alloc;
Damien13ed3a62013-10-08 09:05:10 +0100117 vtype_kind_t *local_vtype;
Damienff8ed772013-10-08 22:18:32 +0100118
119 int stack_info_alloc;
120 stack_info_t *stack_info;
121
Damien13ed3a62013-10-08 09:05:10 +0100122 int stack_start;
123 int stack_size;
124
125 bool last_emit_was_return_value;
126
Damien13ed3a62013-10-08 09:05:10 +0100127 scope_t *scope;
128
Damien3ef4abb2013-10-12 16:53:13 +0100129#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100130 asm_x64_t *as;
Damien3ef4abb2013-10-12 16:53:13 +0100131#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100132 asm_thumb_t *as;
133#endif
134};
135
136emit_t *EXPORT_FUN(new)(uint max_num_labels) {
137 emit_t *emit = m_new(emit_t, 1);
138 emit->do_viper_types = false;
Damienff8ed772013-10-08 22:18:32 +0100139 emit->local_vtype = NULL;
140 emit->stack_info = NULL;
Damien3ef4abb2013-10-12 16:53:13 +0100141#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100142 emit->as = asm_x64_new(max_num_labels);
Damien3ef4abb2013-10-12 16:53:13 +0100143#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100144 emit->as = asm_thumb_new(max_num_labels);
145#endif
146 return emit;
147}
148
Damien George41d02b62014-01-24 22:42:28 +0000149void EXPORT_FUN(free)(emit_t *emit) {
Paul Sokolovskyf46d87a2014-01-24 16:20:11 +0200150#if N_X64
151 asm_x64_free(emit->as, false);
152#elif N_THUMB
153 asm_thumb_free(emit->as, false);
154#endif
155 m_del_obj(emit_t, emit);
156}
157
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200158STATIC void emit_native_set_viper_types(emit_t *emit, bool do_viper_types) {
Damien13ed3a62013-10-08 09:05:10 +0100159 emit->do_viper_types = do_viper_types;
160}
161
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200162STATIC void emit_native_start_pass(emit_t *emit, pass_kind_t pass, scope_t *scope) {
Damien13ed3a62013-10-08 09:05:10 +0100163 emit->pass = pass;
164 emit->stack_start = 0;
165 emit->stack_size = 0;
166 emit->last_emit_was_return_value = false;
Damien13ed3a62013-10-08 09:05:10 +0100167 emit->scope = scope;
168
Damienff8ed772013-10-08 22:18:32 +0100169 if (emit->local_vtype == NULL) {
170 emit->local_vtype_alloc = scope->num_locals + 20; // XXX should be maximum over all scopes
171 emit->local_vtype = m_new(vtype_kind_t, emit->local_vtype_alloc);
172 }
173 if (emit->stack_info == NULL) {
174 emit->stack_info_alloc = scope->stack_size + 50; // XXX don't know stack size on entry, should be maximum over all scopes
175 emit->stack_info = m_new(stack_info_t, emit->stack_info_alloc);
Damien13ed3a62013-10-08 09:05:10 +0100176 }
177
178 if (emit->do_viper_types) {
Damien13ed3a62013-10-08 09:05:10 +0100179 // TODO set types of arguments based on type signature
Damienff8ed772013-10-08 22:18:32 +0100180 for (int i = 0; i < emit->local_vtype_alloc; i++) {
181 emit->local_vtype[i] = VTYPE_UNBOUND;
182 }
183 for (int i = 0; i < emit->stack_info_alloc; i++) {
184 emit->stack_info[i].kind = STACK_VALUE;
185 emit->stack_info[i].vtype = VTYPE_UNBOUND;
186 }
Damien13ed3a62013-10-08 09:05:10 +0100187 } else {
Damienff8ed772013-10-08 22:18:32 +0100188 for (int i = 0; i < emit->local_vtype_alloc; i++) {
189 emit->local_vtype[i] = VTYPE_PYOBJ;
190 }
191 for (int i = 0; i < emit->stack_info_alloc; i++) {
192 emit->stack_info[i].kind = STACK_VALUE;
193 emit->stack_info[i].vtype = VTYPE_PYOBJ;
Damien13ed3a62013-10-08 09:05:10 +0100194 }
195 }
196
Damien3ef4abb2013-10-12 16:53:13 +0100197#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100198 asm_x64_start_pass(emit->as, pass);
Damien3ef4abb2013-10-12 16:53:13 +0100199#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100200 asm_thumb_start_pass(emit->as, pass);
201#endif
202
203 // entry to function
204 int num_locals = 0;
205 if (pass > PASS_1) {
206 num_locals = scope->num_locals - REG_LOCAL_NUM;
207 if (num_locals < 0) {
208 num_locals = 0;
209 }
210 emit->stack_start = num_locals;
211 num_locals += scope->stack_size;
212 }
Damien Georgecd82e022014-02-02 13:11:48 +0000213 if (pass == PASS_2) {
214 // XXX big hack to make sure we have some locals in PASS_2
215 // this is so that on PASS_2 the code emitted in x64 has the right size
216 num_locals += 2;
217 }
Damien3ef4abb2013-10-12 16:53:13 +0100218#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100219 asm_x64_entry(emit->as, num_locals);
Damien3ef4abb2013-10-12 16:53:13 +0100220#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100221 asm_thumb_entry(emit->as, num_locals);
222#endif
223
224 // initialise locals from parameters
Damien3ef4abb2013-10-12 16:53:13 +0100225#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100226 for (int i = 0; i < scope->num_params; i++) {
227 if (i == 0) {
228 asm_x64_mov_r64_to_r64(emit->as, REG_ARG_1, REG_LOCAL_1);
229 } else if (i == 1) {
230 asm_x64_mov_r64_to_local(emit->as, REG_ARG_2, i - 1);
231 } else if (i == 2) {
232 asm_x64_mov_r64_to_local(emit->as, REG_ARG_3, i - 1);
233 } else {
234 // TODO not implemented
235 assert(0);
236 }
237 }
Damien3ef4abb2013-10-12 16:53:13 +0100238#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100239 for (int i = 0; i < scope->num_params; i++) {
240 if (i == 0) {
241 asm_thumb_mov_reg_reg(emit->as, REG_LOCAL_1, REG_ARG_1);
242 } else if (i == 1) {
243 asm_thumb_mov_reg_reg(emit->as, REG_LOCAL_2, REG_ARG_2);
244 } else if (i == 2) {
245 asm_thumb_mov_reg_reg(emit->as, REG_LOCAL_3, REG_ARG_3);
246 } else if (i == 3) {
247 asm_thumb_mov_local_reg(emit->as, i - REG_LOCAL_NUM, REG_ARG_4);
248 } else {
249 // TODO not implemented
250 assert(0);
251 }
252 }
253
254 asm_thumb_mov_reg_i32(emit->as, REG_R7, (machine_uint_t)rt_fun_table);
255#endif
256}
257
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200258STATIC void emit_native_end_pass(emit_t *emit) {
Damien3ef4abb2013-10-12 16:53:13 +0100259#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100260 if (!emit->last_emit_was_return_value) {
261 asm_x64_exit(emit->as);
262 }
263 asm_x64_end_pass(emit->as);
Damien3ef4abb2013-10-12 16:53:13 +0100264#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100265 if (!emit->last_emit_was_return_value) {
266 asm_thumb_exit(emit->as);
267 }
268 asm_thumb_end_pass(emit->as);
269#endif
270
271 // check stack is back to zero size
272 if (emit->stack_size != 0) {
273 printf("ERROR: stack size not back to zero; got %d\n", emit->stack_size);
274 }
275
276 if (emit->pass == PASS_3) {
Damien3ef4abb2013-10-12 16:53:13 +0100277#if N_X64
Damiend99b0522013-12-21 18:17:45 +0000278 void *f = asm_x64_get_code(emit->as);
Damien George2326d522014-03-27 23:26:35 +0000279 mp_emit_glue_assign_native_code(emit->scope->unique_code_id, f, asm_x64_get_code_size(emit->as), emit->scope->num_params);
Damien3ef4abb2013-10-12 16:53:13 +0100280#elif N_THUMB
Damiend99b0522013-12-21 18:17:45 +0000281 void *f = asm_thumb_get_code(emit->as);
Damien George2326d522014-03-27 23:26:35 +0000282 mp_emit_glue_assign_native_code(emit->scope->unique_code_id, f, asm_thumb_get_code_size(emit->as), emit->scope->num_params);
Damien13ed3a62013-10-08 09:05:10 +0100283#endif
284 }
285}
286
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200287STATIC bool emit_native_last_emit_was_return_value(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100288 return emit->last_emit_was_return_value;
289}
290
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200291STATIC int emit_native_get_stack_size(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100292 return emit->stack_size;
293}
294
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200295STATIC void emit_native_set_stack_size(emit_t *emit, int size) {
Damien13ed3a62013-10-08 09:05:10 +0100296 emit->stack_size = size;
297}
298
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200299STATIC void emit_native_set_source_line(emit_t *emit, int source_line) {
Damien George08335002014-01-18 23:24:36 +0000300}
301
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200302STATIC void adjust_stack(emit_t *emit, int stack_size_delta) {
Damien13ed3a62013-10-08 09:05:10 +0100303 emit->stack_size += stack_size_delta;
304 assert(emit->stack_size >= 0);
305 if (emit->pass > PASS_1 && emit->stack_size > emit->scope->stack_size) {
306 emit->scope->stack_size = emit->stack_size;
307 }
308}
309
Damienff8ed772013-10-08 22:18:32 +0100310/*
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200311STATIC void emit_pre_raw(emit_t *emit, int stack_size_delta) {
Damien13ed3a62013-10-08 09:05:10 +0100312 adjust_stack(emit, stack_size_delta);
313 emit->last_emit_was_return_value = false;
314}
Damienff8ed772013-10-08 22:18:32 +0100315*/
Damien13ed3a62013-10-08 09:05:10 +0100316
Damienff8ed772013-10-08 22:18:32 +0100317// this must be called at start of emit functions
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200318// TODO: module-polymorphic function (read: name clash if made global)
Damien13ed3a62013-10-08 09:05:10 +0100319static void emit_pre(emit_t *emit) {
Damienff8ed772013-10-08 22:18:32 +0100320 emit->last_emit_was_return_value = false;
321 // settle the stack
322 /*
323 if (regs_needed != 0) {
324 for (int i = 0; i < emit->stack_size; i++) {
325 switch (emit->stack_info[i].kind) {
326 case STACK_VALUE:
327 break;
328
329 case STACK_REG:
330 // TODO only push reg if in regs_needed
331 emit->stack_info[i].kind = STACK_VALUE;
332 ASM_MOV_REG_TO_LOCAL(emit->stack_info[i].u_reg, emit->stack_start + i);
333 break;
334
335 case STACK_IMM:
336 // don't think we ever need to push imms for settling
337 //ASM_MOV_IMM_TO_LOCAL(emit->last_imm, emit->stack_start + i);
338 break;
339 }
340 }
341 }
342 */
Damien13ed3a62013-10-08 09:05:10 +0100343}
344
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200345STATIC vtype_kind_t peek_vtype(emit_t *emit) {
Damienff8ed772013-10-08 22:18:32 +0100346 return emit->stack_info[emit->stack_size - 1].vtype;
347}
Damien13ed3a62013-10-08 09:05:10 +0100348
Damiend2755ec2013-10-16 23:58:48 +0100349// pos=1 is TOS, pos=2 is next, etc
350// use pos=0 for no skipping
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200351STATIC void need_reg_single(emit_t *emit, int reg_needed, int skip_stack_pos) {
Damiend2755ec2013-10-16 23:58:48 +0100352 skip_stack_pos = emit->stack_size - skip_stack_pos;
Damienff8ed772013-10-08 22:18:32 +0100353 for (int i = 0; i < emit->stack_size; i++) {
Damiend2755ec2013-10-16 23:58:48 +0100354 if (i != skip_stack_pos) {
355 stack_info_t *si = &emit->stack_info[i];
356 if (si->kind == STACK_REG && si->u_reg == reg_needed) {
357 si->kind = STACK_VALUE;
358 ASM_MOV_REG_TO_LOCAL(si->u_reg, emit->stack_start + i);
359 }
Damienff8ed772013-10-08 22:18:32 +0100360 }
361 }
362}
Damien13ed3a62013-10-08 09:05:10 +0100363
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200364STATIC void need_reg_all(emit_t *emit) {
Damienff8ed772013-10-08 22:18:32 +0100365 for (int i = 0; i < emit->stack_size; i++) {
366 stack_info_t *si = &emit->stack_info[i];
367 if (si->kind == STACK_REG) {
368 si->kind = STACK_VALUE;
369 ASM_MOV_REG_TO_LOCAL(si->u_reg, emit->stack_start + i);
370 }
Damien13ed3a62013-10-08 09:05:10 +0100371 }
372}
373
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200374STATIC void need_stack_settled(emit_t *emit) {
Damiend2755ec2013-10-16 23:58:48 +0100375 for (int i = 0; i < emit->stack_size; i++) {
376 stack_info_t *si = &emit->stack_info[i];
377 if (si->kind == STACK_REG) {
378 si->kind = STACK_VALUE;
379 ASM_MOV_REG_TO_LOCAL(si->u_reg, emit->stack_start + i);
380 }
381 }
382 for (int i = 0; i < emit->stack_size; i++) {
383 stack_info_t *si = &emit->stack_info[i];
384 if (si->kind == STACK_IMM) {
385 ASM_MOV_IMM_TO_LOCAL_USING(si->u_imm, emit->stack_start + i, REG_TEMP0);
386 }
387 }
388}
389
390// pos=1 is TOS, pos=2 is next, etc
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200391STATIC void emit_access_stack(emit_t *emit, int pos, vtype_kind_t *vtype, int reg_dest) {
Damiend2755ec2013-10-16 23:58:48 +0100392 need_reg_single(emit, reg_dest, pos);
393 stack_info_t *si = &emit->stack_info[emit->stack_size - pos];
Damienff8ed772013-10-08 22:18:32 +0100394 *vtype = si->vtype;
395 switch (si->kind) {
396 case STACK_VALUE:
Damiend2755ec2013-10-16 23:58:48 +0100397 ASM_MOV_LOCAL_TO_REG(emit->stack_start + emit->stack_size - pos, reg_dest);
Damien13ed3a62013-10-08 09:05:10 +0100398 break;
399
Damienff8ed772013-10-08 22:18:32 +0100400 case STACK_REG:
401 if (si->u_reg != reg_dest) {
402 ASM_MOV_REG_TO_REG(si->u_reg, reg_dest);
Damien13ed3a62013-10-08 09:05:10 +0100403 }
404 break;
405
Damienff8ed772013-10-08 22:18:32 +0100406 case STACK_IMM:
407 ASM_MOV_IMM_TO_REG(si->u_imm, reg_dest);
Damien13ed3a62013-10-08 09:05:10 +0100408 break;
409 }
Damien13ed3a62013-10-08 09:05:10 +0100410}
411
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200412STATIC void emit_pre_pop_reg(emit_t *emit, vtype_kind_t *vtype, int reg_dest) {
Damiend2755ec2013-10-16 23:58:48 +0100413 emit->last_emit_was_return_value = false;
414 emit_access_stack(emit, 1, vtype, reg_dest);
415 adjust_stack(emit, -1);
416}
417
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200418STATIC void emit_pre_pop_reg_reg(emit_t *emit, vtype_kind_t *vtypea, int rega, vtype_kind_t *vtypeb, int regb) {
Damien13ed3a62013-10-08 09:05:10 +0100419 emit_pre_pop_reg(emit, vtypea, rega);
Damienff8ed772013-10-08 22:18:32 +0100420 emit_pre_pop_reg(emit, vtypeb, regb);
Damien13ed3a62013-10-08 09:05:10 +0100421}
422
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200423STATIC void emit_pre_pop_reg_reg_reg(emit_t *emit, vtype_kind_t *vtypea, int rega, vtype_kind_t *vtypeb, int regb, vtype_kind_t *vtypec, int regc) {
Damien13ed3a62013-10-08 09:05:10 +0100424 emit_pre_pop_reg(emit, vtypea, rega);
Damienff8ed772013-10-08 22:18:32 +0100425 emit_pre_pop_reg(emit, vtypeb, regb);
426 emit_pre_pop_reg(emit, vtypec, regc);
Damien13ed3a62013-10-08 09:05:10 +0100427}
428
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200429STATIC void emit_post(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100430}
431
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200432STATIC void emit_post_push_reg(emit_t *emit, vtype_kind_t vtype, int reg) {
Damienff8ed772013-10-08 22:18:32 +0100433 stack_info_t *si = &emit->stack_info[emit->stack_size];
434 si->vtype = vtype;
435 si->kind = STACK_REG;
436 si->u_reg = reg;
Damien13ed3a62013-10-08 09:05:10 +0100437 adjust_stack(emit, 1);
438}
439
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200440STATIC void emit_post_push_imm(emit_t *emit, vtype_kind_t vtype, machine_int_t imm) {
Damienff8ed772013-10-08 22:18:32 +0100441 stack_info_t *si = &emit->stack_info[emit->stack_size];
442 si->vtype = vtype;
443 si->kind = STACK_IMM;
444 si->u_imm = imm;
445 adjust_stack(emit, 1);
446}
447
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200448STATIC void emit_post_push_reg_reg(emit_t *emit, vtype_kind_t vtypea, int rega, vtype_kind_t vtypeb, int regb) {
Damienff8ed772013-10-08 22:18:32 +0100449 emit_post_push_reg(emit, vtypea, rega);
450 emit_post_push_reg(emit, vtypeb, regb);
451}
452
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200453STATIC void emit_post_push_reg_reg_reg(emit_t *emit, vtype_kind_t vtypea, int rega, vtype_kind_t vtypeb, int regb, vtype_kind_t vtypec, int regc) {
Damienff8ed772013-10-08 22:18:32 +0100454 emit_post_push_reg(emit, vtypea, rega);
455 emit_post_push_reg(emit, vtypeb, regb);
456 emit_post_push_reg(emit, vtypec, regc);
Damien13ed3a62013-10-08 09:05:10 +0100457}
458
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200459STATIC void emit_post_push_reg_reg_reg_reg(emit_t *emit, vtype_kind_t vtypea, int rega, vtype_kind_t vtypeb, int regb, vtype_kind_t vtypec, int regc, vtype_kind_t vtyped, int regd) {
Damienff8ed772013-10-08 22:18:32 +0100460 emit_post_push_reg(emit, vtypea, rega);
461 emit_post_push_reg(emit, vtypeb, regb);
462 emit_post_push_reg(emit, vtypec, regc);
463 emit_post_push_reg(emit, vtyped, regd);
Damien13ed3a62013-10-08 09:05:10 +0100464}
465
466// vtype of all n_pop objects is VTYPE_PYOBJ
Damieneb19efb2013-10-10 22:06:54 +0100467// does not use any temporary registers (but may use reg_dest before loading it with stack pointer)
Damien6d4f3462013-11-16 20:44:39 +0000468// TODO this needs some thinking for viper code
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200469STATIC void emit_get_stack_pointer_to_reg_for_pop(emit_t *emit, int reg_dest, int n_pop) {
Damieneb19efb2013-10-10 22:06:54 +0100470 need_reg_all(emit);
Damienff8ed772013-10-08 22:18:32 +0100471 for (int i = 0; i < n_pop; i++) {
Damieneb19efb2013-10-10 22:06:54 +0100472 stack_info_t *si = &emit->stack_info[emit->stack_size - 1 - i];
473 // must push any imm's to stack
Damien6d4f3462013-11-16 20:44:39 +0000474 // must convert them to VTYPE_PYOBJ for viper code
Damieneb19efb2013-10-10 22:06:54 +0100475 if (si->kind == STACK_IMM) {
476 si->kind = STACK_VALUE;
Damien6d4f3462013-11-16 20:44:39 +0000477 switch (si->vtype) {
478 case VTYPE_PYOBJ:
479 ASM_MOV_IMM_TO_LOCAL_USING(si->u_imm, emit->stack_start + emit->stack_size - 1 - i, reg_dest);
480 break;
481 case VTYPE_BOOL:
482 si->vtype = VTYPE_PYOBJ;
483 if (si->u_imm == 0) {
Damiend99b0522013-12-21 18:17:45 +0000484 ASM_MOV_IMM_TO_LOCAL_USING((machine_uint_t)mp_const_false, emit->stack_start + emit->stack_size - 1 - i, reg_dest);
Damien6d4f3462013-11-16 20:44:39 +0000485 } else {
Damiend99b0522013-12-21 18:17:45 +0000486 ASM_MOV_IMM_TO_LOCAL_USING((machine_uint_t)mp_const_true, emit->stack_start + emit->stack_size - 1 - i, reg_dest);
Damien6d4f3462013-11-16 20:44:39 +0000487 }
488 break;
489 case VTYPE_INT:
490 si->vtype = VTYPE_PYOBJ;
491 ASM_MOV_IMM_TO_LOCAL_USING((si->u_imm << 1) | 1, emit->stack_start + emit->stack_size - 1 - i, reg_dest);
492 break;
493 default:
494 // not handled
495 assert(0);
496 }
Damieneb19efb2013-10-10 22:06:54 +0100497 }
498 assert(si->kind == STACK_VALUE);
499 assert(si->vtype == VTYPE_PYOBJ);
Damienff8ed772013-10-08 22:18:32 +0100500 }
Damien13ed3a62013-10-08 09:05:10 +0100501 adjust_stack(emit, -n_pop);
Damien Georgecd82e022014-02-02 13:11:48 +0000502 ASM_MOV_LOCAL_ADDR_TO_REG(emit->stack_start + emit->stack_size, reg_dest);
Damien13ed3a62013-10-08 09:05:10 +0100503}
504
505// vtype of all n_push objects is VTYPE_PYOBJ
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200506STATIC void emit_get_stack_pointer_to_reg_for_push(emit_t *emit, int reg_dest, int n_push) {
Damieneb19efb2013-10-10 22:06:54 +0100507 need_reg_all(emit);
Damien13ed3a62013-10-08 09:05:10 +0100508 for (int i = 0; i < n_push; i++) {
Damien7f5dacf2013-10-10 11:24:39 +0100509 emit->stack_info[emit->stack_size + i].kind = STACK_VALUE;
Damienff8ed772013-10-08 22:18:32 +0100510 emit->stack_info[emit->stack_size + i].vtype = VTYPE_PYOBJ;
Damien13ed3a62013-10-08 09:05:10 +0100511 }
Damien Georgecd82e022014-02-02 13:11:48 +0000512 ASM_MOV_LOCAL_ADDR_TO_REG(emit->stack_start + emit->stack_size, reg_dest);
Damien13ed3a62013-10-08 09:05:10 +0100513 adjust_stack(emit, n_push);
514}
515
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200516STATIC void emit_call(emit_t *emit, rt_fun_kind_t fun_kind, void *fun) {
Damiend2755ec2013-10-16 23:58:48 +0100517 need_reg_all(emit);
Damien3ef4abb2013-10-12 16:53:13 +0100518#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100519 asm_x64_call_ind(emit->as, fun, REG_RAX);
Damien3ef4abb2013-10-12 16:53:13 +0100520#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100521 asm_thumb_bl_ind(emit->as, rt_fun_table[fun_kind], fun_kind, REG_R3);
522#endif
523}
524
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200525STATIC void emit_call_with_imm_arg(emit_t *emit, rt_fun_kind_t fun_kind, void *fun, machine_int_t arg_val, int arg_reg) {
Damieneb19efb2013-10-10 22:06:54 +0100526 need_reg_all(emit);
Damien13ed3a62013-10-08 09:05:10 +0100527 ASM_MOV_IMM_TO_REG(arg_val, arg_reg);
Damiend2755ec2013-10-16 23:58:48 +0100528#if N_X64
529 asm_x64_call_ind(emit->as, fun, REG_RAX);
530#elif N_THUMB
531 asm_thumb_bl_ind(emit->as, rt_fun_table[fun_kind], fun_kind, REG_R3);
532#endif
Damien13ed3a62013-10-08 09:05:10 +0100533}
534
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200535STATIC void emit_call_with_2_imm_args(emit_t *emit, rt_fun_kind_t fun_kind, void *fun, machine_int_t arg_val1, int arg_reg1, machine_int_t arg_val2, int arg_reg2) {
Damien Georgecd82e022014-02-02 13:11:48 +0000536 need_reg_all(emit);
537 ASM_MOV_IMM_TO_REG(arg_val1, arg_reg1);
538 ASM_MOV_IMM_TO_REG(arg_val2, arg_reg2);
539#if N_X64
540 asm_x64_call_ind(emit->as, fun, REG_RAX);
541#elif N_THUMB
542 asm_thumb_bl_ind(emit->as, rt_fun_table[fun_kind], fun_kind, REG_R3);
543#endif
544}
545
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200546STATIC void emit_native_load_id(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100547 // check for built-ins
548 if (strcmp(qstr_str(qstr), "v_int") == 0) {
Damienff8ed772013-10-08 22:18:32 +0100549 assert(0);
Damien13ed3a62013-10-08 09:05:10 +0100550 emit_pre(emit);
551 //emit_post_push_blank(emit, VTYPE_BUILTIN_V_INT);
552
553 // not a built-in, so do usual thing
554 } else {
555 emit_common_load_id(emit, &EXPORT_FUN(method_table), emit->scope, qstr);
556 }
557}
558
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200559STATIC void emit_native_store_id(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100560 // TODO check for built-ins and disallow
561 emit_common_store_id(emit, &EXPORT_FUN(method_table), emit->scope, qstr);
562}
563
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200564STATIC void emit_native_delete_id(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100565 // TODO check for built-ins and disallow
566 emit_common_delete_id(emit, &EXPORT_FUN(method_table), emit->scope, qstr);
567}
568
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200569STATIC void emit_native_label_assign(emit_t *emit, int l) {
Damien6ba13142013-11-02 20:34:54 +0000570 emit_pre(emit);
Damiend2755ec2013-10-16 23:58:48 +0100571 // need to commit stack because we can jump here from elsewhere
572 need_stack_settled(emit);
Damien3ef4abb2013-10-12 16:53:13 +0100573#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100574 asm_x64_label_assign(emit->as, l);
Damien3ef4abb2013-10-12 16:53:13 +0100575#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100576 asm_thumb_label_assign(emit->as, l);
577#endif
Damien6ba13142013-11-02 20:34:54 +0000578 emit_post(emit);
Damien13ed3a62013-10-08 09:05:10 +0100579}
580
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200581STATIC void emit_native_import_name(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100582 // not implemented
583 assert(0);
584}
585
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200586STATIC void emit_native_import_from(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100587 // not implemented
588 assert(0);
589}
590
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200591STATIC void emit_native_import_star(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100592 // not implemented
593 assert(0);
594}
595
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200596STATIC void emit_native_load_const_tok(emit_t *emit, mp_token_kind_t tok) {
Damien13ed3a62013-10-08 09:05:10 +0100597 emit_pre(emit);
598 int vtype;
599 machine_uint_t val;
600 if (emit->do_viper_types) {
601 switch (tok) {
Damiend99b0522013-12-21 18:17:45 +0000602 case MP_TOKEN_KW_NONE: vtype = VTYPE_PTR_NONE; val = 0; break;
603 case MP_TOKEN_KW_FALSE: vtype = VTYPE_BOOL; val = 0; break;
604 case MP_TOKEN_KW_TRUE: vtype = VTYPE_BOOL; val = 1; break;
Damien13ed3a62013-10-08 09:05:10 +0100605 default: assert(0); vtype = 0; val = 0; // shouldn't happen
606 }
607 } else {
608 vtype = VTYPE_PYOBJ;
609 switch (tok) {
Damiend99b0522013-12-21 18:17:45 +0000610 case MP_TOKEN_KW_NONE: val = (machine_uint_t)mp_const_none; break;
611 case MP_TOKEN_KW_FALSE: val = (machine_uint_t)mp_const_false; break;
612 case MP_TOKEN_KW_TRUE: val = (machine_uint_t)mp_const_true; break;
Damien13ed3a62013-10-08 09:05:10 +0100613 default: assert(0); vtype = 0; val = 0; // shouldn't happen
614 }
615 }
616 emit_post_push_imm(emit, vtype, val);
617}
618
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200619STATIC void emit_native_load_const_small_int(emit_t *emit, machine_int_t arg) {
Damien13ed3a62013-10-08 09:05:10 +0100620 emit_pre(emit);
621 if (emit->do_viper_types) {
622 emit_post_push_imm(emit, VTYPE_INT, arg);
623 } else {
624 emit_post_push_imm(emit, VTYPE_PYOBJ, (arg << 1) | 1);
625 }
626}
627
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200628STATIC void emit_native_load_const_int(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100629 // not implemented
630 // load integer, check fits in 32 bits
631 assert(0);
632}
633
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200634STATIC void emit_native_load_const_dec(emit_t *emit, qstr qstr) {
Damien6ba13142013-11-02 20:34:54 +0000635 // for viper, a float/complex is just a Python object
636 emit_pre(emit);
637 emit_call_with_imm_arg(emit, RT_F_LOAD_CONST_DEC, rt_load_const_dec, qstr, REG_ARG_1);
638 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
Damien13ed3a62013-10-08 09:05:10 +0100639}
640
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200641STATIC void emit_native_load_const_id(emit_t *emit, qstr qstr) {
Damien7f5dacf2013-10-10 11:24:39 +0100642 emit_pre(emit);
643 if (emit->do_viper_types) {
644 assert(0);
645 } else {
646 emit_call_with_imm_arg(emit, RT_F_LOAD_CONST_STR, rt_load_const_str, qstr, REG_ARG_1); // TODO
647 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
648 }
Damien13ed3a62013-10-08 09:05:10 +0100649}
650
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200651STATIC void emit_native_load_const_str(emit_t *emit, qstr qstr, bool bytes) {
Damien13ed3a62013-10-08 09:05:10 +0100652 emit_pre(emit);
653 if (emit->do_viper_types) {
654 // not implemented properly
655 // load a pointer to the asciiz string?
656 assert(0);
657 emit_post_push_imm(emit, VTYPE_PTR, (machine_uint_t)qstr_str(qstr));
658 } else {
659 emit_call_with_imm_arg(emit, RT_F_LOAD_CONST_STR, rt_load_const_str, qstr, REG_ARG_1);
660 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
661 }
662}
663
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200664STATIC void emit_native_load_const_verbatim_str(emit_t *emit, const char *str) {
Damien13ed3a62013-10-08 09:05:10 +0100665 // not supported/needed for viper
666 assert(0);
667}
668
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200669STATIC void emit_native_load_fast(emit_t *emit, qstr qstr, int local_num) {
Damien13ed3a62013-10-08 09:05:10 +0100670 vtype_kind_t vtype = emit->local_vtype[local_num];
671 if (vtype == VTYPE_UNBOUND) {
672 printf("ViperTypeError: local %s used before type known\n", qstr_str(qstr));
673 }
674 emit_pre(emit);
Damien3ef4abb2013-10-12 16:53:13 +0100675#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100676 if (local_num == 0) {
677 emit_post_push_reg(emit, vtype, REG_LOCAL_1);
678 } else {
Damiend2755ec2013-10-16 23:58:48 +0100679 need_reg_single(emit, REG_RAX, 0);
Damien13ed3a62013-10-08 09:05:10 +0100680 asm_x64_mov_local_to_r64(emit->as, local_num - 1, REG_RAX);
681 emit_post_push_reg(emit, vtype, REG_RAX);
682 }
Damien3ef4abb2013-10-12 16:53:13 +0100683#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100684 if (local_num == 0) {
685 emit_post_push_reg(emit, vtype, REG_LOCAL_1);
686 } else if (local_num == 1) {
687 emit_post_push_reg(emit, vtype, REG_LOCAL_2);
688 } else if (local_num == 2) {
689 emit_post_push_reg(emit, vtype, REG_LOCAL_3);
690 } else {
Damiend2755ec2013-10-16 23:58:48 +0100691 need_reg_single(emit, REG_R0, 0);
Damien13ed3a62013-10-08 09:05:10 +0100692 asm_thumb_mov_reg_local(emit->as, REG_R0, local_num - 1);
693 emit_post_push_reg(emit, vtype, REG_R0);
694 }
695#endif
696}
697
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200698STATIC void emit_native_load_deref(emit_t *emit, qstr qstr, int local_num) {
Damien9ecbcff2013-12-11 00:41:43 +0000699 // not implemented
700 // in principle could support this quite easily (ldr r0, [r0, #0]) and then get closed over variables!
701 assert(0);
702}
703
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200704STATIC void emit_native_load_closure(emit_t *emit, qstr qstr, int local_num) {
Damien9ecbcff2013-12-11 00:41:43 +0000705 // not implemented
706 assert(0);
707}
708
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200709STATIC void emit_native_load_name(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100710 emit_pre(emit);
711 emit_call_with_imm_arg(emit, RT_F_LOAD_NAME, rt_load_name, qstr, REG_ARG_1);
712 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
713}
714
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200715STATIC void emit_native_load_global(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100716 emit_pre(emit);
717 emit_call_with_imm_arg(emit, RT_F_LOAD_GLOBAL, rt_load_global, qstr, REG_ARG_1);
718 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
719}
720
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200721STATIC void emit_native_load_attr(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100722 // depends on type of subject:
723 // - integer, function, pointer to integers: error
724 // - pointer to structure: get member, quite easy
725 // - Python object: call rt_load_attr, and needs to be typed to convert result
726 vtype_kind_t vtype_base;
727 emit_pre_pop_reg(emit, &vtype_base, REG_ARG_1); // arg1 = base
728 assert(vtype_base == VTYPE_PYOBJ);
729 emit_call_with_imm_arg(emit, RT_F_LOAD_ATTR, rt_load_attr, qstr, REG_ARG_2); // arg2 = attribute name
730 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
731}
732
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200733STATIC void emit_native_load_method(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100734 vtype_kind_t vtype_base;
735 emit_pre_pop_reg(emit, &vtype_base, REG_ARG_1); // arg1 = base
736 assert(vtype_base == VTYPE_PYOBJ);
737 emit_get_stack_pointer_to_reg_for_push(emit, REG_ARG_3, 2); // arg3 = dest ptr
738 emit_call_with_imm_arg(emit, RT_F_LOAD_METHOD, rt_load_method, qstr, REG_ARG_2); // arg2 = method name
739}
740
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200741STATIC void emit_native_load_build_class(emit_t *emit) {
Damien7f5dacf2013-10-10 11:24:39 +0100742 emit_pre(emit);
743 emit_call(emit, RT_F_LOAD_BUILD_CLASS, rt_load_build_class);
744 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
Damien13ed3a62013-10-08 09:05:10 +0100745}
746
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200747STATIC void emit_native_store_fast(emit_t *emit, qstr qstr, int local_num) {
Damien13ed3a62013-10-08 09:05:10 +0100748 vtype_kind_t vtype;
Damien3ef4abb2013-10-12 16:53:13 +0100749#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100750 if (local_num == 0) {
751 emit_pre_pop_reg(emit, &vtype, REG_LOCAL_1);
752 } else {
753 emit_pre_pop_reg(emit, &vtype, REG_RAX);
754 asm_x64_mov_r64_to_local(emit->as, REG_RAX, local_num - 1);
755 }
Damien3ef4abb2013-10-12 16:53:13 +0100756#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100757 if (local_num == 0) {
758 emit_pre_pop_reg(emit, &vtype, REG_LOCAL_1);
759 } else if (local_num == 1) {
760 emit_pre_pop_reg(emit, &vtype, REG_LOCAL_2);
761 } else if (local_num == 2) {
762 emit_pre_pop_reg(emit, &vtype, REG_LOCAL_3);
763 } else {
764 emit_pre_pop_reg(emit, &vtype, REG_R0);
765 asm_thumb_mov_local_reg(emit->as, local_num - 1, REG_R0);
766 }
767#endif
768
769 emit_post(emit);
770
771 // check types
772 if (emit->local_vtype[local_num] == VTYPE_UNBOUND) {
773 // first time this local is assigned, so give it a type of the object stored in it
774 emit->local_vtype[local_num] = vtype;
775 } else if (emit->local_vtype[local_num] != vtype) {
776 // type of local is not the same as object stored in it
777 printf("ViperTypeError: type mismatch, local %s has type %d but source object has type %d\n", qstr_str(qstr), emit->local_vtype[local_num], vtype);
778 }
779}
780
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200781STATIC void emit_native_store_deref(emit_t *emit, qstr qstr, int local_num) {
Damien9ecbcff2013-12-11 00:41:43 +0000782 // not implemented
783 assert(0);
784}
785
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200786STATIC void emit_native_store_name(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100787 // rt_store_name, but needs conversion of object (maybe have rt_viper_store_name(obj, type))
788 vtype_kind_t vtype;
789 emit_pre_pop_reg(emit, &vtype, REG_ARG_2);
790 assert(vtype == VTYPE_PYOBJ);
791 emit_call_with_imm_arg(emit, RT_F_STORE_NAME, rt_store_name, qstr, REG_ARG_1); // arg1 = name
792 emit_post(emit);
793}
794
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200795STATIC void emit_native_store_global(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100796 // not implemented
797 assert(0);
798}
799
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200800STATIC void emit_native_store_attr(emit_t *emit, qstr qstr) {
Damien7f5dacf2013-10-10 11:24:39 +0100801 vtype_kind_t vtype_base, vtype_val;
802 emit_pre_pop_reg_reg(emit, &vtype_base, REG_ARG_1, &vtype_val, REG_ARG_3); // arg1 = base, arg3 = value
803 assert(vtype_base == VTYPE_PYOBJ);
804 assert(vtype_val == VTYPE_PYOBJ);
805 emit_call_with_imm_arg(emit, RT_F_STORE_ATTR, rt_store_attr, qstr, REG_ARG_2); // arg2 = attribute name
806 emit_post(emit);
Damien13ed3a62013-10-08 09:05:10 +0100807}
808
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200809STATIC void emit_native_store_subscr(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100810 // depends on type of subject:
811 // - integer, function, pointer to structure: error
812 // - pointer to integers: store as per array
813 // - Python object: call runtime with converted object or type info
814 vtype_kind_t vtype_index, vtype_base, vtype_value;
815 emit_pre_pop_reg_reg_reg(emit, &vtype_index, REG_ARG_2, &vtype_base, REG_ARG_1, &vtype_value, REG_ARG_3); // index, base, value to store
816 assert(vtype_index == VTYPE_PYOBJ);
817 assert(vtype_base == VTYPE_PYOBJ);
818 assert(vtype_value == VTYPE_PYOBJ);
819 emit_call(emit, RT_F_STORE_SUBSCR, rt_store_subscr);
820}
821
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200822STATIC void emit_native_store_locals(emit_t *emit) {
Damiena3977762013-10-09 23:10:10 +0100823 // not needed
824 vtype_kind_t vtype;
825 emit_pre_pop_reg(emit, &vtype, REG_TEMP0);
826 emit_post(emit);
827}
828
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200829STATIC void emit_native_delete_fast(emit_t *emit, qstr qstr, int local_num) {
Damien13ed3a62013-10-08 09:05:10 +0100830 // not implemented
831 // could support for Python types, just set to None (so GC can reclaim it)
832 assert(0);
833}
834
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200835STATIC void emit_native_delete_deref(emit_t *emit, qstr qstr, int local_num) {
Damien9ecbcff2013-12-11 00:41:43 +0000836 // not supported
837 assert(0);
838}
839
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200840STATIC void emit_native_delete_name(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100841 // not implemented
842 // use rt_delete_name
843 assert(0);
844}
845
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200846STATIC void emit_native_delete_global(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100847 // not implemented
848 // use rt_delete_global
849 assert(0);
850}
851
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200852STATIC void emit_native_delete_attr(emit_t *emit, qstr qstr) {
Damien13ed3a62013-10-08 09:05:10 +0100853 // not supported
854 assert(0);
855}
856
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200857STATIC void emit_native_delete_subscr(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100858 // not supported
859 assert(0);
860}
861
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200862STATIC void emit_native_dup_top(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100863 vtype_kind_t vtype;
864 emit_pre_pop_reg(emit, &vtype, REG_TEMP0);
865 emit_post_push_reg_reg(emit, vtype, REG_TEMP0, vtype, REG_TEMP0);
866}
867
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200868STATIC void emit_native_dup_top_two(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100869 vtype_kind_t vtype0, vtype1;
870 emit_pre_pop_reg_reg(emit, &vtype0, REG_TEMP0, &vtype1, REG_TEMP1);
871 emit_post_push_reg_reg_reg_reg(emit, vtype1, REG_TEMP1, vtype0, REG_TEMP0, vtype1, REG_TEMP1, vtype0, REG_TEMP0);
872}
873
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200874STATIC void emit_native_pop_top(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100875 vtype_kind_t vtype;
876 emit_pre_pop_reg(emit, &vtype, REG_TEMP0);
877 emit_post(emit);
878}
879
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200880STATIC void emit_native_rot_two(emit_t *emit) {
Damienff8ed772013-10-08 22:18:32 +0100881 vtype_kind_t vtype0, vtype1;
882 emit_pre_pop_reg_reg(emit, &vtype0, REG_TEMP0, &vtype1, REG_TEMP1);
883 emit_post_push_reg_reg(emit, vtype0, REG_TEMP0, vtype1, REG_TEMP1);
Damien13ed3a62013-10-08 09:05:10 +0100884}
885
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200886STATIC void emit_native_rot_three(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100887 vtype_kind_t vtype0, vtype1, vtype2;
888 emit_pre_pop_reg_reg_reg(emit, &vtype0, REG_TEMP0, &vtype1, REG_TEMP1, &vtype2, REG_TEMP2);
889 emit_post_push_reg_reg_reg(emit, vtype0, REG_TEMP0, vtype2, REG_TEMP2, vtype1, REG_TEMP1);
890}
891
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200892STATIC void emit_native_jump(emit_t *emit, int label) {
Damien13ed3a62013-10-08 09:05:10 +0100893 emit_pre(emit);
Damien3ef4abb2013-10-12 16:53:13 +0100894#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100895 asm_x64_jmp_label(emit->as, label);
Damien3ef4abb2013-10-12 16:53:13 +0100896#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +0100897 asm_thumb_b_label(emit->as, label);
898#endif
899 emit_post(emit);
900}
901
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200902STATIC void emit_native_pop_jump_pre_helper(emit_t *emit, int label) {
Damien13ed3a62013-10-08 09:05:10 +0100903 vtype_kind_t vtype = peek_vtype(emit);
904 if (vtype == VTYPE_BOOL) {
905 emit_pre_pop_reg(emit, &vtype, REG_RET);
906 } else if (vtype == VTYPE_PYOBJ) {
907 emit_pre_pop_reg(emit, &vtype, REG_ARG_1);
908 emit_call(emit, RT_F_IS_TRUE, rt_is_true);
909 } else {
910 printf("ViperTypeError: expecting a bool or pyobj, got %d\n", vtype);
911 assert(0);
912 }
Damien1a6633a2013-11-03 13:58:19 +0000913}
914
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200915STATIC void emit_native_pop_jump_if_false(emit_t *emit, int label) {
Damien1a6633a2013-11-03 13:58:19 +0000916 emit_native_pop_jump_pre_helper(emit, label);
Damien3ef4abb2013-10-12 16:53:13 +0100917#if N_X64
Damien13ed3a62013-10-08 09:05:10 +0100918 asm_x64_test_r8_with_r8(emit->as, REG_RET, REG_RET);
919 asm_x64_jcc_label(emit->as, JCC_JZ, label);
Damien3ef4abb2013-10-12 16:53:13 +0100920#elif N_THUMB
Damien1a6633a2013-11-03 13:58:19 +0000921 asm_thumb_cmp_rlo_i8(emit->as, REG_RET, 0);
922 asm_thumb_bcc_label(emit->as, THUMB_CC_EQ, label);
Damien13ed3a62013-10-08 09:05:10 +0100923#endif
924 emit_post(emit);
925}
926
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200927STATIC void emit_native_pop_jump_if_true(emit_t *emit, int label) {
Damien1a6633a2013-11-03 13:58:19 +0000928 emit_native_pop_jump_pre_helper(emit, label);
929#if N_X64
930 asm_x64_test_r8_with_r8(emit->as, REG_RET, REG_RET);
931 asm_x64_jcc_label(emit->as, JCC_JNZ, label);
932#elif N_THUMB
933 asm_thumb_cmp_rlo_i8(emit->as, REG_RET, 0);
934 asm_thumb_bcc_label(emit->as, THUMB_CC_NE, label);
935#endif
936 emit_post(emit);
Damien13ed3a62013-10-08 09:05:10 +0100937}
Damien1a6633a2013-11-03 13:58:19 +0000938
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200939STATIC void emit_native_jump_if_true_or_pop(emit_t *emit, int label) {
Damien13ed3a62013-10-08 09:05:10 +0100940 assert(0);
941}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200942STATIC void emit_native_jump_if_false_or_pop(emit_t *emit, int label) {
Damien13ed3a62013-10-08 09:05:10 +0100943 assert(0);
944}
945
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200946STATIC void emit_native_setup_loop(emit_t *emit, int label) {
Damien13ed3a62013-10-08 09:05:10 +0100947 emit_pre(emit);
948 emit_post(emit);
949}
950
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200951STATIC void emit_native_break_loop(emit_t *emit, int label, int except_depth) {
Damien6ba13142013-11-02 20:34:54 +0000952 emit_native_jump(emit, label); // TODO properly
Damien13ed3a62013-10-08 09:05:10 +0100953}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200954STATIC void emit_native_continue_loop(emit_t *emit, int label, int except_depth) {
Damien13ed3a62013-10-08 09:05:10 +0100955 assert(0);
956}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200957STATIC void emit_native_setup_with(emit_t *emit, int label) {
Damien13ed3a62013-10-08 09:05:10 +0100958 // not supported, or could be with runtime call
959 assert(0);
960}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200961STATIC void emit_native_with_cleanup(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100962 assert(0);
963}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200964STATIC void emit_native_setup_except(emit_t *emit, int label) {
Damien13ed3a62013-10-08 09:05:10 +0100965 assert(0);
966}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200967STATIC void emit_native_setup_finally(emit_t *emit, int label) {
Damien13ed3a62013-10-08 09:05:10 +0100968 assert(0);
969}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200970STATIC void emit_native_end_finally(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100971 assert(0);
972}
Damiend2755ec2013-10-16 23:58:48 +0100973
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200974STATIC void emit_native_get_iter(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +0100975 // perhaps the difficult one, as we want to rewrite for loops using native code
976 // in cases where we iterate over a Python object, can we use normal runtime calls?
Damiend2755ec2013-10-16 23:58:48 +0100977
978 vtype_kind_t vtype;
979 emit_pre_pop_reg(emit, &vtype, REG_ARG_1);
980 assert(vtype == VTYPE_PYOBJ);
981 emit_call(emit, RT_F_GETITER, rt_getiter);
982 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
Damien13ed3a62013-10-08 09:05:10 +0100983}
Damiend2755ec2013-10-16 23:58:48 +0100984
Paul Sokolovsky520e2f52014-02-12 18:31:30 +0200985STATIC void emit_native_for_iter(emit_t *emit, int label) {
Damiend2755ec2013-10-16 23:58:48 +0100986 emit_pre(emit);
987 vtype_kind_t vtype;
988 emit_access_stack(emit, 1, &vtype, REG_ARG_1);
989 assert(vtype == VTYPE_PYOBJ);
990 emit_call(emit, RT_F_ITERNEXT, rt_iternext);
Damien George66eaf842014-03-26 19:27:58 +0000991 ASM_MOV_IMM_TO_REG((machine_uint_t)MP_OBJ_NULL, REG_TEMP1);
Damiend2755ec2013-10-16 23:58:48 +0100992#if N_X64
993 asm_x64_cmp_r64_with_r64(emit->as, REG_RET, REG_TEMP1);
994 asm_x64_jcc_label(emit->as, JCC_JE, label);
995#elif N_THUMB
Damiend2755ec2013-10-16 23:58:48 +0100996 asm_thumb_cmp_reg_reg(emit->as, REG_RET, REG_TEMP1);
Damien9b9e9962013-11-03 14:25:43 +0000997 asm_thumb_bcc_label(emit->as, THUMB_CC_EQ, label);
Damiend2755ec2013-10-16 23:58:48 +0100998#endif
999 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
1000}
1001
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001002STATIC void emit_native_for_iter_end(emit_t *emit) {
Damiend2755ec2013-10-16 23:58:48 +01001003 // adjust stack counter (we get here from for_iter ending, which popped the value for us)
1004 emit_pre(emit);
1005 adjust_stack(emit, -1);
1006 emit_post(emit);
Damien13ed3a62013-10-08 09:05:10 +01001007}
1008
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001009STATIC void emit_native_pop_block(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +01001010 emit_pre(emit);
1011 emit_post(emit);
1012}
1013
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001014STATIC void emit_native_pop_except(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +01001015 assert(0);
1016}
1017
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001018STATIC void emit_native_unary_op(emit_t *emit, rt_unary_op_t op) {
Damien13ed3a62013-10-08 09:05:10 +01001019 vtype_kind_t vtype;
1020 emit_pre_pop_reg(emit, &vtype, REG_ARG_2);
1021 assert(vtype == VTYPE_PYOBJ);
1022 emit_call_with_imm_arg(emit, RT_F_UNARY_OP, rt_unary_op, op, REG_ARG_1);
1023 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
1024}
1025
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001026STATIC void emit_native_binary_op(emit_t *emit, rt_binary_op_t op) {
Damien13ed3a62013-10-08 09:05:10 +01001027 vtype_kind_t vtype_lhs, vtype_rhs;
1028 emit_pre_pop_reg_reg(emit, &vtype_rhs, REG_ARG_3, &vtype_lhs, REG_ARG_2);
1029 if (vtype_lhs == VTYPE_INT && vtype_rhs == VTYPE_INT) {
Damien Georgebc1d3692014-01-11 09:47:06 +00001030 if (op == RT_BINARY_OP_ADD || op == RT_BINARY_OP_INPLACE_ADD) {
Damien3ef4abb2013-10-12 16:53:13 +01001031#if N_X64
Damien Georgebc1d3692014-01-11 09:47:06 +00001032 asm_x64_add_r64_to_r64(emit->as, REG_ARG_3, REG_ARG_2);
Damien3ef4abb2013-10-12 16:53:13 +01001033#elif N_THUMB
Damien Georgebc1d3692014-01-11 09:47:06 +00001034 asm_thumb_add_reg_reg_reg(emit->as, REG_ARG_2, REG_ARG_2, REG_ARG_3);
Damien13ed3a62013-10-08 09:05:10 +01001035#endif
Damien Georgebc1d3692014-01-11 09:47:06 +00001036 emit_post_push_reg(emit, VTYPE_INT, REG_ARG_2);
Damien George9aa2a522014-02-01 23:04:09 +00001037 } else if (op == RT_BINARY_OP_LESS) {
Damien Georgebc1d3692014-01-11 09:47:06 +00001038#if N_X64
1039 asm_x64_xor_r64_to_r64(emit->as, REG_RET, REG_RET);
1040 asm_x64_cmp_r64_with_r64(emit->as, REG_ARG_3, REG_ARG_2);
1041 asm_x64_setcc_r8(emit->as, JCC_JL, REG_RET);
1042#elif N_THUMB
1043 asm_thumb_cmp_reg_reg(emit->as, REG_ARG_2, REG_ARG_3);
1044 asm_thumb_ite_ge(emit->as);
1045 asm_thumb_movs_rlo_i8(emit->as, REG_RET, 0); // if r0 >= r1
1046 asm_thumb_movs_rlo_i8(emit->as, REG_RET, 1); // if r0 < r1
1047#endif
1048 emit_post_push_reg(emit, VTYPE_BOOL, REG_RET);
1049 } else {
1050 // TODO other ops not yet implemented
1051 assert(0);
1052 }
Damien13ed3a62013-10-08 09:05:10 +01001053 } else if (vtype_lhs == VTYPE_PYOBJ && vtype_rhs == VTYPE_PYOBJ) {
1054 emit_call_with_imm_arg(emit, RT_F_BINARY_OP, rt_binary_op, op, REG_ARG_1);
1055 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
1056 } else {
1057 printf("ViperTypeError: can't do binary op between types %d and %d\n", vtype_lhs, vtype_rhs);
1058 assert(0);
1059 }
1060}
1061
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001062STATIC void emit_native_build_tuple(emit_t *emit, int n_args) {
Damiend2755ec2013-10-16 23:58:48 +01001063 // for viper: call runtime, with types of args
1064 // if wrapped in byte_array, or something, allocates memory and fills it
1065 emit_pre(emit);
Damien Georgecd82e022014-02-02 13:11:48 +00001066 emit_get_stack_pointer_to_reg_for_pop(emit, REG_ARG_2, n_args); // pointer to items
Damiend2755ec2013-10-16 23:58:48 +01001067 emit_call_with_imm_arg(emit, RT_F_BUILD_TUPLE, rt_build_tuple, n_args, REG_ARG_1);
1068 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET); // new tuple
Damien13ed3a62013-10-08 09:05:10 +01001069}
1070
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001071STATIC void emit_native_build_list(emit_t *emit, int n_args) {
Damien13ed3a62013-10-08 09:05:10 +01001072 emit_pre(emit);
Damien Georgecd82e022014-02-02 13:11:48 +00001073 emit_get_stack_pointer_to_reg_for_pop(emit, REG_ARG_2, n_args); // pointer to items
Damien13ed3a62013-10-08 09:05:10 +01001074 emit_call_with_imm_arg(emit, RT_F_BUILD_LIST, rt_build_list, n_args, REG_ARG_1);
1075 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET); // new list
1076}
1077
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001078STATIC void emit_native_list_append(emit_t *emit, int list_index) {
Damiend2755ec2013-10-16 23:58:48 +01001079 // only used in list comprehension
1080 vtype_kind_t vtype_list, vtype_item;
1081 emit_pre_pop_reg(emit, &vtype_item, REG_ARG_2);
1082 emit_access_stack(emit, list_index, &vtype_list, REG_ARG_1);
1083 assert(vtype_list == VTYPE_PYOBJ);
1084 assert(vtype_item == VTYPE_PYOBJ);
1085 emit_call(emit, RT_F_LIST_APPEND, rt_list_append);
1086 emit_post(emit);
Damien13ed3a62013-10-08 09:05:10 +01001087}
1088
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001089STATIC void emit_native_build_map(emit_t *emit, int n_args) {
Damien13ed3a62013-10-08 09:05:10 +01001090 emit_pre(emit);
1091 emit_call_with_imm_arg(emit, RT_F_BUILD_MAP, rt_build_map, n_args, REG_ARG_1);
1092 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET); // new map
1093}
1094
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001095STATIC void emit_native_store_map(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +01001096 vtype_kind_t vtype_key, vtype_value, vtype_map;
1097 emit_pre_pop_reg_reg_reg(emit, &vtype_key, REG_ARG_2, &vtype_value, REG_ARG_3, &vtype_map, REG_ARG_1); // key, value, map
1098 assert(vtype_key == VTYPE_PYOBJ);
1099 assert(vtype_value == VTYPE_PYOBJ);
1100 assert(vtype_map == VTYPE_PYOBJ);
1101 emit_call(emit, RT_F_STORE_MAP, rt_store_map);
1102 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET); // map
1103}
1104
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001105STATIC void emit_native_map_add(emit_t *emit, int map_index) {
Damiend2755ec2013-10-16 23:58:48 +01001106 // only used in list comprehension
1107 vtype_kind_t vtype_map, vtype_key, vtype_value;
1108 emit_pre_pop_reg_reg(emit, &vtype_key, REG_ARG_2, &vtype_value, REG_ARG_3);
1109 emit_access_stack(emit, map_index, &vtype_map, REG_ARG_1);
1110 assert(vtype_map == VTYPE_PYOBJ);
1111 assert(vtype_key == VTYPE_PYOBJ);
1112 assert(vtype_value == VTYPE_PYOBJ);
1113 emit_call(emit, RT_F_STORE_MAP, rt_store_map);
1114 emit_post(emit);
Damien13ed3a62013-10-08 09:05:10 +01001115}
1116
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001117STATIC void emit_native_build_set(emit_t *emit, int n_args) {
Damien13ed3a62013-10-08 09:05:10 +01001118 emit_pre(emit);
Damien Georgecd82e022014-02-02 13:11:48 +00001119 emit_get_stack_pointer_to_reg_for_pop(emit, REG_ARG_2, n_args); // pointer to items
Damien13ed3a62013-10-08 09:05:10 +01001120 emit_call_with_imm_arg(emit, RT_F_BUILD_SET, rt_build_set, n_args, REG_ARG_1);
1121 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET); // new set
1122}
1123
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001124STATIC void emit_native_set_add(emit_t *emit, int set_index) {
Damiend2755ec2013-10-16 23:58:48 +01001125 // only used in set comprehension
1126 vtype_kind_t vtype_set, vtype_item;
1127 emit_pre_pop_reg(emit, &vtype_item, REG_ARG_2);
1128 emit_access_stack(emit, set_index, &vtype_set, REG_ARG_1);
1129 assert(vtype_set == VTYPE_PYOBJ);
1130 assert(vtype_item == VTYPE_PYOBJ);
1131 emit_call(emit, RT_F_STORE_SET, rt_store_set);
1132 emit_post(emit);
Damien13ed3a62013-10-08 09:05:10 +01001133}
Damiend2755ec2013-10-16 23:58:48 +01001134
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001135STATIC void emit_native_build_slice(emit_t *emit, int n_args) {
Damien13ed3a62013-10-08 09:05:10 +01001136 assert(0);
1137}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001138STATIC void emit_native_unpack_sequence(emit_t *emit, int n_args) {
Damien13ed3a62013-10-08 09:05:10 +01001139 // call runtime, needs type decl
1140 assert(0);
1141}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001142STATIC void emit_native_unpack_ex(emit_t *emit, int n_left, int n_right) {
Damien13ed3a62013-10-08 09:05:10 +01001143 assert(0);
1144}
1145
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001146STATIC void emit_native_make_function(emit_t *emit, scope_t *scope, int n_dict_params, int n_default_params) {
Damien13ed3a62013-10-08 09:05:10 +01001147 // call runtime, with type info for args, or don't support dict/default params, or only support Python objects for them
1148 assert(n_default_params == 0 && n_dict_params == 0);
1149 emit_pre(emit);
1150 emit_call_with_imm_arg(emit, RT_F_MAKE_FUNCTION_FROM_ID, rt_make_function_from_id, scope->unique_code_id, REG_ARG_1);
1151 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
1152}
1153
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001154STATIC void emit_native_make_closure(emit_t *emit, scope_t *scope, int n_dict_params, int n_default_params) {
Damien13ed3a62013-10-08 09:05:10 +01001155 assert(0);
1156}
1157
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001158STATIC void emit_native_call_function(emit_t *emit, int n_positional, int n_keyword, bool have_star_arg, bool have_dbl_star_arg) {
Damien13ed3a62013-10-08 09:05:10 +01001159 // call special viper runtime routine with type info for args, and wanted type info for return
1160 assert(n_keyword == 0 && !have_star_arg && !have_dbl_star_arg);
Damien Georgecd82e022014-02-02 13:11:48 +00001161
1162 /* we no longer have these _n specific call_function's
1163 * they anyway push args into an array
1164 * and they would take too much room in the native dispatch table
Damien13ed3a62013-10-08 09:05:10 +01001165 if (n_positional == 0) {
1166 vtype_kind_t vtype_fun;
1167 emit_pre_pop_reg(emit, &vtype_fun, REG_ARG_1); // the function
1168 assert(vtype_fun == VTYPE_PYOBJ);
1169 emit_call(emit, RT_F_CALL_FUNCTION_0, rt_call_function_0);
1170 } else if (n_positional == 1) {
1171 vtype_kind_t vtype_fun, vtype_arg1;
1172 emit_pre_pop_reg_reg(emit, &vtype_arg1, REG_ARG_2, &vtype_fun, REG_ARG_1); // the single argument, the function
1173 assert(vtype_fun == VTYPE_PYOBJ);
1174 assert(vtype_arg1 == VTYPE_PYOBJ);
1175 emit_call(emit, RT_F_CALL_FUNCTION_1, rt_call_function_1);
1176 } else if (n_positional == 2) {
1177 vtype_kind_t vtype_fun, vtype_arg1, vtype_arg2;
1178 emit_pre_pop_reg_reg_reg(emit, &vtype_arg2, REG_ARG_3, &vtype_arg1, REG_ARG_2, &vtype_fun, REG_ARG_1); // the second argument, the first argument, the function
1179 assert(vtype_fun == VTYPE_PYOBJ);
1180 assert(vtype_arg1 == VTYPE_PYOBJ);
1181 assert(vtype_arg2 == VTYPE_PYOBJ);
1182 emit_call(emit, RT_F_CALL_FUNCTION_2, rt_call_function_2);
1183 } else {
Damieneb19efb2013-10-10 22:06:54 +01001184 */
Damien Georgecd82e022014-02-02 13:11:48 +00001185
1186 emit_pre(emit);
1187 if (n_positional != 0) {
1188 emit_get_stack_pointer_to_reg_for_pop(emit, REG_ARG_3, n_positional); // pointer to args
1189 }
1190 vtype_kind_t vtype_fun;
1191 emit_pre_pop_reg(emit, &vtype_fun, REG_ARG_1); // the function
1192 assert(vtype_fun == VTYPE_PYOBJ);
1193 emit_call_with_imm_arg(emit, RT_F_CALL_FUNCTION_N_KW_FOR_NATIVE, rt_call_function_n_kw_for_native, n_positional, REG_ARG_2);
Damien13ed3a62013-10-08 09:05:10 +01001194 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
1195}
1196
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001197STATIC void emit_native_call_method(emit_t *emit, int n_positional, int n_keyword, bool have_star_arg, bool have_dbl_star_arg) {
Damien13ed3a62013-10-08 09:05:10 +01001198 assert(n_keyword == 0 && !have_star_arg && !have_dbl_star_arg);
Damien Georgecd82e022014-02-02 13:11:48 +00001199
Damieneb19efb2013-10-10 22:06:54 +01001200 /*
Damien13ed3a62013-10-08 09:05:10 +01001201 if (n_positional == 0) {
1202 vtype_kind_t vtype_meth, vtype_self;
1203 emit_pre_pop_reg_reg(emit, &vtype_self, REG_ARG_2, &vtype_meth, REG_ARG_1); // the self object (or NULL), the method
1204 assert(vtype_meth == VTYPE_PYOBJ);
1205 assert(vtype_self == VTYPE_PYOBJ);
1206 emit_call(emit, RT_F_CALL_METHOD_1, rt_call_method_1);
1207 } else if (n_positional == 1) {
1208 vtype_kind_t vtype_meth, vtype_self, vtype_arg1;
1209 emit_pre_pop_reg_reg_reg(emit, &vtype_arg1, REG_ARG_3, &vtype_self, REG_ARG_2, &vtype_meth, REG_ARG_1); // the first argument, the self object (or NULL), the method
1210 assert(vtype_meth == VTYPE_PYOBJ);
1211 assert(vtype_self == VTYPE_PYOBJ);
1212 assert(vtype_arg1 == VTYPE_PYOBJ);
1213 emit_call(emit, RT_F_CALL_METHOD_2, rt_call_method_2);
1214 } else {
Damieneb19efb2013-10-10 22:06:54 +01001215 */
Damien Georgecd82e022014-02-02 13:11:48 +00001216
1217 emit_pre(emit);
1218 emit_get_stack_pointer_to_reg_for_pop(emit, REG_ARG_3, n_positional + 2); // pointer to items, including meth and self
1219 emit_call_with_2_imm_args(emit, RT_F_CALL_METHOD_N_KW, rt_call_method_n_kw, n_positional, REG_ARG_1, n_keyword, REG_ARG_2);
Damien13ed3a62013-10-08 09:05:10 +01001220 emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
1221}
1222
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001223STATIC void emit_native_return_value(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +01001224 // easy. since we don't know who we return to, just return the raw value.
1225 // runtime needs then to know our type signature, but I think that's possible.
1226 vtype_kind_t vtype;
1227 emit_pre_pop_reg(emit, &vtype, REG_RET);
1228 if (emit->do_viper_types) {
1229 assert(vtype == VTYPE_PTR_NONE);
1230 } else {
1231 assert(vtype == VTYPE_PYOBJ);
1232 }
1233 emit->last_emit_was_return_value = true;
Damien3ef4abb2013-10-12 16:53:13 +01001234#if N_X64
Damien13ed3a62013-10-08 09:05:10 +01001235 //asm_x64_call_ind(emit->as, 0, REG_RAX); to seg fault for debugging with gdb
1236 asm_x64_exit(emit->as);
Damien3ef4abb2013-10-12 16:53:13 +01001237#elif N_THUMB
Damien13ed3a62013-10-08 09:05:10 +01001238 //asm_thumb_call_ind(emit->as, 0, REG_R0); to seg fault for debugging with gdb
1239 asm_thumb_exit(emit->as);
1240#endif
1241}
1242
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001243STATIC void emit_native_raise_varargs(emit_t *emit, int n_args) {
Damien13ed3a62013-10-08 09:05:10 +01001244 // call runtime
1245 assert(0);
1246}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001247STATIC void emit_native_yield_value(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +01001248 // not supported (for now)
1249 assert(0);
1250}
Paul Sokolovsky520e2f52014-02-12 18:31:30 +02001251STATIC void emit_native_yield_from(emit_t *emit) {
Damien13ed3a62013-10-08 09:05:10 +01001252 // not supported (for now)
1253 assert(0);
1254}
1255
1256const emit_method_table_t EXPORT_FUN(method_table) = {
1257 emit_native_set_viper_types,
1258 emit_native_start_pass,
1259 emit_native_end_pass,
1260 emit_native_last_emit_was_return_value,
1261 emit_native_get_stack_size,
1262 emit_native_set_stack_size,
Damien George08335002014-01-18 23:24:36 +00001263 emit_native_set_source_line,
Damien13ed3a62013-10-08 09:05:10 +01001264
1265 emit_native_load_id,
1266 emit_native_store_id,
1267 emit_native_delete_id,
1268
1269 emit_native_label_assign,
1270 emit_native_import_name,
1271 emit_native_import_from,
1272 emit_native_import_star,
1273 emit_native_load_const_tok,
1274 emit_native_load_const_small_int,
1275 emit_native_load_const_int,
1276 emit_native_load_const_dec,
1277 emit_native_load_const_id,
1278 emit_native_load_const_str,
Damien13ed3a62013-10-08 09:05:10 +01001279 emit_native_load_const_verbatim_str,
Damien13ed3a62013-10-08 09:05:10 +01001280 emit_native_load_fast,
Damien13ed3a62013-10-08 09:05:10 +01001281 emit_native_load_deref,
1282 emit_native_load_closure,
Damien9ecbcff2013-12-11 00:41:43 +00001283 emit_native_load_name,
1284 emit_native_load_global,
Damien13ed3a62013-10-08 09:05:10 +01001285 emit_native_load_attr,
1286 emit_native_load_method,
1287 emit_native_load_build_class,
1288 emit_native_store_fast,
Damien9ecbcff2013-12-11 00:41:43 +00001289 emit_native_store_deref,
Damien13ed3a62013-10-08 09:05:10 +01001290 emit_native_store_name,
1291 emit_native_store_global,
Damien13ed3a62013-10-08 09:05:10 +01001292 emit_native_store_attr,
Damien13ed3a62013-10-08 09:05:10 +01001293 emit_native_store_subscr,
Damiena3977762013-10-09 23:10:10 +01001294 emit_native_store_locals,
Damien13ed3a62013-10-08 09:05:10 +01001295 emit_native_delete_fast,
Damien9ecbcff2013-12-11 00:41:43 +00001296 emit_native_delete_deref,
Damien13ed3a62013-10-08 09:05:10 +01001297 emit_native_delete_name,
1298 emit_native_delete_global,
Damien13ed3a62013-10-08 09:05:10 +01001299 emit_native_delete_attr,
1300 emit_native_delete_subscr,
1301 emit_native_dup_top,
1302 emit_native_dup_top_two,
1303 emit_native_pop_top,
1304 emit_native_rot_two,
1305 emit_native_rot_three,
1306 emit_native_jump,
1307 emit_native_pop_jump_if_true,
1308 emit_native_pop_jump_if_false,
1309 emit_native_jump_if_true_or_pop,
1310 emit_native_jump_if_false_or_pop,
1311 emit_native_setup_loop,
1312 emit_native_break_loop,
1313 emit_native_continue_loop,
1314 emit_native_setup_with,
1315 emit_native_with_cleanup,
1316 emit_native_setup_except,
1317 emit_native_setup_finally,
1318 emit_native_end_finally,
1319 emit_native_get_iter,
1320 emit_native_for_iter,
1321 emit_native_for_iter_end,
1322 emit_native_pop_block,
1323 emit_native_pop_except,
1324 emit_native_unary_op,
1325 emit_native_binary_op,
Damien13ed3a62013-10-08 09:05:10 +01001326 emit_native_build_tuple,
1327 emit_native_build_list,
1328 emit_native_list_append,
1329 emit_native_build_map,
1330 emit_native_store_map,
1331 emit_native_map_add,
1332 emit_native_build_set,
1333 emit_native_set_add,
1334 emit_native_build_slice,
1335 emit_native_unpack_sequence,
1336 emit_native_unpack_ex,
1337 emit_native_make_function,
1338 emit_native_make_closure,
1339 emit_native_call_function,
1340 emit_native_call_method,
1341 emit_native_return_value,
1342 emit_native_raise_varargs,
1343 emit_native_yield_value,
1344 emit_native_yield_from,
1345};
1346
Damien Georgee67ed5d2014-01-04 13:55:24 +00001347#endif // (MICROPY_EMIT_X64 && N_X64) || (MICROPY_EMIT_THUMB && N_THUMB)