diff --git a/py/objint.c b/py/objint.c index 370c6a5df346fd0cfe00019352e11103333f091b..fd0b2be559eb6d568bdf61ea0b32ec4b12801819 100644 --- a/py/objint.c +++ b/py/objint.c @@ -306,9 +306,19 @@ mp_obj_t mp_obj_new_int_from_uint(mp_uint_t value) { #if MICROPY_PY_BUILTINS_FLOAT mp_obj_t mp_obj_new_int_from_float(mp_float_t val) { - // TODO raise an exception if the int won't fit - mp_int_t i = MICROPY_FLOAT_C_FUN(trunc)(val); - return mp_obj_new_int(i); + int cl = fpclassify(val); + if (cl == FP_INFINITE) { + nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_OverflowError, "can't convert inf to int")); + } else if (cl == FP_NAN) { + nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "can't convert NaN to int")); + } else { + mp_fp_as_int_class_t icl = mp_classify_fp_as_int(val); + if (icl == MP_FP_CLASS_FIT_SMALLINT) { + return MP_OBJ_NEW_SMALL_INT((mp_int_t)val); + } else { + nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "float too big")); + } + } } #endif diff --git a/py/objint_longlong.c b/py/objint_longlong.c index 394ae111adaa226e1b5f25a891680a8fa3f2c0a0..83e1c67d6dd965eeee19644df6e4172d4ac484c0 100644 --- a/py/objint_longlong.c +++ b/py/objint_longlong.c @@ -187,9 +187,21 @@ mp_obj_t mp_obj_new_int_from_ull(unsigned long long val) { #if MICROPY_PY_BUILTINS_FLOAT mp_obj_t mp_obj_new_int_from_float(mp_float_t val) { - // TODO raise an exception if the unsigned long long won't fit - long long i = MICROPY_FLOAT_C_FUN(trunc)(val); - return mp_obj_new_int_from_ll(i); + int cl = fpclassify(val); + if (cl == FP_INFINITE) { + nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_OverflowError, "can't convert inf to int")); + } else if (cl == FP_NAN) { + nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "can't convert NaN to int")); + } else { + mp_fp_as_int_class_t icl = mp_classify_fp_as_int(val); + if (icl == MP_FP_CLASS_FIT_SMALLINT) { + return MP_OBJ_NEW_SMALL_INT((mp_int_t)val); + } else if (icl == MP_FP_CLASS_FIT_LONGINT) { + return mp_obj_new_int_from_ll((long long)val); + } else { + nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "float too big")); + } + } } #endif diff --git a/py/objint_mpz.c b/py/objint_mpz.c index 02b3ec0fe4be150f02b302a5dc6f36777c393774..c46692ec7674df2cce34912f4dba3091ad059900 100644 --- a/py/objint_mpz.c +++ b/py/objint_mpz.c @@ -303,13 +303,15 @@ mp_obj_t mp_obj_new_int_from_float(mp_float_t val) { nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_OverflowError, "can't convert inf to int")); } else if (cl == FP_NAN) { nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "can't convert NaN to int")); - } else if (MICROPY_FLOAT_C_FUN(fabs)(val) < 10000) { - // temporary(?) fix for optimising case where int will be small int - return MP_OBJ_NEW_SMALL_INT(MICROPY_FLOAT_C_FUN(trunc)(val)); } else { - mp_obj_int_t *o = mp_obj_int_new_mpz(); - mpz_set_from_float(&o->mpz, val); - return o; + mp_fp_as_int_class_t icl = mp_classify_fp_as_int(val); + if (icl == MP_FP_CLASS_FIT_SMALLINT) { + return MP_OBJ_NEW_SMALL_INT((mp_int_t)val); + } else { + mp_obj_int_t *o = mp_obj_int_new_mpz(); + mpz_set_from_float(&o->mpz, val); + return o; + } } } #endif