cardboy/Firmware/main/include/display.hpp

//
// Created by Stepan Usatiuk on 02.03.2025.
//

#ifndef CB_DISPLAY_HPP
#define CB_DISPLAY_HPP

#include "config.hpp"

#include "driver/spi_master.h"

#include <array>
#include <bitset>

#include "Surface.hpp"
#include "Window.hpp"

class SMD {
public:
    static constexpr size_t kLineBytes       = DISP_WIDTH / 8;
    static constexpr size_t kLineMultiSingle = (kLineBytes + 2);
    static constexpr size_t kLineDataBytes   = kLineMultiSingle * DISP_HEIGHT + 2;

    static DMA_ATTR uint8_t dma_buf[SMD::kLineDataBytes];

    static SMD& get();
    void        clear();
    void        draw();

    void set_pixel(int x, int y, bool value) {
        assert(x >= 0 && x < DISP_WIDTH && y >= 0 && y < DISP_HEIGHT);

        unsigned lineIdx = 2 + kLineMultiSingle * y + (x / 8);
        unsigned bitIdx  = 1 << (7 - (x % 8)) % 8;

        if (value) {
            dma_buf[lineIdx] &= ~bitIdx;
        } else {
            dma_buf[lineIdx] |= bitIdx;
        }
    }

private:
    SMD();
    static inline spi_device_interface_config_t _devcfg = {
            .mode           = 0,               // SPI mode 0
            .clock_speed_hz = 2 * 1000 * 1000, // Clock out at 10 MHz
            .spics_io_num   = SPI_DISP_CS,     // CS pin
            .flags          = SPI_DEVICE_POSITIVE_CS,
            .queue_size     = 3,
            // .pre_cb = lcd_spi_pre_transfer_callback, //Specify pre-transfer callback to handle D/C line
    };
    spi_device_handle_t _spi;
    bool                _vcom = false;
};

class SMDSurface : public Surface<SMDSurface, BwPixel>, public StandardEventQueue<SMDSurface> {
public:
    using PixelType = BwPixel;

    SMDSurface(EventLoop* loop);

    ~SMDSurface() override;

    void draw_pixel_impl(unsigned x, unsigned y, const BwPixel& pixel);

    void clear_impl();

    int get_width_impl() const;

    int get_height_impl() const;

    template<typename T>
    EventHandlingResult handle(const T& event) {
        return _window->handle(event);
    }

    EventHandlingResult handle(SurfaceResizeEvent event);
};


#endif // DISPLAY_HPP