[Android] Implement asynchronous zero-copy bitmap capture API.

content/browser/android/content_view_core_impl.cc

 // Copyright 2012 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 "content/browser/android/content_view_core_impl.h"
 
 #include "base/android/jni_android.h"
 #include "base/android/jni_array.h"
 #include "base/android/jni_string.h"
-#include "base/android/scoped_java_ref.h"
 #include "base/command_line.h"
 #include "base/json/json_writer.h"
 #include "base/logging.h"
 #include "base/metrics/histogram.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #include "cc/layers/layer.h"
 #include "cc/output/begin_frame_args.h"
 #include "content/browser/android/gesture_event_type.h"
 #include "content/browser/android/interstitial_page_delegate_android.h"
 #include "content/browser/android/load_url_params.h"
 #include "content/browser/frame_host/interstitial_page_impl.h"
 #include "content/browser/frame_host/navigation_controller_impl.h"
 #include "content/browser/frame_host/navigation_entry_impl.h"
 #include "content/browser/media/android/browser_media_player_manager.h"
 #include "content/browser/renderer_host/compositor_impl_android.h"
+#include "content/browser/renderer_host/dip_util.h"
 #include "content/browser/renderer_host/input/web_input_event_builders_android.h"
 #include "content/browser/renderer_host/java/java_bound_object.h"
 #include "content/browser/renderer_host/java/java_bridge_dispatcher_host_manager.h"
 #include "content/browser/renderer_host/render_view_host_impl.h"
 #include "content/browser/renderer_host/render_widget_host_impl.h"
 #include "content/browser/renderer_host/render_widget_host_view_android.h"
 #include "content/browser/ssl/ssl_host_state.h"
 #include "content/browser/web_contents/web_contents_view_android.h"
 #include "content/common/input/web_input_event_traits.h"
 #include "content/common/input_messages.h"
@@ skipping 413 matching lines @@
 }
 
 void ContentViewCoreImpl::OnTabCrashed() {
   JNIEnv* env = AttachCurrentThread();
   ScopedJavaLocalRef<jobject> obj = java_ref_.get(env);
   if (obj.is_null())
     return;
   Java_ContentViewCore_resetVSyncNotification(env, obj.obj());
 }
 
+void ContentViewCoreImpl::OnBitmapReady(bool result, const SkBitmap& bitmap) {
+  JNIEnv* env = AttachCurrentThread();
+  ScopedJavaLocalRef<jobject> obj = java_ref_.get(env);
+  if (obj.is_null())
+      return;
+  java_bitmap_.reset(NULL);
+  Java_ContentViewCore_onBitmapReady(env, obj.obj(),
+                                     result,
+                                     jbitmap_.Release());
+}
+
 // All positions and sizes are in CSS pixels.
 // Note that viewport_width/height is a best effort based.
 // ContentViewCore has the actual information about the physical viewport size.
 void ContentViewCoreImpl::UpdateFrameInfo(
     const gfx::Vector2dF& scroll_offset,
     float page_scale_factor,
     const gfx::Vector2dF& page_scale_factor_limits,
     const gfx::SizeF& content_size,
     const gfx::SizeF& viewport_size,
     const gfx::Vector2dF& controls_offset,
@@ skipping 93 matching lines @@
                 POPUP_ITEM_TYPE_DISABLED));
     env->SetIntArrayRegion(enabled_array.obj(), i, 1, &enabled);
   }
   ScopedJavaLocalRef<jobjectArray> items_array(
       base::android::ToJavaArrayOfStrings(env, labels));
   Java_ContentViewCore_showSelectPopup(env, j_obj.obj(),
                                        items_array.obj(), enabled_array.obj(),
                                        multiple, selected_array.obj());
 }
 
