// SPDX-License-Identifier: GPL-2.0
/*
 * SPI interface to the Ilitek ILI9806E panel.
 *
 * Copyright (c) 2026 Amarula Solutions, Dario Binacchi <dario.binacchi@amarulasolutions.com>
 */

#include <linux/delay.h>
#include <linux/device.h>
#include <linux/media-bus-format.h>
#include <linux/module.h>
#include <linux/spi/spi.h>

#include <drm/drm_mipi_dbi.h>
#include <drm/drm_panel.h>
#include <drm/drm_print.h>

#include <video/mipi_display.h>

#include "panel-ilitek-ili9806e-core.h"

struct ili9806e_spi_panel {
        struct spi_device *spi;
        struct mipi_dbi dbi;
        const struct ili9806e_spi_panel_desc *desc;
};

struct ili9806e_spi_panel_desc {
        const struct drm_display_mode *display_mode;
        u32 bus_format;
        u32 bus_flags;
        void (*init_sequence)(struct ili9806e_spi_panel *ctx);
};

static int ili9806e_spi_off(struct ili9806e_spi_panel *ctx)
{
        struct mipi_dbi *dbi = &ctx->dbi;

        mipi_dbi_command(dbi, MIPI_DCS_SET_DISPLAY_OFF, 0x00);
        mipi_dbi_command(dbi, MIPI_DCS_ENTER_SLEEP_MODE, 0x00);

        return 0;
}

static int ili9806e_spi_unprepare(struct drm_panel *panel)
{
        struct ili9806e_spi_panel *ctx = ili9806e_get_transport(panel);
        struct device *dev = &ctx->spi->dev;
        int ret;

        ili9806e_spi_off(ctx);

        ret = ili9806e_power_off(dev);
        if (ret)
                dev_err(dev, "power off failed: %d\n", ret);

        return 0;
}

static int ili9806e_spi_prepare(struct drm_panel *panel)
{
        struct ili9806e_spi_panel *ctx = ili9806e_get_transport(panel);
        struct device *dev = &ctx->spi->dev;
        int ret;

        ret = ili9806e_power_on(dev);
        if (ret)
                return ret;

        if (ctx->desc->init_sequence)
                ctx->desc->init_sequence(ctx);

        return 0;
}

static int ili9806e_spi_get_modes(struct drm_panel *panel,
                              struct drm_connector *connector)
{
        struct ili9806e_spi_panel *ctx = ili9806e_get_transport(panel);
        const struct ili9806e_spi_panel_desc *desc = ctx->desc;
        struct drm_display_mode *mode;

        mode = drm_mode_duplicate(connector->dev, desc->display_mode);
        if (!mode)
                return -ENOMEM;

        drm_mode_set_name(mode);

        connector->display_info.width_mm = mode->width_mm;
        connector->display_info.height_mm = mode->height_mm;
        connector->display_info.bus_flags = desc->bus_flags;
        drm_display_info_set_bus_formats(&connector->display_info,
                                         &desc->bus_format, 1);

        drm_mode_probed_add(connector, mode);

        return 1;
}

static const struct drm_panel_funcs ili9806e_spi_funcs = {
        .unprepare = ili9806e_spi_unprepare,
        .prepare   = ili9806e_spi_prepare,
        .get_modes = ili9806e_spi_get_modes,
};

static int ili9806e_spi_probe(struct spi_device *spi)
{
        struct device *dev = &spi->dev;
        struct ili9806e_spi_panel *ctx;
        int err;

        ctx = devm_kzalloc(dev, sizeof(struct ili9806e_spi_panel), GFP_KERNEL);
        if (!ctx)
                return -ENOMEM;

        ctx->spi = spi;
        ctx->desc = device_get_match_data(dev);

        err = mipi_dbi_spi_init(spi, &ctx->dbi, NULL);
        if (err)
                return dev_err_probe(dev, err, "MIPI DBI init failed\n");

        return ili9806e_probe(dev, ctx, &ili9806e_spi_funcs,
                              DRM_MODE_CONNECTOR_DPI);
}

static void ili9806e_spi_remove(struct spi_device *spi)
{
        ili9806e_remove(&spi->dev);
}

