Add pixel format 66 (AV_PIX_FMT_YUV422P10LE) to make_part_black().

Part of fix for #1984.

1 files changed, 61 insertions, 0 deletions

diff --git a/test/image_test.cc b/test/image_test.cc
index ed644ce3c..4cc80b7e0 100644
--- a/test/image_test.cc
+++ b/test/image_test.cc
@@ -28,6 +28,8 @@
 
 #include "lib/compose.hpp"
 #include "lib/image.h"
+#include "lib/image_content.h"
+#include "lib/image_decoder.h"
 #include "lib/ffmpeg_image_proxy.h"
 #include "test.h"
 #include <boost/test/unit_test.hpp>
@@ -501,6 +503,65 @@ BOOST_AUTO_TEST_CASE (make_black_test)
 }
 
 
+BOOST_AUTO_TEST_CASE (make_part_black_test)
+{
+	auto proxy = make_shared<FFmpegImageProxy>("test/data/flat_red.png", VideoRange::FULL);
+	auto original = proxy->image().image;
+
+	list<AVPixelFormat> pix_fmts = {
+		AV_PIX_FMT_RGB24,
+		AV_PIX_FMT_ARGB,
+		AV_PIX_FMT_RGBA,
+		AV_PIX_FMT_ABGR,
+		AV_PIX_FMT_BGRA,
+		AV_PIX_FMT_YUV422P10LE,
+	};
+
+	list<std::pair<int, int>> positions = {
+		{ 0, 256 },
+		{ 128, 64 },
+	};
+
+	int N = 0;
+	for (auto i: pix_fmts) {
+		for (auto j: positions) {
+			auto foo = original->convert_pixel_format(dcp::YUVToRGB::REC601, i, true, false);
+			foo->make_part_black (j.first, j.second);
+			auto bar = foo->convert_pixel_format (dcp::YUVToRGB::REC601, AV_PIX_FMT_RGB24, true, false);
+
+			auto p = bar->data()[0];
+			for (int y = 0; y < bar->size().height; ++y) {
+				auto q = p;
+				for (int x = 0; x < bar->size().width; ++x) {
+					int r = *q++;
+					int g = *q++;
+					int b = *q++;
+					if (x >= j.first && x < (j.first + j.second)) {
+						BOOST_CHECK_MESSAGE (
+							r < 3, "red=" << static_cast<int>(r) << " at x=" << x << " format " << i << " from " << j.first << " width " << j.second
+							);
+					} else {
+						BOOST_CHECK_MESSAGE (
+							r > 252, "red=" << static_cast<int>(r) << " at x=" << x << " format " << i << " from " << j.first << " width " << j.second
+							);
+
+					}
+					BOOST_CHECK_MESSAGE (
+						g == 0, "green=" << static_cast<int>(g) << " at x=" << x << " format " << i << " from " << j.first << " width " << j.second
+						);
+					BOOST_CHECK_MESSAGE (
+						b == 0, "blue=" << static_cast<int>(b) << " at x=" << x << " format " << i << " from " << j.first << " width " << j.second
+						);
+				}
+				p += bar->stride()[0];
+			}
+
+			++N;
+		}
+	}
+}
+
+
 /** Make sure the image isn't corrupted if it is cropped too much.  This can happen when a
  *  filler 128x128 black frame is emitted from the FFmpegDecoder and the overall crop in either direction
  *  is greater than 128 pixels.