Update plot_webrtc_test_logs.py:

webrtc/modules/video_coding/codecs/test/plot_webrtc_test_logs.py

 #  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.
 
 """Plots statistics from WebRTC integration test logs.
 
 Usage: $ python plot_webrtc_test_logs.py filename.txt
 """
 
 import numpy
 import sys
 import re
 
 import matplotlib.pyplot as plt
 
 # Log events.
 EVENT_START = 'RUN      ] CodecSettings/PlotVideoProcessorIntegrationTest.'
 EVENT_END = 'OK ] CodecSettings/PlotVideoProcessorIntegrationTest.'
 
 # Metrics to plot, tuple: (name to parse in file, label to use when plotting).
-BITRATE = ('Target Bitrate', 'bitrate (kbps)')
+BITRATE = ('Target Bitrate', 'target bitrate (kbps)')
 WIDTH = ('Width', 'width')
 HEIGHT = ('Height', 'height')
 FILENAME = ('Filename', 'clip')
 CODEC_TYPE = ('Codec type', 'Codec')
 ENCODER_IMPLEMENTATION_NAME = ('Encoder implementation name', 'enc name')
 DECODER_IMPLEMENTATION_NAME = ('Decoder implementation name', 'dec name')
-NUM_FRAMES = ('Total # of frames', 'num frames')
+CODEC_IMPLEMENTATION_NAME = ('Codec implementation name', 'codec name')
 CORES = ('#CPU cores used', 'CPU cores used')
 DENOISING = ('Denoising', 'denoising')
 RESILIENCE = ('Resilience', 'resilience')
 ERROR_CONCEALMENT = ('Error concealment', 'error concealment')
 QP = ('Average QP', 'avg QP')
 PSNR = ('PSNR avg', 'PSNR (dB)')
 SSIM = ('SSIM avg', 'SSIM')
 ENC_BITRATE = ('Encoding bitrate', 'encoded bitrate (kbps)')
 FRAMERATE = ('Frame rate', 'fps')
+NUM_FRAMES = ('Number of processed frames', 'num frames')
 NUM_DROPPED_FRAMES = ('Number of dropped frames', 'num dropped frames')
 NUM_FRAMES_TO_TARGET = ('Number of frames to approach target rate',
                         'frames to reach target rate')
 ENCODE_TIME = ('Encoding time', 'encode time (us)')
 ENCODE_TIME_AVG = ('Encoding time', 'encode time (us) avg')
 DECODE_TIME = ('Decoding time', 'decode time (us)')
 DECODE_TIME_AVG = ('Decoding time', 'decode time (us) avg')
 FRAME_SIZE = ('Frame sizes', 'frame size (bytes)')
 FRAME_SIZE_AVG = ('Frame sizes', 'frame size (bytes) avg')
 AVG_KEY_FRAME_SIZE = ('Average key frame size', 'avg key frame size (bytes)')
@@ skipping 19 matching lines @@
   RESILIENCE,
   ERROR_CONCEALMENT,
   BITRATE,  # TODO(asapersson): Needs to be last.
 ]
 
 # Settings, options for subplots.
 SUBPLOT_SETTINGS = [
   CODEC_TYPE,
   ENCODER_IMPLEMENTATION_NAME,
   DECODER_IMPLEMENTATION_NAME,
+  CODEC_IMPLEMENTATION_NAME,
 ] + X_SETTINGS
 
 # Results.
 RESULTS = [
   PSNR,
   SSIM,
   ENC_BITRATE,
   NUM_DROPPED_FRAMES,
   NUM_FRAMES_TO_TARGET,
   ENCODE_TIME_AVG,
   DECODE_TIME_AVG,
   QP,
   AVG_KEY_FRAME_SIZE,
   AVG_NON_KEY_FRAME_SIZE,
 ]
 
 METRICS_TO_PARSE = SETTINGS + SUBPLOT_SETTINGS + RESULTS
 
 Y_METRICS = [res[1] for res in RESULTS]
 
 # Parameters for plotting.
-FIG_SIZE_SCALE_FACTOR_X = 2
-FIG_SIZE_SCALE_FACTOR_Y = 2.8
+FIG_SIZE_SCALE_FACTOR_X = 1.6
+FIG_SIZE_SCALE_FACTOR_Y = 1.8
 GRID_COLOR = [0.45, 0.45, 0.45]
 
 
 def ParseSetting(filename, setting):
   """Parses setting from file.
 
   Args:
     filename: The name of the file.
     setting: Name of setting to parse (e.g. width).
 
@@ skipping 164 matching lines @@
       "CPU cores used: 1.0": {
         "PSNR (dB)": [25.546029, 29.465518, 34.723535, 36.428493, 38.686551],
         "bitrate (kbps)": [50, 100, 300, 500, 1000]
       },
       "CPU cores used: 2.0": {
         "PSNR (dB)": [25.546029, 29.465518, 34.723535, 36.428493, 38.686551],
         "bitrate (kbps)": [50, 100, 300, 500, 1000]
       },
     }
     """
