mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-12 21:30:45 +01:00
8f01c4e1b6
The purpose of this interface is to allow VideoEncoder to override the bandwidth allocation set by FecController in RtpVideoSender. This CL defines the interface and sends it down to VideoSender. Two upcoming CLs will: 1. Make LibvpxVp8Encoder pass it on to the (injectable) FrameBufferController, where it might be put to good use. 2. Modify RtpVideoSender to respond to the message sent to it via this API. TBR=kwiberg@webrtc.org Bug: webrtc:10769 Change-Id: I2ef82f0ddcde7fd078e32d8aabf6efe43e0f7f8a Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/143962 Commit-Queue: Elad Alon <eladalon@webrtc.org> Reviewed-by: Erik Språng <sprang@webrtc.org> Reviewed-by: Rasmus Brandt <brandtr@webrtc.org> Cr-Commit-Position: refs/heads/master@{#28416}
100 lines
3.1 KiB
C++
100 lines
3.1 KiB
C++
/*
|
|
* Copyright (c) 2013 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 "test/configurable_frame_size_encoder.h"
|
|
|
|
#include <string.h>
|
|
#include <cstdint>
|
|
#include <type_traits>
|
|
#include <utility>
|
|
|
|
#include "api/video/encoded_image.h"
|
|
#include "modules/include/module_common_types.h"
|
|
#include "modules/video_coding/include/video_codec_interface.h"
|
|
#include "modules/video_coding/include/video_error_codes.h"
|
|
#include "rtc_base/checks.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
ConfigurableFrameSizeEncoder::ConfigurableFrameSizeEncoder(
|
|
size_t max_frame_size)
|
|
: callback_(NULL),
|
|
max_frame_size_(max_frame_size),
|
|
current_frame_size_(max_frame_size),
|
|
buffer_(new uint8_t[max_frame_size]),
|
|
codec_type_(kVideoCodecGeneric) {
|
|
memset(buffer_.get(), 0, max_frame_size);
|
|
}
|
|
|
|
ConfigurableFrameSizeEncoder::~ConfigurableFrameSizeEncoder() {}
|
|
|
|
void ConfigurableFrameSizeEncoder::SetFecControllerOverride(
|
|
FecControllerOverride* fec_controller_override) {
|
|
// Ignored.
|
|
}
|
|
|
|
int32_t ConfigurableFrameSizeEncoder::InitEncode(
|
|
const VideoCodec* codec_settings,
|
|
const Settings& settings) {
|
|
return WEBRTC_VIDEO_CODEC_OK;
|
|
}
|
|
|
|
int32_t ConfigurableFrameSizeEncoder::Encode(
|
|
const VideoFrame& inputImage,
|
|
const std::vector<VideoFrameType>* frame_types) {
|
|
EncodedImage encodedImage(buffer_.get(), current_frame_size_,
|
|
max_frame_size_);
|
|
encodedImage._completeFrame = true;
|
|
encodedImage._encodedHeight = inputImage.height();
|
|
encodedImage._encodedWidth = inputImage.width();
|
|
encodedImage._frameType = VideoFrameType::kVideoFrameKey;
|
|
encodedImage.SetTimestamp(inputImage.timestamp());
|
|
encodedImage.capture_time_ms_ = inputImage.render_time_ms();
|
|
RTPFragmentationHeader* fragmentation = NULL;
|
|
CodecSpecificInfo specific{};
|
|
specific.codecType = codec_type_;
|
|
callback_->OnEncodedImage(encodedImage, &specific, fragmentation);
|
|
if (post_encode_callback_) {
|
|
(*post_encode_callback_)();
|
|
}
|
|
return WEBRTC_VIDEO_CODEC_OK;
|
|
}
|
|
|
|
int32_t ConfigurableFrameSizeEncoder::RegisterEncodeCompleteCallback(
|
|
EncodedImageCallback* callback) {
|
|
callback_ = callback;
|
|
return WEBRTC_VIDEO_CODEC_OK;
|
|
}
|
|
|
|
int32_t ConfigurableFrameSizeEncoder::Release() {
|
|
return WEBRTC_VIDEO_CODEC_OK;
|
|
}
|
|
|
|
void ConfigurableFrameSizeEncoder::SetRates(
|
|
const RateControlParameters& parameters) {}
|
|
|
|
int32_t ConfigurableFrameSizeEncoder::SetFrameSize(size_t size) {
|
|
RTC_DCHECK_LE(size, max_frame_size_);
|
|
current_frame_size_ = size;
|
|
return WEBRTC_VIDEO_CODEC_OK;
|
|
}
|
|
|
|
void ConfigurableFrameSizeEncoder::SetCodecType(VideoCodecType codec_type) {
|
|
codec_type_ = codec_type;
|
|
}
|
|
|
|
void ConfigurableFrameSizeEncoder::RegisterPostEncodeCallback(
|
|
std::function<void(void)> post_encode_callback) {
|
|
post_encode_callback_ = std::move(post_encode_callback);
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|