• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 #define LOG_TAG "Bitmap"
2 #include "Bitmap.h"
3 
4 #include "SkBitmap.h"
5 #include "SkPixelRef.h"
6 #include "SkImageEncoder.h"
7 #include "SkImageInfo.h"
8 #include "SkColorPriv.h"
9 #include "GraphicsJNI.h"
10 #include "SkDither.h"
11 #include "SkUnPreMultiply.h"
12 #include "SkStream.h"
13 
14 #include <binder/Parcel.h>
15 #include "android_os_Parcel.h"
16 #include "android_util_Binder.h"
17 #include "android_nio_utils.h"
18 #include "CreateJavaOutputStreamAdaptor.h"
19 #include <Caches.h>
20 #include <hwui/Paint.h>
21 #include <renderthread/RenderProxy.h>
22 
23 #include "core_jni_helpers.h"
24 
25 #include <jni.h>
26 #include <memory>
27 #include <string>
28 #include <sys/mman.h>
29 #include <cutils/ashmem.h>
30 
31 #define DEBUG_PARCEL 0
32 #define ASHMEM_BITMAP_MIN_SIZE (128 * (1 << 10))
33 
34 namespace android {
35 
36 class WrappedPixelRef : public SkPixelRef {
37 public:
WrappedPixelRef(Bitmap * wrapper,void * storage,const SkImageInfo & info,size_t rowBytes,SkColorTable * ctable)38     WrappedPixelRef(Bitmap* wrapper, void* storage,
39             const SkImageInfo& info, size_t rowBytes, SkColorTable* ctable)
40             : SkPixelRef(info)
41             , mBitmap(*wrapper)
42             , mStorage(storage) {
43         reconfigure(info, rowBytes, ctable);
44     }
45 
~WrappedPixelRef()46     ~WrappedPixelRef() {
47         // Tell SkRefCnt that everything is as it expects by forcing
48         // the refcnt to 1
49         internal_dispose_restore_refcnt_to_1();
50         SkSafeUnref(mColorTable);
51     }
52 
reconfigure(const SkImageInfo & newInfo,size_t rowBytes,SkColorTable * ctable)53     void reconfigure(const SkImageInfo& newInfo, size_t rowBytes, SkColorTable* ctable) {
54         if (kIndex_8_SkColorType != newInfo.colorType()) {
55             ctable = nullptr;
56         }
57         mRowBytes = rowBytes;
58         if (mColorTable != ctable) {
59             SkSafeUnref(mColorTable);
60             mColorTable = ctable;
61             SkSafeRef(mColorTable);
62         }
63 
64         // Need to validate the alpha type to filter against the color type
65         // to prevent things like a non-opaque RGB565 bitmap
66         SkAlphaType alphaType;
67         LOG_ALWAYS_FATAL_IF(!SkColorTypeValidateAlphaType(
68                 newInfo.colorType(), newInfo.alphaType(), &alphaType),
69                 "Failed to validate alpha type!");
70 
71         // Dirty hack is dirty
72         // TODO: Figure something out here, Skia's current design makes this
73         // really hard to work with. Skia really, really wants immutable objects,
74         // but with the nested-ref-count hackery going on that's just not
75         // feasible without going insane trying to figure it out
76         SkImageInfo* myInfo = const_cast<SkImageInfo*>(&this->info());
77         *myInfo = newInfo;
78         changeAlphaType(alphaType);
79 
80         // Docs say to only call this in the ctor, but we're going to call
81         // it anyway even if this isn't always the ctor.
82         // TODO: Fix this too as part of the above TODO
83         setPreLocked(mStorage, mRowBytes, mColorTable);
84     }
85 
86     // Can't mark as override since SkPixelRef::rowBytes isn't virtual
87     // but that's OK since we just want BitmapWrapper to be able to rely
88     // on calling rowBytes() on an unlocked pixelref, which it will be
89     // doing on a WrappedPixelRef type, not a SkPixelRef, so static
90     // dispatching will do what we want.
rowBytes() const91     size_t rowBytes() const { return mRowBytes; }
colorTable() const92     SkColorTable* colorTable() const { return mColorTable; }
93 
hasHardwareMipMap() const94     bool hasHardwareMipMap() const {
95         return mHasHardwareMipMap;
96     }
97 
setHasHardwareMipMap(bool hasMipMap)98     void setHasHardwareMipMap(bool hasMipMap) {
99         mHasHardwareMipMap = hasMipMap;
100     }
101 
102 protected:
onNewLockPixels(LockRec * rec)103     virtual bool onNewLockPixels(LockRec* rec) override {
104         rec->fPixels = mStorage;
105         rec->fRowBytes = mRowBytes;
106         rec->fColorTable = mColorTable;
107         return true;
108     }
109 
onUnlockPixels()110     virtual void onUnlockPixels() override {
111         // nothing
112     }
113 
getAllocatedSizeInBytes() const114     virtual size_t getAllocatedSizeInBytes() const override {
115         return info().getSafeSize(mRowBytes);
116     }
117 
118 private:
119     Bitmap& mBitmap;
120     void* mStorage;
121     size_t mRowBytes = 0;
122     SkColorTable* mColorTable = nullptr;
123     bool mHasHardwareMipMap = false;
124 
internal_dispose() const125     virtual void internal_dispose() const override {
126         mBitmap.onStrongRefDestroyed();
127     }
128 };
129 
Bitmap(JNIEnv * env,jbyteArray storageObj,void * address,const SkImageInfo & info,size_t rowBytes,SkColorTable * ctable)130 Bitmap::Bitmap(JNIEnv* env, jbyteArray storageObj, void* address,
131             const SkImageInfo& info, size_t rowBytes, SkColorTable* ctable)
132         : mPixelStorageType(PixelStorageType::Java) {
133     env->GetJavaVM(&mPixelStorage.java.jvm);
134     mPixelStorage.java.jweakRef = env->NewWeakGlobalRef(storageObj);
135     mPixelStorage.java.jstrongRef = nullptr;
136     mPixelRef.reset(new WrappedPixelRef(this, address, info, rowBytes, ctable));
137     // Note: this will trigger a call to onStrongRefDestroyed(), but
138     // we want the pixel ref to have a ref count of 0 at this point
139     mPixelRef->unref();
140 }
141 
Bitmap(void * address,void * context,FreeFunc freeFunc,const SkImageInfo & info,size_t rowBytes,SkColorTable * ctable)142 Bitmap::Bitmap(void* address, void* context, FreeFunc freeFunc,
143             const SkImageInfo& info, size_t rowBytes, SkColorTable* ctable)
144         : mPixelStorageType(PixelStorageType::External) {
145     mPixelStorage.external.address = address;
146     mPixelStorage.external.context = context;
147     mPixelStorage.external.freeFunc = freeFunc;
148     mPixelRef.reset(new WrappedPixelRef(this, address, info, rowBytes, ctable));
149     // Note: this will trigger a call to onStrongRefDestroyed(), but
150     // we want the pixel ref to have a ref count of 0 at this point
151     mPixelRef->unref();
152 }
153 
Bitmap(void * address,int fd,size_t mappedSize,const SkImageInfo & info,size_t rowBytes,SkColorTable * ctable)154 Bitmap::Bitmap(void* address, int fd, size_t mappedSize,
155             const SkImageInfo& info, size_t rowBytes, SkColorTable* ctable)
156         : mPixelStorageType(PixelStorageType::Ashmem) {
157     mPixelStorage.ashmem.address = address;
158     mPixelStorage.ashmem.fd = fd;
159     mPixelStorage.ashmem.size = mappedSize;
160     mPixelRef.reset(new WrappedPixelRef(this, address, info, rowBytes, ctable));
161     // Note: this will trigger a call to onStrongRefDestroyed(), but
162     // we want the pixel ref to have a ref count of 0 at this point
163     mPixelRef->unref();
164 }
~Bitmap()165 Bitmap::~Bitmap() {
166     doFreePixels();
167 }
168 
freePixels()169 void Bitmap::freePixels() {
170     AutoMutex _lock(mLock);
171     if (mPinnedRefCount == 0) {
172         doFreePixels();
173         mPixelStorageType = PixelStorageType::Invalid;
174     }
175 }
176 
doFreePixels()177 void Bitmap::doFreePixels() {
178     switch (mPixelStorageType) {
179     case PixelStorageType::Invalid:
180         // already free'd, nothing to do
181         break;
182     case PixelStorageType::External:
183         mPixelStorage.external.freeFunc(mPixelStorage.external.address,
184                 mPixelStorage.external.context);
185         break;
186     case PixelStorageType::Ashmem:
187         munmap(mPixelStorage.ashmem.address, mPixelStorage.ashmem.size);
188         close(mPixelStorage.ashmem.fd);
189         break;
190     case PixelStorageType::Java:
191         JNIEnv* env = jniEnv();
192         LOG_ALWAYS_FATAL_IF(mPixelStorage.java.jstrongRef,
193                 "Deleting a bitmap wrapper while there are outstanding strong "
194                 "references! mPinnedRefCount = %d", mPinnedRefCount);
195         env->DeleteWeakGlobalRef(mPixelStorage.java.jweakRef);
196         break;
197     }
198 
199     if (android::uirenderer::Caches::hasInstance()) {
200         android::uirenderer::Caches::getInstance().textureCache.releaseTexture(
201                 mPixelRef->getStableID());
202     }
203 }
204 
hasHardwareMipMap()205 bool Bitmap::hasHardwareMipMap() {
206     return mPixelRef->hasHardwareMipMap();
207 }
208 
setHasHardwareMipMap(bool hasMipMap)209 void Bitmap::setHasHardwareMipMap(bool hasMipMap) {
210     mPixelRef->setHasHardwareMipMap(hasMipMap);
211 }
212 
getAshmemFd() const213 int Bitmap::getAshmemFd() const {
214     switch (mPixelStorageType) {
215     case PixelStorageType::Ashmem:
216         return mPixelStorage.ashmem.fd;
217     default:
218         return -1;
219     }
220 }
221 
info() const222 const SkImageInfo& Bitmap::info() const {
223     return mPixelRef->info();
224 }
225 
rowBytes() const226 size_t Bitmap::rowBytes() const {
227     return mPixelRef->rowBytes();
228 }
229 
peekAtPixelRef() const230 SkPixelRef* Bitmap::peekAtPixelRef() const {
231     assertValid();
232     return mPixelRef.get();
233 }
234 
refPixelRef()235 SkPixelRef* Bitmap::refPixelRef() {
236     assertValid();
237     android::AutoMutex _lock(mLock);
238     return refPixelRefLocked();
239 }
240 
refPixelRefLocked()241 SkPixelRef* Bitmap::refPixelRefLocked() {
242     mPixelRef->ref();
243     if (mPixelRef->unique()) {
244         // We just restored this from 0, pin the pixels and inc the strong count
245         // Note that there *might be* an incoming onStrongRefDestroyed from whatever
246         // last unref'd
247         pinPixelsLocked();
248         mPinnedRefCount++;
249     }
250     return mPixelRef.get();
251 }
252 
reconfigure(const SkImageInfo & info,size_t rowBytes,SkColorTable * ctable)253 void Bitmap::reconfigure(const SkImageInfo& info, size_t rowBytes,
254         SkColorTable* ctable) {
255     mPixelRef->reconfigure(info, rowBytes, ctable);
256 }
257 
reconfigure(const SkImageInfo & info)258 void Bitmap::reconfigure(const SkImageInfo& info) {
259     reconfigure(info, info.minRowBytes(), nullptr);
260 }
261 
setAlphaType(SkAlphaType alphaType)262 void Bitmap::setAlphaType(SkAlphaType alphaType) {
263     if (!SkColorTypeValidateAlphaType(info().colorType(), alphaType, &alphaType)) {
264         return;
265     }
266 
267     mPixelRef->changeAlphaType(alphaType);
268 }
269 
detachFromJava()270 void Bitmap::detachFromJava() {
271     bool disposeSelf;
272     {
273         android::AutoMutex _lock(mLock);
274         mAttachedToJava = false;
275         disposeSelf = shouldDisposeSelfLocked();
276     }
277     if (disposeSelf) {
278         delete this;
279     }
280 }
281 
shouldDisposeSelfLocked()282 bool Bitmap::shouldDisposeSelfLocked() {
283     return mPinnedRefCount == 0 && !mAttachedToJava;
284 }
285 
jniEnv()286 JNIEnv* Bitmap::jniEnv() {
287     JNIEnv* env;
288     auto success = mPixelStorage.java.jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
289     LOG_ALWAYS_FATAL_IF(success != JNI_OK,
290         "Failed to get JNIEnv* from JVM: %p", mPixelStorage.java.jvm);
291     return env;
292 }
293 
onStrongRefDestroyed()294 void Bitmap::onStrongRefDestroyed() {
295     bool disposeSelf = false;
296     {
297         android::AutoMutex _lock(mLock);
298         if (mPinnedRefCount > 0) {
299             mPinnedRefCount--;
300             if (mPinnedRefCount == 0) {
301                 unpinPixelsLocked();
302                 disposeSelf = shouldDisposeSelfLocked();
303             }
304         }
305     }
306     if (disposeSelf) {
307         delete this;
308     }
309 }
310 
pinPixelsLocked()311 void Bitmap::pinPixelsLocked() {
312     switch (mPixelStorageType) {
313     case PixelStorageType::Invalid:
314         LOG_ALWAYS_FATAL("Cannot pin invalid pixels!");
315         break;
316     case PixelStorageType::External:
317     case PixelStorageType::Ashmem:
318         // Nothing to do
319         break;
320     case PixelStorageType::Java: {
321         JNIEnv* env = jniEnv();
322         if (!mPixelStorage.java.jstrongRef) {
323             mPixelStorage.java.jstrongRef = reinterpret_cast<jbyteArray>(
324                     env->NewGlobalRef(mPixelStorage.java.jweakRef));
325             if (!mPixelStorage.java.jstrongRef) {
326                 LOG_ALWAYS_FATAL("Failed to acquire strong reference to pixels");
327             }
328         }
329         break;
330     }
331     }
332 }
333 
unpinPixelsLocked()334 void Bitmap::unpinPixelsLocked() {
335     switch (mPixelStorageType) {
336     case PixelStorageType::Invalid:
337         LOG_ALWAYS_FATAL("Cannot unpin invalid pixels!");
338         break;
339     case PixelStorageType::External:
340     case PixelStorageType::Ashmem:
341         // Don't need to do anything
342         break;
343     case PixelStorageType::Java: {
344         JNIEnv* env = jniEnv();
345         if (mPixelStorage.java.jstrongRef) {
346             env->DeleteGlobalRef(mPixelStorage.java.jstrongRef);
347             mPixelStorage.java.jstrongRef = nullptr;
348         }
349         break;
350     }
351     }
352 }
353 
getSkBitmap(SkBitmap * outBitmap)354 void Bitmap::getSkBitmap(SkBitmap* outBitmap) {
355     assertValid();
356     android::AutoMutex _lock(mLock);
357     // Safe because mPixelRef is a WrappedPixelRef type, otherwise rowBytes()
358     // would require locking the pixels first.
359     outBitmap->setInfo(mPixelRef->info(), mPixelRef->rowBytes());
360     outBitmap->setPixelRef(refPixelRefLocked())->unref();
361     outBitmap->setHasHardwareMipMap(hasHardwareMipMap());
362 }
363 
assertValid() const364 void Bitmap::assertValid() const {
365     LOG_ALWAYS_FATAL_IF(mPixelStorageType == PixelStorageType::Invalid,
366             "Error, cannot access an invalid/free'd bitmap here!");
367 }
368 
369 } // namespace android
370 
371 using namespace android;
372 
373 // Convenience class that does not take a global ref on the pixels, relying
374 // on the caller already having a local JNI ref
375 class LocalScopedBitmap {
376 public:
LocalScopedBitmap(jlong bitmapHandle)377     LocalScopedBitmap(jlong bitmapHandle)
378             : mBitmap(reinterpret_cast<Bitmap*>(bitmapHandle)) {}
379 
operator ->()380     Bitmap* operator->() {
381         return mBitmap;
382     }
383 
pixels()384     void* pixels() {
385         return mBitmap->peekAtPixelRef()->pixels();
386     }
387 
valid()388     bool valid() {
389         return mBitmap && mBitmap->valid();
390     }
391 
392 private:
393     Bitmap* mBitmap;
394 };
395 
396 ///////////////////////////////////////////////////////////////////////////////
397 // Conversions to/from SkColor, for get/setPixels, and the create method, which
398 // is basically like setPixels
399 
400 typedef void (*FromColorProc)(void* dst, const SkColor src[], int width,
401                               int x, int y);
402 
FromColor_D32(void * dst,const SkColor src[],int width,int,int)403 static void FromColor_D32(void* dst, const SkColor src[], int width,
404                           int, int) {
405     SkPMColor* d = (SkPMColor*)dst;
406 
407     for (int i = 0; i < width; i++) {
408         *d++ = SkPreMultiplyColor(*src++);
409     }
410 }
411 
FromColor_D32_Raw(void * dst,const SkColor src[],int width,int,int)412 static void FromColor_D32_Raw(void* dst, const SkColor src[], int width,
413                           int, int) {
414     // Needed to thwart the unreachable code detection from clang.
415     static const bool sk_color_ne_zero = SK_COLOR_MATCHES_PMCOLOR_BYTE_ORDER;
416 
417     // SkColor's ordering may be different from SkPMColor
418     if (sk_color_ne_zero) {
419         memcpy(dst, src, width * sizeof(SkColor));
420         return;
421     }
422 
423     // order isn't same, repack each pixel manually
424     SkPMColor* d = (SkPMColor*)dst;
425     for (int i = 0; i < width; i++) {
426         SkColor c = *src++;
427         *d++ = SkPackARGB32NoCheck(SkColorGetA(c), SkColorGetR(c),
428                                    SkColorGetG(c), SkColorGetB(c));
429     }
430 }
431 
FromColor_D565(void * dst,const SkColor src[],int width,int x,int y)432 static void FromColor_D565(void* dst, const SkColor src[], int width,
433                            int x, int y) {
434     uint16_t* d = (uint16_t*)dst;
435 
436     DITHER_565_SCAN(y);
437     for (int stop = x + width; x < stop; x++) {
438         SkColor c = *src++;
439         *d++ = SkDitherRGBTo565(SkColorGetR(c), SkColorGetG(c), SkColorGetB(c),
440                                 DITHER_VALUE(x));
441     }
442 }
443 
FromColor_D4444(void * dst,const SkColor src[],int width,int x,int y)444 static void FromColor_D4444(void* dst, const SkColor src[], int width,
445                             int x, int y) {
446     SkPMColor16* d = (SkPMColor16*)dst;
447 
448     DITHER_4444_SCAN(y);
449     for (int stop = x + width; x < stop; x++) {
450         SkPMColor pmc = SkPreMultiplyColor(*src++);
451         *d++ = SkDitherARGB32To4444(pmc, DITHER_VALUE(x));
452 //        *d++ = SkPixel32ToPixel4444(pmc);
453     }
454 }
455 
FromColor_D4444_Raw(void * dst,const SkColor src[],int width,int x,int y)456 static void FromColor_D4444_Raw(void* dst, const SkColor src[], int width,
457                             int x, int y) {
458     SkPMColor16* d = (SkPMColor16*)dst;
459 
460     DITHER_4444_SCAN(y);
461     for (int stop = x + width; x < stop; x++) {
462         SkColor c = *src++;
463 
464         // SkPMColor is used because the ordering is ARGB32, even though the target actually premultiplied
465         SkPMColor pmc = SkPackARGB32NoCheck(SkColorGetA(c), SkColorGetR(c),
466                                             SkColorGetG(c), SkColorGetB(c));
467         *d++ = SkDitherARGB32To4444(pmc, DITHER_VALUE(x));
468 //        *d++ = SkPixel32ToPixel4444(pmc);
469     }
470 }
471 
FromColor_DA8(void * dst,const SkColor src[],int width,int x,int y)472 static void FromColor_DA8(void* dst, const SkColor src[], int width, int x, int y) {
473     uint8_t* d = (uint8_t*)dst;
474 
475     for (int stop = x + width; x < stop; x++) {
476         *d++ = SkColorGetA(*src++);
477     }
478 }
479 
480 // can return NULL
ChooseFromColorProc(const SkBitmap & bitmap)481 static FromColorProc ChooseFromColorProc(const SkBitmap& bitmap) {
482     switch (bitmap.colorType()) {
483         case kN32_SkColorType:
484             return bitmap.alphaType() == kPremul_SkAlphaType ? FromColor_D32 : FromColor_D32_Raw;
485         case kARGB_4444_SkColorType:
486             return bitmap.alphaType() == kPremul_SkAlphaType ? FromColor_D4444 :
487                     FromColor_D4444_Raw;
488         case kRGB_565_SkColorType:
489             return FromColor_D565;
490         case kAlpha_8_SkColorType:
491             return FromColor_DA8;
492         default:
493             break;
494     }
495     return NULL;
496 }
497 
SetPixels(JNIEnv * env,jintArray srcColors,int srcOffset,int srcStride,int x,int y,int width,int height,const SkBitmap & dstBitmap)498 bool GraphicsJNI::SetPixels(JNIEnv* env, jintArray srcColors, int srcOffset, int srcStride,
499         int x, int y, int width, int height, const SkBitmap& dstBitmap) {
500     SkAutoLockPixels alp(dstBitmap);
501     void* dst = dstBitmap.getPixels();
502     FromColorProc proc = ChooseFromColorProc(dstBitmap);
503 
504     if (NULL == dst || NULL == proc) {
505         return false;
506     }
507 
508     const jint* array = env->GetIntArrayElements(srcColors, NULL);
509     const SkColor* src = (const SkColor*)array + srcOffset;
510 
511     // reset to to actual choice from caller
512     dst = dstBitmap.getAddr(x, y);
513     // now copy/convert each scanline
514     for (int y = 0; y < height; y++) {
515         proc(dst, src, width, x, y);
516         src += srcStride;
517         dst = (char*)dst + dstBitmap.rowBytes();
518     }
519 
520     dstBitmap.notifyPixelsChanged();
521 
522     env->ReleaseIntArrayElements(srcColors, const_cast<jint*>(array),
523                                  JNI_ABORT);
524     return true;
525 }
526 
527 //////////////////// ToColor procs
528 
529 typedef void (*ToColorProc)(SkColor dst[], const void* src, int width,
530                             SkColorTable*);
531 
ToColor_S32_Alpha(SkColor dst[],const void * src,int width,SkColorTable *)532 static void ToColor_S32_Alpha(SkColor dst[], const void* src, int width,
533                               SkColorTable*) {
534     SkASSERT(width > 0);
535     const SkPMColor* s = (const SkPMColor*)src;
536     do {
537         *dst++ = SkUnPreMultiply::PMColorToColor(*s++);
538     } while (--width != 0);
539 }
540 
ToColor_S32_Raw(SkColor dst[],const void * src,int width,SkColorTable *)541 static void ToColor_S32_Raw(SkColor dst[], const void* src, int width,
542                               SkColorTable*) {
543     SkASSERT(width > 0);
544     const SkPMColor* s = (const SkPMColor*)src;
545     do {
546         SkPMColor c = *s++;
547         *dst++ = SkColorSetARGB(SkGetPackedA32(c), SkGetPackedR32(c),
548                                 SkGetPackedG32(c), SkGetPackedB32(c));
549     } while (--width != 0);
550 }
551 
ToColor_S32_Opaque(SkColor dst[],const void * src,int width,SkColorTable *)552 static void ToColor_S32_Opaque(SkColor dst[], const void* src, int width,
553                                SkColorTable*) {
554     SkASSERT(width > 0);
555     const SkPMColor* s = (const SkPMColor*)src;
556     do {
557         SkPMColor c = *s++;
558         *dst++ = SkColorSetRGB(SkGetPackedR32(c), SkGetPackedG32(c),
559                                SkGetPackedB32(c));
560     } while (--width != 0);
561 }
562 
ToColor_S4444_Alpha(SkColor dst[],const void * src,int width,SkColorTable *)563 static void ToColor_S4444_Alpha(SkColor dst[], const void* src, int width,
564                                 SkColorTable*) {
565     SkASSERT(width > 0);
566     const SkPMColor16* s = (const SkPMColor16*)src;
567     do {
568         *dst++ = SkUnPreMultiply::PMColorToColor(SkPixel4444ToPixel32(*s++));
569     } while (--width != 0);
570 }
571 
ToColor_S4444_Raw(SkColor dst[],const void * src,int width,SkColorTable *)572 static void ToColor_S4444_Raw(SkColor dst[], const void* src, int width,
573                                 SkColorTable*) {
574     SkASSERT(width > 0);
575     const SkPMColor16* s = (const SkPMColor16*)src;
576     do {
577         SkPMColor c = SkPixel4444ToPixel32(*s++);
578         *dst++ = SkColorSetARGB(SkGetPackedA32(c), SkGetPackedR32(c),
579                                 SkGetPackedG32(c), SkGetPackedB32(c));
580     } while (--width != 0);
581 }
582 
ToColor_S4444_Opaque(SkColor dst[],const void * src,int width,SkColorTable *)583 static void ToColor_S4444_Opaque(SkColor dst[], const void* src, int width,
584                                  SkColorTable*) {
585     SkASSERT(width > 0);
586     const SkPMColor16* s = (const SkPMColor16*)src;
587     do {
588         SkPMColor c = SkPixel4444ToPixel32(*s++);
589         *dst++ = SkColorSetRGB(SkGetPackedR32(c), SkGetPackedG32(c),
590                                SkGetPackedB32(c));
591     } while (--width != 0);
592 }
593 
ToColor_S565(SkColor dst[],const void * src,int width,SkColorTable *)594 static void ToColor_S565(SkColor dst[], const void* src, int width,
595                          SkColorTable*) {
596     SkASSERT(width > 0);
597     const uint16_t* s = (const uint16_t*)src;
598     do {
599         uint16_t c = *s++;
600         *dst++ =  SkColorSetRGB(SkPacked16ToR32(c), SkPacked16ToG32(c),
601                                 SkPacked16ToB32(c));
602     } while (--width != 0);
603 }
604 
ToColor_SI8_Alpha(SkColor dst[],const void * src,int width,SkColorTable * ctable)605 static void ToColor_SI8_Alpha(SkColor dst[], const void* src, int width,
606                               SkColorTable* ctable) {
607     SkASSERT(width > 0);
608     const uint8_t* s = (const uint8_t*)src;
609     const SkPMColor* colors = ctable->readColors();
610     do {
611         *dst++ = SkUnPreMultiply::PMColorToColor(colors[*s++]);
612     } while (--width != 0);
613 }
614 
ToColor_SI8_Raw(SkColor dst[],const void * src,int width,SkColorTable * ctable)615 static void ToColor_SI8_Raw(SkColor dst[], const void* src, int width,
616                               SkColorTable* ctable) {
617     SkASSERT(width > 0);
618     const uint8_t* s = (const uint8_t*)src;
619     const SkPMColor* colors = ctable->readColors();
620     do {
621         SkPMColor c = colors[*s++];
622         *dst++ = SkColorSetARGB(SkGetPackedA32(c), SkGetPackedR32(c),
623                                 SkGetPackedG32(c), SkGetPackedB32(c));
624     } while (--width != 0);
625 }
626 
ToColor_SI8_Opaque(SkColor dst[],const void * src,int width,SkColorTable * ctable)627 static void ToColor_SI8_Opaque(SkColor dst[], const void* src, int width,
628                                SkColorTable* ctable) {
629     SkASSERT(width > 0);
630     const uint8_t* s = (const uint8_t*)src;
631     const SkPMColor* colors = ctable->readColors();
632     do {
633         SkPMColor c = colors[*s++];
634         *dst++ = SkColorSetRGB(SkGetPackedR32(c), SkGetPackedG32(c),
635                                SkGetPackedB32(c));
636     } while (--width != 0);
637 }
638 
ToColor_SA8(SkColor dst[],const void * src,int width,SkColorTable *)639 static void ToColor_SA8(SkColor dst[], const void* src, int width, SkColorTable*) {
640     SkASSERT(width > 0);
641     const uint8_t* s = (const uint8_t*)src;
642     do {
643         uint8_t c = *s++;
644         *dst++ = SkColorSetARGB(c, c, c, c);
645     } while (--width != 0);
646 }
647 
648 // can return NULL
ChooseToColorProc(const SkBitmap & src)649 static ToColorProc ChooseToColorProc(const SkBitmap& src) {
650     switch (src.colorType()) {
651         case kN32_SkColorType:
652             switch (src.alphaType()) {
653                 case kOpaque_SkAlphaType:
654                     return ToColor_S32_Opaque;
655                 case kPremul_SkAlphaType:
656                     return ToColor_S32_Alpha;
657                 case kUnpremul_SkAlphaType:
658                     return ToColor_S32_Raw;
659                 default:
660                     return NULL;
661             }
662         case kARGB_4444_SkColorType:
663             switch (src.alphaType()) {
664                 case kOpaque_SkAlphaType:
665                     return ToColor_S4444_Opaque;
666                 case kPremul_SkAlphaType:
667                     return ToColor_S4444_Alpha;
668                 case kUnpremul_SkAlphaType:
669                     return ToColor_S4444_Raw;
670                 default:
671                     return NULL;
672             }
673         case kRGB_565_SkColorType:
674             return ToColor_S565;
675         case kIndex_8_SkColorType:
676             if (src.getColorTable() == NULL) {
677                 return NULL;
678             }
679             switch (src.alphaType()) {
680                 case kOpaque_SkAlphaType:
681                     return ToColor_SI8_Opaque;
682                 case kPremul_SkAlphaType:
683                     return ToColor_SI8_Alpha;
684                 case kUnpremul_SkAlphaType:
685                     return ToColor_SI8_Raw;
686                 default:
687                     return NULL;
688             }
689         case kAlpha_8_SkColorType:
690             return ToColor_SA8;
691         default:
692             break;
693     }
694     return NULL;
695 }
696 
697 ///////////////////////////////////////////////////////////////////////////////
698 ///////////////////////////////////////////////////////////////////////////////
699 
getPremulBitmapCreateFlags(bool isMutable)700 static int getPremulBitmapCreateFlags(bool isMutable) {
701     int flags = GraphicsJNI::kBitmapCreateFlag_Premultiplied;
702     if (isMutable) flags |= GraphicsJNI::kBitmapCreateFlag_Mutable;
703     return flags;
704 }
705 
Bitmap_creator(JNIEnv * env,jobject,jintArray jColors,jint offset,jint stride,jint width,jint height,jint configHandle,jboolean isMutable)706 static jobject Bitmap_creator(JNIEnv* env, jobject, jintArray jColors,
707                               jint offset, jint stride, jint width, jint height,
708                               jint configHandle, jboolean isMutable) {
709     SkColorType colorType = GraphicsJNI::legacyBitmapConfigToColorType(configHandle);
710     if (NULL != jColors) {
711         size_t n = env->GetArrayLength(jColors);
712         if (n < SkAbs32(stride) * (size_t)height) {
713             doThrowAIOOBE(env);
714             return NULL;
715         }
716     }
717 
718     // ARGB_4444 is a deprecated format, convert automatically to 8888
719     if (colorType == kARGB_4444_SkColorType) {
720         colorType = kN32_SkColorType;
721     }
722 
723     SkBitmap bitmap;
724     bitmap.setInfo(SkImageInfo::Make(width, height, colorType, kPremul_SkAlphaType));
725 
726     Bitmap* nativeBitmap = GraphicsJNI::allocateJavaPixelRef(env, &bitmap, NULL);
727     if (!nativeBitmap) {
728         return NULL;
729     }
730 
731     if (jColors != NULL) {
732         GraphicsJNI::SetPixels(env, jColors, offset, stride,
733                 0, 0, width, height, bitmap);
734     }
735 
736     return GraphicsJNI::createBitmap(env, nativeBitmap,
737             getPremulBitmapCreateFlags(isMutable));
738 }
739 
Bitmap_copy(JNIEnv * env,jobject,jlong srcHandle,jint dstConfigHandle,jboolean isMutable)740 static jobject Bitmap_copy(JNIEnv* env, jobject, jlong srcHandle,
741                            jint dstConfigHandle, jboolean isMutable) {
742     SkBitmap src;
743     reinterpret_cast<Bitmap*>(srcHandle)->getSkBitmap(&src);
744     SkColorType dstCT = GraphicsJNI::legacyBitmapConfigToColorType(dstConfigHandle);
745     SkBitmap            result;
746     JavaPixelAllocator  allocator(env);
747 
748     if (!src.copyTo(&result, dstCT, &allocator)) {
749         return NULL;
750     }
751     Bitmap* bitmap = allocator.getStorageObjAndReset();
752     return GraphicsJNI::createBitmap(env, bitmap,
753             getPremulBitmapCreateFlags(isMutable));
754 }
755 
Bitmap_copyAshmemImpl(JNIEnv * env,SkBitmap & src,SkColorType & dstCT)756 static Bitmap* Bitmap_copyAshmemImpl(JNIEnv* env, SkBitmap& src, SkColorType& dstCT) {
757     SkBitmap result;
758 
759     AshmemPixelAllocator allocator(env);
760     if (!src.copyTo(&result, dstCT, &allocator)) {
761         return NULL;
762     }
763     Bitmap* bitmap = allocator.getStorageObjAndReset();
764     bitmap->peekAtPixelRef()->setImmutable();
765     return bitmap;
766 }
767 
Bitmap_copyAshmem(JNIEnv * env,jobject,jlong srcHandle)768 static jobject Bitmap_copyAshmem(JNIEnv* env, jobject, jlong srcHandle) {
769     SkBitmap src;
770     reinterpret_cast<Bitmap*>(srcHandle)->getSkBitmap(&src);
771     SkColorType dstCT = src.colorType();
772     Bitmap* bitmap = Bitmap_copyAshmemImpl(env, src, dstCT);
773     jobject ret = GraphicsJNI::createBitmap(env, bitmap, getPremulBitmapCreateFlags(false));
774     return ret;
775 }
776 
Bitmap_copyAshmemConfig(JNIEnv * env,jobject,jlong srcHandle,jint dstConfigHandle)777 static jobject Bitmap_copyAshmemConfig(JNIEnv* env, jobject, jlong srcHandle, jint dstConfigHandle) {
778     SkBitmap src;
779     reinterpret_cast<Bitmap*>(srcHandle)->getSkBitmap(&src);
780     SkColorType dstCT = GraphicsJNI::legacyBitmapConfigToColorType(dstConfigHandle);
781     Bitmap* bitmap = Bitmap_copyAshmemImpl(env, src, dstCT);
782     jobject ret = GraphicsJNI::createBitmap(env, bitmap, getPremulBitmapCreateFlags(false));
783     return ret;
784 }
785 
Bitmap_destruct(Bitmap * bitmap)786 static void Bitmap_destruct(Bitmap* bitmap) {
787     bitmap->detachFromJava();
788 }
789 
Bitmap_getNativeFinalizer(JNIEnv *,jobject)790 static jlong Bitmap_getNativeFinalizer(JNIEnv*, jobject) {
791     return static_cast<jlong>(reinterpret_cast<uintptr_t>(&Bitmap_destruct));
792 }
793 
Bitmap_recycle(JNIEnv * env,jobject,jlong bitmapHandle)794 static jboolean Bitmap_recycle(JNIEnv* env, jobject, jlong bitmapHandle) {
795     LocalScopedBitmap bitmap(bitmapHandle);
796     bitmap->freePixels();
797     return JNI_TRUE;
798 }
799 
Bitmap_reconfigure(JNIEnv * env,jobject clazz,jlong bitmapHandle,jint width,jint height,jint configHandle,jint allocSize,jboolean requestPremul)800 static void Bitmap_reconfigure(JNIEnv* env, jobject clazz, jlong bitmapHandle,
801         jint width, jint height, jint configHandle, jint allocSize,
802         jboolean requestPremul) {
803     LocalScopedBitmap bitmap(bitmapHandle);
804     SkColorType colorType = GraphicsJNI::legacyBitmapConfigToColorType(configHandle);
805 
806     // ARGB_4444 is a deprecated format, convert automatically to 8888
807     if (colorType == kARGB_4444_SkColorType) {
808         colorType = kN32_SkColorType;
809     }
810 
811     if (width * height * SkColorTypeBytesPerPixel(colorType) > allocSize) {
812         // done in native as there's no way to get BytesPerPixel in Java
813         doThrowIAE(env, "Bitmap not large enough to support new configuration");
814         return;
815     }
816     SkAlphaType alphaType;
817     if (bitmap->info().colorType() != kRGB_565_SkColorType
818             && bitmap->info().alphaType() == kOpaque_SkAlphaType) {
819         // If the original bitmap was set to opaque, keep that setting, unless it
820         // was 565, which is required to be opaque.
821         alphaType = kOpaque_SkAlphaType;
822     } else {
823         // Otherwise respect the premultiplied request.
824         alphaType = requestPremul ? kPremul_SkAlphaType : kUnpremul_SkAlphaType;
825     }
826     bitmap->reconfigure(SkImageInfo::Make(width, height, colorType, alphaType));
827 }
828 
829 // These must match the int values in Bitmap.java
830 enum JavaEncodeFormat {
831     kJPEG_JavaEncodeFormat = 0,
832     kPNG_JavaEncodeFormat = 1,
833     kWEBP_JavaEncodeFormat = 2
834 };
835 
Bitmap_compress(JNIEnv * env,jobject clazz,jlong bitmapHandle,jint format,jint quality,jobject jstream,jbyteArray jstorage)836 static jboolean Bitmap_compress(JNIEnv* env, jobject clazz, jlong bitmapHandle,
837                                 jint format, jint quality,
838                                 jobject jstream, jbyteArray jstorage) {
839 
840     LocalScopedBitmap bitmap(bitmapHandle);
841     SkImageEncoder::Type fm;
842 
843     switch (format) {
844     case kJPEG_JavaEncodeFormat:
845         fm = SkImageEncoder::kJPEG_Type;
846         break;
847     case kPNG_JavaEncodeFormat:
848         fm = SkImageEncoder::kPNG_Type;
849         break;
850     case kWEBP_JavaEncodeFormat:
851         fm = SkImageEncoder::kWEBP_Type;
852         break;
853     default:
854         return JNI_FALSE;
855     }
856 
857     if (!bitmap.valid()) {
858         return JNI_FALSE;
859     }
860 
861     bool success = false;
862 
863     std::unique_ptr<SkWStream> strm(CreateJavaOutputStreamAdaptor(env, jstream, jstorage));
864     if (!strm.get()) {
865         return JNI_FALSE;
866     }
867 
868     std::unique_ptr<SkImageEncoder> encoder(SkImageEncoder::Create(fm));
869     if (encoder.get()) {
870         SkBitmap skbitmap;
871         bitmap->getSkBitmap(&skbitmap);
872         success = encoder->encodeStream(strm.get(), skbitmap, quality);
873     }
874     return success ? JNI_TRUE : JNI_FALSE;
875 }
876 
Bitmap_erase(JNIEnv * env,jobject,jlong bitmapHandle,jint color)877 static void Bitmap_erase(JNIEnv* env, jobject, jlong bitmapHandle, jint color) {
878     LocalScopedBitmap bitmap(bitmapHandle);
879     SkBitmap skBitmap;
880     bitmap->getSkBitmap(&skBitmap);
881     skBitmap.eraseColor(color);
882 }
883 
Bitmap_rowBytes(JNIEnv * env,jobject,jlong bitmapHandle)884 static jint Bitmap_rowBytes(JNIEnv* env, jobject, jlong bitmapHandle) {
885     LocalScopedBitmap bitmap(bitmapHandle);
886     return static_cast<jint>(bitmap->rowBytes());
887 }
888 
Bitmap_config(JNIEnv * env,jobject,jlong bitmapHandle)889 static jint Bitmap_config(JNIEnv* env, jobject, jlong bitmapHandle) {
890     LocalScopedBitmap bitmap(bitmapHandle);
891     return GraphicsJNI::colorTypeToLegacyBitmapConfig(bitmap->info().colorType());
892 }
893 
Bitmap_getGenerationId(JNIEnv * env,jobject,jlong bitmapHandle)894 static jint Bitmap_getGenerationId(JNIEnv* env, jobject, jlong bitmapHandle) {
895     LocalScopedBitmap bitmap(bitmapHandle);
896     return static_cast<jint>(bitmap->peekAtPixelRef()->getGenerationID());
897 }
898 
Bitmap_isPremultiplied(JNIEnv * env,jobject,jlong bitmapHandle)899 static jboolean Bitmap_isPremultiplied(JNIEnv* env, jobject, jlong bitmapHandle) {
900     LocalScopedBitmap bitmap(bitmapHandle);
901     if (bitmap->info().alphaType() == kPremul_SkAlphaType) {
902         return JNI_TRUE;
903     }
904     return JNI_FALSE;
905 }
906 
Bitmap_hasAlpha(JNIEnv * env,jobject,jlong bitmapHandle)907 static jboolean Bitmap_hasAlpha(JNIEnv* env, jobject, jlong bitmapHandle) {
908     LocalScopedBitmap bitmap(bitmapHandle);
909     return !bitmap->info().isOpaque() ? JNI_TRUE : JNI_FALSE;
910 }
911 
Bitmap_setHasAlpha(JNIEnv * env,jobject,jlong bitmapHandle,jboolean hasAlpha,jboolean requestPremul)912 static void Bitmap_setHasAlpha(JNIEnv* env, jobject, jlong bitmapHandle,
913         jboolean hasAlpha, jboolean requestPremul) {
914     LocalScopedBitmap bitmap(bitmapHandle);
915     if (hasAlpha) {
916         bitmap->setAlphaType(
917                 requestPremul ? kPremul_SkAlphaType : kUnpremul_SkAlphaType);
918     } else {
919         bitmap->setAlphaType(kOpaque_SkAlphaType);
920     }
921 }
922 
Bitmap_setPremultiplied(JNIEnv * env,jobject,jlong bitmapHandle,jboolean isPremul)923 static void Bitmap_setPremultiplied(JNIEnv* env, jobject, jlong bitmapHandle,
924         jboolean isPremul) {
925     LocalScopedBitmap bitmap(bitmapHandle);
926     if (!bitmap->info().isOpaque()) {
927         if (isPremul) {
928             bitmap->setAlphaType(kPremul_SkAlphaType);
929         } else {
930             bitmap->setAlphaType(kUnpremul_SkAlphaType);
931         }
932     }
933 }
934 
Bitmap_hasMipMap(JNIEnv * env,jobject,jlong bitmapHandle)935 static jboolean Bitmap_hasMipMap(JNIEnv* env, jobject, jlong bitmapHandle) {
936     LocalScopedBitmap bitmap(bitmapHandle);
937     return bitmap->hasHardwareMipMap() ? JNI_TRUE : JNI_FALSE;
938 }
939 
Bitmap_setHasMipMap(JNIEnv * env,jobject,jlong bitmapHandle,jboolean hasMipMap)940 static void Bitmap_setHasMipMap(JNIEnv* env, jobject, jlong bitmapHandle,
941                                 jboolean hasMipMap) {
942     LocalScopedBitmap bitmap(bitmapHandle);
943     bitmap->setHasHardwareMipMap(hasMipMap);
944 }
945 
946 ///////////////////////////////////////////////////////////////////////////////
947 
Bitmap_createFromParcel(JNIEnv * env,jobject,jobject parcel)948 static jobject Bitmap_createFromParcel(JNIEnv* env, jobject, jobject parcel) {
949     if (parcel == NULL) {
950         SkDebugf("-------- unparcel parcel is NULL\n");
951         return NULL;
952     }
953 
954     android::Parcel* p = android::parcelForJavaObject(env, parcel);
955 
956     const bool        isMutable = p->readInt32() != 0;
957     const SkColorType colorType = (SkColorType)p->readInt32();
958     const SkAlphaType alphaType = (SkAlphaType)p->readInt32();
959     const int         width = p->readInt32();
960     const int         height = p->readInt32();
961     const int         rowBytes = p->readInt32();
962     const int         density = p->readInt32();
963 
964     if (kN32_SkColorType != colorType &&
965             kRGB_565_SkColorType != colorType &&
966             kARGB_4444_SkColorType != colorType &&
967             kIndex_8_SkColorType != colorType &&
968             kAlpha_8_SkColorType != colorType) {
969         SkDebugf("Bitmap_createFromParcel unknown colortype: %d\n", colorType);
970         return NULL;
971     }
972 
973     std::unique_ptr<SkBitmap> bitmap(new SkBitmap);
974 
975     if (!bitmap->setInfo(SkImageInfo::Make(width, height, colorType, alphaType), rowBytes)) {
976         return NULL;
977     }
978 
979     SkColorTable* ctable = NULL;
980     if (colorType == kIndex_8_SkColorType) {
981         int count = p->readInt32();
982         if (count < 0 || count > 256) {
983             // The data is corrupt, since SkColorTable enforces a value between 0 and 256,
984             // inclusive.
985             return NULL;
986         }
987         if (count > 0) {
988             size_t size = count * sizeof(SkPMColor);
989             const SkPMColor* src = (const SkPMColor*)p->readInplace(size);
990             if (src == NULL) {
991                 return NULL;
992             }
993             ctable = new SkColorTable(src, count);
994         }
995     }
996 
997     // Read the bitmap blob.
998     size_t size = bitmap->getSize();
999     android::Parcel::ReadableBlob blob;
1000     android::status_t status = p->readBlob(size, &blob);
1001     if (status) {
1002         SkSafeUnref(ctable);
1003         doThrowRE(env, "Could not read bitmap blob.");
1004         return NULL;
1005     }
1006 
1007     // Map the bitmap in place from the ashmem region if possible otherwise copy.
1008     Bitmap* nativeBitmap;
1009     if (blob.fd() >= 0 && (blob.isMutable() || !isMutable) && (size >= ASHMEM_BITMAP_MIN_SIZE)) {
1010 #if DEBUG_PARCEL
1011         ALOGD("Bitmap.createFromParcel: mapped contents of %s bitmap from %s blob "
1012                 "(fds %s)",
1013                 isMutable ? "mutable" : "immutable",
1014                 blob.isMutable() ? "mutable" : "immutable",
1015                 p->allowFds() ? "allowed" : "forbidden");
1016 #endif
1017         // Dup the file descriptor so we can keep a reference to it after the Parcel
1018         // is disposed.
1019         int dupFd = dup(blob.fd());
1020         if (dupFd < 0) {
1021             ALOGE("Error allocating dup fd. Error:%d", errno);
1022             blob.release();
1023             SkSafeUnref(ctable);
1024             doThrowRE(env, "Could not allocate dup blob fd.");
1025             return NULL;
1026         }
1027 
1028         // Map the pixels in place and take ownership of the ashmem region.
1029         nativeBitmap = GraphicsJNI::mapAshmemPixelRef(env, bitmap.get(),
1030                 ctable, dupFd, const_cast<void*>(blob.data()), size, !isMutable);
1031         SkSafeUnref(ctable);
1032         if (!nativeBitmap) {
1033             close(dupFd);
1034             blob.release();
1035             doThrowRE(env, "Could not allocate ashmem pixel ref.");
1036             return NULL;
1037         }
1038 
1039         // Clear the blob handle, don't release it.
1040         blob.clear();
1041     } else {
1042 #if DEBUG_PARCEL
1043         if (blob.fd() >= 0) {
1044             ALOGD("Bitmap.createFromParcel: copied contents of mutable bitmap "
1045                     "from immutable blob (fds %s)",
1046                     p->allowFds() ? "allowed" : "forbidden");
1047         } else {
1048             ALOGD("Bitmap.createFromParcel: copied contents from %s blob "
1049                     "(fds %s)",
1050                     blob.isMutable() ? "mutable" : "immutable",
1051                     p->allowFds() ? "allowed" : "forbidden");
1052         }
1053 #endif
1054 
1055         // Copy the pixels into a new buffer.
1056         nativeBitmap = GraphicsJNI::allocateJavaPixelRef(env, bitmap.get(), ctable);
1057         SkSafeUnref(ctable);
1058         if (!nativeBitmap) {
1059             blob.release();
1060             doThrowRE(env, "Could not allocate java pixel ref.");
1061             return NULL;
1062         }
1063         bitmap->lockPixels();
1064         memcpy(bitmap->getPixels(), blob.data(), size);
1065         bitmap->unlockPixels();
1066 
1067         // Release the blob handle.
1068         blob.release();
1069     }
1070 
1071     return GraphicsJNI::createBitmap(env, nativeBitmap,
1072             getPremulBitmapCreateFlags(isMutable), NULL, NULL, density);
1073 }
1074 
Bitmap_writeToParcel(JNIEnv * env,jobject,jlong bitmapHandle,jboolean isMutable,jint density,jobject parcel)1075 static jboolean Bitmap_writeToParcel(JNIEnv* env, jobject,
1076                                      jlong bitmapHandle,
1077                                      jboolean isMutable, jint density,
1078                                      jobject parcel) {
1079     if (parcel == NULL) {
1080         SkDebugf("------- writeToParcel null parcel\n");
1081         return JNI_FALSE;
1082     }
1083 
1084     android::Parcel* p = android::parcelForJavaObject(env, parcel);
1085     SkBitmap bitmap;
1086 
1087     android::Bitmap* androidBitmap = reinterpret_cast<Bitmap*>(bitmapHandle);
1088     androidBitmap->getSkBitmap(&bitmap);
1089 
1090     p->writeInt32(isMutable);
1091     p->writeInt32(bitmap.colorType());
1092     p->writeInt32(bitmap.alphaType());
1093     p->writeInt32(bitmap.width());
1094     p->writeInt32(bitmap.height());
1095     p->writeInt32(bitmap.rowBytes());
1096     p->writeInt32(density);
1097 
1098     if (bitmap.colorType() == kIndex_8_SkColorType) {
1099         // The bitmap needs to be locked to access its color table.
1100         SkAutoLockPixels alp(bitmap);
1101         SkColorTable* ctable = bitmap.getColorTable();
1102         if (ctable != NULL) {
1103             int count = ctable->count();
1104             p->writeInt32(count);
1105             memcpy(p->writeInplace(count * sizeof(SkPMColor)),
1106                    ctable->readColors(), count * sizeof(SkPMColor));
1107         } else {
1108             p->writeInt32(0);   // indicate no ctable
1109         }
1110     }
1111 
1112     // Transfer the underlying ashmem region if we have one and it's immutable.
1113     android::status_t status;
1114     int fd = androidBitmap->getAshmemFd();
1115     if (fd >= 0 && !isMutable && p->allowFds()) {
1116 #if DEBUG_PARCEL
1117         ALOGD("Bitmap.writeToParcel: transferring immutable bitmap's ashmem fd as "
1118                 "immutable blob (fds %s)",
1119                 p->allowFds() ? "allowed" : "forbidden");
1120 #endif
1121 
1122         status = p->writeDupImmutableBlobFileDescriptor(fd);
1123         if (status) {
1124             doThrowRE(env, "Could not write bitmap blob file descriptor.");
1125             return JNI_FALSE;
1126         }
1127         return JNI_TRUE;
1128     }
1129 
1130     // Copy the bitmap to a new blob.
1131     bool mutableCopy = isMutable;
1132 #if DEBUG_PARCEL
1133     ALOGD("Bitmap.writeToParcel: copying %s bitmap into new %s blob (fds %s)",
1134             isMutable ? "mutable" : "immutable",
1135             mutableCopy ? "mutable" : "immutable",
1136             p->allowFds() ? "allowed" : "forbidden");
1137 #endif
1138 
1139     size_t size = bitmap.getSize();
1140     android::Parcel::WritableBlob blob;
1141     status = p->writeBlob(size, mutableCopy, &blob);
1142     if (status) {
1143         doThrowRE(env, "Could not copy bitmap to parcel blob.");
1144         return JNI_FALSE;
1145     }
1146 
1147     bitmap.lockPixels();
1148     const void* pSrc =  bitmap.getPixels();
1149     if (pSrc == NULL) {
1150         memset(blob.data(), 0, size);
1151     } else {
1152         memcpy(blob.data(), pSrc, size);
1153     }
1154     bitmap.unlockPixels();
1155 
1156     blob.release();
1157     return JNI_TRUE;
1158 }
1159 
Bitmap_extractAlpha(JNIEnv * env,jobject clazz,jlong srcHandle,jlong paintHandle,jintArray offsetXY)1160 static jobject Bitmap_extractAlpha(JNIEnv* env, jobject clazz,
1161                                    jlong srcHandle, jlong paintHandle,
1162                                    jintArray offsetXY) {
1163     SkBitmap src;
1164     reinterpret_cast<Bitmap*>(srcHandle)->getSkBitmap(&src);
1165     const android::Paint* paint = reinterpret_cast<android::Paint*>(paintHandle);
1166     SkIPoint  offset;
1167     SkBitmap dst;
1168     JavaPixelAllocator allocator(env);
1169 
1170     src.extractAlpha(&dst, paint, &allocator, &offset);
1171     // If Skia can't allocate pixels for destination bitmap, it resets
1172     // it, that is set its pixels buffer to NULL, and zero width and height.
1173     if (dst.getPixels() == NULL && src.getPixels() != NULL) {
1174         doThrowOOME(env, "failed to allocate pixels for alpha");
1175         return NULL;
1176     }
1177     if (offsetXY != 0 && env->GetArrayLength(offsetXY) >= 2) {
1178         int* array = env->GetIntArrayElements(offsetXY, NULL);
1179         array[0] = offset.fX;
1180         array[1] = offset.fY;
1181         env->ReleaseIntArrayElements(offsetXY, array, 0);
1182     }
1183 
1184     return GraphicsJNI::createBitmap(env, allocator.getStorageObjAndReset(),
1185             getPremulBitmapCreateFlags(true));
1186 }
1187 
1188 ///////////////////////////////////////////////////////////////////////////////
1189 
Bitmap_getPixel(JNIEnv * env,jobject,jlong bitmapHandle,jint x,jint y)1190 static jint Bitmap_getPixel(JNIEnv* env, jobject, jlong bitmapHandle,
1191         jint x, jint y) {
1192     SkBitmap bitmap;
1193     reinterpret_cast<Bitmap*>(bitmapHandle)->getSkBitmap(&bitmap);
1194     SkAutoLockPixels alp(bitmap);
1195 
1196     ToColorProc proc = ChooseToColorProc(bitmap);
1197     if (NULL == proc) {
1198         return 0;
1199     }
1200     const void* src = bitmap.getAddr(x, y);
1201     if (NULL == src) {
1202         return 0;
1203     }
1204 
1205     SkColor dst[1];
1206     proc(dst, src, 1, bitmap.getColorTable());
1207     return static_cast<jint>(dst[0]);
1208 }
1209 
Bitmap_getPixels(JNIEnv * env,jobject,jlong bitmapHandle,jintArray pixelArray,jint offset,jint stride,jint x,jint y,jint width,jint height)1210 static void Bitmap_getPixels(JNIEnv* env, jobject, jlong bitmapHandle,
1211         jintArray pixelArray, jint offset, jint stride,
1212         jint x, jint y, jint width, jint height) {
1213     SkBitmap bitmap;
1214     reinterpret_cast<Bitmap*>(bitmapHandle)->getSkBitmap(&bitmap);
1215     SkAutoLockPixels alp(bitmap);
1216 
1217     ToColorProc proc = ChooseToColorProc(bitmap);
1218     if (NULL == proc) {
1219         return;
1220     }
1221     const void* src = bitmap.getAddr(x, y);
1222     if (NULL == src) {
1223         return;
1224     }
1225 
1226     SkColorTable* ctable = bitmap.getColorTable();
1227     jint* dst = env->GetIntArrayElements(pixelArray, NULL);
1228     SkColor* d = (SkColor*)dst + offset;
1229     while (--height >= 0) {
1230         proc(d, src, width, ctable);
1231         d += stride;
1232         src = (void*)((const char*)src + bitmap.rowBytes());
1233     }
1234     env->ReleaseIntArrayElements(pixelArray, dst, 0);
1235 }
1236 
1237 ///////////////////////////////////////////////////////////////////////////////
1238 
Bitmap_setPixel(JNIEnv * env,jobject,jlong bitmapHandle,jint x,jint y,jint colorHandle)1239 static void Bitmap_setPixel(JNIEnv* env, jobject, jlong bitmapHandle,
1240         jint x, jint y, jint colorHandle) {
1241     SkBitmap bitmap;
1242     reinterpret_cast<Bitmap*>(bitmapHandle)->getSkBitmap(&bitmap);
1243     SkColor color = static_cast<SkColor>(colorHandle);
1244     SkAutoLockPixels alp(bitmap);
1245     if (NULL == bitmap.getPixels()) {
1246         return;
1247     }
1248 
1249     FromColorProc proc = ChooseFromColorProc(bitmap);
1250     if (NULL == proc) {
1251         return;
1252     }
1253 
1254     proc(bitmap.getAddr(x, y), &color, 1, x, y);
1255     bitmap.notifyPixelsChanged();
1256 }
1257 
Bitmap_setPixels(JNIEnv * env,jobject,jlong bitmapHandle,jintArray pixelArray,jint offset,jint stride,jint x,jint y,jint width,jint height)1258 static void Bitmap_setPixels(JNIEnv* env, jobject, jlong bitmapHandle,
1259         jintArray pixelArray, jint offset, jint stride,
1260         jint x, jint y, jint width, jint height) {
1261     SkBitmap bitmap;
1262     reinterpret_cast<Bitmap*>(bitmapHandle)->getSkBitmap(&bitmap);
1263     GraphicsJNI::SetPixels(env, pixelArray, offset, stride,
1264             x, y, width, height, bitmap);
1265 }
1266 
Bitmap_copyPixelsToBuffer(JNIEnv * env,jobject,jlong bitmapHandle,jobject jbuffer)1267 static void Bitmap_copyPixelsToBuffer(JNIEnv* env, jobject,
1268                                       jlong bitmapHandle, jobject jbuffer) {
1269     SkBitmap bitmap;
1270     reinterpret_cast<Bitmap*>(bitmapHandle)->getSkBitmap(&bitmap);
1271     SkAutoLockPixels alp(bitmap);
1272     const void* src = bitmap.getPixels();
1273 
1274     if (NULL != src) {
1275         android::AutoBufferPointer abp(env, jbuffer, JNI_TRUE);
1276 
1277         // the java side has already checked that buffer is large enough
1278         memcpy(abp.pointer(), src, bitmap.getSize());
1279     }
1280 }
1281 
Bitmap_copyPixelsFromBuffer(JNIEnv * env,jobject,jlong bitmapHandle,jobject jbuffer)1282 static void Bitmap_copyPixelsFromBuffer(JNIEnv* env, jobject,
1283                                         jlong bitmapHandle, jobject jbuffer) {
1284     SkBitmap bitmap;
1285     reinterpret_cast<Bitmap*>(bitmapHandle)->getSkBitmap(&bitmap);
1286     SkAutoLockPixels alp(bitmap);
1287     void* dst = bitmap.getPixels();
1288 
1289     if (NULL != dst) {
1290         android::AutoBufferPointer abp(env, jbuffer, JNI_FALSE);
1291         // the java side has already checked that buffer is large enough
1292         memcpy(dst, abp.pointer(), bitmap.getSize());
1293         bitmap.notifyPixelsChanged();
1294     }
1295 }
1296 
Bitmap_sameAs(JNIEnv * env,jobject,jlong bm0Handle,jlong bm1Handle)1297 static jboolean Bitmap_sameAs(JNIEnv* env, jobject, jlong bm0Handle,
1298                               jlong bm1Handle) {
1299     SkBitmap bm0;
1300     SkBitmap bm1;
1301     reinterpret_cast<Bitmap*>(bm0Handle)->getSkBitmap(&bm0);
1302     reinterpret_cast<Bitmap*>(bm1Handle)->getSkBitmap(&bm1);
1303     if (bm0.width() != bm1.width() ||
1304         bm0.height() != bm1.height() ||
1305         bm0.colorType() != bm1.colorType()) {
1306         return JNI_FALSE;
1307     }
1308 
1309     SkAutoLockPixels alp0(bm0);
1310     SkAutoLockPixels alp1(bm1);
1311 
1312     // if we can't load the pixels, return false
1313     if (NULL == bm0.getPixels() || NULL == bm1.getPixels()) {
1314         return JNI_FALSE;
1315     }
1316 
1317     if (bm0.colorType() == kIndex_8_SkColorType) {
1318         SkColorTable* ct0 = bm0.getColorTable();
1319         SkColorTable* ct1 = bm1.getColorTable();
1320         if (NULL == ct0 || NULL == ct1) {
1321             return JNI_FALSE;
1322         }
1323         if (ct0->count() != ct1->count()) {
1324             return JNI_FALSE;
1325         }
1326 
1327         const size_t size = ct0->count() * sizeof(SkPMColor);
1328         if (memcmp(ct0->readColors(), ct1->readColors(), size) != 0) {
1329             return JNI_FALSE;
1330         }
1331     }
1332 
1333     // now compare each scanline. We can't do the entire buffer at once,
1334     // since we don't care about the pixel values that might extend beyond
1335     // the width (since the scanline might be larger than the logical width)
1336     const int h = bm0.height();
1337     const size_t size = bm0.width() * bm0.bytesPerPixel();
1338     for (int y = 0; y < h; y++) {
1339         // SkBitmap::getAddr(int, int) may return NULL due to unrecognized config
1340         // (ex: kRLE_Index8_Config). This will cause memcmp method to crash. Since bm0
1341         // and bm1 both have pixel data() (have passed NULL == getPixels() check),
1342         // those 2 bitmaps should be valid (only unrecognized), we return JNI_FALSE
1343         // to warn user those 2 unrecognized config bitmaps may be different.
1344         void *bm0Addr = bm0.getAddr(0, y);
1345         void *bm1Addr = bm1.getAddr(0, y);
1346 
1347         if(bm0Addr == NULL || bm1Addr == NULL) {
1348             return JNI_FALSE;
1349         }
1350 
1351         if (memcmp(bm0Addr, bm1Addr, size) != 0) {
1352             return JNI_FALSE;
1353         }
1354     }
1355     return JNI_TRUE;
1356 }
1357 
Bitmap_refPixelRef(JNIEnv * env,jobject,jlong bitmapHandle)1358 static jlong Bitmap_refPixelRef(JNIEnv* env, jobject, jlong bitmapHandle) {
1359     LocalScopedBitmap bitmap(bitmapHandle);
1360     SkPixelRef* pixelRef = bitmap.valid() ? bitmap->peekAtPixelRef() : nullptr;
1361     SkSafeRef(pixelRef);
1362     return reinterpret_cast<jlong>(pixelRef);
1363 }
1364 
Bitmap_prepareToDraw(JNIEnv * env,jobject,jlong bitmapPtr)1365 static void Bitmap_prepareToDraw(JNIEnv* env, jobject, jlong bitmapPtr) {
1366     LocalScopedBitmap bitmapHandle(bitmapPtr);
1367     if (!bitmapHandle.valid()) return;
1368     SkBitmap bitmap;
1369     bitmapHandle->getSkBitmap(&bitmap);
1370     android::uirenderer::renderthread::RenderProxy::prepareToDraw(bitmap);
1371 }
1372 
1373 ///////////////////////////////////////////////////////////////////////////////
1374 
1375 static const JNINativeMethod gBitmapMethods[] = {
1376     {   "nativeCreate",             "([IIIIIIZ)Landroid/graphics/Bitmap;",
1377         (void*)Bitmap_creator },
1378     {   "nativeCopy",               "(JIZ)Landroid/graphics/Bitmap;",
1379         (void*)Bitmap_copy },
1380     {   "nativeCopyAshmem",         "(J)Landroid/graphics/Bitmap;",
1381         (void*)Bitmap_copyAshmem },
1382     {   "nativeCopyAshmemConfig",   "(JI)Landroid/graphics/Bitmap;",
1383         (void*)Bitmap_copyAshmemConfig },
1384     {   "nativeGetNativeFinalizer", "()J", (void*)Bitmap_getNativeFinalizer },
1385     {   "nativeRecycle",            "(J)Z", (void*)Bitmap_recycle },
1386     {   "nativeReconfigure",        "(JIIIIZ)V", (void*)Bitmap_reconfigure },
1387     {   "nativeCompress",           "(JIILjava/io/OutputStream;[B)Z",
1388         (void*)Bitmap_compress },
1389     {   "nativeErase",              "(JI)V", (void*)Bitmap_erase },
1390     {   "nativeRowBytes",           "(J)I", (void*)Bitmap_rowBytes },
1391     {   "nativeConfig",             "(J)I", (void*)Bitmap_config },
1392     {   "nativeHasAlpha",           "(J)Z", (void*)Bitmap_hasAlpha },
1393     {   "nativeIsPremultiplied",    "(J)Z", (void*)Bitmap_isPremultiplied},
1394     {   "nativeSetHasAlpha",        "(JZZ)V", (void*)Bitmap_setHasAlpha},
1395     {   "nativeSetPremultiplied",   "(JZ)V", (void*)Bitmap_setPremultiplied},
1396     {   "nativeHasMipMap",          "(J)Z", (void*)Bitmap_hasMipMap },
1397     {   "nativeSetHasMipMap",       "(JZ)V", (void*)Bitmap_setHasMipMap },
1398     {   "nativeCreateFromParcel",
1399         "(Landroid/os/Parcel;)Landroid/graphics/Bitmap;",
1400         (void*)Bitmap_createFromParcel },
1401     {   "nativeWriteToParcel",      "(JZILandroid/os/Parcel;)Z",
1402         (void*)Bitmap_writeToParcel },
1403     {   "nativeExtractAlpha",       "(JJ[I)Landroid/graphics/Bitmap;",
1404         (void*)Bitmap_extractAlpha },
1405     {   "nativeGenerationId",       "(J)I", (void*)Bitmap_getGenerationId },
1406     {   "nativeGetPixel",           "(JII)I", (void*)Bitmap_getPixel },
1407     {   "nativeGetPixels",          "(J[IIIIIII)V", (void*)Bitmap_getPixels },
1408     {   "nativeSetPixel",           "(JIII)V", (void*)Bitmap_setPixel },
1409     {   "nativeSetPixels",          "(J[IIIIIII)V", (void*)Bitmap_setPixels },
1410     {   "nativeCopyPixelsToBuffer", "(JLjava/nio/Buffer;)V",
1411                                             (void*)Bitmap_copyPixelsToBuffer },
1412     {   "nativeCopyPixelsFromBuffer", "(JLjava/nio/Buffer;)V",
1413                                             (void*)Bitmap_copyPixelsFromBuffer },
1414     {   "nativeSameAs",             "(JJ)Z", (void*)Bitmap_sameAs },
1415     {   "nativeRefPixelRef",        "(J)J", (void*)Bitmap_refPixelRef },
1416     {   "nativePrepareToDraw",      "(J)V", (void*)Bitmap_prepareToDraw },
1417 };
1418 
register_android_graphics_Bitmap(JNIEnv * env)1419 int register_android_graphics_Bitmap(JNIEnv* env)
1420 {
1421     return android::RegisterMethodsOrDie(env, "android/graphics/Bitmap", gBitmapMethods,
1422                                          NELEM(gBitmapMethods));
1423 }
1424