Build AVC decoder configuration record

content/common/gpu/media/avc_config_record_builder_unittest.cc

Index: content/common/gpu/media/avc_config_record_builder_unittest.cc
diff --git a/content/common/gpu/media/avc_config_record_builder_unittest.cc b/content/common/gpu/media/avc_config_record_builder_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..7b926cdbd73cb6d692096222093e292c30f73865
--- /dev/null
+++ b/content/common/gpu/media/avc_config_record_builder_unittest.cc
@@ -0,0 +1,70 @@
+// Copyright (c) 2012 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 "content/common/gpu/media/avc_config_record_builder.h"
+
+#include "content/common/gpu/media/h264_parser.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+const uint8_t kSPSData[] = {
+  0x67, 0x64, 0x00, 0x1f, 0xac, 0x34, 0xec, 0x05,
+  0x00, 0x5b, 0xa1, 0x00, 0x00, 0x03, 0x00, 0x01,
+  0x00, 0x00, 0x03, 0x00, 0x32, 0x0f, 0x18, 0x31,
+  0x38,
+};
+
+const uint8_t kPPSData[] = {
+  0x68, 0xef, 0xb2, 0xc8, 0xb0,
+};
+
+const uint8_t kNALUHeader[] = {
+  0x00, 0x00, 0x00, 0x01
+};
+
+}  // namespace
+
+TEST(AVCConfigRecordBuilderTest, BuildConfig) {
+  std::vector<uint8_t> stream;
+  stream.insert(stream.end(), kNALUHeader,
+                kNALUHeader + arraysize(kNALUHeader));
+  stream.insert(stream.end(), kSPSData, kSPSData + arraysize(kSPSData));
+  stream.insert(stream.end(), kNALUHeader,
+                kNALUHeader + arraysize(kNALUHeader));
+  stream.insert(stream.end(), kPPSData, kPPSData + arraysize(kPPSData));
+
+  content::H264Parser parser;
+  parser.SetStream(&stream[0], stream.size());
+  content::H264NALU nalu;
+  ASSERT_EQ(parser.AdvanceToNextNALU(&nalu), content::H264Parser::kOk);
+
+  content::AVCConfigRecordBuilder config;
+  std::vector<uint8_t> config_data;
+  EXPECT_TRUE(config.ProcessNALU(&parser, nalu, &config_data));
+  EXPECT_TRUE(config_data.empty());
+
+  ASSERT_EQ(parser.AdvanceToNextNALU(&nalu), content::H264Parser::kOk);
+
+  EXPECT_TRUE(config.ProcessNALU(&parser, nalu, &config_data));
+  EXPECT_TRUE(config_data.empty());
+
+  nalu.nal_unit_type = content::H264NALU::kIDRSlice;
+  EXPECT_TRUE(config.ProcessNALU(&parser, nalu, &config_data));
+  EXPECT_FALSE(config_data.empty());
+  EXPECT_EQ(config.coded_width(), 1280);
+  EXPECT_EQ(config.coded_height(), 720);
+
+  const uint8_t EXPECTED_DATA[] = {

[Ami GONE FROM CHROMIUM, 2012/05/30 00:07:54]

kExpectedData

[sail, 2012/06/02 21:05:49]

Done.

+    0x01, 0x64, 0x00, 0x1f,    0xff, 0xe1, 0x00, 0x19,
+    0x67, 0x64, 0x00, 0x1f,    0xac, 0x34, 0xec, 0x05,
+    0x00, 0x5b, 0xa1, 0x00,    0x00, 0x03, 0x00, 0x01,
+    0x00, 0x00, 0x03, 0x00,    0x32, 0x0f, 0x18, 0x31,
+    0x38, 0x01, 0x00, 0x05,    0x68, 0xef, 0xb2, 0xc8,
+    0xb0,
+  };
+  std::vector<uint8_t> expected_config_data(
+        EXPECTED_DATA, EXPECTED_DATA + arraysize(EXPECTED_DATA));
+  EXPECT_EQ(config_data, expected_config_data);
+}