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_FAKE_DESKTOP_CAPTURER_H_
6#define REMOTING_HOST_FAKE_DESKTOP_CAPTURER_H_
7
8#include "base/callback.h"
9#include "base/memory/scoped_ptr.h"
10#include "media/base/media_export.h"
11#include "third_party/webrtc/modules/desktop_capture/desktop_capturer.h"
12#include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h"
13#include "third_party/webrtc/modules/desktop_capture/screen_capture_frame_queue.h"
14
15namespace remoting {
16
17// A FakeDesktopCapturer generates artificial image for testing purpose.
18//
19// FakeDesktopCapturer is double-buffered as required by DesktopCapturer.
20class FakeDesktopCapturer : public webrtc::DesktopCapturer {
21 public:
22  // By default FakeDesktopCapturer generates frames of size kWidth x kHeight,
23  // but custom frame generator set using set_frame_generator() may generate
24  // frames of different size.
25  static const int kWidth = 800;
26  static const int kHeight = 600;
27
28  typedef base::Callback<scoped_ptr<webrtc::DesktopFrame>(
29      webrtc::DesktopCapturer::Callback* callback)> FrameGenerator;
30
31  FakeDesktopCapturer();
32  virtual ~FakeDesktopCapturer();
33
34  void set_frame_generator(const FrameGenerator& frame_generator);
35
36  // webrtc::DesktopCapturer interface.
37  virtual void Start(Callback* callback) OVERRIDE;
38  virtual void Capture(const webrtc::DesktopRegion& rect) OVERRIDE;
39
40 private:
41  FrameGenerator frame_generator_;
42
43  Callback* callback_;
44
45  DISALLOW_COPY_AND_ASSIGN(FakeDesktopCapturer);
46};
47
48}  // namespace remoting
49
50#endif  // REMOTING_HOST_FAKE_DESKTOP_CAPTURER_H_
51