int: Add value accessors: mp_obj_int_get() & mp_obj_int_get_checked().
mp_obj_int_get() can be used when just full resolution of C machine_int_t is required (returns truncated value of long int). mp_obj_int_get_checked() will throw exception if Python int value not representable in machine_int_t.
This commit is contained in:
parent
166bb40fb2
commit
d26b379eec
2
py/obj.c
2
py/obj.c
@ -155,6 +155,8 @@ machine_int_t mp_obj_get_int(mp_obj_t arg) {
|
|||||||
return 1;
|
return 1;
|
||||||
} else if (MP_OBJ_IS_SMALL_INT(arg)) {
|
} else if (MP_OBJ_IS_SMALL_INT(arg)) {
|
||||||
return MP_OBJ_SMALL_INT_VALUE(arg);
|
return MP_OBJ_SMALL_INT_VALUE(arg);
|
||||||
|
} else if (MP_OBJ_IS_TYPE(arg, &int_type)) {
|
||||||
|
return mp_obj_int_get_checked(arg);
|
||||||
#if MICROPY_ENABLE_FLOAT
|
#if MICROPY_ENABLE_FLOAT
|
||||||
} else if (MP_OBJ_IS_TYPE(arg, &float_type)) {
|
} else if (MP_OBJ_IS_TYPE(arg, &float_type)) {
|
||||||
// TODO work out if this should be floor, ceil or trunc
|
// TODO work out if this should be floor, ceil or trunc
|
||||||
|
4
py/obj.h
4
py/obj.h
@ -265,6 +265,10 @@ void mp_obj_cell_set(mp_obj_t self_in, mp_obj_t obj);
|
|||||||
|
|
||||||
// int
|
// int
|
||||||
extern const mp_obj_type_t int_type;
|
extern const mp_obj_type_t int_type;
|
||||||
|
// For long int, returns value truncated to machine_int_t
|
||||||
|
machine_int_t mp_obj_int_get(mp_obj_t self_in);
|
||||||
|
// Will rains exception if value doesn't fit into machine_int_t
|
||||||
|
machine_int_t mp_obj_int_get_checked(mp_obj_t self_in);
|
||||||
|
|
||||||
// exception
|
// exception
|
||||||
extern const mp_obj_type_t exception_type;
|
extern const mp_obj_type_t exception_type;
|
||||||
|
@ -77,4 +77,13 @@ mp_obj_t mp_obj_new_int(machine_int_t value) {
|
|||||||
nlr_jump(mp_obj_new_exception_msg(MP_QSTR_OverflowError, "small int overflow"));
|
nlr_jump(mp_obj_new_exception_msg(MP_QSTR_OverflowError, "small int overflow"));
|
||||||
return mp_const_none;
|
return mp_const_none;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
machine_int_t mp_obj_int_get(mp_obj_t self_in) {
|
||||||
|
return MP_OBJ_SMALL_INT_VALUE(self_in);
|
||||||
|
}
|
||||||
|
|
||||||
|
machine_int_t mp_obj_int_get_checked(mp_obj_t self_in) {
|
||||||
|
return MP_OBJ_SMALL_INT_VALUE(self_in);
|
||||||
|
}
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
@ -120,9 +120,17 @@ mp_obj_t mp_obj_new_int_from_long_str(const char *s) {
|
|||||||
return o;
|
return o;
|
||||||
}
|
}
|
||||||
|
|
||||||
machine_int_t mp_obj_int_get_int(mp_obj_t self_in) {
|
machine_int_t mp_obj_int_get(mp_obj_t self_in) {
|
||||||
|
if (MP_OBJ_IS_SMALL_INT(self_in)) {
|
||||||
|
return MP_OBJ_SMALL_INT_VALUE(self_in);
|
||||||
|
}
|
||||||
mp_obj_int_t *self = self_in;
|
mp_obj_int_t *self = self_in;
|
||||||
return self->val;
|
return self->val;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
machine_int_t mp_obj_int_get_checked(mp_obj_t self_in) {
|
||||||
|
// TODO: Check overflow
|
||||||
|
return mp_obj_int_get(self_in);
|
||||||
|
}
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
Loading…
Reference in New Issue
Block a user