diff --git a/webrtc/tools/py_event_log_analyzer/rtp_analyzer.py b/webrtc/tools/py_event_log_analyzer/rtp_analyzer.py index 1f7b3f84d9..1a9b695c3a 100644 --- a/webrtc/tools/py_event_log_analyzer/rtp_analyzer.py +++ b/webrtc/tools/py_event_log_analyzer/rtp_analyzer.py @@ -13,6 +13,7 @@ from __future__ import print_function import collections import optparse +import os import sys import matplotlib.pyplot as plt @@ -275,9 +276,9 @@ def calculate_delay(start, stop, step, points): for point in points: grouped_delays[rounded_value_index(point.real_send_time_ms) ].append(point.absdelay) - regularized_delays = [numpy.average(arr) if arr else None for arr in + regularized_delays = [numpy.average(arr) if arr else -1 for arr in grouped_delays] - return numpy.ma.masked_values(regularized_delays, None) + return numpy.ma.masked_values(regularized_delays, -1) def main(): @@ -290,13 +291,22 @@ def main(): default=False, action="store_true", help="always query user for real sample rate") + parser.add_option("--working_directory", + default=None, action="store", + help="directory in which to search for relative paths") + (options, args) = parser.parse_args() if len(args) < 1: parser.print_help() sys.exit(0) - data_points = pb_parse.parse_protobuf(args[0]) + input_file = args[0] + + if options.working_directory and not os.path.isabs(input_file): + input_file = os.path.join(options.working_directory, input_file) + + data_points = pb_parse.parse_protobuf(input_file) rtp_stats = RTPStatistics(data_points) if options.dump_header_to_stdout: diff --git a/webrtc/tools/py_event_log_analyzer/rtp_analyzer.sh b/webrtc/tools/py_event_log_analyzer/rtp_analyzer.sh index 7727d597d2..817ca4f1c4 100755 --- a/webrtc/tools/py_event_log_analyzer/rtp_analyzer.sh +++ b/webrtc/tools/py_event_log_analyzer/rtp_analyzer.sh @@ -7,10 +7,11 @@ # in the file PATENTS. All contributing project authors may # be found in the AUTHORS file in the root of the source tree. set -e +WORKING_DIR=$(pwd) cd $(dirname $0) PYTHONPATH="../../third_party/protobuf/python/" if [ -z ${PYTHON_EXECUTABLE+x} ] then PYTHON_EXECUTABLE=python3 fi -exec $PYTHON_EXECUTABLE "rtp_analyzer.py" $@ +exec $PYTHON_EXECUTABLE "rtp_analyzer.py" $@ --working_dir $WORKING_DIR