Tweaked version of BBR for WebRTC.

webrtc/modules/remote_bitrate_estimator/test/plot_dynamics.py

 #!/usr/bin/env python
 # Copyright (c) 2015 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.
 
 # This script is used to plot simulation dynamics. The expected format is
@@ skipping 60 matching lines @@
   def AddSample(self, var_name, ssrc, alg_name, time, value):
     for s in self.subplots:
       s.AddSample(var_name, ssrc, alg_name, time, value)
 
   def PlotFigure(self, fig):
     n = len(self.subplots)
     for i in range(n):
       axis = fig.add_subplot(n, 1, i+1)
       self.subplots[i].PlotSubplot(axis)
 
-
 class Subplot(object):
   def __init__(self, var_names, xlabel, ylabel):
     self.xlabel = xlabel
     self.ylabel = ylabel
     self.var_names = var_names
     self.samples = dict()
 
   def AddSample(self, var_name, ssrc, alg_name, time, value):
     if var_name not in self.var_names:
       return
@@ skipping 12 matching lines @@
     axis.set_ylabel(self.ylabel)
 
     count = 0
     for alg_name in self.samples.keys():
       for ssrc in self.samples[alg_name].keys():
         for var_name in self.samples[alg_name][ssrc].keys():
           x = [sample[0] for sample in self.samples[alg_name][ssrc][var_name]]
           y = [sample[1] for sample in self.samples[alg_name][ssrc][var_name]]
           x = numpy.array(x)
           y = numpy.array(y)
-
           ssrc_count = len(self.samples[alg_name].keys())
           l = GenerateLabel(var_name, ssrc, ssrc_count, alg_name)
-          plt.plot(x, y, label=l, linewidth=2.0)
+          if l == 'MaxThroughput_':
+            plt.plot(x, y, label=l, linewidth=4.0)
+          else:
+            plt.plot(x, y, label=l, linewidth=2.0)
           count += 1
 
     plt.grid(True)
     if count > 1:
       plt.legend(loc='best')
 
 
 def main():
   receiver = Figure("PacketReceiver")
   receiver.AddSubplot(['Throughput_kbps', 'MaxThroughput_', 'Capacity_kbps',
@@ skipping 13 matching lines @@
 
   trendline_state = Figure("TrendlineState")
   trendline_state.AddSubplot(["accumulated_delay_ms", "smoothed_delay_ms"],
                              "Time (s)", "Delay (ms)")
   trendline_state.AddSubplot(["trendline_slope"], "Time (s)", "Slope")
 
   target_bitrate = Figure("TargetBitrate")
   target_bitrate.AddSubplot(['target_bitrate_bps', 'acknowledged_bitrate'],
                             "Time (s)", "Bitrate (bps)")
 
+  min_rtt_state = Figure("MinRttState")
+  min_rtt_state.AddSubplot(['MinRtt'], "Time (s)", "Time (ms)")
+
   # Select which figures to plot here.
-  figures = [receiver, detector_state, trendline_state, target_bitrate]
+  figures = [receiver, detector_state, trendline_state, target_bitrate,
+    min_rtt_state]
 
   # Add samples to the figures.
   for line in sys.stdin:
     if line.startswith("[ RUN      ]"):
       test_name = re.search(r'\.(\w+)', line).group(1)
     if line.startswith("PLOT"):
       try:
         (var_name, ssrc, alg_name, time, value) = ParsePlotLine(line)
         for f in figures:
           # The sample will be ignored bv the figures that don't need it.
           f.AddSample(var_name, ssrc, alg_name, time, value)
       except ParsePlotLineException as e:
         print e.reason
         print e.line
 
   # Plot figures.
   for f in figures:
     fig = plt.figure(f.name)
     f.PlotFigure(fig)
     if SAVE_FIGURE:
       fig.savefig(test_name + f.name + ".png")
   plt.show()
 
 if __name__ == '__main__':
   main()