emitnative.c 45.8 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
// Essentially normal Python has 1 type: Python objects
// Viper has more than 1 type, and is just a more complicated (a superset of) Python.
// If you declare everything in Viper as a Python object (ie omit type decls) then
// it should in principle be exactly the same as Python native.
// Having types means having more opcodes, like binary_op_nat_nat, binary_op_nat_obj etc.
// In practice we won't have a VM but rather do this in asm which is actually very minimal.

// Because it breaks strict Python equivalence it should be a completely separate
// decorator.  It breaks equivalence because overflow on integers wraps around.
// It shouldn't break equivalence if you don't use the new types, but since the
// type decls might be used in normal Python for other reasons, it's probably safest,
// cleanest and clearest to make it a separate decorator.

// Actually, it does break equivalence because integers default to native integers,
// not Python objects.

// for x in l[0:8]: can be compiled into a native loop if l has pointer type

xbe's avatar
xbe committed
19
#include <stdbool.h>
20
21
22
23
24
25
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <assert.h>

#include "misc.h"
26
#include "mpconfig.h"
27
#include "qstr.h"
28
29
30
#include "lexer.h"
#include "parse.h"
#include "scope.h"
31
#include "runtime0.h"
32
#include "emit.h"
33
#include "emitglue.h"
34
35
#include "obj.h"
#include "runtime.h"
36
37

// wrapper around everything in this file
38
#if (MICROPY_EMIT_X64 && N_X64) || (MICROPY_EMIT_THUMB && N_THUMB)
39

40
#if N_X64
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

// x64 specific stuff

#include "asmx64.h"

#define REG_LOCAL_1 (REG_RBX)
#define REG_LOCAL_NUM (1)

#define EXPORT_FUN(name) emit_native_x64_##name

#define REG_TEMP0 (REG_RAX)
#define REG_TEMP1 (REG_RDI)
#define REG_TEMP2 (REG_RSI)
#define ASM_MOV_REG_TO_LOCAL(reg, local_num) asm_x64_mov_r64_to_local(emit->as, (reg), (local_num))
#define ASM_MOV_IMM_TO_REG(imm, reg) asm_x64_mov_i64_to_r64_optimised(emit->as, (imm), (reg))
56
#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)
57
58
59
60
#define ASM_MOV_LOCAL_TO_REG(local_num, reg) asm_x64_mov_local_to_r64(emit->as, (local_num), (reg))
#define ASM_MOV_REG_TO_REG(reg_src, reg_dest) asm_x64_mov_r64_to_r64(emit->as, (reg_src), (reg_dest))
#define ASM_MOV_LOCAL_ADDR_TO_REG(local_num, reg) asm_x64_mov_local_addr_to_r64(emit->as, (local_num), (reg))

61
#elif N_THUMB
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

// thumb specific stuff

#include "asmthumb.h"

#define REG_LOCAL_1 (REG_R4)
#define REG_LOCAL_2 (REG_R5)
#define REG_LOCAL_3 (REG_R6)
#define REG_LOCAL_NUM (3)

#define EXPORT_FUN(name) emit_native_thumb_##name

#define REG_TEMP0 (REG_R0)
#define REG_TEMP1 (REG_R1)
#define REG_TEMP2 (REG_R2)
#define ASM_MOV_REG_TO_LOCAL(reg, local_num) asm_thumb_mov_local_reg(emit->as, (local_num), (reg))
#define ASM_MOV_IMM_TO_REG(imm, reg) asm_thumb_mov_reg_i32_optimised(emit->as, (reg), (imm))
79
#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)
80
81
82
83
84
85
86
#define ASM_MOV_LOCAL_TO_REG(local_num, reg) asm_thumb_mov_reg_local(emit->as, (reg), (local_num))
#define ASM_MOV_REG_TO_REG(reg_src, reg_dest) asm_thumb_mov_reg_reg(emit->as, (reg_dest), (reg_src))
#define ASM_MOV_LOCAL_ADDR_TO_REG(local_num, reg) asm_thumb_mov_reg_local_addr(emit->as, (reg), (local_num))

#endif

typedef enum {
87
88
89
90
    STACK_VALUE,
    STACK_REG,
    STACK_IMM,
} stack_info_kind_t;
91
92
93
94
95
96
97
98
99
100
101

typedef enum {
    VTYPE_UNBOUND,
    VTYPE_PYOBJ,
    VTYPE_BOOL,
    VTYPE_INT,
    VTYPE_PTR,
    VTYPE_PTR_NONE,
    VTYPE_BUILTIN_V_INT,
} vtype_kind_t;

102
103
104
105
106
107
108
109
110
typedef struct _stack_info_t {
    vtype_kind_t vtype;
    stack_info_kind_t kind;
    union {
        int u_reg;
        machine_int_t u_imm;
    };
} stack_info_t;

111
112
113
114
struct _emit_t {
    int pass;

    bool do_viper_types;
115
116

    int local_vtype_alloc;
117
    vtype_kind_t *local_vtype;
118
119
120
121

    int stack_info_alloc;
    stack_info_t *stack_info;

122
123
124
125
126
127
128
    int stack_start;
    int stack_size;

    bool last_emit_was_return_value;

    scope_t *scope;

129
#if N_X64
130
    asm_x64_t *as;
131
#elif N_THUMB
132
133
134
135
136
137
138
    asm_thumb_t *as;
#endif
};

emit_t *EXPORT_FUN(new)(uint max_num_labels) {
    emit_t *emit = m_new(emit_t, 1);
    emit->do_viper_types = false;
139
140
    emit->local_vtype = NULL;
    emit->stack_info = NULL;
141
#if N_X64
142
    emit->as = asm_x64_new(max_num_labels);
143
#elif N_THUMB
144
145
146
147
148
    emit->as = asm_thumb_new(max_num_labels);
#endif
    return emit;
}

