summaryrefslogtreecommitdiff
path: root/py/objint_longlong.c
diff options
context:
space:
mode:
Diffstat (limited to 'py/objint_longlong.c')
-rw-r--r--py/objint_longlong.c23
1 files changed, 8 insertions, 15 deletions
diff --git a/py/objint_longlong.c b/py/objint_longlong.c
index 00fe5636c..1940b8153 100644
--- a/py/objint_longlong.c
+++ b/py/objint_longlong.c
@@ -247,25 +247,21 @@ zero_division:
}
mp_obj_t mp_obj_new_int(mp_int_t value) {
- if (MP_SMALL_INT_FITS(value)) {
- return MP_OBJ_NEW_SMALL_INT(value);
- }
return mp_obj_new_int_from_ll(value);
}
mp_obj_t mp_obj_new_int_from_uint(mp_uint_t value) {
- // SMALL_INT accepts only signed numbers, so make sure the input
- // value fits completely in the small-int positive range.
- if ((value & ~MP_SMALL_INT_POSITIVE_MASK) == 0) {
- return MP_OBJ_NEW_SMALL_INT(value);
- }
return mp_obj_new_int_from_ll(value);
}
mp_obj_t mp_obj_new_int_from_ll(long long val) {
+ if ((long long)(mp_int_t)val == val && MP_SMALL_INT_FITS(val)) {
+ return MP_OBJ_NEW_SMALL_INT(val);
+ }
+
mp_obj_int_t *o = mp_obj_malloc(mp_obj_int_t, &mp_type_int);
o->val = val;
- return o;
+ return MP_OBJ_FROM_PTR(o);
}
mp_obj_t mp_obj_new_int_from_ull(unsigned long long val) {
@@ -273,19 +269,16 @@ mp_obj_t mp_obj_new_int_from_ull(unsigned long long val) {
if (val >> (sizeof(unsigned long long) * 8 - 1) != 0) {
mp_raise_msg(&mp_type_OverflowError, MP_ERROR_TEXT("ulonglong too large"));
}
- mp_obj_int_t *o = mp_obj_malloc(mp_obj_int_t, &mp_type_int);
- o->val = val;
- return o;
+ return mp_obj_new_int_from_ll(val);
}
mp_obj_t mp_obj_new_int_from_str_len(const char **str, size_t len, bool neg, unsigned int base) {
// TODO this does not honor the given length of the string, but it all cases it should anyway be null terminated
// TODO check overflow
- mp_obj_int_t *o = mp_obj_malloc(mp_obj_int_t, &mp_type_int);
char *endptr;
- o->val = strtoll(*str, &endptr, base);
+ mp_obj_t result = mp_obj_new_int_from_ll(strtoll(*str, &endptr, base));
*str = endptr;
- return o;
+ return result;
}
mp_int_t mp_obj_int_get_truncated(mp_const_obj_t self_in) {