Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 /* | 1 /* |
| 2 * Copyright 2015 Google Inc. | 2 * Copyright 2015 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 "SkCodecPriv.h" | 8 #include "SkCodecPriv.h" |
| 9 #include "SkWebpCodec.h" | 9 #include "SkWebpCodec.h" |
| 10 #include "SkTemplates.h" | 10 #include "SkTemplates.h" |
| (...skipping 13 matching lines...) Expand all Loading... | |
| 24 // WEBP starts with the following: | 24 // WEBP starts with the following: |
| 25 // RIFFXXXXWEBPVP | 25 // RIFFXXXXWEBPVP |
| 26 // Where XXXX is unspecified. | 26 // Where XXXX is unspecified. |
| 27 const char* bytes = static_cast<const char*>(buf); | 27 const char* bytes = static_cast<const char*>(buf); |
| 28 return bytesRead >= 14 && !memcmp(bytes, "RIFF", 4) && !memcmp(&bytes[8], "W EBPVP", 6); | 28 return bytesRead >= 14 && !memcmp(bytes, "RIFF", 4) && !memcmp(&bytes[8], "W EBPVP", 6); |
| 29 } | 29 } |
| 30 | 30 |
| 31 // Parse headers of RIFF container, and check for valid Webp (VP8) content. | 31 // Parse headers of RIFF container, and check for valid Webp (VP8) content. |
| 32 // NOTE: This calls peek instead of read, since onGetPixels will need these | 32 // NOTE: This calls peek instead of read, since onGetPixels will need these |
| 33 // bytes again. | 33 // bytes again. |
| 34 static bool webp_parse_header(SkStream* stream, int* width, int* height, SkEncod edInfo* info) { | 34 bool SkWebpCodec::ReadHeader(SkStream* stream, SkCodec** outCodec) { |
| 35 unsigned char buffer[WEBP_VP8_HEADER_SIZE]; | 35 unsigned char buffer[WEBP_VP8_HEADER_SIZE]; |
| 36 SkASSERT(WEBP_VP8_HEADER_SIZE <= SkCodec::MinBufferedBytesNeeded()); | 36 SkASSERT(WEBP_VP8_HEADER_SIZE <= SkCodec::MinBufferedBytesNeeded()); |
| 37 | 37 |
| 38 const size_t bytesPeeked = stream->peek(buffer, WEBP_VP8_HEADER_SIZE); | 38 const size_t bytesPeeked = stream->peek(buffer, WEBP_VP8_HEADER_SIZE); |
| 39 if (bytesPeeked != WEBP_VP8_HEADER_SIZE) { | 39 if (bytesPeeked != WEBP_VP8_HEADER_SIZE) { |
| 40 // Use read + rewind as a backup | 40 // Use read + rewind as a backup |
| 41 if (stream->read(buffer, WEBP_VP8_HEADER_SIZE) != WEBP_VP8_HEADER_SIZE | 41 if (stream->read(buffer, WEBP_VP8_HEADER_SIZE) != WEBP_VP8_HEADER_SIZE |
| 42 || !stream->rewind()) | 42 || !stream->rewind()) |
| 43 return false; | 43 return false; |
| 44 } | 44 } |
| 45 | 45 |
| 46 WebPBitstreamFeatures features; | 46 WebPBitstreamFeatures features; |
| 47 VP8StatusCode status = WebPGetFeatures(buffer, WEBP_VP8_HEADER_SIZE, &featur es); | 47 VP8StatusCode status = WebPGetFeatures(buffer, WEBP_VP8_HEADER_SIZE, &featur es); |
| 48 if (VP8_STATUS_OK != status) { | 48 if (VP8_STATUS_OK != status) { |
| 49 return false; // Invalid WebP file. | 49 return false; // Invalid WebP file. |
| 50 } | 50 } |
| 51 | 51 |
| 52 // sanity check for image size that's about to be decoded. | 52 // sanity check for image size that's about to be decoded. |
| 53 { | 53 { |
| 54 const int64_t size = sk_64_mul(features.width, features.height); | 54 const int64_t size = sk_64_mul(features.width, features.height); |
| 55 if (!sk_64_isS32(size)) { | 55 if (!sk_64_isS32(size)) { |
| 56 return false; | 56 return false; |
| 57 } | 57 } |
| 58 // now check that if we are 4-bytes per pixel, we also don't overflow | 58 // now check that if we are 4-bytes per pixel, we also don't overflow |
| 59 if (sk_64_asS32(size) > (0x7FFFFFFF >> 2)) { | 59 if (sk_64_asS32(size) > (0x7FFFFFFF >> 2)) { |
| 60 return false; | 60 return false; |
| 61 } | 61 } |
| 62 } | 62 } |
| 63 | 63 |
| 64 if (info) { | 64 if (outCodec) { |
| 65 SkEncodedInfo::Color color; | 65 SkEncodedInfo::Color color; |
| 66 SkEncodedInfo::Alpha alpha; | 66 SkEncodedInfo::Alpha alpha; |
| 67 switch (features.format) { | 67 switch (features.format) { |
| 68 case 0: | 68 case 0: |
| 69 // This indicates a "mixed" format. We would see this for | 69 // This indicates a "mixed" format. We would see this for |
| 70 // animated webps or for webps encoded in multiple fragments. | 70 // animated webps or for webps encoded in multiple fragments. |
| 71 // I believe that this is a rare case. | 71 // I believe that this is a rare case. |
| 72 // We could also guess kYUV here, but I think it makes more | 72 // We could also guess kYUV here, but I think it makes more |
| 73 // sense to guess kBGRA which is likely closer to the final | 73 // sense to guess kBGRA which is likely closer to the final |
| 74 // output. Otherwise, we might end up converting | 74 // output. Otherwise, we might end up converting |
| (...skipping 16 matching lines...) Expand all Loading... | |
| 91 // FIXME: Should we check the has_alpha flag here? It looks | 91 // FIXME: Should we check the has_alpha flag here? It looks |
| 92 // like the image is encoded with an alpha channel | 92 // like the image is encoded with an alpha channel |
| 93 // regardless of whether or not the alpha flag is set. | 93 // regardless of whether or not the alpha flag is set. |
| 94 color = SkEncodedInfo::kBGRA_Color; | 94 color = SkEncodedInfo::kBGRA_Color; |
| 95 alpha = SkEncodedInfo::kUnpremul_Alpha; | 95 alpha = SkEncodedInfo::kUnpremul_Alpha; |
| 96 break; | 96 break; |
| 97 default: | 97 default: |
| 98 return false; | 98 return false; |
| 99 } | 99 } |
| 100 | 100 |
| 101 *width = features.width; | 101 SkEncodedInfo info = SkEncodedInfo::Make(color, alpha, 8); |
| 102 *height = features.height; | 102 *outCodec = new SkWebpCodec(features.width, features.height, info, strea m); |
| 103 *info = SkEncodedInfo::Make(color, alpha, 8); | |
| 104 } | 103 } |
| 105 return true; | 104 return true; |
| 106 } | 105 } |
| 107 | 106 |
| 108 SkCodec* SkWebpCodec::NewFromStream(SkStream* stream) { | 107 SkCodec* SkWebpCodec::NewFromStream(SkStream* stream) { |
| 109 SkAutoTDelete<SkStream> streamDeleter(stream); | 108 SkAutoTDelete<SkStream> streamDeleter(stream); |
| 110 int width, height; | 109 SkCodec* outCodec; |
| 111 SkEncodedInfo info; | 110 if (SkWebpCodec::ReadHeader(stream, &outCodec)) { |
|
scroggo
2016/04/25 15:10:43
Since ReadHeader is used only in this method, I do
msarett
2016/04/25 15:18:21
Of course, that's much better :).
| |
| 112 if (webp_parse_header(stream, &width, &height, &info)) { | 111 // Codec has taken ownership of the stream. |
| 113 return new SkWebpCodec(width, height, info, streamDeleter.release()); | 112 SkASSERT(outCodec); |
| 113 streamDeleter.release(); | |
| 114 return outCodec; | |
| 114 } | 115 } |
| 115 return nullptr; | 116 return nullptr; |
| 116 } | 117 } |
| 117 | 118 |
| 118 // This version is slightly different from SkCodecPriv's version of conversion_p ossible. It | 119 // This version is slightly different from SkCodecPriv's version of conversion_p ossible. It |
| 119 // supports both byte orders for 8888. | 120 // supports both byte orders for 8888. |
| 120 static bool webp_conversion_possible(const SkImageInfo& dst, const SkImageInfo& src) { | 121 static bool webp_conversion_possible(const SkImageInfo& dst, const SkImageInfo& src) { |
| 121 // FIXME: skbug.com/4895 | 122 // FIXME: skbug.com/4895 |
| 122 // Currently, we ignore the SkColorProfileType on the SkImageInfo. We | 123 // Currently, we ignore the SkColorProfileType on the SkImageInfo. We |
| 123 // will treat the encoded data as linear regardless of what the client | 124 // will treat the encoded data as linear regardless of what the client |
| (...skipping 158 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 282 default: | 283 default: |
| 283 return kInvalidInput; | 284 return kInvalidInput; |
| 284 } | 285 } |
| 285 } | 286 } |
| 286 } | 287 } |
| 287 | 288 |
| 288 SkWebpCodec::SkWebpCodec(int width, int height, const SkEncodedInfo& info, SkStr eam* stream) | 289 SkWebpCodec::SkWebpCodec(int width, int height, const SkEncodedInfo& info, SkStr eam* stream) |
| 289 // The spec says an unmarked image is sRGB, so we return that space here. | 290 // The spec says an unmarked image is sRGB, so we return that space here. |
| 290 // TODO: Add support for parsing ICC profiles from webps. | 291 // TODO: Add support for parsing ICC profiles from webps. |
| 291 : INHERITED(width, height, info, stream, SkColorSpace::NewNamed(SkColorSpace ::kSRGB_Named)) {} | 292 : INHERITED(width, height, info, stream, SkColorSpace::NewNamed(SkColorSpace ::kSRGB_Named)) {} |
| OLD | NEW |