1 /* 2 * libjingle 3 * Copyright 2004 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 29 #ifndef TALK_MEDIA_WEBRTCVIE_H_ 30 #define TALK_MEDIA_WEBRTCVIE_H_ 31 32 #include "talk/media/webrtc/webrtccommon.h" 33 #include "webrtc/base/common.h" 34 #include "webrtc/common_types.h" 35 #include "webrtc/modules/interface/module_common_types.h" 36 #include "webrtc/modules/video_capture/include/video_capture.h" 37 #include "webrtc/modules/video_coding/codecs/interface/video_codec_interface.h" 38 #include "webrtc/modules/video_render/include/video_render.h" 39 #include "webrtc/video_engine/include/vie_base.h" 40 #include "webrtc/video_engine/include/vie_capture.h" 41 #include "webrtc/video_engine/include/vie_codec.h" 42 #include "webrtc/video_engine/include/vie_errors.h" 43 #include "webrtc/video_engine/include/vie_external_codec.h" 44 #include "webrtc/video_engine/include/vie_image_process.h" 45 #include "webrtc/video_engine/include/vie_network.h" 46 #include "webrtc/video_engine/include/vie_render.h" 47 #include "webrtc/video_engine/include/vie_rtp_rtcp.h" 48 49 namespace cricket { 50 51 // all tracing macros should go to a common file 52 53 // automatically handles lifetime of VideoEngine 54 class scoped_vie_engine { 55 public: scoped_vie_engine(webrtc::VideoEngine * e)56 explicit scoped_vie_engine(webrtc::VideoEngine* e) : ptr(e) {} 57 // VERIFY, to ensure that there are no leaks at shutdown ~scoped_vie_engine()58 ~scoped_vie_engine() { 59 if (ptr) { 60 webrtc::VideoEngine::Delete(ptr); 61 } 62 } get()63 webrtc::VideoEngine* get() const { return ptr; } 64 private: 65 webrtc::VideoEngine* ptr; 66 }; 67 68 // scoped_ptr class to handle obtaining and releasing VideoEngine 69 // interface pointers 70 template<class T> class scoped_vie_ptr { 71 public: scoped_vie_ptr(const scoped_vie_engine & e)72 explicit scoped_vie_ptr(const scoped_vie_engine& e) 73 : ptr(T::GetInterface(e.get())) {} scoped_vie_ptr(T * p)74 explicit scoped_vie_ptr(T* p) : ptr(p) {} ~scoped_vie_ptr()75 ~scoped_vie_ptr() { if (ptr) ptr->Release(); } 76 T* operator->() const { return ptr; } get()77 T* get() const { return ptr; } 78 private: 79 T* ptr; 80 }; 81 82 // Utility class for aggregating the various WebRTC interface. 83 // Fake implementations can also be injected for testing. 84 class ViEWrapper { 85 public: ViEWrapper()86 ViEWrapper() 87 : engine_(webrtc::VideoEngine::Create()), 88 base_(engine_), codec_(engine_), capture_(engine_), 89 network_(engine_), render_(engine_), rtp_(engine_), 90 image_(engine_), ext_codec_(engine_) { 91 } 92 ViEWrapper(webrtc::ViEBase * base,webrtc::ViECodec * codec,webrtc::ViECapture * capture,webrtc::ViENetwork * network,webrtc::ViERender * render,webrtc::ViERTP_RTCP * rtp,webrtc::ViEImageProcess * image,webrtc::ViEExternalCodec * ext_codec)93 ViEWrapper(webrtc::ViEBase* base, webrtc::ViECodec* codec, 94 webrtc::ViECapture* capture, webrtc::ViENetwork* network, 95 webrtc::ViERender* render, webrtc::ViERTP_RTCP* rtp, 96 webrtc::ViEImageProcess* image, 97 webrtc::ViEExternalCodec* ext_codec) 98 : engine_(NULL), 99 base_(base), 100 codec_(codec), 101 capture_(capture), 102 network_(network), 103 render_(render), 104 rtp_(rtp), 105 image_(image), 106 ext_codec_(ext_codec) { 107 } 108 ~ViEWrapper()109 virtual ~ViEWrapper() {} engine()110 webrtc::VideoEngine* engine() { return engine_.get(); } base()111 webrtc::ViEBase* base() { return base_.get(); } codec()112 webrtc::ViECodec* codec() { return codec_.get(); } capture()113 webrtc::ViECapture* capture() { return capture_.get(); } network()114 webrtc::ViENetwork* network() { return network_.get(); } render()115 webrtc::ViERender* render() { return render_.get(); } rtp()116 webrtc::ViERTP_RTCP* rtp() { return rtp_.get(); } image()117 webrtc::ViEImageProcess* image() { return image_.get(); } ext_codec()118 webrtc::ViEExternalCodec* ext_codec() { return ext_codec_.get(); } error()119 int error() { return base_->LastError(); } 120 121 private: 122 scoped_vie_engine engine_; 123 scoped_vie_ptr<webrtc::ViEBase> base_; 124 scoped_vie_ptr<webrtc::ViECodec> codec_; 125 scoped_vie_ptr<webrtc::ViECapture> capture_; 126 scoped_vie_ptr<webrtc::ViENetwork> network_; 127 scoped_vie_ptr<webrtc::ViERender> render_; 128 scoped_vie_ptr<webrtc::ViERTP_RTCP> rtp_; 129 scoped_vie_ptr<webrtc::ViEImageProcess> image_; 130 scoped_vie_ptr<webrtc::ViEExternalCodec> ext_codec_; 131 }; 132 133 // Adds indirection to static WebRtc functions, allowing them to be mocked. 134 class ViETraceWrapper { 135 public: ~ViETraceWrapper()136 virtual ~ViETraceWrapper() {} 137 SetTraceFilter(const unsigned int filter)138 virtual int SetTraceFilter(const unsigned int filter) { 139 return webrtc::VideoEngine::SetTraceFilter(filter); 140 } SetTraceFile(const char * fileNameUTF8)141 virtual int SetTraceFile(const char* fileNameUTF8) { 142 return webrtc::VideoEngine::SetTraceFile(fileNameUTF8); 143 } SetTraceCallback(webrtc::TraceCallback * callback)144 virtual int SetTraceCallback(webrtc::TraceCallback* callback) { 145 return webrtc::VideoEngine::SetTraceCallback(callback); 146 } 147 }; 148 149 } // namespace cricket 150 151 #endif // TALK_MEDIA_WEBRTCVIE_H_ 152