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

Unified Diff: extensions/browser/api/capture_web_contents_function.cc

Issue 1614703003: Revert of Implement webview.captureVisibleRegion() (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 4 years, 11 months 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 side-by-side diff with in-line comments
Download patch
Index: extensions/browser/api/capture_web_contents_function.cc
diff --git a/extensions/browser/api/capture_web_contents_function.cc b/extensions/browser/api/capture_web_contents_function.cc
new file mode 100644
index 0000000000000000000000000000000000000000..e10842d8aee0aa103031261e8dec0ca9045f5479
--- /dev/null
+++ b/extensions/browser/api/capture_web_contents_function.cc
@@ -0,0 +1,150 @@
+// Copyright 2013 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "extensions/browser/api/capture_web_contents_function.h"
+
+#include "base/base64.h"
+#include "base/strings/stringprintf.h"
+#include "content/public/browser/render_widget_host.h"
+#include "content/public/browser/render_widget_host_view.h"
+#include "content/public/browser/web_contents.h"
+#include "extensions/browser/extension_function.h"
+#include "extensions/common/constants.h"
+#include "ui/gfx/codec/jpeg_codec.h"
+#include "ui/gfx/codec/png_codec.h"
+#include "ui/gfx/display.h"
+#include "ui/gfx/geometry/size_conversions.h"
+#include "ui/gfx/screen.h"
+
+using content::RenderWidgetHost;
+using content::RenderWidgetHostView;
+using content::WebContents;
+
+namespace extensions {
+
+using api::extension_types::ImageDetails;
+
+bool CaptureWebContentsFunction::HasPermission() {
+ return true;
+}
+
+bool CaptureWebContentsFunction::RunAsync() {
+ EXTENSION_FUNCTION_VALIDATE(args_);
+
+ context_id_ = extension_misc::kCurrentWindowId;
+ args_->GetInteger(0, &context_id_);
+
+ scoped_ptr<ImageDetails> image_details;
+ if (args_->GetSize() > 1) {
+ base::Value* spec = NULL;
+ EXTENSION_FUNCTION_VALIDATE(args_->Get(1, &spec) && spec);
+ image_details = ImageDetails::FromValue(*spec);
+ }
+
+ if (!IsScreenshotEnabled())
+ return false;
+
+ WebContents* contents = GetWebContentsForID(context_id_);
+ if (!contents)
+ return false;
+
+ // The default format and quality setting used when encoding jpegs.
+ const api::extension_types::ImageFormat kDefaultFormat =
+ api::extension_types::IMAGE_FORMAT_JPEG;
+ const int kDefaultQuality = 90;
+
+ image_format_ = kDefaultFormat;
+ image_quality_ = kDefaultQuality;
+
+ if (image_details) {
+ if (image_details->format != api::extension_types::IMAGE_FORMAT_NONE)
+ image_format_ = image_details->format;
+ if (image_details->quality.get())
+ image_quality_ = *image_details->quality;
+ }
+
+ // TODO(miu): Account for fullscreen render widget? http://crbug.com/419878
+ RenderWidgetHostView* const view = contents->GetRenderWidgetHostView();
+ RenderWidgetHost* const host = view ? view->GetRenderWidgetHost() : nullptr;
+ if (!view || !host) {
+ OnCaptureFailure(FAILURE_REASON_VIEW_INVISIBLE);
+ return false;
+ }
+
+ // By default, the requested bitmap size is the view size in screen
+ // coordinates. However, if there's more pixel detail available on the
+ // current system, increase the requested bitmap size to capture it all.
+ const gfx::Size view_size = view->GetViewBounds().size();
+ gfx::Size bitmap_size = view_size;
+ const gfx::NativeView native_view = view->GetNativeView();
+ gfx::Screen* const screen = gfx::Screen::GetScreenFor(native_view);
+ const float scale =
+ screen->GetDisplayNearestWindow(native_view).device_scale_factor();
+ if (scale > 1.0f)
+ bitmap_size = gfx::ScaleToCeiledSize(view_size, scale);
+
+ host->CopyFromBackingStore(
+ gfx::Rect(view_size),
+ bitmap_size,
+ base::Bind(&CaptureWebContentsFunction::CopyFromBackingStoreComplete,
+ this),
+ kN32_SkColorType);
+ return true;
+}
+
+void CaptureWebContentsFunction::CopyFromBackingStoreComplete(
+ const SkBitmap& bitmap,
+ content::ReadbackResponse response) {
+ if (response == content::READBACK_SUCCESS) {
+ OnCaptureSuccess(bitmap);
+ return;
+ }
+ OnCaptureFailure(FAILURE_REASON_UNKNOWN);
+}
+
+void CaptureWebContentsFunction::OnCaptureSuccess(const SkBitmap& bitmap) {
+ std::vector<unsigned char> data;
+ SkAutoLockPixels screen_capture_lock(bitmap);
+ bool encoded = false;
+ std::string mime_type;
+ switch (image_format_) {
+ case api::extension_types::IMAGE_FORMAT_JPEG:
+ encoded = gfx::JPEGCodec::Encode(
+ reinterpret_cast<unsigned char*>(bitmap.getAddr32(0, 0)),
+ gfx::JPEGCodec::FORMAT_SkBitmap,
+ bitmap.width(),
+ bitmap.height(),
+ static_cast<int>(bitmap.rowBytes()),
+ image_quality_,
+ &data);
+ mime_type = kMimeTypeJpeg;
+ break;
+ case api::extension_types::IMAGE_FORMAT_PNG:
+ encoded =
+ gfx::PNGCodec::EncodeBGRASkBitmap(bitmap,
+ true, // Discard transparency.
+ &data);
+ mime_type = kMimeTypePng;
+ break;
+ default:
+ NOTREACHED() << "Invalid image format.";
+ }
+
+ if (!encoded) {
+ OnCaptureFailure(FAILURE_REASON_ENCODING_FAILED);
+ return;
+ }
+
+ std::string base64_result;
+ base::StringPiece stream_as_string(reinterpret_cast<const char*>(data.data()),
+ data.size());
+
+ base::Base64Encode(stream_as_string, &base64_result);
+ base64_result.insert(
+ 0, base::StringPrintf("data:%s;base64,", mime_type.c_str()));
+ SetResult(new base::StringValue(base64_result));
+ SendResponse(true);
+}
+
+} // namespace extensions

Powered by Google App Engine
This is Rietveld 408576698