X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=test%2Fimage_test.cc;h=7428d51830e87d9201a6fe91f12094e08f890584;hb=da61b62283a95f5937c51024027caed08a358f46;hp=55fbb990774b8c2009eb1852c5e7fb5157341639;hpb=b67a3505b58ffd204ee1a2560be3026688be8f10;p=dcpomatic.git diff --git a/test/image_test.cc b/test/image_test.cc index 55fbb9907..7428d5183 100644 --- a/test/image_test.cc +++ b/test/image_test.cc @@ -1,31 +1,34 @@ /* - Copyright (C) 2012-2014 Carl Hetherington + Copyright (C) 2012-2017 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. - This program is distributed in the hope that it will be useful, + DCP-o-matic is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + along with DCP-o-matic. If not, see . */ /** @file test/image_test.cc - * @brief Tests of the Image class. - * + * @brief Test Image class. + * @ingroup selfcontained * @see test/make_black_test.cc, test/pixel_formats_test.cc */ -#include -#include #include "lib/image.h" +#include "lib/magick_image_proxy.h" +#include "test.h" +#include +#include #include using std::string; @@ -132,58 +135,57 @@ BOOST_AUTO_TEST_CASE (compact_image_test) delete u; } -/** Test Image::alpha_blend */ -BOOST_AUTO_TEST_CASE (alpha_blend_test) +void +alpha_blend_test_one (AVPixelFormat format, string suffix) { - int const stride = 48 * 4; + shared_ptr proxy (new MagickImageProxy (private_data / "prophet_frame.tiff")); + shared_ptr raw = proxy->image(); + shared_ptr background = raw->scale (raw->size(), dcp::YUV_TO_RGB_REC709, format, true, false); - shared_ptr A (new Image (AV_PIX_FMT_RGBA, dcp::Size (48, 48), false)); - A->make_black (); - uint8_t* a = A->data()[0]; + shared_ptr overlay (new Image (AV_PIX_FMT_RGBA, raw->size(), true)); + overlay->make_transparent (); - for (int y = 0; y < 48; ++y) { - uint8_t* p = a + y * stride; - for (int x = 0; x < 16; ++x) { + for (int y = 0; y < 128; ++y) { + uint8_t* p = overlay->data()[0] + y * overlay->stride()[0]; + for (int x = 0; x < 128; ++x) { p[x * 4] = 255; - p[(x + 16) * 4 + 1] = 255; - p[(x + 32) * 4 + 2] = 255; + p[x * 4 + 3] = 255; } } - shared_ptr B (new Image (AV_PIX_FMT_RGBA, dcp::Size (48, 48), true)); - B->make_transparent (); - uint8_t* b = B->data()[0]; - - for (int y = 32; y < 48; ++y) { - uint8_t* p = b + y * stride; - for (int x = 0; x < 48; ++x) { - p[x * 4] = 255; + for (int y = 128; y < 256; ++y) { + uint8_t* p = overlay->data()[0] + y * overlay->stride()[0]; + for (int x = 0; x < 128; ++x) { p[x * 4 + 1] = 255; - p[x * 4 + 2] = 255; p[x * 4 + 3] = 255; } } - A->alpha_blend (B, Position (0, 0)); - - for (int y = 0; y < 32; ++y) { - uint8_t* p = a + y * stride; - for (int x = 0; x < 16; ++x) { - BOOST_CHECK_EQUAL (p[x * 4], 255); - BOOST_CHECK_EQUAL (p[(x + 16) * 4 + 1], 255); - BOOST_CHECK_EQUAL (p[(x + 32) * 4 + 2], 255); + for (int y = 256; y < 384; ++y) { + uint8_t* p = overlay->data()[0] + y * overlay->stride()[0]; + for (int x = 0; x < 128; ++x) { + p[x * 4 + 2] = 255; + p[x * 4 + 3] = 255; } } - for (int y = 32; y < 48; ++y) { - uint8_t* p = a + y * stride; - for (int x = 0; x < 48; ++x) { - BOOST_CHECK_EQUAL (p[x * 4], 255); - BOOST_CHECK_EQUAL (p[x * 4 + 1], 255); - BOOST_CHECK_EQUAL (p[x * 4 + 2], 255); - BOOST_CHECK_EQUAL (p[x * 4 + 3], 255); - } - } + background->alpha_blend (overlay, Position (0, 0)); + + shared_ptr save = background->scale (background->size(), dcp::YUV_TO_RGB_REC709, AV_PIX_FMT_RGB24, false, false); + + write_image (save, "build/test/image_test_" + suffix + ".png", "RGB"); + check_image ("build/test/image_test_" + suffix + ".png", private_data / ("image_test_" + suffix + ".png")); +} + +/** Test Image::alpha_blend */ +BOOST_AUTO_TEST_CASE (alpha_blend_test) +{ + alpha_blend_test_one (AV_PIX_FMT_RGB24, "rgb24"); + alpha_blend_test_one (AV_PIX_FMT_BGRA, "bgra"); + alpha_blend_test_one (AV_PIX_FMT_RGBA, "rgba"); + alpha_blend_test_one (AV_PIX_FMT_RGB48LE, "rgb48le"); + alpha_blend_test_one (AV_PIX_FMT_YUV420P, "yuv420p"); + alpha_blend_test_one (AV_PIX_FMT_YUV420P10LE, "yuv420p10le"); } /** Test merge (list) with a single image */ @@ -253,3 +255,14 @@ BOOST_AUTO_TEST_CASE (merge_test2) BOOST_CHECK_EQUAL (m[(x + 32) * 4 + 3], 255); } } + +/** Test Image::crop_scale_window with YUV420P and some windowing */ +BOOST_AUTO_TEST_CASE (crop_scale_window_test) +{ + shared_ptr proxy(new MagickImageProxy("test/data/flat_red.png")); + shared_ptr raw = proxy->image(); + shared_ptr out = raw->crop_scale_window(Crop(), dcp::Size(1998, 836), dcp::Size(1998, 1080), dcp::YUV_TO_RGB_REC709, AV_PIX_FMT_YUV420P, true, false); + shared_ptr save = out->scale(dcp::Size(1998, 1080), dcp::YUV_TO_RGB_REC709, AV_PIX_FMT_RGB24, false, false); + write_image(save, "build/test/crop_scale_window_test.png", "RGB"); + check_image("test/data/crop_scale_window_test.png", "build/test/crop_scale_window_test.png"); +}