| Index: webrtc/modules/rtp_rtcp/source/bitrate.cc
|
| diff --git a/webrtc/modules/rtp_rtcp/source/bitrate.cc b/webrtc/modules/rtp_rtcp/source/bitrate.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..49a23592bfe8c4a988835f0ca09a211f62f8fbdf
|
| --- /dev/null
|
| +++ b/webrtc/modules/rtp_rtcp/source/bitrate.cc
|
| @@ -0,0 +1,121 @@
|
| +/*
|
| + * Copyright (c) 2012 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 "webrtc/modules/rtp_rtcp/source/bitrate.h"
|
| +
|
| +#include "webrtc/modules/rtp_rtcp/source/rtp_utility.h"
|
| +
|
| +namespace webrtc {
|
| +
|
| +Bitrate::Bitrate(Clock* clock, Observer* observer)
|
| + : clock_(clock),
|
| + packet_rate_(0),
|
| + bitrate_(0),
|
| + bitrate_next_idx_(0),
|
| + time_last_rate_update_(0),
|
| + bytes_count_(0),
|
| + packet_count_(0),
|
| + observer_(observer) {
|
| + memset(packet_rate_array_, 0, sizeof(packet_rate_array_));
|
| + memset(bitrate_diff_ms_, 0, sizeof(bitrate_diff_ms_));
|
| + memset(bitrate_array_, 0, sizeof(bitrate_array_));
|
| +}
|
| +
|
| +Bitrate::~Bitrate() {}
|
| +
|
| +void Bitrate::Update(const size_t bytes) {
|
| + rtc::CritScope cs(&crit_);
|
| + bytes_count_ += bytes;
|
| + packet_count_++;
|
| +}
|
| +
|
| +uint32_t Bitrate::PacketRate() const {
|
| + rtc::CritScope cs(&crit_);
|
| + return packet_rate_;
|
| +}
|
| +
|
| +uint32_t Bitrate::BitrateLast() const {
|
| + rtc::CritScope cs(&crit_);
|
| + return bitrate_;
|
| +}
|
| +
|
| +uint32_t Bitrate::BitrateNow() const {
|
| + rtc::CritScope cs(&crit_);
|
| + int64_t now = clock_->TimeInMilliseconds();
|
| + int64_t diff_ms = now - time_last_rate_update_;
|
| +
|
| + if (diff_ms > 10000) { // 10 seconds.
|
| + // Too high difference, ignore.
|
| + return bitrate_;
|
| + }
|
| + int64_t bits_since_last_rate_update = 8 * bytes_count_ * 1000;
|
| +
|
| + // We have to consider the time when the measurement was done:
|
| + // ((bits/sec * sec) + (bits)) / sec.
|
| + int64_t bitrate = (static_cast<uint64_t>(bitrate_) * 1000 +
|
| + bits_since_last_rate_update) / (1000 + diff_ms);
|
| + return static_cast<uint32_t>(bitrate);
|
| +}
|
| +
|
| +int64_t Bitrate::time_last_rate_update() const {
|
| + rtc::CritScope cs(&crit_);
|
| + return time_last_rate_update_;
|
| +}
|
| +
|
| +// Triggered by timer.
|
| +void Bitrate::Process() {
|
| + BitrateStatistics stats;
|
| + {
|
| + rtc::CritScope cs(&crit_);
|
| + int64_t now = clock_->CurrentNtpInMilliseconds();
|
| + int64_t diff_ms = now - time_last_rate_update_;
|
| +
|
| + if (diff_ms < 100) {
|
| + // Not enough data, wait...
|
| + return;
|
| + }
|
| + if (diff_ms > 10000) { // 10 seconds.
|
| + // Too high difference, ignore.
|
| + time_last_rate_update_ = now;
|
| + bytes_count_ = 0;
|
| + packet_count_ = 0;
|
| + return;
|
| + }
|
| + packet_rate_array_[bitrate_next_idx_] = (packet_count_ * 1000) / diff_ms;
|
| + bitrate_array_[bitrate_next_idx_] = 8 * ((bytes_count_ * 1000) / diff_ms);
|
| + bitrate_diff_ms_[bitrate_next_idx_] = diff_ms;
|
| + bitrate_next_idx_++;
|
| + if (bitrate_next_idx_ >= 10) {
|
| + bitrate_next_idx_ = 0;
|
| + }
|
| + int64_t sum_diffMS = 0;
|
| + int64_t sum_bitrateMS = 0;
|
| + int64_t sum_packetrateMS = 0;
|
| + for (int i = 0; i < 10; i++) {
|
| + sum_diffMS += bitrate_diff_ms_[i];
|
| + sum_bitrateMS += bitrate_array_[i] * bitrate_diff_ms_[i];
|
| + sum_packetrateMS += packet_rate_array_[i] * bitrate_diff_ms_[i];
|
| + }
|
| + time_last_rate_update_ = now;
|
| + bytes_count_ = 0;
|
| + packet_count_ = 0;
|
| + packet_rate_ = static_cast<uint32_t>(sum_packetrateMS / sum_diffMS);
|
| + bitrate_ = static_cast<uint32_t>(sum_bitrateMS / sum_diffMS);
|
| +
|
| + stats.bitrate_bps = bitrate_;
|
| + stats.packet_rate = packet_rate_;
|
| + stats.timestamp_ms = now;
|
| + }
|
| +
|
| + if (observer_)
|
| + observer_->BitrateUpdated(stats);
|
| +}
|
| +
|
| +} // namespace webrtc
|
|
|