usrsw.c 2.47 KB
Newer Older
1
2
3
4
5
6
7
8
#include <stdio.h>
#include <stm32f4xx_hal.h>

#include "misc.h"
#include "mpconfig.h"
#include "qstr.h"
#include "obj.h"
#include "runtime.h"
9
#include "extint.h"
10
#include "pin.h"
11
#include "genhdr/pins.h"
12
#include "usrsw.h"
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

// Usage Model:
//
// pyb.switch() returns True if the user switch is pressed, False otherwise.
//
// pyb.switch(callback) will register a callback to be called when the user
//                      switch is pressed.
//
// pyb.switch(None) will remove the callback.
//
// Example:
//
// def switch_pressed():
//     print("User Switch pressed")
//
// pyb.switch(switch_pressed)

static mp_obj_t switch_user_callback_obj;

static mp_obj_t switch_callback(mp_obj_t line) {
    if (switch_user_callback_obj != mp_const_none) {
Damien George's avatar
Damien George committed
34
        mp_call_function_0(switch_user_callback_obj);
35
36
37
38
39
    }
    return mp_const_none;
}
static MP_DEFINE_CONST_FUN_OBJ_1(switch_callback_obj, switch_callback);

40
41
42
// this function inits the switch GPIO so that it can be used
void switch_init0(void) {
    GPIO_InitTypeDef init;
43
    init.Pin = MICROPY_HW_USRSW_PIN.pin_mask;
44
    init.Mode = GPIO_MODE_INPUT;
45
    init.Pull = MICROPY_HW_USRSW_PULL;
46
    init.Speed = GPIO_SPEED_FAST;
47
    HAL_GPIO_Init(MICROPY_HW_USRSW_PIN.gpio, &init);
48
49
}

50
// this function inits the callback pointer
51
52
53
54
55
void switch_init(void) {
    switch_user_callback_obj = mp_const_none;
}

int switch_get(void) {
56
57
    int val = ((MICROPY_HW_USRSW_PIN.gpio->IDR & MICROPY_HW_USRSW_PIN.pin_mask) != 0);
    return val == MICROPY_HW_USRSW_PRESSED;
58
59
60
61
62
63
64
65
}

/******************************************************************************/
/* Micro Python bindings                                                      */

static mp_obj_t pyb_switch(uint n_args, mp_obj_t *args) {
    if (n_args == 0) {
        return switch_get() ? mp_const_true : mp_const_false;
66
67
68
69
70
    } else {
        switch_user_callback_obj = args[0];
        // Init the EXTI each time this function is called, since the EXTI
        // may have been disabled by an exception in the interrupt, or the
        // user disabling the line explicitly.
71
72
73
74
75
        extint_register((mp_obj_t)&MICROPY_HW_USRSW_PIN,
                        MP_OBJ_NEW_SMALL_INT(MICROPY_HW_USRSW_EXTI_MODE),
                        MP_OBJ_NEW_SMALL_INT(MICROPY_HW_USRSW_PULL),
                        switch_user_callback_obj == mp_const_none ? mp_const_none : (mp_obj_t)&switch_callback_obj,
                        true, NULL);
76
        return mp_const_none;
77
78
79
80
    }
}

MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(pyb_switch_obj, 0, 1, pyb_switch);