Linaro Git Browser
Code Review Sign In
review.linaro.org / lite / micropython / 8854e1fa05d41fe8a3afb0a5c03ed41cc8f7f4a7 / . / py / builtinmath.c
blob: 35c1e45edc1910c5279fa5c7585c8018650c035c [file] [log] [blame]
Damien George0c36da02014-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
21STATIC const mp_obj_float_t mp_math_pi_obj = {{&mp_type_float}, M_PI};
22
23MATH_FUN_1(sqrt, sqrt)
24MATH_FUN_2(pow, pow)
25MATH_FUN_1(exp, exp)
Damien George8854e1f2014-03-12 21:31:41 +0000[diff] [blame^]26MATH_FUN_1(expm1, expm1)
Damien George0c36da02014-03-08 15:24:39 +0000[diff] [blame]27MATH_FUN_1(log, log)
28MATH_FUN_1(log2, log2)
29MATH_FUN_1(log10, log10)
30MATH_FUN_1(cosh, cosh)
31MATH_FUN_1(sinh, sinh)
32MATH_FUN_1(tanh, tanh)
33MATH_FUN_1(acosh, acosh)
34MATH_FUN_1(asinh, asinh)
35MATH_FUN_1(atanh, atanh)
36MATH_FUN_1(cos, cos)
37MATH_FUN_1(sin, sin)
38MATH_FUN_1(tan, tan)
39MATH_FUN_1(acos, acos)
40MATH_FUN_1(asin, asin)
41MATH_FUN_1(atan, atan)
42MATH_FUN_2(atan2, atan2)
43
44STATIC const mp_map_elem_t mp_module_math_globals_table[] = {
45 { MP_OBJ_NEW_QSTR(MP_QSTR___name__), MP_OBJ_NEW_QSTR(MP_QSTR_math) },
46 { MP_OBJ_NEW_QSTR(MP_QSTR_pi), (mp_obj_t)&mp_math_pi_obj },
47 { MP_OBJ_NEW_QSTR(MP_QSTR_sqrt), (mp_obj_t)&mp_math_sqrt_obj },
48 { MP_OBJ_NEW_QSTR(MP_QSTR_pow), (mp_obj_t)&mp_math_pow_obj },
49 { MP_OBJ_NEW_QSTR(MP_QSTR_exp), (mp_obj_t)&mp_math_exp_obj },
Damien George8854e1f2014-03-12 21:31:41 +0000[diff] [blame^]50 { MP_OBJ_NEW_QSTR(MP_QSTR_expm1), (mp_obj_t)&mp_math_expm1_obj },
Damien George0c36da02014-03-08 15:24:39 +0000[diff] [blame]51 { MP_OBJ_NEW_QSTR(MP_QSTR_log), (mp_obj_t)&mp_math_log_obj },
52 { MP_OBJ_NEW_QSTR(MP_QSTR_log2), (mp_obj_t)&mp_math_log2_obj },
53 { MP_OBJ_NEW_QSTR(MP_QSTR_log10), (mp_obj_t)&mp_math_log10_obj },
54 { MP_OBJ_NEW_QSTR(MP_QSTR_cosh), (mp_obj_t)&mp_math_cosh_obj },
55 { MP_OBJ_NEW_QSTR(MP_QSTR_sinh), (mp_obj_t)&mp_math_sinh_obj },
56 { MP_OBJ_NEW_QSTR(MP_QSTR_tanh), (mp_obj_t)&mp_math_tanh_obj },
57 { MP_OBJ_NEW_QSTR(MP_QSTR_acosh), (mp_obj_t)&mp_math_acosh_obj },
58 { MP_OBJ_NEW_QSTR(MP_QSTR_asinh), (mp_obj_t)&mp_math_asinh_obj },
59 { MP_OBJ_NEW_QSTR(MP_QSTR_atanh), (mp_obj_t)&mp_math_atanh_obj },
60 { MP_OBJ_NEW_QSTR(MP_QSTR_cos), (mp_obj_t)&mp_math_cos_obj },
61 { MP_OBJ_NEW_QSTR(MP_QSTR_sin), (mp_obj_t)&mp_math_sin_obj },
62 { MP_OBJ_NEW_QSTR(MP_QSTR_tan), (mp_obj_t)&mp_math_tan_obj },
63 { MP_OBJ_NEW_QSTR(MP_QSTR_acos), (mp_obj_t)&mp_math_acos_obj },
64 { MP_OBJ_NEW_QSTR(MP_QSTR_asin), (mp_obj_t)&mp_math_asin_obj },
65 { MP_OBJ_NEW_QSTR(MP_QSTR_atan), (mp_obj_t)&mp_math_atan_obj },
66 { MP_OBJ_NEW_QSTR(MP_QSTR_atan2), (mp_obj_t)&mp_math_atan2_obj },
67};
68
69STATIC const mp_map_t mp_module_math_globals = {
70 .all_keys_are_qstrs = 1,
71 .table_is_fixed_array = 1,
72 .used = sizeof(mp_module_math_globals_table) / sizeof(mp_map_elem_t),
73 .alloc = sizeof(mp_module_math_globals_table) / sizeof(mp_map_elem_t),
74 .table = (mp_map_elem_t*)mp_module_math_globals_table,
75};
76
77const mp_obj_module_t mp_module_math = {
78 .base = { &mp_type_module },
79 .name = MP_QSTR_math,
80 .globals = (mp_map_t*)&mp_module_math_globals,
81};
82
83#endif // MICROPY_ENABLE_FLOAT