216 lines
		
	
	
	
		
			9.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			216 lines
		
	
	
	
		
			9.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
// Copyright 2023 Nick Brassel (@tzarc)
 | 
						|
// SPDX-License-Identifier: GPL-2.0-or-later
 | 
						|
 | 
						|
#include "qp_internal.h"
 | 
						|
#include "qp_comms.h"
 | 
						|
#include "qp_oled_panel.h"
 | 
						|
#include "qp_sh1106.h"
 | 
						|
#include "qp_sh1106_opcodes.h"
 | 
						|
#include "qp_surface.h"
 | 
						|
#include "qp_surface_internal.h"
 | 
						|
 | 
						|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 | 
						|
// Driver storage
 | 
						|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 | 
						|
 | 
						|
typedef struct sh1106_device_t {
 | 
						|
    oled_panel_painter_device_t oled;
 | 
						|
 | 
						|
    uint8_t framebuffer[SURFACE_REQUIRED_BUFFER_BYTE_SIZE(128, 64, 1)];
 | 
						|
} sh1106_device_t;
 | 
						|
 | 
						|
static sh1106_device_t sh1106_drivers[SH1106_NUM_DEVICES] = {0};
 | 
						|
 | 
						|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 | 
						|
// Quantum Painter API implementations
 | 
						|
 | 
						|
// Initialisation
 | 
						|
