1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ui/snapshot/snapshot.h"
#include <memory>
#include <utility>
#include "base/functional/callback.h"
#include "base/win/windows_version.h"
#include "skia/ext/platform_canvas.h"
#include "skia/ext/skia_utils_win.h"
#include "ui/aura/window.h"
#include "ui/aura/window_tree_host.h"
#include "ui/gfx/geometry/rect.h"
#include "ui/gfx/geometry/size.h"
#include "ui/gfx/geometry/skia_conversions.h"
#include "ui/gfx/geometry/transform.h"
#include "ui/gfx/image/image.h"
namespace ui {
namespace {
void GrabHwndSnapshot(HWND window_handle,
const gfx::Rect& snapshot_bounds_in_pixels,
const gfx::Rect& clip_rect_in_pixels,
gfx::Image* image) {
gfx::Rect snapshot_bounds_in_window =
snapshot_bounds_in_pixels + clip_rect_in_pixels.OffsetFromOrigin();
gfx::Size bitmap_size(snapshot_bounds_in_window.right(),
snapshot_bounds_in_window.bottom());
std::unique_ptr<SkCanvas> canvas = skia::CreatePlatformCanvas(
bitmap_size.width(), bitmap_size.height(), false);
HDC mem_hdc = skia::GetNativeDrawingContext(canvas.get());
// Grab a copy of the window. Use PrintWindow because it works even when the
// window's partially occluded. The PW_RENDERFULLCONTENT flag is undocumented,
// but works starting in Windows 8.1. It allows for capturing the contents of
// the window that are drawn using DirectComposition.
UINT flags = PW_CLIENTONLY | PW_RENDERFULLCONTENT;
BOOL result = PrintWindow(window_handle, mem_hdc, flags);
if (!result) {
PLOG(ERROR) << "Failed to print window";
return;
}
SkBitmap bitmap;
bitmap.allocN32Pixels(snapshot_bounds_in_window.width(),
snapshot_bounds_in_window.height());
canvas->readPixels(bitmap, snapshot_bounds_in_window.x(),
snapshot_bounds_in_window.y());
// Clear the region of the bitmap outside the clip rect to white.
SkCanvas image_canvas(bitmap, SkSurfaceProps{});
SkPaint paint;
paint.setColor(SK_ColorWHITE);
SkRegion region;
gfx::Rect clip_in_bitmap(clip_rect_in_pixels.size());
clip_in_bitmap.Offset(-snapshot_bounds_in_pixels.OffsetFromOrigin());
region.setRect(
gfx::RectToSkIRect(gfx::Rect(snapshot_bounds_in_pixels.size())));
region.op(gfx::RectToSkIRect(clip_in_bitmap), SkRegion::kDifference_Op);
image_canvas.drawRegion(region, paint);
*image = gfx::Image::CreateFrom1xBitmap(bitmap);
return;
}
void GrabNativeWindowSnapshot(gfx::NativeWindow native_window,
const gfx::Rect& snapshot_bounds,
gfx::Image* image) {
DCHECK(native_window);
gfx::Rect window_bounds = native_window->GetBoundsInRootWindow();
aura::WindowTreeHost* host = native_window->GetHost();
DCHECK(host);
HWND hwnd = host->GetAcceleratedWidget();
gfx::Rect snapshot_bounds_in_pixels =
host->GetRootTransform().MapRect(snapshot_bounds);
gfx::Rect expanded_window_bounds_in_pixels =
host->GetRootTransform().MapRect(window_bounds);
RECT client_area;
::GetClientRect(hwnd, &client_area);
gfx::Rect client_area_rect(client_area);
client_area_rect.set_origin(gfx::Point());
expanded_window_bounds_in_pixels.Intersect(client_area_rect);
GrabHwndSnapshot(hwnd, snapshot_bounds_in_pixels,
expanded_window_bounds_in_pixels, image);
}
} // namespace
void GrabWindowSnapshot(gfx::NativeWindow window,
const gfx::Rect& source_rect,
GrabSnapshotImageCallback callback) {
gfx::Image image;
GrabNativeWindowSnapshot(window, source_rect, &image);
std::move(callback).Run(image);
}
void GrabViewSnapshot(gfx::NativeView view,
const gfx::Rect& source_rect,
GrabSnapshotImageCallback callback) {
gfx::Image image;
GrabNativeWindowSnapshot(view, source_rect, &image);
std::move(callback).Run(image);
}
} // namespace ui
|
