• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 #define LOG_TAG "GraphicsJNI"
2 
3 #include "jni.h"
4 #include "JNIHelp.h"
5 #include "GraphicsJNI.h"
6 
7 #include "SkCanvas.h"
8 #include "SkDevice.h"
9 #include "SkPicture.h"
10 #include "SkRegion.h"
11 #include <android_runtime/AndroidRuntime.h>
12 
doThrowNPE(JNIEnv * env)13 void doThrowNPE(JNIEnv* env) {
14     jniThrowNullPointerException(env, NULL);
15 }
16 
doThrowAIOOBE(JNIEnv * env)17 void doThrowAIOOBE(JNIEnv* env) {
18     jniThrowException(env, "java/lang/ArrayIndexOutOfBoundsException", NULL);
19 }
20 
doThrowRE(JNIEnv * env,const char * msg)21 void doThrowRE(JNIEnv* env, const char* msg) {
22     jniThrowRuntimeException(env, msg);
23 }
24 
doThrowIAE(JNIEnv * env,const char * msg)25 void doThrowIAE(JNIEnv* env, const char* msg) {
26     jniThrowException(env, "java/lang/IllegalArgumentException", msg);
27 }
28 
doThrowISE(JNIEnv * env,const char * msg)29 void doThrowISE(JNIEnv* env, const char* msg) {
30     jniThrowException(env, "java/lang/IllegalStateException", msg);
31 }
32 
doThrowOOME(JNIEnv * env,const char * msg)33 void doThrowOOME(JNIEnv* env, const char* msg) {
34     jniThrowException(env, "java/lang/OutOfMemoryError", msg);
35 }
36 
doThrowIOE(JNIEnv * env,const char * msg)37 void doThrowIOE(JNIEnv* env, const char* msg) {
38     jniThrowException(env, "java/io/IOException", msg);
39 }
40 
hasException(JNIEnv * env)41 bool GraphicsJNI::hasException(JNIEnv *env) {
42     if (env->ExceptionCheck() != 0) {
43         ALOGE("*** Uncaught exception returned from Java call!\n");
44         env->ExceptionDescribe();
45         return true;
46     }
47     return false;
48 }
49 
50 ///////////////////////////////////////////////////////////////////////////////
51 
AutoJavaFloatArray(JNIEnv * env,jfloatArray array,int minLength,JNIAccess access)52 AutoJavaFloatArray::AutoJavaFloatArray(JNIEnv* env, jfloatArray array,
53                                        int minLength, JNIAccess access)
54 : fEnv(env), fArray(array), fPtr(NULL), fLen(0) {
55     SkASSERT(env);
56     if (array) {
57         fLen = env->GetArrayLength(array);
58         if (fLen < minLength) {
59             sk_throw();
60         }
61         fPtr = env->GetFloatArrayElements(array, NULL);
62     }
63     fReleaseMode = (access == kRO_JNIAccess) ? JNI_ABORT : 0;
64 }
65 
~AutoJavaFloatArray()66 AutoJavaFloatArray::~AutoJavaFloatArray() {
67     if (fPtr) {
68         fEnv->ReleaseFloatArrayElements(fArray, fPtr, fReleaseMode);
69     }
70 }
71 
AutoJavaIntArray(JNIEnv * env,jintArray array,int minLength)72 AutoJavaIntArray::AutoJavaIntArray(JNIEnv* env, jintArray array,
73                                        int minLength)
74 : fEnv(env), fArray(array), fPtr(NULL), fLen(0) {
75     SkASSERT(env);
76     if (array) {
77         fLen = env->GetArrayLength(array);
78         if (fLen < minLength) {
79             sk_throw();
80         }
81         fPtr = env->GetIntArrayElements(array, NULL);
82     }
83 }
84 
~AutoJavaIntArray()85 AutoJavaIntArray::~AutoJavaIntArray() {
86     if (fPtr) {
87         fEnv->ReleaseIntArrayElements(fArray, fPtr, 0);
88     }
89 }
90 
AutoJavaShortArray(JNIEnv * env,jshortArray array,int minLength,JNIAccess access)91 AutoJavaShortArray::AutoJavaShortArray(JNIEnv* env, jshortArray array,
92                                        int minLength, JNIAccess access)
93 : fEnv(env), fArray(array), fPtr(NULL), fLen(0) {
94     SkASSERT(env);
95     if (array) {
96         fLen = env->GetArrayLength(array);
97         if (fLen < minLength) {
98             sk_throw();
99         }
100         fPtr = env->GetShortArrayElements(array, NULL);
101     }
102     fReleaseMode = (access == kRO_JNIAccess) ? JNI_ABORT : 0;
103 }
104 
~AutoJavaShortArray()105 AutoJavaShortArray::~AutoJavaShortArray() {
106     if (fPtr) {
107         fEnv->ReleaseShortArrayElements(fArray, fPtr, fReleaseMode);
108     }
109 }
110 
AutoJavaByteArray(JNIEnv * env,jbyteArray array,int minLength)111 AutoJavaByteArray::AutoJavaByteArray(JNIEnv* env, jbyteArray array,
112                                        int minLength)
113 : fEnv(env), fArray(array), fPtr(NULL), fLen(0) {
114     SkASSERT(env);
115     if (array) {
116         fLen = env->GetArrayLength(array);
117         if (fLen < minLength) {
118             sk_throw();
119         }
120         fPtr = env->GetByteArrayElements(array, NULL);
121     }
122 }
123 
~AutoJavaByteArray()124 AutoJavaByteArray::~AutoJavaByteArray() {
125     if (fPtr) {
126         fEnv->ReleaseByteArrayElements(fArray, fPtr, 0);
127     }
128 }
129 
130 ///////////////////////////////////////////////////////////////////////////////
131 
132 static jclass   gRect_class;
133 static jfieldID gRect_leftFieldID;
134 static jfieldID gRect_topFieldID;
135 static jfieldID gRect_rightFieldID;
136 static jfieldID gRect_bottomFieldID;
137 
138 static jclass   gRectF_class;
139 static jfieldID gRectF_leftFieldID;
140 static jfieldID gRectF_topFieldID;
141 static jfieldID gRectF_rightFieldID;
142 static jfieldID gRectF_bottomFieldID;
143 
144 static jclass   gPoint_class;
145 static jfieldID gPoint_xFieldID;
146 static jfieldID gPoint_yFieldID;
147 
148 static jclass   gPointF_class;
149 static jfieldID gPointF_xFieldID;
150 static jfieldID gPointF_yFieldID;
151 
152 static jclass   gBitmap_class;
153 static jfieldID gBitmap_nativeInstanceID;
154 static jmethodID gBitmap_constructorMethodID;
155 static jmethodID gBitmap_reinitMethodID;
156 static jmethodID gBitmap_getAllocationByteCountMethodID;
157 
158 static jclass   gBitmapConfig_class;
159 static jfieldID gBitmapConfig_nativeInstanceID;
160 
161 static jclass   gBitmapRegionDecoder_class;
162 static jmethodID gBitmapRegionDecoder_constructorMethodID;
163 
164 static jclass   gCanvas_class;
165 static jfieldID gCanvas_nativeInstanceID;
166 
167 static jclass   gPaint_class;
168 static jfieldID gPaint_nativeInstanceID;
169 
170 static jclass   gPicture_class;
171 static jfieldID gPicture_nativeInstanceID;
172 
173 static jclass   gRegion_class;
174 static jfieldID gRegion_nativeInstanceID;
175 static jmethodID gRegion_constructorMethodID;
176 
177 ///////////////////////////////////////////////////////////////////////////////
178 
get_jrect(JNIEnv * env,jobject obj,int * L,int * T,int * R,int * B)179 void GraphicsJNI::get_jrect(JNIEnv* env, jobject obj, int* L, int* T, int* R, int* B)
180 {
181     SkASSERT(env->IsInstanceOf(obj, gRect_class));
182 
183     *L = env->GetIntField(obj, gRect_leftFieldID);
184     *T = env->GetIntField(obj, gRect_topFieldID);
185     *R = env->GetIntField(obj, gRect_rightFieldID);
186     *B = env->GetIntField(obj, gRect_bottomFieldID);
187 }
188 
set_jrect(JNIEnv * env,jobject obj,int L,int T,int R,int B)189 void GraphicsJNI::set_jrect(JNIEnv* env, jobject obj, int L, int T, int R, int B)
190 {
191     SkASSERT(env->IsInstanceOf(obj, gRect_class));
192 
193     env->SetIntField(obj, gRect_leftFieldID, L);
194     env->SetIntField(obj, gRect_topFieldID, T);
195     env->SetIntField(obj, gRect_rightFieldID, R);
196     env->SetIntField(obj, gRect_bottomFieldID, B);
197 }
198 
jrect_to_irect(JNIEnv * env,jobject obj,SkIRect * ir)199 SkIRect* GraphicsJNI::jrect_to_irect(JNIEnv* env, jobject obj, SkIRect* ir)
200 {
201     SkASSERT(env->IsInstanceOf(obj, gRect_class));
202 
203     ir->set(env->GetIntField(obj, gRect_leftFieldID),
204             env->GetIntField(obj, gRect_topFieldID),
205             env->GetIntField(obj, gRect_rightFieldID),
206             env->GetIntField(obj, gRect_bottomFieldID));
207     return ir;
208 }
209 
irect_to_jrect(const SkIRect & ir,JNIEnv * env,jobject obj)210 void GraphicsJNI::irect_to_jrect(const SkIRect& ir, JNIEnv* env, jobject obj)
211 {
212     SkASSERT(env->IsInstanceOf(obj, gRect_class));
213 
214     env->SetIntField(obj, gRect_leftFieldID, ir.fLeft);
215     env->SetIntField(obj, gRect_topFieldID, ir.fTop);
216     env->SetIntField(obj, gRect_rightFieldID, ir.fRight);
217     env->SetIntField(obj, gRect_bottomFieldID, ir.fBottom);
218 }
219 
jrectf_to_rect(JNIEnv * env,jobject obj,SkRect * r)220 SkRect* GraphicsJNI::jrectf_to_rect(JNIEnv* env, jobject obj, SkRect* r)
221 {
222     SkASSERT(env->IsInstanceOf(obj, gRectF_class));
223 
224     r->set(SkFloatToScalar(env->GetFloatField(obj, gRectF_leftFieldID)),
225            SkFloatToScalar(env->GetFloatField(obj, gRectF_topFieldID)),
226            SkFloatToScalar(env->GetFloatField(obj, gRectF_rightFieldID)),
227            SkFloatToScalar(env->GetFloatField(obj, gRectF_bottomFieldID)));
228     return r;
229 }
230 
jrect_to_rect(JNIEnv * env,jobject obj,SkRect * r)231 SkRect* GraphicsJNI::jrect_to_rect(JNIEnv* env, jobject obj, SkRect* r)
232 {
233     SkASSERT(env->IsInstanceOf(obj, gRect_class));
234 
235     r->set(SkIntToScalar(env->GetIntField(obj, gRect_leftFieldID)),
236            SkIntToScalar(env->GetIntField(obj, gRect_topFieldID)),
237            SkIntToScalar(env->GetIntField(obj, gRect_rightFieldID)),
238            SkIntToScalar(env->GetIntField(obj, gRect_bottomFieldID)));
239     return r;
240 }
241 
rect_to_jrectf(const SkRect & r,JNIEnv * env,jobject obj)242 void GraphicsJNI::rect_to_jrectf(const SkRect& r, JNIEnv* env, jobject obj)
243 {
244     SkASSERT(env->IsInstanceOf(obj, gRectF_class));
245 
246     env->SetFloatField(obj, gRectF_leftFieldID, SkScalarToFloat(r.fLeft));
247     env->SetFloatField(obj, gRectF_topFieldID, SkScalarToFloat(r.fTop));
248     env->SetFloatField(obj, gRectF_rightFieldID, SkScalarToFloat(r.fRight));
249     env->SetFloatField(obj, gRectF_bottomFieldID, SkScalarToFloat(r.fBottom));
250 }
251 
jpoint_to_ipoint(JNIEnv * env,jobject obj,SkIPoint * point)252 SkIPoint* GraphicsJNI::jpoint_to_ipoint(JNIEnv* env, jobject obj, SkIPoint* point)
253 {
254     SkASSERT(env->IsInstanceOf(obj, gPoint_class));
255 
256     point->set(env->GetIntField(obj, gPoint_xFieldID),
257                env->GetIntField(obj, gPoint_yFieldID));
258     return point;
259 }
260 
ipoint_to_jpoint(const SkIPoint & ir,JNIEnv * env,jobject obj)261 void GraphicsJNI::ipoint_to_jpoint(const SkIPoint& ir, JNIEnv* env, jobject obj)
262 {
263     SkASSERT(env->IsInstanceOf(obj, gPoint_class));
264 
265     env->SetIntField(obj, gPoint_xFieldID, ir.fX);
266     env->SetIntField(obj, gPoint_yFieldID, ir.fY);
267 }
268 
jpointf_to_point(JNIEnv * env,jobject obj,SkPoint * point)269 SkPoint* GraphicsJNI::jpointf_to_point(JNIEnv* env, jobject obj, SkPoint* point)
270 {
271     SkASSERT(env->IsInstanceOf(obj, gPointF_class));
272 
273     point->set(SkFloatToScalar(env->GetIntField(obj, gPointF_xFieldID)),
274                SkFloatToScalar(env->GetIntField(obj, gPointF_yFieldID)));
275     return point;
276 }
277 
point_to_jpointf(const SkPoint & r,JNIEnv * env,jobject obj)278 void GraphicsJNI::point_to_jpointf(const SkPoint& r, JNIEnv* env, jobject obj)
279 {
280     SkASSERT(env->IsInstanceOf(obj, gPointF_class));
281 
282     env->SetFloatField(obj, gPointF_xFieldID, SkScalarToFloat(r.fX));
283     env->SetFloatField(obj, gPointF_yFieldID, SkScalarToFloat(r.fY));
284 }
285 
getNativeBitmap(JNIEnv * env,jobject bitmap)286 SkBitmap* GraphicsJNI::getNativeBitmap(JNIEnv* env, jobject bitmap) {
287     SkASSERT(env);
288     SkASSERT(bitmap);
289     SkASSERT(env->IsInstanceOf(bitmap, gBitmap_class));
290     SkBitmap* b = (SkBitmap*)env->GetIntField(bitmap, gBitmap_nativeInstanceID);
291     SkASSERT(b);
292     return b;
293 }
294 
getNativeBitmapConfig(JNIEnv * env,jobject jconfig)295 SkBitmap::Config GraphicsJNI::getNativeBitmapConfig(JNIEnv* env,
296                                                     jobject jconfig) {
297     SkASSERT(env);
298     if (NULL == jconfig) {
299         return SkBitmap::kNo_Config;
300     }
301     SkASSERT(env->IsInstanceOf(jconfig, gBitmapConfig_class));
302     int c = env->GetIntField(jconfig, gBitmapConfig_nativeInstanceID);
303     if (c < 0 || c >= SkBitmap::kConfigCount) {
304         c = SkBitmap::kNo_Config;
305     }
306     return static_cast<SkBitmap::Config>(c);
307 }
308 
getNativeCanvas(JNIEnv * env,jobject canvas)309 SkCanvas* GraphicsJNI::getNativeCanvas(JNIEnv* env, jobject canvas) {
310     SkASSERT(env);
311     SkASSERT(canvas);
312     SkASSERT(env->IsInstanceOf(canvas, gCanvas_class));
313     SkCanvas* c = (SkCanvas*)env->GetIntField(canvas, gCanvas_nativeInstanceID);
314     SkASSERT(c);
315     return c;
316 }
317 
getNativePaint(JNIEnv * env,jobject paint)318 SkPaint* GraphicsJNI::getNativePaint(JNIEnv* env, jobject paint) {
319     SkASSERT(env);
320     SkASSERT(paint);
321     SkASSERT(env->IsInstanceOf(paint, gPaint_class));
322     SkPaint* p = (SkPaint*)env->GetIntField(paint, gPaint_nativeInstanceID);
323     SkASSERT(p);
324     return p;
325 }
326 
getNativePicture(JNIEnv * env,jobject picture)327 SkPicture* GraphicsJNI::getNativePicture(JNIEnv* env, jobject picture)
328 {
329     SkASSERT(env);
330     SkASSERT(picture);
331     SkASSERT(env->IsInstanceOf(picture, gPicture_class));
332     SkPicture* p = (SkPicture*)env->GetIntField(picture, gPicture_nativeInstanceID);
333     SkASSERT(p);
334     return p;
335 }
336 
getNativeRegion(JNIEnv * env,jobject region)337 SkRegion* GraphicsJNI::getNativeRegion(JNIEnv* env, jobject region)
338 {
339     SkASSERT(env);
340     SkASSERT(region);
341     SkASSERT(env->IsInstanceOf(region, gRegion_class));
342     SkRegion* r = (SkRegion*)env->GetIntField(region, gRegion_nativeInstanceID);
343     SkASSERT(r);
344     return r;
345 }
346 
347 ///////////////////////////////////////////////////////////////////////////////////////////
348 
349 // Assert that bitmap's SkAlphaType is consistent with isPremultiplied.
assert_premultiplied(const SkBitmap & bitmap,bool isPremultiplied)350 static void assert_premultiplied(const SkBitmap& bitmap, bool isPremultiplied) {
351     // kOpaque_SkAlphaType and kIgnore_SkAlphaType mean that isPremultiplied is
352     // irrelevant. This just tests to ensure that the SkAlphaType is not
353     // opposite of isPremultiplied.
354     if (isPremultiplied) {
355         SkASSERT(bitmap.alphaType() != kUnpremul_SkAlphaType);
356     } else {
357         SkASSERT(bitmap.alphaType() != kPremul_SkAlphaType);
358     }
359 }
360 
createBitmap(JNIEnv * env,SkBitmap * bitmap,jbyteArray buffer,int bitmapCreateFlags,jbyteArray ninepatch,jintArray layoutbounds,int density)361 jobject GraphicsJNI::createBitmap(JNIEnv* env, SkBitmap* bitmap, jbyteArray buffer,
362         int bitmapCreateFlags, jbyteArray ninepatch, jintArray layoutbounds, int density)
363 {
364     SkASSERT(bitmap);
365     SkASSERT(bitmap->pixelRef());
366     bool isMutable = bitmapCreateFlags & kBitmapCreateFlag_Mutable;
367     bool isPremultiplied = bitmapCreateFlags & kBitmapCreateFlag_Premultiplied;
368 
369     // The caller needs to have already set the alpha type properly, so the
370     // native SkBitmap stays in sync with the Java Bitmap.
371     assert_premultiplied(*bitmap, isPremultiplied);
372 
373     jobject obj = env->NewObject(gBitmap_class, gBitmap_constructorMethodID,
374             static_cast<jint>(reinterpret_cast<uintptr_t>(bitmap)), buffer,
375             bitmap->width(), bitmap->height(), density, isMutable, isPremultiplied,
376             ninepatch, layoutbounds);
377     hasException(env); // For the side effect of logging.
378     return obj;
379 }
380 
createBitmap(JNIEnv * env,SkBitmap * bitmap,int bitmapCreateFlags,jbyteArray ninepatch,int density)381 jobject GraphicsJNI::createBitmap(JNIEnv* env, SkBitmap* bitmap, int bitmapCreateFlags,
382         jbyteArray ninepatch, int density)
383 {
384     return createBitmap(env, bitmap, NULL, bitmapCreateFlags, ninepatch, NULL, density);
385 }
386 
reinitBitmap(JNIEnv * env,jobject javaBitmap,SkBitmap * bitmap,bool isPremultiplied)387 void GraphicsJNI::reinitBitmap(JNIEnv* env, jobject javaBitmap, SkBitmap* bitmap,
388         bool isPremultiplied)
389 {
390     // The caller needs to have already set the alpha type properly, so the
391     // native SkBitmap stays in sync with the Java Bitmap.
392     assert_premultiplied(*bitmap, isPremultiplied);
393 
394     env->CallVoidMethod(javaBitmap, gBitmap_reinitMethodID,
395             bitmap->width(), bitmap->height(), isPremultiplied);
396 }
397 
getBitmapAllocationByteCount(JNIEnv * env,jobject javaBitmap)398 int GraphicsJNI::getBitmapAllocationByteCount(JNIEnv* env, jobject javaBitmap)
399 {
400     return env->CallIntMethod(javaBitmap, gBitmap_getAllocationByteCountMethodID);
401 }
402 
createBitmapRegionDecoder(JNIEnv * env,SkBitmapRegionDecoder * bitmap)403 jobject GraphicsJNI::createBitmapRegionDecoder(JNIEnv* env, SkBitmapRegionDecoder* bitmap)
404 {
405     SkASSERT(bitmap != NULL);
406 
407     jobject obj = env->NewObject(gBitmapRegionDecoder_class,
408             gBitmapRegionDecoder_constructorMethodID,
409             static_cast<jint>(reinterpret_cast<uintptr_t>(bitmap)));
410     hasException(env); // For the side effect of logging.
411     return obj;
412 }
413 
createRegion(JNIEnv * env,SkRegion * region)414 jobject GraphicsJNI::createRegion(JNIEnv* env, SkRegion* region)
415 {
416     SkASSERT(region != NULL);
417     jobject obj = env->NewObject(gRegion_class, gRegion_constructorMethodID,
418             static_cast<jint>(reinterpret_cast<uintptr_t>(region)), 0);
419     hasException(env); // For the side effect of logging.
420     return obj;
421 }
422 
vm2env(JavaVM * vm)423 static JNIEnv* vm2env(JavaVM* vm)
424 {
425     JNIEnv* env = NULL;
426     if (vm->GetEnv((void**)&env, JNI_VERSION_1_4) != JNI_OK || NULL == env)
427     {
428         SkDebugf("------- [%p] vm->GetEnv() failed\n", vm);
429         sk_throw();
430     }
431     return env;
432 }
433 
434 ///////////////////////////////////////////////////////////////////////////////
435 
AndroidPixelRef(JNIEnv * env,const SkImageInfo & info,void * storage,size_t rowBytes,jbyteArray storageObj,SkColorTable * ctable)436 AndroidPixelRef::AndroidPixelRef(JNIEnv* env, const SkImageInfo& info, void* storage,
437         size_t rowBytes, jbyteArray storageObj, SkColorTable* ctable) :
438         SkMallocPixelRef(info, storage, rowBytes, ctable, (storageObj == NULL)),
439         fWrappedPixelRef(NULL) {
440     SkASSERT(storage);
441     SkASSERT(env);
442 
443     if (env->GetJavaVM(&fVM) != JNI_OK) {
444         SkDebugf("------ [%p] env->GetJavaVM failed\n", env);
445         sk_throw();
446     }
447     fStorageObj = storageObj;
448     fHasGlobalRef = false;
449     fGlobalRefCnt = 0;
450 
451     // If storageObj is NULL, the memory was NOT allocated on the Java heap
452     fOnJavaHeap = (storageObj != NULL);
453 }
454 
AndroidPixelRef(AndroidPixelRef & wrappedPixelRef,const SkImageInfo & info,size_t rowBytes,SkColorTable * ctable)455 AndroidPixelRef::AndroidPixelRef(AndroidPixelRef& wrappedPixelRef, const SkImageInfo& info,
456         size_t rowBytes, SkColorTable* ctable) :
457         SkMallocPixelRef(info, wrappedPixelRef.getAddr(), rowBytes, ctable, false),
458         fWrappedPixelRef(wrappedPixelRef.fWrappedPixelRef ?
459                 wrappedPixelRef.fWrappedPixelRef : &wrappedPixelRef)
460 {
461     SkASSERT(fWrappedPixelRef);
462     SkSafeRef(fWrappedPixelRef);
463 
464     // don't need to initialize these, as all the relevant logic delegates to the wrapped ref
465     fStorageObj = NULL;
466     fHasGlobalRef = false;
467     fGlobalRefCnt = 0;
468     fOnJavaHeap = false;
469 }
470 
~AndroidPixelRef()471 AndroidPixelRef::~AndroidPixelRef() {
472     if (fWrappedPixelRef) {
473         SkSafeUnref(fWrappedPixelRef);
474     } else if (fOnJavaHeap) {
475         JNIEnv* env = vm2env(fVM);
476 
477         if (fStorageObj && fHasGlobalRef) {
478             env->DeleteGlobalRef(fStorageObj);
479         }
480         fStorageObj = NULL;
481     }
482 }
getStorageObj()483 jbyteArray AndroidPixelRef::getStorageObj() {
484     if (fWrappedPixelRef) {
485         return fWrappedPixelRef->fStorageObj;
486     }
487     return fStorageObj;
488 }
489 
setLocalJNIRef(jbyteArray arr)490 void AndroidPixelRef::setLocalJNIRef(jbyteArray arr) {
491     if (fWrappedPixelRef) {
492         // delegate java obj management to the wrapped ref
493         fWrappedPixelRef->setLocalJNIRef(arr);
494     } else if (!fHasGlobalRef) {
495         fStorageObj = arr;
496     }
497 }
498 
globalRef(void * localref)499 void AndroidPixelRef::globalRef(void* localref) {
500     if (fWrappedPixelRef) {
501         // delegate java obj management to the wrapped ref
502         fWrappedPixelRef->globalRef(localref);
503 
504         // Note: we only ref and unref the wrapped AndroidPixelRef so that
505         // bitmap->pixelRef()->globalRef() and globalUnref() can be used in a pair, even if
506         // the bitmap has its underlying AndroidPixelRef swapped out/wrapped
507         return;
508     }
509     if (fOnJavaHeap && sk_atomic_inc(&fGlobalRefCnt) == 0) {
510         JNIEnv *env = vm2env(fVM);
511 
512         // If JNI ref was passed, it is always used
513         if (localref) fStorageObj = (jbyteArray) localref;
514 
515         if (fStorageObj == NULL) {
516             SkDebugf("No valid local ref to create a JNI global ref\n");
517             sk_throw();
518         }
519         if (fHasGlobalRef) {
520             // This should never happen
521             SkDebugf("Already holding a JNI global ref");
522             sk_throw();
523         }
524 
525         fStorageObj = (jbyteArray) env->NewGlobalRef(fStorageObj);
526         // TODO: Check for failure here
527         fHasGlobalRef = true;
528     }
529     ref();
530 }
531 
globalUnref()532 void AndroidPixelRef::globalUnref() {
533     if (fWrappedPixelRef) {
534         // delegate java obj management to the wrapped ref
535         fWrappedPixelRef->globalUnref();
536         return;
537     }
538     if (fOnJavaHeap && sk_atomic_dec(&fGlobalRefCnt) == 1) {
539         JNIEnv *env = vm2env(fVM);
540         if (!fHasGlobalRef) {
541             SkDebugf("We don't have a global ref!");
542             sk_throw();
543         }
544         env->DeleteGlobalRef(fStorageObj);
545         fStorageObj = NULL;
546         fHasGlobalRef = false;
547     }
548     unref();
549 }
550 
551 ///////////////////////////////////////////////////////////////////////////////
552 
553 extern "C" jbyte* jniGetNonMovableArrayElements(C_JNIEnv* env, jarray arrayObj);
554 
allocateJavaPixelRef(JNIEnv * env,SkBitmap * bitmap,SkColorTable * ctable)555 jbyteArray GraphicsJNI::allocateJavaPixelRef(JNIEnv* env, SkBitmap* bitmap,
556                                              SkColorTable* ctable) {
557     Sk64 size64 = bitmap->getSize64();
558     if (size64.isNeg() || !size64.is32()) {
559         jniThrowException(env, "java/lang/IllegalArgumentException",
560                           "bitmap size exceeds 32bits");
561         return NULL;
562     }
563 
564     SkImageInfo bitmapInfo;
565     if (!bitmap->asImageInfo(&bitmapInfo)) {
566         jniThrowException(env, "java/lang/IllegalArgumentException",
567                 "unknown bitmap configuration");
568         return NULL;
569     }
570 
571     size_t size = size64.get32();
572     jbyteArray arrayObj = env->NewByteArray(size);
573     if (arrayObj) {
574         // TODO: make this work without jniGetNonMovableArrayElements
575         jbyte* addr = jniGetNonMovableArrayElements(&env->functions, arrayObj);
576         if (addr) {
577             SkPixelRef* pr = new AndroidPixelRef(env, bitmapInfo, (void*) addr,
578                     bitmap->rowBytes(), arrayObj, ctable);
579             bitmap->setPixelRef(pr)->unref();
580             // since we're already allocated, we lockPixels right away
581             // HeapAllocator behaves this way too
582             bitmap->lockPixels();
583         }
584     }
585 
586     return arrayObj;
587 }
588 
589 ///////////////////////////////////////////////////////////////////////////////
590 
JavaPixelAllocator(JNIEnv * env)591 JavaPixelAllocator::JavaPixelAllocator(JNIEnv* env)
592     : fStorageObj(NULL),
593       fAllocCount(0) {
594     if (env->GetJavaVM(&fVM) != JNI_OK) {
595         SkDebugf("------ [%p] env->GetJavaVM failed\n", env);
596         sk_throw();
597     }
598 }
599 
allocPixelRef(SkBitmap * bitmap,SkColorTable * ctable)600 bool JavaPixelAllocator::allocPixelRef(SkBitmap* bitmap, SkColorTable* ctable) {
601     JNIEnv* env = vm2env(fVM);
602 
603     fStorageObj = GraphicsJNI::allocateJavaPixelRef(env, bitmap, ctable);
604     fAllocCount += 1;
605     return fStorageObj != NULL;
606 }
607 
608 ////////////////////////////////////////////////////////////////////////////////
609 
JavaHeapBitmapRef(JNIEnv * env,SkBitmap * nativeBitmap,jbyteArray buffer)610 JavaHeapBitmapRef::JavaHeapBitmapRef(JNIEnv* env, SkBitmap* nativeBitmap, jbyteArray buffer) {
611     fEnv = env;
612     fNativeBitmap = nativeBitmap;
613     fBuffer = buffer;
614 
615     // If the buffer is NULL, the backing memory wasn't allocated on the Java heap
616     if (fBuffer) {
617         ((AndroidPixelRef*) fNativeBitmap->pixelRef())->setLocalJNIRef(fBuffer);
618     }
619 }
620 
~JavaHeapBitmapRef()621 JavaHeapBitmapRef::~JavaHeapBitmapRef() {
622     if (fBuffer) {
623         ((AndroidPixelRef*) fNativeBitmap->pixelRef())->setLocalJNIRef(NULL);
624     }
625 }
626 
627 ////////////////////////////////////////////////////////////////////////////////
628 
make_globalref(JNIEnv * env,const char classname[])629 static jclass make_globalref(JNIEnv* env, const char classname[])
630 {
631     jclass c = env->FindClass(classname);
632     SkASSERT(c);
633     return (jclass)env->NewGlobalRef(c);
634 }
635 
getFieldIDCheck(JNIEnv * env,jclass clazz,const char fieldname[],const char type[])636 static jfieldID getFieldIDCheck(JNIEnv* env, jclass clazz,
637                                 const char fieldname[], const char type[])
638 {
639     jfieldID id = env->GetFieldID(clazz, fieldname, type);
640     SkASSERT(id);
641     return id;
642 }
643 
register_android_graphics_Graphics(JNIEnv * env)644 int register_android_graphics_Graphics(JNIEnv* env)
645 {
646     jmethodID m;
647     jclass c;
648 
649     gRect_class = make_globalref(env, "android/graphics/Rect");
650     gRect_leftFieldID = getFieldIDCheck(env, gRect_class, "left", "I");
651     gRect_topFieldID = getFieldIDCheck(env, gRect_class, "top", "I");
652     gRect_rightFieldID = getFieldIDCheck(env, gRect_class, "right", "I");
653     gRect_bottomFieldID = getFieldIDCheck(env, gRect_class, "bottom", "I");
654 
655     gRectF_class = make_globalref(env, "android/graphics/RectF");
656     gRectF_leftFieldID = getFieldIDCheck(env, gRectF_class, "left", "F");
657     gRectF_topFieldID = getFieldIDCheck(env, gRectF_class, "top", "F");
658     gRectF_rightFieldID = getFieldIDCheck(env, gRectF_class, "right", "F");
659     gRectF_bottomFieldID = getFieldIDCheck(env, gRectF_class, "bottom", "F");
660 
661     gPoint_class = make_globalref(env, "android/graphics/Point");
662     gPoint_xFieldID = getFieldIDCheck(env, gPoint_class, "x", "I");
663     gPoint_yFieldID = getFieldIDCheck(env, gPoint_class, "y", "I");
664 
665     gPointF_class = make_globalref(env, "android/graphics/PointF");
666     gPointF_xFieldID = getFieldIDCheck(env, gPointF_class, "x", "F");
667     gPointF_yFieldID = getFieldIDCheck(env, gPointF_class, "y", "F");
668 
669     gBitmap_class = make_globalref(env, "android/graphics/Bitmap");
670     gBitmap_nativeInstanceID = getFieldIDCheck(env, gBitmap_class, "mNativeBitmap", "I");
671     gBitmap_constructorMethodID = env->GetMethodID(gBitmap_class, "<init>", "(I[BIIIZZ[B[I)V");
672     gBitmap_reinitMethodID = env->GetMethodID(gBitmap_class, "reinit", "(IIZ)V");
673     gBitmap_getAllocationByteCountMethodID = env->GetMethodID(gBitmap_class, "getAllocationByteCount", "()I");
674     gBitmapRegionDecoder_class = make_globalref(env, "android/graphics/BitmapRegionDecoder");
675     gBitmapRegionDecoder_constructorMethodID = env->GetMethodID(gBitmapRegionDecoder_class, "<init>", "(I)V");
676 
677     gBitmapConfig_class = make_globalref(env, "android/graphics/Bitmap$Config");
678     gBitmapConfig_nativeInstanceID = getFieldIDCheck(env, gBitmapConfig_class,
679                                                      "nativeInt", "I");
680 
681     gCanvas_class = make_globalref(env, "android/graphics/Canvas");
682     gCanvas_nativeInstanceID = getFieldIDCheck(env, gCanvas_class, "mNativeCanvas", "I");
683 
684     gPaint_class = make_globalref(env, "android/graphics/Paint");
685     gPaint_nativeInstanceID = getFieldIDCheck(env, gPaint_class, "mNativePaint", "I");
686 
687     gPicture_class = make_globalref(env, "android/graphics/Picture");
688     gPicture_nativeInstanceID = getFieldIDCheck(env, gPicture_class, "mNativePicture", "I");
689 
690     gRegion_class = make_globalref(env, "android/graphics/Region");
691     gRegion_nativeInstanceID = getFieldIDCheck(env, gRegion_class, "mNativeRegion", "I");
692     gRegion_constructorMethodID = env->GetMethodID(gRegion_class, "<init>",
693         "(II)V");
694 
695     return 0;
696 }
697