1 /* 2 * libjingle 3 * Copyright 2010 Google Inc. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright notice, 9 * this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright notice, 11 * this list of conditions and the following disclaimer in the documentation 12 * and/or other materials provided with the distribution. 13 * 3. The name of the author may not be used to endorse or promote products 14 * derived from this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28 #ifndef TALK_MEDIA_WEBRTC_FAKEWEBRTCVIDEOENGINE_H_ 29 #define TALK_MEDIA_WEBRTC_FAKEWEBRTCVIDEOENGINE_H_ 30 31 #include <map> 32 #include <set> 33 #include <vector> 34 35 #include "talk/media/base/codec.h" 36 #include "talk/media/webrtc/fakewebrtccommon.h" 37 #include "talk/media/webrtc/webrtcvideodecoderfactory.h" 38 #include "talk/media/webrtc/webrtcvideoencoderfactory.h" 39 #include "webrtc/base/basictypes.h" 40 #include "webrtc/base/criticalsection.h" 41 #include "webrtc/base/gunit.h" 42 #include "webrtc/base/stringutils.h" 43 #include "webrtc/base/thread_annotations.h" 44 #include "webrtc/modules/video_coding/include/video_error_codes.h" 45 #include "webrtc/video_decoder.h" 46 #include "webrtc/video_encoder.h" 47 48 namespace cricket { 49 50 static const int kMinVideoBitrate = 100; 51 static const int kStartVideoBitrate = 300; 52 static const int kMaxVideoBitrate = 1000; 53 54 // WebRtc channel id and capture id share the same number space. 55 // This is how AddRenderer(renderId, ...) is able to tell if it is adding a 56 // renderer for a channel or it is adding a renderer for a capturer. 57 static const int kViEChannelIdBase = 0; 58 static const int kViEChannelIdMax = 1000; 59 60 // Fake class for mocking out webrtc::VideoDecoder 61 class FakeWebRtcVideoDecoder : public webrtc::VideoDecoder { 62 public: 63 FakeWebRtcVideoDecoder() 64 : num_frames_received_(0) { 65 } 66 67 virtual int32_t InitDecode(const webrtc::VideoCodec*, int32_t) { 68 return WEBRTC_VIDEO_CODEC_OK; 69 } 70 71 virtual int32_t Decode(const webrtc::EncodedImage&, 72 bool, 73 const webrtc::RTPFragmentationHeader*, 74 const webrtc::CodecSpecificInfo*, 75 int64_t) { 76 num_frames_received_++; 77 return WEBRTC_VIDEO_CODEC_OK; 78 } 79 80 virtual int32_t RegisterDecodeCompleteCallback( 81 webrtc::DecodedImageCallback*) { 82 return WEBRTC_VIDEO_CODEC_OK; 83 } 84 85 virtual int32_t Release() { return WEBRTC_VIDEO_CODEC_OK; } 86 87 virtual int32_t Reset() { return WEBRTC_VIDEO_CODEC_OK; } 88 89 int GetNumFramesReceived() const { 90 return num_frames_received_; 91 } 92 93 private: 94 int num_frames_received_; 95 }; 96 97 // Fake class for mocking out WebRtcVideoDecoderFactory. 98 class FakeWebRtcVideoDecoderFactory : public WebRtcVideoDecoderFactory { 99 public: 100 FakeWebRtcVideoDecoderFactory() 101 : num_created_decoders_(0) { 102 } 103 104 virtual webrtc::VideoDecoder* CreateVideoDecoder( 105 webrtc::VideoCodecType type) { 106 if (supported_codec_types_.count(type) == 0) { 107 return NULL; 108 } 109 FakeWebRtcVideoDecoder* decoder = new FakeWebRtcVideoDecoder(); 110 decoders_.push_back(decoder); 111 num_created_decoders_++; 112 return decoder; 113 } 114 115 virtual void DestroyVideoDecoder(webrtc::VideoDecoder* decoder) { 116 decoders_.erase( 117 std::remove(decoders_.begin(), decoders_.end(), decoder), 118 decoders_.end()); 119 delete decoder; 120 } 121 122 void AddSupportedVideoCodecType(webrtc::VideoCodecType type) { 123 supported_codec_types_.insert(type); 124 } 125 126 int GetNumCreatedDecoders() { 127 return num_created_decoders_; 128 } 129 130 const std::vector<FakeWebRtcVideoDecoder*>& decoders() { 131 return decoders_; 132 } 133 134 private: 135 std::set<webrtc::VideoCodecType> supported_codec_types_; 136 std::vector<FakeWebRtcVideoDecoder*> decoders_; 137 int num_created_decoders_; 138 }; 139 140 // Fake class for mocking out webrtc::VideoEnoder 141 class FakeWebRtcVideoEncoder : public webrtc::VideoEncoder { 142 public: 143 FakeWebRtcVideoEncoder() : num_frames_encoded_(0) {} 144 145 virtual int32_t InitEncode(const webrtc::VideoCodec* codecSettings, 146 int32_t numberOfCores, 147 size_t maxPayloadSize) { 148 rtc::CritScope lock(&crit_); 149 codec_settings_ = *codecSettings; 150 return WEBRTC_VIDEO_CODEC_OK; 151 } 152 153 webrtc::VideoCodec GetCodecSettings() { 154 rtc::CritScope lock(&crit_); 155 return codec_settings_; 156 } 157 158 virtual int32_t Encode(const webrtc::VideoFrame& inputImage, 159 const webrtc::CodecSpecificInfo* codecSpecificInfo, 160 const std::vector<webrtc::FrameType>* frame_types) { 161 rtc::CritScope lock(&crit_); 162 ++num_frames_encoded_; 163 return WEBRTC_VIDEO_CODEC_OK; 164 } 165 166 virtual int32_t RegisterEncodeCompleteCallback( 167 webrtc::EncodedImageCallback* callback) { 168 return WEBRTC_VIDEO_CODEC_OK; 169 } 170 171 virtual int32_t Release() { return WEBRTC_VIDEO_CODEC_OK; } 172 173 virtual int32_t SetChannelParameters(uint32_t packetLoss, int64_t rtt) { 174 return WEBRTC_VIDEO_CODEC_OK; 175 } 176 177 virtual int32_t SetRates(uint32_t newBitRate, uint32_t frameRate) { 178 return WEBRTC_VIDEO_CODEC_OK; 179 } 180 181 int GetNumEncodedFrames() { 182 rtc::CritScope lock(&crit_); 183 return num_frames_encoded_; 184 } 185 186 private: 187 rtc::CriticalSection crit_; 188 int num_frames_encoded_ GUARDED_BY(crit_); 189 webrtc::VideoCodec codec_settings_ GUARDED_BY(crit_); 190 }; 191 192 // Fake class for mocking out WebRtcVideoEncoderFactory. 193 class FakeWebRtcVideoEncoderFactory : public WebRtcVideoEncoderFactory { 194 public: 195 FakeWebRtcVideoEncoderFactory() 196 : num_created_encoders_(0), encoders_have_internal_sources_(false) {} 197 198 virtual webrtc::VideoEncoder* CreateVideoEncoder( 199 webrtc::VideoCodecType type) { 200 if (supported_codec_types_.count(type) == 0) { 201 return NULL; 202 } 203 FakeWebRtcVideoEncoder* encoder = new FakeWebRtcVideoEncoder(); 204 encoders_.push_back(encoder); 205 num_created_encoders_++; 206 return encoder; 207 } 208 209 virtual void DestroyVideoEncoder(webrtc::VideoEncoder* encoder) { 210 encoders_.erase( 211 std::remove(encoders_.begin(), encoders_.end(), encoder), 212 encoders_.end()); 213 delete encoder; 214 } 215 216 virtual const std::vector<WebRtcVideoEncoderFactory::VideoCodec>& codecs() 217 const { 218 return codecs_; 219 } 220 221 virtual bool EncoderTypeHasInternalSource( 222 webrtc::VideoCodecType type) const override { 223 return encoders_have_internal_sources_; 224 } 225 226 void set_encoders_have_internal_sources(bool internal_source) { 227 encoders_have_internal_sources_ = internal_source; 228 } 229 230 void AddSupportedVideoCodecType(webrtc::VideoCodecType type, 231 const std::string& name) { 232 supported_codec_types_.insert(type); 233 codecs_.push_back( 234 WebRtcVideoEncoderFactory::VideoCodec(type, name, 1280, 720, 30)); 235 } 236 237 int GetNumCreatedEncoders() { 238 return num_created_encoders_; 239 } 240 241 const std::vector<FakeWebRtcVideoEncoder*>& encoders() { 242 return encoders_; 243 } 244 245 private: 246 std::set<webrtc::VideoCodecType> supported_codec_types_; 247 std::vector<WebRtcVideoEncoderFactory::VideoCodec> codecs_; 248 std::vector<FakeWebRtcVideoEncoder*> encoders_; 249 int num_created_encoders_; 250 bool encoders_have_internal_sources_; 251 }; 252 253 } // namespace cricket 254 255 #endif // TALK_MEDIA_WEBRTC_FAKEWEBRTCVIDEOENGINE_H_ 256