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 + 0] = 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) {
222 p[x * 4 + 2] = 65535;
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_YUV422P9LE, "yuv422p9le");
246 alpha_blend_test_bgra_onto(AV_PIX_FMT_YUV422P10LE, "yuv422p10le");
247 alpha_blend_test_bgra_onto(AV_PIX_FMT_YUV444P9LE, "yuv444p9le");
248 alpha_blend_test_bgra_onto(AV_PIX_FMT_YUV444P10LE, "yuv444p10le");
250 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_RGB24, "rgb24");
251 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_BGRA, "bgra");
252 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_RGBA, "rgba");
253 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_RGB48LE, "rgb48le");
254 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV420P, "yuv420p");
255 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV420P10, "yuv420p10");
256 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV422P9LE, "yuv422p9le");
257 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV422P10LE, "yuv422p10le");
258 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV444P9LE, "yuv444p9le");
259 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV444P10LE, "yuv444p10le");
263 /** Test Image::alpha_blend when blending RGBA onto XYZ12LE */
264 BOOST_AUTO_TEST_CASE(alpha_blend_test_rgba_onto_xyz)
266 Image xyz(AV_PIX_FMT_XYZ12LE, dcp::Size(50, 50), Image::Alignment::PADDED);
269 auto overlay = make_shared<Image>(AV_PIX_FMT_RGBA, dcp::Size(8, 8), Image::Alignment::PADDED);
270 for (int y = 0; y < 8; ++y) {
271 uint8_t* p = overlay->data()[0] + (y * overlay->stride()[0]);
272 for (int x = 0; x < 8; ++x) {
280 xyz.alpha_blend(overlay, Position<int>(4, 4));
282 for (int y = 0; y < 50; ++y) {
283 uint16_t* p = reinterpret_cast<uint16_t*>(xyz.data()[0]) + (y * xyz.stride()[0] / 2);
284 for (int x = 0; x < 50; ++x) {
285 if (4 <= x && x < 12 && 4 <= y && y < 12) {
286 BOOST_REQUIRE_EQUAL(p[0], 45078U);
287 BOOST_REQUIRE_EQUAL(p[1], 34939U);
288 BOOST_REQUIRE_EQUAL(p[2], 13892U);
290 BOOST_REQUIRE_EQUAL(p[0], 0U);
291 BOOST_REQUIRE_EQUAL(p[1], 0U);
292 BOOST_REQUIRE_EQUAL(p[2], 0U);
300 /** Test Image::alpha_blend when blending RGBA64BE onto XYZ12LE */
301 BOOST_AUTO_TEST_CASE(alpha_blend_test_rgba64be_onto_xyz)
303 Image xyz(AV_PIX_FMT_XYZ12LE, dcp::Size(50, 50), Image::Alignment::PADDED);
306 auto overlay = make_shared<Image>(AV_PIX_FMT_RGBA64BE, dcp::Size(8, 8), Image::Alignment::PADDED);
307 for (int y = 0; y < 8; ++y) {
308 auto p = reinterpret_cast<uint16_t*>(overlay->data()[0] + (y * overlay->stride()[0]));
309 for (int x = 0; x < 8; ++x) {
317 xyz.alpha_blend(overlay, Position<int>(4, 4));
319 for (int y = 0; y < 50; ++y) {
320 auto p = reinterpret_cast<uint16_t*>(xyz.data()[0]) + (y * xyz.stride()[0] / 2);
321 for (int x = 0; x < 50; ++x) {
322 if (4 <= x && x < 12 && 4 <= y && y < 12) {
323 BOOST_REQUIRE_EQUAL(p[0], 45078U);
324 BOOST_REQUIRE_EQUAL(p[1], 34939U);
325 BOOST_REQUIRE_EQUAL(p[2], 13892U);
327 BOOST_REQUIRE_EQUAL(p[0], 0U);
328 BOOST_REQUIRE_EQUAL(p[1], 0U);
329 BOOST_REQUIRE_EQUAL(p[2], 0U);
337 BOOST_AUTO_TEST_CASE(alpha_blend_text)
339 Image target(AV_PIX_FMT_RGB24, dcp::Size(1998, 1080), Image::Alignment::PADDED);
342 FFmpegImageProxy subtitle_proxy(TestPaths::private_data() / "16-bit-sub.png");
343 auto subtitle = subtitle_proxy.image(Image::Alignment::COMPACT);
345 target.alpha_blend(subtitle.image, Position<int>(0, 0));
346 write_image(make_shared<Image>(target), "build/test/alpha_blend_text.png");
347 check_image("build/test/alpha_blend_text.png", TestPaths::private_data() / "16-bit-sub-blended.png");
351 /** Test merge (list<PositionImage>) with a single image */
352 BOOST_AUTO_TEST_CASE (merge_test1)
354 int const stride = 48 * 4;
356 auto A = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size (48, 48), Image::Alignment::COMPACT);
357 A->make_transparent ();
358 auto a = A->data()[0];
360 for (int y = 0; y < 48; ++y) {
361 auto p = a + y * stride;
362 for (int x = 0; x < 16; ++x) {
370 list<PositionImage> all;
371 all.push_back (PositionImage (A, Position<int>(0, 0)));
372 auto merged = merge (all, Image::Alignment::COMPACT);
374 BOOST_CHECK (merged.position == Position<int>(0, 0));
375 BOOST_CHECK_EQUAL (memcmp (merged.image->data()[0], A->data()[0], stride * 48), 0);
379 /** Test merge (list<PositionImage>) with two images */
380 BOOST_AUTO_TEST_CASE (merge_test2)
382 auto A = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size (48, 1), Image::Alignment::COMPACT);
383 A->make_transparent ();
384 auto a = A->data()[0];
385 for (int x = 0; x < 16; ++x) {
392 auto B = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size (48, 1), Image::Alignment::COMPACT);
393 B->make_transparent ();
394 auto b = B->data()[0];
395 for (int x = 0; x < 16; ++x) {
397 b[(x + 32) * 4 + 2] = 255;
399 b[(x + 32) * 4 + 3] = 255;
402 list<PositionImage> all;
403 all.push_back (PositionImage(A, Position<int>(0, 0)));
404 all.push_back (PositionImage(B, Position<int>(0, 0)));
405 auto merged = merge (all, Image::Alignment::COMPACT);
407 BOOST_CHECK (merged.position == Position<int>(0, 0));
409 auto m = merged.image->data()[0];
411 for (int x = 0; x < 16; ++x) {
412 BOOST_CHECK_EQUAL (m[x * 4], 255);
413 BOOST_CHECK_EQUAL (m[x * 4 + 3], 255);
414 BOOST_CHECK_EQUAL (m[(x + 16) * 4 + 3], 0);
415 BOOST_CHECK_EQUAL (m[(x + 32) * 4 + 2], 255);
416 BOOST_CHECK_EQUAL (m[(x + 32) * 4 + 3], 255);
421 /** Test Image::crop_scale_window with YUV420P and some windowing */
422 BOOST_AUTO_TEST_CASE (crop_scale_window_test)
424 auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png");
425 auto raw = proxy->image(Image::Alignment::PADDED).image;
426 auto out = raw->crop_scale_window(
427 Crop(), dcp::Size(1998, 836), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_YUV420P, VideoRange::FULL, Image::Alignment::PADDED, false
429 auto save = out->scale(dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::COMPACT, false);
430 write_image(save, "build/test/crop_scale_window_test.png");
431 check_image("test/data/crop_scale_window_test.png", "build/test/crop_scale_window_test.png");
435 /** Special cases of Image::crop_scale_window which triggered some valgrind warnings */
436 BOOST_AUTO_TEST_CASE (crop_scale_window_test2)
438 auto image = make_shared<Image>(AV_PIX_FMT_XYZ12LE, dcp::Size(2048, 858), Image::Alignment::PADDED);
439 image->crop_scale_window (
440 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
442 image->crop_scale_window (
443 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
448 BOOST_AUTO_TEST_CASE (crop_scale_window_test3)
450 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
451 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
452 auto cropped = xyz->crop_scale_window(
453 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
455 write_image(cropped, "build/test/crop_scale_window_test3.png");
456 check_image("test/data/crop_scale_window_test3.png", "build/test/crop_scale_window_test3.png");
460 BOOST_AUTO_TEST_CASE (crop_scale_window_test4)
462 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
463 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
464 auto cropped = xyz->crop_scale_window(
465 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
467 write_image(cropped, "build/test/crop_scale_window_test4.png");
468 check_image("test/data/crop_scale_window_test4.png", "build/test/crop_scale_window_test4.png");
472 BOOST_AUTO_TEST_CASE (crop_scale_window_test5)
474 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
475 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_XYZ12LE, Image::Alignment::PADDED, false);
476 auto cropped = xyz->crop_scale_window(
477 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
479 write_image(cropped, "build/test/crop_scale_window_test5.png");
480 check_image("test/data/crop_scale_window_test5.png", "build/test/crop_scale_window_test5.png");
484 BOOST_AUTO_TEST_CASE (crop_scale_window_test6)
486 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
487 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_XYZ12LE, Image::Alignment::PADDED, false);
488 auto cropped = xyz->crop_scale_window(
489 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
491 write_image(cropped, "build/test/crop_scale_window_test6.png");
492 check_image("test/data/crop_scale_window_test6.png", "build/test/crop_scale_window_test6.png");
496 /** Test some small crops with an image that shows up errors in registration of the YUV planes (#1872) */
497 BOOST_AUTO_TEST_CASE (crop_scale_window_test7)
499 using namespace boost::filesystem;
500 for (int left_crop = 0; left_crop < 8; ++left_crop) {
501 auto proxy = make_shared<FFmpegImageProxy>("test/data/rgb_grey_testcard.png");
502 auto yuv = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_YUV420P, Image::Alignment::PADDED, false);
503 int rounded = left_crop - (left_crop % 2);
504 auto cropped = yuv->crop_scale_window(
505 Crop(left_crop, 0, 0, 0),
506 dcp::Size(1998 - rounded, 1080),
507 dcp::Size(1998 - rounded, 1080),
508 dcp::YUVToRGB::REC709,
512 Image::Alignment::PADDED,
515 path file = String::compose("crop_scale_window_test7-%1.png", left_crop);
516 write_image(cropped, path("build") / "test" / file);
517 check_image(path("test") / "data" / file, path("build") / "test" / file, 10);
522 BOOST_AUTO_TEST_CASE (crop_scale_window_test8)
524 using namespace boost::filesystem;
526 auto image = make_shared<Image>(AV_PIX_FMT_YUV420P, dcp::Size(800, 600), Image::Alignment::PADDED);
527 memset(image->data()[0], 41, image->stride()[0] * 600);
528 memset(image->data()[1], 240, image->stride()[1] * 300);
529 memset(image->data()[2], 41, image->stride()[2] * 300);
530 auto scaled = image->crop_scale_window(
531 Crop(), dcp::Size(1435, 1080), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_YUV420P, VideoRange::FULL, Image::Alignment::PADDED, false
533 auto file = "crop_scale_window_test8.png";
534 write_image(scaled->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::COMPACT, false), path("build") / "test" / file);
535 check_image(path("test") / "data" / file, path("build") / "test" / file, 10);
539 BOOST_AUTO_TEST_CASE (as_png_test)
541 auto proxy = make_shared<FFmpegImageProxy>("test/data/3d_test/000001.png");
542 auto image_rgb = proxy->image(Image::Alignment::PADDED).image;
543 auto image_bgr = image_rgb->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_BGRA, Image::Alignment::PADDED, false);
544 image_as_png(image_rgb).write ("build/test/as_png_rgb.png");
545 image_as_png(image_bgr).write ("build/test/as_png_bgr.png");
547 check_image ("test/data/3d_test/000001.png", "build/test/as_png_rgb.png");
548 check_image ("test/data/3d_test/000001.png", "build/test/as_png_bgr.png");
552 BOOST_AUTO_TEST_CASE (as_jpeg_test)
554 auto proxy = make_shared<FFmpegImageProxy>("test/data/3d_test/000001.png");
555 auto image_rgb = proxy->image(Image::Alignment::PADDED).image;
556 auto image_bgr = image_rgb->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_BGRA, Image::Alignment::PADDED, false);
557 image_as_jpeg(image_rgb, 60).write("build/test/as_jpeg_rgb.jpeg");
558 image_as_jpeg(image_bgr, 60).write("build/test/as_jpeg_bgr.jpeg");
560 check_image ("test/data/as_jpeg_rgb.jpeg", "build/test/as_jpeg_rgb.jpeg");
561 check_image ("test/data/as_jpeg_bgr.jpeg", "build/test/as_jpeg_bgr.jpeg");
565 /* Very dumb test to fade black to make sure it stays black */
567 fade_test_format_black (AVPixelFormat f, string name)
569 Image yuv (f, dcp::Size(640, 480), Image::Alignment::PADDED);
572 string const filename = "fade_test_black_" + name + ".png";
573 image_as_png(yuv.convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGBA, Image::Alignment::PADDED, false)).write("build/test/" + filename);
574 check_image ("test/data/" + filename, "build/test/" + filename);
578 /* Fade red to make sure it stays red */
580 fade_test_format_red (AVPixelFormat f, float amount, string name)
582 auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png");
583 auto red = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, f, Image::Alignment::PADDED, false);
585 string const filename = "fade_test_red_" + name + ".png";
586 image_as_png(red->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGBA, Image::Alignment::PADDED, false)).write("build/test/" + filename);
587 check_image ("test/data/" + filename, "build/test/" + filename);
591 BOOST_AUTO_TEST_CASE (fade_test)
593 fade_test_format_black (AV_PIX_FMT_YUV420P, "yuv420p");
594 fade_test_format_black (AV_PIX_FMT_YUV422P10, "yuv422p10");
595 fade_test_format_black (AV_PIX_FMT_RGB24, "rgb24");
596 fade_test_format_black (AV_PIX_FMT_XYZ12LE, "xyz12le");
597 fade_test_format_black (AV_PIX_FMT_RGB48LE, "rgb48le");
599 fade_test_format_red (AV_PIX_FMT_YUV420P, 0, "yuv420p_0");
600 fade_test_format_red (AV_PIX_FMT_YUV420P, 0.5, "yuv420p_50");
601 fade_test_format_red (AV_PIX_FMT_YUV420P, 1, "yuv420p_100");
602 fade_test_format_red (AV_PIX_FMT_YUV422P10, 0, "yuv422p10_0");
603 fade_test_format_red (AV_PIX_FMT_YUV422P10, 0.5, "yuv422p10_50");
604 fade_test_format_red (AV_PIX_FMT_YUV422P10, 1, "yuv422p10_100");
605 fade_test_format_red (AV_PIX_FMT_RGB24, 0, "rgb24_0");
606 fade_test_format_red (AV_PIX_FMT_RGB24, 0.5, "rgb24_50");
607 fade_test_format_red (AV_PIX_FMT_RGB24, 1, "rgb24_100");
608 fade_test_format_red (AV_PIX_FMT_XYZ12LE, 0, "xyz12le_0");
609 fade_test_format_red (AV_PIX_FMT_XYZ12LE, 0.5, "xyz12le_50");
610 fade_test_format_red (AV_PIX_FMT_XYZ12LE, 1, "xyz12le_100");
611 fade_test_format_red (AV_PIX_FMT_RGB48LE, 0, "rgb48le_0");
612 fade_test_format_red (AV_PIX_FMT_RGB48LE, 0.5, "rgb48le_50");
613 fade_test_format_red (AV_PIX_FMT_RGB48LE, 1, "rgb48le_100");
617 BOOST_AUTO_TEST_CASE (make_black_test)
619 dcp::Size in_size (512, 512);
620 dcp::Size out_size (1024, 1024);
622 list<AVPixelFormat> pix_fmts = {
623 AV_PIX_FMT_RGB24, // 2
628 AV_PIX_FMT_YUV420P, // 0
630 AV_PIX_FMT_YUV422P10LE,
631 AV_PIX_FMT_YUV422P16LE,
632 AV_PIX_FMT_YUV444P9LE,
633 AV_PIX_FMT_YUV444P9BE,
634 AV_PIX_FMT_YUV444P10LE,
635 AV_PIX_FMT_YUV444P10BE,
640 AV_PIX_FMT_YUVA420P9BE,
641 AV_PIX_FMT_YUVA422P9BE,
642 AV_PIX_FMT_YUVA444P9BE,
643 AV_PIX_FMT_YUVA420P9LE,
644 AV_PIX_FMT_YUVA422P9LE,
645 AV_PIX_FMT_YUVA444P9LE,
646 AV_PIX_FMT_YUVA420P10BE,
647 AV_PIX_FMT_YUVA422P10BE,
648 AV_PIX_FMT_YUVA444P10BE,
649 AV_PIX_FMT_YUVA420P10LE,
650 AV_PIX_FMT_YUVA422P10LE,
651 AV_PIX_FMT_YUVA444P10LE,
652 AV_PIX_FMT_YUVA420P16BE,
653 AV_PIX_FMT_YUVA422P16BE,
654 AV_PIX_FMT_YUVA444P16BE,
655 AV_PIX_FMT_YUVA420P16LE,
656 AV_PIX_FMT_YUVA422P16LE,
657 AV_PIX_FMT_YUVA444P16LE,
658 AV_PIX_FMT_RGB555LE, // 46
662 for (auto i: pix_fmts) {
663 auto foo = make_shared<Image>(i, in_size, Image::Alignment::PADDED);
665 auto bar = foo->scale (out_size, dcp::YUVToRGB::REC601, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
667 uint8_t* p = bar->data()[0];
668 for (int y = 0; y < bar->size().height; ++y) {
670 for (int x = 0; x < bar->line_size()[0]; ++x) {
672 std::cerr << "x=" << x << ", (x%3)=" << (x%3) << "\n";
674 BOOST_CHECK_EQUAL (*q++, 0);
676 p += bar->stride()[0];
684 BOOST_AUTO_TEST_CASE (make_part_black_test)
686 auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png");
687 auto original = proxy->image(Image::Alignment::PADDED).image;
689 list<AVPixelFormat> pix_fmts = {
696 AV_PIX_FMT_YUV422P10LE,
697 AV_PIX_FMT_YUV444P10LE
700 list<std::pair<int, int>> positions = {
706 for (auto i: pix_fmts) {
707 for (auto j: positions) {
708 auto foo = original->convert_pixel_format(dcp::YUVToRGB::REC601, i, Image::Alignment::PADDED, false);
709 foo->make_part_black (j.first, j.second);
710 auto bar = foo->convert_pixel_format (dcp::YUVToRGB::REC601, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
712 auto p = bar->data()[0];
713 for (int y = 0; y < bar->size().height; ++y) {
715 for (int x = 0; x < bar->size().width; ++x) {
719 if (x >= j.first && x < (j.first + j.second)) {
720 BOOST_CHECK_MESSAGE (
721 r < 3, "red=" << static_cast<int>(r) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
724 BOOST_CHECK_MESSAGE (
725 r >= 252, "red=" << static_cast<int>(r) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
729 BOOST_CHECK_MESSAGE (
730 g == 0, "green=" << static_cast<int>(g) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
732 BOOST_CHECK_MESSAGE (
733 b == 0, "blue=" << static_cast<int>(b) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
736 p += bar->stride()[0];
745 /** Make sure the image isn't corrupted if it is cropped too much. This can happen when a
746 * filler 128x128 black frame is emitted from the FFmpegDecoder and the overall crop in either direction
747 * is greater than 128 pixels.
749 BOOST_AUTO_TEST_CASE (over_crop_test)
751 auto image = make_shared<Image>(AV_PIX_FMT_RGB24, dcp::Size(128, 128), Image::Alignment::PADDED);
752 image->make_black ();
753 auto scaled = image->crop_scale_window (
754 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
756 string const filename = "over_crop_test.png";
757 write_image (scaled, "build/test/" + filename);
758 check_image ("test/data/" + filename, "build/test/" + filename);