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

Unified 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: Updated README to reflect changes in code. Created 4 years, 7 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 side-by-side diff with in-line comments
Download patch
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")]

Powered by Google App Engine
This is Rietveld 408576698