#if LV_BUILD_TEST
#include "../lvgl.h"
#include "../../lvgl_private.h"


#include "unity/unity.h"

/*Bypassing resolution check*/
#define TEST_DISPLAY_ROTATION_ASSERT_EQUAL_SCREENSHOT(path) TEST_ASSERT_MESSAGE(lv_test_screenshot_compare(path), path);

#ifndef NON_AMD64_BUILD
    #define EXT_NAME ".lp64.png"
#else
    #define EXT_NAME ".lp32.png"
#endif

typedef void (*draw_cb_t)(lv_layer_t *, const lv_matrix_t *);

void setUp(void)
{
    /* Function run before every test */
}

void tearDown(void)
{
    /* Test the cleanup of a specified name */
    lv_draw_unit_send_event("VG_LITE", LV_EVENT_CANCEL, NULL);

    /* Test all cleanup */
    lv_draw_unit_send_event(NULL, LV_EVENT_CANCEL, NULL);

    lv_obj_clean(lv_screen_active());
}

static void draw_shapes(lv_layer_t * layer, const lv_matrix_t * transform)
{
    lv_draw_vector_dsc_t * ctx = lv_draw_vector_dsc_create(layer);
    lv_draw_vector_dsc_set_transform(ctx, transform);

    lv_area_t rect = {0, 0, 640, 480};
    lv_draw_vector_dsc_set_fill_color(ctx, lv_color_white());
    rect = lv_matrix_transform_area(transform, &rect);
    lv_draw_vector_dsc_clear_area(ctx, &rect);

    lv_vector_path_t * path = lv_vector_path_create(LV_VECTOR_PATH_QUALITY_MEDIUM);

    lv_area_t rect1 = {50, 50, 150, 150};
    lv_draw_vector_dsc_set_transform(ctx, transform);
    lv_vector_path_clear(path);
    lv_vector_path_append_rect(path, &rect1, 0, 0);
    lv_draw_vector_dsc_set_fill_color(ctx, lv_color_make(0xff, 0x00, 0x00));
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_vector_path_clear(path);
    lv_vector_path_append_rect(path, &rect1, 20, 20);
    lv_draw_vector_dsc_translate(ctx, 150, 0);
    lv_draw_vector_dsc_set_fill_color(ctx, lv_color_make(0x00, 0x00, 0xff));
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_fpoint_t pc = {100, 100};
    lv_vector_path_clear(path);
    lv_vector_path_append_circle(path, &pc, 50, 50);
    lv_draw_vector_dsc_translate(ctx, 150, 0);
    lv_draw_vector_dsc_set_fill_color32(ctx, lv_color_to_32(lv_color_make(0x00, 0xff, 0x00), 0x80));
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_draw_vector_dsc_set_transform(ctx, transform);
    lv_draw_vector_dsc_translate(ctx, 0, 150);

    lv_grad_stop_t stops[2];
    lv_memzero(stops, sizeof(stops));
    stops[0].color = lv_color_hex(0xffffff);
    stops[0].opa = LV_OPA_COVER;
    stops[0].frac = 0;
    stops[1].color = lv_color_hex(0x000000);
    stops[1].opa = LV_OPA_COVER;
    stops[1].frac = 255;

    lv_matrix_t mt;
    lv_matrix_identity(&mt);
    lv_draw_vector_dsc_set_fill_transform(ctx, &mt);
    lv_draw_vector_dsc_set_fill_radial_gradient(ctx, 100, 100, 50);
    lv_draw_vector_dsc_set_fill_gradient_color_stops(ctx, stops, 2);
    lv_draw_vector_dsc_set_fill_gradient_spread(ctx, LV_VECTOR_GRADIENT_SPREAD_PAD);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_draw_vector_dsc_set_transform(ctx, transform);
    lv_draw_vector_dsc_translate(ctx, 150, 150);
    lv_draw_image_dsc_t img_dsc;
    lv_draw_image_dsc_init(&img_dsc);
    LV_IMAGE_DECLARE(test_image_cogwheel_argb8888);
    img_dsc.header = test_image_cogwheel_argb8888.header;
    img_dsc.src = &test_image_cogwheel_argb8888;
    lv_draw_vector_dsc_set_fill_image(ctx, &img_dsc);
    lv_matrix_identity(&mt);
    lv_matrix_translate(&mt, 50, 50);
    //lv_matrix_rotate(&mt, 45);
    //lv_matrix_scale(&mt, 1.5f, 1.5f);
    lv_matrix_translate(&mt, -50, -50);
    lv_draw_vector_dsc_set_fill_transform(ctx, &mt);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_draw_vector_dsc_set_transform(ctx, transform);
    lv_draw_vector_dsc_translate(ctx, 300, 150);
    lv_draw_vector_dsc_set_fill_color(ctx, lv_color_make(0xff, 0x00, 0x00));
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_vector_path_clear(path);
    lv_vector_path_append_rect(path, &rect1, 20, 20);
    lv_area_t rect2 = {80, 80, 120, 120};
    lv_vector_path_append_rect(path, &rect2, 15, 15);
    lv_draw_vector_dsc_translate(ctx, 50, 0);
    lv_draw_vector_dsc_set_fill_color(ctx, lv_color_make(0x00, 0x80, 0xff));
    lv_draw_vector_dsc_set_blend_mode(ctx, LV_VECTOR_BLEND_MULTIPLY);
    lv_draw_vector_dsc_set_fill_rule(ctx, LV_VECTOR_FILL_EVENODD);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_vector_path_clear(path);
    lv_draw_vector_dsc_set_transform(ctx, transform);
    lv_draw_vector_dsc_set_blend_mode(ctx, LV_VECTOR_BLEND_SRC_OVER);

    rect = (lv_area_t) {
        500, 50, 550, 100
    };
    lv_draw_vector_dsc_set_fill_color(ctx, lv_color_lighten(lv_color_black(), 50));
    rect = lv_matrix_transform_area(transform, &rect);
    lv_draw_vector_dsc_clear_area(ctx, &rect); // clear screen

    lv_fpoint_t p = {500, 100}; /* Center */
    lv_draw_vector_dsc_set_stroke_color(ctx, lv_color_make(0x00, 0xff, 0xff));
    lv_draw_vector_dsc_set_stroke_opa(ctx, LV_OPA_COVER);
    lv_draw_vector_dsc_set_stroke_width(ctx, 2.0f);
    lv_draw_vector_dsc_set_stroke_dash(ctx, NULL, 0);
    lv_draw_vector_dsc_set_fill_opa(ctx, LV_OPA_TRANSP);
    lv_vector_path_move_to(path, &p);
    lv_fpoint_t temp = {p.x + 50, p.y};
    lv_vector_path_line_to(path, &temp);
    lv_vector_path_append_arc(path, &p, 50, 0, -90, false);
    lv_vector_path_line_to(path, &p);
    lv_vector_path_close(path);

    lv_draw_vector_dsc_add_path(ctx, path);

    /* Below code has same effect as above one but with solid fill */
    lv_draw_vector_dsc_set_fill_color(ctx, lv_color_make(0xFF, 0x00, 0x00));
    lv_draw_vector_dsc_set_fill_opa(ctx, LV_OPA_COVER);
    lv_vector_path_clear(path);
    lv_vector_path_append_arc(path, &p, 50, 45, 45, true);
    lv_draw_vector_dsc_add_path(ctx, path); // draw a path

    /* Test image filling with absolute coordinates */
    lv_draw_vector_dsc_set_transform(ctx, transform);
    lv_draw_vector_dsc_set_fill_units(ctx, LV_VECTOR_FILL_UNITS_USER_SPACE_ON_USE);
    lv_draw_vector_dsc_set_fill_image(ctx, &img_dsc);
    lv_draw_vector_dsc_set_fill_opa(ctx, LV_OPA_50);
    lv_draw_vector_dsc_set_stroke_opa(ctx, LV_OPA_TRANSP);
    lv_matrix_identity(&mt);
    lv_matrix_translate(&mt, 50, 350);
    lv_draw_vector_dsc_set_fill_transform(ctx, &mt);

    lv_vector_path_clear(path);
    /* Aligned with translate. Image resolution is 100x100, cropped to 50% of width and height */
    lv_area_t img_area = {50, 350, 50 + 50, 350 + 50};
    lv_vector_path_append_rect(path, &img_area, 0, 0);
    lv_draw_vector_dsc_add_path(ctx, path);

    /* Test PNG image file */
    lv_vector_path_clear(path);
    lv_area_t rect3 = {200, 360, 0, 0};
    lv_area_set_width(&rect3, 200);
    lv_area_set_height(&rect3, 100);
    lv_vector_path_append_rect(path, &rect3, 0, 0);

    lv_draw_image_dsc_init(&img_dsc);
    img_dsc.src = "A:src/test_assets/test_img_lvgl_logo.png";
    lv_draw_vector_dsc_set_transform(ctx, transform);
    lv_draw_vector_dsc_set_fill_units(ctx, LV_VECTOR_FILL_UNITS_OBJECT_BOUNDING_BOX);
    lv_draw_vector_dsc_set_fill_image(ctx, &img_dsc);
    lv_draw_vector_dsc_set_fill_opa(ctx, LV_OPA_COVER);

    lv_matrix_identity(&mt);
    lv_draw_vector_dsc_set_fill_transform(ctx, &mt);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_draw_vector(ctx);
    lv_vector_path_delete(path);
    lv_draw_vector_dsc_delete(ctx);
}

