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 #include "remoting/host/chromeos/aura_desktop_capturer.h"
6
7 #include "ash/shell.h"
8 #include "base/bind.h"
9 #include "cc/output/copy_output_request.h"
10 #include "cc/output/copy_output_result.h"
11 #include "third_party/skia/include/core/SkBitmap.h"
12 #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
13 #include "ui/aura/window.h"
14 #include "ui/aura/window_tree_host.h"
15
16 namespace remoting {
17
18 namespace {
19
20 // DesktopFrame implementation used by screen capture on ChromeOS.
21 // Frame data is stored in a SkBitmap.
22 class SkiaBitmapDesktopFrame : public webrtc::DesktopFrame {
23 public:
24 static SkiaBitmapDesktopFrame* Create(scoped_ptr<SkBitmap> bitmap);
25 virtual ~SkiaBitmapDesktopFrame();
26
27 private:
28 SkiaBitmapDesktopFrame(webrtc::DesktopSize size,
29 int stride,
30 uint8_t* data,
31 scoped_ptr<SkBitmap> bitmap);
32
33 scoped_ptr<SkBitmap> bitmap_;
34
35 DISALLOW_COPY_AND_ASSIGN(SkiaBitmapDesktopFrame);
36 };
37
38 // static
Create(scoped_ptr<SkBitmap> bitmap)39 SkiaBitmapDesktopFrame* SkiaBitmapDesktopFrame::Create(
40 scoped_ptr<SkBitmap> bitmap) {
41
42 webrtc::DesktopSize size(bitmap->width(), bitmap->height());
43 DCHECK_EQ(kRGBA_8888_SkColorType, bitmap->info().colorType())
44 << "DesktopFrame objects always hold RGBA data.";
45
46 uint8_t* bitmap_data = reinterpret_cast<uint8_t*>(bitmap->getPixels());
47
48 SkiaBitmapDesktopFrame* result = new SkiaBitmapDesktopFrame(
49 size, bitmap->rowBytes(), bitmap_data, bitmap.Pass());
50
51 return result;
52 }
53
SkiaBitmapDesktopFrame(webrtc::DesktopSize size,int stride,uint8_t * data,scoped_ptr<SkBitmap> bitmap)54 SkiaBitmapDesktopFrame::SkiaBitmapDesktopFrame(webrtc::DesktopSize size,
55 int stride,
56 uint8_t* data,
57 scoped_ptr<SkBitmap> bitmap)
58 : DesktopFrame(size, stride, data, NULL), bitmap_(bitmap.Pass()) {
59 }
60
~SkiaBitmapDesktopFrame()61 SkiaBitmapDesktopFrame::~SkiaBitmapDesktopFrame() {
62 }
63
64 } // namespace
65
AuraDesktopCapturer()66 AuraDesktopCapturer::AuraDesktopCapturer()
67 : callback_(NULL), desktop_window_(NULL), weak_factory_(this) {
68 }
69
~AuraDesktopCapturer()70 AuraDesktopCapturer::~AuraDesktopCapturer() {
71 }
72
Start(webrtc::DesktopCapturer::Callback * callback)73 void AuraDesktopCapturer::Start(webrtc::DesktopCapturer::Callback* callback) {
74 if (ash::Shell::HasInstance()) {
75 // TODO(kelvinp): Use ash::Shell::GetAllRootWindows() when multiple monitor
76 // support is implemented.
77 desktop_window_ = ash::Shell::GetPrimaryRootWindow();
78 DCHECK(desktop_window_) << "Failed to retrieve the Aura Shell root window";
79 }
80
81 DCHECK(!callback_) << "Start() can only be called once";
82 callback_ = callback;
83 DCHECK(callback_);
84 }
85
Capture(const webrtc::DesktopRegion &)86 void AuraDesktopCapturer::Capture(const webrtc::DesktopRegion&) {
87 scoped_ptr<cc::CopyOutputRequest> request =
88 cc::CopyOutputRequest::CreateBitmapRequest(
89 base::Bind(
90 &AuraDesktopCapturer::OnFrameCaptured,
91 weak_factory_.GetWeakPtr()));
92
93 gfx::Rect window_rect(desktop_window_->bounds().size());
94
95 request->set_area(window_rect);
96 desktop_window_->layer()->RequestCopyOfOutput(request.Pass());
97 }
98
OnFrameCaptured(scoped_ptr<cc::CopyOutputResult> result)99 void AuraDesktopCapturer::OnFrameCaptured(
100 scoped_ptr<cc::CopyOutputResult> result) {
101 DCHECK(result->HasBitmap());
102
103 scoped_ptr<SkBitmap> bitmap = result->TakeBitmap();
104
105 scoped_ptr<webrtc::DesktopFrame> frame(
106 SkiaBitmapDesktopFrame::Create(bitmap.Pass()));
107
108 // |VideoScheduler| will not encode the frame if |updated_region| is empty.
109 const webrtc::DesktopRect& rect = webrtc::DesktopRect::MakeWH(
110 frame->size().width(), frame->size().height());
111
112 // TODO(kelvinp): Set Frame DPI according to the screen resolution.
113 // See cc::Layer::contents_scale_(x|y)() and frame->set_depi().
114 frame->mutable_updated_region()->SetRect(rect);
115
116 callback_->OnCaptureCompleted(frame.release());
117 }
118
119 } // namespace remoting
120