src/image/SkImage.cpp - skia - Git at Google

 /*
  * Copyright 2012 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */

 #include "SkBitmap.h"
 #include "SkCanvas.h"
 #include "SkData.h"
 #include "SkImageGenerator.h"
 #include "SkImagePriv.h"
 #include "SkImage_Base.h"
 #include "SkNextID.h"
 #include "SkPixelRef.h"
 #include "SkPixelSerializer.h"
 #include "SkReadPixelsRec.h"
 #include "SkString.h"
 #include "SkSurface.h"

 #if SK_SUPPORT_GPU
 #include "GrTexture.h"
 #include "GrContext.h"
 #include "SkImage_Gpu.h"
 #endif

 SkImage::SkImage(int width, int height, uint32_t uniqueID)
     : fWidth(width)
     , fHeight(height)
     , fUniqueID(kNeedNewImageUniqueID == uniqueID ? SkNextID::ImageID() : uniqueID)
 {
     SkASSERT(width > 0);
     SkASSERT(height > 0);
 }

 const void* SkImage::peekPixels(SkImageInfo* info, size_t* rowBytes) const {
     SkImageInfo infoStorage;
     size_t rowBytesStorage;
     if (nullptr == info) {
         info = &infoStorage;
     }
     if (nullptr == rowBytes) {
         rowBytes = &rowBytesStorage;
     }
     return as_IB(this)->onPeekPixels(info, rowBytes);
 }

 bool SkImage::readPixels(const SkImageInfo& dstInfo, void* dstPixels, size_t dstRowBytes,
                            int srcX, int srcY) const {
     SkReadPixelsRec rec(dstInfo, dstPixels, dstRowBytes, srcX, srcY);
     if (!rec.trim(this->width(), this->height())) {
         return false;
     }
     return as_IB(this)->onReadPixels(rec.fInfo, rec.fPixels, rec.fRowBytes, rec.fX, rec.fY);
 }

 void SkImage::preroll(GrContext* ctx) const {
     as_IB(this)->onPreroll(ctx);
 }

 SkShader* SkImage::newShader(SkShader::TileMode tileX,
                              SkShader::TileMode tileY,
                              const SkMatrix* localMatrix) const {
     return as_IB(this)->onNewShader(tileX, tileY, localMatrix);
 }

 SkData* SkImage::encode(SkImageEncoder::Type type, int quality) const {
     SkBitmap bm;
     if (as_IB(this)->getROPixels(&bm)) {
         return SkImageEncoder::EncodeData(bm, type, quality);
     }
     return nullptr;
 }

 namespace {

 class DefaultSerializer :  public SkPixelSerializer {
 protected:
     bool onUseEncodedData(const void *data, size_t len) override {
         return true;
     }

     SkData* onEncodePixels(const SkImageInfo& info, const void* pixels, size_t rowBytes) override {
         return SkImageEncoder::EncodeData(info, pixels, rowBytes, SkImageEncoder::kPNG_Type, 100);
     }
 };

 } // anonymous namespace

 SkData* SkImage::encode(SkPixelSerializer* serializer) const {
     DefaultSerializer defaultSerializer;
     SkPixelSerializer* effectiveSerializer = serializer ? serializer : &defaultSerializer;

     SkAutoTUnref<SkData> encoded(this->refEncoded());
     if (encoded && effectiveSerializer->useEncodedData(encoded->data(), encoded->size())) {
         return encoded.detach();
     }

     SkBitmap bm;
     SkAutoPixmapUnlock apu;
     if (as_IB(this)->getROPixels(&bm) && bm.requestLock(&apu)) {
         const SkPixmap& pmap = apu.pixmap();
         return effectiveSerializer->encodePixels(pmap.info(), pmap.addr(), pmap.rowBytes());
     }

     return nullptr;
 }

 SkData* SkImage::refEncoded() const {
     return as_IB(this)->onRefEncoded();
 }

 SkImage* SkImage::NewFromEncoded(SkData* encoded, const SkIRect* subset) {
     if (nullptr == encoded || 0 == encoded->size()) {
         return nullptr;
     }
     SkImageGenerator* generator = SkImageGenerator::NewFromEncoded(encoded);
     return generator ? SkImage::NewFromGenerator(generator, subset) : nullptr;
 }

 SkSurface* SkImage::newSurface(const SkImageInfo& info, const SkSurfaceProps* props) const {
     if (nullptr == props) {
         props = &as_IB(this)->props();
     }
     return as_IB(this)->onNewSurface(info, *props);
 }

 const char* SkImage::toString(SkString* str) const {
     str->appendf("image: (id:%d (%d, %d) %s)", this->uniqueID(), this->width(), this->height(),
                  this->isOpaque() ? "opaque" : "");
     return str->c_str();
 }

 SkImage* SkImage::newImage(int newWidth, int newHeight, const SkIRect* subset,
                            SkFilterQuality quality) const {
     if (newWidth <= 0 || newHeight <= 0) {
         return nullptr;
     }

     const SkIRect bounds = SkIRect::MakeWH(this->width(), this->height());

     if (subset) {
         if (!bounds.contains(*subset)) {
             return nullptr;
         }
         if (bounds == *subset) {
             subset = nullptr;  // and fall through to check below
         }
     }

     if (nullptr == subset && this->width() == newWidth && this->height() == newHeight) {
         return SkRef(const_cast<SkImage*>(this));
     }

     return as_IB(this)->onNewImage(newWidth, newHeight, subset, quality);
 }

 #if SK_SUPPORT_GPU

 GrTexture* SkImage::getTexture() const {
     return as_IB(this)->getTexture();
 }

 bool SkImage::isTextureBacked() const { return SkToBool(as_IB(this)->getTexture()); }

 GrBackendObject SkImage::getTextureHandle(bool flushPendingGrContextIO) const {
     GrTexture* texture = as_IB(this)->getTexture();
     if (texture) {
         GrContext* context = texture->getContext();
         if (context) {
             if (flushPendingGrContextIO) {
                 context->prepareSurfaceForExternalIO(texture);
             }
         }
         return texture->getTextureHandle();
     }
     return 0;
 }

 #else

 GrTexture* SkImage::getTexture() const { return nullptr; }

 bool SkImage::isTextureBacked() const { return false; }

 GrBackendObject SkImage::getTextureHandle(bool) const { return 0; }

 #endif

 ///////////////////////////////////////////////////////////////////////////////

 static bool raster_canvas_supports(const SkImageInfo& info) {
     switch (info.colorType()) {
         case kN32_SkColorType:
             return kUnpremul_SkAlphaType != info.alphaType();
         case kRGB_565_SkColorType:
             return true;
         case kAlpha_8_SkColorType:
             return true;
         default:
             break;
     }
     return false;
 }

 bool SkImage_Base::onReadPixels(const SkImageInfo& dstInfo, void* dstPixels, size_t dstRowBytes,
                                 int srcX, int srcY) const {
     if (!raster_canvas_supports(dstInfo)) {
         return false;
     }

     SkBitmap bm;
     bm.installPixels(dstInfo, dstPixels, dstRowBytes);
     SkCanvas canvas(bm);

     SkPaint paint;
     paint.setXfermodeMode(SkXfermode::kSrc_Mode);
     canvas.drawImage(this, -SkIntToScalar(srcX), -SkIntToScalar(srcY), &paint);

     return true;
 }

 SkImage* SkImage_Base::onNewImage(int newWidth, int newHeight, const SkIRect* subset,
                                   SkFilterQuality quality) const {
     const bool opaque = this->isOpaque();
     const SkImageInfo info = SkImageInfo::Make(newWidth, newHeight, kN32_SkColorType,
                                                opaque ? kOpaque_SkAlphaType : kPremul_SkAlphaType);
     SkAutoTUnref<SkSurface> surface(this->newSurface(info, nullptr));
     if (!surface.get()) {
         return nullptr;
     }

     SkRect src;
     if (subset) {
         src.set(*subset);
     } else {
         src = SkRect::MakeIWH(this->width(), this->height());
     }

     surface->getCanvas()->scale(newWidth / src.width(), newHeight / src.height());
     surface->getCanvas()->translate(-src.x(), -src.y());

     SkPaint paint;
     paint.setXfermodeMode(SkXfermode::kSrc_Mode);
     paint.setFilterQuality(quality);
     surface->getCanvas()->drawImage(this, 0, 0, &paint);
     return surface->newImageSnapshot();
 }

 ///////////////////////////////////////////////////////////////////////////////////////////////////

 bool SkImage::peekPixels(SkPixmap* pmap) const {
     SkImageInfo info;
     size_t rowBytes;
     const void* pixels = this->peekPixels(&info, &rowBytes);
     if (pixels) {
         if (pmap) {
             pmap->reset(info, pixels, rowBytes);
         }
         return true;
     }
     return false;
 }

 bool SkImage::readPixels(const SkPixmap& pmap, int srcX, int srcY) const {
     return this->readPixels(pmap.info(), pmap.writable_addr(), pmap.rowBytes(), srcX, srcY);
 }

 ///////////////////////////////////////////////////////////////////////////////////////////////////

 SkImage* SkImage::NewFromBitmap(const SkBitmap& bm) {
     SkPixelRef* pr = bm.pixelRef();
     if (nullptr == pr) {
         return nullptr;
     }

 #if SK_SUPPORT_GPU
     if (GrTexture* tex = pr->getTexture()) {
         SkAutoTUnref<GrTexture> unrefCopy;
         if (!bm.isImmutable()) {
             const bool notBudgeted = false;
             tex = GrDeepCopyTexture(tex, notBudgeted);
             if (nullptr == tex) {
                 return nullptr;
             }
             unrefCopy.reset(tex);
         }
         const SkImageInfo info = bm.info();
         return new SkImage_Gpu(info.width(), info.height(), bm.getGenerationID(), info.alphaType(),
                                tex, 0, SkSurface::kNo_Budgeted);
     }
 #endif

     // This will check for immutable (share or copy)
     return SkNewImageFromRasterBitmap(bm, nullptr);
 }

 bool SkImage::asLegacyBitmap(SkBitmap* bitmap, LegacyBitmapMode mode) const {
     return as_IB(this)->onAsLegacyBitmap(bitmap, mode);
 }

 bool SkImage_Base::onAsLegacyBitmap(SkBitmap* bitmap, LegacyBitmapMode mode) const {
     // As the base-class, all we can do is make a copy (regardless of mode).
     // Subclasses that want to be more optimal should override.
     SkImageInfo info = SkImageInfo::MakeN32(this->width(), this->height(),
                                     this->isOpaque() ? kOpaque_SkAlphaType : kPremul_SkAlphaType);
     if (!bitmap->tryAllocPixels(info)) {
         return false;
     }
     if (!this->readPixels(bitmap->info(), bitmap->getPixels(), bitmap->rowBytes(), 0, 0)) {
         bitmap->reset();
         return false;
     }

     if (kRO_LegacyBitmapMode == mode) {
         bitmap->setImmutable();
     }
     return true;
 }

 SkImage* SkImage::NewFromPicture(const SkPicture* picture, const SkISize& dimensions,
                                  const SkMatrix* matrix, const SkPaint* paint) {
     if (!picture) {
         return nullptr;
     }
     return NewFromGenerator(SkImageGenerator::NewFromPicture(dimensions, picture, matrix, paint));
 }

 bool SkImage::isLazyGenerated() const {
     return as_IB(this)->onIsLazyGenerated();
 }

 //////////////////////////////////////////////////////////////////////////////////////

 #if !SK_SUPPORT_GPU

 SkImage* SkImage::NewFromTexture(GrContext*, const GrBackendTextureDesc&, SkAlphaType,
                                  TextureReleaseProc, ReleaseContext) {
     return nullptr;
 }

 SkImage* SkImage::NewFromAdoptedTexture(GrContext*, const GrBackendTextureDesc&, SkAlphaType) {
     return nullptr;
 }

 SkImage* SkImage::NewFromTextureCopy(GrContext*, const GrBackendTextureDesc&, SkAlphaType) {
     return nullptr;
 }

 #endif
	/*
	* Copyright 2012 Google Inc.
	*
	* Use of this source code is governed by a BSD-style license that can be
	* found in the LICENSE file.
	*/

	#include "SkBitmap.h"
	#include "SkCanvas.h"
	#include "SkData.h"
	#include "SkImageGenerator.h"
	#include "SkImagePriv.h"
	#include "SkImage_Base.h"
	#include "SkNextID.h"
	#include "SkPixelRef.h"
	#include "SkPixelSerializer.h"
	#include "SkReadPixelsRec.h"
	#include "SkString.h"
	#include "SkSurface.h"

	#if SK_SUPPORT_GPU
	#include "GrTexture.h"
	#include "GrContext.h"
	#include "SkImage_Gpu.h"
	#endif

	SkImage::SkImage(int width, int height, uint32_t uniqueID)
	: fWidth(width)
	, fHeight(height)
	, fUniqueID(kNeedNewImageUniqueID == uniqueID ? SkNextID::ImageID() : uniqueID)
	{
	SkASSERT(width > 0);
	SkASSERT(height > 0);
	}

	const void* SkImage::peekPixels(SkImageInfo* info, size_t* rowBytes) const {
	SkImageInfo infoStorage;
	size_t rowBytesStorage;
	if (nullptr == info) {
	info = &infoStorage;
	}
	if (nullptr == rowBytes) {
	rowBytes = &rowBytesStorage;
	}
	return as_IB(this)->onPeekPixels(info, rowBytes);
	}

	bool SkImage::readPixels(const SkImageInfo& dstInfo, void* dstPixels, size_t dstRowBytes,
	int srcX, int srcY) const {
	SkReadPixelsRec rec(dstInfo, dstPixels, dstRowBytes, srcX, srcY);
	if (!rec.trim(this->width(), this->height())) {
	return false;
	}
	return as_IB(this)->onReadPixels(rec.fInfo, rec.fPixels, rec.fRowBytes, rec.fX, rec.fY);
	}

	void SkImage::preroll(GrContext* ctx) const {
	as_IB(this)->onPreroll(ctx);
	}

	SkShader* SkImage::newShader(SkShader::TileMode tileX,
	SkShader::TileMode tileY,
	const SkMatrix* localMatrix) const {
	return as_IB(this)->onNewShader(tileX, tileY, localMatrix);
	}

	SkData* SkImage::encode(SkImageEncoder::Type type, int quality) const {
	SkBitmap bm;
	if (as_IB(this)->getROPixels(&bm)) {
	return SkImageEncoder::EncodeData(bm, type, quality);
	}
	return nullptr;
	}

	namespace {

	class DefaultSerializer : public SkPixelSerializer {
	protected:
	bool onUseEncodedData(const void *data, size_t len) override {
	return true;
	}

	SkData* onEncodePixels(const SkImageInfo& info, const void* pixels, size_t rowBytes) override {
	return SkImageEncoder::EncodeData(info, pixels, rowBytes, SkImageEncoder::kPNG_Type, 100);
	}
	};

	} // anonymous namespace

	SkData* SkImage::encode(SkPixelSerializer* serializer) const {
	DefaultSerializer defaultSerializer;
	SkPixelSerializer* effectiveSerializer = serializer ? serializer : &defaultSerializer;

	SkAutoTUnref<SkData> encoded(this->refEncoded());
	if (encoded && effectiveSerializer->useEncodedData(encoded->data(), encoded->size())) {
	return encoded.detach();
	}

	SkBitmap bm;
	SkAutoPixmapUnlock apu;
	if (as_IB(this)->getROPixels(&bm) && bm.requestLock(&apu)) {
	const SkPixmap& pmap = apu.pixmap();
	return effectiveSerializer->encodePixels(pmap.info(), pmap.addr(), pmap.rowBytes());
	}

	return nullptr;
	}

	SkData* SkImage::refEncoded() const {
	return as_IB(this)->onRefEncoded();
	}

	SkImage* SkImage::NewFromEncoded(SkData* encoded, const SkIRect* subset) {
	if (nullptr == encoded \|\| 0 == encoded->size()) {
	return nullptr;
	}
	SkImageGenerator* generator = SkImageGenerator::NewFromEncoded(encoded);
	return generator ? SkImage::NewFromGenerator(generator, subset) : nullptr;
	}

	SkSurface* SkImage::newSurface(const SkImageInfo& info, const SkSurfaceProps* props) const {
	if (nullptr == props) {
	props = &as_IB(this)->props();
	}
	return as_IB(this)->onNewSurface(info, *props);
	}

	const char* SkImage::toString(SkString* str) const {
	str->appendf("image: (id:%d (%d, %d) %s)", this->uniqueID(), this->width(), this->height(),
	this->isOpaque() ? "opaque" : "");
	return str->c_str();
	}

	SkImage* SkImage::newImage(int newWidth, int newHeight, const SkIRect* subset,
	SkFilterQuality quality) const {
	if (newWidth <= 0 \|\| newHeight <= 0) {
	return nullptr;
	}

	const SkIRect bounds = SkIRect::MakeWH(this->width(), this->height());

	if (subset) {
	if (!bounds.contains(*subset)) {
	return nullptr;
	}
	if (bounds == *subset) {
	subset = nullptr; // and fall through to check below
	}
	}

	if (nullptr == subset && this->width() == newWidth && this->height() == newHeight) {
	return SkRef(const_cast<SkImage*>(this));
	}

	return as_IB(this)->onNewImage(newWidth, newHeight, subset, quality);
	}

	#if SK_SUPPORT_GPU

	GrTexture* SkImage::getTexture() const {
	return as_IB(this)->getTexture();
	}

	bool SkImage::isTextureBacked() const { return SkToBool(as_IB(this)->getTexture()); }

	GrBackendObject SkImage::getTextureHandle(bool flushPendingGrContextIO) const {
	GrTexture* texture = as_IB(this)->getTexture();
	if (texture) {
	GrContext* context = texture->getContext();
	if (context) {
	if (flushPendingGrContextIO) {
	context->prepareSurfaceForExternalIO(texture);
	}
	}
	return texture->getTextureHandle();
	}
	return 0;
	}

	#else

	GrTexture* SkImage::getTexture() const { return nullptr; }

	bool SkImage::isTextureBacked() const { return false; }

	GrBackendObject SkImage::getTextureHandle(bool) const { return 0; }

	#endif

	///////////////////////////////////////////////////////////////////////////////

	static bool raster_canvas_supports(const SkImageInfo& info) {
	switch (info.colorType()) {
	case kN32_SkColorType:
	return kUnpremul_SkAlphaType != info.alphaType();
	case kRGB_565_SkColorType:
	return true;
	case kAlpha_8_SkColorType:
	return true;
	default:
	break;
	}
	return false;
	}

	bool SkImage_Base::onReadPixels(const SkImageInfo& dstInfo, void* dstPixels, size_t dstRowBytes,
	int srcX, int srcY) const {
	if (!raster_canvas_supports(dstInfo)) {
	return false;
	}

	SkBitmap bm;
	bm.installPixels(dstInfo, dstPixels, dstRowBytes);
	SkCanvas canvas(bm);

	SkPaint paint;
	paint.setXfermodeMode(SkXfermode::kSrc_Mode);
	canvas.drawImage(this, -SkIntToScalar(srcX), -SkIntToScalar(srcY), &paint);

	return true;
	}

	SkImage* SkImage_Base::onNewImage(int newWidth, int newHeight, const SkIRect* subset,
	SkFilterQuality quality) const {
	const bool opaque = this->isOpaque();
	const SkImageInfo info = SkImageInfo::Make(newWidth, newHeight, kN32_SkColorType,
	opaque ? kOpaque_SkAlphaType : kPremul_SkAlphaType);
	SkAutoTUnref<SkSurface> surface(this->newSurface(info, nullptr));
	if (!surface.get()) {
	return nullptr;
	}

	SkRect src;
	if (subset) {
	src.set(*subset);
	} else {
	src = SkRect::MakeIWH(this->width(), this->height());
	}

	surface->getCanvas()->scale(newWidth / src.width(), newHeight / src.height());
	surface->getCanvas()->translate(-src.x(), -src.y());

	SkPaint paint;
	paint.setXfermodeMode(SkXfermode::kSrc_Mode);
	paint.setFilterQuality(quality);
	surface->getCanvas()->drawImage(this, 0, 0, &paint);
	return surface->newImageSnapshot();
	}

	///////////////////////////////////////////////////////////////////////////////////////////////////

	bool SkImage::peekPixels(SkPixmap* pmap) const {
	SkImageInfo info;
	size_t rowBytes;
	const void* pixels = this->peekPixels(&info, &rowBytes);
	if (pixels) {
	if (pmap) {
	pmap->reset(info, pixels, rowBytes);
	}
	return true;
	}
	return false;
	}

	bool SkImage::readPixels(const SkPixmap& pmap, int srcX, int srcY) const {
	return this->readPixels(pmap.info(), pmap.writable_addr(), pmap.rowBytes(), srcX, srcY);
	}

	///////////////////////////////////////////////////////////////////////////////////////////////////

	SkImage* SkImage::NewFromBitmap(const SkBitmap& bm) {
	SkPixelRef* pr = bm.pixelRef();
	if (nullptr == pr) {
	return nullptr;
	}

	#if SK_SUPPORT_GPU
	if (GrTexture* tex = pr->getTexture()) {
	SkAutoTUnref<GrTexture> unrefCopy;
	if (!bm.isImmutable()) {
	const bool notBudgeted = false;
	tex = GrDeepCopyTexture(tex, notBudgeted);
	if (nullptr == tex) {
	return nullptr;
	}
	unrefCopy.reset(tex);
	}
	const SkImageInfo info = bm.info();
	return new SkImage_Gpu(info.width(), info.height(), bm.getGenerationID(), info.alphaType(),
	tex, 0, SkSurface::kNo_Budgeted);
	}
	#endif

	// This will check for immutable (share or copy)
	return SkNewImageFromRasterBitmap(bm, nullptr);
	}

	bool SkImage::asLegacyBitmap(SkBitmap* bitmap, LegacyBitmapMode mode) const {
	return as_IB(this)->onAsLegacyBitmap(bitmap, mode);
	}

	bool SkImage_Base::onAsLegacyBitmap(SkBitmap* bitmap, LegacyBitmapMode mode) const {
	// As the base-class, all we can do is make a copy (regardless of mode).
	// Subclasses that want to be more optimal should override.
	SkImageInfo info = SkImageInfo::MakeN32(this->width(), this->height(),
	this->isOpaque() ? kOpaque_SkAlphaType : kPremul_SkAlphaType);
	if (!bitmap->tryAllocPixels(info)) {
	return false;
	}
	if (!this->readPixels(bitmap->info(), bitmap->getPixels(), bitmap->rowBytes(), 0, 0)) {
	bitmap->reset();
	return false;
	}

	if (kRO_LegacyBitmapMode == mode) {
	bitmap->setImmutable();
	}
	return true;
	}

	SkImage* SkImage::NewFromPicture(const SkPicture* picture, const SkISize& dimensions,
	const SkMatrix* matrix, const SkPaint* paint) {
	if (!picture) {
	return nullptr;
	}
	return NewFromGenerator(SkImageGenerator::NewFromPicture(dimensions, picture, matrix, paint));
	}

	bool SkImage::isLazyGenerated() const {
	return as_IB(this)->onIsLazyGenerated();
	}

	//////////////////////////////////////////////////////////////////////////////////////

	#if !SK_SUPPORT_GPU

	SkImage* SkImage::NewFromTexture(GrContext*, const GrBackendTextureDesc&, SkAlphaType,
	TextureReleaseProc, ReleaseContext) {
	return nullptr;
	}

	SkImage* SkImage::NewFromAdoptedTexture(GrContext*, const GrBackendTextureDesc&, SkAlphaType) {
	return nullptr;
	}

	SkImage* SkImage::NewFromTextureCopy(GrContext*, const GrBackendTextureDesc&, SkAlphaType) {
	return nullptr;
	}

	#endif