| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| (...skipping 622 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 633 } | 633 } |
| 634 }; | 634 }; |
| 635 | 635 |
| 636 class BitrateObserver : public test::EndToEndTest, public test::FakeEncoder { | 636 class BitrateObserver : public test::EndToEndTest, public test::FakeEncoder { |
| 637 public: | 637 public: |
| 638 BitrateObserver() | 638 BitrateObserver() |
| 639 : EndToEndTest(kDefaultTimeoutMs), | 639 : EndToEndTest(kDefaultTimeoutMs), |
| 640 FakeEncoder(Clock::GetRealTimeClock()), | 640 FakeEncoder(Clock::GetRealTimeClock()), |
| 641 time_to_reconfigure_(false, false), | 641 time_to_reconfigure_(false, false), |
| 642 encoder_inits_(0), | 642 encoder_inits_(0), |
| 643 last_set_bitrate_kbps_(0), | 643 last_set_bitrate_(0), |
| 644 send_stream_(nullptr), | 644 send_stream_(nullptr) {} |
| 645 frame_generator_(nullptr) {} | |
| 646 | 645 |
| 647 int32_t InitEncode(const VideoCodec* config, | 646 int32_t InitEncode(const VideoCodec* config, |
| 648 int32_t number_of_cores, | 647 int32_t number_of_cores, |
| 649 size_t max_payload_size) override { | 648 size_t max_payload_size) override { |
| 650 ++encoder_inits_; | 649 ++encoder_inits_; |
| 651 if (encoder_inits_ == 1) { | 650 if (encoder_inits_ == 1) { |
| 652 // First time initialization. Frame size is known. | 651 // First time initialization. Frame size is known. |
| 653 // |expected_bitrate| is affected by bandwidth estimation before the | 652 // |expected_bitrate| is affected by bandwidth estimation before the |
| 654 // first frame arrives to the encoder. | 653 // first frame arrives to the encoder. |
| 655 uint32_t expected_bitrate = last_set_bitrate_kbps_ > 0 | 654 uint32_t expected_bitrate = |
| 656 ? last_set_bitrate_kbps_ | 655 last_set_bitrate_ > 0 ? last_set_bitrate_ : kInitialBitrateKbps; |
| 657 : kInitialBitrateKbps; | |
| 658 EXPECT_EQ(expected_bitrate, config->startBitrate) | 656 EXPECT_EQ(expected_bitrate, config->startBitrate) |
| 659 << "Encoder not initialized at expected bitrate."; | 657 << "Encoder not initialized at expected bitrate."; |
| 660 EXPECT_EQ(kDefaultWidth, config->width); | 658 EXPECT_EQ(kDefaultWidth, config->width); |
| 661 EXPECT_EQ(kDefaultHeight, config->height); | 659 EXPECT_EQ(kDefaultHeight, config->height); |
| 662 } else if (encoder_inits_ == 2) { | 660 } else if (encoder_inits_ == 2) { |
| 663 EXPECT_EQ(2 * kDefaultWidth, config->width); | 661 EXPECT_EQ(2 * kDefaultWidth, config->width); |
| 664 EXPECT_EQ(2 * kDefaultHeight, config->height); | 662 EXPECT_EQ(2 * kDefaultHeight, config->height); |
| 665 EXPECT_GE(last_set_bitrate_kbps_, kReconfigureThresholdKbps); | 663 EXPECT_GE(last_set_bitrate_, kReconfigureThresholdKbps); |
| 666 EXPECT_NEAR(config->startBitrate, last_set_bitrate_kbps_, | 664 EXPECT_NEAR(config->startBitrate, |
| 665 last_set_bitrate_, |
| 667 kPermittedReconfiguredBitrateDiffKbps) | 666 kPermittedReconfiguredBitrateDiffKbps) |
| 668 << "Encoder reconfigured with bitrate too far away from last set."; | 667 << "Encoder reconfigured with bitrate too far away from last set."; |
| 669 observation_complete_.Set(); | 668 observation_complete_.Set(); |
| 670 } | 669 } |
| 671 return FakeEncoder::InitEncode(config, number_of_cores, max_payload_size); | 670 return FakeEncoder::InitEncode(config, number_of_cores, max_payload_size); |
| 672 } | 671 } |
| 673 | 672 |
| 674 int32_t SetRateAllocation(const BitrateAllocation& rate_allocation, | 673 int32_t SetRates(uint32_t new_target_bitrate_kbps, |
| 675 uint32_t framerate) override { | 674 uint32_t framerate) override { |
| 676 last_set_bitrate_kbps_ = rate_allocation.get_sum_kbps(); | 675 last_set_bitrate_ = new_target_bitrate_kbps; |
| 677 if (encoder_inits_ == 1 && | 676 if (encoder_inits_ == 1 && |
| 678 rate_allocation.get_sum_kbps() > kReconfigureThresholdKbps) { | 677 new_target_bitrate_kbps > kReconfigureThresholdKbps) { |
| 679 time_to_reconfigure_.Set(); | 678 time_to_reconfigure_.Set(); |
| 680 } | 679 } |
| 681 return FakeEncoder::SetRateAllocation(rate_allocation, framerate); | 680 return FakeEncoder::SetRates(new_target_bitrate_kbps, framerate); |
| 682 } | 681 } |
| 683 | 682 |
| 684 Call::Config GetSenderCallConfig() override { | 683 Call::Config GetSenderCallConfig() override { |
| 685 Call::Config config = EndToEndTest::GetSenderCallConfig(); | 684 Call::Config config = EndToEndTest::GetSenderCallConfig(); |
| 686 config.event_log = &event_log_; | 685 config.event_log = &event_log_; |
| 687 config.bitrate_config.start_bitrate_bps = kInitialBitrateKbps * 1000; | 686 config.bitrate_config.start_bitrate_bps = kInitialBitrateKbps * 1000; |
| 688 return config; | 687 return config; |
| 689 } | 688 } |
| 690 | 689 |
| 691 void ModifyVideoConfigs( | 690 void ModifyVideoConfigs( |
| (...skipping 25 matching lines...) Expand all Loading... |
| 717 frame_generator_->ChangeResolution(kDefaultWidth * 2, kDefaultHeight * 2); | 716 frame_generator_->ChangeResolution(kDefaultWidth * 2, kDefaultHeight * 2); |
| 718 send_stream_->ReconfigureVideoEncoder(encoder_config_.Copy()); | 717 send_stream_->ReconfigureVideoEncoder(encoder_config_.Copy()); |
| 719 EXPECT_TRUE(Wait()) | 718 EXPECT_TRUE(Wait()) |
| 720 << "Timed out while waiting for a couple of high bitrate estimates " | 719 << "Timed out while waiting for a couple of high bitrate estimates " |
| 721 "after reconfiguring the send stream."; | 720 "after reconfiguring the send stream."; |
| 722 } | 721 } |
| 723 | 722 |
| 724 private: | 723 private: |
| 725 rtc::Event time_to_reconfigure_; | 724 rtc::Event time_to_reconfigure_; |
| 726 int encoder_inits_; | 725 int encoder_inits_; |
| 727 uint32_t last_set_bitrate_kbps_; | 726 uint32_t last_set_bitrate_; |
| 728 VideoSendStream* send_stream_; | 727 VideoSendStream* send_stream_; |
| 729 test::FrameGeneratorCapturer* frame_generator_; | 728 test::FrameGeneratorCapturer* frame_generator_; |
| 730 VideoEncoderConfig encoder_config_; | 729 VideoEncoderConfig encoder_config_; |
| 731 } test; | 730 } test; |
| 732 | 731 |
| 733 RunBaseTest(&test); | 732 RunBaseTest(&test); |
| 734 } | 733 } |
| 735 | 734 |
| 736 } // namespace webrtc | 735 } // namespace webrtc |
| OLD | NEW |