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