2013-12-17 13:27:24 -05:00
# include <stdint.h>
# include <stdlib.h>
# include <stdio.h>
# include <stdarg.h>
# include <string.h>
# include <assert.h>
# include "nlr.h"
# include "misc.h"
2013-12-21 13:17:45 -05:00
# include "mpconfig.h"
# include "obj.h"
# include "runtime0.h"
2013-12-17 13:27:24 -05:00
# include "runtime.h"
# include "map.h"
# include "builtin.h"
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin___build_class__ ( mp_obj_t o_class_fun , mp_obj_t o_class_name ) {
2013-12-17 16:35:38 -05:00
// we differ from CPython: we set the new __locals__ object here
2013-12-21 13:17:45 -05:00
mp_map_t * old_locals = rt_get_map_locals ( ) ;
mp_map_t * class_locals = mp_map_new ( MP_MAP_QSTR , 0 ) ;
2013-12-17 16:35:38 -05:00
rt_set_map_locals ( class_locals ) ;
// call the class code
2013-12-21 13:17:45 -05:00
rt_call_function_1 ( o_class_fun , ( mp_obj_t ) 0xdeadbeef ) ;
2013-12-17 16:35:38 -05:00
// restore old __locals__ object
rt_set_map_locals ( old_locals ) ;
// create and return the new class
2013-12-21 13:17:45 -05:00
return mp_obj_new_class ( class_locals ) ;
2013-12-17 16:35:38 -05:00
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin___import__ ( int n , mp_obj_t * args ) {
2013-12-17 16:35:38 -05:00
printf ( " import: \n " ) ;
for ( int i = 0 ; i < n ; i + + ) {
printf ( " " ) ;
2013-12-21 13:17:45 -05:00
mp_obj_print ( args [ i ] ) ;
2013-12-17 16:35:38 -05:00
printf ( " \n " ) ;
}
2013-12-21 13:17:45 -05:00
return mp_const_none ;
2013-12-17 16:35:38 -05:00
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin___repl_print__ ( mp_obj_t o ) {
if ( o ! = mp_const_none ) {
mp_obj_print ( o ) ;
2013-12-17 13:27:24 -05:00
printf ( " \n " ) ;
}
2013-12-21 13:17:45 -05:00
return mp_const_none ;
2013-12-17 13:27:24 -05:00
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_abs ( mp_obj_t o_in ) {
if ( MP_OBJ_IS_SMALL_INT ( o_in ) ) {
mp_small_int_t val = MP_OBJ_SMALL_INT_VALUE ( o_in ) ;
2013-12-17 16:35:38 -05:00
if ( val < 0 ) {
val = - val ;
2013-12-17 13:27:24 -05:00
}
2013-12-21 13:17:45 -05:00
return MP_OBJ_NEW_SMALL_INT ( val ) ;
2013-12-17 16:35:38 -05:00
# if MICROPY_ENABLE_FLOAT
2013-12-21 13:17:45 -05:00
} else if ( MP_OBJ_IS_TYPE ( o_in , & float_type ) ) {
mp_float_t value = mp_obj_float_get ( o_in ) ;
2013-12-17 16:35:38 -05:00
// TODO check for NaN etc
2013-12-21 13:17:45 -05:00
if ( value < 0 ) {
return mp_obj_new_float ( - value ) ;
2013-12-17 13:27:24 -05:00
} else {
2013-12-17 16:35:38 -05:00
return o_in ;
2013-12-17 13:27:24 -05:00
}
2013-12-21 13:17:45 -05:00
} else if ( MP_OBJ_IS_TYPE ( o_in , & complex_type ) ) {
mp_float_t real , imag ;
mp_obj_complex_get ( o_in , & real , & imag ) ;
return mp_obj_new_float ( machine_sqrt ( real * real + imag * imag ) ) ;
2013-12-17 16:35:38 -05:00
# endif
} else {
assert ( 0 ) ;
2013-12-21 13:17:45 -05:00
return mp_const_none ;
2013-12-17 16:35:38 -05:00
}
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_all ( mp_obj_t o_in ) {
mp_obj_t iterable = rt_getiter ( o_in ) ;
mp_obj_t item ;
while ( ( item = rt_iternext ( iterable ) ) ! = mp_const_stop_iteration ) {
2013-12-17 16:35:38 -05:00
if ( ! rt_is_true ( item ) ) {
2013-12-21 13:17:45 -05:00
return mp_const_false ;
2013-12-17 16:35:38 -05:00
}
}
2013-12-21 13:17:45 -05:00
return mp_const_true ;
2013-12-17 16:35:38 -05:00
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_any ( mp_obj_t o_in ) {
mp_obj_t iterable = rt_getiter ( o_in ) ;
mp_obj_t item ;
while ( ( item = rt_iternext ( iterable ) ) ! = mp_const_stop_iteration ) {
2013-12-17 16:35:38 -05:00
if ( rt_is_true ( item ) ) {
2013-12-21 13:17:45 -05:00
return mp_const_true ;
2013-12-17 16:35:38 -05:00
}
}
2013-12-21 13:17:45 -05:00
return mp_const_false ;
2013-12-17 16:35:38 -05:00
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_bool ( int n_args , const mp_obj_t * args ) {
2013-12-17 16:35:38 -05:00
switch ( n_args ) {
2013-12-21 13:17:45 -05:00
case 0 : return mp_const_false ;
case 1 : if ( rt_is_true ( args [ 0 ] ) ) { return mp_const_true ; } else { return mp_const_false ; }
default : nlr_jump ( mp_obj_new_exception_msg_1_arg ( rt_q_TypeError , " bool() takes at most 1 argument (%d given) " , ( void * ) ( machine_int_t ) n_args ) ) ;
2013-12-17 16:35:38 -05:00
}
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_callable ( mp_obj_t o_in ) {
if ( mp_obj_is_callable ( o_in ) ) {
return mp_const_true ;
2013-12-17 16:35:38 -05:00
} else {
2013-12-21 13:17:45 -05:00
return mp_const_false ;
2013-12-17 16:35:38 -05:00
}
}
# if MICROPY_ENABLE_FLOAT
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_complex ( int n_args , const mp_obj_t * args ) {
2013-12-17 16:35:38 -05:00
switch ( n_args ) {
case 0 :
2013-12-21 13:17:45 -05:00
return mp_obj_new_complex ( 0 , 0 ) ;
2013-12-17 16:35:38 -05:00
case 1 :
// TODO allow string as first arg
2013-12-21 13:17:45 -05:00
if ( MP_OBJ_IS_TYPE ( args [ 0 ] , & complex_type ) ) {
2013-12-17 16:35:38 -05:00
return args [ 0 ] ;
} else {
2013-12-21 13:17:45 -05:00
return mp_obj_new_complex ( mp_obj_get_float ( args [ 0 ] ) , 0 ) ;
2013-12-17 16:35:38 -05:00
}
case 2 :
{
2013-12-21 13:17:45 -05:00
mp_float_t real , imag ;
if ( MP_OBJ_IS_TYPE ( args [ 0 ] , & complex_type ) ) {
mp_obj_get_complex ( args [ 0 ] , & real , & imag ) ;
2013-12-17 16:35:38 -05:00
} else {
2013-12-21 13:17:45 -05:00
real = mp_obj_get_float ( args [ 0 ] ) ;
2013-12-17 16:35:38 -05:00
imag = 0 ;
}
2013-12-21 13:17:45 -05:00
if ( MP_OBJ_IS_TYPE ( args [ 1 ] , & complex_type ) ) {
mp_float_t real2 , imag2 ;
mp_obj_get_complex ( args [ 1 ] , & real2 , & imag2 ) ;
2013-12-17 16:35:38 -05:00
real - = imag2 ;
imag + = real2 ;
} else {
2013-12-21 13:17:45 -05:00
imag + = mp_obj_get_float ( args [ 1 ] ) ;
2013-12-17 16:35:38 -05:00
}
2013-12-21 13:17:45 -05:00
return mp_obj_new_complex ( real , imag ) ;
2013-12-17 16:35:38 -05:00
}
2013-12-21 13:17:45 -05:00
default : nlr_jump ( mp_obj_new_exception_msg_1_arg ( rt_q_TypeError , " comlpex() takes at most 2 arguments (%d given) " , ( void * ) ( machine_int_t ) n_args ) ) ;
2013-12-17 16:35:38 -05:00
}
}
# endif
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_chr ( mp_obj_t o_in ) {
int ord = mp_obj_get_int ( o_in ) ;
2013-12-17 16:35:38 -05:00
if ( 0 < = ord & & ord < = 0x10ffff ) {
char * str = m_new ( char , 2 ) ;
str [ 0 ] = ord ;
str [ 1 ] = ' \0 ' ;
2013-12-21 13:17:45 -05:00
return mp_obj_new_str ( qstr_from_str_take ( str ) ) ;
2013-12-17 16:35:38 -05:00
} else {
2013-12-21 13:17:45 -05:00
nlr_jump ( mp_obj_new_exception_msg ( rt_q_ValueError , " chr() arg not in range(0x110000) " ) ) ;
2013-12-17 13:27:24 -05:00
}
2013-12-17 16:35:38 -05:00
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_dict ( void ) {
2013-12-17 16:35:38 -05:00
// TODO create from an iterable!
return rt_build_map ( 0 ) ;
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_divmod ( mp_obj_t o1_in , mp_obj_t o2_in ) {
if ( MP_OBJ_IS_SMALL_INT ( o1_in ) & & MP_OBJ_IS_SMALL_INT ( o2_in ) ) {
mp_small_int_t i1 = MP_OBJ_SMALL_INT_VALUE ( o1_in ) ;
mp_small_int_t i2 = MP_OBJ_SMALL_INT_VALUE ( o2_in ) ;
mp_obj_t revs_args [ 2 ] ;
revs_args [ 1 ] = MP_OBJ_NEW_SMALL_INT ( i1 / i2 ) ;
revs_args [ 0 ] = MP_OBJ_NEW_SMALL_INT ( i1 % i2 ) ;
2013-12-17 16:35:38 -05:00
return rt_build_tuple ( 2 , revs_args ) ;
} else {
2013-12-21 13:17:45 -05:00
nlr_jump ( mp_obj_new_exception_msg_2_args ( rt_q_TypeError , " unsupported operand type(s) for divmod(): '%s' and '%s' " , mp_obj_get_type_str ( o1_in ) , mp_obj_get_type_str ( o2_in ) ) ) ;
2013-12-17 16:35:38 -05:00
}
}
2013-12-21 13:17:45 -05:00
static mp_obj_t mp_builtin_hash ( mp_obj_t o_in ) {
2013-12-17 16:35:38 -05:00
// TODO hash will generally overflow small integer; can we safely truncate it?
2013-12-21 13:17:45 -05:00
return mp_obj_new_int ( mp_obj_hash ( o_in ) ) ;
2013-12-17 16:35:38 -05:00
}
2013-12-21 13:17:45 -05:00
MP_DEFINE_CONST_FUN_OBJ_1 ( mp_builtin_hash_obj , mp_builtin_hash ) ;
2013-12-17 16:35:38 -05:00
2013-12-21 13:17:45 -05:00
static mp_obj_t mp_builtin_iter ( mp_obj_t o_in ) {
return rt_getiter ( o_in ) ;
2013-12-17 13:27:24 -05:00
}
2013-12-21 13:17:45 -05:00
MP_DEFINE_CONST_FUN_OBJ_1 ( mp_builtin_iter_obj , mp_builtin_iter ) ;
mp_obj_t mp_builtin_len ( mp_obj_t o_in ) {
mp_small_int_t len = 0 ;
if ( MP_OBJ_IS_TYPE ( o_in , & str_type ) ) {
len = strlen ( qstr_str ( mp_obj_str_get ( o_in ) ) ) ;
} else if ( MP_OBJ_IS_TYPE ( o_in , & tuple_type ) ) {
uint seq_len ;
mp_obj_t * seq_items ;
mp_obj_tuple_get ( o_in , & seq_len , & seq_items ) ;
len = seq_len ;
} else if ( MP_OBJ_IS_TYPE ( o_in , & list_type ) ) {
uint seq_len ;
mp_obj_t * seq_items ;
mp_obj_list_get ( o_in , & seq_len , & seq_items ) ;
len = seq_len ;
/* TODO
} else if ( MP_OBJ_IS_TYPE ( o_in , & dict_type ) ) {
mp_obj_base_t * o = o_in ;
2013-12-17 13:27:24 -05:00
len = o - > u_map . used ;
2013-12-21 13:17:45 -05:00
*/
2013-12-17 13:27:24 -05:00
} else {
2013-12-21 13:17:45 -05:00
nlr_jump ( mp_obj_new_exception_msg_1_arg ( rt_q_TypeError , " object of type '%s' has no len() " , mp_obj_get_type_str ( o_in ) ) ) ;
2013-12-17 13:27:24 -05:00
}
2013-12-21 13:17:45 -05:00
return MP_OBJ_NEW_SMALL_INT ( len ) ;
2013-12-17 13:27:24 -05:00
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_list ( int n_args , const mp_obj_t * args ) {
2013-12-17 16:35:38 -05:00
switch ( n_args ) {
case 0 : return rt_build_list ( 0 , NULL ) ;
case 1 :
{
// make list from iterable
2013-12-21 13:17:45 -05:00
mp_obj_t iterable = rt_getiter ( args [ 0 ] ) ;
mp_obj_t list = rt_build_list ( 0 , NULL ) ;
mp_obj_t item ;
while ( ( item = rt_iternext ( iterable ) ) ! = mp_const_stop_iteration ) {
2013-12-17 16:35:38 -05:00
rt_list_append ( list , item ) ;
}
return list ;
2013-12-17 13:27:24 -05:00
}
2013-12-21 13:17:45 -05:00
default : nlr_jump ( mp_obj_new_exception_msg_1_arg ( rt_q_TypeError , " list() takes at most 1 argument (%d given) " , ( void * ) ( machine_int_t ) n_args ) ) ;
2013-12-17 16:35:38 -05:00
}
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_max ( int n_args , const mp_obj_t * args ) {
2013-12-17 16:35:38 -05:00
if ( n_args = = 1 ) {
// given an iterable
2013-12-21 13:17:45 -05:00
mp_obj_t iterable = rt_getiter ( args [ 0 ] ) ;
mp_obj_t max_obj = NULL ;
mp_obj_t item ;
while ( ( item = rt_iternext ( iterable ) ) ! = mp_const_stop_iteration ) {
if ( max_obj = = NULL | | mp_obj_less ( max_obj , item ) ) {
2013-12-17 16:35:38 -05:00
max_obj = item ;
}
2013-12-17 13:27:24 -05:00
}
2013-12-17 16:35:38 -05:00
if ( max_obj = = NULL ) {
2013-12-21 13:17:45 -05:00
nlr_jump ( mp_obj_new_exception_msg ( rt_q_ValueError , " max() arg is an empty sequence " ) ) ;
2013-12-17 16:35:38 -05:00
}
return max_obj ;
2013-12-17 13:27:24 -05:00
} else {
2013-12-17 16:35:38 -05:00
// given many args
2013-12-21 13:17:45 -05:00
mp_obj_t max_obj = args [ 0 ] ;
2013-12-17 16:35:38 -05:00
for ( int i = 1 ; i < n_args ; i + + ) {
2013-12-21 13:17:45 -05:00
if ( mp_obj_less ( max_obj , args [ i ] ) ) {
2013-12-17 16:35:38 -05:00
max_obj = args [ i ] ;
}
}
return max_obj ;
2013-12-17 13:27:24 -05:00
}
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_min ( int n_args , const mp_obj_t * args ) {
2013-12-17 16:35:38 -05:00
if ( n_args = = 1 ) {
// given an iterable
2013-12-21 13:17:45 -05:00
mp_obj_t iterable = rt_getiter ( args [ 0 ] ) ;
mp_obj_t min_obj = NULL ;
mp_obj_t item ;
while ( ( item = rt_iternext ( iterable ) ) ! = mp_const_stop_iteration ) {
if ( min_obj = = NULL | | mp_obj_less ( item , min_obj ) ) {
2013-12-17 16:35:38 -05:00
min_obj = item ;
}
}
if ( min_obj = = NULL ) {
2013-12-21 13:17:45 -05:00
nlr_jump ( mp_obj_new_exception_msg ( rt_q_ValueError , " min() arg is an empty sequence " ) ) ;
2013-12-17 16:35:38 -05:00
}
return min_obj ;
} else {
// given many args
2013-12-21 13:17:45 -05:00
mp_obj_t min_obj = args [ 0 ] ;
2013-12-17 16:35:38 -05:00
for ( int i = 1 ; i < n_args ; i + + ) {
2013-12-21 13:17:45 -05:00
if ( mp_obj_less ( args [ i ] , min_obj ) ) {
2013-12-17 16:35:38 -05:00
min_obj = args [ i ] ;
}
}
return min_obj ;
}
}
2013-12-17 13:27:24 -05:00
2013-12-21 13:38:03 -05:00
static mp_obj_t mp_builtin_next ( mp_obj_t o ) {
mp_obj_t ret = rt_iternext ( o ) ;
if ( ret = = mp_const_stop_iteration ) {
nlr_jump ( mp_obj_new_exception ( qstr_from_str_static ( " StopIteration " ) ) ) ;
} else {
return ret ;
}
2013-12-21 13:17:45 -05:00
}
MP_DEFINE_CONST_FUN_OBJ_1 ( mp_builtin_next_obj , mp_builtin_next ) ;
mp_obj_t mp_builtin_ord ( mp_obj_t o_in ) {
const char * str = qstr_str ( mp_obj_get_qstr ( o_in ) ) ;
2013-12-17 16:35:38 -05:00
if ( strlen ( str ) = = 1 ) {
2013-12-21 13:17:45 -05:00
return mp_obj_new_int ( str [ 0 ] ) ;
2013-12-17 16:35:38 -05:00
} else {
2013-12-21 13:17:45 -05:00
nlr_jump ( mp_obj_new_exception_msg_1_arg ( rt_q_TypeError , " ord() expected a character, but string of length %d found " , ( void * ) ( machine_int_t ) strlen ( str ) ) ) ;
2013-12-17 16:35:38 -05:00
}
}
2013-12-17 13:27:24 -05:00
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_pow ( int n_args , const mp_obj_t * args ) {
2013-12-17 16:35:38 -05:00
switch ( n_args ) {
case 2 : return rt_binary_op ( RT_BINARY_OP_POWER , args [ 0 ] , args [ 1 ] ) ;
case 3 : return rt_binary_op ( RT_BINARY_OP_MODULO , rt_binary_op ( RT_BINARY_OP_POWER , args [ 0 ] , args [ 1 ] ) , args [ 2 ] ) ; // TODO optimise...
2013-12-21 13:17:45 -05:00
default : nlr_jump ( mp_obj_new_exception_msg_1_arg ( rt_q_TypeError , " pow expected at most 3 arguments, got %d " , ( void * ) ( machine_int_t ) n_args ) ) ;
2013-12-17 16:35:38 -05:00
}
}
2013-12-17 13:27:24 -05:00
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_print ( int n_args , const mp_obj_t * args ) {
2013-12-17 16:35:38 -05:00
for ( int i = 0 ; i < n_args ; i + + ) {
if ( i > 0 ) {
printf ( " " ) ;
}
2013-12-21 13:17:45 -05:00
if ( MP_OBJ_IS_TYPE ( args [ i ] , & str_type ) ) {
2013-12-17 16:35:38 -05:00
// special case, print string raw
2013-12-21 13:17:45 -05:00
printf ( " %s " , qstr_str ( mp_obj_str_get ( args [ i ] ) ) ) ;
2013-12-17 16:35:38 -05:00
} else {
// print the object Python style
2013-12-21 13:17:45 -05:00
mp_obj_print ( args [ i ] ) ;
2013-12-17 16:35:38 -05:00
}
}
printf ( " \n " ) ;
2013-12-21 13:17:45 -05:00
return mp_const_none ;
2013-12-17 13:27:24 -05:00
}
2013-12-21 13:17:45 -05:00
mp_obj_t mp_builtin_range ( int n_args , const mp_obj_t * args ) {
2013-12-17 13:27:24 -05:00
switch ( n_args ) {
2013-12-21 13:17:45 -05:00
case 1 : return mp_obj_new_range ( 0 , mp_obj_get_int ( args [ 0 ] ) , 1 ) ;
case 2 : return mp_obj_new_range ( mp_obj_get_int ( args [ 0 ] ) , mp_obj_get_int ( args [ 1 ] ) , 1 ) ;
case 3 : return mp_obj_new_range ( mp_obj_get_int ( args [ 0 ] ) , mp_obj_get_int ( args [ 1 ] ) , mp_obj_get_int ( args [ 2 ] ) ) ;
default : nlr_jump ( mp_obj_new_exception_msg_1_arg ( rt_q_TypeError , " range expected at most 3 arguments, got %d " , ( void * ) ( machine_int_t ) n_args ) ) ;
}
}
static mp_obj_t mp_builtin_set ( int n_args , const mp_obj_t * args ) {
assert ( 0 < = n_args & & n_args < = 1 ) ;
if ( n_args = = 0 ) {
// return a new, empty set
return mp_obj_new_set ( 0 , NULL ) ;
} else {
// 1 argument, an iterable from which we make a new set
mp_obj_t set = mp_obj_new_set ( 0 , NULL ) ;
mp_obj_t iterable = rt_getiter ( args [ 0 ] ) ;
mp_obj_t item ;
while ( ( item = rt_iternext ( iterable ) ) ! = mp_const_stop_iteration ) {
mp_obj_set_store ( set , item ) ;
}
return set ;
2013-12-17 13:27:24 -05:00
}
}
2013-12-17 16:35:38 -05:00
2013-12-21 13:17:45 -05:00
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN ( mp_builtin_set_obj , 0 , 1 , mp_builtin_set ) ;
mp_obj_t mp_builtin_sum ( int n_args , const mp_obj_t * args ) {
mp_obj_t value ;
2013-12-17 16:35:38 -05:00
switch ( n_args ) {
2013-12-21 13:17:45 -05:00
case 1 : value = mp_obj_new_int ( 0 ) ; break ;
2013-12-17 16:35:38 -05:00
case 2 : value = args [ 1 ] ; break ;
2013-12-21 13:17:45 -05:00
default : nlr_jump ( mp_obj_new_exception_msg_1_arg ( rt_q_TypeError , " sum expected at most 2 arguments, got %d " , ( void * ) ( machine_int_t ) n_args ) ) ;
2013-12-17 16:35:38 -05:00
}
2013-12-21 13:17:45 -05:00
mp_obj_t iterable = rt_getiter ( args [ 0 ] ) ;
mp_obj_t item ;
while ( ( item = rt_iternext ( iterable ) ) ! = mp_const_stop_iteration ) {
2013-12-17 16:35:38 -05:00
value = rt_binary_op ( RT_BINARY_OP_ADD , value , item ) ;
}
return value ;
}