Add parser to visualise the ana dump

webrtc/modules/audio_coding/audio_network_adaptor/parse_ana_dump.py

Index: webrtc/modules/audio_coding/audio_network_adaptor/parse_ana_dump.py
diff --git a/webrtc/modules/audio_coding/audio_network_adaptor/parse_ana_dump.py b/webrtc/modules/audio_coding/audio_network_adaptor/parse_ana_dump.py
new file mode 100755
index 0000000000000000000000000000000000000000..a52b065f12c7c7950791695fe80b4d7ee190207d
--- /dev/null
+++ b/webrtc/modules/audio_coding/audio_network_adaptor/parse_ana_dump.py
@@ -0,0 +1,147 @@
+#!/usr/bin/python2
+#  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.
+
+#  To run this script please copy "out/<build_name>//pyproto/webrtc/modules/
+#  audio_coding/audio_network_adaptor/debug_dump_pb2.py" to this folder.
+#  The you can run this script with:
+#  "python parse_ana_dump.py -m uplink_bandwidth_bps -f dump_file.dat"
+#  You can add as may metrics or decisions to the plot as you like.
+#  form more information call:
+#  "python parse_ana_dump.py --help"
+
+import struct
+from optparse import OptionParser
+
+import matplotlib.pyplot as plt
+
+import debug_dump_pb2
+
+
+def GetNextMessageSize(file_to_parse):
+  data = file_to_parse.read(4)
+  if data == '':
+    return 0
+  return struct.unpack('<I', data)[0]
+
+
+def GetNextMessageFromFile(file_to_parse):
+  message_size = GetNextMessageSize(file_to_parse)
+  if message_size == 0:
+    return None
+  try:
+    event = debug_dump_pb2.Event()
+    event.ParseFromString(file_to_parse.read(message_size))
+  except IOError:
+    print 'Invalid message in file'
+    return None
+  return event
+
+
+def InitMetrics():
+  metrics = {}
+  event = debug_dump_pb2.Event()
+  for metric in event.network_metrics.DESCRIPTOR.fields:
+    metrics[metric.name] = {'time': [], 'value': []}
+  return metrics
+
+
+def InitDecisions():
+  decisions = {}
+  event = debug_dump_pb2.Event()
+  for decision in event.encoder_runtime_config.DESCRIPTOR.fields:
+    decisions[decision.name] = {'time': [], 'value': []}
+  return decisions
+
+
+def ParseAnaDump(dump_file_to_parse):
+  with open(dump_file_to_parse, 'rb') as file_to_parse:
+    metrics = InitMetrics()
+    decisions = InitDecisions()
+    first_time_stamp = None
+    while True:
+      event = GetNextMessageFromFile(file_to_parse)
+      if event == None:
+        break
+      if first_time_stamp == None:
+        first_time_stamp = event.timestamp
+      if event.type == debug_dump_pb2.Event.ENCODER_RUNTIME_CONFIG:
+        for decision in event.encoder_runtime_config.DESCRIPTOR.fields:
+          if event.encoder_runtime_config.HasField(decision.name):
+            decisions[decision.name]['time'].append(event.timestamp -
+                                                    first_time_stamp)
+            decisions[decision.name]['value'].append(
+                getattr(event.encoder_runtime_config, decision.name))
+      if event.type == debug_dump_pb2.Event.NETWORK_METRICS:
+        for metric in event.network_metrics.DESCRIPTOR.fields:
+          if event.network_metrics.HasField(metric.name):
+            metrics[metric.name]['time'].append(event.timestamp -
+                                                first_time_stamp)
+            metrics[metric.name]['value'].append(
+                getattr(event.network_metrics, metric.name))
+  return (metrics, decisions)
+
+
+def main():
+  parser = OptionParser()
+  parser.add_option(
+      "-f", "--dump_file", dest="dump_file_to_parse", help="dump file to parse")
+  parser.add_option(
+      '-m',
+      '--metric_plot',
+      default=[],
+      type=str,
+      help='metric key (name of the metric) to plot',
+      dest='metric_keys',
+      action='append')
+
+  parser.add_option(
+      '-d',
+      '--decision_plot',
+      default=[],
+      type=str,
+      help='decision key (name of the decision) to plot',
+      dest='decision_keys',
+      action='append')
+
+  options = parser.parse_args()[0]
+  if options.dump_file_to_parse == None:
+    print "No dump file to parse is set.\n"
+    parser.print_help()
+    exit()
+  (metrics, decisions) = ParseAnaDump(options.dump_file_to_parse)
+  metric_keys = options.metric_keys
+  decision_keys = options.decision_keys
+  plot_count = len(metric_keys) + len(decision_keys)
+  if plot_count == 0:
+    print "You have to set at least one metric or decision to plot.\n"
+    parser.print_help()
+    exit()
+  plots = []
+  if plot_count == 1:
+    f, mp_plot = plt.subplots()
+    plots.append(mp_plot)
+  else:
+    f, mp_plots = plt.subplots(plot_count, sharex=True)
+    plots.extend(mp_plots.tolist())
+
+  for key in metric_keys:
+    plot = plots.pop()
+    plot.grid(True)
+    plot.set_title(key + " (metric)")
+    plot.plot(metrics[key]['time'], metrics[key]['value'])
+  for key in decision_keys:
+    plot = plots.pop()
+    plot.grid(True)
+    plot.set_title(key + " (decision)")
+    plot.plot(decisions[key]['time'], decisions[key]['value'])
+  f.subplots_adjust(hspace=0.3)
+  plt.show()
+
+if __name__ == "__main__":
+  main()