src/third_party/skia/src/images/SkImageRef.cpp

   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 #ifdef DUMP_IMAGEREF_LIFECYCLE
  34     SkDebugf("add ImageRef %p [%d] data=%d\n",
  35               this, this->info().fColorType, (int)stream->getLength());
  36 #endif
  37 }
  38
  39 SkImageRef::~SkImageRef() {
  40
  41 #ifdef DUMP_IMAGEREF_LIFECYCLE
  42     SkDebugf("delete ImageRef %p [%d] data=%d\n",
  43               this, this->info().fColorType, (int)fStream->getLength());
  44 #endif
  45
  46     fStream->unref();
  47     SkSafeUnref(fFactory);
  48 }
  49
  50 bool SkImageRef::getInfo(SkBitmap* bitmap) {
  51     SkAutoMutexAcquire ac(this->mutex());
  52
  53     if (!this->prepareBitmap(SkImageDecoder::kDecodeBounds_Mode)) {
  54         return false;
  55     }
  56
  57     SkASSERT(SkBitmap::kNo_Config != fBitmap.config());
  58     if (bitmap) {
  59         bitmap->setConfig(fBitmap.config(), fBitmap.width(), fBitmap.height());
  60     }
  61     return true;
  62 }
  63
  64 bool SkImageRef::isOpaque(SkBitmap* bitmap) {
  65     if (bitmap && bitmap->pixelRef() == this) {
  66         bitmap->lockPixels();
  67         // what about colortables??????
  68         bitmap->setAlphaType(fBitmap.alphaType());
  69         bitmap->unlockPixels();
  70         return true;
  71     }
  72     return false;
  73 }
  74
  75 SkImageDecoderFactory* SkImageRef::setDecoderFactory(
  76                                                 SkImageDecoderFactory* fact) {
  77     SkRefCnt_SafeAssign(fFactory, fact);
  78     return fact;
  79 }
  80
  81 ///////////////////////////////////////////////////////////////////////////////
  82
  83 bool SkImageRef::onDecode(SkImageDecoder* codec, SkStreamRewindable* stream,
  84                           SkBitmap* bitmap, SkBitmap::Config config,
  85                           SkImageDecoder::Mode mode) {
  86     return codec->decode(stream, bitmap, config, mode);
  87 }
  88
  89 bool SkImageRef::prepareBitmap(SkImageDecoder::Mode mode) {
  90
  91     if (fErrorInDecoding) {
  92         return false;
  93     }
  94
  95     if (NULL != fBitmap.getPixels() ||
  96             (SkBitmap::kNo_Config != fBitmap.config() &&
  97              SkImageDecoder::kDecodeBounds_Mode == mode)) {
  98         return true;
  99     }
 100
 101     SkASSERT(fBitmap.getPixels() == NULL);
 102
 103     if (!fStream->rewind()) {
 104         SkDEBUGF(("Failed to rewind SkImageRef stream!"));
 105         return false;
 106     }
 107
 108     SkImageDecoder* codec;
 109     if (fFactory) {
 110         codec = fFactory->newDecoder(fStream);
 111     } else {
 112         codec = SkImageDecoder::Factory(fStream);
 113     }
 114
 115     if (codec) {
 116         SkAutoTDelete<SkImageDecoder> ad(codec);
 117
 118         codec->setSampleSize(fSampleSize);
 119         codec->setDitherImage(fDoDither);
 120         if (this->onDecode(codec, fStream, &fBitmap, fBitmap.config(), mode)) {
 121             return true;
 122         }
 123     }
 124
 125 #ifdef DUMP_IMAGEREF_LIFECYCLE
 126     if (NULL == codec) {
 127         SkDebugf("--- ImageRef: <%s> failed to find codec\n", this->getURI());
 128     } else {
 129         SkDebugf("--- ImageRef: <%s> failed in codec for %d mode\n",
 130                  this->getURI(), mode);
 131     }
 132 #endif
 133     fErrorInDecoding = true;
 134     fBitmap.reset();
 135     return false;
 136 }
 137
 138 bool SkImageRef::onNewLockPixels(LockRec* rec) {
 139     if (NULL == fBitmap.getPixels()) {
 140         (void)this->prepareBitmap(SkImageDecoder::kDecodePixels_Mode);
 141     }
 142
 143     if (NULL == fBitmap.getPixels()) {
 144         return false;
 145     }
 146     rec->fPixels = fBitmap.getPixels();
 147     rec->fColorTable = NULL;
 148     rec->fRowBytes = fBitmap.rowBytes();
 149     return true;
 150 }
 151
 152 size_t SkImageRef::ramUsed() const {
 153     size_t size = 0;
 154
 155     if (fBitmap.getPixels()) {
 156         size = fBitmap.getSize();
 157         if (fBitmap.getColorTable()) {
 158             size += fBitmap.getColorTable()->count() * sizeof(SkPMColor);
 159         }
 160     }
 161     return size;
 162 }
 163
 164 ///////////////////////////////////////////////////////////////////////////////
 165
 166 SkImageRef::SkImageRef(SkReadBuffer& buffer, SkBaseMutex* mutex)
 167         : INHERITED(buffer, mutex), fErrorInDecoding(false) {
 168     fSampleSize = buffer.readInt();
 169     fDoDither = buffer.readBool();
 170
 171     size_t length = buffer.getArrayCount();
 172     if (buffer.validateAvailable(length)) {
 173         fStream = SkNEW_ARGS(SkMemoryStream, (length));
 174         buffer.readByteArray((void*)fStream->getMemoryBase(), length);
 175     } else {
 176         fStream = NULL;
 177     }
 178
 179     fPrev = fNext = NULL;
 180     fFactory = NULL;
 181 }
 182
 183 void SkImageRef::flatten(SkWriteBuffer& buffer) const {
 184     this->INHERITED::flatten(buffer);
 185
 186     buffer.writeInt(fSampleSize);
 187     buffer.writeBool(fDoDither);
 188     // FIXME: Consider moving this logic should go into writeStream itself.
 189     // writeStream currently has no other callers, so this may be fine for
 190     // now.
 191     if (!fStream->rewind()) {
 192         SkDEBUGF(("Failed to rewind SkImageRef stream!"));
 193         buffer.write32(0);
 194     } else {
 195         // FIXME: Handle getLength properly here. Perhaps this class should
 196         // take an SkStreamAsset.
 197         buffer.writeStream(fStream, fStream->getLength());
 198     }
 199 }