static void rk050hr345_ct106a_init(struct ili9806e_spi_panel *ctx)
{
        struct mipi_dbi *dbi = &ctx->dbi;

        /* Switch to page 1 */
        mipi_dbi_command(dbi, 0xff, 0xff, 0x98, 0x06, 0x04, 0x01);
        /* Interface Settings */
        mipi_dbi_command(dbi, 0x08, 0x10);
        mipi_dbi_command(dbi, 0x21, 0x01);
        /* Panel Settings */
        mipi_dbi_command(dbi, 0x30, 0x01);
        mipi_dbi_command(dbi, 0x31, 0x00);
        /* Power Control */
        mipi_dbi_command(dbi, 0x40, 0x15);
        mipi_dbi_command(dbi, 0x41, 0x44);
        mipi_dbi_command(dbi, 0x42, 0x03);
        mipi_dbi_command(dbi, 0x43, 0x09);
        mipi_dbi_command(dbi, 0x44, 0x09);
        mipi_dbi_command(dbi, 0x50, 0x78);
        mipi_dbi_command(dbi, 0x51, 0x78);
        mipi_dbi_command(dbi, 0x52, 0x00);
        mipi_dbi_command(dbi, 0x53, 0x3a);
        mipi_dbi_command(dbi, 0x57, 0x50);
        /* Timing Control */
        mipi_dbi_command(dbi, 0x60, 0x07);
        mipi_dbi_command(dbi, 0x61, 0x00);
        mipi_dbi_command(dbi, 0x62, 0x08);
        mipi_dbi_command(dbi, 0x63, 0x00);
        /* Gamma Settings */
        mipi_dbi_command(dbi, 0xa0, 0x00);
        mipi_dbi_command(dbi, 0xa1, 0x03);
        mipi_dbi_command(dbi, 0xa2, 0x0b);
        mipi_dbi_command(dbi, 0xa3, 0x0f);
        mipi_dbi_command(dbi, 0xa4, 0x0b);
        mipi_dbi_command(dbi, 0xa5, 0x1b);
        mipi_dbi_command(dbi, 0xa6, 0x0a);
        mipi_dbi_command(dbi, 0xa7, 0x0a);
        mipi_dbi_command(dbi, 0xa8, 0x02);
        mipi_dbi_command(dbi, 0xa9, 0x07);
        mipi_dbi_command(dbi, 0xaa, 0x05);
        mipi_dbi_command(dbi, 0xab, 0x03);
        mipi_dbi_command(dbi, 0xac, 0x0e);
        mipi_dbi_command(dbi, 0xad, 0x32);
        mipi_dbi_command(dbi, 0xae, 0x2d);
        mipi_dbi_command(dbi, 0xaf, 0x00);
        mipi_dbi_command(dbi, 0xc0, 0x00);
        mipi_dbi_command(dbi, 0xc1, 0x03);
        mipi_dbi_command(dbi, 0xc2, 0x0e);
        mipi_dbi_command(dbi, 0xc3, 0x10);
        mipi_dbi_command(dbi, 0xc4, 0x09);
        mipi_dbi_command(dbi, 0xc5, 0x17);
        mipi_dbi_command(dbi, 0xc6, 0x09);
        mipi_dbi_command(dbi, 0xc7, 0x07);
        mipi_dbi_command(dbi, 0xc8, 0x04);
        mipi_dbi_command(dbi, 0xc9, 0x09);
        mipi_dbi_command(dbi, 0xca, 0x06);
        mipi_dbi_command(dbi, 0xcb, 0x06);
        mipi_dbi_command(dbi, 0xcc, 0x0c);
        mipi_dbi_command(dbi, 0xcd, 0x25);
        mipi_dbi_command(dbi, 0xce, 0x20);
        mipi_dbi_command(dbi, 0xcf, 0x00);

        /* Switch to page 6 */
        mipi_dbi_command(dbi, 0xff, 0xff, 0x98, 0x06, 0x04, 0x06);
        /* GIP settings */
        mipi_dbi_command(dbi, 0x00, 0x21);
        mipi_dbi_command(dbi, 0x01, 0x09);
        mipi_dbi_command(dbi, 0x02, 0x00);
        mipi_dbi_command(dbi, 0x03, 0x00);
        mipi_dbi_command(dbi, 0x04, 0x01);
        mipi_dbi_command(dbi, 0x05, 0x01);
        mipi_dbi_command(dbi, 0x06, 0x80);
        mipi_dbi_command(dbi, 0x07, 0x05);
        mipi_dbi_command(dbi, 0x08, 0x02);
        mipi_dbi_command(dbi, 0x09, 0x80);
        mipi_dbi_command(dbi, 0x0a, 0x00);
        mipi_dbi_command(dbi, 0x0b, 0x00);
        mipi_dbi_command(dbi, 0x0c, 0x0a);
        mipi_dbi_command(dbi, 0x0d, 0x0a);
        mipi_dbi_command(dbi, 0x0e, 0x00);
        mipi_dbi_command(dbi, 0x0f, 0x00);
        mipi_dbi_command(dbi, 0x10, 0xe0);
        mipi_dbi_command(dbi, 0x11, 0xe4);
        mipi_dbi_command(dbi, 0x12, 0x04);
        mipi_dbi_command(dbi, 0x13, 0x00);
        mipi_dbi_command(dbi, 0x14, 0x00);
        mipi_dbi_command(dbi, 0x15, 0xc0);
        mipi_dbi_command(dbi, 0x16, 0x08);
        mipi_dbi_command(dbi, 0x17, 0x00);
        mipi_dbi_command(dbi, 0x18, 0x00);
        mipi_dbi_command(dbi, 0x19, 0x00);
        mipi_dbi_command(dbi, 0x1a, 0x00);
        mipi_dbi_command(dbi, 0x1b, 0x00);
        mipi_dbi_command(dbi, 0x1c, 0x00);
        mipi_dbi_command(dbi, 0x1d, 0x00);
        mipi_dbi_command(dbi, 0x20, 0x01);
        mipi_dbi_command(dbi, 0x21, 0x23);
        mipi_dbi_command(dbi, 0x22, 0x45);
        mipi_dbi_command(dbi, 0x23, 0x67);
        mipi_dbi_command(dbi, 0x24, 0x01);
        mipi_dbi_command(dbi, 0x25, 0x23);
        mipi_dbi_command(dbi, 0x26, 0x45);
        mipi_dbi_command(dbi, 0x27, 0x67);
        mipi_dbi_command(dbi, 0x30, 0x01);
        mipi_dbi_command(dbi, 0x31, 0x11);
        mipi_dbi_command(dbi, 0x32, 0x00);
        mipi_dbi_command(dbi, 0x33, 0xee);
        mipi_dbi_command(dbi, 0x34, 0xff);
        mipi_dbi_command(dbi, 0x35, 0xbb);
        mipi_dbi_command(dbi, 0x36, 0xca);
        mipi_dbi_command(dbi, 0x37, 0xdd);
        mipi_dbi_command(dbi, 0x38, 0xac);
        mipi_dbi_command(dbi, 0x39, 0x76);
        mipi_dbi_command(dbi, 0x3a, 0x67);
        mipi_dbi_command(dbi, 0x3b, 0x22);
        mipi_dbi_command(dbi, 0x3c, 0x22);
        mipi_dbi_command(dbi, 0x3d, 0x22);
        mipi_dbi_command(dbi, 0x3e, 0x22);
        mipi_dbi_command(dbi, 0x3f, 0x22);
        mipi_dbi_command(dbi, 0x40, 0x22);
        mipi_dbi_command(dbi, 0x52, 0x10);
        mipi_dbi_command(dbi, 0x53, 0x10);

        /* Switch to page 7 */
        mipi_dbi_command(dbi, 0xff, 0xff, 0x98, 0x06, 0x04, 0x07);
        mipi_dbi_command(dbi, 0x17, 0x22);
        mipi_dbi_command(dbi, 0x02, 0x77);
        mipi_dbi_command(dbi, 0xe1, 0x79);
        mipi_dbi_command(dbi, 0xb3, 0x10);

        /* Switch to page 0 */
        mipi_dbi_command(dbi, 0xff, 0xff, 0x98, 0x06, 0x04, 0x00);
        mipi_dbi_command(dbi, MIPI_DCS_SET_ADDRESS_MODE, 0x00); // 0x36
        mipi_dbi_command(dbi, MIPI_DCS_EXIT_SLEEP_MODE); // 0x11

        msleep(120);

        mipi_dbi_command(dbi, MIPI_DCS_SET_DISPLAY_ON);

        msleep(120);
}

