#include <stdio.h>
#include <string.h>
#include "freertos/FreeRTOS.h"
#include "freertos/task.h"
#include "freertos/queue.h"
#include "freertos/semphr.h"
#include "soc/soc_caps.h"
#include "esp_freertos_hooks.h"
#include "esp_log.h"
#include "esp_lcd_panel_io.h"
#include "esp_lcd_panel_vendor.h"
#include "esp_lcd_panel_ops.h"
#include "esp_lcd_panel_interface.h"
#include "esp_lcd_panel_rgb.h"
#include "driver/gpio.h"
#include "board.h"
#include "lvgl.h"
#include "esp_timer.h"
#include "driver/spi_master.h"
#include "button.h"
#include "mt8901.h"

#define TAG "RGB"

#define ECO_O(y) (y>0)? -1:1
#define ECO_STEP(x) x? ECO_O(x):0

static button_t *g_btn;

static esp_lcd_panel_handle_t g_panel_handle = NULL;

static void __qsmd_rgb_disp_flush(lv_disp_drv_t *disp_drv, const lv_area_t *area, lv_color_t *color_p){
    int offsetx1 = area->x1;
    int offsetx2 = area->x2;
    int offsety1 = area->y1;
    int offsety2 = area->y2;

    esp_lcd_panel_draw_bitmap(g_panel_handle, offsetx1, offsety1, offsetx2 + 1, offsety2 + 1, color_p);
    lv_disp_flush_ready(disp_drv);
}

void qmsd_rgb_init(esp_lcd_rgb_panel_config_t *panel_config){
    static lv_disp_drv_t disp_drv;
    int buffer_size;

    //create the buffers
    void *buf1 = NULL;
    void *buf2 = NULL;

    static lv_disp_draw_buf_t draw_buf;

    lv_init();

    ESP_ERROR_CHECK(esp_lcd_new_rgb_panel(panel_config, &g_panel_handle));
    ESP_ERROR_CHECK(esp_lcd_panel_reset(g_panel_handle));
    ESP_ERROR_CHECK(esp_lcd_panel_init(g_panel_handle));

    buffer_size = panel_config->timings.h_res * panel_config->timings.v_res * 2; //added multiplier for bits per pixel / 8 = 2

    ESP_ERROR_CHECK(esp_lcd_rgb_panel_get_frame_buffer(g_panel_handle, 2, &buf1 , &buf2));

    lv_disp_draw_buf_init(&draw_buf, buf1, buf2, buffer_size);

    lv_disp_drv_init(&disp_drv);
    disp_drv.flush_cb = __qsmd_rgb_disp_flush;
    disp_drv.draw_buf = &draw_buf;
    disp_drv.hor_res = panel_config->timings.h_res;
    disp_drv.ver_res = panel_config->timings.v_res;
    lv_disp_drv_register(&disp_drv);
}

static spi_device_handle_t g_screen_spi;

static void lcd_cmd(spi_device_handle_t spi, const uint16_t data) {
    esp_err_t ret;
    uint16_t tmp_cmd =(data | 0x0000);
    spi_transaction_ext_t trans =(spi_transaction_ext_t) {
            .base = {
                    .flags = SPI_TRANS_VARIABLE_CMD,
                    .cmd = tmp_cmd,
            },
            .command_bits = 9,
    };
    ret = spi_device_transmit(spi,(spi_transaction_t *)&trans);
    assert(ret == ESP_OK);          //Should have had no issues.
}

static void lcd_data(spi_device_handle_t spi, const uint16_t data) {
    esp_err_t ret;
    uint16_t tmp_cmd =(data | 0x0100);
    spi_transaction_ext_t trans =(spi_transaction_ext_t) {
            .base = {
                    .flags = SPI_TRANS_VARIABLE_CMD,
                    .cmd = tmp_cmd,
            },
            .command_bits = 9,
    };
    ret = spi_device_transmit(spi,(spi_transaction_t *)&trans);

    assert(ret == ESP_OK);          //Should have had no issues.
}

static void SPI_WriteComm(uint16_t cmd) {
    lcd_cmd(g_screen_spi, cmd);
}

