forked from adafruit/circuitpython
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path__init__.c
More file actions
83 lines (73 loc) · 2.47 KB
/
__init__.c
File metadata and controls
83 lines (73 loc) · 2.47 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
// This file is part of the CircuitPython project: https://circuitpython.org
//
// SPDX-FileCopyrightText: Copyright (c) 2021 microDev
//
// SPDX-License-Identifier: MIT
#include "py/gc.h"
#include "py/runtime.h"
#include "shared-module/atexit/__init__.h"
static size_t callback_len = 0;
static atexit_callback_t *callback = NULL;
void atexit_reset(void) {
#if MICROPY_MALLOC_USES_ALLOCATED_SIZE
m_free(callback, callback_len * sizeof(atexit_callback_t));
#else
m_free(callback);
#endif
callback_len = 0;
callback = NULL;
}
void atexit_gc_collect(void) {
gc_collect_ptr(callback);
}
void shared_module_atexit_register(mp_obj_t *func, size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
if (!mp_obj_is_callable(func)) {
mp_raise_TypeError_varg(MP_ERROR_TEXT("'%q' object is not callable"), mp_obj_get_type_qstr(func));
}
size_t n_kw_args = (kw_args) ? kw_args->used : 0;
atexit_callback_t cb = {
.n_pos = 0,
.n_kw = 0,
.func = func,
.args = (n_args + n_kw_args) ? m_malloc((n_args + (n_kw_args * 2)) * sizeof(mp_obj_t)) : NULL
};
for (; cb.n_pos < n_args; cb.n_pos++) {
cb.args[cb.n_pos] = pos_args[cb.n_pos];
}
for (size_t i = cb.n_pos; cb.n_kw < n_kw_args; i++, cb.n_kw++) {
cb.args[i] = kw_args->table[cb.n_kw].key;
cb.args[i += 1] = kw_args->table[cb.n_kw].value;
}
callback = (atexit_callback_t *)m_realloc(callback,
#if MICROPY_MALLOC_USES_ALLOCATED_SIZE
callback_len *sizeof(cb), // Old size
#endif
(callback_len + 1) * sizeof(cb));
callback[callback_len++] = cb;
}
void shared_module_atexit_unregister(const mp_obj_t *func) {
for (size_t i = 0; i < callback_len; i++) {
if (callback[i].func == *func) {
callback[i].n_pos = 0;
callback[i].n_kw = 0;
callback[i].func = mp_const_none;
callback[i].args = NULL;
}
}
}
void shared_module_atexit_execute(pyexec_result_t *result) {
if (callback) {
for (size_t i = callback_len; i-- > 0;) {
if (callback[i].func != mp_const_none) {
if (result != NULL) {
pyexec_result_t res;
if (pyexec_exit_handler(&callback[i], &res) == PYEXEC_DEEP_SLEEP) {
*result = res;
}
} else {
pyexec_exit_handler(&callback[i], NULL);
}
}
}
}
}