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..a7ab7025163648467c5b81a39b17c692b67ce01d |
| --- /dev/null |
| +++ b/tools/py_event_log_analyzer/pb_parse.py |
| @@ -0,0 +1,55 @@ |
| +# 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 struct |
| +import sys |
| + |
| +sys.path.append("../../third_party/protobuf/python/") |
| + |
| +import pyproto.webrtc.call.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, size and arrival time. |
| + |
| + RTP header structure is defined in RFC 3550 section 5.1. |
| + """ |
| + self.size = packet_size |
| + self.arrival_timestamp_ms = arrival_timestamp_us / 1000 |
| + (first2header_bytes, self.sequence_number, |
|
phoglund
2016/06/03 08:11:07
Try this instead:
header = struct.unpack_from("!
aleloi
2016/06/09 12:00:44
Here there are also divided views between differen
kwiberg-webrtc
2016/06/09 12:27:20
Pro tip: If two reviewers give you conflicting adv
phoglund
2016/06/09 12:47:08
All right, I won't fight you on this.
|
| + self.timestamp, self.ssrc |
| + ) = struct.unpack_from("!HHII", rtp_header_str, 0) |
| + self.payload_type = first2header_bytes & 0b01111111 |
| + |
| + |
| +def parse_protobuf(file_path): |
| + """Parses RTC event log from protobuf file. |
| + |
| + Args: |
| + file_path: path to protobuf file of RTC event stream |
| + |
| + Returns: |
| + all RTP packet events from the event stream as a list of DataPoints |
| + """ |
| + 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")] |