| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #ifndef WEBRTC_MODULES_CONGESTION_CONTROLLER_PROBE_BITRATE_ESTIMATOR_H_ | 11 #ifndef WEBRTC_MODULES_CONGESTION_CONTROLLER_PROBE_BITRATE_ESTIMATOR_H_ |
| 12 #define WEBRTC_MODULES_CONGESTION_CONTROLLER_PROBE_BITRATE_ESTIMATOR_H_ | 12 #define WEBRTC_MODULES_CONGESTION_CONTROLLER_PROBE_BITRATE_ESTIMATOR_H_ |
| 13 | 13 |
| 14 #include <map> | 14 #include <map> |
| 15 #include <limits> | 15 #include <limits> |
| 16 | 16 |
| 17 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" | 17 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
| 18 | 18 |
| 19 namespace webrtc { | 19 namespace webrtc { |
| 20 | 20 |
| 21 struct ProbingResult { | |
| 22 static constexpr int kNoEstimate = -1; | |
| 23 | |
| 24 ProbingResult(); | |
| 25 ProbingResult(int bps, int64_t timestamp); | |
| 26 bool valid() const; | |
| 27 | |
| 28 int bps; | |
| 29 int64_t timestamp; | |
| 30 }; | |
| 31 | |
| 32 class ProbeBitrateEstimator { | 21 class ProbeBitrateEstimator { |
| 33 public: | 22 public: |
| 34 ProbeBitrateEstimator(); | 23 ProbeBitrateEstimator(); |
| 35 | 24 |
| 36 // Should be called for every packet we receive feedback about. If the | 25 // Should be called for every probe packet we receive feedback about. |
| 37 // packet was used for probing it will validate/calculate the resulting | 26 // Returns the estimated bitrate if the probe completes a valid cluster. |
| 38 // bitrate and return the result. | 27 int HandleProbeAndEstimateBitrate(const PacketInfo& packet_info); |
| 39 ProbingResult PacketFeedback(const PacketInfo& packet_info); | |
| 40 | 28 |
| 41 private: | 29 private: |
| 42 struct AggregatedCluster { | 30 struct AggregatedCluster { |
| 43 int num_probes = 0; | 31 int num_probes = 0; |
| 44 int64_t first_send_ms = std::numeric_limits<int64_t>::max(); | 32 int64_t first_send_ms = std::numeric_limits<int64_t>::max(); |
| 45 int64_t last_send_ms = 0; | 33 int64_t last_send_ms = 0; |
| 46 int64_t first_receive_ms = std::numeric_limits<int64_t>::max(); | 34 int64_t first_receive_ms = std::numeric_limits<int64_t>::max(); |
| 47 int64_t last_receive_ms = 0; | 35 int64_t last_receive_ms = 0; |
| 48 size_t size = 0; | 36 size_t size = 0; |
| 49 }; | 37 }; |
| 50 | 38 |
| 39 // Erases old cluster data that was seen before |timestamp_ms|. |
| 40 void EraseOldClusters(int64_t timestamp_ms); |
| 41 |
| 51 std::map<int, AggregatedCluster> clusters_; | 42 std::map<int, AggregatedCluster> clusters_; |
| 52 int last_valid_cluster_id_; | |
| 53 }; | 43 }; |
| 54 | 44 |
| 55 } // namespace webrtc | 45 } // namespace webrtc |
| 56 | 46 |
| 57 #endif // WEBRTC_MODULES_CONGESTION_CONTROLLER_PROBE_BITRATE_ESTIMATOR_H_ | 47 #endif // WEBRTC_MODULES_CONGESTION_CONTROLLER_PROBE_BITRATE_ESTIMATOR_H_ |
| OLD | NEW |