2014-05-03 18:27:38 -04:00
/*
* This file is part of the Micro Python project , http : //micropython.org/
*
* The MIT License ( MIT )
*
* Copyright ( c ) 2013 , 2014 Damien P . George
*
* Permission is hereby granted , free of charge , to any person obtaining a copy
* of this software and associated documentation files ( the " Software " ) , to deal
* in the Software without restriction , including without limitation the rights
* to use , copy , modify , merge , publish , distribute , sublicense , and / or sell
* copies of the Software , and to permit persons to whom the Software is
* furnished to do so , subject to the following conditions :
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software .
*
* THE SOFTWARE IS PROVIDED " AS IS " , WITHOUT WARRANTY OF ANY KIND , EXPRESS OR
* IMPLIED , INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY ,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT . IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM , DAMAGES OR OTHER
* LIABILITY , WHETHER IN AN ACTION OF CONTRACT , TORT OR OTHERWISE , ARISING FROM ,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE .
*/
2013-10-08 04:05:10 -04:00
// 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
2014-03-16 03:14:26 -04:00
# include <stdbool.h>
2013-10-08 04:05:10 -04:00
# include <stdint.h>
# include <stdio.h>
# include <string.h>
# include <assert.h>
2013-12-21 13:17:45 -05:00
# include "mpconfig.h"
2014-06-30 00:17:25 -04:00
# include "nlr.h"
2014-06-21 15:43:22 -04:00
# include "misc.h"
2014-01-21 16:40:13 -05:00
# include "qstr.h"
2013-10-08 04:05:10 -04:00
# include "lexer.h"
# include "parse.h"
2014-04-13 06:04:33 -04:00
# include "obj.h"
# include "emitglue.h"
2013-10-08 04:05:10 -04:00
# include "scope.h"
2013-12-21 13:17:45 -05:00
# include "runtime0.h"
2013-10-08 04:05:10 -04:00
# include "emit.h"
2013-12-21 13:17:45 -05:00
# include "runtime.h"
2013-10-08 04:05:10 -04:00
2014-04-06 07:58:40 -04:00
#if 0 // print debugging info
# define DEBUG_PRINT (1)
# define DEBUG_printf DEBUG_printf
# else // don't print debugging info
# define DEBUG_printf(...) (void)0
# endif
2013-10-08 04:05:10 -04:00
// wrapper around everything in this file
2014-09-06 18:06:36 -04:00
# if (MICROPY_EMIT_X64 && N_X64) \
| | ( MICROPY_EMIT_X86 & & N_X86 ) \
| | ( MICROPY_EMIT_THUMB & & N_THUMB ) \
| | ( MICROPY_EMIT_ARM & & N_ARM )
2013-10-08 04:05:10 -04:00
2013-10-12 11:53:13 -04:00
# if N_X64
2013-10-08 04:05:10 -04:00
// x64 specific stuff
# include "asmx64.h"
# define EXPORT_FUN(name) emit_native_x64_##name
2014-09-29 11:25:04 -04:00
# define REG_RET ASM_X64_REG_RAX
# define REG_ARG_1 ASM_X64_REG_RDI
# define REG_ARG_2 ASM_X64_REG_RSI
# define REG_ARG_3 ASM_X64_REG_RDX
# define REG_ARG_4 ASM_X64_REG_RCX
2014-09-06 20:06:19 -04:00
// caller-save
2014-09-29 11:25:04 -04:00
# define REG_TEMP0 ASM_X64_REG_RAX
# define REG_TEMP1 ASM_X64_REG_RDI
# define REG_TEMP2 ASM_X64_REG_RSI
2014-09-06 20:06:19 -04:00
// callee-save
2014-09-29 11:25:04 -04:00
# define REG_LOCAL_1 ASM_X64_REG_RBX
# define REG_LOCAL_2 ASM_X64_REG_R12
# define REG_LOCAL_3 ASM_X64_REG_R13
2014-09-06 20:06:19 -04:00
# define REG_LOCAL_NUM (3)
2014-09-06 18:06:36 -04:00
# define ASM_PASS_COMPUTE ASM_X64_PASS_COMPUTE
# define ASM_PASS_EMIT ASM_X64_PASS_EMIT
# define ASM_T asm_x64_t
# define ASM_NEW asm_x64_new
# define ASM_FREE asm_x64_free
# define ASM_GET_CODE asm_x64_get_code
# define ASM_GET_CODE_SIZE asm_x64_get_code_size
# define ASM_START_PASS asm_x64_start_pass
# define ASM_END_PASS asm_x64_end_pass
# define ASM_ENTRY asm_x64_entry
# define ASM_EXIT asm_x64_exit
# define ASM_LABEL_ASSIGN asm_x64_label_assign
# define ASM_JUMP asm_x64_jmp_label
# define ASM_JUMP_IF_REG_ZERO(as, reg, label) \
do { \
asm_x64_test_r8_with_r8 ( as , reg , reg ) ; \
asm_x64_jcc_label ( as , ASM_X64_CC_JZ , label ) ; \
} while ( 0 )
# define ASM_JUMP_IF_REG_NONZERO(as, reg, label) \
do { \
asm_x64_test_r8_with_r8 ( as , reg , reg ) ; \
asm_x64_jcc_label ( as , ASM_X64_CC_JNZ , label ) ; \
} while ( 0 )
# define ASM_JUMP_IF_REG_EQ(as, reg1, reg2, label) \
do { \
asm_x64_cmp_r64_with_r64 ( as , reg1 , reg2 ) ; \
asm_x64_jcc_label ( as , ASM_X64_CC_JE , label ) ; \
} while ( 0 )
2014-09-29 11:25:04 -04:00
# define ASM_CALL_IND(as, ptr, idx) asm_x64_call_ind(as, ptr, ASM_X64_REG_RAX)
2014-09-06 18:06:36 -04:00
# define ASM_MOV_REG_TO_LOCAL asm_x64_mov_r64_to_local
# define ASM_MOV_IMM_TO_REG asm_x64_mov_i64_to_r64_optimised
# define ASM_MOV_ALIGNED_IMM_TO_REG asm_x64_mov_i64_to_r64_aligned
# define ASM_MOV_IMM_TO_LOCAL_USING(as, imm, local_num, reg_temp) \
do { \
asm_x64_mov_i64_to_r64_optimised ( as , ( imm ) , ( reg_temp ) ) ; \
asm_x64_mov_r64_to_local ( as , ( reg_temp ) , ( local_num ) ) ; \
} while ( false )
# define ASM_MOV_LOCAL_TO_REG asm_x64_mov_local_to_r64
2014-09-29 13:45:42 -04:00
# define ASM_MOV_REG_REG(as, reg_dest, reg_src) asm_x64_mov_r64_r64((as), (reg_dest), (reg_src))
2014-09-06 18:06:36 -04:00
# define ASM_MOV_LOCAL_ADDR_TO_REG asm_x64_mov_local_addr_to_r64
2014-09-29 13:45:42 -04:00
# define ASM_LSL_REG(as, reg) asm_x64_shl_r64_cl((as), (reg))
# define ASM_ASR_REG(as, reg) asm_x64_sar_r64_cl((as), (reg))
# define ASM_ADD_REG_REG(as, reg_dest, reg_src) asm_x64_add_r64_r64((as), (reg_dest), (reg_src))
# define ASM_SUB_REG_REG(as, reg_dest, reg_src) asm_x64_sub_r64_r64((as), (reg_dest), (reg_src))
2014-09-06 18:06:36 -04:00
# elif N_X86
// x86 specific stuff
# include "asmx86.h"
STATIC byte mp_f_n_args [ MP_F_NUMBER_OF ] = {
[ MP_F_CONVERT_OBJ_TO_NATIVE ] = 2 ,
[ MP_F_CONVERT_NATIVE_TO_OBJ ] = 2 ,
[ MP_F_LOAD_CONST_INT ] = 1 ,
[ MP_F_LOAD_CONST_DEC ] = 1 ,
[ MP_F_LOAD_CONST_STR ] = 1 ,
[ MP_F_LOAD_CONST_BYTES ] = 1 ,
[ MP_F_LOAD_NAME ] = 1 ,
[ MP_F_LOAD_GLOBAL ] = 1 ,
[ MP_F_LOAD_BUILD_CLASS ] = 0 ,
[ MP_F_LOAD_ATTR ] = 2 ,
[ MP_F_LOAD_METHOD ] = 3 ,
[ MP_F_STORE_NAME ] = 2 ,
[ MP_F_STORE_GLOBAL ] = 2 ,
[ MP_F_STORE_ATTR ] = 3 ,
[ MP_F_OBJ_SUBSCR ] = 3 ,
[ MP_F_OBJ_IS_TRUE ] = 1 ,
[ MP_F_UNARY_OP ] = 2 ,
[ MP_F_BINARY_OP ] = 3 ,
[ MP_F_BUILD_TUPLE ] = 2 ,
[ MP_F_BUILD_LIST ] = 2 ,
[ MP_F_LIST_APPEND ] = 2 ,
[ MP_F_BUILD_MAP ] = 1 ,
[ MP_F_STORE_MAP ] = 3 ,
# if MICROPY_PY_BUILTINS_SET
[ MP_F_BUILD_SET ] = 2 ,
[ MP_F_STORE_SET ] = 2 ,
# endif
[ MP_F_MAKE_FUNCTION_FROM_RAW_CODE ] = 3 ,
[ MP_F_NATIVE_CALL_FUNCTION_N_KW ] = 3 ,
[ MP_F_CALL_METHOD_N_KW ] = 3 ,
[ MP_F_GETITER ] = 1 ,
[ MP_F_ITERNEXT ] = 1 ,
[ MP_F_NLR_PUSH ] = 1 ,
[ MP_F_NLR_POP ] = 0 ,
[ MP_F_NATIVE_RAISE ] = 1 ,
[ MP_F_IMPORT_NAME ] = 3 ,
[ MP_F_IMPORT_FROM ] = 2 ,
[ MP_F_IMPORT_ALL ] = 1 ,
# if MICROPY_PY_BUILTINS_SLICE
[ MP_F_NEW_SLICE ] = 3 ,
# endif
[ MP_F_UNPACK_SEQUENCE ] = 3 ,
[ MP_F_UNPACK_EX ] = 3 ,
[ MP_F_DELETE_NAME ] = 1 ,
[ MP_F_DELETE_GLOBAL ] = 1 ,
} ;
# define EXPORT_FUN(name) emit_native_x86_##name
2014-09-29 11:25:04 -04:00
# define REG_RET ASM_X86_REG_EAX
2014-09-08 18:16:35 -04:00
# define REG_ARG_1 ASM_X86_REG_ARG_1
# define REG_ARG_2 ASM_X86_REG_ARG_2
# define REG_ARG_3 ASM_X86_REG_ARG_3
2014-09-06 20:06:19 -04:00
2014-09-06 19:24:32 -04:00
// caller-save, so can be used as temporaries
2014-09-29 11:25:04 -04:00
# define REG_TEMP0 ASM_X86_REG_EAX
# define REG_TEMP1 ASM_X86_REG_ECX
# define REG_TEMP2 ASM_X86_REG_EDX
2014-09-06 18:06:36 -04:00
2014-09-06 19:24:32 -04:00
// callee-save, so can be used as locals
2014-09-29 11:25:04 -04:00
# define REG_LOCAL_1 ASM_X86_REG_EBX
# define REG_LOCAL_2 ASM_X86_REG_ESI
# define REG_LOCAL_3 ASM_X86_REG_EDI
2014-09-06 19:24:32 -04:00
# define REG_LOCAL_NUM (3)
2014-09-06 18:06:36 -04:00
# define ASM_PASS_COMPUTE ASM_X86_PASS_COMPUTE
# define ASM_PASS_EMIT ASM_X86_PASS_EMIT
# define ASM_T asm_x86_t
# define ASM_NEW asm_x86_new
# define ASM_FREE asm_x86_free
# define ASM_GET_CODE asm_x86_get_code
# define ASM_GET_CODE_SIZE asm_x86_get_code_size
# define ASM_START_PASS asm_x86_start_pass
# define ASM_END_PASS asm_x86_end_pass
# define ASM_ENTRY asm_x86_entry
# define ASM_EXIT asm_x86_exit
# define ASM_LABEL_ASSIGN asm_x86_label_assign
# define ASM_JUMP asm_x86_jmp_label
# define ASM_JUMP_IF_REG_ZERO(as, reg, label) \
do { \
asm_x86_test_r8_with_r8 ( as , reg , reg ) ; \
asm_x86_jcc_label ( as , ASM_X86_CC_JZ , label ) ; \
} while ( 0 )
# define ASM_JUMP_IF_REG_NONZERO(as, reg, label) \
do { \
asm_x86_test_r8_with_r8 ( as , reg , reg ) ; \
asm_x86_jcc_label ( as , ASM_X86_CC_JNZ , label ) ; \
} while ( 0 )
# define ASM_JUMP_IF_REG_EQ(as, reg1, reg2, label) \
do { \
asm_x86_cmp_r32_with_r32 ( as , reg1 , reg2 ) ; \
asm_x86_jcc_label ( as , ASM_X86_CC_JE , label ) ; \
} while ( 0 )
2014-09-29 11:25:04 -04:00
# define ASM_CALL_IND(as, ptr, idx) asm_x86_call_ind(as, ptr, mp_f_n_args[idx], ASM_X86_REG_EAX)
2014-09-06 18:06:36 -04:00
# define ASM_MOV_REG_TO_LOCAL asm_x86_mov_r32_to_local
# define ASM_MOV_IMM_TO_REG asm_x86_mov_i32_to_r32
# define ASM_MOV_ALIGNED_IMM_TO_REG asm_x86_mov_i32_to_r32_aligned
# define ASM_MOV_IMM_TO_LOCAL_USING(as, imm, local_num, reg_temp) \
do { \
asm_x86_mov_i32_to_r32 ( as , ( imm ) , ( reg_temp ) ) ; \
asm_x86_mov_r32_to_local ( as , ( reg_temp ) , ( local_num ) ) ; \
} while ( false )
# define ASM_MOV_LOCAL_TO_REG asm_x86_mov_local_to_r32
2014-09-29 13:45:42 -04:00
# define ASM_MOV_REG_REG(as, reg_dest, reg_src) asm_x86_mov_r32_r32((as), (reg_dest), (reg_src))
2014-09-06 18:06:36 -04:00
# define ASM_MOV_LOCAL_ADDR_TO_REG asm_x86_mov_local_addr_to_r32
2013-10-08 04:05:10 -04:00
2014-09-29 13:45:42 -04:00
# define ASM_LSL_REG(as, reg) asm_x86_shl_r32_cl((as), (reg))
# define ASM_ASR_REG(as, reg) asm_x86_sar_r32_cl((as), (reg))
# define ASM_ADD_REG_REG(as, reg_dest, reg_src) asm_x86_add_r32_r32((as), (reg_dest), (reg_src))
# define ASM_SUB_REG_REG(as, reg_dest, reg_src) asm_x86_sub_r32_r32((as), (reg_dest), (reg_src))
2013-10-12 11:53:13 -04:00
# elif N_THUMB
2013-10-08 04:05:10 -04:00
// thumb specific stuff
# include "asmthumb.h"
# define EXPORT_FUN(name) emit_native_thumb_##name
2014-09-29 11:25:04 -04:00
# define REG_RET ASM_THUMB_REG_R0
# define REG_ARG_1 ASM_THUMB_REG_R0
# define REG_ARG_2 ASM_THUMB_REG_R1
# define REG_ARG_3 ASM_THUMB_REG_R2
# define REG_ARG_4 ASM_THUMB_REG_R3
2014-09-06 20:06:19 -04:00
2014-09-29 11:25:04 -04:00
# define REG_TEMP0 ASM_THUMB_REG_R0
# define REG_TEMP1 ASM_THUMB_REG_R1
# define REG_TEMP2 ASM_THUMB_REG_R2
2014-09-06 18:06:36 -04:00
2014-09-29 11:25:04 -04:00
# define REG_LOCAL_1 ASM_THUMB_REG_R4
# define REG_LOCAL_2 ASM_THUMB_REG_R5
# define REG_LOCAL_3 ASM_THUMB_REG_R6
2014-09-06 18:06:36 -04:00
# define REG_LOCAL_NUM (3)
# define ASM_PASS_COMPUTE ASM_THUMB_PASS_COMPUTE
# define ASM_PASS_EMIT ASM_THUMB_PASS_EMIT
# define ASM_T asm_thumb_t
# define ASM_NEW asm_thumb_new
# define ASM_FREE asm_thumb_free
# define ASM_GET_CODE asm_thumb_get_code
# define ASM_GET_CODE_SIZE asm_thumb_get_code_size
# define ASM_START_PASS asm_thumb_start_pass
# define ASM_END_PASS asm_thumb_end_pass
# define ASM_ENTRY asm_thumb_entry
# define ASM_EXIT asm_thumb_exit
# define ASM_LABEL_ASSIGN asm_thumb_label_assign
# define ASM_JUMP asm_thumb_b_label
# define ASM_JUMP_IF_REG_ZERO(as, reg, label) \
do { \
asm_thumb_cmp_rlo_i8 ( as , reg , 0 ) ; \
2014-09-29 11:25:04 -04:00
asm_thumb_bcc_label ( as , ASM_THUMB_CC_EQ , label ) ; \
2014-09-06 18:06:36 -04:00
} while ( 0 )
# define ASM_JUMP_IF_REG_NONZERO(as, reg, label) \
do { \
asm_thumb_cmp_rlo_i8 ( as , reg , 0 ) ; \
2014-09-29 11:25:04 -04:00
asm_thumb_bcc_label ( as , ASM_THUMB_CC_NE , label ) ; \
2014-09-06 18:06:36 -04:00
} while ( 0 )
# define ASM_JUMP_IF_REG_EQ(as, reg1, reg2, label) \
do { \
asm_thumb_cmp_rlo_rlo ( as , reg1 , reg2 ) ; \
2014-09-29 11:25:04 -04:00
asm_thumb_bcc_label ( as , ASM_THUMB_CC_EQ , label ) ; \
2014-09-06 18:06:36 -04:00
} while ( 0 )
2014-09-29 11:25:04 -04:00
# define ASM_CALL_IND(as, ptr, idx) asm_thumb_bl_ind(as, ptr, idx, ASM_THUMB_REG_R3)
2014-09-06 18:06:36 -04:00
# define ASM_MOV_REG_TO_LOCAL(as, reg, local_num) asm_thumb_mov_local_reg(as, (local_num), (reg))
# define ASM_MOV_IMM_TO_REG(as, imm, reg) asm_thumb_mov_reg_i32_optimised(as, (reg), (imm))
# define ASM_MOV_ALIGNED_IMM_TO_REG(as, imm, reg) asm_thumb_mov_reg_i32_aligned(as, (reg), (imm))
# define ASM_MOV_IMM_TO_LOCAL_USING(as, imm, local_num, reg_temp) \
do { \
asm_thumb_mov_reg_i32_optimised ( as , ( reg_temp ) , ( imm ) ) ; \
asm_thumb_mov_local_reg ( as , ( local_num ) , ( reg_temp ) ) ; \
} while ( false )
# define ASM_MOV_LOCAL_TO_REG(as, local_num, reg) asm_thumb_mov_reg_local(as, (reg), (local_num))
2014-09-29 13:45:42 -04:00
# define ASM_MOV_REG_REG(as, reg_dest, reg_src) asm_thumb_mov_reg_reg((as), (reg_dest), (reg_src))
2014-09-06 18:06:36 -04:00
# define ASM_MOV_LOCAL_ADDR_TO_REG(as, local_num, reg) asm_thumb_mov_reg_local_addr(as, (reg), (local_num))
2013-10-08 04:05:10 -04:00
2014-09-29 13:45:42 -04:00
# define ASM_LSL_REG_REG(as, reg_dest, reg_shift) asm_thumb_format_4((as), ASM_THUMB_FORMAT_4_LSL, (reg_dest), (reg_shift))
# define ASM_ASR_REG_REG(as, reg_dest, reg_shift) asm_thumb_format_4((as), ASM_THUMB_FORMAT_4_ASR, (reg_dest), (reg_shift))
# define ASM_ADD_REG_REG(as, reg_dest, reg_src) asm_thumb_add_rlo_rlo_rlo((as), (reg_dest), (reg_dest), (reg_src))
# define ASM_SUB_REG_REG(as, reg_dest, reg_src) asm_thumb_sub_rlo_rlo_rlo((as), (reg_dest), (reg_dest), (reg_src))
2014-08-16 16:55:53 -04:00
# elif N_ARM
// ARM specific stuff
# include "asmarm.h"
# define EXPORT_FUN(name) emit_native_arm_##name
2014-09-29 11:25:04 -04:00
# define REG_RET ASM_ARM_REG_R0
# define REG_ARG_1 ASM_ARM_REG_R0
# define REG_ARG_2 ASM_ARM_REG_R1
# define REG_ARG_3 ASM_ARM_REG_R2
# define REG_ARG_4 ASM_ARM_REG_R3
2014-09-06 20:06:19 -04:00
2014-09-29 11:25:04 -04:00
# define REG_TEMP0 ASM_ARM_REG_R0
# define REG_TEMP1 ASM_ARM_REG_R1
# define REG_TEMP2 ASM_ARM_REG_R2
2014-09-06 18:06:36 -04:00
2014-09-29 11:25:04 -04:00
# define REG_LOCAL_1 ASM_ARM_REG_R4
# define REG_LOCAL_2 ASM_ARM_REG_R5
# define REG_LOCAL_3 ASM_ARM_REG_R6
2014-09-06 18:06:36 -04:00
# define REG_LOCAL_NUM (3)
# define ASM_PASS_COMPUTE ASM_ARM_PASS_COMPUTE
# define ASM_PASS_EMIT ASM_ARM_PASS_EMIT
# define ASM_T asm_arm_t
# define ASM_NEW asm_arm_new
# define ASM_FREE asm_arm_free
# define ASM_GET_CODE asm_arm_get_code
# define ASM_GET_CODE_SIZE asm_arm_get_code_size
# define ASM_START_PASS asm_arm_start_pass
# define ASM_END_PASS asm_arm_end_pass
# define ASM_ENTRY asm_arm_entry
# define ASM_EXIT asm_arm_exit
# define ASM_LABEL_ASSIGN asm_arm_label_assign
# define ASM_JUMP asm_arm_b_label
# define ASM_JUMP_IF_REG_ZERO(as, reg, label) \
do { \
asm_arm_cmp_reg_i8 ( as , reg , 0 ) ; \
2014-09-29 11:25:04 -04:00
asm_arm_bcc_label ( as , ASM_ARM_CC_EQ , label ) ; \
2014-09-06 18:06:36 -04:00
} while ( 0 )
# define ASM_JUMP_IF_REG_NONZERO(as, reg, label) \
do { \
asm_arm_cmp_reg_i8 ( as , reg , 0 ) ; \
2014-09-29 11:25:04 -04:00
asm_arm_bcc_label ( as , ASM_ARM_CC_NE , label ) ; \
2014-09-06 18:06:36 -04:00
} while ( 0 )
# define ASM_JUMP_IF_REG_EQ(as, reg1, reg2, label) \
do { \
asm_arm_cmp_reg_reg ( as , reg1 , reg2 ) ; \
2014-09-29 11:25:04 -04:00
asm_arm_bcc_label ( as , ASM_ARM_CC_EQ , label ) ; \
2014-09-06 18:06:36 -04:00
} while ( 0 )
2014-09-29 11:25:04 -04:00
# define ASM_CALL_IND(as, ptr, idx) asm_arm_bl_ind(as, ptr, idx, ASM_ARM_REG_R3)
2014-09-06 18:06:36 -04:00
# define ASM_MOV_REG_TO_LOCAL(as, reg, local_num) asm_arm_mov_local_reg(as, (local_num), (reg))
# define ASM_MOV_IMM_TO_REG(as, imm, reg) asm_arm_mov_reg_i32(as, (reg), (imm))
# define ASM_MOV_ALIGNED_IMM_TO_REG(as, imm, reg) asm_arm_mov_reg_i32(as, (reg), (imm))
# define ASM_MOV_IMM_TO_LOCAL_USING(as, imm, local_num, reg_temp) \
do { \
asm_arm_mov_reg_i32 ( as , ( reg_temp ) , ( imm ) ) ; \
asm_arm_mov_local_reg ( as , ( local_num ) , ( reg_temp ) ) ; \
} while ( false )
# define ASM_MOV_LOCAL_TO_REG(as, local_num, reg) asm_arm_mov_reg_local(as, (reg), (local_num))
2014-09-29 13:45:42 -04:00
# define ASM_MOV_REG_REG(as, reg_dest, reg_src) asm_arm_mov_reg_reg((as), (reg_dest), (reg_src))
2014-09-06 18:06:36 -04:00
# define ASM_MOV_LOCAL_ADDR_TO_REG(as, local_num, reg) asm_arm_mov_reg_local_addr(as, (reg), (local_num))
2014-09-29 13:45:42 -04:00
// TODO someone please implement lsl and asr
# define ASM_LSL_REG_REG(as, reg_dest, reg_shift) asm_arm_lsl_((as), (reg_dest), (reg_shift))
# define ASM_ASR_REG_REG(as, reg_dest, reg_shift) asm_arm_asr_((as), (reg_dest), (reg_shift))
# define ASM_ADD_REG_REG(as, reg_dest, reg_src) asm_arm_add_reg_reg_reg((as), (reg_dest), (reg_dest), (reg_src))
# define ASM_SUB_REG_REG(as, reg_dest, reg_src) asm_arm_sub_reg_reg_reg((as), (reg_dest), (reg_dest), (reg_src))
2014-09-06 18:06:36 -04:00
# else
# error unknown native emitter
2014-08-16 16:55:53 -04:00
2013-10-08 04:05:10 -04:00
# endif
typedef enum {
2013-10-08 17:18:32 -04:00
STACK_VALUE ,
STACK_REG ,
STACK_IMM ,
} stack_info_kind_t ;
2013-10-08 04:05:10 -04:00
typedef enum {
2014-08-15 11:45:41 -04:00
VTYPE_PYOBJ = MP_NATIVE_TYPE_OBJ ,
VTYPE_BOOL = MP_NATIVE_TYPE_BOOL ,
VTYPE_INT = MP_NATIVE_TYPE_INT ,
VTYPE_UINT = MP_NATIVE_TYPE_UINT ,
2013-10-08 04:05:10 -04:00
VTYPE_UNBOUND ,
VTYPE_PTR ,
VTYPE_PTR_NONE ,
VTYPE_BUILTIN_V_INT ,
} vtype_kind_t ;
2013-10-08 17:18:32 -04:00
typedef struct _stack_info_t {
vtype_kind_t vtype ;
stack_info_kind_t kind ;
union {
int u_reg ;
2014-07-03 08:25:24 -04:00
mp_int_t u_imm ;
2013-10-08 17:18:32 -04:00
} ;
} stack_info_t ;
2013-10-08 04:05:10 -04:00
struct _emit_t {
int pass ;
bool do_viper_types ;
2013-10-08 17:18:32 -04:00
2014-08-15 11:45:41 -04:00
vtype_kind_t return_vtype ;
2014-09-08 18:05:16 -04:00
mp_uint_t local_vtype_alloc ;
2013-10-08 04:05:10 -04:00
vtype_kind_t * local_vtype ;
2013-10-08 17:18:32 -04:00
2014-09-08 18:05:16 -04:00
mp_uint_t stack_info_alloc ;
2013-10-08 17:18:32 -04:00
stack_info_t * stack_info ;
2013-10-08 04:05:10 -04:00
int stack_start ;
int stack_size ;
bool last_emit_was_return_value ;
scope_t * scope ;
2014-09-06 18:06:36 -04:00
ASM_T * as ;
2013-10-08 04:05:10 -04:00
} ;
2014-09-08 18:05:16 -04:00
emit_t * EXPORT_FUN ( new ) ( mp_uint_t max_num_labels ) {
2014-05-07 12:24:22 -04:00
emit_t * emit = m_new0 ( emit_t , 1 ) ;
2014-09-06 18:06:36 -04:00
emit - > as = ASM_NEW ( max_num_labels ) ;
2013-10-08 04:05:10 -04:00
return emit ;
}
2014-01-24 17:42:28 -05:00
void EXPORT_FUN ( free ) ( emit_t * emit ) {
2014-09-06 18:06:36 -04:00
ASM_FREE ( emit - > as , false ) ;
2014-05-07 12:24:22 -04:00
m_del ( vtype_kind_t , emit - > local_vtype , emit - > local_vtype_alloc ) ;
m_del ( stack_info_t , emit - > stack_info , emit - > stack_info_alloc ) ;
2014-01-24 09:20:11 -05:00
m_del_obj ( emit_t , emit ) ;
}
2014-08-15 11:45:41 -04:00
STATIC void emit_native_set_native_type ( emit_t * emit , mp_uint_t op , mp_uint_t arg1 , qstr arg2 ) {
switch ( op ) {
case MP_EMIT_NATIVE_TYPE_ENABLE :
emit - > do_viper_types = arg1 ;
break ;
default : {
vtype_kind_t type ;
switch ( arg2 ) {
case MP_QSTR_object : type = VTYPE_PYOBJ ; break ;
case MP_QSTR_bool : type = VTYPE_BOOL ; break ;
case MP_QSTR_int : type = VTYPE_INT ; break ;
case MP_QSTR_uint : type = VTYPE_UINT ; break ;
default : printf ( " ViperTypeError: unknown type %s \n " , qstr_str ( arg2 ) ) ; return ;
}
if ( op = = MP_EMIT_NATIVE_TYPE_RETURN ) {
emit - > return_vtype = type ;
} else {
assert ( arg1 < emit - > local_vtype_alloc ) ;
emit - > local_vtype [ arg1 ] = type ;
}
break ;
}
}
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_start_pass ( emit_t * emit , pass_kind_t pass , scope_t * scope ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " start_pass(pass=%u, scope=%p) \n " , pass , scope ) ;
2013-10-08 04:05:10 -04:00
emit - > pass = pass ;
emit - > stack_start = 0 ;
emit - > stack_size = 0 ;
emit - > last_emit_was_return_value = false ;
emit - > scope = scope ;
2014-05-07 12:24:22 -04:00
// allocate memory for keeping track of the types of locals
if ( emit - > local_vtype_alloc < scope - > num_locals ) {
emit - > local_vtype = m_renew ( vtype_kind_t , emit - > local_vtype , emit - > local_vtype_alloc , scope - > num_locals ) ;
emit - > local_vtype_alloc = scope - > num_locals ;
2013-10-08 17:18:32 -04:00
}
2014-05-07 12:24:22 -04:00
// allocate memory for keeping track of the objects on the stack
// XXX don't know stack size on entry, and it should be maximum over all scopes
2013-10-08 17:18:32 -04:00
if ( emit - > stack_info = = NULL ) {
2014-05-07 12:24:22 -04:00
emit - > stack_info_alloc = scope - > stack_size + 50 ;
2013-10-08 17:18:32 -04:00
emit - > stack_info = m_new ( stack_info_t , emit - > stack_info_alloc ) ;
2013-10-08 04:05:10 -04:00
}
2014-08-15 11:45:41 -04:00
// set default type for return and arguments
emit - > return_vtype = VTYPE_PYOBJ ;
2014-08-15 17:39:08 -04:00
for ( mp_uint_t i = 0 ; i < emit - > scope - > num_pos_args ; i + + ) {
2014-08-15 11:45:41 -04:00
emit - > local_vtype [ i ] = VTYPE_PYOBJ ;
}
2014-08-15 17:39:08 -04:00
// local variables begin unbound, and have unknown type
for ( mp_uint_t i = emit - > scope - > num_pos_args ; i < emit - > local_vtype_alloc ; i + + ) {
emit - > local_vtype [ i ] = VTYPE_UNBOUND ;
}
// values on stack begin unbound
for ( mp_uint_t i = 0 ; i < emit - > stack_info_alloc ; i + + ) {
2014-08-15 11:45:41 -04:00
emit - > stack_info [ i ] . kind = STACK_VALUE ;
2014-08-15 17:39:08 -04:00
emit - > stack_info [ i ] . vtype = VTYPE_UNBOUND ;
2013-10-08 04:05:10 -04:00
}
2014-09-06 18:06:36 -04:00
ASM_START_PASS ( emit - > as , pass = = MP_PASS_EMIT ? ASM_PASS_EMIT : ASM_PASS_COMPUTE ) ;
2013-10-08 04:05:10 -04:00
// entry to function
int num_locals = 0 ;
2014-05-07 12:24:22 -04:00
if ( pass > MP_PASS_SCOPE ) {
2013-10-08 04:05:10 -04:00
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 ;
}
2014-09-06 18:06:36 -04:00
ASM_ENTRY ( emit - > as , num_locals ) ;
2013-10-08 04:05:10 -04:00
// initialise locals from parameters
2013-10-12 11:53:13 -04:00
# if N_X64
2014-04-27 10:50:52 -04:00
for ( int i = 0 ; i < scope - > num_pos_args ; i + + ) {
2013-10-08 04:05:10 -04:00
if ( i = = 0 ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_LOCAL_1 , REG_ARG_1 ) ;
2013-10-08 04:05:10 -04:00
} else if ( i = = 1 ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_LOCAL_2 , REG_ARG_2 ) ;
2013-10-08 04:05:10 -04:00
} else if ( i = = 2 ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_LOCAL_3 , REG_ARG_3 ) ;
2014-09-06 20:06:19 -04:00
} else if ( i = = 3 ) {
asm_x64_mov_r64_to_local ( emit - > as , REG_ARG_4 , i - REG_LOCAL_NUM ) ;
2013-10-08 04:05:10 -04:00
} else {
// TODO not implemented
assert ( 0 ) ;
}
}
2014-09-06 18:06:36 -04:00
# elif N_X86
for ( int i = 0 ; i < scope - > num_pos_args ; i + + ) {
if ( i = = 0 ) {
2014-09-06 18:38:50 -04:00
asm_x86_mov_arg_to_r32 ( emit - > as , i , REG_LOCAL_1 ) ;
2014-09-06 18:06:36 -04:00
} else if ( i = = 1 ) {
2014-09-06 18:38:50 -04:00
asm_x86_mov_arg_to_r32 ( emit - > as , i , REG_LOCAL_2 ) ;
2014-09-06 19:24:32 -04:00
} else if ( i = = 2 ) {
asm_x86_mov_arg_to_r32 ( emit - > as , i , REG_LOCAL_3 ) ;
2014-09-06 18:06:36 -04:00
} else {
2014-09-06 18:38:50 -04:00
asm_x86_mov_arg_to_r32 ( emit - > as , i , REG_TEMP0 ) ;
asm_x86_mov_r32_to_local ( emit - > as , REG_TEMP0 , i - REG_LOCAL_NUM ) ;
2014-09-06 18:06:36 -04:00
}
}
2013-10-12 11:53:13 -04:00
# elif N_THUMB
2014-04-27 10:50:52 -04:00
for ( int i = 0 ; i < scope - > num_pos_args ; i + + ) {
2013-10-08 04:05:10 -04:00
if ( i = = 0 ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_LOCAL_1 , REG_ARG_1 ) ;
2013-10-08 04:05:10 -04:00
} else if ( i = = 1 ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_LOCAL_2 , REG_ARG_2 ) ;
2013-10-08 04:05:10 -04:00
} else if ( i = = 2 ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_LOCAL_3 , REG_ARG_3 ) ;
2013-10-08 04:05:10 -04:00
} else if ( i = = 3 ) {
asm_thumb_mov_local_reg ( emit - > as , i - REG_LOCAL_NUM , REG_ARG_4 ) ;
} else {
// TODO not implemented
assert ( 0 ) ;
}
}
2014-09-29 11:25:04 -04:00
asm_thumb_mov_reg_i32 ( emit - > as , ASM_THUMB_REG_R7 , ( mp_uint_t ) mp_fun_table ) ;
2014-08-16 16:55:53 -04:00
# elif N_ARM
for ( int i = 0 ; i < scope - > num_pos_args ; i + + ) {
if ( i = = 0 ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_LOCAL_1 , REG_ARG_1 ) ;
2014-08-16 16:55:53 -04:00
} else if ( i = = 1 ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_LOCAL_2 , REG_ARG_2 ) ;
2014-08-16 16:55:53 -04:00
} else if ( i = = 2 ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_LOCAL_3 , REG_ARG_3 ) ;
2014-08-16 16:55:53 -04:00
} else if ( i = = 3 ) {
asm_arm_mov_local_reg ( emit - > as , i - REG_LOCAL_NUM , REG_ARG_4 ) ;
} else {
// TODO not implemented
assert ( 0 ) ;
}
}
2014-09-29 11:25:04 -04:00
asm_arm_mov_reg_i32 ( emit - > as , ASM_ARM_REG_R7 , ( mp_uint_t ) mp_fun_table ) ;
2014-09-06 18:06:36 -04:00
# else
# error not implemented
2013-10-08 04:05:10 -04:00
# endif
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_end_pass ( emit_t * emit ) {
2014-08-16 16:55:53 -04:00
if ( ! emit - > last_emit_was_return_value ) {
2014-09-06 18:06:36 -04:00
ASM_EXIT ( emit - > as ) ;
2014-08-16 16:55:53 -04:00
}
2014-09-06 18:06:36 -04:00
ASM_END_PASS ( emit - > as ) ;
2013-10-08 04:05:10 -04:00
// 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 ) ;
}
2014-05-07 12:24:22 -04:00
if ( emit - > pass = = MP_PASS_EMIT ) {
2014-09-06 18:06:36 -04:00
void * f = ASM_GET_CODE ( emit - > as ) ;
mp_uint_t f_len = ASM_GET_CODE_SIZE ( emit - > as ) ;
2014-08-15 11:45:41 -04:00
// compute type signature
// TODO check that viper types here convert correctly to valid types for emit glue
mp_uint_t type_sig = emit - > return_vtype & 3 ;
for ( mp_uint_t i = 0 ; i < emit - > scope - > num_pos_args ; i + + ) {
type_sig | = ( emit - > local_vtype [ i ] & 3 ) < < ( i * 2 + 2 ) ;
}
mp_emit_glue_assign_native ( emit - > scope - > raw_code , emit - > do_viper_types ? MP_CODE_NATIVE_VIPER : MP_CODE_NATIVE_PY , f , f_len , emit - > scope - > num_pos_args , type_sig ) ;
2013-10-08 04:05:10 -04:00
}
}
2014-02-12 11:31:30 -05:00
STATIC bool emit_native_last_emit_was_return_value ( emit_t * emit ) {
2013-10-08 04:05:10 -04:00
return emit - > last_emit_was_return_value ;
}
2014-09-23 10:10:03 -04:00
STATIC void adjust_stack ( emit_t * emit , mp_int_t stack_size_delta ) {
DEBUG_printf ( " adjust_stack; stack_size=%d, delta=%d \n " , emit - > stack_size , stack_size_delta ) ;
assert ( ( mp_int_t ) emit - > stack_size + stack_size_delta > = 0 ) ;
2013-10-08 04:05:10 -04:00
emit - > stack_size + = stack_size_delta ;
2014-05-07 12:24:22 -04:00
if ( emit - > pass > MP_PASS_SCOPE & & emit - > stack_size > emit - > scope - > stack_size ) {
2013-10-08 04:05:10 -04:00
emit - > scope - > stack_size = emit - > stack_size ;
}
}
2014-09-23 10:10:03 -04:00
STATIC void emit_native_adjust_stack_size ( emit_t * emit , mp_int_t delta ) {
// If we are adjusting the stack in a positive direction (pushing) then we
// need to fill in values for the stack kind and vtype of the newly-pushed
// entries. These should be set to "value" (ie not reg or imm) because we
// should only need to adjust the stack due to a jump to this part in the
// code (and hence we have settled the stack before the jump).
for ( mp_int_t i = 0 ; i < delta ; i + + ) {
stack_info_t * si = & emit - > stack_info [ emit - > stack_size + i ] ;
si - > kind = STACK_VALUE ;
si - > vtype = VTYPE_PYOBJ ; // XXX we don't know the vtype...
}
adjust_stack ( emit , delta ) ;
}
STATIC void emit_native_set_source_line ( emit_t * emit , mp_uint_t source_line ) {
}
2013-10-08 17:18:32 -04:00
/*
2014-02-12 11:31:30 -05:00
STATIC void emit_pre_raw ( emit_t * emit , int stack_size_delta ) {
2013-10-08 04:05:10 -04:00
adjust_stack ( emit , stack_size_delta ) ;
emit - > last_emit_was_return_value = false ;
}
2013-10-08 17:18:32 -04:00
*/
2013-10-08 04:05:10 -04:00
2013-10-08 17:18:32 -04:00
// this must be called at start of emit functions
2014-03-27 19:30:26 -04:00
STATIC void emit_native_pre ( emit_t * emit ) {
2013-10-08 17:18:32 -04:00
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 ;
2014-09-06 18:06:36 -04:00
ASM_MOV_REG_TO_LOCAL ( emit - > as , emit - > stack_info [ i ] . u_reg , emit - > stack_start + i ) ;
2013-10-08 17:18:32 -04:00
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 ;
}
}
}
*/
2013-10-08 04:05:10 -04:00
}
2014-09-29 13:45:42 -04:00
// depth==0 is top, depth==1 is before top, etc
STATIC stack_info_t * peek_stack ( emit_t * emit , mp_uint_t depth ) {
return & emit - > stack_info [ emit - > stack_size - 1 - depth ] ;
}
// depth==0 is top, depth==1 is before top, etc
STATIC vtype_kind_t peek_vtype ( emit_t * emit , mp_uint_t depth ) {
return peek_stack ( emit , depth ) - > vtype ;
2013-10-08 17:18:32 -04:00
}
2013-10-08 04:05:10 -04:00
2013-10-16 18:58:48 -04:00
// pos=1 is TOS, pos=2 is next, etc
// use pos=0 for no skipping
2014-02-12 11:31:30 -05:00
STATIC void need_reg_single ( emit_t * emit , int reg_needed , int skip_stack_pos ) {
2013-10-16 18:58:48 -04:00
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 ;
2014-09-06 18:06:36 -04:00
ASM_MOV_REG_TO_LOCAL ( emit - > as , si - > u_reg , emit - > stack_start + i ) ;
2013-10-16 18:58:48 -04:00
}
}
}
}
2014-02-12 11:31:30 -05:00
STATIC void need_reg_all ( emit_t * emit ) {
2013-10-08 17:18:32 -04:00
for ( int i = 0 ; i < emit - > stack_size ; i + + ) {
stack_info_t * si = & emit - > stack_info [ i ] ;
2013-10-16 18:58:48 -04:00
if ( si - > kind = = STACK_REG ) {
2013-10-08 17:18:32 -04:00
si - > kind = STACK_VALUE ;
2014-09-06 18:06:36 -04:00
ASM_MOV_REG_TO_LOCAL ( emit - > as , si - > u_reg , emit - > stack_start + i ) ;
2013-10-08 17:18:32 -04:00
}
}
}
2013-10-08 04:05:10 -04:00
2014-02-12 11:31:30 -05:00
STATIC void need_stack_settled ( emit_t * emit ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " need_stack_settled; stack_size=%d \n " , emit - > stack_size ) ;
2013-10-08 17:18:32 -04:00
for ( int i = 0 ; i < emit - > stack_size ; i + + ) {
stack_info_t * si = & emit - > stack_info [ i ] ;
if ( si - > kind = = STACK_REG ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " reg(%u) to local(%u) \n " , si - > u_reg , emit - > stack_start + i ) ;
2013-10-08 17:18:32 -04:00
si - > kind = STACK_VALUE ;
2014-09-06 18:06:36 -04:00
ASM_MOV_REG_TO_LOCAL ( emit - > as , si - > u_reg , emit - > stack_start + i ) ;
2013-10-08 17:18:32 -04:00
}
2013-10-08 04:05:10 -04:00
}
2013-10-16 18:58:48 -04:00
for ( int i = 0 ; i < emit - > stack_size ; i + + ) {
stack_info_t * si = & emit - > stack_info [ i ] ;
if ( si - > kind = = STACK_IMM ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " imm( " INT_FMT " ) to local(%u) \n " , si - > u_imm , emit - > stack_start + i ) ;
2014-08-29 15:05:32 -04:00
si - > kind = STACK_VALUE ;
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_LOCAL_USING ( emit - > as , si - > u_imm , emit - > stack_start + i , REG_TEMP0 ) ;
2013-10-16 18:58:48 -04:00
}
}
2013-10-08 04:05:10 -04:00
}
2013-10-16 18:58:48 -04:00
// pos=1 is TOS, pos=2 is next, etc
2014-02-12 11:31:30 -05:00
STATIC void emit_access_stack ( emit_t * emit , int pos , vtype_kind_t * vtype , int reg_dest ) {
2013-10-16 18:58:48 -04:00
need_reg_single ( emit , reg_dest , pos ) ;
stack_info_t * si = & emit - > stack_info [ emit - > stack_size - pos ] ;
2013-10-08 17:18:32 -04:00
* vtype = si - > vtype ;
switch ( si - > kind ) {
case STACK_VALUE :
2014-09-06 18:06:36 -04:00
ASM_MOV_LOCAL_TO_REG ( emit - > as , emit - > stack_start + emit - > stack_size - pos , reg_dest ) ;
2013-10-08 04:05:10 -04:00
break ;
2013-10-08 17:18:32 -04:00
case STACK_REG :
if ( si - > u_reg ! = reg_dest ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , reg_dest , si - > u_reg ) ;
2013-10-08 04:05:10 -04:00
}
break ;
2013-10-08 17:18:32 -04:00
case STACK_IMM :
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_REG ( emit - > as , si - > u_imm , reg_dest ) ;
2013-10-08 04:05:10 -04:00
break ;
}
}
2014-09-29 13:45:42 -04:00
// If stacked value is in a register, then *reg_dest is set to that register.
// Otherwise, the value is put in *reg_dest.
STATIC void emit_pre_pop_reg_flexible ( emit_t * emit , vtype_kind_t * vtype , int * reg_dest ) {
emit - > last_emit_was_return_value = false ;
stack_info_t * si = peek_stack ( emit , 0 ) ;
if ( si - > kind = = STACK_REG ) {
* vtype = si - > vtype ;
* reg_dest = si - > u_reg ;
need_reg_single ( emit , * reg_dest , 1 ) ;
} else {
emit_access_stack ( emit , 1 , vtype , * reg_dest ) ;
}
adjust_stack ( emit , - 1 ) ;
}
2014-09-06 13:38:20 -04:00
STATIC void emit_pre_pop_discard ( emit_t * emit ) {
2014-05-10 08:40:46 -04:00
emit - > last_emit_was_return_value = false ;
adjust_stack ( emit , - 1 ) ;
}
2014-02-12 11:31:30 -05:00
STATIC void emit_pre_pop_reg ( emit_t * emit , vtype_kind_t * vtype , int reg_dest ) {
2013-10-16 18:58:48 -04:00
emit - > last_emit_was_return_value = false ;
emit_access_stack ( emit , 1 , vtype , reg_dest ) ;
adjust_stack ( emit , - 1 ) ;
}
2014-02-12 11:31:30 -05:00
STATIC void emit_pre_pop_reg_reg ( emit_t * emit , vtype_kind_t * vtypea , int rega , vtype_kind_t * vtypeb , int regb ) {
2013-10-08 04:05:10 -04:00
emit_pre_pop_reg ( emit , vtypea , rega ) ;
2013-10-08 17:18:32 -04:00
emit_pre_pop_reg ( emit , vtypeb , regb ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
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 ) {
2013-10-08 04:05:10 -04:00
emit_pre_pop_reg ( emit , vtypea , rega ) ;
2013-10-08 17:18:32 -04:00
emit_pre_pop_reg ( emit , vtypeb , regb ) ;
emit_pre_pop_reg ( emit , vtypec , regc ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_post ( emit_t * emit ) {
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_post_push_reg ( emit_t * emit , vtype_kind_t vtype , int reg ) {
2013-10-08 17:18:32 -04:00
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 ) ;
2013-10-08 04:05:10 -04:00
}
2014-07-03 08:25:24 -04:00
STATIC void emit_post_push_imm ( emit_t * emit , vtype_kind_t vtype , mp_int_t imm ) {
2013-10-08 17:18:32 -04:00
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 ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_post_push_reg_reg ( emit_t * emit , vtype_kind_t vtypea , int rega , vtype_kind_t vtypeb , int regb ) {
2013-10-08 17:18:32 -04:00
emit_post_push_reg ( emit , vtypea , rega ) ;
emit_post_push_reg ( emit , vtypeb , regb ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
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 ) {
2013-10-08 17:18:32 -04:00
emit_post_push_reg ( emit , vtypea , rega ) ;
emit_post_push_reg ( emit , vtypeb , regb ) ;
emit_post_push_reg ( emit , vtypec , regc ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
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 ) {
2013-10-08 17:18:32 -04:00
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 ) ;
2013-10-08 04:05:10 -04:00
}
2014-08-16 17:31:57 -04:00
STATIC void emit_call ( emit_t * emit , mp_fun_kind_t fun_kind ) {
2013-10-16 18:58:48 -04:00
need_reg_all ( emit ) ;
2014-09-06 18:06:36 -04:00
ASM_CALL_IND ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind ) ;
2013-10-08 04:05:10 -04:00
}
2014-08-16 17:31:57 -04:00
STATIC void emit_call_with_imm_arg ( emit_t * emit , mp_fun_kind_t fun_kind , mp_int_t arg_val , int arg_reg ) {
2013-10-10 17:06:54 -04:00
need_reg_all ( emit ) ;
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_REG ( emit - > as , arg_val , arg_reg ) ;
ASM_CALL_IND ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind ) ;
2013-10-08 04:05:10 -04:00
}
2014-07-03 08:25:24 -04:00
// the first arg is stored in the code aligned on a mp_uint_t boundary
2014-08-16 17:31:57 -04:00
STATIC void emit_call_with_imm_arg_aligned ( emit_t * emit , mp_fun_kind_t fun_kind , mp_int_t arg_val , int arg_reg ) {
2014-05-07 13:30:52 -04:00
need_reg_all ( emit ) ;
2014-09-06 18:06:36 -04:00
ASM_MOV_ALIGNED_IMM_TO_REG ( emit - > as , arg_val , arg_reg ) ;
ASM_CALL_IND ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind ) ;
2014-05-07 13:30:52 -04:00
}
2014-08-16 17:31:57 -04:00
STATIC void emit_call_with_2_imm_args ( emit_t * emit , mp_fun_kind_t fun_kind , mp_int_t arg_val1 , int arg_reg1 , mp_int_t arg_val2 , int arg_reg2 ) {
2014-02-02 08:11:48 -05:00
need_reg_all ( emit ) ;
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_REG ( emit - > as , arg_val1 , arg_reg1 ) ;
ASM_MOV_IMM_TO_REG ( emit - > as , arg_val2 , arg_reg2 ) ;
ASM_CALL_IND ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind ) ;
2014-02-02 08:11:48 -05:00
}
2014-07-03 08:25:24 -04:00
// the first arg is stored in the code aligned on a mp_uint_t boundary
2014-08-16 17:31:57 -04:00
STATIC void emit_call_with_3_imm_args_and_first_aligned ( emit_t * emit , mp_fun_kind_t fun_kind , mp_int_t arg_val1 , int arg_reg1 , mp_int_t arg_val2 , int arg_reg2 , mp_int_t arg_val3 , int arg_reg3 ) {
2014-04-06 07:58:40 -04:00
need_reg_all ( emit ) ;
2014-09-06 18:06:36 -04:00
ASM_MOV_ALIGNED_IMM_TO_REG ( emit - > as , arg_val1 , arg_reg1 ) ;
ASM_MOV_IMM_TO_REG ( emit - > as , arg_val2 , arg_reg2 ) ;
ASM_MOV_IMM_TO_REG ( emit - > as , arg_val3 , arg_reg3 ) ;
ASM_CALL_IND ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind ) ;
2014-04-06 07:58:40 -04:00
}
2014-08-16 17:06:11 -04:00
// vtype of all n_pop objects is VTYPE_PYOBJ
// Will convert any items that are not VTYPE_PYOBJ to this type and put them back on the stack.
// If any conversions of non-immediate values are needed, then it uses REG_ARG_1, REG_ARG_2 and REG_RET.
// Otherwise, it does not use any temporary registers (but may use reg_dest before loading it with stack pointer).
STATIC void emit_get_stack_pointer_to_reg_for_pop ( emit_t * emit , mp_uint_t reg_dest , mp_uint_t n_pop ) {
need_reg_all ( emit ) ;
2013-10-08 04:05:10 -04:00
2014-08-16 17:06:11 -04:00
// First, store any immediate values to their respective place on the stack.
for ( mp_uint_t i = 0 ; i < n_pop ; i + + ) {
stack_info_t * si = & emit - > stack_info [ emit - > stack_size - 1 - i ] ;
// must push any imm's to stack
// must convert them to VTYPE_PYOBJ for viper code
if ( si - > kind = = STACK_IMM ) {
si - > kind = STACK_VALUE ;
switch ( si - > vtype ) {
case VTYPE_PYOBJ :
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_LOCAL_USING ( emit - > as , si - > u_imm , emit - > stack_start + emit - > stack_size - 1 - i , reg_dest ) ;
2014-08-16 17:06:11 -04:00
break ;
case VTYPE_BOOL :
if ( si - > u_imm = = 0 ) {
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_LOCAL_USING ( emit - > as , ( mp_uint_t ) mp_const_false , emit - > stack_start + emit - > stack_size - 1 - i , reg_dest ) ;
2014-08-16 17:06:11 -04:00
} else {
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_LOCAL_USING ( emit - > as , ( mp_uint_t ) mp_const_true , emit - > stack_start + emit - > stack_size - 1 - i , reg_dest ) ;
2014-08-16 17:06:11 -04:00
}
si - > vtype = VTYPE_PYOBJ ;
break ;
case VTYPE_INT :
case VTYPE_UINT :
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_LOCAL_USING ( emit - > as , ( si - > u_imm < < 1 ) | 1 , emit - > stack_start + emit - > stack_size - 1 - i , reg_dest ) ;
2014-08-16 17:06:11 -04:00
si - > vtype = VTYPE_PYOBJ ;
break ;
default :
// not handled
assert ( 0 ) ;
}
}
// verify that this value is on the stack
assert ( si - > kind = = STACK_VALUE ) ;
}
// Second, convert any non-VTYPE_PYOBJ to that type.
for ( mp_uint_t i = 0 ; i < n_pop ; i + + ) {
stack_info_t * si = & emit - > stack_info [ emit - > stack_size - 1 - i ] ;
if ( si - > vtype ! = VTYPE_PYOBJ ) {
mp_uint_t local_num = emit - > stack_start + emit - > stack_size - 1 - i ;
2014-09-06 18:06:36 -04:00
ASM_MOV_LOCAL_TO_REG ( emit - > as , local_num , REG_ARG_1 ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_CONVERT_NATIVE_TO_OBJ , si - > vtype , REG_ARG_2 ) ; // arg2 = type
2014-09-06 18:06:36 -04:00
ASM_MOV_REG_TO_LOCAL ( emit - > as , REG_RET , local_num ) ;
2014-08-16 17:06:11 -04:00
si - > vtype = VTYPE_PYOBJ ;
}
}
// Adujust the stack for a pop of n_pop items, and load the stack pointer into reg_dest.
adjust_stack ( emit , - n_pop ) ;
2014-09-06 18:06:36 -04:00
ASM_MOV_LOCAL_ADDR_TO_REG ( emit - > as , emit - > stack_start + emit - > stack_size , reg_dest ) ;
2014-08-16 17:06:11 -04:00
}
// vtype of all n_push objects is VTYPE_PYOBJ
STATIC void emit_get_stack_pointer_to_reg_for_push ( emit_t * emit , mp_uint_t reg_dest , mp_uint_t n_push ) {
need_reg_all ( emit ) ;
for ( mp_uint_t i = 0 ; i < n_push ; i + + ) {
emit - > stack_info [ emit - > stack_size + i ] . kind = STACK_VALUE ;
emit - > stack_info [ emit - > stack_size + i ] . vtype = VTYPE_PYOBJ ;
2013-10-08 04:05:10 -04:00
}
2014-09-06 18:06:36 -04:00
ASM_MOV_LOCAL_ADDR_TO_REG ( emit - > as , emit - > stack_start + emit - > stack_size , reg_dest ) ;
2014-08-16 17:06:11 -04:00
adjust_stack ( emit , n_push ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_load_id ( emit_t * emit , qstr qst ) {
emit_common_load_id ( emit , & EXPORT_FUN ( method_table ) , emit - > scope , qst ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_store_id ( emit_t * emit , qstr qst ) {
emit_common_store_id ( emit , & EXPORT_FUN ( method_table ) , emit - > scope , qst ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_delete_id ( emit_t * emit , qstr qst ) {
emit_common_delete_id ( emit , & EXPORT_FUN ( method_table ) , emit - > scope , qst ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_label_assign ( emit_t * emit , mp_uint_t l ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " label_assign( " UINT_FMT " ) \n " , l ) ;
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2013-10-16 18:58:48 -04:00
// need to commit stack because we can jump here from elsewhere
need_stack_settled ( emit ) ;
2014-09-06 18:06:36 -04:00
ASM_LABEL_ASSIGN ( emit - > as , l ) ;
2013-11-02 16:34:54 -04:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-04-06 07:58:40 -04:00
STATIC void emit_native_import_name ( emit_t * emit , qstr qst ) {
DEBUG_printf ( " import_name %s \n " , qstr_str ( qst ) ) ;
vtype_kind_t vtype_fromlist ;
vtype_kind_t vtype_level ;
emit_pre_pop_reg_reg ( emit , & vtype_fromlist , REG_ARG_2 , & vtype_level , REG_ARG_3 ) ; // arg2 = fromlist, arg3 = level
assert ( vtype_fromlist = = VTYPE_PYOBJ ) ;
assert ( vtype_level = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_IMPORT_NAME , qst , REG_ARG_1 ) ; // arg1 = import name
2014-04-06 07:58:40 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 04:05:10 -04:00
}
2014-04-06 07:58:40 -04:00
STATIC void emit_native_import_from ( emit_t * emit , qstr qst ) {
DEBUG_printf ( " import_from %s \n " , qstr_str ( qst ) ) ;
emit_native_pre ( emit ) ;
vtype_kind_t vtype_module ;
emit_access_stack ( emit , 1 , & vtype_module , REG_ARG_1 ) ; // arg1 = module
assert ( vtype_module = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_IMPORT_FROM , qst , REG_ARG_2 ) ; // arg2 = import name
2014-04-06 07:58:40 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_import_star ( emit_t * emit ) {
2014-04-06 07:58:40 -04:00
DEBUG_printf ( " import_star \n " ) ;
vtype_kind_t vtype_module ;
emit_pre_pop_reg ( emit , & vtype_module , REG_ARG_1 ) ; // arg1 = module
assert ( vtype_module = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_IMPORT_ALL ) ;
2014-04-06 07:58:40 -04:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_load_const_tok ( emit_t * emit , mp_token_kind_t tok ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " load_const_tok(tok=%u) \n " , tok ) ;
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2013-10-08 04:05:10 -04:00
int vtype ;
2014-07-03 08:25:24 -04:00
mp_uint_t val ;
2013-10-08 04:05:10 -04:00
if ( emit - > do_viper_types ) {
switch ( tok ) {
2013-12-21 13:17:45 -05:00
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 ;
2013-10-08 04:05:10 -04:00
default : assert ( 0 ) ; vtype = 0 ; val = 0 ; // shouldn't happen
}
} else {
vtype = VTYPE_PYOBJ ;
switch ( tok ) {
2014-07-03 08:25:24 -04:00
case MP_TOKEN_KW_NONE : val = ( mp_uint_t ) mp_const_none ; break ;
case MP_TOKEN_KW_FALSE : val = ( mp_uint_t ) mp_const_false ; break ;
case MP_TOKEN_KW_TRUE : val = ( mp_uint_t ) mp_const_true ; break ;
2013-10-08 04:05:10 -04:00
default : assert ( 0 ) ; vtype = 0 ; val = 0 ; // shouldn't happen
}
}
emit_post_push_imm ( emit , vtype , val ) ;
}
2014-07-03 08:25:24 -04:00
STATIC void emit_native_load_const_small_int ( emit_t * emit , mp_int_t arg ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " load_const_small_int(int= " INT_FMT " ) \n " , arg ) ;
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2013-10-08 04:05:10 -04:00
if ( emit - > do_viper_types ) {
emit_post_push_imm ( emit , VTYPE_INT , arg ) ;
} else {
emit_post_push_imm ( emit , VTYPE_PYOBJ , ( arg < < 1 ) | 1 ) ;
}
}
2014-04-06 07:58:40 -04:00
STATIC void emit_native_load_const_int ( emit_t * emit , qstr qst ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " load_const_int %s \n " , qstr_str ( qst ) ) ;
2014-04-06 07:58:40 -04:00
// for viper: load integer, check fits in 32 bits
emit_native_pre ( emit ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_CONST_INT , qst , REG_ARG_1 ) ;
2014-04-06 07:58:40 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_load_const_dec ( emit_t * emit , qstr qst ) {
2013-11-02 16:34:54 -04:00
// for viper, a float/complex is just a Python object
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_CONST_DEC , qst , REG_ARG_1 ) ;
2013-11-02 16:34:54 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_load_const_str ( emit_t * emit , qstr qst , bool bytes ) {
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-09-12 18:15:06 -04:00
// TODO: Eventually we want to be able to work with raw pointers in viper to
// do native array access. For now we just load them as any other object.
/*
2013-10-08 04:05:10 -04:00
if ( emit - > do_viper_types ) {
// not implemented properly
// load a pointer to the asciiz string?
assert ( 0 ) ;
2014-09-08 18:05:16 -04:00
emit_post_push_imm ( emit , VTYPE_PTR , ( mp_uint_t ) qstr_str ( qst ) ) ;
2014-09-12 18:15:06 -04:00
} else
*/
{
2014-06-30 00:17:25 -04:00
if ( bytes ) {
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_CONST_BYTES , qst , REG_ARG_1 ) ;
2014-06-30 00:17:25 -04:00
} else {
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_CONST_STR , qst , REG_ARG_1 ) ;
2014-06-30 00:17:25 -04:00
}
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
}
2014-04-20 12:50:40 -04:00
STATIC void emit_native_load_null ( emit_t * emit ) {
emit_native_pre ( emit ) ;
emit_post_push_imm ( emit , VTYPE_PYOBJ , 0 ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_load_fast ( emit_t * emit , qstr qst , mp_uint_t id_flags , mp_uint_t local_num ) {
2013-10-08 04:05:10 -04:00
vtype_kind_t vtype = emit - > local_vtype [ local_num ] ;
if ( vtype = = VTYPE_UNBOUND ) {
2014-09-08 18:05:16 -04:00
printf ( " ViperTypeError: local %s used before type known \n " , qstr_str ( qst ) ) ;
2013-10-08 04:05:10 -04:00
}
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2013-10-12 11:53:13 -04:00
# if N_X64
2013-10-08 04:05:10 -04:00
if ( local_num = = 0 ) {
emit_post_push_reg ( emit , vtype , REG_LOCAL_1 ) ;
2014-09-06 20:06:19 -04:00
} 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 ) ;
2013-10-08 04:05:10 -04:00
} else {
2014-09-29 11:25:04 -04:00
need_reg_single ( emit , REG_TEMP0 , 0 ) ;
asm_x64_mov_local_to_r64 ( emit - > as , local_num - REG_LOCAL_NUM , REG_TEMP0 ) ;
emit_post_push_reg ( emit , vtype , REG_TEMP0 ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-06 18:06:36 -04:00
# elif N_X86
if ( local_num = = 0 ) {
emit_post_push_reg ( emit , vtype , REG_LOCAL_1 ) ;
2014-09-06 18:38:50 -04:00
} else if ( local_num = = 1 ) {
emit_post_push_reg ( emit , vtype , REG_LOCAL_2 ) ;
2014-09-06 19:24:32 -04:00
} else if ( local_num = = 2 ) {
emit_post_push_reg ( emit , vtype , REG_LOCAL_3 ) ;
2014-09-06 18:06:36 -04:00
} else {
2014-09-29 11:25:04 -04:00
need_reg_single ( emit , REG_TEMP0 , 0 ) ;
asm_x86_mov_local_to_r32 ( emit - > as , local_num - REG_LOCAL_NUM , REG_TEMP0 ) ;
emit_post_push_reg ( emit , vtype , REG_TEMP0 ) ;
2014-09-06 18:06:36 -04:00
}
2013-10-12 11:53:13 -04:00
# elif N_THUMB
2013-10-08 04:05:10 -04:00
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 {
2014-09-29 11:25:04 -04:00
need_reg_single ( emit , REG_TEMP0 , 0 ) ;
asm_thumb_mov_reg_local ( emit - > as , REG_TEMP0 , local_num - REG_LOCAL_NUM ) ;
emit_post_push_reg ( emit , vtype , REG_TEMP0 ) ;
2013-10-08 04:05:10 -04:00
}
2014-08-16 16:55:53 -04:00
# elif N_ARM
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 {
2014-09-29 11:25:04 -04:00
need_reg_single ( emit , REG_TEMP0 , 0 ) ;
asm_arm_mov_reg_local ( emit - > as , REG_TEMP0 , local_num - REG_LOCAL_NUM ) ;
emit_post_push_reg ( emit , vtype , REG_TEMP0 ) ;
2014-08-16 16:55:53 -04:00
}
2014-09-06 18:06:36 -04:00
# else
# error not implemented
2013-10-08 04:05:10 -04:00
# endif
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_load_deref ( emit_t * emit , qstr qst , mp_uint_t local_num ) {
2013-12-10 19:41:43 -05:00
// not implemented
// in principle could support this quite easily (ldr r0, [r0, #0]) and then get closed over variables!
assert ( 0 ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_load_name ( emit_t * emit , qstr qst ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " load_name(%s) \n " , qstr_str ( qst ) ) ;
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_NAME , qst , REG_ARG_1 ) ;
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_load_global ( emit_t * emit , qstr qst ) {
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_GLOBAL , qst , REG_ARG_1 ) ;
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_load_attr ( emit_t * emit , qstr qst ) {
2013-10-08 04:05:10 -04:00
// depends on type of subject:
// - integer, function, pointer to integers: error
// - pointer to structure: get member, quite easy
2014-03-30 08:35:08 -04:00
// - Python object: call mp_load_attr, and needs to be typed to convert result
2013-10-08 04:05:10 -04:00
vtype_kind_t vtype_base ;
emit_pre_pop_reg ( emit , & vtype_base , REG_ARG_1 ) ; // arg1 = base
assert ( vtype_base = = VTYPE_PYOBJ ) ;
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_ATTR , qst , REG_ARG_2 ) ; // arg2 = attribute name
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_load_method ( emit_t * emit , qstr qst ) {
2013-10-08 04:05:10 -04:00
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
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_METHOD , qst , REG_ARG_2 ) ; // arg2 = method name
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_load_build_class ( emit_t * emit ) {
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_LOAD_BUILD_CLASS ) ;
2013-10-10 06:24:39 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 04:05:10 -04:00
}
2014-04-17 17:10:53 -04:00
STATIC void emit_native_load_subscr ( emit_t * emit ) {
vtype_kind_t vtype_lhs , vtype_rhs ;
emit_pre_pop_reg_reg ( emit , & vtype_rhs , REG_ARG_2 , & vtype_lhs , REG_ARG_1 ) ;
if ( vtype_lhs = = VTYPE_PYOBJ & & vtype_rhs = = VTYPE_PYOBJ ) {
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_OBJ_SUBSCR , ( mp_uint_t ) MP_OBJ_SENTINEL , REG_ARG_3 ) ;
2014-04-17 17:10:53 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
} else {
printf ( " ViperTypeError: can't do subscr of types %d and %d \n " , vtype_lhs , vtype_rhs ) ;
}
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_store_fast ( emit_t * emit , qstr qst , mp_uint_t local_num ) {
2013-10-08 04:05:10 -04:00
vtype_kind_t vtype ;
2013-10-12 11:53:13 -04:00
# if N_X64
2013-10-08 04:05:10 -04:00
if ( local_num = = 0 ) {
emit_pre_pop_reg ( emit , & vtype , REG_LOCAL_1 ) ;
2014-09-06 20:06:19 -04:00
} 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 ) ;
2013-10-08 04:05:10 -04:00
} else {
2014-09-29 11:25:04 -04:00
emit_pre_pop_reg ( emit , & vtype , REG_TEMP0 ) ;
asm_x64_mov_r64_to_local ( emit - > as , REG_TEMP0 , local_num - REG_LOCAL_NUM ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-06 18:06:36 -04:00
# elif N_X86
if ( local_num = = 0 ) {
emit_pre_pop_reg ( emit , & vtype , REG_LOCAL_1 ) ;
2014-09-06 18:38:50 -04:00
} else if ( local_num = = 1 ) {
emit_pre_pop_reg ( emit , & vtype , REG_LOCAL_2 ) ;
2014-09-06 19:24:32 -04:00
} else if ( local_num = = 2 ) {
emit_pre_pop_reg ( emit , & vtype , REG_LOCAL_3 ) ;
2014-09-06 18:06:36 -04:00
} else {
2014-09-29 11:25:04 -04:00
emit_pre_pop_reg ( emit , & vtype , REG_TEMP0 ) ;
asm_x86_mov_r32_to_local ( emit - > as , REG_TEMP0 , local_num - REG_LOCAL_NUM ) ;
2014-09-06 18:06:36 -04:00
}
2013-10-12 11:53:13 -04:00
# elif N_THUMB
2013-10-08 04:05:10 -04:00
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 {
2014-09-29 11:25:04 -04:00
emit_pre_pop_reg ( emit , & vtype , REG_TEMP0 ) ;
asm_thumb_mov_local_reg ( emit - > as , local_num - REG_LOCAL_NUM , REG_TEMP0 ) ;
2013-10-08 04:05:10 -04:00
}
2014-08-16 16:55:53 -04:00
# elif N_ARM
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 {
2014-09-29 11:25:04 -04:00
emit_pre_pop_reg ( emit , & vtype , REG_TEMP0 ) ;
asm_arm_mov_local_reg ( emit - > as , local_num - REG_LOCAL_NUM , REG_TEMP0 ) ;
2014-08-16 16:55:53 -04:00
}
2014-09-06 18:06:36 -04:00
# else
# error not implemented
2013-10-08 04:05:10 -04:00
# 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
2014-09-08 18:05:16 -04:00
printf ( " ViperTypeError: type mismatch, local %s has type %d but source object has type %d \n " , qstr_str ( qst ) , emit - > local_vtype [ local_num ] , vtype ) ;
2013-10-08 04:05:10 -04:00
}
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_store_deref ( emit_t * emit , qstr qst , mp_uint_t local_num ) {
2013-12-10 19:41:43 -05:00
// not implemented
assert ( 0 ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_store_name ( emit_t * emit , qstr qst ) {
2014-03-30 08:35:08 -04:00
// mp_store_name, but needs conversion of object (maybe have mp_viper_store_name(obj, type))
2013-10-08 04:05:10 -04:00
vtype_kind_t vtype ;
emit_pre_pop_reg ( emit , & vtype , REG_ARG_2 ) ;
assert ( vtype = = VTYPE_PYOBJ ) ;
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_STORE_NAME , qst , REG_ARG_1 ) ; // arg1 = name
2013-10-08 04:05:10 -04:00
emit_post ( emit ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_store_global ( emit_t * emit , qstr qst ) {
2014-09-29 13:45:42 -04:00
vtype_kind_t vtype = peek_vtype ( emit , 0 ) ;
2014-08-15 18:47:59 -04:00
if ( vtype = = VTYPE_PYOBJ ) {
emit_pre_pop_reg ( emit , & vtype , REG_ARG_2 ) ;
} else {
emit_pre_pop_reg ( emit , & vtype , REG_ARG_1 ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_CONVERT_NATIVE_TO_OBJ , vtype , REG_ARG_2 ) ; // arg2 = type
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_ARG_2 , REG_RET ) ;
2014-08-15 18:47:59 -04:00
}
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_STORE_GLOBAL , qst , REG_ARG_1 ) ; // arg1 = name
2014-08-15 18:47:59 -04:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_store_attr ( emit_t * emit , qstr qst ) {
2013-10-10 06:24:39 -04:00
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 ) ;
2014-09-08 18:05:16 -04:00
emit_call_with_imm_arg ( emit , MP_F_STORE_ATTR , qst , REG_ARG_2 ) ; // arg2 = attribute name
2013-10-10 06:24:39 -04:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_store_subscr ( emit_t * emit ) {
2013-10-08 04:05:10 -04:00
// 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 ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_OBJ_SUBSCR ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_delete_fast ( emit_t * emit , qstr qst , mp_uint_t local_num ) {
2014-09-06 13:38:20 -04:00
// TODO implement me!
2013-10-08 04:05:10 -04:00
// could support for Python types, just set to None (so GC can reclaim it)
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_delete_deref ( emit_t * emit , qstr qst , mp_uint_t local_num ) {
2014-09-06 13:38:20 -04:00
// TODO implement me!
2013-12-10 19:41:43 -05:00
}
2014-09-06 13:38:20 -04:00
STATIC void emit_native_delete_name ( emit_t * emit , qstr qst ) {
emit_native_pre ( emit ) ;
emit_call_with_imm_arg ( emit , MP_F_DELETE_NAME , qst , REG_ARG_1 ) ;
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-06 13:38:20 -04:00
STATIC void emit_native_delete_global ( emit_t * emit , qstr qst ) {
emit_native_pre ( emit ) ;
emit_call_with_imm_arg ( emit , MP_F_DELETE_GLOBAL , qst , REG_ARG_1 ) ;
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-06 13:38:20 -04:00
STATIC void emit_native_delete_attr ( emit_t * emit , qstr qst ) {
2014-06-22 13:35:04 -04:00
vtype_kind_t vtype_base ;
emit_pre_pop_reg ( emit , & vtype_base , REG_ARG_1 ) ; // arg1 = base
assert ( vtype_base = = VTYPE_PYOBJ ) ;
2014-09-06 13:38:20 -04:00
emit_call_with_2_imm_args ( emit , MP_F_STORE_ATTR , qst , REG_ARG_2 , ( mp_uint_t ) MP_OBJ_NULL , REG_ARG_3 ) ; // arg2 = attribute name, arg3 = value (null for delete)
2014-06-22 13:35:04 -04:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_delete_subscr ( emit_t * emit ) {
2014-04-17 17:10:53 -04:00
vtype_kind_t vtype_index , vtype_base ;
emit_pre_pop_reg_reg ( emit , & vtype_index , REG_ARG_2 , & vtype_base , REG_ARG_1 ) ; // index, base
assert ( vtype_index = = VTYPE_PYOBJ ) ;
assert ( vtype_base = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_OBJ_SUBSCR , ( mp_uint_t ) MP_OBJ_NULL , REG_ARG_3 ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_dup_top ( emit_t * emit ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " dup_top \n " ) ;
2013-10-08 04:05:10 -04:00
vtype_kind_t vtype ;
emit_pre_pop_reg ( emit , & vtype , REG_TEMP0 ) ;
emit_post_push_reg_reg ( emit , vtype , REG_TEMP0 , vtype , REG_TEMP0 ) ;
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_dup_top_two ( emit_t * emit ) {
2013-10-08 04:05:10 -04:00
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 ) ;
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_pop_top ( emit_t * emit ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " pop_top \n " ) ;
2014-09-06 13:38:20 -04:00
emit_pre_pop_discard ( emit ) ;
2013-10-08 04:05:10 -04:00
emit_post ( emit ) ;
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_rot_two ( emit_t * emit ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " rot_two \n " ) ;
2013-10-08 17:18:32 -04:00
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 ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_rot_three ( emit_t * emit ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " rot_three \n " ) ;
2013-10-08 04:05:10 -04:00
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 ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_jump ( emit_t * emit , mp_uint_t label ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " jump(label= " UINT_FMT " ) \n " , label ) ;
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-05-07 13:30:52 -04:00
// need to commit stack because we are jumping elsewhere
need_stack_settled ( emit ) ;
2014-09-06 18:06:36 -04:00
ASM_JUMP ( emit - > as , label ) ;
2013-10-08 04:05:10 -04:00
emit_post ( emit ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_jump_helper ( emit_t * emit , mp_uint_t label , bool pop ) {
2014-09-29 13:45:42 -04:00
vtype_kind_t vtype = peek_vtype ( emit , 0 ) ;
2014-09-29 11:41:37 -04:00
switch ( vtype ) {
case VTYPE_PYOBJ :
emit_pre_pop_reg ( emit , & vtype , REG_ARG_1 ) ;
if ( ! pop ) {
adjust_stack ( emit , 1 ) ;
}
emit_call ( emit , MP_F_OBJ_IS_TRUE ) ;
break ;
case VTYPE_BOOL :
case VTYPE_INT :
case VTYPE_UINT :
emit_pre_pop_reg ( emit , & vtype , REG_RET ) ;
if ( ! pop ) {
adjust_stack ( emit , 1 ) ;
}
break ;
default :
printf ( " ViperTypeError: expecting a bool or pyobj, got %d \n " , vtype ) ;
assert ( 0 ) ;
2013-10-08 04:05:10 -04:00
}
2014-05-07 13:30:52 -04:00
// need to commit stack because we may jump elsewhere
need_stack_settled ( emit ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_pop_jump_if_true ( emit_t * emit , mp_uint_t label ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " pop_jump_if_true(label= " UINT_FMT " ) \n " , label ) ;
2014-05-07 13:30:52 -04:00
emit_native_jump_helper ( emit , label , true ) ;
2014-09-06 18:06:36 -04:00
ASM_JUMP_IF_REG_NONZERO ( emit - > as , REG_RET , label ) ;
2014-05-07 13:30:52 -04:00
emit_post ( emit ) ;
2013-11-03 08:58:19 -05:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_pop_jump_if_false ( emit_t * emit , mp_uint_t label ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " pop_jump_if_false(label= " UINT_FMT " ) \n " , label ) ;
2014-05-07 13:30:52 -04:00
emit_native_jump_helper ( emit , label , true ) ;
2014-09-06 18:06:36 -04:00
ASM_JUMP_IF_REG_ZERO ( emit - > as , REG_RET , label ) ;
2013-10-08 04:05:10 -04:00
emit_post ( emit ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_jump_if_true_or_pop ( emit_t * emit , mp_uint_t label ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " jump_if_true_or_pop(label= " UINT_FMT " ) \n " , label ) ;
2014-05-07 13:30:52 -04:00
emit_native_jump_helper ( emit , label , false ) ;
2014-09-06 18:06:36 -04:00
ASM_JUMP_IF_REG_NONZERO ( emit - > as , REG_RET , label ) ;
2014-05-07 13:30:52 -04:00
adjust_stack ( emit , - 1 ) ;
2013-11-03 08:58:19 -05:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2013-11-03 08:58:19 -05:00
2014-09-08 18:05:16 -04:00
STATIC void emit_native_jump_if_false_or_pop ( emit_t * emit , mp_uint_t label ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " jump_if_false_or_pop(label= " UINT_FMT " ) \n " , label ) ;
2014-05-07 13:30:52 -04:00
emit_native_jump_helper ( emit , label , false ) ;
2014-09-06 18:06:36 -04:00
ASM_JUMP_IF_REG_ZERO ( emit - > as , REG_RET , label ) ;
2014-05-07 13:30:52 -04:00
adjust_stack ( emit , - 1 ) ;
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_break_loop ( emit_t * emit , mp_uint_t label , mp_uint_t except_depth ) {
2014-05-30 10:20:41 -04:00
emit_native_jump ( emit , label & ~ MP_EMIT_BREAK_FROM_FOR ) ; // TODO properly
2013-10-08 04:05:10 -04:00
}
2014-05-07 13:30:52 -04:00
2014-09-08 18:05:16 -04:00
STATIC void emit_native_continue_loop ( emit_t * emit , mp_uint_t label , mp_uint_t except_depth ) {
2014-05-07 13:30:52 -04:00
emit_native_jump ( emit , label ) ; // TODO properly
2013-10-08 04:05:10 -04:00
}
2014-05-07 13:30:52 -04:00
2014-09-08 18:05:16 -04:00
STATIC void emit_native_setup_with ( emit_t * emit , mp_uint_t label ) {
2013-10-08 04:05:10 -04:00
// not supported, or could be with runtime call
assert ( 0 ) ;
}
2014-06-30 00:17:25 -04:00
2014-02-12 11:31:30 -05:00
STATIC void emit_native_with_cleanup ( emit_t * emit ) {
2013-10-08 04:05:10 -04:00
assert ( 0 ) ;
}
2014-06-30 00:17:25 -04:00
2014-09-08 18:05:16 -04:00
STATIC void emit_native_setup_except ( emit_t * emit , mp_uint_t label ) {
2014-06-30 00:17:25 -04:00
emit_native_pre ( emit ) ;
// need to commit stack because we may jump elsewhere
need_stack_settled ( emit ) ;
2014-07-03 08:25:24 -04:00
emit_get_stack_pointer_to_reg_for_push ( emit , REG_ARG_1 , sizeof ( nlr_buf_t ) / sizeof ( mp_uint_t ) ) ; // arg1 = pointer to nlr buf
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_NLR_PUSH ) ;
2014-09-06 18:06:36 -04:00
ASM_JUMP_IF_REG_NONZERO ( emit - > as , REG_RET , label ) ;
2014-06-30 00:17:25 -04:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-06-30 00:17:25 -04:00
2014-09-08 18:05:16 -04:00
STATIC void emit_native_setup_finally ( emit_t * emit , mp_uint_t label ) {
2014-09-06 13:38:20 -04:00
emit_native_setup_except ( emit , label ) ;
2013-10-08 04:05:10 -04:00
}
2014-06-30 00:17:25 -04:00
2014-02-12 11:31:30 -05:00
STATIC void emit_native_end_finally ( emit_t * emit ) {
2014-09-06 13:38:20 -04:00
emit_pre_pop_discard ( emit ) ;
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2013-10-16 18:58:48 -04:00
2014-02-12 11:31:30 -05:00
STATIC void emit_native_get_iter ( emit_t * emit ) {
2013-10-08 04:05:10 -04:00
// 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?
2013-10-16 18:58:48 -04:00
vtype_kind_t vtype ;
emit_pre_pop_reg ( emit , & vtype , REG_ARG_1 ) ;
assert ( vtype = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_GETITER ) ;
2013-10-16 18:58:48 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_for_iter ( emit_t * emit , mp_uint_t label ) {
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2013-10-16 18:58:48 -04:00
vtype_kind_t vtype ;
emit_access_stack ( emit , 1 , & vtype , REG_ARG_1 ) ;
assert ( vtype = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_ITERNEXT ) ;
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_REG ( emit - > as , ( mp_uint_t ) MP_OBJ_STOP_ITERATION , REG_TEMP1 ) ;
ASM_JUMP_IF_REG_EQ ( emit - > as , REG_RET , REG_TEMP1 , label ) ;
2013-10-16 18:58:48 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 04:05:10 -04:00
}
2013-10-16 18:58:48 -04:00
2014-02-12 11:31:30 -05:00
STATIC void emit_native_for_iter_end ( emit_t * emit ) {
2013-10-16 18:58:48 -04:00
// adjust stack counter (we get here from for_iter ending, which popped the value for us)
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2013-10-16 18:58:48 -04:00
adjust_stack ( emit , - 1 ) ;
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_pop_block ( emit_t * emit ) {
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_NLR_POP ) ;
2014-07-03 08:25:24 -04:00
adjust_stack ( emit , - ( mp_int_t ) ( sizeof ( nlr_buf_t ) / sizeof ( mp_uint_t ) ) ) ;
2013-10-08 04:05:10 -04:00
emit_post ( emit ) ;
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_pop_except ( emit_t * emit ) {
2014-06-30 00:17:25 -04:00
/*
emit_native_pre ( emit ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_NLR_POP ) ;
2014-07-03 08:25:24 -04:00
adjust_stack ( emit , - ( mp_int_t ) ( sizeof ( nlr_buf_t ) / sizeof ( mp_uint_t ) ) ) ;
2014-06-30 00:17:25 -04:00
emit_post ( emit ) ;
*/
2013-10-08 04:05:10 -04:00
}
2014-03-30 08:35:08 -04:00
STATIC void emit_native_unary_op ( emit_t * emit , mp_unary_op_t op ) {
2014-09-23 10:10:03 -04:00
vtype_kind_t vtype ;
emit_pre_pop_reg ( emit , & vtype , REG_ARG_2 ) ;
assert ( vtype = = VTYPE_PYOBJ ) ;
2014-06-30 00:17:25 -04:00
if ( op = = MP_UNARY_OP_NOT ) {
2014-09-23 10:10:03 -04:00
// we need to synthesise this operation by converting to bool first
emit_call_with_imm_arg ( emit , MP_F_UNARY_OP , MP_UNARY_OP_BOOL , REG_ARG_1 ) ;
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_ARG_2 , REG_RET ) ;
2014-06-30 00:17:25 -04:00
}
2014-09-23 10:10:03 -04:00
emit_call_with_imm_arg ( emit , MP_F_UNARY_OP , op , REG_ARG_1 ) ;
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 04:05:10 -04:00
}
2014-03-30 08:35:08 -04:00
STATIC void emit_native_binary_op ( emit_t * emit , mp_binary_op_t op ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " binary_op( " UINT_FMT " ) \n " , op ) ;
2014-09-29 13:45:42 -04:00
vtype_kind_t vtype_lhs = peek_vtype ( emit , 1 ) ;
vtype_kind_t vtype_rhs = peek_vtype ( emit , 0 ) ;
2013-10-08 04:05:10 -04:00
if ( vtype_lhs = = VTYPE_INT & & vtype_rhs = = VTYPE_INT ) {
2014-09-29 13:45:42 -04:00
# if N_X64 || N_X86
// special cases for x86 and shifting
if ( op = = MP_BINARY_OP_LSHIFT
| | op = = MP_BINARY_OP_INPLACE_LSHIFT
| | op = = MP_BINARY_OP_RSHIFT
| | op = = MP_BINARY_OP_INPLACE_RSHIFT ) {
# if N_X64
emit_pre_pop_reg_reg ( emit , & vtype_rhs , ASM_X64_REG_RCX , & vtype_lhs , REG_RET ) ;
# else
emit_pre_pop_reg_reg ( emit , & vtype_rhs , ASM_X86_REG_ECX , & vtype_lhs , REG_RET ) ;
# endif
if ( op = = MP_BINARY_OP_LSHIFT | | op = = MP_BINARY_OP_INPLACE_LSHIFT ) {
ASM_LSL_REG ( emit - > as , REG_RET ) ;
} else {
ASM_ASR_REG ( emit - > as , REG_RET ) ;
}
emit_post_push_reg ( emit , VTYPE_INT , REG_RET ) ;
return ;
}
# endif
int reg_rhs = REG_ARG_3 ;
emit_pre_pop_reg_flexible ( emit , & vtype_rhs , & reg_rhs ) ;
emit_pre_pop_reg ( emit , & vtype_lhs , REG_ARG_2 ) ;
if ( 0 ) {
// dummy
# if !(N_X64 || N_X86)
} else if ( op = = MP_BINARY_OP_LSHIFT | | op = = MP_BINARY_OP_INPLACE_LSHIFT ) {
ASM_LSL_REG_REG ( emit - > as , REG_ARG_2 , reg_rhs ) ;
2014-01-11 04:47:06 -05:00
emit_post_push_reg ( emit , VTYPE_INT , REG_ARG_2 ) ;
2014-09-29 13:45:42 -04:00
} else if ( op = = MP_BINARY_OP_RSHIFT | | op = = MP_BINARY_OP_INPLACE_RSHIFT ) {
ASM_ASR_REG_REG ( emit - > as , REG_ARG_2 , reg_rhs ) ;
emit_post_push_reg ( emit , VTYPE_INT , REG_ARG_2 ) ;
# endif
} else if ( op = = MP_BINARY_OP_ADD | | op = = MP_BINARY_OP_INPLACE_ADD ) {
ASM_ADD_REG_REG ( emit - > as , REG_ARG_2 , reg_rhs ) ;
emit_post_push_reg ( emit , VTYPE_INT , REG_ARG_2 ) ;
} else if ( op = = MP_BINARY_OP_SUBTRACT | | op = = MP_BINARY_OP_INPLACE_SUBTRACT ) {
ASM_SUB_REG_REG ( emit - > as , REG_ARG_2 , reg_rhs ) ;
emit_post_push_reg ( emit , VTYPE_INT , REG_ARG_2 ) ;
} else if ( MP_BINARY_OP_LESS < = op & & op < = MP_BINARY_OP_NOT_EQUAL ) {
// comparison ops are (in enum order):
// MP_BINARY_OP_LESS
// MP_BINARY_OP_MORE
// MP_BINARY_OP_EQUAL
// MP_BINARY_OP_LESS_EQUAL
// MP_BINARY_OP_MORE_EQUAL
// MP_BINARY_OP_NOT_EQUAL
# if N_X64
asm_x64_xor_r64_r64 ( emit - > as , REG_RET , REG_RET ) ;
asm_x64_cmp_r64_with_r64 ( emit - > as , reg_rhs , REG_ARG_2 ) ;
static byte ops [ 6 ] = {
ASM_X64_CC_JL ,
ASM_X64_CC_JG ,
ASM_X64_CC_JE ,
ASM_X64_CC_JLE ,
ASM_X64_CC_JGE ,
ASM_X64_CC_JNE ,
} ;
asm_x64_setcc_r8 ( emit - > as , ops [ op - MP_BINARY_OP_LESS ] , REG_RET ) ;
# elif N_X86
asm_x86_xor_r32_r32 ( emit - > as , REG_RET , REG_RET ) ;
asm_x86_cmp_r32_with_r32 ( emit - > as , reg_rhs , REG_ARG_2 ) ;
static byte ops [ 6 ] = {
ASM_X86_CC_JL ,
ASM_X86_CC_JG ,
ASM_X86_CC_JE ,
ASM_X86_CC_JLE ,
ASM_X86_CC_JGE ,
ASM_X86_CC_JNE ,
} ;
asm_x86_setcc_r8 ( emit - > as , ops [ op - MP_BINARY_OP_LESS ] , REG_RET ) ;
# elif N_THUMB
asm_thumb_cmp_rlo_rlo ( emit - > as , REG_ARG_2 , reg_rhs ) ;
static uint16_t ops [ 6 ] = {
ASM_THUMB_OP_ITE_GE ,
ASM_THUMB_OP_ITE_GT ,
ASM_THUMB_OP_ITE_EQ ,
ASM_THUMB_OP_ITE_GT ,
ASM_THUMB_OP_ITE_GE ,
ASM_THUMB_OP_ITE_EQ ,
} ;
static byte ret [ 6 ] = { 0 , 1 , 1 , 0 , 1 , 0 , } ;
asm_thumb_op16 ( emit - > as , ops [ op - MP_BINARY_OP_LESS ] ) ;
asm_thumb_mov_rlo_i8 ( emit - > as , REG_RET , ret [ op - MP_BINARY_OP_LESS ] ) ;
asm_thumb_mov_rlo_i8 ( emit - > as , REG_RET , ret [ op - MP_BINARY_OP_LESS ] ^ 1 ) ;
# elif N_ARM
# error generic comparisons for ARM needs implementing
//asm_arm_less_op(emit->as, REG_RET, REG_ARG_2, reg_rhs);
//asm_arm_more_op(emit->as, REG_RET, REG_ARG_2, reg_rhs);
# else
# error not implemented
# endif
2014-01-11 04:47:06 -05:00
emit_post_push_reg ( emit , VTYPE_BOOL , REG_RET ) ;
} else {
// TODO other ops not yet implemented
assert ( 0 ) ;
}
2013-10-08 04:05:10 -04:00
} else if ( vtype_lhs = = VTYPE_PYOBJ & & vtype_rhs = = VTYPE_PYOBJ ) {
2014-09-29 13:45:42 -04:00
emit_pre_pop_reg_reg ( emit , & vtype_rhs , REG_ARG_3 , & vtype_lhs , REG_ARG_2 ) ;
2014-09-23 10:10:03 -04:00
bool invert = false ;
if ( op = = MP_BINARY_OP_NOT_IN ) {
invert = true ;
op = MP_BINARY_OP_IN ;
} else if ( op = = MP_BINARY_OP_IS_NOT ) {
invert = true ;
op = MP_BINARY_OP_IS ;
}
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_BINARY_OP , op , REG_ARG_1 ) ;
2014-09-23 10:10:03 -04:00
if ( invert ) {
2014-09-29 13:45:42 -04:00
ASM_MOV_REG_REG ( emit - > as , REG_ARG_2 , REG_RET ) ;
2014-09-23 10:10:03 -04:00
emit_call_with_imm_arg ( emit , MP_F_UNARY_OP , MP_UNARY_OP_NOT , REG_ARG_1 ) ;
}
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
} else {
printf ( " ViperTypeError: can't do binary op between types %d and %d \n " , vtype_lhs , vtype_rhs ) ;
2014-08-15 17:39:08 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 04:05:10 -04:00
}
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_build_tuple ( emit_t * emit , mp_uint_t n_args ) {
2013-10-16 18:58:48 -04:00
// for viper: call runtime, with types of args
// if wrapped in byte_array, or something, allocates memory and fills it
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-02-02 08:11:48 -05:00
emit_get_stack_pointer_to_reg_for_pop ( emit , REG_ARG_2 , n_args ) ; // pointer to items
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_BUILD_TUPLE , n_args , REG_ARG_1 ) ;
2013-10-16 18:58:48 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // new tuple
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_build_list ( emit_t * emit , mp_uint_t n_args ) {
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-02-02 08:11:48 -05:00
emit_get_stack_pointer_to_reg_for_pop ( emit , REG_ARG_2 , n_args ) ; // pointer to items
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_BUILD_LIST , n_args , REG_ARG_1 ) ;
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // new list
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_list_append ( emit_t * emit , mp_uint_t list_index ) {
2013-10-16 18:58:48 -04:00
// only used in list comprehension
vtype_kind_t vtype_list , vtype_item ;
emit_pre_pop_reg ( emit , & vtype_item , REG_ARG_2 ) ;
emit_access_stack ( emit , list_index , & vtype_list , REG_ARG_1 ) ;
assert ( vtype_list = = VTYPE_PYOBJ ) ;
assert ( vtype_item = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_LIST_APPEND ) ;
2013-10-16 18:58:48 -04:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_build_map ( emit_t * emit , mp_uint_t n_args ) {
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_BUILD_MAP , n_args , REG_ARG_1 ) ;
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // new map
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_store_map ( emit_t * emit ) {
2013-10-08 04:05:10 -04:00
vtype_kind_t vtype_key , vtype_value , vtype_map ;
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
assert ( vtype_key = = VTYPE_PYOBJ ) ;
assert ( vtype_value = = VTYPE_PYOBJ ) ;
assert ( vtype_map = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_STORE_MAP ) ;
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // map
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_map_add ( emit_t * emit , mp_uint_t map_index ) {
2013-10-16 18:58:48 -04:00
// only used in list comprehension
vtype_kind_t vtype_map , vtype_key , vtype_value ;
emit_pre_pop_reg_reg ( emit , & vtype_key , REG_ARG_2 , & vtype_value , REG_ARG_3 ) ;
emit_access_stack ( emit , map_index , & vtype_map , REG_ARG_1 ) ;
assert ( vtype_map = = VTYPE_PYOBJ ) ;
assert ( vtype_key = = VTYPE_PYOBJ ) ;
assert ( vtype_value = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_STORE_MAP ) ;
2013-10-16 18:58:48 -04:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_build_set ( emit_t * emit , mp_uint_t n_args ) {
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-02-02 08:11:48 -05:00
emit_get_stack_pointer_to_reg_for_pop ( emit , REG_ARG_2 , n_args ) ; // pointer to items
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_BUILD_SET , n_args , REG_ARG_1 ) ;
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // new set
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_set_add ( emit_t * emit , mp_uint_t set_index ) {
2013-10-16 18:58:48 -04:00
// only used in set comprehension
vtype_kind_t vtype_set , vtype_item ;
emit_pre_pop_reg ( emit , & vtype_item , REG_ARG_2 ) ;
emit_access_stack ( emit , set_index , & vtype_set , REG_ARG_1 ) ;
assert ( vtype_set = = VTYPE_PYOBJ ) ;
assert ( vtype_item = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_STORE_SET ) ;
2013-10-16 18:58:48 -04:00
emit_post ( emit ) ;
2013-10-08 04:05:10 -04:00
}
2013-10-16 18:58:48 -04:00
2014-09-08 18:05:16 -04:00
STATIC void emit_native_build_slice ( emit_t * emit , mp_uint_t n_args ) {
2014-04-06 07:58:40 -04:00
DEBUG_printf ( " build_slice %d \n " , n_args ) ;
2014-06-30 00:17:25 -04:00
if ( n_args = = 2 ) {
vtype_kind_t vtype_start , vtype_stop ;
emit_pre_pop_reg_reg ( emit , & vtype_stop , REG_ARG_2 , & vtype_start , REG_ARG_1 ) ; // arg1 = start, arg2 = stop
assert ( vtype_start = = VTYPE_PYOBJ ) ;
assert ( vtype_stop = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_NEW_SLICE , ( mp_uint_t ) mp_const_none , REG_ARG_3 ) ; // arg3 = step
2014-06-30 00:17:25 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
} else {
assert ( n_args = = 3 ) ;
vtype_kind_t vtype_start , vtype_stop , vtype_step ;
emit_pre_pop_reg_reg_reg ( emit , & vtype_step , REG_ARG_3 , & vtype_stop , REG_ARG_2 , & vtype_start , REG_ARG_1 ) ; // arg1 = start, arg2 = stop, arg3 = step
assert ( vtype_start = = VTYPE_PYOBJ ) ;
assert ( vtype_stop = = VTYPE_PYOBJ ) ;
assert ( vtype_step = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_NEW_SLICE ) ;
2014-06-30 00:17:25 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2013-10-08 04:05:10 -04:00
}
2014-04-06 07:58:40 -04:00
2014-09-08 18:05:16 -04:00
STATIC void emit_native_unpack_sequence ( emit_t * emit , mp_uint_t n_args ) {
2014-04-06 07:58:40 -04:00
DEBUG_printf ( " unpack_sequence %d \n " , n_args ) ;
vtype_kind_t vtype_base ;
emit_pre_pop_reg ( emit , & vtype_base , REG_ARG_1 ) ; // arg1 = seq
assert ( vtype_base = = VTYPE_PYOBJ ) ;
emit_get_stack_pointer_to_reg_for_push ( emit , REG_ARG_3 , n_args ) ; // arg3 = dest ptr
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_UNPACK_SEQUENCE , n_args , REG_ARG_2 ) ; // arg2 = n_args
2013-10-08 04:05:10 -04:00
}
2014-04-06 07:58:40 -04:00
2014-09-08 18:05:16 -04:00
STATIC void emit_native_unpack_ex ( emit_t * emit , mp_uint_t n_left , mp_uint_t n_right ) {
2014-05-07 13:30:52 -04:00
DEBUG_printf ( " unpack_ex %d %d \n " , n_left , n_right ) ;
vtype_kind_t vtype_base ;
emit_pre_pop_reg ( emit , & vtype_base , REG_ARG_1 ) ; // arg1 = seq
assert ( vtype_base = = VTYPE_PYOBJ ) ;
2014-06-30 00:17:25 -04:00
emit_get_stack_pointer_to_reg_for_push ( emit , REG_ARG_3 , n_left + n_right + 1 ) ; // arg3 = dest ptr
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_UNPACK_EX , n_left | ( n_right < < 8 ) , REG_ARG_2 ) ; // arg2 = n_left + n_right
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_make_function ( emit_t * emit , scope_t * scope , mp_uint_t n_pos_defaults , mp_uint_t n_kw_defaults ) {
2013-10-08 04:05:10 -04:00
// call runtime, with type info for args, or don't support dict/default params, or only support Python objects for them
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-05-07 13:30:52 -04:00
if ( n_pos_defaults = = 0 & & n_kw_defaults = = 0 ) {
2014-08-16 17:31:57 -04:00
emit_call_with_3_imm_args_and_first_aligned ( emit , MP_F_MAKE_FUNCTION_FROM_RAW_CODE , ( mp_uint_t ) scope - > raw_code , REG_ARG_1 , ( mp_uint_t ) MP_OBJ_NULL , REG_ARG_2 , ( mp_uint_t ) MP_OBJ_NULL , REG_ARG_3 ) ;
2014-05-07 13:30:52 -04:00
} else {
vtype_kind_t vtype_def_tuple , vtype_def_dict ;
emit_pre_pop_reg_reg ( emit , & vtype_def_dict , REG_ARG_3 , & vtype_def_tuple , REG_ARG_2 ) ;
assert ( vtype_def_tuple = = VTYPE_PYOBJ ) ;
assert ( vtype_def_dict = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg_aligned ( emit , MP_F_MAKE_FUNCTION_FROM_RAW_CODE , ( mp_uint_t ) scope - > raw_code , REG_ARG_1 ) ;
2014-05-07 13:30:52 -04:00
}
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_make_closure ( emit_t * emit , scope_t * scope , mp_uint_t n_closed_over , mp_uint_t n_pos_defaults , mp_uint_t n_kw_defaults ) {
2013-10-08 04:05:10 -04:00
assert ( 0 ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_call_function ( emit_t * emit , mp_uint_t n_positional , mp_uint_t n_keyword , mp_uint_t star_flags ) {
2014-09-23 10:10:03 -04:00
DEBUG_printf ( " call_function(n_pos= " UINT_FMT " , n_kw= " UINT_FMT " , star_flags= " UINT_FMT " ) \n " , n_positional , n_keyword , star_flags ) ;
2013-10-08 04:05:10 -04:00
// call special viper runtime routine with type info for args, and wanted type info for return
2014-04-09 07:43:17 -04:00
assert ( ! star_flags ) ;
2014-02-02 08:11:48 -05:00
/* we no longer have these _n specific call_function's
* they anyway push args into an array
* and they would take too much room in the native dispatch table
2013-10-08 04:05:10 -04:00
if ( n_positional = = 0 ) {
vtype_kind_t vtype_fun ;
emit_pre_pop_reg ( emit , & vtype_fun , REG_ARG_1 ) ; // the function
assert ( vtype_fun = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_CALL_FUNCTION_0 ) ;
2013-10-08 04:05:10 -04:00
} else if ( n_positional = = 1 ) {
vtype_kind_t vtype_fun , vtype_arg1 ;
emit_pre_pop_reg_reg ( emit , & vtype_arg1 , REG_ARG_2 , & vtype_fun , REG_ARG_1 ) ; // the single argument, the function
assert ( vtype_fun = = VTYPE_PYOBJ ) ;
assert ( vtype_arg1 = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_CALL_FUNCTION_1 ) ;
2013-10-08 04:05:10 -04:00
} else if ( n_positional = = 2 ) {
vtype_kind_t vtype_fun , vtype_arg1 , vtype_arg2 ;
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
assert ( vtype_fun = = VTYPE_PYOBJ ) ;
assert ( vtype_arg1 = = VTYPE_PYOBJ ) ;
assert ( vtype_arg2 = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_CALL_FUNCTION_2 ) ;
2013-10-08 04:05:10 -04:00
} else {
2013-10-10 17:06:54 -04:00
*/
2014-02-02 08:11:48 -05:00
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-04-06 07:58:40 -04:00
if ( n_positional ! = 0 | | n_keyword ! = 0 ) {
emit_get_stack_pointer_to_reg_for_pop ( emit , REG_ARG_3 , n_positional + 2 * n_keyword ) ; // pointer to args
2014-02-02 08:11:48 -05:00
}
vtype_kind_t vtype_fun ;
emit_pre_pop_reg ( emit , & vtype_fun , REG_ARG_1 ) ; // the function
assert ( vtype_fun = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call_with_imm_arg ( emit , MP_F_NATIVE_CALL_FUNCTION_N_KW , n_positional | ( n_keyword < < 8 ) , REG_ARG_2 ) ;
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_call_method ( emit_t * emit , mp_uint_t n_positional , mp_uint_t n_keyword , mp_uint_t star_flags ) {
2014-04-09 07:43:17 -04:00
assert ( ! star_flags ) ;
2014-02-02 08:11:48 -05:00
2013-10-10 17:06:54 -04:00
/*
2013-10-08 04:05:10 -04:00
if ( n_positional = = 0 ) {
vtype_kind_t vtype_meth , vtype_self ;
emit_pre_pop_reg_reg ( emit , & vtype_self , REG_ARG_2 , & vtype_meth , REG_ARG_1 ) ; // the self object (or NULL), the method
assert ( vtype_meth = = VTYPE_PYOBJ ) ;
assert ( vtype_self = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_CALL_METHOD_1 ) ;
2013-10-08 04:05:10 -04:00
} else if ( n_positional = = 1 ) {
vtype_kind_t vtype_meth , vtype_self , vtype_arg1 ;
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
assert ( vtype_meth = = VTYPE_PYOBJ ) ;
assert ( vtype_self = = VTYPE_PYOBJ ) ;
assert ( vtype_arg1 = = VTYPE_PYOBJ ) ;
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_CALL_METHOD_2 ) ;
2013-10-08 04:05:10 -04:00
} else {
2013-10-10 17:06:54 -04:00
*/
2014-02-02 08:11:48 -05:00
2014-03-27 19:30:26 -04:00
emit_native_pre ( emit ) ;
2014-04-06 07:58:40 -04:00
emit_get_stack_pointer_to_reg_for_pop ( emit , REG_ARG_3 , 2 + n_positional + 2 * n_keyword ) ; // pointer to items, including meth and self
2014-08-16 17:31:57 -04:00
emit_call_with_2_imm_args ( emit , MP_F_CALL_METHOD_N_KW , n_positional , REG_ARG_1 , n_keyword , REG_ARG_2 ) ;
2013-10-08 04:05:10 -04:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_return_value ( emit_t * emit ) {
2014-04-06 07:58:40 -04:00
DEBUG_printf ( " return_value \n " ) ;
2013-10-08 04:05:10 -04:00
vtype_kind_t vtype ;
emit_pre_pop_reg ( emit , & vtype , REG_RET ) ;
if ( emit - > do_viper_types ) {
2014-08-15 18:47:59 -04:00
if ( vtype = = VTYPE_PTR_NONE ) {
if ( emit - > return_vtype = = VTYPE_PYOBJ ) {
2014-09-06 18:06:36 -04:00
ASM_MOV_IMM_TO_REG ( emit - > as , ( mp_uint_t ) mp_const_none , REG_RET ) ;
2014-08-15 18:47:59 -04:00
}
} else if ( vtype ! = emit - > return_vtype ) {
2014-08-15 11:45:41 -04:00
printf ( " ViperTypeError: incompatible return type \n " ) ;
}
2013-10-08 04:05:10 -04:00
} else {
assert ( vtype = = VTYPE_PYOBJ ) ;
}
emit - > last_emit_was_return_value = true ;
2014-09-06 18:06:36 -04:00
//ASM_BREAK_POINT(emit->as); // to insert a break-point for debugging
ASM_EXIT ( emit - > as ) ;
2013-10-08 04:05:10 -04:00
}
2014-09-08 18:05:16 -04:00
STATIC void emit_native_raise_varargs ( emit_t * emit , mp_uint_t n_args ) {
2014-06-30 00:17:25 -04:00
assert ( n_args = = 1 ) ;
2014-08-16 17:06:11 -04:00
vtype_kind_t vtype_exc ;
emit_pre_pop_reg ( emit , & vtype_exc , REG_ARG_1 ) ; // arg1 = object to raise
if ( vtype_exc ! = VTYPE_PYOBJ ) {
printf ( " ViperTypeError: must raise an object \n " ) ;
}
// TODO probably make this 1 call to the runtime (which could even call convert, native_raise(obj, type))
2014-08-16 17:31:57 -04:00
emit_call ( emit , MP_F_NATIVE_RAISE ) ;
2013-10-08 04:05:10 -04:00
}
2014-06-30 00:17:25 -04:00
2014-02-12 11:31:30 -05:00
STATIC void emit_native_yield_value ( emit_t * emit ) {
2013-10-08 04:05:10 -04:00
// not supported (for now)
assert ( 0 ) ;
}
2014-02-12 11:31:30 -05:00
STATIC void emit_native_yield_from ( emit_t * emit ) {
2013-10-08 04:05:10 -04:00
// not supported (for now)
assert ( 0 ) ;
}
2014-06-30 00:17:25 -04:00
STATIC void emit_native_start_except_handler ( emit_t * emit ) {
// This instruction follows an nlr_pop, so the stack counter is back to zero, when really
// it should be up by a whole nlr_buf_t. We then want to pop the nlr_buf_t here, but save
// the first 2 elements, so we can get the thrown value.
adjust_stack ( emit , 2 ) ;
vtype_kind_t vtype_nlr ;
emit_pre_pop_reg ( emit , & vtype_nlr , REG_ARG_1 ) ; // get the thrown value
2014-09-06 13:38:20 -04:00
emit_pre_pop_discard ( emit ) ; // discard the linked-list pointer in the nlr_buf
2014-06-30 00:17:25 -04:00
emit_post_push_reg_reg_reg ( emit , VTYPE_PYOBJ , REG_ARG_1 , VTYPE_PYOBJ , REG_ARG_1 , VTYPE_PYOBJ , REG_ARG_1 ) ; // push the 3 exception items
}
STATIC void emit_native_end_except_handler ( emit_t * emit ) {
2014-09-06 13:38:20 -04:00
adjust_stack ( emit , - 2 ) ;
2014-06-30 00:17:25 -04:00
}
2013-10-08 04:05:10 -04:00
const emit_method_table_t EXPORT_FUN ( method_table ) = {
2014-08-15 11:45:41 -04:00
emit_native_set_native_type ,
2013-10-08 04:05:10 -04:00
emit_native_start_pass ,
emit_native_end_pass ,
emit_native_last_emit_was_return_value ,
2014-04-10 13:28:54 -04:00
emit_native_adjust_stack_size ,
2014-01-18 18:24:36 -05:00
emit_native_set_source_line ,
2013-10-08 04:05:10 -04:00
emit_native_load_id ,
emit_native_store_id ,
emit_native_delete_id ,
emit_native_label_assign ,
emit_native_import_name ,
emit_native_import_from ,
emit_native_import_star ,
emit_native_load_const_tok ,
emit_native_load_const_small_int ,
emit_native_load_const_int ,
emit_native_load_const_dec ,
emit_native_load_const_str ,
2014-04-20 12:50:40 -04:00
emit_native_load_null ,
2013-10-08 04:05:10 -04:00
emit_native_load_fast ,
emit_native_load_deref ,
2013-12-10 19:41:43 -05:00
emit_native_load_name ,
emit_native_load_global ,
2013-10-08 04:05:10 -04:00
emit_native_load_attr ,
emit_native_load_method ,
emit_native_load_build_class ,
2014-04-17 17:10:53 -04:00
emit_native_load_subscr ,
2013-10-08 04:05:10 -04:00
emit_native_store_fast ,
2013-12-10 19:41:43 -05:00
emit_native_store_deref ,
2013-10-08 04:05:10 -04:00
emit_native_store_name ,
emit_native_store_global ,
emit_native_store_attr ,
emit_native_store_subscr ,
emit_native_delete_fast ,
2013-12-10 19:41:43 -05:00
emit_native_delete_deref ,
2013-10-08 04:05:10 -04:00
emit_native_delete_name ,
emit_native_delete_global ,
emit_native_delete_attr ,
emit_native_delete_subscr ,
emit_native_dup_top ,
emit_native_dup_top_two ,
emit_native_pop_top ,
emit_native_rot_two ,
emit_native_rot_three ,
emit_native_jump ,
emit_native_pop_jump_if_true ,
emit_native_pop_jump_if_false ,
emit_native_jump_if_true_or_pop ,
emit_native_jump_if_false_or_pop ,
emit_native_break_loop ,
emit_native_continue_loop ,
emit_native_setup_with ,
emit_native_with_cleanup ,
emit_native_setup_except ,
emit_native_setup_finally ,
emit_native_end_finally ,
emit_native_get_iter ,
emit_native_for_iter ,
emit_native_for_iter_end ,
emit_native_pop_block ,
emit_native_pop_except ,
emit_native_unary_op ,
emit_native_binary_op ,
emit_native_build_tuple ,
emit_native_build_list ,
emit_native_list_append ,
emit_native_build_map ,
emit_native_store_map ,
emit_native_map_add ,
emit_native_build_set ,
emit_native_set_add ,
emit_native_build_slice ,
emit_native_unpack_sequence ,
emit_native_unpack_ex ,
emit_native_make_function ,
emit_native_make_closure ,
emit_native_call_function ,
emit_native_call_method ,
emit_native_return_value ,
emit_native_raise_varargs ,
emit_native_yield_value ,
emit_native_yield_from ,
2014-06-30 00:17:25 -04:00
emit_native_start_except_handler ,
emit_native_end_except_handler ,
2013-10-08 04:05:10 -04:00
} ;
2014-09-06 18:06:36 -04:00
# endif