obj.h 30.2 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
/*
 * 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.
 */
26
27
28
29
30
31
#ifndef __MICROPY_INCLUDED_PY_OBJ_H__
#define __MICROPY_INCLUDED_PY_OBJ_H__

#include "py/mpconfig.h"
#include "py/misc.h"
#include "py/qstr.h"
32

33
34
35
36
37
// A Micro Python object is a machine word having the following form:
//  - xxxx...xxx1 : a small int, bits 1 and above are the value
//  - xxxx...xx10 : a qstr, bits 2 and above are the value
//  - xxxx...xx00 : a pointer to an mp_obj_base_t (unless a fake object)

38
39
40
41
42
43
// All Micro Python objects are at least this type
// It must be of pointer size

typedef machine_ptr_t mp_obj_t;
typedef machine_const_ptr_t mp_const_obj_t;

44
// Anything that wants to be a Micro Python object must have
45
// mp_obj_base_t as its first member (except small ints and qstrs)
46

ian-v's avatar
ian-v committed
47
struct _mp_obj_type_t;
48
struct _mp_obj_base_t {
49
    const struct _mp_obj_type_t *type MICROPY_OBJ_BASE_ALIGNMENT;
50
};
ian-v's avatar
ian-v committed
51
typedef struct _mp_obj_base_t mp_obj_base_t;
52

53
54
55
// These fake objects are used to indicate certain things in arguments or return
// values, and should only be used when explicitly allowed.
//
56
//  - MP_OBJ_NULL : used to indicate the absence of an object, or unsupported operation.
57
58
59
60
61
62
63
//  - MP_OBJ_STOP_ITERATION : used instead of throwing a StopIteration, for efficiency.
//  - MP_OBJ_SENTINEL : used for various internal purposes where one needs
//    an object which is unique from all other objects, including MP_OBJ_NULL.
//
// For debugging purposes they are all different.  For non-debug mode, we alias
// as many as we can to MP_OBJ_NULL because it's cheaper to load/compare 0.

64
#ifdef NDEBUG
65
66
67
68
69
#define MP_OBJ_NULL             ((mp_obj_t)0)
#define MP_OBJ_STOP_ITERATION   ((mp_obj_t)0)
#define MP_OBJ_SENTINEL         ((mp_obj_t)4)
#else
#define MP_OBJ_NULL             ((mp_obj_t)0)
70
71
#define MP_OBJ_STOP_ITERATION   ((mp_obj_t)4)
#define MP_OBJ_SENTINEL         ((mp_obj_t)8)
72
#endif
73

74
// These macros check for small int, qstr or object, and access small int and qstr values
75

76
// these macros have now become inline functions; see below
77
78
79
//#define MP_OBJ_IS_SMALL_INT(o) ((((mp_int_t)(o)) & 1) != 0)
//#define MP_OBJ_IS_QSTR(o) ((((mp_int_t)(o)) & 3) == 2)
//#define MP_OBJ_IS_OBJ(o) ((((mp_int_t)(o)) & 3) == 0)
80
#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 int, str or fun; use below macros for that
81
82
#define MP_OBJ_IS_INT(o) (MP_OBJ_IS_SMALL_INT(o) || MP_OBJ_IS_TYPE(o, &mp_type_int))
#define MP_OBJ_IS_STR(o) (MP_OBJ_IS_QSTR(o) || MP_OBJ_IS_TYPE(o, &mp_type_str))
83
#define MP_OBJ_IS_STR_OR_BYTES(o) (MP_OBJ_IS_QSTR(o) || (MP_OBJ_IS_OBJ(o) && ((mp_obj_base_t*)(o))->type->binary_op == mp_obj_str_binary_op))
84
#define MP_OBJ_IS_FUN(o) (MP_OBJ_IS_OBJ(o) && (((mp_obj_base_t*)(o))->type->name == MP_QSTR_function))
85

86
#define MP_OBJ_SMALL_INT_VALUE(o) (((mp_int_t)(o)) >> 1)
87
#define MP_OBJ_NEW_SMALL_INT(small_int) ((mp_obj_t)((((mp_int_t)(small_int)) << 1) | 1))
88

89
#define MP_OBJ_QSTR_VALUE(o) (((mp_uint_t)(o)) >> 2)
90
#define MP_OBJ_NEW_QSTR(qst) ((mp_obj_t)((((mp_uint_t)(qst)) << 2) | 2))
91
92
93
94

// These macros are used to declare and define constant function objects
// You can put "static" in front of the definitions to make them local

95
#define MP_DECLARE_CONST_FUN_OBJ(obj_name) extern const mp_obj_fun_builtin_t obj_name
96

97
#define MP_DEFINE_CONST_FUN_OBJ_VOID_PTR(obj_name, is_kw, n_args_min, n_args_max, fun_name) const mp_obj_fun_builtin_t obj_name = {{&mp_type_fun_builtin}, is_kw, n_args_min, n_args_max, (void *)fun_name}
98
99
100
101
#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)
#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)
#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)
#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)
102
#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)
103
#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)
104
#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)
105

