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
129
130
131
132
133
134
135
136
137
138
139
|
// Copyright 2018-2022 Nick Brassel (@tzarc)
// SPDX-License-Identifier: GPL-3.0-or-later
#include <stdio.h>
#include QMK_KEYBOARD_H
#include "analog.h"
#include "qp.h"
#include "qp_ssd1351.h"
#define NUM_ADC_READS 32
#include "graphics/ghoul-logo.qgf.c"
#include "graphics/ghoul-name.qgf.c"
#include "graphics/lock-caps.qgf.c"
#include "graphics/lock-num.qgf.c"
#include "graphics/lock-scrl.qgf.c"
#include "graphics/thintel15.qff.c"
static painter_device_t oled;
static painter_image_handle_t logo;
static painter_image_handle_t name;
static painter_font_handle_t font;
static painter_image_handle_t lock_caps;
static painter_image_handle_t lock_num;
static painter_image_handle_t lock_scrl;
void ui_init(void) {
oled = qp_ssd1351_make_spi_device(128, 128, OLED_CS_PIN, OLED_DC_PIN, OLED_RST_PIN, 8, 0);
logo = qp_load_image_mem(gfx_ghoul_logo);
name = qp_load_image_mem(gfx_ghoul_name);
font = qp_load_font_mem(font_thintel15);
lock_caps = qp_load_image_mem(gfx_lock_caps);
lock_num = qp_load_image_mem(gfx_lock_num);
lock_scrl = qp_load_image_mem(gfx_lock_scrl);
qp_init(oled, QP_ROTATION_90);
qp_rect(oled, 0, 0, 127, 127, 0, 0, 0, true);
qp_flush(oled);
}
void ui_task(void) {
bool hue_redraw = false;
static uint16_t last_hue = 0xFFFF;
uint8_t curr_hue = rgblight_get_hue();
if (last_hue != curr_hue) {
last_hue = curr_hue;
hue_redraw = true;
}
if (hue_redraw) {
qp_drawimage_recolor(oled, 0, 64 - (name->height / 2), name, curr_hue, 255, 255, curr_hue, 255, 0);
qp_drawimage_recolor(oled, 127 - logo->width, 0, logo, curr_hue, 255, 255, curr_hue, 255, 0);
}
static led_t last_led_state = {0};
if (hue_redraw || last_led_state.raw != host_keyboard_led_state().raw) {
last_led_state.raw = host_keyboard_led_state().raw;
qp_drawimage_recolor(oled, lock_caps->width * 0, 0, lock_caps, curr_hue, 255, last_led_state.caps_lock ? 255 : 32, curr_hue, 255, 0);
qp_drawimage_recolor(oled, lock_caps->width * 1, 0, lock_num, curr_hue, 255, last_led_state.num_lock ? 255 : 32, curr_hue, 255, 0);
qp_drawimage_recolor(oled, lock_caps->width * 2, 0, lock_scrl, curr_hue, 255, last_led_state.scroll_lock ? 255 : 32, curr_hue, 255, 0);
qp_rect(oled, lock_caps->width * 0 + 1, lock_caps->height + 2, lock_caps->width * 1 - 1, lock_caps->height + 3, curr_hue, 255, last_led_state.caps_lock ? 255 : 0, true);
qp_rect(oled, lock_caps->width * 1 + 1, lock_caps->height + 2, lock_caps->width * 2 - 1, lock_caps->height + 3, curr_hue, 255, last_led_state.num_lock ? 255 : 0, true);
qp_rect(oled, lock_caps->width * 2 + 1, lock_caps->height + 2, lock_caps->width * 3 - 1, lock_caps->height + 3, curr_hue, 255, last_led_state.scroll_lock ? 255 : 0, true);
}
#if HAL_USE_ADC
static int16_t current_reads[NUM_ADC_READS] = {0};
static int16_t voltage_reads[NUM_ADC_READS] = {0};
static int write_offset = 0;
static uint32_t last_read = 0;
if (timer_elapsed32(last_read) >= 1) {
// Perform the reads
int16_t current = analogReadPin(ADC_CURRENT_PIN);
int16_t voltage = analogReadPin(ADC_VOLTAGE_PIN);
int16_t current_ma = (int16_t)(((3300 * (int32_t)current) / ADC_SATURATION));
int16_t voltage_mv = (int16_t)((2 * (3300 * (int32_t)voltage)) / ADC_SATURATION);
// Duplicate the first read so that averages work
if (last_read == 0) {
for (int i = 0; i < NUM_ADC_READS; ++i) {
current_reads[i] = current_ma;
voltage_reads[i] = voltage_mv;
}
}
// Dump in the current value
current_reads[write_offset] = current_ma;
voltage_reads[write_offset] = voltage_mv;
write_offset = (write_offset + 1) % NUM_ADC_READS;
static int counter = 0;
counter = (counter + 1) % 2500;
if (counter == 0) {
dprintf("Current: %dmA (%d) -- Voltage: %dmV (%d)\n", (int)current_ma, (int)current, (int)voltage_mv, (int)voltage);
}
last_read = timer_read32();
}
static uint32_t last_draw = 0;
if (hue_redraw || timer_elapsed32(last_draw) >= 250) {
// Accumulate
int32_t total_current_ma = 0;
int32_t total_voltage_mv = 0;
for (int i = 0; i < NUM_ADC_READS; ++i) {
total_current_ma += current_reads[i];
total_voltage_mv += voltage_reads[i];
}
// Get the averages
int16_t avg_current_ma = (int16_t)(total_current_ma / NUM_ADC_READS);
int16_t avg_voltage_mv = (int16_t)(total_voltage_mv / NUM_ADC_READS);
char buf[32] = {0};
sprintf(buf, "Current: %dmA", avg_current_ma);
static int16_t maxlen_curr = 0;
int16_t len = qp_drawtext_recolor(oled, 0, 127 - (font->line_height * 2), font, buf, 0, 0, 32, 0, 0, 0);
if (len < maxlen_curr) {
qp_rect(oled, len, 127 - (font->line_height * 2), maxlen_curr, 127 - (font->line_height * 1), 0, 0, 0, true);
} else if (len > maxlen_curr) {
maxlen_curr = len;
}
static int16_t maxlen_volt = 0;
sprintf(buf, "Voltage: %dmV", avg_voltage_mv);
len = qp_drawtext_recolor(oled, 0, 127 - (font->line_height * 1), font, buf, 0, 0, 32, 0, 0, 0);
if (len < maxlen_volt) {
qp_rect(oled, len, 127 - (font->line_height * 1), maxlen_volt, 127 - (font->line_height * 0), 0, 0, 0, true);
} else if (len > maxlen_volt) {
maxlen_volt = len;
}
qp_flush(oled);
last_draw = timer_read32();
}
#endif // HAL_USE_ADC
}
|
