Home | History | Annotate | Download | only in chromeos
      1 // Copyright 2014 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #ifndef REMOTING_HOST_CHROMEOS_AURA_DESKTOP_CAPTURER_H_
      6 #define REMOTING_HOST_CHROMEOS_AURA_DESKTOP_CAPTURER_H_
      7 
      8 #include "base/memory/weak_ptr.h"
      9 #include "third_party/webrtc/modules/desktop_capture/screen_capturer.h"
     10 
     11 namespace cc {
     12 class CopyOutputResult;
     13 }  // namespace cc
     14 
     15 namespace aura {
     16 class Window;
     17 }  // namespace aura
     18 
     19 namespace remoting {
     20 
     21 // A webrtc::DesktopCapturer that captures pixels from the root window of the
     22 // Aura Shell.  This is implemented by requesting the layer and its substree to
     23 // be rendered to a given data structure.  Start() and Capture() must be called
     24 // on the Browser UI thread.
     25 class AuraDesktopCapturer : public webrtc::DesktopCapturer {
     26  public:
     27   AuraDesktopCapturer();
     28   virtual ~AuraDesktopCapturer();
     29 
     30   // webrtc::DesktopCapturer implementation.
     31   virtual void Start(webrtc::DesktopCapturer::Callback* callback) OVERRIDE;
     32   virtual void Capture(const webrtc::DesktopRegion& region) OVERRIDE;
     33 
     34  private:
     35   friend class AuraDesktopCapturerTest;
     36 
     37   // Called when a copy of the layer is captured.
     38   void OnFrameCaptured(scoped_ptr<cc::CopyOutputResult> result);
     39 
     40   // Points to the callback passed to webrtc::DesktopCapturer::Start().
     41   webrtc::DesktopCapturer::Callback* callback_;
     42 
     43   // The root window of the Aura Shell.
     44   aura::Window* desktop_window_;
     45 
     46   base::WeakPtrFactory<AuraDesktopCapturer> weak_factory_;
     47 
     48   DISALLOW_COPY_AND_ASSIGN(AuraDesktopCapturer);
     49 };
     50 
     51 }  // namespace remoting
     52 
     53 #endif  // REMOTING_HOST_CHROMEOS_AURA_DESKTOP_CAPTURER_H_
     54