| 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 "content/shell/webkit_test_runner.h" | 5 #include "content/shell/webkit_test_runner.h" |
| 6 | 6 |
| 7 #include "base/md5.h" | 7 #include "base/md5.h" |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "base/message_loop.h" | 9 #include "base/message_loop.h" |
| 10 #include "base/stringprintf.h" | 10 #include "base/stringprintf.h" |
| (...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 103 | 103 |
| 104 if (recursive) { | 104 if (recursive) { |
| 105 for (WebFrame* child = frame->firstChild(); child; | 105 for (WebFrame* child = frame->firstChild(); child; |
| 106 child = child->nextSibling()) { | 106 child = child->nextSibling()) { |
| 107 result.append(DumpFrameScrollPosition(child, recursive)); | 107 result.append(DumpFrameScrollPosition(child, recursive)); |
| 108 } | 108 } |
| 109 } | 109 } |
| 110 return result; | 110 return result; |
| 111 } | 111 } |
| 112 | 112 |
| 113 bool PaintViewIntoCanvas(WebView* view, skia::PlatformCanvas& canvas) { | 113 SkCanvas* PaintViewIntoCanvas(WebView* view) { |
| 114 view->layout(); | 114 view->layout(); |
| 115 const WebSize& size = view->size(); | 115 const WebSize& size = view->size(); |
| 116 | 116 |
| 117 if (!canvas.initialize(size.width, size.height, true)) | 117 SkCanvas* canvas = skia::CreatePlatformCanvas(size.width, size.height, true, |
| 118 return false; | 118 0, skia::RETURN_NULL_ON_FAILURE); |
| 119 | 119 if (canvas) { |
| 120 view->paint(webkit_glue::ToWebCanvas(&canvas), | 120 view->paint(webkit_glue::ToWebCanvas(canvas), |
| 121 WebRect(0, 0, size.width, size.height)); | 121 WebRect(0, 0, size.width, size.height)); |
| 122 return true; | 122 } |
| 123 return canvas; |
| 123 } | 124 } |
| 124 | 125 |
| 125 #if !defined(OS_MACOSX) | 126 #if !defined(OS_MACOSX) |
| 126 void MakeBitmapOpaque(SkBitmap* bitmap) { | 127 void MakeBitmapOpaque(SkBitmap* bitmap) { |
| 127 SkAutoLockPixels lock(*bitmap); | 128 SkAutoLockPixels lock(*bitmap); |
| 128 DCHECK(bitmap->config() == SkBitmap::kARGB_8888_Config); | 129 DCHECK(bitmap->config() == SkBitmap::kARGB_8888_Config); |
| 129 for (int y = 0; y < bitmap->height(); ++y) { | 130 for (int y = 0; y < bitmap->height(); ++y) { |
| 130 uint32_t* row = bitmap->getAddr32(0, y); | 131 uint32_t* row = bitmap->getAddr32(0, y); |
| 131 for (int x = 0; x < bitmap->width(); ++x) | 132 for (int x = 0; x < bitmap->width(); ++x) |
| 132 row[x] |= 0xFF000000; // Set alpha bits to 1. | 133 row[x] |= 0xFF000000; // Set alpha bits to 1. |
| 133 } | 134 } |
| 134 } | 135 } |
| 135 #endif | 136 #endif |
| 136 | 137 |
| 137 void CaptureSnapshot(WebView* view, SkBitmap* snapshot) { | 138 void CaptureSnapshot(WebView* view, SkBitmap* snapshot) { |
| 138 skia::PlatformCanvas canvas; | 139 SkCanvas* canvas = PaintViewIntoCanvas(view); |
| 139 if (!PaintViewIntoCanvas(view, canvas)) | 140 if (!canvas) |
| 140 return; | 141 return; |
| 141 | 142 |
| 142 SkDevice* device = skia::GetTopDevice(canvas); | 143 SkDevice* device = skia::GetTopDevice(*canvas); |
| 143 | 144 |
| 144 const SkBitmap& bitmap = device->accessBitmap(false); | 145 const SkBitmap& bitmap = device->accessBitmap(false); |
| 145 bitmap.copyTo(snapshot, SkBitmap::kARGB_8888_Config); | 146 bitmap.copyTo(snapshot, SkBitmap::kARGB_8888_Config); |
| 146 | 147 |
| 147 #if !defined(OS_MACOSX) | 148 #if !defined(OS_MACOSX) |
| 148 // Only the expected PNGs for Mac have a valid alpha channel. | 149 // Only the expected PNGs for Mac have a valid alpha channel. |
| 149 MakeBitmapOpaque(snapshot); | 150 MakeBitmapOpaque(snapshot); |
| 150 #endif | 151 #endif |
| 151 | 152 |
| 152 } | 153 } |
| (...skipping 162 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 315 Send(new ShellViewHostMsg_ImageDump( | 316 Send(new ShellViewHostMsg_ImageDump( |
| 316 routing_id(), actual_pixel_hash, snapshot)); | 317 routing_id(), actual_pixel_hash, snapshot)); |
| 317 } | 318 } |
| 318 | 319 |
| 319 void WebKitTestRunner::OnSetIsMainWindow() { | 320 void WebKitTestRunner::OnSetIsMainWindow() { |
| 320 is_main_window_ = true; | 321 is_main_window_ = true; |
| 321 ShellRenderProcessObserver::GetInstance()->SetMainWindow(render_view(), this); | 322 ShellRenderProcessObserver::GetInstance()->SetMainWindow(render_view(), this); |
| 322 } | 323 } |
| 323 | 324 |
| 324 } // namespace content | 325 } // namespace content |
| OLD | NEW |