qmk_firmware

QMK firmware for my keyboards (Corne, Sweep Ferris) and trackball (Ploopy Adept)
Log | Files | Refs | Submodules | LICENSE

qp_lvgl.c (5636B)


      1 // Copyright 2022 Jose Pablo Ramirez (@jpe230)
      2 // Copyright 2022 Nick Brassel (@tzarc)
      3 // SPDX-License-Identifier: GPL-2.0-or-later
      4 
      5 #include "qp_lvgl.h"
      6 #include "timer.h"
      7 #include "deferred_exec.h"
      8 #include "lvgl.h"
      9 
     10 typedef struct lvgl_state_t {
     11     uint8_t        fnc_id; // Ideally this should be the pointer of the function to run
     12     uint16_t       delay_ms;
     13     deferred_token defer_token;
     14 } lvgl_state_t;
     15 
     16 static deferred_executor_t lvgl_executors[2] = {0}; // For lv_tick_inc and lv_task_handler
     17 static lvgl_state_t        lvgl_states[2]    = {0}; // For lv_tick_inc and lv_task_handler
     18 
     19 painter_device_t selected_display = NULL;
     20 void            *color_buffer     = NULL;
     21 
     22 ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
     23 // Quantum Painter LVGL Integration Internal: qp_lvgl_flush
     24 
     25 void qp_lvgl_flush(lv_disp_drv_t *disp, const lv_area_t *area, lv_color_t *color_p) {
     26     if (selected_display) {
     27         uint32_t number_pixels = (area->x2 - area->x1 + 1) * (area->y2 - area->y1 + 1);
     28         qp_viewport(selected_display, area->x1, area->y1, area->x2, area->y2);
     29         qp_pixdata(selected_display, (void *)color_p, number_pixels);
     30         qp_flush(selected_display);
     31         lv_disp_flush_ready(disp);
     32     }
     33 }
     34 
     35 static uint32_t tick_task_callback(uint32_t trigger_time, void *cb_arg) {
     36     lvgl_state_t   *state     = (lvgl_state_t *)cb_arg;
     37     static uint32_t last_tick = 0;
     38     switch (state->fnc_id) {
     39         case 0: {
     40             uint32_t now = timer_read32();
     41             lv_tick_inc(TIMER_DIFF_32(now, last_tick));
     42             last_tick = now;
     43         } break;
     44         case 1:
     45             lv_task_handler();
     46             break;
     47 
     48         default:
     49             break;
     50     }
     51 
     52     // The tasks should run indefinitely
     53     return state->delay_ms;
     54 }
     55 
     56 ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
     57 // Quantum Painter LVGL Integration API: qp_lvgl_attach
     58 
     59 bool qp_lvgl_attach(painter_device_t device) {
     60     qp_dprintf("qp_lvgl_start: entry\n");
     61     qp_lvgl_detach();
     62 
     63     painter_driver_t *driver = (painter_driver_t *)device;
     64     if (!driver || !driver->validate_ok) {
     65         qp_dprintf("qp_lvgl_attach: fail (validation_ok == false)\n");
     66         qp_lvgl_detach();
     67         return false;
     68     }
     69 
     70     // Setting up the tasks
     71     lvgl_state_t *lv_tick_inc_state = &lvgl_states[0];
     72     lv_tick_inc_state->fnc_id       = 0;
     73     lv_tick_inc_state->delay_ms     = 1;
     74     lv_tick_inc_state->defer_token  = defer_exec_advanced(lvgl_executors, 2, 1, tick_task_callback, lv_tick_inc_state);
     75 
     76     if (lv_tick_inc_state->defer_token == INVALID_DEFERRED_TOKEN) {
     77         qp_dprintf("qp_lvgl_attach: fail (could not set up qp_lvgl executor)\n");
     78         qp_lvgl_detach();
     79         return false;
     80     }
     81 
     82     lvgl_state_t *lv_task_handler_state = &lvgl_states[1];
     83     lv_task_handler_state->fnc_id       = 1;
     84     lv_task_handler_state->delay_ms     = QP_LVGL_TASK_PERIOD;
     85     lv_task_handler_state->defer_token  = defer_exec_advanced(lvgl_executors, 2, QP_LVGL_TASK_PERIOD, tick_task_callback, lv_task_handler_state);
     86 
     87     if (lv_task_handler_state->defer_token == INVALID_DEFERRED_TOKEN) {
     88         qp_dprintf("qp_lvgl_attach: fail (could not set up qp_lvgl executor)\n");
     89         qp_lvgl_detach();
     90         return false;
     91     }
     92 
     93     // Init LVGL
     94     lv_init();
     95 
     96     // Set up lvgl display buffer
     97     static lv_disp_draw_buf_t draw_buf;
     98     // Allocate a buffer for 1/10 screen size
     99     const size_t count_required   = driver->panel_width * driver->panel_height / 10;
    100     void        *new_color_buffer = realloc(color_buffer, sizeof(lv_color_t) * count_required);
    101     if (!new_color_buffer) {
    102         qp_dprintf("qp_lvgl_attach: fail (could not set up memory buffer)\n");
    103         qp_lvgl_detach();
    104         return false;
    105     }
    106     color_buffer = new_color_buffer;
    107     memset(color_buffer, 0, sizeof(lv_color_t) * count_required);
    108     // Initialize the display buffer.
    109     lv_disp_draw_buf_init(&draw_buf, color_buffer, NULL, count_required);
    110 
    111     selected_display = device;
    112 
    113     uint16_t panel_width, panel_height, offset_x, offset_y;
    114     qp_get_geometry(selected_display, &panel_width, &panel_height, NULL, &offset_x, &offset_y);
    115 
    116     // Setting up display driver
    117     static lv_disp_drv_t disp_drv;     /*Descriptor of a display driver*/
    118     lv_disp_drv_init(&disp_drv);       /*Basic initialization*/
    119     disp_drv.flush_cb = qp_lvgl_flush; /*Set your driver function*/
    120     disp_drv.draw_buf = &draw_buf;     /*Assign the buffer to the display*/
    121     disp_drv.hor_res  = panel_width;   /*Set the horizontal resolution of the display*/
    122     disp_drv.ver_res  = panel_height;  /*Set the vertical resolution of the display*/
    123     lv_disp_drv_register(&disp_drv);   /*Finally register the driver*/
    124 
    125     return true;
    126 }
    127 
    128 ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
    129 // Quantum Painter LVGL Integration API: qp_lvgl_detach
    130 
    131 void qp_lvgl_detach(void) {
    132     for (int i = 0; i < 2; ++i) {
    133         cancel_deferred_exec_advanced(lvgl_executors, 2, lvgl_states[i].defer_token);
    134     }
    135     if (color_buffer) {
    136         free(color_buffer);
    137         color_buffer = NULL;
    138     }
    139     selected_display = NULL;
    140 }
    141 
    142 ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
    143 // Quantum Painter LVGL Integration Internal: qp_lvgl_internal_tick
    144 
    145 void qp_lvgl_internal_tick(void) {
    146     static uint32_t last_lvgl_exec = 0;
    147     deferred_exec_advanced_task(lvgl_executors, 2, &last_lvgl_exec);
    148 }