77 lines
2.0 KiB
C++
77 lines
2.0 KiB
C++
// Copyright 2014 Google Inc. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include "packager/mpd/base/bandwidth_estimator.h"
|
|
|
|
#include <cmath>
|
|
|
|
#include "packager/base/logging.h"
|
|
|
|
namespace shaka {
|
|
|
|
BandwidthEstimator::BandwidthEstimator(size_t num_blocks)
|
|
: sliding_queue_(num_blocks) {}
|
|
BandwidthEstimator::~BandwidthEstimator() {}
|
|
|
|
void BandwidthEstimator::AddBlock(uint64_t size, double duration) {
|
|
if (size == 0 || duration == 0) {
|
|
LOG(WARNING) << "Ignore block with size=" << size
|
|
<< ", duration=" << duration;
|
|
return;
|
|
}
|
|
const int kBitsInByte = 8;
|
|
const double bits_per_second_reciprocal = duration / (kBitsInByte * size);
|
|
sliding_queue_.Add(bits_per_second_reciprocal);
|
|
}
|
|
|
|
uint64_t BandwidthEstimator::Estimate() const {
|
|
return sliding_queue_.size() == 0
|
|
? 0
|
|
: static_cast<uint64_t>(
|
|
ceil(sliding_queue_.size() / sliding_queue_.sum()));
|
|
}
|
|
|
|
uint64_t BandwidthEstimator::Max() const {
|
|
// The first element has minimum "bits per second reciprocal", thus the
|
|
// reverse is maximum "bits per second".
|
|
return sliding_queue_.size() == 0
|
|
? 0
|
|
: static_cast<uint64_t>(ceil(1 / sliding_queue_.min()));
|
|
}
|
|
|
|
BandwidthEstimator::SlidingQueue::SlidingQueue(size_t window_size)
|
|
: window_size_(window_size) {}
|
|
|
|
void BandwidthEstimator::SlidingQueue::Add(double value) {
|
|
// Remove elements if needed to form a monotonic non-decreasing sequence.
|
|
while (!min_.empty() && min_.back() > value)
|
|
min_.pop_back();
|
|
min_.push_back(value);
|
|
|
|
if (window_size_ == kUseAllBlocks) {
|
|
size_++;
|
|
sum_ += value;
|
|
min_.resize(1); // Keep only the minimum one.
|
|
return;
|
|
}
|
|
|
|
window_.push_back(value);
|
|
sum_ += value;
|
|
|
|
if (window_.size() <= window_size_) {
|
|
size_++;
|
|
return;
|
|
}
|
|
|
|
if (min_.front() == window_.front())
|
|
min_.pop_front();
|
|
|
|
sum_ -= window_.front();
|
|
window_.pop_front();
|
|
}
|
|
|
|
} // namespace shaka
|