1 files changed, 73 insertions, 0 deletions
diff --git a/dom/media/systemservices/video_engine/tab_capturer.h b/dom/media/systemservices/video_engine/tab_capturer.h
new file mode 100644
index 0000000000..ea896310cb
--- /dev/null
+++ b/dom/media/systemservices/video_engine/tab_capturer.h
@@ -0,0 +1,73 @@
+/*
+ *  Copyright 2018 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 MODULES_DESKTOP_CAPTURE_TAB_CAPTURER_H_
+#define MODULES_DESKTOP_CAPTURE_TAB_CAPTURER_H_
+
+#include <memory>
+#include <string>
+
+#include "modules/desktop_capture/desktop_capture_options.h"
+#include "modules/desktop_capture/desktop_capturer.h"
+#include "mozilla/dom/ImageBitmap.h"
+#include "mozilla/MozPromise.h"
+#include "mozilla/TaskQueue.h"
+#include "nsDeque.h"
+#include "nsThreadUtils.h"
+
+namespace mozilla {
+
+class CaptureFrameRequest;
+class TabCapturedHandler;
+
+class TabCapturerWebrtc : public webrtc::DesktopCapturer {
+ private:
+  ~TabCapturerWebrtc();
+
+ public:
+  friend class CaptureFrameRequest;
+  friend class TabCapturedHandler;
+
+  explicit TabCapturerWebrtc(const webrtc::DesktopCaptureOptions& options);
+
+  static std::unique_ptr<webrtc::DesktopCapturer> CreateRawWindowCapturer(
+      const webrtc::DesktopCaptureOptions& options);
+
+  TabCapturerWebrtc(const TabCapturerWebrtc&) = delete;
+  TabCapturerWebrtc& operator=(const TabCapturerWebrtc&) = delete;
+
+  // DesktopCapturer interface.
+  void Start(Callback* callback) override;
+  void CaptureFrame() override;
+  bool GetSourceList(SourceList* sources) override;
+  bool SelectSource(SourceId id) override;
+  bool FocusOnSelectedSource() override;
+  bool IsOccluded(const webrtc::DesktopVector& pos) override;
+
+ private:
+  // Capture code
+  using CapturePromise = MozPromise<RefPtr<dom::ImageBitmap>, nsresult, true>;
+  RefPtr<CapturePromise> CaptureFrameNow();
+
+  // Helper that checks for overrun requests. Returns true if aRequest had not
+  // been dropped.
+  bool CompleteRequest(CaptureFrameRequest* aRequest);
+
+  const RefPtr<TaskQueue> mMainThreadWorker;
+  webrtc::DesktopCapturer::Callback* mCallback = nullptr;
+  uint64_t mBrowserId = 0;
+
+  // mMainThreadWorker only
+  nsRefPtrDeque<CaptureFrameRequest> mRequests;
+};
+
+}  // namespace mozilla
+
+#endif  // MODULES_DESKTOP_CAPTURE_TAB_CAPTURER_H_