| 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 "SkCodec.h" | 8 #include "SkCodec.h" | 
| 9 #include "SkColorSpaceXform.h" | 9 #include "SkColorSpaceXform.h" | 
| 10 #include "SkColorTable.h" | 10 #include "SkColorTable.h" | 
| 11 #include "SkPngChunkReader.h" | 11 #include "SkPngChunkReader.h" | 
| 12 #include "SkEncodedFormat.h" | 12 #include "SkEncodedFormat.h" | 
| 13 #include "SkImageInfo.h" | 13 #include "SkImageInfo.h" | 
| 14 #include "SkRefCnt.h" | 14 #include "SkRefCnt.h" | 
| 15 #include "SkSwizzler.h" | 15 #include "SkSwizzler.h" | 
| 16 | 16 | 
| 17 // Instead of including png.h here, forward declare the few types we refer to. |  | 
| 18 struct png_struct_def; |  | 
| 19 struct png_info_def; |  | 
| 20 typedef png_struct_def png_struct; |  | 
| 21 typedef png_info_def png_info; |  | 
| 22 |  | 
| 23 class SkStream; | 17 class SkStream; | 
| 24 | 18 | 
| 25 class SkPngCodec : public SkCodec { | 19 class SkPngCodec : public SkCodec { | 
| 26 public: | 20 public: | 
| 27     static bool IsPng(const char*, size_t); | 21     static bool IsPng(const char*, size_t); | 
| 28 | 22 | 
| 29     // Assume IsPng was called and returned true. | 23     // Assume IsPng was called and returned true. | 
| 30     static SkCodec* NewFromStream(SkStream*, SkPngChunkReader* = NULL); | 24     static SkCodec* NewFromStream(SkStream*, SkPngChunkReader* = NULL); | 
| 31 | 25 | 
| 32     virtual ~SkPngCodec(); | 26     virtual ~SkPngCodec(); | 
| 33 | 27 | 
| 34 protected: | 28 protected: | 
|  | 29     // We hold the png_ptr and info_ptr as voidp to avoid having to include png.
    h | 
|  | 30     // or forward declare their types here.  voidp auto-casts to the real pointe
    r types. | 
|  | 31     struct voidp { | 
|  | 32         voidp(void* ptr) : fPtr(ptr) {} | 
|  | 33 | 
|  | 34         template <typename T> | 
|  | 35         operator T*() const { return (T*)fPtr; } | 
|  | 36 | 
|  | 37         explicit operator bool() const { return fPtr != nullptr; } | 
|  | 38 | 
|  | 39         void* fPtr; | 
|  | 40     }; | 
|  | 41 | 
| 35     Result onGetPixels(const SkImageInfo&, void*, size_t, const Options&, SkPMCo
    lor*, int*, int*) | 42     Result onGetPixels(const SkImageInfo&, void*, size_t, const Options&, SkPMCo
    lor*, int*, int*) | 
| 36             override; | 43             override; | 
| 37     SkEncodedFormat onGetEncodedFormat() const override { return kPNG_SkEncodedF
    ormat; } | 44     SkEncodedFormat onGetEncodedFormat() const override { return kPNG_SkEncodedF
    ormat; } | 
| 38     bool onRewind() override; | 45     bool onRewind() override; | 
| 39     uint32_t onGetFillValue(SkColorType) const override; | 46     uint32_t onGetFillValue(SkColorType) const override; | 
| 40 | 47 | 
| 41     // Helper to set up swizzler, color xforms, and color table. Also calls png_
    read_update_info. | 48     // Helper to set up swizzler, color xforms, and color table. Also calls png_
    read_update_info. | 
| 42     bool initializeXforms(const SkImageInfo& requestedInfo, const Options&, SkPM
    Color* colorPtr, | 49     bool initializeXforms(const SkImageInfo& requestedInfo, const Options&, SkPM
    Color* colorPtr, | 
| 43                           int* colorCount); | 50                           int* colorCount); | 
| 44     SkSampler* getSampler(bool createIfNecessary) override { | 51     SkSampler* getSampler(bool createIfNecessary) override { | 
| 45         SkASSERT(fSwizzler); | 52         SkASSERT(fSwizzler); | 
| 46         return fSwizzler; | 53         return fSwizzler; | 
| 47     } | 54     } | 
| 48     void allocateStorage(); | 55     void allocateStorage(); | 
| 49 | 56 | 
| 50     virtual int readRows(const SkImageInfo& dstInfo, void* dst, size_t rowBytes,
     int count, | 57     virtual int readRows(const SkImageInfo& dstInfo, void* dst, size_t rowBytes,
     int count, | 
| 51                          int startRow) = 0; | 58                          int startRow) = 0; | 
| 52 | 59 | 
| 53     SkPngCodec(const SkEncodedInfo&, const SkImageInfo&, SkStream*, SkPngChunkRe
    ader*, | 60     SkPngCodec(const SkEncodedInfo&, const SkImageInfo&, SkStream*, SkPngChunkRe
    ader*, | 
| 54                png_struct*, png_info*, int, int); | 61                void* png_ptr, void* info_ptr, int, int); | 
| 55 | 62 | 
| 56     SkAutoTUnref<SkPngChunkReader>     fPngChunkReader; | 63     SkAutoTUnref<SkPngChunkReader> fPngChunkReader; | 
| 57     png_struct*                        fPng_ptr; | 64     voidp                          fPng_ptr; | 
| 58     png_info*                          fInfo_ptr; | 65     voidp                          fInfo_ptr; | 
| 59 | 66 | 
| 60     // These are stored here so they can be used both by normal decoding and sca
    nline decoding. | 67     // These are stored here so they can be used both by normal decoding and sca
    nline decoding. | 
| 61     SkAutoTUnref<SkColorTable>         fColorTable;    // May be unpremul. | 68     SkAutoTUnref<SkColorTable>         fColorTable;    // May be unpremul. | 
| 62     SkAutoTDelete<SkSwizzler>          fSwizzler; | 69     SkAutoTDelete<SkSwizzler>          fSwizzler; | 
| 63     std::unique_ptr<SkColorSpaceXform> fColorXform; | 70     std::unique_ptr<SkColorSpaceXform> fColorXform; | 
| 64     SkAutoTMalloc<uint8_t>             fStorage; | 71     SkAutoTMalloc<uint8_t>             fStorage; | 
| 65     uint8_t*                           fSwizzlerSrcRow; | 72     uint8_t*                           fSwizzlerSrcRow; | 
| 66     uint32_t*                          fColorXformSrcRow; | 73     uint32_t*                          fColorXformSrcRow; | 
| 67     size_t                             fSrcRowBytes; | 74     size_t                             fSrcRowBytes; | 
| 68 | 75 | 
| 69     const int                          fNumberPasses; | 76     const int                          fNumberPasses; | 
| 70     int                                fBitDepth; | 77     int                                fBitDepth; | 
| 71 | 78 | 
| 72 private: | 79 private: | 
| 73     bool createColorTable(const SkImageInfo& dstInfo, int* ctableCount); | 80     bool createColorTable(const SkImageInfo& dstInfo, int* ctableCount); | 
| 74     void destroyReadStruct(); | 81     void destroyReadStruct(); | 
| 75 | 82 | 
| 76     typedef SkCodec INHERITED; | 83     typedef SkCodec INHERITED; | 
| 77 }; | 84 }; | 
| OLD | NEW | 
|---|