Index: dm/DMSrcSink.cpp |
diff --git a/dm/DMSrcSink.cpp b/dm/DMSrcSink.cpp |
index b44f1d78180e94e9d9f5505789024e5db42c4007..a615b78e7aa34b5296e8332f5ea4036531800a39 100644 |
--- a/dm/DMSrcSink.cpp |
+++ b/dm/DMSrcSink.cpp |
@@ -1,6 +1,7 @@ |
#include "DMSrcSink.h" |
#include "SamplePipeControllers.h" |
#include "SkCommonFlags.h" |
+#include "SkCodec.h" |
#include "SkDocument.h" |
#include "SkMultiPictureDraw.h" |
#include "SkNullCanvas.h" |
@@ -11,6 +12,8 @@ |
#include "SkStream.h" |
#include "SkXMLWriter.h" |
+DEFINE_bool(codec, false, "Use SkCodec instead of SkImageDecoder"); |
+ |
namespace DM { |
GMSrc::GMSrc(skiagm::GMRegistry::Factory factory) : fFactory(factory) {} |
@@ -45,9 +48,32 @@ Error ImageSrc::draw(SkCanvas* canvas) const { |
if (fDivisor == 0) { |
// Decode the full image. |
SkBitmap bitmap; |
- if (!SkImageDecoder::DecodeMemory(encoded->data(), encoded->size(), &bitmap, |
- dstColorType, SkImageDecoder::kDecodePixels_Mode)) { |
- return SkStringPrintf("Couldn't decode %s.", fPath.c_str()); |
+ if (FLAGS_codec) { |
+ SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded)); |
+ if (!codec) { |
+ return SkStringPrintf("Couldn't decode %s.", fPath.c_str()); |
+ } |
+ SkImageInfo info; |
+ if (!codec->getInfo(&info)) { |
+ return SkStringPrintf("Couldn't getInfo %s.", fPath.c_str()); |
+ } |
+ info = info.makeColorType(dstColorType); |
+ if (info.alphaType() == kUnpremul_SkAlphaType) { |
+ // FIXME: Currently we cannot draw unpremultiplied sources. |
+ info = info.makeAlphaType(kPremul_SkAlphaType); |
+ } |
+ bitmap.allocPixels(info); |
+ SkAutoLockPixels alp(bitmap); |
djsollen
2015/02/25 16:02:19
aren't the pixels locked during allocPixels if you
scroggo
2015/02/25 18:00:54
Yes. In fact I initially left this line out, if yo
|
+ const SkImageGenerator::Result result = codec->getPixels(info, bitmap.getPixels(), |
+ bitmap.rowBytes()); |
+ if (result != SkImageGenerator::kSuccess) { |
+ return SkStringPrintf("Couldn't getPixels %s.", fPath.c_str()); |
+ } |
+ } else { |
djsollen
2015/02/25 16:02:19
else if
scroggo
2015/02/25 18:00:54
In this case, I actually like the current code bet
|
+ if (!SkImageDecoder::DecodeMemory(encoded->data(), encoded->size(), &bitmap, |
+ dstColorType, SkImageDecoder::kDecodePixels_Mode)) { |
+ return SkStringPrintf("Couldn't decode %s.", fPath.c_str()); |
+ } |
} |
encoded.reset((SkData*)NULL); // Might as well drop this when we're done with it. |
canvas->drawBitmap(bitmap, 0,0); |