| Index: webrtc/sdk/objc/Framework/Classes/VideoToolbox/encoder.h
|
| diff --git a/webrtc/sdk/objc/Framework/Classes/VideoToolbox/encoder.h b/webrtc/sdk/objc/Framework/Classes/VideoToolbox/encoder.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..92cd6b7d011c9455e70f9bab10d767fa250e3ffc
|
| --- /dev/null
|
| +++ b/webrtc/sdk/objc/Framework/Classes/VideoToolbox/encoder.h
|
| @@ -0,0 +1,97 @@
|
| +/*
|
| + * 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_
|
|
|