Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(128)

Side by Side Diff: src/images/SkImageDecoder_astc.cpp

Issue 647023006: Qualify the return value of SkImageDecoder::decode (Closed) Base URL: https://skia.googlesource.com/skia.git/+/master
Patch Set: fix a sample Created 6 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « src/images/SkImageDecoder.cpp ('k') | src/images/SkImageDecoder_ktx.cpp » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 /* 1 /*
2 * Copyright 2014 Google Inc. 2 * Copyright 2014 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 "SkEndian.h" 8 #include "SkEndian.h"
9 #include "SkColorPriv.h" 9 #include "SkColorPriv.h"
10 #include "SkImageDecoder.h" 10 #include "SkImageDecoder.h"
11 #include "SkScaledBitmapSampler.h" 11 #include "SkScaledBitmapSampler.h"
12 #include "SkStream.h" 12 #include "SkStream.h"
13 #include "SkStreamPriv.h" 13 #include "SkStreamPriv.h"
14 #include "SkTypes.h" 14 #include "SkTypes.h"
15 15
16 #include "SkTextureCompressor.h" 16 #include "SkTextureCompressor.h"
17 17
18 class SkASTCImageDecoder : public SkImageDecoder { 18 class SkASTCImageDecoder : public SkImageDecoder {
19 public: 19 public:
20 SkASTCImageDecoder() { } 20 SkASTCImageDecoder() { }
21 21
22 virtual Format getFormat() const SK_OVERRIDE { 22 virtual Format getFormat() const SK_OVERRIDE {
23 return kASTC_Format; 23 return kASTC_Format;
24 } 24 }
25 25
26 protected: 26 protected:
27 virtual bool onDecode(SkStream* stream, SkBitmap* bm, Mode) SK_OVERRIDE; 27 virtual Result onDecode(SkStream* stream, SkBitmap* bm, Mode) SK_OVERRIDE;
28 28
29 private: 29 private:
30 typedef SkImageDecoder INHERITED; 30 typedef SkImageDecoder INHERITED;
31 }; 31 };
32 32
33 //////////////////////////////////////////////////////////////////////////////// ///////// 33 //////////////////////////////////////////////////////////////////////////////// /////////
34 34
35 static const uint32_t kASTCMagicNumber = 0x5CA1AB13; 35 static const uint32_t kASTCMagicNumber = 0x5CA1AB13;
36 36
37 static inline int read_24bit(const uint8_t* buf) { 37 static inline int read_24bit(const uint8_t* buf) {
38 // Assume everything is little endian... 38 // Assume everything is little endian...
39 return 39 return
40 static_cast<int>(buf[0]) | 40 static_cast<int>(buf[0]) |
41 (static_cast<int>(buf[1]) << 8) | 41 (static_cast<int>(buf[1]) << 8) |
42 (static_cast<int>(buf[2]) << 16); 42 (static_cast<int>(buf[2]) << 16);
43 } 43 }
44 44
45 bool SkASTCImageDecoder::onDecode(SkStream* stream, SkBitmap* bm, Mode mode) { 45 SkImageDecoder::Result SkASTCImageDecoder::onDecode(SkStream* stream, SkBitmap* bm, Mode mode) {
46 SkAutoMalloc autoMal; 46 SkAutoMalloc autoMal;
47 const size_t length = SkCopyStreamToStorage(&autoMal, stream); 47 const size_t length = SkCopyStreamToStorage(&autoMal, stream);
48 if (0 == length) { 48 if (0 == length) {
49 return false; 49 return kFailure;
50 } 50 }
51 51
52 unsigned char* buf = (unsigned char*)autoMal.get(); 52 unsigned char* buf = (unsigned char*)autoMal.get();
53 53
54 // Make sure that the magic header is there... 54 // Make sure that the magic header is there...
55 SkASSERT(SkEndian_SwapLE32(*(reinterpret_cast<uint32_t*>(buf))) == kASTCMagi cNumber); 55 SkASSERT(SkEndian_SwapLE32(*(reinterpret_cast<uint32_t*>(buf))) == kASTCMagi cNumber);
56 56
57 // Advance past the magic header 57 // Advance past the magic header
58 buf += 4; 58 buf += 4;
59 59
60 const int blockDimX = buf[0]; 60 const int blockDimX = buf[0];
61 const int blockDimY = buf[1]; 61 const int blockDimY = buf[1];
62 const int blockDimZ = buf[2]; 62 const int blockDimZ = buf[2];
63 63
64 if (1 != blockDimZ) { 64 if (1 != blockDimZ) {
65 // We don't support decoding 3D 65 // We don't support decoding 3D
66 return false; 66 return kFailure;
67 } 67 }
68 68
69 // Choose the proper ASTC format 69 // Choose the proper ASTC format
70 SkTextureCompressor::Format astcFormat; 70 SkTextureCompressor::Format astcFormat;
71 if (4 == blockDimX && 4 == blockDimY) { 71 if (4 == blockDimX && 4 == blockDimY) {
72 astcFormat = SkTextureCompressor::kASTC_4x4_Format; 72 astcFormat = SkTextureCompressor::kASTC_4x4_Format;
73 } else if (5 == blockDimX && 4 == blockDimY) { 73 } else if (5 == blockDimX && 4 == blockDimY) {
74 astcFormat = SkTextureCompressor::kASTC_5x4_Format; 74 astcFormat = SkTextureCompressor::kASTC_5x4_Format;
75 } else if (5 == blockDimX && 5 == blockDimY) { 75 } else if (5 == blockDimX && 5 == blockDimY) {
76 astcFormat = SkTextureCompressor::kASTC_5x5_Format; 76 astcFormat = SkTextureCompressor::kASTC_5x5_Format;
(...skipping 14 matching lines...) Expand all
91 } else if (10 == blockDimX && 8 == blockDimY) { 91 } else if (10 == blockDimX && 8 == blockDimY) {
92 astcFormat = SkTextureCompressor::kASTC_10x8_Format; 92 astcFormat = SkTextureCompressor::kASTC_10x8_Format;
93 } else if (10 == blockDimX && 10 == blockDimY) { 93 } else if (10 == blockDimX && 10 == blockDimY) {
94 astcFormat = SkTextureCompressor::kASTC_10x10_Format; 94 astcFormat = SkTextureCompressor::kASTC_10x10_Format;
95 } else if (12 == blockDimX && 10 == blockDimY) { 95 } else if (12 == blockDimX && 10 == blockDimY) {
96 astcFormat = SkTextureCompressor::kASTC_12x10_Format; 96 astcFormat = SkTextureCompressor::kASTC_12x10_Format;
97 } else if (12 == blockDimX && 12 == blockDimY) { 97 } else if (12 == blockDimX && 12 == blockDimY) {
98 astcFormat = SkTextureCompressor::kASTC_12x12_Format; 98 astcFormat = SkTextureCompressor::kASTC_12x12_Format;
99 } else { 99 } else {
100 // We don't support any other block dimensions.. 100 // We don't support any other block dimensions..
101 return false; 101 return kFailure;
102 } 102 }
103 103
104 // Advance buf past the block dimensions 104 // Advance buf past the block dimensions
105 buf += 3; 105 buf += 3;
106 106
107 // Read the width/height/depth from the buffer... 107 // Read the width/height/depth from the buffer...
108 const int width = read_24bit(buf); 108 const int width = read_24bit(buf);
109 const int height = read_24bit(buf + 3); 109 const int height = read_24bit(buf + 3);
110 const int depth = read_24bit(buf + 6); 110 const int depth = read_24bit(buf + 6);
111 111
112 if (1 != depth) { 112 if (1 != depth) {
113 // We don't support decoding 3D. 113 // We don't support decoding 3D.
114 return false; 114 return kFailure;
115 } 115 }
116 116
117 // Advance the buffer past the image dimensions 117 // Advance the buffer past the image dimensions
118 buf += 9; 118 buf += 9;
119 119
120 #ifdef SK_SUPPORT_LEGACY_IMAGEDECODER_CHOOSER 120 #ifdef SK_SUPPORT_LEGACY_IMAGEDECODER_CHOOSER
121 // should we allow the Chooser (if present) to pick a config for us??? 121 // should we allow the Chooser (if present) to pick a config for us???
122 if (!this->chooseFromOneChoice(kN32_SkColorType, width, height)) { 122 if (!this->chooseFromOneChoice(kN32_SkColorType, width, height)) {
123 return false; 123 return kFailure;
124 } 124 }
125 #endif 125 #endif
126 126
127 // Setup the sampler... 127 // Setup the sampler...
128 SkScaledBitmapSampler sampler(width, height, this->getSampleSize()); 128 SkScaledBitmapSampler sampler(width, height, this->getSampleSize());
129 129
130 // Determine the alpha of the bitmap... 130 // Determine the alpha of the bitmap...
131 SkAlphaType alphaType = kOpaque_SkAlphaType; 131 SkAlphaType alphaType = kOpaque_SkAlphaType;
132 if (this->getRequireUnpremultipliedColors()) { 132 if (this->getRequireUnpremultipliedColors()) {
133 alphaType = kUnpremul_SkAlphaType; 133 alphaType = kUnpremul_SkAlphaType;
134 } else { 134 } else {
135 alphaType = kPremul_SkAlphaType; 135 alphaType = kPremul_SkAlphaType;
136 } 136 }
137 137
138 // Set the config... 138 // Set the config...
139 bm->setInfo(SkImageInfo::MakeN32(sampler.scaledWidth(), sampler.scaledHeight (), alphaType)); 139 bm->setInfo(SkImageInfo::MakeN32(sampler.scaledWidth(), sampler.scaledHeight (), alphaType));
140 140
141 if (SkImageDecoder::kDecodeBounds_Mode == mode) { 141 if (SkImageDecoder::kDecodeBounds_Mode == mode) {
142 return true; 142 return kSuccess;
143 } 143 }
144 144
145 if (!this->allocPixelRef(bm, NULL)) { 145 if (!this->allocPixelRef(bm, NULL)) {
146 return false; 146 return kFailure;
147 } 147 }
148 148
149 // Lock the pixels, since we're about to write to them... 149 // Lock the pixels, since we're about to write to them...
150 SkAutoLockPixels alp(*bm); 150 SkAutoLockPixels alp(*bm);
151 151
152 if (!sampler.begin(bm, SkScaledBitmapSampler::kRGBA, *this)) { 152 if (!sampler.begin(bm, SkScaledBitmapSampler::kRGBA, *this)) {
153 return false; 153 return kFailure;
154 } 154 }
155 155
156 // ASTC Data is encoded as RGBA pixels, so we should extract it as such 156 // ASTC Data is encoded as RGBA pixels, so we should extract it as such
157 int nPixels = width * height; 157 int nPixels = width * height;
158 SkAutoMalloc outRGBAData(nPixels * 4); 158 SkAutoMalloc outRGBAData(nPixels * 4);
159 uint8_t *outRGBADataPtr = reinterpret_cast<uint8_t *>(outRGBAData.get()); 159 uint8_t *outRGBADataPtr = reinterpret_cast<uint8_t *>(outRGBAData.get());
160 160
161 // Decode ASTC 161 // Decode ASTC
162 if (!SkTextureCompressor::DecompressBufferFromFormat( 162 if (!SkTextureCompressor::DecompressBufferFromFormat(
163 outRGBADataPtr, width*4, buf, width, height, astcFormat)) { 163 outRGBADataPtr, width*4, buf, width, height, astcFormat)) {
164 return kFailure;
164 } 165 }
165 166
166 // Set each of the pixels... 167 // Set each of the pixels...
167 const int srcRowBytes = width * 4; 168 const int srcRowBytes = width * 4;
168 const int dstHeight = sampler.scaledHeight(); 169 const int dstHeight = sampler.scaledHeight();
169 const uint8_t *srcRow = reinterpret_cast<uint8_t *>(outRGBADataPtr); 170 const uint8_t *srcRow = reinterpret_cast<uint8_t *>(outRGBADataPtr);
170 srcRow += sampler.srcY0() * srcRowBytes; 171 srcRow += sampler.srcY0() * srcRowBytes;
171 for (int y = 0; y < dstHeight; ++y) { 172 for (int y = 0; y < dstHeight; ++y) {
172 sampler.next(srcRow); 173 sampler.next(srcRow);
173 srcRow += sampler.srcDY() * srcRowBytes; 174 srcRow += sampler.srcDY() * srcRowBytes;
174 } 175 }
175 176
176 return true; 177 return kSuccess;
177 } 178 }
178 179
179 //////////////////////////////////////////////////////////////////////////////// ///////// 180 //////////////////////////////////////////////////////////////////////////////// /////////
180 DEFINE_DECODER_CREATOR(ASTCImageDecoder); 181 DEFINE_DECODER_CREATOR(ASTCImageDecoder);
181 //////////////////////////////////////////////////////////////////////////////// ///////// 182 //////////////////////////////////////////////////////////////////////////////// /////////
182 183
183 static bool is_astc(SkStreamRewindable* stream) { 184 static bool is_astc(SkStreamRewindable* stream) {
184 // Read the ASTC header and make sure it's valid. 185 // Read the ASTC header and make sure it's valid.
185 uint32_t magic; 186 uint32_t magic;
186 if (stream->read((void*)&magic, 4) != 4) { 187 if (stream->read((void*)&magic, 4) != 4) {
(...skipping 13 matching lines...) Expand all
200 static SkImageDecoder_DecodeReg gReg(sk_libastc_dfactory); 201 static SkImageDecoder_DecodeReg gReg(sk_libastc_dfactory);
201 202
202 static SkImageDecoder::Format get_format_astc(SkStreamRewindable* stream) { 203 static SkImageDecoder::Format get_format_astc(SkStreamRewindable* stream) {
203 if (is_astc(stream)) { 204 if (is_astc(stream)) {
204 return SkImageDecoder::kASTC_Format; 205 return SkImageDecoder::kASTC_Format;
205 } 206 }
206 return SkImageDecoder::kUnknown_Format; 207 return SkImageDecoder::kUnknown_Format;
207 } 208 }
208 209
209 static SkImageDecoder_FormatReg gFormatReg(get_format_astc); 210 static SkImageDecoder_FormatReg gFormatReg(get_format_astc);
OLDNEW
« no previous file with comments | « src/images/SkImageDecoder.cpp ('k') | src/images/SkImageDecoder_ktx.cpp » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698