// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * lms501kf03 TFT LCD panel driver.
 *
 * Copyright (c) 2012 Samsung Electronics Co., Ltd.
 * Author: Jingoo Han  <jg1.han@samsung.com>
 */

#include <linux/delay.h>
#include <linux/lcd.h>
#include <linux/module.h>
#include <linux/spi/spi.h>
#include <linux/wait.h>

#define COMMAND_ONLY            0x00
#define DATA_ONLY               0x01

struct lms501kf03 {
        struct device                   *dev;
        struct spi_device               *spi;
        unsigned int                    power;
        struct lcd_device               *ld;
        struct lcd_platform_data        *lcd_pd;
};

static const unsigned char seq_password[] = {
        0xb9, 0xff, 0x83, 0x69,
};

static const unsigned char seq_power[] = {
        0xb1, 0x01, 0x00, 0x34, 0x06, 0x00, 0x14, 0x14, 0x20, 0x28,
        0x12, 0x12, 0x17, 0x0a, 0x01, 0xe6, 0xe6, 0xe6, 0xe6, 0xe6,
};

static const unsigned char seq_display[] = {
        0xb2, 0x00, 0x2b, 0x03, 0x03, 0x70, 0x00, 0xff, 0x00, 0x00,
        0x00, 0x00, 0x03, 0x03, 0x00, 0x01,
};

static const unsigned char seq_rgb_if[] = {
        0xb3, 0x09,
};

static const unsigned char seq_display_inv[] = {
        0xb4, 0x01, 0x08, 0x77, 0x0e, 0x06,
};

static const unsigned char seq_vcom[] = {
        0xb6, 0x4c, 0x2e,
};

static const unsigned char seq_gate[] = {
        0xd5, 0x00, 0x05, 0x03, 0x29, 0x01, 0x07, 0x17, 0x68, 0x13,
        0x37, 0x20, 0x31, 0x8a, 0x46, 0x9b, 0x57, 0x13, 0x02, 0x75,
        0xb9, 0x64, 0xa8, 0x07, 0x0f, 0x04, 0x07,
};

static const unsigned char seq_panel[] = {
        0xcc, 0x02,
};

static const unsigned char seq_col_mod[] = {
        0x3a, 0x77,
};

static const unsigned char seq_w_gamma[] = {
        0xe0, 0x00, 0x04, 0x09, 0x0f, 0x1f, 0x3f, 0x1f, 0x2f, 0x0a,
        0x0f, 0x10, 0x16, 0x18, 0x16, 0x17, 0x0d, 0x15, 0x00, 0x04,
        0x09, 0x0f, 0x38, 0x3f, 0x20, 0x39, 0x0a, 0x0f, 0x10, 0x16,
        0x18, 0x16, 0x17, 0x0d, 0x15,
};

static const unsigned char seq_rgb_gamma[] = {
        0xc1, 0x01, 0x03, 0x07, 0x0f, 0x1a, 0x22, 0x2c, 0x33, 0x3c,
        0x46, 0x4f, 0x58, 0x60, 0x69, 0x71, 0x79, 0x82, 0x89, 0x92,
        0x9a, 0xa1, 0xa9, 0xb1, 0xb9, 0xc1, 0xc9, 0xcf, 0xd6, 0xde,
        0xe5, 0xec, 0xf3, 0xf9, 0xff, 0xdd, 0x39, 0x07, 0x1c, 0xcb,
        0xab, 0x5f, 0x49, 0x80, 0x03, 0x07, 0x0f, 0x19, 0x20, 0x2a,
        0x31, 0x39, 0x42, 0x4b, 0x53, 0x5b, 0x63, 0x6b, 0x73, 0x7b,
        0x83, 0x8a, 0x92, 0x9b, 0xa2, 0xaa, 0xb2, 0xba, 0xc2, 0xca,
        0xd0, 0xd8, 0xe1, 0xe8, 0xf0, 0xf8, 0xff, 0xf7, 0xd8, 0xbe,
        0xa7, 0x39, 0x40, 0x85, 0x8c, 0xc0, 0x04, 0x07, 0x0c, 0x17,
        0x1c, 0x23, 0x2b, 0x34, 0x3b, 0x43, 0x4c, 0x54, 0x5b, 0x63,
        0x6a, 0x73, 0x7a, 0x82, 0x8a, 0x91, 0x98, 0xa1, 0xa8, 0xb0,
        0xb7, 0xc1, 0xc9, 0xcf, 0xd9, 0xe3, 0xea, 0xf4, 0xff, 0x00,
        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
};

static const unsigned char seq_sleep_out[] = {
        0x11,
};

static const unsigned char seq_display_on[] = {
        0x29,
};

static const unsigned char seq_display_off[] = {
        0x10,
};

