codal_port/modmicrobit: Add helper funcs to query button/pin types.

Signed-off-by: Damien George <damien@micropython.org>

Author: dpgeorge

src/codal_port/microbit_button.c

@@ -64,7 +64,7 @@ STATIC const mp_map_elem_t microbit_button_locals_dict_table[] = {
 
 STATIC MP_DEFINE_CONST_DICT(microbit_button_locals_dict, microbit_button_locals_dict_table);
 
-STATIC const mp_obj_type_t microbit_button_type = {
+const mp_obj_type_t microbit_button_type = {
     { &mp_type_type },
     .name = MP_QSTR_MicroBitButton,
     .locals_dict = (mp_obj_dict_t *)&microbit_button_locals_dict,
@@ -81,3 +81,8 @@ const microbit_button_obj_t microbit_button_b_obj = {
     .pin = &microbit_p11_obj,
     .button_id = 1,
 };
+
+// This function assumes "button" is of type microbit_button_type.
+uint8_t microbit_obj_get_button_id(mp_obj_t button) {
+    return ((microbit_button_obj_t *)MP_OBJ_TO_PTR(button))->button_id;
+}

src/codal_port/microbit_pin.c

@@ -266,7 +266,7 @@ const mp_obj_type_t microbit_touch_only_pin_type = {
 
 const microbit_pin_obj_t *microbit_obj_get_pin(mp_const_obj_t o) {
     const mp_obj_type_t *type = mp_obj_get_type(o);
-    if (type == &microbit_touch_pin_type || type == &microbit_ad_pin_type || type == &microbit_dig_pin_type) {
+    if (microbit_obj_type_is_pin(type)) {
         return (microbit_pin_obj_t*)o;
     } else {
         mp_raise_TypeError(MP_ERROR_TEXT("expecting a pin"));

src/codal_port/modmicrobit.h

@@ -80,6 +80,7 @@ typedef struct _microbit_soundevent_obj_t microbit_soundevent_obj_t;
 
 extern const microbit_pinmode_t microbit_pinmodes[];
 
+extern const mp_obj_type_t microbit_button_type;
 extern const mp_obj_type_t microbit_ad_pin_type;
 extern const mp_obj_type_t microbit_dig_pin_type;
 extern const mp_obj_type_t microbit_touch_pin_type;
@@ -192,6 +193,17 @@ extern const struct _microbit_microphone_obj_t microbit_microphone_obj;
 extern const struct _microbit_button_obj_t microbit_button_a_obj;
 extern const struct _microbit_button_obj_t microbit_button_b_obj;
 
+static inline bool microbit_obj_type_is_button(const mp_obj_type_t *type) {
+    return type == &microbit_button_type;
+}
+
+static inline bool microbit_obj_type_is_pin(const mp_obj_type_t *type) {
+    return type == &microbit_touch_pin_type || type == &microbit_ad_pin_type || type == &microbit_dig_pin_type;
+}
+
+// This function assumes "button" is of type microbit_button_type.
+uint8_t microbit_obj_get_button_id(mp_obj_t button);
+
 const microbit_pin_obj_t *microbit_obj_get_pin(mp_const_obj_t o);
 uint8_t microbit_obj_get_pin_name(mp_obj_t o);