//

// display

//

#include "display.h"

#include "esp_err.h"

#include "esp_log.h"

#include "esp_check.h"

#include "driver/i2c.h"

#include "driver/gpio.h"

#include "driver/spi_master.h"

#include "esp_lcd_panel_io.h"

#include "esp_lcd_panel_vendor.h"

#include "esp_lcd_panel_ops.h"

#include "esp_lvgl_port.h"

#include <stdio.h>

#include "esp_timer.h"

#include "esp_heap_caps.h"

#include "freertos/FreeRTOS.h"

#include "freertos/task.h"

// #include "ui_helpers.h"

static const char *TAG = "LVGL_SETUP";

static void lvgl_timer_task(void *arg);

static lv_obj_t *ui_Screen1;

static lv_obj_t *ui_redsquare;

static lv_obj_t *meter;

static lv_obj_t *ue_img_logo;

static lv_obj_t *esp_img_logo;

static lv_obj_t *ui_Dropdown2;

LV_IMG_DECLARE(ue_logo)

LV_IMG_DECLARE(esp_logo)

LV_IMG_DECLARE(red_square)

#define BSP_NULL_CHECK(x, ret) assert(x)

static SemaphoreHandle_t lvgl_mux;  // LVGL mutex

static SemaphoreHandle_t touch_mux; // Touch mutex

static void bsp_touchpad_read(lv_indev_drv_t *drv, lv_indev_data_t *data)

{

    uint16_t touchpad_x[1] = {0};

    uint16_t touchpad_y[1] = {0};

    uint8_t touchpad_cnt = 0;

    /* Read data from touch controller into memory */

    if (xSemaphoreTake(touch_mux, 0) == pdTRUE)

    {

        esp_lcd_touch_read_data(drv->user_data);

    }

    /* Get coordinates */

    bool touchpad_pressed = esp_lcd_touch_get_coordinates(drv->user_data, touchpad_x, touchpad_y, NULL, &touchpad_cnt, 1);

    if (touchpad_pressed && touchpad_cnt > 0)

    {

        data->point.x = touchpad_x[0];

        data->point.y = touchpad_y[0];

        data->state = LV_INDEV_STATE_PRESSED;

        printf("data->point.x = %u \n", data->point.x);

        printf("data->point.y = %u \n", data->point.y);

    }

    else

    {

        data->state = LV_INDEV_STATE_RELEASED;

    }

}

static void touch_callback(esp_lcd_touch_handle_t tp)

{

    BaseType_t xHigherPriorityTaskWoken = pdFALSE;

    xSemaphoreGiveFromISR(touch_mux, &xHigherPriorityTaskWoken);

    if (xHigherPriorityTaskWoken)

    {

        portYIELD_FROM_ISR();

    }

}

static void set_value(void *indic, int32_t v)

{

    lv_meter_set_indicator_end_value(meter, indic, v);

}

static bool example_notify_lvgl_flush_ready(esp_lcd_panel_io_handle_t panel_io, esp_lcd_panel_io_event_data_t *edata, void *user_ctx)

{

    lv_disp_drv_t *disp_driver = (lv_disp_drv_t *)user_ctx;

    lv_disp_flush_ready(disp_driver);

    return false;

}

static void example_lvgl_flush_cb(lv_disp_drv_t *drv, const lv_area_t *area, lv_color_t *color_map)

{

    esp_lcd_panel_handle_t panel_handle = (esp_lcd_panel_handle_t)drv->user_data;

    int offsetx1 = area->x1;

    int offsetx2 = area->x2;

    int offsety1 = area->y1;

    int offsety2 = area->y2;

    // copy a buffer's content to a specific area of the display

    esp_lcd_panel_draw_bitmap(panel_handle, offsetx1, offsety1, offsetx2 + 1, offsety2 + 1, color_map);

}

static void example_increase_lvgl_tick(void *arg)

{

    /* Tell LVGL how many milliseconds has elapsed */

    lv_tick_inc(EXAMPLE_LVGL_TICK_PERIOD_MS);

}

void display_init()

