| OLD | NEW |
| 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/scoped_nsobject.h" |
| 13 #include "base/mac/sdk_forward_declarations.h" | 13 #include "base/mac/sdk_forward_declarations.h" |
| 14 #include "base/task_runner.h" | 14 #include "base/task_runner.h" |
| 15 #include "ui/gfx/geometry/rect.h" | 15 #include "ui/gfx/geometry/rect.h" |
| 16 #include "ui/gfx/image/image.h" | 16 #include "ui/gfx/image/image.h" |
| 17 | 17 |
| 18 namespace ui { | 18 namespace ui { |
| 19 | 19 |
| 20 bool GrabViewSnapshot(gfx::NativeView view, | 20 bool GrabViewSnapshot(gfx::NativeView view, |
| 21 std::vector<unsigned char>* png_representation, | 21 std::vector<unsigned char>* png_representation, |
| 22 const gfx::Rect& snapshot_bounds) { | 22 const gfx::Rect& snapshot_bounds, |
| 23 SnapshotEncoding encoding, |
| 24 SnapshotQuality quality) { |
| 23 NSWindow* window = [view window]; | 25 NSWindow* window = [view window]; |
| 24 NSScreen* screen = [[NSScreen screens] firstObject]; | 26 NSScreen* screen = [[NSScreen screens] firstObject]; |
| 25 gfx::Rect screen_bounds = gfx::Rect(NSRectToCGRect([screen frame])); | 27 gfx::Rect screen_bounds = gfx::Rect(NSRectToCGRect([screen frame])); |
| 26 | 28 |
| 27 | 29 |
| 28 // Get the view bounds relative to the screen | 30 // Get the view bounds relative to the screen |
| 29 NSRect frame = [view convertRect:[view bounds] toView:nil]; | 31 NSRect frame = [view convertRect:[view bounds] toView:nil]; |
| 30 frame = [window convertRectToScreen:frame]; | 32 frame = [window convertRectToScreen:frame]; |
| 31 | 33 |
| 32 gfx::Rect view_bounds = gfx::Rect(NSRectToCGRect(frame)); | 34 gfx::Rect view_bounds = gfx::Rect(NSRectToCGRect(frame)); |
| (...skipping 15 matching lines...) Expand all Loading... |
| 48 base::ScopedCFTypeRef<CGImageRef> windowSnapshot( | 50 base::ScopedCFTypeRef<CGImageRef> windowSnapshot( |
| 49 CGWindowListCreateImage(screen_snapshot_bounds.ToCGRect(), | 51 CGWindowListCreateImage(screen_snapshot_bounds.ToCGRect(), |
| 50 kCGWindowListOptionIncludingWindow, | 52 kCGWindowListOptionIncludingWindow, |
| 51 [window windowNumber], | 53 [window windowNumber], |
| 52 kCGWindowImageBoundsIgnoreFraming)); | 54 kCGWindowImageBoundsIgnoreFraming)); |
| 53 if (CGImageGetWidth(windowSnapshot) <= 0) | 55 if (CGImageGetWidth(windowSnapshot) <= 0) |
| 54 return false; | 56 return false; |
| 55 | 57 |
| 56 base::scoped_nsobject<NSBitmapImageRep> rep( | 58 base::scoped_nsobject<NSBitmapImageRep> rep( |
| 57 [[NSBitmapImageRep alloc] initWithCGImage:windowSnapshot]); | 59 [[NSBitmapImageRep alloc] initWithCGImage:windowSnapshot]); |
| 58 NSData* data = [rep representationUsingType:NSPNGFileType properties:@{}]; | 60 NSData* data; |
| 61 float compression_factor = 0; |
| 62 NSDictionary* options = nullptr; |
| 63 switch (encoding) { |
| 64 case SnapshotEncoding::PNG: |
| 65 data = [rep representationUsingType:NSPNGFileType properties:@{}]; |
| 66 break; |
| 67 case SnapshotEncoding::JPEG: |
| 68 compression_factor = static_cast<float>(quality) / 100; |
| 69 options = [NSDictionary |
| 70 dictionaryWithObject:[NSNumber numberWithFloat:compression_factor] |
| 71 forKey:NSImageCompressionFactor]; |
| 72 data = [rep representationUsingType:NSJPEGFileType properties:options]; |
| 73 break; |
| 74 default: |
| 75 NOTREACHED(); |
| 76 } |
| 59 const unsigned char* buf = static_cast<const unsigned char*>([data bytes]); | 77 const unsigned char* buf = static_cast<const unsigned char*>([data bytes]); |
| 60 NSUInteger length = [data length]; | 78 NSUInteger length = [data length]; |
| 61 if (buf == NULL || length == 0) | 79 if (buf == NULL || length == 0) |
| 62 return false; | 80 return false; |
| 63 | 81 |
| 64 png_representation->assign(buf, buf + length); | 82 png_representation->assign(buf, buf + length); |
| 65 DCHECK(!png_representation->empty()); | 83 DCHECK(!png_representation->empty()); |
| 66 | 84 |
| 67 return true; | 85 return true; |
| 68 } | 86 } |
| 69 | 87 |
| 70 bool GrabWindowSnapshot(gfx::NativeWindow window, | 88 bool GrabWindowSnapshot(gfx::NativeWindow window, |
| 71 std::vector<unsigned char>* png_representation, | 89 std::vector<unsigned char>* png_representation, |
| 72 const gfx::Rect& snapshot_bounds) { | 90 const gfx::Rect& snapshot_bounds, |
| 91 SnapshotEncoding encoding, |
| 92 SnapshotQuality quality) { |
| 73 // Make sure to grab the "window frame" view so we get current tab + | 93 // Make sure to grab the "window frame" view so we get current tab + |
| 74 // tabstrip. | 94 // tabstrip. |
| 75 return GrabViewSnapshot([[window contentView] superview], png_representation, | 95 return GrabViewSnapshot([[window contentView] superview], png_representation, |
| 76 snapshot_bounds); | 96 snapshot_bounds, encoding, quality); |
| 77 } | 97 } |
| 78 | 98 |
| 79 void GrabWindowSnapshotAndScaleAsync( | 99 void GrabWindowSnapshotAndScaleAsync( |
| 80 gfx::NativeWindow window, | 100 gfx::NativeWindow window, |
| 81 const gfx::Rect& snapshot_bounds, | 101 const gfx::Rect& snapshot_bounds, |
| 82 const gfx::Size& target_size, | 102 const gfx::Size& target_size, |
| 83 scoped_refptr<base::TaskRunner> background_task_runner, | 103 scoped_refptr<base::TaskRunner> background_task_runner, |
| 84 GrabWindowSnapshotAsyncCallback callback) { | 104 GrabWindowSnapshotAsyncCallback callback) { |
| 85 callback.Run(gfx::Image()); | 105 callback.Run(gfx::Image()); |
| 86 } | 106 } |
| 87 | 107 |
| 88 void GrabViewSnapshotAsync( | 108 void GrabViewSnapshotAsync( |
| 89 gfx::NativeView view, | 109 gfx::NativeView view, |
| 90 const gfx::Rect& source_rect, | 110 const gfx::Rect& source_rect, |
| 91 scoped_refptr<base::TaskRunner> background_task_runner, | 111 scoped_refptr<base::TaskRunner> background_task_runner, |
| 92 const GrabWindowSnapshotAsyncPNGCallback& callback) { | 112 const GrabWindowSnapshotAsyncEncodedCallback& callback, |
| 113 SnapshotEncoding encoding, |
| 114 SnapshotQuality quality) { |
| 93 callback.Run(scoped_refptr<base::RefCountedBytes>()); | 115 callback.Run(scoped_refptr<base::RefCountedBytes>()); |
| 94 } | 116 } |
| 95 | 117 |
| 96 void GrabWindowSnapshotAsync( | 118 void GrabWindowSnapshotAsync( |
| 97 gfx::NativeWindow window, | 119 gfx::NativeWindow window, |
| 98 const gfx::Rect& source_rect, | 120 const gfx::Rect& source_rect, |
| 99 scoped_refptr<base::TaskRunner> background_task_runner, | 121 scoped_refptr<base::TaskRunner> background_task_runner, |
| 100 const GrabWindowSnapshotAsyncPNGCallback& callback) { | 122 const GrabWindowSnapshotAsyncEncodedCallback& callback, |
| 123 SnapshotEncoding encoding, |
| 124 SnapshotQuality quality) { |
| 101 return GrabViewSnapshotAsync([[window contentView] superview], source_rect, | 125 return GrabViewSnapshotAsync([[window contentView] superview], source_rect, |
| 102 background_task_runner, callback); | 126 background_task_runner, callback, encoding, |
| 127 quality); |
| 103 } | 128 } |
| 104 | 129 |
| 105 } // namespace ui | 130 } // namespace ui |
| OLD | NEW |