summaryrefslogtreecommitdiffstats
path: root/third_party/libwebrtc/test/test_video_capturer.h
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/libwebrtc/test/test_video_capturer.h')
-rw-r--r--third_party/libwebrtc/test/test_video_capturer.h64
1 files changed, 64 insertions, 0 deletions
diff --git a/third_party/libwebrtc/test/test_video_capturer.h b/third_party/libwebrtc/test/test_video_capturer.h
new file mode 100644
index 0000000000..6fafd96efb
--- /dev/null
+++ b/third_party/libwebrtc/test/test_video_capturer.h
@@ -0,0 +1,64 @@
+/*
+ * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
+ *
+ * Use of this source code is governed by a BSD-style license
+ * that can be found in the LICENSE file in the root of the source
+ * tree. An additional intellectual property rights grant can be found
+ * in the file PATENTS. All contributing project authors may
+ * be found in the AUTHORS file in the root of the source tree.
+ */
+#ifndef TEST_TEST_VIDEO_CAPTURER_H_
+#define TEST_TEST_VIDEO_CAPTURER_H_
+
+#include <stddef.h>
+
+#include <memory>
+
+#include "api/video/video_frame.h"
+#include "api/video/video_source_interface.h"
+#include "media/base/video_adapter.h"
+#include "media/base/video_broadcaster.h"
+#include "rtc_base/synchronization/mutex.h"
+
+namespace webrtc {
+namespace test {
+
+class TestVideoCapturer : public rtc::VideoSourceInterface<VideoFrame> {
+ public:
+ class FramePreprocessor {
+ public:
+ virtual ~FramePreprocessor() = default;
+
+ virtual VideoFrame Preprocess(const VideoFrame& frame) = 0;
+ };
+
+ ~TestVideoCapturer() override;
+
+ void AddOrUpdateSink(rtc::VideoSinkInterface<VideoFrame>* sink,
+ const rtc::VideoSinkWants& wants) override;
+ void RemoveSink(rtc::VideoSinkInterface<VideoFrame>* sink) override;
+ void SetFramePreprocessor(std::unique_ptr<FramePreprocessor> preprocessor) {
+ MutexLock lock(&lock_);
+ preprocessor_ = std::move(preprocessor);
+ }
+ void OnOutputFormatRequest(int width,
+ int height,
+ const absl::optional<int>& max_fps);
+
+ protected:
+ void OnFrame(const VideoFrame& frame);
+ rtc::VideoSinkWants GetSinkWants();
+
+ private:
+ void UpdateVideoAdapter();
+ VideoFrame MaybePreprocess(const VideoFrame& frame);
+
+ Mutex lock_;
+ std::unique_ptr<FramePreprocessor> preprocessor_ RTC_GUARDED_BY(lock_);
+ rtc::VideoBroadcaster broadcaster_;
+ cricket::VideoAdapter video_adapter_;
+};
+} // namespace test
+} // namespace webrtc
+
+#endif // TEST_TEST_VIDEO_CAPTURER_H_