11320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// Copyright 2014 The Chromium Authors. All rights reserved. 21320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// Use of this source code is governed by a BSD-style license that can be 31320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// found in the LICENSE file. 41320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 51320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "remoting/host/chromeos/aura_desktop_capturer.h" 61320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 71320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "ash/shell.h" 81320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/bind.h" 91320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "cc/output/copy_output_request.h" 101320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "cc/output/copy_output_result.h" 111320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "third_party/skia/include/core/SkBitmap.h" 121320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" 131320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "ui/aura/window.h" 141320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "ui/aura/window_tree_host.h" 151320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 161320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccinamespace remoting { 171320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 181320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccinamespace { 191320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 201320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// DesktopFrame implementation used by screen capture on ChromeOS. 211320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// Frame data is stored in a SkBitmap. 221320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucciclass SkiaBitmapDesktopFrame : public webrtc::DesktopFrame { 231320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci public: 241320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci static SkiaBitmapDesktopFrame* Create(scoped_ptr<SkBitmap> bitmap); 251320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci virtual ~SkiaBitmapDesktopFrame(); 261320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 271320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci private: 281320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci SkiaBitmapDesktopFrame(webrtc::DesktopSize size, 291320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci int stride, 301320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci uint8_t* data, 311320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<SkBitmap> bitmap); 321320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 331320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<SkBitmap> bitmap_; 341320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 351320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DISALLOW_COPY_AND_ASSIGN(SkiaBitmapDesktopFrame); 361320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci}; 371320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 381320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// static 391320f92c476a1ad9d19dba2a48c72b75566198e9Primiano TucciSkiaBitmapDesktopFrame* SkiaBitmapDesktopFrame::Create( 401320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<SkBitmap> bitmap) { 411320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 421320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci webrtc::DesktopSize size(bitmap->width(), bitmap->height()); 431320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DCHECK_EQ(kRGBA_8888_SkColorType, bitmap->info().colorType()) 441320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci << "DesktopFrame objects always hold RGBA data."; 451320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 461320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci uint8_t* bitmap_data = reinterpret_cast<uint8_t*>(bitmap->getPixels()); 471320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 481320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci SkiaBitmapDesktopFrame* result = new SkiaBitmapDesktopFrame( 491320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci size, bitmap->rowBytes(), bitmap_data, bitmap.Pass()); 501320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 511320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci return result; 521320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 531320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 541320f92c476a1ad9d19dba2a48c72b75566198e9Primiano TucciSkiaBitmapDesktopFrame::SkiaBitmapDesktopFrame(webrtc::DesktopSize size, 551320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci int stride, 561320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci uint8_t* data, 571320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<SkBitmap> bitmap) 581320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci : DesktopFrame(size, stride, data, NULL), bitmap_(bitmap.Pass()) { 591320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 601320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 611320f92c476a1ad9d19dba2a48c72b75566198e9Primiano TucciSkiaBitmapDesktopFrame::~SkiaBitmapDesktopFrame() { 621320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 631320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 641320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} // namespace 651320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 661320f92c476a1ad9d19dba2a48c72b75566198e9Primiano TucciAuraDesktopCapturer::AuraDesktopCapturer() 671320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci : callback_(NULL), desktop_window_(NULL), weak_factory_(this) { 681320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 691320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 701320f92c476a1ad9d19dba2a48c72b75566198e9Primiano TucciAuraDesktopCapturer::~AuraDesktopCapturer() { 711320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 721320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 731320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccivoid AuraDesktopCapturer::Start(webrtc::DesktopCapturer::Callback* callback) { 741320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci if (ash::Shell::HasInstance()) { 751320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // TODO(kelvinp): Use ash::Shell::GetAllRootWindows() when multiple monitor 761320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // support is implemented. 771320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci desktop_window_ = ash::Shell::GetPrimaryRootWindow(); 781320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DCHECK(desktop_window_) << "Failed to retrieve the Aura Shell root window"; 791320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci } 801320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 811320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DCHECK(!callback_) << "Start() can only be called once"; 821320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci callback_ = callback; 831320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DCHECK(callback_); 841320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 851320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 861320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccivoid AuraDesktopCapturer::Capture(const webrtc::DesktopRegion&) { 871320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<cc::CopyOutputRequest> request = 881320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci cc::CopyOutputRequest::CreateBitmapRequest( 891320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci base::Bind( 901320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci &AuraDesktopCapturer::OnFrameCaptured, 911320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci weak_factory_.GetWeakPtr())); 921320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 931320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci gfx::Rect window_rect(desktop_window_->bounds().size()); 941320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 951320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci request->set_area(window_rect); 961320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci desktop_window_->layer()->RequestCopyOfOutput(request.Pass()); 971320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 981320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 991320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccivoid AuraDesktopCapturer::OnFrameCaptured( 1001320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<cc::CopyOutputResult> result) { 1011320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci DCHECK(result->HasBitmap()); 1021320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 1031320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<SkBitmap> bitmap = result->TakeBitmap(); 1041320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 1051320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<webrtc::DesktopFrame> frame( 1061320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci SkiaBitmapDesktopFrame::Create(bitmap.Pass())); 1071320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 1081320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // |VideoScheduler| will not encode the frame if |updated_region| is empty. 1091320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci const webrtc::DesktopRect& rect = webrtc::DesktopRect::MakeWH( 1101320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci frame->size().width(), frame->size().height()); 1111320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 1121320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // TODO(kelvinp): Set Frame DPI according to the screen resolution. 1131320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // See cc::Layer::contents_scale_(x|y)() and frame->set_depi(). 1141320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci frame->mutable_updated_region()->SetRect(rect); 1151320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 1161320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci callback_->OnCaptureCompleted(frame.release()); 1171320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 1181320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 1191320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} // namespace remoting 120