2026-05-12 23:40:22 +02:00

352 lines
9.4 KiB
C

#include <u8g2.h>
#include <ch32fun.h>
#include <stdio.h>
#include "display.h"
#include "lib_i2c.h"
static u8g2_t u8g2;
static const char digits_lut[] =
"0001020304050607080910111213141516171819"
"2021222324252627282930313233343536373839"
"4041424344454647484950515253545556575859"
"6061626364656667686970717273747576777879"
"8081828384858687888990919293949596979899";
// GPIO and delay callback for u8g2/u8x8 display driver
static uint8_t u8x8_gpio_and_delay(u8x8_t *u8x8, uint8_t msg, uint8_t arg_int, void *arg_ptr)
{
switch(msg) {
case U8X8_MSG_GPIO_AND_DELAY_INIT:
// called once during init phase of u8g2/u8x8
// can be used to setup pins
printf("TODO: U8X8_MSG_GPIO_AND_DELAY_INIT\n");
break;
case U8X8_MSG_DELAY_NANO:
// delay arg_int * 1 nano second
// at 48MHz 1 cycle = 20ns, 1 nop = 1 cycle
// the for lopp takes about 2 cycles per iteration
for (int i = 0; i < (arg_int/(20*3)); i++) {
asm("nop");
}
break;
case U8X8_MSG_DELAY_100NANO:
// delay arg_int * 100 nano seconds
// at 48MHz 1 cycle = 20ns, 1 nop = 1 cycle
// the for lopp takes about 2 cycles per iteration
for (int i = 0; i < ((arg_int*100)/(20*3)); i++) {
asm("nop");
}
break;
case U8X8_MSG_DELAY_10MICRO:
// delay arg_int * 10 micro seconds
Delay_Us(arg_int*10);
break;
case U8X8_MSG_DELAY_MILLI:
// delay arg_int * 1 milli second
Delay_Ms(arg_int);
break;
case U8X8_MSG_DELAY_I2C:
// arg_int is the I2C speed in 100KHz, e.g. 4 = 400 KHz
// arg_int=1: delay by 5us, arg_int = 4: delay by 1.25us
printf("TODO: U8X8_MSG_DELAY_I2C\n");
break;
case U8X8_MSG_GPIO_D0:
// D0 or SPI clock pin: Output level in arg_int
//case U8X8_MSG_GPIO_SPI_CLOCK:
break;
case U8X8_MSG_GPIO_D1:
// D1 or SPI data pin: Output level in arg_int
//case U8X8_MSG_GPIO_SPI_DATA:
break;
case U8X8_MSG_GPIO_D2:
// D2 pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_D3:
// D3 pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_D4:
// D4 pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_D5:
// D5 pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_D6:
// D6 pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_D7:
// D7 pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_E:
// E/WR pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_CS:
// CS (chip select) pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_DC:
// DC (data/cmd, A0, register select) pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_RESET:
funDigitalWrite(PIN_DISP_RST, arg_int);
// Reset pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_CS1:
// CS1 (chip select) pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_CS2:
// CS2 (chip select) pin: Output level in arg_int
break;
case U8X8_MSG_GPIO_I2C_CLOCK:
// arg_int=0: Output low at I2C clock pin
// arg_int=1: Input dir with pullup high for I2C clock pin
break;
case U8X8_MSG_GPIO_I2C_DATA:
// arg_int=0: Output low at I2C data pin
// arg_int=1: Input dir with pullup high for I2C data pin
break;
case U8X8_MSG_GPIO_MENU_SELECT:
// get menu select pin state
u8x8_SetGPIOResult(u8x8, 0);
break;
case U8X8_MSG_GPIO_MENU_NEXT:
// get menu next pin state
u8x8_SetGPIOResult(u8x8, 0);
break;
case U8X8_MSG_GPIO_MENU_PREV:
// get menu prev pin state
u8x8_SetGPIOResult(u8x8, 0);
break;
case U8X8_MSG_GPIO_MENU_HOME:
// get menu home pin state
u8x8_SetGPIOResult(u8x8, 0);
break;
default:
// default return value
u8x8_SetGPIOResult(u8x8, 1);
break;
}
return 1;
}
static uint8_t u8x8_byte_i2c(u8x8_t *u8x8, uint8_t msg, uint8_t arg_int, void *arg_ptr)
{
static uint8_t buffer[32];
static uint8_t buffer_idx;
uint8_t *data;
switch(msg) {
case U8X8_MSG_BYTE_INIT:
break;
case U8X8_MSG_BYTE_SEND:
data = (uint8_t*)arg_ptr;
while (arg_int > 0) {
buffer[buffer_idx++] = *data;
data++;
arg_int--;
}
break;
case U8X8_MSG_BYTE_START_TRANSFER:
buffer_idx = 0;
break;
case U8X8_MSG_BYTE_END_TRANSFER:
i2c_sendBytes(I2C_TARGET, u8x8_GetI2CAddress(u8x8) >> 1, buffer, buffer_idx);
break;
default:
return 0;
}
return 1;
}
u8g2_t* display_init(void)
{
#if defined(SSD1306_128X32) && SSD1306_128X32
u8g2_Setup_ssd1306_i2c_128x32_univision_f(&u8g2, U8G2_R0, u8x8_byte_i2c, u8x8_gpio_and_delay);
#elif defined(SSD1312_96X16) && SSD1312_96X16
// NOTE: display size is wrong, hardware is 96x16, but driver is configured for 120x28
u8g2_Setup_ssd1312_i2c_128x32_f(&u8g2, U8G2_R0, u8x8_byte_i2c, u8x8_gpio_and_delay);
#else
static_assert(0, "unsupported display size");
#endif
// TODO: log errors and return NULL on failure
u8g2_InitDisplay(&u8g2);
u8g2_SetPowerSave(&u8g2, 0);
u8g2_SetContrast(&u8g2, 255);
u8g2_SetBitmapMode(&u8g2, 1);
u8g2_SetFontMode(&u8g2, 1);
return &u8g2;
}
const char* u16toa(uint16_t value)
{
// Max uint16_t is 65535 (5 digits) + null terminator
static char buf[6];
char *p = &buf[5];
*p = '\0';
// Process two digits at a time using the LUT
while (value >= 100) {
const unsigned int idx = (value % 100) * 2;
value /= 100;
*--p = digits_lut[idx + 1];
*--p = digits_lut[idx];
}
// Handle the remaining value (< 100)
if (value < 10) {
*--p = (char)('0' + value);
} else {
const unsigned int idx = value * 2;
*--p = digits_lut[idx + 1];
*--p = digits_lut[idx];
}
return p;
}
const char* i16toa(int16_t value)
{
static char buf[7];
uint16_t uval;
bool negative = false;
if (value < 0) {
negative = true;
uval = (uint16_t)-value;
} else {
uval = (uint16_t)value;
}
char *p = &buf[6];
*p = '\0';
// Same LUT logic as unsigned
while (uval >= 100) {
const unsigned int idx = (uval % 100) * 2;
uval /= 100;
*--p = digits_lut[idx + 1];
*--p = digits_lut[idx];
}
if (uval < 10) {
*--p = (char)('0' + uval);
} else {
const unsigned int idx = uval * 2;
*--p = digits_lut[idx + 1];
*--p = digits_lut[idx];
}
if (negative) {
*--p = '-';
}
return p;
}
static char str_buf[32];
uint8_t str_len;
int buf_putc(char c) { str_buf[str_len++] = c; return 0; }
const char* buf_get(void) { str_buf[str_len] = '\0'; str_len = 0; return str_buf; }
// 7 segment display segments
static const uint8_t seg_left_xbm[] = {0x01,0x03,0x03,0x03,0x03,0x01};
static const uint8_t seg_right_xbm[] = {0x02,0x03,0x03,0x03,0x03,0x02};
static const uint8_t seg_top_xbm[] = {0x3f,0x1e};
static const uint8_t seg_bot_xbm[] = {0x1e,0x3f};
static const uint8_t seg_center_xbm[] = {0x1e,0x3f,0x1e};
static void draw_7seg(u8g2_t *u8g2, uint8_t x, uint8_t y, uint8_t n)
{
switch (n) {
case 0:
u8g2_DrawXBM(u8g2, x+1, y+0, 6, 2, seg_top_xbm);
u8g2_DrawXBM(u8g2, x+6, y+1, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+6, y+8, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+1, y+13, 6, 2, seg_bot_xbm);
u8g2_DrawXBM(u8g2, x+0, y+1, 2, 6, seg_left_xbm);
u8g2_DrawXBM(u8g2, x+0, y+8, 2, 6, seg_left_xbm);
break;
case 1:
u8g2_DrawXBM(u8g2, x+6, y+1, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+6, y+8, 2, 6, seg_right_xbm);
break;
case 2:
u8g2_DrawXBM(u8g2, x+1, y+0, 6, 2, seg_top_xbm);
u8g2_DrawXBM(u8g2, x+6, y+1, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+1, y+13, 6, 2, seg_bot_xbm);
u8g2_DrawXBM(u8g2, x+0, y+8, 2, 6, seg_left_xbm);
u8g2_DrawXBM(u8g2, x+1, y+6, 6, 3, seg_center_xbm);
break;
case 3:
u8g2_DrawXBM(u8g2, x+1, y+0, 6, 2, seg_top_xbm);
u8g2_DrawXBM(u8g2, x+6, y+1, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+6, y+8, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+1, y+13, 6, 2, seg_bot_xbm);
u8g2_DrawXBM(u8g2, x+1, y+6, 6, 3, seg_center_xbm);
break;
case 4:
u8g2_DrawXBM(u8g2, x+6, y+1, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+6, y+8, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+0, y+1, 2, 6, seg_left_xbm);
u8g2_DrawXBM(u8g2, x+1, y+6, 6, 3, seg_center_xbm);
break;
case 5:
u8g2_DrawXBM(u8g2, x+1, y+0, 6, 2, seg_top_xbm);
u8g2_DrawXBM(u8g2, x+6, y+8, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+1, y+13, 6, 2, seg_bot_xbm);
u8g2_DrawXBM(u8g2, x+0, y+1, 2, 6, seg_left_xbm);
u8g2_DrawXBM(u8g2, x+1, y+6, 6, 3, seg_center_xbm);
break;
case 6:
u8g2_DrawXBM(u8g2, x+1, y+0, 6, 2, seg_top_xbm);
u8g2_DrawXBM(u8g2, x+6, y+8, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+1, y+13, 6, 2, seg_bot_xbm);
u8g2_DrawXBM(u8g2, x+0, y+1, 2, 6, seg_left_xbm);
u8g2_DrawXBM(u8g2, x+0, y+8, 2, 6, seg_left_xbm);
u8g2_DrawXBM(u8g2, x+1, y+6, 6, 3, seg_center_xbm);
break;
case 7:
u8g2_DrawXBM(u8g2, x+1, y+0, 6, 2, seg_top_xbm);
u8g2_DrawXBM(u8g2, x+6, y+1, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+6, y+8, 2, 6, seg_right_xbm);
break;
case 8:
u8g2_DrawXBM(u8g2, x+1, y+0, 6, 2, seg_top_xbm);
u8g2_DrawXBM(u8g2, x+6, y+1, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+6, y+8, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+1, y+13, 6, 2, seg_bot_xbm);
u8g2_DrawXBM(u8g2, x+0, y+1, 2, 6, seg_left_xbm);
u8g2_DrawXBM(u8g2, x+0, y+8, 2, 6, seg_left_xbm);
u8g2_DrawXBM(u8g2, x+1, y+6, 6, 3, seg_center_xbm);
break;
case 9:
u8g2_DrawXBM(u8g2, x+1, y+0, 6, 2, seg_top_xbm);
u8g2_DrawXBM(u8g2, x+6, y+1, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+6, y+8, 2, 6, seg_right_xbm);
u8g2_DrawXBM(u8g2, x+1, y+13, 6, 2, seg_bot_xbm);
u8g2_DrawXBM(u8g2, x+0, y+1, 2, 6, seg_left_xbm);
u8g2_DrawXBM(u8g2, x+1, y+6, 6, 3, seg_center_xbm);
break;
default:
break;
}
}
void draw_temp(u8g2_t *u8g2, uint8_t x, uint8_t y, int16_t temp)
{
if (temp > 999 || temp < 0) temp = 0;
const char *s = u8x8_u16toa(temp, 3);
draw_7seg(u8g2, x+0, y+0, s[0]-'0');
draw_7seg(u8g2, x+9, y+0, s[1]-'0');
draw_7seg(u8g2, x+18, y+0, s[2]-'0');
}