/*
 * SPDX-FileCopyrightText: 2022-2023 Espressif Systems (Shanghai) CO LTD
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#ifndef _IOT_SCREEN_DRIVER_H_
#define _IOT_SCREEN_DRIVER_H_

#include "scr_interface_driver.h"

#ifdef __cplusplus
extern "C" {
#endif

/* Color definitions, RGB565 format */
#define COLOR_BLACK       0x0000
#define COLOR_NAVY        0x000F
#define COLOR_DARKGREEN   0x03E0
#define COLOR_DARKCYAN    0x03EF
#define COLOR_MAROON      0x7800
#define COLOR_PURPLE      0x780F
#define COLOR_OLIVE       0x7BE0
#define COLOR_LIGHTGREY   0xC618
#define COLOR_DARKGREY    0x7BEF
#define COLOR_BLUE        0x001F
#define COLOR_GREEN       0x07E0
#define COLOR_CYAN        0x07FF
#define COLOR_RED         0xF800
#define COLOR_MAGENTA     0xF81F
#define COLOR_YELLOW      0xFFE0
#define COLOR_WHITE       0xFFFF
#define COLOR_ORANGE      0xFD20
#define COLOR_GREENYELLOW 0xAFE5
#define COLOR_PINK        0xF81F
#define COLOR_SILVER      0xC618
#define COLOR_GRAY        0x8410
#define COLOR_LIME        0x07E0
#define COLOR_TEAL        0x0410
#define COLOR_FUCHSIA     0xF81F
#define COLOR_ESP_BKGD    0xD185

/**
 * @brief Define all screen direction
 *
 */
typedef enum {
    /* @---> X
       |
       Y
    */
    SCR_DIR_LRTB,   /**< From left to right then from top to bottom, this consider as the original direction of the screen */

    /*  Y
        |
        @---> X
    */
    SCR_DIR_LRBT,   /**< From left to right then from bottom to top */

    /* X <---@
             |
             Y
    */
    SCR_DIR_RLTB,   /**< From right to left then from top to bottom */

    /*       Y
             |
       X <---@
    */
    SCR_DIR_RLBT,   /**< From right to left then from bottom to top */

    /* @---> Y
       |
       X
    */
    SCR_DIR_TBLR,   /**< From top to bottom then from left to right */

    /*  X
        |
        @---> Y
    */
    SCR_DIR_BTLR,   /**< From bottom to top then from left to right */

    /* Y <---@
             |
             X
    */
    SCR_DIR_TBRL,   /**< From top to bottom then from right to left */

    /*       X
             |
       Y <---@
    */
    SCR_DIR_BTRL,   /**< From bottom to top then from right to left */

    SCR_DIR_MAX,

    /* Another way to represent rotation with 3 bit*/
    SCR_MIRROR_X = 0x40, /**< Mirror X-axis */
    SCR_MIRROR_Y = 0x20, /**< Mirror Y-axis */
    SCR_SWAP_XY  = 0x80, /**< Swap XY axis */
} scr_dir_t;

/**
 * @brief The types of colors that can be displayed on the screen
 *
 */
typedef enum {
    SCR_COLOR_TYPE_MONO,     /**< The screen is monochrome */
    SCR_COLOR_TYPE_GRAY,     /**< The screen is gray */
    SCR_COLOR_TYPE_RGB565,   /**< The screen is colorful */
} scr_color_type_t;

/**
 * @brief All supported screen controllers
 *
 */
typedef enum {
    /* color screen */
    SCREEN_CONTROLLER_ILI9341,
    SCREEN_CONTROLLER_ILI9342,
    SCREEN_CONTROLLER_ILI9806,
    SCREEN_CONTROLLER_ILI9486,
    SCREEN_CONTROLLER_ILI9488,
    SCREEN_CONTROLLER_NT35510,
    SCREEN_CONTROLLER_RM68120,
    SCREEN_CONTROLLER_ST7789,
    SCREEN_CONTROLLER_ST7796,
    SCREEN_CONTROLLER_SSD1351,
    SCREEN_CONTROLLER_SSD1963,

    /* monochrome screen */
    SCREEN_CONTROLLER_SSD1306,
    SCREEN_CONTROLLER_SSD1307,
    SCREEN_CONTROLLER_SSD1322,

} scr_controller_t;

/**
 * @brief configuration of screen controller
 *
 */
