Add pixel format 66 (AV_PIX_FMT_YUV422P10LE) to make_part_black().
authorCarl Hetherington <cth@carlh.net>
Sun, 2 May 2021 09:30:16 +0000 (11:30 +0200)
committerCarl Hetherington <cth@carlh.net>
Sun, 2 May 2021 23:39:32 +0000 (01:39 +0200)
Part of fix for #1984.

src/lib/image.cc
src/lib/image.h
test/image_test.cc

index 03fe494..63ae34c 100644 (file)
@@ -64,6 +64,15 @@ using dcp::Size;
 /** The memory alignment, in bytes, used for each row of an image if aligment is requested */
 #define ALIGNMENT 64
 
+/* U/V black value for 8-bit colour */
+static uint8_t const eight_bit_uv =    (1 << 7) - 1;
+/* U/V black value for 9-bit colour */
+static uint16_t const nine_bit_uv =    (1 << 8) - 1;
+/* U/V black value for 10-bit colour */
+static uint16_t const ten_bit_uv =     (1 << 9) - 1;
+/* U/V black value for 16-bit colour */
+static uint16_t const sixteen_bit_uv = (1 << 15) - 1;
+
 
 int
 Image::vertical_factor (int n) const
@@ -381,7 +390,7 @@ Image::swap_16 (uint16_t v)
 }
 
 void
-Image::make_part_black (int x, int w)
+Image::make_part_black (int const start, int const width)
 {
        switch (_pixel_format) {
        case AV_PIX_FMT_RGB24:
@@ -399,12 +408,33 @@ Image::make_part_black (int x, int w)
                int const s = stride()[0];
                uint8_t* p = data()[0];
                for (int y = 0; y < h; y++) {
-                       memset (p + x * bpp, 0, w * bpp);
+                       memset (p + start * bpp, 0, width * bpp);
                        p += s;
                }
                break;
        }
-
+       case AV_PIX_FMT_YUV422P10LE:
+       {
+               int const bpp_0 = bytes_per_pixel(0);
+               int const h_0 = sample_size(0).height;
+               int const stride_0 = stride()[0];
+               auto p = data()[0];
+               for (int y = 0; y < h_0; ++y) {
+                       memset (p + start * bpp_0, 0xff, width * bpp_0);
+                       p += stride_0;
+               }
+               for (int i = 1; i < 3; ++i) {
+                       auto p = reinterpret_cast<int16_t*>(data()[i]);
+                       int const lines = sample_size(i).height;
+                       for (int y = 0; y < lines; ++y) {
+                               for (int x = start / 2; x < (start + width) / 2; ++x) {
+                                       p[x] = ten_bit_uv;
+                               }
+                               p += stride()[i] / 2;
+                       }
+               }
+               break;
+       }
        default:
                throw PixelFormatError ("make_part_black()", _pixel_format);
        }
@@ -413,15 +443,6 @@ Image::make_part_black (int x, int w)
 void
 Image::make_black ()
 {
-       /* U/V black value for 8-bit colour */
-       static uint8_t const eight_bit_uv =     (1 << 7) - 1;
-       /* U/V black value for 9-bit colour */
-       static uint16_t const nine_bit_uv =     (1 << 8) - 1;
-       /* U/V black value for 10-bit colour */
-       static uint16_t const ten_bit_uv =      (1 << 9) - 1;
-       /* U/V black value for 16-bit colour */
-       static uint16_t const sixteen_bit_uv =  (1 << 15) - 1;
-
        switch (_pixel_format) {
        case AV_PIX_FMT_YUV420P:
        case AV_PIX_FMT_YUV422P:
index a582365..7edea21 100644 (file)
@@ -99,6 +99,7 @@ public:
 
 private:
        friend struct pixel_formats_test;
+       friend struct make_part_black_test;
 
        void allocate ();
        void swap (Image &);
index ed644ce..4cc80b7 100644 (file)
@@ -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.