android_view_GraphicBuffer.cpp revision 5c3d927e17e98e8fd4a9f3c86f7f4def0bcfa816 
1/*
2 * Copyright (C) 2013 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#define LOG_TAG "GraphicBuffer"
18
19#include "jni.h"
20#include "JNIHelp.h"
21
22#include "android_os_Parcel.h"
23#include "android_view_GraphicBuffer.h"
24#include "android/graphics/GraphicsJNI.h"
25
26#include <android_runtime/AndroidRuntime.h>
27
28#include <binder/Parcel.h>
29
30#include <ui/GraphicBuffer.h>
31#include <ui/PixelFormat.h>
32
33#include <gui/IGraphicBufferAlloc.h>
34#include <gui/ISurfaceComposer.h>
35
36#include <SkCanvas.h>
37#include <SkBitmap.h>
38
39#include <private/gui/ComposerService.h>
40
41namespace android {
42
43// ----------------------------------------------------------------------------
44// Defines
45// ----------------------------------------------------------------------------
46
47// Debug
48#define DEBUG_GRAPHIC_BUFFER 0
49
50// Debug
51#if DEBUG_GRAPHIC_BUFFER
52    #define GB_LOGD(...) ALOGD(__VA_ARGS__)
53    #define GB_LOGW(...) ALOGW(__VA_ARGS__)
54#else
55    #define GB_LOGD(...)
56    #define GB_LOGW(...)
57#endif
58
59#define LOCK_CANVAS_USAGE GraphicBuffer::USAGE_SW_READ_OFTEN | GraphicBuffer::USAGE_SW_WRITE_OFTEN
60
61// ----------------------------------------------------------------------------
62// JNI Helpers
63// ----------------------------------------------------------------------------
64
65static struct {
66    jfieldID mNativeObject;
67} gGraphicBufferClassInfo;
68
69static struct {
70    jmethodID set;
71    jfieldID left;
72    jfieldID top;
73    jfieldID right;
74    jfieldID bottom;
75} gRectClassInfo;
76
77static struct {
78    jfieldID mSurfaceFormat;
79    jmethodID setNativeBitmap;
80} gCanvasClassInfo;
81
82#define GET_INT(object, field) \
83    env->GetIntField(object, field)
84
85#define SET_INT(object, field, value) \
86    env->SetIntField(object, field, value)
87
88#define GET_LONG(object, field) \
89    env->GetLongField(object, field)
90
91#define SET_LONG(object, field, value) \
92    env->SetLongField(object, field, value)
93
94#define INVOKEV(object, method, ...) \
95    env->CallVoidMethod(object, method, __VA_ARGS__)
96
97// ----------------------------------------------------------------------------
98// Types
99// ----------------------------------------------------------------------------
100
101class GraphicBufferWrapper {
102public:
103    GraphicBufferWrapper(const sp<GraphicBuffer>& buffer): buffer(buffer) {
104    }
105
106    sp<GraphicBuffer> buffer;
107};
108
109// ----------------------------------------------------------------------------
110// GraphicBuffer lifecycle
111// ----------------------------------------------------------------------------
112
113static jlong android_view_GraphiceBuffer_create(JNIEnv* env, jobject clazz,
114        jint width, jint height, jint format, jint usage) {
115
116    sp<ISurfaceComposer> composer(ComposerService::getComposerService());
117    sp<IGraphicBufferAlloc> alloc(composer->createGraphicBufferAlloc());
118    if (alloc == NULL) {
119        GB_LOGW("createGraphicBufferAlloc() failed in GraphicBuffer.create()");
120        return NULL;
121    }
122
123    status_t error;
124    sp<GraphicBuffer> buffer(alloc->createGraphicBuffer(width, height, format, usage, &error));
125    if (buffer == NULL) {
126        GB_LOGW("createGraphicBuffer() failed in GraphicBuffer.create()");
127        return NULL;
128    }
129
130    GraphicBufferWrapper* wrapper = new GraphicBufferWrapper(buffer);
131    return reinterpret_cast<jlong>(wrapper);
132}
133
134static void android_view_GraphiceBuffer_destroy(JNIEnv* env, jobject clazz,
135        jlong wrapperHandle) {
136    GraphicBufferWrapper* wrapper =
137                reinterpret_cast<GraphicBufferWrapper*>(wrapperHandle);
138    delete wrapper;
139}
140
141// ----------------------------------------------------------------------------
142// Canvas management
143// ----------------------------------------------------------------------------
144
145static inline SkBitmap::Config convertPixelFormat(int32_t format) {
146    switch (format) {
147        case PIXEL_FORMAT_RGBA_8888:
148            return SkBitmap::kARGB_8888_Config;
149        case PIXEL_FORMAT_RGBX_8888:
150            return SkBitmap::kARGB_8888_Config;
151        case PIXEL_FORMAT_RGB_565:
152            return SkBitmap::kRGB_565_Config;
153        default:
154            return SkBitmap::kNo_Config;
155    }
156}
157
158static jboolean android_view_GraphicBuffer_lockCanvas(JNIEnv* env, jobject,
159        jlong wrapperHandle, jobject canvas, jobject dirtyRect) {
160
161    GraphicBufferWrapper* wrapper =
162                reinterpret_cast<GraphicBufferWrapper*>(wrapperHandle);
163    if (!wrapper) {
164        return JNI_FALSE;
165    }
166
167    sp<GraphicBuffer> buffer(wrapper->buffer);
168
169    Rect rect;
170    if (dirtyRect) {
171        rect.left = GET_INT(dirtyRect, gRectClassInfo.left);
172        rect.top = GET_INT(dirtyRect, gRectClassInfo.top);
173        rect.right = GET_INT(dirtyRect, gRectClassInfo.right);
174        rect.bottom = GET_INT(dirtyRect, gRectClassInfo.bottom);
175    } else {
176        rect.set(Rect(buffer->getWidth(), buffer->getHeight()));
177    }
178
179    void* bits = NULL;
180    status_t status = buffer->lock(LOCK_CANVAS_USAGE, rect, &bits);
181
182    if (status) return JNI_FALSE;
183    if (!bits) {
184        buffer->unlock();
185        return JNI_FALSE;
186    }
187
188    ssize_t bytesCount = buffer->getStride() * bytesPerPixel(buffer->getPixelFormat());
189
190    SkBitmap bitmap;
191    bitmap.setConfig(convertPixelFormat(buffer->getPixelFormat()),
192            buffer->getWidth(), buffer->getHeight(), bytesCount);
193
194    if (buffer->getWidth() > 0 && buffer->getHeight() > 0) {
195        bitmap.setPixels(bits);
196    } else {
197        bitmap.setPixels(NULL);
198    }
199
200    SET_INT(canvas, gCanvasClassInfo.mSurfaceFormat, buffer->getPixelFormat());
201    INVOKEV(canvas, gCanvasClassInfo.setNativeBitmap, reinterpret_cast<jlong>(&bitmap));
202
203    SkRect clipRect;
204    clipRect.set(rect.left, rect.top, rect.right, rect.bottom);
205    SkCanvas* nativeCanvas = GraphicsJNI::getNativeCanvas(env, canvas);
206    nativeCanvas->clipRect(clipRect);
207
208    if (dirtyRect) {
209        INVOKEV(dirtyRect, gRectClassInfo.set,
210                int(rect.left), int(rect.top), int(rect.right), int(rect.bottom));
211    }
212
213    return JNI_TRUE;
214}
215
216static jboolean android_view_GraphicBuffer_unlockCanvasAndPost(JNIEnv* env, jobject,
217        jlong wrapperHandle, jobject canvas) {
218
219    GraphicBufferWrapper* wrapper =
220                reinterpret_cast<GraphicBufferWrapper*>(wrapperHandle);
221    INVOKEV(canvas, gCanvasClassInfo.setNativeBitmap, (jlong)0);
222
223    if (wrapper) {
224        status_t status = wrapper->buffer->unlock();
225        return status == 0 ? JNI_TRUE : JNI_FALSE;
226    }
227
228    return JNI_FALSE;
229}
230
231// ----------------------------------------------------------------------------
232// Serialization
233// ----------------------------------------------------------------------------
234
235static void android_view_GraphiceBuffer_write(JNIEnv* env, jobject clazz,
236        jlong wrapperHandle, jobject dest) {
237    GraphicBufferWrapper* wrapper =
238                reinterpret_cast<GraphicBufferWrapper*>(wrapperHandle);
239    Parcel* parcel = parcelForJavaObject(env, dest);
240    if (parcel) {
241        parcel->write(*wrapper->buffer);
242    }
243}
244
245static jlong android_view_GraphiceBuffer_read(JNIEnv* env, jobject clazz,
246        jobject in) {
247
248    Parcel* parcel = parcelForJavaObject(env, in);
249    if (parcel) {
250        sp<GraphicBuffer> buffer = new GraphicBuffer();
251        parcel->read(*buffer);
252        return reinterpret_cast<jlong>(new GraphicBufferWrapper(buffer));
253    }
254
255    return NULL;
256}
257
258// ----------------------------------------------------------------------------
259// External helpers
260// ----------------------------------------------------------------------------
261
262sp<GraphicBuffer> graphicBufferForJavaObject(JNIEnv* env, jobject obj) {
263    if (obj) {
264        jlong nativeObject = env->GetLongField(obj, gGraphicBufferClassInfo.mNativeObject);
265        GraphicBufferWrapper* wrapper = (GraphicBufferWrapper*) nativeObject;
266        if (wrapper != NULL) {
267            sp<GraphicBuffer> buffer(wrapper->buffer);
268            return buffer;
269        }
270    }
271    return NULL;
272}
273
274// ----------------------------------------------------------------------------
275// JNI Glue
276// ----------------------------------------------------------------------------
277
278#define FIND_CLASS(var, className) \
279        var = env->FindClass(className); \
280        LOG_FATAL_IF(! var, "Unable to find class " className);
281
282#define GET_FIELD_ID(var, clazz, fieldName, fieldDescriptor) \
283        var = env->GetFieldID(clazz, fieldName, fieldDescriptor); \
284        LOG_FATAL_IF(! var, "Unable to find field " fieldName);
285
286#define GET_METHOD_ID(var, clazz, methodName, methodDescriptor) \
287        var = env->GetMethodID(clazz, methodName, methodDescriptor); \
288        LOG_FATAL_IF(!var, "Unable to find method " methodName);
289
290const char* const kClassPathName = "android/view/GraphicBuffer";
291
292static JNINativeMethod gMethods[] = {
293    { "nCreateGraphicBuffer",  "(IIII)J", (void*) android_view_GraphiceBuffer_create },
294    { "nDestroyGraphicBuffer", "(J)V",    (void*) android_view_GraphiceBuffer_destroy },
295
296    { "nWriteGraphicBufferToParcel",  "(JLandroid/os/Parcel;)V",
297            (void*) android_view_GraphiceBuffer_write },
298    { "nReadGraphicBufferFromParcel", "(Landroid/os/Parcel;)J",
299            (void*) android_view_GraphiceBuffer_read },
300
301    { "nLockCanvas", "(JLandroid/graphics/Canvas;Landroid/graphics/Rect;)Z",
302            (void*) android_view_GraphicBuffer_lockCanvas },
303    { "nUnlockCanvasAndPost", "(JLandroid/graphics/Canvas;)Z",
304            (void*) android_view_GraphicBuffer_unlockCanvasAndPost },
305};
306
307int register_android_view_GraphicBuffer(JNIEnv* env) {
308    jclass clazz;
309    FIND_CLASS(clazz, "android/view/GraphicBuffer");
310    GET_FIELD_ID(gGraphicBufferClassInfo.mNativeObject, clazz, "mNativeObject", "J");
311
312    FIND_CLASS(clazz, "android/graphics/Rect");
313    GET_METHOD_ID(gRectClassInfo.set, clazz, "set", "(IIII)V");
314    GET_FIELD_ID(gRectClassInfo.left, clazz, "left", "I");
315    GET_FIELD_ID(gRectClassInfo.top, clazz, "top", "I");
316    GET_FIELD_ID(gRectClassInfo.right, clazz, "right", "I");
317    GET_FIELD_ID(gRectClassInfo.bottom, clazz, "bottom", "I");
318
319    FIND_CLASS(clazz, "android/graphics/Canvas");
320    GET_FIELD_ID(gCanvasClassInfo.mSurfaceFormat, clazz, "mSurfaceFormat", "I");
321    GET_METHOD_ID(gCanvasClassInfo.setNativeBitmap, clazz, "setNativeBitmap", "(J)V");
322
323    return AndroidRuntime::registerNativeMethods(env, kClassPathName, gMethods, NELEM(gMethods));
324}
325
326};
327