| Damien George | 0c36da0 | 2014-03-08 15:24:39 +0000 | [diff] [blame] | 1 | #include <stdint.h> |
| 2 | #include <math.h> |
| 3 | |
| 4 | #include "misc.h" |
| 5 | #include "mpconfig.h" |
| 6 | #include "qstr.h" |
| 7 | #include "obj.h" |
| 8 | #include "map.h" |
| 9 | #include "builtin.h" |
| 10 | |
| 11 | #if MICROPY_ENABLE_FLOAT |
| 12 | |
| 13 | #define MATH_FUN_1(py_name, c_name) \ |
| 14 | mp_obj_t mp_math_ ## py_name(mp_obj_t x_obj) { return mp_obj_new_float(MICROPY_FLOAT_C_FUN(c_name)(mp_obj_get_float(x_obj))); } \ |
| 15 | STATIC MP_DEFINE_CONST_FUN_OBJ_1(mp_math_## py_name ## _obj, mp_math_ ## py_name); |
| 16 | |
| 17 | #define MATH_FUN_2(py_name, c_name) \ |
| 18 | mp_obj_t mp_math_ ## py_name(mp_obj_t x_obj, mp_obj_t y_obj) { return mp_obj_new_float(MICROPY_FLOAT_C_FUN(c_name)(mp_obj_get_float(x_obj), mp_obj_get_float(y_obj))); } \ |
| 19 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(mp_math_## py_name ## _obj, mp_math_ ## py_name); |
| 20 | |
| 21 | STATIC const mp_obj_float_t mp_math_pi_obj = {{&mp_type_float}, M_PI}; |
| 22 | |
| 23 | MATH_FUN_1(sqrt, sqrt) |
| 24 | MATH_FUN_2(pow, pow) |
| 25 | MATH_FUN_1(exp, exp) |
| 26 | MATH_FUN_1(log, log) |
| 27 | MATH_FUN_1(log2, log2) |
| 28 | MATH_FUN_1(log10, log10) |
| 29 | MATH_FUN_1(cosh, cosh) |
| 30 | MATH_FUN_1(sinh, sinh) |
| 31 | MATH_FUN_1(tanh, tanh) |
| 32 | MATH_FUN_1(acosh, acosh) |
| 33 | MATH_FUN_1(asinh, asinh) |
| 34 | MATH_FUN_1(atanh, atanh) |
| 35 | MATH_FUN_1(cos, cos) |
| 36 | MATH_FUN_1(sin, sin) |
| 37 | MATH_FUN_1(tan, tan) |
| 38 | MATH_FUN_1(acos, acos) |
| 39 | MATH_FUN_1(asin, asin) |
| 40 | MATH_FUN_1(atan, atan) |
| 41 | MATH_FUN_2(atan2, atan2) |
| 42 | |
| 43 | STATIC const mp_map_elem_t mp_module_math_globals_table[] = { |
| 44 | { MP_OBJ_NEW_QSTR(MP_QSTR___name__), MP_OBJ_NEW_QSTR(MP_QSTR_math) }, |
| 45 | { MP_OBJ_NEW_QSTR(MP_QSTR_pi), (mp_obj_t)&mp_math_pi_obj }, |
| 46 | { MP_OBJ_NEW_QSTR(MP_QSTR_sqrt), (mp_obj_t)&mp_math_sqrt_obj }, |
| 47 | { MP_OBJ_NEW_QSTR(MP_QSTR_pow), (mp_obj_t)&mp_math_pow_obj }, |
| 48 | { MP_OBJ_NEW_QSTR(MP_QSTR_exp), (mp_obj_t)&mp_math_exp_obj }, |
| 49 | { MP_OBJ_NEW_QSTR(MP_QSTR_log), (mp_obj_t)&mp_math_log_obj }, |
| 50 | { MP_OBJ_NEW_QSTR(MP_QSTR_log2), (mp_obj_t)&mp_math_log2_obj }, |
| 51 | { MP_OBJ_NEW_QSTR(MP_QSTR_log10), (mp_obj_t)&mp_math_log10_obj }, |
| 52 | { MP_OBJ_NEW_QSTR(MP_QSTR_cosh), (mp_obj_t)&mp_math_cosh_obj }, |
| 53 | { MP_OBJ_NEW_QSTR(MP_QSTR_sinh), (mp_obj_t)&mp_math_sinh_obj }, |
| 54 | { MP_OBJ_NEW_QSTR(MP_QSTR_tanh), (mp_obj_t)&mp_math_tanh_obj }, |
| 55 | { MP_OBJ_NEW_QSTR(MP_QSTR_acosh), (mp_obj_t)&mp_math_acosh_obj }, |
| 56 | { MP_OBJ_NEW_QSTR(MP_QSTR_asinh), (mp_obj_t)&mp_math_asinh_obj }, |
| 57 | { MP_OBJ_NEW_QSTR(MP_QSTR_atanh), (mp_obj_t)&mp_math_atanh_obj }, |
| 58 | { MP_OBJ_NEW_QSTR(MP_QSTR_cos), (mp_obj_t)&mp_math_cos_obj }, |
| 59 | { MP_OBJ_NEW_QSTR(MP_QSTR_sin), (mp_obj_t)&mp_math_sin_obj }, |
| 60 | { MP_OBJ_NEW_QSTR(MP_QSTR_tan), (mp_obj_t)&mp_math_tan_obj }, |
| 61 | { MP_OBJ_NEW_QSTR(MP_QSTR_acos), (mp_obj_t)&mp_math_acos_obj }, |
| 62 | { MP_OBJ_NEW_QSTR(MP_QSTR_asin), (mp_obj_t)&mp_math_asin_obj }, |
| 63 | { MP_OBJ_NEW_QSTR(MP_QSTR_atan), (mp_obj_t)&mp_math_atan_obj }, |
| 64 | { MP_OBJ_NEW_QSTR(MP_QSTR_atan2), (mp_obj_t)&mp_math_atan2_obj }, |
| 65 | }; |
| 66 | |
| 67 | STATIC const mp_map_t mp_module_math_globals = { |
| 68 | .all_keys_are_qstrs = 1, |
| 69 | .table_is_fixed_array = 1, |
| 70 | .used = sizeof(mp_module_math_globals_table) / sizeof(mp_map_elem_t), |
| 71 | .alloc = sizeof(mp_module_math_globals_table) / sizeof(mp_map_elem_t), |
| 72 | .table = (mp_map_elem_t*)mp_module_math_globals_table, |
| 73 | }; |
| 74 | |
| 75 | const mp_obj_module_t mp_module_math = { |
| 76 | .base = { &mp_type_module }, |
| 77 | .name = MP_QSTR_math, |
| 78 | .globals = (mp_map_t*)&mp_module_math_globals, |
| 79 | }; |
| 80 | |
| 81 | #endif // MICROPY_ENABLE_FLOAT |