You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.6 KiB
50 lines
1.6 KiB
/*
|
|
* This file is part of the micropython-usermod project,
|
|
*
|
|
* https://github.com/v923z/micropython-usermod
|
|
*
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2019 Zoltán Vörös
|
|
*/
|
|
|
|
#include <math.h>
|
|
#include <stdio.h>
|
|
#include "py/obj.h"
|
|
#include "py/runtime.h"
|
|
#include "mphalport.h" // needed for mp_hal_ticks_cpu()
|
|
#include "py/builtin.h" // needed for mp_micropython_mem_info()
|
|
|
|
STATIC mp_obj_t measure_cpu(mp_obj_t _x, mp_obj_t _y, mp_obj_t _z) {
|
|
size_t start, middle, end;
|
|
start = m_get_current_bytes_allocated();
|
|
|
|
float x = mp_obj_get_float(_x);
|
|
float y = mp_obj_get_float(_y);
|
|
float z = mp_obj_get_float(_z);
|
|
middle = m_get_current_bytes_allocated();
|
|
|
|
float hypo = sqrtf(x*x + y*y + z*z);
|
|
end = m_get_current_bytes_allocated();
|
|
mp_obj_t tuple[4];
|
|
tuple[0] = MP_OBJ_NEW_SMALL_INT(start);
|
|
tuple[1] = MP_OBJ_NEW_SMALL_INT(middle);
|
|
tuple[2] = MP_OBJ_NEW_SMALL_INT(end);
|
|
tuple[3] = mp_obj_new_float(hypo);
|
|
return mp_obj_new_tuple(4, tuple);
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_3(measure_cpu_obj, measure_cpu);
|
|
|
|
STATIC const mp_rom_map_elem_t profiling_module_globals_table[] = {
|
|
{ MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_profiling) },
|
|
{ MP_ROM_QSTR(MP_QSTR_measure), MP_ROM_PTR(&measure_cpu_obj) },
|
|
};
|
|
STATIC MP_DEFINE_CONST_DICT(profiling_module_globals, profiling_module_globals_table);
|
|
|
|
const mp_obj_module_t profiling_user_cmodule = {
|
|
.base = { &mp_type_module },
|
|
.globals = (mp_obj_dict_t*)&profiling_module_globals,
|
|
};
|
|
|
|
MP_REGISTER_MODULE(MP_QSTR_profiling, profiling_user_cmodule, MODULE_PROFILING_ENABLED);
|
|
|