msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2015 Google Inc. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
| 7 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 8 | /* |
| 9 | * Copyright (C) 2006 Apple Computer, Inc. All rights reserved. |
| 10 | * |
| 11 | * Redistribution and use in source and binary forms, with or without |
| 12 | * modification, are permitted provided that the following conditions |
| 13 | * are met: |
| 14 | * 1. Redistributions of source code must retain the above copyright |
| 15 | * notice, this list of conditions and the following disclaimer. |
| 16 | * 2. Redistributions in binary form must reproduce the above copyright |
| 17 | * notice, this list of conditions and the following disclaimer in the |
| 18 | * documentation and/or other materials provided with the distribution. |
| 19 | * |
| 20 | * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY |
| 21 | * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| 22 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
| 23 | * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR |
| 24 | * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
| 25 | * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
| 26 | * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
| 27 | * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY |
| 28 | * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 29 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 30 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 31 | */ |
| 32 | |
| 33 | #include "SkCodecAnimation.h" |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 34 | #include "SkCodecPriv.h" |
| 35 | #include "SkColorPriv.h" |
| 36 | #include "SkColorTable.h" |
msarett | 1a46467 | 2016-01-07 13:17:19 -0800 | [diff] [blame] | 37 | #include "SkGifCodec.h" |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 38 | #include "SkStream.h" |
| 39 | #include "SkSwizzler.h" |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 40 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 41 | #include <algorithm> |
| 42 | |
| 43 | #define GIF87_STAMP "GIF87a" |
| 44 | #define GIF89_STAMP "GIF89a" |
| 45 | #define GIF_STAMP_LEN 6 |
msarett | 39b2d5a | 2016-02-17 08:26:31 -0800 | [diff] [blame] | 46 | |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 47 | /* |
| 48 | * Checks the start of the stream to see if the image is a gif |
| 49 | */ |
scroggo | db30be2 | 2015-12-08 18:54:13 -0800 | [diff] [blame] | 50 | bool SkGifCodec::IsGif(const void* buf, size_t bytesRead) { |
| 51 | if (bytesRead >= GIF_STAMP_LEN) { |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 52 | if (memcmp(GIF87_STAMP, buf, GIF_STAMP_LEN) == 0 || |
bungeman | 0153dea | 2015-08-27 16:43:42 -0700 | [diff] [blame] | 53 | memcmp(GIF89_STAMP, buf, GIF_STAMP_LEN) == 0) |
| 54 | { |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 55 | return true; |
| 56 | } |
| 57 | } |
| 58 | return false; |
| 59 | } |
| 60 | |
| 61 | /* |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 62 | * Error function |
| 63 | */ |
bungeman | 0153dea | 2015-08-27 16:43:42 -0700 | [diff] [blame] | 64 | static SkCodec::Result gif_error(const char* msg, SkCodec::Result result = SkCodec::kInvalidInput) { |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 65 | SkCodecPrintf("Gif Error: %s\n", msg); |
| 66 | return result; |
| 67 | } |
| 68 | |
msarett | 438b2ad | 2015-04-09 12:43:10 -0700 | [diff] [blame] | 69 | /* |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 70 | * Assumes IsGif was called and returned true |
| 71 | * Creates a gif decoder |
| 72 | * Reads enough of the stream to determine the image format |
| 73 | */ |
| 74 | SkCodec* SkGifCodec::NewFromStream(SkStream* stream) { |
scroggo | 3d3a65c | 2016-10-24 12:28:30 -0700 | [diff] [blame] | 75 | std::unique_ptr<SkGifImageReader> reader(new SkGifImageReader(stream)); |
scroggo | f9acbe2 | 2016-10-25 12:43:21 -0700 | [diff] [blame] | 76 | if (!reader->parse(SkGifImageReader::SkGIFSizeQuery)) { |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 77 | // Not enough data to determine the size. |
| 78 | return nullptr; |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 79 | } |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 80 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 81 | if (0 == reader->screenWidth() || 0 == reader->screenHeight()) { |
| 82 | return nullptr; |
| 83 | } |
| 84 | |
| 85 | const auto alpha = reader->firstFrameHasAlpha() ? SkEncodedInfo::kBinary_Alpha |
| 86 | : SkEncodedInfo::kOpaque_Alpha; |
| 87 | // Use kPalette since Gifs are encoded with a color table. |
| 88 | // FIXME: Gifs can actually be encoded with 4-bits per pixel. Using 8 works, but we could skip |
| 89 | // expanding to 8 bits and take advantage of the SkSwizzler to work from 4. |
| 90 | const auto encodedInfo = SkEncodedInfo::Make(SkEncodedInfo::kPalette_Color, alpha, 8); |
| 91 | |
| 92 | // Although the encodedInfo is always kPalette_Color, it is possible that kIndex_8 is |
| 93 | // unsupported if the frame is subset and there is no transparent pixel. |
| 94 | const auto colorType = reader->firstFrameSupportsIndex8() ? kIndex_8_SkColorType |
| 95 | : kN32_SkColorType; |
| 96 | // The choice of unpremul versus premul is arbitrary, since all colors are either fully |
| 97 | // opaque or fully transparent (i.e. kBinary), but we stored the transparent colors as all |
| 98 | // zeroes, which is arguably premultiplied. |
| 99 | const auto alphaType = reader->firstFrameHasAlpha() ? kUnpremul_SkAlphaType |
| 100 | : kOpaque_SkAlphaType; |
Matt Sarett | 7f650bd | 2016-10-30 21:25:34 -0400 | [diff] [blame] | 101 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 102 | const auto imageInfo = SkImageInfo::Make(reader->screenWidth(), reader->screenHeight(), |
Matt Sarett | 7f650bd | 2016-10-30 21:25:34 -0400 | [diff] [blame] | 103 | colorType, alphaType, |
| 104 | SkColorSpace::MakeNamed(SkColorSpace::kSRGB_Named)); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 105 | return new SkGifCodec(encodedInfo, imageInfo, reader.release()); |
| 106 | } |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 107 | |
scroggo | b427db1 | 2015-08-12 07:24:13 -0700 | [diff] [blame] | 108 | bool SkGifCodec::onRewind() { |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 109 | fReader->clearDecodeState(); |
scroggo | b427db1 | 2015-08-12 07:24:13 -0700 | [diff] [blame] | 110 | return true; |
| 111 | } |
| 112 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 113 | SkGifCodec::SkGifCodec(const SkEncodedInfo& encodedInfo, const SkImageInfo& imageInfo, |
scroggo | 3d3a65c | 2016-10-24 12:28:30 -0700 | [diff] [blame] | 114 | SkGifImageReader* reader) |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 115 | : INHERITED(encodedInfo, imageInfo, nullptr) |
| 116 | , fReader(reader) |
| 117 | , fTmpBuffer(nullptr) |
| 118 | , fSwizzler(nullptr) |
| 119 | , fCurrColorTable(nullptr) |
| 120 | , fCurrColorTableIsReal(false) |
| 121 | , fFilledBackground(false) |
| 122 | , fFirstCallToIncrementalDecode(false) |
| 123 | , fDst(nullptr) |
| 124 | , fDstRowBytes(0) |
| 125 | , fRowsDecoded(0) |
| 126 | { |
| 127 | reader->setClient(this); |
msarett | 8c8f22a | 2015-04-01 06:58:48 -0700 | [diff] [blame] | 128 | } |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 129 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 130 | std::vector<SkCodec::FrameInfo> SkGifCodec::onGetFrameInfo() { |
scroggo | f9acbe2 | 2016-10-25 12:43:21 -0700 | [diff] [blame] | 131 | fReader->parse(SkGifImageReader::SkGIFFrameCountQuery); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 132 | const size_t size = fReader->imagesCount(); |
| 133 | std::vector<FrameInfo> result(size); |
| 134 | for (size_t i = 0; i < size; i++) { |
scroggo | f9acbe2 | 2016-10-25 12:43:21 -0700 | [diff] [blame] | 135 | const SkGIFFrameContext* frameContext = fReader->frameContext(i); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 136 | result[i].fDuration = frameContext->delayTime(); |
| 137 | result[i].fRequiredFrame = frameContext->getRequiredFrame(); |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 138 | } |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 139 | return result; |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 140 | } |
| 141 | |
Leon Scroggins III | fc49b40 | 2016-10-31 14:08:56 -0400 | [diff] [blame^] | 142 | void SkGifCodec::initializeColorTable(const SkImageInfo& dstInfo, size_t frameIndex) { |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 143 | fCurrColorTable = fReader->getColorTable(dstInfo.colorType(), frameIndex); |
| 144 | fCurrColorTableIsReal = fCurrColorTable; |
| 145 | if (!fCurrColorTable) { |
Leon Scroggins III | a049ac4 | 2016-10-27 11:16:11 -0400 | [diff] [blame] | 146 | // This is possible for an empty frame. Create a dummy with one value (transparent). |
| 147 | SkPMColor color = SK_ColorTRANSPARENT; |
| 148 | fCurrColorTable.reset(new SkColorTable(&color, 1)); |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 149 | } |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 150 | } |
| 151 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 152 | |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 153 | SkCodec::Result SkGifCodec::prepareToDecode(const SkImageInfo& dstInfo, SkPMColor* inputColorPtr, |
| 154 | int* inputColorCount, const Options& opts) { |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 155 | // Check for valid input parameters |
msarett | 2ecc35f | 2016-09-08 11:55:16 -0700 | [diff] [blame] | 156 | if (!conversion_possible_ignore_color_space(dstInfo, this->getInfo())) { |
| 157 | return gif_error("Cannot convert input type to output type.\n", kInvalidConversion); |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 158 | } |
| 159 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 160 | if (dstInfo.colorType() == kRGBA_F16_SkColorType) { |
| 161 | // FIXME: This should be supported. |
| 162 | return gif_error("GIF does not yet support F16.\n", kInvalidConversion); |
| 163 | } |
msarett | 5af4e0b | 2015-11-17 11:18:03 -0800 | [diff] [blame] | 164 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 165 | if (opts.fSubset) { |
| 166 | return gif_error("Subsets not supported.\n", kUnimplemented); |
| 167 | } |
| 168 | |
| 169 | const size_t frameIndex = opts.fFrameIndex; |
scroggo | 53f63b6 | 2016-10-27 08:29:13 -0700 | [diff] [blame] | 170 | if (frameIndex > 0) { |
| 171 | switch (dstInfo.colorType()) { |
| 172 | case kIndex_8_SkColorType: |
| 173 | // FIXME: It is possible that a later frame can be decoded to index8, if it does one |
| 174 | // of the following: |
| 175 | // - Covers the entire previous frame |
| 176 | // - Shares a color table (and transparent index) with any prior frames that are |
| 177 | // showing. |
| 178 | // We must support index8 for the first frame to be backwards compatible on Android, |
| 179 | // but we do not (currently) need to support later frames as index8. |
| 180 | return gif_error("Cannot decode multiframe gif (except frame 0) as index 8.\n", |
| 181 | kInvalidConversion); |
| 182 | case kRGB_565_SkColorType: |
| 183 | // FIXME: In theory, we might be able to support this, but it's not clear that it |
| 184 | // is necessary (Chromium does not decode to 565, and Android does not decode |
| 185 | // frames beyond the first). Disabling it because it is somewhat difficult: |
| 186 | // - If there is a transparent pixel, and this frame draws on top of another frame |
| 187 | // (if the frame is independent with a transparent pixel, we should not decode to |
| 188 | // 565 anyway, since it is not opaque), we need to skip drawing the transparent |
| 189 | // pixels (see writeTransparentPixels in haveDecodedRow). We currently do this by |
| 190 | // first swizzling into temporary memory, then copying into the destination. (We |
| 191 | // let the swizzler handle it first because it may need to sample.) After |
| 192 | // swizzling to 565, we do not know which pixels in our temporary memory |
| 193 | // correspond to the transparent pixel, so we do not know what to skip. We could |
| 194 | // special case the non-sampled case (no need to swizzle), but as this is |
| 195 | // currently unused we can just not support it. |
| 196 | return gif_error("Cannot decode multiframe gif (except frame 0) as 565.\n", |
| 197 | kInvalidConversion); |
| 198 | default: |
| 199 | break; |
| 200 | } |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 201 | } |
| 202 | |
scroggo | f9acbe2 | 2016-10-25 12:43:21 -0700 | [diff] [blame] | 203 | fReader->parse((SkGifImageReader::SkGIFParseQuery) frameIndex); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 204 | |
| 205 | if (frameIndex >= fReader->imagesCount()) { |
| 206 | return gif_error("frame index out of range!\n", kIncompleteInput); |
| 207 | } |
| 208 | |
| 209 | fTmpBuffer.reset(new uint8_t[dstInfo.minRowBytes()]); |
| 210 | |
Leon Scroggins III | fc49b40 | 2016-10-31 14:08:56 -0400 | [diff] [blame^] | 211 | this->initializeColorTable(dstInfo, frameIndex); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 212 | this->initializeSwizzler(dstInfo, frameIndex); |
Leon Scroggins III | fc49b40 | 2016-10-31 14:08:56 -0400 | [diff] [blame^] | 213 | |
| 214 | SkASSERT(fCurrColorTable); |
| 215 | if (inputColorCount) { |
| 216 | *inputColorCount = fCurrColorTable->count(); |
| 217 | } |
| 218 | copy_color_table(dstInfo, fCurrColorTable.get(), inputColorPtr, inputColorCount); |
| 219 | |
msarett | b30d698 | 2016-02-15 10:18:45 -0800 | [diff] [blame] | 220 | return kSuccess; |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 221 | } |
| 222 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 223 | void SkGifCodec::initializeSwizzler(const SkImageInfo& dstInfo, size_t frameIndex) { |
scroggo | f9acbe2 | 2016-10-25 12:43:21 -0700 | [diff] [blame] | 224 | const SkGIFFrameContext* frame = fReader->frameContext(frameIndex); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 225 | // This is only called by prepareToDecode, which ensures frameIndex is in range. |
| 226 | SkASSERT(frame); |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 227 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 228 | const int xBegin = frame->xOffset(); |
| 229 | const int xEnd = std::min(static_cast<int>(frame->xOffset() + frame->width()), |
| 230 | static_cast<int>(fReader->screenWidth())); |
| 231 | |
| 232 | // CreateSwizzler only reads left and right of the frame. We cannot use the frame's raw |
| 233 | // frameRect, since it might extend beyond the edge of the frame. |
| 234 | SkIRect swizzleRect = SkIRect::MakeLTRB(xBegin, 0, xEnd, 0); |
| 235 | |
| 236 | // The default Options should be fine: |
| 237 | // - we'll ignore if the memory is zero initialized - unless we're the first frame, this won't |
| 238 | // matter anyway. |
| 239 | // - subsets are not supported for gif |
| 240 | // - the swizzler does not need to know about the frame. |
| 241 | // We may not be able to use the real Options anyway, since getPixels does not store it (due to |
| 242 | // a bug). |
| 243 | fSwizzler.reset(SkSwizzler::CreateSwizzler(this->getEncodedInfo(), |
| 244 | fCurrColorTable->readColors(), dstInfo, Options(), &swizzleRect)); |
| 245 | SkASSERT(fSwizzler.get()); |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 246 | } |
| 247 | |
| 248 | /* |
| 249 | * Initiates the gif decode |
| 250 | */ |
| 251 | SkCodec::Result SkGifCodec::onGetPixels(const SkImageInfo& dstInfo, |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 252 | void* pixels, size_t dstRowBytes, |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 253 | const Options& opts, |
| 254 | SkPMColor* inputColorPtr, |
msarett | e6dd004 | 2015-10-09 11:07:34 -0700 | [diff] [blame] | 255 | int* inputColorCount, |
| 256 | int* rowsDecoded) { |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 257 | Result result = this->prepareToDecode(dstInfo, inputColorPtr, inputColorCount, opts); |
| 258 | if (kSuccess != result) { |
| 259 | return result; |
| 260 | } |
| 261 | |
| 262 | if (dstInfo.dimensions() != this->getInfo().dimensions()) { |
| 263 | return gif_error("Scaling not supported.\n", kInvalidScale); |
| 264 | } |
| 265 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 266 | fDst = pixels; |
| 267 | fDstRowBytes = dstRowBytes; |
| 268 | |
| 269 | return this->decodeFrame(true, opts, rowsDecoded); |
| 270 | } |
| 271 | |
| 272 | SkCodec::Result SkGifCodec::onStartIncrementalDecode(const SkImageInfo& dstInfo, |
| 273 | void* pixels, size_t dstRowBytes, |
| 274 | const SkCodec::Options& opts, |
| 275 | SkPMColor* inputColorPtr, |
| 276 | int* inputColorCount) { |
| 277 | Result result = this->prepareToDecode(dstInfo, inputColorPtr, inputColorCount, opts); |
| 278 | if (result != kSuccess) { |
| 279 | return result; |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 280 | } |
| 281 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 282 | fDst = pixels; |
| 283 | fDstRowBytes = dstRowBytes; |
| 284 | |
| 285 | fFirstCallToIncrementalDecode = true; |
| 286 | |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 287 | return kSuccess; |
| 288 | } |
| 289 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 290 | SkCodec::Result SkGifCodec::onIncrementalDecode(int* rowsDecoded) { |
| 291 | // It is possible the client has appended more data. Parse, if needed. |
| 292 | const auto& options = this->options(); |
| 293 | const size_t frameIndex = options.fFrameIndex; |
scroggo | f9acbe2 | 2016-10-25 12:43:21 -0700 | [diff] [blame] | 294 | fReader->parse((SkGifImageReader::SkGIFParseQuery) frameIndex); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 295 | |
| 296 | const bool firstCallToIncrementalDecode = fFirstCallToIncrementalDecode; |
| 297 | fFirstCallToIncrementalDecode = false; |
| 298 | return this->decodeFrame(firstCallToIncrementalDecode, options, rowsDecoded); |
msarett | e6dd004 | 2015-10-09 11:07:34 -0700 | [diff] [blame] | 299 | } |
| 300 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 301 | SkCodec::Result SkGifCodec::decodeFrame(bool firstAttempt, const Options& opts, int* rowsDecoded) { |
| 302 | const SkImageInfo& dstInfo = this->dstInfo(); |
| 303 | const size_t frameIndex = opts.fFrameIndex; |
| 304 | SkASSERT(frameIndex < fReader->imagesCount()); |
scroggo | f9acbe2 | 2016-10-25 12:43:21 -0700 | [diff] [blame] | 305 | const SkGIFFrameContext* frameContext = fReader->frameContext(frameIndex); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 306 | if (firstAttempt) { |
| 307 | // rowsDecoded reports how many rows have been initialized, so a layer above |
| 308 | // can fill the rest. In some cases, we fill the background before decoding |
| 309 | // (or it is already filled for us), so we report rowsDecoded to be the full |
| 310 | // height. |
| 311 | bool filledBackground = false; |
| 312 | if (frameContext->getRequiredFrame() == kNone) { |
| 313 | // We may need to clear to transparent for one of the following reasons: |
| 314 | // - The frameRect does not cover the full bounds. haveDecodedRow will |
| 315 | // only draw inside the frameRect, so we need to clear the rest. |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 316 | // - The frame is interlaced. There is no obvious way to fill |
| 317 | // afterwards for an incomplete image. (FIXME: Does the first pass |
| 318 | // cover all rows? If so, we do not have to fill here.) |
scroggo | 8bce117 | 2016-10-25 13:08:40 -0700 | [diff] [blame] | 319 | // - There is no color table for this frame. In that case will not |
| 320 | // draw anything, so we need to fill. |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 321 | if (frameContext->frameRect() != this->getInfo().bounds() |
scroggo | 8bce117 | 2016-10-25 13:08:40 -0700 | [diff] [blame] | 322 | || frameContext->interlaced() || !fCurrColorTableIsReal) { |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 323 | // fill ignores the width (replaces it with the actual, scaled width). |
| 324 | // But we need to scale in Y. |
| 325 | const int scaledHeight = get_scaled_dimension(dstInfo.height(), |
| 326 | fSwizzler->sampleY()); |
| 327 | auto fillInfo = dstInfo.makeWH(0, scaledHeight); |
| 328 | fSwizzler->fill(fillInfo, fDst, fDstRowBytes, this->getFillValue(dstInfo), |
| 329 | opts.fZeroInitialized); |
| 330 | filledBackground = true; |
| 331 | } |
| 332 | } else { |
| 333 | // Not independent |
| 334 | if (!opts.fHasPriorFrame) { |
| 335 | // Decode that frame into pixels. |
| 336 | Options prevFrameOpts(opts); |
| 337 | prevFrameOpts.fFrameIndex = frameContext->getRequiredFrame(); |
| 338 | prevFrameOpts.fHasPriorFrame = false; |
Leon Scroggins III | fc49b40 | 2016-10-31 14:08:56 -0400 | [diff] [blame^] | 339 | // The prior frame may have a different color table, so update it and the |
| 340 | // swizzler. |
| 341 | this->initializeColorTable(dstInfo, prevFrameOpts.fFrameIndex); |
| 342 | this->initializeSwizzler(dstInfo, prevFrameOpts.fFrameIndex); |
| 343 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 344 | const Result prevResult = this->decodeFrame(true, prevFrameOpts, nullptr); |
| 345 | switch (prevResult) { |
| 346 | case kSuccess: |
| 347 | // Prior frame succeeded. Carry on. |
| 348 | break; |
| 349 | case kIncompleteInput: |
| 350 | // Prior frame was incomplete. So this frame cannot be decoded. |
| 351 | return kInvalidInput; |
| 352 | default: |
| 353 | return prevResult; |
| 354 | } |
Leon Scroggins III | fc49b40 | 2016-10-31 14:08:56 -0400 | [diff] [blame^] | 355 | |
| 356 | // Go back to using the correct color table for this frame. |
| 357 | this->initializeColorTable(dstInfo, frameIndex); |
| 358 | this->initializeSwizzler(dstInfo, frameIndex); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 359 | } |
| 360 | const auto* prevFrame = fReader->frameContext(frameContext->getRequiredFrame()); |
| 361 | if (prevFrame->getDisposalMethod() == SkCodecAnimation::RestoreBGColor_DisposalMethod) { |
| 362 | const SkIRect prevRect = prevFrame->frameRect(); |
| 363 | auto left = get_scaled_dimension(prevRect.fLeft, fSwizzler->sampleX()); |
| 364 | auto top = get_scaled_dimension(prevRect.fTop, fSwizzler->sampleY()); |
| 365 | void* const eraseDst = SkTAddOffset<void>(fDst, top * fDstRowBytes |
| 366 | + left * SkColorTypeBytesPerPixel(dstInfo.colorType())); |
| 367 | auto width = get_scaled_dimension(prevRect.width(), fSwizzler->sampleX()); |
| 368 | auto height = get_scaled_dimension(prevRect.height(), fSwizzler->sampleY()); |
| 369 | // fSwizzler->fill() would fill to the scaled width of the frame, but we want to |
| 370 | // fill to the scaled with of the width of the PRIOR frame, so we do all the scaling |
| 371 | // ourselves and call the static version. |
| 372 | SkSampler::Fill(dstInfo.makeWH(width, height), eraseDst, |
| 373 | fDstRowBytes, this->getFillValue(dstInfo), kNo_ZeroInitialized); |
| 374 | } |
| 375 | filledBackground = true; |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 376 | } |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 377 | |
| 378 | fFilledBackground = filledBackground; |
| 379 | if (filledBackground) { |
| 380 | // Report the full (scaled) height, since the client will never need to fill. |
| 381 | fRowsDecoded = get_scaled_dimension(dstInfo.height(), fSwizzler->sampleY()); |
| 382 | } else { |
| 383 | // This will be updated by haveDecodedRow. |
| 384 | fRowsDecoded = 0; |
| 385 | } |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 386 | } |
msarett | e6dd004 | 2015-10-09 11:07:34 -0700 | [diff] [blame] | 387 | |
scroggo | 3d3a65c | 2016-10-24 12:28:30 -0700 | [diff] [blame] | 388 | // Note: there is a difference between the following call to SkGifImageReader::decode |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 389 | // returning false and leaving frameDecoded false: |
| 390 | // - If the method returns false, there was an error in the stream. We still treat this as |
| 391 | // incomplete, since we have already decoded some rows. |
| 392 | // - If frameDecoded is false, that just means that we do not have enough data. If more data |
| 393 | // is supplied, we may be able to continue decoding this frame. We also treat this as |
| 394 | // incomplete. |
| 395 | // FIXME: Ensure that we do not attempt to continue decoding if the method returns false and |
| 396 | // more data is supplied. |
| 397 | bool frameDecoded = false; |
| 398 | if (!fReader->decode(frameIndex, &frameDecoded) || !frameDecoded) { |
| 399 | if (rowsDecoded) { |
| 400 | *rowsDecoded = fRowsDecoded; |
| 401 | } |
| 402 | return kIncompleteInput; |
| 403 | } |
| 404 | |
| 405 | return kSuccess; |
msarett | 10522ff | 2015-09-07 08:54:01 -0700 | [diff] [blame] | 406 | } |
scroggo | 46c5747 | 2015-09-30 08:57:13 -0700 | [diff] [blame] | 407 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 408 | uint64_t SkGifCodec::onGetFillValue(const SkImageInfo& dstInfo) const { |
| 409 | // Note: Using fCurrColorTable relies on having called initializeColorTable already. |
| 410 | // This is (currently) safe because this method is only called when filling, after |
| 411 | // initializeColorTable has been called. |
| 412 | // FIXME: Is there a way to make this less fragile? |
| 413 | if (dstInfo.colorType() == kIndex_8_SkColorType && fCurrColorTableIsReal) { |
| 414 | // We only support index 8 for the first frame, for backwards |
| 415 | // compatibity on Android, so we are using the color table for the first frame. |
| 416 | SkASSERT(this->options().fFrameIndex == 0); |
| 417 | // Use the transparent index for the first frame. |
| 418 | const size_t transPixel = fReader->frameContext(0)->transparentPixel(); |
| 419 | if (transPixel < (size_t) fCurrColorTable->count()) { |
| 420 | return transPixel; |
| 421 | } |
| 422 | // Fall through to return SK_ColorTRANSPARENT (i.e. 0). This choice is arbitrary, |
| 423 | // but we have to pick something inside the color table, and this one is as good |
| 424 | // as any. |
| 425 | } |
| 426 | // Using transparent as the fill value matches the behavior in Chromium, |
| 427 | // which ignores the background color. |
| 428 | // If the colorType is kIndex_8, and there was no color table (i.e. |
| 429 | // fCurrColorTableIsReal is false), this value (zero) corresponds to the |
| 430 | // only entry in the dummy color table provided to the client. |
| 431 | return SK_ColorTRANSPARENT; |
| 432 | } |
msarett | 72261c0 | 2015-11-19 15:29:26 -0800 | [diff] [blame] | 433 | |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 434 | bool SkGifCodec::haveDecodedRow(size_t frameIndex, const unsigned char* rowBegin, |
| 435 | size_t rowNumber, unsigned repeatCount, bool writeTransparentPixels) |
| 436 | { |
scroggo | f9acbe2 | 2016-10-25 12:43:21 -0700 | [diff] [blame] | 437 | const SkGIFFrameContext* frameContext = fReader->frameContext(frameIndex); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 438 | // The pixel data and coordinates supplied to us are relative to the frame's |
| 439 | // origin within the entire image size, i.e. |
| 440 | // (frameContext->xOffset, frameContext->yOffset). There is no guarantee |
| 441 | // that width == (size().width() - frameContext->xOffset), so |
| 442 | // we must ensure we don't run off the end of either the source data or the |
| 443 | // row's X-coordinates. |
| 444 | const size_t width = frameContext->width(); |
| 445 | const int xBegin = frameContext->xOffset(); |
| 446 | const int yBegin = frameContext->yOffset() + rowNumber; |
| 447 | const int xEnd = std::min(static_cast<int>(frameContext->xOffset() + width), |
| 448 | this->getInfo().width()); |
| 449 | const int yEnd = std::min(static_cast<int>(frameContext->yOffset() + rowNumber + repeatCount), |
| 450 | this->getInfo().height()); |
| 451 | // FIXME: No need to make the checks on width/xBegin/xEnd for every row. We could instead do |
| 452 | // this once in prepareToDecode. |
| 453 | if (!width || (xBegin < 0) || (yBegin < 0) || (xEnd <= xBegin) || (yEnd <= yBegin)) |
| 454 | return true; |
| 455 | |
| 456 | // yBegin is the first row in the non-sampled image. dstRow will be the row in the output, |
| 457 | // after potentially scaling it. |
| 458 | int dstRow = yBegin; |
| 459 | |
| 460 | const int sampleY = fSwizzler->sampleY(); |
| 461 | if (sampleY > 1) { |
| 462 | // Check to see whether this row or one that falls in the repeatCount is needed in the |
| 463 | // output. |
| 464 | bool foundNecessaryRow = false; |
| 465 | for (unsigned i = 0; i < repeatCount; i++) { |
| 466 | const int potentialRow = yBegin + i; |
| 467 | if (fSwizzler->rowNeeded(potentialRow)) { |
| 468 | dstRow = potentialRow / sampleY; |
| 469 | const int scaledHeight = get_scaled_dimension(this->dstInfo().height(), sampleY); |
| 470 | if (dstRow >= scaledHeight) { |
| 471 | return true; |
| 472 | } |
| 473 | |
| 474 | foundNecessaryRow = true; |
| 475 | repeatCount -= i; |
| 476 | |
| 477 | repeatCount = (repeatCount - 1) / sampleY + 1; |
| 478 | |
| 479 | // Make sure the repeatCount does not take us beyond the end of the dst |
| 480 | if (dstRow + (int) repeatCount > scaledHeight) { |
| 481 | repeatCount = scaledHeight - dstRow; |
| 482 | SkASSERT(repeatCount >= 1); |
| 483 | } |
| 484 | break; |
| 485 | } |
| 486 | } |
| 487 | |
| 488 | if (!foundNecessaryRow) { |
| 489 | return true; |
| 490 | } |
Matt Sarett | 8a4e9c5 | 2016-10-25 14:24:50 -0400 | [diff] [blame] | 491 | } else { |
| 492 | // Make sure the repeatCount does not take us beyond the end of the dst |
| 493 | SkASSERT(this->dstInfo().height() >= yBegin); |
| 494 | repeatCount = SkTMin(repeatCount, (unsigned) (this->dstInfo().height() - yBegin)); |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 495 | } |
| 496 | |
| 497 | if (!fFilledBackground) { |
| 498 | // At this point, we are definitely going to write the row, so count it towards the number |
| 499 | // of rows decoded. |
| 500 | // We do not consider the repeatCount, which only happens for interlaced, in which case we |
| 501 | // have already set fRowsDecoded to the proper value (reflecting that we have filled the |
| 502 | // background). |
| 503 | fRowsDecoded++; |
| 504 | } |
| 505 | |
| 506 | if (!fCurrColorTableIsReal) { |
| 507 | // No color table, so nothing to draw this frame. |
| 508 | // FIXME: We can abort even earlier - no need to decode this frame. |
| 509 | return true; |
| 510 | } |
| 511 | |
| 512 | // The swizzler takes care of offsetting into the dst width-wise. |
| 513 | void* dstLine = SkTAddOffset<void>(fDst, dstRow * fDstRowBytes); |
| 514 | |
| 515 | // We may or may not need to write transparent pixels to the buffer. |
scroggo | 1285f41 | 2016-10-26 13:48:03 -0700 | [diff] [blame] | 516 | // If we're compositing against a previous image, it's wrong, but if |
| 517 | // we're decoding an interlaced gif and displaying it "Haeberli"-style, |
| 518 | // we must write these for passes beyond the first, or the initial passes |
| 519 | // will "show through" the later ones. |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 520 | const auto dstInfo = this->dstInfo(); |
scroggo | 53f63b6 | 2016-10-27 08:29:13 -0700 | [diff] [blame] | 521 | if (writeTransparentPixels) { |
scroggo | 19b9153 | 2016-10-24 09:03:26 -0700 | [diff] [blame] | 522 | fSwizzler->swizzle(dstLine, rowBegin); |
| 523 | } else { |
| 524 | // We cannot swizzle directly into the dst, since that will write the transparent pixels. |
| 525 | // Instead, swizzle into a temporary buffer, and copy that into the dst. |
| 526 | { |
| 527 | void* const memsetDst = fTmpBuffer.get(); |
| 528 | // Although onGetFillValue returns a uint64_t, we only use the low eight bits. The |
| 529 | // return value is either an 8 bit index (for index8) or SK_ColorTRANSPARENT, which is |
| 530 | // all zeroes. |
| 531 | const int fillValue = (uint8_t) this->onGetFillValue(dstInfo); |
| 532 | const size_t rb = dstInfo.minRowBytes(); |
| 533 | if (fillValue == 0) { |
| 534 | // FIXME: This special case should be unnecessary, and in fact sk_bzero just calls |
| 535 | // memset. But without it, the compiler thinks this is trying to pass a zero length |
| 536 | // to memset, causing an error. |
| 537 | sk_bzero(memsetDst, rb); |
| 538 | } else { |
| 539 | memset(memsetDst, fillValue, rb); |
| 540 | } |
| 541 | } |
| 542 | fSwizzler->swizzle(fTmpBuffer.get(), rowBegin); |
| 543 | |
| 544 | const size_t offsetBytes = fSwizzler->swizzleOffsetBytes(); |
| 545 | switch (dstInfo.colorType()) { |
| 546 | case kBGRA_8888_SkColorType: |
| 547 | case kRGBA_8888_SkColorType: { |
| 548 | uint32_t* dstPixel = SkTAddOffset<uint32_t>(dstLine, offsetBytes); |
| 549 | uint32_t* srcPixel = SkTAddOffset<uint32_t>(fTmpBuffer.get(), offsetBytes); |
| 550 | for (int i = 0; i < fSwizzler->swizzleWidth(); i++) { |
| 551 | // Technically SK_ColorTRANSPARENT is an SkPMColor, and srcPixel would have |
| 552 | // the opposite swizzle for the non-native swizzle, but TRANSPARENT is all |
| 553 | // zeroes, which is the same either way. |
| 554 | if (*srcPixel != SK_ColorTRANSPARENT) { |
| 555 | *dstPixel = *srcPixel; |
| 556 | } |
| 557 | dstPixel++; |
| 558 | srcPixel++; |
| 559 | } |
| 560 | break; |
| 561 | } |
| 562 | case kIndex_8_SkColorType: { |
| 563 | uint8_t* dstPixel = SkTAddOffset<uint8_t>(dstLine, offsetBytes); |
| 564 | uint8_t* srcPixel = SkTAddOffset<uint8_t>(fTmpBuffer.get(), offsetBytes); |
| 565 | for (int i = 0; i < fSwizzler->swizzleWidth(); i++) { |
| 566 | if (*srcPixel != frameContext->transparentPixel()) { |
| 567 | *dstPixel = *srcPixel; |
| 568 | } |
| 569 | dstPixel++; |
| 570 | srcPixel++; |
| 571 | } |
| 572 | break; |
| 573 | } |
| 574 | default: |
| 575 | SkASSERT(false); |
| 576 | break; |
| 577 | } |
| 578 | } |
| 579 | |
| 580 | // Tell the frame to copy the row data if need be. |
| 581 | if (repeatCount > 1) { |
| 582 | const size_t bytesPerPixel = SkColorTypeBytesPerPixel(this->dstInfo().colorType()); |
| 583 | const size_t bytesToCopy = fSwizzler->swizzleWidth() * bytesPerPixel; |
| 584 | void* copiedLine = SkTAddOffset<void>(dstLine, fSwizzler->swizzleOffsetBytes()); |
| 585 | void* dst = copiedLine; |
| 586 | for (unsigned i = 1; i < repeatCount; i++) { |
| 587 | dst = SkTAddOffset<void>(dst, fDstRowBytes); |
| 588 | memcpy(dst, copiedLine, bytesToCopy); |
msarett | 72261c0 | 2015-11-19 15:29:26 -0800 | [diff] [blame] | 589 | } |
| 590 | } |
| 591 | |
| 592 | return true; |
| 593 | } |