+void ContentViewCoreImpl::ProxyBitmapAllocator(SkBitmap*& bitmap,
+                                               const gfx::Size& computed_size,
+                                               SkBitmap::Config bitmap_config)
+{
+  const gfx::Display& display =
+      gfx::Screen::GetNativeScreen()->GetPrimaryDisplay();
+  float device_scale_factor = display.device_scale_factor();
+  gfx::Size dst_size_in_pixel =
+      ConvertRectToPixel(device_scale_factor, gfx::Rect(computed_size)).size();
+
+  base::android::ScopedJavaLocalRef<jobject> new_bitmap =
+      gfx::CreateJavaBitmap(dst_size_in_pixel.width(),
+                            dst_size_in_pixel.height(),
+                            (bitmap_config == SkBitmap::kRGB_565_Config));
+  jbitmap_.Reset(AttachCurrentThread(), new_bitmap.obj());
+  java_bitmap_.reset(new gfx::JavaBitmap(jbitmap_.obj()));
+
+  bitmap = new SkBitmap();
+  bitmap->setConfig(bitmap_config, dst_size_in_pixel.width() ,
+                    dst_size_in_pixel.height());
+  bitmap->setPixels(java_bitmap_->pixels());
+}
+
 void ContentViewCoreImpl::ConfirmTouchEvent(InputEventAckState ack_result) {
   touch_disposition_gesture_filter_.OnTouchEventAck(ack_result);
 }
 
 void ContentViewCoreImpl::OnGestureEventAck(const blink::WebGestureEvent& event,
                                             InputEventAckState ack_result) {
   JNIEnv* env = AttachCurrentThread();
   ScopedJavaLocalRef<jobject> j_obj = java_ref_.get(env);
   if (j_obj.is_null())
     return;
@@ skipping 108 matching lines @@
   ScopedJavaLocalRef<jobject> obj = java_ref_.get(env);
   if (obj.is_null())
     return;
   Java_ContentViewCore_showPastePopup(env, obj.obj(),
                                       static_cast<jint>(x_dip),
                                       static_cast<jint>(y_dip));
 }
 
 void ContentViewCoreImpl::GetScaledContentBitmap(
     float scale,
-    const base::Callback<void(bool, const SkBitmap&)>& result_callback) {
+    const base::Callback<void(bool, const SkBitmap&)>& result_callback,
+    SkBitmap::Config bitmap_config,
+    const gfx::Rect& bounding_rect,
+    const BitmapAllocator bitmap_allocator) {
   RenderWidgetHostViewAndroid* view = GetRenderWidgetHostViewAndroid();
   if (!view) {
     result_callback.Run(false, SkBitmap());
     return;
   }
 
-  view->GetScaledContentBitmap(scale, result_callback);
+  view->GetScaledContentBitmap(scale, result_callback,
+                               bitmap_config, bounding_rect, bitmap_allocator);
 }
 
 void ContentViewCoreImpl::StartContentIntent(const GURL& content_url) {
   JNIEnv* env = AttachCurrentThread();
   ScopedJavaLocalRef<jobject> j_obj = java_ref_.get(env);
   if (j_obj.is_null())
     return;
   ScopedJavaLocalRef<jstring> jcontent_url =
       ConvertUTF8ToJavaString(env, content_url.spec());
   Java_ContentViewCore_startContentIntent(env,
@@ skipping 743 matching lines @@
 jboolean ContentViewCoreImpl::PopulateBitmapFromCompositor(JNIEnv* env,
                                                            jobject obj,
                                                            jobject jbitmap) {
   RenderWidgetHostViewAndroid* view = GetRenderWidgetHostViewAndroid();
   if (!view)
     return false;
 
   return view->PopulateBitmapWithContents(jbitmap);
 }
 
+void ContentViewCoreImpl::PopulateBitmapFromCompositorAsync(
+    JNIEnv* env,
+    jobject obj,
+    jint x,
+    jint y,
+    jint width,
+    jint height,
+    jfloat scale,
+    jobject jbitmap_config) {
+  RenderWidgetHostViewAndroid* view = GetRenderWidgetHostViewAndroid();
+  if (!view)
+      return;
+
+  base::Callback<void(bool, const SkBitmap&)> callback =
+    base::Bind(&ContentViewCoreImpl::OnBitmapReady,
+               base::Unretained(this));
+
+  BitmapAllocator bitmap_allocator =
+    base::Bind(&ContentViewCoreImpl::ProxyBitmapAllocator,
+               base::Unretained(this));
+
+  //SkBitmap::Config bitmap_config = gfx::ConvertToSkiaConfig(jbitmap_config);

[vivekg, 2014/02/26 01:44:46]

This would be enabled once https://codereview.chromium.org/157033007/ lands.

+  SkBitmap::Config bitmap_config = SkBitmap::kARGB_8888_Config;
+
+  GetScaledContentBitmap(scale, callback, bitmap_config,
+                         gfx::Rect(x, y, width, height),
+                         bitmap_allocator);
+}
+
 void ContentViewCoreImpl::WasResized(JNIEnv* env, jobject obj) {
   RenderWidgetHostViewAndroid* view = GetRenderWidgetHostViewAndroid();
   if (view)
     view->WasResized();
 }
 
 void ContentViewCoreImpl::ShowInterstitialPage(
     JNIEnv* env, jobject obj, jstring jurl, jint delegate_ptr) {
   GURL url(base::android::ConvertJavaStringToUTF8(env, jurl));
   InterstitialPageDelegateAndroid* delegate =
@@ skipping 350 matching lines @@
       reinterpret_cast<ui::ViewAndroid*>(view_android),
       reinterpret_cast<ui::WindowAndroid*>(window_android));
   return reinterpret_cast<intptr_t>(view);
 }
 
 bool RegisterContentViewCore(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }
 
 }  // namespace content