drumkit/src/lcd.c

/*
 * lcd.c
 *
 * Copyright (c) 2024 Thomas Buck (thomas@xythobuz.de)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * See <http://www.gnu.org/licenses/>.
 */

#include <stdio.h>
#include <string.h>

#include "hardware/i2c.h"
#include "hardware/watchdog.h"

#include "ssd1306.h"
#include "git.h"

#include "config.h"
#include "buttons.h"
#include "console.h"
#include "main.h"
#include "usb.h"
#include "lcd.h"

static i2c_inst_t *gpio_i2c_proto = i2c0;
static const uint gpio_num_proto[2] = { 0, 1 };

static i2c_inst_t *gpio_i2c_v2 = i2c0;
static const uint gpio_num_v2[2] = { 16, 17 };

#define LCD_ADDR 0x3C

static ssd1306_t disp = {0};
static bool buttons[NUM_BTNS] = {0};
static bool changed = true;
static bool found = false;

static void lcd_debug_buttons_callback(enum buttons btn, bool v) {
    buttons[btn] = v;
    changed = true;
}

void lcd_debug_buttons(void) {
    if (!found) {
        return;
    }

    buttons_callback(lcd_debug_buttons_callback);

    while (1) {
        watchdog_update();
        buttons_run();
        usb_run();
        cnsl_run();

        if (changed) {
            changed = false;

            ssd1306_clear(&disp);
            ssd1306_draw_string(&disp, 0, 0, 3, "Buttons");

            for (uint i = 0; i < NUM_BTNS; i++) {
                if ((hw_type == HW_PROTOTYPE) && (i >= BTN_D) && (i <= BTN_H)) {
                    continue;
                }

                ssd1306_draw_char(&disp,
                                  i * 12, LCD_HEIGHT - 20 - 16 - 1,
                                  2, '0' + i);

                if (buttons[i]) {
                    ssd1306_draw_square(&disp,
                                        i * 12, LCD_HEIGHT - 20 - 1,
                                        10, 20);
                } else {
                    ssd1306_draw_empty_square(&disp,
                                              i * 12, LCD_HEIGHT - 20 - 1,
                                              10, 20);
                }
            }

            ssd1306_show(&disp);
        }
    }
}

void lcd_draw_bitmap(uint8_t *data, int width, int height, int x_off, int y_off) {
    if (!found) {
        return;
    }

    ssd1306_clear(&disp);

    for (int y = 0; y < height; y++) {
        for (int x = 0; x < width; x++) {
            const uint pos = y * width + x;
            const uint bit = 7 - (pos % 8);
            if (data[pos / 8] & (1 << bit)) {
                ssd1306_draw_pixel(&disp, x + x_off, y + y_off);
            }
        }
    }

    ssd1306_show(&disp);
}

void lcd_init(void) {
    if (hw_type == HW_PROTOTYPE) {
        i2c_init(gpio_i2c_proto, 2UL * 1000UL * 1000UL);

        for (uint i = 0; i < sizeof(gpio_num_proto) / sizeof(gpio_num_proto[0]); i++) {
            gpio_set_function(gpio_num_proto[i], GPIO_FUNC_I2C);
            gpio_pull_up(gpio_num_proto[i]);
        }

        bool r = ssd1306_init(&disp,
                              LCD_WIDTH, LCD_HEIGHT,
                              LCD_ADDR, gpio_i2c_proto);
        found = r;
    } else if (hw_type == HW_V2) {
        i2c_init(gpio_i2c_v2, 2UL * 1000UL * 1000UL);

        for (uint i = 0; i < sizeof(gpio_num_v2) / sizeof(gpio_num_v2[0]); i++) {
            gpio_set_function(gpio_num_v2[i], GPIO_FUNC_I2C);
            gpio_pull_up(gpio_num_v2[i]);
        }

        bool r = ssd1306_init(&disp,
                              LCD_WIDTH, LCD_HEIGHT,
                              LCD_ADDR, gpio_i2c_v2);
        found = r;
    }
}

void lcd_draw(const char *mode, const char *val, const char *bat) {
    if (!found) {
        return;
    }

    ssd1306_clear(&disp);
    ssd1306_draw_string(&disp, 0, 0, 2, mode);
    ssd1306_draw_string(&disp, 0, 20, 4, val);
    ssd1306_draw_string(&disp, 0, LCD_HEIGHT - 8, 1, bat);
    ssd1306_show(&disp);
}

void lcd_draw_bye(void) {
    if (!found) {
        return;
    }

    ssd1306_clear(&disp);
    ssd1306_draw_string(&disp, 6, 5, 3, " Boot-");
    ssd1306_draw_string(&disp, 8, LCD_HEIGHT / 2 + 5, 3, "loader");
    ssd1306_show(&disp);
}

void lcd_draw_version(void) {
    if (!found) {
        return;
    }

    ssd1306_clear(&disp);

    ssd1306_draw_string(&disp, 0, 0, 2,
                        "LARS");

    ssd1306_draw_string(&disp, 0, FONT_HEIGHT * 2 + 1, 1,
                        "Release: " VERSION_STR);

    ssd1306_draw_string(&disp, 0, FONT_HEIGHT * 2 + 1 + FONT_HEIGHT + 1, 1,
                        __DATE__ " " __TIME__);

    if (git_IsPopulated()) {
        const char *hash = git_CommitSHA1();
        char short_hash[6 + 7 + 6 + 1] = {0};
        memcpy(short_hash, "Hash: ", 6);
        memcpy(short_hash + 6, hash, 7);
        if (git_AnyUncommittedChanges()) {
            memcpy(short_hash + 6 + 7, " dirty", 6);
        }
        ssd1306_draw_string(&disp, 0, FONT_HEIGHT * 2 + 1 + (FONT_HEIGHT + 1) * 2, 1,
                            short_hash);
    } else {
        ssd1306_draw_string(&disp, 0, FONT_HEIGHT * 2 + 1 + (FONT_HEIGHT + 1) * 2, 1,
                            "No Git Repo");
    }

    char hw_id_str[42] = {0};
    if (hw_type == HW_PROTOTYPE) {
        snprintf(hw_id_str, sizeof(hw_id_str) - 1,
                 "HW Prototype");
    } else if (hw_type == HW_V2) {
        snprintf(hw_id_str, sizeof(hw_id_str) - 1,
                 "HW V2");
    } else {
        snprintf(hw_id_str, sizeof(hw_id_str) - 1,
                 "HW unknown %X", hw_type);
    }
    ssd1306_draw_string(&disp, 0, FONT_HEIGHT * 2 + 1 + (FONT_HEIGHT + 1) * 3, 1,
                        hw_id_str);

    ssd1306_show(&disp);
}