a80c16a67c
This reverts commit c3771cc4d37f5573fe53b7c7cff295a4f0f9560f. (breaks downstream internal project) BUG=webrtc:7634 NOTRY=True NOPRESUBMIT=True Review-Url: https://codereview.webrtc.org/2972463002 . Cr-Commit-Position: refs/heads/master@{#18873}
98 lines
3.4 KiB
C++
98 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) 2015 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 WEBRTC_SDK_OBJC_FRAMEWORK_CLASSES_VIDEOTOOLBOX_ENCODER_H_
|
|
#define WEBRTC_SDK_OBJC_FRAMEWORK_CLASSES_VIDEOTOOLBOX_ENCODER_H_
|
|
|
|
#include "webrtc/api/video/video_rotation.h"
|
|
#include "webrtc/base/criticalsection.h"
|
|
#include "webrtc/common_video/h264/h264_bitstream_parser.h"
|
|
#include "webrtc/common_video/include/bitrate_adjuster.h"
|
|
#include "webrtc/media/base/codec.h"
|
|
#include "webrtc/modules/video_coding/codecs/h264/include/h264.h"
|
|
#include "webrtc/modules/video_coding/utility/quality_scaler.h"
|
|
|
|
#include <VideoToolbox/VideoToolbox.h>
|
|
#include <vector>
|
|
|
|
// This file provides a H264 encoder implementation using the VideoToolbox
|
|
// APIs. Since documentation is almost non-existent, this is largely based on
|
|
// the information in the VideoToolbox header files, a talk from WWDC 2014 and
|
|
// experimentation.
|
|
|
|
namespace webrtc {
|
|
|
|
class H264VideoToolboxEncoder : public H264Encoder {
|
|
public:
|
|
explicit H264VideoToolboxEncoder(const cricket::VideoCodec& codec);
|
|
|
|
~H264VideoToolboxEncoder() override;
|
|
|
|
int InitEncode(const VideoCodec* codec_settings,
|
|
int number_of_cores,
|
|
size_t max_payload_size) override;
|
|
|
|
int Encode(const VideoFrame& input_image,
|
|
const CodecSpecificInfo* codec_specific_info,
|
|
const std::vector<FrameType>* frame_types) override;
|
|
|
|
int RegisterEncodeCompleteCallback(EncodedImageCallback* callback) override;
|
|
|
|
int SetChannelParameters(uint32_t packet_loss, int64_t rtt) override;
|
|
|
|
int SetRates(uint32_t new_bitrate_kbit, uint32_t frame_rate) override;
|
|
|
|
int Release() override;
|
|
|
|
const char* ImplementationName() const override;
|
|
|
|
bool SupportsNativeHandle() const override;
|
|
|
|
void OnEncodedFrame(OSStatus status,
|
|
VTEncodeInfoFlags info_flags,
|
|
CMSampleBufferRef sample_buffer,
|
|
CodecSpecificInfo codec_specific_info,
|
|
int32_t width,
|
|
int32_t height,
|
|
int64_t render_time_ms,
|
|
uint32_t timestamp,
|
|
VideoRotation rotation);
|
|
|
|
ScalingSettings GetScalingSettings() const override;
|
|
|
|
private:
|
|
int ResetCompressionSession();
|
|
void ConfigureCompressionSession();
|
|
void DestroyCompressionSession();
|
|
rtc::scoped_refptr<VideoFrameBuffer> GetScaledBufferOnEncode(
|
|
const rtc::scoped_refptr<VideoFrameBuffer>& frame);
|
|
void SetBitrateBps(uint32_t bitrate_bps);
|
|
void SetEncoderBitrateBps(uint32_t bitrate_bps);
|
|
|
|
EncodedImageCallback* callback_;
|
|
VTCompressionSessionRef compression_session_;
|
|
BitrateAdjuster bitrate_adjuster_;
|
|
H264PacketizationMode packetization_mode_;
|
|
uint32_t target_bitrate_bps_;
|
|
uint32_t encoder_bitrate_bps_;
|
|
int32_t width_;
|
|
int32_t height_;
|
|
VideoCodecMode mode_;
|
|
const CFStringRef profile_;
|
|
|
|
H264BitstreamParser h264_bitstream_parser_;
|
|
std::vector<uint8_t> nv12_scale_buffer_;
|
|
}; // H264VideoToolboxEncoder
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // WEBRTC_SDK_OBJC_FRAMEWORK_CLASSES_VIDEOTOOLBOX_ENCODER_H_
|