#include "catch.hpp"

// mapnik
#include <mapnik/image_any.hpp>
#include <mapnik/color.hpp>
#include <mapnik/image_util.hpp>

TEST_CASE("image premultiply")
{
    SECTION("test rgba8")
    {
        mapnik::image_rgba8 im(4, 4);
        mapnik::image_rgba8 im2(4, 4, true, true); // Initialize as already premultiplied
        mapnik::image_any im_any(mapnik::image_rgba8(4, 4));
        mapnik::image_any im2_any(mapnik::image_rgba8(4, 4, true, true));

        // First test that the default state is correct for each
        CHECK_FALSE(im.get_premultiplied());
        CHECK_FALSE(im_any.get_premultiplied());
        CHECK(im2.get_premultiplied());
        CHECK(im2_any.get_premultiplied());

        // Set the image to premultiplied
        im.set_premultiplied(true);
        CHECK(im.get_premultiplied());
        // Set back to not premultiplied
        im.set_premultiplied(false);
        CHECK_FALSE(im.get_premultiplied());

        mapnik::set_premultiplied_alpha(im, true);
        CHECK(im.get_premultiplied());
        mapnik::set_premultiplied_alpha(im, false);
        CHECK_FALSE(im.get_premultiplied());

        // Fill the images with meaningfull values
        mapnik::color c1(57, 70, 128, 128);       // This color is not premultiplied
        mapnik::color c2(57, 70, 128, 128, true); // This color is premultiplied
        mapnik::fill(im, c2);                     // Because c2 is premultiplied it will make the image premultiplied
        mapnik::fill(im_any, c2);                 // Because c2 is premultiplied it will make the image premultiplied
        mapnik::fill(im2, c1);     // Because c1 is not premultiplied it will make the image not premultiplied
        mapnik::fill(im2_any, c1); // Because c1 is not premultiplied it will make the image not premultiplied

        // Demultipled via image_util
        CHECK(mapnik::demultiply_alpha(im));            // Should return true as was premultiplied
        CHECK(mapnik::demultiply_alpha(im_any));        // Should return true as was premultiplied
        CHECK_FALSE(mapnik::demultiply_alpha(im2));     // Should return false as was demultiplied
        CHECK_FALSE(mapnik::demultiply_alpha(im2_any)); // Should return false as was demultiplied

        mapnik::color out;
        // This will be higher because it became demultiplied!
        out = mapnik::get_pixel<mapnik::color>(im, 0, 0);
        CHECK(static_cast<int>(out.red()) == 113);
        CHECK(static_cast<int>(out.green()) == 139);
        CHECK(static_cast<int>(out.blue()) == 255);
        CHECK(static_cast<int>(out.alpha()) == 128);
        out = mapnik::get_pixel<mapnik::color>(im_any, 0, 0);
        CHECK(static_cast<int>(out.red()) == 113);
        CHECK(static_cast<int>(out.green()) == 139);
        CHECK(static_cast<int>(out.blue()) == 255);
        CHECK(static_cast<int>(out.alpha()) == 128);
        // This will be the same because it was already demultiplied
        out = mapnik::get_pixel<mapnik::color>(im2, 0, 0);
        CHECK(static_cast<int>(out.red()) == 57);
        CHECK(static_cast<int>(out.green()) == 70);
        CHECK(static_cast<int>(out.blue()) == 128);
        CHECK(static_cast<int>(out.alpha()) == 128);
        out = mapnik::get_pixel<mapnik::color>(im2_any, 0, 0);
        CHECK(static_cast<int>(out.red()) == 57);
        CHECK(static_cast<int>(out.green()) == 70);
        CHECK(static_cast<int>(out.blue()) == 128);
        CHECK(static_cast<int>(out.alpha()) == 128);

        // Set back to im2s to "premultiplied" with out changing underlying values
        mapnik::set_premultiplied_alpha(im2, true);
        mapnik::set_premultiplied_alpha(im2_any, true);

        // Demultipled via image_util
        CHECK(mapnik::premultiply_alpha(im));            // Should return true as was demultiplied
        CHECK(mapnik::premultiply_alpha(im_any));        // Should return true as was demultiplied
        CHECK_FALSE(mapnik::premultiply_alpha(im2));     // Should return false as was premultiplied
        CHECK_FALSE(mapnik::premultiply_alpha(im2_any)); // Should return false as was premultiplied

        // This will be the same because it was already demultiplied
        out = mapnik::get_pixel<mapnik::color>(im, 0, 0);
        CHECK(static_cast<int>(out.red()) == 57);
        CHECK(static_cast<int>(out.green()) == 70);
        CHECK(static_cast<int>(out.blue()) == 128);
        CHECK(static_cast<int>(out.alpha()) == 128);
        out = mapnik::get_pixel<mapnik::color>(im_any, 0, 0);
        CHECK(static_cast<int>(out.red()) == 57);
        CHECK(static_cast<int>(out.green()) == 70);
        CHECK(static_cast<int>(out.blue()) == 128);
        CHECK(static_cast<int>(out.alpha()) == 128);
        // This will be the same because it was already demultiplied
        out = mapnik::get_pixel<mapnik::color>(im2, 0, 0);
        CHECK(static_cast<int>(out.red()) == 57);
        CHECK(static_cast<int>(out.green()) == 70);
        CHECK(static_cast<int>(out.blue()) == 128);
        CHECK(static_cast<int>(out.alpha()) == 128);
        out = mapnik::get_pixel<mapnik::color>(im2_any, 0, 0);
        CHECK(static_cast<int>(out.red()) == 57);
        CHECK(static_cast<int>(out.green()) == 70);
        CHECK(static_cast<int>(out.blue()) == 128);
        CHECK(static_cast<int>(out.alpha()) == 128);

    } // END SECTION

    SECTION("test gray8")
    {
        mapnik::image_gray8 im(4, 4);
        mapnik::image_gray8 im2(4, 4, true, true); // Initialize as already premultiplied
        mapnik::image_any im_any(mapnik::image_gray8(4, 4));
        mapnik::image_any im2_any(mapnik::image_gray8(4, 4, true, true));

        // First test that the default state is correct for each
        CHECK_FALSE(im.get_premultiplied());
        CHECK_FALSE(im_any.get_premultiplied());
        CHECK(im2.get_premultiplied());
        CHECK(im2_any.get_premultiplied());

        // Set the image to premultiplied
        im.set_premultiplied(true);
        CHECK(im.get_premultiplied());
        // Set back to not premultiplied
        im.set_premultiplied(false);
        CHECK_FALSE(im.get_premultiplied());

        mapnik::set_premultiplied_alpha(im, true);
        CHECK(im.get_premultiplied());
        mapnik::set_premultiplied_alpha(im, false);
        CHECK_FALSE(im.get_premultiplied());

        // Always fails on demultiply since its gray8
        CHECK_FALSE(mapnik::demultiply_alpha(im));
        CHECK_FALSE(mapnik::demultiply_alpha(im_any));
        CHECK_FALSE(mapnik::demultiply_alpha(im2));
        CHECK_FALSE(mapnik::demultiply_alpha(im2_any));

        // Always fails on premultiply since its gray8
        CHECK_FALSE(mapnik::premultiply_alpha(im));
        CHECK_FALSE(mapnik::premultiply_alpha(im_any));
        CHECK_FALSE(mapnik::premultiply_alpha(im2));
        CHECK_FALSE(mapnik::premultiply_alpha(im2_any));

    } // END SECTION
} // END TEST_CASE