149
void EXPORT_FUN(free)(emit_t *emit) {
150
151
152
153
154
155
156
157
#if N_X64
    asm_x64_free(emit->as, false);
#elif N_THUMB
    asm_thumb_free(emit->as, false);
#endif
    m_del_obj(emit_t, emit);
}

158
STATIC void emit_native_set_viper_types(emit_t *emit, bool do_viper_types) {
159
160
161
    emit->do_viper_types = do_viper_types;
}

162
STATIC void emit_native_start_pass(emit_t *emit, pass_kind_t pass, scope_t *scope) {
163
164
165
166
167
168
    emit->pass = pass;
    emit->stack_start = 0;
    emit->stack_size = 0;
    emit->last_emit_was_return_value = false;
    emit->scope = scope;

169
170
171
172
173
174
175
    if (emit->local_vtype == NULL) {
        emit->local_vtype_alloc = scope->num_locals + 20; // XXX should be maximum over all scopes
        emit->local_vtype = m_new(vtype_kind_t, emit->local_vtype_alloc);
    }
    if (emit->stack_info == NULL) {
        emit->stack_info_alloc = scope->stack_size + 50; // XXX don't know stack size on entry, should be maximum over all scopes
        emit->stack_info = m_new(stack_info_t, emit->stack_info_alloc);
176
177
178
179
    }

    if (emit->do_viper_types) {
        // TODO set types of arguments based on type signature
180
181
182
183
184
185
186
        for (int i = 0; i < emit->local_vtype_alloc; i++) {
            emit->local_vtype[i] = VTYPE_UNBOUND;
        }
        for (int i = 0; i < emit->stack_info_alloc; i++) {
            emit->stack_info[i].kind = STACK_VALUE;
            emit->stack_info[i].vtype = VTYPE_UNBOUND;
        }
187
    } else {
188
189
190
191
192
193
        for (int i = 0; i < emit->local_vtype_alloc; i++) {
            emit->local_vtype[i] = VTYPE_PYOBJ;
        }
        for (int i = 0; i < emit->stack_info_alloc; i++) {
            emit->stack_info[i].kind = STACK_VALUE;
            emit->stack_info[i].vtype = VTYPE_PYOBJ;
194
195
196
        }
    }

197
#if N_X64
198
    asm_x64_start_pass(emit->as, pass);
199
#elif N_THUMB
200
201
202
203
204
205
206
207
208
209
210
211
212
    asm_thumb_start_pass(emit->as, pass);
#endif

    // entry to function
    int num_locals = 0;
    if (pass > PASS_1) {
        num_locals = scope->num_locals - REG_LOCAL_NUM;
        if (num_locals < 0) {
            num_locals = 0;
        }
        emit->stack_start = num_locals;
        num_locals += scope->stack_size;
    }
213
214
215
216
217
    if (pass == PASS_2) {
        // XXX big hack to make sure we have some locals in PASS_2
        // this is so that on PASS_2 the code emitted in x64 has the right size
        num_locals += 2;
    }
218
#if N_X64
219
    asm_x64_entry(emit->as, num_locals);
220
#elif N_THUMB
221
222
223
224
    asm_thumb_entry(emit->as, num_locals);
#endif

    // initialise locals from parameters
225
#if N_X64
226
227
228
229
230
231
232
233
234
235
236
237
    for (int i = 0; i < scope->num_params; i++) {
        if (i == 0) {
            asm_x64_mov_r64_to_r64(emit->as, REG_ARG_1, REG_LOCAL_1);
        } else if (i == 1) {
            asm_x64_mov_r64_to_local(emit->as, REG_ARG_2, i - 1);
        } else if (i == 2) {
            asm_x64_mov_r64_to_local(emit->as, REG_ARG_3, i - 1);
        } else {
            // TODO not implemented
            assert(0);
        }
    }
238
#elif N_THUMB
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
    for (int i = 0; i < scope->num_params; i++) {
        if (i == 0) {
            asm_thumb_mov_reg_reg(emit->as, REG_LOCAL_1, REG_ARG_1);
        } else if (i == 1) {
            asm_thumb_mov_reg_reg(emit->as, REG_LOCAL_2, REG_ARG_2);
        } else if (i == 2) {
            asm_thumb_mov_reg_reg(emit->as, REG_LOCAL_3, REG_ARG_3);
        } else if (i == 3) {
            asm_thumb_mov_local_reg(emit->as, i - REG_LOCAL_NUM, REG_ARG_4);
        } else {
            // TODO not implemented
            assert(0);
        }
    }

Damien George's avatar
Damien George committed
254
    asm_thumb_mov_reg_i32(emit->as, REG_R7, (machine_uint_t)mp_fun_table);
255
256
257
#endif
}

258
STATIC void emit_native_end_pass(emit_t *emit) {
259
#if N_X64
260
261
262
263
    if (!emit->last_emit_was_return_value) {
        asm_x64_exit(emit->as);
    }
    asm_x64_end_pass(emit->as);
264
#elif N_THUMB
265
266
267
268
269
270
271
272
273
274
275
276
    if (!emit->last_emit_was_return_value) {
        asm_thumb_exit(emit->as);
    }
    asm_thumb_end_pass(emit->as);
#endif

    // check stack is back to zero size
    if (emit->stack_size != 0) {
        printf("ERROR: stack size not back to zero; got %d\n", emit->stack_size);
    }

    if (emit->pass == PASS_3) {
277
#if N_X64
278
        void *f = asm_x64_get_code(emit->as);
279
        mp_emit_glue_assign_native_code(emit->scope->unique_code_id, f, asm_x64_get_code_size(emit->as), emit->scope->num_params);
280
#elif N_THUMB
281
        void *f = asm_thumb_get_code(emit->as);
282
        mp_emit_glue_assign_native_code(emit->scope->unique_code_id, f, asm_thumb_get_code_size(emit->as), emit->scope->num_params);
283
284
285
286
#endif
    }
}

