diff options
Diffstat (limited to 'py/objcomplex.c')
| -rw-r--r-- | py/objcomplex.c | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/py/objcomplex.c b/py/objcomplex.c index 8e0f638c3e..2768a01e65 100644 --- a/py/objcomplex.c +++ b/py/objcomplex.c @@ -25,6 +25,7 @@ */ #include <stdlib.h> +#include <stdio.h> #include <assert.h> #include "mpconfig.h" @@ -55,21 +56,26 @@ mp_obj_t mp_obj_new_complex(mp_float_t real, mp_float_t imag); STATIC void complex_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t o_in, mp_print_kind_t kind) { mp_obj_complex_t *o = o_in; #if MICROPY_FLOAT_IMPL == MICROPY_FLOAT_IMPL_FLOAT - char buf[32]; + char buf[16]; if (o->real == 0) { - format_float(o->imag, buf, sizeof(buf), 'g', 6, '\0'); + format_float(o->imag, buf, sizeof(buf), 'g', 7, '\0'); print(env, "%sj", buf); } else { - format_float(o->real, buf, sizeof(buf), 'g', 6, '\0'); + format_float(o->real, buf, sizeof(buf), 'g', 7, '\0'); print(env, "(%s+", buf); - format_float(o->imag, buf, sizeof(buf), 'g', 6, '\0'); + format_float(o->imag, buf, sizeof(buf), 'g', 7, '\0'); print(env, "%sj)", buf); } #else + char buf[32]; if (o->real == 0) { - print(env, "%.8gj", (double) o->imag); + sprintf(buf, "%.16g", (double)o->imag); + print(env, "%sj", buf); } else { - print(env, "(%.8g+%.8gj)", (double) o->real, (double) o->imag); + sprintf(buf, "%.16g", (double)o->real); + print(env, "(%s+", buf); + sprintf(buf, "%.16g", (double)o->imag); + print(env, "%sj)", buf); } #endif } |