Index: media/gpu/h264_decoder_unittest.cc |
diff --git a/media/gpu/h264_decoder_unittest.cc b/media/gpu/h264_decoder_unittest.cc |
deleted file mode 100644 |
index fdd01f3715fb356c30d4d4cc2ca141478d3ff3ff..0000000000000000000000000000000000000000 |
--- a/media/gpu/h264_decoder_unittest.cc |
+++ /dev/null |
@@ -1,332 +0,0 @@ |
-// Copyright 2017 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 <stdint.h> |
-#include <string.h> |
- |
-#include <memory> |
-#include <queue> |
-#include <string> |
- |
-#include "base/command_line.h" |
-#include "base/files/file_util.h" |
-#include "base/logging.h" |
-#include "media/base/test_data_util.h" |
-#include "media/gpu/h264_decoder.h" |
-#include "testing/gmock/include/gmock/gmock.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-using ::testing::_; |
-using ::testing::Expectation; |
-using ::testing::InSequence; |
-using ::testing::Invoke; |
-using ::testing::MakeMatcher; |
-using ::testing::Matcher; |
-using ::testing::MatcherInterface; |
-using ::testing::MatchResultListener; |
-using ::testing::Mock; |
-using ::testing::Return; |
- |
-namespace media { |
-namespace { |
- |
-const std::string kBaselineFrame0 = "bear-320x192-baseline-frame-0.h264"; |
-const std::string kBaselineFrame1 = "bear-320x192-baseline-frame-1.h264"; |
-const std::string kBaselineFrame2 = "bear-320x192-baseline-frame-2.h264"; |
-const std::string kBaselineFrame3 = "bear-320x192-baseline-frame-3.h264"; |
-const std::string kHighFrame0 = "bear-320x192-high-frame-0.h264"; |
-const std::string kHighFrame1 = "bear-320x192-high-frame-1.h264"; |
-const std::string kHighFrame2 = "bear-320x192-high-frame-2.h264"; |
-const std::string kHighFrame3 = "bear-320x192-high-frame-3.h264"; |
- |
-class MockH264Accelerator : public H264Decoder::H264Accelerator { |
- public: |
- MockH264Accelerator() {} |
- |
- MOCK_METHOD0(CreateH264Picture, scoped_refptr<H264Picture>()); |
- MOCK_METHOD1(SubmitDecode, bool(const scoped_refptr<H264Picture>& pic)); |
- MOCK_METHOD1(OutputPicture, bool(const scoped_refptr<H264Picture>& pic)); |
- |
- bool SubmitFrameMetadata(const H264SPS* sps, |
- const H264PPS* pps, |
- const H264DPB& dpb, |
- const H264Picture::Vector& ref_pic_listp0, |
- const H264Picture::Vector& ref_pic_listb0, |
- const H264Picture::Vector& ref_pic_listb1, |
- const scoped_refptr<H264Picture>& pic) override { |
- return true; |
- } |
- |
- bool SubmitSlice(const H264PPS* pps, |
- const H264SliceHeader* slice_hdr, |
- const H264Picture::Vector& ref_pic_list0, |
- const H264Picture::Vector& ref_pic_list1, |
- const scoped_refptr<H264Picture>& pic, |
- const uint8_t* data, |
- size_t size) override { |
- return true; |
- } |
- |
- void Reset() override {} |
-}; |
- |
-// Test H264Decoder by feeding different of h264 frame sequences and make |
-// sure it behaves as expected. |
-class H264DecoderTest : public ::testing::Test { |
- public: |
- H264DecoderTest() = default; |
- |
- void SetUp() override; |
- |
- // Sets the bitstreams to be decoded, frame by frame. The content of each |
- // file is the encoded bitstream of a single video frame. |
- void SetInputFrameFiles(const std::vector<std::string>& frame_files); |
- |
- // Keeps decoding the input bitstream set at |SetInputFrameFiles| until the |
- // decoder has consumed all bitstreams or returned from |
- // |H264Decoder::Decode|. Returns the same result as |H264Decoder::Decode|. |
- AcceleratedVideoDecoder::DecodeResult Decode(); |
- |
- protected: |
- std::unique_ptr<H264Decoder> decoder_; |
- MockH264Accelerator accelerator_; |
- |
- private: |
- std::queue<std::string> input_frame_files_; |
- std::string bitstream_; |
-}; |
- |
-void H264DecoderTest::SetUp() { |
- decoder_.reset(new H264Decoder(&accelerator_)); |
- |
- // Sets default behaviors for mock methods for convenience. |
- ON_CALL(accelerator_, CreateH264Picture()).WillByDefault(Invoke([]() { |
- return new H264Picture(); |
- })); |
- ON_CALL(accelerator_, SubmitDecode(_)).WillByDefault(Return(true)); |
- ON_CALL(accelerator_, OutputPicture(_)).WillByDefault(Return(true)); |
-} |
- |
-void H264DecoderTest::SetInputFrameFiles( |
- const std::vector<std::string>& input_frame_files) { |
- for (auto f : input_frame_files) |
- input_frame_files_.push(f); |
-} |
- |
-AcceleratedVideoDecoder::DecodeResult H264DecoderTest::Decode() { |
- while (true) { |
- auto result = decoder_->Decode(); |
- if (result != AcceleratedVideoDecoder::kRanOutOfStreamData || |
- input_frame_files_.empty()) |
- return result; |
- auto input_file = GetTestDataFilePath(input_frame_files_.front()); |
- input_frame_files_.pop(); |
- CHECK(base::ReadFileToString(input_file, &bitstream_)); |
- decoder_->SetStream(reinterpret_cast<const uint8_t*>(bitstream_.data()), |
- bitstream_.size()); |
- } |
-} |
- |
-// To have better description on mismatch. |
-class WithPocMatcher |
- : public MatcherInterface<const scoped_refptr<H264Picture>&> { |
- public: |
- explicit WithPocMatcher(int expected_poc) : expected_poc_(expected_poc) {} |
- |
- bool MatchAndExplain(const scoped_refptr<H264Picture>& p, |
- MatchResultListener* listener) const override { |
- if (p->pic_order_cnt == expected_poc_) |
- return true; |
- *listener << "with poc: " << p->pic_order_cnt; |
- return false; |
- } |
- |
- void DescribeTo(std::ostream* os) const override { |
- *os << "with poc " << expected_poc_; |
- } |
- |
- private: |
- int expected_poc_; |
-}; |
- |
-inline Matcher<const scoped_refptr<H264Picture>&> WithPoc(int expected_poc) { |
- return MakeMatcher(new WithPocMatcher(expected_poc)); |
-} |
- |
-// Test Cases |
- |
-TEST_F(H264DecoderTest, DecodeSingleFrame) { |
- SetInputFrameFiles({kBaselineFrame0}); |
- ASSERT_EQ(AcceleratedVideoDecoder::kAllocateNewSurfaces, Decode()); |
- EXPECT_EQ(gfx::Size(320, 192), decoder_->GetPicSize()); |
- EXPECT_LE(9u, decoder_->GetRequiredNumOfPictures()); |
- |
- EXPECT_CALL(accelerator_, CreateH264Picture()).WillOnce(Return(nullptr)); |
- ASSERT_EQ(AcceleratedVideoDecoder::kRanOutOfSurfaces, Decode()); |
- ASSERT_TRUE(Mock::VerifyAndClearExpectations(&accelerator_)); |
- |
- { |
- InSequence sequence; |
- EXPECT_CALL(accelerator_, CreateH264Picture()); |
- EXPECT_CALL(accelerator_, SubmitDecode(_)); |
- EXPECT_CALL(accelerator_, OutputPicture(_)); |
- } |
- ASSERT_EQ(AcceleratedVideoDecoder::kRanOutOfStreamData, Decode()); |
- ASSERT_TRUE(decoder_->Flush()); |
-} |
- |
-TEST_F(H264DecoderTest, SkipNonIDRFrames) { |
- SetInputFrameFiles({kBaselineFrame1, kBaselineFrame2, kBaselineFrame0}); |
- ASSERT_EQ(AcceleratedVideoDecoder::kAllocateNewSurfaces, Decode()); |
- EXPECT_EQ(gfx::Size(320, 192), decoder_->GetPicSize()); |
- EXPECT_LE(9u, decoder_->GetRequiredNumOfPictures()); |
- { |
- InSequence sequence; |
- EXPECT_CALL(accelerator_, CreateH264Picture()); |
- EXPECT_CALL(accelerator_, SubmitDecode(_)); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(0))); |
- } |
- ASSERT_EQ(AcceleratedVideoDecoder::kRanOutOfStreamData, Decode()); |
- ASSERT_TRUE(decoder_->Flush()); |
-} |
- |
-TEST_F(H264DecoderTest, DecodeProfileBaseline) { |
- SetInputFrameFiles({ |
- kBaselineFrame0, kBaselineFrame1, kBaselineFrame2, kBaselineFrame3, |
- }); |
- ASSERT_EQ(AcceleratedVideoDecoder::kAllocateNewSurfaces, Decode()); |
- EXPECT_EQ(gfx::Size(320, 192), decoder_->GetPicSize()); |
- EXPECT_LE(9u, decoder_->GetRequiredNumOfPictures()); |
- |
- EXPECT_CALL(accelerator_, CreateH264Picture()).Times(4); |
- Expectation decode_poc0, decode_poc2, decode_poc4, decode_poc6; |
- { |
- InSequence decode_order; |
- decode_poc0 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(0))); |
- decode_poc2 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(2))); |
- decode_poc4 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(4))); |
- decode_poc6 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(6))); |
- } |
- { |
- InSequence display_order; |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(0))).After(decode_poc0); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(2))).After(decode_poc2); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(4))).After(decode_poc4); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(6))).After(decode_poc6); |
- } |
- ASSERT_EQ(AcceleratedVideoDecoder::kRanOutOfStreamData, Decode()); |
- ASSERT_TRUE(decoder_->Flush()); |
-} |
- |
-TEST_F(H264DecoderTest, DecodeProfileHigh) { |
- SetInputFrameFiles({kHighFrame0, kHighFrame1, kHighFrame2, kHighFrame3}); |
- ASSERT_EQ(AcceleratedVideoDecoder::kAllocateNewSurfaces, Decode()); |
- EXPECT_EQ(gfx::Size(320, 192), decoder_->GetPicSize()); |
- EXPECT_LE(16u, decoder_->GetRequiredNumOfPictures()); |
- |
- // Two pictures will be kept in DPB for reordering. The first picture should |
- // be outputted after feeding the third frame. |
- EXPECT_CALL(accelerator_, CreateH264Picture()).Times(4); |
- Expectation decode_poc0, decode_poc2, decode_poc4, decode_poc6; |
- { |
- InSequence decode_order; |
- decode_poc0 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(0))); |
- decode_poc4 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(4))); |
- decode_poc2 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(2))); |
- decode_poc6 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(6))); |
- } |
- { |
- InSequence display_order; |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(0))).After(decode_poc0); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(2))).After(decode_poc2); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(4))).After(decode_poc4); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(6))).After(decode_poc6); |
- } |
- ASSERT_EQ(AcceleratedVideoDecoder::kRanOutOfStreamData, Decode()); |
- ASSERT_TRUE(decoder_->Flush()); |
-} |
- |
-TEST_F(H264DecoderTest, SwitchBaselineToHigh) { |
- SetInputFrameFiles({ |
- kBaselineFrame0, kHighFrame0, kHighFrame1, kHighFrame2, kHighFrame3, |
- }); |
- ASSERT_EQ(AcceleratedVideoDecoder::kAllocateNewSurfaces, Decode()); |
- EXPECT_EQ(gfx::Size(320, 192), decoder_->GetPicSize()); |
- EXPECT_LE(9u, decoder_->GetRequiredNumOfPictures()); |
- |
- EXPECT_CALL(accelerator_, CreateH264Picture()); |
- { |
- InSequence sequence; |
- EXPECT_CALL(accelerator_, SubmitDecode(_)); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(0))); |
- } |
- ASSERT_EQ(AcceleratedVideoDecoder::kAllocateNewSurfaces, Decode()); |
- EXPECT_EQ(gfx::Size(320, 192), decoder_->GetPicSize()); |
- EXPECT_LE(16u, decoder_->GetRequiredNumOfPictures()); |
- |
- ASSERT_TRUE(Mock::VerifyAndClearExpectations(&accelerator_)); |
- |
- EXPECT_CALL(accelerator_, CreateH264Picture()).Times(4); |
- Expectation decode_poc0, decode_poc2, decode_poc4, decode_poc6; |
- { |
- InSequence decode_order; |
- decode_poc0 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(0))); |
- decode_poc4 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(4))); |
- decode_poc2 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(2))); |
- decode_poc6 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(6))); |
- } |
- { |
- InSequence display_order; |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(0))).After(decode_poc0); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(2))).After(decode_poc2); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(4))).After(decode_poc4); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(6))).After(decode_poc6); |
- } |
- ASSERT_EQ(AcceleratedVideoDecoder::kRanOutOfStreamData, Decode()); |
- ASSERT_TRUE(decoder_->Flush()); |
-} |
- |
-TEST_F(H264DecoderTest, SwitchHighToBaseline) { |
- SetInputFrameFiles({ |
- kHighFrame0, kBaselineFrame0, kBaselineFrame1, kBaselineFrame2, |
- kBaselineFrame3, |
- }); |
- ASSERT_EQ(AcceleratedVideoDecoder::kAllocateNewSurfaces, Decode()); |
- EXPECT_EQ(gfx::Size(320, 192), decoder_->GetPicSize()); |
- EXPECT_LE(16u, decoder_->GetRequiredNumOfPictures()); |
- |
- EXPECT_CALL(accelerator_, CreateH264Picture()); |
- { |
- InSequence sequence; |
- EXPECT_CALL(accelerator_, SubmitDecode(_)); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(0))); |
- } |
- ASSERT_EQ(AcceleratedVideoDecoder::kAllocateNewSurfaces, Decode()); |
- EXPECT_EQ(gfx::Size(320, 192), decoder_->GetPicSize()); |
- EXPECT_LE(9u, decoder_->GetRequiredNumOfPictures()); |
- |
- ASSERT_TRUE(Mock::VerifyAndClearExpectations(&accelerator_)); |
- |
- EXPECT_CALL(accelerator_, CreateH264Picture()).Times(4); |
- Expectation decode_poc0, decode_poc2, decode_poc4, decode_poc6; |
- { |
- InSequence decode_order; |
- decode_poc0 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(0))); |
- decode_poc2 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(2))); |
- decode_poc4 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(4))); |
- decode_poc6 = EXPECT_CALL(accelerator_, SubmitDecode(WithPoc(6))); |
- } |
- { |
- InSequence display_order; |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(0))).After(decode_poc0); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(2))).After(decode_poc2); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(4))).After(decode_poc4); |
- EXPECT_CALL(accelerator_, OutputPicture(WithPoc(6))).After(decode_poc6); |
- } |
- ASSERT_EQ(AcceleratedVideoDecoder::kRanOutOfStreamData, Decode()); |
- ASSERT_TRUE(decoder_->Flush()); |
-} |
- |
-} // namespace |
-} // namespace media |