| Index: src/ports/SkImageGenerator_skia.cpp
|
| diff --git a/src/ports/SkImageGenerator_skia.cpp b/src/ports/SkImageGenerator_skia.cpp
|
| index b6ddee935ad7bdde59570b8938f0c785d4f1134a..8dbad555dd1d7f35ae7955c6187b34a97052f11e 100644
|
| --- a/src/ports/SkImageGenerator_skia.cpp
|
| +++ b/src/ports/SkImageGenerator_skia.cpp
|
| @@ -6,8 +6,100 @@
|
| */
|
|
|
| #include "SkData.h"
|
| -#include "SkCodecImageGenerator.h"
|
| +#include "SkImageDecoder.h"
|
| +#include "SkImageGenerator.h"
|
| +#include "SkStream.h"
|
| +
|
| +class BareMemoryAllocator : public SkBitmap::Allocator {
|
| + const SkImageInfo fInfo;
|
| + void* const fMemory;
|
| + const size_t fRowBytes;
|
| +
|
| +public:
|
| + BareMemoryAllocator(const SkImageInfo& info, void* memory, size_t rowBytes)
|
| + : fInfo(info), fMemory(memory), fRowBytes(rowBytes)
|
| + {}
|
| +
|
| +protected:
|
| + bool allocPixelRef(SkBitmap* bm, SkColorTable* ctable) override {
|
| + const SkImageInfo bmi = bm->info();
|
| + if (bmi.width() != fInfo.width() || bmi.height() != fInfo.height() ||
|
| + bmi.colorType() != fInfo.colorType())
|
| + {
|
| + return false;
|
| + }
|
| + return bm->installPixels(bmi, fMemory, fRowBytes, ctable, nullptr, nullptr);
|
| + }
|
| +};
|
| +
|
| +class SkImageDecoderGenerator : public SkImageGenerator {
|
| + const SkImageInfo fInfo;
|
| + SkAutoTDelete<SkImageDecoder> fDecoder;
|
| + SkAutoTUnref<SkData> fData;
|
| +
|
| +public:
|
| + SkImageDecoderGenerator(const SkImageInfo& info, SkImageDecoder* decoder, SkData* data)
|
| + : INHERITED(info), fInfo(info), fDecoder(decoder), fData(SkRef(data))
|
| + {}
|
| +
|
| +protected:
|
| + SkData* onRefEncodedData(SK_REFENCODEDDATA_CTXPARAM) override {
|
| + return SkRef(fData.get());
|
| + }
|
| + bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes,
|
| + SkPMColor ctableEntries[], int* ctableCount) override {
|
| + SkMemoryStream stream(fData->data(), fData->size(), false);
|
| + SkAutoTUnref<BareMemoryAllocator> allocator(
|
| + new BareMemoryAllocator(info, pixels, rowBytes));
|
| + fDecoder->setAllocator(allocator);
|
| + fDecoder->setRequireUnpremultipliedColors(kUnpremul_SkAlphaType == info.alphaType());
|
| +
|
| + SkBitmap bm;
|
| + const SkImageDecoder::Result result = fDecoder->decode(&stream, &bm, info.colorType(),
|
| + SkImageDecoder::kDecodePixels_Mode);
|
| + if (SkImageDecoder::kFailure == result) {
|
| + return false;
|
| + }
|
| +
|
| + SkASSERT(info.colorType() == bm.info().colorType());
|
| +
|
| + if (kIndex_8_SkColorType == info.colorType()) {
|
| + SkASSERT(ctableEntries);
|
| +
|
| + SkColorTable* ctable = bm.getColorTable();
|
| + if (nullptr == ctable) {
|
| + return false;
|
| + }
|
| + const int count = ctable->count();
|
| + memcpy(ctableEntries, ctable->readColors(), count * sizeof(SkPMColor));
|
| + *ctableCount = count;
|
| + }
|
| + return true;
|
| + }
|
| +
|
| + bool onGetYUV8Planes(SkISize sizes[3], void* planes[3], size_t rowBytes[3],
|
| + SkYUVColorSpace* colorSpace) override {
|
| + SkMemoryStream stream(fData->data(), fData->size(), false);
|
| + return fDecoder->decodeYUV8Planes(&stream, sizes, planes, rowBytes, colorSpace);
|
| + }
|
| +
|
| +private:
|
| + typedef SkImageGenerator INHERITED;
|
| +};
|
|
|
| SkImageGenerator* SkImageGenerator::NewFromEncodedImpl(SkData* data) {
|
| - return SkCodecImageGenerator::NewFromEncodedCodec(data);
|
| + SkMemoryStream stream(data->data(), data->size(), false);
|
| + SkImageDecoder* decoder = SkImageDecoder::Factory(&stream);
|
| + if (nullptr == decoder) {
|
| + return nullptr;
|
| + }
|
| +
|
| + SkBitmap bm;
|
| + stream.rewind();
|
| + if (!decoder->decode(&stream, &bm, kUnknown_SkColorType, SkImageDecoder::kDecodeBounds_Mode)) {
|
| + delete decoder;
|
| + return nullptr;
|
| + }
|
| +
|
| + return new SkImageDecoderGenerator(bm.info(), decoder, data);
|
| }
|
|
|