OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | |
3 * | |
4 * Use of this source code is governed by a BSD-style license | |
5 * that can be found in the LICENSE file in the root of the source | |
6 * tree. An additional intellectual property rights grant can be found | |
7 * in the file PATENTS. All contributing project authors may | |
8 * be found in the AUTHORS file in the root of the source tree. | |
9 * | |
10 */ | |
11 | |
12 #ifndef WEBRTC_SDK_OBJC_FRAMEWORK_CLASSES_H264_VIDEO_TOOLBOX_ENCODER_H_ | |
13 #define WEBRTC_SDK_OBJC_FRAMEWORK_CLASSES_H264_VIDEO_TOOLBOX_ENCODER_H_ | |
14 | |
15 #include "webrtc/base/criticalsection.h" | |
16 #include "webrtc/common_video/h264/h264_bitstream_parser.h" | |
17 #include "webrtc/common_video/include/bitrate_adjuster.h" | |
18 #include "webrtc/common_video/rotation.h" | |
19 #include "webrtc/modules/video_coding/codecs/h264/include/h264.h" | |
20 #include "webrtc/modules/video_coding/utility/quality_scaler.h" | |
21 | |
22 #include <VideoToolbox/VideoToolbox.h> | |
23 #include <vector> | |
24 | |
25 // This file provides a H264 encoder implementation using the VideoToolbox | |
26 // APIs. Since documentation is almost non-existent, this is largely based on | |
27 // the information in the VideoToolbox header files, a talk from WWDC 2014 and | |
28 // experimentation. | |
29 | |
30 namespace webrtc { | |
31 | |
32 class H264VideoToolboxEncoder : public H264Encoder { | |
33 public: | |
34 H264VideoToolboxEncoder(); | |
35 | |
36 ~H264VideoToolboxEncoder() override; | |
37 | |
38 int InitEncode(const VideoCodec* codec_settings, | |
39 int number_of_cores, | |
40 size_t max_payload_size) override; | |
41 | |
42 int Encode(const VideoFrame& input_image, | |
43 const CodecSpecificInfo* codec_specific_info, | |
44 const std::vector<FrameType>* frame_types) override; | |
45 | |
46 int RegisterEncodeCompleteCallback(EncodedImageCallback* callback) override; | |
47 | |
48 void OnDroppedFrame() override; | |
49 int SetChannelParameters(uint32_t packet_loss, int64_t rtt) override; | |
50 | |
51 int SetRates(uint32_t new_bitrate_kbit, uint32_t frame_rate) override; | |
52 | |
53 int Release() override; | |
54 | |
55 const char* ImplementationName() const override; | |
56 | |
57 bool SupportsNativeHandle() const override; | |
58 | |
59 void OnEncodedFrame(OSStatus status, | |
60 VTEncodeInfoFlags info_flags, | |
61 CMSampleBufferRef sample_buffer, | |
62 CodecSpecificInfo codec_specific_info, | |
63 int32_t width, | |
64 int32_t height, | |
65 int64_t render_time_ms, | |
66 uint32_t timestamp, | |
67 VideoRotation rotation); | |
68 | |
69 private: | |
70 int ResetCompressionSession(); | |
71 void ConfigureCompressionSession(); | |
72 void DestroyCompressionSession(); | |
73 rtc::scoped_refptr<VideoFrameBuffer> GetScaledBufferOnEncode( | |
74 const rtc::scoped_refptr<VideoFrameBuffer>& frame); | |
75 void SetBitrateBps(uint32_t bitrate_bps); | |
76 void SetEncoderBitrateBps(uint32_t bitrate_bps); | |
77 | |
78 EncodedImageCallback* callback_; | |
79 VTCompressionSessionRef compression_session_; | |
80 BitrateAdjuster bitrate_adjuster_; | |
81 uint32_t target_bitrate_bps_; | |
82 uint32_t encoder_bitrate_bps_; | |
83 int32_t width_; | |
84 int32_t height_; | |
85 | |
86 rtc::CriticalSection quality_scaler_crit_; | |
87 QualityScaler quality_scaler_ GUARDED_BY(quality_scaler_crit_); | |
88 H264BitstreamParser h264_bitstream_parser_; | |
89 bool enable_scaling_; | |
90 std::vector<uint8_t> nv12_scale_buffer_; | |
91 }; // H264VideoToolboxEncoder | |
92 | |
93 } // namespace webrtc | |
94 | |
95 #endif // WEBRTC_SDK_OBJC_FRAMEWORK_CLASSES_H264_VIDEO_TOOLBOX_ENCODER_H_ | |
OLD | NEW |