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 3 // found in the LICENSE file. 4 5 #include "ui/snapshot/snapshot.h" 6 7 #include "base/logging.h" 8 #include "base/safe_numerics.h" 9 #include "third_party/skia/include/core/SkBitmap.h" 10 #include "third_party/skia/include/core/SkPixelRef.h" 11 #include "ui/aura/root_window.h" 12 #include "ui/aura/window.h" 13 #include "ui/compositor/compositor.h" 14 #include "ui/compositor/dip_util.h" 15 #include "ui/compositor/layer.h" 16 #include "ui/gfx/codec/png_codec.h" 17 #include "ui/gfx/display.h" 18 #include "ui/gfx/rect.h" 19 #include "ui/gfx/rect_conversions.h" 20 #include "ui/gfx/rect_f.h" 21 #include "ui/gfx/screen.h" 22 #include "ui/gfx/skbitmap_operations.h" 23 #include "ui/gfx/transform.h" 24 25 namespace ui { 26 27 bool GrabViewSnapshot(gfx::NativeView view, 28 std::vector<unsigned char>* png_representation, 29 const gfx::Rect& snapshot_bounds) { 30 return GrabWindowSnapshot(view, png_representation, snapshot_bounds); 31 } 32 33 bool GrabWindowSnapshot(gfx::NativeWindow window, 34 std::vector<unsigned char>* png_representation, 35 const gfx::Rect& snapshot_bounds) { 36 ui::Compositor* compositor = window->layer()->GetCompositor(); 37 38 gfx::RectF read_pixels_bounds = snapshot_bounds; 39 40 // We must take into account the window's position on the desktop. 41 read_pixels_bounds.Offset( 42 window->GetBoundsInRootWindow().origin().OffsetFromOrigin()); 43 aura::RootWindow* root_window = window->GetRootWindow(); 44 if (root_window) 45 root_window->GetRootTransform().TransformRect(&read_pixels_bounds); 46 47 gfx::Rect read_pixels_bounds_in_pixel = 48 gfx::ToEnclosingRect(read_pixels_bounds); 49 50 // Sometimes (i.e. when using Aero on Windows) the compositor's size is 51 // smaller than the window bounds. So trim appropriately. 52 read_pixels_bounds_in_pixel.Intersect(gfx::Rect(compositor->size())); 53 54 DCHECK_LE(0, read_pixels_bounds.x()); 55 DCHECK_LE(0, read_pixels_bounds.y()); 56 57 SkBitmap bitmap; 58 if (!compositor->ReadPixels(&bitmap, read_pixels_bounds_in_pixel)) 59 return false; 60 61 gfx::Display display = 62 gfx::Screen::GetScreenFor(window)->GetDisplayNearestWindow(window); 63 switch (display.rotation()) { 64 case gfx::Display::ROTATE_0: 65 break; 66 case gfx::Display::ROTATE_90: 67 bitmap = SkBitmapOperations::Rotate( 68 bitmap, SkBitmapOperations::ROTATION_270_CW); 69 break; 70 case gfx::Display::ROTATE_180: 71 bitmap = SkBitmapOperations::Rotate( 72 bitmap, SkBitmapOperations::ROTATION_180_CW); 73 break; 74 case gfx::Display::ROTATE_270: 75 bitmap = SkBitmapOperations::Rotate( 76 bitmap, SkBitmapOperations::ROTATION_90_CW); 77 break; 78 } 79 80 unsigned char* pixels = reinterpret_cast<unsigned char*>( 81 bitmap.pixelRef()->pixels()); 82 gfx::PNGCodec::Encode(pixels, gfx::PNGCodec::FORMAT_BGRA, 83 gfx::Size(bitmap.width(), bitmap.height()), 84 base::checked_numeric_cast<int>(bitmap.rowBytes()), 85 true, std::vector<gfx::PNGCodec::Comment>(), 86 png_representation); 87 return true; 88 } 89 90 } // namespace ui 91
