| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/browser/android/content_readback_handler.h" | 5 #include "content/browser/android/content_readback_handler.h" |
| 6 | 6 |
| 7 #include "base/android/jni_android.h" | 7 #include "base/android/jni_android.h" |
| 8 #include "base/bind.h" | 8 #include "base/bind.h" |
| 9 #include "cc/output/copy_output_request.h" | 9 #include "cc/output/copy_output_request.h" |
| 10 #include "cc/output/copy_output_result.h" | 10 #include "cc/output/copy_output_result.h" |
| 11 #include "content/browser/android/content_view_core_impl.h" | 11 #include "content/browser/android/content_view_core_impl.h" |
| 12 #include "jni/ContentReadbackHandler_jni.h" | 12 #include "jni/ContentReadbackHandler_jni.h" |
| 13 #include "third_party/skia/include/core/SkBitmap.h" | 13 #include "third_party/skia/include/core/SkBitmap.h" |
| 14 #include "ui/base/android/window_android.h" | 14 #include "ui/base/android/window_android.h" |
| 15 #include "ui/base/android/window_android_compositor.h" | 15 #include "ui/base/android/window_android_compositor.h" |
| 16 #include "ui/gfx/android/java_bitmap.h" | 16 #include "ui/gfx/android/java_bitmap.h" |
| 17 #include "ui/gfx/rect.h" | 17 #include "ui/gfx/rect.h" |
| 18 | 18 |
| 19 namespace content { | 19 namespace content { |
| 20 | 20 |
| 21 namespace { | 21 namespace { |
| 22 | 22 |
| 23 typedef base::Callback<void(bool, const SkBitmap&)> ResultCallback; | |
| 24 | |
| 25 void OnFinishCopyOutputRequest( | 23 void OnFinishCopyOutputRequest( |
| 26 const ResultCallback& result_callback, | 24 const ReadbackRequestCallback& result_callback, |
| 27 scoped_ptr<cc::CopyOutputResult> copy_output_result) { | 25 scoped_ptr<cc::CopyOutputResult> copy_output_result) { |
| 28 if (!copy_output_result->HasBitmap()) { | 26 if (!copy_output_result->HasBitmap()) { |
| 29 result_callback.Run(false, SkBitmap()); | 27 result_callback.Run(SkBitmap(), READBACK_FAILED); |
| 30 return; | 28 return; |
| 31 } | 29 } |
| 32 | 30 |
| 33 scoped_ptr<SkBitmap> bitmap = copy_output_result->TakeBitmap(); | 31 scoped_ptr<SkBitmap> bitmap = copy_output_result->TakeBitmap(); |
| 34 result_callback.Run(true, *bitmap); | 32 result_callback.Run(*bitmap, READBACK_SUCCESS); |
| 35 } | 33 } |
| 36 | 34 |
| 37 } // anonymous namespace | 35 } // anonymous namespace |
| 38 | 36 |
| 39 // static | 37 // static |
| 40 bool ContentReadbackHandler::RegisterContentReadbackHandler(JNIEnv* env) { | 38 bool ContentReadbackHandler::RegisterContentReadbackHandler(JNIEnv* env) { |
| 41 return RegisterNativesImpl(env); | 39 return RegisterNativesImpl(env); |
| 42 } | 40 } |
| 43 | 41 |
| 44 ContentReadbackHandler::ContentReadbackHandler(JNIEnv* env, jobject obj) | 42 ContentReadbackHandler::ContentReadbackHandler(JNIEnv* env, jobject obj) |
| (...skipping 12 matching lines...) Expand all Loading... |
| 57 jobject color_type, | 55 jobject color_type, |
| 58 jfloat x, | 56 jfloat x, |
| 59 jfloat y, | 57 jfloat y, |
| 60 jfloat width, | 58 jfloat width, |
| 61 jfloat height, | 59 jfloat height, |
| 62 jobject content_view_core) { | 60 jobject content_view_core) { |
| 63 ContentViewCore* view = | 61 ContentViewCore* view = |
| 64 ContentViewCore::GetNativeContentViewCore(env, content_view_core); | 62 ContentViewCore::GetNativeContentViewCore(env, content_view_core); |
| 65 DCHECK(view); | 63 DCHECK(view); |
| 66 | 64 |
| 67 ResultCallback result_callback = | 65 ReadbackRequestCallback result_callback = |
| 68 base::Bind(&ContentReadbackHandler::OnFinishReadback, | 66 base::Bind(&ContentReadbackHandler::OnFinishReadback, |
| 69 weak_factory_.GetWeakPtr(), | 67 weak_factory_.GetWeakPtr(), readback_id); |
| 70 readback_id); | |
| 71 | 68 |
| 72 SkColorType sk_color_type = gfx::ConvertToSkiaColorType(color_type); | 69 SkColorType sk_color_type = gfx::ConvertToSkiaColorType(color_type); |
| 73 view->GetScaledContentBitmap( | 70 view->GetScaledContentBitmap( |
| 74 scale, sk_color_type, gfx::Rect(x, y, width, height), result_callback); | 71 scale, sk_color_type, gfx::Rect(x, y, width, height), result_callback); |
| 75 } | 72 } |
| 76 | 73 |
| 77 void ContentReadbackHandler::GetCompositorBitmap(JNIEnv* env, | 74 void ContentReadbackHandler::GetCompositorBitmap(JNIEnv* env, |
| 78 jobject obj, | 75 jobject obj, |
| 79 jint readback_id, | 76 jint readback_id, |
| 80 jlong native_window_android) { | 77 jlong native_window_android) { |
| 81 ui::WindowAndroid* window_android = | 78 ui::WindowAndroid* window_android = |
| 82 reinterpret_cast<ui::WindowAndroid*>(native_window_android); | 79 reinterpret_cast<ui::WindowAndroid*>(native_window_android); |
| 83 DCHECK(window_android); | 80 DCHECK(window_android); |
| 84 | 81 |
| 85 ResultCallback result_callback = | 82 ReadbackRequestCallback result_callback = |
| 86 base::Bind(&ContentReadbackHandler::OnFinishReadback, | 83 base::Bind(&ContentReadbackHandler::OnFinishReadback, |
| 87 weak_factory_.GetWeakPtr(), | 84 weak_factory_.GetWeakPtr(), readback_id); |
| 88 readback_id); | |
| 89 | 85 |
| 90 base::Callback<void(scoped_ptr<cc::CopyOutputResult>)> copy_output_callback = | 86 base::Callback<void(scoped_ptr<cc::CopyOutputResult>)> copy_output_callback = |
| 91 base::Bind(&OnFinishCopyOutputRequest, | 87 base::Bind(&OnFinishCopyOutputRequest, |
| 92 result_callback); | 88 result_callback); |
| 93 | 89 |
| 94 ui::WindowAndroidCompositor* compositor = window_android->GetCompositor(); | 90 ui::WindowAndroidCompositor* compositor = window_android->GetCompositor(); |
| 95 | 91 |
| 96 if (!compositor) { | 92 if (!compositor) { |
| 97 copy_output_callback.Run(cc::CopyOutputResult::CreateEmptyResult()); | 93 copy_output_callback.Run(cc::CopyOutputResult::CreateEmptyResult()); |
| 98 return; | 94 return; |
| 99 } | 95 } |
| 100 | 96 |
| 101 compositor->RequestCopyOfOutputOnRootLayer( | 97 compositor->RequestCopyOfOutputOnRootLayer( |
| 102 cc::CopyOutputRequest::CreateBitmapRequest(copy_output_callback)); | 98 cc::CopyOutputRequest::CreateBitmapRequest(copy_output_callback)); |
| 103 } | 99 } |
| 104 | 100 |
| 105 ContentReadbackHandler::~ContentReadbackHandler() {} | 101 ContentReadbackHandler::~ContentReadbackHandler() {} |
| 106 | 102 |
| 107 void ContentReadbackHandler::OnFinishReadback(int readback_id, | 103 void ContentReadbackHandler::OnFinishReadback( |
| 108 bool success, | 104 int readback_id, |
| 109 const SkBitmap& bitmap) { | 105 const SkBitmap& bitmap, |
| 106 const ReadbackResponse& response) { |
| 110 JNIEnv* env = base::android::AttachCurrentThread(); | 107 JNIEnv* env = base::android::AttachCurrentThread(); |
| 111 ScopedJavaLocalRef<jobject> java_bitmap; | 108 ScopedJavaLocalRef<jobject> java_bitmap; |
| 112 if (success) | 109 if (response == READBACK_SUCCESS) |
| 113 java_bitmap = gfx::ConvertToJavaBitmap(&bitmap); | 110 java_bitmap = gfx::ConvertToJavaBitmap(&bitmap); |
| 114 | 111 |
| 115 Java_ContentReadbackHandler_notifyGetBitmapFinished( | 112 Java_ContentReadbackHandler_notifyGetBitmapFinished( |
| 116 env, java_obj_.obj(), readback_id, java_bitmap.obj()); | 113 env, java_obj_.obj(), readback_id, java_bitmap.obj()); |
| 117 } | 114 } |
| 118 | 115 |
| 119 // static | 116 // static |
| 120 static jlong Init(JNIEnv* env, jobject obj) { | 117 static jlong Init(JNIEnv* env, jobject obj) { |
| 121 ContentReadbackHandler* content_readback_handler = | 118 ContentReadbackHandler* content_readback_handler = |
| 122 new ContentReadbackHandler(env, obj); | 119 new ContentReadbackHandler(env, obj); |
| 123 return reinterpret_cast<intptr_t>(content_readback_handler); | 120 return reinterpret_cast<intptr_t>(content_readback_handler); |
| 124 } | 121 } |
| 125 | 122 |
| 126 } // namespace content | 123 } // namespace content |
| OLD | NEW |