287
STATIC bool emit_native_last_emit_was_return_value(emit_t *emit) {
288
289
290
    return emit->last_emit_was_return_value;
}

291
STATIC int emit_native_get_stack_size(emit_t *emit) {
292
293
294
    return emit->stack_size;
}

295
STATIC void emit_native_set_stack_size(emit_t *emit, int size) {
296
297
298
    emit->stack_size = size;
}

299
STATIC void emit_native_set_source_line(emit_t *emit, int source_line) {
300
301
}

302
STATIC void adjust_stack(emit_t *emit, int stack_size_delta) {
303
304
305
306
307
308
309
    emit->stack_size += stack_size_delta;
    assert(emit->stack_size >= 0);
    if (emit->pass > PASS_1 && emit->stack_size > emit->scope->stack_size) {
        emit->scope->stack_size = emit->stack_size;
    }
}

310
/*
311
STATIC void emit_pre_raw(emit_t *emit, int stack_size_delta) {
312
313
314
    adjust_stack(emit, stack_size_delta);
    emit->last_emit_was_return_value = false;
}
315
*/
316

317
// this must be called at start of emit functions
318
STATIC void emit_native_pre(emit_t *emit) {
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
    emit->last_emit_was_return_value = false;
    // settle the stack
    /*
    if (regs_needed != 0) {
        for (int i = 0; i < emit->stack_size; i++) {
            switch (emit->stack_info[i].kind) {
                case STACK_VALUE:
                    break;

                case STACK_REG:
                    // TODO only push reg if in regs_needed
                    emit->stack_info[i].kind = STACK_VALUE;
                    ASM_MOV_REG_TO_LOCAL(emit->stack_info[i].u_reg, emit->stack_start + i);
                    break;

                case STACK_IMM:
                    // don't think we ever need to push imms for settling
                    //ASM_MOV_IMM_TO_LOCAL(emit->last_imm, emit->stack_start + i);
                    break;
            }
        }
    }
    */
342
343
}

344
STATIC vtype_kind_t peek_vtype(emit_t *emit) {
345
346
    return emit->stack_info[emit->stack_size - 1].vtype;
}
347

348
349
// pos=1 is TOS, pos=2 is next, etc
// use pos=0 for no skipping
350
STATIC void need_reg_single(emit_t *emit, int reg_needed, int skip_stack_pos) {
351
352
353
354
355
356
357
358
359
360
361
362
    skip_stack_pos = emit->stack_size - skip_stack_pos;
    for (int i = 0; i < emit->stack_size; i++) {
        if (i != skip_stack_pos) {
            stack_info_t *si = &emit->stack_info[i];
            if (si->kind == STACK_REG && si->u_reg == reg_needed) {
                si->kind = STACK_VALUE;
                ASM_MOV_REG_TO_LOCAL(si->u_reg, emit->stack_start + i);
            }
        }
    }
}

363
STATIC void need_reg_all(emit_t *emit) {
364
365
    for (int i = 0; i < emit->stack_size; i++) {
        stack_info_t *si = &emit->stack_info[i];
366
        if (si->kind == STACK_REG) {
367
368
369
370
371
            si->kind = STACK_VALUE;
            ASM_MOV_REG_TO_LOCAL(si->u_reg, emit->stack_start + i);
        }
    }
}
372

373
STATIC void need_stack_settled(emit_t *emit) {
374
375
376
377
378
379
    for (int i = 0; i < emit->stack_size; i++) {
        stack_info_t *si = &emit->stack_info[i];
        if (si->kind == STACK_REG) {
            si->kind = STACK_VALUE;
            ASM_MOV_REG_TO_LOCAL(si->u_reg, emit->stack_start + i);
        }
380
    }
381
382
383
384
385
386
    for (int i = 0; i < emit->stack_size; i++) {
        stack_info_t *si = &emit->stack_info[i];
        if (si->kind == STACK_IMM) {
            ASM_MOV_IMM_TO_LOCAL_USING(si->u_imm, emit->stack_start + i, REG_TEMP0);
        }
    }
387
388
}

389
// pos=1 is TOS, pos=2 is next, etc
390
STATIC void emit_access_stack(emit_t *emit, int pos, vtype_kind_t *vtype, int reg_dest) {
391
392
    need_reg_single(emit, reg_dest, pos);
    stack_info_t *si = &emit->stack_info[emit->stack_size - pos];
393
394
395
    *vtype = si->vtype;
    switch (si->kind) {
        case STACK_VALUE:
396
            ASM_MOV_LOCAL_TO_REG(emit->stack_start + emit->stack_size - pos, reg_dest);
397
398
            break;

399
400
401
        case STACK_REG:
            if (si->u_reg != reg_dest) {
                ASM_MOV_REG_TO_REG(si->u_reg, reg_dest);
402
403
404
            }
            break;

405
406
        case STACK_IMM:
            ASM_MOV_IMM_TO_REG(si->u_imm, reg_dest);
407
408
409
410
            break;
    }
}

411
STATIC void emit_pre_pop_reg(emit_t *emit, vtype_kind_t *vtype, int reg_dest) {
412
413
414
415
416
    emit->last_emit_was_return_value = false;
    emit_access_stack(emit, 1, vtype, reg_dest);
    adjust_stack(emit, -1);
}

417
STATIC void emit_pre_pop_reg_reg(emit_t *emit, vtype_kind_t *vtypea, int rega, vtype_kind_t *vtypeb, int regb) {
418
    emit_pre_pop_reg(emit, vtypea, rega);
419
    emit_pre_pop_reg(emit, vtypeb, regb);
420
421
}

