| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2013 Google Inc. | 2 * Copyright 2013 Google Inc. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license that can be | 4 * Use of this source code is governed by a BSD-style license that can be |
| 5 * found in the LICENSE file. | 5 * found in the LICENSE file. |
| 6 */ | 6 */ |
| 7 | 7 |
| 8 #include "SkDiscardablePixelRef.h" | 8 #include "SkDiscardablePixelRef.h" |
| 9 #include "SkDiscardableMemory.h" | 9 #include "SkDiscardableMemory.h" |
| 10 #include "SkImageGenerator.h" | 10 #include "SkImageGenerator.h" |
| 11 | 11 |
| 12 SkDiscardablePixelRef::SkDiscardablePixelRef(const SkImageInfo& info, | 12 SkDiscardablePixelRef::SkDiscardablePixelRef(const SkImageInfo& info, |
| 13 SkImageGenerator* generator, | 13 SkImageGenerator* generator, |
| 14 size_t rowBytes, | 14 size_t rowBytes, |
| 15 SkDiscardableMemory::Factory* fact) | 15 SkDiscardableMemory::Factory* fact) |
| 16 : INHERITED(info) | 16 : INHERITED(info) |
| 17 , fGenerator(generator) | 17 , fGenerator(generator) |
| 18 , fDMFactory(fact) | 18 , fDMFactory(fact) |
| 19 , fRowBytes(rowBytes) | 19 , fRowBytes(rowBytes) |
| 20 , fDiscardableMemory(NULL) | 20 , fDiscardableMemory(NULL) |
| 21 , fDiscardableMemoryIsLocked(false) | |
| 22 { | 21 { |
| 23 SkASSERT(fGenerator != NULL); | 22 SkASSERT(fGenerator != NULL); |
| 24 SkASSERT(fRowBytes > 0); | 23 SkASSERT(fRowBytes > 0); |
| 25 // The SkImageGenerator contract requires fGenerator to always | 24 // The SkImageGenerator contract requires fGenerator to always |
| 26 // decode the same image on each call to getPixels(). | 25 // decode the same image on each call to getPixels(). |
| 27 this->setImmutable(); | 26 this->setImmutable(); |
| 28 SkSafeRef(fDMFactory); | 27 SkSafeRef(fDMFactory); |
| 29 } | 28 } |
| 30 | 29 |
| 31 SkDiscardablePixelRef::~SkDiscardablePixelRef() { | 30 SkDiscardablePixelRef::~SkDiscardablePixelRef() { |
| 32 if (fDiscardableMemoryIsLocked) { | 31 if (this->isLocked()) { |
| 33 fDiscardableMemory->unlock(); | 32 fDiscardableMemory->unlock(); |
| 34 fDiscardableMemoryIsLocked = false; | |
| 35 } | 33 } |
| 36 SkDELETE(fDiscardableMemory); | 34 SkDELETE(fDiscardableMemory); |
| 37 SkSafeUnref(fDMFactory); | 35 SkSafeUnref(fDMFactory); |
| 38 SkDELETE(fGenerator); | 36 SkDELETE(fGenerator); |
| 39 } | 37 } |
| 40 | 38 |
| 41 bool SkDiscardablePixelRef::onNewLockPixels(LockRec* rec) { | 39 bool SkDiscardablePixelRef::onNewLockPixels(LockRec* rec) { |
| 42 if (fDiscardableMemory != NULL) { | 40 if (fDiscardableMemory != NULL) { |
| 43 if (fDiscardableMemory->lock()) { | 41 if (fDiscardableMemory->lock()) { |
| 44 fDiscardableMemoryIsLocked = true; | |
| 45 rec->fPixels = fDiscardableMemory->data(); | 42 rec->fPixels = fDiscardableMemory->data(); |
| 46 rec->fColorTable = fCTable.get(); | 43 rec->fColorTable = fCTable.get(); |
| 47 rec->fRowBytes = fRowBytes; | 44 rec->fRowBytes = fRowBytes; |
| 48 return true; | 45 return true; |
| 49 } | 46 } |
| 50 SkDELETE(fDiscardableMemory); | 47 SkDELETE(fDiscardableMemory); |
| 51 fDiscardableMemory = NULL; | 48 fDiscardableMemory = NULL; |
| 52 fDiscardableMemoryIsLocked = false; | |
| 53 } | 49 } |
| 54 | 50 |
| 55 const size_t size = this->info().getSafeSize(fRowBytes); | 51 const size_t size = this->info().getSafeSize(fRowBytes); |
| 56 | 52 |
| 57 if (fDMFactory != NULL) { | 53 if (fDMFactory != NULL) { |
| 58 fDiscardableMemory = fDMFactory->create(size); | 54 fDiscardableMemory = fDMFactory->create(size); |
| 59 fDiscardableMemoryIsLocked = true; | |
| 60 } else { | 55 } else { |
| 61 fDiscardableMemory = SkDiscardableMemory::Create(size); | 56 fDiscardableMemory = SkDiscardableMemory::Create(size); |
| 62 fDiscardableMemoryIsLocked = true; | |
| 63 } | 57 } |
| 64 if (NULL == fDiscardableMemory) { | 58 if (NULL == fDiscardableMemory) { |
| 65 fDiscardableMemoryIsLocked = false; | |
| 66 return false; // Memory allocation failed. | 59 return false; // Memory allocation failed. |
| 67 } | 60 } |
| 68 | 61 |
| 69 void* pixels = fDiscardableMemory->data(); | 62 void* pixels = fDiscardableMemory->data(); |
| 70 const SkImageInfo& info = this->info(); | 63 const SkImageInfo& info = this->info(); |
| 71 SkPMColor colors[256]; | 64 SkPMColor colors[256]; |
| 72 int colorCount = 0; | 65 int colorCount = 0; |
| 73 | 66 |
| 74 const SkImageGenerator::Result result = fGenerator->getPixels(info, pixels,
fRowBytes, | 67 const SkImageGenerator::Result result = fGenerator->getPixels(info, pixels,
fRowBytes, |
| 75 colors, &color
Count); | 68 colors, &color
Count); |
| 76 switch (result) { | 69 switch (result) { |
| 77 case SkImageGenerator::kSuccess: | 70 case SkImageGenerator::kSuccess: |
| 78 case SkImageGenerator::kIncompleteInput: | 71 case SkImageGenerator::kIncompleteInput: |
| 79 break; | 72 break; |
| 80 default: | 73 default: |
| 81 fDiscardableMemory->unlock(); | 74 fDiscardableMemory->unlock(); |
| 82 fDiscardableMemoryIsLocked = false; | |
| 83 SkDELETE(fDiscardableMemory); | 75 SkDELETE(fDiscardableMemory); |
| 84 fDiscardableMemory = NULL; | 76 fDiscardableMemory = NULL; |
| 85 return false; | 77 return false; |
| 86 } | 78 } |
| 87 | 79 |
| 88 // Note: our ctable is not purgeable, as it is not stored in the discardable
memory block. | 80 // Note: our ctable is not purgeable, as it is not stored in the discardable
memory block. |
| 89 // This is because SkColorTable is refcntable, and therefore our caller coul
d hold onto it | 81 // This is because SkColorTable is refcntable, and therefore our caller coul
d hold onto it |
| 90 // beyond the scope of a lock/unlock. If we change the API/lifecycle for SkC
olorTable, we | 82 // beyond the scope of a lock/unlock. If we change the API/lifecycle for SkC
olorTable, we |
| 91 // could move it into the block, but then again perhaps it is small enough t
hat this doesn't | 83 // could move it into the block, but then again perhaps it is small enough t
hat this doesn't |
| 92 // really matter. | 84 // really matter. |
| 93 if (colorCount > 0) { | 85 if (colorCount > 0) { |
| 94 fCTable.reset(SkNEW_ARGS(SkColorTable, (colors, colorCount))); | 86 fCTable.reset(SkNEW_ARGS(SkColorTable, (colors, colorCount))); |
| 95 } else { | 87 } else { |
| 96 fCTable.reset(NULL); | 88 fCTable.reset(NULL); |
| 97 } | 89 } |
| 98 | 90 |
| 99 rec->fPixels = pixels; | 91 rec->fPixels = pixels; |
| 100 rec->fColorTable = fCTable.get(); | 92 rec->fColorTable = fCTable.get(); |
| 101 rec->fRowBytes = fRowBytes; | 93 rec->fRowBytes = fRowBytes; |
| 102 return true; | 94 return true; |
| 103 } | 95 } |
| 104 | 96 |
| 105 void SkDiscardablePixelRef::onUnlockPixels() { | 97 void SkDiscardablePixelRef::onUnlockPixels() { |
| 106 fDiscardableMemory->unlock(); | 98 fDiscardableMemory->unlock(); |
| 107 fDiscardableMemoryIsLocked = false; | |
| 108 } | 99 } |
| 109 | 100 |
| 110 bool SkInstallDiscardablePixelRef(SkImageGenerator* generator, SkBitmap* dst, | 101 bool SkInstallDiscardablePixelRef(SkImageGenerator* generator, SkBitmap* dst, |
| 111 SkDiscardableMemory::Factory* factory) { | 102 SkDiscardableMemory::Factory* factory) { |
| 112 SkImageInfo info; | 103 SkImageInfo info; |
| 113 SkAutoTDelete<SkImageGenerator> autoGenerator(generator); | 104 SkAutoTDelete<SkImageGenerator> autoGenerator(generator); |
| 114 if ((NULL == autoGenerator.get()) | 105 if ((NULL == autoGenerator.get()) |
| 115 || (!autoGenerator->getInfo(&info)) | 106 || (!autoGenerator->getInfo(&info)) |
| 116 || info.isEmpty() | 107 || info.isEmpty() |
| 117 || (!dst->setInfo(info))) { | 108 || (!dst->setInfo(info))) { |
| (...skipping 16 matching lines...) Expand all Loading... |
| 134 // These are the public API | 125 // These are the public API |
| 135 | 126 |
| 136 bool SkInstallDiscardablePixelRef(SkImageGenerator* generator, SkBitmap* dst) { | 127 bool SkInstallDiscardablePixelRef(SkImageGenerator* generator, SkBitmap* dst) { |
| 137 return SkInstallDiscardablePixelRef(generator, dst, NULL); | 128 return SkInstallDiscardablePixelRef(generator, dst, NULL); |
| 138 } | 129 } |
| 139 | 130 |
| 140 bool SkInstallDiscardablePixelRef(SkData* encoded, SkBitmap* dst) { | 131 bool SkInstallDiscardablePixelRef(SkData* encoded, SkBitmap* dst) { |
| 141 SkImageGenerator* generator = SkImageGenerator::NewFromData(encoded); | 132 SkImageGenerator* generator = SkImageGenerator::NewFromData(encoded); |
| 142 return generator ? SkInstallDiscardablePixelRef(generator, dst, NULL) : fals
e; | 133 return generator ? SkInstallDiscardablePixelRef(generator, dst, NULL) : fals
e; |
| 143 } | 134 } |
| OLD | NEW |