bongo_cat.c

/*
 * Copyright (c) 2024 The ZMK Contributors
 *
 * SPDX-License-Identifier: MIT
 */

#include <zephyr/kernel.h>
#include <zephyr/bluetooth/services/bas.h>

#include <zephyr/logging/log.h>
LOG_MODULE_DECLARE(zmk, CONFIG_ZMK_LOG_LEVEL);

#include <zmk/display.h>
#include <zmk/event_manager.h>
#include <zmk/events/wpm_state_changed.h>
#include <zmk/wpm.h>

#include "bongo_cat.h"

#define SRC(array) (const void **)array, sizeof(array) / sizeof(lv_img_dsc_t *)

static sys_slist_t widgets = SYS_SLIST_STATIC_INIT(&widgets);

/*
const lv_img_dsc_t *5_output_images_rotate_flip_images[17] = {
    &bongo_cat_double_tap1_01,
    &bongo_cat_double_tap1_02,
    &bongo_cat_double_tap1_03,
    &bongo_cat_double_tap1_04,
    &bongo_cat_double_tap1_05,
    &bongo_cat_double_tap1_06,
    &bongo_cat_double_tap2_01,
    &bongo_cat_double_tap2_02,
    &bongo_cat_double_tap2_03,
    &bongo_cat_tap1_01,
    &bongo_cat_tap1_02,
    &bongo_cat_tap1_03,
    &bongo_cat_tap1_04,
    &bongo_cat_tap2_01,
    &bongo_cat_tap2_02,
    &bongo_cat_tap2_03,
    &bongo_cat_tap2_04
};
*/

LV_IMG_DECLARE(bongo_cat_double_tap1_01);
LV_IMG_DECLARE(bongo_cat_double_tap1_02);
LV_IMG_DECLARE(bongo_cat_double_tap1_03);
LV_IMG_DECLARE(bongo_cat_double_tap1_04);
LV_IMG_DECLARE(bongo_cat_double_tap1_05);
LV_IMG_DECLARE(bongo_cat_double_tap1_06);
LV_IMG_DECLARE(bongo_cat_double_tap2_01);
LV_IMG_DECLARE(bongo_cat_double_tap2_02);
LV_IMG_DECLARE(bongo_cat_double_tap2_03);
LV_IMG_DECLARE(bongo_cat_tap1_01);
LV_IMG_DECLARE(bongo_cat_tap1_02);
LV_IMG_DECLARE(bongo_cat_tap1_03);
LV_IMG_DECLARE(bongo_cat_tap1_04);
LV_IMG_DECLARE(bongo_cat_tap2_01);
LV_IMG_DECLARE(bongo_cat_tap2_02);
LV_IMG_DECLARE(bongo_cat_tap2_03);
LV_IMG_DECLARE(bongo_cat_tap2_04);

#define ANIMATION_SPEED_IDLE 10000
const lv_img_dsc_t *idle_imgs[] = {
    &bongo_cat_double_tap1_06,
};

#define ANIMATION_SPEED_SLOW 2000
const lv_img_dsc_t *slow_imgs[] = {
    &bongo_cat_tap1_03,
    &bongo_cat_tap2_03,
};

#define ANIMATION_SPEED_MID 500
const lv_img_dsc_t *mid_imgs[] = {
    &bongo_cat_tap1_03,
    &bongo_cat_tap2_03,
};

// The fast state dominates CPU usage while typing, so it is the one exposed
// through Kconfig (see docs/OPTIMIZE.md).
#define ANIMATION_SPEED_FAST CONFIG_NICE_OLED_WIDGET_WPM_BONGO_CAT_ANIMATION_MS
const lv_img_dsc_t *fast_imgs[] = {
    &bongo_cat_double_tap2_02,
    &bongo_cat_double_tap1_03,
};

struct wpm_bongo_cat_status_state {
    uint8_t wpm;
};

enum anim_state {
    anim_state_none,
    anim_state_idle,
    anim_state_slow,
    anim_state_mid,
    anim_state_fast
} current_anim_state;

static void set_animation(lv_obj_t *animing, struct wpm_bongo_cat_status_state state) {
    if (state.wpm < 5) {
        if (current_anim_state != anim_state_idle) {
            lv_animimg_set_src(animing, SRC(idle_imgs));
            lv_animimg_set_duration(animing, ANIMATION_SPEED_IDLE);
            lv_animimg_set_repeat_count(animing, LV_ANIM_REPEAT_INFINITE);
            lv_animimg_start(animing);
            current_anim_state = anim_state_idle;
        }
    } else if (state.wpm < 30) {
        if (current_anim_state != anim_state_slow) {
            lv_animimg_set_src(animing, SRC(slow_imgs));
            lv_animimg_set_duration(animing, ANIMATION_SPEED_SLOW);
            lv_animimg_set_repeat_count(animing, LV_ANIM_REPEAT_INFINITE);
            lv_animimg_start(animing);
            current_anim_state = anim_state_slow;
        }
    } else if (state.wpm < 70) {
        if (current_anim_state != anim_state_mid) {
            lv_animimg_set_src(animing, SRC(mid_imgs));
            lv_animimg_set_duration(animing, ANIMATION_SPEED_MID);
            lv_animimg_set_repeat_count(animing, LV_ANIM_REPEAT_INFINITE);
            lv_animimg_start(animing);
            current_anim_state = anim_state_mid;
        }
    } else {
        if (current_anim_state != anim_state_fast) {
            lv_animimg_set_src(animing, SRC(fast_imgs));
            lv_animimg_set_duration(animing, ANIMATION_SPEED_FAST);
            lv_animimg_set_repeat_count(animing, LV_ANIM_REPEAT_INFINITE);
            lv_animimg_start(animing);
            current_anim_state = anim_state_fast;
        }
    }
}

struct wpm_bongo_cat_status_state wpm_bongo_cat_status_get_state(const zmk_event_t *eh) {
    struct zmk_wpm_state_changed *ev = as_zmk_wpm_state_changed(eh);
    return (struct wpm_bongo_cat_status_state){.wpm = ev->state};
};

void wpm_bongo_cat_status_update_cb(struct wpm_bongo_cat_status_state state) {
    struct zmk_widget_wpm_bongo_cat *widget;
    SYS_SLIST_FOR_EACH_CONTAINER(&widgets, widget, node) { set_animation(widget->obj, state); }
}

ZMK_DISPLAY_WIDGET_LISTENER(widget_wpm_bongo_cat, struct wpm_bongo_cat_status_state,
                            wpm_bongo_cat_status_update_cb, wpm_bongo_cat_status_get_state)

ZMK_SUBSCRIPTION(widget_wpm_bongo_cat, zmk_wpm_state_changed);

int zmk_widget_wpm_bongo_cat_init(struct zmk_widget_wpm_bongo_cat *widget, lv_obj_t *parent) {
    widget->obj = lv_animimg_create(parent);
    lv_obj_center(widget->obj);

    sys_slist_append(&widgets, &widget->node);

    widget_wpm_bongo_cat_init();

    return 0;
}

lv_obj_t *zmk_widget_wpm_bongo_cat_obj(struct zmk_widget_wpm_bongo_cat *widget) {
    return widget->obj;
}