Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(76)

Side by Side Diff: ui/snapshot/snapshot_mac.mm

Issue 2650903003: Revert of [devtools] Support different encodings for Page.CaptureScreenshot. (Closed)
Patch Set: Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « ui/snapshot/snapshot_ios.mm ('k') | ui/snapshot/snapshot_mac_unittest.mm » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "ui/snapshot/snapshot.h" 5 #include "ui/snapshot/snapshot.h"
6 6
7 #import <Cocoa/Cocoa.h> 7 #import <Cocoa/Cocoa.h>
8 8
9 #include "base/callback.h" 9 #include "base/callback.h"
10 #include "base/logging.h" 10 #include "base/logging.h"
11 #include "base/mac/scoped_cftyperef.h" 11 #include "base/mac/scoped_cftyperef.h"
12 #include "base/mac/scoped_nsobject.h"
12 #include "base/mac/sdk_forward_declarations.h" 13 #include "base/mac/sdk_forward_declarations.h"
13 #include "base/task_runner.h" 14 #include "base/task_runner.h"
14 #include "ui/gfx/geometry/rect.h" 15 #include "ui/gfx/geometry/rect.h"
15 #include "ui/gfx/image/image.h" 16 #include "ui/gfx/image/image.h"
16 17
17 namespace ui { 18 namespace ui {
18 19
19 bool GrabViewSnapshot(gfx::NativeView view, 20 bool GrabViewSnapshot(gfx::NativeView view,
20 const gfx::Rect& snapshot_bounds, 21 std::vector<unsigned char>* png_representation,
21 gfx::Image* image) { 22 const gfx::Rect& snapshot_bounds) {
22 NSWindow* window = [view window]; 23 NSWindow* window = [view window];
23 NSScreen* screen = [[NSScreen screens] firstObject]; 24 NSScreen* screen = [[NSScreen screens] firstObject];
24 gfx::Rect screen_bounds = gfx::Rect(NSRectToCGRect([screen frame])); 25 gfx::Rect screen_bounds = gfx::Rect(NSRectToCGRect([screen frame]));
25 26
26 27
27 // Get the view bounds relative to the screen 28 // Get the view bounds relative to the screen
28 NSRect frame = [view convertRect:[view bounds] toView:nil]; 29 NSRect frame = [view convertRect:[view bounds] toView:nil];
29 frame = [window convertRectToScreen:frame]; 30 frame = [window convertRectToScreen:frame];
30 31
31 gfx::Rect view_bounds = gfx::Rect(NSRectToCGRect(frame)); 32 gfx::Rect view_bounds = gfx::Rect(NSRectToCGRect(frame));
32 33
33 // Flip window coordinates based on the primary screen. 34 // Flip window coordinates based on the primary screen.
34 view_bounds.set_y( 35 view_bounds.set_y(
35 screen_bounds.height() - view_bounds.y() - view_bounds.height()); 36 screen_bounds.height() - view_bounds.y() - view_bounds.height());
36 37
37 // Convert snapshot bounds relative to window into bounds relative to 38 // Convert snapshot bounds relative to window into bounds relative to
38 // screen. 39 // screen.
39 gfx::Rect screen_snapshot_bounds = snapshot_bounds; 40 gfx::Rect screen_snapshot_bounds = snapshot_bounds;
40 screen_snapshot_bounds.Offset(view_bounds.OffsetFromOrigin()); 41 screen_snapshot_bounds.Offset(view_bounds.OffsetFromOrigin());
41 42
42 DCHECK_LE(screen_snapshot_bounds.right(), view_bounds.right()); 43 DCHECK_LE(screen_snapshot_bounds.right(), view_bounds.right());
43 DCHECK_LE(screen_snapshot_bounds.bottom(), view_bounds.bottom()); 44 DCHECK_LE(screen_snapshot_bounds.bottom(), view_bounds.bottom());
44 45
46 png_representation->clear();
47
45 base::ScopedCFTypeRef<CGImageRef> windowSnapshot( 48 base::ScopedCFTypeRef<CGImageRef> windowSnapshot(
46 CGWindowListCreateImage(screen_snapshot_bounds.ToCGRect(), 49 CGWindowListCreateImage(screen_snapshot_bounds.ToCGRect(),
47 kCGWindowListOptionIncludingWindow, 50 kCGWindowListOptionIncludingWindow,
48 [window windowNumber], 51 [window windowNumber],
49 kCGWindowImageBoundsIgnoreFraming)); 52 kCGWindowImageBoundsIgnoreFraming));
50 if (CGImageGetWidth(windowSnapshot) <= 0) 53 if (CGImageGetWidth(windowSnapshot) <= 0)
51 return false; 54 return false;
52 55
53 NSImage* nsImage = 56 base::scoped_nsobject<NSBitmapImageRep> rep(
54 [[NSImage alloc] initWithCGImage:windowSnapshot size:NSZeroSize]; 57 [[NSBitmapImageRep alloc] initWithCGImage:windowSnapshot]);
55 *image = gfx::Image(nsImage); 58 NSData* data = [rep representationUsingType:NSPNGFileType properties:@{}];
59 const unsigned char* buf = static_cast<const unsigned char*>([data bytes]);
60 NSUInteger length = [data length];
61 if (buf == NULL || length == 0)
62 return false;
63
64 png_representation->assign(buf, buf + length);
65 DCHECK(!png_representation->empty());
66
56 return true; 67 return true;
57 } 68 }
58 69
59 bool GrabWindowSnapshot(gfx::NativeWindow window, 70 bool GrabWindowSnapshot(gfx::NativeWindow window,
60 const gfx::Rect& snapshot_bounds, 71 std::vector<unsigned char>* png_representation,
61 gfx::Image* image) { 72 const gfx::Rect& snapshot_bounds) {
62 // Make sure to grab the "window frame" view so we get current tab + 73 // Make sure to grab the "window frame" view so we get current tab +
63 // tabstrip. 74 // tabstrip.
64 return GrabViewSnapshot([[window contentView] superview], snapshot_bounds, 75 return GrabViewSnapshot([[window contentView] superview], png_representation,
65 image); 76 snapshot_bounds);
66 } 77 }
67 78
68 void GrabWindowSnapshotAndScaleAsync( 79 void GrabWindowSnapshotAndScaleAsync(
69 gfx::NativeWindow window, 80 gfx::NativeWindow window,
70 const gfx::Rect& snapshot_bounds, 81 const gfx::Rect& snapshot_bounds,
71 const gfx::Size& target_size, 82 const gfx::Size& target_size,
72 scoped_refptr<base::TaskRunner> background_task_runner, 83 scoped_refptr<base::TaskRunner> background_task_runner,
73 GrabWindowSnapshotAsyncCallback callback) { 84 GrabWindowSnapshotAsyncCallback callback) {
74 callback.Run(gfx::Image()); 85 callback.Run(gfx::Image());
75 } 86 }
76 87
77 void GrabViewSnapshotAsync(gfx::NativeView view, 88 void GrabViewSnapshotAsync(
78 const gfx::Rect& source_rect, 89 gfx::NativeView view,
79 const GrabWindowSnapshotAsyncCallback& callback) { 90 const gfx::Rect& source_rect,
80 callback.Run(gfx::Image()); 91 scoped_refptr<base::TaskRunner> background_task_runner,
92 const GrabWindowSnapshotAsyncPNGCallback& callback) {
93 callback.Run(scoped_refptr<base::RefCountedBytes>());
81 } 94 }
82 95
83 void GrabWindowSnapshotAsync(gfx::NativeWindow window, 96 void GrabWindowSnapshotAsync(
84 const gfx::Rect& source_rect, 97 gfx::NativeWindow window,
85 const GrabWindowSnapshotAsyncCallback& callback) { 98 const gfx::Rect& source_rect,
99 scoped_refptr<base::TaskRunner> background_task_runner,
100 const GrabWindowSnapshotAsyncPNGCallback& callback) {
86 return GrabViewSnapshotAsync([[window contentView] superview], source_rect, 101 return GrabViewSnapshotAsync([[window contentView] superview], source_rect,
87 callback); 102 background_task_runner, callback);
88 } 103 }
89 104
90 } // namespace ui 105 } // namespace ui
OLDNEW
« no previous file with comments | « ui/snapshot/snapshot_ios.mm ('k') | ui/snapshot/snapshot_mac_unittest.mm » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698