106
// These macros are used to define constant map/dict objects
107
108
// You can put "static" in front of the definition to make it local

109
110
111
#define MP_DEFINE_CONST_MAP(map_name, table_name) \
    const mp_map_t map_name = { \
        .all_keys_are_qstrs = 1, \
112
113
        .is_fixed = 1, \
        .is_ordered = 1, \
114
115
116
117
118
        .used = MP_ARRAY_SIZE(table_name), \
        .alloc = MP_ARRAY_SIZE(table_name), \
        .table = (mp_map_elem_t*)table_name, \
    }

119
120
#define MP_DEFINE_CONST_DICT(dict_name, table_name) \
    const mp_obj_dict_t dict_name = { \
121
        .base = {&mp_type_dict}, \
122
123
        .map = { \
            .all_keys_are_qstrs = 1, \
124
125
            .is_fixed = 1, \
            .is_ordered = 1, \
126
127
            .used = MP_ARRAY_SIZE(table_name), \
            .alloc = MP_ARRAY_SIZE(table_name), \
128
129
130
131
            .table = (mp_map_elem_t*)table_name, \
        }, \
    }

132
133
134
// These macros are used to declare and define constant staticmethond and classmethod objects
// You can put "static" in front of the definitions to make them local

135
136
#define MP_DECLARE_CONST_STATICMETHOD_OBJ(obj_name) extern const mp_obj_static_class_method_t obj_name
#define MP_DECLARE_CONST_CLASSMETHOD_OBJ(obj_name) extern const mp_obj_static_class_method_t obj_name
137

138
139
#define MP_DEFINE_CONST_STATICMETHOD_OBJ(obj_name, fun_name) const mp_obj_static_class_method_t obj_name = {{&mp_type_staticmethod}, fun_name}
#define MP_DEFINE_CONST_CLASSMETHOD_OBJ(obj_name, fun_name) const mp_obj_static_class_method_t obj_name = {{&mp_type_classmethod}, fun_name}
140

Damien George's avatar
Damien George committed
141
142
143
144
145
146
147
148
149
150
151
152
153
// Underlying map/hash table implementation (not dict object or map function)

typedef struct _mp_map_elem_t {
    mp_obj_t key;
    mp_obj_t value;
} mp_map_elem_t;

// TODO maybe have a truncated mp_map_t for fixed tables, since alloc=used
// put alloc last in the structure, so the truncated version does not need it
// this would save 1 ROM word for all ROM objects that have a locals_dict
// would also need a trucated dict structure

typedef struct _mp_map_t {
154
    mp_uint_t all_keys_are_qstrs : 1;
155
156
157
    mp_uint_t is_fixed : 1;     // a fixed array that can't be modified; must also be ordered
    mp_uint_t is_ordered : 1;   // an ordered array
    mp_uint_t used : (8 * sizeof(mp_uint_t) - 3);
158
    mp_uint_t alloc;
Damien George's avatar
Damien George committed
159
160
161
    mp_map_elem_t *table;
} mp_map_t;

162
// mp_set_lookup requires these constants to have the values they do
Damien George's avatar
Damien George committed
163
typedef enum _mp_map_lookup_kind_t {
164
165
166
167
    MP_MAP_LOOKUP = 0,
    MP_MAP_LOOKUP_ADD_IF_NOT_FOUND = 1,
    MP_MAP_LOOKUP_REMOVE_IF_FOUND = 2,
    MP_MAP_LOOKUP_ADD_IF_NOT_FOUND_OR_REMOVE_IF_FOUND = 3, // only valid for mp_set_lookup
Damien George's avatar
Damien George committed
168
169
} mp_map_lookup_kind_t;

Paul Sokolovsky's avatar
Paul Sokolovsky committed
170
171
extern const mp_map_t mp_const_empty_map;

172
static inline bool MP_MAP_SLOT_IS_FILLED(const mp_map_t *map, mp_uint_t pos) { return ((map)->table[pos].key != MP_OBJ_NULL && (map)->table[pos].key != MP_OBJ_SENTINEL); }
173

174
175
176
void mp_map_init(mp_map_t *map, mp_uint_t n);
void mp_map_init_fixed_table(mp_map_t *map, mp_uint_t n, const mp_obj_t *table);
mp_map_t *mp_map_new(mp_uint_t n);
Damien George's avatar
Damien George committed
177
178
179
180
void mp_map_deinit(mp_map_t *map);
void mp_map_free(mp_map_t *map);
mp_map_elem_t* mp_map_lookup(mp_map_t *map, mp_obj_t index, mp_map_lookup_kind_t lookup_kind);
void mp_map_clear(mp_map_t *map);
181
void mp_map_dump(mp_map_t *map);
Damien George's avatar
Damien George committed
182
183
184
185

// Underlying set implementation (not set object)

typedef struct _mp_set_t {
186
187
    mp_uint_t alloc;
    mp_uint_t used;
Damien George's avatar
Damien George committed
188
189
190
    mp_obj_t *table;
} mp_set_t;

