Index: chrome/renderer/automation/automation_renderer_helper.cc |
=================================================================== |
--- chrome/renderer/automation/automation_renderer_helper.cc (revision 107646) |
+++ chrome/renderer/automation/automation_renderer_helper.cc (working copy) |
@@ -4,120 +4,21 @@ |
#include "chrome/renderer/automation/automation_renderer_helper.h" |
-#include <algorithm> |
- |
#include "base/basictypes.h" |
#include "chrome/common/automation_messages.h" |
-#include "content/public/renderer/render_view.h" |
-#include "skia/ext/platform_canvas.h" |
#include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h" |
-#include "third_party/WebKit/Source/WebKit/chromium/public/WebSize.h" |
#include "third_party/WebKit/Source/WebKit/chromium/public/WebURL.h" |
-#include "third_party/WebKit/Source/WebKit/chromium/public/WebView.h" |
-#include "ui/gfx/codec/png_codec.h" |
-#include "ui/gfx/rect.h" |
-#include "webkit/glue/webkit_glue.h" |
using WebKit::WebFrame; |
-using WebKit::WebSize; |
using WebKit::WebURL; |
-using WebKit::WebView; |
AutomationRendererHelper::AutomationRendererHelper( |
content::RenderView* render_view) |
- : content::RenderViewObserver(render_view), |
- content::RenderViewObserverTracker<AutomationRendererHelper>( |
- render_view) { |
+ : content::RenderViewObserver(render_view) { |
} |
AutomationRendererHelper::~AutomationRendererHelper() { } |
-bool AutomationRendererHelper::SnapshotEntirePage( |
- WebView* view, |
- std::vector<unsigned char>* png_data, |
- std::string* error_msg) { |
- WebFrame* frame = view->mainFrame(); |
- WebSize old_size = view->size(); |
- WebSize new_size = frame->contentsSize(); |
- // For RTL, the minimum scroll offset may be negative. |
- WebSize min_scroll = frame->minimumScrollOffset(); |
- WebSize old_scroll = frame->scrollOffset(); |
- bool fixed_layout_enabled = view->isFixedLayoutModeEnabled(); |
- WebSize fixed_size = view->fixedLayoutSize(); |
- |
- frame->setCanHaveScrollbars(false); |
- view->setFixedLayoutSize(old_size); |
- view->enableFixedLayoutMode(true); |
- view->resize(new_size); |
- view->layout(); |
- frame->setScrollOffset(WebSize(0, 0)); |
- |
- skia::PlatformCanvas canvas( |
- new_size.width, new_size.height, true /* is_opaque */); |
- view->paint(webkit_glue::ToWebCanvas(&canvas), |
- gfx::Rect(0, 0, new_size.width, new_size.height)); |
- |
- frame->setCanHaveScrollbars(true); |
- view->setFixedLayoutSize(fixed_size); |
- view->enableFixedLayoutMode(fixed_layout_enabled); |
- view->resize(old_size); |
- view->layout(); |
- frame->setScrollOffset(WebSize(old_scroll.width - min_scroll.width, |
- old_scroll.height - min_scroll.height)); |
- |
- const SkBitmap& bmp = skia::GetTopDevice(canvas)->accessBitmap(false); |
- SkAutoLockPixels lock_pixels(bmp); |
- // EncodeBGRA uses FORMAT_SkBitmap, which doesn't work on windows for some |
- // cases dealing with transparency. See crbug.com/96317. Use FORMAT_BGRA. |
- bool encode_success = gfx::PNGCodec::Encode( |
- reinterpret_cast<unsigned char*>(bmp.getPixels()), |
- gfx::PNGCodec::FORMAT_BGRA, |
- gfx::Size(bmp.width(), bmp.height()), |
- bmp.rowBytes(), |
- true, // discard_transparency |
- std::vector<gfx::PNGCodec::Comment>(), |
- png_data); |
- if (!encode_success) |
- *error_msg = "failed to encode image as png"; |
- return encode_success; |
-} |
- |
-void AutomationRendererHelper::OnSnapshotEntirePage() { |
- std::vector<unsigned char> png_data; |
- std::string error_msg; |
- bool success = false; |
- if (render_view()->GetWebView()) { |
- success = SnapshotEntirePage( |
- render_view()->GetWebView(), &png_data, &error_msg); |
- } else { |
- error_msg = "cannot snapshot page because webview is null"; |
- } |
- |
- // Check that the image is not too large, allowing a 1kb buffer for other |
- // message data. |
- if (success && png_data.size() > IPC::Channel::kMaximumMessageSize - 1024) { |
- png_data.clear(); |
- success = false; |
- error_msg = "image is too large to be transferred over ipc"; |
- } |
- Send(new AutomationMsg_SnapshotEntirePageACK( |
- routing_id(), success, png_data, error_msg)); |
-} |
- |
-bool AutomationRendererHelper::OnMessageReceived(const IPC::Message& message) { |
- bool handled = true; |
- bool deserialize_success = true; |
- IPC_BEGIN_MESSAGE_MAP_EX(AutomationRendererHelper, message, |
- deserialize_success) |
- IPC_MESSAGE_HANDLER(AutomationMsg_SnapshotEntirePage, OnSnapshotEntirePage) |
- IPC_MESSAGE_UNHANDLED(handled = false) |
- IPC_END_MESSAGE_MAP_EX() |
- if (!deserialize_success) { |
- LOG(ERROR) << "Failed to deserialize an IPC message"; |
- } |
- return handled; |
-} |
- |
void AutomationRendererHelper::WillPerformClientRedirect( |
WebFrame* frame, const WebURL& from, const WebURL& to, double interval, |
double fire_time) { |