blob: 2d726664ad5fa0fc50d33e71aefcf6207b99b5a1 [file] [log] [blame]
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <memory>
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "third_party/webrtc/modules/desktop_capture/desktop_capturer.h"
namespace viz {
class CopyOutputResult;
} // namespace cc
namespace aura {
class Window;
} // namespace aura
namespace remoting {
// A webrtc::DesktopCapturer that captures pixels from the root window of the
// Aura Shell. This is implemented by requesting the layer and its substree to
// be rendered to a given data structure. Start() and Capture() must be called
// on the Browser UI thread.
class AuraDesktopCapturer : public webrtc::DesktopCapturer {
~AuraDesktopCapturer() override;
// webrtc::DesktopCapturer implementation.
void Start(webrtc::DesktopCapturer::Callback* callback) override;
void CaptureFrame() override;
bool GetSourceList(SourceList* sources) override;
bool SelectSource(SourceId id) override;
friend class AuraDesktopCapturerTest;
// Called when a copy of the layer is captured.
void OnFrameCaptured(std::unique_ptr<viz::CopyOutputResult> result);
// Points to the callback passed to webrtc::DesktopCapturer::Start().
webrtc::DesktopCapturer::Callback* callback_;
// The root window of the Aura Shell.
aura::Window* desktop_window_;
base::WeakPtrFactory<AuraDesktopCapturer> weak_factory_;
} // namespace remoting