191
static inline bool MP_SET_SLOT_IS_FILLED(const mp_set_t *set, mp_uint_t pos) { return ((set)->table[pos] != MP_OBJ_NULL && (set)->table[pos] != MP_OBJ_SENTINEL); }
192

193
void mp_set_init(mp_set_t *set, mp_uint_t n);
Damien George's avatar
Damien George committed
194
mp_obj_t mp_set_lookup(mp_set_t *set, mp_obj_t index, mp_map_lookup_kind_t lookup_kind);
195
mp_obj_t mp_set_remove_first(mp_set_t *set);
Damien George's avatar
Damien George committed
196
void mp_set_clear(mp_set_t *set);
197
198
199
200
201
202

// Type definitions for methods

typedef mp_obj_t (*mp_fun_0_t)(void);
typedef mp_obj_t (*mp_fun_1_t)(mp_obj_t);
typedef mp_obj_t (*mp_fun_2_t)(mp_obj_t, mp_obj_t);
203
typedef mp_obj_t (*mp_fun_3_t)(mp_obj_t, mp_obj_t, mp_obj_t);
204
205
typedef mp_obj_t (*mp_fun_var_t)(mp_uint_t n, const mp_obj_t *);
typedef mp_obj_t (*mp_fun_kw_t)(mp_uint_t n, const mp_obj_t *, mp_map_t *);
206

207
typedef enum {
208
209
210
    PRINT_STR = 0,
    PRINT_REPR = 1,
    PRINT_EXC = 2, // Special format for printing exception in unhandled exception message
211
212
    PRINT_JSON = 3,
    PRINT_EXC_SUBCLASS = 0x80, // Internal flag for printing exception subclasses
213
214
215
} mp_print_kind_t;

typedef void (*mp_print_fun_t)(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t o, mp_print_kind_t kind);
216
217
218
219
typedef mp_obj_t (*mp_make_new_fun_t)(mp_obj_t type_in, mp_uint_t n_args, mp_uint_t n_kw, const mp_obj_t *args);
typedef mp_obj_t (*mp_call_fun_t)(mp_obj_t fun, mp_uint_t n_args, mp_uint_t n_kw, const mp_obj_t *args);
typedef mp_obj_t (*mp_unary_op_fun_t)(mp_uint_t op, mp_obj_t);
typedef mp_obj_t (*mp_binary_op_fun_t)(mp_uint_t op, mp_obj_t, mp_obj_t);
220
typedef 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
221
typedef 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
222
typedef mp_obj_t (*mp_subscr_fun_t)(mp_obj_t self_in, mp_obj_t index, mp_obj_t value);
223
224

typedef struct _mp_method_t {
225
    qstr name;
226
227
228
    mp_const_obj_t fun;
} mp_method_t;

229
// Buffer protocol
230
typedef struct _mp_buffer_info_t {
231
232
233
234
235
    // if we'd bother to support various versions of structure
    // (with different number of fields), we can distinguish
    // them with ver = sizeof(struct). Cons: overkill for *micro*?
    //int ver; // ?

236
237
238
    void *buf;      // can be NULL if len == 0
    mp_uint_t len;  // in bytes
    int typecode;   // as per binary.h
239
240
241
242

    // Rationale: to load arbitrary-sized sprites directly to LCD
    // Cons: a bit adhoc usecase
    // int stride;
243
244
245
246
} mp_buffer_info_t;
#define MP_BUFFER_READ  (1)
#define MP_BUFFER_WRITE (2)
#define MP_BUFFER_RW (MP_BUFFER_READ | MP_BUFFER_WRITE)
247
typedef struct _mp_buffer_p_t {
248
    mp_int_t (*get_buffer)(mp_obj_t obj, mp_buffer_info_t *bufinfo, mp_uint_t flags);
249
} mp_buffer_p_t;
250
251
bool mp_get_buffer(mp_obj_t obj, mp_buffer_info_t *bufinfo, mp_uint_t flags);
void mp_get_buffer_raise(mp_obj_t obj, mp_buffer_info_t *bufinfo, mp_uint_t flags);
252
253

// Stream protocol
254
#define MP_STREAM_ERROR ((mp_uint_t)-1)
255
typedef struct _mp_stream_p_t {
256
257
258
259
    // On error, functions should return MP_STREAM_ERROR and fill in *errcode (values
    // are implementation-dependent, but will be exposed to user, e.g. via exception).
    mp_uint_t (*read)(mp_obj_t obj, void *buf, mp_uint_t size, int *errcode);
    mp_uint_t (*write)(mp_obj_t obj, const void *buf, mp_uint_t size, int *errcode);
260
    mp_uint_t (*ioctl)(mp_obj_t obj, mp_uint_t request, mp_uint_t arg, int *errcode);
261
    mp_uint_t is_text : 1; // default is bytes, set this for text stream
262
263
} mp_stream_p_t;

264
265
266
267
268
269
270
271
272
273
274
// Stream ioctl request codes
#define MP_STREAM_FLUSH (1)
#define MP_STREAM_SEEK  (2)
#define MP_STREAM_POLL  (3)

