Index: webrtc/modules/remote_bitrate_estimator/transport_feedback_adapter.cc |
diff --git a/webrtc/modules/remote_bitrate_estimator/transport_feedback_adapter.cc b/webrtc/modules/remote_bitrate_estimator/transport_feedback_adapter.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..66ef7f04953524b5fb44531f644c91f48b853e22 |
--- /dev/null |
+++ b/webrtc/modules/remote_bitrate_estimator/transport_feedback_adapter.cc |
@@ -0,0 +1,148 @@ |
+/* |
+ * Copyright (c) 2015 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/remote_bitrate_estimator/transport_feedback_adapter.h" |
+ |
+#include <algorithm> |
+#include <limits> |
+ |
+#include "webrtc/base/checks.h" |
+#include "webrtc/base/logging.h" |
+#include "webrtc/modules/remote_bitrate_estimator/include/remote_bitrate_estimator.h" |
+#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h" |
+#include "webrtc/modules/utility/include/process_thread.h" |
+ |
+namespace webrtc { |
+ |
+const int64_t kNoTimestamp = -1; |
+const int64_t kSendTimeHistoryWindowMs = 10000; |
+const int64_t kBaseTimestampScaleFactor = |
+ rtcp::TransportFeedback::kDeltaScaleFactor * (1 << 8); |
+const int64_t kBaseTimestampRangeSizeUs = kBaseTimestampScaleFactor * (1 << 24); |
+ |
+class PacketInfoComparator { |
+ public: |
+ inline bool operator()(const PacketInfo& lhs, const PacketInfo& rhs) { |
+ if (lhs.arrival_time_ms != rhs.arrival_time_ms) |
+ return lhs.arrival_time_ms < rhs.arrival_time_ms; |
+ if (lhs.send_time_ms != rhs.send_time_ms) |
+ return lhs.send_time_ms < rhs.send_time_ms; |
+ return lhs.sequence_number < rhs.sequence_number; |
+ } |
+}; |
+ |
+TransportFeedbackAdapter::TransportFeedbackAdapter( |
+ Clock* clock) |
+ : send_time_history_(clock, kSendTimeHistoryWindowMs), |
+ clock_(clock), |
+ current_offset_ms_(kNoTimestamp), |
+ last_timestamp_us_(kNoTimestamp) {} |
+ |
+TransportFeedbackAdapter::~TransportFeedbackAdapter() { |
+} |
+ |
+void TransportFeedbackAdapter::SetBitrateEstimator( |
+ RemoteBitrateEstimator* rbe) { |
+ if (bitrate_estimator_.get() != rbe) { |
+ bitrate_estimator_.reset(rbe); |
+ } |
+} |
+ |
+void TransportFeedbackAdapter::AddPacket(uint16_t sequence_number, |
+ size_t length, |
+ int probe_cluster_id) { |
+ rtc::CritScope cs(&lock_); |
+ send_time_history_.AddAndRemoveOld(sequence_number, length, probe_cluster_id); |
+} |
+ |
+void TransportFeedbackAdapter::OnSentPacket(uint16_t sequence_number, |
+ int64_t send_time_ms) { |
+ rtc::CritScope cs(&lock_); |
+ send_time_history_.OnSentPacket(sequence_number, send_time_ms); |
+} |
+ |
+std::vector<PacketInfo> TransportFeedbackAdapter::GetPacketFeedbackVector( |
+ const rtcp::TransportFeedback& feedback) { |
+ int64_t timestamp_us = feedback.GetBaseTimeUs(); |
+ // Add timestamp deltas to a local time base selected on first packet arrival. |
+ // This won't be the true time base, but makes it easier to manually inspect |
+ // time stamps. |
+ if (last_timestamp_us_ == kNoTimestamp) { |
+ current_offset_ms_ = clock_->TimeInMilliseconds(); |
+ } else { |
+ int64_t delta = timestamp_us - last_timestamp_us_; |
+ |
+ // Detect and compensate for wrap-arounds in base time. |
+ if (std::abs(delta - kBaseTimestampRangeSizeUs) < std::abs(delta)) { |
+ delta -= kBaseTimestampRangeSizeUs; // Wrap backwards. |
+ } else if (std::abs(delta + kBaseTimestampRangeSizeUs) < std::abs(delta)) { |
+ delta += kBaseTimestampRangeSizeUs; // Wrap forwards. |
+ } |
+ |
+ current_offset_ms_ += delta / 1000; |
+ } |
+ last_timestamp_us_ = timestamp_us; |
+ |
+ uint16_t sequence_number = feedback.GetBaseSequence(); |
+ std::vector<int64_t> delta_vec = feedback.GetReceiveDeltasUs(); |
+ auto delta_it = delta_vec.begin(); |
+ std::vector<PacketInfo> packet_feedback_vector; |
+ packet_feedback_vector.reserve(delta_vec.size()); |
+ |
+ { |
+ rtc::CritScope cs(&lock_); |
+ size_t failed_lookups = 0; |
+ int64_t offset_us = 0; |
+ for (auto symbol : feedback.GetStatusVector()) { |
+ if (symbol != rtcp::TransportFeedback::StatusSymbol::kNotReceived) { |
+ RTC_DCHECK(delta_it != delta_vec.end()); |
+ offset_us += *(delta_it++); |
+ int64_t timestamp_ms = current_offset_ms_ + (offset_us / 1000); |
+ PacketInfo info(timestamp_ms, sequence_number); |
+ if (send_time_history_.GetInfo(&info, true) && info.send_time_ms >= 0) { |
+ packet_feedback_vector.push_back(info); |
+ } else { |
+ ++failed_lookups; |
+ } |
+ } |
+ ++sequence_number; |
+ } |
+ std::sort(packet_feedback_vector.begin(), packet_feedback_vector.end(), |
+ PacketInfoComparator()); |
+ RTC_DCHECK(delta_it == delta_vec.end()); |
+ if (failed_lookups > 0) { |
+ LOG(LS_WARNING) << "Failed to lookup send time for " << failed_lookups |
+ << " packet" << (failed_lookups > 1 ? "s" : "") |
+ << ". Send time history too small?"; |
+ } |
+ } |
+ return packet_feedback_vector; |
+} |
+ |
+void TransportFeedbackAdapter::OnTransportFeedback( |
+ const rtcp::TransportFeedback& feedback) { |
+ last_packet_feedback_vector_ = GetPacketFeedbackVector(feedback); |
+ if (bitrate_estimator_.get()) |
+ bitrate_estimator_->IncomingPacketFeedbackVector( |
+ last_packet_feedback_vector_); |
+} |
+ |
+std::vector<PacketInfo> TransportFeedbackAdapter::GetTransportFeedbackVector() |
+ const { |
+ return last_packet_feedback_vector_; |
+} |
+ |
+void TransportFeedbackAdapter::OnRttUpdate(int64_t avg_rtt_ms, |
+ int64_t max_rtt_ms) { |
+ RTC_DCHECK(bitrate_estimator_.get() != nullptr); |
+ bitrate_estimator_->OnRttUpdate(avg_rtt_ms, max_rtt_ms); |
+} |
+ |
+} // namespace webrtc |