static void draw_lines(lv_layer_t * layer, const lv_matrix_t * transform)
{
    lv_draw_vector_dsc_t * ctx = lv_draw_vector_dsc_create(layer);
    lv_draw_vector_dsc_set_transform(ctx, transform);

    lv_area_t rect = {0, 0, 640, 480};
    lv_draw_vector_dsc_set_fill_color(ctx, lv_color_white());
    rect = lv_matrix_transform_area(transform, &rect);
    lv_draw_vector_dsc_clear_area(ctx, &rect);

    lv_vector_path_t * path = lv_vector_path_create(LV_VECTOR_PATH_QUALITY_MEDIUM);

    lv_draw_vector_dsc_set_stroke_opa(ctx, LV_OPA_COVER);
    lv_draw_vector_dsc_set_fill_opa(ctx, LV_OPA_0);

    lv_fpoint_t pts1[] = {{50, 50}, {400, 50}};
    lv_vector_path_clear(path);
    lv_vector_path_move_to(path, &pts1[0]);
    lv_vector_path_line_to(path, &pts1[1]);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_fpoint_t pts2[] = {{50, 80}, {400, 80}};
    lv_vector_path_clear(path);
    lv_vector_path_move_to(path, &pts2[0]);
    lv_vector_path_line_to(path, &pts2[1]);
    lv_draw_vector_dsc_set_stroke_color(ctx, lv_color_make(0xff, 0x00, 0x00));
    lv_draw_vector_dsc_set_stroke_width(ctx, 3.0f);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_fpoint_t pts3[] = {{50, 120}, {400, 120}};
    lv_vector_path_clear(path);
    lv_vector_path_move_to(path, &pts3[0]);
    lv_vector_path_line_to(path, &pts3[1]);
    lv_draw_vector_dsc_set_stroke_color(ctx, lv_color_make(0x00, 0xff, 0x00));
    lv_draw_vector_dsc_set_stroke_width(ctx, 5.0f);
    float dashes[] = {10, 15, 20, 12};
    lv_draw_vector_dsc_set_stroke_dash(ctx, dashes, 4);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_fpoint_t pts4[] = {{100, 150}, {50, 250}, {150, 250}};
    lv_vector_path_clear(path);
    lv_vector_path_move_to(path, &pts4[0]);
    lv_vector_path_line_to(path, &pts4[1]);
    lv_vector_path_line_to(path, &pts4[2]);
    lv_draw_vector_dsc_set_stroke_color(ctx, lv_color_make(0x00, 0x00, 0x00));
    lv_draw_vector_dsc_set_stroke_width(ctx, 10.0f);
    lv_draw_vector_dsc_set_stroke_dash(ctx, NULL, 0);
    lv_draw_vector_dsc_set_stroke_join(ctx, LV_VECTOR_STROKE_JOIN_MITER);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_draw_vector_dsc_translate(ctx, 150, 0);
    lv_draw_vector_dsc_set_stroke_join(ctx, LV_VECTOR_STROKE_JOIN_BEVEL);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_draw_vector_dsc_translate(ctx, 150, 0);
    lv_draw_vector_dsc_set_stroke_join(ctx, LV_VECTOR_STROKE_JOIN_ROUND);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_fpoint_t pts5[] = {{50, 300}, {150, 300}};
    lv_draw_vector_dsc_set_transform(ctx, transform);
    lv_vector_path_clear(path);
    lv_vector_path_move_to(path, &pts5[0]);
    lv_vector_path_line_to(path, &pts5[1]);
    lv_draw_vector_dsc_set_stroke_cap(ctx, LV_VECTOR_STROKE_CAP_BUTT);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_draw_vector_dsc_translate(ctx, 0, 40);
    lv_draw_vector_dsc_set_stroke_cap(ctx, LV_VECTOR_STROKE_CAP_SQUARE);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_draw_vector_dsc_translate(ctx, 0, 40);
    lv_draw_vector_dsc_set_stroke_cap(ctx, LV_VECTOR_STROKE_CAP_ROUND);
    lv_draw_vector_dsc_add_path(ctx, path);

    lv_area_t rect1 = {250, 300, 350, 400};
    lv_draw_vector_dsc_set_transform(ctx, transform);
    lv_vector_path_clear(path);
    lv_vector_path_append_rect(path, &rect1, 0, 0);

    lv_grad_stop_t stops[2];
    lv_memzero(stops, sizeof(stops));
    stops[0].color = lv_color_hex(0xff0000);
    stops[0].opa = LV_OPA_COVER;
    stops[0].frac = 0;
    stops[1].color = lv_color_hex(0x00ff00);
    stops[1].opa = LV_OPA_COVER;
    stops[1].frac = 255;

    lv_matrix_t mt;
    lv_matrix_identity(&mt);
    lv_matrix_rotate(&mt, 5);
    lv_matrix_translate(&mt, 20, 20);
    lv_draw_vector_dsc_set_stroke_transform(ctx, &mt);
    lv_draw_vector_dsc_set_stroke_join(ctx, LV_VECTOR_STROKE_JOIN_MITER);
    lv_draw_vector_dsc_set_stroke_linear_gradient(ctx, 250, 300, 350, 300);
    lv_draw_vector_dsc_set_stroke_gradient_color_stops(ctx, stops, 2);
    lv_draw_vector_dsc_set_stroke_gradient_spread(ctx, LV_VECTOR_GRADIENT_SPREAD_REFLECT);
    lv_draw_vector_dsc_add_path(ctx, path); // draw a path

    lv_draw_vector(ctx);
    lv_vector_path_delete(path);
    lv_draw_vector_dsc_delete(ctx);
}

