scroggo | 6f5e619 | 2015-06-18 12:53:43 -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 | |
| 8 | #ifndef SkWebpCodec_DEFINED |
| 9 | #define SkWebpCodec_DEFINED |
| 10 | |
| 11 | #include "SkCodec.h" |
msarett | ad8bcfe | 2016-03-07 07:09:03 -0800 | [diff] [blame] | 12 | #include "SkColorSpace.h" |
Hal Canary | db68301 | 2016-11-23 08:55:18 -0700 | [diff] [blame] | 13 | #include "SkEncodedImageFormat.h" |
Leon Scroggins III | 557fbbe | 2017-05-23 09:37:21 -0400 | [diff] [blame] | 14 | #include "SkFrameHolder.h" |
scroggo | 6f5e619 | 2015-06-18 12:53:43 -0700 | [diff] [blame] | 15 | #include "SkImageInfo.h" |
| 16 | #include "SkTypes.h" |
| 17 | |
Leon Scroggins III | 557fbbe | 2017-05-23 09:37:21 -0400 | [diff] [blame] | 18 | #include <vector> |
| 19 | |
scroggo | 6f5e619 | 2015-06-18 12:53:43 -0700 | [diff] [blame] | 20 | class SkStream; |
msarett | ff2a6c8 | 2016-09-07 11:23:28 -0700 | [diff] [blame] | 21 | extern "C" { |
| 22 | struct WebPDemuxer; |
| 23 | void WebPDemuxDelete(WebPDemuxer* dmux); |
| 24 | } |
scroggo | 6f5e619 | 2015-06-18 12:53:43 -0700 | [diff] [blame] | 25 | |
| 26 | class SkWebpCodec final : public SkCodec { |
| 27 | public: |
| 28 | // Assumes IsWebp was called and returned true. |
Mike Reed | ede7bac | 2017-07-23 15:30:02 -0400 | [diff] [blame] | 29 | static std::unique_ptr<SkCodec> MakeFromStream(std::unique_ptr<SkStream>, Result*); |
scroggo | db30be2 | 2015-12-08 18:54:13 -0800 | [diff] [blame] | 30 | static bool IsWebp(const void*, size_t); |
scroggo | 6f5e619 | 2015-06-18 12:53:43 -0700 | [diff] [blame] | 31 | protected: |
Leon Scroggins | 571b30f | 2017-07-11 17:35:31 +0000 | [diff] [blame] | 32 | Result onGetPixels(const SkImageInfo&, void*, size_t, const Options&, int*) override; |
Hal Canary | db68301 | 2016-11-23 08:55:18 -0700 | [diff] [blame] | 33 | SkEncodedImageFormat onGetEncodedFormat() const override { return SkEncodedImageFormat::kWEBP; } |
scroggo | 6f5e619 | 2015-06-18 12:53:43 -0700 | [diff] [blame] | 34 | |
scroggo | 6f5e619 | 2015-06-18 12:53:43 -0700 | [diff] [blame] | 35 | SkISize onGetScaledDimensions(float desiredScale) const override; |
scroggo | b636b45 | 2015-07-22 07:16:20 -0700 | [diff] [blame] | 36 | |
scroggo | e7fc14b | 2015-10-02 13:14:46 -0700 | [diff] [blame] | 37 | bool onDimensionsSupported(const SkISize&) override; |
| 38 | |
scroggo | b636b45 | 2015-07-22 07:16:20 -0700 | [diff] [blame] | 39 | bool onGetValidSubset(SkIRect* /* desiredSubset */) const override; |
Leon Scroggins III | 557fbbe | 2017-05-23 09:37:21 -0400 | [diff] [blame] | 40 | |
| 41 | int onGetFrameCount() override; |
| 42 | bool onGetFrameInfo(int, FrameInfo*) const override; |
| 43 | int onGetRepetitionCount() override; |
| 44 | |
Leon Scroggins III | 1f6af6b | 2017-06-12 16:41:09 -0400 | [diff] [blame] | 45 | const SkFrameHolder* getFrameHolder() const override { |
| 46 | return &fFrameHolder; |
| 47 | } |
| 48 | |
scroggo | 6f5e619 | 2015-06-18 12:53:43 -0700 | [diff] [blame] | 49 | private: |
Mike Reed | ede7bac | 2017-07-23 15:30:02 -0400 | [diff] [blame] | 50 | SkWebpCodec(int width, int height, const SkEncodedInfo&, sk_sp<SkColorSpace>, |
| 51 | std::unique_ptr<SkStream>, WebPDemuxer*, sk_sp<SkData>); |
msarett | ff2a6c8 | 2016-09-07 11:23:28 -0700 | [diff] [blame] | 52 | |
| 53 | SkAutoTCallVProc<WebPDemuxer, WebPDemuxDelete> fDemux; |
| 54 | |
| 55 | // fDemux has a pointer into this data. |
| 56 | // This should not be freed until the decode is completed. |
| 57 | sk_sp<SkData> fData; |
scroggo | 6f5e619 | 2015-06-18 12:53:43 -0700 | [diff] [blame] | 58 | |
Leon Scroggins III | 557fbbe | 2017-05-23 09:37:21 -0400 | [diff] [blame] | 59 | class Frame : public SkFrame { |
| 60 | public: |
| 61 | Frame(int i, bool alpha) |
| 62 | : INHERITED(i) |
| 63 | , fReportsAlpha(alpha) |
| 64 | {} |
| 65 | Frame(Frame&& other) |
| 66 | : INHERITED(other.frameId()) |
| 67 | , fReportsAlpha(other.fReportsAlpha) |
| 68 | {} |
| 69 | |
| 70 | protected: |
| 71 | bool onReportsAlpha() const override { |
| 72 | return fReportsAlpha; |
| 73 | } |
| 74 | |
| 75 | private: |
| 76 | const bool fReportsAlpha; |
| 77 | |
| 78 | typedef SkFrame INHERITED; |
| 79 | }; |
| 80 | |
| 81 | class FrameHolder : public SkFrameHolder { |
| 82 | public: |
| 83 | ~FrameHolder() override {} |
| 84 | void setScreenSize(int w, int h) { |
| 85 | fScreenWidth = w; |
| 86 | fScreenHeight = h; |
| 87 | } |
| 88 | Frame* appendNewFrame(bool hasAlpha); |
| 89 | const Frame* frame(int i) const; |
| 90 | int size() const { |
| 91 | return static_cast<int>(fFrames.size()); |
| 92 | } |
| 93 | void reserve(int size) { |
| 94 | fFrames.reserve(size); |
| 95 | } |
| 96 | |
| 97 | protected: |
| 98 | const SkFrame* onGetFrame(int i) const override; |
| 99 | |
| 100 | private: |
| 101 | std::vector<Frame> fFrames; |
| 102 | }; |
| 103 | |
| 104 | FrameHolder fFrameHolder; |
| 105 | // Set to true if WebPDemuxGetFrame fails. This only means |
| 106 | // that we will cap the frame count to the frames that |
| 107 | // succeed. |
| 108 | bool fFailed; |
| 109 | |
scroggo | 6f5e619 | 2015-06-18 12:53:43 -0700 | [diff] [blame] | 110 | typedef SkCodec INHERITED; |
| 111 | }; |
| 112 | #endif // SkWebpCodec_DEFINED |