summaryrefslogtreecommitdiffstats
path: root/keyboards/keychron/bluetooth/bat_level_animation.c
blob: 94a21f11e15a639f473824b81ded8b41a3b255f7 (plain)
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

#include "quantum.h"
#include "bluetooth.h"
#include "indicator.h"
#include "lpm.h"
#if defined(PROTOCOL_CHIBIOS)
#    include <usb_main.h>
#elif if defined(PROTOCOL_LUFA)
#    include "lufa.h"
#endif

#ifndef BAT_LEVEL_GROWING_INTERVAL
#    define BAT_LEVEL_GROWING_INTERVAL 150
#endif

#ifndef BAT_LEVEL_ON_INTERVAL
#    define BAT_LEVEL_ON_INTERVAL 3000
#endif

enum {
    BAT_LVL_ANI_NONE,
    BAT_LVL_ANI_GROWING,
    BAT_LVL_ANI_BLINK_OFF,
    BAT_LVL_ANI_BLINK_ON,
};

static uint8_t  animation_state;
static uint32_t bat_lvl_ani_timer_buffer;
static uint8_t  bat_percentage;
static uint8_t  cur_percentage;
static uint8_t  bat_lvl_led_list[10] = BAT_LEVEL_LED_LIST;
static uint32_t time_interval;
#ifdef RGB_MATRIX_ENABLE
static uint8_t r, g, b;
#endif

extern indicator_config_t indicator_config;
extern backlight_state_t  original_backlight_state;

void bat_level_animiation_start(uint8_t percentage) {
    /* Turn on backlight mode for indicator */
    indicator_enable();

    animation_state          = BAT_LVL_ANI_GROWING;
    bat_percentage           = percentage;
    bat_lvl_ani_timer_buffer = sync_timer_read32();
    cur_percentage           = 0;
    time_interval            = BAT_LEVEL_GROWING_INTERVAL;
#ifdef RGB_MATRIX_ENABLE
    r = g = b = 255;
#endif
}

void bat_level_animiation_stop(void) {
    animation_state = BAT_LVL_ANI_NONE;
}

bool bat_level_animiation_actived(void) {
    return animation_state;
}

void bat_level_animiation_indicate(void) {
#ifdef LED_MATRIX_ENABLE
    for (uint8_t i = 0; i <= DRIVER_LED_TOTAL; i++) {
        led_matrix_set_value(i, 0);
    }

    if (animation_state == BAT_LVL_ANI_GROWING || animation_state == BAT_LVL_ANI_BLINK_ON)
        for (uint8_t i = 0; i < cur_percentage / 10; i++)
            led_matrix_set_value(bat_lvl_led_list[i], 255);
#endif

#ifdef RGB_MATRIX_ENABLE
    for (uint8_t i = 0; i <= DRIVER_LED_TOTAL; i++) {
        rgb_matrix_set_color(i, 0, 0, 0);
    }

    if (animation_state == BAT_LVL_ANI_GROWING || animation_state == BAT_LVL_ANI_BLINK_ON)
        for (uint8_t i = 0; i < cur_percentage / 10; i++) {
            rgb_matrix_set_color(bat_lvl_led_list[i], r, g, b);
        }
#endif
}

void bat_level_animiation_update(void) {
    switch (animation_state) {
        case BAT_LVL_ANI_GROWING:
            if (cur_percentage < bat_percentage)
                cur_percentage += 10;
            else {
                if (cur_percentage == 0) cur_percentage = 10;
                animation_state = BAT_LVL_ANI_BLINK_OFF;
            }
            break;

        case BAT_LVL_ANI_BLINK_OFF:
#ifdef RGB_MATRIX_ENABLE
            if (bat_percentage < 30) {
                r = 255;
                b = g = 0;
            } else {
                r = b = 0;
                g     = 255;
            }
#endif
            time_interval   = BAT_LEVEL_ON_INTERVAL;
            animation_state = BAT_LVL_ANI_BLINK_ON;
            break;

        case BAT_LVL_ANI_BLINK_ON:
            animation_state = BAT_LVL_ANI_NONE;
            if (indicator_config.value == 0 && indicator_is_backlit_enabled_eeprom()) {
                indicator_disable();
            }
            break;

        default:
            break;
    }

    bat_lvl_ani_timer_buffer = sync_timer_read32();
}

void bat_level_animiation_task(void) {
    if (animation_state && sync_timer_elapsed32(bat_lvl_ani_timer_buffer) > time_interval) {
        bat_level_animiation_update();
    }
}