static void SPI_WriteData(uint16_t data) {
    lcd_data(g_screen_spi, data);
}

static void rgb_driver_init(void) {
    //these commands taken from the Arduino GFX library ST7701 type7 init instructions:

    SPI_WriteComm(0xFF);
    SPI_WriteData(0x77);
    SPI_WriteData(0x01);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x10);

    SPI_WriteComm(0xC0);  //FOUND
    SPI_WriteData(0x3B);//Scan line
    SPI_WriteData(0x00);

    SPI_WriteComm(0xC1);  //FOUND
    SPI_WriteData(0x0B);	//VBP
    SPI_WriteData(0x02);

    SPI_WriteComm(0xC2);  //FOUND
    SPI_WriteData(0x07);
    SPI_WriteData(0x02);

    SPI_WriteComm(0xCC);  //FOUND
    SPI_WriteData(0x10);

    SPI_WriteComm(0xCD);//RGB format  //FOUND
    SPI_WriteData(0x08);        //?565???    666??

    SPI_WriteComm(0xB0); // IPS  FOUND
    SPI_WriteData(0x00); // 255
    SPI_WriteData(0x11); // 251
    SPI_WriteData(0x16); // 247  down
    SPI_WriteData(0x0e); // 239
    SPI_WriteData(0x11); // 231
    SPI_WriteData(0x06); // 203
    SPI_WriteData(0x05); // 175
    SPI_WriteData(0x09); // 147
    SPI_WriteData(0x08); // 108
    SPI_WriteData(0x21); // 80
    SPI_WriteData(0x06); // 52
    SPI_WriteData(0x13); // 24
    SPI_WriteData(0x10); // 16
    SPI_WriteData(0x29); // 8    down
    SPI_WriteData(0x31); // 4
    SPI_WriteData(0x18); // 0

    SPI_WriteComm(0xB1);//  IPS FOUND
    SPI_WriteData(0x00);//  255
    SPI_WriteData(0x11);//  251
    SPI_WriteData(0x16);//  247   down
    SPI_WriteData(0x0e);//  239
    SPI_WriteData(0x11);//  231
    SPI_WriteData(0x07);//  203
    SPI_WriteData(0x05);//  175
    SPI_WriteData(0x09);//  147
    SPI_WriteData(0x09);//  108
    SPI_WriteData(0x21);//  80
    SPI_WriteData(0x05);//  52
    SPI_WriteData(0x13);//  24
    SPI_WriteData(0x11);//  16
    SPI_WriteData(0x2a);//  8  down
    SPI_WriteData(0x31);//  4
    SPI_WriteData(0x18);//  0

    SPI_WriteComm(0xFF);
    SPI_WriteData(0x77);
    SPI_WriteData(0x01);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x11);

    SPI_WriteComm(0xB0);  //VOP  3.5375+ *x 0.0125
    SPI_WriteData(0x6d);  //5D

    SPI_WriteComm(0xB1); 	//VCOM amplitude setting
    SPI_WriteData(0x37);  //

    SPI_WriteComm(0xB2); 	//VGH Voltage setting
    SPI_WriteData(0x81);	//12V

    SPI_WriteComm(0xB3);
    SPI_WriteData(0x80);

    SPI_WriteComm(0xB5); 	//VGL Voltage setting
    SPI_WriteData(0x43);	//-8.3V

    SPI_WriteComm(0xB7);
    SPI_WriteData(0x85);

    SPI_WriteComm(0xB8);
    SPI_WriteData(0x20);

    SPI_WriteComm(0xC1);
    SPI_WriteData(0x78);

    SPI_WriteComm(0xC2);
    SPI_WriteData(0x78);

    SPI_WriteComm(0xD0);
    SPI_WriteData(0x88);

    SPI_WriteComm(0xE0);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x02);

    SPI_WriteComm(0xE1);
    SPI_WriteData(0x03);
    SPI_WriteData(0xA0);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x04);
    SPI_WriteData(0xA0);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x20);
    SPI_WriteData(0x20);

    SPI_WriteComm(0xE2);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);

    SPI_WriteComm(0xE3);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x11);
    SPI_WriteData(0x00);

    SPI_WriteComm(0xE4);
    SPI_WriteData(0x22);
    SPI_WriteData(0x00);

    SPI_WriteComm(0xE5);
    SPI_WriteData(0x05);
    SPI_WriteData(0xEC);
    SPI_WriteData(0xA0);
    SPI_WriteData(0xA0);
    SPI_WriteData(0x07);
    SPI_WriteData(0xEE);
    SPI_WriteData(0xA0);
    SPI_WriteData(0xA0);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);

    SPI_WriteComm(0xE6);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x11);
    SPI_WriteData(0x00);

    SPI_WriteComm(0xE7);
    SPI_WriteData(0x22);
    SPI_WriteData(0x00);

    SPI_WriteComm(0xE8);
    SPI_WriteData(0x06);
    SPI_WriteData(0xED);
    SPI_WriteData(0xA0);
    SPI_WriteData(0xA0);
    SPI_WriteData(0x08);
    SPI_WriteData(0xEF);
    SPI_WriteData(0xA0);
    SPI_WriteData(0xA0);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);

    SPI_WriteComm(0xEB);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x40);
    SPI_WriteData(0x40);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);

    SPI_WriteComm(0xED);
    SPI_WriteData(0xFF);
    SPI_WriteData(0xFF);
    SPI_WriteData(0xFF);
    SPI_WriteData(0xBA);
    SPI_WriteData(0x0A);
    SPI_WriteData(0xBF);
    SPI_WriteData(0x45);
    SPI_WriteData(0xFF);
    SPI_WriteData(0xFF);
    SPI_WriteData(0x54);
    SPI_WriteData(0xFB);
    SPI_WriteData(0xA0);
    SPI_WriteData(0xAB);
    SPI_WriteData(0xFF);
    SPI_WriteData(0xFF);
    SPI_WriteData(0xFF);

    SPI_WriteComm(0xEF);
    SPI_WriteData(0x10);
    SPI_WriteData(0x0D);
    SPI_WriteData(0x04);
    SPI_WriteData(0x08);
    SPI_WriteData(0x3F);
    SPI_WriteData(0x1F);

    SPI_WriteComm(0xFF);
    SPI_WriteData(0x77);
    SPI_WriteData(0x01);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x13);

    SPI_WriteComm(0xEF);
    SPI_WriteData(0x08);

    SPI_WriteComm(0xFF);
    SPI_WriteData(0x77);
    SPI_WriteData(0x01);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);
    SPI_WriteData(0x00);

    SPI_WriteComm(0x36);
    SPI_WriteData(0x00);

    SPI_WriteComm(0x3A);
    SPI_WriteData(0x66);  //55/50=16bit(RGB565);66=18bit(RGB666);77?????3AH?=24bit(RGB888)

    SPI_WriteComm(0x11);
    SPI_WriteData(0x00);

    SPI_WriteComm(0x11);

    vTaskDelay(pdMS_TO_TICKS(120));
    SPI_WriteComm(0x29);
    vTaskDelay(pdMS_TO_TICKS(20));
}