static int lms501kf03_spi_write_byte(struct lms501kf03 *lcd, int addr, int data)
{
        u16 buf[1];
        struct spi_message msg;

        struct spi_transfer xfer = {
                .len            = 2,
                .tx_buf         = buf,
        };

        buf[0] = (addr << 8) | data;

        spi_message_init(&msg);
        spi_message_add_tail(&xfer, &msg);

        return spi_sync(lcd->spi, &msg);
}

static int lms501kf03_spi_write(struct lms501kf03 *lcd, unsigned char address,
                                unsigned char command)
{
        return lms501kf03_spi_write_byte(lcd, address, command);
}

static int lms501kf03_panel_send_sequence(struct lms501kf03 *lcd,
                                        const unsigned char *wbuf,
                                        unsigned int len)
{
        int ret = 0, i = 0;

        while (i < len) {
                if (i == 0)
                        ret = lms501kf03_spi_write(lcd, COMMAND_ONLY, wbuf[i]);
                else
                        ret = lms501kf03_spi_write(lcd, DATA_ONLY, wbuf[i]);
                if (ret)
                        break;
                i += 1;
        }

        return ret;
}

static int lms501kf03_ldi_init(struct lms501kf03 *lcd)
{
        int ret, i;
        static const unsigned char *init_seq[] = {
                seq_password,
                seq_power,
                seq_display,
                seq_rgb_if,
                seq_display_inv,
                seq_vcom,
                seq_gate,
                seq_panel,
                seq_col_mod,
                seq_w_gamma,
                seq_rgb_gamma,
                seq_sleep_out,
        };

        static const unsigned int size_seq[] = {
                ARRAY_SIZE(seq_password),
                ARRAY_SIZE(seq_power),
                ARRAY_SIZE(seq_display),
                ARRAY_SIZE(seq_rgb_if),
                ARRAY_SIZE(seq_display_inv),
                ARRAY_SIZE(seq_vcom),
                ARRAY_SIZE(seq_gate),
                ARRAY_SIZE(seq_panel),
                ARRAY_SIZE(seq_col_mod),
                ARRAY_SIZE(seq_w_gamma),
                ARRAY_SIZE(seq_rgb_gamma),
                ARRAY_SIZE(seq_sleep_out),
        };

        for (i = 0; i < ARRAY_SIZE(init_seq); i++) {
                ret = lms501kf03_panel_send_sequence(lcd, init_seq[i],
                                                size_seq[i]);
                if (ret)
                        break;
        }
        /*
         * According to the datasheet, 120ms delay time is required.
         * After sleep out sequence, command is blocked for 120ms.
         * Thus, LDI should wait for 120ms.
         */
        msleep(120);

        return ret;
}

static int lms501kf03_ldi_enable(struct lms501kf03 *lcd)
{
        return lms501kf03_panel_send_sequence(lcd, seq_display_on,
                                        ARRAY_SIZE(seq_display_on));
}

static int lms501kf03_ldi_disable(struct lms501kf03 *lcd)
{
        return lms501kf03_panel_send_sequence(lcd, seq_display_off,
                                        ARRAY_SIZE(seq_display_off));
}

static int lms501kf03_power_is_on(int power)
{
        return (power) <= LCD_POWER_REDUCED;
}

static int lms501kf03_power_on(struct lms501kf03 *lcd)
{
        int ret = 0;
        struct lcd_platform_data *pd;

        pd = lcd->lcd_pd;

        if (!pd->power_on) {
                dev_err(lcd->dev, "power_on is NULL.\n");
                return -EINVAL;
        }

        pd->power_on(lcd->ld, 1);
        msleep(pd->power_on_delay);

        if (!pd->reset) {
                dev_err(lcd->dev, "reset is NULL.\n");
                return -EINVAL;
        }

        pd->reset(lcd->ld);
        msleep(pd->reset_delay);

        ret = lms501kf03_ldi_init(lcd);
        if (ret) {
                dev_err(lcd->dev, "failed to initialize ldi.\n");
                return ret;
        }

        ret = lms501kf03_ldi_enable(lcd);
        if (ret) {
                dev_err(lcd->dev, "failed to enable ldi.\n");
                return ret;
        }

        return 0;
}

static int lms501kf03_power_off(struct lms501kf03 *lcd)
{
        int ret = 0;
        struct lcd_platform_data *pd;

        pd = lcd->lcd_pd;

        ret = lms501kf03_ldi_disable(lcd);
        if (ret) {
                dev_err(lcd->dev, "lcd setting failed.\n");
                return -EIO;
        }

        msleep(pd->power_off_delay);

        pd->power_on(lcd->ld, 0);

        return 0;
}

