OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef WEBRTC_VIDEO_ENCODER_H_ | 11 #ifndef WEBRTC_VIDEO_ENCODER_H_ |
12 #define WEBRTC_VIDEO_ENCODER_H_ | 12 #define WEBRTC_VIDEO_ENCODER_H_ |
13 | 13 |
14 #include <memory> | 14 #include <memory> |
15 #include <string> | 15 #include <string> |
16 #include <vector> | 16 #include <vector> |
17 | 17 |
18 #include "webrtc/base/checks.h" | 18 #include "webrtc/base/checks.h" |
19 #include "webrtc/common_types.h" | 19 #include "webrtc/common_types.h" |
20 #include "webrtc/typedefs.h" | 20 #include "webrtc/typedefs.h" |
21 #include "webrtc/video_frame.h" | 21 #include "webrtc/video_frame.h" |
| 22 #include "webrtc/base/optional.h" |
22 | 23 |
23 namespace webrtc { | 24 namespace webrtc { |
24 | 25 |
25 class RTPFragmentationHeader; | 26 class RTPFragmentationHeader; |
26 // TODO(pbos): Expose these through a public (root) header or change these APIs. | 27 // TODO(pbos): Expose these through a public (root) header or change these APIs. |
27 struct CodecSpecificInfo; | 28 struct CodecSpecificInfo; |
28 class VideoCodec; | 29 class VideoCodec; |
29 | 30 |
30 class EncodedImageCallback { | 31 class EncodedImageCallback { |
31 public: | 32 public: |
(...skipping 20 matching lines...) Expand all Loading... |
52 | 53 |
53 // Tells the encoder that the next frame is should be dropped. | 54 // Tells the encoder that the next frame is should be dropped. |
54 bool drop_next_frame = false; | 55 bool drop_next_frame = false; |
55 }; | 56 }; |
56 | 57 |
57 // Callback function which is called when an image has been encoded. | 58 // Callback function which is called when an image has been encoded. |
58 virtual Result OnEncodedImage( | 59 virtual Result OnEncodedImage( |
59 const EncodedImage& encoded_image, | 60 const EncodedImage& encoded_image, |
60 const CodecSpecificInfo* codec_specific_info, | 61 const CodecSpecificInfo* codec_specific_info, |
61 const RTPFragmentationHeader* fragmentation) = 0; | 62 const RTPFragmentationHeader* fragmentation) = 0; |
| 63 |
| 64 virtual void OnDroppedFrame() {} |
62 }; | 65 }; |
63 | 66 |
64 class VideoEncoder { | 67 class VideoEncoder { |
65 public: | 68 public: |
| 69 enum EncoderType { |
| 70 kH264, |
| 71 kVp8, |
| 72 kVp9, |
| 73 kUnsupportedCodec, |
| 74 }; |
| 75 struct QpThresholds { |
| 76 QpThresholds(int l, int h) : low(l), high(h) {} |
| 77 QpThresholds() : low(-1), high(-1) {} |
| 78 int low; |
| 79 int high; |
| 80 }; |
| 81 struct ScalingSettings { |
| 82 ScalingSettings(bool on, int low, int high) |
| 83 : enabled(on), |
| 84 thresholds(rtc::Optional<QpThresholds>(QpThresholds(low, high))) {} |
| 85 explicit ScalingSettings(bool on) : enabled(on) {} |
| 86 const bool enabled; |
| 87 const rtc::Optional<QpThresholds> thresholds; |
| 88 }; |
| 89 static VideoEncoder* Create(EncoderType codec_type); |
| 90 // Returns true if this type of encoder can be created using |
| 91 // VideoEncoder::Create. |
| 92 static bool IsSupportedSoftware(EncoderType codec_type); |
| 93 static EncoderType CodecToEncoderType(VideoCodecType codec_type); |
| 94 |
66 static VideoCodecVP8 GetDefaultVp8Settings(); | 95 static VideoCodecVP8 GetDefaultVp8Settings(); |
67 static VideoCodecVP9 GetDefaultVp9Settings(); | 96 static VideoCodecVP9 GetDefaultVp9Settings(); |
68 static VideoCodecH264 GetDefaultH264Settings(); | 97 static VideoCodecH264 GetDefaultH264Settings(); |
69 | 98 |
70 virtual ~VideoEncoder() {} | 99 virtual ~VideoEncoder() {} |
71 | 100 |
72 // Initialize the encoder with the information from the codecSettings | 101 // Initialize the encoder with the information from the codecSettings |
73 // | 102 // |
74 // Input: | 103 // Input: |
75 // - codec_settings : Codec settings | 104 // - codec_settings : Codec settings |
(...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
141 return -1; | 170 return -1; |
142 } | 171 } |
143 | 172 |
144 // Default fallback: Just use the sum of bitrates as the single target rate. | 173 // Default fallback: Just use the sum of bitrates as the single target rate. |
145 // TODO(sprang): Remove this default implementation when we remove SetRates(). | 174 // TODO(sprang): Remove this default implementation when we remove SetRates(). |
146 virtual int32_t SetRateAllocation(const BitrateAllocation& allocation, | 175 virtual int32_t SetRateAllocation(const BitrateAllocation& allocation, |
147 uint32_t framerate) { | 176 uint32_t framerate) { |
148 return SetRates(allocation.get_sum_kbps(), framerate); | 177 return SetRates(allocation.get_sum_kbps(), framerate); |
149 } | 178 } |
150 | 179 |
| 180 // Any encoder implementation wishing to use the WebRTC provided |
| 181 // quality scaler must implement this method. |
| 182 virtual ScalingSettings GetScalingSettings() const { |
| 183 return ScalingSettings(false); |
| 184 } |
| 185 |
151 virtual int32_t SetPeriodicKeyFrames(bool enable) { return -1; } | 186 virtual int32_t SetPeriodicKeyFrames(bool enable) { return -1; } |
152 virtual void OnDroppedFrame() {} | |
153 virtual bool SupportsNativeHandle() const { return false; } | 187 virtual bool SupportsNativeHandle() const { return false; } |
154 virtual const char* ImplementationName() const { return "unknown"; } | 188 virtual const char* ImplementationName() const { return "unknown"; } |
155 }; | 189 }; |
156 | 190 |
157 } // namespace webrtc | 191 } // namespace webrtc |
158 #endif // WEBRTC_VIDEO_ENCODER_H_ | 192 #endif // WEBRTC_VIDEO_ENCODER_H_ |
OLD | NEW |