src/remoting/host/ipc_video_frame_capturer.h

   1 // Copyright (c) 2012 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_IPC_VIDEO_FRAME_CAPTURER_H_
   6 #define REMOTING_HOST_IPC_VIDEO_FRAME_CAPTURER_H_
   7
   8 #include "base/memory/ref_counted.h"
   9 #include "base/memory/weak_ptr.h"
  10 #include "base/memory/scoped_ptr.h"
  11 #include "third_party/webrtc/modules/desktop_capture/desktop_capturer.h"
  12
  13 namespace remoting {
  14
  15 class DesktopSessionProxy;
  16
  17 // Routes webrtc::DesktopCapturer calls though the IPC channel to the desktop
  18 // session agent running in the desktop integration process.
  19 class IpcVideoFrameCapturer : public webrtc::DesktopCapturer {
  20  public:
  21   explicit IpcVideoFrameCapturer(
  22       scoped_refptr<DesktopSessionProxy> desktop_session_proxy);
  23   virtual ~IpcVideoFrameCapturer();
  24
  25   // webrtc::DesktopCapturer interface.
  26   virtual void Start(Callback* callback) OVERRIDE;
  27   virtual void Capture(const webrtc::DesktopRegion& region) OVERRIDE;
  28
  29   // Called when a video |frame| has been captured.
  30   void OnCaptureCompleted(scoped_ptr<webrtc::DesktopFrame> frame);
  31
  32  private:
  33   // Points to the callback passed to webrtc::DesktopCapturer::Start().
  34   webrtc::DesktopCapturer::Callback* callback_;
  35
  36   // Wraps the IPC channel to the desktop session agent.
  37   scoped_refptr<DesktopSessionProxy> desktop_session_proxy_;
  38
  39   // Set to true when a frame is being captured.
  40   bool capture_pending_;
  41
  42   // Used to cancel tasks pending on the capturer when it is stopped.
  43   base::WeakPtrFactory<IpcVideoFrameCapturer> weak_factory_;
  44
  45   DISALLOW_COPY_AND_ASSIGN(IpcVideoFrameCapturer);
  46 };
  47
  48 }  // namespace remoting
  49
  50 #endif  // REMOTING_HOST_IPC_VIDEO_FRAME_CAPTURER_H_