__attribute__((weak)) bool qp_sh1106_init(painter_device_t device, painter_rotation_t rotation) {
 | 
						|
    sh1106_device_t *driver = (sh1106_device_t *)device;
 | 
						|
 | 
						|
    // Change the surface geometry based on the panel rotation
 | 
						|
    if (rotation == QP_ROTATION_90 || rotation == QP_ROTATION_270) {
 | 
						|
        driver->oled.surface.base.panel_width  = driver->oled.base.panel_height;
 | 
						|
        driver->oled.surface.base.panel_height = driver->oled.base.panel_width;
 | 
						|
    } else {
 | 
						|
        driver->oled.surface.base.panel_width  = driver->oled.base.panel_width;
 | 
						|
        driver->oled.surface.base.panel_height = driver->oled.base.panel_height;
 | 
						|
    }
 | 
						|
 | 
						|
    // Init the internal surface
 | 
						|
    if (!qp_init(&driver->oled.surface.base, QP_ROTATION_0)) {
 | 
						|
        qp_dprintf("Failed to init internal surface in qp_sh1106_init\n");
 | 
						|
        return false;
 | 
						|
    }
 | 
						|
 | 
						|
    // clang-format off
 | 
						|
    uint8_t sh1106_init_sequence[] = {
 | 
						|
        // Command,                 Delay,  N, Data[N]
 | 
						|
        SH1106_SET_MUX_RATIO,           0,  1, 0x3F,
 | 
						|
        SH1106_DISPLAY_OFFSET,          0,  1, 0x00,
 | 
						|
        SH1106_DISPLAY_START_LINE,      0,  0,
 | 
						|
        SH1106_SET_SEGMENT_REMAP_INV,   0,  0,
 | 
						|
        SH1106_COM_SCAN_DIR_DEC,        0,  0,
 | 
						|
        SH1106_COM_PADS_HW_CFG,         0,  1, 0x12,
 | 
						|
        SH1106_SET_CONTRAST,            0,  1, 0x7F,
 | 
						|
        SH1106_ALL_ON_RESUME,           0,  0,
 | 
						|
        SH1106_NON_INVERTING_DISPLAY,   0,  0,
 | 
						|
        SH1106_SET_OSC_DIVFREQ,         0,  1, 0x80,
 | 
						|
        SH1106_SET_CHARGE_PUMP,         0,  1, 0x14,
 | 
						|
        SH1106_DISPLAY_ON,              0,  0,
 | 
						|
    };
 | 
						|
    // clang-format on
 | 
						|
 | 
						|
    // If the display height is anything other than the default 64 pixels, change SH1106_SET_MUX_RATIO data byte to the correct value
 | 
						|
    if (driver->oled.base.panel_height != 64) {
 | 
						|
        sh1106_init_sequence[3] = driver->oled.base.panel_height - 1;
 | 
						|
    }
 | 
						|
 | 
						|
    // For 128x32 or 96x16 displays, change SH1106_COM_PADS_HW_CFG data byte from alternative (0x12) to sequential (0x02) configuration
 | 
						|
    if (driver->oled.base.panel_height <= 32) {
 | 
						|
        sh1106_init_sequence[20] = 0x02;
 | 
						|
    }
 | 
						|
 | 
						|
    qp_comms_bulk_command_sequence(device, sh1106_init_sequence, sizeof(sh1106_init_sequence));
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
// Screen flush
 | 
						|
bool qp_sh1106_flush(painter_device_t device) {
 | 
						|
    sh1106_device_t *driver = (sh1106_device_t *)device;
 | 
						|
 | 
						|
    if (!driver->oled.surface.dirty.is_dirty) {
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
 | 
						|
    switch (driver->oled.base.rotation) {
 | 
						|
        default:
 | 
						|
        case QP_ROTATION_0:
 | 
						|
            qp_oled_panel_page_column_flush_rot0(device, &driver->oled.surface.dirty, driver->framebuffer);
 | 
						|
            break;
 | 
						|
        case QP_ROTATION_90:
 | 
						|
            qp_oled_panel_page_column_flush_rot90(device, &driver->oled.surface.dirty, driver->framebuffer);
 | 
						|
            break;
 | 
						|
        case QP_ROTATION_180:
 | 
						|
            qp_oled_panel_page_column_flush_rot180(device, &driver->oled.surface.dirty, driver->framebuffer);
 | 
						|
            break;
 | 
						|
        case QP_ROTATION_270:
 | 
						|
            qp_oled_panel_page_column_flush_rot270(device, &driver->oled.surface.dirty, driver->framebuffer);
 | 
						|
            break;
 | 
						|
    }
 | 
						|
 | 
						|
    // Clear the dirty area
 | 
						|
    qp_flush(&driver->oled.surface);
 | 
						|
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 | 
						|
// Driver vtable
 | 
						|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 | 
						|
 | 
						|
const oled_panel_painter_driver_vtable_t sh1106_driver_vtable = {
 | 
						|
    .base =
 | 
						|
        {
 | 
						|
            .init            = qp_sh1106_init,
 | 
						|
            .power           = qp_oled_panel_power,
 | 
						|
            .clear           = qp_oled_panel_clear,
 | 
						|
            .flush           = qp_sh1106_flush,
 | 
						|
            .pixdata         = qp_oled_panel_passthru_pixdata,
 | 
						|
            .viewport        = qp_oled_panel_passthru_viewport,
 | 
						|
            .palette_convert = qp_oled_panel_passthru_palette_convert,
 | 
						|
            .append_pixels   = qp_oled_panel_passthru_append_pixels,
 | 
						|
            .append_pixdata  = qp_oled_panel_passthru_append_pixdata,
 | 
						|
        },
 | 
						|
    .opcodes =
 | 
						|
        {
 | 
						|
            .display_on     = SH1106_DISPLAY_ON,
 | 
						|
            .display_off    = SH1106_DISPLAY_OFF,
 | 
						|
            .set_page       = SH1106_PAGE_ADDR,
 | 
						|
            .set_column_lsb = SH1106_SETCOLUMN_LSB,
 | 
						|
            .set_column_msb = SH1106_SETCOLUMN_MSB,
 | 
						|
        },
 | 
						|
};
 | 
						|
 | 
						|
#ifdef QUANTUM_PAINTER_SH1106_SPI_ENABLE
 | 
						|
// Factory function for creating a handle to the SH1106 device
 | 
						|
painter_device_t qp_sh1106_make_spi_device(uint16_t panel_width, uint16_t panel_height, pin_t chip_select_pin, pin_t dc_pin, pin_t reset_pin, uint16_t spi_divisor, int spi_mode) {
 | 
						|
    for (uint32_t i = 0; i < SH1106_NUM_DEVICES; ++i) {
 | 
						|
        sh1106_device_t *driver = &sh1106_drivers[i];
 | 
						|
        if (!driver->oled.base.driver_vtable) {
 | 
						|
            painter_device_t surface = qp_make_mono1bpp_surface_advanced(&driver->oled.surface, 1, panel_width, panel_height, driver->framebuffer);
 | 
						|
            if (!surface) {
 | 
						|
                return NULL;
 | 
						|
            }
 | 
						|
 | 
						|
            // Setup the OLED device
 | 
						|
            driver->oled.base.driver_vtable         = (const painter_driver_vtable_t *)&sh1106_driver_vtable;
 | 
						|
            driver->oled.base.comms_vtable          = (const painter_comms_vtable_t *)&spi_comms_with_dc_vtable;
 | 
						|
            driver->oled.base.native_bits_per_pixel = 1; // 1bpp mono
 | 
						|
            driver->oled.base.panel_width           = panel_width;
 | 
						|
            driver->oled.base.panel_height          = panel_height;
 | 
						|
            driver->oled.base.rotation              = QP_ROTATION_0;
 | 
						|
            driver->oled.base.offset_x              = 0;
 | 
						|
            driver->oled.base.offset_y              = 0;
 | 
						|
 | 
						|
            // SPI and other pin configuration
 | 
						|
            driver->oled.base.comms_config                                   = &driver->oled.spi_dc_reset_config;
 | 
						|
            driver->oled.spi_dc_reset_config.spi_config.chip_select_pin      = chip_select_pin;
 | 
						|
            driver->oled.spi_dc_reset_config.spi_config.divisor              = spi_divisor;
 | 
						|
            driver->oled.spi_dc_reset_config.spi_config.lsb_first            = false;
 | 
						|
            driver->oled.spi_dc_reset_config.spi_config.mode                 = spi_mode;
 | 
						|
            driver->oled.spi_dc_reset_config.dc_pin                          = dc_pin;
 | 
						|
            driver->oled.spi_dc_reset_config.reset_pin                       = reset_pin;
 | 
						|
            driver->oled.spi_dc_reset_config.command_params_uses_command_pin = true;
 | 
						|
 | 
						|
            if (!qp_internal_register_device((painter_device_t)driver)) {
 | 
						|
                memset(driver, 0, sizeof(sh1106_device_t));
 | 
						|
                return NULL;
 | 
						|
            }
 | 
						|
 | 
						|
            return (painter_device_t)driver;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
#endif // QUANTUM_PAINTER_SH1106_SPI_ENABLE
 | 
						|
 | 
						|
#ifdef QUANTUM_PAINTER_SH1106_I2C_ENABLE
 | 
						|
// Factory function for creating a handle to the SH1106 device
 | 
						|
painter_device_t qp_sh1106_make_i2c_device(uint16_t panel_width, uint16_t panel_height, uint8_t i2c_address) {
 | 
						|
    for (uint32_t i = 0; i < SH1106_NUM_DEVICES; ++i) {
 | 
						|
        sh1106_device_t *driver = &sh1106_drivers[i];
 | 
						|
        if (!driver->oled.base.driver_vtable) {
 | 
						|
            // Instantiate the surface, intentional swap of width/high due to transpose
 | 
						|
            painter_device_t surface = qp_make_mono1bpp_surface_advanced(&driver->oled.surface, 1, panel_width, panel_height, driver->framebuffer);
 | 
						|
            if (!surface) {
 | 
						|
                return NULL;
 | 
						|
            }
 | 
						|
 | 
						|
            // Setup the OLED device
 | 
						|
            driver->oled.base.driver_vtable         = (const painter_driver_vtable_t *)&sh1106_driver_vtable;
 | 
						|
            driver->oled.base.comms_vtable          = (const painter_comms_vtable_t *)&i2c_comms_cmddata_vtable;
 | 
						|
            driver->oled.base.native_bits_per_pixel = 1; // 1bpp mono
 | 
						|
            driver->oled.base.panel_width           = panel_width;
 | 
						|
            driver->oled.base.panel_height          = panel_height;
 | 
						|
            driver->oled.base.rotation              = QP_ROTATION_0;
 | 
						|
            driver->oled.base.offset_x              = 0;
 | 
						|
            driver->oled.base.offset_y              = 0;
 | 
						|
 | 
						|
            // I2C configuration
 | 
						|
            driver->oled.base.comms_config       = &driver->oled.i2c_config;
 | 
						|
            driver->oled.i2c_config.chip_address = i2c_address;
 | 
						|
 | 
						|
            if (!qp_internal_register_device((painter_device_t)driver)) {
 | 
						|
                memset(driver, 0, sizeof(sh1106_device_t));
 | 
						|
                return NULL;
 | 
						|
            }
 | 
						|
 | 
						|
            return (painter_device_t)driver;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
#endif // QUANTUM_PAINTER_SH1106_I2C_ENABLE
 |