Add color metadata info to VideoDecoderConfig.

media/formats/webm/webm_stream_parser_unittest.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "media/formats/webm/webm_stream_parser.h"
 
 #include <memory>
 
 #include "base/bind.h"
 #include "media/base/decoder_buffer.h"
@@ skipping 135 matching lines @@
   StreamParser::InitParameters params(kInfiniteDuration);
   params.detected_audio_track_count = 1;
   params.detected_video_track_count = 1;
   params.detected_text_track_count = 1;
   ParseWebMFile("bear-vp8-webvtt.webm", params);
   EXPECT_EQ(media_tracks_->tracks().size(), 2u);
   EXPECT_EQ(media_tracks_->tracks()[0]->type(), MediaTrack::Video);
   EXPECT_EQ(media_tracks_->tracks()[1]->type(), MediaTrack::Audio);
 }
 
+TEST_F(WebMStreamParserTest, ColourElement) {
+  EXPECT_MEDIA_LOG(testing::HasSubstr("Estimating WebM block duration"))
+      .Times(testing::AnyNumber());
+  StreamParser::InitParameters params(kInfiniteDuration);
+  params.detected_audio_track_count = 0;
+  params.detected_video_track_count = 1;
+  params.detected_text_track_count = 0;
+  ParseWebMFile("colour.webm", params);
+  EXPECT_EQ(media_tracks_->tracks().size(), 1u);
+
+  const auto& video_track = media_tracks_->tracks()[0];
+  EXPECT_EQ(video_track->type(), MediaTrack::Video);
+
+  const VideoDecoderConfig& video_config =
+      media_tracks_->getVideoConfig(video_track->bytestream_track_id());
+
+  gfx::ColorSpace expected_color_space(gfx::ColorSpace::PrimaryID::SMPTEST428_1,
+                                       gfx::ColorSpace::TransferID::LOG,
+                                       gfx::ColorSpace::MatrixID::RGB,
+                                       gfx::ColorSpace::RangeID::FULL);
+  EXPECT_EQ(video_config.color_space_info(), expected_color_space);
+
+  base::Optional<HDRMetadata> hdr_metadata = video_config.hdr_metadata();
+  EXPECT_TRUE(hdr_metadata.has_value());
+  EXPECT_EQ(hdr_metadata->max_cll, 11u);
+  EXPECT_EQ(hdr_metadata->max_fall, 12u);
+
+  const MasteringMetadata& mmdata = hdr_metadata->mastering_metadata;
+  EXPECT_FLOAT_EQ(mmdata.primary_r_chromaticity_x, 0.1f);
+  EXPECT_FLOAT_EQ(mmdata.primary_r_chromaticity_y, 0.2f);
+  EXPECT_FLOAT_EQ(mmdata.primary_g_chromaticity_x, 0.1f);
+  EXPECT_FLOAT_EQ(mmdata.primary_g_chromaticity_y, 0.2f);
+  EXPECT_FLOAT_EQ(mmdata.primary_b_chromaticity_x, 0.1f);
+  EXPECT_FLOAT_EQ(mmdata.primary_b_chromaticity_y, 0.2f);
+  EXPECT_FLOAT_EQ(mmdata.white_point_chromaticity_x, 0.1f);
+  EXPECT_FLOAT_EQ(mmdata.white_point_chromaticity_y, 0.2f);
+  EXPECT_EQ(mmdata.luminance_max, 40);
+  EXPECT_EQ(mmdata.luminance_min, 30);
+}
+
 }  // namespace media