{

    static lv_disp_draw_buf_t disp_buf; // contains internal graphic buffer(s) called draw buffer(s)

    static lv_disp_drv_t disp_drv;      // contains callback functions

    gpio_config_t pwr_gpio_config =

        {

            .mode = GPIO_MODE_OUTPUT,

            .pin_bit_mask = 1ULL << EXAMPLE_PIN_NUM_POWER};

    ESP_ERROR_CHECK(gpio_config(&pwr_gpio_config));

    gpio_set_level(EXAMPLE_PIN_NUM_POWER, EXAMPLE_LCD_BK_LIGHT_ON_LEVEL);

    gpio_config_t input_conf =

        {

            .mode = GPIO_MODE_INPUT,

            .pull_up_en = GPIO_PULLUP_ENABLE,

            .pin_bit_mask = 1ULL << PIN_LCD_RD};

    ESP_ERROR_CHECK(gpio_config(&input_conf));

    ESP_ERROR_CHECK(i2c_driver_install(EXAMPLE_I2C_NUM, i2c_conf.mode, 0, 0, 0));

    i2c_cmd_handle_t cmd;

    for (int i = 0; i < 0x7f; i++)

    {

        cmd = i2c_cmd_link_create();

        i2c_master_start(cmd);

        i2c_master_write_byte(cmd, (i << 1) | I2C_MASTER_WRITE, true);

        i2c_master_stop(cmd);

        if (i2c_master_cmd_begin(EXAMPLE_I2C_NUM, cmd, portMAX_DELAY) == ESP_OK)

        {

            ESP_LOGW("I2C_TEST", "%02X", i);

        }

        i2c_cmd_link_delete(cmd);

    }

    esp_lcd_panel_io_i2c_config_t tp_io_config = ESP_LCD_TOUCH_IO_I2C_CST816S_CONFIG();

    ESP_LOGI(TAG, "esp_lcd_new_panel_io_i2c");

    ESP_ERROR_CHECK(esp_lcd_new_panel_io_i2c((esp_lcd_i2c_bus_handle_t)EXAMPLE_I2C_NUM, &tp_io_config, &tp_io_handle));

    esp_lcd_touch_config_t tp_cfg = {

        .x_max = 170,

        .y_max = 320,

        .rst_gpio_num = 21,

        .int_gpio_num = 16,

        .levels = {

            .reset = 0,

            .interrupt = 0,

        },

        .flags = {

            .swap_xy = 1,

            .mirror_x = 0,

            .mirror_y = 1,

        },

        .interrupt_callback = touch_callback,

    };

    ESP_LOGI(TAG, "esp_lcd_touch_new_i2c_cst816s");

    esp_lcd_touch_new_i2c_cst816s(tp_io_handle, &tp_cfg, &tp);

    // END OF CST816 TOUCH TESTING FUNCTION

#endif

    lv_init();

    lvgl_mux = xSemaphoreCreateMutex();

    BSP_NULL_CHECK(lvgl_mux, NULL);

    touch_mux = xSemaphoreCreateBinary();

    BSP_NULL_CHECK(touch_mux, NULL);

    // it's recommended to choose the size of the draw buffer(s) to be at least 1/10 screen sized

    lv_color_t *buf1 = heap_caps_malloc(LVGL_LCD_BUF_SIZE * sizeof(lv_color_t), MALLOC_CAP_DMA);

    assert(buf1);

    lv_disp_draw_buf_init(&disp_buf, buf1, NULL, LVGL_LCD_BUF_SIZE);

    ESP_LOGI(TAG, "Register display driver to LVGL");

    lv_disp_drv_init(&disp_drv);

    disp_drv.hor_res = EXAMPLE_LCD_H_RES;

    disp_drv.ver_res = EXAMPLE_LCD_V_RES;

    disp_drv.flush_cb = example_lvgl_flush_cb;

    disp_drv.draw_buf = &disp_buf;

    disp_drv.user_data = panel_handle;

    disp_drv.sw_rotate = 1;

    lv_disp_t *disp = lv_disp_drv_register(&disp_drv);

    ESP_LOGI(TAG, "Install LVGL tick timer");

    // Tick interface for LVGL (using esp_timer to generate 2ms periodic event)

    const esp_timer_create_args_t lvgl_tick_timer_args = {

        .callback = &example_increase_lvgl_tick,

        .name = "lvgl_tick"};

    esp_timer_handle_t lvgl_tick_timer = NULL;

    ESP_ERROR_CHECK(esp_timer_create(&lvgl_tick_timer_args, &lvgl_tick_timer));

    ESP_ERROR_CHECK(esp_timer_start_periodic(lvgl_tick_timer, EXAMPLE_LVGL_TICK_PERIOD_MS * 1000));

#if USE_TOUCH_DISPLAY

    static lv_indev_drv_t indev_drv; // Input device driver (Touch)

    lv_indev_t *indev_touchpad;

    lv_indev_drv_init(&indev_drv);

    indev_drv.type = LV_INDEV_TYPE_POINTER;

    indev_drv.disp = disp;

    indev_drv.read_cb = bsp_touchpad_read;

    indev_drv.user_data = tp;

    indev_touchpad = lv_indev_drv_register(&indev_drv);

    BSP_NULL_CHECK(indev_touchpad, ESP_ERR_NO_MEM);

#endif

    xTaskCreatePinnedToCore(lvgl_timer_task, "lvgl Timer", 10000, NULL, 4, NULL, 1);

}

