New media/base/decode_capabilities.h|cc

media/base/decode_capabilities_unittest.cc

Index: media/base/decode_capabilities_unittest.cc
diff --git a/media/base/decode_capabilities_unittest.cc b/media/base/decode_capabilities_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..c54881426683c58cefb4845fe857092ae042efeb
--- /dev/null
+++ b/media/base/decode_capabilities_unittest.cc
@@ -0,0 +1,128 @@
+// Copyright 2017 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/base/decode_capabilities.h"
+
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace media {
+
+TEST(DecodeCapabilitiesTest, IsSupportedVideoConfigBasics) {
+  // Default to common 709.
+  const media::VideoColorSpace kColorSpace = media::VideoColorSpace::REC709();
+
+  // Some codecs do not have a notion of level.
+  const int kUnspecifiedLevel = 0;
+
+  // Expect support for baseline configuration of known codecs.
+  EXPECT_TRUE(IsSupportedVideoConfig(
+      {media::kCodecH264, media::H264PROFILE_BASELINE, 1, kColorSpace}));
+  EXPECT_TRUE(IsSupportedVideoConfig({media::kCodecVP8, media::VP8PROFILE_ANY,
+                                      kUnspecifiedLevel, kColorSpace}));
+  EXPECT_TRUE(
+      IsSupportedVideoConfig({media::kCodecVP9, media::VP9PROFILE_PROFILE0,
+                              kUnspecifiedLevel, kColorSpace}));
+  EXPECT_TRUE(IsSupportedVideoConfig({media::kCodecTheora,
+                                      media::VIDEO_CODEC_PROFILE_UNKNOWN,
+                                      kUnspecifiedLevel, kColorSpace}));
+
+  // Expect non-support for the following.
+  EXPECT_FALSE(IsSupportedVideoConfig({media::kUnknownVideoCodec,
+                                       media::VIDEO_CODEC_PROFILE_UNKNOWN,
+                                       kUnspecifiedLevel, kColorSpace}));
+  EXPECT_FALSE(IsSupportedVideoConfig({media::kCodecVC1,
+                                       media::VIDEO_CODEC_PROFILE_UNKNOWN,
+                                       kUnspecifiedLevel, kColorSpace}));
+  EXPECT_FALSE(IsSupportedVideoConfig({media::kCodecMPEG2,
+                                       media::VIDEO_CODEC_PROFILE_UNKNOWN,
+                                       kUnspecifiedLevel, kColorSpace}));
+  EXPECT_FALSE(IsSupportedVideoConfig({media::kCodecMPEG4,
+                                       media::VIDEO_CODEC_PROFILE_UNKNOWN,
+                                       kUnspecifiedLevel, kColorSpace}));
+  EXPECT_FALSE(IsSupportedVideoConfig({media::kCodecHEVC,
+                                       media::VIDEO_CODEC_PROFILE_UNKNOWN,
+                                       kUnspecifiedLevel, kColorSpace}));
+}
+
+TEST(DecodeCapabilitiesTest, IsSupportedVideoConfig_VP9TransferFunctions) {
+  size_t num_found = 0;
+  // TODO(hubbe): Verify support for HDR codecs when color management enabled.
+  const std::set<media::VideoColorSpace::TransferID> kSupportedTransfers = {
+      media::VideoColorSpace::TransferID::GAMMA22,
+      media::VideoColorSpace::TransferID::UNSPECIFIED,
+      media::VideoColorSpace::TransferID::BT709,
+      media::VideoColorSpace::TransferID::SMPTE170M,
+      media::VideoColorSpace::TransferID::BT2020_10,
+      media::VideoColorSpace::TransferID::BT2020_12,
+      media::VideoColorSpace::TransferID::IEC61966_2_1,
+  };
+
+  for (int i = 0; i <= (1 << (8 * sizeof(media::VideoColorSpace::TransferID)));
+       i++) {
+    media::VideoColorSpace color_space = media::VideoColorSpace::REC709();
+    color_space.transfer = media::VideoColorSpace::GetTransferID(i);
+    bool found = kSupportedTransfers.find(color_space.transfer) !=
+                 kSupportedTransfers.end();
+    if (found)
+      num_found++;
+    EXPECT_EQ(found, IsSupportedVideoConfig({media::kCodecVP9,
+                                             media::VP9PROFILE_PROFILE0, 1,
+                                             color_space}));
+  }
+  EXPECT_EQ(kSupportedTransfers.size(), num_found);
+}
+
+TEST(DecodeCapabilitiesTest, IsSupportedVideoConfig_VP9Primaries) {
+  size_t num_found = 0;
+  // TODO(hubbe): Verify support for HDR codecs when color management enabled.
+  const std::set<media::VideoColorSpace::PrimaryID> kSupportedPrimaries = {
+      media::VideoColorSpace::PrimaryID::BT709,
+      media::VideoColorSpace::PrimaryID::UNSPECIFIED,
+      media::VideoColorSpace::PrimaryID::BT470M,
+      media::VideoColorSpace::PrimaryID::BT470BG,
+      media::VideoColorSpace::PrimaryID::SMPTE170M,
+  };
+
+  for (int i = 0; i <= (1 << (8 * sizeof(media::VideoColorSpace::PrimaryID)));
+       i++) {
+    media::VideoColorSpace color_space = media::VideoColorSpace::REC709();
+    color_space.primaries = media::VideoColorSpace::GetPrimaryID(i);
+    bool found = kSupportedPrimaries.find(color_space.primaries) !=
+                 kSupportedPrimaries.end();
+    if (found)
+      num_found++;
+    EXPECT_EQ(found, IsSupportedVideoConfig({media::kCodecVP9,
+                                             media::VP9PROFILE_PROFILE0, 1,
+                                             color_space}));
+  }
+  EXPECT_EQ(kSupportedPrimaries.size(), num_found);
+}
+
+TEST(DecodeCapabilitiesTest, IsSupportedVideoConfig_VP9Matrix) {
+  size_t num_found = 0;
+  // TODO(hubbe): Verify support for HDR codecs when color management enabled.
+  const std::set<media::VideoColorSpace::MatrixID> kSupportedMatrix = {
+      media::VideoColorSpace::MatrixID::BT709,
+      media::VideoColorSpace::MatrixID::UNSPECIFIED,
+      media::VideoColorSpace::MatrixID::BT470BG,
+      media::VideoColorSpace::MatrixID::SMPTE170M,
+      media::VideoColorSpace::MatrixID::BT2020_NCL,
+  };
+
+  for (int i = 0; i <= (1 << (8 * sizeof(media::VideoColorSpace::MatrixID)));
+       i++) {
+    media::VideoColorSpace color_space = media::VideoColorSpace::REC709();
+    color_space.matrix = media::VideoColorSpace::GetMatrixID(i);
+    bool found =
+        kSupportedMatrix.find(color_space.matrix) != kSupportedMatrix.end();
+    if (found)
+      num_found++;
+    EXPECT_EQ(found, IsSupportedVideoConfig({media::kCodecVP9,
+                                             media::VP9PROFILE_PROFILE0, 1,
+                                             color_space}));
+  }
+  EXPECT_EQ(kSupportedMatrix.size(), num_found);
+}
+
+}  // namespace media