| Index: webrtc/modules/video_coding/frame_buffer2_unittest.cc
|
| diff --git a/webrtc/modules/video_coding/frame_buffer2_unittest.cc b/webrtc/modules/video_coding/frame_buffer2_unittest.cc
|
| deleted file mode 100644
|
| index 67706ce05819b51ec8fc6befeea2ac374c23dd9f..0000000000000000000000000000000000000000
|
| --- a/webrtc/modules/video_coding/frame_buffer2_unittest.cc
|
| +++ /dev/null
|
| @@ -1,329 +0,0 @@
|
| -/*
|
| - * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
|
| - *
|
| - * Use of this source code is governed by a BSD-style license
|
| - * that can be found in the LICENSE file in the root of the source
|
| - * tree. An additional intellectual property rights grant can be found
|
| - * in the file PATENTS. All contributing project authors may
|
| - * be found in the AUTHORS file in the root of the source tree.
|
| - */
|
| -
|
| -#include "webrtc/modules/video_coding/frame_buffer2.h"
|
| -
|
| -#include <algorithm>
|
| -#include <cstring>
|
| -#include <limits>
|
| -#include <vector>
|
| -
|
| -#include "testing/gmock/include/gmock/gmock.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -#include "webrtc/base/platform_thread.h"
|
| -#include "webrtc/base/random.h"
|
| -#include "webrtc/modules/video_coding/frame_object.h"
|
| -#include "webrtc/modules/video_coding/jitter_estimator.h"
|
| -#include "webrtc/modules/video_coding/sequence_number_util.h"
|
| -#include "webrtc/modules/video_coding/timing.h"
|
| -#include "webrtc/system_wrappers/include/clock.h"
|
| -
|
| -namespace webrtc {
|
| -namespace video_coding {
|
| -
|
| -class VCMTimingFake : public VCMTiming {
|
| - public:
|
| - explicit VCMTimingFake(Clock* clock) : VCMTiming(clock) {}
|
| -
|
| - int64_t RenderTimeMs(uint32_t frame_timestamp,
|
| - int64_t now_ms) const override {
|
| - if (last_ms_ == -1) {
|
| - last_ms_ = now_ms + kDelayMs;
|
| - last_timestamp_ = frame_timestamp;
|
| - }
|
| -
|
| - uint32_t diff = MinDiff(frame_timestamp, last_timestamp_);
|
| - if (AheadOf(frame_timestamp, last_timestamp_))
|
| - last_ms_ += diff / 90;
|
| - else
|
| - last_ms_ -= diff / 90;
|
| -
|
| - last_timestamp_ = frame_timestamp;
|
| - return last_ms_;
|
| - }
|
| -
|
| - uint32_t MaxWaitingTime(int64_t render_time_ms,
|
| - int64_t now_ms) const override {
|
| - return std::max<int>(0, render_time_ms - now_ms - kDecodeTime);
|
| - }
|
| -
|
| - private:
|
| - static constexpr int kDelayMs = 50;
|
| - static constexpr int kDecodeTime = kDelayMs / 2;
|
| - mutable uint32_t last_timestamp_ = 0;
|
| - mutable int64_t last_ms_ = -1;
|
| -};
|
| -
|
| -class VCMJitterEstimatorMock : public VCMJitterEstimator {
|
| - public:
|
| - explicit VCMJitterEstimatorMock(Clock* clock) : VCMJitterEstimator(clock) {}
|
| -
|
| - MOCK_METHOD1(UpdateRtt, void(int64_t rttMs));
|
| - MOCK_METHOD3(UpdateEstimate,
|
| - void(int64_t frameDelayMs,
|
| - uint32_t frameSizeBytes,
|
| - bool incompleteFrame));
|
| -};
|
| -
|
| -class FrameObjectMock : public FrameObject {
|
| - public:
|
| - MOCK_CONST_METHOD1(GetBitstream, bool(uint8_t* destination));
|
| -};
|
| -
|
| -class TestFrameBuffer2 : public ::testing::Test {
|
| - protected:
|
| - static constexpr int kMaxReferences = 5;
|
| - static constexpr int kFps1 = 1000;
|
| - static constexpr int kFps10 = kFps1 / 10;
|
| - static constexpr int kFps20 = kFps1 / 20;
|
| -
|
| - TestFrameBuffer2()
|
| - : clock_(0),
|
| - timing_(&clock_),
|
| - jitter_estimator_(&clock_),
|
| - buffer_(&clock_, &jitter_estimator_, &timing_),
|
| - rand_(0x34678213),
|
| - tear_down_(false),
|
| - extract_thread_(&ExtractLoop, this, "Extract Thread"),
|
| - trigger_extract_event_(false, false),
|
| - crit_acquired_event_(false, false) {}
|
| -
|
| - void SetUp() override { extract_thread_.Start(); }
|
| -
|
| - void TearDown() override {
|
| - tear_down_ = true;
|
| - trigger_extract_event_.Set();
|
| - extract_thread_.Stop();
|
| - }
|
| -
|
| - template <typename... T>
|
| - void InsertFrame(uint16_t picture_id,
|
| - uint8_t spatial_layer,
|
| - int64_t ts_ms,
|
| - bool inter_layer_predicted,
|
| - T... refs) {
|
| - static_assert(sizeof...(refs) <= kMaxReferences,
|
| - "To many references specified for FrameObject.");
|
| - std::array<uint16_t, sizeof...(refs)> references = {{refs...}};
|
| -
|
| - std::unique_ptr<FrameObjectMock> frame(new FrameObjectMock());
|
| - frame->picture_id = picture_id;
|
| - frame->spatial_layer = spatial_layer;
|
| - frame->timestamp = ts_ms * 90;
|
| - frame->num_references = references.size();
|
| - frame->inter_layer_predicted = inter_layer_predicted;
|
| - for (size_t r = 0; r < references.size(); ++r)
|
| - frame->references[r] = references[r];
|
| -
|
| - buffer_.InsertFrame(std::move(frame));
|
| - }
|
| -
|
| - void ExtractFrame(int64_t max_wait_time = 0) {
|
| - crit_.Enter();
|
| - if (max_wait_time == 0) {
|
| - frames_.emplace_back(buffer_.NextFrame(0));
|
| - crit_.Leave();
|
| - } else {
|
| - max_wait_time_ = max_wait_time;
|
| - trigger_extract_event_.Set();
|
| - crit_.Leave();
|
| - // Make sure |crit_| is aquired by |extract_thread_| before returning.
|
| - crit_acquired_event_.Wait(rtc::Event::kForever);
|
| - }
|
| - }
|
| -
|
| - void CheckFrame(size_t index, int picture_id, int spatial_layer) {
|
| - rtc::CritScope lock(&crit_);
|
| - ASSERT_LT(index, frames_.size());
|
| - ASSERT_TRUE(frames_[index]);
|
| - ASSERT_EQ(picture_id, frames_[index]->picture_id);
|
| - ASSERT_EQ(spatial_layer, frames_[index]->spatial_layer);
|
| - }
|
| -
|
| - void CheckNoFrame(size_t index) {
|
| - rtc::CritScope lock(&crit_);
|
| - ASSERT_LT(index, frames_.size());
|
| - ASSERT_FALSE(frames_[index]);
|
| - }
|
| -
|
| - static bool ExtractLoop(void* obj) {
|
| - TestFrameBuffer2* tfb = static_cast<TestFrameBuffer2*>(obj);
|
| - while (true) {
|
| - tfb->trigger_extract_event_.Wait(rtc::Event::kForever);
|
| - {
|
| - rtc::CritScope lock(&tfb->crit_);
|
| - tfb->crit_acquired_event_.Set();
|
| - if (tfb->tear_down_)
|
| - return false;
|
| -
|
| - tfb->frames_.emplace_back(tfb->buffer_.NextFrame(tfb->max_wait_time_));
|
| - }
|
| - }
|
| - }
|
| -
|
| - uint32_t Rand() { return rand_.Rand<uint32_t>(); }
|
| -
|
| - SimulatedClock clock_;
|
| - VCMTimingFake timing_;
|
| - VCMJitterEstimatorMock jitter_estimator_;
|
| - FrameBuffer buffer_;
|
| - std::vector<std::unique_ptr<FrameObject>> frames_;
|
| - Random rand_;
|
| -
|
| - int64_t max_wait_time_;
|
| - bool tear_down_;
|
| - rtc::PlatformThread extract_thread_;
|
| - rtc::Event trigger_extract_event_;
|
| - rtc::Event crit_acquired_event_;
|
| - rtc::CriticalSection crit_;
|
| -};
|
| -
|
| -TEST_F(TestFrameBuffer2, ExtractFromEmptyBuffer) {
|
| - ExtractFrame();
|
| - CheckNoFrame(0);
|
| -}
|
| -
|
| -TEST_F(TestFrameBuffer2, WaitForFrame) {
|
| - uint16_t pid = Rand();
|
| - uint32_t ts = Rand();
|
| -
|
| - ExtractFrame(20);
|
| - InsertFrame(pid, 0, ts, false);
|
| - CheckFrame(0, pid, 0);
|
| -}
|
| -
|
| -TEST_F(TestFrameBuffer2, OneSuperFrame) {
|
| - uint16_t pid = Rand();
|
| - uint32_t ts = Rand();
|
| -
|
| - ExtractFrame(20);
|
| - InsertFrame(pid, 1, ts, true);
|
| - InsertFrame(pid, 0, ts, false);
|
| - ExtractFrame();
|
| -
|
| - CheckFrame(0, pid, 0);
|
| - CheckFrame(1, pid, 1);
|
| -}
|
| -
|
| -TEST_F(TestFrameBuffer2, OneLayerStream) {
|
| - uint16_t pid = Rand();
|
| - uint32_t ts = Rand();
|
| -
|
| - InsertFrame(pid, 0, ts, false);
|
| - ExtractFrame();
|
| - CheckFrame(0, pid, 0);
|
| - for (int i = 1; i < 10; ++i) {
|
| - InsertFrame(pid + i, 0, ts + i * kFps10, false, pid + i - 1);
|
| - ExtractFrame();
|
| - clock_.AdvanceTimeMilliseconds(kFps10);
|
| - CheckFrame(i, pid + i, 0);
|
| - }
|
| -}
|
| -
|
| -TEST_F(TestFrameBuffer2, OneLayerStreamReordered) {
|
| - uint16_t pid = Rand();
|
| - uint32_t ts = Rand();
|
| -
|
| - InsertFrame(pid, 0, ts, false);
|
| - ExtractFrame();
|
| - CheckFrame(0, pid, 0);
|
| - for (int i = 1; i < 10; i += 2) {
|
| - ExtractFrame(15);
|
| - InsertFrame(pid + i + 1, 0, ts + (i + 1) * kFps10, false, pid + i);
|
| - clock_.AdvanceTimeMilliseconds(kFps10);
|
| - InsertFrame(pid + i, 0, ts + i * kFps10, false, pid + i - 1);
|
| - clock_.AdvanceTimeMilliseconds(kFps10);
|
| - ExtractFrame();
|
| - CheckFrame(i, pid + i, 0);
|
| - CheckFrame(i + 1, pid + i + 1, 0);
|
| - }
|
| -}
|
| -
|
| -TEST_F(TestFrameBuffer2, DropTemporalLayerSlowDecoder) {
|
| - uint16_t pid = Rand();
|
| - uint32_t ts = Rand();
|
| -
|
| - InsertFrame(pid, 0, ts, false);
|
| - InsertFrame(pid + 1, 0, ts + kFps20, false);
|
| - for (int i = 2; i < 10; i += 2) {
|
| - uint32_t ts_tl0 = ts + i / 2 * kFps10;
|
| - InsertFrame(pid + i, 0, ts_tl0, false, pid + i - 2);
|
| - InsertFrame(pid + i + 1, 0, ts_tl0 + kFps20, false, pid + i, pid + i - 1);
|
| - }
|
| -
|
| - for (int i = 0; i < 10; ++i) {
|
| - ExtractFrame();
|
| - clock_.AdvanceTimeMilliseconds(60);
|
| - }
|
| -
|
| - CheckFrame(0, pid, 0);
|
| - CheckFrame(1, pid + 1, 0);
|
| - CheckFrame(2, pid + 2, 0);
|
| - CheckFrame(3, pid + 4, 0);
|
| - CheckFrame(4, pid + 6, 0);
|
| - CheckFrame(5, pid + 8, 0);
|
| - CheckNoFrame(6);
|
| - CheckNoFrame(7);
|
| - CheckNoFrame(8);
|
| - CheckNoFrame(9);
|
| -}
|
| -
|
| -TEST_F(TestFrameBuffer2, DropSpatialLayerSlowDecoder) {
|
| - uint16_t pid = Rand();
|
| - uint32_t ts = Rand();
|
| -
|
| - InsertFrame(pid, 0, ts, false);
|
| - InsertFrame(pid, 1, ts, false);
|
| - for (int i = 1; i < 6; ++i) {
|
| - uint32_t ts_tl0 = ts + i * kFps10;
|
| - InsertFrame(pid + i, 0, ts_tl0, false, pid + i - 1);
|
| - InsertFrame(pid + i, 1, ts_tl0, false, pid + i - 1);
|
| - }
|
| -
|
| - ExtractFrame();
|
| - ExtractFrame();
|
| - clock_.AdvanceTimeMilliseconds(55);
|
| - for (int i = 2; i < 12; ++i) {
|
| - ExtractFrame();
|
| - clock_.AdvanceTimeMilliseconds(55);
|
| - }
|
| -
|
| - CheckFrame(0, pid, 0);
|
| - CheckFrame(1, pid, 1);
|
| - CheckFrame(2, pid + 1, 0);
|
| - CheckFrame(3, pid + 1, 1);
|
| - CheckFrame(4, pid + 2, 0);
|
| - CheckFrame(5, pid + 2, 1);
|
| - CheckFrame(6, pid + 3, 0);
|
| - CheckFrame(7, pid + 4, 0);
|
| - CheckFrame(8, pid + 5, 0);
|
| - CheckNoFrame(9);
|
| - CheckNoFrame(10);
|
| - CheckNoFrame(11);
|
| -}
|
| -
|
| -TEST_F(TestFrameBuffer2, InsertLateFrame) {
|
| - uint16_t pid = Rand();
|
| - uint32_t ts = Rand();
|
| -
|
| - InsertFrame(pid, 0, ts, false);
|
| - ExtractFrame();
|
| - InsertFrame(pid + 2, 0, ts, false);
|
| - ExtractFrame();
|
| - InsertFrame(pid + 1, 0, ts, false, pid);
|
| - ExtractFrame();
|
| -
|
| - CheckFrame(0, pid, 0);
|
| - CheckFrame(1, pid + 2, 0);
|
| - CheckNoFrame(2);
|
| -}
|
| -
|
| -} // namespace video_coding
|
| -} // namespace webrtc
|
|
|