1 /* 2 * Copyright (c) 2016 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 MEDIA_BASE_VIDEO_SOURCE_BASE_H_ 12 #define MEDIA_BASE_VIDEO_SOURCE_BASE_H_ 13 14 #include <vector> 15 16 #include "api/video/video_frame.h" 17 #include "api/video/video_sink_interface.h" 18 #include "api/video/video_source_interface.h" 19 #include "rtc_base/thread_checker.h" 20 21 namespace rtc { 22 23 // VideoSourceBase is not thread safe. 24 class VideoSourceBase : public VideoSourceInterface<webrtc::VideoFrame> { 25 public: 26 VideoSourceBase(); 27 ~VideoSourceBase() override; 28 void AddOrUpdateSink(VideoSinkInterface<webrtc::VideoFrame>* sink, 29 const VideoSinkWants& wants) override; 30 void RemoveSink(VideoSinkInterface<webrtc::VideoFrame>* sink) override; 31 32 protected: 33 struct SinkPair { SinkPairSinkPair34 SinkPair(VideoSinkInterface<webrtc::VideoFrame>* sink, VideoSinkWants wants) 35 : sink(sink), wants(wants) {} 36 VideoSinkInterface<webrtc::VideoFrame>* sink; 37 VideoSinkWants wants; 38 }; 39 SinkPair* FindSinkPair(const VideoSinkInterface<webrtc::VideoFrame>* sink); 40 sink_pairs()41 const std::vector<SinkPair>& sink_pairs() const { return sinks_; } 42 43 private: 44 std::vector<SinkPair> sinks_; 45 }; 46 47 } // namespace rtc 48 49 #endif // MEDIA_BASE_VIDEO_SOURCE_BASE_H_ 50