void qmsd_rgb_spi_init() {
    spi_bus_config_t buscfg = {
            .sclk_io_num = LCD_SPI_CLK,
            .mosi_io_num = LCD_SPI_DATA0,
            .miso_io_num = -1,
            .quadwp_io_num = -1,
            .quadhd_io_num = -1,
            .max_transfer_sz = 2046,  // was 10*1024
    };
    ESP_ERROR_CHECK(spi_bus_initialize(SPI2_HOST, &buscfg, SPI_DMA_CH_AUTO));  // was SPI_DMA_CH_AUTO

    spi_device_interface_config_t devcfg = {
            .clock_speed_hz = 6000000,   //Clock out at 10 MHz? originally SPI_MASTER_FREQ_10M
            .mode = 0,                               //SPI mode 0?
            .spics_io_num = LCD_SPI_CS,              //CS pin
            .queue_size = 7,                         //We want to be able to queue 7 transactions at a time
    };

    ESP_ERROR_CHECK(spi_bus_add_device(SPI2_HOST, &devcfg, &g_screen_spi));
    rgb_driver_init();

    spi_bus_remove_device(g_screen_spi);
    spi_bus_free(SPI2_HOST);
}

void __qmsd_encoder_read(lv_indev_drv_t *drv, lv_indev_data_t *data)
{
    static int16_t cont_last = 0;
    int16_t cont_now = mt8901_get_count();
    data->enc_diff = ECO_STEP(cont_now - cont_last);
    cont_last = cont_now;
    if (button_isPressed(g_btn)){
        data->state = LV_INDEV_STATE_PR;
    } else {
        data->state = LV_INDEV_STATE_REL;
    }
}