typedef struct {
    scr_interface_driver_t *interface_drv;   /*!< Interface driver for screen */
    int8_t pin_num_rst;                      /*!< Pin to hardreset LCD*/
    int8_t pin_num_bckl;                     /*!< Pin for control backlight */
    uint8_t rst_active_level;                /*!< Reset pin active level */
    uint8_t bckl_active_level;               /*!< Backlight active level */
    uint16_t width;                          /*!< Screen width */
    uint16_t height;                         /*!< Screen height */
    uint16_t offset_hor;                     /*!< Offset of horizontal */
    uint16_t offset_ver;                     /*!< Offset of vertical */
    scr_dir_t rotate;                        /*!< Screen rotate direction */
} scr_controller_config_t;

/**
 * @brief Information of screen
 *
 */
typedef struct {
    uint16_t width;                      /*!< Current screen width, it may change when apply to rotate */
    uint16_t height;                     /*!< Current screen height, it may change when apply to rotate */
    scr_dir_t dir;                       /*!< Current screen direction */
    scr_color_type_t color_type;         /*!< Color type of the screen, See scr_color_type_t struct */
    uint8_t bpp;                         /*!< Bits per pixel */
    const char *name;                    /*!< Name of the screen */
} scr_info_t;

/**
 * @brief Define a screen common function
 *
 */
typedef struct {
    /**
    * @brief Initialize screen
    *
    * @param lcd_conf Pointer to a structure with lcd config arguments. see struct scr_controller_config_t
    *
    * @return
    *      - ESP_OK on success
    *      - ESP_FAIL Driver not installed
    */
    esp_err_t (*init)(const scr_controller_config_t *lcd_conf);

    /**
    * @brief Deinitialize screen
    *
    * @return
    *      - ESP_OK on success
    *      - ESP_FAIL Deinitialize failed
    *      - ESP_ERR_NOT_SUPPORTED unsupported
    */
    esp_err_t (*deinit)(void);

    /**
    * @brief Set screen direction of rotation
    *
    * @param dir Pointer to a scr_dir_t structure.
    * You can set the direction in two ways, for example, set it to "SCR_DIR_LRBT" or "SCR_MIRROR_Y", They are the same, depending on which expression you want to use
    *
    * @note Not all screens support eight directions, it depends on the screen controller.
    *
    * @return
    *      - ESP_OK on success
    *      - ESP_FAIL Failed
    */
    esp_err_t (*set_direction)(scr_dir_t dir);

    /**
    * @brief Set screen window
    *
    * @param x0 Starting point in X direction
    * @param y0 Starting point in Y direction
    * @param x1 End point in X direction
    * @param y1 End point in Y direction
    *
    * @note When the BPP of the screen controller is less than 8, the coordinate value is limited to a multiple of some number
    *
    * @return
    *      - ESP_OK on success
    *      - ESP_FAIL Failed
    */
    esp_err_t (*set_window)(uint16_t x0, uint16_t y0, uint16_t x1, uint16_t y1);

    /**
    * @brief Write a RAM data
    *
    * @param color New color of a pixel
    *
    * @return
    *      - ESP_OK on success
    *      - ESP_FAIL Failed
    */
    esp_err_t (*write_ram_data)(uint16_t color);

    /**
    * @brief Draw one pixel in screen with color
    *
    * @param x X co-ordinate of set orientation
    * @param y Y co-ordinate of set orientation
    * @param color New color of the pixel
    *
    * @return
    *      - ESP_OK on success
    *      - ESP_FAIL Failed
    */
    esp_err_t (*draw_pixel)(uint16_t x, uint16_t y, uint16_t color);

    /**
    * @brief Fill the pixels on LCD screen with bitmap
    *
    * @param x Starting point in X direction
    * @param y Starting point in Y direction
    * @param w width of image in bitmap array
    * @param h height of image in bitmap array
    * @param bitmap pointer to bitmap array
    *
    * @return
    *      - ESP_OK on success
    *      - ESP_FAIL Failed
    */
    esp_err_t (*draw_bitmap)(uint16_t x, uint16_t y, uint16_t w, uint16_t h, uint16_t *bitmap);

    /**
    * @brief Get screen information
    *
    * @param info Pointer to a scr_info_t structure.
    *
    * @return
    *      - ESP_OK on success
    *      - ESP_FAIL Failed
    */
    esp_err_t (*get_info)(scr_info_t *info);
} scr_driver_t;

/**
 * @brief Find a screen driver
 *
 * @param controller Screen controller to initialize
 * @param out_screen Pointer to a screen driver
 *
 * @return
 *      - ESP_OK on success
 *      - ESP_ERR_INVALID_ARG Arguments is NULL.
 *      - ESP_ERR_NOT_FOUND Screen controller was not found.
 */
esp_err_t scr_find_driver(scr_controller_t controller, scr_driver_t *out_screen);

#ifdef __cplusplus
}
#endif

#endif