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

dpgeorge · dpgeorge · commit aaef1851a748 · 2015-10-20T12:35:17.000+01:00
diff --git a/py/modbuiltins.c b/py/modbuiltins.c
@@ -93,7 +93,7 @@ STATIC mp_obj_t mp_builtin_abs(mp_obj_t o_in) {
     if (0) {
         // dummy
 #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);
         // TODO check for NaN etc
         if (value < 0) {
diff --git a/py/obj.c b/py/obj.c
@@ -268,7 +268,7 @@ mp_float_t mp_obj_get_float(mp_obj_t arg) {
         return MP_OBJ_SMALL_INT_VALUE(arg);
     } else if (MP_OBJ_IS_TYPE(arg, &mp_type_int)) {
         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);
     } else {
         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)) {
         *real = mp_obj_int_as_float(arg);
         *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);
         *imag = 0;
     } else if (MP_OBJ_IS_TYPE(arg, &mp_type_complex)) {
diff --git a/py/obj.h b/py/obj.h
@@ -568,6 +568,7 @@ typedef struct _mp_obj_float_t {
     mp_obj_base_t base;
     mp_float_t value;
 } 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_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
 
diff --git a/py/objfloat.c b/py/objfloat.c
@@ -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;
                 const char *s = mp_obj_str_get_data(args[0], &l);
                 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
                 return args[0];
             } 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) {
-    assert(MP_OBJ_IS_TYPE(self_in, &mp_type_float));
+    assert(mp_obj_is_float(self_in));
     mp_obj_float_t *self = self_in;
     return self->value;
 }
diff --git a/py/objint.c b/py/objint.c
@@ -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);
                 return mp_parse_num_integer(s, l, 0, NULL);
 #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]));
 #endif
             } else {
diff --git a/py/objint_mpz.c b/py/objint_mpz.c
@@ -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)) {
         zrhs = &((mp_obj_int_t*)rhs_in)->mpz;
 #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);
 #if MICROPY_PY_BUILTINS_COMPLEX
     } else if (MP_OBJ_IS_TYPE(rhs_in, &mp_type_complex)) {
diff --git a/py/objstr.c b/py/objstr.c
@@ -827,15 +827,15 @@ STATIC bool arg_looks_integer(mp_obj_t arg) {
 STATIC bool arg_looks_numeric(mp_obj_t arg) {
     return arg_looks_integer(arg)
 #if MICROPY_PY_BUILTINS_FLOAT
-        || MP_OBJ_IS_TYPE(arg, &mp_type_float)
+        || mp_obj_is_float(arg)
 #endif
     ;
 }
 
 STATIC mp_obj_t arg_as_int(mp_obj_t arg) {
 #if MICROPY_PY_BUILTINS_FLOAT
-    if (MP_OBJ_IS_TYPE(arg, &mp_type_float)) {
-        return mp_obj_new_int_from_float(mp_obj_get_float(arg));
+    if (mp_obj_is_float(arg)) {
+        return mp_obj_new_int_from_float(mp_obj_float_get(arg));
     }
 #endif
     return arg;
diff --git a/py/runtime.c b/py/runtime.c
@@ -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);
             }
 #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);
             if (res == MP_OBJ_NULL) {
                 goto unsupported_op;

Original file line number	Diff line number	Diff line change
`@@ -827,15 +827,15 @@ STATIC bool arg_looks_integer(mp_obj_t arg) {`
`827`	`827`	`STATIC bool arg_looks_numeric(mp_obj_t arg) {`
`828`	`828`	`return arg_looks_integer(arg)`
`829`	`829`	`#if MICROPY_PY_BUILTINS_FLOAT`
`830`		`- \|\| MP_OBJ_IS_TYPE(arg, &mp_type_float)`
	`830`	`+ \|\| mp_obj_is_float(arg)`
`831`	`831`	`#endif`
`832`	`832`	`;`
`833`	`833`	`}`
`834`	`834`
`835`	`835`	`STATIC mp_obj_t arg_as_int(mp_obj_t arg) {`
`836`	`836`	`#if MICROPY_PY_BUILTINS_FLOAT`
`837`		`- if (MP_OBJ_IS_TYPE(arg, &mp_type_float)) {`
`838`		`- return mp_obj_new_int_from_float(mp_obj_get_float(arg));`
	`837`	`+ if (mp_obj_is_float(arg)) {`
	`838`	`+ return mp_obj_new_int_from_float(mp_obj_float_get(arg));`
`839`	`839`	`}`
`840`	`840`	`#endif`
`841`	`841`	`return arg;`
Original file line number	Diff line number	Diff line change
`@@ -458,7 +458,7 @@ mp_obj_t mp_binary_op(mp_uint_t op, mp_obj_t lhs, mp_obj_t rhs) {`
`458`	`458`	`return mp_obj_new_int(lhs_val);`
`459`	`459`	`}`
`460`	`460`	`#if MICROPY_PY_BUILTINS_FLOAT`
`461`		`- } else if (MP_OBJ_IS_TYPE(rhs, &mp_type_float)) {`
	`461`	`+ } else if (mp_obj_is_float(rhs)) {`
`462`	`462`	`mp_obj_t res = mp_obj_float_binary_op(op, lhs_val, rhs);`
`463`	`463`	`if (res == MP_OBJ_NULL) {`
`464`	`464`	`goto unsupported_op;`