mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
Add time module
This commit is contained in:
parent
9e6524f70c
commit
ffb3dac484
81
src/py/py_clock.c
Normal file
81
src/py/py_clock.c
Normal file
@ -0,0 +1,81 @@
|
||||
#include <libmp.h>
|
||||
#include "systick.h"
|
||||
#include "py_clock.h"
|
||||
|
||||
/* Clock Type */
|
||||
typedef struct _py_clock_obj_t {
|
||||
mp_obj_base_t base;
|
||||
uint32_t t_start;
|
||||
uint32_t t_ticks;
|
||||
uint32_t t_frame;
|
||||
} py_clock_obj_t;
|
||||
|
||||
mp_obj_t py_clock_tick(py_clock_obj_t *clock)
|
||||
{
|
||||
clock->t_start = systick_current_millis();
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
mp_obj_t py_clock_fps(py_clock_obj_t *clock)
|
||||
{
|
||||
clock->t_frame++;
|
||||
clock->t_ticks += (systick_current_millis()-clock->t_start);
|
||||
return mp_obj_new_float(1000.0f/(clock->t_ticks/(float)clock->t_frame));
|
||||
}
|
||||
|
||||
mp_obj_t py_clock_avg(py_clock_obj_t *clock)
|
||||
{
|
||||
clock->t_frame++;
|
||||
clock->t_ticks += (systick_current_millis()-clock->t_start);
|
||||
return mp_obj_new_float(clock->t_ticks/(float)clock->t_frame);
|
||||
}
|
||||
|
||||
mp_obj_t py_clock_reset(py_clock_obj_t *clock)
|
||||
{
|
||||
clock->t_start = 0;
|
||||
clock->t_ticks = 0;
|
||||
clock->t_frame = 0;
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
static void py_clock_print(void (*print)(void *env, const char *fmt, ...),
|
||||
void *env, mp_obj_t self_in, mp_print_kind_t kind)
|
||||
{
|
||||
py_clock_obj_t *self = self_in;
|
||||
|
||||
/* print some info */
|
||||
print(env, "t_start:%d t_ticks:%d t_frame:%d\n",
|
||||
self->t_start, self->t_ticks, self->t_frame);
|
||||
}
|
||||
|
||||
static MP_DEFINE_CONST_FUN_OBJ_1(py_clock_tick_obj, py_clock_tick);
|
||||
static MP_DEFINE_CONST_FUN_OBJ_1(py_clock_fps_obj, py_clock_fps);
|
||||
static MP_DEFINE_CONST_FUN_OBJ_1(py_clock_avg_obj, py_clock_avg);
|
||||
static MP_DEFINE_CONST_FUN_OBJ_1(py_clock_reset_obj, py_clock_reset);
|
||||
|
||||
static const mp_method_t py_clock_methods[] = {
|
||||
{ "tick", &py_clock_tick_obj},
|
||||
{ "fps", &py_clock_fps_obj},
|
||||
{ "avg", &py_clock_avg_obj},
|
||||
{ "reset", &py_clock_reset_obj},
|
||||
{ NULL, NULL },
|
||||
};
|
||||
|
||||
static const mp_obj_type_t py_clock_type = {
|
||||
{ &mp_type_type },
|
||||
.name = MP_QSTR_Clock,
|
||||
.print = py_clock_print,
|
||||
.methods = py_clock_methods,
|
||||
};
|
||||
|
||||
mp_obj_t py_clock()
|
||||
{
|
||||
py_clock_obj_t *clock =NULL;
|
||||
clock = m_new_obj(py_clock_obj_t);
|
||||
clock->base.type = &py_clock_type;
|
||||
clock->t_start = 0;
|
||||
clock->t_ticks = 0;
|
||||
clock->t_frame = 0;
|
||||
|
||||
return clock;
|
||||
}
|
||||
4
src/py/py_clock.h
Normal file
4
src/py/py_clock.h
Normal file
@ -0,0 +1,4 @@
|
||||
#ifndef __PY_CLOCK_H__
|
||||
#define __PY_CLOCK_H__
|
||||
mp_obj_t py_clock();
|
||||
#endif /* __PY_CLOCK_H__ */
|
||||
32
src/py/py_time.c
Normal file
32
src/py/py_time.c
Normal file
@ -0,0 +1,32 @@
|
||||
#include <libmp.h>
|
||||
#include "systick.h"
|
||||
#include "py_time.h"
|
||||
#include "py_clock.h"
|
||||
|
||||
mp_obj_t py_time_ticks()
|
||||
{
|
||||
return mp_obj_new_int(systick_current_millis());
|
||||
}
|
||||
|
||||
mp_obj_t py_time_clock()
|
||||
{
|
||||
return py_clock();
|
||||
}
|
||||
|
||||
mp_obj_t py_time_sleep(mp_obj_t ms)
|
||||
{
|
||||
systick_sleep(mp_obj_get_int(ms));
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
mp_obj_t py_time_init()
|
||||
{
|
||||
/* Create module */
|
||||
mp_obj_t m = mp_obj_new_module(qstr_from_str("time"));
|
||||
|
||||
/* Store module functions */
|
||||
rt_store_attr(m, qstr_from_str("clock"), rt_make_function_n(0, py_time_clock));
|
||||
rt_store_attr(m, qstr_from_str("ticks"), rt_make_function_n(0, py_time_ticks));
|
||||
rt_store_attr(m, qstr_from_str("sleep"), rt_make_function_n(1, py_time_sleep));
|
||||
return m;
|
||||
}
|
||||
4
src/py/py_time.h
Normal file
4
src/py/py_time.h
Normal file
@ -0,0 +1,4 @@
|
||||
#ifndef __PY_TIME_H__
|
||||
#define __PY_TIME_H__
|
||||
mp_obj_t py_time_init();
|
||||
#endif /* __PY_TIME_H__ */
|
||||
Loading…
Reference in New Issue
Block a user