blob: af6ef83189b561289b164a91c2074c93b04b3a50 [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
24#define MP_OBJ_NULL ((mp_obj_t)NULL)
25
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)
Damiend99b0522013-12-21 18:17:45 +000040#define MP_OBJ_IS_SMALL_INT(o) ((((mp_small_int_t)(o)) & 1) != 0)
Damien George38a2da62014-01-08 17:33:12 +000041#define MP_OBJ_IS_QSTR(o) ((((mp_small_int_t)(o)) & 3) == 2)
42#define MP_OBJ_IS_OBJ(o) ((((mp_small_int_t)(o)) & 3) == 0)
Damien George7c9c6672014-01-25 00:17:36 +000043#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
Damien George3e1a5c12014-03-29 13:43:38 +000044#define MP_OBJ_IS_INT(o) (MP_OBJ_IS_SMALL_INT(o) || MP_OBJ_IS_TYPE(o, &mp_type_int))
45#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 +000046
Damiend99b0522013-12-21 18:17:45 +000047#define MP_OBJ_SMALL_INT_VALUE(o) (((mp_small_int_t)(o)) >> 1)
Damien George38a2da62014-01-08 17:33:12 +000048#define MP_OBJ_NEW_SMALL_INT(small_int) ((mp_obj_t)(((small_int) << 1) | 1))
49
50#define MP_OBJ_QSTR_VALUE(o) (((mp_small_int_t)(o)) >> 2)
51#define MP_OBJ_NEW_QSTR(qstr) ((mp_obj_t)((((machine_uint_t)qstr) << 2) | 2))
Damiend99b0522013-12-21 18:17:45 +000052
53// These macros are used to declare and define constant function objects
54// You can put "static" in front of the definitions to make them local
55
56#define MP_DECLARE_CONST_FUN_OBJ(obj_name) extern const mp_obj_fun_native_t obj_name
57
Damien George3e1a5c12014-03-29 13:43:38 +000058#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 +000059#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)
60#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)
61#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)
62#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 +000063#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 +000064#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 +000065#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 +000066
Damien George9b196cd2014-03-26 21:47:19 +000067// This macro is used to define constant dict objects
68// You can put "static" in front of the definition to make it local
69
70#define MP_DEFINE_CONST_DICT(dict_name, table_name) \
71 const mp_obj_dict_t dict_name = { \
Damien George3e1a5c12014-03-29 13:43:38 +000072 .base = {&mp_type_dict}, \
Damien George9b196cd2014-03-26 21:47:19 +000073 .map = { \
74 .all_keys_are_qstrs = 1, \
75 .table_is_fixed_array = 1, \
76 .used = sizeof(table_name) / sizeof(mp_map_elem_t), \
77 .alloc = sizeof(table_name) / sizeof(mp_map_elem_t), \
78 .table = (mp_map_elem_t*)table_name, \
79 }, \
80 }
81
Damien Georgeeae16442014-01-11 19:22:29 +000082// These macros are used to declare and define constant staticmethond and classmethod objects
83// You can put "static" in front of the definitions to make them local
84
Damien George64131f32014-02-06 20:31:44 +000085#define MP_DECLARE_CONST_STATICMETHOD_OBJ(obj_name) extern const mp_obj_static_class_method_t obj_name
86#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 +000087
Damien George64131f32014-02-06 20:31:44 +000088#define MP_DEFINE_CONST_STATICMETHOD_OBJ(obj_name, fun_name) const mp_obj_static_class_method_t obj_name = {{&mp_type_staticmethod}, fun_name}
89#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 +000090
Damien Georgedf6567e2014-03-30 13:54:02 +010091// Underlying map/hash table implementation (not dict object or map function)
92
93typedef struct _mp_map_elem_t {
94 mp_obj_t key;
95 mp_obj_t value;
96} mp_map_elem_t;
97
98// TODO maybe have a truncated mp_map_t for fixed tables, since alloc=used
99// put alloc last in the structure, so the truncated version does not need it
100// this would save 1 ROM word for all ROM objects that have a locals_dict
101// would also need a trucated dict structure
102
103typedef struct _mp_map_t {
104 machine_uint_t all_keys_are_qstrs : 1;
105 machine_uint_t table_is_fixed_array : 1;
106 machine_uint_t used : (8 * sizeof(machine_uint_t) - 2);
107 machine_uint_t alloc;
108 mp_map_elem_t *table;
109} mp_map_t;
110
Damien George95004e52014-04-05 17:17:19 +0100111// These can be or'd together
Damien Georgedf6567e2014-03-30 13:54:02 +0100112typedef enum _mp_map_lookup_kind_t {
113 MP_MAP_LOOKUP, // 0
114 MP_MAP_LOOKUP_ADD_IF_NOT_FOUND, // 1
115 MP_MAP_LOOKUP_REMOVE_IF_FOUND, // 2
Damien Georgedf6567e2014-03-30 13:54:02 +0100116} mp_map_lookup_kind_t;
117
Damien George8b0535e2014-04-05 21:53:54 +0100118#define MP_MAP_SLOT_IS_FILLED(map, pos) ((map)->table[pos].key != MP_OBJ_NULL && (map)->table[pos].key != MP_OBJ_SENTINEL)
119
Damien Georgedf6567e2014-03-30 13:54:02 +0100120void mp_map_init(mp_map_t *map, int n);
121void mp_map_init_fixed_table(mp_map_t *map, int n, const mp_obj_t *table);
122mp_map_t *mp_map_new(int n);
123void mp_map_deinit(mp_map_t *map);
124void mp_map_free(mp_map_t *map);
125mp_map_elem_t* mp_map_lookup(mp_map_t *map, mp_obj_t index, mp_map_lookup_kind_t lookup_kind);
126void mp_map_clear(mp_map_t *map);
Paul Sokolovskye3f58c82014-04-05 04:14:22 +0300127void mp_map_dump(mp_map_t *map);
Damien Georgedf6567e2014-03-30 13:54:02 +0100128
129// Underlying set implementation (not set object)
130
131typedef struct _mp_set_t {
132 machine_uint_t alloc;
133 machine_uint_t used;
134 mp_obj_t *table;
135} mp_set_t;
136
Damien George8b0535e2014-04-05 21:53:54 +0100137#define MP_SET_SLOT_IS_FILLED(set, pos) ((set)->table[pos] != MP_OBJ_NULL && (set)->table[pos] != MP_OBJ_SENTINEL)
138
Damien Georgedf6567e2014-03-30 13:54:02 +0100139void mp_set_init(mp_set_t *set, int n);
140mp_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 +0100141mp_obj_t mp_set_remove_first(mp_set_t *set);
Damien Georgedf6567e2014-03-30 13:54:02 +0100142void mp_set_clear(mp_set_t *set);
Damiend99b0522013-12-21 18:17:45 +0000143
144// Type definitions for methods
145
146typedef mp_obj_t (*mp_fun_0_t)(void);
147typedef mp_obj_t (*mp_fun_1_t)(mp_obj_t);
148typedef mp_obj_t (*mp_fun_2_t)(mp_obj_t, mp_obj_t);
John R. Lenton45a87442014-01-04 01:15:01 +0000149typedef mp_obj_t (*mp_fun_3_t)(mp_obj_t, mp_obj_t, mp_obj_t);
Damiend99b0522013-12-21 18:17:45 +0000150typedef mp_obj_t (*mp_fun_t)(void);
Damien Georgea11ceca2014-01-19 16:02:09 +0000151typedef mp_obj_t (*mp_fun_var_t)(uint n, const mp_obj_t *);
Damien Georgedf6567e2014-03-30 13:54:02 +0100152typedef mp_obj_t (*mp_fun_kw_t)(uint n, const mp_obj_t *, mp_map_t *);
Damiend99b0522013-12-21 18:17:45 +0000153
Paul Sokolovsky76d982e2014-01-13 19:19:16 +0200154typedef enum {
Paul Sokolovskya96d3d02014-03-31 01:10:10 +0300155 PRINT_STR,
156 PRINT_REPR,
157 PRINT_EXC, // Special format for printing exception in unhandled exception message
Paul Sokolovsky76d982e2014-01-13 19:19:16 +0200158} mp_print_kind_t;
159
160typedef 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 +0000161typedef mp_obj_t (*mp_make_new_fun_t)(mp_obj_t type_in, uint n_args, uint n_kw, const mp_obj_t *args);
162typedef 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 +0000163typedef mp_obj_t (*mp_unary_op_fun_t)(int op, mp_obj_t);
164typedef mp_obj_t (*mp_binary_op_fun_t)(int op, mp_obj_t, mp_obj_t);
Damien George20006db2014-01-18 14:10:48 +0000165typedef 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 +0100166typedef 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
167typedef 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 +0000168
169typedef struct _mp_method_t {
Damien Georgec12b2212014-03-26 20:15:40 +0000170 qstr name;
Damiend99b0522013-12-21 18:17:45 +0000171 mp_const_obj_t fun;
172} mp_method_t;
173
Paul Sokolovsky5b15daf2014-01-07 20:12:26 +0200174// Buffer protocol
175typedef struct _buffer_info_t {
176 // if we'd bother to support various versions of structure
177 // (with different number of fields), we can distinguish
178 // them with ver = sizeof(struct). Cons: overkill for *micro*?
179 //int ver; // ?
180
181 void *buf;
182 machine_int_t len;
183
184 // Rationale: have array.array and have SIMD operations on them
185 // Cons: users can pass item size to processing functions themselves,
186 // though that's not "plug&play"
187 // int itemsize;
188
189 // Rationale: to load arbitrary-sized sprites directly to LCD
190 // Cons: a bit adhoc usecase
191 // int stride;
192} buffer_info_t;
193#define BUFFER_READ (1)
194#define BUFFER_WRITE (2)
195#define BUFFER_RW (BUFFER_READ | BUFFER_WRITE)
196typedef struct _mp_buffer_p_t {
197 machine_int_t (*get_buffer)(mp_obj_t obj, buffer_info_t *bufinfo, int flags);
198} mp_buffer_p_t;
Paul Sokolovsky3aa8ee72014-04-09 00:25:28 +0300199bool mp_get_buffer(mp_obj_t obj, buffer_info_t *bufinfo);
200void mp_get_buffer_raise(mp_obj_t obj, buffer_info_t *bufinfo);
Paul Sokolovsky5b15daf2014-01-07 20:12:26 +0200201
202// Stream protocol
203typedef struct _mp_stream_p_t {
204 // On error, functions should return -1 and fill in *errcode (values are
205 // implementation-dependent, but will be exposed to user, e.g. via exception).
206 machine_int_t (*read)(mp_obj_t obj, void *buf, machine_uint_t size, int *errcode);
207 machine_int_t (*write)(mp_obj_t obj, const void *buf, machine_uint_t size, int *errcode);
208 // add seek() ?
209} mp_stream_p_t;
210
Damiend99b0522013-12-21 18:17:45 +0000211struct _mp_obj_type_t {
212 mp_obj_base_t base;
Damien Georgea71c83a2014-02-15 11:34:50 +0000213 qstr name;
Damiend99b0522013-12-21 18:17:45 +0000214 mp_print_fun_t print;
Damien George71c51812014-01-04 20:21:15 +0000215 mp_make_new_fun_t make_new; // to make an instance of the type
Damiend99b0522013-12-21 18:17:45 +0000216
Damien George20006db2014-01-18 14:10:48 +0000217 mp_call_fun_t call;
Damiend99b0522013-12-21 18:17:45 +0000218 mp_unary_op_fun_t unary_op; // can return NULL if op not supported
219 mp_binary_op_fun_t binary_op; // can return NULL if op not supported
220
Damien Georgea71c83a2014-02-15 11:34:50 +0000221 mp_load_attr_fun_t load_attr;
Damien George1d24ea52014-04-08 21:11:49 +0100222 mp_store_attr_fun_t store_attr; // if value is MP_OBJ_NULL, then delete that attribute
Damien Georgef4c9b332014-04-08 21:32:29 +0100223
224 // Implements container[index] = val. If val == MP_OBJ_NULL, then it's a delete.
225 // Note that load_item is implemented by binary_op(RT_BINARY_OP_SUBSCR)
Damien Georgea71c83a2014-02-15 11:34:50 +0000226 mp_store_item_fun_t store_item;
227
Damiend99b0522013-12-21 18:17:45 +0000228 mp_fun_1_t getiter;
Damien George66eaf842014-03-26 19:27:58 +0000229 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 +0000230
Paul Sokolovsky5b15daf2014-01-07 20:12:26 +0200231 // Alternatively, pointer(s) to interfaces to save space
232 // in mp_obj_type_t at the expense of extra pointer and extra dereference
233 // when actually used.
234 mp_buffer_p_t buffer_p;
Damien George27e735f2014-04-05 23:02:23 +0100235 const mp_stream_p_t *stream_p;
Paul Sokolovsky5b15daf2014-01-07 20:12:26 +0200236
Damien George004cdce2014-01-09 21:43:51 +0000237 // these are for dynamically created types (classes)
238 mp_obj_t bases_tuple;
239 mp_obj_t locals_dict;
Damien George062478e2014-01-09 20:57:50 +0000240
Damiend99b0522013-12-21 18:17:45 +0000241 /*
242 What we might need to add here:
243
Damiend99b0522013-12-21 18:17:45 +0000244 store_subscr list dict
245
246 len str tuple list map
247 abs float complex
248 hash bool int none str
249 equal int str
Damiend99b0522013-12-21 18:17:45 +0000250
251 unpack seq list tuple
Damiend99b0522013-12-21 18:17:45 +0000252 */
253};
254
ian-v7a16fad2014-01-06 09:52:29 -0800255typedef struct _mp_obj_type_t mp_obj_type_t;
Damiend99b0522013-12-21 18:17:45 +0000256
Damien Georgec5966122014-02-15 16:10:44 +0000257// Constant types, globally accessible
Damien Georgec5966122014-02-15 16:10:44 +0000258extern const mp_obj_type_t mp_type_type;
Damien George3e1a5c12014-03-29 13:43:38 +0000259extern const mp_obj_type_t mp_type_object;
260extern const mp_obj_type_t mp_type_NoneType;
261extern const mp_obj_type_t mp_type_bool;
262extern const mp_obj_type_t mp_type_int;
263extern const mp_obj_type_t mp_type_str;
264extern const mp_obj_type_t mp_type_bytes;
Paul Sokolovsky4dcb6052014-04-08 22:09:14 +0300265extern const mp_obj_type_t mp_type_bytearray;
Damien George3e1a5c12014-03-29 13:43:38 +0000266extern const mp_obj_type_t mp_type_float;
267extern const mp_obj_type_t mp_type_complex;
268extern const mp_obj_type_t mp_type_tuple;
269extern const mp_obj_type_t mp_type_list;
270extern const mp_obj_type_t mp_type_map; // map (the python builtin, not the dict implementation detail)
271extern const mp_obj_type_t mp_type_enumerate;
272extern const mp_obj_type_t mp_type_filter;
273extern const mp_obj_type_t mp_type_dict;
274extern const mp_obj_type_t mp_type_set;
275extern const mp_obj_type_t mp_type_slice;
276extern const mp_obj_type_t mp_type_zip;
277extern const mp_obj_type_t mp_type_array;
278extern const mp_obj_type_t mp_type_super;
279extern const mp_obj_type_t mp_type_gen_instance;
280extern const mp_obj_type_t mp_type_fun_native;
281extern const mp_obj_type_t mp_type_fun_bc;
282extern const mp_obj_type_t mp_type_module;
283extern const mp_obj_type_t mp_type_staticmethod;
284extern const mp_obj_type_t mp_type_classmethod;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000285
286// Exceptions
Damien Georgec5966122014-02-15 16:10:44 +0000287extern const mp_obj_type_t mp_type_BaseException;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000288extern const mp_obj_type_t mp_type_ArithmeticError;
289extern const mp_obj_type_t mp_type_AssertionError;
290extern const mp_obj_type_t mp_type_AttributeError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000291extern const mp_obj_type_t mp_type_EOFError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000292extern const mp_obj_type_t mp_type_Exception;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000293extern const mp_obj_type_t mp_type_GeneratorExit;
294extern const mp_obj_type_t mp_type_IOError;
295extern const mp_obj_type_t mp_type_ImportError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000296extern const mp_obj_type_t mp_type_IndentationError;
297extern const mp_obj_type_t mp_type_IndexError;
298extern const mp_obj_type_t mp_type_KeyError;
299extern const mp_obj_type_t mp_type_LookupError;
300extern const mp_obj_type_t mp_type_MemoryError;
301extern const mp_obj_type_t mp_type_NameError;
302extern const mp_obj_type_t mp_type_NotImplementedError;
303extern const mp_obj_type_t mp_type_OSError;
304extern const mp_obj_type_t mp_type_OverflowError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000305extern const mp_obj_type_t mp_type_RuntimeError;
Damien Georgeffb5cfc2014-03-25 14:29:40 +0000306extern const mp_obj_type_t mp_type_StopIteration;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000307extern const mp_obj_type_t mp_type_SyntaxError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000308extern const mp_obj_type_t mp_type_SystemError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000309extern const mp_obj_type_t mp_type_TypeError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000310extern const mp_obj_type_t mp_type_ValueError;
Rachel Dowdall721c55d2014-03-22 15:28:16 +0000311extern const mp_obj_type_t mp_type_ZeroDivisionError;
312
Damiend99b0522013-12-21 18:17:45 +0000313// Constant objects, globally accessible
Damien George07ddab52014-03-29 13:15:08 +0000314// The macros are for convenience only
315#define mp_const_none ((mp_obj_t)&mp_const_none_obj)
316#define mp_const_false ((mp_obj_t)&mp_const_false_obj)
317#define mp_const_true ((mp_obj_t)&mp_const_true_obj)
318#define mp_const_empty_tuple ((mp_obj_t)&mp_const_empty_tuple_obj)
319extern const struct _mp_obj_none_t mp_const_none_obj;
320extern const struct _mp_obj_bool_t mp_const_false_obj;
321extern const struct _mp_obj_bool_t mp_const_true_obj;
322extern const struct _mp_obj_tuple_t mp_const_empty_tuple_obj;
323extern const struct _mp_obj_ellipsis_t mp_const_ellipsis_obj;
Damien George6902eed2014-04-04 10:52:59 +0000324extern const struct _mp_obj_exception_t mp_const_MemoryError_obj;
Damien George07ddab52014-03-29 13:15:08 +0000325extern const struct _mp_obj_exception_t mp_const_GeneratorExit_obj;
Damiend99b0522013-12-21 18:17:45 +0000326
Damiend99b0522013-12-21 18:17:45 +0000327// General API for objects
328
Damien Georgea71c83a2014-02-15 11:34:50 +0000329mp_obj_t mp_obj_new_type(qstr name, mp_obj_t bases_tuple, mp_obj_t locals_dict);
Damiend99b0522013-12-21 18:17:45 +0000330mp_obj_t mp_obj_new_none(void);
331mp_obj_t mp_obj_new_bool(bool value);
Damien George6baf76e2013-12-30 22:32:17 +0000332mp_obj_t mp_obj_new_cell(mp_obj_t obj);
Damiend99b0522013-12-21 18:17:45 +0000333mp_obj_t mp_obj_new_int(machine_int_t value);
Paul Sokolovsky48b35722014-01-12 17:30:48 +0200334mp_obj_t mp_obj_new_int_from_uint(machine_uint_t value);
335mp_obj_t mp_obj_new_int_from_long_str(const char *s);
Damien George9d68e9c2014-03-12 15:38:15 +0000336mp_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 +0000337mp_obj_t mp_obj_new_str(const byte* data, uint len, bool make_qstr_if_not_already);
Paul Sokolovsky91fb1c92014-01-24 22:50:40 +0200338mp_obj_t mp_obj_new_bytes(const byte* data, uint len);
Damiend99b0522013-12-21 18:17:45 +0000339#if MICROPY_ENABLE_FLOAT
340mp_obj_t mp_obj_new_float(mp_float_t val);
341mp_obj_t mp_obj_new_complex(mp_float_t real, mp_float_t imag);
342#endif
Damien Georgec5966122014-02-15 16:10:44 +0000343mp_obj_t mp_obj_new_exception(const mp_obj_type_t *exc_type);
Paul Sokolovsky962b1cd2014-03-23 21:48:29 +0200344mp_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 +0000345mp_obj_t mp_obj_new_exception_msg(const mp_obj_type_t *exc_type, const char *msg);
346mp_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 +0000347mp_obj_t mp_obj_new_range(int start, int stop, int step);
348mp_obj_t mp_obj_new_range_iterator(int cur, int stop, int step);
Damien Georgebee17b02014-03-27 11:07:04 +0000349mp_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 +0000350mp_obj_t mp_obj_new_fun_asm(uint n_args, void *fun);
Damien Georged0691cc2014-01-29 20:30:52 +0000351mp_obj_t mp_obj_new_gen_wrap(mp_obj_t fun);
Damiend99b0522013-12-21 18:17:45 +0000352mp_obj_t mp_obj_new_closure(mp_obj_t fun, mp_obj_t closure_tuple);
John R. Lentonc06763a2014-01-07 17:29:16 +0000353mp_obj_t mp_obj_new_tuple(uint n, const mp_obj_t *items);
Damiend99b0522013-12-21 18:17:45 +0000354mp_obj_t mp_obj_new_list(uint n, mp_obj_t *items);
Damiend99b0522013-12-21 18:17:45 +0000355mp_obj_t mp_obj_new_dict(int n_args);
356mp_obj_t mp_obj_new_set(int n_args, mp_obj_t *items);
Paul Sokolovsky1c6de112014-01-03 02:41:17 +0200357mp_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 +0000358mp_obj_t mp_obj_new_super(mp_obj_t type, mp_obj_t obj);
Damien George20006db2014-01-18 14:10:48 +0000359mp_obj_t mp_obj_new_bound_meth(mp_obj_t meth, mp_obj_t self);
Damien George7c9c6672014-01-25 00:17:36 +0000360mp_obj_t mp_obj_new_getitem_iter(mp_obj_t *args);
Damien George28708622014-01-02 21:30:26 +0000361mp_obj_t mp_obj_new_module(qstr module_name);
Damiend99b0522013-12-21 18:17:45 +0000362
Damien George5fa93b62014-01-22 14:35:10 +0000363mp_obj_type_t *mp_obj_get_type(mp_obj_t o_in);
Damiend99b0522013-12-21 18:17:45 +0000364const char *mp_obj_get_type_str(mp_obj_t o_in);
Damien George9e6e9352014-03-26 18:37:06 +0000365bool 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 +0000366
Paul Sokolovsky76d982e2014-01-13 19:19:16 +0200367void mp_obj_print_helper(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t o_in, mp_print_kind_t kind);
368void mp_obj_print(mp_obj_t o, mp_print_kind_t kind);
Damien George136b1492014-01-19 12:38:49 +0000369void mp_obj_print_exception(mp_obj_t exc);
Damiend99b0522013-12-21 18:17:45 +0000370
Damien Georged17926d2014-03-30 13:35:08 +0100371int mp_obj_is_true(mp_obj_t arg);
Damien Georgeb5fbd0b2014-04-09 19:55:33 +0100372bool mp_obj_is_integer(mp_obj_t o_in); // returns true if o_in is bool, small int, or long int
Damiend99b0522013-12-21 18:17:45 +0000373bool mp_obj_is_callable(mp_obj_t o_in);
374machine_int_t mp_obj_hash(mp_obj_t o_in);
375bool mp_obj_equal(mp_obj_t o1, mp_obj_t o2);
Damiend99b0522013-12-21 18:17:45 +0000376
377machine_int_t mp_obj_get_int(mp_obj_t arg);
Damien George8270e382014-04-03 11:00:54 +0000378bool mp_obj_get_int_maybe(mp_obj_t arg, machine_int_t *value);
Damiend99b0522013-12-21 18:17:45 +0000379#if MICROPY_ENABLE_FLOAT
380mp_float_t mp_obj_get_float(mp_obj_t self_in);
381void mp_obj_get_complex(mp_obj_t self_in, mp_float_t *real, mp_float_t *imag);
382#endif
Damien George5fa93b62014-01-22 14:35:10 +0000383//qstr mp_obj_get_qstr(mp_obj_t arg);
Damien George24ff0632014-03-24 10:47:13 +0000384void mp_obj_get_array(mp_obj_t o, uint *len, mp_obj_t **items);
385void mp_obj_get_array_fixed_n(mp_obj_t o, uint len, mp_obj_t **items);
xbe9e1e8cd2014-03-12 22:57:16 -0700386uint mp_get_index(const mp_obj_type_t *type, machine_uint_t len, mp_obj_t index, bool is_slice);
John R. Lenton4bee76e2014-01-10 11:25:03 +0000387mp_obj_t mp_obj_len_maybe(mp_obj_t o_in); /* may return NULL */
Damiend99b0522013-12-21 18:17:45 +0000388
Damiend99b0522013-12-21 18:17:45 +0000389// bool
John R. Lentonb8698fc2014-01-11 00:58:59 +0000390#define MP_BOOL(x) (x ? mp_const_true : mp_const_false)
Damiend99b0522013-12-21 18:17:45 +0000391
392// cell
393mp_obj_t mp_obj_cell_get(mp_obj_t self_in);
394void mp_obj_cell_set(mp_obj_t self_in, mp_obj_t obj);
395
Damien George71c51812014-01-04 20:21:15 +0000396// int
Paul Sokolovskyd26b3792014-01-18 16:07:16 +0200397// For long int, returns value truncated to machine_int_t
398machine_int_t mp_obj_int_get(mp_obj_t self_in);
Damien Georgeeabdf672014-03-22 20:54:01 +0000399#if MICROPY_ENABLE_FLOAT
400mp_float_t mp_obj_int_as_float(mp_obj_t self_in);
401#endif
Paul Sokolovskyd26b3792014-01-18 16:07:16 +0200402// Will rains exception if value doesn't fit into machine_int_t
403machine_int_t mp_obj_int_get_checked(mp_obj_t self_in);
Damien George71c51812014-01-04 20:21:15 +0000404
Damienb86e3f92013-12-29 17:17:43 +0000405// exception
Damien Georgec5966122014-02-15 16:10:44 +0000406bool mp_obj_is_exception_type(mp_obj_t self_in);
407bool mp_obj_is_exception_instance(mp_obj_t self_in);
Paul Sokolovsky962b1cd2014-03-23 21:48:29 +0200408bool mp_obj_exception_match(mp_obj_t exc, const mp_obj_type_t *exc_type);
Damien Georgec5966122014-02-15 16:10:44 +0000409void mp_obj_exception_clear_traceback(mp_obj_t self_in);
Damien George136b1492014-01-19 12:38:49 +0000410void mp_obj_exception_add_traceback(mp_obj_t self_in, qstr file, machine_uint_t line, qstr block);
411void 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 +0200412mp_obj_t mp_obj_exception_get_value(mp_obj_t self_in);
Damienb86e3f92013-12-29 17:17:43 +0000413
Damiend99b0522013-12-21 18:17:45 +0000414// str
Paul Sokolovsky91fb1c92014-01-24 22:50:40 +0200415mp_obj_t mp_obj_str_builder_start(const mp_obj_type_t *type, uint len, byte **data);
Damien George5fa93b62014-01-22 14:35:10 +0000416mp_obj_t mp_obj_str_builder_end(mp_obj_t o_in);
417bool mp_obj_str_equal(mp_obj_t s1, mp_obj_t s2);
418uint mp_obj_str_get_hash(mp_obj_t self_in);
419uint mp_obj_str_get_len(mp_obj_t self_in);
Damien Georgeb829b5c2014-01-25 13:51:19 +0000420qstr 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 +0000421const 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 +0000422const char *mp_obj_str_get_data(mp_obj_t self_in, uint *len);
Paul Sokolovsky0b7e29c2014-01-28 03:40:06 +0200423void 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 +0000424
425#if MICROPY_ENABLE_FLOAT
426// float
Damien George0c36da02014-03-08 15:24:39 +0000427typedef struct _mp_obj_float_t {
428 mp_obj_base_t base;
429 mp_float_t value;
430} mp_obj_float_t;
Damiend99b0522013-12-21 18:17:45 +0000431mp_float_t mp_obj_float_get(mp_obj_t self_in);
Damien Georgeae491052014-04-10 20:08:11 +0100432mp_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 +0000433
434// complex
Damiend99b0522013-12-21 18:17:45 +0000435void mp_obj_complex_get(mp_obj_t self_in, mp_float_t *real, mp_float_t *imag);
Damien Georgeae491052014-04-10 20:08:11 +0100436mp_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 +0000437#endif
438
439// tuple
Damiend99b0522013-12-21 18:17:45 +0000440void mp_obj_tuple_get(mp_obj_t self_in, uint *len, mp_obj_t **items);
John R. Lenton07205ec2014-01-13 02:31:00 +0000441void mp_obj_tuple_del(mp_obj_t self_in);
Damien George7f8be592014-03-20 19:20:59 +0000442machine_int_t mp_obj_tuple_hash(mp_obj_t self_in);
Paul Sokolovsky55ca0752014-03-30 17:35:53 +0300443mp_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 +0000444
445// list
Damiend99b0522013-12-21 18:17:45 +0000446mp_obj_t mp_obj_list_append(mp_obj_t self_in, mp_obj_t arg);
447void mp_obj_list_get(mp_obj_t self_in, uint *len, mp_obj_t **items);
Damien George495d7812014-04-08 17:51:47 +0100448void mp_obj_list_set_len(mp_obj_t self_in, uint len);
Damiend99b0522013-12-21 18:17:45 +0000449void mp_obj_list_store(mp_obj_t self_in, mp_obj_t index, mp_obj_t value);
Damien Georgedf6567e2014-03-30 13:54:02 +0100450mp_obj_t mp_obj_list_sort(uint n_args, const mp_obj_t *args, mp_map_t *kwargs);
Damiend99b0522013-12-21 18:17:45 +0000451
452// dict
Damien Georgedf6567e2014-03-30 13:54:02 +0100453typedef struct _mp_obj_dict_t {
454 mp_obj_base_t base;
455 mp_map_t map;
456} mp_obj_dict_t;
Damien George8b0535e2014-04-05 21:53:54 +0100457void mp_obj_dict_init(mp_obj_dict_t *dict, int n_args);
Damiendae7eb72013-12-29 22:32:51 +0000458uint mp_obj_dict_len(mp_obj_t self_in);
Damiend99b0522013-12-21 18:17:45 +0000459mp_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 +0100460mp_obj_t mp_obj_dict_delete(mp_obj_t self_in, mp_obj_t key);
Damien Georgedf6567e2014-03-30 13:54:02 +0100461mp_map_t *mp_obj_dict_get_map(mp_obj_t self_in);
Damiend99b0522013-12-21 18:17:45 +0000462
463// set
464void mp_obj_set_store(mp_obj_t self_in, mp_obj_t item);
465
Paul Sokolovsky1c6de112014-01-03 02:41:17 +0200466// slice
Paul Sokolovsky1c6de112014-01-03 02:41:17 +0200467void mp_obj_slice_get(mp_obj_t self_in, machine_int_t *start, machine_int_t *stop, machine_int_t *step);
468
Paul Sokolovsky427905c2014-01-18 19:24:47 +0200469// array
Paul Sokolovsky33996682014-01-21 23:30:10 +0200470uint mp_obj_array_len(mp_obj_t self_in);
Paul Sokolovsky7f11c792014-01-29 00:21:41 +0200471mp_obj_t mp_obj_new_bytearray_by_ref(uint n, void *items);
John R. Lenton07205ec2014-01-13 02:31:00 +0000472
Damiend99b0522013-12-21 18:17:45 +0000473// functions
Damien Georged5e81822014-02-26 17:47:05 +0000474#define MP_OBJ_FUN_ARGS_MAX (0xffff) // to set maximum value in n_args_max below
Damien George66028ab2014-01-03 14:03:48 +0000475typedef struct _mp_obj_fun_native_t { // need this so we can define const objects (to go in ROM)
Damiend99b0522013-12-21 18:17:45 +0000476 mp_obj_base_t base;
Damien George51047752014-02-26 17:40:52 +0000477 bool is_kw : 1;
Damien Georged5e81822014-02-26 17:47:05 +0000478 uint n_args_min : 15; // inclusive
479 uint n_args_max : 16; // inclusive
Damien660365e2013-12-17 18:27:24 +0000480 void *fun;
Damien George66028ab2014-01-03 14:03:48 +0000481 // TODO add mp_map_t *globals
482 // for const function objects, make an empty, const map
483 // such functions won't be able to access the global scope, but that's probably okay
Damiend99b0522013-12-21 18:17:45 +0000484} mp_obj_fun_native_t;
Damien George97209d32014-01-07 15:58:30 +0000485
Damien Georgebee17b02014-03-27 11:07:04 +0000486void mp_obj_fun_bc_get(mp_obj_t self_in, int *n_args, const byte **code);
Paul Sokolovsky7fafb282014-03-30 20:21:28 +0300487bool mp_obj_fun_prepare_simple_args(mp_obj_t self_in, uint n_args, uint n_kw, const mp_obj_t *args,
488 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 +0000489
Paul Sokolovskydff3f892014-01-20 18:37:30 +0200490mp_obj_t mp_identity(mp_obj_t self);
Paul Sokolovsky557c9d52014-02-08 21:57:19 +0200491MP_DECLARE_CONST_FUN_OBJ(mp_identity_obj);
Paul Sokolovskydff3f892014-01-20 18:37:30 +0200492
Damien George28708622014-01-02 21:30:26 +0000493// module
Damien George0c36da02014-03-08 15:24:39 +0000494typedef struct _mp_obj_module_t {
495 mp_obj_base_t base;
496 qstr name;
Damien George8b0535e2014-04-05 21:53:54 +0100497 mp_obj_dict_t *globals;
Damien George0c36da02014-03-08 15:24:39 +0000498} mp_obj_module_t;
Damien George8b0535e2014-04-05 21:53:54 +0100499mp_obj_dict_t *mp_obj_module_get_globals(mp_obj_t self_in);
Damien Georgeeae16442014-01-11 19:22:29 +0000500
501// staticmethod and classmethod types; defined here so we can make const versions
Damien George64131f32014-02-06 20:31:44 +0000502// this structure is used for instances of both staticmethod and classmethod
503typedef struct _mp_obj_static_class_method_t {
Damien Georgeeae16442014-01-11 19:22:29 +0000504 mp_obj_base_t base;
505 mp_obj_t fun;
Damien George64131f32014-02-06 20:31:44 +0000506} mp_obj_static_class_method_t;
Paul Sokolovsky439542f2014-01-21 00:19:19 +0200507
508// sequence helpers
509void mp_seq_multiply(const void *items, uint item_sz, uint len, uint times, void *dest);
Paul Sokolovsky7364af22014-02-02 02:38:22 +0200510bool 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 +0200511#define m_seq_copy(dest, src, len, item_t) memcpy(dest, src, len * sizeof(item_t))
Paul Sokolovsky55ca0752014-03-30 17:35:53 +0300512#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 +0200513bool mp_seq_cmp_bytes(int op, const byte *data1, uint len1, const byte *data2, uint len2);
Paul Sokolovsky1a996c42014-02-08 22:49:46 +0200514bool 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 +0200515mp_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 +0200516mp_obj_t mp_seq_count_obj(const mp_obj_t *items, uint len, mp_obj_t value);