// Argument structure for MP_STREAM_SEEK
struct mp_stream_seek_t {
    mp_off_t offset;
    int whence;
};

275
276
struct _mp_obj_type_t {
    mp_obj_base_t base;
277
    qstr name;
278
    mp_print_fun_t print;
279
    mp_make_new_fun_t make_new;     // to make an instance of the type
280

281
    mp_call_fun_t call;
282
283
    mp_unary_op_fun_t unary_op;     // can return MP_OBJ_NULL if op not supported
    mp_binary_op_fun_t binary_op;   // can return MP_OBJ_NULL if op not supported
284

285
    mp_load_attr_fun_t load_attr;
286
    mp_store_attr_fun_t store_attr; // if value is MP_OBJ_NULL, then delete that attribute
287

288
289
    mp_subscr_fun_t subscr;         // implements load, store, delete subscripting
                                    // value=MP_OBJ_NULL means delete, value=MP_OBJ_SENTINEL means load, else store
290
                                    // can return MP_OBJ_NULL if op not supported
291

292
    mp_fun_1_t getiter;             // corresponds to __iter__ special method
293
    mp_fun_1_t iternext; // may return MP_OBJ_STOP_ITERATION as an optimisation instead of raising StopIteration() (with no args)
294

295
    mp_buffer_p_t buffer_p;
296
    const mp_stream_p_t *stream_p;
297

298
299
300
    // these are for dynamically created types (classes)
    mp_obj_t bases_tuple;
    mp_obj_t locals_dict;
301

302
303
304
305
306
307
308
309
310
311
312
313
    /*
    What we might need to add here:

    len             str tuple list map
    abs             float complex
    hash            bool int none str
    equal           int str

    unpack seq      list tuple
    */
};

ian-v's avatar
ian-v committed
314
typedef struct _mp_obj_type_t mp_obj_type_t;
315

316
317
// Constant types, globally accessible
extern const mp_obj_type_t mp_type_type;
318
319
320
321
322
323
extern const mp_obj_type_t mp_type_object;
extern const mp_obj_type_t mp_type_NoneType;
extern const mp_obj_type_t mp_type_bool;
extern const mp_obj_type_t mp_type_int;
extern const mp_obj_type_t mp_type_str;
extern const mp_obj_type_t mp_type_bytes;
324
extern const mp_obj_type_t mp_type_bytearray;
325
extern const mp_obj_type_t mp_type_memoryview;
326
327
328
329
330
331
332
333
extern const mp_obj_type_t mp_type_float;
extern const mp_obj_type_t mp_type_complex;
extern const mp_obj_type_t mp_type_tuple;
extern const mp_obj_type_t mp_type_list;
extern const mp_obj_type_t mp_type_map; // map (the python builtin, not the dict implementation detail)
extern const mp_obj_type_t mp_type_enumerate;
extern const mp_obj_type_t mp_type_filter;
extern const mp_obj_type_t mp_type_dict;
334
extern const mp_obj_type_t mp_type_ordereddict;
335
extern const mp_obj_type_t mp_type_range;
336
extern const mp_obj_type_t mp_type_set;
337
extern const mp_obj_type_t mp_type_frozenset;
338
339
340
341
342
extern const mp_obj_type_t mp_type_slice;
extern const mp_obj_type_t mp_type_zip;
extern const mp_obj_type_t mp_type_array;
extern const mp_obj_type_t mp_type_super;
extern const mp_obj_type_t mp_type_gen_instance;
343
extern const mp_obj_type_t mp_type_fun_builtin;
344
345
346
347
extern const mp_obj_type_t mp_type_fun_bc;
extern const mp_obj_type_t mp_type_module;
extern const mp_obj_type_t mp_type_staticmethod;
extern const mp_obj_type_t mp_type_classmethod;
348
extern const mp_obj_type_t mp_type_property;
349
extern const mp_obj_type_t mp_type_stringio;
350
extern const mp_obj_type_t mp_type_bytesio;
351
extern const mp_obj_type_t mp_type_reversed;
352
353

// Exceptions
354
extern const mp_obj_type_t mp_type_BaseException;
355
356
357
358
359
360
361
362
363
extern const mp_obj_type_t mp_type_ArithmeticError;
extern const mp_obj_type_t mp_type_AssertionError;
extern const mp_obj_type_t mp_type_AttributeError;
extern const mp_obj_type_t mp_type_EOFError;
extern const mp_obj_type_t mp_type_Exception;
extern const mp_obj_type_t mp_type_GeneratorExit;
extern const mp_obj_type_t mp_type_ImportError;
extern const mp_obj_type_t mp_type_IndentationError;
extern const mp_obj_type_t mp_type_IndexError;
364
extern const mp_obj_type_t mp_type_KeyboardInterrupt;
365
366
367
368
369
370
371
372
extern const mp_obj_type_t mp_type_KeyError;
extern const mp_obj_type_t mp_type_LookupError;
extern const mp_obj_type_t mp_type_MemoryError;
extern const mp_obj_type_t mp_type_NameError;
extern const mp_obj_type_t mp_type_NotImplementedError;
extern const mp_obj_type_t mp_type_OSError;
extern const mp_obj_type_t mp_type_OverflowError;
extern const mp_obj_type_t mp_type_RuntimeError;
373
extern const mp_obj_type_t mp_type_StopIteration;
374
extern const mp_obj_type_t mp_type_SyntaxError;
375
extern const mp_obj_type_t mp_type_SystemExit;
376
extern const mp_obj_type_t mp_type_TypeError;
Paul Sokolovsky's avatar
Paul Sokolovsky committed
377
extern const mp_obj_type_t mp_type_UnicodeError;
378
379
380
extern const mp_obj_type_t mp_type_ValueError;
extern const mp_obj_type_t mp_type_ZeroDivisionError;

