| OLD | NEW |
|---|
| (Empty) |
| 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 | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 579583003:
Revert of Remote Assistance on Chrome OS Part I - Aura Desktop Capturer (Closed)
Base URL: https://chromium.googlesource.com/chromium/src.git@master