static void lvgl_timer_task(void *arg)

{

    ESP_LOGI(TAG, "Starting LVGL task");

    while (1)

    {

        bsp_display_lock(0);

        uint32_t task_delay_ms = lv_timer_handler();

        bsp_display_unlock();

        if (task_delay_ms > 500)

        {

            task_delay_ms = 500;

        }

        else if (task_delay_ms < 5)

        {

            task_delay_ms = 5;

        }

        vTaskDelay(pdMS_TO_TICKS(task_delay_ms));

    }

}

void display_meter()

{

    meter = lv_meter_create(lv_scr_act());

    lv_obj_center(meter);

    lv_obj_set_size(meter, 170, 170);

    /*Remove the circle from the middle*/

    lv_obj_remove_style(meter, NULL, LV_PART_INDICATOR);

    /*Add a scale first*/

    lv_meter_scale_t *scale = lv_meter_add_scale(meter);

    lv_meter_set_scale_ticks(meter, scale, 11, 2, 10, lv_palette_main(LV_PALETTE_GREY));

    lv_meter_set_scale_major_ticks(meter, scale, 1, 2, 15, lv_color_hex3(0xeee), 10);

    lv_meter_set_scale_range(meter, scale, 0, 100, 270, 90);

    /*Add a three arc indicator*/

    lv_meter_indicator_t *indic1 = lv_meter_add_arc(meter, scale, 10, lv_color_hex3(0x00F), 0);

    // lv_meter_indicator_t *indic2 = lv_meter_add_arc(meter, scale, 10, lv_color_hex3(0x0F0), -10);

    // lv_meter_indicator_t *indic3 = lv_meter_add_arc(meter, scale, 10, lv_color_hex3(0xF00), -20);

    /*Create an animation to set the value*/

    lv_anim_t a;

    lv_anim_init(&a);

    lv_anim_set_exec_cb(&a, set_value);

    lv_anim_set_values(&a, 0, 100);

    lv_anim_set_repeat_delay(&a, 100);

    /*Create a label below the slider*/

    slider_label = lv_label_create(lv_scr_act());

    lv_label_set_text(slider_label, "0%");

    lv_obj_align_to(slider_label, slider, LV_ALIGN_OUT_BOTTOM_MID, 0, 10);

}

static void slider_event_cb(lv_event_t *e)

{

    lv_obj_t *slider = lv_event_get_target(e);

    char buf[8];

    lv_snprintf(buf, sizeof(buf), "%d%%", (int)lv_slider_get_value(slider));

    lv_label_set_text(slider_label, buf);

    lv_obj_align_to(slider_label, slider, LV_ALIGN_OUT_BOTTOM_MID, 0, 10);

}

void get_img_color()