422
STATIC 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) {
423
    emit_pre_pop_reg(emit, vtypea, rega);
424
425
    emit_pre_pop_reg(emit, vtypeb, regb);
    emit_pre_pop_reg(emit, vtypec, regc);
426
427
}

428
STATIC void emit_post(emit_t *emit) {
429
430
}

431
STATIC void emit_post_push_reg(emit_t *emit, vtype_kind_t vtype, int reg) {
432
433
434
435
436
    stack_info_t *si = &emit->stack_info[emit->stack_size];
    si->vtype = vtype;
    si->kind = STACK_REG;
    si->u_reg = reg;
    adjust_stack(emit, 1);
437
438
}

439
STATIC void emit_post_push_imm(emit_t *emit, vtype_kind_t vtype, machine_int_t imm) {
440
441
442
443
444
    stack_info_t *si = &emit->stack_info[emit->stack_size];
    si->vtype = vtype;
    si->kind = STACK_IMM;
    si->u_imm = imm;
    adjust_stack(emit, 1);
445
446
}

447
STATIC void emit_post_push_reg_reg(emit_t *emit, vtype_kind_t vtypea, int rega, vtype_kind_t vtypeb, int regb) {
448
449
    emit_post_push_reg(emit, vtypea, rega);
    emit_post_push_reg(emit, vtypeb, regb);
450
451
}

452
STATIC 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) {
453
454
455
    emit_post_push_reg(emit, vtypea, rega);
    emit_post_push_reg(emit, vtypeb, regb);
    emit_post_push_reg(emit, vtypec, regc);
456
457
}

458
STATIC 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) {
459
460
461
462
    emit_post_push_reg(emit, vtypea, rega);
    emit_post_push_reg(emit, vtypeb, regb);
    emit_post_push_reg(emit, vtypec, regc);
    emit_post_push_reg(emit, vtyped, regd);
463
464
465
}

// vtype of all n_pop objects is VTYPE_PYOBJ
466
// does not use any temporary registers (but may use reg_dest before loading it with stack pointer)
467
// TODO this needs some thinking for viper code
468
STATIC void emit_get_stack_pointer_to_reg_for_pop(emit_t *emit, int reg_dest, int n_pop) {
469
    need_reg_all(emit);
470
    for (int i = 0; i < n_pop; i++) {
471
472
        stack_info_t *si = &emit->stack_info[emit->stack_size - 1 - i];
        // must push any imm's to stack
473
        // must convert them to VTYPE_PYOBJ for viper code
474
475
        if (si->kind == STACK_IMM) {
            si->kind = STACK_VALUE;
476
477
478
479
480
481
482
            switch (si->vtype) {
                case VTYPE_PYOBJ:
                    ASM_MOV_IMM_TO_LOCAL_USING(si->u_imm, emit->stack_start + emit->stack_size - 1 - i, reg_dest);
                    break;
                case VTYPE_BOOL:
                    si->vtype = VTYPE_PYOBJ;
                    if (si->u_imm == 0) {
483
                        ASM_MOV_IMM_TO_LOCAL_USING((machine_uint_t)mp_const_false, emit->stack_start + emit->stack_size - 1 - i, reg_dest);
484
                    } else {
485
                        ASM_MOV_IMM_TO_LOCAL_USING((machine_uint_t)mp_const_true, emit->stack_start + emit->stack_size - 1 - i, reg_dest);
486
487
488
489
490
491
492
493
494
495
                    }
                    break;
                case VTYPE_INT:
                    si->vtype = VTYPE_PYOBJ;
                    ASM_MOV_IMM_TO_LOCAL_USING((si->u_imm << 1) | 1, emit->stack_start + emit->stack_size - 1 - i, reg_dest);
                    break;
                default:
                    // not handled
                    assert(0);
            }
496
497
498
        }
        assert(si->kind == STACK_VALUE);
        assert(si->vtype == VTYPE_PYOBJ);
499
    }
500
    adjust_stack(emit, -n_pop);
501
    ASM_MOV_LOCAL_ADDR_TO_REG(emit->stack_start + emit->stack_size, reg_dest);
502
503
504
}

// vtype of all n_push objects is VTYPE_PYOBJ
505
STATIC void emit_get_stack_pointer_to_reg_for_push(emit_t *emit, int reg_dest, int n_push) {
506
    need_reg_all(emit);
507
    for (int i = 0; i < n_push; i++) {
508
        emit->stack_info[emit->stack_size + i].kind = STACK_VALUE;
509
        emit->stack_info[emit->stack_size + i].vtype = VTYPE_PYOBJ;
510
    }
511
    ASM_MOV_LOCAL_ADDR_TO_REG(emit->stack_start + emit->stack_size, reg_dest);
512
513
514
    adjust_stack(emit, n_push);
}

Damien George's avatar
Damien George committed
515
STATIC void emit_call(emit_t *emit, mp_fun_kind_t fun_kind, void *fun) {
516
    need_reg_all(emit);
517
#if N_X64
518
    asm_x64_call_ind(emit->as, fun, REG_RAX);
519
#elif N_THUMB
Damien George's avatar
Damien George committed
520
    asm_thumb_bl_ind(emit->as, mp_fun_table[fun_kind], fun_kind, REG_R3);
521
522
523
#endif
}

Damien George's avatar
Damien George committed
524
STATIC void emit_call_with_imm_arg(emit_t *emit, mp_fun_kind_t fun_kind, void *fun, machine_int_t arg_val, int arg_reg) {
525
    need_reg_all(emit);
526
    ASM_MOV_IMM_TO_REG(arg_val, arg_reg);
527
528
529
#if N_X64
    asm_x64_call_ind(emit->as, fun, REG_RAX);
#elif N_THUMB
Damien George's avatar
Damien George committed
530
    asm_thumb_bl_ind(emit->as, mp_fun_table[fun_kind], fun_kind, REG_R3);
531
#endif
532
533
}

