media/filters/h264_parser_unittest.cc

   1 // Copyright 2014 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 "base/command_line.h"
   6 #include "base/files/memory_mapped_file.h"
   7 #include "base/logging.h"
   8 #include "base/path_service.h"
   9 #include "base/strings/string_number_conversions.h"
  10 #include "media/base/test_data_util.h"
  11 #include "media/filters/h264_parser.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 namespace media {
  15
  16 TEST(H264ParserTest, StreamFileParsing) {
  17   base::FilePath file_path = GetTestDataFilePath("test-25fps.h264");
  18   // Number of NALUs in the test stream to be parsed.
  19   int num_nalus = 759;
  20
  21   base::MemoryMappedFile stream;
  22   ASSERT_TRUE(stream.Initialize(file_path))
  23       << "Couldn't open stream file: " << file_path.MaybeAsASCII();
  24
  25   H264Parser parser;
  26   parser.SetStream(stream.data(), stream.length());
  27
  28   // Parse until the end of stream/unsupported stream/error in stream is found.
  29   int num_parsed_nalus = 0;
  30   while (true) {
  31     media::H264SliceHeader shdr;
  32     media::H264SEIMessage sei_msg;
  33     H264NALU nalu;
  34     H264Parser::Result res = parser.AdvanceToNextNALU(&nalu);
  35     if (res == H264Parser::kEOStream) {
  36       DVLOG(1) << "Number of successfully parsed NALUs before EOS: "
  37                << num_parsed_nalus;
  38       ASSERT_EQ(num_nalus, num_parsed_nalus);
  39       return;
  40     }
  41     ASSERT_EQ(res, H264Parser::kOk);
  42
  43     ++num_parsed_nalus;
  44
  45     int id;
  46     switch (nalu.nal_unit_type) {
  47       case H264NALU::kIDRSlice:
  48       case H264NALU::kNonIDRSlice:
  49         ASSERT_EQ(parser.ParseSliceHeader(nalu, &shdr), H264Parser::kOk);
  50         break;
  51
  52       case H264NALU::kSPS:
  53         ASSERT_EQ(parser.ParseSPS(&id), H264Parser::kOk);
  54         break;
  55
  56       case H264NALU::kPPS:
  57         ASSERT_EQ(parser.ParsePPS(&id), H264Parser::kOk);
  58         break;
  59
  60       case H264NALU::kSEIMessage:
  61         ASSERT_EQ(parser.ParseSEI(&sei_msg), H264Parser::kOk);
  62         break;
  63
  64       default:
  65         // Skip unsupported NALU.
  66         DVLOG(4) << "Skipping unsupported NALU";
  67         break;
  68     }
  69   }
  70 }
  71
  72 }  // namespace media