#include "luat_base.h"
#include "luat_lcd.h"
#include "luat_gpio.h"
#include "luat_spi.h"
#include "luat_malloc.h"
#include "luat_timer.h"

#define LUAT_LOG_TAG "st7735s"
#include "luat_log.h"

#define LCD_W 128
#define LCD_H 160
#define LCD_DIRECTION 0

static int st7735s_sleep(luat_lcd_conf_t* conf) {
    luat_gpio_set(conf->pin_pwr, Luat_GPIO_LOW);
    luat_timer_mdelay(5);
    lcd_write_cmd(conf,0x10);
    return 0;
}

static int st7735s_wakeup(luat_lcd_conf_t* conf) {
    luat_gpio_set(conf->pin_pwr, Luat_GPIO_HIGH);
    luat_timer_mdelay(5);
    lcd_write_cmd(conf,0x11);
    //luat_timer_mdelay(120); // 外部休眠就好了吧
    return 0;
}

static int st7735s_close(luat_lcd_conf_t* conf) {
    luat_gpio_set(conf->pin_pwr, Luat_GPIO_LOW);
    return 0;
}
static int st7735s_init(luat_lcd_conf_t* conf) {
    if (conf->w == 0)
        conf->w = LCD_W;
    if (conf->h == 0)
        conf->h = LCD_H;

    luat_gpio_mode(conf->pin_dc, Luat_GPIO_OUTPUT, Luat_GPIO_PULLUP, 0); // DC
    luat_gpio_mode(conf->pin_pwr, Luat_GPIO_OUTPUT, Luat_GPIO_PULLUP, 0); // POWER
    luat_gpio_mode(conf->pin_rst, Luat_GPIO_OUTPUT, Luat_GPIO_PULLUP, 0); // RST

    luat_gpio_mode(conf->pin_dc, Luat_GPIO_OUTPUT, Luat_GPIO_DEFAULT, Luat_GPIO_HIGH); // DC
    luat_gpio_mode(conf->pin_pwr, Luat_GPIO_OUTPUT, Luat_GPIO_DEFAULT, Luat_GPIO_LOW); // POWER
    luat_gpio_mode(conf->pin_rst, Luat_GPIO_OUTPUT, Luat_GPIO_DEFAULT, Luat_GPIO_LOW); // RST

    luat_gpio_set(conf->pin_pwr, Luat_GPIO_LOW);
    luat_gpio_set(conf->pin_rst, Luat_GPIO_LOW);
    luat_timer_mdelay(100);
    luat_gpio_set(conf->pin_rst, Luat_GPIO_HIGH);
    luat_timer_mdelay(100);

	lcd_write_cmd(conf,0x11);     //Sleep out
	luat_timer_mdelay(120);                //Delay 120ms
	lcd_write_cmd(conf,0xB1);     //Normal mode
	lcd_write_data(conf,0x05);
	lcd_write_data(conf,0x3C);
	lcd_write_data(conf,0x3C);
	lcd_write_cmd(conf,0xB2);     //Idle mode
	lcd_write_data(conf,0x05);
	lcd_write_data(conf,0x3C);
	lcd_write_data(conf,0x3C);
	lcd_write_cmd(conf,0xB3);     //Partial mode
	lcd_write_data(conf,0x05);
	lcd_write_data(conf,0x3C);
	lcd_write_data(conf,0x3C);
	lcd_write_data(conf,0x05);
	lcd_write_data(conf,0x3C);
	lcd_write_data(conf,0x3C);
	lcd_write_cmd(conf,0xB4);     //Dot inversion
	lcd_write_data(conf,0x03);
	lcd_write_cmd(conf,0xC0);     //AVDD GVDD
	lcd_write_data(conf,0xAB);
	lcd_write_data(conf,0x0B);
	lcd_write_data(conf,0x04);
	lcd_write_cmd(conf,0xC1);     //VGH VGL
	lcd_write_data(conf,0xC5);   //C0
	lcd_write_cmd(conf,0xC2);     //Normal Mode
	lcd_write_data(conf,0x0D);
	lcd_write_data(conf,0x00);
	lcd_write_cmd(conf,0xC3);     //Idle
	lcd_write_data(conf,0x8D);
	lcd_write_data(conf,0x6A);
	lcd_write_cmd(conf,0xC4);     //Partial+Full
	lcd_write_data(conf,0x8D);
	lcd_write_data(conf,0xEE);
	lcd_write_cmd(conf,0xC5);     //VCOM
	lcd_write_data(conf,0x0F);
	lcd_write_cmd(conf,0xE0);     //positive gamma
	lcd_write_data(conf,0x07);
	lcd_write_data(conf,0x0E);
	lcd_write_data(conf,0x08);
	lcd_write_data(conf,0x07);
	lcd_write_data(conf,0x10);
	lcd_write_data(conf,0x07);
	lcd_write_data(conf,0x02);
	lcd_write_data(conf,0x07);
	lcd_write_data(conf,0x09);
	lcd_write_data(conf,0x0F);
	lcd_write_data(conf,0x25);
	lcd_write_data(conf,0x36);
	lcd_write_data(conf,0x00);
	lcd_write_data(conf,0x08);
	lcd_write_data(conf,0x04);
	lcd_write_data(conf,0x10);
	lcd_write_cmd(conf,0xE1);     //negative gamma
	lcd_write_data(conf,0x0A);
	lcd_write_data(conf,0x0D);
	lcd_write_data(conf,0x08);
	lcd_write_data(conf,0x07);
	lcd_write_data(conf,0x0F);
	lcd_write_data(conf,0x07);
	lcd_write_data(conf,0x02);
	lcd_write_data(conf,0x07);
	lcd_write_data(conf,0x09);
	lcd_write_data(conf,0x0F);
	lcd_write_data(conf,0x25);
	lcd_write_data(conf,0x35);
	lcd_write_data(conf,0x00);
	lcd_write_data(conf,0x09);
	lcd_write_data(conf,0x04);
	lcd_write_data(conf,0x10);

	lcd_write_cmd(conf,0xFC);
	lcd_write_data(conf,0x80);

	lcd_write_cmd(conf,0x3A);
	lcd_write_data(conf,0x05);
	lcd_write_cmd(conf,0x36);
    if(conf->direction==0)lcd_write_data(conf,0xC8);
    else if(conf->direction==1)lcd_write_data(conf,0x78);
    else if(conf->direction==2)lcd_write_data(conf,0x08);
    else lcd_write_data(conf,0xA8);
	lcd_write_cmd(conf,0x21);     //Display inversion
	lcd_write_cmd(conf,0x29);     //Display on
	lcd_write_cmd(conf,0x2A);     //Set Column Address
	lcd_write_data(conf,0x00);
	lcd_write_data(conf,conf->xoffset);  //26
	lcd_write_data(conf,0x00);
	lcd_write_data(conf,conf->w+conf->xoffset-1);   //105
	lcd_write_cmd(conf,0x2B);     //Set Page Address
	lcd_write_data(conf,0x00);
	lcd_write_data(conf,conf->yoffset);    //1
	lcd_write_data(conf,0x00);
	lcd_write_data(conf,conf->h+conf->yoffset-1);    //160
	lcd_write_cmd(conf,0x2C);

    luat_lcd_clear(conf,WHITE);
    /* display on */
    luat_lcd_display_on(conf);
    return 0;
};

static int st7735s_draw(luat_lcd_conf_t* conf, uint16_t x_start, uint16_t y_start, uint16_t x_end, uint16_t y_end, luat_color_t* color) {
    uint32_t size = size = (x_end - x_start+1) * (y_end - y_start+1) * 2;
    luat_lcd_set_address(conf,x_start+conf->xoffset, y_start+conf->yoffset, x_end+conf->xoffset, y_end+conf->yoffset);
    luat_gpio_set(conf->pin_dc, Luat_GPIO_HIGH);
    luat_spi_send(conf->port, (const char*)color, size);
    return 0;
}

const luat_lcd_opts_t lcd_opts_st7735s = {
    .name = "st7735s",
    .init = st7735s_init,
    .close = st7735s_close,
    .draw = st7735s_draw,
    .sleep = st7735s_sleep,
    .wakeup = st7735s_wakeup,
};