Index: webrtc/modules/congestion_controller/acknowledged_bitrate_estimator.cc |
diff --git a/webrtc/modules/congestion_controller/acknowledged_bitrate_estimator.cc b/webrtc/modules/congestion_controller/acknowledged_bitrate_estimator.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..9994f53dc49a8bd46c7257e2ec3c58535a71e881 |
--- /dev/null |
+++ b/webrtc/modules/congestion_controller/acknowledged_bitrate_estimator.cc |
@@ -0,0 +1,82 @@ |
+/* |
+ * Copyright (c) 2017 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/congestion_controller/acknowledged_bitrate_estimator.h" |
+ |
+#include <utility> |
+ |
+#include "webrtc/base/ptr_util.h" |
+#include "webrtc/base/timeutils.h" |
+#include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
+ |
+namespace webrtc { |
+ |
+namespace { |
+bool IsInSendTimeHistory(const PacketFeedback& packet) { |
+ return packet.send_time_ms >= 0; |
+} |
+} // namespace |
+ |
+std::unique_ptr<BitrateEstimator> BitrateEstimatorCreator::Create() const { |
+ return rtc::MakeUnique<BitrateEstimator>(); |
+} |
+ |
+AcknowledgedBitrateEstimator::AcknowledgedBitrateEstimator( |
+ std::unique_ptr<BitrateEstimatorCreator> bitrate_estimator_creator) |
+ : was_in_alr_(false), |
+ bitrate_estimator_creator_( |
+ bitrate_estimator_creator |
+ ? std::move(bitrate_estimator_creator) |
+ : rtc::MakeUnique<BitrateEstimatorCreator>()), |
+ bitrate_estimator_(bitrate_estimator_creator_->Create()) {} |
+ |
+void AcknowledgedBitrateEstimator::IncomingPacketFeedbackVector( |
+ const std::vector<PacketFeedback>& packet_feedback_vector, |
+ bool currently_in_alr) { |
+ RTC_DCHECK(std::is_sorted(packet_feedback_vector.begin(), |
+ packet_feedback_vector.end(), |
+ PacketFeedbackComparator())); |
+ MaybeResetBitrateEstimator(currently_in_alr); |
+ for (const auto& packet : packet_feedback_vector) { |
+ if (IsInSendTimeHistory(packet) && !SentBeforeAlrEnded(packet)) |
+ bitrate_estimator_->Update(packet.arrival_time_ms, packet.payload_size); |
+ } |
+} |
+ |
+rtc::Optional<uint32_t> AcknowledgedBitrateEstimator::bitrate_bps() const { |
+ return bitrate_estimator_->bitrate_bps(); |
+} |
+ |
+bool AcknowledgedBitrateEstimator::SentBeforeAlrEnded( |
+ const PacketFeedback& packet) { |
+ if (alr_ended_time_ms_) { |
+ if (*alr_ended_time_ms_ > packet.send_time_ms) { |
+ return true; |
+ } else { |
+ alr_ended_time_ms_.reset(); |
philipel
2017/06/14 09:34:05
What happens if feedback messages are reordered? W
tschumi
2017/06/14 10:57:46
Right, this reset is kind of senseless.
|
+ } |
+ } |
+ return false; |
+} |
+ |
+bool AcknowledgedBitrateEstimator::AlrEnded(bool currently_in_alr) const { |
+ return was_in_alr_ && !currently_in_alr; |
+} |
+ |
+void AcknowledgedBitrateEstimator::MaybeResetBitrateEstimator( |
+ bool currently_in_alr) { |
+ if (AlrEnded(currently_in_alr)) { |
+ bitrate_estimator_ = bitrate_estimator_creator_->Create(); |
+ alr_ended_time_ms_ = rtc::Optional<int64_t>(rtc::TimeMillis()); |
+ } |
+ was_in_alr_ = currently_in_alr; |
+} |
+ |
+} // namespace webrtc |