| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * Copyright (c) 2014 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 #ifndef WEBRTC_VIDEO_ENCODER_H_ | |
| 12 #define WEBRTC_VIDEO_ENCODER_H_ | |
| 13 | |
| 14 #include <memory> | |
| 15 #include <string> | |
| 16 #include <vector> | |
| 17 | |
| 18 #include "webrtc/base/checks.h" | |
| 19 #include "webrtc/common_types.h" | |
| 20 #include "webrtc/typedefs.h" | |
| 21 #include "webrtc/video_frame.h" | |
| 22 #include "webrtc/base/optional.h" | |
| 23 | |
| 24 namespace webrtc { | |
| 25 | |
| 26 class RTPFragmentationHeader; | |
| 27 // TODO(pbos): Expose these through a public (root) header or change these APIs. | |
| 28 struct CodecSpecificInfo; | |
| 29 class VideoCodec; | |
| 30 | |
| 31 class EncodedImageCallback { | |
| 32 public: | |
| 33 virtual ~EncodedImageCallback() {} | |
| 34 | |
| 35 struct Result { | |
| 36 enum Error { | |
| 37 OK, | |
| 38 | |
| 39 // Failed to send the packet. | |
| 40 ERROR_SEND_FAILED, | |
| 41 }; | |
| 42 | |
| 43 Result(Error error) : error(error) {} | |
| 44 Result(Error error, uint32_t frame_id) : error(error), frame_id(frame_id) {} | |
| 45 | |
| 46 Error error; | |
| 47 | |
| 48 // Frame ID assigned to the frame. The frame ID should be the same as the ID | |
| 49 // seen by the receiver for this frame. RTP timestamp of the frame is used | |
| 50 // as frame ID when RTP is used to send video. Must be used only when | |
| 51 // error=OK. | |
| 52 uint32_t frame_id = 0; | |
| 53 | |
| 54 // Tells the encoder that the next frame is should be dropped. | |
| 55 bool drop_next_frame = false; | |
| 56 }; | |
| 57 | |
| 58 // Callback function which is called when an image has been encoded. | |
| 59 virtual Result OnEncodedImage( | |
| 60 const EncodedImage& encoded_image, | |
| 61 const CodecSpecificInfo* codec_specific_info, | |
| 62 const RTPFragmentationHeader* fragmentation) = 0; | |
| 63 | |
| 64 virtual void OnDroppedFrame() {} | |
| 65 }; | |
| 66 | |
| 67 class VideoEncoder { | |
| 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 | |
| 95 static VideoCodecVP8 GetDefaultVp8Settings(); | |
| 96 static VideoCodecVP9 GetDefaultVp9Settings(); | |
| 97 static VideoCodecH264 GetDefaultH264Settings(); | |
| 98 | |
| 99 virtual ~VideoEncoder() {} | |
| 100 | |
| 101 // Initialize the encoder with the information from the codecSettings | |
| 102 // | |
| 103 // Input: | |
| 104 // - codec_settings : Codec settings | |
| 105 // - number_of_cores : Number of cores available for the encoder | |
| 106 // - max_payload_size : The maximum size each payload is allowed | |
| 107 // to have. Usually MTU - overhead. | |
| 108 // | |
| 109 // Return value : Set bit rate if OK | |
| 110 // <0 - Errors: | |
| 111 // WEBRTC_VIDEO_CODEC_ERR_PARAMETER | |
| 112 // WEBRTC_VIDEO_CODEC_ERR_SIZE | |
| 113 // WEBRTC_VIDEO_CODEC_LEVEL_EXCEEDED | |
| 114 // WEBRTC_VIDEO_CODEC_MEMORY | |
| 115 // WEBRTC_VIDEO_CODEC_ERROR | |
| 116 virtual int32_t InitEncode(const VideoCodec* codec_settings, | |
| 117 int32_t number_of_cores, | |
| 118 size_t max_payload_size) = 0; | |
| 119 | |
| 120 // Register an encode complete callback object. | |
| 121 // | |
| 122 // Input: | |
| 123 // - callback : Callback object which handles encoded images. | |
| 124 // | |
| 125 // Return value : WEBRTC_VIDEO_CODEC_OK if OK, < 0 otherwise. | |
| 126 virtual int32_t RegisterEncodeCompleteCallback( | |
| 127 EncodedImageCallback* callback) = 0; | |
| 128 | |
| 129 // Free encoder memory. | |
| 130 // Return value : WEBRTC_VIDEO_CODEC_OK if OK, < 0 otherwise. | |
| 131 virtual int32_t Release() = 0; | |
| 132 | |
| 133 // Encode an I420 image (as a part of a video stream). The encoded image | |
| 134 // will be returned to the user through the encode complete callback. | |
| 135 // | |
| 136 // Input: | |
| 137 // - frame : Image to be encoded | |
| 138 // - frame_types : Frame type to be generated by the encoder. | |
| 139 // | |
| 140 // Return value : WEBRTC_VIDEO_CODEC_OK if OK | |
| 141 // <0 - Errors: | |
| 142 // WEBRTC_VIDEO_CODEC_ERR_PARAMETER | |
| 143 // WEBRTC_VIDEO_CODEC_MEMORY | |
| 144 // WEBRTC_VIDEO_CODEC_ERROR | |
| 145 // WEBRTC_VIDEO_CODEC_TIMEOUT | |
| 146 virtual int32_t Encode(const VideoFrame& frame, | |
| 147 const CodecSpecificInfo* codec_specific_info, | |
| 148 const std::vector<FrameType>* frame_types) = 0; | |
| 149 | |
| 150 // Inform the encoder of the new packet loss rate and the round-trip time of | |
| 151 // the network. | |
| 152 // | |
| 153 // Input: | |
| 154 // - packet_loss : Fraction lost | |
| 155 // (loss rate in percent = 100 * packetLoss / 255) | |
| 156 // - rtt : Round-trip time in milliseconds | |
| 157 // Return value : WEBRTC_VIDEO_CODEC_OK if OK | |
| 158 // <0 - Errors: WEBRTC_VIDEO_CODEC_ERROR | |
| 159 virtual int32_t SetChannelParameters(uint32_t packet_loss, int64_t rtt) = 0; | |
| 160 | |
| 161 // Inform the encoder about the new target bit rate. | |
| 162 // | |
| 163 // Input: | |
| 164 // - bitrate : New target bit rate | |
| 165 // - framerate : The target frame rate | |
| 166 // | |
| 167 // Return value : WEBRTC_VIDEO_CODEC_OK if OK, < 0 otherwise. | |
| 168 virtual int32_t SetRates(uint32_t bitrate, uint32_t framerate) { | |
| 169 RTC_NOTREACHED() << "SetRate(uint32_t, uint32_t) is deprecated."; | |
| 170 return -1; | |
| 171 } | |
| 172 | |
| 173 // Default fallback: Just use the sum of bitrates as the single target rate. | |
| 174 // TODO(sprang): Remove this default implementation when we remove SetRates(). | |
| 175 virtual int32_t SetRateAllocation(const BitrateAllocation& allocation, | |
| 176 uint32_t framerate) { | |
| 177 return SetRates(allocation.get_sum_kbps(), framerate); | |
| 178 } | |
| 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 | |
| 186 virtual int32_t SetPeriodicKeyFrames(bool enable) { return -1; } | |
| 187 virtual bool SupportsNativeHandle() const { return false; } | |
| 188 virtual const char* ImplementationName() const { return "unknown"; } | |
| 189 }; | |
| 190 | |
| 191 } // namespace webrtc | |
| 192 #endif // WEBRTC_VIDEO_ENCODER_H_ | |
| OLD | NEW |