Add image_as_jpeg()

[dcpomatic.git] / test / test.cc

diff --git a/test/test.cc b/test/test.cc
index 6b2e584a200af4bec0f4c70b30ec2d9dfdef9328..7136b038d99bbb9c28bc1a8569cdb0f6b226e72f 100644 (file)
--- a/test/test.cc
+++ b/test/test.cc
@@ -369,6 +369,8 @@ rms_error (boost::filesystem::path ref, boost::filesystem::path check)
        FFmpegImageProxy check_proxy (check);
        auto check_image = check_proxy.image(Image::Alignment::COMPACT).image;
 
+       BOOST_REQUIRE_EQUAL (ref_image->planes(), check_image->planes());
+
        BOOST_REQUIRE_EQUAL (ref_image->pixel_format(), check_image->pixel_format());
        auto const format = ref_image->pixel_format();
 
@@ -417,6 +419,19 @@ rms_error (boost::filesystem::path ref, boost::filesystem::path check)
                        }
                        break;
                }
+               case AV_PIX_FMT_YUVJ420P:
+               {
+                       for (int c = 0; c < ref_image->planes(); ++c) {
+                               for (int y = 0; y < height / ref_image->vertical_factor(c); ++y) {
+                                       auto p = ref_image->data()[c] + y * ref_image->stride()[c];
+                                       auto q = check_image->data()[c] + y * check_image->stride()[c];
+                                       for (int x = 0; x < ref_image->line_size()[c]; ++x) {
+                                               sum_square += pow((*p++ - *q++), 2);
+                                       }
+                               }
+                       }
+                       break;
+               }
                default:
                        BOOST_REQUIRE_MESSAGE (false, "unrecognised pixel format " << format);
        }