void __qsmd_encoder_init(void)
{
    static lv_indev_drv_t indev_drv;

    g_btn = button_attch(3, 1, 10);
    mt8901_init(5,6);

    lv_indev_drv_init(&indev_drv);
    indev_drv.read_cb = __qmsd_encoder_read;
    indev_drv.type = LV_INDEV_TYPE_ENCODER;
    lv_indev_drv_register(&indev_drv);
}

void screen_init(void) {
    gpio_config_t bk_gpio_config = {
            .mode = GPIO_MODE_OUTPUT,
            .pin_bit_mask = 1ULL << LCD_PIN_BK_LIGHT
    };
    // Initialize the GPIO of backlight
    ESP_ERROR_CHECK(gpio_config(&bk_gpio_config));

    ESP_ERROR_CHECK(gpio_set_level(LCD_PIN_BK_LIGHT, LCD_BK_LIGHT_OFF_LEVEL));

    qmsd_rgb_spi_init();

    esp_lcd_rgb_panel_config_t panel_config = {
            .data_width = 16,
            .bits_per_pixel = 16,  //added
            .psram_trans_align = 32, //originally 64
            .sram_trans_align = 8,  //originally not defined
            .pclk_gpio_num = LCD_PCLK_GPIO,
            .vsync_gpio_num = LCD_VSYNC_GPIO,
            .hsync_gpio_num = LCD_HSYNC_GPIO,
            .de_gpio_num = LCD_DE_GPIO,
            .disp_gpio_num = LCD_DISP_EN_GPIO,
            .data_gpio_nums = {
                    LCD_DATA0_GPIO,
                    LCD_DATA1_GPIO,
                    LCD_DATA2_GPIO,
                    LCD_DATA3_GPIO,
                    LCD_DATA4_GPIO,
                    LCD_DATA5_GPIO,
                    LCD_DATA6_GPIO,
                    LCD_DATA7_GPIO,
                    LCD_DATA8_GPIO,
                    LCD_DATA9_GPIO,
                    LCD_DATA10_GPIO,
                    LCD_DATA11_GPIO,
                    LCD_DATA12_GPIO,
                    LCD_DATA13_GPIO,
                    LCD_DATA14_GPIO,
                    LCD_DATA15_GPIO,
            },
            .timings = {
                    .pclk_hz = 12000000,  //originally 15000000
                    .h_res = 480,
                    .v_res = 480,
                    .hsync_pulse_width = 10,   //originally 10
                    .hsync_back_porch = 10,  // originally 10
                    .hsync_front_porch = 10, // originally 10
                    .vsync_pulse_width = 2,  // originally 2
                    .vsync_back_porch = 12,  // originally 12
                    .vsync_front_porch = 14,  // originally 14
                    .flags.hsync_idle_low = 0,
                    .flags.vsync_idle_low = 0,
                    .flags.de_idle_high = 0,
                    .flags.pclk_active_neg = 0,
                    .flags.pclk_idle_high = 0,
            },
            .flags.fb_in_psram = 1,
            .flags.double_fb = 1,
            .flags.disp_active_low = 0,
            .flags.refresh_on_demand = 0,   // Manually control refresh operation
            .bounce_buffer_size_px = 0,  // originally 0
            .clk_src = LCD_CLK_SRC_PLL240M,  //originally LCD_CLK_SRC_PLL160M
    };

    qmsd_rgb_init(&panel_config);
    __qsmd_encoder_init();

    ESP_ERROR_CHECK(gpio_set_level(LCD_PIN_BK_LIGHT, LCD_BK_LIGHT_ON_LEVEL));
}