// 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");