py/persistentcode: Allow to compile with complex numbers disabled.

72732fea · Damien George · 4d2778c9 · 72732fea
Commit 72732fea authored Jun 08, 2017 by Damien George
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 4 deletions

py/persistentcode.c py/persistentcode.c +6 -4

No files found.
--- a/py/persistentcode.c
+++ b/py/persistentcode.c
@@ -286,11 +286,13 @@ STATIC void save_obj(mp_print_t *print, mp_obj_t o) {
        byte obj_type;
        if (MP_OBJ_IS_TYPE(o, &mp_type_int)) {
            obj_type = 'i';
-        } else if (mp_obj_is_float(o)) {
+        #if MICROPY_PY_BUILTINS_COMPLEX
-            obj_type = 'f';
+        } else if (MP_OBJ_IS_TYPE(o, &mp_type_complex)) {
-        } else {
-            assert(MP_OBJ_IS_TYPE(o, &mp_type_complex));
            obj_type = 'c';
+        #endif
+        } else {
+            assert(mp_obj_is_float(o));
+            obj_type = 'f';
        }
        vstr_t vstr;
        mp_print_t pr;