qmk_firmware/keyboards/tzarc/ghoul/keymaps/default/ui.c

140 lines
5.6 KiB
C

// 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
}