1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/gfx/skia/trunk/src/images/SkImageRef.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,212 @@
1.4 +
1.5 +/*
1.6 + * Copyright 2011 Google Inc.
1.7 + *
1.8 + * Use of this source code is governed by a BSD-style license that can be
1.9 + * found in the LICENSE file.
1.10 + */
1.11 +#include "SkImageRef.h"
1.12 +#include "SkBitmap.h"
1.13 +#include "SkReadBuffer.h"
1.14 +#include "SkWriteBuffer.h"
1.15 +#include "SkImageDecoder.h"
1.16 +#include "SkStream.h"
1.17 +#include "SkTemplates.h"
1.18 +#include "SkThread.h"
1.19 +
1.20 +//#define DUMP_IMAGEREF_LIFECYCLE
1.21 +
1.22 +///////////////////////////////////////////////////////////////////////////////
1.23 +
1.24 +SkImageRef::SkImageRef(const SkImageInfo& info, SkStreamRewindable* stream,
1.25 + int sampleSize, SkBaseMutex* mutex)
1.26 + : INHERITED(info, mutex), fErrorInDecoding(false)
1.27 +{
1.28 + SkASSERT(stream);
1.29 + stream->ref();
1.30 + fStream = stream;
1.31 + fSampleSize = sampleSize;
1.32 + fDoDither = true;
1.33 + fPrev = fNext = NULL;
1.34 + fFactory = NULL;
1.35 +
1.36 + // This sets the colortype/alphatype to exactly match our info, so that this
1.37 + // can get communicated down to the codec.
1.38 + fBitmap.setConfig(info);
1.39 +
1.40 +#ifdef DUMP_IMAGEREF_LIFECYCLE
1.41 + SkDebugf("add ImageRef %p [%d] data=%d\n",
1.42 + this, this->info().fColorType, (int)stream->getLength());
1.43 +#endif
1.44 +}
1.45 +
1.46 +SkImageRef::~SkImageRef() {
1.47 +
1.48 +#ifdef DUMP_IMAGEREF_LIFECYCLE
1.49 + SkDebugf("delete ImageRef %p [%d] data=%d\n",
1.50 + this, this->info().fColorType, (int)fStream->getLength());
1.51 +#endif
1.52 +
1.53 + fStream->unref();
1.54 + SkSafeUnref(fFactory);
1.55 +}
1.56 +
1.57 +bool SkImageRef::getInfo(SkBitmap* bitmap) {
1.58 + SkAutoMutexAcquire ac(this->mutex());
1.59 +
1.60 + if (!this->prepareBitmap(SkImageDecoder::kDecodeBounds_Mode)) {
1.61 + return false;
1.62 + }
1.63 +
1.64 + SkASSERT(SkBitmap::kNo_Config != fBitmap.config());
1.65 + if (bitmap) {
1.66 + bitmap->setConfig(fBitmap.config(), fBitmap.width(), fBitmap.height());
1.67 + }
1.68 + return true;
1.69 +}
1.70 +
1.71 +bool SkImageRef::isOpaque(SkBitmap* bitmap) {
1.72 + if (bitmap && bitmap->pixelRef() == this) {
1.73 + bitmap->lockPixels();
1.74 + // what about colortables??????
1.75 + bitmap->setAlphaType(fBitmap.alphaType());
1.76 + bitmap->unlockPixels();
1.77 + return true;
1.78 + }
1.79 + return false;
1.80 +}
1.81 +
1.82 +SkImageDecoderFactory* SkImageRef::setDecoderFactory(
1.83 + SkImageDecoderFactory* fact) {
1.84 + SkRefCnt_SafeAssign(fFactory, fact);
1.85 + return fact;
1.86 +}
1.87 +
1.88 +///////////////////////////////////////////////////////////////////////////////
1.89 +
1.90 +bool SkImageRef::onDecode(SkImageDecoder* codec, SkStreamRewindable* stream,
1.91 + SkBitmap* bitmap, SkBitmap::Config config,
1.92 + SkImageDecoder::Mode mode) {
1.93 + return codec->decode(stream, bitmap, config, mode);
1.94 +}
1.95 +
1.96 +bool SkImageRef::prepareBitmap(SkImageDecoder::Mode mode) {
1.97 +
1.98 + if (fErrorInDecoding) {
1.99 + return false;
1.100 + }
1.101 +
1.102 + if (NULL != fBitmap.getPixels() ||
1.103 + (SkBitmap::kNo_Config != fBitmap.config() &&
1.104 + SkImageDecoder::kDecodeBounds_Mode == mode)) {
1.105 + return true;
1.106 + }
1.107 +
1.108 + SkASSERT(fBitmap.getPixels() == NULL);
1.109 +
1.110 + if (!fStream->rewind()) {
1.111 + SkDEBUGF(("Failed to rewind SkImageRef stream!"));
1.112 + return false;
1.113 + }
1.114 +
1.115 + SkImageDecoder* codec;
1.116 + if (fFactory) {
1.117 + codec = fFactory->newDecoder(fStream);
1.118 + } else {
1.119 + codec = SkImageDecoder::Factory(fStream);
1.120 + }
1.121 +
1.122 + if (codec) {
1.123 + SkAutoTDelete<SkImageDecoder> ad(codec);
1.124 +
1.125 + codec->setSampleSize(fSampleSize);
1.126 + codec->setDitherImage(fDoDither);
1.127 + codec->setRequireUnpremultipliedColors(this->info().fAlphaType == kUnpremul_SkAlphaType);
1.128 + if (this->onDecode(codec, fStream, &fBitmap, fBitmap.config(), mode)) {
1.129 + if (kOpaque_SkAlphaType == fBitmap.alphaType()) {
1.130 + this->changeAlphaType(kOpaque_SkAlphaType);
1.131 + }
1.132 + SkASSERT(this->info() == fBitmap.info());
1.133 + return true;
1.134 + }
1.135 + }
1.136 +
1.137 +#ifdef DUMP_IMAGEREF_LIFECYCLE
1.138 + if (NULL == codec) {
1.139 + SkDebugf("--- ImageRef: <%s> failed to find codec\n", this->getURI());
1.140 + } else {
1.141 + SkDebugf("--- ImageRef: <%s> failed in codec for %d mode\n",
1.142 + this->getURI(), mode);
1.143 + }
1.144 +#endif
1.145 + fErrorInDecoding = true;
1.146 + fBitmap.reset();
1.147 + return false;
1.148 +}
1.149 +
1.150 +bool SkImageRef::onNewLockPixels(LockRec* rec) {
1.151 + if (NULL == fBitmap.getPixels()) {
1.152 + (void)this->prepareBitmap(SkImageDecoder::kDecodePixels_Mode);
1.153 + }
1.154 +
1.155 + if (NULL == fBitmap.getPixels()) {
1.156 + return false;
1.157 + }
1.158 + rec->fPixels = fBitmap.getPixels();
1.159 + rec->fColorTable = NULL;
1.160 + rec->fRowBytes = fBitmap.rowBytes();
1.161 + return true;
1.162 +}
1.163 +
1.164 +size_t SkImageRef::ramUsed() const {
1.165 + size_t size = 0;
1.166 +
1.167 + if (fBitmap.getPixels()) {
1.168 + size = fBitmap.getSize();
1.169 + if (fBitmap.getColorTable()) {
1.170 + size += fBitmap.getColorTable()->count() * sizeof(SkPMColor);
1.171 + }
1.172 + }
1.173 + return size;
1.174 +}
1.175 +
1.176 +///////////////////////////////////////////////////////////////////////////////
1.177 +
1.178 +SkImageRef::SkImageRef(SkReadBuffer& buffer, SkBaseMutex* mutex)
1.179 + : INHERITED(buffer, mutex), fErrorInDecoding(false) {
1.180 + fSampleSize = buffer.readInt();
1.181 + fDoDither = buffer.readBool();
1.182 +
1.183 + size_t length = buffer.getArrayCount();
1.184 + if (buffer.validateAvailable(length)) {
1.185 + fStream = SkNEW_ARGS(SkMemoryStream, (length));
1.186 + buffer.readByteArray((void*)fStream->getMemoryBase(), length);
1.187 + } else {
1.188 + fStream = NULL;
1.189 + }
1.190 +
1.191 + fPrev = fNext = NULL;
1.192 + fFactory = NULL;
1.193 +
1.194 + // This sets the colortype/alphatype to exactly match our info, so that this
1.195 + // can get communicated down to the codec.
1.196 + fBitmap.setConfig(this->info());
1.197 +}
1.198 +
1.199 +void SkImageRef::flatten(SkWriteBuffer& buffer) const {
1.200 + this->INHERITED::flatten(buffer);
1.201 +
1.202 + buffer.writeInt(fSampleSize);
1.203 + buffer.writeBool(fDoDither);
1.204 + // FIXME: Consider moving this logic should go into writeStream itself.
1.205 + // writeStream currently has no other callers, so this may be fine for
1.206 + // now.
1.207 + if (!fStream->rewind()) {
1.208 + SkDEBUGF(("Failed to rewind SkImageRef stream!"));
1.209 + buffer.write32(0);
1.210 + } else {
1.211 + // FIXME: Handle getLength properly here. Perhaps this class should
1.212 + // take an SkStreamAsset.
1.213 + buffer.writeStream(fStream, fStream->getLength());
1.214 + }
1.215 +}
