mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-16 23:30:48 +01:00

Since the webrtc_common build target does not have visibility set, we cannot easily use BitrateAllocation in other parts of Chromium. This is currently blocking parts of chromium:794608, and I know of other usage outside webrtc already, so moving it to api/ should be warranted. Also, since there's some naming confusion and this class is video specific rename it VideoBitrateAllocation. This also fits with the standard interface for producing these: VideoBitrateAllocator. Bug: chromium:794608 Change-Id: I4c0fae40f9365e860c605a76a4f67ecc9b9cf9fe Reviewed-on: https://webrtc-review.googlesource.com/70783 Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Reviewed-by: Niels Moller <nisse@webrtc.org> Commit-Queue: Erik Språng <sprang@webrtc.org> Cr-Commit-Position: refs/heads/master@{#22986}
103 lines
3.5 KiB
C++
103 lines
3.5 KiB
C++
/*
|
|
* Copyright (c) 2018 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.
|
|
*/
|
|
|
|
#ifndef TEST_ENCODER_PROXY_FACTORY_H_
|
|
#define TEST_ENCODER_PROXY_FACTORY_H_
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "api/video_codecs/video_encoder.h"
|
|
#include "api/video_codecs/video_encoder_factory.h"
|
|
#include "rtc_base/ptr_util.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
// An encoder factory with a single underlying VideoEncoder object,
|
|
// intended for test purposes. Each call to CreateVideoEncoder returns
|
|
// a proxy for the same encoder, typically an instance of FakeEncoder.
|
|
class EncoderProxyFactory final : public VideoEncoderFactory {
|
|
public:
|
|
explicit EncoderProxyFactory(VideoEncoder* encoder) : encoder_(encoder) {
|
|
codec_info_.is_hardware_accelerated = false;
|
|
codec_info_.has_internal_source = false;
|
|
}
|
|
|
|
// Unused by tests.
|
|
std::vector<SdpVideoFormat> GetSupportedFormats() const override {
|
|
RTC_NOTREACHED();
|
|
return {};
|
|
}
|
|
|
|
CodecInfo QueryVideoEncoder(const SdpVideoFormat& format) const override {
|
|
return codec_info_;
|
|
}
|
|
|
|
std::unique_ptr<VideoEncoder> CreateVideoEncoder(
|
|
const SdpVideoFormat& format) override {
|
|
return rtc::MakeUnique<EncoderProxy>(encoder_);
|
|
}
|
|
|
|
void SetIsHardwareAccelerated(bool is_hardware_accelerated) {
|
|
codec_info_.is_hardware_accelerated = is_hardware_accelerated;
|
|
}
|
|
void SetHasInternalSource(bool has_internal_source) {
|
|
codec_info_.has_internal_source = has_internal_source;
|
|
}
|
|
|
|
private:
|
|
// Wrapper class, since CreateVideoEncoder needs to surrender
|
|
// ownership to the object it returns.
|
|
class EncoderProxy final : public VideoEncoder {
|
|
public:
|
|
explicit EncoderProxy(VideoEncoder* encoder) : encoder_(encoder) {}
|
|
|
|
private:
|
|
int32_t Encode(const VideoFrame& input_image,
|
|
const CodecSpecificInfo* codec_specific_info,
|
|
const std::vector<FrameType>* frame_types) override {
|
|
return encoder_->Encode(input_image, codec_specific_info, frame_types);
|
|
}
|
|
int32_t InitEncode(const VideoCodec* config,
|
|
int32_t number_of_cores,
|
|
size_t max_payload_size) override {
|
|
return encoder_->InitEncode(config, number_of_cores, max_payload_size);
|
|
}
|
|
VideoEncoder::ScalingSettings GetScalingSettings() const override {
|
|
return encoder_->GetScalingSettings();
|
|
}
|
|
int32_t RegisterEncodeCompleteCallback(
|
|
EncodedImageCallback* callback) override {
|
|
return encoder_->RegisterEncodeCompleteCallback(callback);
|
|
}
|
|
int32_t Release() override { return encoder_->Release(); }
|
|
int32_t SetChannelParameters(uint32_t packet_loss, int64_t rtt) override {
|
|
return encoder_->SetChannelParameters(packet_loss, rtt);
|
|
}
|
|
int32_t SetRateAllocation(const VideoBitrateAllocation& rate_allocation,
|
|
uint32_t framerate) override {
|
|
return encoder_->SetRateAllocation(rate_allocation, framerate);
|
|
}
|
|
const char* ImplementationName() const override {
|
|
return encoder_->ImplementationName();
|
|
}
|
|
|
|
VideoEncoder* const encoder_;
|
|
};
|
|
|
|
VideoEncoder* const encoder_;
|
|
CodecInfo codec_info_;
|
|
};
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|
|
|
|
#endif // TEST_ENCODER_PROXY_FACTORY_H_
|