Damien George's avatar
Damien George committed
534
STATIC void emit_call_with_2_imm_args(emit_t *emit, mp_fun_kind_t fun_kind, void *fun, machine_int_t arg_val1, int arg_reg1, machine_int_t arg_val2, int arg_reg2) {
535
536
537
538
539
540
    need_reg_all(emit);
    ASM_MOV_IMM_TO_REG(arg_val1, arg_reg1);
    ASM_MOV_IMM_TO_REG(arg_val2, arg_reg2);
#if N_X64
    asm_x64_call_ind(emit->as, fun, REG_RAX);
#elif N_THUMB
Damien George's avatar
Damien George committed
541
    asm_thumb_bl_ind(emit->as, mp_fun_table[fun_kind], fun_kind, REG_R3);
542
543
544
#endif
}

545
STATIC void emit_native_load_id(emit_t *emit, qstr qstr) {
546
547
    // check for built-ins
    if (strcmp(qstr_str(qstr), "v_int") == 0) {
548
        assert(0);
549
        emit_native_pre(emit);
550
551
552
553
554
555
556
557
        //emit_post_push_blank(emit, VTYPE_BUILTIN_V_INT);

    // not a built-in, so do usual thing
    } else {
        emit_common_load_id(emit, &EXPORT_FUN(method_table), emit->scope, qstr);
    }
}

558
STATIC void emit_native_store_id(emit_t *emit, qstr qstr) {
559
560
561
562
    // TODO check for built-ins and disallow
    emit_common_store_id(emit, &EXPORT_FUN(method_table), emit->scope, qstr);
}

563
STATIC void emit_native_delete_id(emit_t *emit, qstr qstr) {
564
565
566
567
    // TODO check for built-ins and disallow
    emit_common_delete_id(emit, &EXPORT_FUN(method_table), emit->scope, qstr);
}

568
STATIC void emit_native_label_assign(emit_t *emit, int l) {
569
    emit_native_pre(emit);
570
571
    // need to commit stack because we can jump here from elsewhere
    need_stack_settled(emit);
572
#if N_X64
573
    asm_x64_label_assign(emit->as, l);
574
#elif N_THUMB
575
576
    asm_thumb_label_assign(emit->as, l);
#endif
577
    emit_post(emit);
578
579
}

580
STATIC void emit_native_import_name(emit_t *emit, qstr qstr) {
581
582
583
584
    // not implemented
    assert(0);
}

585
STATIC void emit_native_import_from(emit_t *emit, qstr qstr) {
586
587
588
589
    // not implemented
    assert(0);
}

590
STATIC void emit_native_import_star(emit_t *emit) {
591
592
593
594
    // not implemented
    assert(0);
}

595
STATIC void emit_native_load_const_tok(emit_t *emit, mp_token_kind_t tok) {
596
    emit_native_pre(emit);
597
598
599
600
    int vtype;
    machine_uint_t val;
    if (emit->do_viper_types) {
        switch (tok) {
601
602
603
            case MP_TOKEN_KW_NONE: vtype = VTYPE_PTR_NONE; val = 0; break;
            case MP_TOKEN_KW_FALSE: vtype = VTYPE_BOOL; val = 0; break;
            case MP_TOKEN_KW_TRUE: vtype = VTYPE_BOOL; val = 1; break;
604
605
606
607
608
            default: assert(0); vtype = 0; val = 0; // shouldn't happen
        }
    } else {
        vtype = VTYPE_PYOBJ;
        switch (tok) {
609
610
611
            case MP_TOKEN_KW_NONE: val = (machine_uint_t)mp_const_none; break;
            case MP_TOKEN_KW_FALSE: val = (machine_uint_t)mp_const_false; break;
            case MP_TOKEN_KW_TRUE: val = (machine_uint_t)mp_const_true; break;
612
613
614
615
616
617
            default: assert(0); vtype = 0; val = 0; // shouldn't happen
        }
    }
    emit_post_push_imm(emit, vtype, val);
}

618
STATIC void emit_native_load_const_small_int(emit_t *emit, machine_int_t arg) {
619
    emit_native_pre(emit);
620
621
622
623
624
625
626
    if (emit->do_viper_types) {
        emit_post_push_imm(emit, VTYPE_INT, arg);
    } else {
        emit_post_push_imm(emit, VTYPE_PYOBJ, (arg << 1) | 1);
    }
}

627
STATIC void emit_native_load_const_int(emit_t *emit, qstr qstr) {
628
629
630
631
632
    // not implemented
    // load integer, check fits in 32 bits
    assert(0);
}

633
STATIC void emit_native_load_const_dec(emit_t *emit, qstr qstr) {
634
    // for viper, a float/complex is just a Python object
635
    emit_native_pre(emit);
Damien George's avatar
Damien George committed
636
    emit_call_with_imm_arg(emit, MP_F_LOAD_CONST_DEC, mp_load_const_dec, qstr, REG_ARG_1);
637
    emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
638
639
}

640
STATIC void emit_native_load_const_id(emit_t *emit, qstr qstr) {
641
    emit_native_pre(emit);
642
643
644
    if (emit->do_viper_types) {
        assert(0);
    } else {
Damien George's avatar
Damien George committed
645
        emit_call_with_imm_arg(emit, MP_F_LOAD_CONST_STR, mp_load_const_str, qstr, REG_ARG_1); // TODO
646
647
        emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
    }
648
649
}