381
// Constant objects, globally accessible
382
383
384
385
// The macros are for convenience only
#define mp_const_none ((mp_obj_t)&mp_const_none_obj)
#define mp_const_false ((mp_obj_t)&mp_const_false_obj)
#define mp_const_true ((mp_obj_t)&mp_const_true_obj)
386
#define mp_const_empty_bytes ((mp_obj_t)&mp_const_empty_bytes_obj)
387
388
389
390
#define mp_const_empty_tuple ((mp_obj_t)&mp_const_empty_tuple_obj)
extern const struct _mp_obj_none_t mp_const_none_obj;
extern const struct _mp_obj_bool_t mp_const_false_obj;
extern const struct _mp_obj_bool_t mp_const_true_obj;
391
extern const struct _mp_obj_str_t mp_const_empty_bytes_obj;
392
393
extern const struct _mp_obj_tuple_t mp_const_empty_tuple_obj;
extern const struct _mp_obj_ellipsis_t mp_const_ellipsis_obj;
394
extern const struct _mp_obj_exception_t mp_const_MemoryError_obj;
395
extern const struct _mp_obj_exception_t mp_const_GeneratorExit_obj;
396
397
398

// General API for objects

399
mp_obj_t mp_obj_new_type(qstr name, mp_obj_t bases_tuple, mp_obj_t locals_dict);
400
401
mp_obj_t mp_obj_new_none(void);
mp_obj_t mp_obj_new_bool(bool value);
Damien George's avatar
Damien George committed
402
mp_obj_t mp_obj_new_cell(mp_obj_t obj);
403
404
mp_obj_t mp_obj_new_int(mp_int_t value);
mp_obj_t mp_obj_new_int_from_uint(mp_uint_t value);
405
mp_obj_t mp_obj_new_int_from_str_len(const char **str, mp_uint_t len, bool neg, mp_uint_t base);
406
mp_obj_t mp_obj_new_int_from_ll(long long val); // this must return a multi-precision integer object (or raise an overflow exception)
407
mp_obj_t mp_obj_new_int_from_ull(unsigned long long val); // this must return a multi-precision integer object (or raise an overflow exception)
408
mp_obj_t mp_obj_new_str(const char* data, mp_uint_t len, bool make_qstr_if_not_already);
409
mp_obj_t mp_obj_new_str_from_vstr(const mp_obj_type_t *type, vstr_t *vstr);
410
mp_obj_t mp_obj_new_bytes(const byte* data, mp_uint_t len);
411
412
mp_obj_t mp_obj_new_bytearray(mp_uint_t n, void *items);
mp_obj_t mp_obj_new_bytearray_by_ref(mp_uint_t n, void *items);
413
#if MICROPY_PY_BUILTINS_FLOAT
414
mp_obj_t mp_obj_new_int_from_float(mp_float_t val);
415
416
mp_obj_t mp_obj_new_float(mp_float_t val);
mp_obj_t mp_obj_new_complex(mp_float_t real, mp_float_t imag);
417
#endif
418
mp_obj_t mp_obj_new_exception(const mp_obj_type_t *exc_type);
419
mp_obj_t mp_obj_new_exception_arg1(const mp_obj_type_t *exc_type, mp_obj_t arg);
420
mp_obj_t mp_obj_new_exception_args(const mp_obj_type_t *exc_type, mp_uint_t n_args, const mp_obj_t *args);
421
422
mp_obj_t mp_obj_new_exception_msg(const mp_obj_type_t *exc_type, const char *msg);
mp_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!)
423
mp_obj_t mp_obj_new_fun_bc(mp_uint_t scope_flags, mp_uint_t n_pos_args, mp_uint_t n_kwonly_args, mp_obj_t def_args, mp_obj_t def_kw_args, const byte *code);
424
425
426
mp_obj_t mp_obj_new_fun_native(mp_uint_t n_args, void *fun_data);
mp_obj_t mp_obj_new_fun_viper(mp_uint_t n_args, void *fun_data, mp_uint_t type_sig);
mp_obj_t mp_obj_new_fun_asm(mp_uint_t n_args, void *fun_data);
427
mp_obj_t mp_obj_new_gen_wrap(mp_obj_t fun);
428
mp_obj_t mp_obj_new_closure(mp_obj_t fun, mp_uint_t n_closed, const mp_obj_t *closed);
429
430
mp_obj_t mp_obj_new_tuple(mp_uint_t n, const mp_obj_t *items);
mp_obj_t mp_obj_new_list(mp_uint_t n, mp_obj_t *items);
431
432
mp_obj_t mp_obj_new_dict(mp_uint_t n_args);
mp_obj_t mp_obj_new_set(mp_uint_t n_args, mp_obj_t *items);
433
mp_obj_t mp_obj_new_slice(mp_obj_t start, mp_obj_t stop, mp_obj_t step);
Damien George's avatar
Damien George committed
434
mp_obj_t mp_obj_new_super(mp_obj_t type, mp_obj_t obj);
435
mp_obj_t mp_obj_new_bound_meth(mp_obj_t meth, mp_obj_t self);
436
mp_obj_t mp_obj_new_getitem_iter(mp_obj_t *args);
437
mp_obj_t mp_obj_new_module(qstr module_name);
438
mp_obj_t mp_obj_new_memoryview(byte typecode, mp_uint_t nitems, void *items);
439