static const struct drm_display_mode rk050hr345_ct106a_mode = {
        .width_mm    = 62,
        .height_mm   = 110,
        .clock       = 27000,
        .hdisplay    = 480,
        .hsync_start = 480 + 10,
        .hsync_end   = 480 + 10 + 10,
        .htotal      = 480 + 10 + 10 + 10,
        .vdisplay    = 854,
        .vsync_start = 854 + 10,
        .vsync_end   = 854 + 10 + 10,
        .vtotal      = 854 + 10 + 10 + 10,
        .flags       = DRM_MODE_FLAG_NHSYNC | DRM_MODE_FLAG_NVSYNC,
        .type        = DRM_MODE_TYPE_PREFERRED | DRM_MODE_TYPE_DRIVER,
};

static const struct ili9806e_spi_panel_desc rk050hr345_ct106a_desc = {
        .init_sequence = rk050hr345_ct106a_init,
        .display_mode = &rk050hr345_ct106a_mode,
        .bus_format = MEDIA_BUS_FMT_RGB888_1X24,
        .bus_flags = DRM_BUS_FLAG_DE_HIGH |
                     DRM_BUS_FLAG_PIXDATA_DRIVE_POSEDGE,
};

static const struct of_device_id ili9806e_spi_of_match[] = {
        { .compatible = "rocktech,rk050hr345-ct106a", .data = &rk050hr345_ct106a_desc },
        { }
};
MODULE_DEVICE_TABLE(of, ili9806e_spi_of_match);

static const struct spi_device_id ili9806e_spi_ids[] = {
        { "rk050hr345-ct106a", },
        { /* sentinel */ }
};
MODULE_DEVICE_TABLE(spi, ili9806e_spi_ids);

static struct spi_driver ili9806e_spi_driver = {
        .driver = {
                .name = "ili9806e-spi",
                .of_match_table = ili9806e_spi_of_match,
        },
        .probe = ili9806e_spi_probe,
        .remove = ili9806e_spi_remove,
        .id_table = ili9806e_spi_ids,
};
module_spi_driver(ili9806e_spi_driver);

MODULE_AUTHOR("Dario Binacchi <dario.binacchi@amarulasolutions.com>");
MODULE_DESCRIPTION("Ilitek ILI9806E LCD SPI Driver");
MODULE_LICENSE("GPL");