2014-05-03 23:27:38 +01: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 09:05:10 +01: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 07:14:26 +00:00
# include <stdbool.h>
2013-10-08 09:05:10 +01:00
# include <stdint.h>
# include <stdio.h>
# include <string.h>
# include <assert.h>
2013-12-21 18:17:45 +00:00
# include "mpconfig.h"
2014-06-30 05:17:25 +01:00
# include "nlr.h"
2014-06-21 20:43:22 +01:00
# include "misc.h"
2014-01-21 21:40:13 +00:00
# include "qstr.h"
2013-10-08 09:05:10 +01:00
# include "lexer.h"
# include "parse.h"
2014-04-13 11:04:33 +01:00
# include "obj.h"
# include "emitglue.h"
2013-10-08 09:05:10 +01:00
# include "scope.h"
2013-12-21 18:17:45 +00:00
# include "runtime0.h"
2013-10-08 09:05:10 +01:00
# include "emit.h"
2013-12-21 18:17:45 +00:00
# include "runtime.h"
2013-10-08 09:05:10 +01:00
2014-04-06 12:58:40 +01: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 09:05:10 +01:00
// wrapper around everything in this file
2014-08-16 21:55:53 +01:00
# if (MICROPY_EMIT_X64 && N_X64) || (MICROPY_EMIT_THUMB && N_THUMB) || (MICROPY_EMIT_ARM && N_ARM)
2013-10-08 09:05:10 +01:00
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
// x64 specific stuff
# include "asmx64.h"
# define REG_LOCAL_1 (REG_RBX)
# define REG_LOCAL_NUM (1)
# define EXPORT_FUN(name) emit_native_x64_##name
# define REG_TEMP0 (REG_RAX)
# define REG_TEMP1 (REG_RDI)
# define REG_TEMP2 (REG_RSI)
# define ASM_MOV_REG_TO_LOCAL(reg, local_num) asm_x64_mov_r64_to_local(emit->as, (reg), (local_num))
# define ASM_MOV_IMM_TO_REG(imm, reg) asm_x64_mov_i64_to_r64_optimised(emit->as, (imm), (reg))
2014-05-07 17:24:22 +01:00
# define ASM_MOV_ALIGNED_IMM_TO_REG(imm, reg) asm_x64_mov_i64_to_r64_aligned(emit->as, (imm), (reg))
2013-10-10 22:06:54 +01:00
# define ASM_MOV_IMM_TO_LOCAL_USING(imm, local_num, reg_temp) do { asm_x64_mov_i64_to_r64_optimised(emit->as, (imm), (reg_temp)); asm_x64_mov_r64_to_local(emit->as, (reg_temp), (local_num)); } while (false)
2013-10-08 09:05:10 +01:00
# define ASM_MOV_LOCAL_TO_REG(local_num, reg) asm_x64_mov_local_to_r64(emit->as, (local_num), (reg))
# define ASM_MOV_REG_TO_REG(reg_src, reg_dest) asm_x64_mov_r64_to_r64(emit->as, (reg_src), (reg_dest))
# define ASM_MOV_LOCAL_ADDR_TO_REG(local_num, reg) asm_x64_mov_local_addr_to_r64(emit->as, (local_num), (reg))
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01:00
// thumb specific stuff
# include "asmthumb.h"
# define REG_LOCAL_1 (REG_R4)
# define REG_LOCAL_2 (REG_R5)
# define REG_LOCAL_3 (REG_R6)
# define REG_LOCAL_NUM (3)
# define EXPORT_FUN(name) emit_native_thumb_##name
# define REG_TEMP0 (REG_R0)
# define REG_TEMP1 (REG_R1)
# define REG_TEMP2 (REG_R2)
# define ASM_MOV_REG_TO_LOCAL(reg, local_num) asm_thumb_mov_local_reg(emit->as, (local_num), (reg))
# define ASM_MOV_IMM_TO_REG(imm, reg) asm_thumb_mov_reg_i32_optimised(emit->as, (reg), (imm))
2014-05-07 17:24:22 +01:00
# define ASM_MOV_ALIGNED_IMM_TO_REG(imm, reg) asm_thumb_mov_reg_i32_aligned(emit->as, (reg), (imm))
2013-10-10 22:06:54 +01:00
# define ASM_MOV_IMM_TO_LOCAL_USING(imm, local_num, reg_temp) do { asm_thumb_mov_reg_i32_optimised(emit->as, (reg_temp), (imm)); asm_thumb_mov_local_reg(emit->as, (local_num), (reg_temp)); } while (false)
2013-10-08 09:05:10 +01:00
# define ASM_MOV_LOCAL_TO_REG(local_num, reg) asm_thumb_mov_reg_local(emit->as, (reg), (local_num))
# define ASM_MOV_REG_TO_REG(reg_src, reg_dest) asm_thumb_mov_reg_reg(emit->as, (reg_dest), (reg_src))
# define ASM_MOV_LOCAL_ADDR_TO_REG(local_num, reg) asm_thumb_mov_reg_local_addr(emit->as, (reg), (local_num))
2014-08-16 21:55:53 +01:00
# elif N_ARM
// ARM specific stuff
# include "asmarm.h"
# define REG_LOCAL_1 (REG_R4)
# define REG_LOCAL_2 (REG_R5)
# define REG_LOCAL_3 (REG_R6)
# define REG_LOCAL_NUM (3)
# define EXPORT_FUN(name) emit_native_arm_##name
# define REG_TEMP0 (REG_R0)
# define REG_TEMP1 (REG_R1)
# define REG_TEMP2 (REG_R2)
# define ASM_MOV_REG_TO_LOCAL(reg, local_num) asm_arm_mov_local_reg(emit->as, (local_num), (reg))
# define ASM_MOV_IMM_TO_REG(imm, reg) asm_arm_mov_reg_i32(emit->as, (reg), (imm))
# define ASM_MOV_ALIGNED_IMM_TO_REG(imm, reg) asm_arm_mov_reg_i32(emit->as, (reg), (imm))
# define ASM_MOV_IMM_TO_LOCAL_USING(imm, local_num, reg_temp) do { asm_arm_mov_reg_i32(emit->as, (reg_temp), (imm)); asm_arm_mov_local_reg(emit->as, (local_num), (reg_temp)); } while (false)
# define ASM_MOV_LOCAL_TO_REG(local_num, reg) asm_arm_mov_reg_local(emit->as, (reg), (local_num))
# define ASM_MOV_REG_TO_REG(reg_src, reg_dest) asm_arm_mov_reg_reg(emit->as, (reg_dest), (reg_src))
# define ASM_MOV_LOCAL_ADDR_TO_REG(local_num, reg) asm_arm_mov_reg_local_addr(emit->as, (reg), (local_num))
2013-10-08 09:05:10 +01:00
# endif
typedef enum {
2013-10-08 22:18:32 +01:00
STACK_VALUE ,
STACK_REG ,
STACK_IMM ,
} stack_info_kind_t ;
2013-10-08 09:05:10 +01:00
typedef enum {
2014-08-15 16:45:41 +01: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 09:05:10 +01:00
VTYPE_UNBOUND ,
VTYPE_PTR ,
VTYPE_PTR_NONE ,
VTYPE_BUILTIN_V_INT ,
} vtype_kind_t ;
2013-10-08 22:18:32 +01:00
typedef struct _stack_info_t {
vtype_kind_t vtype ;
stack_info_kind_t kind ;
union {
int u_reg ;
2014-07-03 13:25:24 +01:00
mp_int_t u_imm ;
2013-10-08 22:18:32 +01:00
} ;
} stack_info_t ;
2013-10-08 09:05:10 +01:00
struct _emit_t {
int pass ;
bool do_viper_types ;
2013-10-08 22:18:32 +01:00
2014-08-15 16:45:41 +01:00
vtype_kind_t return_vtype ;
2014-05-07 17:24:22 +01:00
uint local_vtype_alloc ;
2013-10-08 09:05:10 +01:00
vtype_kind_t * local_vtype ;
2013-10-08 22:18:32 +01:00
2014-05-07 17:24:22 +01:00
uint stack_info_alloc ;
2013-10-08 22:18:32 +01:00
stack_info_t * stack_info ;
2013-10-08 09:05:10 +01:00
int stack_start ;
int stack_size ;
bool last_emit_was_return_value ;
scope_t * scope ;
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
asm_x64_t * as ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01:00
asm_thumb_t * as ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_t * as ;
2013-10-08 09:05:10 +01:00
# endif
} ;
emit_t * EXPORT_FUN ( new ) ( uint max_num_labels ) {
2014-05-07 17:24:22 +01:00
emit_t * emit = m_new0 ( emit_t , 1 ) ;
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
emit - > as = asm_x64_new ( max_num_labels ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01:00
emit - > as = asm_thumb_new ( max_num_labels ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
emit - > as = asm_arm_new ( max_num_labels ) ;
2013-10-08 09:05:10 +01:00
# endif
return emit ;
}
2014-01-24 22:42:28 +00:00
void EXPORT_FUN ( free ) ( emit_t * emit ) {
2014-01-24 14:20:11 +00:00
# if N_X64
asm_x64_free ( emit - > as , false ) ;
# elif N_THUMB
asm_thumb_free ( emit - > as , false ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_free ( emit - > as , false ) ;
2014-01-24 14:20:11 +00:00
# endif
2014-05-07 17:24:22 +01: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 14:20:11 +00:00
m_del_obj ( emit_t , emit ) ;
}
2014-08-15 16:45:41 +01: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 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_start_pass ( emit_t * emit , pass_kind_t pass , scope_t * scope ) {
2013-10-08 09:05:10 +01: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 17:24:22 +01: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 22:18:32 +01:00
}
2014-05-07 17:24:22 +01: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 22:18:32 +01:00
if ( emit - > stack_info = = NULL ) {
2014-05-07 17:24:22 +01:00
emit - > stack_info_alloc = scope - > stack_size + 50 ;
2013-10-08 22:18:32 +01:00
emit - > stack_info = m_new ( stack_info_t , emit - > stack_info_alloc ) ;
2013-10-08 09:05:10 +01:00
}
2014-08-15 16:45:41 +01:00
// set default type for return and arguments
emit - > return_vtype = VTYPE_PYOBJ ;
2014-08-15 22:39:08 +01:00
for ( mp_uint_t i = 0 ; i < emit - > scope - > num_pos_args ; i + + ) {
2014-08-15 16:45:41 +01:00
emit - > local_vtype [ i ] = VTYPE_PYOBJ ;
}
2014-08-15 22:39:08 +01: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 16:45:41 +01:00
emit - > stack_info [ i ] . kind = STACK_VALUE ;
2014-08-15 22:39:08 +01:00
emit - > stack_info [ i ] . vtype = VTYPE_UNBOUND ;
2013-10-08 09:05:10 +01:00
}
2013-10-12 16:53:13 +01:00
# if N_X64
2014-05-07 17:24:22 +01:00
asm_x64_start_pass ( emit - > as , pass = = MP_PASS_EMIT ? ASM_X64_PASS_EMIT : ASM_X64_PASS_COMPUTE ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2014-05-07 17:24:22 +01:00
asm_thumb_start_pass ( emit - > as , pass = = MP_PASS_EMIT ? ASM_THUMB_PASS_EMIT : ASM_THUMB_PASS_COMPUTE ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_start_pass ( emit - > as , pass = = MP_PASS_EMIT ? ASM_ARM_PASS_EMIT : ASM_ARM_PASS_COMPUTE ) ;
2013-10-08 09:05:10 +01:00
# endif
// entry to function
int num_locals = 0 ;
2014-05-07 17:24:22 +01:00
if ( pass > MP_PASS_SCOPE ) {
2013-10-08 09:05:10 +01: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 ;
}
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
asm_x64_entry ( emit - > as , num_locals ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01:00
asm_thumb_entry ( emit - > as , num_locals ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_entry ( emit - > as , num_locals ) ;
2013-10-08 09:05:10 +01:00
# endif
// initialise locals from parameters
2013-10-12 16:53:13 +01:00
# if N_X64
2014-04-27 15:50:52 +01:00
for ( int i = 0 ; i < scope - > num_pos_args ; i + + ) {
2013-10-08 09:05:10 +01:00
if ( i = = 0 ) {
asm_x64_mov_r64_to_r64 ( emit - > as , REG_ARG_1 , REG_LOCAL_1 ) ;
} else if ( i = = 1 ) {
2014-05-07 23:27:45 +01:00
asm_x64_mov_r64_to_local ( emit - > as , REG_ARG_2 , i - REG_LOCAL_NUM ) ;
2013-10-08 09:05:10 +01:00
} else if ( i = = 2 ) {
2014-05-07 23:27:45 +01:00
asm_x64_mov_r64_to_local ( emit - > as , REG_ARG_3 , i - REG_LOCAL_NUM ) ;
2013-10-08 09:05:10 +01:00
} else {
// TODO not implemented
assert ( 0 ) ;
}
}
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2014-04-27 15:50:52 +01:00
for ( int i = 0 ; i < scope - > num_pos_args ; i + + ) {
2013-10-08 09:05:10 +01:00
if ( i = = 0 ) {
asm_thumb_mov_reg_reg ( emit - > as , REG_LOCAL_1 , REG_ARG_1 ) ;
} else if ( i = = 1 ) {
asm_thumb_mov_reg_reg ( emit - > as , REG_LOCAL_2 , REG_ARG_2 ) ;
} else if ( i = = 2 ) {
asm_thumb_mov_reg_reg ( emit - > as , REG_LOCAL_3 , REG_ARG_3 ) ;
} else if ( i = = 3 ) {
asm_thumb_mov_local_reg ( emit - > as , i - REG_LOCAL_NUM , REG_ARG_4 ) ;
} else {
// TODO not implemented
assert ( 0 ) ;
}
}
2014-07-03 13:25:24 +01:00
asm_thumb_mov_reg_i32 ( emit - > as , REG_R7 , ( mp_uint_t ) mp_fun_table ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
for ( int i = 0 ; i < scope - > num_pos_args ; i + + ) {
if ( i = = 0 ) {
asm_arm_mov_reg_reg ( emit - > as , REG_LOCAL_1 , REG_ARG_1 ) ;
} else if ( i = = 1 ) {
asm_arm_mov_reg_reg ( emit - > as , REG_LOCAL_2 , REG_ARG_2 ) ;
} else if ( i = = 2 ) {
asm_arm_mov_reg_reg ( emit - > as , REG_LOCAL_3 , REG_ARG_3 ) ;
} else if ( i = = 3 ) {
asm_arm_mov_local_reg ( emit - > as , i - REG_LOCAL_NUM , REG_ARG_4 ) ;
} else {
// TODO not implemented
assert ( 0 ) ;
}
}
asm_arm_mov_reg_i32 ( emit - > as , REG_R7 , ( mp_uint_t ) mp_fun_table ) ;
2013-10-08 09:05:10 +01:00
# endif
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_end_pass ( emit_t * emit ) {
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
if ( ! emit - > last_emit_was_return_value ) {
asm_x64_exit ( emit - > as ) ;
}
asm_x64_end_pass ( emit - > as ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01:00
if ( ! emit - > last_emit_was_return_value ) {
asm_thumb_exit ( emit - > as ) ;
}
asm_thumb_end_pass ( emit - > as ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
if ( ! emit - > last_emit_was_return_value ) {
asm_arm_exit ( emit - > as ) ;
}
asm_arm_end_pass ( emit - > as ) ;
2013-10-08 09:05:10 +01:00
# endif
// check stack is back to zero size
if ( emit - > stack_size ! = 0 ) {
printf ( " ERROR: stack size not back to zero; got %d \n " , emit - > stack_size ) ;
}
2014-05-07 17:24:22 +01:00
if ( emit - > pass = = MP_PASS_EMIT ) {
2013-10-12 16:53:13 +01:00
# if N_X64
2013-12-21 18:17:45 +00:00
void * f = asm_x64_get_code ( emit - > as ) ;
2014-08-15 16:45:41 +01:00
mp_uint_t f_len = asm_x64_get_code_size ( emit - > as ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-12-21 18:17:45 +00:00
void * f = asm_thumb_get_code ( emit - > as ) ;
2014-08-15 16:45:41 +01:00
mp_uint_t f_len = asm_thumb_get_code_size ( emit - > as ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
void * f = asm_arm_get_code ( emit - > as ) ;
mp_uint_t f_len = asm_arm_get_code_size ( emit - > as ) ;
2013-10-08 09:05:10 +01:00
# endif
2014-08-15 16:45:41 +01: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 09:05:10 +01:00
}
}
2014-02-12 16:31:30 +00:00
STATIC bool emit_native_last_emit_was_return_value ( emit_t * emit ) {
2013-10-08 09:05:10 +01:00
return emit - > last_emit_was_return_value ;
}
2014-04-10 18:28:54 +01:00
STATIC void emit_native_adjust_stack_size ( emit_t * emit , int delta ) {
emit - > stack_size + = delta ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_set_source_line ( emit_t * emit , int source_line ) {
2014-01-18 23:24:36 +00:00
}
2014-02-12 16:31:30 +00:00
STATIC void adjust_stack ( emit_t * emit , int stack_size_delta ) {
2014-04-06 12:58:40 +01:00
DEBUG_printf ( " adjust stack: stack:%d + delta:%d \n " , emit - > stack_size , stack_size_delta ) ;
2014-04-09 12:27:39 +01:00
assert ( ( int ) emit - > stack_size + stack_size_delta > = 0 ) ;
2013-10-08 09:05:10 +01:00
emit - > stack_size + = stack_size_delta ;
2014-05-07 17:24:22 +01:00
if ( emit - > pass > MP_PASS_SCOPE & & emit - > stack_size > emit - > scope - > stack_size ) {
2013-10-08 09:05:10 +01:00
emit - > scope - > stack_size = emit - > stack_size ;
}
}
2013-10-08 22:18:32 +01:00
/*
2014-02-12 16:31:30 +00:00
STATIC void emit_pre_raw ( emit_t * emit , int stack_size_delta ) {
2013-10-08 09:05:10 +01:00
adjust_stack ( emit , stack_size_delta ) ;
emit - > last_emit_was_return_value = false ;
}
2013-10-08 22:18:32 +01:00
*/
2013-10-08 09:05:10 +01:00
2013-10-08 22:18:32 +01:00
// this must be called at start of emit functions
2014-03-27 23:30:26 +00:00
STATIC void emit_native_pre ( emit_t * emit ) {
2013-10-08 22:18:32 +01: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 ;
ASM_MOV_REG_TO_LOCAL ( emit - > stack_info [ i ] . u_reg , emit - > stack_start + i ) ;
break ;
case STACK_IMM :
// don't think we ever need to push imms for settling
//ASM_MOV_IMM_TO_LOCAL(emit->last_imm, emit->stack_start + i);
break ;
}
}
}
*/
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC vtype_kind_t peek_vtype ( emit_t * emit ) {
2013-10-08 22:18:32 +01:00
return emit - > stack_info [ emit - > stack_size - 1 ] . vtype ;
}
2013-10-08 09:05:10 +01:00
2013-10-16 23:58:48 +01:00
// pos=1 is TOS, pos=2 is next, etc
// use pos=0 for no skipping
2014-02-12 16:31:30 +00:00
STATIC void need_reg_single ( emit_t * emit , int reg_needed , int skip_stack_pos ) {
2013-10-16 23:58:48 +01: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 ;
ASM_MOV_REG_TO_LOCAL ( si - > u_reg , emit - > stack_start + i ) ;
}
}
}
}
2014-02-12 16:31:30 +00:00
STATIC void need_reg_all ( emit_t * emit ) {
2013-10-08 22:18:32 +01:00
for ( int i = 0 ; i < emit - > stack_size ; i + + ) {
stack_info_t * si = & emit - > stack_info [ i ] ;
2013-10-16 23:58:48 +01:00
if ( si - > kind = = STACK_REG ) {
2013-10-08 22:18:32 +01:00
si - > kind = STACK_VALUE ;
ASM_MOV_REG_TO_LOCAL ( si - > u_reg , emit - > stack_start + i ) ;
}
}
}
2013-10-08 09:05:10 +01:00
2014-02-12 16:31:30 +00:00
STATIC void need_stack_settled ( emit_t * emit ) {
2013-10-08 22:18:32 +01:00
for ( int i = 0 ; i < emit - > stack_size ; i + + ) {
stack_info_t * si = & emit - > stack_info [ i ] ;
if ( si - > kind = = STACK_REG ) {
si - > kind = STACK_VALUE ;
ASM_MOV_REG_TO_LOCAL ( si - > u_reg , emit - > stack_start + i ) ;
}
2013-10-08 09:05:10 +01:00
}
2013-10-16 23:58:48 +01:00
for ( int i = 0 ; i < emit - > stack_size ; i + + ) {
stack_info_t * si = & emit - > stack_info [ i ] ;
if ( si - > kind = = STACK_IMM ) {
ASM_MOV_IMM_TO_LOCAL_USING ( si - > u_imm , emit - > stack_start + i , REG_TEMP0 ) ;
}
}
2013-10-08 09:05:10 +01:00
}
2013-10-16 23:58:48 +01:00
// pos=1 is TOS, pos=2 is next, etc
2014-02-12 16:31:30 +00:00
STATIC void emit_access_stack ( emit_t * emit , int pos , vtype_kind_t * vtype , int reg_dest ) {
2013-10-16 23:58:48 +01:00
need_reg_single ( emit , reg_dest , pos ) ;
stack_info_t * si = & emit - > stack_info [ emit - > stack_size - pos ] ;
2013-10-08 22:18:32 +01:00
* vtype = si - > vtype ;
switch ( si - > kind ) {
case STACK_VALUE :
2013-10-16 23:58:48 +01:00
ASM_MOV_LOCAL_TO_REG ( emit - > stack_start + emit - > stack_size - pos , reg_dest ) ;
2013-10-08 09:05:10 +01:00
break ;
2013-10-08 22:18:32 +01:00
case STACK_REG :
if ( si - > u_reg ! = reg_dest ) {
ASM_MOV_REG_TO_REG ( si - > u_reg , reg_dest ) ;
2013-10-08 09:05:10 +01:00
}
break ;
2013-10-08 22:18:32 +01:00
case STACK_IMM :
ASM_MOV_IMM_TO_REG ( si - > u_imm , reg_dest ) ;
2013-10-08 09:05:10 +01:00
break ;
}
}
2014-05-10 13:40:46 +01:00
STATIC void emit_pre_pop_discard ( emit_t * emit , vtype_kind_t * vtype ) {
emit - > last_emit_was_return_value = false ;
adjust_stack ( emit , - 1 ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_pre_pop_reg ( emit_t * emit , vtype_kind_t * vtype , int reg_dest ) {
2013-10-16 23:58:48 +01:00
emit - > last_emit_was_return_value = false ;
emit_access_stack ( emit , 1 , vtype , reg_dest ) ;
adjust_stack ( emit , - 1 ) ;
}
2014-02-12 16:31:30 +00: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 09:05:10 +01:00
emit_pre_pop_reg ( emit , vtypea , rega ) ;
2013-10-08 22:18:32 +01:00
emit_pre_pop_reg ( emit , vtypeb , regb ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00: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 09:05:10 +01:00
emit_pre_pop_reg ( emit , vtypea , rega ) ;
2013-10-08 22:18:32 +01:00
emit_pre_pop_reg ( emit , vtypeb , regb ) ;
emit_pre_pop_reg ( emit , vtypec , regc ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_post ( emit_t * emit ) {
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_post_push_reg ( emit_t * emit , vtype_kind_t vtype , int reg ) {
2013-10-08 22:18:32 +01: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 09:05:10 +01:00
}
2014-07-03 13:25:24 +01:00
STATIC void emit_post_push_imm ( emit_t * emit , vtype_kind_t vtype , mp_int_t imm ) {
2013-10-08 22:18:32 +01: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 09:05:10 +01:00
}
2014-02-12 16:31:30 +00: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 22:18:32 +01:00
emit_post_push_reg ( emit , vtypea , rega ) ;
emit_post_push_reg ( emit , vtypeb , regb ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00: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 22:18:32 +01: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 09:05:10 +01:00
}
2014-02-12 16:31:30 +00: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 22:18:32 +01: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 09:05:10 +01:00
}
2014-08-16 22:31:57 +01:00
STATIC void emit_call ( emit_t * emit , mp_fun_kind_t fun_kind ) {
2013-10-16 23:58:48 +01:00
need_reg_all ( emit ) ;
2013-10-12 16:53:13 +01:00
# if N_X64
2014-08-16 22:31:57 +01:00
asm_x64_call_ind ( emit - > as , mp_fun_table [ fun_kind ] , REG_RAX ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2014-03-30 13:35:08 +01:00
asm_thumb_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2013-10-08 09:05:10 +01:00
# endif
}
2014-08-16 22:31:57 +01: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 22:06:54 +01:00
need_reg_all ( emit ) ;
2013-10-08 09:05:10 +01:00
ASM_MOV_IMM_TO_REG ( arg_val , arg_reg ) ;
2013-10-16 23:58:48 +01:00
# if N_X64
2014-08-16 22:31:57 +01:00
asm_x64_call_ind ( emit - > as , mp_fun_table [ fun_kind ] , REG_RAX ) ;
2013-10-16 23:58:48 +01:00
# elif N_THUMB
2014-03-30 13:35:08 +01:00
asm_thumb_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2013-10-16 23:58:48 +01:00
# endif
2013-10-08 09:05:10 +01:00
}
2014-07-03 13:25:24 +01:00
// the first arg is stored in the code aligned on a mp_uint_t boundary
2014-08-16 22:31:57 +01: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 18:30:52 +01:00
need_reg_all ( emit ) ;
ASM_MOV_ALIGNED_IMM_TO_REG ( arg_val , arg_reg ) ;
# if N_X64
2014-08-16 22:31:57 +01:00
asm_x64_call_ind ( emit - > as , mp_fun_table [ fun_kind ] , REG_RAX ) ;
2014-05-07 18:30:52 +01:00
# elif N_THUMB
asm_thumb_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2014-05-07 18:30:52 +01:00
# endif
}
2014-08-16 22:31:57 +01: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 13:11:48 +00:00
need_reg_all ( emit ) ;
ASM_MOV_IMM_TO_REG ( arg_val1 , arg_reg1 ) ;
ASM_MOV_IMM_TO_REG ( arg_val2 , arg_reg2 ) ;
# if N_X64
2014-08-16 22:31:57 +01:00
asm_x64_call_ind ( emit - > as , mp_fun_table [ fun_kind ] , REG_RAX ) ;
2014-02-02 13:11:48 +00:00
# elif N_THUMB
2014-03-30 13:35:08 +01:00
asm_thumb_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2014-02-02 13:11:48 +00:00
# endif
}
2014-07-03 13:25:24 +01:00
// the first arg is stored in the code aligned on a mp_uint_t boundary
2014-08-16 22:31:57 +01: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 12:58:40 +01:00
need_reg_all ( emit ) ;
2014-05-07 17:24:22 +01:00
ASM_MOV_ALIGNED_IMM_TO_REG ( arg_val1 , arg_reg1 ) ;
2014-04-06 12:58:40 +01:00
ASM_MOV_IMM_TO_REG ( arg_val2 , arg_reg2 ) ;
ASM_MOV_IMM_TO_REG ( arg_val3 , arg_reg3 ) ;
# if N_X64
2014-08-16 22:31:57 +01:00
asm_x64_call_ind ( emit - > as , mp_fun_table [ fun_kind ] , REG_RAX ) ;
2014-04-06 12:58:40 +01:00
# elif N_THUMB
asm_thumb_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_bl_ind ( emit - > as , mp_fun_table [ fun_kind ] , fun_kind , REG_R3 ) ;
2014-04-06 12:58:40 +01:00
# endif
}
2014-08-16 22:06:11 +01: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 09:05:10 +01:00
2014-08-16 22:06:11 +01: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 :
ASM_MOV_IMM_TO_LOCAL_USING ( si - > u_imm , emit - > stack_start + emit - > stack_size - 1 - i , reg_dest ) ;
break ;
case VTYPE_BOOL :
if ( si - > u_imm = = 0 ) {
ASM_MOV_IMM_TO_LOCAL_USING ( ( mp_uint_t ) mp_const_false , emit - > stack_start + emit - > stack_size - 1 - i , reg_dest ) ;
} else {
ASM_MOV_IMM_TO_LOCAL_USING ( ( mp_uint_t ) mp_const_true , emit - > stack_start + emit - > stack_size - 1 - i , reg_dest ) ;
}
si - > vtype = VTYPE_PYOBJ ;
break ;
case VTYPE_INT :
case VTYPE_UINT :
ASM_MOV_IMM_TO_LOCAL_USING ( ( si - > u_imm < < 1 ) | 1 , emit - > stack_start + emit - > stack_size - 1 - i , reg_dest ) ;
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 ;
ASM_MOV_LOCAL_TO_REG ( local_num , REG_ARG_1 ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_CONVERT_NATIVE_TO_OBJ , si - > vtype , REG_ARG_2 ) ; // arg2 = type
2014-08-16 22:06:11 +01:00
ASM_MOV_REG_TO_LOCAL ( REG_RET , local_num ) ;
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 ) ;
ASM_MOV_LOCAL_ADDR_TO_REG ( emit - > stack_start + emit - > stack_size , reg_dest ) ;
}
// 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 09:05:10 +01:00
}
2014-08-16 22:06:11 +01:00
ASM_MOV_LOCAL_ADDR_TO_REG ( emit - > stack_start + emit - > stack_size , reg_dest ) ;
adjust_stack ( emit , n_push ) ;
}
STATIC void emit_native_load_id ( emit_t * emit , qstr qstr ) {
emit_common_load_id ( emit , & EXPORT_FUN ( method_table ) , emit - > scope , qstr ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_store_id ( emit_t * emit , qstr qstr ) {
2013-10-08 09:05:10 +01:00
emit_common_store_id ( emit , & EXPORT_FUN ( method_table ) , emit - > scope , qstr ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_delete_id ( emit_t * emit , qstr qstr ) {
2013-10-08 09:05:10 +01:00
emit_common_delete_id ( emit , & EXPORT_FUN ( method_table ) , emit - > scope , qstr ) ;
}
2014-04-10 14:11:31 +01:00
STATIC void emit_native_label_assign ( emit_t * emit , uint l ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2013-10-16 23:58:48 +01:00
// need to commit stack because we can jump here from elsewhere
need_stack_settled ( emit ) ;
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
asm_x64_label_assign ( emit - > as , l ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01:00
asm_thumb_label_assign ( emit - > as , l ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_label_assign ( emit - > as , l ) ;
2013-10-08 09:05:10 +01:00
# endif
2013-11-02 20:34:54 +00:00
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-04-06 12:58:40 +01: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 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_IMPORT_NAME , qst , REG_ARG_1 ) ; // arg1 = import name
2014-04-06 12:58:40 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 09:05:10 +01:00
}
2014-04-06 12:58:40 +01: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 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_IMPORT_FROM , qst , REG_ARG_2 ) ; // arg2 = import name
2014-04-06 12:58:40 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_import_star ( emit_t * emit ) {
2014-04-06 12:58:40 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_IMPORT_ALL ) ;
2014-04-06 12:58:40 +01:00
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_load_const_tok ( emit_t * emit , mp_token_kind_t tok ) {
2014-04-06 12:58:40 +01:00
DEBUG_printf ( " load_const_tok %d \n " , tok ) ;
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2013-10-08 09:05:10 +01:00
int vtype ;
2014-07-03 13:25:24 +01:00
mp_uint_t val ;
2013-10-08 09:05:10 +01:00
if ( emit - > do_viper_types ) {
switch ( tok ) {
2013-12-21 18:17:45 +00: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 09:05:10 +01:00
default : assert ( 0 ) ; vtype = 0 ; val = 0 ; // shouldn't happen
}
} else {
vtype = VTYPE_PYOBJ ;
switch ( tok ) {
2014-07-03 13:25:24 +01: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 09:05:10 +01:00
default : assert ( 0 ) ; vtype = 0 ; val = 0 ; // shouldn't happen
}
}
emit_post_push_imm ( emit , vtype , val ) ;
}
2014-07-03 13:25:24 +01:00
STATIC void emit_native_load_const_small_int ( emit_t * emit , mp_int_t arg ) {
2014-04-06 12:58:40 +01:00
DEBUG_printf ( " load_const_small_int %d \n " , arg ) ;
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2013-10-08 09:05:10 +01: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 12:58:40 +01:00
STATIC void emit_native_load_const_int ( emit_t * emit , qstr qst ) {
DEBUG_printf ( " load_const_int %s \n " , qstr_str ( st ) ) ;
// for viper: load integer, check fits in 32 bits
emit_native_pre ( emit ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_CONST_INT , qst , REG_ARG_1 ) ;
2014-04-06 12:58:40 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_load_const_dec ( emit_t * emit , qstr qstr ) {
2013-11-02 20:34:54 +00:00
// for viper, a float/complex is just a Python object
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_CONST_DEC , qstr , REG_ARG_1 ) ;
2013-11-02 20:34:54 +00:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_load_const_str ( emit_t * emit , qstr qstr , bool bytes ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2013-10-08 09:05:10 +01:00
if ( emit - > do_viper_types ) {
// not implemented properly
// load a pointer to the asciiz string?
assert ( 0 ) ;
2014-07-03 13:25:24 +01:00
emit_post_push_imm ( emit , VTYPE_PTR , ( mp_uint_t ) qstr_str ( qstr ) ) ;
2013-10-08 09:05:10 +01:00
} else {
2014-06-30 05:17:25 +01:00
if ( bytes ) {
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_CONST_BYTES , qstr , REG_ARG_1 ) ;
2014-06-30 05:17:25 +01:00
} else {
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_CONST_STR , qstr , REG_ARG_1 ) ;
2014-06-30 05:17:25 +01:00
}
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
}
2014-04-20 17:50:40 +01:00
STATIC void emit_native_load_null ( emit_t * emit ) {
emit_native_pre ( emit ) ;
emit_post_push_imm ( emit , VTYPE_PYOBJ , 0 ) ;
}
2014-04-09 15:26:46 +01:00
STATIC void emit_native_load_fast ( emit_t * emit , qstr qstr , uint id_flags , int local_num ) {
2013-10-08 09:05:10 +01:00
vtype_kind_t vtype = emit - > local_vtype [ local_num ] ;
if ( vtype = = VTYPE_UNBOUND ) {
printf ( " ViperTypeError: local %s used before type known \n " , qstr_str ( qstr ) ) ;
}
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
if ( local_num = = 0 ) {
emit_post_push_reg ( emit , vtype , REG_LOCAL_1 ) ;
} else {
2013-10-16 23:58:48 +01:00
need_reg_single ( emit , REG_RAX , 0 ) ;
2014-05-07 23:27:45 +01:00
asm_x64_mov_local_to_r64 ( emit - > as , local_num - REG_LOCAL_NUM , REG_RAX ) ;
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , vtype , REG_RAX ) ;
}
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01: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 {
2013-10-16 23:58:48 +01:00
need_reg_single ( emit , REG_R0 , 0 ) ;
2014-05-07 23:27:45 +01:00
asm_thumb_mov_reg_local ( emit - > as , REG_R0 , local_num - REG_LOCAL_NUM ) ;
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , vtype , REG_R0 ) ;
}
2014-08-16 21:55:53 +01: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 {
need_reg_single ( emit , REG_R0 , 0 ) ;
asm_arm_mov_reg_local ( emit - > as , REG_R0 , local_num - REG_LOCAL_NUM ) ;
emit_post_push_reg ( emit , vtype , REG_R0 ) ;
}
2013-10-08 09:05:10 +01:00
# endif
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_load_deref ( emit_t * emit , qstr qstr , int local_num ) {
2013-12-11 00:41:43 +00:00
// not implemented
// in principle could support this quite easily (ldr r0, [r0, #0]) and then get closed over variables!
assert ( 0 ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_load_name ( emit_t * emit , qstr qstr ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_NAME , qstr , REG_ARG_1 ) ;
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_load_global ( emit_t * emit , qstr qstr ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_GLOBAL , qstr , REG_ARG_1 ) ;
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_load_attr ( emit_t * emit , qstr qstr ) {
2013-10-08 09:05:10 +01:00
// depends on type of subject:
// - integer, function, pointer to integers: error
// - pointer to structure: get member, quite easy
2014-03-30 13:35:08 +01:00
// - Python object: call mp_load_attr, and needs to be typed to convert result
2013-10-08 09:05:10 +01:00
vtype_kind_t vtype_base ;
emit_pre_pop_reg ( emit , & vtype_base , REG_ARG_1 ) ; // arg1 = base
assert ( vtype_base = = VTYPE_PYOBJ ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_ATTR , qstr , REG_ARG_2 ) ; // arg2 = attribute name
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_load_method ( emit_t * emit , qstr qstr ) {
2013-10-08 09:05:10 +01: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-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_LOAD_METHOD , qstr , REG_ARG_2 ) ; // arg2 = method name
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_load_build_class ( emit_t * emit ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-08-16 22:31:57 +01:00
emit_call ( emit , MP_F_LOAD_BUILD_CLASS ) ;
2013-10-10 11:24:39 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 09:05:10 +01:00
}
2014-04-17 22:10:53 +01: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 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_OBJ_SUBSCR , ( mp_uint_t ) MP_OBJ_SENTINEL , REG_ARG_3 ) ;
2014-04-17 22:10:53 +01: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-02-12 16:31:30 +00:00
STATIC void emit_native_store_fast ( emit_t * emit , qstr qstr , int local_num ) {
2013-10-08 09:05:10 +01:00
vtype_kind_t vtype ;
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
if ( local_num = = 0 ) {
emit_pre_pop_reg ( emit , & vtype , REG_LOCAL_1 ) ;
} else {
emit_pre_pop_reg ( emit , & vtype , REG_RAX ) ;
2014-05-07 23:27:45 +01:00
asm_x64_mov_r64_to_local ( emit - > as , REG_RAX , local_num - REG_LOCAL_NUM ) ;
2013-10-08 09:05:10 +01:00
}
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01: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 {
emit_pre_pop_reg ( emit , & vtype , REG_R0 ) ;
2014-05-07 23:27:45 +01:00
asm_thumb_mov_local_reg ( emit - > as , local_num - REG_LOCAL_NUM , REG_R0 ) ;
2013-10-08 09:05:10 +01:00
}
2014-08-16 21:55:53 +01: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 {
emit_pre_pop_reg ( emit , & vtype , REG_R0 ) ;
asm_arm_mov_local_reg ( emit - > as , local_num - REG_LOCAL_NUM , REG_R0 ) ;
}
2013-10-08 09:05:10 +01: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
printf ( " ViperTypeError: type mismatch, local %s has type %d but source object has type %d \n " , qstr_str ( qstr ) , emit - > local_vtype [ local_num ] , vtype ) ;
}
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_store_deref ( emit_t * emit , qstr qstr , int local_num ) {
2013-12-11 00:41:43 +00:00
// not implemented
assert ( 0 ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_store_name ( emit_t * emit , qstr qstr ) {
2014-03-30 13:35:08 +01:00
// mp_store_name, but needs conversion of object (maybe have mp_viper_store_name(obj, type))
2013-10-08 09:05:10 +01:00
vtype_kind_t vtype ;
emit_pre_pop_reg ( emit , & vtype , REG_ARG_2 ) ;
assert ( vtype = = VTYPE_PYOBJ ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_STORE_NAME , qstr , REG_ARG_1 ) ; // arg1 = name
2013-10-08 09:05:10 +01:00
emit_post ( emit ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_store_global ( emit_t * emit , qstr qstr ) {
2014-08-15 23:47:59 +01:00
vtype_kind_t vtype = peek_vtype ( emit ) ;
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 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_CONVERT_NATIVE_TO_OBJ , vtype , REG_ARG_2 ) ; // arg2 = type
2014-08-15 23:47:59 +01:00
ASM_MOV_REG_TO_REG ( REG_RET , REG_ARG_2 ) ;
}
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_STORE_GLOBAL , qstr , REG_ARG_1 ) ; // arg1 = name
2014-08-15 23:47:59 +01:00
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_store_attr ( emit_t * emit , qstr qstr ) {
2013-10-10 11:24:39 +01: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-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_STORE_ATTR , qstr , REG_ARG_2 ) ; // arg2 = attribute name
2013-10-10 11:24:39 +01:00
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_store_subscr ( emit_t * emit ) {
2013-10-08 09:05:10 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_OBJ_SUBSCR ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_delete_fast ( emit_t * emit , qstr qstr , int local_num ) {
2013-10-08 09:05:10 +01:00
// not implemented
// could support for Python types, just set to None (so GC can reclaim it)
assert ( 0 ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_delete_deref ( emit_t * emit , qstr qstr , int local_num ) {
2013-12-11 00:41:43 +00:00
// not supported
assert ( 0 ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_delete_name ( emit_t * emit , qstr qstr ) {
2013-10-08 09:05:10 +01:00
// not implemented
2014-03-30 13:35:08 +01:00
// use mp_delete_name
2013-10-08 09:05:10 +01:00
assert ( 0 ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_delete_global ( emit_t * emit , qstr qstr ) {
2013-10-08 09:05:10 +01:00
// not implemented
2014-03-30 13:35:08 +01:00
// use mp_delete_global
2013-10-08 09:05:10 +01:00
assert ( 0 ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_delete_attr ( emit_t * emit , qstr qstr ) {
2014-06-22 18:35:04 +01:00
vtype_kind_t vtype_base ;
emit_pre_pop_reg ( emit , & vtype_base , REG_ARG_1 ) ; // arg1 = base
assert ( vtype_base = = VTYPE_PYOBJ ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_2_imm_args ( emit , MP_F_STORE_ATTR , qstr , REG_ARG_2 , ( mp_uint_t ) MP_OBJ_NULL , REG_ARG_3 ) ; // arg2 = attribute name, arg3 = value (null for delete)
2014-06-22 18:35:04 +01:00
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_delete_subscr ( emit_t * emit ) {
2014-04-17 22:10:53 +01: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 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_OBJ_SUBSCR , ( mp_uint_t ) MP_OBJ_NULL , REG_ARG_3 ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_dup_top ( emit_t * emit ) {
2013-10-08 09:05:10 +01: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 16:31:30 +00:00
STATIC void emit_native_dup_top_two ( emit_t * emit ) {
2013-10-08 09:05:10 +01: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 16:31:30 +00:00
STATIC void emit_native_pop_top ( emit_t * emit ) {
2013-10-08 09:05:10 +01:00
vtype_kind_t vtype ;
2014-05-10 13:40:46 +01:00
emit_pre_pop_discard ( emit , & vtype ) ;
2013-10-08 09:05:10 +01:00
emit_post ( emit ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_rot_two ( emit_t * emit ) {
2013-10-08 22:18:32 +01: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 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_rot_three ( emit_t * emit ) {
2013-10-08 09:05:10 +01: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-04-10 14:11:31 +01:00
STATIC void emit_native_jump ( emit_t * emit , uint label ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-05-07 18:30:52 +01:00
// need to commit stack because we are jumping elsewhere
need_stack_settled ( emit ) ;
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
asm_x64_jmp_label ( emit - > as , label ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01:00
asm_thumb_b_label ( emit - > as , label ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_b_label ( emit - > as , label ) ;
2013-10-08 09:05:10 +01:00
# endif
emit_post ( emit ) ;
}
2014-05-07 18:30:52 +01:00
STATIC void emit_native_jump_helper ( emit_t * emit , uint label , bool pop ) {
2013-10-08 09:05:10 +01:00
vtype_kind_t vtype = peek_vtype ( emit ) ;
if ( vtype = = VTYPE_BOOL ) {
emit_pre_pop_reg ( emit , & vtype , REG_RET ) ;
2014-05-07 18:30:52 +01:00
if ( ! pop ) {
adjust_stack ( emit , 1 ) ;
}
2013-10-08 09:05:10 +01:00
} else if ( vtype = = VTYPE_PYOBJ ) {
emit_pre_pop_reg ( emit , & vtype , REG_ARG_1 ) ;
2014-08-16 22:31:57 +01:00
emit_call ( emit , MP_F_OBJ_IS_TRUE ) ;
2014-05-07 18:30:52 +01:00
if ( ! pop ) {
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2013-10-08 09:05:10 +01:00
} else {
printf ( " ViperTypeError: expecting a bool or pyobj, got %d \n " , vtype ) ;
assert ( 0 ) ;
}
2014-05-07 18:30:52 +01:00
// need to commit stack because we may jump elsewhere
need_stack_settled ( emit ) ;
}
STATIC void emit_native_pop_jump_if_true ( emit_t * emit , uint label ) {
emit_native_jump_helper ( emit , label , true ) ;
# if N_X64
asm_x64_test_r8_with_r8 ( emit - > as , REG_RET , REG_RET ) ;
asm_x64_jcc_label ( emit - > as , JCC_JNZ , label ) ;
# elif N_THUMB
asm_thumb_cmp_rlo_i8 ( emit - > as , REG_RET , 0 ) ;
asm_thumb_bcc_label ( emit - > as , THUMB_CC_NE , label ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_cmp_reg_i8 ( emit - > as , REG_RET , 0 ) ;
asm_arm_bcc_label ( emit - > as , ARM_CC_NE , label ) ;
2014-05-07 18:30:52 +01:00
# endif
emit_post ( emit ) ;
2013-11-03 13:58:19 +00:00
}
2014-04-10 14:11:31 +01:00
STATIC void emit_native_pop_jump_if_false ( emit_t * emit , uint label ) {
2014-05-07 18:30:52 +01:00
emit_native_jump_helper ( emit , label , true ) ;
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
asm_x64_test_r8_with_r8 ( emit - > as , REG_RET , REG_RET ) ;
asm_x64_jcc_label ( emit - > as , JCC_JZ , label ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-11-03 13:58:19 +00:00
asm_thumb_cmp_rlo_i8 ( emit - > as , REG_RET , 0 ) ;
asm_thumb_bcc_label ( emit - > as , THUMB_CC_EQ , label ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_cmp_reg_i8 ( emit - > as , REG_RET , 0 ) ;
asm_arm_bcc_label ( emit - > as , ARM_CC_EQ , label ) ;
2013-10-08 09:05:10 +01:00
# endif
emit_post ( emit ) ;
}
2014-05-07 18:30:52 +01:00
STATIC void emit_native_jump_if_true_or_pop ( emit_t * emit , uint label ) {
emit_native_jump_helper ( emit , label , false ) ;
2013-11-03 13:58:19 +00:00
# if N_X64
asm_x64_test_r8_with_r8 ( emit - > as , REG_RET , REG_RET ) ;
asm_x64_jcc_label ( emit - > as , JCC_JNZ , label ) ;
# elif N_THUMB
asm_thumb_cmp_rlo_i8 ( emit - > as , REG_RET , 0 ) ;
asm_thumb_bcc_label ( emit - > as , THUMB_CC_NE , label ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_cmp_reg_i8 ( emit - > as , REG_RET , 0 ) ;
asm_arm_bcc_label ( emit - > as , ARM_CC_NE , label ) ;
2013-11-03 13:58:19 +00:00
# endif
2014-05-07 18:30:52 +01:00
adjust_stack ( emit , - 1 ) ;
2013-11-03 13:58:19 +00:00
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2013-11-03 13:58:19 +00:00
2014-04-10 14:11:31 +01:00
STATIC void emit_native_jump_if_false_or_pop ( emit_t * emit , uint label ) {
2014-05-07 18:30:52 +01:00
emit_native_jump_helper ( emit , label , false ) ;
# if N_X64
asm_x64_test_r8_with_r8 ( emit - > as , REG_RET , REG_RET ) ;
asm_x64_jcc_label ( emit - > as , JCC_JZ , label ) ;
# elif N_THUMB
asm_thumb_cmp_rlo_i8 ( emit - > as , REG_RET , 0 ) ;
asm_thumb_bcc_label ( emit - > as , THUMB_CC_EQ , label ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_cmp_reg_i8 ( emit - > as , REG_RET , 0 ) ;
asm_arm_bcc_label ( emit - > as , ARM_CC_EQ , label ) ;
2014-05-07 18:30:52 +01:00
# endif
adjust_stack ( emit , - 1 ) ;
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-04-10 14:11:31 +01:00
STATIC void emit_native_break_loop ( emit_t * emit , uint label , int except_depth ) {
2014-05-30 15:20:41 +01:00
emit_native_jump ( emit , label & ~ MP_EMIT_BREAK_FROM_FOR ) ; // TODO properly
2013-10-08 09:05:10 +01:00
}
2014-05-07 18:30:52 +01:00
2014-04-10 14:11:31 +01:00
STATIC void emit_native_continue_loop ( emit_t * emit , uint label , int except_depth ) {
2014-05-07 18:30:52 +01:00
emit_native_jump ( emit , label ) ; // TODO properly
2013-10-08 09:05:10 +01:00
}
2014-05-07 18:30:52 +01:00
2014-04-10 14:11:31 +01:00
STATIC void emit_native_setup_with ( emit_t * emit , uint label ) {
2013-10-08 09:05:10 +01:00
// not supported, or could be with runtime call
assert ( 0 ) ;
}
2014-06-30 05:17:25 +01:00
2014-02-12 16:31:30 +00:00
STATIC void emit_native_with_cleanup ( emit_t * emit ) {
2013-10-08 09:05:10 +01:00
assert ( 0 ) ;
}
2014-06-30 05:17:25 +01:00
2014-04-10 14:11:31 +01:00
STATIC void emit_native_setup_except ( emit_t * emit , uint label ) {
2014-06-30 05:17:25 +01:00
emit_native_pre ( emit ) ;
// need to commit stack because we may jump elsewhere
need_stack_settled ( emit ) ;
2014-07-03 13:25:24 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_NLR_PUSH ) ;
2014-06-30 05:17:25 +01:00
# if N_X64
asm_x64_test_r8_with_r8 ( emit - > as , REG_RET , REG_RET ) ;
asm_x64_jcc_label ( emit - > as , JCC_JNZ , label ) ;
# elif N_THUMB
asm_thumb_cmp_rlo_i8 ( emit - > as , REG_RET , 0 ) ;
asm_thumb_bcc_label ( emit - > as , THUMB_CC_NE , label ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_cmp_reg_i8 ( emit - > as , REG_RET , 0 ) ;
asm_arm_bcc_label ( emit - > as , ARM_CC_NE , label ) ;
2014-06-30 05:17:25 +01:00
# endif
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-06-30 05:17:25 +01:00
2014-04-10 14:11:31 +01:00
STATIC void emit_native_setup_finally ( emit_t * emit , uint label ) {
2013-10-08 09:05:10 +01:00
assert ( 0 ) ;
}
2014-06-30 05:17:25 +01:00
2014-02-12 16:31:30 +00:00
STATIC void emit_native_end_finally ( emit_t * emit ) {
2014-06-30 05:17:25 +01:00
//assert(0);
2013-10-08 09:05:10 +01:00
}
2013-10-16 23:58:48 +01:00
2014-02-12 16:31:30 +00:00
STATIC void emit_native_get_iter ( emit_t * emit ) {
2013-10-08 09:05:10 +01: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 23:58:48 +01:00
vtype_kind_t vtype ;
emit_pre_pop_reg ( emit , & vtype , REG_ARG_1 ) ;
assert ( vtype = = VTYPE_PYOBJ ) ;
2014-08-16 22:31:57 +01:00
emit_call ( emit , MP_F_GETITER ) ;
2013-10-16 23:58:48 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-04-10 14:11:31 +01:00
STATIC void emit_native_for_iter ( emit_t * emit , uint label ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2013-10-16 23:58:48 +01:00
vtype_kind_t vtype ;
emit_access_stack ( emit , 1 , & vtype , REG_ARG_1 ) ;
assert ( vtype = = VTYPE_PYOBJ ) ;
2014-08-16 22:31:57 +01:00
emit_call ( emit , MP_F_ITERNEXT ) ;
2014-07-03 13:25:24 +01:00
ASM_MOV_IMM_TO_REG ( ( mp_uint_t ) MP_OBJ_STOP_ITERATION , REG_TEMP1 ) ;
2013-10-16 23:58:48 +01:00
# if N_X64
asm_x64_cmp_r64_with_r64 ( emit - > as , REG_RET , REG_TEMP1 ) ;
asm_x64_jcc_label ( emit - > as , JCC_JE , label ) ;
# elif N_THUMB
2014-04-13 00:30:32 +01:00
asm_thumb_cmp_rlo_rlo ( emit - > as , REG_RET , REG_TEMP1 ) ;
2013-11-03 14:25:43 +00:00
asm_thumb_bcc_label ( emit - > as , THUMB_CC_EQ , label ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_cmp_reg_i8 ( emit - > as , REG_RET , 0 ) ;
asm_arm_bcc_label ( emit - > as , ARM_CC_EQ , label ) ;
2013-10-16 23:58:48 +01:00
# endif
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 09:05:10 +01:00
}
2013-10-16 23:58:48 +01:00
2014-02-12 16:31:30 +00:00
STATIC void emit_native_for_iter_end ( emit_t * emit ) {
2013-10-16 23:58:48 +01:00
// adjust stack counter (we get here from for_iter ending, which popped the value for us)
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2013-10-16 23:58:48 +01:00
adjust_stack ( emit , - 1 ) ;
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_pop_block ( emit_t * emit ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-08-16 22:31:57 +01:00
emit_call ( emit , MP_F_NLR_POP ) ;
2014-07-03 13:25:24 +01:00
adjust_stack ( emit , - ( mp_int_t ) ( sizeof ( nlr_buf_t ) / sizeof ( mp_uint_t ) ) ) ;
2013-10-08 09:05:10 +01:00
emit_post ( emit ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_pop_except ( emit_t * emit ) {
2014-06-30 05:17:25 +01:00
/*
emit_native_pre ( emit ) ;
2014-08-16 22:31:57 +01:00
emit_call ( emit , MP_F_NLR_POP ) ;
2014-07-03 13:25:24 +01:00
adjust_stack ( emit , - ( mp_int_t ) ( sizeof ( nlr_buf_t ) / sizeof ( mp_uint_t ) ) ) ;
2014-06-30 05:17:25 +01:00
emit_post ( emit ) ;
*/
2013-10-08 09:05:10 +01:00
}
2014-03-30 13:35:08 +01:00
STATIC void emit_native_unary_op ( emit_t * emit , mp_unary_op_t op ) {
2014-06-30 05:17:25 +01:00
if ( op = = MP_UNARY_OP_NOT ) {
// we need to synthesise this operation
assert ( 0 ) ;
} else {
vtype_kind_t vtype ;
emit_pre_pop_reg ( emit , & vtype , REG_ARG_2 ) ;
assert ( vtype = = VTYPE_PYOBJ ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_UNARY_OP , op , REG_ARG_1 ) ;
2014-06-30 05:17:25 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2013-10-08 09:05:10 +01:00
}
2014-03-30 13:35:08 +01:00
STATIC void emit_native_binary_op ( emit_t * emit , mp_binary_op_t op ) {
2013-10-08 09:05:10 +01:00
vtype_kind_t vtype_lhs , vtype_rhs ;
emit_pre_pop_reg_reg ( emit , & vtype_rhs , REG_ARG_3 , & vtype_lhs , REG_ARG_2 ) ;
if ( vtype_lhs = = VTYPE_INT & & vtype_rhs = = VTYPE_INT ) {
2014-03-30 13:35:08 +01:00
if ( op = = MP_BINARY_OP_ADD | | op = = MP_BINARY_OP_INPLACE_ADD ) {
2013-10-12 16:53:13 +01:00
# if N_X64
2014-01-11 09:47:06 +00:00
asm_x64_add_r64_to_r64 ( emit - > as , REG_ARG_3 , REG_ARG_2 ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2014-04-12 17:54:52 +01:00
asm_thumb_add_rlo_rlo_rlo ( emit - > as , REG_ARG_2 , REG_ARG_2 , REG_ARG_3 ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_add_reg ( emit - > as , REG_ARG_2 , REG_ARG_2 , REG_ARG_3 ) ;
2013-10-08 09:05:10 +01:00
# endif
2014-01-11 09:47:06 +00:00
emit_post_push_reg ( emit , VTYPE_INT , REG_ARG_2 ) ;
2014-03-30 13:35:08 +01:00
} else if ( op = = MP_BINARY_OP_LESS ) {
2014-01-11 09:47:06 +00:00
# if N_X64
asm_x64_xor_r64_to_r64 ( emit - > as , REG_RET , REG_RET ) ;
asm_x64_cmp_r64_with_r64 ( emit - > as , REG_ARG_3 , REG_ARG_2 ) ;
asm_x64_setcc_r8 ( emit - > as , JCC_JL , REG_RET ) ;
# elif N_THUMB
2014-04-13 00:30:32 +01:00
asm_thumb_cmp_rlo_rlo ( emit - > as , REG_ARG_2 , REG_ARG_3 ) ;
2014-01-11 09:47:06 +00:00
asm_thumb_ite_ge ( emit - > as ) ;
2014-04-13 00:30:32 +01:00
asm_thumb_mov_rlo_i8 ( emit - > as , REG_RET , 0 ) ; // if r0 >= r1
asm_thumb_mov_rlo_i8 ( emit - > as , REG_RET , 1 ) ; // if r0 < r1
2014-08-16 21:55:53 +01:00
# elif N_ARM
asm_arm_less_op ( emit - > as , REG_ARG_2 , REG_ARG_3 ) ;
2014-01-11 09:47:06 +00:00
# endif
emit_post_push_reg ( emit , VTYPE_BOOL , REG_RET ) ;
} else {
// TODO other ops not yet implemented
assert ( 0 ) ;
}
2013-10-08 09:05:10 +01:00
} else if ( vtype_lhs = = VTYPE_PYOBJ & & vtype_rhs = = VTYPE_PYOBJ ) {
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_BINARY_OP , op , REG_ARG_1 ) ;
2013-10-08 09:05:10 +01: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 22:39:08 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
2013-10-08 09:05:10 +01:00
}
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_build_tuple ( emit_t * emit , int n_args ) {
2013-10-16 23:58:48 +01:00
// for viper: call runtime, with types of args
// if wrapped in byte_array, or something, allocates memory and fills it
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-02-02 13:11:48 +00:00
emit_get_stack_pointer_to_reg_for_pop ( emit , REG_ARG_2 , n_args ) ; // pointer to items
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_BUILD_TUPLE , n_args , REG_ARG_1 ) ;
2013-10-16 23:58:48 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // new tuple
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_build_list ( emit_t * emit , int n_args ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-02-02 13:11:48 +00:00
emit_get_stack_pointer_to_reg_for_pop ( emit , REG_ARG_2 , n_args ) ; // pointer to items
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_BUILD_LIST , n_args , REG_ARG_1 ) ;
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // new list
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_list_append ( emit_t * emit , int list_index ) {
2013-10-16 23:58:48 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_LIST_APPEND ) ;
2013-10-16 23:58:48 +01:00
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_build_map ( emit_t * emit , int n_args ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_BUILD_MAP , n_args , REG_ARG_1 ) ;
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // new map
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_store_map ( emit_t * emit ) {
2013-10-08 09:05:10 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_STORE_MAP ) ;
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // map
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_map_add ( emit_t * emit , int map_index ) {
2013-10-16 23:58:48 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_STORE_MAP ) ;
2013-10-16 23:58:48 +01:00
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_build_set ( emit_t * emit , int n_args ) {
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-02-02 13:11:48 +00:00
emit_get_stack_pointer_to_reg_for_pop ( emit , REG_ARG_2 , n_args ) ; // pointer to items
2014-08-16 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_BUILD_SET , n_args , REG_ARG_1 ) ;
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ; // new set
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_set_add ( emit_t * emit , int set_index ) {
2013-10-16 23:58:48 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_STORE_SET ) ;
2013-10-16 23:58:48 +01:00
emit_post ( emit ) ;
2013-10-08 09:05:10 +01:00
}
2013-10-16 23:58:48 +01:00
2014-02-12 16:31:30 +00:00
STATIC void emit_native_build_slice ( emit_t * emit , int n_args ) {
2014-04-06 12:58:40 +01:00
DEBUG_printf ( " build_slice %d \n " , n_args ) ;
2014-06-30 05:17:25 +01: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 22:31:57 +01: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 05:17:25 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_NEW_SLICE ) ;
2014-06-30 05:17:25 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2013-10-08 09:05:10 +01:00
}
2014-04-06 12:58:40 +01:00
2014-02-12 16:31:30 +00:00
STATIC void emit_native_unpack_sequence ( emit_t * emit , int n_args ) {
2014-04-06 12:58:40 +01: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 22:31:57 +01:00
emit_call_with_imm_arg ( emit , MP_F_UNPACK_SEQUENCE , n_args , REG_ARG_2 ) ; // arg2 = n_args
2013-10-08 09:05:10 +01:00
}
2014-04-06 12:58:40 +01:00
2014-02-12 16:31:30 +00:00
STATIC void emit_native_unpack_ex ( emit_t * emit , int n_left , int n_right ) {
2014-05-07 18:30:52 +01: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 05:17:25 +01:00
emit_get_stack_pointer_to_reg_for_push ( emit , REG_ARG_3 , n_left + n_right + 1 ) ; // arg3 = dest ptr
2014-08-16 22:31:57 +01: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 09:05:10 +01:00
}
2014-03-31 11:30:17 +01:00
STATIC void emit_native_make_function ( emit_t * emit , scope_t * scope , uint n_pos_defaults , uint n_kw_defaults ) {
2013-10-08 09:05:10 +01: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 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-05-07 18:30:52 +01:00
if ( n_pos_defaults = = 0 & & n_kw_defaults = = 0 ) {
2014-08-16 22:31:57 +01: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 18:30:52 +01: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 22:31:57 +01: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 18:30:52 +01:00
}
2013-10-08 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-04-20 17:50:40 +01:00
STATIC void emit_native_make_closure ( emit_t * emit , scope_t * scope , uint n_closed_over , uint n_pos_defaults , uint n_kw_defaults ) {
2013-10-08 09:05:10 +01:00
assert ( 0 ) ;
}
2014-04-09 12:43:17 +01:00
STATIC void emit_native_call_function ( emit_t * emit , int n_positional , int n_keyword , uint star_flags ) {
2013-10-08 09:05:10 +01:00
// call special viper runtime routine with type info for args, and wanted type info for return
2014-04-09 12:43:17 +01:00
assert ( ! star_flags ) ;
2014-02-02 13:11:48 +00: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 09:05:10 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_CALL_FUNCTION_0 ) ;
2013-10-08 09:05:10 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_CALL_FUNCTION_1 ) ;
2013-10-08 09:05:10 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_CALL_FUNCTION_2 ) ;
2013-10-08 09:05:10 +01:00
} else {
2013-10-10 22:06:54 +01:00
*/
2014-02-02 13:11:48 +00:00
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-04-06 12:58:40 +01: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 13:11:48 +00: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 22:31:57 +01: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 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-04-09 12:43:17 +01:00
STATIC void emit_native_call_method ( emit_t * emit , int n_positional , int n_keyword , uint star_flags ) {
assert ( ! star_flags ) ;
2014-02-02 13:11:48 +00:00
2013-10-10 22:06:54 +01:00
/*
2013-10-08 09:05:10 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_CALL_METHOD_1 ) ;
2013-10-08 09:05:10 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_CALL_METHOD_2 ) ;
2013-10-08 09:05:10 +01:00
} else {
2013-10-10 22:06:54 +01:00
*/
2014-02-02 13:11:48 +00:00
2014-03-27 23:30:26 +00:00
emit_native_pre ( emit ) ;
2014-04-06 12:58:40 +01: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 22:31:57 +01: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 09:05:10 +01:00
emit_post_push_reg ( emit , VTYPE_PYOBJ , REG_RET ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_return_value ( emit_t * emit ) {
2014-04-06 12:58:40 +01:00
DEBUG_printf ( " return_value \n " ) ;
2013-10-08 09:05:10 +01:00
vtype_kind_t vtype ;
emit_pre_pop_reg ( emit , & vtype , REG_RET ) ;
if ( emit - > do_viper_types ) {
2014-08-15 23:47:59 +01:00
if ( vtype = = VTYPE_PTR_NONE ) {
if ( emit - > return_vtype = = VTYPE_PYOBJ ) {
ASM_MOV_IMM_TO_REG ( ( mp_uint_t ) mp_const_none , REG_RET ) ;
}
} else if ( vtype ! = emit - > return_vtype ) {
2014-08-15 16:45:41 +01:00
printf ( " ViperTypeError: incompatible return type \n " ) ;
}
2013-10-08 09:05:10 +01:00
} else {
assert ( vtype = = VTYPE_PYOBJ ) ;
}
emit - > last_emit_was_return_value = true ;
2013-10-12 16:53:13 +01:00
# if N_X64
2013-10-08 09:05:10 +01:00
//asm_x64_call_ind(emit->as, 0, REG_RAX); to seg fault for debugging with gdb
asm_x64_exit ( emit - > as ) ;
2013-10-12 16:53:13 +01:00
# elif N_THUMB
2013-10-08 09:05:10 +01:00
//asm_thumb_call_ind(emit->as, 0, REG_R0); to seg fault for debugging with gdb
asm_thumb_exit ( emit - > as ) ;
2014-08-16 21:55:53 +01:00
# elif N_ARM
//asm_arm_bkpt(emit->as); to insert a bkpt and not segfault for debugging
asm_arm_exit ( emit - > as ) ;
2013-10-08 09:05:10 +01:00
# endif
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_raise_varargs ( emit_t * emit , int n_args ) {
2014-06-30 05:17:25 +01:00
assert ( n_args = = 1 ) ;
2014-08-16 22:06:11 +01: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 22:31:57 +01:00
emit_call ( emit , MP_F_NATIVE_RAISE ) ;
2013-10-08 09:05:10 +01:00
}
2014-06-30 05:17:25 +01:00
2014-02-12 16:31:30 +00:00
STATIC void emit_native_yield_value ( emit_t * emit ) {
2013-10-08 09:05:10 +01:00
// not supported (for now)
assert ( 0 ) ;
}
2014-02-12 16:31:30 +00:00
STATIC void emit_native_yield_from ( emit_t * emit ) {
2013-10-08 09:05:10 +01:00
// not supported (for now)
assert ( 0 ) ;
}
2014-06-30 05:17:25 +01: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
emit_pre_pop_discard ( emit , & vtype_nlr ) ; // discard the linked-list pointer in the nlr_buf
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 ) {
adjust_stack ( emit , - 3 ) ; // stack adjust (not sure why it's this much...)
}
2013-10-08 09:05:10 +01:00
const emit_method_table_t EXPORT_FUN ( method_table ) = {
2014-08-15 16:45:41 +01:00
emit_native_set_native_type ,
2013-10-08 09:05:10 +01:00
emit_native_start_pass ,
emit_native_end_pass ,
emit_native_last_emit_was_return_value ,
2014-04-10 18:28:54 +01:00
emit_native_adjust_stack_size ,
2014-01-18 23:24:36 +00:00
emit_native_set_source_line ,
2013-10-08 09:05:10 +01: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 17:50:40 +01:00
emit_native_load_null ,
2013-10-08 09:05:10 +01:00
emit_native_load_fast ,
emit_native_load_deref ,
2013-12-11 00:41:43 +00:00
emit_native_load_name ,
emit_native_load_global ,
2013-10-08 09:05:10 +01:00
emit_native_load_attr ,
emit_native_load_method ,
emit_native_load_build_class ,
2014-04-17 22:10:53 +01:00
emit_native_load_subscr ,
2013-10-08 09:05:10 +01:00
emit_native_store_fast ,
2013-12-11 00:41:43 +00:00
emit_native_store_deref ,
2013-10-08 09:05:10 +01:00
emit_native_store_name ,
emit_native_store_global ,
emit_native_store_attr ,
emit_native_store_subscr ,
emit_native_delete_fast ,
2013-12-11 00:41:43 +00:00
emit_native_delete_deref ,
2013-10-08 09:05:10 +01: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 05:17:25 +01:00
emit_native_start_except_handler ,
emit_native_end_except_handler ,
2013-10-08 09:05:10 +01:00
} ;
2014-08-16 21:55:53 +01:00
# endif // (MICROPY_EMIT_X64 && N_X64) || (MICROPY_EMIT_THUMB && N_THUMB) || (MICROPY_EMIT_ARM && N_ARM)