stmhal: Enable cmath module.

dpgeorge · dpgeorge · commit 93e51b594a14 · 2014-05-03T18:40:50.000+01:00
diff --git a/py/modcmath.c b/py/modcmath.c
@@ -65,7 +65,7 @@ mp_obj_t mp_cmath_sqrt(mp_obj_t z_obj) {
     mp_obj_get_complex(z_obj, &real, &imag);
     mp_float_t sqrt_abs = MICROPY_FLOAT_C_FUN(pow)(real*real + imag*imag, 0.25);
     mp_float_t theta = 0.5 * MICROPY_FLOAT_C_FUN(atan2)(imag, real);
-    return mp_obj_new_complex(sqrt_abs * cos(theta), sqrt_abs * sin(theta));
+    return mp_obj_new_complex(sqrt_abs * MICROPY_FLOAT_C_FUN(cos)(theta), sqrt_abs * MICROPY_FLOAT_C_FUN(sin)(theta));
 }
 STATIC MP_DEFINE_CONST_FUN_OBJ_1(mp_cmath_sqrt_obj, mp_cmath_sqrt);
 
diff --git a/stmhal/mpconfigport.h b/stmhal/mpconfigport.h
@@ -19,6 +19,7 @@
 #define MICROPY_ENABLE_LFN          (1)
 #define MICROPY_LFN_CODE_PAGE       (437) /* 1=SFN/ANSI 437=LFN/U.S.(OEM) */
 #define MICROPY_MOD_SYS_STDFILES    (1)
+#define MICROPY_ENABLE_MOD_CMATH    (1)
 
 // extra built in names to add to the global namespace
 extern const struct _mp_obj_fun_native_t mp_builtin_help_obj;

Original file line number	Diff line number	Diff line change
`@@ -65,7 +65,7 @@ mp_obj_t mp_cmath_sqrt(mp_obj_t z_obj) {`
`65`	`65`	`mp_obj_get_complex(z_obj, &real, &imag);`
`66`	`66`	`mp_float_t sqrt_abs = MICROPY_FLOAT_C_FUN(pow)(realreal + imagimag, 0.25);`
`67`	`67`	`mp_float_t theta = 0.5 * MICROPY_FLOAT_C_FUN(atan2)(imag, real);`
`68`		`- return mp_obj_new_complex(sqrt_abs * cos(theta), sqrt_abs * sin(theta));`
	`68`	`+ return mp_obj_new_complex(sqrt_abs * MICROPY_FLOAT_C_FUN(cos)(theta), sqrt_abs * MICROPY_FLOAT_C_FUN(sin)(theta));`
`69`	`69`	`}`
`70`	`70`	`STATIC MP_DEFINE_CONST_FUN_OBJ_1(mp_cmath_sqrt_obj, mp_cmath_sqrt);`
`71`	`71`