/*******************************************************************************
 * This file is part of the "Enduro2D"
 * For conditions of distribution and use, see copyright notice in LICENSE.md
 * Copyright (C) 2018 Matvey Cherevko
 ******************************************************************************/

#include "_utils.hpp"
using namespace e2d;

TEST_CASE("images") {
    {
        image i;
        REQUIRE(i.size() == v2u::zero());
        REQUIRE(i.format() == image_data_format::rgba8);
        REQUIRE(i.data().empty());
        REQUIRE(i.empty());
    }
    {
        const u8 img[] = {1,2,3,4,5,6,7,8};
        image i0(v2u(2,2), image_data_format::g8, {img,4});
        image i1(v2u(2,1), image_data_format::ga8, {img,4});
        image i2(v2u(1,2), image_data_format::ga8, {img,4});
        image i3(v2u(2,1), image_data_format::rgb8, {img,6});
        image i4(v2u(1,2), image_data_format::rgb8, {img,6});
        image i5(v2u(2,1), image_data_format::rgba8, {img,8});
        image i6(v2u(1,2), image_data_format::rgba8, {img,8});
        REQUIRE(i0.format() == image_data_format::g8);
        REQUIRE(i1.format() == image_data_format::ga8);
        REQUIRE(i2.format() == image_data_format::ga8);
        REQUIRE(i3.format() == image_data_format::rgb8);
        REQUIRE(i4.format() == image_data_format::rgb8);
        REQUIRE(i5.format() == image_data_format::rgba8);
        REQUIRE(i6.format() == image_data_format::rgba8);
    }
    {
        REQUIRE(filesystem::remove_file("image_save_test.jpg"));
        REQUIRE(filesystem::remove_file("image_save_test.png"));
        REQUIRE(filesystem::remove_file("image_save_test.tga"));
        const u8 img_data[] = {255,0,0, 0,255,0, 0,0,255};
        image img(v2u(3,1), image_data_format::rgb8, buffer(img_data, sizeof(img_data)));
        REQUIRE(images::try_save_image(
            img,
            image_file_format::jpg,
            make_write_file("image_save_test.jpg", false)));
        REQUIRE(filesystem::file_exists("image_save_test.jpg"));
        REQUIRE(images::try_save_image(
            img,
            image_file_format::png,
            make_write_file("image_save_test.png", false)));
        REQUIRE(filesystem::file_exists("image_save_test.png"));
        REQUIRE(images::try_save_image(
            img,
            image_file_format::tga,
            make_write_file("image_save_test.tga", false)));
        REQUIRE(filesystem::file_exists("image_save_test.tga"));
    }
    {
        image img;
//        https://github.com/nothings/stb/issues/706
//        REQUIRE(images::try_load_image(img, make_read_file("image_save_test.jpg")));
//        REQUIRE(img.size() == v2u(3,1));
//        REQUIRE(img.format() == image_data_format::rgb8);
//        REQUIRE(math::approximately(img.pixel32(0,0), color32::red(),   10u));
//        REQUIRE(math::approximately(img.pixel32(1,0), color32::green(), 10u));
//        REQUIRE(math::approximately(img.pixel32(2,0), color32::blue(),  10u));

        REQUIRE(images::try_load_image(img, make_read_file("image_save_test.png")));
        REQUIRE(img.size() == v2u(3,1));
        REQUIRE(img.format() == image_data_format::rgb8);
        REQUIRE(math::approximately(img.pixel32(0,0), color32::red(),   0u));
        REQUIRE(math::approximately(img.pixel32(1,0), color32::green(), 0u));
        REQUIRE(math::approximately(img.pixel32(2,0), color32::blue(),  0u));

        REQUIRE(images::try_load_image(img, make_read_file("image_save_test.tga")));
        REQUIRE(img.size() == v2u(3,1));
        REQUIRE(img.format() == image_data_format::rgb8);
        REQUIRE(math::approximately(img.pixel32(0,0), color32::red(),   0u));
        REQUIRE(math::approximately(img.pixel32(1,0), color32::green(), 0u));
        REQUIRE(math::approximately(img.pixel32(2,0), color32::blue(),  0u));
    }
    {
        REQUIRE(filesystem::remove_file("image_save_test.jpg"));
        REQUIRE(filesystem::remove_file("image_save_test.png"));
        REQUIRE(filesystem::remove_file("image_save_test.tga"));
        const u8 img_data[] = {255,0,0, 0,255,0, 0,0,255};
        image img(v2u(3,1), image_data_format::rgb8, buffer(img_data, sizeof(img_data)));
        buffer buf;
        REQUIRE(images::try_save_image(
            img,
            image_file_format::jpg,
            buf));
        REQUIRE(filesystem::try_write_all(buf, "image_save_test.jpg", false));
        REQUIRE(images::try_save_image(
            img,
            image_file_format::png,
            buf));
        REQUIRE(filesystem::try_write_all(buf, "image_save_test.png", false));
        REQUIRE(images::try_save_image(
            img,
            image_file_format::tga,
            buf));
        REQUIRE(filesystem::try_write_all(buf, "image_save_test.tga", false));
    }
    {
        image img;
        buffer buf;
//        https://github.com/nothings/stb/issues/706
//        REQUIRE(filesystem::try_read_all(buf, "image_save_test.jpg"));
//        REQUIRE(images::try_load_image(img, buf));
//        REQUIRE(img.size() == v2u(3,1));
//        REQUIRE(img.format() == image_data_format::rgb8);
//        REQUIRE(math::approximately(img.pixel32(0,0), color32::red(),   10));
//        REQUIRE(math::approximately(img.pixel32(1,0), color32::green(), 10));
//        REQUIRE(math::approximately(img.pixel32(2,0), color32::blue(),  10));

        REQUIRE(filesystem::try_read_all(buf, "image_save_test.png"));
        REQUIRE(images::try_load_image(img, buf));
        REQUIRE(img.size() == v2u(3,1));
        REQUIRE(img.format() == image_data_format::rgb8);
        REQUIRE(math::approximately(img.pixel32(0,0), color32::red(),   0));
        REQUIRE(math::approximately(img.pixel32(1,0), color32::green(), 0));
        REQUIRE(math::approximately(img.pixel32(2,0), color32::blue(),  0));

        REQUIRE(filesystem::try_read_all(buf, "image_save_test.tga"));
        REQUIRE(images::try_load_image(img, buf));
        REQUIRE(img.size() == v2u(3,1));
        REQUIRE(img.format() == image_data_format::rgb8);
        REQUIRE(math::approximately(img.pixel32(0,0), color32::red(),   0));
        REQUIRE(math::approximately(img.pixel32(1,0), color32::green(), 0));
        REQUIRE(math::approximately(img.pixel32(2,0), color32::blue(),  0));
    }
}