Chromium Code Reviews| Index: third_party/WebKit/Source/platform/image-decoders/gif/GIFImageDecoder.h |
| diff --git a/third_party/WebKit/Source/platform/image-decoders/gif/GIFImageDecoder.h b/third_party/WebKit/Source/platform/image-decoders/gif/GIFImageDecoder.h |
| index e84b1b488dc443a73ab341e658cac98e49b771a7..dc2739023a306bbb3c701a6575293b0fa8e21e4d 100644 |
| --- a/third_party/WebKit/Source/platform/image-decoders/gif/GIFImageDecoder.h |
| +++ b/third_party/WebKit/Source/platform/image-decoders/gif/GIFImageDecoder.h |
| @@ -26,13 +26,13 @@ |
| #ifndef GIFImageDecoder_h |
| #define GIFImageDecoder_h |
| +#include <algorithm> |
| +#include <memory> |
| #include "platform/image-decoders/ImageDecoder.h" |
| +#include "third_party/skia/include/codec/SkCodec.h" |
| +#include "third_party/skia/include/core/SkStream.h" |
| #include "wtf/Noncopyable.h" |
| -#include <memory> |
| - |
| -class GIFImageReader; |
| - |
| -typedef Vector<unsigned char> GIFRow; |
| +#include "wtf/RefPtr.h" |
| namespace blink { |
| @@ -44,56 +44,123 @@ class PLATFORM_EXPORT GIFImageDecoder final : public ImageDecoder { |
| GIFImageDecoder(AlphaOption, const ColorBehavior&, size_t maxDecodedBytes); |
| ~GIFImageDecoder() override; |
| - enum GIFParseQuery { GIFSizeQuery, GIFFrameCountQuery }; |
| - |
| // ImageDecoder: |
| String filenameExtension() const override { return "gif"; } |
| void onSetData(SegmentReader* data) override; |
| int repetitionCount() const override; |
| bool frameIsCompleteAtIndex(size_t) const override; |
| float frameDurationAtIndex(size_t) const override; |
| - // CAUTION: setFailed() deletes |m_reader|. Be careful to avoid |
| - // accessing deleted memory, especially when calling this from inside |
| - // GIFImageReader! |
| - bool setFailed() override; |
| - |
| - // Callbacks from the GIF reader. |
| - bool haveDecodedRow(size_t frameIndex, |
| - GIFRow::const_iterator rowBegin, |
| - size_t width, |
| - size_t rowNumber, |
| - unsigned repeatCount, |
| - bool writeTransparentPixels); |
| - bool frameComplete(size_t frameIndex); |
| - |
| - // For testing. |
| - bool parseCompleted() const; |
| private: |
| // ImageDecoder: |
| - void clearFrameBuffer(size_t frameIndex) override; |
| - virtual void decodeSize() { parse(GIFSizeQuery); } |
| + void decodeSize() override {} |
| size_t decodeFrameCount() override; |
| - void initializeNewFrame(size_t) override; |
| void decode(size_t) override; |
| - |
| - // Parses as much as is needed to answer the query, ignoring bitmap |
| - // data. If parsing fails, sets the "decode failure" flag. |
| - void parse(GIFParseQuery); |
| - |
| - // Reset the alpha tracker for this frame. Before calling this method, the |
| - // caller must verify that the frame exists. |
| - void onInitFrameBuffer(size_t) override; |
| - |
| + bool frameStatusSufficientForSuccessors(size_t index) { |
| + DCHECK(index < m_frameBufferCache.size()); |
| + return m_frameBufferCache[index].getStatus() == ImageFrame::FrameComplete; |
| + } |
| // When the disposal method of the frame is DisposeOverWritePrevious, the |
| // next frame will use the previous frame's buffer as its starting state, so |
| // we can't take over the data in that case. Before calling this method, the |
| // caller must verify that the frame exists. |
| bool canReusePreviousFrameBuffer(size_t) const override; |
| - bool m_currentBufferSawAlpha; |
| - mutable int m_repetitionCount; |
| - std::unique_ptr<GIFImageReader> m_reader; |
| + class SegmentStream : public SkStream { |
| + public: |
| + SegmentStream() : m_reader(), m_position(0), m_hasReadAllContents(true) {} |
| + |
| + void setReader(SegmentReader* reader, bool allContentsReceived) { |
| + m_reader = reader; |
| + m_hasReadAllContents = reader->size() == m_position; |
| + } |
| + |
| + size_t read(void* dst, size_t len) { |
| + len = std::min(len, m_reader->size()); |
|
scroggo_chromium
2017/02/24 19:04:30
I think the second parameter here should be
m_r
cblume
2017/03/05 19:17:42
Done.
cblume
2017/03/06 08:30:52
I also updated this in peek().
|
| + |
| + size_t bytesAdvanced = 0; |
| + if (!dst) { // skipping, not reading |
| + bytesAdvanced = len; |
| + } else { |
| + bytesAdvanced = peek(dst, len); |
| + } |
| + |
| + m_position += bytesAdvanced; |
| + m_hasReadAllContents = m_position == m_reader->size(); |
| + |
| + return bytesAdvanced; |
| + } |
| + |
| + size_t peek(void* buffer, size_t size) const override { |
| + size = std::min(size, m_reader->size()); |
| + |
| + size_t peekPosition = m_position; |
| + size_t totalBytesPeeked = 0; |
| + char* bufferAsCharPtr = reinterpret_cast<char*>(buffer); |
| + while (size) { |
| + const char* segment = nullptr; |
| + size_t bytesPeeked = m_reader->getSomeData(segment, peekPosition); |
| + if (!bytesPeeked) { |
| + break; |
| + } |
| + if (bytesPeeked > size) { |
| + bytesPeeked = size; |
| + } |
| + |
| + memcpy(bufferAsCharPtr, segment, bytesPeeked); |
| + bufferAsCharPtr += bytesPeeked; |
| + size -= bytesPeeked; |
| + totalBytesPeeked += bytesPeeked; |
| + peekPosition += bytesPeeked; |
| + } |
| + |
| + return totalBytesPeeked; |
| + } |
| + |
| + bool isAtEnd() const override { return m_hasReadAllContents; } |
| + |
| + bool rewind() override { |
| + m_position = 0; |
| + return true; |
| + } |
| + |
| + bool hasPosition() const override { return true; } |
| + size_t getPosition() const override { return m_position; } |
| + |
| + bool seek(size_t position) override { |
| + position = std::min(position, m_reader->size()); |
| + m_position = position; |
| + |
| + return true; |
| + } |
| + |
| + bool move(long offset) override { |
|
scroggo_chromium
2017/02/24 19:04:30
Need to update m_hasReadAllContents
cblume
2017/03/05 19:17:42
Done.
|
| + long absolutePosition = m_position + offset; |
| + |
| + // clamp inside the bounds of the buffer size |
| + absolutePosition = std::max(absolutePosition, 0l); |
| + absolutePosition = |
| + std::min(static_cast<size_t>(absolutePosition), m_reader->size()); |
| + |
| + m_position = absolutePosition; |
| + |
| + return true; |
| + } |
| + |
| + bool hasLength() const override { return true; } |
| + size_t getLength() const override { return m_reader->size(); } |
| + |
| + private: |
| + WTF::RefPtr<SegmentReader> m_reader; |
| + size_t m_position; |
| + bool m_hasReadAllContents; |
| + }; |
| + |
| + std::unique_ptr<SkCodec> m_codec; |
| + SegmentStream* m_segmentStream; |
| + // m_segmentStream is a raw pointer because it passes ownership to |
|
scroggo_chromium
2017/02/24 19:04:30
Move this comment above the line it applies to?
cblume
2017/03/05 19:17:42
Done.
|
| + // m_codec when m_codec is created. However, we still need the |
| + // reference so we can append more data as it arrives. |
| }; |
| } // namespace blink |