1 files changed, 116 insertions, 0 deletions
diff --git a/chromium/third_party/webrtc/modules/remote_bitrate_estimator/tools/bwe_rtp_play.cc b/chromium/third_party/webrtc/modules/remote_bitrate_estimator/tools/bwe_rtp_play.cc
new file mode 100644
index 00000000000..9ea3f08eab5
--- /dev/null
+++ b/chromium/third_party/webrtc/modules/remote_bitrate_estimator/tools/bwe_rtp_play.cc
@@ -0,0 +1,116 @@
+/*
+ *  Copyright (c) 2014 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.
+ */
+
+#include <stdio.h>
+
+#include "webrtc/modules/remote_bitrate_estimator/include/remote_bitrate_estimator.h"
+#include "webrtc/modules/remote_bitrate_estimator/tools/bwe_rtp.h"
+#include "webrtc/modules/rtp_rtcp/interface/rtp_header_parser.h"
+#include "webrtc/modules/rtp_rtcp/interface/rtp_payload_registry.h"
+#include "webrtc/modules/video_coding/main/test/rtp_file_reader.h"
+#include "webrtc/modules/video_coding/main/test/rtp_player.h"
+#include "webrtc/system_wrappers/interface/scoped_ptr.h"
+
+using webrtc::rtpplayer::RtpPacketSourceInterface;
+
+class Observer : public webrtc::RemoteBitrateObserver {
+ public:
+  explicit Observer(webrtc::Clock* clock) : clock_(clock) {}
+
+  // Called when a receive channel group has a new bitrate estimate for the
+  // incoming streams.
+  virtual void OnReceiveBitrateChanged(const std::vector<unsigned int>& ssrcs,
+                                       unsigned int bitrate) {
+    printf("[%u] Num SSRCs: %d, bitrate: %u\n",
+           static_cast<uint32_t>(clock_->TimeInMilliseconds()),
+           static_cast<int>(ssrcs.size()), bitrate);
+  }
+
+  virtual ~Observer() {}
+
+ private:
+  webrtc::Clock* clock_;
+};
+
+int main(int argc, char** argv) {
+  if (argc < 4) {
+    printf("Usage: bwe_rtp_play <extension type> <extension id> "
+           "<input_file.rtp>\n");
+    printf("<extension type> can either be:\n"
+           "  abs for absolute send time or\n"
+           "  tsoffset for timestamp offset.\n"
+           "<extension id> is the id associated with the extension.\n");
+    return -1;
+  }
+  RtpPacketSourceInterface* reader;
+  webrtc::RemoteBitrateEstimator* estimator;
+  webrtc::RtpHeaderParser* parser;
+  std::string estimator_used;
+  webrtc::SimulatedClock clock(0);
+  Observer observer(&clock);
+  if (!ParseArgsAndSetupEstimator(argc, argv, &clock, &observer, &reader,
+                                  &parser, &estimator, &estimator_used)) {
+    return -1;
+  }
+  webrtc::scoped_ptr<RtpPacketSourceInterface> rtp_reader(reader);
+  webrtc::scoped_ptr<webrtc::RtpHeaderParser> rtp_parser(parser);
+  webrtc::scoped_ptr<webrtc::RemoteBitrateEstimator> rbe(estimator);
+
+  // Process the file.
+  int packet_counter = 0;
+  int64_t next_process_time_ms = 0;
+  int64_t next_rtp_time_ms = 0;
+  int64_t first_rtp_time_ms = -1;
+  const uint32_t kMaxPacketSize = 1500;
+  uint8_t packet_buffer[kMaxPacketSize];
+  uint8_t* packet = packet_buffer;
+  int non_zero_abs_send_time = 0;
+  int non_zero_ts_offsets = 0;
+  while (true) {
+    uint32_t next_rtp_time;
+    if (next_rtp_time_ms <= clock.TimeInMilliseconds()) {
+      uint32_t packet_length = kMaxPacketSize;
+      if (rtp_reader->NextPacket(packet, &packet_length,
+                                 &next_rtp_time) == -1) {
+        break;
+      }
+      if (first_rtp_time_ms == -1)
+        first_rtp_time_ms = next_rtp_time;
+      next_rtp_time_ms = next_rtp_time - first_rtp_time_ms;
+      webrtc::RTPHeader header;
+      parser->Parse(packet, packet_length, &header);
+      if (header.extension.absoluteSendTime != 0)
+        ++non_zero_abs_send_time;
+      if (header.extension.transmissionTimeOffset != 0)
+        ++non_zero_ts_offsets;
+      rbe->IncomingPacket(clock.TimeInMilliseconds(),
+                          packet_length - header.headerLength,
+                          header);
+      ++packet_counter;
+    }
+    next_process_time_ms = rbe->TimeUntilNextProcess() +
+        clock.TimeInMilliseconds();
+    if (next_process_time_ms <= clock.TimeInMilliseconds()) {
+      rbe->Process();
+    }
+    int time_until_next_event =
+        std::min(next_process_time_ms, next_rtp_time_ms) -
+        clock.TimeInMilliseconds();
+    clock.AdvanceTimeMilliseconds(std::max(time_until_next_event, 0));
+  }
+  printf("Parsed %d packets\nTime passed: %u ms\n", packet_counter,
+         static_cast<uint32_t>(clock.TimeInMilliseconds()));
+  printf("Estimator used: %s\n", estimator_used.c_str());
+  printf("Packets with non-zero absolute send time: %d\n",
+         non_zero_abs_send_time);
+  printf("Packets with non-zero timestamp offset: %d\n",
+         non_zero_ts_offsets);
+  return 0;
+}