media/mp4/avc_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <string.h>
   6
   7 #include "base/basictypes.h"
   8 #include "media/base/stream_parser_buffer.h"
   9 #include "media/mp4/avc.h"
  10 #include "media/mp4/box_definitions.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12 #include "testing/gtest/include/gtest/gtest-param-test.h"
  13
  14 namespace media {
  15 namespace mp4 {
  16
  17 static const uint8 kNALU1[] = { 0x01, 0x02, 0x03 };
  18 static const uint8 kNALU2[] = { 0x04, 0x05, 0x06, 0x07 };
  19 static const uint8 kExpected[] = {
  20   0x00, 0x00, 0x00, 0x01, 0x01, 0x02, 0x03,
  21   0x00, 0x00, 0x00, 0x01, 0x04, 0x05, 0x06, 0x07 };
  22
  23 static const uint8 kExpectedParamSets[] = {
  24   0x00, 0x00, 0x00, 0x01, 0x67, 0x12,
  25   0x00, 0x00, 0x00, 0x01, 0x67, 0x34,
  26   0x00, 0x00, 0x00, 0x01, 0x68, 0x56, 0x78};
  27
  28 class AVCConversionTest : public testing::TestWithParam<int> {
  29  protected:
  30   void MakeInputForLength(int length_size, std::vector<uint8>* buf) {
  31     buf->clear();
  32     for (int i = 1; i < length_size; i++)
  33       buf->push_back(0);
  34     buf->push_back(sizeof(kNALU1));
  35     buf->insert(buf->end(), kNALU1, kNALU1 + sizeof(kNALU1));
  36
  37     for (int i = 1; i < length_size; i++)
  38       buf->push_back(0);
  39     buf->push_back(sizeof(kNALU2));
  40     buf->insert(buf->end(), kNALU2, kNALU2 + sizeof(kNALU2));
  41   }
  42 };
  43
  44 TEST_P(AVCConversionTest, ParseCorrectly) {
  45   std::vector<uint8> buf;
  46   MakeInputForLength(GetParam(), &buf);
  47   EXPECT_TRUE(AVC::ConvertFrameToAnnexB(GetParam(), &buf));
  48   EXPECT_EQ(buf.size(), sizeof(kExpected));
  49   EXPECT_EQ(0, memcmp(kExpected, &buf[0], sizeof(kExpected)));
  50 }
  51
  52 TEST_P(AVCConversionTest, ParsePartial) {
  53   std::vector<uint8> buf;
  54   MakeInputForLength(GetParam(), &buf);
  55   buf.pop_back();
  56   EXPECT_FALSE(AVC::ConvertFrameToAnnexB(GetParam(), &buf));
  57   // This tests a buffer ending in the middle of a NAL length. For length size
  58   // of one, this can't happen, so we skip that case.
  59   if (GetParam() != 1) {
  60     MakeInputForLength(GetParam(), &buf);
  61     buf.erase(buf.end() - (sizeof(kNALU2) + 1), buf.end());
  62     EXPECT_FALSE(AVC::ConvertFrameToAnnexB(GetParam(), &buf));
  63   }
  64 }
  65
  66 TEST_P(AVCConversionTest, ParseEmpty) {
  67   std::vector<uint8> buf;
  68   EXPECT_TRUE(AVC::ConvertFrameToAnnexB(GetParam(), &buf));
  69   EXPECT_EQ(0u, buf.size());
  70 }
  71
  72 INSTANTIATE_TEST_CASE_P(AVCConversionTestValues,
  73                         AVCConversionTest,
  74                         ::testing::Values(1, 2, 4));
  75
  76 TEST_F(AVCConversionTest, ConvertConfigToAnnexB) {
  77   AVCDecoderConfigurationRecord avc_config;
  78   avc_config.sps_list.resize(2);
  79   avc_config.sps_list[0].push_back(0x67);
  80   avc_config.sps_list[0].push_back(0x12);
  81   avc_config.sps_list[1].push_back(0x67);
  82   avc_config.sps_list[1].push_back(0x34);
  83   avc_config.pps_list.resize(1);
  84   avc_config.pps_list[0].push_back(0x68);
  85   avc_config.pps_list[0].push_back(0x56);
  86   avc_config.pps_list[0].push_back(0x78);
  87
  88   std::vector<uint8> buf;
  89   EXPECT_TRUE(AVC::ConvertConfigToAnnexB(avc_config, &buf));
  90   EXPECT_EQ(0, memcmp(kExpectedParamSets, &buf[0],
  91                       sizeof(kExpectedParamSets)));
  92 }
  93
  94 }  // namespace mp4
  95 }  // namespace media