440
441
mp_obj_type_t *mp_obj_get_type(mp_const_obj_t o_in);
const char *mp_obj_get_type_str(mp_const_obj_t o_in);
442
bool mp_obj_is_subclass_fast(mp_const_obj_t object, mp_const_obj_t classinfo); // arguments should be type objects
443
mp_obj_t mp_instance_cast_to_native_base(mp_const_obj_t self_in, mp_const_obj_t native_type);
444

445
446
void mp_obj_print_helper(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t o_in, mp_print_kind_t kind);
void mp_obj_print(mp_obj_t o, mp_print_kind_t kind);
447
void mp_obj_print_exception(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t exc);
448

449
bool mp_obj_is_true(mp_obj_t arg);
450
451

// TODO make these all lower case when they have proven themselves
452
453
static inline bool MP_OBJ_IS_OBJ(mp_const_obj_t o) { return ((((mp_int_t)(o)) & 3) == 0); }
static inline bool MP_OBJ_IS_SMALL_INT(mp_const_obj_t o) { return ((((mp_int_t)(o)) & 1) != 0); }
454
455
456
//static inline 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
//static inline 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
static inline 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
457
static inline bool MP_OBJ_IS_QSTR(mp_const_obj_t o) { return ((((mp_int_t)(o)) & 3) == 2); }
458
//static inline bool MP_OBJ_IS_STR(mp_const_obj_t o) { return (MP_OBJ_IS_QSTR(o) || MP_OBJ_IS_TYPE(o, &mp_type_str)); }
459

460
bool mp_obj_is_callable(mp_obj_t o_in);
461
mp_int_t mp_obj_hash(mp_obj_t o_in);
462
bool mp_obj_equal(mp_obj_t o1, mp_obj_t o2);
463

464
465
mp_int_t mp_obj_get_int(mp_const_obj_t arg);
bool mp_obj_get_int_maybe(mp_const_obj_t arg, mp_int_t *value);
466
#if MICROPY_PY_BUILTINS_FLOAT
467
468
mp_float_t mp_obj_get_float(mp_obj_t self_in);
void mp_obj_get_complex(mp_obj_t self_in, mp_float_t *real, mp_float_t *imag);
469
#endif
470
//qstr mp_obj_get_qstr(mp_obj_t arg);
471
472
473
void mp_obj_get_array(mp_obj_t o, mp_uint_t *len, mp_obj_t **items);
void mp_obj_get_array_fixed_n(mp_obj_t o, mp_uint_t len, mp_obj_t **items);
mp_uint_t mp_get_index(const mp_obj_type_t *type, mp_uint_t len, mp_obj_t index, bool is_slice);
474
mp_obj_t mp_obj_id(mp_obj_t o_in);
475
mp_obj_t mp_obj_len(mp_obj_t o_in);
476
mp_obj_t mp_obj_len_maybe(mp_obj_t o_in); /* may return MP_OBJ_NULL */
477
mp_obj_t mp_obj_subscr(mp_obj_t base, mp_obj_t index, mp_obj_t val);
478
479

// bool
480
// TODO make lower case when it has proven itself
481
static inline mp_obj_t MP_BOOL(mp_int_t x) { return x ? mp_const_true : mp_const_false; }
482
483
484
485
486

// cell
mp_obj_t mp_obj_cell_get(mp_obj_t self_in);
void mp_obj_cell_set(mp_obj_t self_in, mp_obj_t obj);

487
// int
488
// For long int, returns value truncated to mp_int_t
489
490
491
mp_int_t mp_obj_int_get_truncated(mp_const_obj_t self_in);
// Will raise exception if value doesn't fit into mp_int_t
mp_int_t mp_obj_int_get_checked(mp_const_obj_t self_in);
492
#if MICROPY_PY_BUILTINS_FLOAT
493
494
mp_float_t mp_obj_int_as_float(mp_obj_t self_in);
#endif
495

