Add pixel format 0 (AV_PIX_FMT_YUV420P) to make_part_black().

[dcpomatic.git] / test / make_black_test.cc

diff --git a/test/make_black_test.cc b/test/make_black_test.cc
index 88bf4bb7a5e1100bd529102d650bb6fb069011e5..ee99031099b525119e188915dcb72289d9d23b28 100644 (file)
--- a/test/make_black_test.cc
+++ b/test/make_black_test.cc
@@ -19,22 +19,22 @@
 */
 
 /** @file  test/make_black_test.cc
- *  @brief Check that Image::make_black works, and doesn't use values which crash
+ *  @brief Check that Image::make_black works and doesn't use values which crash
  *  sws_scale().
- *
+ *  @ingroup selfcontained
  *  @see test/image_test.cc
  */
 
+#include <boost/test/unit_test.hpp>
 #include <dcp/util.h>
 extern "C" {
 #include <libavutil/pixfmt.h>
 }
+#include "lib/ffmpeg_image_proxy.h"
 #include "lib/image.h"
-#include <boost/make_shared.hpp>
-#include <boost/test/unit_test.hpp>
 
 using std::list;
-using boost::make_shared;
+using boost::shared_ptr;
 
 BOOST_AUTO_TEST_CASE (make_black_test)
 {
@@ -81,7 +81,7 @@ BOOST_AUTO_TEST_CASE (make_black_test)
 
        int N = 0;
        for (list<AVPixelFormat>::const_iterator i = pix_fmts.begin(); i != pix_fmts.end(); ++i) {
-               boost::shared_ptr<Image> foo = make_shared<Image> (*i, in_size, true);
+               boost::shared_ptr<Image> foo (new Image (*i, in_size, true));
                foo->make_black ();
                boost::shared_ptr<Image> bar = foo->scale (out_size, dcp::YUV_TO_RGB_REC601, AV_PIX_FMT_RGB24, true, false);
 
@@ -100,3 +100,64 @@ BOOST_AUTO_TEST_CASE (make_black_test)
                ++N;
        }
 }
+
+
+BOOST_AUTO_TEST_CASE (make_part_black_test)
+{
+       shared_ptr<FFmpegImageProxy> proxy (new FFmpegImageProxy("test/data/flat_red.png"));
+       shared_ptr<Image> original = proxy->image().first;
+
+       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_YUV420P,
+               AV_PIX_FMT_YUV422P10LE,
+       };
+
+       list<std::pair<int, int> > positions = {
+               { 0, 256 },
+               { 128, 64 },
+       };
+
+       int N = 0;
+       for (list<AVPixelFormat>::const_iterator i = pix_fmts.begin(); i != pix_fmts.end(); ++i) {
+               for (list<std::pair<int, int> >::const_iterator j = positions.begin(); j != positions.end(); ++j) {
+                       shared_ptr<Image> foo = original->convert_pixel_format(dcp::YUV_TO_RGB_REC601, *i, true, false);
+                       foo->make_part_black (j->first, j->second);
+                       shared_ptr<Image> bar = foo->convert_pixel_format (dcp::YUV_TO_RGB_REC601, AV_PIX_FMT_RGB24, true, false);
+
+                       uint8_t* p = bar->data()[0];
+                       for (int y = 0; y < bar->size().height; ++y) {
+                               uint8_t* 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;
+               }
+       }
+}
+