650
STATIC void emit_native_load_const_str(emit_t *emit, qstr qstr, bool bytes) {
651
    emit_native_pre(emit);
652
653
654
655
656
657
    if (emit->do_viper_types) {
        // not implemented properly
        // load a pointer to the asciiz string?
        assert(0);
        emit_post_push_imm(emit, VTYPE_PTR, (machine_uint_t)qstr_str(qstr));
    } else {
Damien George's avatar
Damien George committed
658
        emit_call_with_imm_arg(emit, MP_F_LOAD_CONST_STR, mp_load_const_str, qstr, REG_ARG_1);
659
660
661
662
        emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
    }
}

663
STATIC void emit_native_load_const_verbatim_str(emit_t *emit, const char *str) {
664
665
666
667
    // not supported/needed for viper
    assert(0);
}

668
STATIC void emit_native_load_fast(emit_t *emit, qstr qstr, int local_num) {
669
670
671
672
    vtype_kind_t vtype = emit->local_vtype[local_num];
    if (vtype == VTYPE_UNBOUND) {
        printf("ViperTypeError: local %s used before type known\n", qstr_str(qstr));
    }
673
    emit_native_pre(emit);
674
#if N_X64
675
676
677
    if (local_num == 0) {
        emit_post_push_reg(emit, vtype, REG_LOCAL_1);
    } else {
678
        need_reg_single(emit, REG_RAX, 0);
679
680
681
        asm_x64_mov_local_to_r64(emit->as, local_num - 1, REG_RAX);
        emit_post_push_reg(emit, vtype, REG_RAX);
    }
682
#elif N_THUMB
683
684
685
686
687
688
689
    if (local_num == 0) {
        emit_post_push_reg(emit, vtype, REG_LOCAL_1);
    } else if (local_num == 1) {
        emit_post_push_reg(emit, vtype, REG_LOCAL_2);
    } else if (local_num == 2) {
        emit_post_push_reg(emit, vtype, REG_LOCAL_3);
    } else {
690
        need_reg_single(emit, REG_R0, 0);
691
692
693
694
695
696
        asm_thumb_mov_reg_local(emit->as, REG_R0, local_num - 1);
        emit_post_push_reg(emit, vtype, REG_R0);
    }
#endif
}

697
STATIC void emit_native_load_deref(emit_t *emit, qstr qstr, int local_num) {
Damien's avatar
Damien committed
698
699
700
701
702
    // not implemented
    // in principle could support this quite easily (ldr r0, [r0, #0]) and then get closed over variables!
    assert(0);
}

703
STATIC void emit_native_load_closure(emit_t *emit, qstr qstr, int local_num) {
Damien's avatar
Damien committed
704
705
706
707
    // not implemented
    assert(0);
}

708
STATIC void emit_native_load_name(emit_t *emit, qstr qstr) {
709
    emit_native_pre(emit);
Damien George's avatar
Damien George committed
710
    emit_call_with_imm_arg(emit, MP_F_LOAD_NAME, mp_load_name, qstr, REG_ARG_1);
711
712
713
    emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
}

714
STATIC void emit_native_load_global(emit_t *emit, qstr qstr) {
715
    emit_native_pre(emit);
Damien George's avatar
Damien George committed
716
    emit_call_with_imm_arg(emit, MP_F_LOAD_GLOBAL, mp_load_global, qstr, REG_ARG_1);
717
718
719
    emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
}

720
STATIC void emit_native_load_attr(emit_t *emit, qstr qstr) {
721
722
723
    // depends on type of subject:
    //  - integer, function, pointer to integers: error
    //  - pointer to structure: get member, quite easy
Damien George's avatar
Damien George committed
724
    //  - Python object: call mp_load_attr, and needs to be typed to convert result
725
726
727
    vtype_kind_t vtype_base;
    emit_pre_pop_reg(emit, &vtype_base, REG_ARG_1); // arg1 = base
    assert(vtype_base == VTYPE_PYOBJ);
Damien George's avatar
Damien George committed
728
    emit_call_with_imm_arg(emit, MP_F_LOAD_ATTR, mp_load_attr, qstr, REG_ARG_2); // arg2 = attribute name
729
730
731
    emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
}

732
STATIC void emit_native_load_method(emit_t *emit, qstr qstr) {
733
734
735
736
    vtype_kind_t vtype_base;
    emit_pre_pop_reg(emit, &vtype_base, REG_ARG_1); // arg1 = base
    assert(vtype_base == VTYPE_PYOBJ);
    emit_get_stack_pointer_to_reg_for_push(emit, REG_ARG_3, 2); // arg3 = dest ptr
Damien George's avatar
Damien George committed
737
    emit_call_with_imm_arg(emit, MP_F_LOAD_METHOD, mp_load_method, qstr, REG_ARG_2); // arg2 = method name
738
739
}

740
STATIC void emit_native_load_build_class(emit_t *emit) {
741
    emit_native_pre(emit);
Damien George's avatar
Damien George committed
742
    emit_call(emit, MP_F_LOAD_BUILD_CLASS, mp_load_build_class);
743
    emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
744
745
}

746
STATIC void emit_native_store_fast(emit_t *emit, qstr qstr, int local_num) {
747
    vtype_kind_t vtype;
748
#if N_X64
749
750
751
752
753
754
    if (local_num == 0) {
        emit_pre_pop_reg(emit, &vtype, REG_LOCAL_1);
    } else {
        emit_pre_pop_reg(emit, &vtype, REG_RAX);
        asm_x64_mov_r64_to_local(emit->as, REG_RAX, local_num - 1);
    }
755
#elif N_THUMB
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
    if (local_num == 0) {
        emit_pre_pop_reg(emit, &vtype, REG_LOCAL_1);
    } else if (local_num == 1) {
        emit_pre_pop_reg(emit, &vtype, REG_LOCAL_2);
    } else if (local_num == 2) {
        emit_pre_pop_reg(emit, &vtype, REG_LOCAL_3);
    } else {
        emit_pre_pop_reg(emit, &vtype, REG_R0);
        asm_thumb_mov_local_reg(emit->as, local_num - 1, REG_R0);
    }
#endif

    emit_post(emit);

    // check types
    if (emit->local_vtype[local_num] == VTYPE_UNBOUND) {
        // first time this local is assigned, so give it a type of the object stored in it
        emit->local_vtype[local_num] = vtype;
    } else if (emit->local_vtype[local_num] != vtype) {
        // type of local is not the same as object stored in it
        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);
    }
}

