snapshot_aura.cc revision 1e9bf3e0803691d0a228da41fc608347b6db4340 
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
25namespace ui {
26
27bool 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
33bool 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::WindowEventDispatcher* dispatcher = window->GetDispatcher();
44  if (dispatcher)
45    dispatcher->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