zmk/app/src/display/widgets/bongo_cat.c

82 lines
2.2 KiB
C

/*
* Copyright (c) 2022 The ZMK Contributors
*
* SPDX-License-Identifier: MIT
*/
#include <logging/log.h>
LOG_MODULE_DECLARE(zmk, CONFIG_ZMK_LOG_LEVEL);
#include <zmk/display.h>
#include <zmk/display/widgets/bongo_cat.h>
#include <zmk/event_manager.h>
#include <zmk/events/position_state_changed.h>
static sys_slist_t widgets = SYS_SLIST_STATIC_INIT(&widgets);
enum bongo_state {
bongo_state_none, /* no hands down */
bongo_state_left, /* left hand down */
bongo_state_right, /* right hand down */
} current_bongo_state;
LV_IMG_DECLARE(none);
LV_IMG_DECLARE(left);
LV_IMG_DECLARE(right);
LV_IMG_DECLARE(both);
const void *images[] = {
&none,
&left,
&right,
&both
};
int zmk_widget_bongo_cat_init(struct zmk_widget_bongo_cat *widget, lv_obj_t *parent) {
widget->obj = lv_img_create(parent, NULL);
lv_img_set_src(widget->obj, &none);
current_bongo_state = bongo_state_none;
sys_slist_append(&widgets, &widget->node);
return 0;
}
lv_obj_t *zmk_widget_bongo_cat_obj(struct zmk_widget_bongo_cat *widget) {
return widget->obj;
}
void set_bongo_state(struct zmk_widget_bongo_cat *widget, struct zmk_position_state_changed *ev) {
if (ev == NULL) {
return;
}
uint8_t tmp = bongo_state_left << widget->is_right;
if (ev->state) {
if (current_bongo_state & (bongo_state_left | bongo_state_right)) {
tmp = bongo_state_left | bongo_state_right;
}
} else {
if (current_bongo_state ^ (bongo_state_left | bongo_state_right)) {
tmp = bongo_state_none;
widget->is_right = !widget->is_right; /* switch hand when return to none state */
}
}
if (current_bongo_state == tmp) {
return;
}
current_bongo_state = tmp;
lv_img_set_src(widget->obj, images[current_bongo_state]);
}
int bongo_cat_listener(const zmk_event_t *eh) {
struct zmk_widget_bongo_cat *widget;
struct zmk_position_state_changed *ev = as_zmk_position_state_changed(eh);
SYS_SLIST_FOR_EACH_CONTAINER(&widgets, widget, node) { set_bongo_state(widget, ev); }
return ZMK_EV_EVENT_BUBBLE;
}
ZMK_LISTENER(widget_bongo_cat, bongo_cat_listener)
ZMK_SUBSCRIPTION(widget_bongo_cat, zmk_position_state_changed);