modtime.c 2.41 KB
Newer Older
xbe's avatar
xbe committed
1
#include <unistd.h>
2
3
#include <string.h>
#include <time.h>
4
5
#include <sys/time.h>
#include <math.h>
6
7
8
9
10
11
12

#include "misc.h"
#include "mpconfig.h"
#include "qstr.h"
#include "obj.h"
#include "runtime.h"

13
STATIC mp_obj_t mod_time_time() {
14
15
16
17
18
19
#if MICROPY_ENABLE_FLOAT
    struct timeval tv;
    gettimeofday(&tv, NULL);
    mp_float_t val = tv.tv_sec + (mp_float_t)tv.tv_usec / 1000000;
    return mp_obj_new_float(val);
#else
20
    return mp_obj_new_int((machine_int_t)time(NULL));
21
#endif
22
}
23
STATIC MP_DEFINE_CONST_FUN_OBJ_0(mod_time_time_obj, mod_time_time);
24
25

// Note: this is deprecated since CPy3.3, but pystone still uses it.
26
STATIC mp_obj_t mod_time_clock() {
27
28
#if MICROPY_ENABLE_FLOAT
    // POSIX requires CLOCKS_PER_SEC equals 1000000, so that's what we assume.
29
30
31
32
    // float cannot represent full range of int32 precisely, so we pre-divide
    // int to reduce resolution, and then actually do float division hoping
    // to preserve integer part resolution.
    return mp_obj_new_float((float)(clock() / 1000) / 1000.0);
33
34
35
#else
    return mp_obj_new_int((machine_int_t)clock());
#endif
36
}
37
STATIC MP_DEFINE_CONST_FUN_OBJ_0(mod_time_clock_obj, mod_time_clock);
38

39
STATIC mp_obj_t mod_time_sleep(mp_obj_t arg) {
40
41
#if MICROPY_ENABLE_FLOAT
    struct timeval tv;
42
    mp_float_t val = mp_obj_get_float(arg);
43
44
45
46
47
48
49
50
51
    double ipart;
    tv.tv_usec = round(modf(val, &ipart) * 1000000);
    tv.tv_sec = ipart;
    select(0, NULL, NULL, NULL, &tv);
#else
    sleep(mp_obj_get_int(arg));
#endif
    return mp_const_none;
}
52
STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_time_sleep_obj, mod_time_sleep);
53

54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
STATIC const mp_map_elem_t mp_module_time_globals_table[] = {
    { MP_OBJ_NEW_QSTR(MP_QSTR___name__), MP_OBJ_NEW_QSTR(MP_QSTR_time) },
    { MP_OBJ_NEW_QSTR(MP_QSTR_clock), (mp_obj_t)&mod_time_clock_obj },
    { MP_OBJ_NEW_QSTR(MP_QSTR_sleep), (mp_obj_t)&mod_time_sleep_obj },
    { MP_OBJ_NEW_QSTR(MP_QSTR_time), (mp_obj_t)&mod_time_time_obj },
};

STATIC const mp_obj_dict_t mp_module_time_globals = {
    .base = {&mp_type_dict},
    .map = {
        .all_keys_are_qstrs = 1,
        .table_is_fixed_array = 1,
        .used = sizeof(mp_module_time_globals_table) / sizeof(mp_map_elem_t),
        .alloc = sizeof(mp_module_time_globals_table) / sizeof(mp_map_elem_t),
        .table = (mp_map_elem_t*)mp_module_time_globals_table,
    },
};

const mp_obj_module_t mp_module_time = {
    .base = { &mp_type_module },
    .name = MP_QSTR_time,
    .globals = (mp_obj_dict_t*)&mp_module_time_globals,
};