|
| 1 | +// SPDX-FileCopyrightText: 2024 M5Stack Technology CO LTD |
| 2 | +// |
| 3 | +// SPDX-License-Identifier: MIT |
| 4 | + |
| 5 | +#include <stdio.h> |
| 6 | +#include <string.h> |
| 7 | +#include <sys/time.h> |
| 8 | + |
| 9 | +#include "py/runtime.h" |
| 10 | + |
| 11 | +STATIC mp_obj_t m5_utils_remap(size_t n_args, const mp_obj_t *args) { |
| 12 | + mp_float_t x = mp_obj_get_float(args[0]); |
| 13 | + mp_float_t in_min = mp_obj_get_float(args[1]); |
| 14 | + mp_float_t in_max = mp_obj_get_float(args[2]); |
| 15 | + mp_float_t out_min = mp_obj_get_float(args[3]); |
| 16 | + mp_float_t out_max = mp_obj_get_float(args[4]); |
| 17 | + |
| 18 | + if (in_max == in_min) { |
| 19 | + // 抛出异常或返回错误 |
| 20 | + mp_raise_ValueError("Input range cannot be zero"); |
| 21 | + return mp_const_none; |
| 22 | + } |
| 23 | + |
| 24 | + mp_float_t result = (x - in_min) * (out_max - out_min) / (in_max - in_min) + out_min; |
| 25 | + return mp_obj_new_float(result); |
| 26 | +} |
| 27 | +STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(m5_utils_remap_obj, 5, 5, m5_utils_remap); |
| 28 | + |
| 29 | + |
| 30 | +STATIC const mp_rom_map_elem_t m5utils_module_globals_table[] = { |
| 31 | + { MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_m5utils) }, |
| 32 | + |
| 33 | + { MP_ROM_QSTR(MP_QSTR_remap), MP_ROM_PTR(&m5_utils_remap_obj) }, |
| 34 | +}; |
| 35 | + |
| 36 | +STATIC MP_DEFINE_CONST_DICT(m5utils_module_globals, m5utils_module_globals_table); |
| 37 | + |
| 38 | +const mp_obj_module_t m5utils_module = { |
| 39 | + .base = { &mp_type_module }, |
| 40 | + .globals = (mp_obj_dict_t *)&m5utils_module_globals, |
| 41 | +}; |
| 42 | + |
| 43 | +MP_REGISTER_MODULE(MP_QSTR_m5utils, m5utils_module); |
0 commit comments