780
STATIC void emit_native_store_deref(emit_t *emit, qstr qstr, int local_num) {
Damien's avatar
Damien committed
781
782
783
784
    // not implemented
    assert(0);
}

785
STATIC void emit_native_store_name(emit_t *emit, qstr qstr) {
Damien George's avatar
Damien George committed
786
    // mp_store_name, but needs conversion of object (maybe have mp_viper_store_name(obj, type))
787
788
789
    vtype_kind_t vtype;
    emit_pre_pop_reg(emit, &vtype, REG_ARG_2);
    assert(vtype == VTYPE_PYOBJ);
Damien George's avatar
Damien George committed
790
    emit_call_with_imm_arg(emit, MP_F_STORE_NAME, mp_store_name, qstr, REG_ARG_1); // arg1 = name
791
792
793
    emit_post(emit);
}

794
STATIC void emit_native_store_global(emit_t *emit, qstr qstr) {
795
796
797
798
    // not implemented
    assert(0);
}

799
STATIC void emit_native_store_attr(emit_t *emit, qstr qstr) {
800
801
802
803
    vtype_kind_t vtype_base, vtype_val;
    emit_pre_pop_reg_reg(emit, &vtype_base, REG_ARG_1, &vtype_val, REG_ARG_3); // arg1 = base, arg3 = value
    assert(vtype_base == VTYPE_PYOBJ);
    assert(vtype_val == VTYPE_PYOBJ);
Damien George's avatar
Damien George committed
804
    emit_call_with_imm_arg(emit, MP_F_STORE_ATTR, mp_store_attr, qstr, REG_ARG_2); // arg2 = attribute name
805
    emit_post(emit);
806
807
}

808
STATIC void emit_native_store_subscr(emit_t *emit) {
809
810
811
812
813
814
815
816
817
    // depends on type of subject:
    //  - integer, function, pointer to structure: error
    //  - pointer to integers: store as per array
    //  - Python object: call runtime with converted object or type info
    vtype_kind_t vtype_index, vtype_base, vtype_value;
    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
    assert(vtype_index == VTYPE_PYOBJ);
    assert(vtype_base == VTYPE_PYOBJ);
    assert(vtype_value == VTYPE_PYOBJ);
Damien George's avatar
Damien George committed
818
    emit_call(emit, MP_F_STORE_SUBSCR, mp_store_subscr);
819
820
}

821
STATIC void emit_native_delete_fast(emit_t *emit, qstr qstr, int local_num) {
822
823
824
825
826
    // not implemented
    // could support for Python types, just set to None (so GC can reclaim it)
    assert(0);
}

827
STATIC void emit_native_delete_deref(emit_t *emit, qstr qstr, int local_num) {
Damien's avatar
Damien committed
828
829
830
831
    // not supported
    assert(0);
}

832
STATIC void emit_native_delete_name(emit_t *emit, qstr qstr) {
833
    // not implemented
Damien George's avatar
Damien George committed
834
    // use mp_delete_name
835
836
837
    assert(0);
}

838
STATIC void emit_native_delete_global(emit_t *emit, qstr qstr) {
839
    // not implemented
Damien George's avatar
Damien George committed
840
    // use mp_delete_global
841
842
843
    assert(0);
}

844
STATIC void emit_native_delete_attr(emit_t *emit, qstr qstr) {
845
846
847
848
    // not supported
    assert(0);
}

849
STATIC void emit_native_delete_subscr(emit_t *emit) {
850
851
852
853
    // not supported
    assert(0);
}

854
STATIC void emit_native_dup_top(emit_t *emit) {
855
856
857
858
859
    vtype_kind_t vtype;
    emit_pre_pop_reg(emit, &vtype, REG_TEMP0);
    emit_post_push_reg_reg(emit, vtype, REG_TEMP0, vtype, REG_TEMP0);
}

860
STATIC void emit_native_dup_top_two(emit_t *emit) {
861
862
863
864
865
    vtype_kind_t vtype0, vtype1;
    emit_pre_pop_reg_reg(emit, &vtype0, REG_TEMP0, &vtype1, REG_TEMP1);
    emit_post_push_reg_reg_reg_reg(emit, vtype1, REG_TEMP1, vtype0, REG_TEMP0, vtype1, REG_TEMP1, vtype0, REG_TEMP0);
}

866
STATIC void emit_native_pop_top(emit_t *emit) {
867
868
869
870
871
    vtype_kind_t vtype;
    emit_pre_pop_reg(emit, &vtype, REG_TEMP0);
    emit_post(emit);
}

872
STATIC void emit_native_rot_two(emit_t *emit) {
873
874
875
    vtype_kind_t vtype0, vtype1;
    emit_pre_pop_reg_reg(emit, &vtype0, REG_TEMP0, &vtype1, REG_TEMP1);
    emit_post_push_reg_reg(emit, vtype0, REG_TEMP0, vtype1, REG_TEMP1);
876
877
}

