Index: src/images/SkImageDecoder_pkm.cpp |
diff --git a/src/images/SkImageDecoder_pkm.cpp b/src/images/SkImageDecoder_pkm.cpp |
new file mode 100644 |
index 0000000000000000000000000000000000000000..af68f20d9775fea69a11ff7d85f096d7c2de6361 |
--- /dev/null |
+++ b/src/images/SkImageDecoder_pkm.cpp |
@@ -0,0 +1,128 @@ |
+/* |
+ * Copyright 2014 Google Inc. |
+ * |
+ * Use of this source code is governed by a BSD-style license that can be |
+ * found in the LICENSE file. |
+ */ |
+ |
+#include "SkColorPriv.h" |
+#include "SkData.h" |
+#include "SkImageDecoder.h" |
+#include "SkScaledBitmapSampler.h" |
+#include "SkStream.h" |
+#include "SkStreamPriv.h" |
+#include "SkTextureCompressor.h" |
+#include "SkTypes.h" |
+ |
+#include "etc1.h" |
+ |
+class SkPKMImageDecoder : public SkImageDecoder { |
+public: |
+ SkPKMImageDecoder() { } |
+ |
+ Format getFormat() const override { |
+ return kPKM_Format; |
+ } |
+ |
+protected: |
+ Result onDecode(SkStream* stream, SkBitmap* bm, Mode) override; |
+ |
+private: |
+ typedef SkImageDecoder INHERITED; |
+}; |
+ |
+///////////////////////////////////////////////////////////////////////////////////////// |
+ |
+SkImageDecoder::Result SkPKMImageDecoder::onDecode(SkStream* stream, SkBitmap* bm, Mode mode) { |
+ sk_sp<SkData> data(SkCopyStreamToData(stream)); |
+ if (!data || !data->size()) { |
+ return kFailure; |
+ } |
+ |
+ unsigned char* buf = (unsigned char*) data->data(); |
+ |
+ // Make sure original PKM header is there... |
+ SkASSERT(etc1_pkm_is_valid(buf)); |
+ |
+ const unsigned short width = etc1_pkm_get_width(buf); |
+ const unsigned short height = etc1_pkm_get_height(buf); |
+ |
+ // Setup the sampler... |
+ SkScaledBitmapSampler sampler(width, height, this->getSampleSize()); |
+ |
+ // Set the config... |
+ bm->setInfo(SkImageInfo::MakeN32(sampler.scaledWidth(), sampler.scaledHeight(), |
+ kOpaque_SkAlphaType)); |
+ if (SkImageDecoder::kDecodeBounds_Mode == mode) { |
+ return kSuccess; |
+ } |
+ |
+ if (!this->allocPixelRef(bm, nullptr)) { |
+ return kFailure; |
+ } |
+ |
+ // Lock the pixels, since we're about to write to them... |
+ SkAutoLockPixels alp(*bm); |
+ |
+ if (!sampler.begin(bm, SkScaledBitmapSampler::kRGB, *this)) { |
+ return kFailure; |
+ } |
+ |
+ // Advance buffer past the header |
+ buf += ETC_PKM_HEADER_SIZE; |
+ |
+ // ETC1 Data is encoded as RGB pixels, so we should extract it as such |
+ int nPixels = width * height; |
+ SkAutoMalloc outRGBData(nPixels * 3); |
+ uint8_t *outRGBDataPtr = reinterpret_cast<uint8_t *>(outRGBData.get()); |
+ |
+ // Decode ETC1 |
+ if (!SkTextureCompressor::DecompressBufferFromFormat( |
+ outRGBDataPtr, width*3, buf, width, height, SkTextureCompressor::kETC1_Format)) { |
+ return kFailure; |
+ } |
+ |
+ // Set each of the pixels... |
+ const int srcRowBytes = width * 3; |
+ const int dstHeight = sampler.scaledHeight(); |
+ const uint8_t *srcRow = reinterpret_cast<uint8_t *>(outRGBDataPtr); |
+ srcRow += sampler.srcY0() * srcRowBytes; |
+ for (int y = 0; y < dstHeight; ++y) { |
+ sampler.next(srcRow); |
+ srcRow += sampler.srcDY() * srcRowBytes; |
+ } |
+ |
+ return kSuccess; |
+} |
+ |
+///////////////////////////////////////////////////////////////////////////////////////// |
+DEFINE_DECODER_CREATOR(PKMImageDecoder); |
+///////////////////////////////////////////////////////////////////////////////////////// |
+ |
+static bool is_pkm(SkStreamRewindable* stream) { |
+ // Read the PKM header and make sure it's valid. |
+ unsigned char buf[ETC_PKM_HEADER_SIZE]; |
+ if (stream->read((void*)buf, ETC_PKM_HEADER_SIZE) != ETC_PKM_HEADER_SIZE) { |
+ return false; |
+ } |
+ |
+ return SkToBool(etc1_pkm_is_valid(buf)); |
+} |
+ |
+static SkImageDecoder* sk_libpkm_dfactory(SkStreamRewindable* stream) { |
+ if (is_pkm(stream)) { |
+ return new SkPKMImageDecoder; |
+ } |
+ return nullptr; |
+} |
+ |
+static SkImageDecoder_DecodeReg gReg(sk_libpkm_dfactory); |
+ |
+static SkImageDecoder::Format get_format_pkm(SkStreamRewindable* stream) { |
+ if (is_pkm(stream)) { |
+ return SkImageDecoder::kPKM_Format; |
+ } |
+ return SkImageDecoder::kUnknown_Format; |
+} |
+ |
+static SkImageDecoder_FormatReg gFormatReg(get_format_pkm); |