{

    lv_color_t pixel_color = lv_img_buf_get_px_color(&red_square, 50, 50, lv_color_make(0, 0, 0));

    uint32_t c32 = lv_color_to32(pixel_color);

    printf("Pixel color: %lu \n", c32);

}

bool bsp_display_lock(uint32_t timeout_ms)

{

    BSP_NULL_CHECK(lvgl_mux, NULL);

    const TickType_t timeout_ticks = (timeout_ms == 0) ? portMAX_DELAY : pdMS_TO_TICKS(timeout_ms);

    return xSemaphoreTake(lvgl_mux, timeout_ticks) == pdTRUE;

}

void bsp_display_unlock(void)

{

    BSP_NULL_CHECK(lvgl_mux, NULL);

    xSemaphoreGive(lvgl_mux);

}

lv_obj_t *label;

void display_update_text(char* str)

{

    bsp_display_lock(0);

    lv_label_set_text(label, str);

    bsp_display_unlock();

}

void display_process(void)

{

    bsp_display_lock(0);

    lv_label_set_text(label, "Yup");

    // lv_obj_align_to(label, lv_scr_act(), LV_ALIGN_OUT_BOTTOM_MID, 0, 10);

    // display_slider();

    //Call one at a time to see examples

    display_meter();

    // display_image();

    // display_window();

    //display_dropdown();

    bsp_display_unlock();

#define EXAMPLE_LCD_BK_LIGHT_ON_LEVEL 1

#define EXAMPLE_LCD_BK_LIGHT_OFF_LEVEL !EXAMPLE_LCD_BK_LIGHT_ON_LEVEL

#define EXAMPLE_PIN_NUM_DATA0 39    // 6

#define EXAMPLE_PIN_NUM_DATA1 40    // 7

#define EXAMPLE_PIN_NUM_DATA2 41    // 8

#define EXAMPLE_PIN_NUM_DATA3 42    // 9

#define EXAMPLE_PIN_NUM_DATA4 45    // 10

#define EXAMPLE_PIN_NUM_DATA5 46    // 11

#define EXAMPLE_PIN_NUM_DATA6 47    // 12

#define EXAMPLE_PIN_NUM_DATA7 48    // 13

#define EXAMPLE_PIN_NUM_PCLK 8      // 5

#define EXAMPLE_PIN_NUM_CS 6        // 3

#define EXAMPLE_PIN_NUM_DC 7        // 4

#define EXAMPLE_PIN_NUM_RST 5       // 2

#define EXAMPLE_PIN_NUM_BK_LIGHT 38 // 1

#define EXAMPLE_PIN_NUM_POWER 15

#define PIN_LCD_RD 9

// The pixel number in horizontal and vertical

#define EXAMPLE_LCD_H_RES 320

#define EXAMPLE_LCD_V_RES 170

#define LVGL_LCD_BUF_SIZE            (EXAMPLE_LCD_H_RES * EXAMPLE_LCD_V_RES)

// Bit number used to represent command and parameter

#define EXAMPLE_LCD_CMD_BITS 8

#define EXAMPLE_LCD_PARAM_BITS 8

#define EXAMPLE_LVGL_TICK_PERIOD_MS 2

// Supported alignment: 16, 32, 64. A higher alignment can enables higher burst transfer size, thus a higher i80 bus throughput.

#define EXAMPLE_PSRAM_DATA_ALIGNMENT 32

//Change this to 0 if you do not wish to use Touch.

#define USE_TOUCH_DISPLAY 0

void display_meter();

void display_window();

void display_image();

void display_slider();

void display_red_square();

void display_dropdown();

void get_img_color();

void bsp_display_unlock(void);

bool bsp_display_lock(uint32_t timeout_ms);

void display_init(void);

void display_update_text(char* string);

void display_process(void);

#endif

//

// Protofusion ESP32S3 Template

//

#include <stdint.h>

#include "freertos/FreeRTOS.h"

#include "freertos/task.h"

#include "freertos/event_groups.h"

#include "esp_system.h"

#include "esp_event.h"

#include "esp_log.h"

#include "nvs_flash.h"

#include "sdkconfig.h"

#include "wifi.h"

#include "can.h"

#include "display.h"

// Private variables

static const char *TAG = "main";

// Application entry point

void app_main(void)

{

    // Initialize usb-cdc interface

    // Initialize NVS

    esp_err_t ret = nvs_flash_init();

    if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {

        ESP_ERROR_CHECK(nvs_flash_erase());

        ret = nvs_flash_init();

    }

    ESP_ERROR_CHECK(ret);

    // Initialize display

    display_init();

    display_process();

    // Connect to wifi

    wifi_init();

    // Initialize canbus

    //can_init();

    while(1)

    {

    }

}

//

// wifi

//

#include "wifi.h"

#include "esp_wifi.h"

#include "esp_event.h"

#include "esp_log.h"

#include "lwip/err.h"

#include "lwip/sys.h"

#include "display.h"

#include "freertos/FreeRTOS.h"

#include "freertos/task.h"

#include "freertos/event_groups.h"

// Private variables

// FreeRTOS event group to signal when we are connected

static EventGroupHandle_t s_wifi_event_group;

static const char *TAG = "wifi station";

static int s_retry_num = 0;

// Handler for wifi events

static void __event_handler(void* arg, esp_event_base_t event_base, int32_t event_id, void* event_data)

{

    if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_START) {

        esp_wifi_connect();

    }

    else if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_DISCONNECTED) 

    {

        if (s_retry_num < EXAMPLE_ESP_MAXIMUM_RETRY) 

        {

            esp_wifi_connect();

            s_retry_num++;

            ESP_LOGI(TAG, "retry to connect to the AP");

        } 

        else 

        {

            xEventGroupSetBits(s_wifi_event_group, WIFI_FAIL_BIT);

        }

        ESP_LOGI(TAG,"connect to the AP fail");

    } 

    else if (event_base == IP_EVENT && event_id == IP_EVENT_STA_GOT_IP) 

    {

        ip_event_got_ip_t* event = (ip_event_got_ip_t*) event_data;

        char out[128] = {0};

        display_update_text(out);

        s_retry_num = 0;

        xEventGroupSetBits(s_wifi_event_group, WIFI_CONNECTED_BIT);

    }

}

