niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 1 | /* |
stefan@webrtc.org | f27916a | 2012-01-20 14:04:13 +0000 | [diff] [blame] | 2 | * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 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 | |
Henrik Kjellander | 2557b86 | 2015-11-18 22:00:21 +0100 | [diff] [blame] | 11 | #ifndef WEBRTC_MODULES_VIDEO_CODING_CODEC_DATABASE_H_ |
| 12 | #define WEBRTC_MODULES_VIDEO_CODING_CODEC_DATABASE_H_ |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 13 | |
stefan@webrtc.org | f27916a | 2012-01-20 14:04:13 +0000 | [diff] [blame] | 14 | #include <map> |
| 15 | |
kwiberg@webrtc.org | 00b8f6b | 2015-02-26 14:34:55 +0000 | [diff] [blame] | 16 | #include "webrtc/base/scoped_ptr.h" |
Henrik Kjellander | 2557b86 | 2015-11-18 22:00:21 +0100 | [diff] [blame] | 17 | #include "webrtc/modules/video_coding/include/video_codec_interface.h" |
| 18 | #include "webrtc/modules/video_coding/include/video_coding.h" |
| 19 | #include "webrtc/modules/video_coding/generic_decoder.h" |
| 20 | #include "webrtc/modules/video_coding/generic_encoder.h" |
pbos@webrtc.org | a440732 | 2013-07-16 12:32:05 +0000 | [diff] [blame] | 21 | #include "webrtc/typedefs.h" |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 22 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 23 | namespace webrtc { |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 24 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 25 | struct VCMDecoderMapItem { |
| 26 | public: |
| 27 | VCMDecoderMapItem(VideoCodec* settings, |
| 28 | int number_of_cores, |
| 29 | bool require_key_frame); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 30 | |
kwiberg@webrtc.org | 00b8f6b | 2015-02-26 14:34:55 +0000 | [diff] [blame] | 31 | rtc::scoped_ptr<VideoCodec> settings; |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 32 | int number_of_cores; |
| 33 | bool require_key_frame; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 34 | }; |
| 35 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 36 | struct VCMExtDecoderMapItem { |
| 37 | public: |
| 38 | VCMExtDecoderMapItem(VideoDecoder* external_decoder_instance, |
| 39 | uint8_t payload_type, |
| 40 | bool internal_render_timing); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 41 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 42 | uint8_t payload_type; |
| 43 | VideoDecoder* external_decoder_instance; |
| 44 | bool internal_render_timing; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 45 | }; |
| 46 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 47 | class VCMCodecDataBase { |
| 48 | public: |
Peter Boström | 4f5db11 | 2015-10-29 16:53:59 +0100 | [diff] [blame] | 49 | VCMCodecDataBase(VideoEncoderRateObserver* encoder_rate_observer, |
| 50 | VCMEncodedFrameCallback* encoded_frame_callback); |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 51 | ~VCMCodecDataBase(); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 52 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 53 | // Sender Side |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 54 | // Returns the default settings for the codec with type |codec_type|. |
Peter Boström | 92f8dbd | 2015-11-24 13:55:55 +0100 | [diff] [blame] | 55 | static void Codec(VideoCodecType codec_type, VideoCodec* settings); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 56 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 57 | void ResetSender(); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 58 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 59 | // Sets the sender side codec and initiates the desired codec given the |
| 60 | // VideoCodec struct. |
| 61 | // Returns true if the codec was successfully registered, false otherwise. |
pbos@webrtc.org | b9bb3d1 | 2013-05-16 18:40:48 +0000 | [diff] [blame] | 62 | bool SetSendCodec(const VideoCodec* send_codec, |
| 63 | int number_of_cores, |
Peter Boström | 4f5db11 | 2015-10-29 16:53:59 +0100 | [diff] [blame] | 64 | size_t max_payload_size); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 65 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 66 | // Gets the current send codec. Relevant for internal codecs only. |
| 67 | // Returns true if there is a send codec, false otherwise. |
| 68 | bool SendCodec(VideoCodec* current_send_codec) const; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 69 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 70 | // Gets current send side codec type. Relevant for internal codecs only. |
| 71 | // Returns kVideoCodecUnknown if there is no send codec. |
| 72 | VideoCodecType SendCodec() const; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 73 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 74 | // Registers and initializes an external encoder object. |
| 75 | // |internal_source| should be set to true if the codec has an internal |
| 76 | // video source and doesn't need the user to provide it with frames via |
| 77 | // the Encode() method. |
| 78 | void RegisterExternalEncoder(VideoEncoder* external_encoder, |
| 79 | uint8_t payload_type, |
| 80 | bool internal_source); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 81 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 82 | // Deregisters an external encoder. Returns true if the encoder was |
| 83 | // found and deregistered, false otherwise. |was_send_codec| is set to true |
| 84 | // if the external encoder was the send codec before being deregistered. |
| 85 | bool DeregisterExternalEncoder(uint8_t payload_type, bool* was_send_codec); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 86 | |
pbos@webrtc.org | b9bb3d1 | 2013-05-16 18:40:48 +0000 | [diff] [blame] | 87 | VCMGenericEncoder* GetEncoder(); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 88 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 89 | bool SetPeriodicKeyFrames(bool enable); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 90 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 91 | // Receiver Side |
| 92 | void ResetReceiver(); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 93 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 94 | // Deregisters an external decoder object specified by |payload_type|. |
| 95 | bool DeregisterExternalDecoder(uint8_t payload_type); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 96 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 97 | // Registers an external decoder object to the payload type |payload_type|. |
| 98 | // |internal_render_timing| is set to true if the |external_decoder| has |
| 99 | // built in rendering which is able to obey the render timestamps of the |
| 100 | // encoded frames. |
| 101 | bool RegisterExternalDecoder(VideoDecoder* external_decoder, |
| 102 | uint8_t payload_type, |
| 103 | bool internal_render_timing); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 104 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 105 | bool DecoderRegistered() const; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 106 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 107 | bool RegisterReceiveCodec(const VideoCodec* receive_codec, |
| 108 | int number_of_cores, |
| 109 | bool require_key_frame); |
| 110 | |
| 111 | bool DeregisterReceiveCodec(uint8_t payload_type); |
| 112 | |
| 113 | // Get current receive side codec. Relevant for internal codecs only. |
| 114 | bool ReceiveCodec(VideoCodec* current_receive_codec) const; |
| 115 | |
| 116 | // Get current receive side codec type. Relevant for internal codecs only. |
| 117 | VideoCodecType ReceiveCodec() const; |
| 118 | |
| 119 | // Returns a decoder specified by |payload_type|. The decoded frame callback |
| 120 | // of the encoder is set to |decoded_frame_callback|. If no such decoder |
| 121 | // already exists an instance will be created and initialized. |
| 122 | // NULL is returned if no encoder with the specified payload type was found |
| 123 | // and the function failed to create one. |
| 124 | VCMGenericDecoder* GetDecoder( |
| 125 | uint8_t payload_type, VCMDecodedFrameCallback* decoded_frame_callback); |
| 126 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 127 | // Deletes the memory of the decoder instance |decoder|. Used to delete |
| 128 | // deep copies returned by CreateDecoderCopy(). |
| 129 | void ReleaseDecoder(VCMGenericDecoder* decoder) const; |
| 130 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 131 | // Returns true if the currently active decoder supports render scheduling, |
| 132 | // that is, it is able to render frames according to the render timestamp of |
| 133 | // the encoded frames. |
| 134 | bool SupportsRenderScheduling() const; |
| 135 | |
pbos@webrtc.org | 67a9e40 | 2015-03-05 13:57:37 +0000 | [diff] [blame] | 136 | bool MatchesCurrentResolution(int width, int height) const; |
| 137 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 138 | private: |
| 139 | typedef std::map<uint8_t, VCMDecoderMapItem*> DecoderMap; |
| 140 | typedef std::map<uint8_t, VCMExtDecoderMapItem*> ExternalDecoderMap; |
| 141 | |
| 142 | VCMGenericDecoder* CreateAndInitDecoder(uint8_t payload_type, |
Stefan Holmer | 79064e5 | 2015-03-25 15:20:32 +0100 | [diff] [blame] | 143 | VideoCodec* new_codec) const; |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 144 | |
pbos@webrtc.org | b9bb3d1 | 2013-05-16 18:40:48 +0000 | [diff] [blame] | 145 | // Determines whether a new codec has to be created or not. |
| 146 | // Checks every setting apart from maxFramerate and startBitrate. |
| 147 | bool RequiresEncoderReset(const VideoCodec& send_codec); |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 148 | |
| 149 | void DeleteEncoder(); |
| 150 | |
| 151 | // Create an internal Decoder given a codec type |
| 152 | VCMGenericDecoder* CreateDecoder(VideoCodecType type) const; |
| 153 | |
| 154 | const VCMDecoderMapItem* FindDecoderItem(uint8_t payload_type) const; |
| 155 | |
| 156 | const VCMExtDecoderMapItem* FindExternalDecoderItem( |
| 157 | uint8_t payload_type) const; |
| 158 | |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 159 | int number_of_cores_; |
pkasting@chromium.org | 4591fbd | 2014-11-20 22:28:14 +0000 | [diff] [blame] | 160 | size_t max_payload_size_; |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 161 | bool periodic_key_frames_; |
pbos@webrtc.org | b9bb3d1 | 2013-05-16 18:40:48 +0000 | [diff] [blame] | 162 | bool pending_encoder_reset_; |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 163 | VideoCodec send_codec_; |
| 164 | VideoCodec receive_codec_; |
Peter Boström | ab73d13 | 2015-10-15 12:01:38 +0200 | [diff] [blame] | 165 | uint8_t encoder_payload_type_; |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 166 | VideoEncoder* external_encoder_; |
| 167 | bool internal_source_; |
pbos@webrtc.org | 891d483 | 2015-02-26 13:15:22 +0000 | [diff] [blame] | 168 | VideoEncoderRateObserver* const encoder_rate_observer_; |
Peter Boström | 4f5db11 | 2015-10-29 16:53:59 +0100 | [diff] [blame] | 169 | VCMEncodedFrameCallback* const encoded_frame_callback_; |
Peter Boström | ab73d13 | 2015-10-15 12:01:38 +0200 | [diff] [blame] | 170 | rtc::scoped_ptr<VCMGenericEncoder> ptr_encoder_; |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 171 | VCMGenericDecoder* ptr_decoder_; |
stefan@webrtc.org | fa7e868 | 2012-10-11 11:21:38 +0000 | [diff] [blame] | 172 | DecoderMap dec_map_; |
| 173 | ExternalDecoderMap dec_external_map_; |
| 174 | }; // VCMCodecDataBase |
| 175 | |
| 176 | } // namespace webrtc |
| 177 | |
Henrik Kjellander | 2557b86 | 2015-11-18 22:00:21 +0100 | [diff] [blame] | 178 | #endif // WEBRTC_MODULES_VIDEO_CODING_CODEC_DATABASE_H_ |