OLD | NEW |
(Empty) | |
| 1 /* |
| 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
| 3 * |
| 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 |
| 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ |
| 10 |
| 11 #include "webrtc/modules/remote_bitrate_estimator/transport_feedback_adapter.h" |
| 12 |
| 13 #include <algorithm> |
| 14 #include <limits> |
| 15 |
| 16 #include "webrtc/base/checks.h" |
| 17 #include "webrtc/base/logging.h" |
| 18 #include "webrtc/modules/remote_bitrate_estimator/include/remote_bitrate_estimat
or.h" |
| 19 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h" |
| 20 #include "webrtc/modules/utility/include/process_thread.h" |
| 21 |
| 22 namespace webrtc { |
| 23 |
| 24 const int64_t kNoTimestamp = -1; |
| 25 const int64_t kSendTimeHistoryWindowMs = 10000; |
| 26 const int64_t kBaseTimestampScaleFactor = |
| 27 rtcp::TransportFeedback::kDeltaScaleFactor * (1 << 8); |
| 28 const int64_t kBaseTimestampRangeSizeUs = kBaseTimestampScaleFactor * (1 << 24); |
| 29 |
| 30 class PacketInfoComparator { |
| 31 public: |
| 32 inline bool operator()(const PacketInfo& lhs, const PacketInfo& rhs) { |
| 33 if (lhs.arrival_time_ms != rhs.arrival_time_ms) |
| 34 return lhs.arrival_time_ms < rhs.arrival_time_ms; |
| 35 if (lhs.send_time_ms != rhs.send_time_ms) |
| 36 return lhs.send_time_ms < rhs.send_time_ms; |
| 37 return lhs.sequence_number < rhs.sequence_number; |
| 38 } |
| 39 }; |
| 40 |
| 41 TransportFeedbackAdapter::TransportFeedbackAdapter( |
| 42 Clock* clock) |
| 43 : send_time_history_(clock, kSendTimeHistoryWindowMs), |
| 44 clock_(clock), |
| 45 current_offset_ms_(kNoTimestamp), |
| 46 last_timestamp_us_(kNoTimestamp) {} |
| 47 |
| 48 TransportFeedbackAdapter::~TransportFeedbackAdapter() { |
| 49 } |
| 50 |
| 51 void TransportFeedbackAdapter::SetBitrateEstimator( |
| 52 RemoteBitrateEstimator* rbe) { |
| 53 if (bitrate_estimator_.get() != rbe) { |
| 54 bitrate_estimator_.reset(rbe); |
| 55 } |
| 56 } |
| 57 |
| 58 void TransportFeedbackAdapter::AddPacket(uint16_t sequence_number, |
| 59 size_t length, |
| 60 int probe_cluster_id) { |
| 61 rtc::CritScope cs(&lock_); |
| 62 send_time_history_.AddAndRemoveOld(sequence_number, length, probe_cluster_id); |
| 63 } |
| 64 |
| 65 void TransportFeedbackAdapter::OnSentPacket(uint16_t sequence_number, |
| 66 int64_t send_time_ms) { |
| 67 rtc::CritScope cs(&lock_); |
| 68 send_time_history_.OnSentPacket(sequence_number, send_time_ms); |
| 69 } |
| 70 |
| 71 std::vector<PacketInfo> TransportFeedbackAdapter::GetPacketFeedbackVector( |
| 72 const rtcp::TransportFeedback& feedback) { |
| 73 int64_t timestamp_us = feedback.GetBaseTimeUs(); |
| 74 // Add timestamp deltas to a local time base selected on first packet arrival. |
| 75 // This won't be the true time base, but makes it easier to manually inspect |
| 76 // time stamps. |
| 77 if (last_timestamp_us_ == kNoTimestamp) { |
| 78 current_offset_ms_ = clock_->TimeInMilliseconds(); |
| 79 } else { |
| 80 int64_t delta = timestamp_us - last_timestamp_us_; |
| 81 |
| 82 // Detect and compensate for wrap-arounds in base time. |
| 83 if (std::abs(delta - kBaseTimestampRangeSizeUs) < std::abs(delta)) { |
| 84 delta -= kBaseTimestampRangeSizeUs; // Wrap backwards. |
| 85 } else if (std::abs(delta + kBaseTimestampRangeSizeUs) < std::abs(delta)) { |
| 86 delta += kBaseTimestampRangeSizeUs; // Wrap forwards. |
| 87 } |
| 88 |
| 89 current_offset_ms_ += delta / 1000; |
| 90 } |
| 91 last_timestamp_us_ = timestamp_us; |
| 92 |
| 93 uint16_t sequence_number = feedback.GetBaseSequence(); |
| 94 std::vector<int64_t> delta_vec = feedback.GetReceiveDeltasUs(); |
| 95 auto delta_it = delta_vec.begin(); |
| 96 std::vector<PacketInfo> packet_feedback_vector; |
| 97 packet_feedback_vector.reserve(delta_vec.size()); |
| 98 |
| 99 { |
| 100 rtc::CritScope cs(&lock_); |
| 101 size_t failed_lookups = 0; |
| 102 int64_t offset_us = 0; |
| 103 for (auto symbol : feedback.GetStatusVector()) { |
| 104 if (symbol != rtcp::TransportFeedback::StatusSymbol::kNotReceived) { |
| 105 RTC_DCHECK(delta_it != delta_vec.end()); |
| 106 offset_us += *(delta_it++); |
| 107 int64_t timestamp_ms = current_offset_ms_ + (offset_us / 1000); |
| 108 PacketInfo info(timestamp_ms, sequence_number); |
| 109 if (send_time_history_.GetInfo(&info, true) && info.send_time_ms >= 0) { |
| 110 packet_feedback_vector.push_back(info); |
| 111 } else { |
| 112 ++failed_lookups; |
| 113 } |
| 114 } |
| 115 ++sequence_number; |
| 116 } |
| 117 std::sort(packet_feedback_vector.begin(), packet_feedback_vector.end(), |
| 118 PacketInfoComparator()); |
| 119 RTC_DCHECK(delta_it == delta_vec.end()); |
| 120 if (failed_lookups > 0) { |
| 121 LOG(LS_WARNING) << "Failed to lookup send time for " << failed_lookups |
| 122 << " packet" << (failed_lookups > 1 ? "s" : "") |
| 123 << ". Send time history too small?"; |
| 124 } |
| 125 } |
| 126 return packet_feedback_vector; |
| 127 } |
| 128 |
| 129 void TransportFeedbackAdapter::OnTransportFeedback( |
| 130 const rtcp::TransportFeedback& feedback) { |
| 131 last_packet_feedback_vector_ = GetPacketFeedbackVector(feedback); |
| 132 if (bitrate_estimator_.get()) |
| 133 bitrate_estimator_->IncomingPacketFeedbackVector( |
| 134 last_packet_feedback_vector_); |
| 135 } |
| 136 |
| 137 std::vector<PacketInfo> TransportFeedbackAdapter::GetTransportFeedbackVector() |
| 138 const { |
| 139 return last_packet_feedback_vector_; |
| 140 } |
| 141 |
| 142 void TransportFeedbackAdapter::OnRttUpdate(int64_t avg_rtt_ms, |
| 143 int64_t max_rtt_ms) { |
| 144 RTC_DCHECK(bitrate_estimator_.get() != nullptr); |
| 145 bitrate_estimator_->OnRttUpdate(avg_rtt_ms, max_rtt_ms); |
| 146 } |
| 147 |
| 148 } // namespace webrtc |
OLD | NEW |