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, "yuv422p10le");
247 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_RGB24, "rgb24");
248 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_BGRA, "bgra");
249 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_RGBA, "rgba");
250 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_RGB48LE, "rgb48le");
251 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV420P, "yuv420p");
252 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV420P10, "yuv420p10");
253 alpha_blend_test_rgba64be_onto(AV_PIX_FMT_YUV422P10LE, "yuv422p10le");
257 /** Test Image::alpha_blend when blending RGBA onto XYZ12LE */
258 BOOST_AUTO_TEST_CASE(alpha_blend_test_rgba_onto_xyz)
260 Image xyz(AV_PIX_FMT_XYZ12LE, dcp::Size(50, 50), Image::Alignment::PADDED);
263 auto overlay = make_shared<Image>(AV_PIX_FMT_RGBA, dcp::Size(8, 8), Image::Alignment::PADDED);
264 for (int y = 0; y < 8; ++y) {
265 uint8_t* p = overlay->data()[0] + (y * overlay->stride()[0]);
266 for (int x = 0; x < 8; ++x) {
274 xyz.alpha_blend(overlay, Position<int>(4, 4));
276 for (int y = 0; y < 50; ++y) {
277 uint16_t* p = reinterpret_cast<uint16_t*>(xyz.data()[0]) + (y * xyz.stride()[0] / 2);
278 for (int x = 0; x < 50; ++x) {
279 if (4 <= x && x < 12 && 4 <= y && y < 12) {
280 BOOST_REQUIRE_EQUAL(p[0], 45078U);
281 BOOST_REQUIRE_EQUAL(p[1], 34939U);
282 BOOST_REQUIRE_EQUAL(p[2], 13892U);
284 BOOST_REQUIRE_EQUAL(p[0], 0U);
285 BOOST_REQUIRE_EQUAL(p[1], 0U);
286 BOOST_REQUIRE_EQUAL(p[2], 0U);
294 /** Test Image::alpha_blend when blending RGBA64BE onto XYZ12LE */
295 BOOST_AUTO_TEST_CASE(alpha_blend_test_rgba64be_onto_xyz)
297 Image xyz(AV_PIX_FMT_XYZ12LE, dcp::Size(50, 50), Image::Alignment::PADDED);
300 auto overlay = make_shared<Image>(AV_PIX_FMT_RGBA64BE, dcp::Size(8, 8), Image::Alignment::PADDED);
301 for (int y = 0; y < 8; ++y) {
302 auto p = reinterpret_cast<uint16_t*>(overlay->data()[0] + (y * overlay->stride()[0]));
303 for (int x = 0; x < 8; ++x) {
311 xyz.alpha_blend(overlay, Position<int>(4, 4));
313 for (int y = 0; y < 50; ++y) {
314 auto p = reinterpret_cast<uint16_t*>(xyz.data()[0]) + (y * xyz.stride()[0] / 2);
315 for (int x = 0; x < 50; ++x) {
316 if (4 <= x && x < 12 && 4 <= y && y < 12) {
317 BOOST_REQUIRE_EQUAL(p[0], 45078U);
318 BOOST_REQUIRE_EQUAL(p[1], 34939U);
319 BOOST_REQUIRE_EQUAL(p[2], 13892U);
321 BOOST_REQUIRE_EQUAL(p[0], 0U);
322 BOOST_REQUIRE_EQUAL(p[1], 0U);
323 BOOST_REQUIRE_EQUAL(p[2], 0U);
331 BOOST_AUTO_TEST_CASE(alpha_blend_text)
333 Image target(AV_PIX_FMT_RGB24, dcp::Size(1998, 1080), Image::Alignment::PADDED);
336 FFmpegImageProxy subtitle_proxy(TestPaths::private_data() / "16-bit-sub.png");
337 auto subtitle = subtitle_proxy.image(Image::Alignment::COMPACT);
339 target.alpha_blend(subtitle.image, Position<int>(0, 0));
340 write_image(make_shared<Image>(target), "build/test/alpha_blend_text.png");
341 check_image("build/test/alpha_blend_text.png", TestPaths::private_data() / "16-bit-sub-blended.png");
345 /** Test merge (list<PositionImage>) with a single image */
346 BOOST_AUTO_TEST_CASE (merge_test1)
348 int const stride = 48 * 4;
350 auto A = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size (48, 48), Image::Alignment::COMPACT);
351 A->make_transparent ();
352 auto a = A->data()[0];
354 for (int y = 0; y < 48; ++y) {
355 auto p = a + y * stride;
356 for (int x = 0; x < 16; ++x) {
364 list<PositionImage> all;
365 all.push_back (PositionImage (A, Position<int>(0, 0)));
366 auto merged = merge (all, Image::Alignment::COMPACT);
368 BOOST_CHECK (merged.position == Position<int>(0, 0));
369 BOOST_CHECK_EQUAL (memcmp (merged.image->data()[0], A->data()[0], stride * 48), 0);
373 /** Test merge (list<PositionImage>) with two images */
374 BOOST_AUTO_TEST_CASE (merge_test2)
376 auto A = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size (48, 1), Image::Alignment::COMPACT);
377 A->make_transparent ();
378 auto a = A->data()[0];
379 for (int x = 0; x < 16; ++x) {
386 auto B = make_shared<Image>(AV_PIX_FMT_BGRA, dcp::Size (48, 1), Image::Alignment::COMPACT);
387 B->make_transparent ();
388 auto b = B->data()[0];
389 for (int x = 0; x < 16; ++x) {
391 b[(x + 32) * 4 + 2] = 255;
393 b[(x + 32) * 4 + 3] = 255;
396 list<PositionImage> all;
397 all.push_back (PositionImage(A, Position<int>(0, 0)));
398 all.push_back (PositionImage(B, Position<int>(0, 0)));
399 auto merged = merge (all, Image::Alignment::COMPACT);
401 BOOST_CHECK (merged.position == Position<int>(0, 0));
403 auto m = merged.image->data()[0];
405 for (int x = 0; x < 16; ++x) {
406 BOOST_CHECK_EQUAL (m[x * 4], 255);
407 BOOST_CHECK_EQUAL (m[x * 4 + 3], 255);
408 BOOST_CHECK_EQUAL (m[(x + 16) * 4 + 3], 0);
409 BOOST_CHECK_EQUAL (m[(x + 32) * 4 + 2], 255);
410 BOOST_CHECK_EQUAL (m[(x + 32) * 4 + 3], 255);
415 /** Test Image::crop_scale_window with YUV420P and some windowing */
416 BOOST_AUTO_TEST_CASE (crop_scale_window_test)
418 auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png");
419 auto raw = proxy->image(Image::Alignment::PADDED).image;
420 auto out = raw->crop_scale_window(
421 Crop(), dcp::Size(1998, 836), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_YUV420P, VideoRange::FULL, Image::Alignment::PADDED, false
423 auto save = out->scale(dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::COMPACT, false);
424 write_image(save, "build/test/crop_scale_window_test.png");
425 check_image("test/data/crop_scale_window_test.png", "build/test/crop_scale_window_test.png");
429 /** Special cases of Image::crop_scale_window which triggered some valgrind warnings */
430 BOOST_AUTO_TEST_CASE (crop_scale_window_test2)
432 auto image = make_shared<Image>(AV_PIX_FMT_XYZ12LE, dcp::Size(2048, 858), Image::Alignment::PADDED);
433 image->crop_scale_window (
434 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
436 image->crop_scale_window (
437 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
442 BOOST_AUTO_TEST_CASE (crop_scale_window_test3)
444 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
445 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
446 auto cropped = xyz->crop_scale_window(
447 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
449 write_image(cropped, "build/test/crop_scale_window_test3.png");
450 check_image("test/data/crop_scale_window_test3.png", "build/test/crop_scale_window_test3.png");
454 BOOST_AUTO_TEST_CASE (crop_scale_window_test4)
456 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
457 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
458 auto cropped = xyz->crop_scale_window(
459 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
461 write_image(cropped, "build/test/crop_scale_window_test4.png");
462 check_image("test/data/crop_scale_window_test4.png", "build/test/crop_scale_window_test4.png");
466 BOOST_AUTO_TEST_CASE (crop_scale_window_test5)
468 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
469 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_XYZ12LE, Image::Alignment::PADDED, false);
470 auto cropped = xyz->crop_scale_window(
471 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
473 write_image(cropped, "build/test/crop_scale_window_test5.png");
474 check_image("test/data/crop_scale_window_test5.png", "build/test/crop_scale_window_test5.png");
478 BOOST_AUTO_TEST_CASE (crop_scale_window_test6)
480 auto proxy = make_shared<FFmpegImageProxy>(TestPaths::private_data() / "player_seek_test_0.png");
481 auto xyz = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_XYZ12LE, Image::Alignment::PADDED, false);
482 auto cropped = xyz->crop_scale_window(
483 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
485 write_image(cropped, "build/test/crop_scale_window_test6.png");
486 check_image("test/data/crop_scale_window_test6.png", "build/test/crop_scale_window_test6.png");
490 /** Test some small crops with an image that shows up errors in registration of the YUV planes (#1872) */
491 BOOST_AUTO_TEST_CASE (crop_scale_window_test7)
493 using namespace boost::filesystem;
494 for (int left_crop = 0; left_crop < 8; ++left_crop) {
495 auto proxy = make_shared<FFmpegImageProxy>("test/data/rgb_grey_testcard.png");
496 auto yuv = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_YUV420P, Image::Alignment::PADDED, false);
497 int rounded = left_crop - (left_crop % 2);
498 auto cropped = yuv->crop_scale_window(
499 Crop(left_crop, 0, 0, 0),
500 dcp::Size(1998 - rounded, 1080),
501 dcp::Size(1998 - rounded, 1080),
502 dcp::YUVToRGB::REC709,
506 Image::Alignment::PADDED,
509 path file = String::compose("crop_scale_window_test7-%1.png", left_crop);
510 write_image(cropped, path("build") / "test" / file);
511 check_image(path("test") / "data" / file, path("build") / "test" / file, 10);
516 BOOST_AUTO_TEST_CASE (crop_scale_window_test8)
518 using namespace boost::filesystem;
520 auto image = make_shared<Image>(AV_PIX_FMT_YUV420P, dcp::Size(800, 600), Image::Alignment::PADDED);
521 memset(image->data()[0], 41, image->stride()[0] * 600);
522 memset(image->data()[1], 240, image->stride()[1] * 300);
523 memset(image->data()[2], 41, image->stride()[2] * 300);
524 auto scaled = image->crop_scale_window(
525 Crop(), dcp::Size(1435, 1080), dcp::Size(1998, 1080), dcp::YUVToRGB::REC709, VideoRange::FULL, AV_PIX_FMT_YUV420P, VideoRange::FULL, Image::Alignment::PADDED, false
527 auto file = "crop_scale_window_test8.png";
528 write_image(scaled->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::COMPACT, false), path("build") / "test" / file);
529 check_image(path("test") / "data" / file, path("build") / "test" / file, 10);
533 BOOST_AUTO_TEST_CASE (as_png_test)
535 auto proxy = make_shared<FFmpegImageProxy>("test/data/3d_test/000001.png");
536 auto image_rgb = proxy->image(Image::Alignment::PADDED).image;
537 auto image_bgr = image_rgb->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_BGRA, Image::Alignment::PADDED, false);
538 image_as_png(image_rgb).write ("build/test/as_png_rgb.png");
539 image_as_png(image_bgr).write ("build/test/as_png_bgr.png");
541 check_image ("test/data/3d_test/000001.png", "build/test/as_png_rgb.png");
542 check_image ("test/data/3d_test/000001.png", "build/test/as_png_bgr.png");
546 BOOST_AUTO_TEST_CASE (as_jpeg_test)
548 auto proxy = make_shared<FFmpegImageProxy>("test/data/3d_test/000001.png");
549 auto image_rgb = proxy->image(Image::Alignment::PADDED).image;
550 auto image_bgr = image_rgb->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_BGRA, Image::Alignment::PADDED, false);
551 image_as_jpeg(image_rgb, 60).write("build/test/as_jpeg_rgb.jpeg");
552 image_as_jpeg(image_bgr, 60).write("build/test/as_jpeg_bgr.jpeg");
554 check_image ("test/data/as_jpeg_rgb.jpeg", "build/test/as_jpeg_rgb.jpeg");
555 check_image ("test/data/as_jpeg_bgr.jpeg", "build/test/as_jpeg_bgr.jpeg");
559 /* Very dumb test to fade black to make sure it stays black */
561 fade_test_format_black (AVPixelFormat f, string name)
563 Image yuv (f, dcp::Size(640, 480), Image::Alignment::PADDED);
566 string const filename = "fade_test_black_" + name + ".png";
567 image_as_png(yuv.convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGBA, Image::Alignment::PADDED, false)).write("build/test/" + filename);
568 check_image ("test/data/" + filename, "build/test/" + filename);
572 /* Fade red to make sure it stays red */
574 fade_test_format_red (AVPixelFormat f, float amount, string name)
576 auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png");
577 auto red = proxy->image(Image::Alignment::PADDED).image->convert_pixel_format(dcp::YUVToRGB::REC709, f, Image::Alignment::PADDED, false);
579 string const filename = "fade_test_red_" + name + ".png";
580 image_as_png(red->convert_pixel_format(dcp::YUVToRGB::REC709, AV_PIX_FMT_RGBA, Image::Alignment::PADDED, false)).write("build/test/" + filename);
581 check_image ("test/data/" + filename, "build/test/" + filename);
585 BOOST_AUTO_TEST_CASE (fade_test)
587 fade_test_format_black (AV_PIX_FMT_YUV420P, "yuv420p");
588 fade_test_format_black (AV_PIX_FMT_YUV422P10, "yuv422p10");
589 fade_test_format_black (AV_PIX_FMT_RGB24, "rgb24");
590 fade_test_format_black (AV_PIX_FMT_XYZ12LE, "xyz12le");
591 fade_test_format_black (AV_PIX_FMT_RGB48LE, "rgb48le");
593 fade_test_format_red (AV_PIX_FMT_YUV420P, 0, "yuv420p_0");
594 fade_test_format_red (AV_PIX_FMT_YUV420P, 0.5, "yuv420p_50");
595 fade_test_format_red (AV_PIX_FMT_YUV420P, 1, "yuv420p_100");
596 fade_test_format_red (AV_PIX_FMT_YUV422P10, 0, "yuv422p10_0");
597 fade_test_format_red (AV_PIX_FMT_YUV422P10, 0.5, "yuv422p10_50");
598 fade_test_format_red (AV_PIX_FMT_YUV422P10, 1, "yuv422p10_100");
599 fade_test_format_red (AV_PIX_FMT_RGB24, 0, "rgb24_0");
600 fade_test_format_red (AV_PIX_FMT_RGB24, 0.5, "rgb24_50");
601 fade_test_format_red (AV_PIX_FMT_RGB24, 1, "rgb24_100");
602 fade_test_format_red (AV_PIX_FMT_XYZ12LE, 0, "xyz12le_0");
603 fade_test_format_red (AV_PIX_FMT_XYZ12LE, 0.5, "xyz12le_50");
604 fade_test_format_red (AV_PIX_FMT_XYZ12LE, 1, "xyz12le_100");
605 fade_test_format_red (AV_PIX_FMT_RGB48LE, 0, "rgb48le_0");
606 fade_test_format_red (AV_PIX_FMT_RGB48LE, 0.5, "rgb48le_50");
607 fade_test_format_red (AV_PIX_FMT_RGB48LE, 1, "rgb48le_100");
611 BOOST_AUTO_TEST_CASE (make_black_test)
613 dcp::Size in_size (512, 512);
614 dcp::Size out_size (1024, 1024);
616 list<AVPixelFormat> pix_fmts = {
617 AV_PIX_FMT_RGB24, // 2
622 AV_PIX_FMT_YUV420P, // 0
624 AV_PIX_FMT_YUV422P10LE,
625 AV_PIX_FMT_YUV422P16LE,
626 AV_PIX_FMT_YUV444P9LE,
627 AV_PIX_FMT_YUV444P9BE,
628 AV_PIX_FMT_YUV444P10LE,
629 AV_PIX_FMT_YUV444P10BE,
634 AV_PIX_FMT_YUVA420P9BE,
635 AV_PIX_FMT_YUVA422P9BE,
636 AV_PIX_FMT_YUVA444P9BE,
637 AV_PIX_FMT_YUVA420P9LE,
638 AV_PIX_FMT_YUVA422P9LE,
639 AV_PIX_FMT_YUVA444P9LE,
640 AV_PIX_FMT_YUVA420P10BE,
641 AV_PIX_FMT_YUVA422P10BE,
642 AV_PIX_FMT_YUVA444P10BE,
643 AV_PIX_FMT_YUVA420P10LE,
644 AV_PIX_FMT_YUVA422P10LE,
645 AV_PIX_FMT_YUVA444P10LE,
646 AV_PIX_FMT_YUVA420P16BE,
647 AV_PIX_FMT_YUVA422P16BE,
648 AV_PIX_FMT_YUVA444P16BE,
649 AV_PIX_FMT_YUVA420P16LE,
650 AV_PIX_FMT_YUVA422P16LE,
651 AV_PIX_FMT_YUVA444P16LE,
652 AV_PIX_FMT_RGB555LE, // 46
656 for (auto i: pix_fmts) {
657 auto foo = make_shared<Image>(i, in_size, Image::Alignment::PADDED);
659 auto bar = foo->scale (out_size, dcp::YUVToRGB::REC601, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
661 uint8_t* p = bar->data()[0];
662 for (int y = 0; y < bar->size().height; ++y) {
664 for (int x = 0; x < bar->line_size()[0]; ++x) {
666 std::cerr << "x=" << x << ", (x%3)=" << (x%3) << "\n";
668 BOOST_CHECK_EQUAL (*q++, 0);
670 p += bar->stride()[0];
678 BOOST_AUTO_TEST_CASE (make_part_black_test)
680 auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png");
681 auto original = proxy->image(Image::Alignment::PADDED).image;
683 list<AVPixelFormat> pix_fmts = {
690 AV_PIX_FMT_YUV422P10LE,
691 AV_PIX_FMT_YUV444P10LE
694 list<std::pair<int, int>> positions = {
700 for (auto i: pix_fmts) {
701 for (auto j: positions) {
702 auto foo = original->convert_pixel_format(dcp::YUVToRGB::REC601, i, Image::Alignment::PADDED, false);
703 foo->make_part_black (j.first, j.second);
704 auto bar = foo->convert_pixel_format (dcp::YUVToRGB::REC601, AV_PIX_FMT_RGB24, Image::Alignment::PADDED, false);
706 auto p = bar->data()[0];
707 for (int y = 0; y < bar->size().height; ++y) {
709 for (int x = 0; x < bar->size().width; ++x) {
713 if (x >= j.first && x < (j.first + j.second)) {
714 BOOST_CHECK_MESSAGE (
715 r < 3, "red=" << static_cast<int>(r) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
718 BOOST_CHECK_MESSAGE (
719 r >= 252, "red=" << static_cast<int>(r) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
723 BOOST_CHECK_MESSAGE (
724 g == 0, "green=" << static_cast<int>(g) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
726 BOOST_CHECK_MESSAGE (
727 b == 0, "blue=" << static_cast<int>(b) << " at (" << x << "," << y << ") format " << i << " from " << j.first << " width " << j.second
730 p += bar->stride()[0];
739 /** Make sure the image isn't corrupted if it is cropped too much. This can happen when a
740 * filler 128x128 black frame is emitted from the FFmpegDecoder and the overall crop in either direction
741 * is greater than 128 pixels.
743 BOOST_AUTO_TEST_CASE (over_crop_test)
745 auto image = make_shared<Image>(AV_PIX_FMT_RGB24, dcp::Size(128, 128), Image::Alignment::PADDED);
746 image->make_black ();
747 auto scaled = image->crop_scale_window (
748 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
750 string const filename = "over_crop_test.png";
751 write_image (scaled, "build/test/" + filename);
752 check_image ("test/data/" + filename, "build/test/" + filename);