static int lms501kf03_power(struct lms501kf03 *lcd, int power)
{
        int ret = 0;

        if (lms501kf03_power_is_on(power) &&
                !lms501kf03_power_is_on(lcd->power))
                ret = lms501kf03_power_on(lcd);
        else if (!lms501kf03_power_is_on(power) &&
                lms501kf03_power_is_on(lcd->power))
                ret = lms501kf03_power_off(lcd);

        if (!ret)
                lcd->power = power;

        return ret;
}

static int lms501kf03_get_power(struct lcd_device *ld)
{
        struct lms501kf03 *lcd = lcd_get_data(ld);

        return lcd->power;
}

static int lms501kf03_set_power(struct lcd_device *ld, int power)
{
        struct lms501kf03 *lcd = lcd_get_data(ld);

        if (power != LCD_POWER_ON && power != LCD_POWER_OFF &&
                power != LCD_POWER_REDUCED) {
                dev_err(lcd->dev, "power value should be 0, 1 or 4.\n");
                return -EINVAL;
        }

        return lms501kf03_power(lcd, power);
}

static const struct lcd_ops lms501kf03_lcd_ops = {
        .get_power = lms501kf03_get_power,
        .set_power = lms501kf03_set_power,
};

static int lms501kf03_probe(struct spi_device *spi)
{
        struct lms501kf03 *lcd = NULL;
        struct lcd_device *ld = NULL;
        int ret = 0;

        lcd = devm_kzalloc(&spi->dev, sizeof(struct lms501kf03), GFP_KERNEL);
        if (!lcd)
                return -ENOMEM;

        /* lms501kf03 lcd panel uses 3-wire 9-bit SPI Mode. */
        spi->bits_per_word = 9;

        ret = spi_setup(spi);
        if (ret < 0) {
                dev_err(&spi->dev, "spi setup failed.\n");
                return ret;
        }

        lcd->spi = spi;
        lcd->dev = &spi->dev;

        lcd->lcd_pd = dev_get_platdata(&spi->dev);
        if (!lcd->lcd_pd) {
                dev_err(&spi->dev, "platform data is NULL\n");
                return -EINVAL;
        }

        ld = devm_lcd_device_register(&spi->dev, "lms501kf03", &spi->dev, lcd,
                                        &lms501kf03_lcd_ops);
        if (IS_ERR(ld))
                return PTR_ERR(ld);

        lcd->ld = ld;

        if (!lcd->lcd_pd->lcd_enabled) {
                /*
                 * if lcd panel was off from bootloader then
                 * current lcd status is powerdown and then
                 * it enables lcd panel.
                 */
                lcd->power = LCD_POWER_OFF;

                lms501kf03_power(lcd, LCD_POWER_ON);
        } else {
                lcd->power = LCD_POWER_ON;
        }

        spi_set_drvdata(spi, lcd);

        dev_info(&spi->dev, "lms501kf03 panel driver has been probed.\n");

        return 0;
}

static void lms501kf03_remove(struct spi_device *spi)
{
        struct lms501kf03 *lcd = spi_get_drvdata(spi);

        lms501kf03_power(lcd, LCD_POWER_OFF);
}

#ifdef CONFIG_PM_SLEEP
static int lms501kf03_suspend(struct device *dev)
{
        struct lms501kf03 *lcd = dev_get_drvdata(dev);

        dev_dbg(dev, "lcd->power = %d\n", lcd->power);

        /*
         * when lcd panel is suspend, lcd panel becomes off
         * regardless of status.
         */
        return lms501kf03_power(lcd, LCD_POWER_OFF);
}

static int lms501kf03_resume(struct device *dev)
{
        struct lms501kf03 *lcd = dev_get_drvdata(dev);

        lcd->power = LCD_POWER_OFF;

        return lms501kf03_power(lcd, LCD_POWER_ON);
}
#endif

static SIMPLE_DEV_PM_OPS(lms501kf03_pm_ops, lms501kf03_suspend,
                        lms501kf03_resume);

static void lms501kf03_shutdown(struct spi_device *spi)
{
        struct lms501kf03 *lcd = spi_get_drvdata(spi);

        lms501kf03_power(lcd, LCD_POWER_OFF);
}

static struct spi_driver lms501kf03_driver = {
        .driver = {
                .name   = "lms501kf03",
                .pm     = &lms501kf03_pm_ops,
        },
        .probe          = lms501kf03_probe,
        .remove         = lms501kf03_remove,
        .shutdown       = lms501kf03_shutdown,
};

module_spi_driver(lms501kf03_driver);

MODULE_AUTHOR("Jingoo Han <jg1.han@samsung.com>");
MODULE_DESCRIPTION("lms501kf03 LCD Driver");
MODULE_LICENSE("GPL");