496
// exception
497
#define mp_obj_is_native_exception_instance(o) (mp_obj_get_type(o)->make_new == mp_obj_exception_make_new)
498
499
bool mp_obj_is_exception_type(mp_obj_t self_in);
bool mp_obj_is_exception_instance(mp_obj_t self_in);
500
bool mp_obj_exception_match(mp_obj_t exc, mp_const_obj_t exc_type);
501
void mp_obj_exception_clear_traceback(mp_obj_t self_in);
502
503
void mp_obj_exception_add_traceback(mp_obj_t self_in, qstr file, mp_uint_t line, qstr block);
void mp_obj_exception_get_traceback(mp_obj_t self_in, mp_uint_t *n, mp_uint_t **values);
504
mp_obj_t mp_obj_exception_get_value(mp_obj_t self_in);
505
mp_obj_t mp_obj_exception_make_new(mp_obj_t type_in, mp_uint_t n_args, mp_uint_t n_kw, const mp_obj_t *args);
506
507
mp_obj_t mp_alloc_emergency_exception_buf(mp_obj_t size_in);
void mp_init_emergency_exception_buf(void);
508

509
// str
510
bool mp_obj_str_equal(mp_obj_t s1, mp_obj_t s2);
511
512
mp_uint_t mp_obj_str_get_hash(mp_obj_t self_in);
mp_uint_t mp_obj_str_get_len(mp_obj_t self_in);
513
qstr mp_obj_str_get_qstr(mp_obj_t self_in); // use this if you will anyway convert the string to a qstr
514
const char *mp_obj_str_get_str(mp_obj_t self_in); // use this only if you need the string to be null terminated
515
const char *mp_obj_str_get_data(mp_obj_t self_in, mp_uint_t *len);
516
mp_obj_t mp_obj_str_intern(mp_obj_t str);
517
void mp_str_print_quoted(void (*print)(void *env, const char *fmt, ...), void *env, const byte *str_data, mp_uint_t str_len, bool is_bytes);
518

519
#if MICROPY_PY_BUILTINS_FLOAT
520
// float
521
522
523
524
typedef struct _mp_obj_float_t {
    mp_obj_base_t base;
    mp_float_t value;
} mp_obj_float_t;
525
mp_float_t mp_obj_float_get(mp_obj_t self_in);
526
mp_obj_t mp_obj_float_binary_op(mp_uint_t op, mp_float_t lhs_val, mp_obj_t rhs); // can return MP_OBJ_NULL if op not supported
527
void mp_obj_float_divmod(mp_float_t *x, mp_float_t *y);
528
529
530

// complex
void mp_obj_complex_get(mp_obj_t self_in, mp_float_t *real, mp_float_t *imag);
531
mp_obj_t mp_obj_complex_binary_op(mp_uint_t op, mp_float_t lhs_real, mp_float_t lhs_imag, mp_obj_t rhs_in); // can return MP_OBJ_NULL if op not supported
532
533
534
#endif

// tuple
535
void mp_obj_tuple_get(mp_obj_t self_in, mp_uint_t *len, mp_obj_t **items);
John R. Lenton's avatar
John R. Lenton committed
536
void mp_obj_tuple_del(mp_obj_t self_in);
537
mp_int_t mp_obj_tuple_hash(mp_obj_t self_in);
538
539

// list
540
struct _mp_obj_list_t;
541
void mp_obj_list_init(struct _mp_obj_list_t *o, mp_uint_t n);
542
mp_obj_t mp_obj_list_append(mp_obj_t self_in, mp_obj_t arg);
543
mp_obj_t mp_obj_list_remove(mp_obj_t self_in, mp_obj_t value);
544
545
void mp_obj_list_get(mp_obj_t self_in, mp_uint_t *len, mp_obj_t **items);
void mp_obj_list_set_len(mp_obj_t self_in, mp_uint_t len);
546
void mp_obj_list_store(mp_obj_t self_in, mp_obj_t index, mp_obj_t value);
547
mp_obj_t mp_obj_list_sort(mp_uint_t n_args, const mp_obj_t *args, mp_map_t *kwargs);
548
549

// dict
Damien George's avatar
Damien George committed
550
551
552
553
typedef struct _mp_obj_dict_t {
    mp_obj_base_t base;
    mp_map_t map;
} mp_obj_dict_t;
554
555
void mp_obj_dict_init(mp_obj_dict_t *dict, mp_uint_t n_args);
mp_uint_t mp_obj_dict_len(mp_obj_t self_in);
556
mp_obj_t mp_obj_dict_get(mp_obj_t self_in, mp_obj_t index);
557
mp_obj_t mp_obj_dict_store(mp_obj_t self_in, mp_obj_t key, mp_obj_t value);
558
mp_obj_t mp_obj_dict_delete(mp_obj_t self_in, mp_obj_t key);
Damien George's avatar
Damien George committed
559
mp_map_t *mp_obj_dict_get_map(mp_obj_t self_in);
560
561
562
563

// set
void mp_obj_set_store(mp_obj_t self_in, mp_obj_t item);

