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 |