remoting/host/chromeos/aura_desktop_capturer.cc

   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
  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
  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
  61 SkiaBitmapDesktopFrame::~SkiaBitmapDesktopFrame() {
  62 }
  63
  64 }  // namespace
  65
  66 AuraDesktopCapturer::AuraDesktopCapturer()
  67     : callback_(NULL), desktop_window_(NULL), weak_factory_(this) {
  68 }
  69
  70 AuraDesktopCapturer::~AuraDesktopCapturer() {
  71 }
  72
  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
  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
  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