| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "ui/snapshot/snapshot_aura.h" |
| |
| #include <memory> |
| #include <utility> |
| |
| #include "base/bind.h" |
| #include "base/callback.h" |
| #include "base/task_runner_util.h" |
| #include "components/viz/common/frame_sinks/copy_output_request.h" |
| #include "third_party/skia/include/core/SkBitmap.h" |
| #include "ui/aura/window.h" |
| #include "ui/aura/window_tracker.h" |
| #include "ui/compositor/compositor.h" |
| #include "ui/compositor/dip_util.h" |
| #include "ui/compositor/layer.h" |
| #include "ui/snapshot/snapshot_async.h" |
| |
| namespace ui { |
| |
| bool GrabWindowSnapshotAura(aura::Window* window, |
| const gfx::Rect& snapshot_bounds, |
| gfx::Image* image) { |
| // Not supported in Aura. Callers should fall back to the async version. |
| return false; |
| } |
| |
| static void MakeAsyncCopyRequest( |
| Layer* layer, |
| const gfx::Rect& source_rect, |
| viz::CopyOutputRequest::CopyOutputRequestCallback callback) { |
| std::unique_ptr<viz::CopyOutputRequest> request = |
| std::make_unique<viz::CopyOutputRequest>( |
| viz::CopyOutputRequest::ResultFormat::RGBA_BITMAP, |
| std::move(callback)); |
| request->set_area(source_rect); |
| layer->RequestCopyOfOutput(std::move(request)); |
| } |
| |
| static void FinishedAsyncCopyRequest( |
| std::unique_ptr<aura::WindowTracker> tracker, |
| const gfx::Rect& source_rect, |
| viz::CopyOutputRequest::CopyOutputRequestCallback callback, |
| int retry_count, |
| std::unique_ptr<viz::CopyOutputResult> result) { |
| static const int kMaxRetries = 5; |
| // Retry the copy request if the previous one failed for some reason. |
| if (!tracker->windows().empty() && (retry_count < kMaxRetries) && |
| result->IsEmpty()) { |
| // Look up window before calling MakeAsyncRequest. Otherwise, due |
| // to undefined (favorably right to left) argument evaluation |
| // order, the tracker might have been passed and set to NULL |
| // before the window is looked up which results in a NULL pointer |
| // dereference. |
| aura::Window* window = tracker->windows()[0]; |
| MakeAsyncCopyRequest( |
| window->layer(), source_rect, |
| base::BindOnce(&FinishedAsyncCopyRequest, std::move(tracker), |
| source_rect, std::move(callback), retry_count + 1)); |
| return; |
| } |
| |
| std::move(callback).Run(std::move(result)); |
| } |
| |
| static void MakeInitialAsyncCopyRequest( |
| aura::Window* window, |
| const gfx::Rect& source_rect, |
| viz::CopyOutputRequest::CopyOutputRequestCallback callback) { |
| auto tracker = std::make_unique<aura::WindowTracker>(); |
| tracker->Add(window); |
| MakeAsyncCopyRequest( |
| window->layer(), source_rect, |
| base::BindOnce(&FinishedAsyncCopyRequest, std::move(tracker), source_rect, |
| std::move(callback), 0)); |
| } |
| |
| void GrabWindowSnapshotAndScaleAsyncAura( |
| aura::Window* window, |
| const gfx::Rect& source_rect, |
| const gfx::Size& target_size, |
| const GrabWindowSnapshotAsyncCallback& callback) { |
| MakeInitialAsyncCopyRequest( |
| window, source_rect, |
| base::BindOnce(&SnapshotAsync::ScaleCopyOutputResult, callback, |
| target_size)); |
| } |
| |
| void GrabWindowSnapshotAsyncAura( |
| aura::Window* window, |
| const gfx::Rect& source_rect, |
| const GrabWindowSnapshotAsyncCallback& callback) { |
| MakeInitialAsyncCopyRequest( |
| window, source_rect, |
| base::BindOnce(&SnapshotAsync::RunCallbackWithCopyOutputResult, |
| callback)); |
| } |
| |
| #if !defined(OS_WIN) |
| bool GrabWindowSnapshot(gfx::NativeWindow window, |
| const gfx::Rect& snapshot_bounds, |
| gfx::Image* image) { |
| // Not supported in Aura. Callers should fall back to the async version. |
| return false; |
| } |
| |
| bool GrabViewSnapshot(gfx::NativeView view, |
| const gfx::Rect& snapshot_bounds, |
| gfx::Image* image) { |
| return GrabWindowSnapshot(view, snapshot_bounds, image); |
| } |
| |
| void GrabWindowSnapshotAndScaleAsync( |
| gfx::NativeWindow window, |
| const gfx::Rect& source_rect, |
| const gfx::Size& target_size, |
| const GrabWindowSnapshotAsyncCallback& callback) { |
| GrabWindowSnapshotAndScaleAsyncAura(window, source_rect, target_size, |
| callback); |
| } |
| |
| void GrabWindowSnapshotAsync(gfx::NativeWindow window, |
| const gfx::Rect& source_rect, |
| const GrabWindowSnapshotAsyncCallback& callback) { |
| GrabWindowSnapshotAsyncAura(window, source_rect, callback); |
| } |
| |
| void GrabViewSnapshotAsync(gfx::NativeView view, |
| const gfx::Rect& source_rect, |
| const GrabWindowSnapshotAsyncCallback& callback) { |
| GrabWindowSnapshotAsyncAura(view, source_rect, callback); |
| } |
| |
| void GrabLayerSnapshotAsync(ui::Layer* layer, |
| const gfx::Rect& source_rect, |
| const GrabWindowSnapshotAsyncCallback& callback) { |
| MakeAsyncCopyRequest( |
| layer, source_rect, |
| base::BindOnce(&SnapshotAsync::RunCallbackWithCopyOutputResult, |
| callback)); |
| } |
| |
| #endif |
| |
| } // namespace ui |