Chromium Code Reviews| Index: tools/py_event_log_analyzer/pb_parse.py |
| diff --git a/tools/py_event_log_analyzer/pb_parse.py b/tools/py_event_log_analyzer/pb_parse.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..9635197fa37a872ab260a960e2ad91f1e051f087 |
| --- /dev/null |
| +++ b/tools/py_event_log_analyzer/pb_parse.py |
| @@ -0,0 +1,63 @@ |
| +# Copyright (c) 2016 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. |
| + |
| +"""Parses protobuf RTC dumps. |
| +""" |
| + |
| +from __future__ import division |
| + |
| +import os |
| +import struct |
| +import sys |
| + |
| + |
| +# makes sure to import correct protobuf |
| +webrtc_root = "../../" |
| +sys.path.append(os.path.join(webrtc_root, "out/Debug/pyproto/webrtc/call/")) |
| +sys.path.append(os.path.join(webrtc_root, "out/Release/pyproto/webrtc/call/")) |
| +sys.path.append(os.path.join(webrtc_root, "third_party/protobuf/python/")) |
| +import rtc_event_log_pb2 as rtc_pb |
| + |
| + |
| +class DataPoint(object): |
| + """Simple container class for RTP events. |
| + """ |
| + |
| + def __init__(self, rtp_header_str, packet_size, |
| + arrival_timestamp_us): |
| + """Builds a data point by parsing an RTP header. |
| + |
| + As defined in RFC 3550 section 5.1. |
|
hlundin-webrtc
2016/05/25 12:54:23
Technically, the arrival timestamp is not part of
aleloi2
2016/05/30 14:57:55
Updated a little.
|
| + """ |
| + self.size = packet_size |
| + self.arrival_timestamp_ms = arrival_timestamp_us / 1000 |
| + unpacked = struct.unpack_from("!HHII", rtp_header_str, 0) |
| + (first2header_bytes, self.sequence_number, |
| + self.timestamp, self.ssrc |
| + ) = unpacked |
|
kwiberg-webrtc
2016/05/25 12:42:41
Eliminate "unpacked"?
(first2header_bytes, self.s
aleloi2
2016/05/30 14:57:55
Done.
|
| + self.payload_type = first2header_bytes & 0b01111111 |
| + |
| + |
| +def parse_protobuf(file_path): |
| + """Parses rtc event log from protobuf file. |
|
peah-webrtc
2016/05/26 06:44:18
I think the abbreviation rtc should be capitalized
aleloi2
2016/05/30 14:57:55
Done.
|
| + |
| + Args: |
| + file_path: path to protobuf file of RTC event stream |
| + |
| + Returns: |
| + all RtpPacket events from the event stream as a list of DataPoints |
|
peah-webrtc
2016/05/26 06:44:18
Rtp should be capitalized according to how it is c
aleloi2
2016/05/30 14:57:55
Done.
|
| + """ |
| + event_stream = rtc_pb.EventStream() |
| + with open(file_path, "rb") as f: |
| + event_stream.ParseFromString(f.read()) |
| + |
| + return [DataPoint(event.rtp_packet.header, |
| + event.rtp_packet.packet_length, |
| + event.timestamp_us) |
| + for event in event_stream.stream |
| + if event.HasField("rtp_packet")] |