static void canvas_draw(const char * name, draw_cb_t draw_cb)
{
    LV_UNUSED(name);
    lv_obj_t * canvas = lv_canvas_create(lv_screen_active());
    lv_draw_buf_t * draw_buf = lv_draw_buf_create(640, 480, LV_COLOR_FORMAT_ARGB8888, LV_STRIDE_AUTO);
    TEST_ASSERT_NOT_NULL(draw_buf);

    lv_draw_buf_clear(draw_buf, NULL);
    lv_canvas_set_draw_buf(canvas, draw_buf);
    lv_canvas_fill_bg(canvas, lv_color_make(0xff, 0xff, 0xff), 255);

    lv_layer_t layer;
    lv_canvas_init_layer(canvas, &layer);

    lv_matrix_t matrix;
    lv_matrix_identity(&matrix);
    draw_cb(&layer, &matrix);

    lv_canvas_finish_layer(canvas, &layer);

    char fn_buf[64];
    lv_snprintf(fn_buf, sizeof(fn_buf), "draw/vector_%s" EXT_NAME, name);
    TEST_ASSERT_EQUAL_SCREENSHOT(fn_buf);

    lv_image_cache_drop(draw_buf);
    lv_draw_buf_destroy(draw_buf);
    lv_obj_delete(canvas);
}

