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