// Initialize wifi and connect to network

void wifi_init(void)

{

    s_wifi_event_group = xEventGroupCreate();

    ESP_ERROR_CHECK(esp_netif_init());

    ESP_ERROR_CHECK(esp_event_loop_create_default());

    esp_netif_create_default_wifi_sta();

    wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();

    ESP_ERROR_CHECK(esp_wifi_init(&cfg));

    esp_event_handler_instance_t instance_any_id;

    esp_event_handler_instance_t instance_got_ip;

    ESP_ERROR_CHECK(esp_event_handler_instance_register(WIFI_EVENT,

                                                        ESP_EVENT_ANY_ID,

                                                        &__event_handler,

                                                        NULL,

                                                        &instance_any_id));

    ESP_ERROR_CHECK(esp_event_handler_instance_register(IP_EVENT,

                                                        IP_EVENT_STA_GOT_IP,

                                                        &__event_handler,

                                                        NULL,

                                                        &instance_got_ip));

    wifi_config_t wifi_config = {

        .sta = {

            .ssid = EXAMPLE_ESP_WIFI_SSID,

            .password = EXAMPLE_ESP_WIFI_PASS,

            /* Authmode threshold resets to WPA2 as default if password matches WPA2 standards (pasword len => 8).

             * If you want to connect the device to deprecated WEP/WPA networks, Please set the threshold value

             * to WIFI_AUTH_WEP/WIFI_AUTH_WPA_PSK and set the password with length and format matching to

             * WIFI_AUTH_WEP/WIFI_AUTH_WPA_PSK standards.

             */

            .threshold.authmode = ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD,

            .sae_pwe_h2e = ESP_WIFI_SAE_MODE,

            .sae_h2e_identifier = EXAMPLE_H2E_IDENTIFIER,

        },

    };

    ESP_ERROR_CHECK(esp_wifi_set_mode(WIFI_MODE_STA) );

    ESP_ERROR_CHECK(esp_wifi_set_config(WIFI_IF_STA, &wifi_config) );