vp8_parser_unittest.cc 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. // Copyright 2015 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. #include <stddef.h>
  5. #include <stdint.h>
  6. #include "base/files/memory_mapped_file.h"
  7. #include "base/logging.h"
  8. #include "media/base/test_data_util.h"
  9. #include "media/filters/ivf_parser.h"
  10. #include "media/parsers/vp8_parser.h"
  11. #include "testing/gtest/include/gtest/gtest.h"
  12. namespace media {
  13. TEST(Vp8ParserTest, StreamFileParsing) {
  14. base::FilePath file_path = GetTestDataFilePath("test-25fps.vp8");
  15. base::MemoryMappedFile stream;
  16. ASSERT_TRUE(stream.Initialize(file_path))
  17. << "Couldn't open stream file: " << file_path.MaybeAsASCII();
  18. IvfParser ivf_parser;
  19. IvfFileHeader ivf_file_header = {};
  20. ASSERT_TRUE(
  21. ivf_parser.Initialize(stream.data(), stream.length(), &ivf_file_header));
  22. ASSERT_EQ(ivf_file_header.fourcc, 0x30385056u); // VP80
  23. Vp8Parser vp8_parser;
  24. IvfFrameHeader ivf_frame_header = {};
  25. size_t num_parsed_frames = 0;
  26. // Parse until the end of stream/unsupported stream/error in stream is found.
  27. const uint8_t* payload = nullptr;
  28. while (ivf_parser.ParseNextFrame(&ivf_frame_header, &payload)) {
  29. Vp8FrameHeader fhdr;
  30. ASSERT_TRUE(
  31. vp8_parser.ParseFrame(payload, ivf_frame_header.frame_size, &fhdr));
  32. ++num_parsed_frames;
  33. }
  34. DVLOG(1) << "Number of successfully parsed frames before EOS: "
  35. << num_parsed_frames;
  36. EXPECT_EQ(ivf_file_header.num_frames, num_parsed_frames);
  37. }
  38. } // namespace media