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 #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
    120