void test_transform(void)
{
    lv_matrix_t matrix;
    lv_matrix_identity(&matrix);
    lv_matrix_translate(&matrix, 100, 100);

    lv_fpoint_t p = {10, 10};
    lv_matrix_transform_point(&matrix, &p);

    TEST_ASSERT_EQUAL_FLOAT(110.0f, p.x);
    TEST_ASSERT_EQUAL_FLOAT(110.0f, p.y);

    lv_vector_path_t * path = lv_vector_path_create(LV_VECTOR_PATH_QUALITY_MEDIUM);
    lv_vector_path_move_to(path, &p);

    lv_fpoint_t p2 = {20, 20};
    lv_vector_path_line_to(path, &p2);
    lv_matrix_transform_path(&matrix, path);

    lv_fpoint_t * pt = lv_array_at(&path->points, 0);

    TEST_ASSERT_EQUAL_FLOAT(210.0f, pt[0].x);
    TEST_ASSERT_EQUAL_FLOAT(210.0f, pt[0].y);
    TEST_ASSERT_EQUAL_FLOAT(120.0f, pt[1].x);
    TEST_ASSERT_EQUAL_FLOAT(120.0f, pt[1].y);

    lv_vector_path_delete(path);
}

void test_draw_lines(void)
{
    canvas_draw("draw_lines", draw_lines);
}

