2 Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
4 This file is part of DCP-o-matic.
6 DCP-o-matic is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
11 DCP-o-matic is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with DCP-o-matic. If not, see <http://www.gnu.org/licenses/>.
22 /** @file test/image_test.cc
23 * @brief Test Image class.
24 * @ingroup selfcontained
25 * @see test/pixel_formats_test.cc
29 #include "lib/compose.hpp"
30 #include "lib/image.h"
31 #include "lib/image_content.h"
32 #include "lib/image_decoder.h"
33 #include "lib/image_jpeg.h"
34 #include "lib/image_png.h"
35 #include "lib/ffmpeg_image_proxy.h"
37 #include <boost/test/unit_test.hpp>
43 using std::make_shared;
47 BOOST_AUTO_TEST_CASE (aligned_image_test)
49 auto s = new Image (AV_PIX_FMT_RGB24, dcp::Size (50, 50), Image::Alignment::PADDED);
50 BOOST_CHECK_EQUAL (s->planes(), 1);
51 /* 192 is 150 aligned to the nearest 64 bytes */
52 BOOST_CHECK_EQUAL (s->stride()[0], 192);
53 BOOST_CHECK_EQUAL (s->line_size()[0], 150);
54 BOOST_CHECK (s->data()[0]);
55 BOOST_CHECK (!s->data()[1]);
56 BOOST_CHECK (!s->data()[2]);
57 BOOST_CHECK (!s->data()[3]);
59 /* copy constructor */
60 auto t = new Image (*s);
61 BOOST_CHECK_EQUAL (t->planes(), 1);
62 BOOST_CHECK_EQUAL (t->stride()[0], 192);
63 BOOST_CHECK_EQUAL (t->line_size()[0], 150);
64 BOOST_CHECK (t->data()[0]);
65 BOOST_CHECK (!t->data()[1]);
66 BOOST_CHECK (!t->data()[2]);
67 BOOST_CHECK (!t->data()[3]);
68 BOOST_CHECK (t->data() != s->data());
69 BOOST_CHECK (t->data()[0] != s->data()[0]);
70 BOOST_CHECK (t->line_size() != s->line_size());
71 BOOST_CHECK_EQUAL (t->line_size()[0], s->line_size()[0]);
72 BOOST_CHECK (t->stride() != s->stride());
73 BOOST_CHECK_EQUAL (t->stride()[0], s->stride()[0]);
75 /* assignment operator */
76 auto u = new Image (AV_PIX_FMT_YUV422P, dcp::Size (150, 150), Image::Alignment::COMPACT);
78 BOOST_CHECK_EQUAL (u->planes(), 1);
79 BOOST_CHECK_EQUAL (u->stride()[0], 192);
80 BOOST_CHECK_EQUAL (u->line_size()[0], 150);
81 BOOST_CHECK (u->data()[0]);
82 BOOST_CHECK (!u->data()[1]);
83 BOOST_CHECK (!u->data()[2]);
84 BOOST_CHECK (!u->data()[3]);
85 BOOST_CHECK (u->data() != s->data());
86 BOOST_CHECK (u->data()[0] != s->data()[0]);
87 BOOST_CHECK (u->line_size() != s->line_size());
88 BOOST_CHECK_EQUAL (u->line_size()[0], s->line_size()[0]);
89 BOOST_CHECK (u->stride() != s->stride());
90 BOOST_CHECK_EQUAL (u->stride()[0], s->stride()[0]);
98 BOOST_AUTO_TEST_CASE (compact_image_test)
100 auto s = new Image (AV_PIX_FMT_RGB24, dcp::Size (50, 50), Image::Alignment::COMPACT);
101 BOOST_CHECK_EQUAL (s->planes(), 1);
102 BOOST_CHECK_EQUAL (s->stride()[0], 50 * 3);
103 BOOST_CHECK_EQUAL (s->line_size()[0], 50 * 3);
104 BOOST_CHECK (s->data()[0]);
105 BOOST_CHECK (!s->data()[1]);
106 BOOST_CHECK (!s->data()[2]);
107 BOOST_CHECK (!s->data()[3]);
109 /* copy constructor */
110 auto t = new Image (*s);
111 BOOST_CHECK_EQUAL (t->planes(), 1);
112 BOOST_CHECK_EQUAL (t->stride()[0], 50 * 3);
113 BOOST_CHECK_EQUAL (t->line_size()[0], 50 * 3);
114 BOOST_CHECK (t->data()[0]);
115 BOOST_CHECK (!t->data()[1]);
116 BOOST_CHECK (!t->data()[2]);
117 BOOST_CHECK (!t->data()[3]);
118 BOOST_CHECK (t->data() != s->data());
119 BOOST_CHECK (t->data()[0] != s->data()[0]);
120 BOOST_CHECK (t->line_size() != s->line_size());
121 BOOST_CHECK_EQUAL (t->line_size()[0], s->line_size()[0]);
122 BOOST_CHECK (t->stride() != s->stride());
123 BOOST_CHECK_EQUAL (t->stride()[0], s->stride()[0]);
125 /* assignment operator */
126 auto u = new Image (AV_PIX_FMT_YUV422P, dcp::Size (150, 150), Image::Alignment::PADDED);
128 BOOST_CHECK_EQUAL (u->planes(), 1);
129 BOOST_CHECK_EQUAL (u->stride()[0], 50 * 3);
130 BOOST_CHECK_EQUAL (u->line_size()[0], 50 * 3);
131 BOOST_CHECK (u->data()[0]);
132 BOOST_CHECK (!u->data()[1]);
133 BOOST_CHECK (!u->data()[2]);
134 BOOST_CHECK (!u->data()[3]);
135 BOOST_CHECK (u->data() != s->data());
136 BOOST_CHECK (u->data()[0] != s->data()[0]);
137 BOOST_CHECK (u->line_size() != s->line_size());
138 BOOST_CHECK_EQUAL (u->line_size()[0], s->line_size()[0]);
139 BOOST_CHECK (u->stride() != s->stride());
140 BOOST_CHECK_EQUAL (u->stride()[0], s->stride()[0]);
150 alpha_blend_test_bgra_onto(AVPixelFormat format, string suffix)
152 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "prophet_frame.tiff");
153 auto raw = proxy->image(Image::Alignment::PADDED).image;
154 auto background = raw->convert_pixel_format (dcp::YUVToRGB::REC709, format, Image::Alignment::PADDED, false);
156 auto overlay = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size(431, 891), Image::Alignment::PADDED);
157 overlay->make_transparent ();
159 for (int y = 0; y < 128; ++y) {
160 auto p = overlay->data()[0] + y * overlay->stride()[0];
161 for (int x = 0; x < 128; ++x) {
167 for (int y = 128; y < 256; ++y) {
168 auto p = overlay->data()[0] + y * overlay->stride()[0];
169 for (int x = 0; x < 128; ++x) {
175 for (int y = 256; y < 384; ++y) {
176 auto p = overlay->data()[0] + y * overlay->stride()[0];
177 for (int x = 0; x < 128; ++x) {
183 background->alpha_blend (overlay, Position<int> (13, 17));
185 auto save = background->convert_pixel_format (dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::COMPACT, false);
187 write_image(save, "build/test/image_test_bgra_" + suffix + ".png");
188 check_image("build/test/image_test_bgra_" + suffix + ".png", TestPaths::private_data() / ("image_test_bgra_" + suffix + ".png"));
194 alpha_blend_test_rgba64be_onto(AVPixelFormat format, string suffix)
196 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "prophet_frame.tiff");
197 auto raw = proxy->image(Image::Alignment::PADDED).image;
198 auto background = raw->convert_pixel_format (dcp::YUVToRGB::REC709, format, Image::Alignment::PADDED, false);
200 auto overlay = make_shared<Image>(AV_PIX_FMT_RGBA64BE, dcp::Size(431, 891), Image::Alignment::PADDED);
201 overlay->make_transparent();
203 for (int y = 0; y < 128; ++y) {
204 auto p = reinterpret_cast<uint16_t*>(overlay->data()[0] + y * overlay->stride()[0]);
205 for (int x = 0; x < 128; ++x) {
206 p[x * 4 + 2] = 65535;
207 p[x * 4 + 3] = 65535;
211 for (int y = 128; y < 256; ++y) {
212 auto p = reinterpret_cast<uint16_t*>(overlay->data()[0] + y * overlay->stride()[0]);
213 for (int x = 0; x < 128; ++x) {
214 p[x * 4 + 1] = 65535;
215 p[x * 4 + 3] = 65535;
219 for (int y = 256; y < 384; ++y) {
220 auto p = reinterpret_cast<uint16_t*>(overlay->data()[0] + y * overlay->stride()[0]);
221 for (int x = 0; x < 128; ++x) {
223 p[x * 4 + 3] = 65535;
227 background->alpha_blend(overlay, Position<int>(13, 17));
229 auto save = background->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::COMPACT, false);
231 write_image(save, "build/test/image_test_rgba64_" + suffix + ".png");
232 check_image("build/test/image_test_rgba64_" + suffix + ".png", TestPaths::private_data() / ("image_test_rgba64_" + suffix + ".png"));
236 /** Test Image::alpha_blend */
237 BOOST_AUTO_TEST_CASE (alpha_blend_test)
239 alpha_blend_test_bgra_onto(AV_PIX_FMT_RGB24, "rgb24");
240 alpha_blend_test_bgra_onto(AV_PIX_FMT_BGRA, "bgra");
241 alpha_blend_test_bgra_onto(AV_PIX_FMT_RGBA, "rgba");
242 alpha_blend_test_bgra_onto(AV_PIX_FMT_RGB48LE, "rgb48le");
243 alpha_blend_test_bgra_onto(AV_PIX_FMT_YUV420P, "yuv420p");
244 alpha_blend_test_bgra_onto(AV_PIX_FMT_YUV420P10, "yuv420p10");
245 alpha_blend_test_bgra_onto(AV_PIX_FMT_YUV422P10LE, "yuv422p9le");
246 alpha_blend_test_bgra_onto(AV_PIX_FMT_YUV422P10LE, "yuv422p10le");
248 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_RGB24, "rgb24");
249 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_BGRA, "bgra");
250 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_RGBA, "rgba");
251 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_RGB48LE, "rgb48le");
252 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV420P, "yuv420p");
253 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV420P10, "yuv420p10");
254 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV422P10LE, "yuv422p9le");
255 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV422P10LE, "yuv422p10le");
259 /** Test Image::alpha_blend when blending RGBA onto XYZ12LE */
260 BOOST_AUTO_TEST_CASE(alpha_blend_test_rgba_onto_xyz)
262 Image xyz(AV_PIX_FMT_XYZ12LE, dcp::Size(50, 50), Image::Alignment::PADDED);
265 auto overlay = make_shared<Image>(AV_PIX_FMT_RGBA, dcp::Size(8, 8), Image::Alignment::PADDED);
266 for (int y = 0; y < 8; ++y) {
267 uint8_t* p = overlay->data()[0] + (y * overlay->stride()[0]);
268 for (int x = 0; x < 8; ++x) {
276 xyz.alpha_blend(overlay, Position<int>(4, 4));
278 for (int y = 0; y < 50; ++y) {
279 uint16_t* p = reinterpret_cast<uint16_t*>(xyz.data()[0]) + (y * xyz.stride()[0] / 2);
280 for (int x = 0; x < 50; ++x) {
281 if (4 <= x && x < 12 && 4 <= y && y < 12) {
282 BOOST_REQUIRE_EQUAL(p[0], 45078U);
283 BOOST_REQUIRE_EQUAL(p[1], 34939U);
284 BOOST_REQUIRE_EQUAL(p[2], 13892U);
286 BOOST_REQUIRE_EQUAL(p[0], 0U);
287 BOOST_REQUIRE_EQUAL(p[1], 0U);
288 BOOST_REQUIRE_EQUAL(p[2], 0U);
296 /** Test Image::alpha_blend when blending RGBA64BE onto XYZ12LE */
297 BOOST_AUTO_TEST_CASE(alpha_blend_test_rgba64be_onto_xyz)
299 Image xyz(AV_PIX_FMT_XYZ12LE, dcp::Size(50, 50), Image::Alignment::PADDED);
302 auto overlay = make_shared<Image>(AV_PIX_FMT_RGBA64BE, dcp::Size(8, 8), Image::Alignment::PADDED);
303 for (int y = 0; y < 8; ++y) {
304 auto p = reinterpret_cast<uint16_t*>(overlay->data()[0] + (y * overlay->stride()[0]));
305 for (int x = 0; x < 8; ++x) {
313 xyz.alpha_blend(overlay, Position<int>(4, 4));
315 for (int y = 0; y < 50; ++y) {
316 auto p = reinterpret_cast<uint16_t*>(xyz.data()[0]) + (y * xyz.stride()[0] / 2);
317 for (int x = 0; x < 50; ++x) {
318 if (4 <= x && x < 12 && 4 <= y && y < 12) {
319 BOOST_REQUIRE_EQUAL(p[0], 45078U);
320 BOOST_REQUIRE_EQUAL(p[1], 34939U);
321 BOOST_REQUIRE_EQUAL(p[2], 13892U);
323 BOOST_REQUIRE_EQUAL(p[0], 0U);
324 BOOST_REQUIRE_EQUAL(p[1], 0U);
325 BOOST_REQUIRE_EQUAL(p[2], 0U);
333 BOOST_AUTO_TEST_CASE(alpha_blend_text)
335 Image target(AV_PIX_FMT_RGB24, dcp::Size(1998, 1080), Image::Alignment::PADDED);
338 FFmpegImageProxy subtitle_proxy(TestPaths::private_data() / "16-bit-sub.png");
339 auto subtitle = subtitle_proxy.image(Image::Alignment::COMPACT);
341 target.alpha_blend(subtitle.image, Position<int>(0, 0));
342 write_image(make_shared<Image>(target), "build/test/alpha_blend_text.png");
343 check_image("build/test/alpha_blend_text.png", TestPaths::private_data() / "16-bit-sub-blended.png");
347 /** Test merge (list<PositionImage>) with a single image */
348 BOOST_AUTO_TEST_CASE (merge_test1)
350 int const stride = 48 * 4;
352 auto A = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size (48, 48), Image::Alignment::COMPACT);
353 A->make_transparent ();
354 auto a = A->data()[0];
356 for (int y = 0; y < 48; ++y) {
357 auto p = a + y * stride;
358 for (int x = 0; x < 16; ++x) {
366 list<PositionImage> all;
367 all.push_back (PositionImage (A, Position<int>(0, 0)));
368 auto merged = merge (all, Image::Alignment::COMPACT);
370 BOOST_CHECK (merged.position == Position<int>(0, 0));
371 BOOST_CHECK_EQUAL (memcmp (merged.image->data()[0], A->data()[0], stride * 48), 0);
375 /** Test merge (list<PositionImage>) with two images */
376 BOOST_AUTO_TEST_CASE (merge_test2)
378 auto A = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size (48, 1), Image::Alignment::COMPACT);
379 A->make_transparent ();
380 auto a = A->data()[0];
381 for (int x = 0; x < 16; ++x) {
388 auto B = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size (48, 1), Image::Alignment::COMPACT);
389 B->make_transparent ();
390 auto b = B->data()[0];
391 for (int x = 0; x < 16; ++x) {
393 b[(x + 32) * 4 + 2] = 255;
395 b[(x + 32) * 4 + 3] = 255;
398 list<PositionImage> all;
399 all.push_back (PositionImage(A, Position<int>(0, 0)));
400 all.push_back (PositionImage(B, Position<int>(0, 0)));
401 auto merged = merge (all, Image::Alignment::COMPACT);
403 BOOST_CHECK (merged.position == Position<int>(0, 0));
405 auto m = merged.image->data()[0];
407 for (int x = 0; x < 16; ++x) {
408 BOOST_CHECK_EQUAL (m[x * 4], 255);
409 BOOST_CHECK_EQUAL (m[x * 4 + 3], 255);
410 BOOST_CHECK_EQUAL (m[(x + 16) * 4 + 3], 0);
411 BOOST_CHECK_EQUAL (m[(x + 32) * 4 + 2], 255);
412 BOOST_CHECK_EQUAL (m[(x + 32) * 4 + 3], 255);
417 /** Test Image::crop_scale_window with YUV420P and some windowing */
418 BOOST_AUTO_TEST_CASE (crop_scale_window_test)
420 auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png");
421 auto raw = proxy->image(Image::Alignment::PADDED).image;
422 auto out = raw->crop_scale_window(
423 Crop(), dcp::Size(1998, 836), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_YUV420P, VideoRange::FULL, Image::Alignment::PADDED, false
425 auto save = out->scale(dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::COMPACT, false);
426 write_image(save, "build/test/crop_scale_window_test.png");
427 check_image("test/data/crop_scale_window_test.png", "build/test/crop_scale_window_test.png");
431 /** Special cases of Image::crop_scale_window which triggered some valgrind warnings */
432 BOOST_AUTO_TEST_CASE (crop_scale_window_test2)
434 auto image = make_shared<Image>(AV_PIX_FMT_XYZ12LE, dcp::Size(2048, 858), Image::Alignment::PADDED);
435 image->crop_scale_window (
436 Crop(279, 0, 0, 0), dcp::Size(1069, 448), dcp::Size(1069, 578), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_RGB24, VideoRange::FULL, Image::Alignment::COMPACT, false
438 image->crop_scale_window (
439 Crop(2048, 0, 0, 0), dcp::Size(1069, 448), dcp::Size(1069, 578), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_RGB24, VideoRange::FULL, Image::Alignment::COMPACT, false
444 BOOST_AUTO_TEST_CASE (crop_scale_window_test3)
446 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
447 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
448 auto cropped = xyz->crop_scale_window(
449 Crop(512, 0, 0, 0), dcp::Size(1486, 1080), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_RGB24, VideoRange::FULL, Image::Alignment::COMPACT, false
451 write_image(cropped, "build/test/crop_scale_window_test3.png");
452 check_image("test/data/crop_scale_window_test3.png", "build/test/crop_scale_window_test3.png");
456 BOOST_AUTO_TEST_CASE (crop_scale_window_test4)
458 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
459 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
460 auto cropped = xyz->crop_scale_window(
461 Crop(512, 0, 0, 0), dcp::Size(1486, 1080), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_XYZ12LE, VideoRange::FULL, Image::Alignment::COMPACT, false
463 write_image(cropped, "build/test/crop_scale_window_test4.png");
464 check_image("test/data/crop_scale_window_test4.png", "build/test/crop_scale_window_test4.png");
468 BOOST_AUTO_TEST_CASE (crop_scale_window_test5)
470 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
471 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_XYZ12LE, Image::Alignment::PADDED, false);
472 auto cropped = xyz->crop_scale_window(
473 Crop(512, 0, 0, 0), dcp::Size(1486, 1080), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_RGB24, VideoRange::FULL, Image::Alignment::COMPACT, false
475 write_image(cropped, "build/test/crop_scale_window_test5.png");
476 check_image("test/data/crop_scale_window_test5.png", "build/test/crop_scale_window_test5.png");
480 BOOST_AUTO_TEST_CASE (crop_scale_window_test6)
482 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
483 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_XYZ12LE, Image::Alignment::PADDED, false);
484 auto cropped = xyz->crop_scale_window(
485 Crop(512, 0, 0, 0), dcp::Size(1486, 1080), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_XYZ12LE, VideoRange::FULL, Image::Alignment::COMPACT, false
487 write_image(cropped, "build/test/crop_scale_window_test6.png");
488 check_image("test/data/crop_scale_window_test6.png", "build/test/crop_scale_window_test6.png");
492 /** Test some small crops with an image that shows up errors in registration of the YUV planes (#1872) */
493 BOOST_AUTO_TEST_CASE (crop_scale_window_test7)
495 using namespace boost::filesystem;
496 for (int left_crop = 0; left_crop < 8; ++left_crop) {
497 auto proxy = make_shared<FFmpegImageProxy>("test/data/rgb_grey_testcard.png");
498 auto yuv = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_YUV420P, Image::Alignment::PADDED, false);
499 int rounded = left_crop - (left_crop % 2);
500 auto cropped = yuv->crop_scale_window(
501 Crop(left_crop, 0, 0, 0),
502 dcp::Size(1998 - rounded, 1080),
503 dcp::Size(1998 - rounded, 1080),
504 dcp::YUVToRGB::REC709,
508 Image::Alignment::PADDED,
511 path file = String::compose("crop_scale_window_test7-%1.png", left_crop);
512 write_image(cropped, path("build") / "test" / file);
513 check_image(path("test") / "data" / file, path("build") / "test" / file, 10);
518 BOOST_AUTO_TEST_CASE (crop_scale_window_test8)
520 using namespace boost::filesystem;
522 auto image = make_shared<Image>(AV_PIX_FMT_YUV420P, dcp::Size(800, 600), Image::Alignment::PADDED);
523 memset(image->data()[0], 41, image->stride()[0] * 600);
524 memset(image->data()[1], 240, image->stride()[1] * 300);
525 memset(image->data()[2], 41, image->stride()[2] * 300);
526 auto scaled = image->crop_scale_window(
527 Crop(), dcp::Size(1435, 1080), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_YUV420P, VideoRange::FULL, Image::Alignment::PADDED, false
529 auto file = "crop_scale_window_test8.png";
530 write_image(scaled->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::COMPACT, false), path("build") / "test" / file);
531 check_image(path("test") / "data" / file, path("build") / "test" / file, 10);
535 BOOST_AUTO_TEST_CASE (as_png_test)
537 auto proxy = make_shared<FFmpegImageProxy>("test/data/3d_test/000001.png");
538 auto image_rgb = proxy->image(Image::Alignment::PADDED).image;
539 auto image_bgr = image_rgb->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_BGRA, Image::Alignment::PADDED, false);
540 image_as_png(image_rgb).write ("build/test/as_png_rgb.png");
541 image_as_png(image_bgr).write ("build/test/as_png_bgr.png");
543 check_image ("test/data/3d_test/000001.png", "build/test/as_png_rgb.png");
544 check_image ("test/data/3d_test/000001.png", "build/test/as_png_bgr.png");
548 BOOST_AUTO_TEST_CASE (as_jpeg_test)
550 auto proxy = make_shared<FFmpegImageProxy>("test/data/3d_test/000001.png");
551 auto image_rgb = proxy->image(Image::Alignment::PADDED).image;
552 auto image_bgr = image_rgb->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_BGRA, Image::Alignment::PADDED, false);
553 image_as_jpeg(image_rgb, 60).write("build/test/as_jpeg_rgb.jpeg");
554 image_as_jpeg(image_bgr, 60).write("build/test/as_jpeg_bgr.jpeg");
556 check_image ("test/data/as_jpeg_rgb.jpeg", "build/test/as_jpeg_rgb.jpeg");
557 check_image ("test/data/as_jpeg_bgr.jpeg", "build/test/as_jpeg_bgr.jpeg");
561 /* Very dumb test to fade black to make sure it stays black */
563 fade_test_format_black (AVPixelFormat f, string name)
565 Image yuv (f, dcp::Size(640, 480), Image::Alignment::PADDED);
568 string const filename = "fade_test_black_" + name + ".png";
569 image_as_png(yuv.convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGBA, Image::Alignment::PADDED, false)).write("build/test/" + filename);
570 check_image ("test/data/" + filename, "build/test/" + filename);
574 /* Fade red to make sure it stays red */
576 fade_test_format_red (AVPixelFormat f, float amount, string name)
578 auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png");
579 auto red = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, f, Image::Alignment::PADDED, false);
581 string const filename = "fade_test_red_" + name + ".png";
582 image_as_png(red->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGBA, Image::Alignment::PADDED, false)).write("build/test/" + filename);
583 check_image ("test/data/" + filename, "build/test/" + filename);
587 BOOST_AUTO_TEST_CASE (fade_test)
589 fade_test_format_black (AV_PIX_FMT_YUV420P, "yuv420p");
590 fade_test_format_black (AV_PIX_FMT_YUV422P10, "yuv422p10");
591 fade_test_format_black (AV_PIX_FMT_RGB24, "rgb24");
592 fade_test_format_black (AV_PIX_FMT_XYZ12LE, "xyz12le");
593 fade_test_format_black (AV_PIX_FMT_RGB48LE, "rgb48le");
595 fade_test_format_red (AV_PIX_FMT_YUV420P, 0, "yuv420p_0");
596 fade_test_format_red (AV_PIX_FMT_YUV420P, 0.5, "yuv420p_50");
597 fade_test_format_red (AV_PIX_FMT_YUV420P, 1, "yuv420p_100");
598 fade_test_format_red (AV_PIX_FMT_YUV422P10, 0, "yuv422p10_0");
599 fade_test_format_red (AV_PIX_FMT_YUV422P10, 0.5, "yuv422p10_50");
600 fade_test_format_red (AV_PIX_FMT_YUV422P10, 1, "yuv422p10_100");
601 fade_test_format_red (AV_PIX_FMT_RGB24, 0, "rgb24_0");
602 fade_test_format_red (AV_PIX_FMT_RGB24, 0.5, "rgb24_50");
603 fade_test_format_red (AV_PIX_FMT_RGB24, 1, "rgb24_100");
604 fade_test_format_red (AV_PIX_FMT_XYZ12LE, 0, "xyz12le_0");
605 fade_test_format_red (AV_PIX_FMT_XYZ12LE, 0.5, "xyz12le_50");
606 fade_test_format_red (AV_PIX_FMT_XYZ12LE, 1, "xyz12le_100");
607 fade_test_format_red (AV_PIX_FMT_RGB48LE, 0, "rgb48le_0");
608 fade_test_format_red (AV_PIX_FMT_RGB48LE, 0.5, "rgb48le_50");
609 fade_test_format_red (AV_PIX_FMT_RGB48LE, 1, "rgb48le_100");
613 BOOST_AUTO_TEST_CASE (make_black_test)
615 dcp::Size in_size (512, 512);
616 dcp::Size out_size (1024, 1024);
618 list<AVPixelFormat> pix_fmts = {
619 AV_PIX_FMT_RGB24, // 2
624 AV_PIX_FMT_YUV420P, // 0
626 AV_PIX_FMT_YUV422P10LE,
627 AV_PIX_FMT_YUV422P16LE,
628 AV_PIX_FMT_YUV444P9LE,
629 AV_PIX_FMT_YUV444P9BE,
630 AV_PIX_FMT_YUV444P10LE,
631 AV_PIX_FMT_YUV444P10BE,
636 AV_PIX_FMT_YUVA420P9BE,
637 AV_PIX_FMT_YUVA422P9BE,
638 AV_PIX_FMT_YUVA444P9BE,
639 AV_PIX_FMT_YUVA420P9LE,
640 AV_PIX_FMT_YUVA422P9LE,
641 AV_PIX_FMT_YUVA444P9LE,
642 AV_PIX_FMT_YUVA420P10BE,
643 AV_PIX_FMT_YUVA422P10BE,
644 AV_PIX_FMT_YUVA444P10BE,
645 AV_PIX_FMT_YUVA420P10LE,
646 AV_PIX_FMT_YUVA422P10LE,
647 AV_PIX_FMT_YUVA444P10LE,
648 AV_PIX_FMT_YUVA420P16BE,
649 AV_PIX_FMT_YUVA422P16BE,
650 AV_PIX_FMT_YUVA444P16BE,
651 AV_PIX_FMT_YUVA420P16LE,
652 AV_PIX_FMT_YUVA422P16LE,
653 AV_PIX_FMT_YUVA444P16LE,
654 AV_PIX_FMT_RGB555LE, // 46
658 for (auto i: pix_fmts) {
659 auto foo = make_shared<Image>(i, in_size, Image::Alignment::PADDED);
661 auto bar = foo->scale (out_size, dcp::YUVToRGB::REC601, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
663 uint8_t* p = bar->data()[0];
664 for (int y = 0; y < bar->size().height; ++y) {
666 for (int x = 0; x < bar->line_size()[0]; ++x) {
668 std::cerr << "x=" << x << ", (x%3)=" << (x%3) << "\n";
670 BOOST_CHECK_EQUAL (*q++, 0);
672 p += bar->stride()[0];
680 BOOST_AUTO_TEST_CASE (make_part_black_test)
682 auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png");
683 auto original = proxy->image(Image::Alignment::PADDED).image;
685 list<AVPixelFormat> pix_fmts = {
692 AV_PIX_FMT_YUV422P10LE,
693 AV_PIX_FMT_YUV444P10LE
696 list<std::pair<int, int>> positions = {
702 for (auto i: pix_fmts) {
703 for (auto j: positions) {
704 auto foo = original->convert_pixel_format(dcp::YUVToRGB::REC601, i, Image::Alignment::PADDED, false);
705 foo->make_part_black (j.first, j.second);
706 auto bar = foo->convert_pixel_format (dcp::YUVToRGB::REC601, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
708 auto p = bar->data()[0];
709 for (int y = 0; y < bar->size().height; ++y) {
711 for (int x = 0; x < bar->size().width; ++x) {
715 if (x >= j.first && x < (j.first + j.second)) {
716 BOOST_CHECK_MESSAGE (
717 r < 3, "red=" << static_cast<int>(r) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
720 BOOST_CHECK_MESSAGE (
721 r >= 252, "red=" << static_cast<int>(r) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
725 BOOST_CHECK_MESSAGE (
726 g == 0, "green=" << static_cast<int>(g) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
728 BOOST_CHECK_MESSAGE (
729 b == 0, "blue=" << static_cast<int>(b) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
732 p += bar->stride()[0];
741 /** Make sure the image isn't corrupted if it is cropped too much. This can happen when a
742 * filler 128x128 black frame is emitted from the FFmpegDecoder and the overall crop in either direction
743 * is greater than 128 pixels.
745 BOOST_AUTO_TEST_CASE (over_crop_test)
747 auto image = make_shared<Image>(AV_PIX_FMT_RGB24, dcp::Size(128, 128), Image::Alignment::PADDED);
748 image->make_black ();
749 auto scaled = image->crop_scale_window (
750 Crop(0, 0, 128, 128), dcp::Size(1323, 565), dcp::Size(1349, 565), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_RGB24, VideoRange::FULL, Image::Alignment::PADDED, true
752 string const filename = "over_crop_test.png";
753 write_image (scaled, "build/test/" + filename);
754 check_image ("test/data/" + filename, "build/test/" + filename);