-  for key in metrics:
+  for key in sorted(metrics):
     data = metrics[key]
     if y_metric not in data:
       print "Failed to find metric: %s" % y_metric
       continue
 
     y = numpy.array(data[y_metric])
     x = numpy.array(data[x_metric])
     if len(y) != len(x):
       print "Length mismatch for %s, %s" % (y, x)
       continue
 
     label = y_metric + ' - ' + str(key)
 
     plt.plot(x, y, label=label, linewidth=1.5, marker='o', markersize=5,
              markeredgewidth=0.0)
 
 
 def PlotFigure(settings, y_metrics, x_metric, metrics, title):
   """Plots metrics in y_metrics list. One figure is plotted and each entry
   in the list is plotted in a subplot (and sorted per settings).
 
   For example:
     settings = ['width: 128.0', 'width: 160.0']. Sort subplot per setting.
     y_metrics = ['PSNR (dB)', 'PSNR (dB)']. Metric to plot per subplot.
     x_metric = 'bitrate (kbps)'
 
   """
 
   plt.figure()
-  plt.suptitle(title, fontsize='small', fontweight='bold')
+  plt.suptitle(title, fontsize='large', fontweight='bold')
+  settings.sort()
   rows = len(settings)
   cols = 1
   pos = 1
   while pos <= rows:
     plt.rc('grid', color=GRID_COLOR)
     ax = plt.subplot(rows, cols, pos)
     plt.grid()
-    plt.setp(ax.get_xticklabels(), visible=(pos == rows), fontsize='small')
-    plt.setp(ax.get_yticklabels(), fontsize='small')
+    plt.setp(ax.get_xticklabels(), visible=(pos == rows), fontsize='large')
+    plt.setp(ax.get_yticklabels(), fontsize='large')
     setting = settings[pos - 1]
     Plot(y_metrics[pos - 1], x_metric, metrics[setting])
-    plt.title(setting, fontsize='x-small')
-    plt.legend(fontsize='xx-small')
+    if setting.startswith(WIDTH[1]):
+      plt.title(setting, fontsize='medium')
+    plt.legend(fontsize='large', loc='best')
     pos += 1
 
-  plt.xlabel(x_metric, fontsize='small')
-  plt.subplots_adjust(left=0.04, right=0.98, bottom=0.04, top=0.96, hspace=0.1)
+  plt.xlabel(x_metric, fontsize='large')
+  plt.subplots_adjust(left=0.06, right=0.98, bottom=0.05, top=0.94, hspace=0.08)
 
 
-def GetTitle(filename):
+def GetTitle(filename, setting):
   title = ''
-  codec_types = ParseSetting(filename, CODEC_TYPE[1])
-  for i in range(0, len(codec_types)):
-    title += codec_types[i] + ', '
+  if setting != CODEC_IMPLEMENTATION_NAME[1] and setting != CODEC_TYPE[1]:
+    codec_types = ParseSetting(filename, CODEC_TYPE[1])
+    for i in range(0, len(codec_types)):
+      title += codec_types[i] + ', '
 
-  cores = ParseSetting(filename, CORES[1])
-  for i in range(0, len(cores)):
-    title += cores[i].split('.')[0] + ', '
+  if setting != CORES[1]:
+    cores = ParseSetting(filename, CORES[1])
+    for i in range(0, len(cores)):
+      title += cores[i].split('.')[0] + ', '
 
-  framerate = ParseSetting(filename, FRAMERATE[1])
-  for i in range(0, len(framerate)):
-    title += framerate[i].split('.')[0] + ', '
+  if setting != FRAMERATE[1]:
+    framerate = ParseSetting(filename, FRAMERATE[1])
+    for i in range(0, len(framerate)):
+      title += framerate[i].split('.')[0] + ', '
 
-  enc_names = ParseSetting(filename, ENCODER_IMPLEMENTATION_NAME[1])
-  for i in range(0, len(enc_names)):
-    title += enc_names[i] + ', '
+  if (setting != CODEC_IMPLEMENTATION_NAME[1] and
+      setting != ENCODER_IMPLEMENTATION_NAME[1]):
+    enc_names = ParseSetting(filename, ENCODER_IMPLEMENTATION_NAME[1])
+    for i in range(0, len(enc_names)):
+      title += enc_names[i] + ', '
 
-  dec_names = ParseSetting(filename, DECODER_IMPLEMENTATION_NAME[1])
-  for i in range(0, len(dec_names)):
-    title += dec_names[i] + ', '
+  if (setting != CODEC_IMPLEMENTATION_NAME[1] and
+      setting != DECODER_IMPLEMENTATION_NAME[1]):
+    dec_names = ParseSetting(filename, DECODER_IMPLEMENTATION_NAME[1])
+    for i in range(0, len(dec_names)):
+      title += dec_names[i] + ', '
 
   filenames = ParseSetting(filename, FILENAME[1])
   title += filenames[0].split('_')[0]
 
   num_frames = ParseSetting(filename, NUM_FRAMES[1])
   for i in range(0, len(num_frames)):
     title += ' (' + num_frames[i].split('.')[0] + ')'
 
   return title
 
@@ skipping 54 matching lines @@
       x_metric = X_SETTINGS[idx][1]
 
   metrics = ParseMetrics(filename, setting1, setting2)
 
   # Stretch fig size.
   figsize = plt.rcParams["figure.figsize"]
   figsize[0] *= FIG_SIZE_SCALE_FACTOR_X
   figsize[1] *= FIG_SIZE_SCALE_FACTOR_Y
   plt.rcParams["figure.figsize"] = figsize
 
-  PlotFigure(sub_keys, y_metrics, x_metric, metrics, GetTitle(filename))
+  PlotFigure(sub_keys, y_metrics, x_metric, metrics,
+             GetTitle(filename, setting2))
 
   plt.show()
 
 
 if __name__ == '__main__':
   main()