Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(63)

Side by Side Diff: tools/py_event_log_analyzer/pb_parse.py

Issue 1999113002: New rtc dump analyzing tool in Python (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: Formatting, licence and usage string. Created 4 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 # Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
2 #
3 # Use of this source code is governed by a BSD-style license
4 # that can be found in the LICENSE file in the root of the source
5 # tree. An additional intellectual property rights grant can be found
6 # in the file PATENTS. All contributing project authors may
7 # be found in the AUTHORS file in the root of the source tree.
8
9 """Parses protobuf RTC dumps."""
10
11 from __future__ import division
12
13 import struct
14 import sys
15
16 sys.path.append("../../third_party/protobuf/python/")
17
18 import pyproto.webrtc.call.rtc_event_log_pb2 as rtc_pb
19
20
21 class DataPoint(object):
22 """Simple container class for RTP events."""
23
24 def __init__(self, rtp_header_str, packet_size,
25 arrival_timestamp_us):
26 """Builds a data point by parsing an RTP header, size and arrival time.
27
28 RTP header structure is defined in RFC 3550 section 5.1.
29 """
30 self.size = packet_size
31 self.arrival_timestamp_ms = arrival_timestamp_us / 1000
32 (first2header_bytes, self.sequence_number,
33 self.timestamp, self.ssrc
34 ) = struct.unpack_from("!HHII", rtp_header_str, 0)
35 self.payload_type = first2header_bytes & 0b01111111
36
37
38 def parse_protobuf(file_path):
39 """Parses RTC event log from protobuf file.
40
41 Args:
42 file_path: path to protobuf file of RTC event stream
43
44 Returns:
45 all RTP packet events from the event stream as a list of DataPoints
46 """
47 event_stream = rtc_pb.EventStream()
48 with open(file_path, "rb") as f:
49 event_stream.ParseFromString(f.read())
50
51 return [DataPoint(event.rtp_packet.header,
52 event.rtp_packet.packet_length,
53 event.timestamp_us)
54 for event in event_stream.stream
55 if event.HasField("rtp_packet")]
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698