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
|
// Copyright 2012 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"
#import <UIKit/UIKit.h>
#include "base/functional/callback.h"
#include "ui/gfx/geometry/rect.h"
#include "ui/gfx/image/image.h"
#include "ui/gfx/image/image_util.h"
namespace ui {
namespace {
UIImage* GetViewSnapshot(UIView* view, CGRect bounds) {
UIGraphicsImageRendererFormat* format =
[UIGraphicsImageRendererFormat preferredFormat];
format.scale = [[UIScreen mainScreen] scale];
format.opaque = NO;
UIGraphicsImageRenderer* renderer =
[[UIGraphicsImageRenderer alloc] initWithSize:bounds.size format:format];
UIImage* snapshot =
[renderer imageWithActions:^(UIGraphicsImageRendererContext* context) {
[view drawViewHierarchyInRect:bounds afterScreenUpdates:YES];
}];
return snapshot;
}
} // namespace
bool GrabViewSnapshot(gfx::NativeView view,
const gfx::Rect& snapshot_bounds,
gfx::Image* image) {
UIView* source_view = view.Get();
if (source_view == nil) {
return false;
}
UIImage* snapshot = GetViewSnapshot(source_view, snapshot_bounds.ToCGRect());
if (snapshot) {
*image = gfx::Image(snapshot);
return true;
}
return false;
}
bool GrabWindowSnapshot(gfx::NativeWindow window,
const gfx::Rect& snapshot_bounds,
gfx::Image* image) {
UIWindow* source_window = window.Get();
if (source_window == nil) {
return false;
}
UIImage* snapshot = GetViewSnapshot(source_window.rootViewController.view,
snapshot_bounds.ToCGRect());
if (snapshot) {
*image = gfx::Image(snapshot);
return true;
}
return false;
}
void GrabWindowSnapshotAndScaleAsync(
gfx::NativeWindow window,
const gfx::Rect& snapshot_bounds,
const gfx::Size& target_size,
GrabWindowSnapshotAsyncCallback callback) {
gfx::Image image;
if (GrabWindowSnapshot(window, snapshot_bounds, &image)) {
gfx::Image resized_image = gfx::ResizedImage(image, target_size);
std::move(callback).Run(resized_image);
return;
}
std::move(callback).Run(image);
}
void GrabViewSnapshotAsync(gfx::NativeView view,
const gfx::Rect& source_rect,
GrabWindowSnapshotAsyncCallback callback) {
gfx::Image image;
GrabViewSnapshot(view, source_rect, &image);
std::move(callback).Run(image);
}
void GrabWindowSnapshotAsync(gfx::NativeWindow window,
const gfx::Rect& source_rect,
GrabWindowSnapshotAsyncCallback callback) {
gfx::Image image;
GrabWindowSnapshot(window, source_rect, &image);
std::move(callback).Run(image);
}
} // namespace ui
|
