X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=test%2Fimage_test.cc;h=6a35be07f315b369073073d13ae0ef61b4981f8f;hb=f1d209319f90adc50e0f3d2a853216052f290bd5;hp=82ebe0b1ff5e6825c0db29bbd92f8fc2495092f8;hpb=d6be9f27147113dc6355ed0de7c99d1312aaeebd;p=dcpomatic.git diff --git a/test/image_test.cc b/test/image_test.cc index 82ebe0b1f..6a35be07f 100644 --- a/test/image_test.cc +++ b/test/image_test.cc @@ -1,32 +1,45 @@ /* - Copyright (C) 2012 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 . */ -#include +/** @file test/image_test.cc + * @brief Test Image class. + * @ingroup selfcontained + * @see test/make_black_test.cc, test/pixel_formats_test.cc + */ + #include "lib/image.h" -#include "lib/scaler.h" +#include "lib/magick_image_proxy.h" +#include "test.h" +#include +#include +#include +using std::string; +using std::list; +using std::cout; using boost::shared_ptr; BOOST_AUTO_TEST_CASE (aligned_image_test) { - Image* s = new Image (PIX_FMT_RGB24, libdcp::Size (50, 50), true); - BOOST_CHECK_EQUAL (s->components(), 1); + Image* s = new Image (AV_PIX_FMT_RGB24, dcp::Size (50, 50), true); + BOOST_CHECK_EQUAL (s->planes(), 1); /* 160 is 150 aligned to the nearest 32 bytes */ BOOST_CHECK_EQUAL (s->stride()[0], 160); BOOST_CHECK_EQUAL (s->line_size()[0], 150); @@ -37,7 +50,7 @@ BOOST_AUTO_TEST_CASE (aligned_image_test) /* copy constructor */ Image* t = new Image (*s); - BOOST_CHECK_EQUAL (t->components(), 1); + BOOST_CHECK_EQUAL (t->planes(), 1); BOOST_CHECK_EQUAL (t->stride()[0], 160); BOOST_CHECK_EQUAL (t->line_size()[0], 150); BOOST_CHECK (t->data()[0]); @@ -47,14 +60,14 @@ BOOST_AUTO_TEST_CASE (aligned_image_test) BOOST_CHECK (t->data() != s->data()); BOOST_CHECK (t->data()[0] != s->data()[0]); BOOST_CHECK (t->line_size() != s->line_size()); - BOOST_CHECK (t->line_size()[0] == s->line_size()[0]); + BOOST_CHECK_EQUAL (t->line_size()[0], s->line_size()[0]); BOOST_CHECK (t->stride() != s->stride()); - BOOST_CHECK (t->stride()[0] == s->stride()[0]); + BOOST_CHECK_EQUAL (t->stride()[0], s->stride()[0]); /* assignment operator */ - Image* u = new Image (PIX_FMT_YUV422P, libdcp::Size (150, 150), false); + Image* u = new Image (AV_PIX_FMT_YUV422P, dcp::Size (150, 150), false); *u = *s; - BOOST_CHECK_EQUAL (u->components(), 1); + BOOST_CHECK_EQUAL (u->planes(), 1); BOOST_CHECK_EQUAL (u->stride()[0], 160); BOOST_CHECK_EQUAL (u->line_size()[0], 150); BOOST_CHECK (u->data()[0]); @@ -64,9 +77,9 @@ BOOST_AUTO_TEST_CASE (aligned_image_test) BOOST_CHECK (u->data() != s->data()); BOOST_CHECK (u->data()[0] != s->data()[0]); BOOST_CHECK (u->line_size() != s->line_size()); - BOOST_CHECK (u->line_size()[0] == s->line_size()[0]); + BOOST_CHECK_EQUAL (u->line_size()[0], s->line_size()[0]); BOOST_CHECK (u->stride() != s->stride()); - BOOST_CHECK (u->stride()[0] == s->stride()[0]); + BOOST_CHECK_EQUAL (u->stride()[0], s->stride()[0]); delete s; delete t; @@ -75,8 +88,8 @@ BOOST_AUTO_TEST_CASE (aligned_image_test) BOOST_AUTO_TEST_CASE (compact_image_test) { - Image* s = new Image (PIX_FMT_RGB24, libdcp::Size (50, 50), false); - BOOST_CHECK_EQUAL (s->components(), 1); + Image* s = new Image (AV_PIX_FMT_RGB24, dcp::Size (50, 50), false); + BOOST_CHECK_EQUAL (s->planes(), 1); BOOST_CHECK_EQUAL (s->stride()[0], 50 * 3); BOOST_CHECK_EQUAL (s->line_size()[0], 50 * 3); BOOST_CHECK (s->data()[0]); @@ -86,7 +99,7 @@ BOOST_AUTO_TEST_CASE (compact_image_test) /* copy constructor */ Image* t = new Image (*s); - BOOST_CHECK_EQUAL (t->components(), 1); + BOOST_CHECK_EQUAL (t->planes(), 1); BOOST_CHECK_EQUAL (t->stride()[0], 50 * 3); BOOST_CHECK_EQUAL (t->line_size()[0], 50 * 3); BOOST_CHECK (t->data()[0]); @@ -96,14 +109,14 @@ BOOST_AUTO_TEST_CASE (compact_image_test) BOOST_CHECK (t->data() != s->data()); BOOST_CHECK (t->data()[0] != s->data()[0]); BOOST_CHECK (t->line_size() != s->line_size()); - BOOST_CHECK (t->line_size()[0] == s->line_size()[0]); + BOOST_CHECK_EQUAL (t->line_size()[0], s->line_size()[0]); BOOST_CHECK (t->stride() != s->stride()); - BOOST_CHECK (t->stride()[0] == s->stride()[0]); + BOOST_CHECK_EQUAL (t->stride()[0], s->stride()[0]); /* assignment operator */ - Image* u = new Image (PIX_FMT_YUV422P, libdcp::Size (150, 150), true); + Image* u = new Image (AV_PIX_FMT_YUV422P, dcp::Size (150, 150), true); *u = *s; - BOOST_CHECK_EQUAL (u->components(), 1); + BOOST_CHECK_EQUAL (u->planes(), 1); BOOST_CHECK_EQUAL (u->stride()[0], 50 * 3); BOOST_CHECK_EQUAL (u->line_size()[0], 50 * 3); BOOST_CHECK (u->data()[0]); @@ -113,51 +126,128 @@ BOOST_AUTO_TEST_CASE (compact_image_test) BOOST_CHECK (u->data() != s->data()); BOOST_CHECK (u->data()[0] != s->data()[0]); BOOST_CHECK (u->line_size() != s->line_size()); - BOOST_CHECK (u->line_size()[0] == s->line_size()[0]); + BOOST_CHECK_EQUAL (u->line_size()[0], s->line_size()[0]); BOOST_CHECK (u->stride() != s->stride()); - BOOST_CHECK (u->stride()[0] == s->stride()[0]); + BOOST_CHECK_EQUAL (u->stride()[0], s->stride()[0]); delete s; delete t; delete u; } -BOOST_AUTO_TEST_CASE (crop_image_test) +void +alpha_blend_test_one (AVPixelFormat format, string suffix) { - /* This was to check out a bug with valgrind, and is probably not very useful */ - shared_ptr image (new Image (PIX_FMT_YUV420P, libdcp::Size (16, 16), true)); - image->make_black (); - Crop crop; - crop.top = 3; - image->crop (crop, false); + 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 overlay (new Image (AV_PIX_FMT_RGBA, raw->size(), true)); + overlay->make_transparent (); + + 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 * 4 + 3] = 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 + 3] = 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; + } + } + + 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 cropping of a YUV 4:2:0 image by 1 pixel, which used to fail because - the U/V copying was not rounded up to the next sample. -*/ -BOOST_AUTO_TEST_CASE (crop_image_test2) +/** Test Image::alpha_blend */ +BOOST_AUTO_TEST_CASE (alpha_blend_test) { - /* Here's a 1998 x 1080 image which is black */ - shared_ptr image (new Image (PIX_FMT_YUV420P, libdcp::Size (1998, 1080), true)); - image->make_black (); - - /* Crop it by 1 pixel */ - Crop crop; - crop.left = 1; - image = image->crop (crop, true); - - /* Convert it back to RGB to make comparison to black easier */ - image = image->scale (image->size(), Scaler::from_id ("bicubic"), PIX_FMT_RGB24, true); - - /* Check that its still black after the crop */ - uint8_t* p = image->data()[0]; - for (int y = 0; y < image->size().height; ++y) { - uint8_t* q = p; - for (int x = 0; x < image->size().width; ++x) { - BOOST_CHECK_EQUAL (*q++, 0); - BOOST_CHECK_EQUAL (*q++, 0); - BOOST_CHECK_EQUAL (*q++, 0); + alpha_blend_test_one (AV_PIX_FMT_RGBA, "rgba"); + alpha_blend_test_one (AV_PIX_FMT_YUV420P, "yuv420p"); +} + +/** Test merge (list) with a single image */ +BOOST_AUTO_TEST_CASE (merge_test1) +{ + int const stride = 48 * 4; + + shared_ptr A (new Image (AV_PIX_FMT_RGBA, dcp::Size (48, 48), false)); + A->make_transparent (); + uint8_t* a = A->data()[0]; + + for (int y = 0; y < 48; ++y) { + uint8_t* p = a + y * stride; + for (int x = 0; x < 16; ++x) { + /* red */ + p[x * 4] = 255; + /* opaque */ + p[x * 4 + 3] = 255; } - p += image->stride()[0]; + } + + list all; + all.push_back (PositionImage (A, Position (0, 0))); + PositionImage merged = merge (all); + + BOOST_CHECK (merged.position == Position (0, 0)); + BOOST_CHECK_EQUAL (memcmp (merged.image->data()[0], A->data()[0], stride * 48), 0); +} + +/** Test merge (list) with two images */ +BOOST_AUTO_TEST_CASE (merge_test2) +{ + shared_ptr A (new Image (AV_PIX_FMT_RGBA, dcp::Size (48, 1), false)); + A->make_transparent (); + uint8_t* a = A->data()[0]; + for (int x = 0; x < 16; ++x) { + /* red */ + a[x * 4] = 255; + /* opaque */ + a[x * 4 + 3] = 255; + } + + shared_ptr B (new Image (AV_PIX_FMT_RGBA, dcp::Size (48, 1), false)); + B->make_transparent (); + uint8_t* b = B->data()[0]; + for (int x = 0; x < 16; ++x) { + /* blue */ + b[(x + 32) * 4 + 2] = 255; + /* opaque */ + b[(x + 32) * 4 + 3] = 255; + } + + list all; + all.push_back (PositionImage (A, Position (0, 0))); + all.push_back (PositionImage (B, Position (0, 0))); + PositionImage merged = merge (all); + + BOOST_CHECK (merged.position == Position (0, 0)); + + uint8_t* m = merged.image->data()[0]; + + for (int x = 0; x < 16; ++x) { + BOOST_CHECK_EQUAL (m[x * 4], 255); + BOOST_CHECK_EQUAL (m[x * 4 + 3], 255); + BOOST_CHECK_EQUAL (m[(x + 16) * 4 + 3], 0); + BOOST_CHECK_EQUAL (m[(x + 32) * 4 + 2], 255); + BOOST_CHECK_EQUAL (m[(x + 32) * 4 + 3], 255); } }