878
STATIC void emit_native_rot_three(emit_t *emit) {
879
880
881
882
883
    vtype_kind_t vtype0, vtype1, vtype2;
    emit_pre_pop_reg_reg_reg(emit, &vtype0, REG_TEMP0, &vtype1, REG_TEMP1, &vtype2, REG_TEMP2);
    emit_post_push_reg_reg_reg(emit, vtype0, REG_TEMP0, vtype2, REG_TEMP2, vtype1, REG_TEMP1);
}

884
STATIC void emit_native_jump(emit_t *emit, int label) {
885
    emit_native_pre(emit);
886
#if N_X64
887
    asm_x64_jmp_label(emit->as, label);
888
#elif N_THUMB
889
890
891
892
893
    asm_thumb_b_label(emit->as, label);
#endif
    emit_post(emit);
}

894
STATIC void emit_native_pop_jump_pre_helper(emit_t *emit, int label) {
895
896
897
898
899
    vtype_kind_t vtype = peek_vtype(emit);
    if (vtype == VTYPE_BOOL) {
        emit_pre_pop_reg(emit, &vtype, REG_RET);
    } else if (vtype == VTYPE_PYOBJ) {
        emit_pre_pop_reg(emit, &vtype, REG_ARG_1);
Damien George's avatar
Damien George committed
900
        emit_call(emit, MP_F_OBJ_IS_TRUE, mp_obj_is_true);
901
902
903
904
    } else {
        printf("ViperTypeError: expecting a bool or pyobj, got %d\n", vtype);
        assert(0);
    }
905
906
}

907
STATIC void emit_native_pop_jump_if_false(emit_t *emit, int label) {
908
    emit_native_pop_jump_pre_helper(emit, label);
909
#if N_X64
910
911
    asm_x64_test_r8_with_r8(emit->as, REG_RET, REG_RET);
    asm_x64_jcc_label(emit->as, JCC_JZ, label);
912
#elif N_THUMB
913
914
    asm_thumb_cmp_rlo_i8(emit->as, REG_RET, 0);
    asm_thumb_bcc_label(emit->as, THUMB_CC_EQ, label);
915
916
917
918
#endif
    emit_post(emit);
}

919
STATIC void emit_native_pop_jump_if_true(emit_t *emit, int label) {
920
921
922
923
924
925
926
927
928
    emit_native_pop_jump_pre_helper(emit, label);
#if N_X64
    asm_x64_test_r8_with_r8(emit->as, REG_RET, REG_RET);
    asm_x64_jcc_label(emit->as, JCC_JNZ, label);
#elif N_THUMB
    asm_thumb_cmp_rlo_i8(emit->as, REG_RET, 0);
    asm_thumb_bcc_label(emit->as, THUMB_CC_NE, label);
#endif
    emit_post(emit);
929
}
930

931
STATIC void emit_native_jump_if_true_or_pop(emit_t *emit, int label) {
932
933
    assert(0);
}
934
STATIC void emit_native_jump_if_false_or_pop(emit_t *emit, int label) {
935
936
937
    assert(0);
}

938
STATIC void emit_native_setup_loop(emit_t *emit, int label) {
939
    emit_native_pre(emit);
940
941
942
    emit_post(emit);
}

943
STATIC void emit_native_break_loop(emit_t *emit, int label, int except_depth) {
944
    emit_native_jump(emit, label); // TODO properly
945
}
946
STATIC void emit_native_continue_loop(emit_t *emit, int label, int except_depth) {
947
948
    assert(0);
}
949
STATIC void emit_native_setup_with(emit_t *emit, int label) {
950
951
952
    // not supported, or could be with runtime call
    assert(0);
}
953
STATIC void emit_native_with_cleanup(emit_t *emit) {
954
955
    assert(0);
}
956
STATIC void emit_native_setup_except(emit_t *emit, int label) {
957
958
    assert(0);
}
959
STATIC void emit_native_setup_finally(emit_t *emit, int label) {
960
961
    assert(0);
}
962
STATIC void emit_native_end_finally(emit_t *emit) {
963
964
    assert(0);
}
965

966
STATIC void emit_native_get_iter(emit_t *emit) {
967
968
    // perhaps the difficult one, as we want to rewrite for loops using native code
    // in cases where we iterate over a Python object, can we use normal runtime calls?
969
970
971
972

    vtype_kind_t vtype;
    emit_pre_pop_reg(emit, &vtype, REG_ARG_1);
    assert(vtype == VTYPE_PYOBJ);
Damien George's avatar
Damien George committed
973
    emit_call(emit, MP_F_GETITER, mp_getiter);
974
975
976
    emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
}

977
STATIC void emit_native_for_iter(emit_t *emit, int label) {
978
    emit_native_pre(emit);
979
980
981
    vtype_kind_t vtype;
    emit_access_stack(emit, 1, &vtype, REG_ARG_1);
    assert(vtype == VTYPE_PYOBJ);
Damien George's avatar
Damien George committed
982
    emit_call(emit, MP_F_ITERNEXT, mp_iternext);
983
    ASM_MOV_IMM_TO_REG((machine_uint_t)MP_OBJ_NULL, REG_TEMP1);
984
985
986
987
988
#if N_X64
    asm_x64_cmp_r64_with_r64(emit->as, REG_RET, REG_TEMP1);
    asm_x64_jcc_label(emit->as, JCC_JE, label);
#elif N_THUMB
    asm_thumb_cmp_reg_reg(emit->as, REG_RET, REG_TEMP1);
Damien's avatar
Damien committed
989
    asm_thumb_bcc_label(emit->as, THUMB_CC_EQ, label);
990
991
#endif
    emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
992
}
993

994
STATIC void emit_native_for_iter_end(emit_t *emit) {
995
    // adjust stack counter (we get here from for_iter ending, which popped the value for us)
996
    emit_native_pre(emit);
997
998
    adjust_stack(emit, -1);
    emit_post(emit);
999
1000
}

For faster browsing, not all history is shown. View entire blame