W3SLAV/micropython
1
Fork 0
mirror of https://github.com/W3SLAV/micropython.git synced 2025-06-20 04:25:34 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

py: Add mp_obj_is_float function (macro) and use it where appropriate.

Browse Source
This commit is contained in:
Damien George 2015-08-20 23:30:12 +01:00
parent 60401d461a
commit aaef1851a7
8 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
py/modbuiltins.c
View File

@ -93,7 +93,7 @@ STATIC mp_obj_t mp_builtin_abs(mp_obj_t o_in) {
if (0) { if (0) {
// dummy // dummy
#if MICROPY_PY_BUILTINS_FLOAT #if MICROPY_PY_BUILTINS_FLOAT
} else if (MP_OBJ_IS_TYPE(o_in, &mp_type_float)) { } else if (mp_obj_is_float(o_in)) {
mp_float_t value = mp_obj_float_get(o_in); mp_float_t value = mp_obj_float_get(o_in);
// TODO check for NaN etc // TODO check for NaN etc
if (value < 0) { if (value < 0) {

4
py/obj.c
View File

@ -268,7 +268,7 @@ mp_float_t mp_obj_get_float(mp_obj_t arg) {
return MP_OBJ_SMALL_INT_VALUE(arg); return MP_OBJ_SMALL_INT_VALUE(arg);
} else if (MP_OBJ_IS_TYPE(arg, &mp_type_int)) { } else if (MP_OBJ_IS_TYPE(arg, &mp_type_int)) {
return mp_obj_int_as_float(arg); return mp_obj_int_as_float(arg);
} else if (MP_OBJ_IS_TYPE(arg, &mp_type_float)) { } else if (mp_obj_is_float(arg)) {
return mp_obj_float_get(arg); return mp_obj_float_get(arg);
} else { } else {
if (MICROPY_ERROR_REPORTING == MICROPY_ERROR_REPORTING_TERSE) { if (MICROPY_ERROR_REPORTING == MICROPY_ERROR_REPORTING_TERSE) {
@ -295,7 +295,7 @@ void mp_obj_get_complex(mp_obj_t arg, mp_float_t *real, mp_float_t *imag) {
} else if (MP_OBJ_IS_TYPE(arg, &mp_type_int)) { } else if (MP_OBJ_IS_TYPE(arg, &mp_type_int)) {
*real = mp_obj_int_as_float(arg); *real = mp_obj_int_as_float(arg);
*imag = 0; *imag = 0;
} else if (MP_OBJ_IS_TYPE(arg, &mp_type_float)) { } else if (mp_obj_is_float(arg)) {
*real = mp_obj_float_get(arg); *real = mp_obj_float_get(arg);
*imag = 0; *imag = 0;
} else if (MP_OBJ_IS_TYPE(arg, &mp_type_complex)) { } else if (MP_OBJ_IS_TYPE(arg, &mp_type_complex)) {

1
py/obj.h
View File

@ -568,6 +568,7 @@ typedef struct _mp_obj_float_t {
mp_obj_base_t base; mp_obj_base_t base;
mp_float_t value; mp_float_t value;
} mp_obj_float_t; } mp_obj_float_t;
#define mp_obj_is_float(o) MP_OBJ_IS_TYPE((o), &mp_type_float)
mp_float_t mp_obj_float_get(mp_obj_t self_in); mp_float_t mp_obj_float_get(mp_obj_t self_in);
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 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

4
py/objfloat.c
View File

@ -72,7 +72,7 @@ STATIC mp_obj_t float_make_new(mp_obj_t type_in, mp_uint_t n_args, mp_uint_t n_k
mp_uint_t l; mp_uint_t l;
const char *s = mp_obj_str_get_data(args[0], &l); const char *s = mp_obj_str_get_data(args[0], &l);
return mp_parse_num_decimal(s, l, false, false, NULL); return mp_parse_num_decimal(s, l, false, false, NULL);
} else if (MP_OBJ_IS_TYPE(args[0], &mp_type_float)) { } else if (mp_obj_is_float(args[0])) {
// a float, just return it // a float, just return it
return args[0]; return args[0];
} else { } else {
@ -121,7 +121,7 @@ mp_obj_t mp_obj_new_float(mp_float_t value) {
} }
mp_float_t mp_obj_float_get(mp_obj_t self_in) { mp_float_t mp_obj_float_get(mp_obj_t self_in) {
assert(MP_OBJ_IS_TYPE(self_in, &mp_type_float)); assert(mp_obj_is_float(self_in));
mp_obj_float_t *self = self_in; mp_obj_float_t *self = self_in;
return self->value; return self->value;
} }

2
py/objint.c
View File

@ -60,7 +60,7 @@ STATIC mp_obj_t mp_obj_int_make_new(mp_obj_t type_in, mp_uint_t n_args, mp_uint_
const char *s = mp_obj_str_get_data(args[0], &l); const char *s = mp_obj_str_get_data(args[0], &l);
return mp_parse_num_integer(s, l, 0, NULL); return mp_parse_num_integer(s, l, 0, NULL);
#if MICROPY_PY_BUILTINS_FLOAT #if MICROPY_PY_BUILTINS_FLOAT
} else if (MP_OBJ_IS_TYPE(args[0], &mp_type_float)) { } else if (mp_obj_is_float(args[0])) {
return mp_obj_new_int_from_float(mp_obj_float_get(args[0])); return mp_obj_new_int_from_float(mp_obj_float_get(args[0]));
#endif #endif
} else { } else {

2
py/objint_mpz.c
View File

@ -178,7 +178,7 @@ mp_obj_t mp_obj_int_binary_op(mp_uint_t op, mp_obj_t lhs_in, mp_obj_t rhs_in) {
} else if (MP_OBJ_IS_TYPE(rhs_in, &mp_type_int)) { } else if (MP_OBJ_IS_TYPE(rhs_in, &mp_type_int)) {
zrhs = &((mp_obj_int_t*)rhs_in)->mpz; zrhs = &((mp_obj_int_t*)rhs_in)->mpz;
#if MICROPY_PY_BUILTINS_FLOAT #if MICROPY_PY_BUILTINS_FLOAT
} else if (MP_OBJ_IS_TYPE(rhs_in, &mp_type_float)) { } else if (mp_obj_is_float(rhs_in)) {
return mp_obj_float_binary_op(op, mpz_as_float(zlhs), rhs_in); return mp_obj_float_binary_op(op, mpz_as_float(zlhs), rhs_in);
#if MICROPY_PY_BUILTINS_COMPLEX #if MICROPY_PY_BUILTINS_COMPLEX
} else if (MP_OBJ_IS_TYPE(rhs_in, &mp_type_complex)) { } else if (MP_OBJ_IS_TYPE(rhs_in, &mp_type_complex)) {

6
py/objstr.c
View File

@ -827,15 +827,15 @@ STATIC bool arg_looks_integer(mp_obj_t arg) {
STATIC bool arg_looks_numeric(mp_obj_t arg) { STATIC bool arg_looks_numeric(mp_obj_t arg) {
return arg_looks_integer(arg) return arg_looks_integer(arg)
#if MICROPY_PY_BUILTINS_FLOAT #if MICROPY_PY_BUILTINS_FLOAT
|| MP_OBJ_IS_TYPE(arg, &mp_type_float) || mp_obj_is_float(arg)
#endif #endif
; ;
} }
STATIC mp_obj_t arg_as_int(mp_obj_t arg) { STATIC mp_obj_t arg_as_int(mp_obj_t arg) {
#if MICROPY_PY_BUILTINS_FLOAT #if MICROPY_PY_BUILTINS_FLOAT
if (MP_OBJ_IS_TYPE(arg, &mp_type_float)) { if (mp_obj_is_float(arg)) {
return mp_obj_new_int_from_float(mp_obj_get_float(arg)); return mp_obj_new_int_from_float(mp_obj_float_get(arg));
} }
#endif #endif
return arg; return arg;

2
py/runtime.c
View File

@ -458,7 +458,7 @@ mp_obj_t mp_binary_op(mp_uint_t op, mp_obj_t lhs, mp_obj_t rhs) {
return mp_obj_new_int(lhs_val); return mp_obj_new_int(lhs_val);
} }
#if MICROPY_PY_BUILTINS_FLOAT #if MICROPY_PY_BUILTINS_FLOAT
} else if (MP_OBJ_IS_TYPE(rhs, &mp_type_float)) { } else if (mp_obj_is_float(rhs)) {
mp_obj_t res = mp_obj_float_binary_op(op, lhs_val, rhs); mp_obj_t res = mp_obj_float_binary_op(op, lhs_val, rhs);
if (res == MP_OBJ_NULL) { if (res == MP_OBJ_NULL) {
goto unsupported_op; goto unsupported_op;