564
// slice
565
void mp_obj_slice_get(mp_obj_t self_in, mp_obj_t *start, mp_obj_t *stop, mp_obj_t *step);
566

567
// functions
568
#define MP_OBJ_FUN_ARGS_MAX (0xffff) // to set maximum value in n_args_max below
569
typedef struct _mp_obj_fun_builtin_t { // use this to make const objects that go in ROM
570
    mp_obj_base_t base;
571
    bool is_kw : 1;
572
573
574
575
    mp_uint_t n_args_min : 15; // inclusive
    mp_uint_t n_args_max : 16; // inclusive
    void *fun; // must be a pointer to a callable function in ROM
} mp_obj_fun_builtin_t;
576

577
578
qstr mp_obj_fun_get_name(mp_const_obj_t fun);
qstr mp_obj_code_get_name(const byte *code_info);
579

580
mp_obj_t mp_identity(mp_obj_t self);
581
MP_DECLARE_CONST_FUN_OBJ(mp_identity_obj);
582

583
// module
584
585
586
typedef struct _mp_obj_module_t {
    mp_obj_base_t base;
    qstr name;
587
    mp_obj_dict_t *globals;
588
} mp_obj_module_t;
589
mp_obj_dict_t *mp_obj_module_get_globals(mp_obj_t self_in);
590
591
// check if given module object is a package
bool mp_obj_is_package(mp_obj_t module);
592
593

// staticmethod and classmethod types; defined here so we can make const versions
594
595
// this structure is used for instances of both staticmethod and classmethod
typedef struct _mp_obj_static_class_method_t {
596
597
    mp_obj_base_t base;
    mp_obj_t fun;
598
} mp_obj_static_class_method_t;
599

600
601
602
// property
const mp_obj_t *mp_obj_property_get(mp_obj_t self_in);

603
// sequence helpers
604
605
606

// slice indexes resolved to particular sequence
typedef struct {
607
608
609
    mp_uint_t start;
    mp_uint_t stop;
    mp_int_t step;
610
611
} mp_bound_slice_t;

612
void mp_seq_multiply(const void *items, mp_uint_t item_sz, mp_uint_t len, mp_uint_t times, void *dest);
613
#if MICROPY_PY_BUILTINS_SLICE
614
bool mp_seq_get_fast_slice_indexes(mp_uint_t len, mp_obj_t slice, mp_bound_slice_t *indexes);
615
#endif
616
617
#define mp_seq_copy(dest, src, len, item_t) memcpy(dest, src, len * sizeof(item_t))
#define mp_seq_cat(dest, src1, len1, src2, len2, item_t) { memcpy(dest, src1, (len1) * sizeof(item_t)); memcpy(dest + (len1), src2, (len2) * sizeof(item_t)); }
618
619
bool mp_seq_cmp_bytes(mp_uint_t op, const byte *data1, mp_uint_t len1, const byte *data2, mp_uint_t len2);
bool mp_seq_cmp_objs(mp_uint_t op, const mp_obj_t *items1, mp_uint_t len1, const mp_obj_t *items2, mp_uint_t len2);
620
621
622
mp_obj_t mp_seq_index_obj(const mp_obj_t *items, mp_uint_t len, mp_uint_t n_args, const mp_obj_t *args);
mp_obj_t mp_seq_count_obj(const mp_obj_t *items, mp_uint_t len, mp_obj_t value);
mp_obj_t mp_seq_extract_slice(mp_uint_t len, const mp_obj_t *seq, mp_bound_slice_t *indexes);
623
624
// Helper to clear stale pointers from allocated, but unused memory, to preclude GC problems
#define mp_seq_clear(start, len, alloc_len, item_sz) memset((byte*)(start) + (len) * (item_sz), 0, ((alloc_len) - (len)) * (item_sz))
625
626
627
628
629
630
631
632
633
634
#define mp_seq_replace_slice_no_grow(dest, dest_len, beg, end, slice, slice_len, item_sz) \
    /*printf("memcpy(%p, %p, %d)\n", dest + beg, slice, slice_len * (item_sz));*/ \
    memcpy(((char*)dest) + (beg) * (item_sz), slice, slice_len * (item_sz)); \
    /*printf("memmove(%p, %p, %d)\n", dest + (beg + slice_len), dest + end, (dest_len - end) * (item_sz));*/ \
    memmove(((char*)dest) + (beg + slice_len) * (item_sz), ((char*)dest) + (end) * (item_sz), (dest_len - end) * (item_sz));

#define mp_seq_replace_slice_grow_inplace(dest, dest_len, beg, end, slice, slice_len, len_adj, item_sz) \
    /*printf("memmove(%p, %p, %d)\n", dest + beg + len_adj, dest + beg, (dest_len - beg) * (item_sz));*/ \
    memmove(((char*)dest) + (beg + len_adj) * (item_sz), ((char*)dest) + (beg) * (item_sz), (dest_len - beg) * (item_sz)); \
    memcpy(((char*)dest) + (beg) * (item_sz), slice, slice_len * (item_sz));
635
636

#endif // __MICROPY_INCLUDED_PY_OBJ_H__