ivf_parser_unittest.cc 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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 "media/base/test_data_util.h"
  8. #include "media/filters/ivf_parser.h"
  9. #include "testing/gtest/include/gtest/gtest.h"
  10. namespace media {
  11. TEST(IvfParserTest, StreamFileParsing) {
  12. base::FilePath file_path = GetTestDataFilePath("test-25fps.vp8");
  13. base::MemoryMappedFile stream;
  14. ASSERT_TRUE(stream.Initialize(file_path)) << "Couldn't open stream file: "
  15. << file_path.MaybeAsASCII();
  16. IvfParser parser;
  17. IvfFileHeader file_header = {};
  18. EXPECT_TRUE(parser.Initialize(stream.data(), stream.length(), &file_header));
  19. // Check file header fields.
  20. EXPECT_EQ(0, memcmp(file_header.signature, kIvfHeaderSignature,
  21. sizeof(file_header.signature)));
  22. EXPECT_EQ(0, file_header.version);
  23. EXPECT_EQ(sizeof(IvfFileHeader), file_header.header_size);
  24. EXPECT_EQ(0x30385056u, file_header.fourcc); // VP80
  25. EXPECT_EQ(320u, file_header.width);
  26. EXPECT_EQ(240u, file_header.height);
  27. EXPECT_EQ(50u, file_header.timebase_denum);
  28. EXPECT_EQ(2u, file_header.timebase_num);
  29. EXPECT_EQ(250u, file_header.num_frames);
  30. IvfFrameHeader frame_header;
  31. size_t num_parsed_frames = 0;
  32. const uint8_t* payload = nullptr;
  33. while (parser.ParseNextFrame(&frame_header, &payload)) {
  34. ++num_parsed_frames;
  35. EXPECT_TRUE(payload != nullptr);
  36. // Only check the first frame.
  37. if (num_parsed_frames == 1u) {
  38. EXPECT_EQ(14788u, frame_header.frame_size);
  39. EXPECT_EQ(0u, frame_header.timestamp);
  40. EXPECT_EQ(
  41. static_cast<ptrdiff_t>(sizeof(file_header) + sizeof(frame_header)),
  42. payload - stream.data());
  43. }
  44. }
  45. EXPECT_EQ(file_header.num_frames, num_parsed_frames);
  46. }
  47. } // namespace media