void test_draw_shapes(void)
{
    canvas_draw("draw_shapes", draw_shapes);
}

static void event_cb(lv_event_t * e)
{
    lv_layer_t * layer = lv_event_get_layer(e);
    lv_obj_t * obj = lv_event_get_current_target_obj(e);
    draw_cb_t draw_cb = (draw_cb_t)(lv_uintptr_t)lv_event_get_user_data(e);
    LV_ASSERT_NULL(draw_cb);

    lv_matrix_t matrix;
    lv_matrix_identity(&matrix);
    lv_matrix_translate(&matrix, obj->coords.x1, obj->coords.y1);
    draw_cb(layer, &matrix);
}

static void draw_during_rendering(const char * name, draw_cb_t draw_cb, lv_opa_t opa)
{
    lv_obj_t * obj = lv_obj_create(lv_screen_active());
    lv_obj_set_size(obj, lv_pct(90), lv_pct(90));
    lv_obj_center(obj);
    lv_obj_set_style_opa(obj, opa, 0);
    lv_obj_add_event_cb(obj, event_cb, LV_EVENT_DRAW_MAIN, (void *)(lv_uintptr_t)draw_cb);

    char fn_buf[128];
    lv_snprintf(fn_buf, sizeof(fn_buf), "draw/vector_draw_%s_during_rendering" EXT_NAME, name);

    TEST_DISPLAY_ROTATION_ASSERT_EQUAL_SCREENSHOT(fn_buf);

    lv_obj_delete(obj);
}

void test_draw_during_rendering(void)
{
    draw_during_rendering("shapes", draw_shapes, LV_OPA_COVER);
    draw_during_rendering("lines", draw_lines, LV_OPA_COVER);
    draw_during_rendering("shapes_opa_50", draw_shapes, LV_OPA_50);
    draw_during_rendering("lines_opa_50", draw_lines, LV_OPA_50);
}

void test_draw_display_matrix_rotation(void)
{
#if LV_DRAW_TRANSFORM_USE_MATRIX
    lv_display_t * disp = lv_display_get_default();
    lv_display_set_matrix_rotation(disp, true);
    TEST_ASSERT_TRUE(lv_display_get_matrix_rotation(disp));

    lv_display_set_rotation(disp, LV_DISPLAY_ROTATION_0);
    draw_during_rendering("shapes_disp_rotation_0", draw_shapes, LV_OPA_COVER);
    draw_during_rendering("lines_disp_rotation_0", draw_lines, LV_OPA_COVER);

    lv_display_set_rotation(disp, LV_DISPLAY_ROTATION_90);
    draw_during_rendering("shapes_disp_rotation_90", draw_shapes, LV_OPA_COVER);
    draw_during_rendering("lines_disp_rotation_90", draw_lines, LV_OPA_COVER);

    lv_display_set_rotation(disp, LV_DISPLAY_ROTATION_180);
    draw_during_rendering("shapes_disp_rotation_180", draw_shapes, LV_OPA_COVER);
    draw_during_rendering("lines_disp_rotation_180", draw_lines, LV_OPA_COVER);

    lv_display_set_rotation(disp, LV_DISPLAY_ROTATION_270);
    draw_during_rendering("shapes_disp_rotation_270", draw_shapes, LV_OPA_COVER);
    draw_during_rendering("lines_disp_rotation_270", draw_lines, LV_OPA_COVER);

    lv_display_set_matrix_rotation(disp, false);
    TEST_ASSERT_FALSE(lv_display_get_matrix_rotation(disp));
    lv_display_set_rotation(disp, LV_DISPLAY_ROTATION_0);
    draw_during_rendering("shapes_disp_rotation_0", draw_shapes, LV_OPA_COVER);
    draw_during_rendering("lines_disp_rotation_0", draw_lines, LV_OPA_COVER);
#else
    TEST_PASS();
#endif
}

#endif
