OLD | NEW |
1 /* | 1 /* |
2 * Copyright 2014 Google Inc. | 2 * Copyright 2014 Google Inc. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license that can be | 4 * Use of this source code is governed by a BSD-style license that can be |
5 * found in the LICENSE file. | 5 * found in the LICENSE file. |
6 */ | 6 */ |
7 | 7 |
8 #include "SkColorPriv.h" | 8 #include "SkColorPriv.h" |
9 #include "SkImageDecoder.h" | 9 #include "SkImageDecoder.h" |
10 #include "SkImageGenerator.h" | 10 #include "SkImageGenerator.h" |
(...skipping 311 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
322 return SkImageDecoder::kUnknown_Format; | 322 return SkImageDecoder::kUnknown_Format; |
323 } | 323 } |
324 | 324 |
325 SkImageEncoder* sk_libktx_efactory(SkImageEncoder::Type t) { | 325 SkImageEncoder* sk_libktx_efactory(SkImageEncoder::Type t) { |
326 return (SkImageEncoder::kKTX_Type == t) ? new SkKTXImageEncoder : nullptr; | 326 return (SkImageEncoder::kKTX_Type == t) ? new SkKTXImageEncoder : nullptr; |
327 } | 327 } |
328 | 328 |
329 static SkImageDecoder_DecodeReg gReg(sk_libktx_dfactory); | 329 static SkImageDecoder_DecodeReg gReg(sk_libktx_dfactory); |
330 static SkImageDecoder_FormatReg gFormatReg(get_format_ktx); | 330 static SkImageDecoder_FormatReg gFormatReg(get_format_ktx); |
331 static SkImageEncoder_EncodeReg gEReg(sk_libktx_efactory); | 331 static SkImageEncoder_EncodeReg gEReg(sk_libktx_efactory); |
332 | |
333 ////////////////////////////////////////////////////////////////////////////////
///////// | |
334 // Old implementation of SkImageGenerator::NewFromEncoded which uses SkImageDeco
der. | |
335 // Here because it is only needed by DM and tests for Ktx. | |
336 class BareMemoryAllocator : public SkBitmap::Allocator { | |
337 const SkImageInfo fInfo; | |
338 void* const fMemory; | |
339 const size_t fRowBytes; | |
340 | |
341 public: | |
342 BareMemoryAllocator(const SkImageInfo& info, void* memory, size_t rowBytes) | |
343 : fInfo(info), fMemory(memory), fRowBytes(rowBytes) | |
344 {} | |
345 | |
346 protected: | |
347 bool allocPixelRef(SkBitmap* bm, SkColorTable* ctable) override { | |
348 const SkImageInfo bmi = bm->info(); | |
349 if (bmi.width() != fInfo.width() || bmi.height() != fInfo.height() || | |
350 bmi.colorType() != fInfo.colorType()) | |
351 { | |
352 return false; | |
353 } | |
354 return bm->installPixels(bmi, fMemory, fRowBytes, ctable, nullptr, nullp
tr); | |
355 } | |
356 }; | |
357 | |
358 class SkImageDecoderGenerator : public SkImageGenerator { | |
359 const SkImageInfo fInfo; | |
360 SkAutoTDelete<SkImageDecoder> fDecoder; | |
361 SkAutoTUnref<SkData> fData; | |
362 | |
363 public: | |
364 SkImageDecoderGenerator(const SkImageInfo& info, SkImageDecoder* decoder, Sk
Data* data) | |
365 : INHERITED(info), fInfo(info), fDecoder(decoder), fData(SkRef(data)) | |
366 {} | |
367 | |
368 protected: | |
369 SkData* onRefEncodedData(SK_REFENCODEDDATA_CTXPARAM) override { | |
370 return SkRef(fData.get()); | |
371 } | |
372 bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes, | |
373 SkPMColor ctableEntries[], int* ctableCount) override { | |
374 SkMemoryStream stream(fData->data(), fData->size(), false); | |
375 SkAutoTUnref<BareMemoryAllocator> allocator( | |
376 new BareMemoryAllocator(info, pixels, rowBytes)); | |
377 fDecoder->setAllocator(allocator); | |
378 fDecoder->setRequireUnpremultipliedColors(kUnpremul_SkAlphaType == info.
alphaType()); | |
379 | |
380 SkBitmap bm; | |
381 const SkImageDecoder::Result result = fDecoder->decode(&stream, &bm, inf
o.colorType(), | |
382 SkImageDecoder::k
DecodePixels_Mode); | |
383 if (SkImageDecoder::kFailure == result) { | |
384 return false; | |
385 } | |
386 | |
387 SkASSERT(info.colorType() == bm.info().colorType()); | |
388 | |
389 if (kIndex_8_SkColorType == info.colorType()) { | |
390 SkASSERT(ctableEntries); | |
391 | |
392 SkColorTable* ctable = bm.getColorTable(); | |
393 if (nullptr == ctable) { | |
394 return false; | |
395 } | |
396 const int count = ctable->count(); | |
397 memcpy(ctableEntries, ctable->readColors(), count * sizeof(SkPMColor
)); | |
398 *ctableCount = count; | |
399 } | |
400 return true; | |
401 } | |
402 | |
403 bool onGetYUV8Planes(SkISize sizes[3], void* planes[3], size_t rowBytes[3], | |
404 SkYUVColorSpace* colorSpace) override { | |
405 SkMemoryStream stream(fData->data(), fData->size(), false); | |
406 return fDecoder->decodeYUV8Planes(&stream, sizes, planes, rowBytes, colo
rSpace); | |
407 } | |
408 | |
409 private: | |
410 typedef SkImageGenerator INHERITED; | |
411 }; | |
412 | |
413 SkImageGenerator* decoder_image_generator(SkData* data) { | |
414 SkMemoryStream stream(data->data(), data->size(), false); | |
415 SkImageDecoder* decoder = SkImageDecoder::Factory(&stream); | |
416 if (nullptr == decoder) { | |
417 return nullptr; | |
418 } | |
419 | |
420 SkBitmap bm; | |
421 stream.rewind(); | |
422 if (!decoder->decode(&stream, &bm, kUnknown_SkColorType, SkImageDecoder::kDe
codeBounds_Mode)) { | |
423 delete decoder; | |
424 return nullptr; | |
425 } | |
426 | |
427 return new SkImageDecoderGenerator(bm.info(), decoder, data); | |
428 } | |
429 | |
OLD | NEW |