Tweaked version of BBR for WebRTC.

webrtc/modules/remote_bitrate_estimator/test/estimators/bbr.h

 /*
  *  Copyright (c) 2017 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.
  *
  */
@@ skipping 12 matching lines @@
 #include "webrtc/rtc_base/random.h"
 
 namespace webrtc {
 namespace testing {
 namespace bwe {
 class MaxBandwidthFilter;
 class MinRttFilter;
 class CongestionWindow;
 class BbrBweSender : public BweSender {
  public:
-  explicit BbrBweSender(Clock* clock);
+  explicit BbrBweSender(BitrateObserver* observer, Clock* clock);
   virtual ~BbrBweSender();
   enum Mode {
     // Startup phase.
     STARTUP,
     // Queue draining phase, which where created during startup.
     DRAIN,
     // Cruising, probing new bandwidth.
     PROBE_BW,
     // Temporarily limiting congestion window size in order to measure
     // minimum RTT.
@@ skipping 62 matching lines @@
     uint64_t round;
   };
   void OnPacketsSent(const Packets& packets) override;
   int GetFeedbackIntervalMs() const override;
   void GiveFeedback(const FeedbackPacket& feedback) override;
   int64_t TimeUntilNextProcess() override;
   void Process() override;
 
  private:
   void EnterStartup();
-  bool UpdateBandwidthAndMinRtt(int64_t now_ms,
-                                const std::vector<uint64_t>& feedback_vector,
+  void UpdateBandwidthAndMinRtt(int64_t now_ms,
+                                const std::vector<uint16_t>& feedback_vector,
                                 int64_t bytes_acked);
   void TryExitingStartup();
   void TryExitingDrain(int64_t now_ms);
   void EnterProbeBw(int64_t now_ms);
   void TryUpdatingCyclePhase(int64_t now_ms);
   void TryEnteringProbeRtt(int64_t now_ms);
   void TryExitingProbeRtt(int64_t now_ms, int64_t round);
   void TryEnteringRecovery(bool new_round_started);
   void TryExitingRecovery(bool new_round_started);
   size_t TargetCongestionWindow(float gain);
@@ skipping 10 matching lines @@
   // phase. Motivation of having a seperate bucket for high gain phase is to
   // achieve quicker ramp up. Slight overestimations may happen due to window
   // not being as large as usual.
   void AddSampleForHighGain();
 
   // Declares lost packets as acked. Implements simple logic by looking at the
   // gap between sequence numbers. If there is a gap between sequence numbers we
   // declare those packets as lost immediately.
   void HandleLoss(uint64_t last_acked_packet, uint64_t recently_acked_packet);
   void AddToPastRtts(int64_t rtt_sample_ms);
+  BitrateObserver* observer_;
   Clock* const clock_;
   Mode mode_;
   std::unique_ptr<MaxBandwidthFilter> max_bandwidth_filter_;
   std::unique_ptr<MinRttFilter> min_rtt_filter_;
   std::unique_ptr<CongestionWindow> congestion_window_;
   std::unique_ptr<Random> rand_;
   uint64_t round_count_;
   uint64_t round_trip_end_;
   float pacing_gain_;
   float congestion_window_gain_;
@@ skipping 64 matching lines @@
   std::list<AverageRtt> past_rtts_;
 };
 
 class BbrBweReceiver : public BweReceiver {
  public:
   explicit BbrBweReceiver(int flow_id);
   virtual ~BbrBweReceiver();
   void ReceivePacket(int64_t arrival_time_ms,
                      const MediaPacket& media_packet) override;
   FeedbackPacket* GetFeedback(int64_t now_ms) override;
-
  private:
   SimulatedClock clock_;
-  std::vector<uint64_t> packet_feedbacks_;
+  std::vector<uint16_t> packet_feedbacks_;
+  int64_t last_feedback_ms_;
 };
 }  // namespace bwe
 }  // namespace testing
 }  // namespace webrtc
 
 #endif  // WEBRTC_MODULES_REMOTE_BITRATE_ESTIMATOR_TEST_ESTIMATORS_BBR_H_