1 /* 2 * Copyright (c) 2016 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 #ifndef MODULES_CONGESTION_CONTROLLER_PROBE_BITRATE_ESTIMATOR_H_ 12 #define MODULES_CONGESTION_CONTROLLER_PROBE_BITRATE_ESTIMATOR_H_ 13 14 #include <map> 15 #include <limits> 16 17 #include "modules/rtp_rtcp/include/rtp_rtcp_defines.h" 18 19 namespace webrtc { 20 class RtcEventLog; 21 22 class ProbeBitrateEstimator { 23 public: 24 explicit ProbeBitrateEstimator(RtcEventLog* event_log); 25 ~ProbeBitrateEstimator(); 26 27 // Should be called for every probe packet we receive feedback about. 28 // Returns the estimated bitrate if the probe completes a valid cluster. 29 int HandleProbeAndEstimateBitrate(const PacketFeedback& packet_feedback); 30 31 rtc::Optional<int> FetchAndResetLastEstimatedBitrateBps(); 32 33 private: 34 struct AggregatedCluster { 35 int num_probes = 0; 36 int64_t first_send_ms = std::numeric_limits<int64_t>::max(); 37 int64_t last_send_ms = 0; 38 int64_t first_receive_ms = std::numeric_limits<int64_t>::max(); 39 int64_t last_receive_ms = 0; 40 int size_last_send = 0; 41 int size_first_receive = 0; 42 int size_total = 0; 43 }; 44 45 // Erases old cluster data that was seen before |timestamp_ms|. 46 void EraseOldClusters(int64_t timestamp_ms); 47 48 std::map<int, AggregatedCluster> clusters_; 49 RtcEventLog* const event_log_; 50 rtc::Optional<int> estimated_bitrate_bps_; 51 }; 52 53 } // namespace webrtc 54 55 #endif // MODULES_CONGESTION_CONTROLLER_PROBE_BITRATE_ESTIMATOR_H_ 56