blob: fb64d849410154c11243b5ff3f0279fa1b797e4b [file] [log] [blame]
Damiend99b0522013-12-21 18:17:45 +00001// All Micro Python objects are at least this type
2// It must be of pointer size
3
4typedef machine_ptr_t mp_obj_t;
5typedef machine_const_ptr_t mp_const_obj_t;
6
7// Integers that fit in a pointer have this type
8// (do we need to expose this in the public API?)
9
10typedef machine_int_t mp_small_int_t;
11
Damien George97209d32014-01-07 15:58:30 +000012// Anything that wants to be a Micro Python object must have
13// mp_obj_base_t as its first member (except NULL and small ints)
Damien660365e2013-12-17 18:27:24 +000014
ian-v7a16fad2014-01-06 09:52:29 -080015struct _mp_obj_type_t;
Damiend99b0522013-12-21 18:17:45 +000016struct _mp_obj_base_t {
ian-v7a16fad2014-01-06 09:52:29 -080017 const struct _mp_obj_type_t *type;
Damiend99b0522013-12-21 18:17:45 +000018};
ian-v7a16fad2014-01-06 09:52:29 -080019typedef struct _mp_obj_base_t mp_obj_base_t;
Damiend99b0522013-12-21 18:17:45 +000020
21// The NULL object is used to indicate the absence of an object
22// It *cannot* be used when an mp_obj_t is expected, except where explicitly allowed
23
Damien Georgea5c82a82014-04-11 11:16:53 +000024#define MP_OBJ_NULL ((mp_obj_t)0)
Damiend99b0522013-12-21 18:17:45 +000025
Damien George95004e52014-04-05 17:17:19 +010026// The SENTINEL object is used for various internal purposes where one needs
27// an object which is unique from all other objects, including MP_OBJ_NULL.
28
29#define MP_OBJ_SENTINEL ((mp_obj_t)8)
30
Damien George38a2da62014-01-08 17:33:12 +000031// These macros check for small int, qstr or object, and access small int and qstr values
32// - xxxx...xxx1: a small int, bits 1 and above are the value
33// - xxxx...xx10: a qstr, bits 2 and above are the value
34// - xxxx...xx00: a pointer to an mp_obj_base_t
Damiend99b0522013-12-21 18:17:45 +000035
Paul Sokolovsky757ac812014-01-12 17:06:25 +020036// In SMALL_INT, next-to-highest bits is used as sign, so both must match for value in range
Damien George9d68e9c2014-03-12 15:38:15 +000037#define MP_SMALL_INT_MIN ((mp_small_int_t)(((machine_int_t)WORD_MSBIT_HIGH) >> 1))
38#define MP_SMALL_INT_MAX ((mp_small_int_t)(~(MP_SMALL_INT_MIN)))
Paul Sokolovsky757ac812014-01-12 17:06:25 +020039#define MP_OBJ_FITS_SMALL_INT(n) ((((n) ^ ((n) << 1)) & WORD_MSBIT_HIGH) == 0)
Damien Georgea5c82a82014-04-11 11:16:53 +000040// these macros have now become inline functions; see below
41//#define MP_OBJ_IS_SMALL_INT(o) ((((mp_small_int_t)(o)) & 1) != 0)
42//#define MP_OBJ_IS_QSTR(o) ((((mp_small_int_t)(o)) & 3) == 2)
43//#define MP_OBJ_IS_OBJ(o) ((((mp_small_int_t)(o)) & 3) == 0)
44//#define MP_OBJ_IS_TYPE(o, t) (MP_OBJ_IS_OBJ(o) && (((mp_obj_base_t*)(o))->type == (t))) // this does not work for checking a string, use below macro for that
45//#define MP_OBJ_IS_INT(o) (MP_OBJ_IS_SMALL_INT(o) || MP_OBJ_IS_TYPE(o, &mp_type_int))
46//#define MP_OBJ_IS_STR(o) (MP_OBJ_IS_QSTR(o) || MP_OBJ_IS_TYPE(o, &mp_type_str))
Damien George38a2da62014-01-08 17:33:12 +000047
Damiend99b0522013-12-21 18:17:45 +000048#define MP_OBJ_SMALL_INT_VALUE(o) (((mp_small_int_t)(o)) >> 1)
Damien George38a2da62014-01-08 17:33:12 +000049#define MP_OBJ_NEW_SMALL_INT(small_int) ((mp_obj_t)(((small_int) << 1) | 1))
50
51#define MP_OBJ_QSTR_VALUE(o) (((mp_small_int_t)(o)) >> 2)
52#define MP_OBJ_NEW_QSTR(qstr) ((mp_obj_t)((((machine_uint_t)qstr) << 2) | 2))
Damiend99b0522013-12-21 18:17:45 +000053
54// These macros are used to declare and define constant function objects
55// You can put "static" in front of the definitions to make them local
56
57#define MP_DECLARE_CONST_FUN_OBJ(obj_name) extern const mp_obj_fun_native_t obj_name
58
Damien George3e1a5c12014-03-29 13:43:38 +000059#define MP_DEFINE_CONST_FUN_OBJ_VOID_PTR(obj_name, is_kw, n_args_min, n_args_max, fun_name) const mp_obj_fun_native_t obj_name = {{&mp_type_fun_native}, is_kw, n_args_min, n_args_max, (void *)fun_name}
John R. Lenton270112f2014-01-07 18:01:08 +000060#define MP_DEFINE_CONST_FUN_OBJ_0(obj_name, fun_name) MP_DEFINE_CONST_FUN_OBJ_VOID_PTR(obj_name, false, 0, 0, (mp_fun_0_t)fun_name)
61#define MP_DEFINE_CONST_FUN_OBJ_1(obj_name, fun_name) MP_DEFINE_CONST_FUN_OBJ_VOID_PTR(obj_name, false, 1, 1, (mp_fun_1_t)fun_name)
62#define MP_DEFINE_CONST_FUN_OBJ_2(obj_name, fun_name) MP_DEFINE_CONST_FUN_OBJ_VOID_PTR(obj_name, false, 2, 2, (mp_fun_2_t)fun_name)
63#define MP_DEFINE_CONST_FUN_OBJ_3(obj_name, fun_name) MP_DEFINE_CONST_FUN_OBJ_VOID_PTR(obj_name, false, 3, 3, (mp_fun_3_t)fun_name)
Damien Georged5e81822014-02-26 17:47:05 +000064#define MP_DEFINE_CONST_FUN_OBJ_VAR(obj_name, n_args_min, fun_name) MP_DEFINE_CONST_FUN_OBJ_VOID_PTR(obj_name, false, n_args_min, MP_OBJ_FUN_ARGS_MAX, (mp_fun_var_t)fun_name)
John R. Lenton270112f2014-01-07 18:01:08 +000065#define MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(obj_name, n_args_min, n_args_max, fun_name) MP_DEFINE_CONST_FUN_OBJ_VOID_PTR(obj_name, false, n_args_min, n_args_max, (mp_fun_var_t)fun_name)
Damien Georged5e81822014-02-26 17:47:05 +000066#define MP_DEFINE_CONST_FUN_OBJ_KW(obj_name, n_args_min, fun_name) MP_DEFINE_CONST_FUN_OBJ_VOID_PTR(obj_name, true, n_args_min, MP_OBJ_FUN_ARGS_MAX, (mp_fun_kw_t)fun_name)
John R. Lentonc06763a2014-01-07 17:29:16 +000067
Damien George9b196cd2014-03-26 21:47:19 +000068// This macro is used to define constant dict objects
69// You can put "static" in front of the definition to make it local
70
71#define MP_DEFINE_CONST_DICT(dict_name, table_name) \
72 const mp_obj_dict_t dict_name = { \
Damien George3e1a5c12014-03-29 13:43:38 +000073 .base = {&mp_type_dict}, \
Damien George9b196cd2014-03-26 21:47:19 +000074 .map = { \
75 .all_keys_are_qstrs = 1, \
76 .table_is_fixed_array = 1, \
77 .used = sizeof(table_name) / sizeof(mp_map_elem_t), \
78 .alloc = sizeof(table_name) / sizeof(mp_map_elem_t), \
79 .table = (mp_map_elem_t*)table_name, \
80 }, \
81 }
82
Damien Georgeeae16442014-01-11 19:22:29 +000083// These macros are used to declare and define constant staticmethond and classmethod objects
84// You can put "static" in front of the definitions to make them local
85
Damien George64131f32014-02-06 20:31:44 +000086#define MP_DECLARE_CONST_STATICMETHOD_OBJ(obj_name) extern const mp_obj_static_class_method_t obj_name
87#define MP_DECLARE_CONST_CLASSMETHOD_OBJ(obj_name) extern const mp_obj_static_class_method_t obj_name
Damien Georgeeae16442014-01-11 19:22:29 +000088
Damien George64131f32014-02-06 20:31:44 +000089#define MP_DEFINE_CONST_STATICMETHOD_OBJ(obj_name, fun_name) const mp_obj_static_class_method_t obj_name = {{&mp_type_staticmethod}, fun_name}
90#define MP_DEFINE_CONST_CLASSMETHOD_OBJ(obj_name, fun_name) const mp_obj_static_class_method_t obj_name = {{&mp_type_classmethod}, fun_name}
Damien Georgeeae16442014-01-11 19:22:29 +000091
Damien Georgedf6567e2014-03-30 13:54:02 +010092// Underlying map/hash table implementation (not dict object or map function)
93
94typedef struct _mp_map_elem_t {
95 mp_obj_t key;
96 mp_obj_t value;
97} mp_map_elem_t;
98
99// TODO maybe have a truncated mp_map_t for fixed tables, since alloc=used
100// put alloc last in the structure, so the truncated version does not need it
101// this would save 1 ROM word for all ROM objects that have a locals_dict
102// would also need a trucated dict structure
103
104typedef struct _mp_map_t {
105 machine_uint_t all_keys_are_qstrs : 1;
106 machine_uint_t table_is_fixed_array : 1;
107 machine_uint_t used : (8 * sizeof(machine_uint_t) - 2);
108 machine_uint_t alloc;
109 mp_map_elem_t *table;
110} mp_map_t;
111
Damien George95004e52014-04-05 17:17:19 +0100112// These can be or'd together
Damien Georgedf6567e2014-03-30 13:54:02 +0100113typedef enum _mp_map_lookup_kind_t {
114 MP_MAP_LOOKUP, // 0
115 MP_MAP_LOOKUP_ADD_IF_NOT_FOUND, // 1
116 MP_MAP_LOOKUP_REMOVE_IF_FOUND, // 2
Damien Georgedf6567e2014-03-30 13:54:02 +0100117} mp_map_lookup_kind_t;
118
Damien Georgea5c82a82014-04-11 11:16:53 +0000119inline bool MP_MAP_SLOT_IS_FILLED(mp_map_t *map, machine_uint_t pos) { return ((map)->table[pos].key != MP_OBJ_NULL && (map)->table[pos].key != MP_OBJ_SENTINEL); }
Damien George8b0535e2014-04-05 21:53:54 +0100120
Damien Georgedf6567e2014-03-30 13:54:02 +0100121void mp_map_init(mp_map_t *map, int n);
122void mp_map_init_fixed_table(mp_map_t *map, int n, const mp_obj_t *table);
123mp_map_t *mp_map_new(int n);
124void mp_map_deinit(mp_map_t *map);
125void mp_map_free(mp_map_t *map);
126mp_map_elem_t* mp_map_lookup(mp_map_t *map, mp_obj_t index, mp_map_lookup_kind_t lookup_kind);
127void mp_map_clear(mp_map_t *map);
Paul Sokolovskye3f58c82014-04-05 04:14:22 +0300128void mp_map_dump(mp_map_t *map);
Damien Georgedf6567e2014-03-30 13:54:02 +0100129
130// Underlying set implementation (not set object)
131
132typedef struct _mp_set_t {
133 machine_uint_t alloc;
134 machine_uint_t used;
135 mp_obj_t *table;
136} mp_set_t;
137
Damien Georgea5c82a82014-04-11 11:16:53 +0000138inline bool MP_SET_SLOT_IS_FILLED(mp_set_t *set, machine_uint_t pos) { return ((set)->table[pos] != MP_OBJ_NULL && (set)->table[pos] != MP_OBJ_SENTINEL); }
Damien George8b0535e2014-04-05 21:53:54 +0100139
Damien Georgedf6567e2014-03-30 13:54:02 +0100140void mp_set_init(mp_set_t *set, int n);
141mp_obj_t mp_set_lookup(mp_set_t *set, mp_obj_t index, mp_map_lookup_kind_t lookup_kind);
Damien George95004e52014-04-05 17:17:19 +0100142mp_obj_t mp_set_remove_first(mp_set_t *set);
Damien Georgedf6567e2014-03-30 13:54:02 +0100143void mp_set_clear(mp_set_t *set);
Damiend99b0522013-12-21 18:17:45 +0000144
145// Type definitions for methods
146
147typedef mp_obj_t (*mp_fun_0_t)(void);
148typedef mp_obj_t (*mp_fun_1_t)(mp_obj_t);
149typedef mp_obj_t (*mp_fun_2_t)(mp_obj_t, mp_obj_t);
John R. Lenton45a87442014-01-04 01:15:01 +0000150typedef mp_obj_t (*mp_fun_3_t)(mp_obj_t, mp_obj_t, mp_obj_t);
Damiend99b0522013-12-21 18:17:45 +0000151typedef mp_obj_t (*mp_fun_t)(void);
Damien Georgea11ceca2014-01-19 16:02:09 +0000152typedef mp_obj_t (*mp_fun_var_t)(uint n, const mp_obj_t *);
Damien Georgedf6567e2014-03-30 13:54:02 +0100153typedef mp_obj_t (*mp_fun_kw_t)(uint n, const mp_obj_t *, mp_map_t *);
Damiend99b0522013-12-21 18:17:45 +0000154
Paul Sokolovsky76d982e2014-01-13 19:19:16 +0200155typedef enum {
Paul Sokolovskya96d3d02014-03-31 01:10:10 +0300156 PRINT_STR,
157 PRINT_REPR,
158 PRINT_EXC, // Special format for printing exception in unhandled exception message
Paul Sokolovsky76d982e2014-01-13 19:19:16 +0200159} mp_print_kind_t;
160
161typedef void (*mp_print_fun_t)(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t o, mp_print_kind_t kind);
Damien George20006db2014-01-18 14:10:48 +0000162typedef mp_obj_t (*mp_make_new_fun_t)(mp_obj_t type_in, uint n_args, uint n_kw, const mp_obj_t *args);
163typedef mp_obj_t (*mp_call_fun_t)(mp_obj_t fun, uint n_args, uint n_kw, const mp_obj_t *args);
Damiend99b0522013-12-21 18:17:45 +0000164typedef mp_obj_t (*mp_unary_op_fun_t)(int op, mp_obj_t);
165typedef mp_obj_t (*mp_binary_op_fun_t)(int op, mp_obj_t, mp_obj_t);
Damien George20006db2014-01-18 14:10:48 +0000166typedef void (*mp_load_attr_fun_t)(mp_obj_t self_in, qstr attr, mp_obj_t *dest); // for fail, do nothing; for attr, dest[0] = value; for method, dest[0] = method, dest[1] = self
Damien Georgef4c9b332014-04-08 21:32:29 +0100167typedef bool (*mp_store_attr_fun_t)(mp_obj_t self_in, qstr attr, mp_obj_t value); // return true if store succeeded; if value==MP_OBJ_NULL then delete
168typedef bool (*mp_store_item_fun_t)(mp_obj_t self_in, mp_obj_t index, mp_obj_t value); // return true if store succeeded; if value==MP_OBJ_NULL then delete
Damiend99b0522013-12-21 18:17:45 +0000169
170typedef struct _mp_method_t {
Damien Georgec12b2212014-03-26 20:15:40 +0000171 qstr name;
Damiend99b0522013-12-21 18:17:45 +0000172 mp_const_obj_t fun;
173} mp_method_t;
174
Paul Sokolovsky5b15daf2014-01-07 20:12:26 +0200175// Buffer protocol
176typedef struct _buffer_info_t {
177 // if we'd bother to support various versions of structure
178 // (with different number of fields), we can distinguish
179 // them with ver = sizeof(struct). Cons: overkill for *micro*?
180 //int ver; // ?
181
182 void *buf;
183 machine_int_t len;
184
185 // Rationale: have array.array and have SIMD operations on them
186 // Cons: users can pass item size to processing functions themselves,
187 // though that's not "plug&play"
188 // int itemsize;
189
190 // Rationale: to load arbitrary-sized sprites directly to LCD
191 // Cons: a bit adhoc usecase
192 // int stride;
193} buffer_info_t;
194#define BUFFER_READ (1)
195#define BUFFER_WRITE (2)
196#define BUFFER_RW (BUFFER_READ | BUFFER_WRITE)
197typedef struct _mp_buffer_p_t {
198 machine_int_t (*get_buffer)(mp_obj_t obj, buffer_info_t *bufinfo, int flags);
199} mp_buffer_p_t;
Paul Sokolovsky3aa8ee72014-04-09 00:25:28 +0300200bool mp_get_buffer(mp_obj_t obj, buffer_info_t *bufinfo);
201void mp_get_buffer_raise(mp_obj_t obj, buffer_info_t *bufinfo);
Paul Sokolovsky5b15daf2014-01-07 20:12:26 +0200202
203// Stream protocol
204typedef struct _mp_stream_p_t {
205 // On error, functions should return -1 and fill in *errcode (values are
206 // implementation-dependent, but will be exposed to user, e.g. via exception).
207 machine_int_t (*read)(mp_obj_t obj, void *buf, machine_uint_t size, int *errcode);
208 machine_int_t (*write)(mp_obj_t obj, const void *buf, machine_uint_t size, int *errcode);
209 // add seek() ?
210} mp_stream_p_t;
211
Damiend99b0522013-12-21 18:17:45 +0000212struct _mp_obj_type_t {
213 mp_obj_base_t base;
Damien Georgea71c83a2014-02-15 11:34:50 +0000214 qstr name;
Damiend99b0522013-12-21 18:17:45 +0000215 mp_print_fun_t print;
Damien George71c51812014-01-04 20:21:15 +0000216 mp_make_new_fun_t make_new; // to make an instance of the type
Damiend99b0522013-12-21 18:17:45 +0000217
Damien George20006db2014-01-18 14:10:48 +0000218 mp_call_fun_t call;
Damiend99b0522013-12-21 18:17:45 +0000219 mp_unary_op_fun_t unary_op; // can return NULL if op not supported
220 mp_binary_op_fun_t binary_op; // can return NULL if op not supported
221
Damien Georgea71c83a2014-02-15 11:34:50 +0000222 mp_load_attr_fun_t load_attr;
Damien George1d24ea52014-04-08 21:11:49 +0100223 mp_store_attr_fun_t store_attr; // if value is MP_OBJ_NULL, then delete that attribute
Damien Georgef4c9b332014-04-08 21:32:29 +0100224
225 // Implements container[index] = val. If val == MP_OBJ_NULL, then it's a delete.
226 // Note that load_item is implemented by binary_op(RT_BINARY_OP_SUBSCR)
Damien Georgea71c83a2014-02-15 11:34:50 +0000227 mp_store_item_fun_t store_item;
228
Damiend99b0522013-12-21 18:17:45 +0000229 mp_fun_1_t getiter;
Damien George66eaf842014-03-26 19:27:58 +0000230 mp_fun_1_t iternext; // may return MP_OBJ_NULL as an optimisation instead of raising StopIteration() (with no args)
Damiend99b0522013-12-21 18:17:45 +0000231
Paul Sokolovsky5b15daf2014-01-07 20:12:26 +0200232 // Alternatively, pointer(s) to interfaces to save space
233 // in mp_obj_type_t at the expense of extra pointer and extra dereference
234 // when actually used.
235 mp_buffer_p_t buffer_p;
Damien George27e735f2014-04-05 23:02:23 +0100236 const mp_stream_p_t *stream_p;
Paul Sokolovsky5b15daf2014-01-07 20:12:26 +0200237
Damien George004cdce2014-01-09 21:43:51 +0000238 // these are for dynamically created types (classes)
239 mp_obj_t bases_tuple;
240 mp_obj_t locals_dict;
Damien George062478e2014-01-09 20:57:50 +0000241
Damiend99b0522013-12-21 18:17:45 +0000242 /*
243 What we might need to add here:
244
Damiend99b0522013-12-21 18:17:45 +0000245 store_subscr list dict
246
247 len str tuple list map
248 abs float complex
249 hash bool int none str
250 equal int str
Damiend99b0522013-12-21 18:17:45 +0000251
252 unpack seq list tuple
Damiend99b0522013-12-21 18:17:45 +0000253 */
254};
255
ian-v7a16fad2014-01-06 09:52:29 -0800256typedef struct _mp_obj_type_t mp_obj_type_t;
Damiend99b0522013-12-21 18:17:45 +0000257
Damien Georgec5966122014-02-15 16:10:44 +0000258// Constant types, globally accessible
Damien Georgec5966122014-02-15 16:10:44 +0000259extern const mp_obj_type_t mp_type_type;
Damien George3e1a5c12014-03-29 13:43:38 +0000260extern const mp_obj_type_t mp_type_object;
261extern const mp_obj_type_t mp_type_NoneType;
262extern const mp_obj_type_t mp_type_bool;
263extern const mp_obj_type_t mp_type_int;
264extern const mp_obj_type_t mp_type_str;
265extern const mp_obj_type_t mp_type_bytes;
Paul Sokolovsky4dcb6052014-04-08 22:09:14 +0300266extern const mp_obj_type_t mp_type_bytearray;
Damien George3e1a5c12014-03-29 13:43:38 +0000267extern const mp_obj_type_t mp_type_float;
268extern const mp_obj_type_t mp_type_complex;
269extern const mp_obj_type_t mp_type_tuple;
270extern const mp_obj_type_t mp_type_list;
271extern const mp_obj_type_t mp_type_map; // map (the python builtin, not the dict implementation detail)
272extern const mp_obj_type_t mp_type_enumerate;
273extern const mp_obj_type_t mp_type_filter;
274extern const mp_obj_type_t mp_type_dict;
275extern const mp_obj_type_t mp_type_set;
276extern const mp_obj_type_t mp_type_slice;
277extern const mp_obj_type_t mp_type_zip;
278extern const mp_obj_type_t mp_type_array;
279extern const mp_obj_type_t mp_type_super;
280extern const mp_obj_type_t mp_type_gen_instance;
281extern const mp_obj_type_t mp_type_fun_native;
282extern const mp_obj_type_t mp_type_fun_bc;
283extern const mp_obj_type_t mp_type_module;
284extern const mp_obj_type_t mp_type_staticmethod;
285extern const mp_obj_type_t mp_type_classmethod;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000286
287// Exceptions
Damien Georgec5966122014-02-15 16:10:44 +0000288extern const mp_obj_type_t mp_type_BaseException;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000289extern const mp_obj_type_t mp_type_ArithmeticError;
290extern const mp_obj_type_t mp_type_AssertionError;
291extern const mp_obj_type_t mp_type_AttributeError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000292extern const mp_obj_type_t mp_type_EOFError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000293extern const mp_obj_type_t mp_type_Exception;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000294extern const mp_obj_type_t mp_type_GeneratorExit;
295extern const mp_obj_type_t mp_type_IOError;
296extern const mp_obj_type_t mp_type_ImportError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000297extern const mp_obj_type_t mp_type_IndentationError;
298extern const mp_obj_type_t mp_type_IndexError;
299extern const mp_obj_type_t mp_type_KeyError;
300extern const mp_obj_type_t mp_type_LookupError;
301extern const mp_obj_type_t mp_type_MemoryError;
302extern const mp_obj_type_t mp_type_NameError;
303extern const mp_obj_type_t mp_type_NotImplementedError;
304extern const mp_obj_type_t mp_type_OSError;
305extern const mp_obj_type_t mp_type_OverflowError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000306extern const mp_obj_type_t mp_type_RuntimeError;
Damien Georgeffb5cfc2014-03-25 14:29:40 +0000307extern const mp_obj_type_t mp_type_StopIteration;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000308extern const mp_obj_type_t mp_type_SyntaxError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000309extern const mp_obj_type_t mp_type_SystemError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000310extern const mp_obj_type_t mp_type_TypeError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000311extern const mp_obj_type_t mp_type_ValueError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000312extern const mp_obj_type_t mp_type_ZeroDivisionError;
313
Damiend99b0522013-12-21 18:17:45 +0000314// Constant objects, globally accessible
Damien George07ddab52014-03-29 13:15:08 +0000315// The macros are for convenience only
316#define mp_const_none ((mp_obj_t)&mp_const_none_obj)
317#define mp_const_false ((mp_obj_t)&mp_const_false_obj)
318#define mp_const_true ((mp_obj_t)&mp_const_true_obj)
319#define mp_const_empty_tuple ((mp_obj_t)&mp_const_empty_tuple_obj)
320extern const struct _mp_obj_none_t mp_const_none_obj;
321extern const struct _mp_obj_bool_t mp_const_false_obj;
322extern const struct _mp_obj_bool_t mp_const_true_obj;
323extern const struct _mp_obj_tuple_t mp_const_empty_tuple_obj;
324extern const struct _mp_obj_ellipsis_t mp_const_ellipsis_obj;
Damien George6902eed2014-04-04 10:52:59 +0000325extern const struct _mp_obj_exception_t mp_const_MemoryError_obj;
Damien George07ddab52014-03-29 13:15:08 +0000326extern const struct _mp_obj_exception_t mp_const_GeneratorExit_obj;
Damiend99b0522013-12-21 18:17:45 +0000327
Damiend99b0522013-12-21 18:17:45 +0000328// General API for objects
329
Damien Georgea71c83a2014-02-15 11:34:50 +0000330mp_obj_t mp_obj_new_type(qstr name, mp_obj_t bases_tuple, mp_obj_t locals_dict);
Damiend99b0522013-12-21 18:17:45 +0000331mp_obj_t mp_obj_new_none(void);
332mp_obj_t mp_obj_new_bool(bool value);
Damien George6baf76e2013-12-30 22:32:17 +0000333mp_obj_t mp_obj_new_cell(mp_obj_t obj);
Damiend99b0522013-12-21 18:17:45 +0000334mp_obj_t mp_obj_new_int(machine_int_t value);
Paul Sokolovsky48b35722014-01-12 17:30:48 +0200335mp_obj_t mp_obj_new_int_from_uint(machine_uint_t value);
336mp_obj_t mp_obj_new_int_from_long_str(const char *s);
Damien George9d68e9c2014-03-12 15:38:15 +0000337mp_obj_t mp_obj_new_int_from_ll(long long val); // this must return a multi-precision integer object (or raise an overflow exception)
Damien George5fa93b62014-01-22 14:35:10 +0000338mp_obj_t mp_obj_new_str(const byte* data, uint len, bool make_qstr_if_not_already);
Paul Sokolovsky91fb1c92014-01-24 22:50:40 +0200339mp_obj_t mp_obj_new_bytes(const byte* data, uint len);
Damiend99b0522013-12-21 18:17:45 +0000340#if MICROPY_ENABLE_FLOAT
341mp_obj_t mp_obj_new_float(mp_float_t val);
342mp_obj_t mp_obj_new_complex(mp_float_t real, mp_float_t imag);
343#endif
Damien Georgec5966122014-02-15 16:10:44 +0000344mp_obj_t mp_obj_new_exception(const mp_obj_type_t *exc_type);
Paul Sokolovsky962b1cd2014-03-23 21:48:29 +0200345mp_obj_t mp_obj_new_exception_args(const mp_obj_type_t *exc_type, uint n_args, const mp_obj_t *args);
Damien Georgec5966122014-02-15 16:10:44 +0000346mp_obj_t mp_obj_new_exception_msg(const mp_obj_type_t *exc_type, const char *msg);
347mp_obj_t mp_obj_new_exception_msg_varg(const mp_obj_type_t *exc_type, const char *fmt, ...); // counts args by number of % symbols in fmt, excluding %%; can only handle void* sizes (ie no float/double!)
Damiend99b0522013-12-21 18:17:45 +0000348mp_obj_t mp_obj_new_range(int start, int stop, int step);
349mp_obj_t mp_obj_new_range_iterator(int cur, int stop, int step);
Damien Georgebee17b02014-03-27 11:07:04 +0000350mp_obj_t mp_obj_new_fun_bc(uint scope_flags, qstr *args, uint n_args, mp_obj_t def_args, const byte *code);
Damiend99b0522013-12-21 18:17:45 +0000351mp_obj_t mp_obj_new_fun_asm(uint n_args, void *fun);
Damien Georged0691cc2014-01-29 20:30:52 +0000352mp_obj_t mp_obj_new_gen_wrap(mp_obj_t fun);
Damiend99b0522013-12-21 18:17:45 +0000353mp_obj_t mp_obj_new_closure(mp_obj_t fun, mp_obj_t closure_tuple);
John R. Lentonc06763a2014-01-07 17:29:16 +0000354mp_obj_t mp_obj_new_tuple(uint n, const mp_obj_t *items);
Damiend99b0522013-12-21 18:17:45 +0000355mp_obj_t mp_obj_new_list(uint n, mp_obj_t *items);
Damiend99b0522013-12-21 18:17:45 +0000356mp_obj_t mp_obj_new_dict(int n_args);
357mp_obj_t mp_obj_new_set(int n_args, mp_obj_t *items);
Paul Sokolovsky1c6de112014-01-03 02:41:17 +0200358mp_obj_t mp_obj_new_slice(mp_obj_t start, mp_obj_t stop, mp_obj_t step);
Damien George35e2a4e2014-02-05 00:51:47 +0000359mp_obj_t mp_obj_new_super(mp_obj_t type, mp_obj_t obj);
Damien George20006db2014-01-18 14:10:48 +0000360mp_obj_t mp_obj_new_bound_meth(mp_obj_t meth, mp_obj_t self);
Damien George7c9c6672014-01-25 00:17:36 +0000361mp_obj_t mp_obj_new_getitem_iter(mp_obj_t *args);
Damien George28708622014-01-02 21:30:26 +0000362mp_obj_t mp_obj_new_module(qstr module_name);
Damiend99b0522013-12-21 18:17:45 +0000363
Damien George5fa93b62014-01-22 14:35:10 +0000364mp_obj_type_t *mp_obj_get_type(mp_obj_t o_in);
Damiend99b0522013-12-21 18:17:45 +0000365const char *mp_obj_get_type_str(mp_obj_t o_in);
Damien George9e6e9352014-03-26 18:37:06 +0000366bool mp_obj_is_subclass_fast(mp_const_obj_t object, mp_const_obj_t classinfo); // arguments should be type objects
Damiend99b0522013-12-21 18:17:45 +0000367
Paul Sokolovsky76d982e2014-01-13 19:19:16 +0200368void mp_obj_print_helper(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t o_in, mp_print_kind_t kind);
369void mp_obj_print(mp_obj_t o, mp_print_kind_t kind);
Damien George136b1492014-01-19 12:38:49 +0000370void mp_obj_print_exception(mp_obj_t exc);
Damiend99b0522013-12-21 18:17:45 +0000371
Damien Georged17926d2014-03-30 13:35:08 +0100372int mp_obj_is_true(mp_obj_t arg);
Damien Georgea5c82a82014-04-11 11:16:53 +0000373
374// TODO make these all lower case when they have proven themselves
375inline bool MP_OBJ_IS_OBJ(mp_const_obj_t o) { return ((((mp_small_int_t)(o)) & 3) == 0); }
376inline bool MP_OBJ_IS_TYPE(mp_const_obj_t o, const mp_obj_type_t *t) { return (MP_OBJ_IS_OBJ(o) && (((mp_obj_base_t*)(o))->type == (t))); } // this does not work for checking a string, use below macro for that
377inline bool MP_OBJ_IS_SMALL_INT(mp_const_obj_t o) { return ((((mp_small_int_t)(o)) & 1) != 0); }
378inline bool MP_OBJ_IS_INT(mp_const_obj_t o) { return (MP_OBJ_IS_SMALL_INT(o) || MP_OBJ_IS_TYPE(o, &mp_type_int)); } // returns true if o is a small int or long int
379inline bool mp_obj_is_integer(mp_const_obj_t o) { return MP_OBJ_IS_INT(o) || MP_OBJ_IS_TYPE(o, &mp_type_bool); } // returns true if o is bool, small int or long int
380inline bool MP_OBJ_IS_QSTR(mp_const_obj_t o) { return ((((mp_small_int_t)(o)) & 3) == 2); }
381inline bool MP_OBJ_IS_STR(mp_const_obj_t o) { return (MP_OBJ_IS_QSTR(o) || MP_OBJ_IS_TYPE(o, &mp_type_str)); }
382
Damiend99b0522013-12-21 18:17:45 +0000383bool mp_obj_is_callable(mp_obj_t o_in);
384machine_int_t mp_obj_hash(mp_obj_t o_in);
385bool mp_obj_equal(mp_obj_t o1, mp_obj_t o2);
Damiend99b0522013-12-21 18:17:45 +0000386
387machine_int_t mp_obj_get_int(mp_obj_t arg);
Damien George8270e382014-04-03 11:00:54 +0000388bool mp_obj_get_int_maybe(mp_obj_t arg, machine_int_t *value);
Damiend99b0522013-12-21 18:17:45 +0000389#if MICROPY_ENABLE_FLOAT
390mp_float_t mp_obj_get_float(mp_obj_t self_in);
391void mp_obj_get_complex(mp_obj_t self_in, mp_float_t *real, mp_float_t *imag);
392#endif
Damien George5fa93b62014-01-22 14:35:10 +0000393//qstr mp_obj_get_qstr(mp_obj_t arg);
Damien George24ff0632014-03-24 10:47:13 +0000394void mp_obj_get_array(mp_obj_t o, uint *len, mp_obj_t **items);
395void mp_obj_get_array_fixed_n(mp_obj_t o, uint len, mp_obj_t **items);
xbe9e1e8cd2014-03-12 22:57:16 -0700396uint mp_get_index(const mp_obj_type_t *type, machine_uint_t len, mp_obj_t index, bool is_slice);
Damien Georgea5c82a82014-04-11 11:16:53 +0000397mp_obj_t mp_obj_len_maybe(mp_obj_t o_in); /* may return MP_OBJ_NULL */
Damiend99b0522013-12-21 18:17:45 +0000398
Damiend99b0522013-12-21 18:17:45 +0000399// bool
Damien Georgea5c82a82014-04-11 11:16:53 +0000400// TODO make lower case when it has proven itself
401inline mp_obj_t MP_BOOL(machine_int_t x) { return x ? mp_const_true : mp_const_false; }
Damiend99b0522013-12-21 18:17:45 +0000402
403// cell
404mp_obj_t mp_obj_cell_get(mp_obj_t self_in);
405void mp_obj_cell_set(mp_obj_t self_in, mp_obj_t obj);
406
Damien George71c51812014-01-04 20:21:15 +0000407// int
Paul Sokolovskyd26b3792014-01-18 16:07:16 +0200408// For long int, returns value truncated to machine_int_t
409machine_int_t mp_obj_int_get(mp_obj_t self_in);
Damien Georgeeabdf672014-03-22 20:54:01 +0000410#if MICROPY_ENABLE_FLOAT
411mp_float_t mp_obj_int_as_float(mp_obj_t self_in);
412#endif
Paul Sokolovskyd26b3792014-01-18 16:07:16 +0200413// Will rains exception if value doesn't fit into machine_int_t
414machine_int_t mp_obj_int_get_checked(mp_obj_t self_in);
Damien George71c51812014-01-04 20:21:15 +0000415
Damienb86e3f92013-12-29 17:17:43 +0000416// exception
Damien Georgec5966122014-02-15 16:10:44 +0000417bool mp_obj_is_exception_type(mp_obj_t self_in);
418bool mp_obj_is_exception_instance(mp_obj_t self_in);
Paul Sokolovsky962b1cd2014-03-23 21:48:29 +0200419bool mp_obj_exception_match(mp_obj_t exc, const mp_obj_type_t *exc_type);
Damien Georgec5966122014-02-15 16:10:44 +0000420void mp_obj_exception_clear_traceback(mp_obj_t self_in);
Damien George136b1492014-01-19 12:38:49 +0000421void mp_obj_exception_add_traceback(mp_obj_t self_in, qstr file, machine_uint_t line, qstr block);
422void mp_obj_exception_get_traceback(mp_obj_t self_in, machine_uint_t *n, machine_uint_t **values);
Paul Sokolovskyaf1ae302014-03-26 19:17:20 +0200423mp_obj_t mp_obj_exception_get_value(mp_obj_t self_in);
Damienb86e3f92013-12-29 17:17:43 +0000424
Damiend99b0522013-12-21 18:17:45 +0000425// str
Paul Sokolovsky91fb1c92014-01-24 22:50:40 +0200426mp_obj_t mp_obj_str_builder_start(const mp_obj_type_t *type, uint len, byte **data);
Damien George5fa93b62014-01-22 14:35:10 +0000427mp_obj_t mp_obj_str_builder_end(mp_obj_t o_in);
428bool mp_obj_str_equal(mp_obj_t s1, mp_obj_t s2);
429uint mp_obj_str_get_hash(mp_obj_t self_in);
430uint mp_obj_str_get_len(mp_obj_t self_in);
Damien Georgeb829b5c2014-01-25 13:51:19 +0000431qstr mp_obj_str_get_qstr(mp_obj_t self_in); // use this if you will anyway convert the string to a qstr
Damien George5fa93b62014-01-22 14:35:10 +0000432const char *mp_obj_str_get_str(mp_obj_t self_in); // use this only if you need the string to be null terminated
Damien George698ec212014-02-08 18:17:23 +0000433const char *mp_obj_str_get_data(mp_obj_t self_in, uint *len);
Paul Sokolovsky0b7e29c2014-01-28 03:40:06 +0200434void mp_str_print_quoted(void (*print)(void *env, const char *fmt, ...), void *env, const byte *str_data, uint str_len);
Damiend99b0522013-12-21 18:17:45 +0000435
436#if MICROPY_ENABLE_FLOAT
437// float
Damien George0c36da02014-03-08 15:24:39 +0000438typedef struct _mp_obj_float_t {
439 mp_obj_base_t base;
440 mp_float_t value;
441} mp_obj_float_t;
Damiend99b0522013-12-21 18:17:45 +0000442mp_float_t mp_obj_float_get(mp_obj_t self_in);
Damien Georgeae491052014-04-10 20:08:11 +0100443mp_obj_t mp_obj_float_binary_op(int op, mp_float_t lhs_val, mp_obj_t rhs); // can return MP_OBJ_NULL
Damiend99b0522013-12-21 18:17:45 +0000444
445// complex
Damiend99b0522013-12-21 18:17:45 +0000446void mp_obj_complex_get(mp_obj_t self_in, mp_float_t *real, mp_float_t *imag);
Damien Georgeae491052014-04-10 20:08:11 +0100447mp_obj_t mp_obj_complex_binary_op(int op, mp_float_t lhs_real, mp_float_t lhs_imag, mp_obj_t rhs_in); // can return MP_OBJ_NULL
Damiend99b0522013-12-21 18:17:45 +0000448#endif
449
450// tuple
Damiend99b0522013-12-21 18:17:45 +0000451void mp_obj_tuple_get(mp_obj_t self_in, uint *len, mp_obj_t **items);
John R. Lenton07205ec2014-01-13 02:31:00 +0000452void mp_obj_tuple_del(mp_obj_t self_in);
Damien George7f8be592014-03-20 19:20:59 +0000453machine_int_t mp_obj_tuple_hash(mp_obj_t self_in);
Paul Sokolovsky55ca0752014-03-30 17:35:53 +0300454mp_obj_t mp_obj_tuple_make_new(mp_obj_t type_in, uint n_args, uint n_kw, const mp_obj_t *args);
Damiend99b0522013-12-21 18:17:45 +0000455
456// list
Damiend99b0522013-12-21 18:17:45 +0000457mp_obj_t mp_obj_list_append(mp_obj_t self_in, mp_obj_t arg);
458void mp_obj_list_get(mp_obj_t self_in, uint *len, mp_obj_t **items);
Damien George495d7812014-04-08 17:51:47 +0100459void mp_obj_list_set_len(mp_obj_t self_in, uint len);
Damiend99b0522013-12-21 18:17:45 +0000460void mp_obj_list_store(mp_obj_t self_in, mp_obj_t index, mp_obj_t value);
Damien Georgedf6567e2014-03-30 13:54:02 +0100461mp_obj_t mp_obj_list_sort(uint n_args, const mp_obj_t *args, mp_map_t *kwargs);
Damiend99b0522013-12-21 18:17:45 +0000462
463// dict
Damien Georgedf6567e2014-03-30 13:54:02 +0100464typedef struct _mp_obj_dict_t {
465 mp_obj_base_t base;
466 mp_map_t map;
467} mp_obj_dict_t;
Damien George8b0535e2014-04-05 21:53:54 +0100468void mp_obj_dict_init(mp_obj_dict_t *dict, int n_args);
Damiendae7eb72013-12-29 22:32:51 +0000469uint mp_obj_dict_len(mp_obj_t self_in);
Damiend99b0522013-12-21 18:17:45 +0000470mp_obj_t mp_obj_dict_store(mp_obj_t self_in, mp_obj_t key, mp_obj_t value);
Damien George66edc5d2014-04-05 13:25:13 +0100471mp_obj_t mp_obj_dict_delete(mp_obj_t self_in, mp_obj_t key);
Damien Georgedf6567e2014-03-30 13:54:02 +0100472mp_map_t *mp_obj_dict_get_map(mp_obj_t self_in);
Damiend99b0522013-12-21 18:17:45 +0000473
474// set
475void mp_obj_set_store(mp_obj_t self_in, mp_obj_t item);
476
Paul Sokolovsky1c6de112014-01-03 02:41:17 +0200477// slice
Paul Sokolovsky1c6de112014-01-03 02:41:17 +0200478void mp_obj_slice_get(mp_obj_t self_in, machine_int_t *start, machine_int_t *stop, machine_int_t *step);
479
Paul Sokolovsky427905c2014-01-18 19:24:47 +0200480// array
Paul Sokolovsky33996682014-01-21 23:30:10 +0200481uint mp_obj_array_len(mp_obj_t self_in);
Paul Sokolovsky7f11c792014-01-29 00:21:41 +0200482mp_obj_t mp_obj_new_bytearray_by_ref(uint n, void *items);
John R. Lenton07205ec2014-01-13 02:31:00 +0000483
Damiend99b0522013-12-21 18:17:45 +0000484// functions
Damien Georged5e81822014-02-26 17:47:05 +0000485#define MP_OBJ_FUN_ARGS_MAX (0xffff) // to set maximum value in n_args_max below
Damien George66028ab2014-01-03 14:03:48 +0000486typedef struct _mp_obj_fun_native_t { // need this so we can define const objects (to go in ROM)
Damiend99b0522013-12-21 18:17:45 +0000487 mp_obj_base_t base;
Damien George51047752014-02-26 17:40:52 +0000488 bool is_kw : 1;
Damien Georged5e81822014-02-26 17:47:05 +0000489 uint n_args_min : 15; // inclusive
490 uint n_args_max : 16; // inclusive
Damien660365e2013-12-17 18:27:24 +0000491 void *fun;
Damien George66028ab2014-01-03 14:03:48 +0000492 // TODO add mp_map_t *globals
493 // for const function objects, make an empty, const map
494 // such functions won't be able to access the global scope, but that's probably okay
Damiend99b0522013-12-21 18:17:45 +0000495} mp_obj_fun_native_t;
Damien George97209d32014-01-07 15:58:30 +0000496
Damien Georgebee17b02014-03-27 11:07:04 +0000497void mp_obj_fun_bc_get(mp_obj_t self_in, int *n_args, const byte **code);
Paul Sokolovsky7fafb282014-03-30 20:21:28 +0300498bool mp_obj_fun_prepare_simple_args(mp_obj_t self_in, uint n_args, uint n_kw, const mp_obj_t *args,
499 uint *out_args1_len, const mp_obj_t **out_args1, uint *out_args2_len, const mp_obj_t **out_args2);
Damien660365e2013-12-17 18:27:24 +0000500
Paul Sokolovskydff3f892014-01-20 18:37:30 +0200501mp_obj_t mp_identity(mp_obj_t self);
Paul Sokolovsky557c9d52014-02-08 21:57:19 +0200502MP_DECLARE_CONST_FUN_OBJ(mp_identity_obj);
Paul Sokolovskydff3f892014-01-20 18:37:30 +0200503
Damien George28708622014-01-02 21:30:26 +0000504// module
Damien George0c36da02014-03-08 15:24:39 +0000505typedef struct _mp_obj_module_t {
506 mp_obj_base_t base;
507 qstr name;
Damien George8b0535e2014-04-05 21:53:54 +0100508 mp_obj_dict_t *globals;
Damien George0c36da02014-03-08 15:24:39 +0000509} mp_obj_module_t;
Damien George8b0535e2014-04-05 21:53:54 +0100510mp_obj_dict_t *mp_obj_module_get_globals(mp_obj_t self_in);
Damien Georgeeae16442014-01-11 19:22:29 +0000511
512// staticmethod and classmethod types; defined here so we can make const versions
Damien George64131f32014-02-06 20:31:44 +0000513// this structure is used for instances of both staticmethod and classmethod
514typedef struct _mp_obj_static_class_method_t {
Damien Georgeeae16442014-01-11 19:22:29 +0000515 mp_obj_base_t base;
516 mp_obj_t fun;
Damien George64131f32014-02-06 20:31:44 +0000517} mp_obj_static_class_method_t;
Paul Sokolovsky439542f2014-01-21 00:19:19 +0200518
519// sequence helpers
520void mp_seq_multiply(const void *items, uint item_sz, uint len, uint times, void *dest);
Paul Sokolovsky7364af22014-02-02 02:38:22 +0200521bool m_seq_get_fast_slice_indexes(machine_uint_t len, mp_obj_t slice, machine_uint_t *begin, machine_uint_t *end);
Paul Sokolovskyee4aaf72014-02-08 23:17:51 +0200522#define m_seq_copy(dest, src, len, item_t) memcpy(dest, src, len * sizeof(item_t))
Paul Sokolovsky55ca0752014-03-30 17:35:53 +0300523#define m_seq_cat(dest, src1, len1, src2, len2, item_t) { memcpy(dest, src1, (len1) * sizeof(item_t)); memcpy(dest + (len1), src2, (len2) * sizeof(item_t)); }
Paul Sokolovsky87e85b72014-02-02 08:24:07 +0200524bool mp_seq_cmp_bytes(int op, const byte *data1, uint len1, const byte *data2, uint len2);
Paul Sokolovsky1a996c42014-02-08 22:49:46 +0200525bool mp_seq_cmp_objs(int op, const mp_obj_t *items1, uint len1, const mp_obj_t *items2, uint len2);
Paul Sokolovsky0cd1dc02014-02-10 06:37:11 +0200526mp_obj_t mp_seq_index_obj(const mp_obj_t *items, uint len, uint n_args, const mp_obj_t *args);
Paul Sokolovskyac0134d2014-02-10 07:10:55 +0200527mp_obj_t mp_seq_count_obj(const mp_obj_t *items, uint len, mp_obj_t value);