OLD | NEW |
---|---|
1 # Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 1 # Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
2 # | 2 # |
3 # Use of this source code is governed by a BSD-style license | 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 | 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 | 5 # tree. An additional intellectual property rights grant can be found |
6 # in the file PATENTS. All contributing project authors may | 6 # in the file PATENTS. All contributing project authors may |
7 # be found in the AUTHORS file in the root of the source tree. | 7 # be found in the AUTHORS file in the root of the source tree. |
8 | 8 |
9 """Parses protobuf RTC dumps.""" | 9 """Parses protobuf RTC dumps.""" |
10 | 10 |
11 from __future__ import division | 11 from __future__ import division |
12 import struct | 12 import struct |
13 import pyproto.webrtc.call.rtc_event_log_pb2 as rtc_pb | 13 import pyproto.webrtc.call.rtc_event_log_pb2 as rtc_pb |
14 | 14 |
15 | 15 |
16 class DataPoint(object): | 16 class DataPoint(object): |
17 """Simple container class for RTP events.""" | 17 """Simple container class for RTP events.""" |
18 | 18 |
19 def __init__(self, rtp_header_str, packet_size, | 19 def __init__(self, rtp_header_str, packet_size, |
20 arrival_timestamp_us): | 20 arrival_timestamp_us, incoming): |
21 """Builds a data point by parsing an RTP header, size and arrival time. | 21 """Builds a data point by parsing an RTP header, size and arrival time. |
22 | 22 |
23 RTP header structure is defined in RFC 3550 section 5.1. | 23 RTP header structure is defined in RFC 3550 section 5.1. |
24 """ | 24 """ |
25 self.size = packet_size | 25 self.size = packet_size |
26 self.arrival_timestamp_ms = arrival_timestamp_us / 1000 | 26 self.arrival_timestamp_ms = arrival_timestamp_us / 1000 |
27 self.incoming = incoming | |
27 header = struct.unpack_from("!HHII", rtp_header_str, 0) | 28 header = struct.unpack_from("!HHII", rtp_header_str, 0) |
28 (first2header_bytes, self.sequence_number, self.timestamp, | 29 (first2header_bytes, self.sequence_number, self.timestamp, |
29 self.ssrc) = header | 30 self.ssrc) = header |
30 self.payload_type = first2header_bytes & 0b01111111 | 31 self.payload_type = first2header_bytes & 0b01111111 |
31 self.marker_bit = (first2header_bytes & 0b10000000) >> 7 | 32 self.marker_bit = (first2header_bytes & 0b10000000) >> 7 |
32 | 33 |
33 | 34 |
34 def parse_protobuf(file_path): | 35 def parse_protobuf(file_path): |
35 """Parses RTC event log from protobuf file. | 36 """Parses RTC event log from protobuf file. |
36 | 37 |
37 Args: | 38 Args: |
38 file_path: path to protobuf file of RTC event stream | 39 file_path: path to protobuf file of RTC event stream |
39 | 40 |
40 Returns: | 41 Returns: |
41 all RTP packet events from the event stream as a list of DataPoints | 42 all RTP packet events from the event stream as a list of DataPoints |
42 """ | 43 """ |
43 event_stream = rtc_pb.EventStream() | 44 event_stream = rtc_pb.EventStream() |
44 with open(file_path, "rb") as f: | 45 with open(file_path, "rb") as f: |
45 event_stream.ParseFromString(f.read()) | 46 event_stream.ParseFromString(f.read()) |
46 | 47 |
47 return [DataPoint(event.rtp_packet.header, | 48 return [DataPoint(event.rtp_packet.header, |
phoglund
2016/09/06 13:14:50
Nit: remove extra space
| |
48 event.rtp_packet.packet_length, | 49 event.rtp_packet.packet_length, |
49 event.timestamp_us) | 50 event.timestamp_us, event.rtp_packet.incoming) |
50 for event in event_stream.stream | 51 for event in event_stream.stream |
51 if event.HasField("rtp_packet")] | 52 if event.HasField("rtp_packet")] |
OLD | NEW |