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

Side by Side Diff: chrome/browser/ui/window_snapshot/window_snapshot_win.cc

Issue 8523022: Change snasphotting API to take in a bounds Rect (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 9 years, 1 month 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 | Annotate | Revision Log
OLDNEW
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2011 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 "chrome/browser/ui/window_snapshot/window_snapshot.h" 5 #include "chrome/browser/ui/window_snapshot/window_snapshot.h"
6 6
7 #include "base/win/scoped_gdi_object.h" 7 #include "base/win/scoped_gdi_object.h"
8 #include "base/win/scoped_hdc.h" 8 #include "base/win/scoped_hdc.h"
9 #include "base/win/scoped_select_object.h" 9 #include "base/win/scoped_select_object.h"
10 #include "ui/gfx/codec/png_codec.h" 10 #include "ui/gfx/codec/png_codec.h"
11 #include "ui/gfx/gdi_util.h" 11 #include "ui/gfx/gdi_util.h"
12 #include "ui/gfx/rect.h" 12 #include "ui/gfx/rect.h"
13 #include "ui/gfx/size.h" 13 #include "ui/gfx/size.h"
14 14
15 namespace browser { 15 namespace browser {
16 16
17 gfx::Rect GrabWindowSnapshot(gfx::NativeWindow window_handle, 17 gfx::Rect GrabWindowSnapshot(gfx::NativeWindow window_handle,
18 std::vector<unsigned char>* png_representation) { 18 std::vector<unsigned char>* png_representation,
19 gfx::Rect snapshot_bounds) {
19 // Create a memory DC that's compatible with the window. 20 // Create a memory DC that's compatible with the window.
20 HDC window_hdc = GetWindowDC(window_handle); 21 HDC window_hdc = GetWindowDC(window_handle);
21 base::win::ScopedCreateDC mem_hdc(CreateCompatibleDC(window_hdc)); 22 base::win::ScopedCreateDC mem_hdc(CreateCompatibleDC(window_hdc));
22 23
23 // Create a DIB that's the same size as the window. 24 // Create a DIB that's the same size as the window.
24 RECT content_rect = {0, 0, 0, 0}; 25 RECT content_rect = {0, 0, 0, 0};
25 ::GetWindowRect(window_handle, &content_rect); 26 ::GetWindowRect(window_handle, &content_rect);
26 content_rect.right++; // Match what PrintWindow wants. 27 content_rect.right++; // Match what PrintWindow wants.
27 int width = content_rect.right - content_rect.left; 28
28 int height = content_rect.bottom - content_rect.top; 29 if (snapshot_bounds.IsEmpty()) {
30 snapshot_bounds(gfx::Size(content_rect.right - content_rect.left,
31 content_rect.bottom - content_rect.top));
32 } else {
33 snapshot_bounds.Offset(content_rect.left, content_rect.top);
34 DCHECK(snapshot_bounds.right() <= content_rect.right);
35 DCHECK(snapshot_bounds.bottom() <= content_rect.bottom);
36 }
37
29 BITMAPINFOHEADER hdr; 38 BITMAPINFOHEADER hdr;
30 gfx::CreateBitmapHeader(width, height, &hdr); 39 gfx::CreateBitmapHeader(snapshot_bounds.width(),
40 snapshot_bounds.height(),
41 &hdr);
31 unsigned char *bit_ptr = NULL; 42 unsigned char *bit_ptr = NULL;
32 base::win::ScopedBitmap bitmap( 43 base::win::ScopedBitmap bitmap(
33 CreateDIBSection(mem_hdc, 44 CreateDIBSection(mem_hdc,
34 reinterpret_cast<BITMAPINFO*>(&hdr), 45 reinterpret_cast<BITMAPINFO*>(&hdr),
35 DIB_RGB_COLORS, 46 DIB_RGB_COLORS,
36 reinterpret_cast<void **>(&bit_ptr), 47 reinterpret_cast<void **>(&bit_ptr),
37 NULL, 0)); 48 NULL, 0));
38 49
39 base::win::ScopedSelectObject select_bitmap(mem_hdc, bitmap); 50 base::win::ScopedSelectObject select_bitmap(mem_hdc, bitmap);
40 // Clear the bitmap to white (so that rounded corners on windows 51 // Clear the bitmap to white (so that rounded corners on windows
41 // show up on a white background, and strangely-shaped windows 52 // show up on a white background, and strangely-shaped windows
42 // look reasonable). Not capturing an alpha mask saves a 53 // look reasonable). Not capturing an alpha mask saves a
43 // bit of space. 54 // bit of space.
44 PatBlt(mem_hdc, 0, 0, width, height, WHITENESS); 55 PatBlt(mem_hdc, 0, 0, snapshot_bounds.width(), snapshot_bounds.height(),
56 WHITENESS);
45 // Grab a copy of the window 57 // Grab a copy of the window
46 // First, see if PrintWindow is defined (it's not in Windows 2000). 58 // First, see if PrintWindow is defined (it's not in Windows 2000).
47 typedef BOOL (WINAPI *PrintWindowPointer)(HWND, HDC, UINT); 59 typedef BOOL (WINAPI *PrintWindowPointer)(HWND, HDC, UINT);
48 PrintWindowPointer print_window = 60 PrintWindowPointer print_window =
49 reinterpret_cast<PrintWindowPointer>( 61 reinterpret_cast<PrintWindowPointer>(
50 GetProcAddress(GetModuleHandle(L"User32.dll"), "PrintWindow")); 62 GetProcAddress(GetModuleHandle(L"User32.dll"), "PrintWindow"));
51 63
52 // If PrintWindow is defined, use it. It will work on partially 64 // If PrintWindow is defined, use it. It will work on partially
53 // obscured windows, and works better for out of process sub-windows. 65 // obscured windows, and works better for out of process sub-windows.
54 // Otherwise grab the bits we can get with BitBlt; it's better 66 // Otherwise grab the bits we can get with BitBlt; it's better
55 // than nothing and will work fine in the average case (window is 67 // than nothing and will work fine in the average case (window is
56 // completely on screen). 68 // completely on screen).
57 if (print_window) 69 if (snapshot_bounds.origin() != gfx::Point() && print_window)
58 (*print_window)(window_handle, mem_hdc, 0); 70 (*print_window)(window_handle, mem_hdc, 0);
59 else 71 else
60 BitBlt(mem_hdc, 0, 0, width, height, window_hdc, 0, 0, SRCCOPY); 72 BitBlt(mem_hdc, 0, 0, snapshot_bounds.width(), snapshot_bounds.height(),
73 window_hdc, snapshot_bounds.x(), snapshot_bounds.y(), SRCCOPY);
61 74
62 // We now have a copy of the window contents in a DIB, so 75 // We now have a copy of the window contents in a DIB, so
63 // encode it into a useful format for posting to the bug report 76 // encode it into a useful format for posting to the bug report
64 // server. 77 // server.
65 gfx::PNGCodec::Encode(bit_ptr, gfx::PNGCodec::FORMAT_BGRA, 78 gfx::PNGCodec::Encode(bit_ptr, gfx::PNGCodec::FORMAT_BGRA,
66 gfx::Size(width, height), width * 4, true, 79 snapshot_bounds.size(), width * 4, true,
67 std::vector<gfx::PNGCodec::Comment>(), 80 std::vector<gfx::PNGCodec::Comment>(),
68 png_representation); 81 png_representation);
69 82
70 ReleaseDC(window_handle, window_hdc); 83 ReleaseDC(window_handle, window_hdc);
71 84
72 return gfx::Rect(width, height); 85 return gfx::Rect(snapshot_bounds.size());
73 } 86 }
74 87
75 } // namespace browser 88 } // namespace browser
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698