| OLD | NEW |
| (Empty) |
| 1 | |
| 2 /* | |
| 3 * Copyright 2011 Google Inc. | |
| 4 * | |
| 5 * Use of this source code is governed by a BSD-style license that can be | |
| 6 * found in the LICENSE file. | |
| 7 */ | |
| 8 #include "SkImageRef.h" | |
| 9 #include "SkBitmap.h" | |
| 10 #include "SkReadBuffer.h" | |
| 11 #include "SkWriteBuffer.h" | |
| 12 #include "SkImageDecoder.h" | |
| 13 #include "SkStream.h" | |
| 14 #include "SkTemplates.h" | |
| 15 #include "SkThread.h" | |
| 16 | |
| 17 //#define DUMP_IMAGEREF_LIFECYCLE | |
| 18 | |
| 19 /////////////////////////////////////////////////////////////////////////////// | |
| 20 | |
| 21 SkImageRef::SkImageRef(const SkImageInfo& info, SkStreamRewindable* stream, | |
| 22 int sampleSize, SkBaseMutex* mutex) | |
| 23 : INHERITED(info, mutex), fErrorInDecoding(false) | |
| 24 { | |
| 25 SkASSERT(stream); | |
| 26 stream->ref(); | |
| 27 fStream = stream; | |
| 28 fSampleSize = sampleSize; | |
| 29 fDoDither = true; | |
| 30 fPrev = fNext = NULL; | |
| 31 fFactory = NULL; | |
| 32 | |
| 33 // This sets the colortype/alphatype to exactly match our info, so that this | |
| 34 // can get communicated down to the codec. | |
| 35 fBitmap.setInfo(info); | |
| 36 | |
| 37 #ifdef DUMP_IMAGEREF_LIFECYCLE | |
| 38 SkDebugf("add ImageRef %p [%d] data=%d\n", | |
| 39 this, this->info().fColorType, (int)stream->getLength()); | |
| 40 #endif | |
| 41 } | |
| 42 | |
| 43 SkImageRef::~SkImageRef() { | |
| 44 | |
| 45 #ifdef DUMP_IMAGEREF_LIFECYCLE | |
| 46 SkDebugf("delete ImageRef %p [%d] data=%d\n", | |
| 47 this, this->info().fColorType, (int)fStream->getLength()); | |
| 48 #endif | |
| 49 | |
| 50 fStream->unref(); | |
| 51 SkSafeUnref(fFactory); | |
| 52 } | |
| 53 | |
| 54 bool SkImageRef::getInfo(SkBitmap* bitmap) { | |
| 55 SkAutoMutexAcquire ac(this->mutex()); | |
| 56 | |
| 57 if (!this->prepareBitmap(SkImageDecoder::kDecodeBounds_Mode)) { | |
| 58 return false; | |
| 59 } | |
| 60 | |
| 61 SkASSERT(SkBitmap::kNo_Config != fBitmap.config()); | |
| 62 if (bitmap) { | |
| 63 bitmap->setConfig(fBitmap.config(), fBitmap.width(), fBitmap.height()); | |
| 64 } | |
| 65 return true; | |
| 66 } | |
| 67 | |
| 68 bool SkImageRef::isOpaque(SkBitmap* bitmap) { | |
| 69 if (bitmap && bitmap->pixelRef() == this) { | |
| 70 bitmap->lockPixels(); | |
| 71 // what about colortables?????? | |
| 72 bitmap->setAlphaType(fBitmap.alphaType()); | |
| 73 bitmap->unlockPixels(); | |
| 74 return true; | |
| 75 } | |
| 76 return false; | |
| 77 } | |
| 78 | |
| 79 SkImageDecoderFactory* SkImageRef::setDecoderFactory( | |
| 80 SkImageDecoderFactory* fact) { | |
| 81 SkRefCnt_SafeAssign(fFactory, fact); | |
| 82 return fact; | |
| 83 } | |
| 84 | |
| 85 /////////////////////////////////////////////////////////////////////////////// | |
| 86 | |
| 87 bool SkImageRef::onDecode(SkImageDecoder* codec, SkStreamRewindable* stream, | |
| 88 SkBitmap* bitmap, SkBitmap::Config config, | |
| 89 SkImageDecoder::Mode mode) { | |
| 90 return codec->decode(stream, bitmap, config, mode); | |
| 91 } | |
| 92 | |
| 93 bool SkImageRef::prepareBitmap(SkImageDecoder::Mode mode) { | |
| 94 | |
| 95 if (fErrorInDecoding) { | |
| 96 return false; | |
| 97 } | |
| 98 | |
| 99 if (NULL != fBitmap.getPixels() || | |
| 100 (SkBitmap::kNo_Config != fBitmap.config() && | |
| 101 SkImageDecoder::kDecodeBounds_Mode == mode)) { | |
| 102 return true; | |
| 103 } | |
| 104 | |
| 105 SkASSERT(fBitmap.getPixels() == NULL); | |
| 106 | |
| 107 if (!fStream->rewind()) { | |
| 108 SkDEBUGF(("Failed to rewind SkImageRef stream!")); | |
| 109 return false; | |
| 110 } | |
| 111 | |
| 112 SkImageDecoder* codec; | |
| 113 if (fFactory) { | |
| 114 codec = fFactory->newDecoder(fStream); | |
| 115 } else { | |
| 116 codec = SkImageDecoder::Factory(fStream); | |
| 117 } | |
| 118 | |
| 119 if (codec) { | |
| 120 SkAutoTDelete<SkImageDecoder> ad(codec); | |
| 121 | |
| 122 codec->setSampleSize(fSampleSize); | |
| 123 codec->setDitherImage(fDoDither); | |
| 124 codec->setRequireUnpremultipliedColors(this->info().fAlphaType == kUnpre
mul_SkAlphaType); | |
| 125 if (this->onDecode(codec, fStream, &fBitmap, fBitmap.config(), mode)) { | |
| 126 if (kOpaque_SkAlphaType == fBitmap.alphaType()) { | |
| 127 this->changeAlphaType(kOpaque_SkAlphaType); | |
| 128 } | |
| 129 SkASSERT(this->info() == fBitmap.info()); | |
| 130 return true; | |
| 131 } | |
| 132 } | |
| 133 | |
| 134 #ifdef DUMP_IMAGEREF_LIFECYCLE | |
| 135 if (NULL == codec) { | |
| 136 SkDebugf("--- ImageRef: <%s> failed to find codec\n", this->getURI()); | |
| 137 } else { | |
| 138 SkDebugf("--- ImageRef: <%s> failed in codec for %d mode\n", | |
| 139 this->getURI(), mode); | |
| 140 } | |
| 141 #endif | |
| 142 fErrorInDecoding = true; | |
| 143 fBitmap.reset(); | |
| 144 return false; | |
| 145 } | |
| 146 | |
| 147 bool SkImageRef::onNewLockPixels(LockRec* rec) { | |
| 148 if (NULL == fBitmap.getPixels()) { | |
| 149 (void)this->prepareBitmap(SkImageDecoder::kDecodePixels_Mode); | |
| 150 } | |
| 151 | |
| 152 if (NULL == fBitmap.getPixels()) { | |
| 153 return false; | |
| 154 } | |
| 155 rec->fPixels = fBitmap.getPixels(); | |
| 156 rec->fColorTable = NULL; | |
| 157 rec->fRowBytes = fBitmap.rowBytes(); | |
| 158 return true; | |
| 159 } | |
| 160 | |
| 161 size_t SkImageRef::ramUsed() const { | |
| 162 size_t size = 0; | |
| 163 | |
| 164 if (fBitmap.getPixels()) { | |
| 165 size = fBitmap.getSize(); | |
| 166 if (fBitmap.getColorTable()) { | |
| 167 size += fBitmap.getColorTable()->count() * sizeof(SkPMColor); | |
| 168 } | |
| 169 } | |
| 170 return size; | |
| 171 } | |
| 172 | |
| 173 /////////////////////////////////////////////////////////////////////////////// | |
| 174 | |
| 175 SkImageRef::SkImageRef(SkReadBuffer& buffer, SkBaseMutex* mutex) | |
| 176 : INHERITED(buffer, mutex), fErrorInDecoding(false) { | |
| 177 fSampleSize = buffer.readInt(); | |
| 178 fDoDither = buffer.readBool(); | |
| 179 | |
| 180 size_t length = buffer.getArrayCount(); | |
| 181 if (buffer.validateAvailable(length)) { | |
| 182 fStream = SkNEW_ARGS(SkMemoryStream, (length)); | |
| 183 buffer.readByteArray((void*)fStream->getMemoryBase(), length); | |
| 184 } else { | |
| 185 fStream = NULL; | |
| 186 } | |
| 187 | |
| 188 fPrev = fNext = NULL; | |
| 189 fFactory = NULL; | |
| 190 | |
| 191 // This sets the colortype/alphatype to exactly match our info, so that this | |
| 192 // can get communicated down to the codec. | |
| 193 fBitmap.setInfo(this->info()); | |
| 194 } | |
| 195 | |
| 196 void SkImageRef::flatten(SkWriteBuffer& buffer) const { | |
| 197 this->INHERITED::flatten(buffer); | |
| 198 | |
| 199 buffer.writeInt(fSampleSize); | |
| 200 buffer.writeBool(fDoDither); | |
| 201 // FIXME: Consider moving this logic should go into writeStream itself. | |
| 202 // writeStream currently has no other callers, so this may be fine for | |
| 203 // now. | |
| 204 if (!fStream->rewind()) { | |
| 205 SkDEBUGF(("Failed to rewind SkImageRef stream!")); | |
| 206 buffer.write32(0); | |
| 207 } else { | |
| 208 // FIXME: Handle getLength properly here. Perhaps this class should | |
| 209 // take an SkStreamAsset. | |
| 210 buffer.writeStream(fStream, fStream->getLength()); | |
| 211 } | |
| 212 } | |
| OLD | NEW |