1*89c4ff92SAndroid Build Coastguard Worker //
2*89c4ff92SAndroid Build Coastguard Worker // Copyright (c) 2017 Sean Barrett
3*89c4ff92SAndroid Build Coastguard Worker // SPDX-License-Identifier: MIT
4*89c4ff92SAndroid Build Coastguard Worker //
5*89c4ff92SAndroid Build Coastguard Worker
6*89c4ff92SAndroid Build Coastguard Worker /* stb_image - v2.16 - public domain image loader - http://nothings.org/stb_image.h
7*89c4ff92SAndroid Build Coastguard Worker no warranty implied; use at your own risk
8*89c4ff92SAndroid Build Coastguard Worker
9*89c4ff92SAndroid Build Coastguard Worker Do this:
10*89c4ff92SAndroid Build Coastguard Worker #define STB_IMAGE_IMPLEMENTATION
11*89c4ff92SAndroid Build Coastguard Worker before you include this file in *one* C or C++ file to create the implementation.
12*89c4ff92SAndroid Build Coastguard Worker
13*89c4ff92SAndroid Build Coastguard Worker // i.e. it should look like this:
14*89c4ff92SAndroid Build Coastguard Worker #include ...
15*89c4ff92SAndroid Build Coastguard Worker #include ...
16*89c4ff92SAndroid Build Coastguard Worker #include ...
17*89c4ff92SAndroid Build Coastguard Worker #define STB_IMAGE_IMPLEMENTATION
18*89c4ff92SAndroid Build Coastguard Worker #include "stb_image.h"
19*89c4ff92SAndroid Build Coastguard Worker
20*89c4ff92SAndroid Build Coastguard Worker You can #define STBI_ASSERT(x) before the #include to avoid using assert.h.
21*89c4ff92SAndroid Build Coastguard Worker And #define STBI_MALLOC, STBI_REALLOC, and STBI_FREE to avoid using malloc,realloc,free
22*89c4ff92SAndroid Build Coastguard Worker
23*89c4ff92SAndroid Build Coastguard Worker
24*89c4ff92SAndroid Build Coastguard Worker QUICK NOTES:
25*89c4ff92SAndroid Build Coastguard Worker Primarily of interest to game developers and other people who can
26*89c4ff92SAndroid Build Coastguard Worker avoid problematic images and only need the trivial interface
27*89c4ff92SAndroid Build Coastguard Worker
28*89c4ff92SAndroid Build Coastguard Worker JPEG baseline & progressive (12 bpc/arithmetic not supported, same as stock IJG lib)
29*89c4ff92SAndroid Build Coastguard Worker PNG 1/2/4/8/16-bit-per-channel
30*89c4ff92SAndroid Build Coastguard Worker
31*89c4ff92SAndroid Build Coastguard Worker TGA (not sure what subset, if a subset)
32*89c4ff92SAndroid Build Coastguard Worker BMP non-1bpp, non-RLE
33*89c4ff92SAndroid Build Coastguard Worker PSD (composited view only, no extra channels, 8/16 bit-per-channel)
34*89c4ff92SAndroid Build Coastguard Worker
35*89c4ff92SAndroid Build Coastguard Worker GIF (*comp always reports as 4-channel)
36*89c4ff92SAndroid Build Coastguard Worker HDR (radiance rgbE format)
37*89c4ff92SAndroid Build Coastguard Worker PIC (Softimage PIC)
38*89c4ff92SAndroid Build Coastguard Worker PNM (PPM and PGM binary only)
39*89c4ff92SAndroid Build Coastguard Worker
40*89c4ff92SAndroid Build Coastguard Worker Animated GIF still needs a proper API, but here's one way to do it:
41*89c4ff92SAndroid Build Coastguard Worker http://gist.github.com/urraka/685d9a6340b26b830d49
42*89c4ff92SAndroid Build Coastguard Worker
43*89c4ff92SAndroid Build Coastguard Worker - decode from memory or through FILE (define STBI_NO_STDIO to remove code)
44*89c4ff92SAndroid Build Coastguard Worker - decode from arbitrary I/O callbacks
45*89c4ff92SAndroid Build Coastguard Worker - SIMD acceleration on x86/x64 (SSE2) and ARM (NEON)
46*89c4ff92SAndroid Build Coastguard Worker
47*89c4ff92SAndroid Build Coastguard Worker Full documentation under "DOCUMENTATION" below.
48*89c4ff92SAndroid Build Coastguard Worker
49*89c4ff92SAndroid Build Coastguard Worker
50*89c4ff92SAndroid Build Coastguard Worker LICENSE
51*89c4ff92SAndroid Build Coastguard Worker
52*89c4ff92SAndroid Build Coastguard Worker See end of file for license information.
53*89c4ff92SAndroid Build Coastguard Worker
54*89c4ff92SAndroid Build Coastguard Worker RECENT REVISION HISTORY:
55*89c4ff92SAndroid Build Coastguard Worker
56*89c4ff92SAndroid Build Coastguard Worker 2.16 (2017-07-23) all functions have 16-bit variants; optimizations; bugfixes
57*89c4ff92SAndroid Build Coastguard Worker 2.15 (2017-03-18) fix png-1,2,4; all Imagenet JPGs; no runtime SSE detection on GCC
58*89c4ff92SAndroid Build Coastguard Worker 2.14 (2017-03-03) remove deprecated STBI_JPEG_OLD; fixes for Imagenet JPGs
59*89c4ff92SAndroid Build Coastguard Worker 2.13 (2016-12-04) experimental 16-bit API, only for PNG so far; fixes
60*89c4ff92SAndroid Build Coastguard Worker 2.12 (2016-04-02) fix typo in 2.11 PSD fix that caused crashes
61*89c4ff92SAndroid Build Coastguard Worker 2.11 (2016-04-02) 16-bit PNGS; enable SSE2 in non-gcc x64
62*89c4ff92SAndroid Build Coastguard Worker RGB-format JPEG; remove white matting in PSD;
63*89c4ff92SAndroid Build Coastguard Worker allocate large structures on the stack;
64*89c4ff92SAndroid Build Coastguard Worker correct channel count for PNG & BMP
65*89c4ff92SAndroid Build Coastguard Worker 2.10 (2016-01-22) avoid warning introduced in 2.09
66*89c4ff92SAndroid Build Coastguard Worker 2.09 (2016-01-16) 16-bit TGA; comments in PNM files; STBI_REALLOC_SIZED
67*89c4ff92SAndroid Build Coastguard Worker
68*89c4ff92SAndroid Build Coastguard Worker See end of file for full revision history.
69*89c4ff92SAndroid Build Coastguard Worker
70*89c4ff92SAndroid Build Coastguard Worker
71*89c4ff92SAndroid Build Coastguard Worker ============================ Contributors =========================
72*89c4ff92SAndroid Build Coastguard Worker
73*89c4ff92SAndroid Build Coastguard Worker Image formats Extensions, features
74*89c4ff92SAndroid Build Coastguard Worker Sean Barrett (jpeg, png, bmp) Jetro Lauha (stbi_info)
75*89c4ff92SAndroid Build Coastguard Worker Nicolas Schulz (hdr, psd) Martin "SpartanJ" Golini (stbi_info)
76*89c4ff92SAndroid Build Coastguard Worker Jonathan Dummer (tga) James "moose2000" Brown (iPhone PNG)
77*89c4ff92SAndroid Build Coastguard Worker Jean-Marc Lienher (gif) Ben "Disch" Wenger (io callbacks)
78*89c4ff92SAndroid Build Coastguard Worker Tom Seddon (pic) Omar Cornut (1/2/4-bit PNG)
79*89c4ff92SAndroid Build Coastguard Worker Thatcher Ulrich (psd) Nicolas Guillemot (vertical flip)
80*89c4ff92SAndroid Build Coastguard Worker Ken Miller (pgm, ppm) Richard Mitton (16-bit PSD)
81*89c4ff92SAndroid Build Coastguard Worker github:urraka (animated gif) Junggon Kim (PNM comments)
82*89c4ff92SAndroid Build Coastguard Worker Daniel Gibson (16-bit TGA)
83*89c4ff92SAndroid Build Coastguard Worker socks-the-fox (16-bit PNG)
84*89c4ff92SAndroid Build Coastguard Worker Jeremy Sawicki (handle all ImageNet JPGs)
85*89c4ff92SAndroid Build Coastguard Worker Optimizations & bugfixes
86*89c4ff92SAndroid Build Coastguard Worker Fabian "ryg" Giesen
87*89c4ff92SAndroid Build Coastguard Worker Arseny Kapoulkine
88*89c4ff92SAndroid Build Coastguard Worker John-Mark Allen
89*89c4ff92SAndroid Build Coastguard Worker
90*89c4ff92SAndroid Build Coastguard Worker Bug & warning fixes
91*89c4ff92SAndroid Build Coastguard Worker Marc LeBlanc David Woo Guillaume George Martins Mozeiko
92*89c4ff92SAndroid Build Coastguard Worker Christpher Lloyd Jerry Jansson Joseph Thomson Phil Jordan
93*89c4ff92SAndroid Build Coastguard Worker Dave Moore Roy Eltham Hayaki Saito Nathan Reed
94*89c4ff92SAndroid Build Coastguard Worker Won Chun Luke Graham Johan Duparc Nick Verigakis
95*89c4ff92SAndroid Build Coastguard Worker the Horde3D community Thomas Ruf Ronny Chevalier Baldur Karlsson
96*89c4ff92SAndroid Build Coastguard Worker Janez Zemva John Bartholomew Michal Cichon github:rlyeh
97*89c4ff92SAndroid Build Coastguard Worker Jonathan Blow Ken Hamada Tero Hanninen github:romigrou
98*89c4ff92SAndroid Build Coastguard Worker Laurent Gomila Cort Stratton Sergio Gonzalez github:svdijk
99*89c4ff92SAndroid Build Coastguard Worker Aruelien Pocheville Thibault Reuille Cass Everitt github:snagar
100*89c4ff92SAndroid Build Coastguard Worker Ryamond Barbiero Paul Du Bois Engin Manap github:Zelex
101*89c4ff92SAndroid Build Coastguard Worker Michaelangel007@github Philipp Wiesemann Dale Weiler github:grim210
102*89c4ff92SAndroid Build Coastguard Worker Oriol Ferrer Mesia Josh Tobin Matthew Gregan github:sammyhw
103*89c4ff92SAndroid Build Coastguard Worker Blazej Dariusz Roszkowski Gregory Mullen github:phprus
104*89c4ff92SAndroid Build Coastguard Worker Christian Floisand Kevin Schmidt github:poppolopoppo
105*89c4ff92SAndroid Build Coastguard Worker */
106*89c4ff92SAndroid Build Coastguard Worker
107*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_INCLUDE_STB_IMAGE_H
108*89c4ff92SAndroid Build Coastguard Worker #define STBI_INCLUDE_STB_IMAGE_H
109*89c4ff92SAndroid Build Coastguard Worker
110*89c4ff92SAndroid Build Coastguard Worker // DOCUMENTATION
111*89c4ff92SAndroid Build Coastguard Worker //
112*89c4ff92SAndroid Build Coastguard Worker // Limitations:
113*89c4ff92SAndroid Build Coastguard Worker // - no 16-bit-per-channel PNG
114*89c4ff92SAndroid Build Coastguard Worker // - no 12-bit-per-channel JPEG
115*89c4ff92SAndroid Build Coastguard Worker // - no JPEGs with arithmetic coding
116*89c4ff92SAndroid Build Coastguard Worker // - no 1-bit BMP
117*89c4ff92SAndroid Build Coastguard Worker // - GIF always returns *comp=4
118*89c4ff92SAndroid Build Coastguard Worker //
119*89c4ff92SAndroid Build Coastguard Worker // Basic usage (see HDR discussion below for HDR usage):
120*89c4ff92SAndroid Build Coastguard Worker // int x,y,n;
121*89c4ff92SAndroid Build Coastguard Worker // unsigned char *data = stbi_load(filename, &x, &y, &n, 0);
122*89c4ff92SAndroid Build Coastguard Worker // // ... process data if not NULL ...
123*89c4ff92SAndroid Build Coastguard Worker // // ... x = width, y = height, n = # 8-bit components per pixel ...
124*89c4ff92SAndroid Build Coastguard Worker // // ... replace '0' with '1'..'4' to force that many components per pixel
125*89c4ff92SAndroid Build Coastguard Worker // // ... but 'n' will always be the number that it would have been if you said 0
126*89c4ff92SAndroid Build Coastguard Worker // stbi_image_free(data)
127*89c4ff92SAndroid Build Coastguard Worker //
128*89c4ff92SAndroid Build Coastguard Worker // Standard parameters:
129*89c4ff92SAndroid Build Coastguard Worker // int *x -- outputs image width in pixels
130*89c4ff92SAndroid Build Coastguard Worker // int *y -- outputs image height in pixels
131*89c4ff92SAndroid Build Coastguard Worker // int *channels_in_file -- outputs # of image components in image file
132*89c4ff92SAndroid Build Coastguard Worker // int desired_channels -- if non-zero, # of image components requested in result
133*89c4ff92SAndroid Build Coastguard Worker //
134*89c4ff92SAndroid Build Coastguard Worker // The return value from an image loader is an 'unsigned char *' which points
135*89c4ff92SAndroid Build Coastguard Worker // to the pixel data, or NULL on an allocation failure or if the image is
136*89c4ff92SAndroid Build Coastguard Worker // corrupt or invalid. The pixel data consists of *y scanlines of *x pixels,
137*89c4ff92SAndroid Build Coastguard Worker // with each pixel consisting of N interleaved 8-bit components; the first
138*89c4ff92SAndroid Build Coastguard Worker // pixel pointed to is top-left-most in the image. There is no padding between
139*89c4ff92SAndroid Build Coastguard Worker // image scanlines or between pixels, regardless of format. The number of
140*89c4ff92SAndroid Build Coastguard Worker // components N is 'desired_channels' if desired_channels is non-zero, or
141*89c4ff92SAndroid Build Coastguard Worker // *channels_in_file otherwise. If desired_channels is non-zero,
142*89c4ff92SAndroid Build Coastguard Worker // *channels_in_file has the number of components that _would_ have been
143*89c4ff92SAndroid Build Coastguard Worker // output otherwise. E.g. if you set desired_channels to 4, you will always
144*89c4ff92SAndroid Build Coastguard Worker // get RGBA output, but you can check *channels_in_file to see if it's trivially
145*89c4ff92SAndroid Build Coastguard Worker // opaque because e.g. there were only 3 channels in the source image.
146*89c4ff92SAndroid Build Coastguard Worker //
147*89c4ff92SAndroid Build Coastguard Worker // An output image with N components has the following components interleaved
148*89c4ff92SAndroid Build Coastguard Worker // in this order in each pixel:
149*89c4ff92SAndroid Build Coastguard Worker //
150*89c4ff92SAndroid Build Coastguard Worker // N=#comp components
151*89c4ff92SAndroid Build Coastguard Worker // 1 grey
152*89c4ff92SAndroid Build Coastguard Worker // 2 grey, alpha
153*89c4ff92SAndroid Build Coastguard Worker // 3 red, green, blue
154*89c4ff92SAndroid Build Coastguard Worker // 4 red, green, blue, alpha
155*89c4ff92SAndroid Build Coastguard Worker //
156*89c4ff92SAndroid Build Coastguard Worker // If image loading fails for any reason, the return value will be NULL,
157*89c4ff92SAndroid Build Coastguard Worker // and *x, *y, *channels_in_file will be unchanged. The function
158*89c4ff92SAndroid Build Coastguard Worker // stbi_failure_reason() can be queried for an extremely brief, end-user
159*89c4ff92SAndroid Build Coastguard Worker // unfriendly explanation of why the load failed. Define STBI_NO_FAILURE_STRINGS
160*89c4ff92SAndroid Build Coastguard Worker // to avoid compiling these strings at all, and STBI_FAILURE_USERMSG to get slightly
161*89c4ff92SAndroid Build Coastguard Worker // more user-friendly ones.
162*89c4ff92SAndroid Build Coastguard Worker //
163*89c4ff92SAndroid Build Coastguard Worker // Paletted PNG, BMP, GIF, and PIC images are automatically depalettized.
164*89c4ff92SAndroid Build Coastguard Worker //
165*89c4ff92SAndroid Build Coastguard Worker // ===========================================================================
166*89c4ff92SAndroid Build Coastguard Worker //
167*89c4ff92SAndroid Build Coastguard Worker // Philosophy
168*89c4ff92SAndroid Build Coastguard Worker //
169*89c4ff92SAndroid Build Coastguard Worker // stb libraries are designed with the following priorities:
170*89c4ff92SAndroid Build Coastguard Worker //
171*89c4ff92SAndroid Build Coastguard Worker // 1. easy to use
172*89c4ff92SAndroid Build Coastguard Worker // 2. easy to maintain
173*89c4ff92SAndroid Build Coastguard Worker // 3. good performance
174*89c4ff92SAndroid Build Coastguard Worker //
175*89c4ff92SAndroid Build Coastguard Worker // Sometimes I let "good performance" creep up in priority over "easy to maintain",
176*89c4ff92SAndroid Build Coastguard Worker // and for best performance I may provide less-easy-to-use APIs that give higher
177*89c4ff92SAndroid Build Coastguard Worker // performance, in addition to the easy to use ones. Nevertheless, it's important
178*89c4ff92SAndroid Build Coastguard Worker // to keep in mind that from the standpoint of you, a client of this library,
179*89c4ff92SAndroid Build Coastguard Worker // all you care about is #1 and #3, and stb libraries DO NOT emphasize #3 above all.
180*89c4ff92SAndroid Build Coastguard Worker //
181*89c4ff92SAndroid Build Coastguard Worker // Some secondary priorities arise directly from the first two, some of which
182*89c4ff92SAndroid Build Coastguard Worker // make more explicit reasons why performance can't be emphasized.
183*89c4ff92SAndroid Build Coastguard Worker //
184*89c4ff92SAndroid Build Coastguard Worker // - Portable ("ease of use")
185*89c4ff92SAndroid Build Coastguard Worker // - Small source code footprint ("easy to maintain")
186*89c4ff92SAndroid Build Coastguard Worker // - No dependencies ("ease of use")
187*89c4ff92SAndroid Build Coastguard Worker //
188*89c4ff92SAndroid Build Coastguard Worker // ===========================================================================
189*89c4ff92SAndroid Build Coastguard Worker //
190*89c4ff92SAndroid Build Coastguard Worker // I/O callbacks
191*89c4ff92SAndroid Build Coastguard Worker //
192*89c4ff92SAndroid Build Coastguard Worker // I/O callbacks allow you to read from arbitrary sources, like packaged
193*89c4ff92SAndroid Build Coastguard Worker // files or some other source. Data read from callbacks are processed
194*89c4ff92SAndroid Build Coastguard Worker // through a small internal buffer (currently 128 bytes) to try to reduce
195*89c4ff92SAndroid Build Coastguard Worker // overhead.
196*89c4ff92SAndroid Build Coastguard Worker //
197*89c4ff92SAndroid Build Coastguard Worker // The three functions you must define are "read" (reads some bytes of data),
198*89c4ff92SAndroid Build Coastguard Worker // "skip" (skips some bytes of data), "eof" (reports if the stream is at the end).
199*89c4ff92SAndroid Build Coastguard Worker //
200*89c4ff92SAndroid Build Coastguard Worker // ===========================================================================
201*89c4ff92SAndroid Build Coastguard Worker //
202*89c4ff92SAndroid Build Coastguard Worker // SIMD support
203*89c4ff92SAndroid Build Coastguard Worker //
204*89c4ff92SAndroid Build Coastguard Worker // The JPEG decoder will try to automatically use SIMD kernels on x86 when
205*89c4ff92SAndroid Build Coastguard Worker // supported by the compiler. For ARM Neon support, you must explicitly
206*89c4ff92SAndroid Build Coastguard Worker // request it.
207*89c4ff92SAndroid Build Coastguard Worker //
208*89c4ff92SAndroid Build Coastguard Worker // (The old do-it-yourself SIMD API is no longer supported in the current
209*89c4ff92SAndroid Build Coastguard Worker // code.)
210*89c4ff92SAndroid Build Coastguard Worker //
211*89c4ff92SAndroid Build Coastguard Worker // On x86, SSE2 will automatically be used when available based on a run-time
212*89c4ff92SAndroid Build Coastguard Worker // test; if not, the generic C versions are used as a fall-back. On ARM targets,
213*89c4ff92SAndroid Build Coastguard Worker // the typical path is to have separate builds for NEON and non-NEON devices
214*89c4ff92SAndroid Build Coastguard Worker // (at least this is true for iOS and Android). Therefore, the NEON support is
215*89c4ff92SAndroid Build Coastguard Worker // toggled by a build flag: define STBI_NEON to get NEON loops.
216*89c4ff92SAndroid Build Coastguard Worker //
217*89c4ff92SAndroid Build Coastguard Worker // If for some reason you do not want to use any of SIMD code, or if
218*89c4ff92SAndroid Build Coastguard Worker // you have issues compiling it, you can disable it entirely by
219*89c4ff92SAndroid Build Coastguard Worker // defining STBI_NO_SIMD.
220*89c4ff92SAndroid Build Coastguard Worker //
221*89c4ff92SAndroid Build Coastguard Worker // ===========================================================================
222*89c4ff92SAndroid Build Coastguard Worker //
223*89c4ff92SAndroid Build Coastguard Worker // HDR image support (disable by defining STBI_NO_HDR)
224*89c4ff92SAndroid Build Coastguard Worker //
225*89c4ff92SAndroid Build Coastguard Worker // stb_image now supports loading HDR images in general, and currently
226*89c4ff92SAndroid Build Coastguard Worker // the Radiance .HDR file format, although the support is provided
227*89c4ff92SAndroid Build Coastguard Worker // generically. You can still load any file through the existing interface;
228*89c4ff92SAndroid Build Coastguard Worker // if you attempt to load an HDR file, it will be automatically remapped to
229*89c4ff92SAndroid Build Coastguard Worker // LDR, assuming gamma 2.2 and an arbitrary scale factor defaulting to 1;
230*89c4ff92SAndroid Build Coastguard Worker // both of these constants can be reconfigured through this interface:
231*89c4ff92SAndroid Build Coastguard Worker //
232*89c4ff92SAndroid Build Coastguard Worker // stbi_hdr_to_ldr_gamma(2.2f);
233*89c4ff92SAndroid Build Coastguard Worker // stbi_hdr_to_ldr_scale(1.0f);
234*89c4ff92SAndroid Build Coastguard Worker //
235*89c4ff92SAndroid Build Coastguard Worker // (note, do not use _inverse_ constants; stbi_image will invert them
236*89c4ff92SAndroid Build Coastguard Worker // appropriately).
237*89c4ff92SAndroid Build Coastguard Worker //
238*89c4ff92SAndroid Build Coastguard Worker // Additionally, there is a new, parallel interface for loading files as
239*89c4ff92SAndroid Build Coastguard Worker // (linear) floats to preserve the full dynamic range:
240*89c4ff92SAndroid Build Coastguard Worker //
241*89c4ff92SAndroid Build Coastguard Worker // float *data = stbi_loadf(filename, &x, &y, &n, 0);
242*89c4ff92SAndroid Build Coastguard Worker //
243*89c4ff92SAndroid Build Coastguard Worker // If you load LDR images through this interface, those images will
244*89c4ff92SAndroid Build Coastguard Worker // be promoted to floating point values, run through the inverse of
245*89c4ff92SAndroid Build Coastguard Worker // constants corresponding to the above:
246*89c4ff92SAndroid Build Coastguard Worker //
247*89c4ff92SAndroid Build Coastguard Worker // stbi_ldr_to_hdr_scale(1.0f);
248*89c4ff92SAndroid Build Coastguard Worker // stbi_ldr_to_hdr_gamma(2.2f);
249*89c4ff92SAndroid Build Coastguard Worker //
250*89c4ff92SAndroid Build Coastguard Worker // Finally, given a filename (or an open file or memory block--see header
251*89c4ff92SAndroid Build Coastguard Worker // file for details) containing image data, you can query for the "most
252*89c4ff92SAndroid Build Coastguard Worker // appropriate" interface to use (that is, whether the image is HDR or
253*89c4ff92SAndroid Build Coastguard Worker // not), using:
254*89c4ff92SAndroid Build Coastguard Worker //
255*89c4ff92SAndroid Build Coastguard Worker // stbi_is_hdr(char *filename);
256*89c4ff92SAndroid Build Coastguard Worker //
257*89c4ff92SAndroid Build Coastguard Worker // ===========================================================================
258*89c4ff92SAndroid Build Coastguard Worker //
259*89c4ff92SAndroid Build Coastguard Worker // iPhone PNG support:
260*89c4ff92SAndroid Build Coastguard Worker //
261*89c4ff92SAndroid Build Coastguard Worker // By default we convert iphone-formatted PNGs back to RGB, even though
262*89c4ff92SAndroid Build Coastguard Worker // they are internally encoded differently. You can disable this conversion
263*89c4ff92SAndroid Build Coastguard Worker // by by calling stbi_convert_iphone_png_to_rgb(0), in which case
264*89c4ff92SAndroid Build Coastguard Worker // you will always just get the native iphone "format" through (which
265*89c4ff92SAndroid Build Coastguard Worker // is BGR stored in RGB).
266*89c4ff92SAndroid Build Coastguard Worker //
267*89c4ff92SAndroid Build Coastguard Worker // Call stbi_set_unpremultiply_on_load(1) as well to force a divide per
268*89c4ff92SAndroid Build Coastguard Worker // pixel to remove any premultiplied alpha *only* if the image file explicitly
269*89c4ff92SAndroid Build Coastguard Worker // says there's premultiplied data (currently only happens in iPhone images,
270*89c4ff92SAndroid Build Coastguard Worker // and only if iPhone convert-to-rgb processing is on).
271*89c4ff92SAndroid Build Coastguard Worker //
272*89c4ff92SAndroid Build Coastguard Worker // ===========================================================================
273*89c4ff92SAndroid Build Coastguard Worker //
274*89c4ff92SAndroid Build Coastguard Worker // ADDITIONAL CONFIGURATION
275*89c4ff92SAndroid Build Coastguard Worker //
276*89c4ff92SAndroid Build Coastguard Worker // - You can suppress implementation of any of the decoders to reduce
277*89c4ff92SAndroid Build Coastguard Worker // your code footprint by #defining one or more of the following
278*89c4ff92SAndroid Build Coastguard Worker // symbols before creating the implementation.
279*89c4ff92SAndroid Build Coastguard Worker //
280*89c4ff92SAndroid Build Coastguard Worker // STBI_NO_JPEG
281*89c4ff92SAndroid Build Coastguard Worker // STBI_NO_PNG
282*89c4ff92SAndroid Build Coastguard Worker // STBI_NO_BMP
283*89c4ff92SAndroid Build Coastguard Worker // STBI_NO_PSD
284*89c4ff92SAndroid Build Coastguard Worker // STBI_NO_TGA
285*89c4ff92SAndroid Build Coastguard Worker // STBI_NO_GIF
286*89c4ff92SAndroid Build Coastguard Worker // STBI_NO_HDR
287*89c4ff92SAndroid Build Coastguard Worker // STBI_NO_PIC
288*89c4ff92SAndroid Build Coastguard Worker // STBI_NO_PNM (.ppm and .pgm)
289*89c4ff92SAndroid Build Coastguard Worker //
290*89c4ff92SAndroid Build Coastguard Worker // - You can request *only* certain decoders and suppress all other ones
291*89c4ff92SAndroid Build Coastguard Worker // (this will be more forward-compatible, as addition of new decoders
292*89c4ff92SAndroid Build Coastguard Worker // doesn't require you to disable them explicitly):
293*89c4ff92SAndroid Build Coastguard Worker //
294*89c4ff92SAndroid Build Coastguard Worker // STBI_ONLY_JPEG
295*89c4ff92SAndroid Build Coastguard Worker // STBI_ONLY_PNG
296*89c4ff92SAndroid Build Coastguard Worker // STBI_ONLY_BMP
297*89c4ff92SAndroid Build Coastguard Worker // STBI_ONLY_PSD
298*89c4ff92SAndroid Build Coastguard Worker // STBI_ONLY_TGA
299*89c4ff92SAndroid Build Coastguard Worker // STBI_ONLY_GIF
300*89c4ff92SAndroid Build Coastguard Worker // STBI_ONLY_HDR
301*89c4ff92SAndroid Build Coastguard Worker // STBI_ONLY_PIC
302*89c4ff92SAndroid Build Coastguard Worker // STBI_ONLY_PNM (.ppm and .pgm)
303*89c4ff92SAndroid Build Coastguard Worker //
304*89c4ff92SAndroid Build Coastguard Worker // - If you use STBI_NO_PNG (or _ONLY_ without PNG), and you still
305*89c4ff92SAndroid Build Coastguard Worker // want the zlib decoder to be available, #define STBI_SUPPORT_ZLIB
306*89c4ff92SAndroid Build Coastguard Worker //
307*89c4ff92SAndroid Build Coastguard Worker
308*89c4ff92SAndroid Build Coastguard Worker
309*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
310*89c4ff92SAndroid Build Coastguard Worker #include <stdio.h>
311*89c4ff92SAndroid Build Coastguard Worker #endif // STBI_NO_STDIO
312*89c4ff92SAndroid Build Coastguard Worker
313*89c4ff92SAndroid Build Coastguard Worker #define STBI_VERSION 1
314*89c4ff92SAndroid Build Coastguard Worker
315*89c4ff92SAndroid Build Coastguard Worker enum
316*89c4ff92SAndroid Build Coastguard Worker {
317*89c4ff92SAndroid Build Coastguard Worker STBI_default = 0, // only used for desired_channels
318*89c4ff92SAndroid Build Coastguard Worker
319*89c4ff92SAndroid Build Coastguard Worker STBI_grey = 1,
320*89c4ff92SAndroid Build Coastguard Worker STBI_grey_alpha = 2,
321*89c4ff92SAndroid Build Coastguard Worker STBI_rgb = 3,
322*89c4ff92SAndroid Build Coastguard Worker STBI_rgb_alpha = 4
323*89c4ff92SAndroid Build Coastguard Worker };
324*89c4ff92SAndroid Build Coastguard Worker
325*89c4ff92SAndroid Build Coastguard Worker typedef unsigned char stbi_uc;
326*89c4ff92SAndroid Build Coastguard Worker typedef unsigned short stbi_us;
327*89c4ff92SAndroid Build Coastguard Worker
328*89c4ff92SAndroid Build Coastguard Worker #ifdef __cplusplus
329*89c4ff92SAndroid Build Coastguard Worker extern "C" {
330*89c4ff92SAndroid Build Coastguard Worker #endif
331*89c4ff92SAndroid Build Coastguard Worker
332*89c4ff92SAndroid Build Coastguard Worker #ifdef STB_IMAGE_STATIC
333*89c4ff92SAndroid Build Coastguard Worker #define STBIDEF static
334*89c4ff92SAndroid Build Coastguard Worker #else
335*89c4ff92SAndroid Build Coastguard Worker #define STBIDEF extern
336*89c4ff92SAndroid Build Coastguard Worker #endif
337*89c4ff92SAndroid Build Coastguard Worker
338*89c4ff92SAndroid Build Coastguard Worker //////////////////////////////////////////////////////////////////////////////
339*89c4ff92SAndroid Build Coastguard Worker //
340*89c4ff92SAndroid Build Coastguard Worker // PRIMARY API - works on images of any type
341*89c4ff92SAndroid Build Coastguard Worker //
342*89c4ff92SAndroid Build Coastguard Worker
343*89c4ff92SAndroid Build Coastguard Worker //
344*89c4ff92SAndroid Build Coastguard Worker // load image by filename, open file, or memory buffer
345*89c4ff92SAndroid Build Coastguard Worker //
346*89c4ff92SAndroid Build Coastguard Worker
347*89c4ff92SAndroid Build Coastguard Worker typedef struct
348*89c4ff92SAndroid Build Coastguard Worker {
349*89c4ff92SAndroid Build Coastguard Worker int (*read) (void *user,char *data,int size); // fill 'data' with 'size' bytes. return number of bytes actually read
350*89c4ff92SAndroid Build Coastguard Worker void (*skip) (void *user,int n); // skip the next 'n' bytes, or 'unget' the last -n bytes if negative
351*89c4ff92SAndroid Build Coastguard Worker int (*eof) (void *user); // returns nonzero if we are at end of file/data
352*89c4ff92SAndroid Build Coastguard Worker } stbi_io_callbacks;
353*89c4ff92SAndroid Build Coastguard Worker
354*89c4ff92SAndroid Build Coastguard Worker ////////////////////////////////////
355*89c4ff92SAndroid Build Coastguard Worker //
356*89c4ff92SAndroid Build Coastguard Worker // 8-bits-per-channel interface
357*89c4ff92SAndroid Build Coastguard Worker //
358*89c4ff92SAndroid Build Coastguard Worker
359*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_uc *stbi_load_from_memory (stbi_uc const *buffer, int len , int *x, int *y, int *channels_in_file, int desired_channels);
360*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_uc *stbi_load_from_callbacks(stbi_io_callbacks const *clbk , void *user, int *x, int *y, int *channels_in_file, int desired_channels);
361*89c4ff92SAndroid Build Coastguard Worker
362*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
363*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_uc *stbi_load (char const *filename, int *x, int *y, int *channels_in_file, int desired_channels);
364*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_uc *stbi_load_from_file (FILE *f, int *x, int *y, int *channels_in_file, int desired_channels);
365*89c4ff92SAndroid Build Coastguard Worker // for stbi_load_from_file, file pointer is left pointing immediately after image
366*89c4ff92SAndroid Build Coastguard Worker #endif
367*89c4ff92SAndroid Build Coastguard Worker
368*89c4ff92SAndroid Build Coastguard Worker ////////////////////////////////////
369*89c4ff92SAndroid Build Coastguard Worker //
370*89c4ff92SAndroid Build Coastguard Worker // 16-bits-per-channel interface
371*89c4ff92SAndroid Build Coastguard Worker //
372*89c4ff92SAndroid Build Coastguard Worker
373*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_us *stbi_load_16_from_memory (stbi_uc const *buffer, int len, int *x, int *y, int *channels_in_file, int desired_channels);
374*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_us *stbi_load_16_from_callbacks(stbi_io_callbacks const *clbk, void *user, int *x, int *y, int *channels_in_file, int desired_channels);
375*89c4ff92SAndroid Build Coastguard Worker
376*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
377*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_us *stbi_load_16 (char const *filename, int *x, int *y, int *channels_in_file, int desired_channels);
378*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_us *stbi_load_from_file_16(FILE *f, int *x, int *y, int *channels_in_file, int desired_channels);
379*89c4ff92SAndroid Build Coastguard Worker #endif
380*89c4ff92SAndroid Build Coastguard Worker
381*89c4ff92SAndroid Build Coastguard Worker ////////////////////////////////////
382*89c4ff92SAndroid Build Coastguard Worker //
383*89c4ff92SAndroid Build Coastguard Worker // float-per-channel interface
384*89c4ff92SAndroid Build Coastguard Worker //
385*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_LINEAR
386*89c4ff92SAndroid Build Coastguard Worker STBIDEF float *stbi_loadf_from_memory (stbi_uc const *buffer, int len, int *x, int *y, int *channels_in_file, int desired_channels);
387*89c4ff92SAndroid Build Coastguard Worker STBIDEF float *stbi_loadf_from_callbacks (stbi_io_callbacks const *clbk, void *user, int *x, int *y, int *channels_in_file, int desired_channels);
388*89c4ff92SAndroid Build Coastguard Worker
389*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
390*89c4ff92SAndroid Build Coastguard Worker STBIDEF float *stbi_loadf (char const *filename, int *x, int *y, int *channels_in_file, int desired_channels);
391*89c4ff92SAndroid Build Coastguard Worker STBIDEF float *stbi_loadf_from_file (FILE *f, int *x, int *y, int *channels_in_file, int desired_channels);
392*89c4ff92SAndroid Build Coastguard Worker #endif
393*89c4ff92SAndroid Build Coastguard Worker #endif
394*89c4ff92SAndroid Build Coastguard Worker
395*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
396*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_hdr_to_ldr_gamma(float gamma);
397*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_hdr_to_ldr_scale(float scale);
398*89c4ff92SAndroid Build Coastguard Worker #endif // STBI_NO_HDR
399*89c4ff92SAndroid Build Coastguard Worker
400*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_LINEAR
401*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_ldr_to_hdr_gamma(float gamma);
402*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_ldr_to_hdr_scale(float scale);
403*89c4ff92SAndroid Build Coastguard Worker #endif // STBI_NO_LINEAR
404*89c4ff92SAndroid Build Coastguard Worker
405*89c4ff92SAndroid Build Coastguard Worker // stbi_is_hdr is always defined, but always returns false if STBI_NO_HDR
406*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_is_hdr_from_callbacks(stbi_io_callbacks const *clbk, void *user);
407*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_is_hdr_from_memory(stbi_uc const *buffer, int len);
408*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
409*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_is_hdr (char const *filename);
410*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_is_hdr_from_file(FILE *f);
411*89c4ff92SAndroid Build Coastguard Worker #endif // STBI_NO_STDIO
412*89c4ff92SAndroid Build Coastguard Worker
413*89c4ff92SAndroid Build Coastguard Worker
414*89c4ff92SAndroid Build Coastguard Worker // get a VERY brief reason for failure
415*89c4ff92SAndroid Build Coastguard Worker // NOT THREADSAFE
416*89c4ff92SAndroid Build Coastguard Worker STBIDEF const char *stbi_failure_reason (void);
417*89c4ff92SAndroid Build Coastguard Worker
418*89c4ff92SAndroid Build Coastguard Worker // free the loaded image -- this is just free()
419*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_image_free (void *retval_from_stbi_load);
420*89c4ff92SAndroid Build Coastguard Worker
421*89c4ff92SAndroid Build Coastguard Worker // get image dimensions & components without fully decoding
422*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_info_from_memory(stbi_uc const *buffer, int len, int *x, int *y, int *comp);
423*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_info_from_callbacks(stbi_io_callbacks const *clbk, void *user, int *x, int *y, int *comp);
424*89c4ff92SAndroid Build Coastguard Worker
425*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
426*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_info (char const *filename, int *x, int *y, int *comp);
427*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_info_from_file (FILE *f, int *x, int *y, int *comp);
428*89c4ff92SAndroid Build Coastguard Worker
429*89c4ff92SAndroid Build Coastguard Worker #endif
430*89c4ff92SAndroid Build Coastguard Worker
431*89c4ff92SAndroid Build Coastguard Worker
432*89c4ff92SAndroid Build Coastguard Worker
433*89c4ff92SAndroid Build Coastguard Worker // for image formats that explicitly notate that they have premultiplied alpha,
434*89c4ff92SAndroid Build Coastguard Worker // we just return the colors as stored in the file. set this flag to force
435*89c4ff92SAndroid Build Coastguard Worker // unpremultiplication. results are undefined if the unpremultiply overflow.
436*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_set_unpremultiply_on_load(int flag_true_if_should_unpremultiply);
437*89c4ff92SAndroid Build Coastguard Worker
438*89c4ff92SAndroid Build Coastguard Worker // indicate whether we should process iphone images back to canonical format,
439*89c4ff92SAndroid Build Coastguard Worker // or just pass them through "as-is"
440*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_convert_iphone_png_to_rgb(int flag_true_if_should_convert);
441*89c4ff92SAndroid Build Coastguard Worker
442*89c4ff92SAndroid Build Coastguard Worker // flip the image vertically, so the first pixel in the output array is the bottom left
443*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_set_flip_vertically_on_load(int flag_true_if_should_flip);
444*89c4ff92SAndroid Build Coastguard Worker
445*89c4ff92SAndroid Build Coastguard Worker // ZLIB client - used by PNG, available for other purposes
446*89c4ff92SAndroid Build Coastguard Worker
447*89c4ff92SAndroid Build Coastguard Worker STBIDEF char *stbi_zlib_decode_malloc_guesssize(const char *buffer, int len, int initial_size, int *outlen);
448*89c4ff92SAndroid Build Coastguard Worker STBIDEF char *stbi_zlib_decode_malloc_guesssize_headerflag(const char *buffer, int len, int initial_size, int *outlen, int parse_header);
449*89c4ff92SAndroid Build Coastguard Worker STBIDEF char *stbi_zlib_decode_malloc(const char *buffer, int len, int *outlen);
450*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_zlib_decode_buffer(char *obuffer, int olen, const char *ibuffer, int ilen);
451*89c4ff92SAndroid Build Coastguard Worker
452*89c4ff92SAndroid Build Coastguard Worker STBIDEF char *stbi_zlib_decode_noheader_malloc(const char *buffer, int len, int *outlen);
453*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_zlib_decode_noheader_buffer(char *obuffer, int olen, const char *ibuffer, int ilen);
454*89c4ff92SAndroid Build Coastguard Worker
455*89c4ff92SAndroid Build Coastguard Worker
456*89c4ff92SAndroid Build Coastguard Worker #ifdef __cplusplus
457*89c4ff92SAndroid Build Coastguard Worker }
458*89c4ff92SAndroid Build Coastguard Worker #endif
459*89c4ff92SAndroid Build Coastguard Worker
460*89c4ff92SAndroid Build Coastguard Worker //
461*89c4ff92SAndroid Build Coastguard Worker //
462*89c4ff92SAndroid Build Coastguard Worker //// end header file /////////////////////////////////////////////////////
463*89c4ff92SAndroid Build Coastguard Worker #endif // STBI_INCLUDE_STB_IMAGE_H
464*89c4ff92SAndroid Build Coastguard Worker
465*89c4ff92SAndroid Build Coastguard Worker #ifdef STB_IMAGE_IMPLEMENTATION
466*89c4ff92SAndroid Build Coastguard Worker
467*89c4ff92SAndroid Build Coastguard Worker #if defined(STBI_ONLY_JPEG) || defined(STBI_ONLY_PNG) || defined(STBI_ONLY_BMP) \
468*89c4ff92SAndroid Build Coastguard Worker || defined(STBI_ONLY_TGA) || defined(STBI_ONLY_GIF) || defined(STBI_ONLY_PSD) \
469*89c4ff92SAndroid Build Coastguard Worker || defined(STBI_ONLY_HDR) || defined(STBI_ONLY_PIC) || defined(STBI_ONLY_PNM) \
470*89c4ff92SAndroid Build Coastguard Worker || defined(STBI_ONLY_ZLIB)
471*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ONLY_JPEG
472*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_JPEG
473*89c4ff92SAndroid Build Coastguard Worker #endif
474*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ONLY_PNG
475*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_PNG
476*89c4ff92SAndroid Build Coastguard Worker #endif
477*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ONLY_BMP
478*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_BMP
479*89c4ff92SAndroid Build Coastguard Worker #endif
480*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ONLY_PSD
481*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_PSD
482*89c4ff92SAndroid Build Coastguard Worker #endif
483*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ONLY_TGA
484*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_TGA
485*89c4ff92SAndroid Build Coastguard Worker #endif
486*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ONLY_GIF
487*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_GIF
488*89c4ff92SAndroid Build Coastguard Worker #endif
489*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ONLY_HDR
490*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_HDR
491*89c4ff92SAndroid Build Coastguard Worker #endif
492*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ONLY_PIC
493*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_PIC
494*89c4ff92SAndroid Build Coastguard Worker #endif
495*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ONLY_PNM
496*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_PNM
497*89c4ff92SAndroid Build Coastguard Worker #endif
498*89c4ff92SAndroid Build Coastguard Worker #endif
499*89c4ff92SAndroid Build Coastguard Worker
500*89c4ff92SAndroid Build Coastguard Worker #if defined(STBI_NO_PNG) && !defined(STBI_SUPPORT_ZLIB) && !defined(STBI_NO_ZLIB)
501*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_ZLIB
502*89c4ff92SAndroid Build Coastguard Worker #endif
503*89c4ff92SAndroid Build Coastguard Worker
504*89c4ff92SAndroid Build Coastguard Worker
505*89c4ff92SAndroid Build Coastguard Worker #include <stdarg.h>
506*89c4ff92SAndroid Build Coastguard Worker #include <stddef.h> // ptrdiff_t on osx
507*89c4ff92SAndroid Build Coastguard Worker #include <stdlib.h>
508*89c4ff92SAndroid Build Coastguard Worker #include <string.h>
509*89c4ff92SAndroid Build Coastguard Worker #include <limits.h>
510*89c4ff92SAndroid Build Coastguard Worker
511*89c4ff92SAndroid Build Coastguard Worker #if !defined(STBI_NO_LINEAR) || !defined(STBI_NO_HDR)
512*89c4ff92SAndroid Build Coastguard Worker #include <math.h> // ldexp
513*89c4ff92SAndroid Build Coastguard Worker #endif
514*89c4ff92SAndroid Build Coastguard Worker
515*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
516*89c4ff92SAndroid Build Coastguard Worker #include <stdio.h>
517*89c4ff92SAndroid Build Coastguard Worker #endif
518*89c4ff92SAndroid Build Coastguard Worker
519*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_ASSERT
520*89c4ff92SAndroid Build Coastguard Worker #include <assert.h>
521*89c4ff92SAndroid Build Coastguard Worker #define STBI_ASSERT(x) assert(x)
522*89c4ff92SAndroid Build Coastguard Worker #endif
523*89c4ff92SAndroid Build Coastguard Worker
524*89c4ff92SAndroid Build Coastguard Worker
525*89c4ff92SAndroid Build Coastguard Worker #ifndef _MSC_VER
526*89c4ff92SAndroid Build Coastguard Worker #ifdef __cplusplus
527*89c4ff92SAndroid Build Coastguard Worker #define stbi_inline inline
528*89c4ff92SAndroid Build Coastguard Worker #else
529*89c4ff92SAndroid Build Coastguard Worker #define stbi_inline
530*89c4ff92SAndroid Build Coastguard Worker #endif
531*89c4ff92SAndroid Build Coastguard Worker #else
532*89c4ff92SAndroid Build Coastguard Worker #define stbi_inline __forceinline
533*89c4ff92SAndroid Build Coastguard Worker #endif
534*89c4ff92SAndroid Build Coastguard Worker
535*89c4ff92SAndroid Build Coastguard Worker
536*89c4ff92SAndroid Build Coastguard Worker #ifdef _MSC_VER
537*89c4ff92SAndroid Build Coastguard Worker typedef unsigned short stbi__uint16;
538*89c4ff92SAndroid Build Coastguard Worker typedef signed short stbi__int16;
539*89c4ff92SAndroid Build Coastguard Worker typedef unsigned int stbi__uint32;
540*89c4ff92SAndroid Build Coastguard Worker typedef signed int stbi__int32;
541*89c4ff92SAndroid Build Coastguard Worker #else
542*89c4ff92SAndroid Build Coastguard Worker #include <stdint.h>
543*89c4ff92SAndroid Build Coastguard Worker typedef uint16_t stbi__uint16;
544*89c4ff92SAndroid Build Coastguard Worker typedef int16_t stbi__int16;
545*89c4ff92SAndroid Build Coastguard Worker typedef uint32_t stbi__uint32;
546*89c4ff92SAndroid Build Coastguard Worker typedef int32_t stbi__int32;
547*89c4ff92SAndroid Build Coastguard Worker #endif
548*89c4ff92SAndroid Build Coastguard Worker
549*89c4ff92SAndroid Build Coastguard Worker // should produce compiler error if size is wrong
550*89c4ff92SAndroid Build Coastguard Worker typedef unsigned char validate_uint32[sizeof(stbi__uint32)==4 ? 1 : -1];
551*89c4ff92SAndroid Build Coastguard Worker
552*89c4ff92SAndroid Build Coastguard Worker #ifdef _MSC_VER
553*89c4ff92SAndroid Build Coastguard Worker #define STBI_NOTUSED(v) (void)(v)
554*89c4ff92SAndroid Build Coastguard Worker #else
555*89c4ff92SAndroid Build Coastguard Worker #define STBI_NOTUSED(v) (void)sizeof(v)
556*89c4ff92SAndroid Build Coastguard Worker #endif
557*89c4ff92SAndroid Build Coastguard Worker
558*89c4ff92SAndroid Build Coastguard Worker #ifdef _MSC_VER
559*89c4ff92SAndroid Build Coastguard Worker #define STBI_HAS_LROTL
560*89c4ff92SAndroid Build Coastguard Worker #endif
561*89c4ff92SAndroid Build Coastguard Worker
562*89c4ff92SAndroid Build Coastguard Worker #ifdef STBI_HAS_LROTL
563*89c4ff92SAndroid Build Coastguard Worker #define stbi_lrot(x,y) _lrotl(x,y)
564*89c4ff92SAndroid Build Coastguard Worker #else
565*89c4ff92SAndroid Build Coastguard Worker #define stbi_lrot(x,y) (((x) << (y)) | ((x) >> (32 - (y))))
566*89c4ff92SAndroid Build Coastguard Worker #endif
567*89c4ff92SAndroid Build Coastguard Worker
568*89c4ff92SAndroid Build Coastguard Worker #if defined(STBI_MALLOC) && defined(STBI_FREE) && (defined(STBI_REALLOC) || defined(STBI_REALLOC_SIZED))
569*89c4ff92SAndroid Build Coastguard Worker // ok
570*89c4ff92SAndroid Build Coastguard Worker #elif !defined(STBI_MALLOC) && !defined(STBI_FREE) && !defined(STBI_REALLOC) && !defined(STBI_REALLOC_SIZED)
571*89c4ff92SAndroid Build Coastguard Worker // ok
572*89c4ff92SAndroid Build Coastguard Worker #else
573*89c4ff92SAndroid Build Coastguard Worker #error "Must define all or none of STBI_MALLOC, STBI_FREE, and STBI_REALLOC (or STBI_REALLOC_SIZED)."
574*89c4ff92SAndroid Build Coastguard Worker #endif
575*89c4ff92SAndroid Build Coastguard Worker
576*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_MALLOC
577*89c4ff92SAndroid Build Coastguard Worker #define STBI_MALLOC(sz) malloc(sz)
578*89c4ff92SAndroid Build Coastguard Worker #define STBI_REALLOC(p,newsz) realloc(p,newsz)
579*89c4ff92SAndroid Build Coastguard Worker #define STBI_FREE(p) free(p)
580*89c4ff92SAndroid Build Coastguard Worker #endif
581*89c4ff92SAndroid Build Coastguard Worker
582*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_REALLOC_SIZED
583*89c4ff92SAndroid Build Coastguard Worker #define STBI_REALLOC_SIZED(p,oldsz,newsz) STBI_REALLOC(p,newsz)
584*89c4ff92SAndroid Build Coastguard Worker #endif
585*89c4ff92SAndroid Build Coastguard Worker
586*89c4ff92SAndroid Build Coastguard Worker // x86/x64 detection
587*89c4ff92SAndroid Build Coastguard Worker #if defined(__x86_64__) || defined(_M_X64)
588*89c4ff92SAndroid Build Coastguard Worker #define STBI__X64_TARGET
589*89c4ff92SAndroid Build Coastguard Worker #elif defined(__i386) || defined(_M_IX86)
590*89c4ff92SAndroid Build Coastguard Worker #define STBI__X86_TARGET
591*89c4ff92SAndroid Build Coastguard Worker #endif
592*89c4ff92SAndroid Build Coastguard Worker
593*89c4ff92SAndroid Build Coastguard Worker #if defined(__GNUC__) && defined(STBI__X86_TARGET) && !defined(__SSE2__) && !defined(STBI_NO_SIMD)
594*89c4ff92SAndroid Build Coastguard Worker // gcc doesn't support sse2 intrinsics unless you compile with -msse2,
595*89c4ff92SAndroid Build Coastguard Worker // which in turn means it gets to use SSE2 everywhere. This is unfortunate,
596*89c4ff92SAndroid Build Coastguard Worker // but previous attempts to provide the SSE2 functions with runtime
597*89c4ff92SAndroid Build Coastguard Worker // detection caused numerous issues. The way architecture extensions are
598*89c4ff92SAndroid Build Coastguard Worker // exposed in GCC/Clang is, sadly, not really suited for one-file libs.
599*89c4ff92SAndroid Build Coastguard Worker // New behavior: if compiled with -msse2, we use SSE2 without any
600*89c4ff92SAndroid Build Coastguard Worker // detection; if not, we don't use it at all.
601*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_SIMD
602*89c4ff92SAndroid Build Coastguard Worker #endif
603*89c4ff92SAndroid Build Coastguard Worker
604*89c4ff92SAndroid Build Coastguard Worker #if defined(__MINGW32__) && defined(STBI__X86_TARGET) && !defined(STBI_MINGW_ENABLE_SSE2) && !defined(STBI_NO_SIMD)
605*89c4ff92SAndroid Build Coastguard Worker // Note that __MINGW32__ doesn't actually mean 32-bit, so we have to avoid STBI__X64_TARGET
606*89c4ff92SAndroid Build Coastguard Worker //
607*89c4ff92SAndroid Build Coastguard Worker // 32-bit MinGW wants ESP to be 16-byte aligned, but this is not in the
608*89c4ff92SAndroid Build Coastguard Worker // Windows ABI and VC++ as well as Windows DLLs don't maintain that invariant.
609*89c4ff92SAndroid Build Coastguard Worker // As a result, enabling SSE2 on 32-bit MinGW is dangerous when not
610*89c4ff92SAndroid Build Coastguard Worker // simultaneously enabling "-mstackrealign".
611*89c4ff92SAndroid Build Coastguard Worker //
612*89c4ff92SAndroid Build Coastguard Worker // See https://github.com/nothings/stb/issues/81 for more information.
613*89c4ff92SAndroid Build Coastguard Worker //
614*89c4ff92SAndroid Build Coastguard Worker // So default to no SSE2 on 32-bit MinGW. If you've read this far and added
615*89c4ff92SAndroid Build Coastguard Worker // -mstackrealign to your build settings, feel free to #define STBI_MINGW_ENABLE_SSE2.
616*89c4ff92SAndroid Build Coastguard Worker #define STBI_NO_SIMD
617*89c4ff92SAndroid Build Coastguard Worker #endif
618*89c4ff92SAndroid Build Coastguard Worker
619*89c4ff92SAndroid Build Coastguard Worker #if !defined(STBI_NO_SIMD) && (defined(STBI__X86_TARGET) || defined(STBI__X64_TARGET))
620*89c4ff92SAndroid Build Coastguard Worker #define STBI_SSE2
621*89c4ff92SAndroid Build Coastguard Worker #include <emmintrin.h>
622*89c4ff92SAndroid Build Coastguard Worker
623*89c4ff92SAndroid Build Coastguard Worker #ifdef _MSC_VER
624*89c4ff92SAndroid Build Coastguard Worker
625*89c4ff92SAndroid Build Coastguard Worker #if _MSC_VER >= 1400 // not VC6
626*89c4ff92SAndroid Build Coastguard Worker #include <intrin.h> // __cpuid
stbi__cpuid3(void)627*89c4ff92SAndroid Build Coastguard Worker static int stbi__cpuid3(void)
628*89c4ff92SAndroid Build Coastguard Worker {
629*89c4ff92SAndroid Build Coastguard Worker int info[4];
630*89c4ff92SAndroid Build Coastguard Worker __cpuid(info,1);
631*89c4ff92SAndroid Build Coastguard Worker return info[3];
632*89c4ff92SAndroid Build Coastguard Worker }
633*89c4ff92SAndroid Build Coastguard Worker #else
stbi__cpuid3(void)634*89c4ff92SAndroid Build Coastguard Worker static int stbi__cpuid3(void)
635*89c4ff92SAndroid Build Coastguard Worker {
636*89c4ff92SAndroid Build Coastguard Worker int res;
637*89c4ff92SAndroid Build Coastguard Worker __asm {
638*89c4ff92SAndroid Build Coastguard Worker mov eax,1
639*89c4ff92SAndroid Build Coastguard Worker cpuid
640*89c4ff92SAndroid Build Coastguard Worker mov res,edx
641*89c4ff92SAndroid Build Coastguard Worker }
642*89c4ff92SAndroid Build Coastguard Worker return res;
643*89c4ff92SAndroid Build Coastguard Worker }
644*89c4ff92SAndroid Build Coastguard Worker #endif
645*89c4ff92SAndroid Build Coastguard Worker
646*89c4ff92SAndroid Build Coastguard Worker #define STBI_SIMD_ALIGN(type, name) __declspec(align(16)) type name
647*89c4ff92SAndroid Build Coastguard Worker
stbi__sse2_available(void)648*89c4ff92SAndroid Build Coastguard Worker static int stbi__sse2_available(void)
649*89c4ff92SAndroid Build Coastguard Worker {
650*89c4ff92SAndroid Build Coastguard Worker int info3 = stbi__cpuid3();
651*89c4ff92SAndroid Build Coastguard Worker return ((info3 >> 26) & 1) != 0;
652*89c4ff92SAndroid Build Coastguard Worker }
653*89c4ff92SAndroid Build Coastguard Worker #else // assume GCC-style if not VC++
654*89c4ff92SAndroid Build Coastguard Worker #define STBI_SIMD_ALIGN(type, name) type name __attribute__((aligned(16)))
655*89c4ff92SAndroid Build Coastguard Worker
stbi__sse2_available(void)656*89c4ff92SAndroid Build Coastguard Worker static int stbi__sse2_available(void)
657*89c4ff92SAndroid Build Coastguard Worker {
658*89c4ff92SAndroid Build Coastguard Worker // If we're even attempting to compile this on GCC/Clang, that means
659*89c4ff92SAndroid Build Coastguard Worker // -msse2 is on, which means the compiler is allowed to use SSE2
660*89c4ff92SAndroid Build Coastguard Worker // instructions at will, and so are we.
661*89c4ff92SAndroid Build Coastguard Worker return 1;
662*89c4ff92SAndroid Build Coastguard Worker }
663*89c4ff92SAndroid Build Coastguard Worker #endif
664*89c4ff92SAndroid Build Coastguard Worker #endif
665*89c4ff92SAndroid Build Coastguard Worker
666*89c4ff92SAndroid Build Coastguard Worker // ARM NEON
667*89c4ff92SAndroid Build Coastguard Worker #if defined(STBI_NO_SIMD) && defined(STBI_NEON)
668*89c4ff92SAndroid Build Coastguard Worker #undef STBI_NEON
669*89c4ff92SAndroid Build Coastguard Worker #endif
670*89c4ff92SAndroid Build Coastguard Worker
671*89c4ff92SAndroid Build Coastguard Worker #ifdef STBI_NEON
672*89c4ff92SAndroid Build Coastguard Worker #include <arm_neon.h>
673*89c4ff92SAndroid Build Coastguard Worker // assume GCC or Clang on ARM targets
674*89c4ff92SAndroid Build Coastguard Worker #define STBI_SIMD_ALIGN(type, name) type name __attribute__((aligned(16)))
675*89c4ff92SAndroid Build Coastguard Worker #endif
676*89c4ff92SAndroid Build Coastguard Worker
677*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_SIMD_ALIGN
678*89c4ff92SAndroid Build Coastguard Worker #define STBI_SIMD_ALIGN(type, name) type name
679*89c4ff92SAndroid Build Coastguard Worker #endif
680*89c4ff92SAndroid Build Coastguard Worker
681*89c4ff92SAndroid Build Coastguard Worker ///////////////////////////////////////////////
682*89c4ff92SAndroid Build Coastguard Worker //
683*89c4ff92SAndroid Build Coastguard Worker // stbi__context struct and start_xxx functions
684*89c4ff92SAndroid Build Coastguard Worker
685*89c4ff92SAndroid Build Coastguard Worker // stbi__context structure is our basic context used by all images, so it
686*89c4ff92SAndroid Build Coastguard Worker // contains all the IO context, plus some basic image information
687*89c4ff92SAndroid Build Coastguard Worker typedef struct
688*89c4ff92SAndroid Build Coastguard Worker {
689*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 img_x, img_y;
690*89c4ff92SAndroid Build Coastguard Worker int img_n, img_out_n;
691*89c4ff92SAndroid Build Coastguard Worker
692*89c4ff92SAndroid Build Coastguard Worker stbi_io_callbacks io;
693*89c4ff92SAndroid Build Coastguard Worker void *io_user_data;
694*89c4ff92SAndroid Build Coastguard Worker
695*89c4ff92SAndroid Build Coastguard Worker int read_from_callbacks;
696*89c4ff92SAndroid Build Coastguard Worker int buflen;
697*89c4ff92SAndroid Build Coastguard Worker stbi_uc buffer_start[128];
698*89c4ff92SAndroid Build Coastguard Worker
699*89c4ff92SAndroid Build Coastguard Worker stbi_uc *img_buffer, *img_buffer_end;
700*89c4ff92SAndroid Build Coastguard Worker stbi_uc *img_buffer_original, *img_buffer_original_end;
701*89c4ff92SAndroid Build Coastguard Worker } stbi__context;
702*89c4ff92SAndroid Build Coastguard Worker
703*89c4ff92SAndroid Build Coastguard Worker
704*89c4ff92SAndroid Build Coastguard Worker static void stbi__refill_buffer(stbi__context *s);
705*89c4ff92SAndroid Build Coastguard Worker
706*89c4ff92SAndroid Build Coastguard Worker // initialize a memory-decode context
stbi__start_mem(stbi__context * s,stbi_uc const * buffer,int len)707*89c4ff92SAndroid Build Coastguard Worker static void stbi__start_mem(stbi__context *s, stbi_uc const *buffer, int len)
708*89c4ff92SAndroid Build Coastguard Worker {
709*89c4ff92SAndroid Build Coastguard Worker s->io.read = NULL;
710*89c4ff92SAndroid Build Coastguard Worker s->read_from_callbacks = 0;
711*89c4ff92SAndroid Build Coastguard Worker s->img_buffer = s->img_buffer_original = (stbi_uc *) buffer;
712*89c4ff92SAndroid Build Coastguard Worker s->img_buffer_end = s->img_buffer_original_end = (stbi_uc *) buffer+len;
713*89c4ff92SAndroid Build Coastguard Worker }
714*89c4ff92SAndroid Build Coastguard Worker
715*89c4ff92SAndroid Build Coastguard Worker // initialize a callback-based context
stbi__start_callbacks(stbi__context * s,stbi_io_callbacks * c,void * user)716*89c4ff92SAndroid Build Coastguard Worker static void stbi__start_callbacks(stbi__context *s, stbi_io_callbacks *c, void *user)
717*89c4ff92SAndroid Build Coastguard Worker {
718*89c4ff92SAndroid Build Coastguard Worker s->io = *c;
719*89c4ff92SAndroid Build Coastguard Worker s->io_user_data = user;
720*89c4ff92SAndroid Build Coastguard Worker s->buflen = sizeof(s->buffer_start);
721*89c4ff92SAndroid Build Coastguard Worker s->read_from_callbacks = 1;
722*89c4ff92SAndroid Build Coastguard Worker s->img_buffer_original = s->buffer_start;
723*89c4ff92SAndroid Build Coastguard Worker stbi__refill_buffer(s);
724*89c4ff92SAndroid Build Coastguard Worker s->img_buffer_original_end = s->img_buffer_end;
725*89c4ff92SAndroid Build Coastguard Worker }
726*89c4ff92SAndroid Build Coastguard Worker
727*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
728*89c4ff92SAndroid Build Coastguard Worker
stbi__stdio_read(void * user,char * data,int size)729*89c4ff92SAndroid Build Coastguard Worker static int stbi__stdio_read(void *user, char *data, int size)
730*89c4ff92SAndroid Build Coastguard Worker {
731*89c4ff92SAndroid Build Coastguard Worker return (int) fread(data,1,size,(FILE*) user);
732*89c4ff92SAndroid Build Coastguard Worker }
733*89c4ff92SAndroid Build Coastguard Worker
stbi__stdio_skip(void * user,int n)734*89c4ff92SAndroid Build Coastguard Worker static void stbi__stdio_skip(void *user, int n)
735*89c4ff92SAndroid Build Coastguard Worker {
736*89c4ff92SAndroid Build Coastguard Worker fseek((FILE*) user, n, SEEK_CUR);
737*89c4ff92SAndroid Build Coastguard Worker }
738*89c4ff92SAndroid Build Coastguard Worker
stbi__stdio_eof(void * user)739*89c4ff92SAndroid Build Coastguard Worker static int stbi__stdio_eof(void *user)
740*89c4ff92SAndroid Build Coastguard Worker {
741*89c4ff92SAndroid Build Coastguard Worker return feof((FILE*) user);
742*89c4ff92SAndroid Build Coastguard Worker }
743*89c4ff92SAndroid Build Coastguard Worker
744*89c4ff92SAndroid Build Coastguard Worker static stbi_io_callbacks stbi__stdio_callbacks =
745*89c4ff92SAndroid Build Coastguard Worker {
746*89c4ff92SAndroid Build Coastguard Worker stbi__stdio_read,
747*89c4ff92SAndroid Build Coastguard Worker stbi__stdio_skip,
748*89c4ff92SAndroid Build Coastguard Worker stbi__stdio_eof,
749*89c4ff92SAndroid Build Coastguard Worker };
750*89c4ff92SAndroid Build Coastguard Worker
stbi__start_file(stbi__context * s,FILE * f)751*89c4ff92SAndroid Build Coastguard Worker static void stbi__start_file(stbi__context *s, FILE *f)
752*89c4ff92SAndroid Build Coastguard Worker {
753*89c4ff92SAndroid Build Coastguard Worker stbi__start_callbacks(s, &stbi__stdio_callbacks, (void *) f);
754*89c4ff92SAndroid Build Coastguard Worker }
755*89c4ff92SAndroid Build Coastguard Worker
756*89c4ff92SAndroid Build Coastguard Worker //static void stop_file(stbi__context *s) { }
757*89c4ff92SAndroid Build Coastguard Worker
758*89c4ff92SAndroid Build Coastguard Worker #endif // !STBI_NO_STDIO
759*89c4ff92SAndroid Build Coastguard Worker
stbi__rewind(stbi__context * s)760*89c4ff92SAndroid Build Coastguard Worker static void stbi__rewind(stbi__context *s)
761*89c4ff92SAndroid Build Coastguard Worker {
762*89c4ff92SAndroid Build Coastguard Worker // conceptually rewind SHOULD rewind to the beginning of the stream,
763*89c4ff92SAndroid Build Coastguard Worker // but we just rewind to the beginning of the initial buffer, because
764*89c4ff92SAndroid Build Coastguard Worker // we only use it after doing 'test', which only ever looks at at most 92 bytes
765*89c4ff92SAndroid Build Coastguard Worker s->img_buffer = s->img_buffer_original;
766*89c4ff92SAndroid Build Coastguard Worker s->img_buffer_end = s->img_buffer_original_end;
767*89c4ff92SAndroid Build Coastguard Worker }
768*89c4ff92SAndroid Build Coastguard Worker
769*89c4ff92SAndroid Build Coastguard Worker enum
770*89c4ff92SAndroid Build Coastguard Worker {
771*89c4ff92SAndroid Build Coastguard Worker STBI_ORDER_RGB,
772*89c4ff92SAndroid Build Coastguard Worker STBI_ORDER_BGR
773*89c4ff92SAndroid Build Coastguard Worker };
774*89c4ff92SAndroid Build Coastguard Worker
775*89c4ff92SAndroid Build Coastguard Worker typedef struct
776*89c4ff92SAndroid Build Coastguard Worker {
777*89c4ff92SAndroid Build Coastguard Worker int bits_per_channel;
778*89c4ff92SAndroid Build Coastguard Worker int num_channels;
779*89c4ff92SAndroid Build Coastguard Worker int channel_order;
780*89c4ff92SAndroid Build Coastguard Worker } stbi__result_info;
781*89c4ff92SAndroid Build Coastguard Worker
782*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_JPEG
783*89c4ff92SAndroid Build Coastguard Worker static int stbi__jpeg_test(stbi__context *s);
784*89c4ff92SAndroid Build Coastguard Worker static void *stbi__jpeg_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri);
785*89c4ff92SAndroid Build Coastguard Worker static int stbi__jpeg_info(stbi__context *s, int *x, int *y, int *comp);
786*89c4ff92SAndroid Build Coastguard Worker #endif
787*89c4ff92SAndroid Build Coastguard Worker
788*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PNG
789*89c4ff92SAndroid Build Coastguard Worker static int stbi__png_test(stbi__context *s);
790*89c4ff92SAndroid Build Coastguard Worker static void *stbi__png_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri);
791*89c4ff92SAndroid Build Coastguard Worker static int stbi__png_info(stbi__context *s, int *x, int *y, int *comp);
792*89c4ff92SAndroid Build Coastguard Worker #endif
793*89c4ff92SAndroid Build Coastguard Worker
794*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_BMP
795*89c4ff92SAndroid Build Coastguard Worker static int stbi__bmp_test(stbi__context *s);
796*89c4ff92SAndroid Build Coastguard Worker static void *stbi__bmp_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri);
797*89c4ff92SAndroid Build Coastguard Worker static int stbi__bmp_info(stbi__context *s, int *x, int *y, int *comp);
798*89c4ff92SAndroid Build Coastguard Worker #endif
799*89c4ff92SAndroid Build Coastguard Worker
800*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_TGA
801*89c4ff92SAndroid Build Coastguard Worker static int stbi__tga_test(stbi__context *s);
802*89c4ff92SAndroid Build Coastguard Worker static void *stbi__tga_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri);
803*89c4ff92SAndroid Build Coastguard Worker static int stbi__tga_info(stbi__context *s, int *x, int *y, int *comp);
804*89c4ff92SAndroid Build Coastguard Worker #endif
805*89c4ff92SAndroid Build Coastguard Worker
806*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PSD
807*89c4ff92SAndroid Build Coastguard Worker static int stbi__psd_test(stbi__context *s);
808*89c4ff92SAndroid Build Coastguard Worker static void *stbi__psd_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri, int bpc);
809*89c4ff92SAndroid Build Coastguard Worker static int stbi__psd_info(stbi__context *s, int *x, int *y, int *comp);
810*89c4ff92SAndroid Build Coastguard Worker #endif
811*89c4ff92SAndroid Build Coastguard Worker
812*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
813*89c4ff92SAndroid Build Coastguard Worker static int stbi__hdr_test(stbi__context *s);
814*89c4ff92SAndroid Build Coastguard Worker static float *stbi__hdr_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri);
815*89c4ff92SAndroid Build Coastguard Worker static int stbi__hdr_info(stbi__context *s, int *x, int *y, int *comp);
816*89c4ff92SAndroid Build Coastguard Worker #endif
817*89c4ff92SAndroid Build Coastguard Worker
818*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PIC
819*89c4ff92SAndroid Build Coastguard Worker static int stbi__pic_test(stbi__context *s);
820*89c4ff92SAndroid Build Coastguard Worker static void *stbi__pic_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri);
821*89c4ff92SAndroid Build Coastguard Worker static int stbi__pic_info(stbi__context *s, int *x, int *y, int *comp);
822*89c4ff92SAndroid Build Coastguard Worker #endif
823*89c4ff92SAndroid Build Coastguard Worker
824*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_GIF
825*89c4ff92SAndroid Build Coastguard Worker static int stbi__gif_test(stbi__context *s);
826*89c4ff92SAndroid Build Coastguard Worker static void *stbi__gif_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri);
827*89c4ff92SAndroid Build Coastguard Worker static int stbi__gif_info(stbi__context *s, int *x, int *y, int *comp);
828*89c4ff92SAndroid Build Coastguard Worker #endif
829*89c4ff92SAndroid Build Coastguard Worker
830*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PNM
831*89c4ff92SAndroid Build Coastguard Worker static int stbi__pnm_test(stbi__context *s);
832*89c4ff92SAndroid Build Coastguard Worker static void *stbi__pnm_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri);
833*89c4ff92SAndroid Build Coastguard Worker static int stbi__pnm_info(stbi__context *s, int *x, int *y, int *comp);
834*89c4ff92SAndroid Build Coastguard Worker #endif
835*89c4ff92SAndroid Build Coastguard Worker
836*89c4ff92SAndroid Build Coastguard Worker // this is not threadsafe
837*89c4ff92SAndroid Build Coastguard Worker static const char *stbi__g_failure_reason;
838*89c4ff92SAndroid Build Coastguard Worker
stbi_failure_reason(void)839*89c4ff92SAndroid Build Coastguard Worker STBIDEF const char *stbi_failure_reason(void)
840*89c4ff92SAndroid Build Coastguard Worker {
841*89c4ff92SAndroid Build Coastguard Worker return stbi__g_failure_reason;
842*89c4ff92SAndroid Build Coastguard Worker }
843*89c4ff92SAndroid Build Coastguard Worker
stbi__err(const char * str)844*89c4ff92SAndroid Build Coastguard Worker static int stbi__err(const char *str)
845*89c4ff92SAndroid Build Coastguard Worker {
846*89c4ff92SAndroid Build Coastguard Worker stbi__g_failure_reason = str;
847*89c4ff92SAndroid Build Coastguard Worker return 0;
848*89c4ff92SAndroid Build Coastguard Worker }
849*89c4ff92SAndroid Build Coastguard Worker
stbi__malloc(size_t size)850*89c4ff92SAndroid Build Coastguard Worker static void *stbi__malloc(size_t size)
851*89c4ff92SAndroid Build Coastguard Worker {
852*89c4ff92SAndroid Build Coastguard Worker return STBI_MALLOC(size);
853*89c4ff92SAndroid Build Coastguard Worker }
854*89c4ff92SAndroid Build Coastguard Worker
855*89c4ff92SAndroid Build Coastguard Worker // stb_image uses ints pervasively, including for offset calculations.
856*89c4ff92SAndroid Build Coastguard Worker // therefore the largest decoded image size we can support with the
857*89c4ff92SAndroid Build Coastguard Worker // current code, even on 64-bit targets, is INT_MAX. this is not a
858*89c4ff92SAndroid Build Coastguard Worker // significant limitation for the intended use case.
859*89c4ff92SAndroid Build Coastguard Worker //
860*89c4ff92SAndroid Build Coastguard Worker // we do, however, need to make sure our size calculations don't
861*89c4ff92SAndroid Build Coastguard Worker // overflow. hence a few helper functions for size calculations that
862*89c4ff92SAndroid Build Coastguard Worker // multiply integers together, making sure that they're non-negative
863*89c4ff92SAndroid Build Coastguard Worker // and no overflow occurs.
864*89c4ff92SAndroid Build Coastguard Worker
865*89c4ff92SAndroid Build Coastguard Worker // return 1 if the sum is valid, 0 on overflow.
866*89c4ff92SAndroid Build Coastguard Worker // negative terms are considered invalid.
stbi__addsizes_valid(int a,int b)867*89c4ff92SAndroid Build Coastguard Worker static int stbi__addsizes_valid(int a, int b)
868*89c4ff92SAndroid Build Coastguard Worker {
869*89c4ff92SAndroid Build Coastguard Worker if (b < 0) return 0;
870*89c4ff92SAndroid Build Coastguard Worker // now 0 <= b <= INT_MAX, hence also
871*89c4ff92SAndroid Build Coastguard Worker // 0 <= INT_MAX - b <= INTMAX.
872*89c4ff92SAndroid Build Coastguard Worker // And "a + b <= INT_MAX" (which might overflow) is the
873*89c4ff92SAndroid Build Coastguard Worker // same as a <= INT_MAX - b (no overflow)
874*89c4ff92SAndroid Build Coastguard Worker return a <= INT_MAX - b;
875*89c4ff92SAndroid Build Coastguard Worker }
876*89c4ff92SAndroid Build Coastguard Worker
877*89c4ff92SAndroid Build Coastguard Worker // returns 1 if the product is valid, 0 on overflow.
878*89c4ff92SAndroid Build Coastguard Worker // negative factors are considered invalid.
stbi__mul2sizes_valid(int a,int b)879*89c4ff92SAndroid Build Coastguard Worker static int stbi__mul2sizes_valid(int a, int b)
880*89c4ff92SAndroid Build Coastguard Worker {
881*89c4ff92SAndroid Build Coastguard Worker if (a < 0 || b < 0) return 0;
882*89c4ff92SAndroid Build Coastguard Worker if (b == 0) return 1; // mul-by-0 is always safe
883*89c4ff92SAndroid Build Coastguard Worker // portable way to check for no overflows in a*b
884*89c4ff92SAndroid Build Coastguard Worker return a <= INT_MAX/b;
885*89c4ff92SAndroid Build Coastguard Worker }
886*89c4ff92SAndroid Build Coastguard Worker
887*89c4ff92SAndroid Build Coastguard Worker // returns 1 if "a*b + add" has no negative terms/factors and doesn't overflow
stbi__mad2sizes_valid(int a,int b,int add)888*89c4ff92SAndroid Build Coastguard Worker static int stbi__mad2sizes_valid(int a, int b, int add)
889*89c4ff92SAndroid Build Coastguard Worker {
890*89c4ff92SAndroid Build Coastguard Worker return stbi__mul2sizes_valid(a, b) && stbi__addsizes_valid(a*b, add);
891*89c4ff92SAndroid Build Coastguard Worker }
892*89c4ff92SAndroid Build Coastguard Worker
893*89c4ff92SAndroid Build Coastguard Worker // returns 1 if "a*b*c + add" has no negative terms/factors and doesn't overflow
stbi__mad3sizes_valid(int a,int b,int c,int add)894*89c4ff92SAndroid Build Coastguard Worker static int stbi__mad3sizes_valid(int a, int b, int c, int add)
895*89c4ff92SAndroid Build Coastguard Worker {
896*89c4ff92SAndroid Build Coastguard Worker return stbi__mul2sizes_valid(a, b) && stbi__mul2sizes_valid(a*b, c) &&
897*89c4ff92SAndroid Build Coastguard Worker stbi__addsizes_valid(a*b*c, add);
898*89c4ff92SAndroid Build Coastguard Worker }
899*89c4ff92SAndroid Build Coastguard Worker
900*89c4ff92SAndroid Build Coastguard Worker // returns 1 if "a*b*c*d + add" has no negative terms/factors and doesn't overflow
stbi__mad4sizes_valid(int a,int b,int c,int d,int add)901*89c4ff92SAndroid Build Coastguard Worker static int stbi__mad4sizes_valid(int a, int b, int c, int d, int add)
902*89c4ff92SAndroid Build Coastguard Worker {
903*89c4ff92SAndroid Build Coastguard Worker return stbi__mul2sizes_valid(a, b) && stbi__mul2sizes_valid(a*b, c) &&
904*89c4ff92SAndroid Build Coastguard Worker stbi__mul2sizes_valid(a*b*c, d) && stbi__addsizes_valid(a*b*c*d, add);
905*89c4ff92SAndroid Build Coastguard Worker }
906*89c4ff92SAndroid Build Coastguard Worker
907*89c4ff92SAndroid Build Coastguard Worker // mallocs with size overflow checking
stbi__malloc_mad2(int a,int b,int add)908*89c4ff92SAndroid Build Coastguard Worker static void *stbi__malloc_mad2(int a, int b, int add)
909*89c4ff92SAndroid Build Coastguard Worker {
910*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad2sizes_valid(a, b, add)) return NULL;
911*89c4ff92SAndroid Build Coastguard Worker return stbi__malloc(a*b + add);
912*89c4ff92SAndroid Build Coastguard Worker }
913*89c4ff92SAndroid Build Coastguard Worker
stbi__malloc_mad3(int a,int b,int c,int add)914*89c4ff92SAndroid Build Coastguard Worker static void *stbi__malloc_mad3(int a, int b, int c, int add)
915*89c4ff92SAndroid Build Coastguard Worker {
916*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad3sizes_valid(a, b, c, add)) return NULL;
917*89c4ff92SAndroid Build Coastguard Worker return stbi__malloc(a*b*c + add);
918*89c4ff92SAndroid Build Coastguard Worker }
919*89c4ff92SAndroid Build Coastguard Worker
stbi__malloc_mad4(int a,int b,int c,int d,int add)920*89c4ff92SAndroid Build Coastguard Worker static void *stbi__malloc_mad4(int a, int b, int c, int d, int add)
921*89c4ff92SAndroid Build Coastguard Worker {
922*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad4sizes_valid(a, b, c, d, add)) return NULL;
923*89c4ff92SAndroid Build Coastguard Worker return stbi__malloc(a*b*c*d + add);
924*89c4ff92SAndroid Build Coastguard Worker }
925*89c4ff92SAndroid Build Coastguard Worker
926*89c4ff92SAndroid Build Coastguard Worker // stbi__err - error
927*89c4ff92SAndroid Build Coastguard Worker // stbi__errpf - error returning pointer to float
928*89c4ff92SAndroid Build Coastguard Worker // stbi__errpuc - error returning pointer to unsigned char
929*89c4ff92SAndroid Build Coastguard Worker
930*89c4ff92SAndroid Build Coastguard Worker #ifdef STBI_NO_FAILURE_STRINGS
931*89c4ff92SAndroid Build Coastguard Worker #define stbi__err(x,y) 0
932*89c4ff92SAndroid Build Coastguard Worker #elif defined(STBI_FAILURE_USERMSG)
933*89c4ff92SAndroid Build Coastguard Worker #define stbi__err(x,y) stbi__err(y)
934*89c4ff92SAndroid Build Coastguard Worker #else
935*89c4ff92SAndroid Build Coastguard Worker #define stbi__err(x,y) stbi__err(x)
936*89c4ff92SAndroid Build Coastguard Worker #endif
937*89c4ff92SAndroid Build Coastguard Worker
938*89c4ff92SAndroid Build Coastguard Worker #define stbi__errpf(x,y) ((float *)(size_t) (stbi__err(x,y)?NULL:NULL))
939*89c4ff92SAndroid Build Coastguard Worker #define stbi__errpuc(x,y) ((unsigned char *)(size_t) (stbi__err(x,y)?NULL:NULL))
940*89c4ff92SAndroid Build Coastguard Worker
stbi_image_free(void * retval_from_stbi_load)941*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_image_free(void *retval_from_stbi_load)
942*89c4ff92SAndroid Build Coastguard Worker {
943*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(retval_from_stbi_load);
944*89c4ff92SAndroid Build Coastguard Worker }
945*89c4ff92SAndroid Build Coastguard Worker
946*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_LINEAR
947*89c4ff92SAndroid Build Coastguard Worker static float *stbi__ldr_to_hdr(stbi_uc *data, int x, int y, int comp);
948*89c4ff92SAndroid Build Coastguard Worker #endif
949*89c4ff92SAndroid Build Coastguard Worker
950*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
951*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__hdr_to_ldr(float *data, int x, int y, int comp);
952*89c4ff92SAndroid Build Coastguard Worker #endif
953*89c4ff92SAndroid Build Coastguard Worker
954*89c4ff92SAndroid Build Coastguard Worker static int stbi__vertically_flip_on_load = 0;
955*89c4ff92SAndroid Build Coastguard Worker
stbi_set_flip_vertically_on_load(int flag_true_if_should_flip)956*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_set_flip_vertically_on_load(int flag_true_if_should_flip)
957*89c4ff92SAndroid Build Coastguard Worker {
958*89c4ff92SAndroid Build Coastguard Worker stbi__vertically_flip_on_load = flag_true_if_should_flip;
959*89c4ff92SAndroid Build Coastguard Worker }
960*89c4ff92SAndroid Build Coastguard Worker
stbi__load_main(stbi__context * s,int * x,int * y,int * comp,int req_comp,stbi__result_info * ri,int bpc)961*89c4ff92SAndroid Build Coastguard Worker static void *stbi__load_main(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri, int bpc)
962*89c4ff92SAndroid Build Coastguard Worker {
963*89c4ff92SAndroid Build Coastguard Worker memset(ri, 0, sizeof(*ri)); // make sure it's initialized if we add new fields
964*89c4ff92SAndroid Build Coastguard Worker ri->bits_per_channel = 8; // default is 8 so most paths don't have to be changed
965*89c4ff92SAndroid Build Coastguard Worker ri->channel_order = STBI_ORDER_RGB; // all current input & output are this, but this is here so we can add BGR order
966*89c4ff92SAndroid Build Coastguard Worker ri->num_channels = 0;
967*89c4ff92SAndroid Build Coastguard Worker
968*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_JPEG
969*89c4ff92SAndroid Build Coastguard Worker if (stbi__jpeg_test(s)) return stbi__jpeg_load(s,x,y,comp,req_comp, ri);
970*89c4ff92SAndroid Build Coastguard Worker #endif
971*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PNG
972*89c4ff92SAndroid Build Coastguard Worker if (stbi__png_test(s)) return stbi__png_load(s,x,y,comp,req_comp, ri);
973*89c4ff92SAndroid Build Coastguard Worker #endif
974*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_BMP
975*89c4ff92SAndroid Build Coastguard Worker if (stbi__bmp_test(s)) return stbi__bmp_load(s,x,y,comp,req_comp, ri);
976*89c4ff92SAndroid Build Coastguard Worker #endif
977*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_GIF
978*89c4ff92SAndroid Build Coastguard Worker if (stbi__gif_test(s)) return stbi__gif_load(s,x,y,comp,req_comp, ri);
979*89c4ff92SAndroid Build Coastguard Worker #endif
980*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PSD
981*89c4ff92SAndroid Build Coastguard Worker if (stbi__psd_test(s)) return stbi__psd_load(s,x,y,comp,req_comp, ri, bpc);
982*89c4ff92SAndroid Build Coastguard Worker #endif
983*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PIC
984*89c4ff92SAndroid Build Coastguard Worker if (stbi__pic_test(s)) return stbi__pic_load(s,x,y,comp,req_comp, ri);
985*89c4ff92SAndroid Build Coastguard Worker #endif
986*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PNM
987*89c4ff92SAndroid Build Coastguard Worker if (stbi__pnm_test(s)) return stbi__pnm_load(s,x,y,comp,req_comp, ri);
988*89c4ff92SAndroid Build Coastguard Worker #endif
989*89c4ff92SAndroid Build Coastguard Worker
990*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
991*89c4ff92SAndroid Build Coastguard Worker if (stbi__hdr_test(s)) {
992*89c4ff92SAndroid Build Coastguard Worker float *hdr = stbi__hdr_load(s, x,y,comp,req_comp, ri);
993*89c4ff92SAndroid Build Coastguard Worker return stbi__hdr_to_ldr(hdr, *x, *y, req_comp ? req_comp : *comp);
994*89c4ff92SAndroid Build Coastguard Worker }
995*89c4ff92SAndroid Build Coastguard Worker #endif
996*89c4ff92SAndroid Build Coastguard Worker
997*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_TGA
998*89c4ff92SAndroid Build Coastguard Worker // test tga last because it's a crappy test!
999*89c4ff92SAndroid Build Coastguard Worker if (stbi__tga_test(s))
1000*89c4ff92SAndroid Build Coastguard Worker return stbi__tga_load(s,x,y,comp,req_comp, ri);
1001*89c4ff92SAndroid Build Coastguard Worker #endif
1002*89c4ff92SAndroid Build Coastguard Worker
1003*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("unknown image type", "Image not of any known type, or corrupt");
1004*89c4ff92SAndroid Build Coastguard Worker }
1005*89c4ff92SAndroid Build Coastguard Worker
stbi__convert_16_to_8(stbi__uint16 * orig,int w,int h,int channels)1006*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__convert_16_to_8(stbi__uint16 *orig, int w, int h, int channels)
1007*89c4ff92SAndroid Build Coastguard Worker {
1008*89c4ff92SAndroid Build Coastguard Worker int i;
1009*89c4ff92SAndroid Build Coastguard Worker int img_len = w * h * channels;
1010*89c4ff92SAndroid Build Coastguard Worker stbi_uc *reduced;
1011*89c4ff92SAndroid Build Coastguard Worker
1012*89c4ff92SAndroid Build Coastguard Worker reduced = (stbi_uc *) stbi__malloc(img_len);
1013*89c4ff92SAndroid Build Coastguard Worker if (reduced == NULL) return stbi__errpuc("outofmem", "Out of memory");
1014*89c4ff92SAndroid Build Coastguard Worker
1015*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < img_len; ++i)
1016*89c4ff92SAndroid Build Coastguard Worker reduced[i] = (stbi_uc)((orig[i] >> 8) & 0xFF); // top half of each byte is sufficient approx of 16->8 bit scaling
1017*89c4ff92SAndroid Build Coastguard Worker
1018*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(orig);
1019*89c4ff92SAndroid Build Coastguard Worker return reduced;
1020*89c4ff92SAndroid Build Coastguard Worker }
1021*89c4ff92SAndroid Build Coastguard Worker
stbi__convert_8_to_16(stbi_uc * orig,int w,int h,int channels)1022*89c4ff92SAndroid Build Coastguard Worker static stbi__uint16 *stbi__convert_8_to_16(stbi_uc *orig, int w, int h, int channels)
1023*89c4ff92SAndroid Build Coastguard Worker {
1024*89c4ff92SAndroid Build Coastguard Worker int i;
1025*89c4ff92SAndroid Build Coastguard Worker int img_len = w * h * channels;
1026*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *enlarged;
1027*89c4ff92SAndroid Build Coastguard Worker
1028*89c4ff92SAndroid Build Coastguard Worker enlarged = (stbi__uint16 *) stbi__malloc(img_len*2);
1029*89c4ff92SAndroid Build Coastguard Worker if (enlarged == NULL) return (stbi__uint16 *) stbi__errpuc("outofmem", "Out of memory");
1030*89c4ff92SAndroid Build Coastguard Worker
1031*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < img_len; ++i)
1032*89c4ff92SAndroid Build Coastguard Worker enlarged[i] = (stbi__uint16)((orig[i] << 8) + orig[i]); // replicate to high and low byte, maps 0->0, 255->0xffff
1033*89c4ff92SAndroid Build Coastguard Worker
1034*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(orig);
1035*89c4ff92SAndroid Build Coastguard Worker return enlarged;
1036*89c4ff92SAndroid Build Coastguard Worker }
1037*89c4ff92SAndroid Build Coastguard Worker
stbi__vertical_flip(void * image,int w,int h,int bytes_per_pixel)1038*89c4ff92SAndroid Build Coastguard Worker static void stbi__vertical_flip(void *image, int w, int h, int bytes_per_pixel)
1039*89c4ff92SAndroid Build Coastguard Worker {
1040*89c4ff92SAndroid Build Coastguard Worker int row;
1041*89c4ff92SAndroid Build Coastguard Worker size_t bytes_per_row = (size_t)w * bytes_per_pixel;
1042*89c4ff92SAndroid Build Coastguard Worker stbi_uc temp[2048];
1043*89c4ff92SAndroid Build Coastguard Worker stbi_uc *bytes = (stbi_uc *)image;
1044*89c4ff92SAndroid Build Coastguard Worker
1045*89c4ff92SAndroid Build Coastguard Worker for (row = 0; row < (h>>1); row++) {
1046*89c4ff92SAndroid Build Coastguard Worker stbi_uc *row0 = bytes + row*bytes_per_row;
1047*89c4ff92SAndroid Build Coastguard Worker stbi_uc *row1 = bytes + (h - row - 1)*bytes_per_row;
1048*89c4ff92SAndroid Build Coastguard Worker // swap row0 with row1
1049*89c4ff92SAndroid Build Coastguard Worker size_t bytes_left = bytes_per_row;
1050*89c4ff92SAndroid Build Coastguard Worker while (bytes_left) {
1051*89c4ff92SAndroid Build Coastguard Worker size_t bytes_copy = (bytes_left < sizeof(temp)) ? bytes_left : sizeof(temp);
1052*89c4ff92SAndroid Build Coastguard Worker memcpy(temp, row0, bytes_copy);
1053*89c4ff92SAndroid Build Coastguard Worker memcpy(row0, row1, bytes_copy);
1054*89c4ff92SAndroid Build Coastguard Worker memcpy(row1, temp, bytes_copy);
1055*89c4ff92SAndroid Build Coastguard Worker row0 += bytes_copy;
1056*89c4ff92SAndroid Build Coastguard Worker row1 += bytes_copy;
1057*89c4ff92SAndroid Build Coastguard Worker bytes_left -= bytes_copy;
1058*89c4ff92SAndroid Build Coastguard Worker }
1059*89c4ff92SAndroid Build Coastguard Worker }
1060*89c4ff92SAndroid Build Coastguard Worker }
1061*89c4ff92SAndroid Build Coastguard Worker
stbi__load_and_postprocess_8bit(stbi__context * s,int * x,int * y,int * comp,int req_comp)1062*89c4ff92SAndroid Build Coastguard Worker static unsigned char *stbi__load_and_postprocess_8bit(stbi__context *s, int *x, int *y, int *comp, int req_comp)
1063*89c4ff92SAndroid Build Coastguard Worker {
1064*89c4ff92SAndroid Build Coastguard Worker stbi__result_info ri;
1065*89c4ff92SAndroid Build Coastguard Worker void *result = stbi__load_main(s, x, y, comp, req_comp, &ri, 8);
1066*89c4ff92SAndroid Build Coastguard Worker
1067*89c4ff92SAndroid Build Coastguard Worker if (result == NULL)
1068*89c4ff92SAndroid Build Coastguard Worker return NULL;
1069*89c4ff92SAndroid Build Coastguard Worker
1070*89c4ff92SAndroid Build Coastguard Worker if (ri.bits_per_channel != 8) {
1071*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(ri.bits_per_channel == 16);
1072*89c4ff92SAndroid Build Coastguard Worker result = stbi__convert_16_to_8((stbi__uint16 *) result, *x, *y, req_comp == 0 ? *comp : req_comp);
1073*89c4ff92SAndroid Build Coastguard Worker ri.bits_per_channel = 8;
1074*89c4ff92SAndroid Build Coastguard Worker }
1075*89c4ff92SAndroid Build Coastguard Worker
1076*89c4ff92SAndroid Build Coastguard Worker // @TODO: move stbi__convert_format to here
1077*89c4ff92SAndroid Build Coastguard Worker
1078*89c4ff92SAndroid Build Coastguard Worker if (stbi__vertically_flip_on_load) {
1079*89c4ff92SAndroid Build Coastguard Worker int channels = req_comp ? req_comp : *comp;
1080*89c4ff92SAndroid Build Coastguard Worker stbi__vertical_flip(result, *x, *y, channels * sizeof(stbi_uc));
1081*89c4ff92SAndroid Build Coastguard Worker }
1082*89c4ff92SAndroid Build Coastguard Worker
1083*89c4ff92SAndroid Build Coastguard Worker return (unsigned char *) result;
1084*89c4ff92SAndroid Build Coastguard Worker }
1085*89c4ff92SAndroid Build Coastguard Worker
stbi__load_and_postprocess_16bit(stbi__context * s,int * x,int * y,int * comp,int req_comp)1086*89c4ff92SAndroid Build Coastguard Worker static stbi__uint16 *stbi__load_and_postprocess_16bit(stbi__context *s, int *x, int *y, int *comp, int req_comp)
1087*89c4ff92SAndroid Build Coastguard Worker {
1088*89c4ff92SAndroid Build Coastguard Worker stbi__result_info ri;
1089*89c4ff92SAndroid Build Coastguard Worker void *result = stbi__load_main(s, x, y, comp, req_comp, &ri, 16);
1090*89c4ff92SAndroid Build Coastguard Worker
1091*89c4ff92SAndroid Build Coastguard Worker if (result == NULL)
1092*89c4ff92SAndroid Build Coastguard Worker return NULL;
1093*89c4ff92SAndroid Build Coastguard Worker
1094*89c4ff92SAndroid Build Coastguard Worker if (ri.bits_per_channel != 16) {
1095*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(ri.bits_per_channel == 8);
1096*89c4ff92SAndroid Build Coastguard Worker result = stbi__convert_8_to_16((stbi_uc *) result, *x, *y, req_comp == 0 ? *comp : req_comp);
1097*89c4ff92SAndroid Build Coastguard Worker ri.bits_per_channel = 16;
1098*89c4ff92SAndroid Build Coastguard Worker }
1099*89c4ff92SAndroid Build Coastguard Worker
1100*89c4ff92SAndroid Build Coastguard Worker // @TODO: move stbi__convert_format16 to here
1101*89c4ff92SAndroid Build Coastguard Worker // @TODO: special case RGB-to-Y (and RGBA-to-YA) for 8-bit-to-16-bit case to keep more precision
1102*89c4ff92SAndroid Build Coastguard Worker
1103*89c4ff92SAndroid Build Coastguard Worker if (stbi__vertically_flip_on_load) {
1104*89c4ff92SAndroid Build Coastguard Worker int channels = req_comp ? req_comp : *comp;
1105*89c4ff92SAndroid Build Coastguard Worker stbi__vertical_flip(result, *x, *y, channels * sizeof(stbi__uint16));
1106*89c4ff92SAndroid Build Coastguard Worker }
1107*89c4ff92SAndroid Build Coastguard Worker
1108*89c4ff92SAndroid Build Coastguard Worker return (stbi__uint16 *) result;
1109*89c4ff92SAndroid Build Coastguard Worker }
1110*89c4ff92SAndroid Build Coastguard Worker
1111*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
stbi__float_postprocess(float * result,int * x,int * y,int * comp,int req_comp)1112*89c4ff92SAndroid Build Coastguard Worker static void stbi__float_postprocess(float *result, int *x, int *y, int *comp, int req_comp)
1113*89c4ff92SAndroid Build Coastguard Worker {
1114*89c4ff92SAndroid Build Coastguard Worker if (stbi__vertically_flip_on_load && result != NULL) {
1115*89c4ff92SAndroid Build Coastguard Worker int channels = req_comp ? req_comp : *comp;
1116*89c4ff92SAndroid Build Coastguard Worker stbi__vertical_flip(result, *x, *y, channels * sizeof(float));
1117*89c4ff92SAndroid Build Coastguard Worker }
1118*89c4ff92SAndroid Build Coastguard Worker }
1119*89c4ff92SAndroid Build Coastguard Worker #endif
1120*89c4ff92SAndroid Build Coastguard Worker
1121*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
1122*89c4ff92SAndroid Build Coastguard Worker
stbi__fopen(char const * filename,char const * mode)1123*89c4ff92SAndroid Build Coastguard Worker static FILE *stbi__fopen(char const *filename, char const *mode)
1124*89c4ff92SAndroid Build Coastguard Worker {
1125*89c4ff92SAndroid Build Coastguard Worker FILE *f;
1126*89c4ff92SAndroid Build Coastguard Worker #if defined(_MSC_VER) && _MSC_VER >= 1400
1127*89c4ff92SAndroid Build Coastguard Worker if (0 != fopen_s(&f, filename, mode))
1128*89c4ff92SAndroid Build Coastguard Worker f=0;
1129*89c4ff92SAndroid Build Coastguard Worker #else
1130*89c4ff92SAndroid Build Coastguard Worker f = fopen(filename, mode);
1131*89c4ff92SAndroid Build Coastguard Worker #endif
1132*89c4ff92SAndroid Build Coastguard Worker return f;
1133*89c4ff92SAndroid Build Coastguard Worker }
1134*89c4ff92SAndroid Build Coastguard Worker
1135*89c4ff92SAndroid Build Coastguard Worker
stbi_load(char const * filename,int * x,int * y,int * comp,int req_comp)1136*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_uc *stbi_load(char const *filename, int *x, int *y, int *comp, int req_comp)
1137*89c4ff92SAndroid Build Coastguard Worker {
1138*89c4ff92SAndroid Build Coastguard Worker FILE *f = stbi__fopen(filename, "rb");
1139*89c4ff92SAndroid Build Coastguard Worker unsigned char *result;
1140*89c4ff92SAndroid Build Coastguard Worker if (!f) return stbi__errpuc("can't fopen", "Unable to open file");
1141*89c4ff92SAndroid Build Coastguard Worker result = stbi_load_from_file(f,x,y,comp,req_comp);
1142*89c4ff92SAndroid Build Coastguard Worker fclose(f);
1143*89c4ff92SAndroid Build Coastguard Worker return result;
1144*89c4ff92SAndroid Build Coastguard Worker }
1145*89c4ff92SAndroid Build Coastguard Worker
stbi_load_from_file(FILE * f,int * x,int * y,int * comp,int req_comp)1146*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_uc *stbi_load_from_file(FILE *f, int *x, int *y, int *comp, int req_comp)
1147*89c4ff92SAndroid Build Coastguard Worker {
1148*89c4ff92SAndroid Build Coastguard Worker unsigned char *result;
1149*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1150*89c4ff92SAndroid Build Coastguard Worker stbi__start_file(&s,f);
1151*89c4ff92SAndroid Build Coastguard Worker result = stbi__load_and_postprocess_8bit(&s,x,y,comp,req_comp);
1152*89c4ff92SAndroid Build Coastguard Worker if (result) {
1153*89c4ff92SAndroid Build Coastguard Worker // need to 'unget' all the characters in the IO buffer
1154*89c4ff92SAndroid Build Coastguard Worker fseek(f, - (int) (s.img_buffer_end - s.img_buffer), SEEK_CUR);
1155*89c4ff92SAndroid Build Coastguard Worker }
1156*89c4ff92SAndroid Build Coastguard Worker return result;
1157*89c4ff92SAndroid Build Coastguard Worker }
1158*89c4ff92SAndroid Build Coastguard Worker
stbi_load_from_file_16(FILE * f,int * x,int * y,int * comp,int req_comp)1159*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi__uint16 *stbi_load_from_file_16(FILE *f, int *x, int *y, int *comp, int req_comp)
1160*89c4ff92SAndroid Build Coastguard Worker {
1161*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *result;
1162*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1163*89c4ff92SAndroid Build Coastguard Worker stbi__start_file(&s,f);
1164*89c4ff92SAndroid Build Coastguard Worker result = stbi__load_and_postprocess_16bit(&s,x,y,comp,req_comp);
1165*89c4ff92SAndroid Build Coastguard Worker if (result) {
1166*89c4ff92SAndroid Build Coastguard Worker // need to 'unget' all the characters in the IO buffer
1167*89c4ff92SAndroid Build Coastguard Worker fseek(f, - (int) (s.img_buffer_end - s.img_buffer), SEEK_CUR);
1168*89c4ff92SAndroid Build Coastguard Worker }
1169*89c4ff92SAndroid Build Coastguard Worker return result;
1170*89c4ff92SAndroid Build Coastguard Worker }
1171*89c4ff92SAndroid Build Coastguard Worker
stbi_load_16(char const * filename,int * x,int * y,int * comp,int req_comp)1172*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_us *stbi_load_16(char const *filename, int *x, int *y, int *comp, int req_comp)
1173*89c4ff92SAndroid Build Coastguard Worker {
1174*89c4ff92SAndroid Build Coastguard Worker FILE *f = stbi__fopen(filename, "rb");
1175*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *result;
1176*89c4ff92SAndroid Build Coastguard Worker if (!f) return (stbi_us *) stbi__errpuc("can't fopen", "Unable to open file");
1177*89c4ff92SAndroid Build Coastguard Worker result = stbi_load_from_file_16(f,x,y,comp,req_comp);
1178*89c4ff92SAndroid Build Coastguard Worker fclose(f);
1179*89c4ff92SAndroid Build Coastguard Worker return result;
1180*89c4ff92SAndroid Build Coastguard Worker }
1181*89c4ff92SAndroid Build Coastguard Worker
1182*89c4ff92SAndroid Build Coastguard Worker
1183*89c4ff92SAndroid Build Coastguard Worker #endif //!STBI_NO_STDIO
1184*89c4ff92SAndroid Build Coastguard Worker
stbi_load_16_from_memory(stbi_uc const * buffer,int len,int * x,int * y,int * channels_in_file,int desired_channels)1185*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_us *stbi_load_16_from_memory(stbi_uc const *buffer, int len, int *x, int *y, int *channels_in_file, int desired_channels)
1186*89c4ff92SAndroid Build Coastguard Worker {
1187*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1188*89c4ff92SAndroid Build Coastguard Worker stbi__start_mem(&s,buffer,len);
1189*89c4ff92SAndroid Build Coastguard Worker return stbi__load_and_postprocess_16bit(&s,x,y,channels_in_file,desired_channels);
1190*89c4ff92SAndroid Build Coastguard Worker }
1191*89c4ff92SAndroid Build Coastguard Worker
stbi_load_16_from_callbacks(stbi_io_callbacks const * clbk,void * user,int * x,int * y,int * channels_in_file,int desired_channels)1192*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_us *stbi_load_16_from_callbacks(stbi_io_callbacks const *clbk, void *user, int *x, int *y, int *channels_in_file, int desired_channels)
1193*89c4ff92SAndroid Build Coastguard Worker {
1194*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1195*89c4ff92SAndroid Build Coastguard Worker stbi__start_callbacks(&s, (stbi_io_callbacks *)clbk, user);
1196*89c4ff92SAndroid Build Coastguard Worker return stbi__load_and_postprocess_16bit(&s,x,y,channels_in_file,desired_channels);
1197*89c4ff92SAndroid Build Coastguard Worker }
1198*89c4ff92SAndroid Build Coastguard Worker
stbi_load_from_memory(stbi_uc const * buffer,int len,int * x,int * y,int * comp,int req_comp)1199*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_uc *stbi_load_from_memory(stbi_uc const *buffer, int len, int *x, int *y, int *comp, int req_comp)
1200*89c4ff92SAndroid Build Coastguard Worker {
1201*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1202*89c4ff92SAndroid Build Coastguard Worker stbi__start_mem(&s,buffer,len);
1203*89c4ff92SAndroid Build Coastguard Worker return stbi__load_and_postprocess_8bit(&s,x,y,comp,req_comp);
1204*89c4ff92SAndroid Build Coastguard Worker }
1205*89c4ff92SAndroid Build Coastguard Worker
stbi_load_from_callbacks(stbi_io_callbacks const * clbk,void * user,int * x,int * y,int * comp,int req_comp)1206*89c4ff92SAndroid Build Coastguard Worker STBIDEF stbi_uc *stbi_load_from_callbacks(stbi_io_callbacks const *clbk, void *user, int *x, int *y, int *comp, int req_comp)
1207*89c4ff92SAndroid Build Coastguard Worker {
1208*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1209*89c4ff92SAndroid Build Coastguard Worker stbi__start_callbacks(&s, (stbi_io_callbacks *) clbk, user);
1210*89c4ff92SAndroid Build Coastguard Worker return stbi__load_and_postprocess_8bit(&s,x,y,comp,req_comp);
1211*89c4ff92SAndroid Build Coastguard Worker }
1212*89c4ff92SAndroid Build Coastguard Worker
1213*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_LINEAR
stbi__loadf_main(stbi__context * s,int * x,int * y,int * comp,int req_comp)1214*89c4ff92SAndroid Build Coastguard Worker static float *stbi__loadf_main(stbi__context *s, int *x, int *y, int *comp, int req_comp)
1215*89c4ff92SAndroid Build Coastguard Worker {
1216*89c4ff92SAndroid Build Coastguard Worker unsigned char *data;
1217*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
1218*89c4ff92SAndroid Build Coastguard Worker if (stbi__hdr_test(s)) {
1219*89c4ff92SAndroid Build Coastguard Worker stbi__result_info ri;
1220*89c4ff92SAndroid Build Coastguard Worker float *hdr_data = stbi__hdr_load(s,x,y,comp,req_comp, &ri);
1221*89c4ff92SAndroid Build Coastguard Worker if (hdr_data)
1222*89c4ff92SAndroid Build Coastguard Worker stbi__float_postprocess(hdr_data,x,y,comp,req_comp);
1223*89c4ff92SAndroid Build Coastguard Worker return hdr_data;
1224*89c4ff92SAndroid Build Coastguard Worker }
1225*89c4ff92SAndroid Build Coastguard Worker #endif
1226*89c4ff92SAndroid Build Coastguard Worker data = stbi__load_and_postprocess_8bit(s, x, y, comp, req_comp);
1227*89c4ff92SAndroid Build Coastguard Worker if (data)
1228*89c4ff92SAndroid Build Coastguard Worker return stbi__ldr_to_hdr(data, *x, *y, req_comp ? req_comp : *comp);
1229*89c4ff92SAndroid Build Coastguard Worker return stbi__errpf("unknown image type", "Image not of any known type, or corrupt");
1230*89c4ff92SAndroid Build Coastguard Worker }
1231*89c4ff92SAndroid Build Coastguard Worker
stbi_loadf_from_memory(stbi_uc const * buffer,int len,int * x,int * y,int * comp,int req_comp)1232*89c4ff92SAndroid Build Coastguard Worker STBIDEF float *stbi_loadf_from_memory(stbi_uc const *buffer, int len, int *x, int *y, int *comp, int req_comp)
1233*89c4ff92SAndroid Build Coastguard Worker {
1234*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1235*89c4ff92SAndroid Build Coastguard Worker stbi__start_mem(&s,buffer,len);
1236*89c4ff92SAndroid Build Coastguard Worker return stbi__loadf_main(&s,x,y,comp,req_comp);
1237*89c4ff92SAndroid Build Coastguard Worker }
1238*89c4ff92SAndroid Build Coastguard Worker
stbi_loadf_from_callbacks(stbi_io_callbacks const * clbk,void * user,int * x,int * y,int * comp,int req_comp)1239*89c4ff92SAndroid Build Coastguard Worker STBIDEF float *stbi_loadf_from_callbacks(stbi_io_callbacks const *clbk, void *user, int *x, int *y, int *comp, int req_comp)
1240*89c4ff92SAndroid Build Coastguard Worker {
1241*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1242*89c4ff92SAndroid Build Coastguard Worker stbi__start_callbacks(&s, (stbi_io_callbacks *) clbk, user);
1243*89c4ff92SAndroid Build Coastguard Worker return stbi__loadf_main(&s,x,y,comp,req_comp);
1244*89c4ff92SAndroid Build Coastguard Worker }
1245*89c4ff92SAndroid Build Coastguard Worker
1246*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
stbi_loadf(char const * filename,int * x,int * y,int * comp,int req_comp)1247*89c4ff92SAndroid Build Coastguard Worker STBIDEF float *stbi_loadf(char const *filename, int *x, int *y, int *comp, int req_comp)
1248*89c4ff92SAndroid Build Coastguard Worker {
1249*89c4ff92SAndroid Build Coastguard Worker float *result;
1250*89c4ff92SAndroid Build Coastguard Worker FILE *f = stbi__fopen(filename, "rb");
1251*89c4ff92SAndroid Build Coastguard Worker if (!f) return stbi__errpf("can't fopen", "Unable to open file");
1252*89c4ff92SAndroid Build Coastguard Worker result = stbi_loadf_from_file(f,x,y,comp,req_comp);
1253*89c4ff92SAndroid Build Coastguard Worker fclose(f);
1254*89c4ff92SAndroid Build Coastguard Worker return result;
1255*89c4ff92SAndroid Build Coastguard Worker }
1256*89c4ff92SAndroid Build Coastguard Worker
stbi_loadf_from_file(FILE * f,int * x,int * y,int * comp,int req_comp)1257*89c4ff92SAndroid Build Coastguard Worker STBIDEF float *stbi_loadf_from_file(FILE *f, int *x, int *y, int *comp, int req_comp)
1258*89c4ff92SAndroid Build Coastguard Worker {
1259*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1260*89c4ff92SAndroid Build Coastguard Worker stbi__start_file(&s,f);
1261*89c4ff92SAndroid Build Coastguard Worker return stbi__loadf_main(&s,x,y,comp,req_comp);
1262*89c4ff92SAndroid Build Coastguard Worker }
1263*89c4ff92SAndroid Build Coastguard Worker #endif // !STBI_NO_STDIO
1264*89c4ff92SAndroid Build Coastguard Worker
1265*89c4ff92SAndroid Build Coastguard Worker #endif // !STBI_NO_LINEAR
1266*89c4ff92SAndroid Build Coastguard Worker
1267*89c4ff92SAndroid Build Coastguard Worker // these is-hdr-or-not is defined independent of whether STBI_NO_LINEAR is
1268*89c4ff92SAndroid Build Coastguard Worker // defined, for API simplicity; if STBI_NO_LINEAR is defined, it always
1269*89c4ff92SAndroid Build Coastguard Worker // reports false!
1270*89c4ff92SAndroid Build Coastguard Worker
stbi_is_hdr_from_memory(stbi_uc const * buffer,int len)1271*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_is_hdr_from_memory(stbi_uc const *buffer, int len)
1272*89c4ff92SAndroid Build Coastguard Worker {
1273*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
1274*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1275*89c4ff92SAndroid Build Coastguard Worker stbi__start_mem(&s,buffer,len);
1276*89c4ff92SAndroid Build Coastguard Worker return stbi__hdr_test(&s);
1277*89c4ff92SAndroid Build Coastguard Worker #else
1278*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(buffer);
1279*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(len);
1280*89c4ff92SAndroid Build Coastguard Worker return 0;
1281*89c4ff92SAndroid Build Coastguard Worker #endif
1282*89c4ff92SAndroid Build Coastguard Worker }
1283*89c4ff92SAndroid Build Coastguard Worker
1284*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
stbi_is_hdr(char const * filename)1285*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_is_hdr (char const *filename)
1286*89c4ff92SAndroid Build Coastguard Worker {
1287*89c4ff92SAndroid Build Coastguard Worker FILE *f = stbi__fopen(filename, "rb");
1288*89c4ff92SAndroid Build Coastguard Worker int result=0;
1289*89c4ff92SAndroid Build Coastguard Worker if (f) {
1290*89c4ff92SAndroid Build Coastguard Worker result = stbi_is_hdr_from_file(f);
1291*89c4ff92SAndroid Build Coastguard Worker fclose(f);
1292*89c4ff92SAndroid Build Coastguard Worker }
1293*89c4ff92SAndroid Build Coastguard Worker return result;
1294*89c4ff92SAndroid Build Coastguard Worker }
1295*89c4ff92SAndroid Build Coastguard Worker
stbi_is_hdr_from_file(FILE * f)1296*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_is_hdr_from_file(FILE *f)
1297*89c4ff92SAndroid Build Coastguard Worker {
1298*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
1299*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1300*89c4ff92SAndroid Build Coastguard Worker stbi__start_file(&s,f);
1301*89c4ff92SAndroid Build Coastguard Worker return stbi__hdr_test(&s);
1302*89c4ff92SAndroid Build Coastguard Worker #else
1303*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(f);
1304*89c4ff92SAndroid Build Coastguard Worker return 0;
1305*89c4ff92SAndroid Build Coastguard Worker #endif
1306*89c4ff92SAndroid Build Coastguard Worker }
1307*89c4ff92SAndroid Build Coastguard Worker #endif // !STBI_NO_STDIO
1308*89c4ff92SAndroid Build Coastguard Worker
stbi_is_hdr_from_callbacks(stbi_io_callbacks const * clbk,void * user)1309*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_is_hdr_from_callbacks(stbi_io_callbacks const *clbk, void *user)
1310*89c4ff92SAndroid Build Coastguard Worker {
1311*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
1312*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
1313*89c4ff92SAndroid Build Coastguard Worker stbi__start_callbacks(&s, (stbi_io_callbacks *) clbk, user);
1314*89c4ff92SAndroid Build Coastguard Worker return stbi__hdr_test(&s);
1315*89c4ff92SAndroid Build Coastguard Worker #else
1316*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(clbk);
1317*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(user);
1318*89c4ff92SAndroid Build Coastguard Worker return 0;
1319*89c4ff92SAndroid Build Coastguard Worker #endif
1320*89c4ff92SAndroid Build Coastguard Worker }
1321*89c4ff92SAndroid Build Coastguard Worker
1322*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_LINEAR
1323*89c4ff92SAndroid Build Coastguard Worker static float stbi__l2h_gamma=2.2f, stbi__l2h_scale=1.0f;
1324*89c4ff92SAndroid Build Coastguard Worker
stbi_ldr_to_hdr_gamma(float gamma)1325*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_ldr_to_hdr_gamma(float gamma) { stbi__l2h_gamma = gamma; }
stbi_ldr_to_hdr_scale(float scale)1326*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_ldr_to_hdr_scale(float scale) { stbi__l2h_scale = scale; }
1327*89c4ff92SAndroid Build Coastguard Worker #endif
1328*89c4ff92SAndroid Build Coastguard Worker
1329*89c4ff92SAndroid Build Coastguard Worker static float stbi__h2l_gamma_i=1.0f/2.2f, stbi__h2l_scale_i=1.0f;
1330*89c4ff92SAndroid Build Coastguard Worker
stbi_hdr_to_ldr_gamma(float gamma)1331*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_hdr_to_ldr_gamma(float gamma) { stbi__h2l_gamma_i = 1/gamma; }
stbi_hdr_to_ldr_scale(float scale)1332*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_hdr_to_ldr_scale(float scale) { stbi__h2l_scale_i = 1/scale; }
1333*89c4ff92SAndroid Build Coastguard Worker
1334*89c4ff92SAndroid Build Coastguard Worker
1335*89c4ff92SAndroid Build Coastguard Worker //////////////////////////////////////////////////////////////////////////////
1336*89c4ff92SAndroid Build Coastguard Worker //
1337*89c4ff92SAndroid Build Coastguard Worker // Common code used by all image loaders
1338*89c4ff92SAndroid Build Coastguard Worker //
1339*89c4ff92SAndroid Build Coastguard Worker
1340*89c4ff92SAndroid Build Coastguard Worker enum
1341*89c4ff92SAndroid Build Coastguard Worker {
1342*89c4ff92SAndroid Build Coastguard Worker STBI__SCAN_load=0,
1343*89c4ff92SAndroid Build Coastguard Worker STBI__SCAN_type,
1344*89c4ff92SAndroid Build Coastguard Worker STBI__SCAN_header
1345*89c4ff92SAndroid Build Coastguard Worker };
1346*89c4ff92SAndroid Build Coastguard Worker
stbi__refill_buffer(stbi__context * s)1347*89c4ff92SAndroid Build Coastguard Worker static void stbi__refill_buffer(stbi__context *s)
1348*89c4ff92SAndroid Build Coastguard Worker {
1349*89c4ff92SAndroid Build Coastguard Worker int n = (s->io.read)(s->io_user_data,(char*)s->buffer_start,s->buflen);
1350*89c4ff92SAndroid Build Coastguard Worker if (n == 0) {
1351*89c4ff92SAndroid Build Coastguard Worker // at end of file, treat same as if from memory, but need to handle case
1352*89c4ff92SAndroid Build Coastguard Worker // where s->img_buffer isn't pointing to safe memory, e.g. 0-byte file
1353*89c4ff92SAndroid Build Coastguard Worker s->read_from_callbacks = 0;
1354*89c4ff92SAndroid Build Coastguard Worker s->img_buffer = s->buffer_start;
1355*89c4ff92SAndroid Build Coastguard Worker s->img_buffer_end = s->buffer_start+1;
1356*89c4ff92SAndroid Build Coastguard Worker *s->img_buffer = 0;
1357*89c4ff92SAndroid Build Coastguard Worker } else {
1358*89c4ff92SAndroid Build Coastguard Worker s->img_buffer = s->buffer_start;
1359*89c4ff92SAndroid Build Coastguard Worker s->img_buffer_end = s->buffer_start + n;
1360*89c4ff92SAndroid Build Coastguard Worker }
1361*89c4ff92SAndroid Build Coastguard Worker }
1362*89c4ff92SAndroid Build Coastguard Worker
stbi__get8(stbi__context * s)1363*89c4ff92SAndroid Build Coastguard Worker stbi_inline static stbi_uc stbi__get8(stbi__context *s)
1364*89c4ff92SAndroid Build Coastguard Worker {
1365*89c4ff92SAndroid Build Coastguard Worker if (s->img_buffer < s->img_buffer_end)
1366*89c4ff92SAndroid Build Coastguard Worker return *s->img_buffer++;
1367*89c4ff92SAndroid Build Coastguard Worker if (s->read_from_callbacks) {
1368*89c4ff92SAndroid Build Coastguard Worker stbi__refill_buffer(s);
1369*89c4ff92SAndroid Build Coastguard Worker return *s->img_buffer++;
1370*89c4ff92SAndroid Build Coastguard Worker }
1371*89c4ff92SAndroid Build Coastguard Worker return 0;
1372*89c4ff92SAndroid Build Coastguard Worker }
1373*89c4ff92SAndroid Build Coastguard Worker
stbi__at_eof(stbi__context * s)1374*89c4ff92SAndroid Build Coastguard Worker stbi_inline static int stbi__at_eof(stbi__context *s)
1375*89c4ff92SAndroid Build Coastguard Worker {
1376*89c4ff92SAndroid Build Coastguard Worker if (s->io.read) {
1377*89c4ff92SAndroid Build Coastguard Worker if (!(s->io.eof)(s->io_user_data)) return 0;
1378*89c4ff92SAndroid Build Coastguard Worker // if feof() is true, check if buffer = end
1379*89c4ff92SAndroid Build Coastguard Worker // special case: we've only got the special 0 character at the end
1380*89c4ff92SAndroid Build Coastguard Worker if (s->read_from_callbacks == 0) return 1;
1381*89c4ff92SAndroid Build Coastguard Worker }
1382*89c4ff92SAndroid Build Coastguard Worker
1383*89c4ff92SAndroid Build Coastguard Worker return s->img_buffer >= s->img_buffer_end;
1384*89c4ff92SAndroid Build Coastguard Worker }
1385*89c4ff92SAndroid Build Coastguard Worker
stbi__skip(stbi__context * s,int n)1386*89c4ff92SAndroid Build Coastguard Worker static void stbi__skip(stbi__context *s, int n)
1387*89c4ff92SAndroid Build Coastguard Worker {
1388*89c4ff92SAndroid Build Coastguard Worker if (n < 0) {
1389*89c4ff92SAndroid Build Coastguard Worker s->img_buffer = s->img_buffer_end;
1390*89c4ff92SAndroid Build Coastguard Worker return;
1391*89c4ff92SAndroid Build Coastguard Worker }
1392*89c4ff92SAndroid Build Coastguard Worker if (s->io.read) {
1393*89c4ff92SAndroid Build Coastguard Worker int blen = (int) (s->img_buffer_end - s->img_buffer);
1394*89c4ff92SAndroid Build Coastguard Worker if (blen < n) {
1395*89c4ff92SAndroid Build Coastguard Worker s->img_buffer = s->img_buffer_end;
1396*89c4ff92SAndroid Build Coastguard Worker (s->io.skip)(s->io_user_data, n - blen);
1397*89c4ff92SAndroid Build Coastguard Worker return;
1398*89c4ff92SAndroid Build Coastguard Worker }
1399*89c4ff92SAndroid Build Coastguard Worker }
1400*89c4ff92SAndroid Build Coastguard Worker s->img_buffer += n;
1401*89c4ff92SAndroid Build Coastguard Worker }
1402*89c4ff92SAndroid Build Coastguard Worker
stbi__getn(stbi__context * s,stbi_uc * buffer,int n)1403*89c4ff92SAndroid Build Coastguard Worker static int stbi__getn(stbi__context *s, stbi_uc *buffer, int n)
1404*89c4ff92SAndroid Build Coastguard Worker {
1405*89c4ff92SAndroid Build Coastguard Worker if (s->io.read) {
1406*89c4ff92SAndroid Build Coastguard Worker int blen = (int) (s->img_buffer_end - s->img_buffer);
1407*89c4ff92SAndroid Build Coastguard Worker if (blen < n) {
1408*89c4ff92SAndroid Build Coastguard Worker int res, count;
1409*89c4ff92SAndroid Build Coastguard Worker
1410*89c4ff92SAndroid Build Coastguard Worker memcpy(buffer, s->img_buffer, blen);
1411*89c4ff92SAndroid Build Coastguard Worker
1412*89c4ff92SAndroid Build Coastguard Worker count = (s->io.read)(s->io_user_data, (char*) buffer + blen, n - blen);
1413*89c4ff92SAndroid Build Coastguard Worker res = (count == (n-blen));
1414*89c4ff92SAndroid Build Coastguard Worker s->img_buffer = s->img_buffer_end;
1415*89c4ff92SAndroid Build Coastguard Worker return res;
1416*89c4ff92SAndroid Build Coastguard Worker }
1417*89c4ff92SAndroid Build Coastguard Worker }
1418*89c4ff92SAndroid Build Coastguard Worker
1419*89c4ff92SAndroid Build Coastguard Worker if (s->img_buffer+n <= s->img_buffer_end) {
1420*89c4ff92SAndroid Build Coastguard Worker memcpy(buffer, s->img_buffer, n);
1421*89c4ff92SAndroid Build Coastguard Worker s->img_buffer += n;
1422*89c4ff92SAndroid Build Coastguard Worker return 1;
1423*89c4ff92SAndroid Build Coastguard Worker } else
1424*89c4ff92SAndroid Build Coastguard Worker return 0;
1425*89c4ff92SAndroid Build Coastguard Worker }
1426*89c4ff92SAndroid Build Coastguard Worker
stbi__get16be(stbi__context * s)1427*89c4ff92SAndroid Build Coastguard Worker static int stbi__get16be(stbi__context *s)
1428*89c4ff92SAndroid Build Coastguard Worker {
1429*89c4ff92SAndroid Build Coastguard Worker int z = stbi__get8(s);
1430*89c4ff92SAndroid Build Coastguard Worker return (z << 8) + stbi__get8(s);
1431*89c4ff92SAndroid Build Coastguard Worker }
1432*89c4ff92SAndroid Build Coastguard Worker
stbi__get32be(stbi__context * s)1433*89c4ff92SAndroid Build Coastguard Worker static stbi__uint32 stbi__get32be(stbi__context *s)
1434*89c4ff92SAndroid Build Coastguard Worker {
1435*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 z = stbi__get16be(s);
1436*89c4ff92SAndroid Build Coastguard Worker return (z << 16) + stbi__get16be(s);
1437*89c4ff92SAndroid Build Coastguard Worker }
1438*89c4ff92SAndroid Build Coastguard Worker
1439*89c4ff92SAndroid Build Coastguard Worker #if defined(STBI_NO_BMP) && defined(STBI_NO_TGA) && defined(STBI_NO_GIF)
1440*89c4ff92SAndroid Build Coastguard Worker // nothing
1441*89c4ff92SAndroid Build Coastguard Worker #else
stbi__get16le(stbi__context * s)1442*89c4ff92SAndroid Build Coastguard Worker static int stbi__get16le(stbi__context *s)
1443*89c4ff92SAndroid Build Coastguard Worker {
1444*89c4ff92SAndroid Build Coastguard Worker int z = stbi__get8(s);
1445*89c4ff92SAndroid Build Coastguard Worker return z + (stbi__get8(s) << 8);
1446*89c4ff92SAndroid Build Coastguard Worker }
1447*89c4ff92SAndroid Build Coastguard Worker #endif
1448*89c4ff92SAndroid Build Coastguard Worker
1449*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_BMP
stbi__get32le(stbi__context * s)1450*89c4ff92SAndroid Build Coastguard Worker static stbi__uint32 stbi__get32le(stbi__context *s)
1451*89c4ff92SAndroid Build Coastguard Worker {
1452*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 z = stbi__get16le(s);
1453*89c4ff92SAndroid Build Coastguard Worker return z + (stbi__get16le(s) << 16);
1454*89c4ff92SAndroid Build Coastguard Worker }
1455*89c4ff92SAndroid Build Coastguard Worker #endif
1456*89c4ff92SAndroid Build Coastguard Worker
1457*89c4ff92SAndroid Build Coastguard Worker #define STBI__BYTECAST(x) ((stbi_uc) ((x) & 255)) // truncate int to byte without warnings
1458*89c4ff92SAndroid Build Coastguard Worker
1459*89c4ff92SAndroid Build Coastguard Worker
1460*89c4ff92SAndroid Build Coastguard Worker //////////////////////////////////////////////////////////////////////////////
1461*89c4ff92SAndroid Build Coastguard Worker //
1462*89c4ff92SAndroid Build Coastguard Worker // generic converter from built-in img_n to req_comp
1463*89c4ff92SAndroid Build Coastguard Worker // individual types do this automatically as much as possible (e.g. jpeg
1464*89c4ff92SAndroid Build Coastguard Worker // does all cases internally since it needs to colorspace convert anyway,
1465*89c4ff92SAndroid Build Coastguard Worker // and it never has alpha, so very few cases ). png can automatically
1466*89c4ff92SAndroid Build Coastguard Worker // interleave an alpha=255 channel, but falls back to this for other cases
1467*89c4ff92SAndroid Build Coastguard Worker //
1468*89c4ff92SAndroid Build Coastguard Worker // assume data buffer is malloced, so malloc a new one and free that one
1469*89c4ff92SAndroid Build Coastguard Worker // only failure mode is malloc failing
1470*89c4ff92SAndroid Build Coastguard Worker
stbi__compute_y(int r,int g,int b)1471*89c4ff92SAndroid Build Coastguard Worker static stbi_uc stbi__compute_y(int r, int g, int b)
1472*89c4ff92SAndroid Build Coastguard Worker {
1473*89c4ff92SAndroid Build Coastguard Worker return (stbi_uc) (((r*77) + (g*150) + (29*b)) >> 8);
1474*89c4ff92SAndroid Build Coastguard Worker }
1475*89c4ff92SAndroid Build Coastguard Worker
stbi__convert_format(unsigned char * data,int img_n,int req_comp,unsigned int x,unsigned int y)1476*89c4ff92SAndroid Build Coastguard Worker static unsigned char *stbi__convert_format(unsigned char *data, int img_n, int req_comp, unsigned int x, unsigned int y)
1477*89c4ff92SAndroid Build Coastguard Worker {
1478*89c4ff92SAndroid Build Coastguard Worker int i,j;
1479*89c4ff92SAndroid Build Coastguard Worker unsigned char *good;
1480*89c4ff92SAndroid Build Coastguard Worker
1481*89c4ff92SAndroid Build Coastguard Worker if (req_comp == img_n) return data;
1482*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(req_comp >= 1 && req_comp <= 4);
1483*89c4ff92SAndroid Build Coastguard Worker
1484*89c4ff92SAndroid Build Coastguard Worker good = (unsigned char *) stbi__malloc_mad3(req_comp, x, y, 0);
1485*89c4ff92SAndroid Build Coastguard Worker if (good == NULL) {
1486*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(data);
1487*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("outofmem", "Out of memory");
1488*89c4ff92SAndroid Build Coastguard Worker }
1489*89c4ff92SAndroid Build Coastguard Worker
1490*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < (int) y; ++j) {
1491*89c4ff92SAndroid Build Coastguard Worker unsigned char *src = data + j * x * img_n ;
1492*89c4ff92SAndroid Build Coastguard Worker unsigned char *dest = good + j * x * req_comp;
1493*89c4ff92SAndroid Build Coastguard Worker
1494*89c4ff92SAndroid Build Coastguard Worker #define STBI__COMBO(a,b) ((a)*8+(b))
1495*89c4ff92SAndroid Build Coastguard Worker #define STBI__CASE(a,b) case STBI__COMBO(a,b): for(i=x-1; i >= 0; --i, src += a, dest += b)
1496*89c4ff92SAndroid Build Coastguard Worker // convert source image with img_n components to one with req_comp components;
1497*89c4ff92SAndroid Build Coastguard Worker // avoid switch per pixel, so use switch per scanline and massive macros
1498*89c4ff92SAndroid Build Coastguard Worker switch (STBI__COMBO(img_n, req_comp)) {
1499*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(1,2) { dest[0]=src[0], dest[1]=255; } break;
1500*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(1,3) { dest[0]=dest[1]=dest[2]=src[0]; } break;
1501*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(1,4) { dest[0]=dest[1]=dest[2]=src[0], dest[3]=255; } break;
1502*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(2,1) { dest[0]=src[0]; } break;
1503*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(2,3) { dest[0]=dest[1]=dest[2]=src[0]; } break;
1504*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(2,4) { dest[0]=dest[1]=dest[2]=src[0], dest[3]=src[1]; } break;
1505*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(3,4) { dest[0]=src[0],dest[1]=src[1],dest[2]=src[2],dest[3]=255; } break;
1506*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(3,1) { dest[0]=stbi__compute_y(src[0],src[1],src[2]); } break;
1507*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(3,2) { dest[0]=stbi__compute_y(src[0],src[1],src[2]), dest[1] = 255; } break;
1508*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(4,1) { dest[0]=stbi__compute_y(src[0],src[1],src[2]); } break;
1509*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(4,2) { dest[0]=stbi__compute_y(src[0],src[1],src[2]), dest[1] = src[3]; } break;
1510*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(4,3) { dest[0]=src[0],dest[1]=src[1],dest[2]=src[2]; } break;
1511*89c4ff92SAndroid Build Coastguard Worker default: STBI_ASSERT(0);
1512*89c4ff92SAndroid Build Coastguard Worker }
1513*89c4ff92SAndroid Build Coastguard Worker #undef STBI__CASE
1514*89c4ff92SAndroid Build Coastguard Worker }
1515*89c4ff92SAndroid Build Coastguard Worker
1516*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(data);
1517*89c4ff92SAndroid Build Coastguard Worker return good;
1518*89c4ff92SAndroid Build Coastguard Worker }
1519*89c4ff92SAndroid Build Coastguard Worker
stbi__compute_y_16(int r,int g,int b)1520*89c4ff92SAndroid Build Coastguard Worker static stbi__uint16 stbi__compute_y_16(int r, int g, int b)
1521*89c4ff92SAndroid Build Coastguard Worker {
1522*89c4ff92SAndroid Build Coastguard Worker return (stbi__uint16) (((r*77) + (g*150) + (29*b)) >> 8);
1523*89c4ff92SAndroid Build Coastguard Worker }
1524*89c4ff92SAndroid Build Coastguard Worker
stbi__convert_format16(stbi__uint16 * data,int img_n,int req_comp,unsigned int x,unsigned int y)1525*89c4ff92SAndroid Build Coastguard Worker static stbi__uint16 *stbi__convert_format16(stbi__uint16 *data, int img_n, int req_comp, unsigned int x, unsigned int y)
1526*89c4ff92SAndroid Build Coastguard Worker {
1527*89c4ff92SAndroid Build Coastguard Worker int i,j;
1528*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *good;
1529*89c4ff92SAndroid Build Coastguard Worker
1530*89c4ff92SAndroid Build Coastguard Worker if (req_comp == img_n) return data;
1531*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(req_comp >= 1 && req_comp <= 4);
1532*89c4ff92SAndroid Build Coastguard Worker
1533*89c4ff92SAndroid Build Coastguard Worker good = (stbi__uint16 *) stbi__malloc(req_comp * x * y * 2);
1534*89c4ff92SAndroid Build Coastguard Worker if (good == NULL) {
1535*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(data);
1536*89c4ff92SAndroid Build Coastguard Worker return (stbi__uint16 *) stbi__errpuc("outofmem", "Out of memory");
1537*89c4ff92SAndroid Build Coastguard Worker }
1538*89c4ff92SAndroid Build Coastguard Worker
1539*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < (int) y; ++j) {
1540*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *src = data + j * x * img_n ;
1541*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *dest = good + j * x * req_comp;
1542*89c4ff92SAndroid Build Coastguard Worker
1543*89c4ff92SAndroid Build Coastguard Worker #define STBI__COMBO(a,b) ((a)*8+(b))
1544*89c4ff92SAndroid Build Coastguard Worker #define STBI__CASE(a,b) case STBI__COMBO(a,b): for(i=x-1; i >= 0; --i, src += a, dest += b)
1545*89c4ff92SAndroid Build Coastguard Worker // convert source image with img_n components to one with req_comp components;
1546*89c4ff92SAndroid Build Coastguard Worker // avoid switch per pixel, so use switch per scanline and massive macros
1547*89c4ff92SAndroid Build Coastguard Worker switch (STBI__COMBO(img_n, req_comp)) {
1548*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(1,2) { dest[0]=src[0], dest[1]=0xffff; } break;
1549*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(1,3) { dest[0]=dest[1]=dest[2]=src[0]; } break;
1550*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(1,4) { dest[0]=dest[1]=dest[2]=src[0], dest[3]=0xffff; } break;
1551*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(2,1) { dest[0]=src[0]; } break;
1552*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(2,3) { dest[0]=dest[1]=dest[2]=src[0]; } break;
1553*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(2,4) { dest[0]=dest[1]=dest[2]=src[0], dest[3]=src[1]; } break;
1554*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(3,4) { dest[0]=src[0],dest[1]=src[1],dest[2]=src[2],dest[3]=0xffff; } break;
1555*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(3,1) { dest[0]=stbi__compute_y_16(src[0],src[1],src[2]); } break;
1556*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(3,2) { dest[0]=stbi__compute_y_16(src[0],src[1],src[2]), dest[1] = 0xffff; } break;
1557*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(4,1) { dest[0]=stbi__compute_y_16(src[0],src[1],src[2]); } break;
1558*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(4,2) { dest[0]=stbi__compute_y_16(src[0],src[1],src[2]), dest[1] = src[3]; } break;
1559*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(4,3) { dest[0]=src[0],dest[1]=src[1],dest[2]=src[2]; } break;
1560*89c4ff92SAndroid Build Coastguard Worker default: STBI_ASSERT(0);
1561*89c4ff92SAndroid Build Coastguard Worker }
1562*89c4ff92SAndroid Build Coastguard Worker #undef STBI__CASE
1563*89c4ff92SAndroid Build Coastguard Worker }
1564*89c4ff92SAndroid Build Coastguard Worker
1565*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(data);
1566*89c4ff92SAndroid Build Coastguard Worker return good;
1567*89c4ff92SAndroid Build Coastguard Worker }
1568*89c4ff92SAndroid Build Coastguard Worker
1569*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_LINEAR
stbi__ldr_to_hdr(stbi_uc * data,int x,int y,int comp)1570*89c4ff92SAndroid Build Coastguard Worker static float *stbi__ldr_to_hdr(stbi_uc *data, int x, int y, int comp)
1571*89c4ff92SAndroid Build Coastguard Worker {
1572*89c4ff92SAndroid Build Coastguard Worker int i,k,n;
1573*89c4ff92SAndroid Build Coastguard Worker float *output;
1574*89c4ff92SAndroid Build Coastguard Worker if (!data) return NULL;
1575*89c4ff92SAndroid Build Coastguard Worker output = (float *) stbi__malloc_mad4(x, y, comp, sizeof(float), 0);
1576*89c4ff92SAndroid Build Coastguard Worker if (output == NULL) { STBI_FREE(data); return stbi__errpf("outofmem", "Out of memory"); }
1577*89c4ff92SAndroid Build Coastguard Worker // compute number of non-alpha components
1578*89c4ff92SAndroid Build Coastguard Worker if (comp & 1) n = comp; else n = comp-1;
1579*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < x*y; ++i) {
1580*89c4ff92SAndroid Build Coastguard Worker for (k=0; k < n; ++k) {
1581*89c4ff92SAndroid Build Coastguard Worker output[i*comp + k] = (float) (pow(data[i*comp+k]/255.0f, stbi__l2h_gamma) * stbi__l2h_scale);
1582*89c4ff92SAndroid Build Coastguard Worker }
1583*89c4ff92SAndroid Build Coastguard Worker if (k < comp) output[i*comp + k] = data[i*comp+k]/255.0f;
1584*89c4ff92SAndroid Build Coastguard Worker }
1585*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(data);
1586*89c4ff92SAndroid Build Coastguard Worker return output;
1587*89c4ff92SAndroid Build Coastguard Worker }
1588*89c4ff92SAndroid Build Coastguard Worker #endif
1589*89c4ff92SAndroid Build Coastguard Worker
1590*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
1591*89c4ff92SAndroid Build Coastguard Worker #define stbi__float2int(x) ((int) (x))
stbi__hdr_to_ldr(float * data,int x,int y,int comp)1592*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__hdr_to_ldr(float *data, int x, int y, int comp)
1593*89c4ff92SAndroid Build Coastguard Worker {
1594*89c4ff92SAndroid Build Coastguard Worker int i,k,n;
1595*89c4ff92SAndroid Build Coastguard Worker stbi_uc *output;
1596*89c4ff92SAndroid Build Coastguard Worker if (!data) return NULL;
1597*89c4ff92SAndroid Build Coastguard Worker output = (stbi_uc *) stbi__malloc_mad3(x, y, comp, 0);
1598*89c4ff92SAndroid Build Coastguard Worker if (output == NULL) { STBI_FREE(data); return stbi__errpuc("outofmem", "Out of memory"); }
1599*89c4ff92SAndroid Build Coastguard Worker // compute number of non-alpha components
1600*89c4ff92SAndroid Build Coastguard Worker if (comp & 1) n = comp; else n = comp-1;
1601*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < x*y; ++i) {
1602*89c4ff92SAndroid Build Coastguard Worker for (k=0; k < n; ++k) {
1603*89c4ff92SAndroid Build Coastguard Worker float z = (float) pow(data[i*comp+k]*stbi__h2l_scale_i, stbi__h2l_gamma_i) * 255 + 0.5f;
1604*89c4ff92SAndroid Build Coastguard Worker if (z < 0) z = 0;
1605*89c4ff92SAndroid Build Coastguard Worker if (z > 255) z = 255;
1606*89c4ff92SAndroid Build Coastguard Worker output[i*comp + k] = (stbi_uc) stbi__float2int(z);
1607*89c4ff92SAndroid Build Coastguard Worker }
1608*89c4ff92SAndroid Build Coastguard Worker if (k < comp) {
1609*89c4ff92SAndroid Build Coastguard Worker float z = data[i*comp+k] * 255 + 0.5f;
1610*89c4ff92SAndroid Build Coastguard Worker if (z < 0) z = 0;
1611*89c4ff92SAndroid Build Coastguard Worker if (z > 255) z = 255;
1612*89c4ff92SAndroid Build Coastguard Worker output[i*comp + k] = (stbi_uc) stbi__float2int(z);
1613*89c4ff92SAndroid Build Coastguard Worker }
1614*89c4ff92SAndroid Build Coastguard Worker }
1615*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(data);
1616*89c4ff92SAndroid Build Coastguard Worker return output;
1617*89c4ff92SAndroid Build Coastguard Worker }
1618*89c4ff92SAndroid Build Coastguard Worker #endif
1619*89c4ff92SAndroid Build Coastguard Worker
1620*89c4ff92SAndroid Build Coastguard Worker //////////////////////////////////////////////////////////////////////////////
1621*89c4ff92SAndroid Build Coastguard Worker //
1622*89c4ff92SAndroid Build Coastguard Worker // "baseline" JPEG/JFIF decoder
1623*89c4ff92SAndroid Build Coastguard Worker //
1624*89c4ff92SAndroid Build Coastguard Worker // simple implementation
1625*89c4ff92SAndroid Build Coastguard Worker // - doesn't support delayed output of y-dimension
1626*89c4ff92SAndroid Build Coastguard Worker // - simple interface (only one output format: 8-bit interleaved RGB)
1627*89c4ff92SAndroid Build Coastguard Worker // - doesn't try to recover corrupt jpegs
1628*89c4ff92SAndroid Build Coastguard Worker // - doesn't allow partial loading, loading multiple at once
1629*89c4ff92SAndroid Build Coastguard Worker // - still fast on x86 (copying globals into locals doesn't help x86)
1630*89c4ff92SAndroid Build Coastguard Worker // - allocates lots of intermediate memory (full size of all components)
1631*89c4ff92SAndroid Build Coastguard Worker // - non-interleaved case requires this anyway
1632*89c4ff92SAndroid Build Coastguard Worker // - allows good upsampling (see next)
1633*89c4ff92SAndroid Build Coastguard Worker // high-quality
1634*89c4ff92SAndroid Build Coastguard Worker // - upsampled channels are bilinearly interpolated, even across blocks
1635*89c4ff92SAndroid Build Coastguard Worker // - quality integer IDCT derived from IJG's 'slow'
1636*89c4ff92SAndroid Build Coastguard Worker // performance
1637*89c4ff92SAndroid Build Coastguard Worker // - fast huffman; reasonable integer IDCT
1638*89c4ff92SAndroid Build Coastguard Worker // - some SIMD kernels for common paths on targets with SSE2/NEON
1639*89c4ff92SAndroid Build Coastguard Worker // - uses a lot of intermediate memory, could cache poorly
1640*89c4ff92SAndroid Build Coastguard Worker
1641*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_JPEG
1642*89c4ff92SAndroid Build Coastguard Worker
1643*89c4ff92SAndroid Build Coastguard Worker // huffman decoding acceleration
1644*89c4ff92SAndroid Build Coastguard Worker #define FAST_BITS 9 // larger handles more cases; smaller stomps less cache
1645*89c4ff92SAndroid Build Coastguard Worker
1646*89c4ff92SAndroid Build Coastguard Worker typedef struct
1647*89c4ff92SAndroid Build Coastguard Worker {
1648*89c4ff92SAndroid Build Coastguard Worker stbi_uc fast[1 << FAST_BITS];
1649*89c4ff92SAndroid Build Coastguard Worker // weirdly, repacking this into AoS is a 10% speed loss, instead of a win
1650*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 code[256];
1651*89c4ff92SAndroid Build Coastguard Worker stbi_uc values[256];
1652*89c4ff92SAndroid Build Coastguard Worker stbi_uc size[257];
1653*89c4ff92SAndroid Build Coastguard Worker unsigned int maxcode[18];
1654*89c4ff92SAndroid Build Coastguard Worker int delta[17]; // old 'firstsymbol' - old 'firstcode'
1655*89c4ff92SAndroid Build Coastguard Worker } stbi__huffman;
1656*89c4ff92SAndroid Build Coastguard Worker
1657*89c4ff92SAndroid Build Coastguard Worker typedef struct
1658*89c4ff92SAndroid Build Coastguard Worker {
1659*89c4ff92SAndroid Build Coastguard Worker stbi__context *s;
1660*89c4ff92SAndroid Build Coastguard Worker stbi__huffman huff_dc[4];
1661*89c4ff92SAndroid Build Coastguard Worker stbi__huffman huff_ac[4];
1662*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 dequant[4][64];
1663*89c4ff92SAndroid Build Coastguard Worker stbi__int16 fast_ac[4][1 << FAST_BITS];
1664*89c4ff92SAndroid Build Coastguard Worker
1665*89c4ff92SAndroid Build Coastguard Worker // sizes for components, interleaved MCUs
1666*89c4ff92SAndroid Build Coastguard Worker int img_h_max, img_v_max;
1667*89c4ff92SAndroid Build Coastguard Worker int img_mcu_x, img_mcu_y;
1668*89c4ff92SAndroid Build Coastguard Worker int img_mcu_w, img_mcu_h;
1669*89c4ff92SAndroid Build Coastguard Worker
1670*89c4ff92SAndroid Build Coastguard Worker // definition of jpeg image component
1671*89c4ff92SAndroid Build Coastguard Worker struct
1672*89c4ff92SAndroid Build Coastguard Worker {
1673*89c4ff92SAndroid Build Coastguard Worker int id;
1674*89c4ff92SAndroid Build Coastguard Worker int h,v;
1675*89c4ff92SAndroid Build Coastguard Worker int tq;
1676*89c4ff92SAndroid Build Coastguard Worker int hd,ha;
1677*89c4ff92SAndroid Build Coastguard Worker int dc_pred;
1678*89c4ff92SAndroid Build Coastguard Worker
1679*89c4ff92SAndroid Build Coastguard Worker int x,y,w2,h2;
1680*89c4ff92SAndroid Build Coastguard Worker stbi_uc *data;
1681*89c4ff92SAndroid Build Coastguard Worker void *raw_data, *raw_coeff;
1682*89c4ff92SAndroid Build Coastguard Worker stbi_uc *linebuf;
1683*89c4ff92SAndroid Build Coastguard Worker short *coeff; // progressive only
1684*89c4ff92SAndroid Build Coastguard Worker int coeff_w, coeff_h; // number of 8x8 coefficient blocks
1685*89c4ff92SAndroid Build Coastguard Worker } img_comp[4];
1686*89c4ff92SAndroid Build Coastguard Worker
1687*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 code_buffer; // jpeg entropy-coded buffer
1688*89c4ff92SAndroid Build Coastguard Worker int code_bits; // number of valid bits
1689*89c4ff92SAndroid Build Coastguard Worker unsigned char marker; // marker seen while filling entropy buffer
1690*89c4ff92SAndroid Build Coastguard Worker int nomore; // flag if we saw a marker so must stop
1691*89c4ff92SAndroid Build Coastguard Worker
1692*89c4ff92SAndroid Build Coastguard Worker int progressive;
1693*89c4ff92SAndroid Build Coastguard Worker int spec_start;
1694*89c4ff92SAndroid Build Coastguard Worker int spec_end;
1695*89c4ff92SAndroid Build Coastguard Worker int succ_high;
1696*89c4ff92SAndroid Build Coastguard Worker int succ_low;
1697*89c4ff92SAndroid Build Coastguard Worker int eob_run;
1698*89c4ff92SAndroid Build Coastguard Worker int jfif;
1699*89c4ff92SAndroid Build Coastguard Worker int app14_color_transform; // Adobe APP14 tag
1700*89c4ff92SAndroid Build Coastguard Worker int rgb;
1701*89c4ff92SAndroid Build Coastguard Worker
1702*89c4ff92SAndroid Build Coastguard Worker int scan_n, order[4];
1703*89c4ff92SAndroid Build Coastguard Worker int restart_interval, todo;
1704*89c4ff92SAndroid Build Coastguard Worker
1705*89c4ff92SAndroid Build Coastguard Worker // kernels
1706*89c4ff92SAndroid Build Coastguard Worker void (*idct_block_kernel)(stbi_uc *out, int out_stride, short data[64]);
1707*89c4ff92SAndroid Build Coastguard Worker void (*YCbCr_to_RGB_kernel)(stbi_uc *out, const stbi_uc *y, const stbi_uc *pcb, const stbi_uc *pcr, int count, int step);
1708*89c4ff92SAndroid Build Coastguard Worker stbi_uc *(*resample_row_hv_2_kernel)(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far, int w, int hs);
1709*89c4ff92SAndroid Build Coastguard Worker } stbi__jpeg;
1710*89c4ff92SAndroid Build Coastguard Worker
stbi__build_huffman(stbi__huffman * h,int * count)1711*89c4ff92SAndroid Build Coastguard Worker static int stbi__build_huffman(stbi__huffman *h, int *count)
1712*89c4ff92SAndroid Build Coastguard Worker {
1713*89c4ff92SAndroid Build Coastguard Worker int i,j,k=0,code;
1714*89c4ff92SAndroid Build Coastguard Worker // build size list for each symbol (from JPEG spec)
1715*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < 16; ++i)
1716*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < count[i]; ++j)
1717*89c4ff92SAndroid Build Coastguard Worker h->size[k++] = (stbi_uc) (i+1);
1718*89c4ff92SAndroid Build Coastguard Worker h->size[k] = 0;
1719*89c4ff92SAndroid Build Coastguard Worker
1720*89c4ff92SAndroid Build Coastguard Worker // compute actual symbols (from jpeg spec)
1721*89c4ff92SAndroid Build Coastguard Worker code = 0;
1722*89c4ff92SAndroid Build Coastguard Worker k = 0;
1723*89c4ff92SAndroid Build Coastguard Worker for(j=1; j <= 16; ++j) {
1724*89c4ff92SAndroid Build Coastguard Worker // compute delta to add to code to compute symbol id
1725*89c4ff92SAndroid Build Coastguard Worker h->delta[j] = k - code;
1726*89c4ff92SAndroid Build Coastguard Worker if (h->size[k] == j) {
1727*89c4ff92SAndroid Build Coastguard Worker while (h->size[k] == j)
1728*89c4ff92SAndroid Build Coastguard Worker h->code[k++] = (stbi__uint16) (code++);
1729*89c4ff92SAndroid Build Coastguard Worker if (code-1 >= (1 << j)) return stbi__err("bad code lengths","Corrupt JPEG");
1730*89c4ff92SAndroid Build Coastguard Worker }
1731*89c4ff92SAndroid Build Coastguard Worker // compute largest code + 1 for this size, preshifted as needed later
1732*89c4ff92SAndroid Build Coastguard Worker h->maxcode[j] = code << (16-j);
1733*89c4ff92SAndroid Build Coastguard Worker code <<= 1;
1734*89c4ff92SAndroid Build Coastguard Worker }
1735*89c4ff92SAndroid Build Coastguard Worker h->maxcode[j] = 0xffffffff;
1736*89c4ff92SAndroid Build Coastguard Worker
1737*89c4ff92SAndroid Build Coastguard Worker // build non-spec acceleration table; 255 is flag for not-accelerated
1738*89c4ff92SAndroid Build Coastguard Worker memset(h->fast, 255, 1 << FAST_BITS);
1739*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < k; ++i) {
1740*89c4ff92SAndroid Build Coastguard Worker int s = h->size[i];
1741*89c4ff92SAndroid Build Coastguard Worker if (s <= FAST_BITS) {
1742*89c4ff92SAndroid Build Coastguard Worker int c = h->code[i] << (FAST_BITS-s);
1743*89c4ff92SAndroid Build Coastguard Worker int m = 1 << (FAST_BITS-s);
1744*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < m; ++j) {
1745*89c4ff92SAndroid Build Coastguard Worker h->fast[c+j] = (stbi_uc) i;
1746*89c4ff92SAndroid Build Coastguard Worker }
1747*89c4ff92SAndroid Build Coastguard Worker }
1748*89c4ff92SAndroid Build Coastguard Worker }
1749*89c4ff92SAndroid Build Coastguard Worker return 1;
1750*89c4ff92SAndroid Build Coastguard Worker }
1751*89c4ff92SAndroid Build Coastguard Worker
1752*89c4ff92SAndroid Build Coastguard Worker // build a table that decodes both magnitude and value of small ACs in
1753*89c4ff92SAndroid Build Coastguard Worker // one go.
stbi__build_fast_ac(stbi__int16 * fast_ac,stbi__huffman * h)1754*89c4ff92SAndroid Build Coastguard Worker static void stbi__build_fast_ac(stbi__int16 *fast_ac, stbi__huffman *h)
1755*89c4ff92SAndroid Build Coastguard Worker {
1756*89c4ff92SAndroid Build Coastguard Worker int i;
1757*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < (1 << FAST_BITS); ++i) {
1758*89c4ff92SAndroid Build Coastguard Worker stbi_uc fast = h->fast[i];
1759*89c4ff92SAndroid Build Coastguard Worker fast_ac[i] = 0;
1760*89c4ff92SAndroid Build Coastguard Worker if (fast < 255) {
1761*89c4ff92SAndroid Build Coastguard Worker int rs = h->values[fast];
1762*89c4ff92SAndroid Build Coastguard Worker int run = (rs >> 4) & 15;
1763*89c4ff92SAndroid Build Coastguard Worker int magbits = rs & 15;
1764*89c4ff92SAndroid Build Coastguard Worker int len = h->size[fast];
1765*89c4ff92SAndroid Build Coastguard Worker
1766*89c4ff92SAndroid Build Coastguard Worker if (magbits && len + magbits <= FAST_BITS) {
1767*89c4ff92SAndroid Build Coastguard Worker // magnitude code followed by receive_extend code
1768*89c4ff92SAndroid Build Coastguard Worker int k = ((i << len) & ((1 << FAST_BITS) - 1)) >> (FAST_BITS - magbits);
1769*89c4ff92SAndroid Build Coastguard Worker int m = 1 << (magbits - 1);
1770*89c4ff92SAndroid Build Coastguard Worker if (k < m) k += (~0U << magbits) + 1;
1771*89c4ff92SAndroid Build Coastguard Worker // if the result is small enough, we can fit it in fast_ac table
1772*89c4ff92SAndroid Build Coastguard Worker if (k >= -128 && k <= 127)
1773*89c4ff92SAndroid Build Coastguard Worker fast_ac[i] = (stbi__int16) ((k << 8) + (run << 4) + (len + magbits));
1774*89c4ff92SAndroid Build Coastguard Worker }
1775*89c4ff92SAndroid Build Coastguard Worker }
1776*89c4ff92SAndroid Build Coastguard Worker }
1777*89c4ff92SAndroid Build Coastguard Worker }
1778*89c4ff92SAndroid Build Coastguard Worker
stbi__grow_buffer_unsafe(stbi__jpeg * j)1779*89c4ff92SAndroid Build Coastguard Worker static void stbi__grow_buffer_unsafe(stbi__jpeg *j)
1780*89c4ff92SAndroid Build Coastguard Worker {
1781*89c4ff92SAndroid Build Coastguard Worker do {
1782*89c4ff92SAndroid Build Coastguard Worker int b = j->nomore ? 0 : stbi__get8(j->s);
1783*89c4ff92SAndroid Build Coastguard Worker if (b == 0xff) {
1784*89c4ff92SAndroid Build Coastguard Worker int c = stbi__get8(j->s);
1785*89c4ff92SAndroid Build Coastguard Worker while (c == 0xff) c = stbi__get8(j->s); // consume fill bytes
1786*89c4ff92SAndroid Build Coastguard Worker if (c != 0) {
1787*89c4ff92SAndroid Build Coastguard Worker j->marker = (unsigned char) c;
1788*89c4ff92SAndroid Build Coastguard Worker j->nomore = 1;
1789*89c4ff92SAndroid Build Coastguard Worker return;
1790*89c4ff92SAndroid Build Coastguard Worker }
1791*89c4ff92SAndroid Build Coastguard Worker }
1792*89c4ff92SAndroid Build Coastguard Worker j->code_buffer |= b << (24 - j->code_bits);
1793*89c4ff92SAndroid Build Coastguard Worker j->code_bits += 8;
1794*89c4ff92SAndroid Build Coastguard Worker } while (j->code_bits <= 24);
1795*89c4ff92SAndroid Build Coastguard Worker }
1796*89c4ff92SAndroid Build Coastguard Worker
1797*89c4ff92SAndroid Build Coastguard Worker // (1 << n) - 1
1798*89c4ff92SAndroid Build Coastguard Worker static stbi__uint32 stbi__bmask[17]={0,1,3,7,15,31,63,127,255,511,1023,2047,4095,8191,16383,32767,65535};
1799*89c4ff92SAndroid Build Coastguard Worker
1800*89c4ff92SAndroid Build Coastguard Worker // decode a jpeg huffman value from the bitstream
stbi__jpeg_huff_decode(stbi__jpeg * j,stbi__huffman * h)1801*89c4ff92SAndroid Build Coastguard Worker stbi_inline static int stbi__jpeg_huff_decode(stbi__jpeg *j, stbi__huffman *h)
1802*89c4ff92SAndroid Build Coastguard Worker {
1803*89c4ff92SAndroid Build Coastguard Worker unsigned int temp;
1804*89c4ff92SAndroid Build Coastguard Worker int c,k;
1805*89c4ff92SAndroid Build Coastguard Worker
1806*89c4ff92SAndroid Build Coastguard Worker if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
1807*89c4ff92SAndroid Build Coastguard Worker
1808*89c4ff92SAndroid Build Coastguard Worker // look at the top FAST_BITS and determine what symbol ID it is,
1809*89c4ff92SAndroid Build Coastguard Worker // if the code is <= FAST_BITS
1810*89c4ff92SAndroid Build Coastguard Worker c = (j->code_buffer >> (32 - FAST_BITS)) & ((1 << FAST_BITS)-1);
1811*89c4ff92SAndroid Build Coastguard Worker k = h->fast[c];
1812*89c4ff92SAndroid Build Coastguard Worker if (k < 255) {
1813*89c4ff92SAndroid Build Coastguard Worker int s = h->size[k];
1814*89c4ff92SAndroid Build Coastguard Worker if (s > j->code_bits)
1815*89c4ff92SAndroid Build Coastguard Worker return -1;
1816*89c4ff92SAndroid Build Coastguard Worker j->code_buffer <<= s;
1817*89c4ff92SAndroid Build Coastguard Worker j->code_bits -= s;
1818*89c4ff92SAndroid Build Coastguard Worker return h->values[k];
1819*89c4ff92SAndroid Build Coastguard Worker }
1820*89c4ff92SAndroid Build Coastguard Worker
1821*89c4ff92SAndroid Build Coastguard Worker // naive test is to shift the code_buffer down so k bits are
1822*89c4ff92SAndroid Build Coastguard Worker // valid, then test against maxcode. To speed this up, we've
1823*89c4ff92SAndroid Build Coastguard Worker // preshifted maxcode left so that it has (16-k) 0s at the
1824*89c4ff92SAndroid Build Coastguard Worker // end; in other words, regardless of the number of bits, it
1825*89c4ff92SAndroid Build Coastguard Worker // wants to be compared against something shifted to have 16;
1826*89c4ff92SAndroid Build Coastguard Worker // that way we don't need to shift inside the loop.
1827*89c4ff92SAndroid Build Coastguard Worker temp = j->code_buffer >> 16;
1828*89c4ff92SAndroid Build Coastguard Worker for (k=FAST_BITS+1 ; ; ++k)
1829*89c4ff92SAndroid Build Coastguard Worker if (temp < h->maxcode[k])
1830*89c4ff92SAndroid Build Coastguard Worker break;
1831*89c4ff92SAndroid Build Coastguard Worker if (k == 17) {
1832*89c4ff92SAndroid Build Coastguard Worker // error! code not found
1833*89c4ff92SAndroid Build Coastguard Worker j->code_bits -= 16;
1834*89c4ff92SAndroid Build Coastguard Worker return -1;
1835*89c4ff92SAndroid Build Coastguard Worker }
1836*89c4ff92SAndroid Build Coastguard Worker
1837*89c4ff92SAndroid Build Coastguard Worker if (k > j->code_bits)
1838*89c4ff92SAndroid Build Coastguard Worker return -1;
1839*89c4ff92SAndroid Build Coastguard Worker
1840*89c4ff92SAndroid Build Coastguard Worker // convert the huffman code to the symbol id
1841*89c4ff92SAndroid Build Coastguard Worker c = ((j->code_buffer >> (32 - k)) & stbi__bmask[k]) + h->delta[k];
1842*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT((((j->code_buffer) >> (32 - h->size[c])) & stbi__bmask[h->size[c]]) == h->code[c]);
1843*89c4ff92SAndroid Build Coastguard Worker
1844*89c4ff92SAndroid Build Coastguard Worker // convert the id to a symbol
1845*89c4ff92SAndroid Build Coastguard Worker j->code_bits -= k;
1846*89c4ff92SAndroid Build Coastguard Worker j->code_buffer <<= k;
1847*89c4ff92SAndroid Build Coastguard Worker return h->values[c];
1848*89c4ff92SAndroid Build Coastguard Worker }
1849*89c4ff92SAndroid Build Coastguard Worker
1850*89c4ff92SAndroid Build Coastguard Worker // bias[n] = (-1<<n) + 1
1851*89c4ff92SAndroid Build Coastguard Worker static int const stbi__jbias[16] = {0,-1,-3,-7,-15,-31,-63,-127,-255,-511,-1023,-2047,-4095,-8191,-16383,-32767};
1852*89c4ff92SAndroid Build Coastguard Worker
1853*89c4ff92SAndroid Build Coastguard Worker // combined JPEG 'receive' and JPEG 'extend', since baseline
1854*89c4ff92SAndroid Build Coastguard Worker // always extends everything it receives.
stbi__extend_receive(stbi__jpeg * j,int n)1855*89c4ff92SAndroid Build Coastguard Worker stbi_inline static int stbi__extend_receive(stbi__jpeg *j, int n)
1856*89c4ff92SAndroid Build Coastguard Worker {
1857*89c4ff92SAndroid Build Coastguard Worker unsigned int k;
1858*89c4ff92SAndroid Build Coastguard Worker int sgn;
1859*89c4ff92SAndroid Build Coastguard Worker if (j->code_bits < n) stbi__grow_buffer_unsafe(j);
1860*89c4ff92SAndroid Build Coastguard Worker
1861*89c4ff92SAndroid Build Coastguard Worker sgn = (stbi__int32)j->code_buffer >> 31; // sign bit is always in MSB
1862*89c4ff92SAndroid Build Coastguard Worker k = stbi_lrot(j->code_buffer, n);
1863*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(n >= 0 && n < (int) (sizeof(stbi__bmask)/sizeof(*stbi__bmask)));
1864*89c4ff92SAndroid Build Coastguard Worker j->code_buffer = k & ~stbi__bmask[n];
1865*89c4ff92SAndroid Build Coastguard Worker k &= stbi__bmask[n];
1866*89c4ff92SAndroid Build Coastguard Worker j->code_bits -= n;
1867*89c4ff92SAndroid Build Coastguard Worker return k + (stbi__jbias[n] & ~sgn);
1868*89c4ff92SAndroid Build Coastguard Worker }
1869*89c4ff92SAndroid Build Coastguard Worker
1870*89c4ff92SAndroid Build Coastguard Worker // get some unsigned bits
stbi__jpeg_get_bits(stbi__jpeg * j,int n)1871*89c4ff92SAndroid Build Coastguard Worker stbi_inline static int stbi__jpeg_get_bits(stbi__jpeg *j, int n)
1872*89c4ff92SAndroid Build Coastguard Worker {
1873*89c4ff92SAndroid Build Coastguard Worker unsigned int k;
1874*89c4ff92SAndroid Build Coastguard Worker if (j->code_bits < n) stbi__grow_buffer_unsafe(j);
1875*89c4ff92SAndroid Build Coastguard Worker k = stbi_lrot(j->code_buffer, n);
1876*89c4ff92SAndroid Build Coastguard Worker j->code_buffer = k & ~stbi__bmask[n];
1877*89c4ff92SAndroid Build Coastguard Worker k &= stbi__bmask[n];
1878*89c4ff92SAndroid Build Coastguard Worker j->code_bits -= n;
1879*89c4ff92SAndroid Build Coastguard Worker return k;
1880*89c4ff92SAndroid Build Coastguard Worker }
1881*89c4ff92SAndroid Build Coastguard Worker
stbi__jpeg_get_bit(stbi__jpeg * j)1882*89c4ff92SAndroid Build Coastguard Worker stbi_inline static int stbi__jpeg_get_bit(stbi__jpeg *j)
1883*89c4ff92SAndroid Build Coastguard Worker {
1884*89c4ff92SAndroid Build Coastguard Worker unsigned int k;
1885*89c4ff92SAndroid Build Coastguard Worker if (j->code_bits < 1) stbi__grow_buffer_unsafe(j);
1886*89c4ff92SAndroid Build Coastguard Worker k = j->code_buffer;
1887*89c4ff92SAndroid Build Coastguard Worker j->code_buffer <<= 1;
1888*89c4ff92SAndroid Build Coastguard Worker --j->code_bits;
1889*89c4ff92SAndroid Build Coastguard Worker return k & 0x80000000;
1890*89c4ff92SAndroid Build Coastguard Worker }
1891*89c4ff92SAndroid Build Coastguard Worker
1892*89c4ff92SAndroid Build Coastguard Worker // given a value that's at position X in the zigzag stream,
1893*89c4ff92SAndroid Build Coastguard Worker // where does it appear in the 8x8 matrix coded as row-major?
1894*89c4ff92SAndroid Build Coastguard Worker static stbi_uc stbi__jpeg_dezigzag[64+15] =
1895*89c4ff92SAndroid Build Coastguard Worker {
1896*89c4ff92SAndroid Build Coastguard Worker 0, 1, 8, 16, 9, 2, 3, 10,
1897*89c4ff92SAndroid Build Coastguard Worker 17, 24, 32, 25, 18, 11, 4, 5,
1898*89c4ff92SAndroid Build Coastguard Worker 12, 19, 26, 33, 40, 48, 41, 34,
1899*89c4ff92SAndroid Build Coastguard Worker 27, 20, 13, 6, 7, 14, 21, 28,
1900*89c4ff92SAndroid Build Coastguard Worker 35, 42, 49, 56, 57, 50, 43, 36,
1901*89c4ff92SAndroid Build Coastguard Worker 29, 22, 15, 23, 30, 37, 44, 51,
1902*89c4ff92SAndroid Build Coastguard Worker 58, 59, 52, 45, 38, 31, 39, 46,
1903*89c4ff92SAndroid Build Coastguard Worker 53, 60, 61, 54, 47, 55, 62, 63,
1904*89c4ff92SAndroid Build Coastguard Worker // let corrupt input sample past end
1905*89c4ff92SAndroid Build Coastguard Worker 63, 63, 63, 63, 63, 63, 63, 63,
1906*89c4ff92SAndroid Build Coastguard Worker 63, 63, 63, 63, 63, 63, 63
1907*89c4ff92SAndroid Build Coastguard Worker };
1908*89c4ff92SAndroid Build Coastguard Worker
1909*89c4ff92SAndroid Build Coastguard Worker // decode one 64-entry block--
stbi__jpeg_decode_block(stbi__jpeg * j,short data[64],stbi__huffman * hdc,stbi__huffman * hac,stbi__int16 * fac,int b,stbi__uint16 * dequant)1910*89c4ff92SAndroid Build Coastguard Worker static int stbi__jpeg_decode_block(stbi__jpeg *j, short data[64], stbi__huffman *hdc, stbi__huffman *hac, stbi__int16 *fac, int b, stbi__uint16 *dequant)
1911*89c4ff92SAndroid Build Coastguard Worker {
1912*89c4ff92SAndroid Build Coastguard Worker int diff,dc,k;
1913*89c4ff92SAndroid Build Coastguard Worker int t;
1914*89c4ff92SAndroid Build Coastguard Worker
1915*89c4ff92SAndroid Build Coastguard Worker if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
1916*89c4ff92SAndroid Build Coastguard Worker t = stbi__jpeg_huff_decode(j, hdc);
1917*89c4ff92SAndroid Build Coastguard Worker if (t < 0) return stbi__err("bad huffman code","Corrupt JPEG");
1918*89c4ff92SAndroid Build Coastguard Worker
1919*89c4ff92SAndroid Build Coastguard Worker // 0 all the ac values now so we can do it 32-bits at a time
1920*89c4ff92SAndroid Build Coastguard Worker memset(data,0,64*sizeof(data[0]));
1921*89c4ff92SAndroid Build Coastguard Worker
1922*89c4ff92SAndroid Build Coastguard Worker diff = t ? stbi__extend_receive(j, t) : 0;
1923*89c4ff92SAndroid Build Coastguard Worker dc = j->img_comp[b].dc_pred + diff;
1924*89c4ff92SAndroid Build Coastguard Worker j->img_comp[b].dc_pred = dc;
1925*89c4ff92SAndroid Build Coastguard Worker data[0] = (short) (dc * dequant[0]);
1926*89c4ff92SAndroid Build Coastguard Worker
1927*89c4ff92SAndroid Build Coastguard Worker // decode AC components, see JPEG spec
1928*89c4ff92SAndroid Build Coastguard Worker k = 1;
1929*89c4ff92SAndroid Build Coastguard Worker do {
1930*89c4ff92SAndroid Build Coastguard Worker unsigned int zig;
1931*89c4ff92SAndroid Build Coastguard Worker int c,r,s;
1932*89c4ff92SAndroid Build Coastguard Worker if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
1933*89c4ff92SAndroid Build Coastguard Worker c = (j->code_buffer >> (32 - FAST_BITS)) & ((1 << FAST_BITS)-1);
1934*89c4ff92SAndroid Build Coastguard Worker r = fac[c];
1935*89c4ff92SAndroid Build Coastguard Worker if (r) { // fast-AC path
1936*89c4ff92SAndroid Build Coastguard Worker k += (r >> 4) & 15; // run
1937*89c4ff92SAndroid Build Coastguard Worker s = r & 15; // combined length
1938*89c4ff92SAndroid Build Coastguard Worker j->code_buffer <<= s;
1939*89c4ff92SAndroid Build Coastguard Worker j->code_bits -= s;
1940*89c4ff92SAndroid Build Coastguard Worker // decode into unzigzag'd location
1941*89c4ff92SAndroid Build Coastguard Worker zig = stbi__jpeg_dezigzag[k++];
1942*89c4ff92SAndroid Build Coastguard Worker data[zig] = (short) ((r >> 8) * dequant[zig]);
1943*89c4ff92SAndroid Build Coastguard Worker } else {
1944*89c4ff92SAndroid Build Coastguard Worker int rs = stbi__jpeg_huff_decode(j, hac);
1945*89c4ff92SAndroid Build Coastguard Worker if (rs < 0) return stbi__err("bad huffman code","Corrupt JPEG");
1946*89c4ff92SAndroid Build Coastguard Worker s = rs & 15;
1947*89c4ff92SAndroid Build Coastguard Worker r = rs >> 4;
1948*89c4ff92SAndroid Build Coastguard Worker if (s == 0) {
1949*89c4ff92SAndroid Build Coastguard Worker if (rs != 0xf0) break; // end block
1950*89c4ff92SAndroid Build Coastguard Worker k += 16;
1951*89c4ff92SAndroid Build Coastguard Worker } else {
1952*89c4ff92SAndroid Build Coastguard Worker k += r;
1953*89c4ff92SAndroid Build Coastguard Worker // decode into unzigzag'd location
1954*89c4ff92SAndroid Build Coastguard Worker zig = stbi__jpeg_dezigzag[k++];
1955*89c4ff92SAndroid Build Coastguard Worker data[zig] = (short) (stbi__extend_receive(j,s) * dequant[zig]);
1956*89c4ff92SAndroid Build Coastguard Worker }
1957*89c4ff92SAndroid Build Coastguard Worker }
1958*89c4ff92SAndroid Build Coastguard Worker } while (k < 64);
1959*89c4ff92SAndroid Build Coastguard Worker return 1;
1960*89c4ff92SAndroid Build Coastguard Worker }
1961*89c4ff92SAndroid Build Coastguard Worker
stbi__jpeg_decode_block_prog_dc(stbi__jpeg * j,short data[64],stbi__huffman * hdc,int b)1962*89c4ff92SAndroid Build Coastguard Worker static int stbi__jpeg_decode_block_prog_dc(stbi__jpeg *j, short data[64], stbi__huffman *hdc, int b)
1963*89c4ff92SAndroid Build Coastguard Worker {
1964*89c4ff92SAndroid Build Coastguard Worker int diff,dc;
1965*89c4ff92SAndroid Build Coastguard Worker int t;
1966*89c4ff92SAndroid Build Coastguard Worker if (j->spec_end != 0) return stbi__err("can't merge dc and ac", "Corrupt JPEG");
1967*89c4ff92SAndroid Build Coastguard Worker
1968*89c4ff92SAndroid Build Coastguard Worker if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
1969*89c4ff92SAndroid Build Coastguard Worker
1970*89c4ff92SAndroid Build Coastguard Worker if (j->succ_high == 0) {
1971*89c4ff92SAndroid Build Coastguard Worker // first scan for DC coefficient, must be first
1972*89c4ff92SAndroid Build Coastguard Worker memset(data,0,64*sizeof(data[0])); // 0 all the ac values now
1973*89c4ff92SAndroid Build Coastguard Worker t = stbi__jpeg_huff_decode(j, hdc);
1974*89c4ff92SAndroid Build Coastguard Worker diff = t ? stbi__extend_receive(j, t) : 0;
1975*89c4ff92SAndroid Build Coastguard Worker
1976*89c4ff92SAndroid Build Coastguard Worker dc = j->img_comp[b].dc_pred + diff;
1977*89c4ff92SAndroid Build Coastguard Worker j->img_comp[b].dc_pred = dc;
1978*89c4ff92SAndroid Build Coastguard Worker data[0] = (short) (dc << j->succ_low);
1979*89c4ff92SAndroid Build Coastguard Worker } else {
1980*89c4ff92SAndroid Build Coastguard Worker // refinement scan for DC coefficient
1981*89c4ff92SAndroid Build Coastguard Worker if (stbi__jpeg_get_bit(j))
1982*89c4ff92SAndroid Build Coastguard Worker data[0] += (short) (1 << j->succ_low);
1983*89c4ff92SAndroid Build Coastguard Worker }
1984*89c4ff92SAndroid Build Coastguard Worker return 1;
1985*89c4ff92SAndroid Build Coastguard Worker }
1986*89c4ff92SAndroid Build Coastguard Worker
1987*89c4ff92SAndroid Build Coastguard Worker // @OPTIMIZE: store non-zigzagged during the decode passes,
1988*89c4ff92SAndroid Build Coastguard Worker // and only de-zigzag when dequantizing
stbi__jpeg_decode_block_prog_ac(stbi__jpeg * j,short data[64],stbi__huffman * hac,stbi__int16 * fac)1989*89c4ff92SAndroid Build Coastguard Worker static int stbi__jpeg_decode_block_prog_ac(stbi__jpeg *j, short data[64], stbi__huffman *hac, stbi__int16 *fac)
1990*89c4ff92SAndroid Build Coastguard Worker {
1991*89c4ff92SAndroid Build Coastguard Worker int k;
1992*89c4ff92SAndroid Build Coastguard Worker if (j->spec_start == 0) return stbi__err("can't merge dc and ac", "Corrupt JPEG");
1993*89c4ff92SAndroid Build Coastguard Worker
1994*89c4ff92SAndroid Build Coastguard Worker if (j->succ_high == 0) {
1995*89c4ff92SAndroid Build Coastguard Worker int shift = j->succ_low;
1996*89c4ff92SAndroid Build Coastguard Worker
1997*89c4ff92SAndroid Build Coastguard Worker if (j->eob_run) {
1998*89c4ff92SAndroid Build Coastguard Worker --j->eob_run;
1999*89c4ff92SAndroid Build Coastguard Worker return 1;
2000*89c4ff92SAndroid Build Coastguard Worker }
2001*89c4ff92SAndroid Build Coastguard Worker
2002*89c4ff92SAndroid Build Coastguard Worker k = j->spec_start;
2003*89c4ff92SAndroid Build Coastguard Worker do {
2004*89c4ff92SAndroid Build Coastguard Worker unsigned int zig;
2005*89c4ff92SAndroid Build Coastguard Worker int c,r,s;
2006*89c4ff92SAndroid Build Coastguard Worker if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
2007*89c4ff92SAndroid Build Coastguard Worker c = (j->code_buffer >> (32 - FAST_BITS)) & ((1 << FAST_BITS)-1);
2008*89c4ff92SAndroid Build Coastguard Worker r = fac[c];
2009*89c4ff92SAndroid Build Coastguard Worker if (r) { // fast-AC path
2010*89c4ff92SAndroid Build Coastguard Worker k += (r >> 4) & 15; // run
2011*89c4ff92SAndroid Build Coastguard Worker s = r & 15; // combined length
2012*89c4ff92SAndroid Build Coastguard Worker j->code_buffer <<= s;
2013*89c4ff92SAndroid Build Coastguard Worker j->code_bits -= s;
2014*89c4ff92SAndroid Build Coastguard Worker zig = stbi__jpeg_dezigzag[k++];
2015*89c4ff92SAndroid Build Coastguard Worker data[zig] = (short) ((r >> 8) << shift);
2016*89c4ff92SAndroid Build Coastguard Worker } else {
2017*89c4ff92SAndroid Build Coastguard Worker int rs = stbi__jpeg_huff_decode(j, hac);
2018*89c4ff92SAndroid Build Coastguard Worker if (rs < 0) return stbi__err("bad huffman code","Corrupt JPEG");
2019*89c4ff92SAndroid Build Coastguard Worker s = rs & 15;
2020*89c4ff92SAndroid Build Coastguard Worker r = rs >> 4;
2021*89c4ff92SAndroid Build Coastguard Worker if (s == 0) {
2022*89c4ff92SAndroid Build Coastguard Worker if (r < 15) {
2023*89c4ff92SAndroid Build Coastguard Worker j->eob_run = (1 << r);
2024*89c4ff92SAndroid Build Coastguard Worker if (r)
2025*89c4ff92SAndroid Build Coastguard Worker j->eob_run += stbi__jpeg_get_bits(j, r);
2026*89c4ff92SAndroid Build Coastguard Worker --j->eob_run;
2027*89c4ff92SAndroid Build Coastguard Worker break;
2028*89c4ff92SAndroid Build Coastguard Worker }
2029*89c4ff92SAndroid Build Coastguard Worker k += 16;
2030*89c4ff92SAndroid Build Coastguard Worker } else {
2031*89c4ff92SAndroid Build Coastguard Worker k += r;
2032*89c4ff92SAndroid Build Coastguard Worker zig = stbi__jpeg_dezigzag[k++];
2033*89c4ff92SAndroid Build Coastguard Worker data[zig] = (short) (stbi__extend_receive(j,s) << shift);
2034*89c4ff92SAndroid Build Coastguard Worker }
2035*89c4ff92SAndroid Build Coastguard Worker }
2036*89c4ff92SAndroid Build Coastguard Worker } while (k <= j->spec_end);
2037*89c4ff92SAndroid Build Coastguard Worker } else {
2038*89c4ff92SAndroid Build Coastguard Worker // refinement scan for these AC coefficients
2039*89c4ff92SAndroid Build Coastguard Worker
2040*89c4ff92SAndroid Build Coastguard Worker short bit = (short) (1 << j->succ_low);
2041*89c4ff92SAndroid Build Coastguard Worker
2042*89c4ff92SAndroid Build Coastguard Worker if (j->eob_run) {
2043*89c4ff92SAndroid Build Coastguard Worker --j->eob_run;
2044*89c4ff92SAndroid Build Coastguard Worker for (k = j->spec_start; k <= j->spec_end; ++k) {
2045*89c4ff92SAndroid Build Coastguard Worker short *p = &data[stbi__jpeg_dezigzag[k]];
2046*89c4ff92SAndroid Build Coastguard Worker if (*p != 0)
2047*89c4ff92SAndroid Build Coastguard Worker if (stbi__jpeg_get_bit(j))
2048*89c4ff92SAndroid Build Coastguard Worker if ((*p & bit)==0) {
2049*89c4ff92SAndroid Build Coastguard Worker if (*p > 0)
2050*89c4ff92SAndroid Build Coastguard Worker *p += bit;
2051*89c4ff92SAndroid Build Coastguard Worker else
2052*89c4ff92SAndroid Build Coastguard Worker *p -= bit;
2053*89c4ff92SAndroid Build Coastguard Worker }
2054*89c4ff92SAndroid Build Coastguard Worker }
2055*89c4ff92SAndroid Build Coastguard Worker } else {
2056*89c4ff92SAndroid Build Coastguard Worker k = j->spec_start;
2057*89c4ff92SAndroid Build Coastguard Worker do {
2058*89c4ff92SAndroid Build Coastguard Worker int r,s;
2059*89c4ff92SAndroid Build Coastguard Worker int rs = stbi__jpeg_huff_decode(j, hac); // @OPTIMIZE see if we can use the fast path here, advance-by-r is so slow, eh
2060*89c4ff92SAndroid Build Coastguard Worker if (rs < 0) return stbi__err("bad huffman code","Corrupt JPEG");
2061*89c4ff92SAndroid Build Coastguard Worker s = rs & 15;
2062*89c4ff92SAndroid Build Coastguard Worker r = rs >> 4;
2063*89c4ff92SAndroid Build Coastguard Worker if (s == 0) {
2064*89c4ff92SAndroid Build Coastguard Worker if (r < 15) {
2065*89c4ff92SAndroid Build Coastguard Worker j->eob_run = (1 << r) - 1;
2066*89c4ff92SAndroid Build Coastguard Worker if (r)
2067*89c4ff92SAndroid Build Coastguard Worker j->eob_run += stbi__jpeg_get_bits(j, r);
2068*89c4ff92SAndroid Build Coastguard Worker r = 64; // force end of block
2069*89c4ff92SAndroid Build Coastguard Worker } else {
2070*89c4ff92SAndroid Build Coastguard Worker // r=15 s=0 should write 16 0s, so we just do
2071*89c4ff92SAndroid Build Coastguard Worker // a run of 15 0s and then write s (which is 0),
2072*89c4ff92SAndroid Build Coastguard Worker // so we don't have to do anything special here
2073*89c4ff92SAndroid Build Coastguard Worker }
2074*89c4ff92SAndroid Build Coastguard Worker } else {
2075*89c4ff92SAndroid Build Coastguard Worker if (s != 1) return stbi__err("bad huffman code", "Corrupt JPEG");
2076*89c4ff92SAndroid Build Coastguard Worker // sign bit
2077*89c4ff92SAndroid Build Coastguard Worker if (stbi__jpeg_get_bit(j))
2078*89c4ff92SAndroid Build Coastguard Worker s = bit;
2079*89c4ff92SAndroid Build Coastguard Worker else
2080*89c4ff92SAndroid Build Coastguard Worker s = -bit;
2081*89c4ff92SAndroid Build Coastguard Worker }
2082*89c4ff92SAndroid Build Coastguard Worker
2083*89c4ff92SAndroid Build Coastguard Worker // advance by r
2084*89c4ff92SAndroid Build Coastguard Worker while (k <= j->spec_end) {
2085*89c4ff92SAndroid Build Coastguard Worker short *p = &data[stbi__jpeg_dezigzag[k++]];
2086*89c4ff92SAndroid Build Coastguard Worker if (*p != 0) {
2087*89c4ff92SAndroid Build Coastguard Worker if (stbi__jpeg_get_bit(j))
2088*89c4ff92SAndroid Build Coastguard Worker if ((*p & bit)==0) {
2089*89c4ff92SAndroid Build Coastguard Worker if (*p > 0)
2090*89c4ff92SAndroid Build Coastguard Worker *p += bit;
2091*89c4ff92SAndroid Build Coastguard Worker else
2092*89c4ff92SAndroid Build Coastguard Worker *p -= bit;
2093*89c4ff92SAndroid Build Coastguard Worker }
2094*89c4ff92SAndroid Build Coastguard Worker } else {
2095*89c4ff92SAndroid Build Coastguard Worker if (r == 0) {
2096*89c4ff92SAndroid Build Coastguard Worker *p = (short) s;
2097*89c4ff92SAndroid Build Coastguard Worker break;
2098*89c4ff92SAndroid Build Coastguard Worker }
2099*89c4ff92SAndroid Build Coastguard Worker --r;
2100*89c4ff92SAndroid Build Coastguard Worker }
2101*89c4ff92SAndroid Build Coastguard Worker }
2102*89c4ff92SAndroid Build Coastguard Worker } while (k <= j->spec_end);
2103*89c4ff92SAndroid Build Coastguard Worker }
2104*89c4ff92SAndroid Build Coastguard Worker }
2105*89c4ff92SAndroid Build Coastguard Worker return 1;
2106*89c4ff92SAndroid Build Coastguard Worker }
2107*89c4ff92SAndroid Build Coastguard Worker
2108*89c4ff92SAndroid Build Coastguard Worker // take a -128..127 value and stbi__clamp it and convert to 0..255
stbi__clamp(int x)2109*89c4ff92SAndroid Build Coastguard Worker stbi_inline static stbi_uc stbi__clamp(int x)
2110*89c4ff92SAndroid Build Coastguard Worker {
2111*89c4ff92SAndroid Build Coastguard Worker // trick to use a single test to catch both cases
2112*89c4ff92SAndroid Build Coastguard Worker if ((unsigned int) x > 255) {
2113*89c4ff92SAndroid Build Coastguard Worker if (x < 0) return 0;
2114*89c4ff92SAndroid Build Coastguard Worker if (x > 255) return 255;
2115*89c4ff92SAndroid Build Coastguard Worker }
2116*89c4ff92SAndroid Build Coastguard Worker return (stbi_uc) x;
2117*89c4ff92SAndroid Build Coastguard Worker }
2118*89c4ff92SAndroid Build Coastguard Worker
2119*89c4ff92SAndroid Build Coastguard Worker #define stbi__f2f(x) ((int) (((x) * 4096 + 0.5)))
2120*89c4ff92SAndroid Build Coastguard Worker #define stbi__fsh(x) ((x) << 12)
2121*89c4ff92SAndroid Build Coastguard Worker
2122*89c4ff92SAndroid Build Coastguard Worker // derived from jidctint -- DCT_ISLOW
2123*89c4ff92SAndroid Build Coastguard Worker #define STBI__IDCT_1D(s0,s1,s2,s3,s4,s5,s6,s7) \
2124*89c4ff92SAndroid Build Coastguard Worker int t0,t1,t2,t3,p1,p2,p3,p4,p5,x0,x1,x2,x3; \
2125*89c4ff92SAndroid Build Coastguard Worker p2 = s2; \
2126*89c4ff92SAndroid Build Coastguard Worker p3 = s6; \
2127*89c4ff92SAndroid Build Coastguard Worker p1 = (p2+p3) * stbi__f2f(0.5411961f); \
2128*89c4ff92SAndroid Build Coastguard Worker t2 = p1 + p3*stbi__f2f(-1.847759065f); \
2129*89c4ff92SAndroid Build Coastguard Worker t3 = p1 + p2*stbi__f2f( 0.765366865f); \
2130*89c4ff92SAndroid Build Coastguard Worker p2 = s0; \
2131*89c4ff92SAndroid Build Coastguard Worker p3 = s4; \
2132*89c4ff92SAndroid Build Coastguard Worker t0 = stbi__fsh(p2+p3); \
2133*89c4ff92SAndroid Build Coastguard Worker t1 = stbi__fsh(p2-p3); \
2134*89c4ff92SAndroid Build Coastguard Worker x0 = t0+t3; \
2135*89c4ff92SAndroid Build Coastguard Worker x3 = t0-t3; \
2136*89c4ff92SAndroid Build Coastguard Worker x1 = t1+t2; \
2137*89c4ff92SAndroid Build Coastguard Worker x2 = t1-t2; \
2138*89c4ff92SAndroid Build Coastguard Worker t0 = s7; \
2139*89c4ff92SAndroid Build Coastguard Worker t1 = s5; \
2140*89c4ff92SAndroid Build Coastguard Worker t2 = s3; \
2141*89c4ff92SAndroid Build Coastguard Worker t3 = s1; \
2142*89c4ff92SAndroid Build Coastguard Worker p3 = t0+t2; \
2143*89c4ff92SAndroid Build Coastguard Worker p4 = t1+t3; \
2144*89c4ff92SAndroid Build Coastguard Worker p1 = t0+t3; \
2145*89c4ff92SAndroid Build Coastguard Worker p2 = t1+t2; \
2146*89c4ff92SAndroid Build Coastguard Worker p5 = (p3+p4)*stbi__f2f( 1.175875602f); \
2147*89c4ff92SAndroid Build Coastguard Worker t0 = t0*stbi__f2f( 0.298631336f); \
2148*89c4ff92SAndroid Build Coastguard Worker t1 = t1*stbi__f2f( 2.053119869f); \
2149*89c4ff92SAndroid Build Coastguard Worker t2 = t2*stbi__f2f( 3.072711026f); \
2150*89c4ff92SAndroid Build Coastguard Worker t3 = t3*stbi__f2f( 1.501321110f); \
2151*89c4ff92SAndroid Build Coastguard Worker p1 = p5 + p1*stbi__f2f(-0.899976223f); \
2152*89c4ff92SAndroid Build Coastguard Worker p2 = p5 + p2*stbi__f2f(-2.562915447f); \
2153*89c4ff92SAndroid Build Coastguard Worker p3 = p3*stbi__f2f(-1.961570560f); \
2154*89c4ff92SAndroid Build Coastguard Worker p4 = p4*stbi__f2f(-0.390180644f); \
2155*89c4ff92SAndroid Build Coastguard Worker t3 += p1+p4; \
2156*89c4ff92SAndroid Build Coastguard Worker t2 += p2+p3; \
2157*89c4ff92SAndroid Build Coastguard Worker t1 += p2+p4; \
2158*89c4ff92SAndroid Build Coastguard Worker t0 += p1+p3;
2159*89c4ff92SAndroid Build Coastguard Worker
stbi__idct_block(stbi_uc * out,int out_stride,short data[64])2160*89c4ff92SAndroid Build Coastguard Worker static void stbi__idct_block(stbi_uc *out, int out_stride, short data[64])
2161*89c4ff92SAndroid Build Coastguard Worker {
2162*89c4ff92SAndroid Build Coastguard Worker int i,val[64],*v=val;
2163*89c4ff92SAndroid Build Coastguard Worker stbi_uc *o;
2164*89c4ff92SAndroid Build Coastguard Worker short *d = data;
2165*89c4ff92SAndroid Build Coastguard Worker
2166*89c4ff92SAndroid Build Coastguard Worker // columns
2167*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < 8; ++i,++d, ++v) {
2168*89c4ff92SAndroid Build Coastguard Worker // if all zeroes, shortcut -- this avoids dequantizing 0s and IDCTing
2169*89c4ff92SAndroid Build Coastguard Worker if (d[ 8]==0 && d[16]==0 && d[24]==0 && d[32]==0
2170*89c4ff92SAndroid Build Coastguard Worker && d[40]==0 && d[48]==0 && d[56]==0) {
2171*89c4ff92SAndroid Build Coastguard Worker // no shortcut 0 seconds
2172*89c4ff92SAndroid Build Coastguard Worker // (1|2|3|4|5|6|7)==0 0 seconds
2173*89c4ff92SAndroid Build Coastguard Worker // all separate -0.047 seconds
2174*89c4ff92SAndroid Build Coastguard Worker // 1 && 2|3 && 4|5 && 6|7: -0.047 seconds
2175*89c4ff92SAndroid Build Coastguard Worker int dcterm = d[0] << 2;
2176*89c4ff92SAndroid Build Coastguard Worker v[0] = v[8] = v[16] = v[24] = v[32] = v[40] = v[48] = v[56] = dcterm;
2177*89c4ff92SAndroid Build Coastguard Worker } else {
2178*89c4ff92SAndroid Build Coastguard Worker STBI__IDCT_1D(d[ 0],d[ 8],d[16],d[24],d[32],d[40],d[48],d[56])
2179*89c4ff92SAndroid Build Coastguard Worker // constants scaled things up by 1<<12; let's bring them back
2180*89c4ff92SAndroid Build Coastguard Worker // down, but keep 2 extra bits of precision
2181*89c4ff92SAndroid Build Coastguard Worker x0 += 512; x1 += 512; x2 += 512; x3 += 512;
2182*89c4ff92SAndroid Build Coastguard Worker v[ 0] = (x0+t3) >> 10;
2183*89c4ff92SAndroid Build Coastguard Worker v[56] = (x0-t3) >> 10;
2184*89c4ff92SAndroid Build Coastguard Worker v[ 8] = (x1+t2) >> 10;
2185*89c4ff92SAndroid Build Coastguard Worker v[48] = (x1-t2) >> 10;
2186*89c4ff92SAndroid Build Coastguard Worker v[16] = (x2+t1) >> 10;
2187*89c4ff92SAndroid Build Coastguard Worker v[40] = (x2-t1) >> 10;
2188*89c4ff92SAndroid Build Coastguard Worker v[24] = (x3+t0) >> 10;
2189*89c4ff92SAndroid Build Coastguard Worker v[32] = (x3-t0) >> 10;
2190*89c4ff92SAndroid Build Coastguard Worker }
2191*89c4ff92SAndroid Build Coastguard Worker }
2192*89c4ff92SAndroid Build Coastguard Worker
2193*89c4ff92SAndroid Build Coastguard Worker for (i=0, v=val, o=out; i < 8; ++i,v+=8,o+=out_stride) {
2194*89c4ff92SAndroid Build Coastguard Worker // no fast case since the first 1D IDCT spread components out
2195*89c4ff92SAndroid Build Coastguard Worker STBI__IDCT_1D(v[0],v[1],v[2],v[3],v[4],v[5],v[6],v[7])
2196*89c4ff92SAndroid Build Coastguard Worker // constants scaled things up by 1<<12, plus we had 1<<2 from first
2197*89c4ff92SAndroid Build Coastguard Worker // loop, plus horizontal and vertical each scale by sqrt(8) so together
2198*89c4ff92SAndroid Build Coastguard Worker // we've got an extra 1<<3, so 1<<17 total we need to remove.
2199*89c4ff92SAndroid Build Coastguard Worker // so we want to round that, which means adding 0.5 * 1<<17,
2200*89c4ff92SAndroid Build Coastguard Worker // aka 65536. Also, we'll end up with -128 to 127 that we want
2201*89c4ff92SAndroid Build Coastguard Worker // to encode as 0..255 by adding 128, so we'll add that before the shift
2202*89c4ff92SAndroid Build Coastguard Worker x0 += 65536 + (128<<17);
2203*89c4ff92SAndroid Build Coastguard Worker x1 += 65536 + (128<<17);
2204*89c4ff92SAndroid Build Coastguard Worker x2 += 65536 + (128<<17);
2205*89c4ff92SAndroid Build Coastguard Worker x3 += 65536 + (128<<17);
2206*89c4ff92SAndroid Build Coastguard Worker // tried computing the shifts into temps, or'ing the temps to see
2207*89c4ff92SAndroid Build Coastguard Worker // if any were out of range, but that was slower
2208*89c4ff92SAndroid Build Coastguard Worker o[0] = stbi__clamp((x0+t3) >> 17);
2209*89c4ff92SAndroid Build Coastguard Worker o[7] = stbi__clamp((x0-t3) >> 17);
2210*89c4ff92SAndroid Build Coastguard Worker o[1] = stbi__clamp((x1+t2) >> 17);
2211*89c4ff92SAndroid Build Coastguard Worker o[6] = stbi__clamp((x1-t2) >> 17);
2212*89c4ff92SAndroid Build Coastguard Worker o[2] = stbi__clamp((x2+t1) >> 17);
2213*89c4ff92SAndroid Build Coastguard Worker o[5] = stbi__clamp((x2-t1) >> 17);
2214*89c4ff92SAndroid Build Coastguard Worker o[3] = stbi__clamp((x3+t0) >> 17);
2215*89c4ff92SAndroid Build Coastguard Worker o[4] = stbi__clamp((x3-t0) >> 17);
2216*89c4ff92SAndroid Build Coastguard Worker }
2217*89c4ff92SAndroid Build Coastguard Worker }
2218*89c4ff92SAndroid Build Coastguard Worker
2219*89c4ff92SAndroid Build Coastguard Worker #ifdef STBI_SSE2
2220*89c4ff92SAndroid Build Coastguard Worker // sse2 integer IDCT. not the fastest possible implementation but it
2221*89c4ff92SAndroid Build Coastguard Worker // produces bit-identical results to the generic C version so it's
2222*89c4ff92SAndroid Build Coastguard Worker // fully "transparent".
stbi__idct_simd(stbi_uc * out,int out_stride,short data[64])2223*89c4ff92SAndroid Build Coastguard Worker static void stbi__idct_simd(stbi_uc *out, int out_stride, short data[64])
2224*89c4ff92SAndroid Build Coastguard Worker {
2225*89c4ff92SAndroid Build Coastguard Worker // This is constructed to match our regular (generic) integer IDCT exactly.
2226*89c4ff92SAndroid Build Coastguard Worker __m128i row0, row1, row2, row3, row4, row5, row6, row7;
2227*89c4ff92SAndroid Build Coastguard Worker __m128i tmp;
2228*89c4ff92SAndroid Build Coastguard Worker
2229*89c4ff92SAndroid Build Coastguard Worker // dot product constant: even elems=x, odd elems=y
2230*89c4ff92SAndroid Build Coastguard Worker #define dct_const(x,y) _mm_setr_epi16((x),(y),(x),(y),(x),(y),(x),(y))
2231*89c4ff92SAndroid Build Coastguard Worker
2232*89c4ff92SAndroid Build Coastguard Worker // out(0) = c0[even]*x + c0[odd]*y (c0, x, y 16-bit, out 32-bit)
2233*89c4ff92SAndroid Build Coastguard Worker // out(1) = c1[even]*x + c1[odd]*y
2234*89c4ff92SAndroid Build Coastguard Worker #define dct_rot(out0,out1, x,y,c0,c1) \
2235*89c4ff92SAndroid Build Coastguard Worker __m128i c0##lo = _mm_unpacklo_epi16((x),(y)); \
2236*89c4ff92SAndroid Build Coastguard Worker __m128i c0##hi = _mm_unpackhi_epi16((x),(y)); \
2237*89c4ff92SAndroid Build Coastguard Worker __m128i out0##_l = _mm_madd_epi16(c0##lo, c0); \
2238*89c4ff92SAndroid Build Coastguard Worker __m128i out0##_h = _mm_madd_epi16(c0##hi, c0); \
2239*89c4ff92SAndroid Build Coastguard Worker __m128i out1##_l = _mm_madd_epi16(c0##lo, c1); \
2240*89c4ff92SAndroid Build Coastguard Worker __m128i out1##_h = _mm_madd_epi16(c0##hi, c1)
2241*89c4ff92SAndroid Build Coastguard Worker
2242*89c4ff92SAndroid Build Coastguard Worker // out = in << 12 (in 16-bit, out 32-bit)
2243*89c4ff92SAndroid Build Coastguard Worker #define dct_widen(out, in) \
2244*89c4ff92SAndroid Build Coastguard Worker __m128i out##_l = _mm_srai_epi32(_mm_unpacklo_epi16(_mm_setzero_si128(), (in)), 4); \
2245*89c4ff92SAndroid Build Coastguard Worker __m128i out##_h = _mm_srai_epi32(_mm_unpackhi_epi16(_mm_setzero_si128(), (in)), 4)
2246*89c4ff92SAndroid Build Coastguard Worker
2247*89c4ff92SAndroid Build Coastguard Worker // wide add
2248*89c4ff92SAndroid Build Coastguard Worker #define dct_wadd(out, a, b) \
2249*89c4ff92SAndroid Build Coastguard Worker __m128i out##_l = _mm_add_epi32(a##_l, b##_l); \
2250*89c4ff92SAndroid Build Coastguard Worker __m128i out##_h = _mm_add_epi32(a##_h, b##_h)
2251*89c4ff92SAndroid Build Coastguard Worker
2252*89c4ff92SAndroid Build Coastguard Worker // wide sub
2253*89c4ff92SAndroid Build Coastguard Worker #define dct_wsub(out, a, b) \
2254*89c4ff92SAndroid Build Coastguard Worker __m128i out##_l = _mm_sub_epi32(a##_l, b##_l); \
2255*89c4ff92SAndroid Build Coastguard Worker __m128i out##_h = _mm_sub_epi32(a##_h, b##_h)
2256*89c4ff92SAndroid Build Coastguard Worker
2257*89c4ff92SAndroid Build Coastguard Worker // butterfly a/b, add bias, then shift by "s" and pack
2258*89c4ff92SAndroid Build Coastguard Worker #define dct_bfly32o(out0, out1, a,b,bias,s) \
2259*89c4ff92SAndroid Build Coastguard Worker { \
2260*89c4ff92SAndroid Build Coastguard Worker __m128i abiased_l = _mm_add_epi32(a##_l, bias); \
2261*89c4ff92SAndroid Build Coastguard Worker __m128i abiased_h = _mm_add_epi32(a##_h, bias); \
2262*89c4ff92SAndroid Build Coastguard Worker dct_wadd(sum, abiased, b); \
2263*89c4ff92SAndroid Build Coastguard Worker dct_wsub(dif, abiased, b); \
2264*89c4ff92SAndroid Build Coastguard Worker out0 = _mm_packs_epi32(_mm_srai_epi32(sum_l, s), _mm_srai_epi32(sum_h, s)); \
2265*89c4ff92SAndroid Build Coastguard Worker out1 = _mm_packs_epi32(_mm_srai_epi32(dif_l, s), _mm_srai_epi32(dif_h, s)); \
2266*89c4ff92SAndroid Build Coastguard Worker }
2267*89c4ff92SAndroid Build Coastguard Worker
2268*89c4ff92SAndroid Build Coastguard Worker // 8-bit interleave step (for transposes)
2269*89c4ff92SAndroid Build Coastguard Worker #define dct_interleave8(a, b) \
2270*89c4ff92SAndroid Build Coastguard Worker tmp = a; \
2271*89c4ff92SAndroid Build Coastguard Worker a = _mm_unpacklo_epi8(a, b); \
2272*89c4ff92SAndroid Build Coastguard Worker b = _mm_unpackhi_epi8(tmp, b)
2273*89c4ff92SAndroid Build Coastguard Worker
2274*89c4ff92SAndroid Build Coastguard Worker // 16-bit interleave step (for transposes)
2275*89c4ff92SAndroid Build Coastguard Worker #define dct_interleave16(a, b) \
2276*89c4ff92SAndroid Build Coastguard Worker tmp = a; \
2277*89c4ff92SAndroid Build Coastguard Worker a = _mm_unpacklo_epi16(a, b); \
2278*89c4ff92SAndroid Build Coastguard Worker b = _mm_unpackhi_epi16(tmp, b)
2279*89c4ff92SAndroid Build Coastguard Worker
2280*89c4ff92SAndroid Build Coastguard Worker #define dct_pass(bias,shift) \
2281*89c4ff92SAndroid Build Coastguard Worker { \
2282*89c4ff92SAndroid Build Coastguard Worker /* even part */ \
2283*89c4ff92SAndroid Build Coastguard Worker dct_rot(t2e,t3e, row2,row6, rot0_0,rot0_1); \
2284*89c4ff92SAndroid Build Coastguard Worker __m128i sum04 = _mm_add_epi16(row0, row4); \
2285*89c4ff92SAndroid Build Coastguard Worker __m128i dif04 = _mm_sub_epi16(row0, row4); \
2286*89c4ff92SAndroid Build Coastguard Worker dct_widen(t0e, sum04); \
2287*89c4ff92SAndroid Build Coastguard Worker dct_widen(t1e, dif04); \
2288*89c4ff92SAndroid Build Coastguard Worker dct_wadd(x0, t0e, t3e); \
2289*89c4ff92SAndroid Build Coastguard Worker dct_wsub(x3, t0e, t3e); \
2290*89c4ff92SAndroid Build Coastguard Worker dct_wadd(x1, t1e, t2e); \
2291*89c4ff92SAndroid Build Coastguard Worker dct_wsub(x2, t1e, t2e); \
2292*89c4ff92SAndroid Build Coastguard Worker /* odd part */ \
2293*89c4ff92SAndroid Build Coastguard Worker dct_rot(y0o,y2o, row7,row3, rot2_0,rot2_1); \
2294*89c4ff92SAndroid Build Coastguard Worker dct_rot(y1o,y3o, row5,row1, rot3_0,rot3_1); \
2295*89c4ff92SAndroid Build Coastguard Worker __m128i sum17 = _mm_add_epi16(row1, row7); \
2296*89c4ff92SAndroid Build Coastguard Worker __m128i sum35 = _mm_add_epi16(row3, row5); \
2297*89c4ff92SAndroid Build Coastguard Worker dct_rot(y4o,y5o, sum17,sum35, rot1_0,rot1_1); \
2298*89c4ff92SAndroid Build Coastguard Worker dct_wadd(x4, y0o, y4o); \
2299*89c4ff92SAndroid Build Coastguard Worker dct_wadd(x5, y1o, y5o); \
2300*89c4ff92SAndroid Build Coastguard Worker dct_wadd(x6, y2o, y5o); \
2301*89c4ff92SAndroid Build Coastguard Worker dct_wadd(x7, y3o, y4o); \
2302*89c4ff92SAndroid Build Coastguard Worker dct_bfly32o(row0,row7, x0,x7,bias,shift); \
2303*89c4ff92SAndroid Build Coastguard Worker dct_bfly32o(row1,row6, x1,x6,bias,shift); \
2304*89c4ff92SAndroid Build Coastguard Worker dct_bfly32o(row2,row5, x2,x5,bias,shift); \
2305*89c4ff92SAndroid Build Coastguard Worker dct_bfly32o(row3,row4, x3,x4,bias,shift); \
2306*89c4ff92SAndroid Build Coastguard Worker }
2307*89c4ff92SAndroid Build Coastguard Worker
2308*89c4ff92SAndroid Build Coastguard Worker __m128i rot0_0 = dct_const(stbi__f2f(0.5411961f), stbi__f2f(0.5411961f) + stbi__f2f(-1.847759065f));
2309*89c4ff92SAndroid Build Coastguard Worker __m128i rot0_1 = dct_const(stbi__f2f(0.5411961f) + stbi__f2f( 0.765366865f), stbi__f2f(0.5411961f));
2310*89c4ff92SAndroid Build Coastguard Worker __m128i rot1_0 = dct_const(stbi__f2f(1.175875602f) + stbi__f2f(-0.899976223f), stbi__f2f(1.175875602f));
2311*89c4ff92SAndroid Build Coastguard Worker __m128i rot1_1 = dct_const(stbi__f2f(1.175875602f), stbi__f2f(1.175875602f) + stbi__f2f(-2.562915447f));
2312*89c4ff92SAndroid Build Coastguard Worker __m128i rot2_0 = dct_const(stbi__f2f(-1.961570560f) + stbi__f2f( 0.298631336f), stbi__f2f(-1.961570560f));
2313*89c4ff92SAndroid Build Coastguard Worker __m128i rot2_1 = dct_const(stbi__f2f(-1.961570560f), stbi__f2f(-1.961570560f) + stbi__f2f( 3.072711026f));
2314*89c4ff92SAndroid Build Coastguard Worker __m128i rot3_0 = dct_const(stbi__f2f(-0.390180644f) + stbi__f2f( 2.053119869f), stbi__f2f(-0.390180644f));
2315*89c4ff92SAndroid Build Coastguard Worker __m128i rot3_1 = dct_const(stbi__f2f(-0.390180644f), stbi__f2f(-0.390180644f) + stbi__f2f( 1.501321110f));
2316*89c4ff92SAndroid Build Coastguard Worker
2317*89c4ff92SAndroid Build Coastguard Worker // rounding biases in column/row passes, see stbi__idct_block for explanation.
2318*89c4ff92SAndroid Build Coastguard Worker __m128i bias_0 = _mm_set1_epi32(512);
2319*89c4ff92SAndroid Build Coastguard Worker __m128i bias_1 = _mm_set1_epi32(65536 + (128<<17));
2320*89c4ff92SAndroid Build Coastguard Worker
2321*89c4ff92SAndroid Build Coastguard Worker // load
2322*89c4ff92SAndroid Build Coastguard Worker row0 = _mm_load_si128((const __m128i *) (data + 0*8));
2323*89c4ff92SAndroid Build Coastguard Worker row1 = _mm_load_si128((const __m128i *) (data + 1*8));
2324*89c4ff92SAndroid Build Coastguard Worker row2 = _mm_load_si128((const __m128i *) (data + 2*8));
2325*89c4ff92SAndroid Build Coastguard Worker row3 = _mm_load_si128((const __m128i *) (data + 3*8));
2326*89c4ff92SAndroid Build Coastguard Worker row4 = _mm_load_si128((const __m128i *) (data + 4*8));
2327*89c4ff92SAndroid Build Coastguard Worker row5 = _mm_load_si128((const __m128i *) (data + 5*8));
2328*89c4ff92SAndroid Build Coastguard Worker row6 = _mm_load_si128((const __m128i *) (data + 6*8));
2329*89c4ff92SAndroid Build Coastguard Worker row7 = _mm_load_si128((const __m128i *) (data + 7*8));
2330*89c4ff92SAndroid Build Coastguard Worker
2331*89c4ff92SAndroid Build Coastguard Worker // column pass
2332*89c4ff92SAndroid Build Coastguard Worker dct_pass(bias_0, 10);
2333*89c4ff92SAndroid Build Coastguard Worker
2334*89c4ff92SAndroid Build Coastguard Worker {
2335*89c4ff92SAndroid Build Coastguard Worker // 16bit 8x8 transpose pass 1
2336*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row0, row4);
2337*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row1, row5);
2338*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row2, row6);
2339*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row3, row7);
2340*89c4ff92SAndroid Build Coastguard Worker
2341*89c4ff92SAndroid Build Coastguard Worker // transpose pass 2
2342*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row0, row2);
2343*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row1, row3);
2344*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row4, row6);
2345*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row5, row7);
2346*89c4ff92SAndroid Build Coastguard Worker
2347*89c4ff92SAndroid Build Coastguard Worker // transpose pass 3
2348*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row0, row1);
2349*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row2, row3);
2350*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row4, row5);
2351*89c4ff92SAndroid Build Coastguard Worker dct_interleave16(row6, row7);
2352*89c4ff92SAndroid Build Coastguard Worker }
2353*89c4ff92SAndroid Build Coastguard Worker
2354*89c4ff92SAndroid Build Coastguard Worker // row pass
2355*89c4ff92SAndroid Build Coastguard Worker dct_pass(bias_1, 17);
2356*89c4ff92SAndroid Build Coastguard Worker
2357*89c4ff92SAndroid Build Coastguard Worker {
2358*89c4ff92SAndroid Build Coastguard Worker // pack
2359*89c4ff92SAndroid Build Coastguard Worker __m128i p0 = _mm_packus_epi16(row0, row1); // a0a1a2a3...a7b0b1b2b3...b7
2360*89c4ff92SAndroid Build Coastguard Worker __m128i p1 = _mm_packus_epi16(row2, row3);
2361*89c4ff92SAndroid Build Coastguard Worker __m128i p2 = _mm_packus_epi16(row4, row5);
2362*89c4ff92SAndroid Build Coastguard Worker __m128i p3 = _mm_packus_epi16(row6, row7);
2363*89c4ff92SAndroid Build Coastguard Worker
2364*89c4ff92SAndroid Build Coastguard Worker // 8bit 8x8 transpose pass 1
2365*89c4ff92SAndroid Build Coastguard Worker dct_interleave8(p0, p2); // a0e0a1e1...
2366*89c4ff92SAndroid Build Coastguard Worker dct_interleave8(p1, p3); // c0g0c1g1...
2367*89c4ff92SAndroid Build Coastguard Worker
2368*89c4ff92SAndroid Build Coastguard Worker // transpose pass 2
2369*89c4ff92SAndroid Build Coastguard Worker dct_interleave8(p0, p1); // a0c0e0g0...
2370*89c4ff92SAndroid Build Coastguard Worker dct_interleave8(p2, p3); // b0d0f0h0...
2371*89c4ff92SAndroid Build Coastguard Worker
2372*89c4ff92SAndroid Build Coastguard Worker // transpose pass 3
2373*89c4ff92SAndroid Build Coastguard Worker dct_interleave8(p0, p2); // a0b0c0d0...
2374*89c4ff92SAndroid Build Coastguard Worker dct_interleave8(p1, p3); // a4b4c4d4...
2375*89c4ff92SAndroid Build Coastguard Worker
2376*89c4ff92SAndroid Build Coastguard Worker // store
2377*89c4ff92SAndroid Build Coastguard Worker _mm_storel_epi64((__m128i *) out, p0); out += out_stride;
2378*89c4ff92SAndroid Build Coastguard Worker _mm_storel_epi64((__m128i *) out, _mm_shuffle_epi32(p0, 0x4e)); out += out_stride;
2379*89c4ff92SAndroid Build Coastguard Worker _mm_storel_epi64((__m128i *) out, p2); out += out_stride;
2380*89c4ff92SAndroid Build Coastguard Worker _mm_storel_epi64((__m128i *) out, _mm_shuffle_epi32(p2, 0x4e)); out += out_stride;
2381*89c4ff92SAndroid Build Coastguard Worker _mm_storel_epi64((__m128i *) out, p1); out += out_stride;
2382*89c4ff92SAndroid Build Coastguard Worker _mm_storel_epi64((__m128i *) out, _mm_shuffle_epi32(p1, 0x4e)); out += out_stride;
2383*89c4ff92SAndroid Build Coastguard Worker _mm_storel_epi64((__m128i *) out, p3); out += out_stride;
2384*89c4ff92SAndroid Build Coastguard Worker _mm_storel_epi64((__m128i *) out, _mm_shuffle_epi32(p3, 0x4e));
2385*89c4ff92SAndroid Build Coastguard Worker }
2386*89c4ff92SAndroid Build Coastguard Worker
2387*89c4ff92SAndroid Build Coastguard Worker #undef dct_const
2388*89c4ff92SAndroid Build Coastguard Worker #undef dct_rot
2389*89c4ff92SAndroid Build Coastguard Worker #undef dct_widen
2390*89c4ff92SAndroid Build Coastguard Worker #undef dct_wadd
2391*89c4ff92SAndroid Build Coastguard Worker #undef dct_wsub
2392*89c4ff92SAndroid Build Coastguard Worker #undef dct_bfly32o
2393*89c4ff92SAndroid Build Coastguard Worker #undef dct_interleave8
2394*89c4ff92SAndroid Build Coastguard Worker #undef dct_interleave16
2395*89c4ff92SAndroid Build Coastguard Worker #undef dct_pass
2396*89c4ff92SAndroid Build Coastguard Worker }
2397*89c4ff92SAndroid Build Coastguard Worker
2398*89c4ff92SAndroid Build Coastguard Worker #endif // STBI_SSE2
2399*89c4ff92SAndroid Build Coastguard Worker
2400*89c4ff92SAndroid Build Coastguard Worker #ifdef STBI_NEON
2401*89c4ff92SAndroid Build Coastguard Worker
2402*89c4ff92SAndroid Build Coastguard Worker // NEON integer IDCT. should produce bit-identical
2403*89c4ff92SAndroid Build Coastguard Worker // results to the generic C version.
stbi__idct_simd(stbi_uc * out,int out_stride,short data[64])2404*89c4ff92SAndroid Build Coastguard Worker static void stbi__idct_simd(stbi_uc *out, int out_stride, short data[64])
2405*89c4ff92SAndroid Build Coastguard Worker {
2406*89c4ff92SAndroid Build Coastguard Worker int16x8_t row0, row1, row2, row3, row4, row5, row6, row7;
2407*89c4ff92SAndroid Build Coastguard Worker
2408*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot0_0 = vdup_n_s16(stbi__f2f(0.5411961f));
2409*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot0_1 = vdup_n_s16(stbi__f2f(-1.847759065f));
2410*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot0_2 = vdup_n_s16(stbi__f2f( 0.765366865f));
2411*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot1_0 = vdup_n_s16(stbi__f2f( 1.175875602f));
2412*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot1_1 = vdup_n_s16(stbi__f2f(-0.899976223f));
2413*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot1_2 = vdup_n_s16(stbi__f2f(-2.562915447f));
2414*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot2_0 = vdup_n_s16(stbi__f2f(-1.961570560f));
2415*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot2_1 = vdup_n_s16(stbi__f2f(-0.390180644f));
2416*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot3_0 = vdup_n_s16(stbi__f2f( 0.298631336f));
2417*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot3_1 = vdup_n_s16(stbi__f2f( 2.053119869f));
2418*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot3_2 = vdup_n_s16(stbi__f2f( 3.072711026f));
2419*89c4ff92SAndroid Build Coastguard Worker int16x4_t rot3_3 = vdup_n_s16(stbi__f2f( 1.501321110f));
2420*89c4ff92SAndroid Build Coastguard Worker
2421*89c4ff92SAndroid Build Coastguard Worker #define dct_long_mul(out, inq, coeff) \
2422*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_l = vmull_s16(vget_low_s16(inq), coeff); \
2423*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_h = vmull_s16(vget_high_s16(inq), coeff)
2424*89c4ff92SAndroid Build Coastguard Worker
2425*89c4ff92SAndroid Build Coastguard Worker #define dct_long_mac(out, acc, inq, coeff) \
2426*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_l = vmlal_s16(acc##_l, vget_low_s16(inq), coeff); \
2427*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_h = vmlal_s16(acc##_h, vget_high_s16(inq), coeff)
2428*89c4ff92SAndroid Build Coastguard Worker
2429*89c4ff92SAndroid Build Coastguard Worker #define dct_widen(out, inq) \
2430*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_l = vshll_n_s16(vget_low_s16(inq), 12); \
2431*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_h = vshll_n_s16(vget_high_s16(inq), 12)
2432*89c4ff92SAndroid Build Coastguard Worker
2433*89c4ff92SAndroid Build Coastguard Worker // wide add
2434*89c4ff92SAndroid Build Coastguard Worker #define dct_wadd(out, a, b) \
2435*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_l = vaddq_s32(a##_l, b##_l); \
2436*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_h = vaddq_s32(a##_h, b##_h)
2437*89c4ff92SAndroid Build Coastguard Worker
2438*89c4ff92SAndroid Build Coastguard Worker // wide sub
2439*89c4ff92SAndroid Build Coastguard Worker #define dct_wsub(out, a, b) \
2440*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_l = vsubq_s32(a##_l, b##_l); \
2441*89c4ff92SAndroid Build Coastguard Worker int32x4_t out##_h = vsubq_s32(a##_h, b##_h)
2442*89c4ff92SAndroid Build Coastguard Worker
2443*89c4ff92SAndroid Build Coastguard Worker // butterfly a/b, then shift using "shiftop" by "s" and pack
2444*89c4ff92SAndroid Build Coastguard Worker #define dct_bfly32o(out0,out1, a,b,shiftop,s) \
2445*89c4ff92SAndroid Build Coastguard Worker { \
2446*89c4ff92SAndroid Build Coastguard Worker dct_wadd(sum, a, b); \
2447*89c4ff92SAndroid Build Coastguard Worker dct_wsub(dif, a, b); \
2448*89c4ff92SAndroid Build Coastguard Worker out0 = vcombine_s16(shiftop(sum_l, s), shiftop(sum_h, s)); \
2449*89c4ff92SAndroid Build Coastguard Worker out1 = vcombine_s16(shiftop(dif_l, s), shiftop(dif_h, s)); \
2450*89c4ff92SAndroid Build Coastguard Worker }
2451*89c4ff92SAndroid Build Coastguard Worker
2452*89c4ff92SAndroid Build Coastguard Worker #define dct_pass(shiftop, shift) \
2453*89c4ff92SAndroid Build Coastguard Worker { \
2454*89c4ff92SAndroid Build Coastguard Worker /* even part */ \
2455*89c4ff92SAndroid Build Coastguard Worker int16x8_t sum26 = vaddq_s16(row2, row6); \
2456*89c4ff92SAndroid Build Coastguard Worker dct_long_mul(p1e, sum26, rot0_0); \
2457*89c4ff92SAndroid Build Coastguard Worker dct_long_mac(t2e, p1e, row6, rot0_1); \
2458*89c4ff92SAndroid Build Coastguard Worker dct_long_mac(t3e, p1e, row2, rot0_2); \
2459*89c4ff92SAndroid Build Coastguard Worker int16x8_t sum04 = vaddq_s16(row0, row4); \
2460*89c4ff92SAndroid Build Coastguard Worker int16x8_t dif04 = vsubq_s16(row0, row4); \
2461*89c4ff92SAndroid Build Coastguard Worker dct_widen(t0e, sum04); \
2462*89c4ff92SAndroid Build Coastguard Worker dct_widen(t1e, dif04); \
2463*89c4ff92SAndroid Build Coastguard Worker dct_wadd(x0, t0e, t3e); \
2464*89c4ff92SAndroid Build Coastguard Worker dct_wsub(x3, t0e, t3e); \
2465*89c4ff92SAndroid Build Coastguard Worker dct_wadd(x1, t1e, t2e); \
2466*89c4ff92SAndroid Build Coastguard Worker dct_wsub(x2, t1e, t2e); \
2467*89c4ff92SAndroid Build Coastguard Worker /* odd part */ \
2468*89c4ff92SAndroid Build Coastguard Worker int16x8_t sum15 = vaddq_s16(row1, row5); \
2469*89c4ff92SAndroid Build Coastguard Worker int16x8_t sum17 = vaddq_s16(row1, row7); \
2470*89c4ff92SAndroid Build Coastguard Worker int16x8_t sum35 = vaddq_s16(row3, row5); \
2471*89c4ff92SAndroid Build Coastguard Worker int16x8_t sum37 = vaddq_s16(row3, row7); \
2472*89c4ff92SAndroid Build Coastguard Worker int16x8_t sumodd = vaddq_s16(sum17, sum35); \
2473*89c4ff92SAndroid Build Coastguard Worker dct_long_mul(p5o, sumodd, rot1_0); \
2474*89c4ff92SAndroid Build Coastguard Worker dct_long_mac(p1o, p5o, sum17, rot1_1); \
2475*89c4ff92SAndroid Build Coastguard Worker dct_long_mac(p2o, p5o, sum35, rot1_2); \
2476*89c4ff92SAndroid Build Coastguard Worker dct_long_mul(p3o, sum37, rot2_0); \
2477*89c4ff92SAndroid Build Coastguard Worker dct_long_mul(p4o, sum15, rot2_1); \
2478*89c4ff92SAndroid Build Coastguard Worker dct_wadd(sump13o, p1o, p3o); \
2479*89c4ff92SAndroid Build Coastguard Worker dct_wadd(sump24o, p2o, p4o); \
2480*89c4ff92SAndroid Build Coastguard Worker dct_wadd(sump23o, p2o, p3o); \
2481*89c4ff92SAndroid Build Coastguard Worker dct_wadd(sump14o, p1o, p4o); \
2482*89c4ff92SAndroid Build Coastguard Worker dct_long_mac(x4, sump13o, row7, rot3_0); \
2483*89c4ff92SAndroid Build Coastguard Worker dct_long_mac(x5, sump24o, row5, rot3_1); \
2484*89c4ff92SAndroid Build Coastguard Worker dct_long_mac(x6, sump23o, row3, rot3_2); \
2485*89c4ff92SAndroid Build Coastguard Worker dct_long_mac(x7, sump14o, row1, rot3_3); \
2486*89c4ff92SAndroid Build Coastguard Worker dct_bfly32o(row0,row7, x0,x7,shiftop,shift); \
2487*89c4ff92SAndroid Build Coastguard Worker dct_bfly32o(row1,row6, x1,x6,shiftop,shift); \
2488*89c4ff92SAndroid Build Coastguard Worker dct_bfly32o(row2,row5, x2,x5,shiftop,shift); \
2489*89c4ff92SAndroid Build Coastguard Worker dct_bfly32o(row3,row4, x3,x4,shiftop,shift); \
2490*89c4ff92SAndroid Build Coastguard Worker }
2491*89c4ff92SAndroid Build Coastguard Worker
2492*89c4ff92SAndroid Build Coastguard Worker // load
2493*89c4ff92SAndroid Build Coastguard Worker row0 = vld1q_s16(data + 0*8);
2494*89c4ff92SAndroid Build Coastguard Worker row1 = vld1q_s16(data + 1*8);
2495*89c4ff92SAndroid Build Coastguard Worker row2 = vld1q_s16(data + 2*8);
2496*89c4ff92SAndroid Build Coastguard Worker row3 = vld1q_s16(data + 3*8);
2497*89c4ff92SAndroid Build Coastguard Worker row4 = vld1q_s16(data + 4*8);
2498*89c4ff92SAndroid Build Coastguard Worker row5 = vld1q_s16(data + 5*8);
2499*89c4ff92SAndroid Build Coastguard Worker row6 = vld1q_s16(data + 6*8);
2500*89c4ff92SAndroid Build Coastguard Worker row7 = vld1q_s16(data + 7*8);
2501*89c4ff92SAndroid Build Coastguard Worker
2502*89c4ff92SAndroid Build Coastguard Worker // add DC bias
2503*89c4ff92SAndroid Build Coastguard Worker row0 = vaddq_s16(row0, vsetq_lane_s16(1024, vdupq_n_s16(0), 0));
2504*89c4ff92SAndroid Build Coastguard Worker
2505*89c4ff92SAndroid Build Coastguard Worker // column pass
2506*89c4ff92SAndroid Build Coastguard Worker dct_pass(vrshrn_n_s32, 10);
2507*89c4ff92SAndroid Build Coastguard Worker
2508*89c4ff92SAndroid Build Coastguard Worker // 16bit 8x8 transpose
2509*89c4ff92SAndroid Build Coastguard Worker {
2510*89c4ff92SAndroid Build Coastguard Worker // these three map to a single VTRN.16, VTRN.32, and VSWP, respectively.
2511*89c4ff92SAndroid Build Coastguard Worker // whether compilers actually get this is another story, sadly.
2512*89c4ff92SAndroid Build Coastguard Worker #define dct_trn16(x, y) { int16x8x2_t t = vtrnq_s16(x, y); x = t.val[0]; y = t.val[1]; }
2513*89c4ff92SAndroid Build Coastguard Worker #define dct_trn32(x, y) { int32x4x2_t t = vtrnq_s32(vreinterpretq_s32_s16(x), vreinterpretq_s32_s16(y)); x = vreinterpretq_s16_s32(t.val[0]); y = vreinterpretq_s16_s32(t.val[1]); }
2514*89c4ff92SAndroid Build Coastguard Worker #define dct_trn64(x, y) { int16x8_t x0 = x; int16x8_t y0 = y; x = vcombine_s16(vget_low_s16(x0), vget_low_s16(y0)); y = vcombine_s16(vget_high_s16(x0), vget_high_s16(y0)); }
2515*89c4ff92SAndroid Build Coastguard Worker
2516*89c4ff92SAndroid Build Coastguard Worker // pass 1
2517*89c4ff92SAndroid Build Coastguard Worker dct_trn16(row0, row1); // a0b0a2b2a4b4a6b6
2518*89c4ff92SAndroid Build Coastguard Worker dct_trn16(row2, row3);
2519*89c4ff92SAndroid Build Coastguard Worker dct_trn16(row4, row5);
2520*89c4ff92SAndroid Build Coastguard Worker dct_trn16(row6, row7);
2521*89c4ff92SAndroid Build Coastguard Worker
2522*89c4ff92SAndroid Build Coastguard Worker // pass 2
2523*89c4ff92SAndroid Build Coastguard Worker dct_trn32(row0, row2); // a0b0c0d0a4b4c4d4
2524*89c4ff92SAndroid Build Coastguard Worker dct_trn32(row1, row3);
2525*89c4ff92SAndroid Build Coastguard Worker dct_trn32(row4, row6);
2526*89c4ff92SAndroid Build Coastguard Worker dct_trn32(row5, row7);
2527*89c4ff92SAndroid Build Coastguard Worker
2528*89c4ff92SAndroid Build Coastguard Worker // pass 3
2529*89c4ff92SAndroid Build Coastguard Worker dct_trn64(row0, row4); // a0b0c0d0e0f0g0h0
2530*89c4ff92SAndroid Build Coastguard Worker dct_trn64(row1, row5);
2531*89c4ff92SAndroid Build Coastguard Worker dct_trn64(row2, row6);
2532*89c4ff92SAndroid Build Coastguard Worker dct_trn64(row3, row7);
2533*89c4ff92SAndroid Build Coastguard Worker
2534*89c4ff92SAndroid Build Coastguard Worker #undef dct_trn16
2535*89c4ff92SAndroid Build Coastguard Worker #undef dct_trn32
2536*89c4ff92SAndroid Build Coastguard Worker #undef dct_trn64
2537*89c4ff92SAndroid Build Coastguard Worker }
2538*89c4ff92SAndroid Build Coastguard Worker
2539*89c4ff92SAndroid Build Coastguard Worker // row pass
2540*89c4ff92SAndroid Build Coastguard Worker // vrshrn_n_s32 only supports shifts up to 16, we need
2541*89c4ff92SAndroid Build Coastguard Worker // 17. so do a non-rounding shift of 16 first then follow
2542*89c4ff92SAndroid Build Coastguard Worker // up with a rounding shift by 1.
2543*89c4ff92SAndroid Build Coastguard Worker dct_pass(vshrn_n_s32, 16);
2544*89c4ff92SAndroid Build Coastguard Worker
2545*89c4ff92SAndroid Build Coastguard Worker {
2546*89c4ff92SAndroid Build Coastguard Worker // pack and round
2547*89c4ff92SAndroid Build Coastguard Worker uint8x8_t p0 = vqrshrun_n_s16(row0, 1);
2548*89c4ff92SAndroid Build Coastguard Worker uint8x8_t p1 = vqrshrun_n_s16(row1, 1);
2549*89c4ff92SAndroid Build Coastguard Worker uint8x8_t p2 = vqrshrun_n_s16(row2, 1);
2550*89c4ff92SAndroid Build Coastguard Worker uint8x8_t p3 = vqrshrun_n_s16(row3, 1);
2551*89c4ff92SAndroid Build Coastguard Worker uint8x8_t p4 = vqrshrun_n_s16(row4, 1);
2552*89c4ff92SAndroid Build Coastguard Worker uint8x8_t p5 = vqrshrun_n_s16(row5, 1);
2553*89c4ff92SAndroid Build Coastguard Worker uint8x8_t p6 = vqrshrun_n_s16(row6, 1);
2554*89c4ff92SAndroid Build Coastguard Worker uint8x8_t p7 = vqrshrun_n_s16(row7, 1);
2555*89c4ff92SAndroid Build Coastguard Worker
2556*89c4ff92SAndroid Build Coastguard Worker // again, these can translate into one instruction, but often don't.
2557*89c4ff92SAndroid Build Coastguard Worker #define dct_trn8_8(x, y) { uint8x8x2_t t = vtrn_u8(x, y); x = t.val[0]; y = t.val[1]; }
2558*89c4ff92SAndroid Build Coastguard Worker #define dct_trn8_16(x, y) { uint16x4x2_t t = vtrn_u16(vreinterpret_u16_u8(x), vreinterpret_u16_u8(y)); x = vreinterpret_u8_u16(t.val[0]); y = vreinterpret_u8_u16(t.val[1]); }
2559*89c4ff92SAndroid Build Coastguard Worker #define dct_trn8_32(x, y) { uint32x2x2_t t = vtrn_u32(vreinterpret_u32_u8(x), vreinterpret_u32_u8(y)); x = vreinterpret_u8_u32(t.val[0]); y = vreinterpret_u8_u32(t.val[1]); }
2560*89c4ff92SAndroid Build Coastguard Worker
2561*89c4ff92SAndroid Build Coastguard Worker // sadly can't use interleaved stores here since we only write
2562*89c4ff92SAndroid Build Coastguard Worker // 8 bytes to each scan line!
2563*89c4ff92SAndroid Build Coastguard Worker
2564*89c4ff92SAndroid Build Coastguard Worker // 8x8 8-bit transpose pass 1
2565*89c4ff92SAndroid Build Coastguard Worker dct_trn8_8(p0, p1);
2566*89c4ff92SAndroid Build Coastguard Worker dct_trn8_8(p2, p3);
2567*89c4ff92SAndroid Build Coastguard Worker dct_trn8_8(p4, p5);
2568*89c4ff92SAndroid Build Coastguard Worker dct_trn8_8(p6, p7);
2569*89c4ff92SAndroid Build Coastguard Worker
2570*89c4ff92SAndroid Build Coastguard Worker // pass 2
2571*89c4ff92SAndroid Build Coastguard Worker dct_trn8_16(p0, p2);
2572*89c4ff92SAndroid Build Coastguard Worker dct_trn8_16(p1, p3);
2573*89c4ff92SAndroid Build Coastguard Worker dct_trn8_16(p4, p6);
2574*89c4ff92SAndroid Build Coastguard Worker dct_trn8_16(p5, p7);
2575*89c4ff92SAndroid Build Coastguard Worker
2576*89c4ff92SAndroid Build Coastguard Worker // pass 3
2577*89c4ff92SAndroid Build Coastguard Worker dct_trn8_32(p0, p4);
2578*89c4ff92SAndroid Build Coastguard Worker dct_trn8_32(p1, p5);
2579*89c4ff92SAndroid Build Coastguard Worker dct_trn8_32(p2, p6);
2580*89c4ff92SAndroid Build Coastguard Worker dct_trn8_32(p3, p7);
2581*89c4ff92SAndroid Build Coastguard Worker
2582*89c4ff92SAndroid Build Coastguard Worker // store
2583*89c4ff92SAndroid Build Coastguard Worker vst1_u8(out, p0); out += out_stride;
2584*89c4ff92SAndroid Build Coastguard Worker vst1_u8(out, p1); out += out_stride;
2585*89c4ff92SAndroid Build Coastguard Worker vst1_u8(out, p2); out += out_stride;
2586*89c4ff92SAndroid Build Coastguard Worker vst1_u8(out, p3); out += out_stride;
2587*89c4ff92SAndroid Build Coastguard Worker vst1_u8(out, p4); out += out_stride;
2588*89c4ff92SAndroid Build Coastguard Worker vst1_u8(out, p5); out += out_stride;
2589*89c4ff92SAndroid Build Coastguard Worker vst1_u8(out, p6); out += out_stride;
2590*89c4ff92SAndroid Build Coastguard Worker vst1_u8(out, p7);
2591*89c4ff92SAndroid Build Coastguard Worker
2592*89c4ff92SAndroid Build Coastguard Worker #undef dct_trn8_8
2593*89c4ff92SAndroid Build Coastguard Worker #undef dct_trn8_16
2594*89c4ff92SAndroid Build Coastguard Worker #undef dct_trn8_32
2595*89c4ff92SAndroid Build Coastguard Worker }
2596*89c4ff92SAndroid Build Coastguard Worker
2597*89c4ff92SAndroid Build Coastguard Worker #undef dct_long_mul
2598*89c4ff92SAndroid Build Coastguard Worker #undef dct_long_mac
2599*89c4ff92SAndroid Build Coastguard Worker #undef dct_widen
2600*89c4ff92SAndroid Build Coastguard Worker #undef dct_wadd
2601*89c4ff92SAndroid Build Coastguard Worker #undef dct_wsub
2602*89c4ff92SAndroid Build Coastguard Worker #undef dct_bfly32o
2603*89c4ff92SAndroid Build Coastguard Worker #undef dct_pass
2604*89c4ff92SAndroid Build Coastguard Worker }
2605*89c4ff92SAndroid Build Coastguard Worker
2606*89c4ff92SAndroid Build Coastguard Worker #endif // STBI_NEON
2607*89c4ff92SAndroid Build Coastguard Worker
2608*89c4ff92SAndroid Build Coastguard Worker #define STBI__MARKER_none 0xff
2609*89c4ff92SAndroid Build Coastguard Worker // if there's a pending marker from the entropy stream, return that
2610*89c4ff92SAndroid Build Coastguard Worker // otherwise, fetch from the stream and get a marker. if there's no
2611*89c4ff92SAndroid Build Coastguard Worker // marker, return 0xff, which is never a valid marker value
stbi__get_marker(stbi__jpeg * j)2612*89c4ff92SAndroid Build Coastguard Worker static stbi_uc stbi__get_marker(stbi__jpeg *j)
2613*89c4ff92SAndroid Build Coastguard Worker {
2614*89c4ff92SAndroid Build Coastguard Worker stbi_uc x;
2615*89c4ff92SAndroid Build Coastguard Worker if (j->marker != STBI__MARKER_none) { x = j->marker; j->marker = STBI__MARKER_none; return x; }
2616*89c4ff92SAndroid Build Coastguard Worker x = stbi__get8(j->s);
2617*89c4ff92SAndroid Build Coastguard Worker if (x != 0xff) return STBI__MARKER_none;
2618*89c4ff92SAndroid Build Coastguard Worker while (x == 0xff)
2619*89c4ff92SAndroid Build Coastguard Worker x = stbi__get8(j->s); // consume repeated 0xff fill bytes
2620*89c4ff92SAndroid Build Coastguard Worker return x;
2621*89c4ff92SAndroid Build Coastguard Worker }
2622*89c4ff92SAndroid Build Coastguard Worker
2623*89c4ff92SAndroid Build Coastguard Worker // in each scan, we'll have scan_n components, and the order
2624*89c4ff92SAndroid Build Coastguard Worker // of the components is specified by order[]
2625*89c4ff92SAndroid Build Coastguard Worker #define STBI__RESTART(x) ((x) >= 0xd0 && (x) <= 0xd7)
2626*89c4ff92SAndroid Build Coastguard Worker
2627*89c4ff92SAndroid Build Coastguard Worker // after a restart interval, stbi__jpeg_reset the entropy decoder and
2628*89c4ff92SAndroid Build Coastguard Worker // the dc prediction
stbi__jpeg_reset(stbi__jpeg * j)2629*89c4ff92SAndroid Build Coastguard Worker static void stbi__jpeg_reset(stbi__jpeg *j)
2630*89c4ff92SAndroid Build Coastguard Worker {
2631*89c4ff92SAndroid Build Coastguard Worker j->code_bits = 0;
2632*89c4ff92SAndroid Build Coastguard Worker j->code_buffer = 0;
2633*89c4ff92SAndroid Build Coastguard Worker j->nomore = 0;
2634*89c4ff92SAndroid Build Coastguard Worker j->img_comp[0].dc_pred = j->img_comp[1].dc_pred = j->img_comp[2].dc_pred = j->img_comp[3].dc_pred = 0;
2635*89c4ff92SAndroid Build Coastguard Worker j->marker = STBI__MARKER_none;
2636*89c4ff92SAndroid Build Coastguard Worker j->todo = j->restart_interval ? j->restart_interval : 0x7fffffff;
2637*89c4ff92SAndroid Build Coastguard Worker j->eob_run = 0;
2638*89c4ff92SAndroid Build Coastguard Worker // no more than 1<<31 MCUs if no restart_interal? that's plenty safe,
2639*89c4ff92SAndroid Build Coastguard Worker // since we don't even allow 1<<30 pixels
2640*89c4ff92SAndroid Build Coastguard Worker }
2641*89c4ff92SAndroid Build Coastguard Worker
stbi__parse_entropy_coded_data(stbi__jpeg * z)2642*89c4ff92SAndroid Build Coastguard Worker static int stbi__parse_entropy_coded_data(stbi__jpeg *z)
2643*89c4ff92SAndroid Build Coastguard Worker {
2644*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg_reset(z);
2645*89c4ff92SAndroid Build Coastguard Worker if (!z->progressive) {
2646*89c4ff92SAndroid Build Coastguard Worker if (z->scan_n == 1) {
2647*89c4ff92SAndroid Build Coastguard Worker int i,j;
2648*89c4ff92SAndroid Build Coastguard Worker STBI_SIMD_ALIGN(short, data[64]);
2649*89c4ff92SAndroid Build Coastguard Worker int n = z->order[0];
2650*89c4ff92SAndroid Build Coastguard Worker // non-interleaved data, we just need to process one block at a time,
2651*89c4ff92SAndroid Build Coastguard Worker // in trivial scanline order
2652*89c4ff92SAndroid Build Coastguard Worker // number of blocks to do just depends on how many actual "pixels" this
2653*89c4ff92SAndroid Build Coastguard Worker // component has, independent of interleaved MCU blocking and such
2654*89c4ff92SAndroid Build Coastguard Worker int w = (z->img_comp[n].x+7) >> 3;
2655*89c4ff92SAndroid Build Coastguard Worker int h = (z->img_comp[n].y+7) >> 3;
2656*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < h; ++j) {
2657*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < w; ++i) {
2658*89c4ff92SAndroid Build Coastguard Worker int ha = z->img_comp[n].ha;
2659*89c4ff92SAndroid Build Coastguard Worker if (!stbi__jpeg_decode_block(z, data, z->huff_dc+z->img_comp[n].hd, z->huff_ac+ha, z->fast_ac[ha], n, z->dequant[z->img_comp[n].tq])) return 0;
2660*89c4ff92SAndroid Build Coastguard Worker z->idct_block_kernel(z->img_comp[n].data+z->img_comp[n].w2*j*8+i*8, z->img_comp[n].w2, data);
2661*89c4ff92SAndroid Build Coastguard Worker // every data block is an MCU, so countdown the restart interval
2662*89c4ff92SAndroid Build Coastguard Worker if (--z->todo <= 0) {
2663*89c4ff92SAndroid Build Coastguard Worker if (z->code_bits < 24) stbi__grow_buffer_unsafe(z);
2664*89c4ff92SAndroid Build Coastguard Worker // if it's NOT a restart, then just bail, so we get corrupt data
2665*89c4ff92SAndroid Build Coastguard Worker // rather than no data
2666*89c4ff92SAndroid Build Coastguard Worker if (!STBI__RESTART(z->marker)) return 1;
2667*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg_reset(z);
2668*89c4ff92SAndroid Build Coastguard Worker }
2669*89c4ff92SAndroid Build Coastguard Worker }
2670*89c4ff92SAndroid Build Coastguard Worker }
2671*89c4ff92SAndroid Build Coastguard Worker return 1;
2672*89c4ff92SAndroid Build Coastguard Worker } else { // interleaved
2673*89c4ff92SAndroid Build Coastguard Worker int i,j,k,x,y;
2674*89c4ff92SAndroid Build Coastguard Worker STBI_SIMD_ALIGN(short, data[64]);
2675*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < z->img_mcu_y; ++j) {
2676*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->img_mcu_x; ++i) {
2677*89c4ff92SAndroid Build Coastguard Worker // scan an interleaved mcu... process scan_n components in order
2678*89c4ff92SAndroid Build Coastguard Worker for (k=0; k < z->scan_n; ++k) {
2679*89c4ff92SAndroid Build Coastguard Worker int n = z->order[k];
2680*89c4ff92SAndroid Build Coastguard Worker // scan out an mcu's worth of this component; that's just determined
2681*89c4ff92SAndroid Build Coastguard Worker // by the basic H and V specified for the component
2682*89c4ff92SAndroid Build Coastguard Worker for (y=0; y < z->img_comp[n].v; ++y) {
2683*89c4ff92SAndroid Build Coastguard Worker for (x=0; x < z->img_comp[n].h; ++x) {
2684*89c4ff92SAndroid Build Coastguard Worker int x2 = (i*z->img_comp[n].h + x)*8;
2685*89c4ff92SAndroid Build Coastguard Worker int y2 = (j*z->img_comp[n].v + y)*8;
2686*89c4ff92SAndroid Build Coastguard Worker int ha = z->img_comp[n].ha;
2687*89c4ff92SAndroid Build Coastguard Worker if (!stbi__jpeg_decode_block(z, data, z->huff_dc+z->img_comp[n].hd, z->huff_ac+ha, z->fast_ac[ha], n, z->dequant[z->img_comp[n].tq])) return 0;
2688*89c4ff92SAndroid Build Coastguard Worker z->idct_block_kernel(z->img_comp[n].data+z->img_comp[n].w2*y2+x2, z->img_comp[n].w2, data);
2689*89c4ff92SAndroid Build Coastguard Worker }
2690*89c4ff92SAndroid Build Coastguard Worker }
2691*89c4ff92SAndroid Build Coastguard Worker }
2692*89c4ff92SAndroid Build Coastguard Worker // after all interleaved components, that's an interleaved MCU,
2693*89c4ff92SAndroid Build Coastguard Worker // so now count down the restart interval
2694*89c4ff92SAndroid Build Coastguard Worker if (--z->todo <= 0) {
2695*89c4ff92SAndroid Build Coastguard Worker if (z->code_bits < 24) stbi__grow_buffer_unsafe(z);
2696*89c4ff92SAndroid Build Coastguard Worker if (!STBI__RESTART(z->marker)) return 1;
2697*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg_reset(z);
2698*89c4ff92SAndroid Build Coastguard Worker }
2699*89c4ff92SAndroid Build Coastguard Worker }
2700*89c4ff92SAndroid Build Coastguard Worker }
2701*89c4ff92SAndroid Build Coastguard Worker return 1;
2702*89c4ff92SAndroid Build Coastguard Worker }
2703*89c4ff92SAndroid Build Coastguard Worker } else {
2704*89c4ff92SAndroid Build Coastguard Worker if (z->scan_n == 1) {
2705*89c4ff92SAndroid Build Coastguard Worker int i,j;
2706*89c4ff92SAndroid Build Coastguard Worker int n = z->order[0];
2707*89c4ff92SAndroid Build Coastguard Worker // non-interleaved data, we just need to process one block at a time,
2708*89c4ff92SAndroid Build Coastguard Worker // in trivial scanline order
2709*89c4ff92SAndroid Build Coastguard Worker // number of blocks to do just depends on how many actual "pixels" this
2710*89c4ff92SAndroid Build Coastguard Worker // component has, independent of interleaved MCU blocking and such
2711*89c4ff92SAndroid Build Coastguard Worker int w = (z->img_comp[n].x+7) >> 3;
2712*89c4ff92SAndroid Build Coastguard Worker int h = (z->img_comp[n].y+7) >> 3;
2713*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < h; ++j) {
2714*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < w; ++i) {
2715*89c4ff92SAndroid Build Coastguard Worker short *data = z->img_comp[n].coeff + 64 * (i + j * z->img_comp[n].coeff_w);
2716*89c4ff92SAndroid Build Coastguard Worker if (z->spec_start == 0) {
2717*89c4ff92SAndroid Build Coastguard Worker if (!stbi__jpeg_decode_block_prog_dc(z, data, &z->huff_dc[z->img_comp[n].hd], n))
2718*89c4ff92SAndroid Build Coastguard Worker return 0;
2719*89c4ff92SAndroid Build Coastguard Worker } else {
2720*89c4ff92SAndroid Build Coastguard Worker int ha = z->img_comp[n].ha;
2721*89c4ff92SAndroid Build Coastguard Worker if (!stbi__jpeg_decode_block_prog_ac(z, data, &z->huff_ac[ha], z->fast_ac[ha]))
2722*89c4ff92SAndroid Build Coastguard Worker return 0;
2723*89c4ff92SAndroid Build Coastguard Worker }
2724*89c4ff92SAndroid Build Coastguard Worker // every data block is an MCU, so countdown the restart interval
2725*89c4ff92SAndroid Build Coastguard Worker if (--z->todo <= 0) {
2726*89c4ff92SAndroid Build Coastguard Worker if (z->code_bits < 24) stbi__grow_buffer_unsafe(z);
2727*89c4ff92SAndroid Build Coastguard Worker if (!STBI__RESTART(z->marker)) return 1;
2728*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg_reset(z);
2729*89c4ff92SAndroid Build Coastguard Worker }
2730*89c4ff92SAndroid Build Coastguard Worker }
2731*89c4ff92SAndroid Build Coastguard Worker }
2732*89c4ff92SAndroid Build Coastguard Worker return 1;
2733*89c4ff92SAndroid Build Coastguard Worker } else { // interleaved
2734*89c4ff92SAndroid Build Coastguard Worker int i,j,k,x,y;
2735*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < z->img_mcu_y; ++j) {
2736*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->img_mcu_x; ++i) {
2737*89c4ff92SAndroid Build Coastguard Worker // scan an interleaved mcu... process scan_n components in order
2738*89c4ff92SAndroid Build Coastguard Worker for (k=0; k < z->scan_n; ++k) {
2739*89c4ff92SAndroid Build Coastguard Worker int n = z->order[k];
2740*89c4ff92SAndroid Build Coastguard Worker // scan out an mcu's worth of this component; that's just determined
2741*89c4ff92SAndroid Build Coastguard Worker // by the basic H and V specified for the component
2742*89c4ff92SAndroid Build Coastguard Worker for (y=0; y < z->img_comp[n].v; ++y) {
2743*89c4ff92SAndroid Build Coastguard Worker for (x=0; x < z->img_comp[n].h; ++x) {
2744*89c4ff92SAndroid Build Coastguard Worker int x2 = (i*z->img_comp[n].h + x);
2745*89c4ff92SAndroid Build Coastguard Worker int y2 = (j*z->img_comp[n].v + y);
2746*89c4ff92SAndroid Build Coastguard Worker short *data = z->img_comp[n].coeff + 64 * (x2 + y2 * z->img_comp[n].coeff_w);
2747*89c4ff92SAndroid Build Coastguard Worker if (!stbi__jpeg_decode_block_prog_dc(z, data, &z->huff_dc[z->img_comp[n].hd], n))
2748*89c4ff92SAndroid Build Coastguard Worker return 0;
2749*89c4ff92SAndroid Build Coastguard Worker }
2750*89c4ff92SAndroid Build Coastguard Worker }
2751*89c4ff92SAndroid Build Coastguard Worker }
2752*89c4ff92SAndroid Build Coastguard Worker // after all interleaved components, that's an interleaved MCU,
2753*89c4ff92SAndroid Build Coastguard Worker // so now count down the restart interval
2754*89c4ff92SAndroid Build Coastguard Worker if (--z->todo <= 0) {
2755*89c4ff92SAndroid Build Coastguard Worker if (z->code_bits < 24) stbi__grow_buffer_unsafe(z);
2756*89c4ff92SAndroid Build Coastguard Worker if (!STBI__RESTART(z->marker)) return 1;
2757*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg_reset(z);
2758*89c4ff92SAndroid Build Coastguard Worker }
2759*89c4ff92SAndroid Build Coastguard Worker }
2760*89c4ff92SAndroid Build Coastguard Worker }
2761*89c4ff92SAndroid Build Coastguard Worker return 1;
2762*89c4ff92SAndroid Build Coastguard Worker }
2763*89c4ff92SAndroid Build Coastguard Worker }
2764*89c4ff92SAndroid Build Coastguard Worker }
2765*89c4ff92SAndroid Build Coastguard Worker
stbi__jpeg_dequantize(short * data,stbi__uint16 * dequant)2766*89c4ff92SAndroid Build Coastguard Worker static void stbi__jpeg_dequantize(short *data, stbi__uint16 *dequant)
2767*89c4ff92SAndroid Build Coastguard Worker {
2768*89c4ff92SAndroid Build Coastguard Worker int i;
2769*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < 64; ++i)
2770*89c4ff92SAndroid Build Coastguard Worker data[i] *= dequant[i];
2771*89c4ff92SAndroid Build Coastguard Worker }
2772*89c4ff92SAndroid Build Coastguard Worker
stbi__jpeg_finish(stbi__jpeg * z)2773*89c4ff92SAndroid Build Coastguard Worker static void stbi__jpeg_finish(stbi__jpeg *z)
2774*89c4ff92SAndroid Build Coastguard Worker {
2775*89c4ff92SAndroid Build Coastguard Worker if (z->progressive) {
2776*89c4ff92SAndroid Build Coastguard Worker // dequantize and idct the data
2777*89c4ff92SAndroid Build Coastguard Worker int i,j,n;
2778*89c4ff92SAndroid Build Coastguard Worker for (n=0; n < z->s->img_n; ++n) {
2779*89c4ff92SAndroid Build Coastguard Worker int w = (z->img_comp[n].x+7) >> 3;
2780*89c4ff92SAndroid Build Coastguard Worker int h = (z->img_comp[n].y+7) >> 3;
2781*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < h; ++j) {
2782*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < w; ++i) {
2783*89c4ff92SAndroid Build Coastguard Worker short *data = z->img_comp[n].coeff + 64 * (i + j * z->img_comp[n].coeff_w);
2784*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg_dequantize(data, z->dequant[z->img_comp[n].tq]);
2785*89c4ff92SAndroid Build Coastguard Worker z->idct_block_kernel(z->img_comp[n].data+z->img_comp[n].w2*j*8+i*8, z->img_comp[n].w2, data);
2786*89c4ff92SAndroid Build Coastguard Worker }
2787*89c4ff92SAndroid Build Coastguard Worker }
2788*89c4ff92SAndroid Build Coastguard Worker }
2789*89c4ff92SAndroid Build Coastguard Worker }
2790*89c4ff92SAndroid Build Coastguard Worker }
2791*89c4ff92SAndroid Build Coastguard Worker
stbi__process_marker(stbi__jpeg * z,int m)2792*89c4ff92SAndroid Build Coastguard Worker static int stbi__process_marker(stbi__jpeg *z, int m)
2793*89c4ff92SAndroid Build Coastguard Worker {
2794*89c4ff92SAndroid Build Coastguard Worker int L;
2795*89c4ff92SAndroid Build Coastguard Worker switch (m) {
2796*89c4ff92SAndroid Build Coastguard Worker case STBI__MARKER_none: // no marker found
2797*89c4ff92SAndroid Build Coastguard Worker return stbi__err("expected marker","Corrupt JPEG");
2798*89c4ff92SAndroid Build Coastguard Worker
2799*89c4ff92SAndroid Build Coastguard Worker case 0xDD: // DRI - specify restart interval
2800*89c4ff92SAndroid Build Coastguard Worker if (stbi__get16be(z->s) != 4) return stbi__err("bad DRI len","Corrupt JPEG");
2801*89c4ff92SAndroid Build Coastguard Worker z->restart_interval = stbi__get16be(z->s);
2802*89c4ff92SAndroid Build Coastguard Worker return 1;
2803*89c4ff92SAndroid Build Coastguard Worker
2804*89c4ff92SAndroid Build Coastguard Worker case 0xDB: // DQT - define quantization table
2805*89c4ff92SAndroid Build Coastguard Worker L = stbi__get16be(z->s)-2;
2806*89c4ff92SAndroid Build Coastguard Worker while (L > 0) {
2807*89c4ff92SAndroid Build Coastguard Worker int q = stbi__get8(z->s);
2808*89c4ff92SAndroid Build Coastguard Worker int p = q >> 4, sixteen = (p != 0);
2809*89c4ff92SAndroid Build Coastguard Worker int t = q & 15,i;
2810*89c4ff92SAndroid Build Coastguard Worker if (p != 0 && p != 1) return stbi__err("bad DQT type","Corrupt JPEG");
2811*89c4ff92SAndroid Build Coastguard Worker if (t > 3) return stbi__err("bad DQT table","Corrupt JPEG");
2812*89c4ff92SAndroid Build Coastguard Worker
2813*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < 64; ++i)
2814*89c4ff92SAndroid Build Coastguard Worker z->dequant[t][stbi__jpeg_dezigzag[i]] = (stbi__uint16)(sixteen ? stbi__get16be(z->s) : stbi__get8(z->s));
2815*89c4ff92SAndroid Build Coastguard Worker L -= (sixteen ? 129 : 65);
2816*89c4ff92SAndroid Build Coastguard Worker }
2817*89c4ff92SAndroid Build Coastguard Worker return L==0;
2818*89c4ff92SAndroid Build Coastguard Worker
2819*89c4ff92SAndroid Build Coastguard Worker case 0xC4: // DHT - define huffman table
2820*89c4ff92SAndroid Build Coastguard Worker L = stbi__get16be(z->s)-2;
2821*89c4ff92SAndroid Build Coastguard Worker while (L > 0) {
2822*89c4ff92SAndroid Build Coastguard Worker stbi_uc *v;
2823*89c4ff92SAndroid Build Coastguard Worker int sizes[16],i,n=0;
2824*89c4ff92SAndroid Build Coastguard Worker int q = stbi__get8(z->s);
2825*89c4ff92SAndroid Build Coastguard Worker int tc = q >> 4;
2826*89c4ff92SAndroid Build Coastguard Worker int th = q & 15;
2827*89c4ff92SAndroid Build Coastguard Worker if (tc > 1 || th > 3) return stbi__err("bad DHT header","Corrupt JPEG");
2828*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < 16; ++i) {
2829*89c4ff92SAndroid Build Coastguard Worker sizes[i] = stbi__get8(z->s);
2830*89c4ff92SAndroid Build Coastguard Worker n += sizes[i];
2831*89c4ff92SAndroid Build Coastguard Worker }
2832*89c4ff92SAndroid Build Coastguard Worker L -= 17;
2833*89c4ff92SAndroid Build Coastguard Worker if (tc == 0) {
2834*89c4ff92SAndroid Build Coastguard Worker if (!stbi__build_huffman(z->huff_dc+th, sizes)) return 0;
2835*89c4ff92SAndroid Build Coastguard Worker v = z->huff_dc[th].values;
2836*89c4ff92SAndroid Build Coastguard Worker } else {
2837*89c4ff92SAndroid Build Coastguard Worker if (!stbi__build_huffman(z->huff_ac+th, sizes)) return 0;
2838*89c4ff92SAndroid Build Coastguard Worker v = z->huff_ac[th].values;
2839*89c4ff92SAndroid Build Coastguard Worker }
2840*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < n; ++i)
2841*89c4ff92SAndroid Build Coastguard Worker v[i] = stbi__get8(z->s);
2842*89c4ff92SAndroid Build Coastguard Worker if (tc != 0)
2843*89c4ff92SAndroid Build Coastguard Worker stbi__build_fast_ac(z->fast_ac[th], z->huff_ac + th);
2844*89c4ff92SAndroid Build Coastguard Worker L -= n;
2845*89c4ff92SAndroid Build Coastguard Worker }
2846*89c4ff92SAndroid Build Coastguard Worker return L==0;
2847*89c4ff92SAndroid Build Coastguard Worker }
2848*89c4ff92SAndroid Build Coastguard Worker
2849*89c4ff92SAndroid Build Coastguard Worker // check for comment block or APP blocks
2850*89c4ff92SAndroid Build Coastguard Worker if ((m >= 0xE0 && m <= 0xEF) || m == 0xFE) {
2851*89c4ff92SAndroid Build Coastguard Worker L = stbi__get16be(z->s);
2852*89c4ff92SAndroid Build Coastguard Worker if (L < 2) {
2853*89c4ff92SAndroid Build Coastguard Worker if (m == 0xFE)
2854*89c4ff92SAndroid Build Coastguard Worker return stbi__err("bad COM len","Corrupt JPEG");
2855*89c4ff92SAndroid Build Coastguard Worker else
2856*89c4ff92SAndroid Build Coastguard Worker return stbi__err("bad APP len","Corrupt JPEG");
2857*89c4ff92SAndroid Build Coastguard Worker }
2858*89c4ff92SAndroid Build Coastguard Worker L -= 2;
2859*89c4ff92SAndroid Build Coastguard Worker
2860*89c4ff92SAndroid Build Coastguard Worker if (m == 0xE0 && L >= 5) { // JFIF APP0 segment
2861*89c4ff92SAndroid Build Coastguard Worker static const unsigned char tag[5] = {'J','F','I','F','\0'};
2862*89c4ff92SAndroid Build Coastguard Worker int ok = 1;
2863*89c4ff92SAndroid Build Coastguard Worker int i;
2864*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < 5; ++i)
2865*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(z->s) != tag[i])
2866*89c4ff92SAndroid Build Coastguard Worker ok = 0;
2867*89c4ff92SAndroid Build Coastguard Worker L -= 5;
2868*89c4ff92SAndroid Build Coastguard Worker if (ok)
2869*89c4ff92SAndroid Build Coastguard Worker z->jfif = 1;
2870*89c4ff92SAndroid Build Coastguard Worker } else if (m == 0xEE && L >= 12) { // Adobe APP14 segment
2871*89c4ff92SAndroid Build Coastguard Worker static const unsigned char tag[6] = {'A','d','o','b','e','\0'};
2872*89c4ff92SAndroid Build Coastguard Worker int ok = 1;
2873*89c4ff92SAndroid Build Coastguard Worker int i;
2874*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < 6; ++i)
2875*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(z->s) != tag[i])
2876*89c4ff92SAndroid Build Coastguard Worker ok = 0;
2877*89c4ff92SAndroid Build Coastguard Worker L -= 6;
2878*89c4ff92SAndroid Build Coastguard Worker if (ok) {
2879*89c4ff92SAndroid Build Coastguard Worker stbi__get8(z->s); // version
2880*89c4ff92SAndroid Build Coastguard Worker stbi__get16be(z->s); // flags0
2881*89c4ff92SAndroid Build Coastguard Worker stbi__get16be(z->s); // flags1
2882*89c4ff92SAndroid Build Coastguard Worker z->app14_color_transform = stbi__get8(z->s); // color transform
2883*89c4ff92SAndroid Build Coastguard Worker L -= 6;
2884*89c4ff92SAndroid Build Coastguard Worker }
2885*89c4ff92SAndroid Build Coastguard Worker }
2886*89c4ff92SAndroid Build Coastguard Worker
2887*89c4ff92SAndroid Build Coastguard Worker stbi__skip(z->s, L);
2888*89c4ff92SAndroid Build Coastguard Worker return 1;
2889*89c4ff92SAndroid Build Coastguard Worker }
2890*89c4ff92SAndroid Build Coastguard Worker
2891*89c4ff92SAndroid Build Coastguard Worker return stbi__err("unknown marker","Corrupt JPEG");
2892*89c4ff92SAndroid Build Coastguard Worker }
2893*89c4ff92SAndroid Build Coastguard Worker
2894*89c4ff92SAndroid Build Coastguard Worker // after we see SOS
stbi__process_scan_header(stbi__jpeg * z)2895*89c4ff92SAndroid Build Coastguard Worker static int stbi__process_scan_header(stbi__jpeg *z)
2896*89c4ff92SAndroid Build Coastguard Worker {
2897*89c4ff92SAndroid Build Coastguard Worker int i;
2898*89c4ff92SAndroid Build Coastguard Worker int Ls = stbi__get16be(z->s);
2899*89c4ff92SAndroid Build Coastguard Worker z->scan_n = stbi__get8(z->s);
2900*89c4ff92SAndroid Build Coastguard Worker if (z->scan_n < 1 || z->scan_n > 4 || z->scan_n > (int) z->s->img_n) return stbi__err("bad SOS component count","Corrupt JPEG");
2901*89c4ff92SAndroid Build Coastguard Worker if (Ls != 6+2*z->scan_n) return stbi__err("bad SOS len","Corrupt JPEG");
2902*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->scan_n; ++i) {
2903*89c4ff92SAndroid Build Coastguard Worker int id = stbi__get8(z->s), which;
2904*89c4ff92SAndroid Build Coastguard Worker int q = stbi__get8(z->s);
2905*89c4ff92SAndroid Build Coastguard Worker for (which = 0; which < z->s->img_n; ++which)
2906*89c4ff92SAndroid Build Coastguard Worker if (z->img_comp[which].id == id)
2907*89c4ff92SAndroid Build Coastguard Worker break;
2908*89c4ff92SAndroid Build Coastguard Worker if (which == z->s->img_n) return 0; // no match
2909*89c4ff92SAndroid Build Coastguard Worker z->img_comp[which].hd = q >> 4; if (z->img_comp[which].hd > 3) return stbi__err("bad DC huff","Corrupt JPEG");
2910*89c4ff92SAndroid Build Coastguard Worker z->img_comp[which].ha = q & 15; if (z->img_comp[which].ha > 3) return stbi__err("bad AC huff","Corrupt JPEG");
2911*89c4ff92SAndroid Build Coastguard Worker z->order[i] = which;
2912*89c4ff92SAndroid Build Coastguard Worker }
2913*89c4ff92SAndroid Build Coastguard Worker
2914*89c4ff92SAndroid Build Coastguard Worker {
2915*89c4ff92SAndroid Build Coastguard Worker int aa;
2916*89c4ff92SAndroid Build Coastguard Worker z->spec_start = stbi__get8(z->s);
2917*89c4ff92SAndroid Build Coastguard Worker z->spec_end = stbi__get8(z->s); // should be 63, but might be 0
2918*89c4ff92SAndroid Build Coastguard Worker aa = stbi__get8(z->s);
2919*89c4ff92SAndroid Build Coastguard Worker z->succ_high = (aa >> 4);
2920*89c4ff92SAndroid Build Coastguard Worker z->succ_low = (aa & 15);
2921*89c4ff92SAndroid Build Coastguard Worker if (z->progressive) {
2922*89c4ff92SAndroid Build Coastguard Worker if (z->spec_start > 63 || z->spec_end > 63 || z->spec_start > z->spec_end || z->succ_high > 13 || z->succ_low > 13)
2923*89c4ff92SAndroid Build Coastguard Worker return stbi__err("bad SOS", "Corrupt JPEG");
2924*89c4ff92SAndroid Build Coastguard Worker } else {
2925*89c4ff92SAndroid Build Coastguard Worker if (z->spec_start != 0) return stbi__err("bad SOS","Corrupt JPEG");
2926*89c4ff92SAndroid Build Coastguard Worker if (z->succ_high != 0 || z->succ_low != 0) return stbi__err("bad SOS","Corrupt JPEG");
2927*89c4ff92SAndroid Build Coastguard Worker z->spec_end = 63;
2928*89c4ff92SAndroid Build Coastguard Worker }
2929*89c4ff92SAndroid Build Coastguard Worker }
2930*89c4ff92SAndroid Build Coastguard Worker
2931*89c4ff92SAndroid Build Coastguard Worker return 1;
2932*89c4ff92SAndroid Build Coastguard Worker }
2933*89c4ff92SAndroid Build Coastguard Worker
stbi__free_jpeg_components(stbi__jpeg * z,int ncomp,int why)2934*89c4ff92SAndroid Build Coastguard Worker static int stbi__free_jpeg_components(stbi__jpeg *z, int ncomp, int why)
2935*89c4ff92SAndroid Build Coastguard Worker {
2936*89c4ff92SAndroid Build Coastguard Worker int i;
2937*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < ncomp; ++i) {
2938*89c4ff92SAndroid Build Coastguard Worker if (z->img_comp[i].raw_data) {
2939*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(z->img_comp[i].raw_data);
2940*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].raw_data = NULL;
2941*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].data = NULL;
2942*89c4ff92SAndroid Build Coastguard Worker }
2943*89c4ff92SAndroid Build Coastguard Worker if (z->img_comp[i].raw_coeff) {
2944*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(z->img_comp[i].raw_coeff);
2945*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].raw_coeff = 0;
2946*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].coeff = 0;
2947*89c4ff92SAndroid Build Coastguard Worker }
2948*89c4ff92SAndroid Build Coastguard Worker if (z->img_comp[i].linebuf) {
2949*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(z->img_comp[i].linebuf);
2950*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].linebuf = NULL;
2951*89c4ff92SAndroid Build Coastguard Worker }
2952*89c4ff92SAndroid Build Coastguard Worker }
2953*89c4ff92SAndroid Build Coastguard Worker return why;
2954*89c4ff92SAndroid Build Coastguard Worker }
2955*89c4ff92SAndroid Build Coastguard Worker
stbi__process_frame_header(stbi__jpeg * z,int scan)2956*89c4ff92SAndroid Build Coastguard Worker static int stbi__process_frame_header(stbi__jpeg *z, int scan)
2957*89c4ff92SAndroid Build Coastguard Worker {
2958*89c4ff92SAndroid Build Coastguard Worker stbi__context *s = z->s;
2959*89c4ff92SAndroid Build Coastguard Worker int Lf,p,i,q, h_max=1,v_max=1,c;
2960*89c4ff92SAndroid Build Coastguard Worker Lf = stbi__get16be(s); if (Lf < 11) return stbi__err("bad SOF len","Corrupt JPEG"); // JPEG
2961*89c4ff92SAndroid Build Coastguard Worker p = stbi__get8(s); if (p != 8) return stbi__err("only 8-bit","JPEG format not supported: 8-bit only"); // JPEG baseline
2962*89c4ff92SAndroid Build Coastguard Worker s->img_y = stbi__get16be(s); if (s->img_y == 0) return stbi__err("no header height", "JPEG format not supported: delayed height"); // Legal, but we don't handle it--but neither does IJG
2963*89c4ff92SAndroid Build Coastguard Worker s->img_x = stbi__get16be(s); if (s->img_x == 0) return stbi__err("0 width","Corrupt JPEG"); // JPEG requires
2964*89c4ff92SAndroid Build Coastguard Worker c = stbi__get8(s);
2965*89c4ff92SAndroid Build Coastguard Worker if (c != 3 && c != 1 && c != 4) return stbi__err("bad component count","Corrupt JPEG");
2966*89c4ff92SAndroid Build Coastguard Worker s->img_n = c;
2967*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < c; ++i) {
2968*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].data = NULL;
2969*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].linebuf = NULL;
2970*89c4ff92SAndroid Build Coastguard Worker }
2971*89c4ff92SAndroid Build Coastguard Worker
2972*89c4ff92SAndroid Build Coastguard Worker if (Lf != 8+3*s->img_n) return stbi__err("bad SOF len","Corrupt JPEG");
2973*89c4ff92SAndroid Build Coastguard Worker
2974*89c4ff92SAndroid Build Coastguard Worker z->rgb = 0;
2975*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < s->img_n; ++i) {
2976*89c4ff92SAndroid Build Coastguard Worker static unsigned char rgb[3] = { 'R', 'G', 'B' };
2977*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].id = stbi__get8(s);
2978*89c4ff92SAndroid Build Coastguard Worker if (s->img_n == 3 && z->img_comp[i].id == rgb[i])
2979*89c4ff92SAndroid Build Coastguard Worker ++z->rgb;
2980*89c4ff92SAndroid Build Coastguard Worker q = stbi__get8(s);
2981*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].h = (q >> 4); if (!z->img_comp[i].h || z->img_comp[i].h > 4) return stbi__err("bad H","Corrupt JPEG");
2982*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].v = q & 15; if (!z->img_comp[i].v || z->img_comp[i].v > 4) return stbi__err("bad V","Corrupt JPEG");
2983*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].tq = stbi__get8(s); if (z->img_comp[i].tq > 3) return stbi__err("bad TQ","Corrupt JPEG");
2984*89c4ff92SAndroid Build Coastguard Worker }
2985*89c4ff92SAndroid Build Coastguard Worker
2986*89c4ff92SAndroid Build Coastguard Worker if (scan != STBI__SCAN_load) return 1;
2987*89c4ff92SAndroid Build Coastguard Worker
2988*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad3sizes_valid(s->img_x, s->img_y, s->img_n, 0)) return stbi__err("too large", "Image too large to decode");
2989*89c4ff92SAndroid Build Coastguard Worker
2990*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < s->img_n; ++i) {
2991*89c4ff92SAndroid Build Coastguard Worker if (z->img_comp[i].h > h_max) h_max = z->img_comp[i].h;
2992*89c4ff92SAndroid Build Coastguard Worker if (z->img_comp[i].v > v_max) v_max = z->img_comp[i].v;
2993*89c4ff92SAndroid Build Coastguard Worker }
2994*89c4ff92SAndroid Build Coastguard Worker
2995*89c4ff92SAndroid Build Coastguard Worker // compute interleaved mcu info
2996*89c4ff92SAndroid Build Coastguard Worker z->img_h_max = h_max;
2997*89c4ff92SAndroid Build Coastguard Worker z->img_v_max = v_max;
2998*89c4ff92SAndroid Build Coastguard Worker z->img_mcu_w = h_max * 8;
2999*89c4ff92SAndroid Build Coastguard Worker z->img_mcu_h = v_max * 8;
3000*89c4ff92SAndroid Build Coastguard Worker // these sizes can't be more than 17 bits
3001*89c4ff92SAndroid Build Coastguard Worker z->img_mcu_x = (s->img_x + z->img_mcu_w-1) / z->img_mcu_w;
3002*89c4ff92SAndroid Build Coastguard Worker z->img_mcu_y = (s->img_y + z->img_mcu_h-1) / z->img_mcu_h;
3003*89c4ff92SAndroid Build Coastguard Worker
3004*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < s->img_n; ++i) {
3005*89c4ff92SAndroid Build Coastguard Worker // number of effective pixels (e.g. for non-interleaved MCU)
3006*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].x = (s->img_x * z->img_comp[i].h + h_max-1) / h_max;
3007*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].y = (s->img_y * z->img_comp[i].v + v_max-1) / v_max;
3008*89c4ff92SAndroid Build Coastguard Worker // to simplify generation, we'll allocate enough memory to decode
3009*89c4ff92SAndroid Build Coastguard Worker // the bogus oversized data from using interleaved MCUs and their
3010*89c4ff92SAndroid Build Coastguard Worker // big blocks (e.g. a 16x16 iMCU on an image of width 33); we won't
3011*89c4ff92SAndroid Build Coastguard Worker // discard the extra data until colorspace conversion
3012*89c4ff92SAndroid Build Coastguard Worker //
3013*89c4ff92SAndroid Build Coastguard Worker // img_mcu_x, img_mcu_y: <=17 bits; comp[i].h and .v are <=4 (checked earlier)
3014*89c4ff92SAndroid Build Coastguard Worker // so these muls can't overflow with 32-bit ints (which we require)
3015*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].w2 = z->img_mcu_x * z->img_comp[i].h * 8;
3016*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].h2 = z->img_mcu_y * z->img_comp[i].v * 8;
3017*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].coeff = 0;
3018*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].raw_coeff = 0;
3019*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].linebuf = NULL;
3020*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].raw_data = stbi__malloc_mad2(z->img_comp[i].w2, z->img_comp[i].h2, 15);
3021*89c4ff92SAndroid Build Coastguard Worker if (z->img_comp[i].raw_data == NULL)
3022*89c4ff92SAndroid Build Coastguard Worker return stbi__free_jpeg_components(z, i+1, stbi__err("outofmem", "Out of memory"));
3023*89c4ff92SAndroid Build Coastguard Worker // align blocks for idct using mmx/sse
3024*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].data = (stbi_uc*) (((size_t) z->img_comp[i].raw_data + 15) & ~15);
3025*89c4ff92SAndroid Build Coastguard Worker if (z->progressive) {
3026*89c4ff92SAndroid Build Coastguard Worker // w2, h2 are multiples of 8 (see above)
3027*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].coeff_w = z->img_comp[i].w2 / 8;
3028*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].coeff_h = z->img_comp[i].h2 / 8;
3029*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].raw_coeff = stbi__malloc_mad3(z->img_comp[i].w2, z->img_comp[i].h2, sizeof(short), 15);
3030*89c4ff92SAndroid Build Coastguard Worker if (z->img_comp[i].raw_coeff == NULL)
3031*89c4ff92SAndroid Build Coastguard Worker return stbi__free_jpeg_components(z, i+1, stbi__err("outofmem", "Out of memory"));
3032*89c4ff92SAndroid Build Coastguard Worker z->img_comp[i].coeff = (short*) (((size_t) z->img_comp[i].raw_coeff + 15) & ~15);
3033*89c4ff92SAndroid Build Coastguard Worker }
3034*89c4ff92SAndroid Build Coastguard Worker }
3035*89c4ff92SAndroid Build Coastguard Worker
3036*89c4ff92SAndroid Build Coastguard Worker return 1;
3037*89c4ff92SAndroid Build Coastguard Worker }
3038*89c4ff92SAndroid Build Coastguard Worker
3039*89c4ff92SAndroid Build Coastguard Worker // use comparisons since in some cases we handle more than one case (e.g. SOF)
3040*89c4ff92SAndroid Build Coastguard Worker #define stbi__DNL(x) ((x) == 0xdc)
3041*89c4ff92SAndroid Build Coastguard Worker #define stbi__SOI(x) ((x) == 0xd8)
3042*89c4ff92SAndroid Build Coastguard Worker #define stbi__EOI(x) ((x) == 0xd9)
3043*89c4ff92SAndroid Build Coastguard Worker #define stbi__SOF(x) ((x) == 0xc0 || (x) == 0xc1 || (x) == 0xc2)
3044*89c4ff92SAndroid Build Coastguard Worker #define stbi__SOS(x) ((x) == 0xda)
3045*89c4ff92SAndroid Build Coastguard Worker
3046*89c4ff92SAndroid Build Coastguard Worker #define stbi__SOF_progressive(x) ((x) == 0xc2)
3047*89c4ff92SAndroid Build Coastguard Worker
stbi__decode_jpeg_header(stbi__jpeg * z,int scan)3048*89c4ff92SAndroid Build Coastguard Worker static int stbi__decode_jpeg_header(stbi__jpeg *z, int scan)
3049*89c4ff92SAndroid Build Coastguard Worker {
3050*89c4ff92SAndroid Build Coastguard Worker int m;
3051*89c4ff92SAndroid Build Coastguard Worker z->jfif = 0;
3052*89c4ff92SAndroid Build Coastguard Worker z->app14_color_transform = -1; // valid values are 0,1,2
3053*89c4ff92SAndroid Build Coastguard Worker z->marker = STBI__MARKER_none; // initialize cached marker to empty
3054*89c4ff92SAndroid Build Coastguard Worker m = stbi__get_marker(z);
3055*89c4ff92SAndroid Build Coastguard Worker if (!stbi__SOI(m)) return stbi__err("no SOI","Corrupt JPEG");
3056*89c4ff92SAndroid Build Coastguard Worker if (scan == STBI__SCAN_type) return 1;
3057*89c4ff92SAndroid Build Coastguard Worker m = stbi__get_marker(z);
3058*89c4ff92SAndroid Build Coastguard Worker while (!stbi__SOF(m)) {
3059*89c4ff92SAndroid Build Coastguard Worker if (!stbi__process_marker(z,m)) return 0;
3060*89c4ff92SAndroid Build Coastguard Worker m = stbi__get_marker(z);
3061*89c4ff92SAndroid Build Coastguard Worker while (m == STBI__MARKER_none) {
3062*89c4ff92SAndroid Build Coastguard Worker // some files have extra padding after their blocks, so ok, we'll scan
3063*89c4ff92SAndroid Build Coastguard Worker if (stbi__at_eof(z->s)) return stbi__err("no SOF", "Corrupt JPEG");
3064*89c4ff92SAndroid Build Coastguard Worker m = stbi__get_marker(z);
3065*89c4ff92SAndroid Build Coastguard Worker }
3066*89c4ff92SAndroid Build Coastguard Worker }
3067*89c4ff92SAndroid Build Coastguard Worker z->progressive = stbi__SOF_progressive(m);
3068*89c4ff92SAndroid Build Coastguard Worker if (!stbi__process_frame_header(z, scan)) return 0;
3069*89c4ff92SAndroid Build Coastguard Worker return 1;
3070*89c4ff92SAndroid Build Coastguard Worker }
3071*89c4ff92SAndroid Build Coastguard Worker
3072*89c4ff92SAndroid Build Coastguard Worker // decode image to YCbCr format
stbi__decode_jpeg_image(stbi__jpeg * j)3073*89c4ff92SAndroid Build Coastguard Worker static int stbi__decode_jpeg_image(stbi__jpeg *j)
3074*89c4ff92SAndroid Build Coastguard Worker {
3075*89c4ff92SAndroid Build Coastguard Worker int m;
3076*89c4ff92SAndroid Build Coastguard Worker for (m = 0; m < 4; m++) {
3077*89c4ff92SAndroid Build Coastguard Worker j->img_comp[m].raw_data = NULL;
3078*89c4ff92SAndroid Build Coastguard Worker j->img_comp[m].raw_coeff = NULL;
3079*89c4ff92SAndroid Build Coastguard Worker }
3080*89c4ff92SAndroid Build Coastguard Worker j->restart_interval = 0;
3081*89c4ff92SAndroid Build Coastguard Worker if (!stbi__decode_jpeg_header(j, STBI__SCAN_load)) return 0;
3082*89c4ff92SAndroid Build Coastguard Worker m = stbi__get_marker(j);
3083*89c4ff92SAndroid Build Coastguard Worker while (!stbi__EOI(m)) {
3084*89c4ff92SAndroid Build Coastguard Worker if (stbi__SOS(m)) {
3085*89c4ff92SAndroid Build Coastguard Worker if (!stbi__process_scan_header(j)) return 0;
3086*89c4ff92SAndroid Build Coastguard Worker if (!stbi__parse_entropy_coded_data(j)) return 0;
3087*89c4ff92SAndroid Build Coastguard Worker if (j->marker == STBI__MARKER_none ) {
3088*89c4ff92SAndroid Build Coastguard Worker // handle 0s at the end of image data from IP Kamera 9060
3089*89c4ff92SAndroid Build Coastguard Worker while (!stbi__at_eof(j->s)) {
3090*89c4ff92SAndroid Build Coastguard Worker int x = stbi__get8(j->s);
3091*89c4ff92SAndroid Build Coastguard Worker if (x == 255) {
3092*89c4ff92SAndroid Build Coastguard Worker j->marker = stbi__get8(j->s);
3093*89c4ff92SAndroid Build Coastguard Worker break;
3094*89c4ff92SAndroid Build Coastguard Worker }
3095*89c4ff92SAndroid Build Coastguard Worker }
3096*89c4ff92SAndroid Build Coastguard Worker // if we reach eof without hitting a marker, stbi__get_marker() below will fail and we'll eventually return 0
3097*89c4ff92SAndroid Build Coastguard Worker }
3098*89c4ff92SAndroid Build Coastguard Worker } else if (stbi__DNL(m)) {
3099*89c4ff92SAndroid Build Coastguard Worker int Ld = stbi__get16be(j->s);
3100*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 NL = stbi__get16be(j->s);
3101*89c4ff92SAndroid Build Coastguard Worker if (Ld != 4) stbi__err("bad DNL len", "Corrupt JPEG");
3102*89c4ff92SAndroid Build Coastguard Worker if (NL != j->s->img_y) stbi__err("bad DNL height", "Corrupt JPEG");
3103*89c4ff92SAndroid Build Coastguard Worker } else {
3104*89c4ff92SAndroid Build Coastguard Worker if (!stbi__process_marker(j, m)) return 0;
3105*89c4ff92SAndroid Build Coastguard Worker }
3106*89c4ff92SAndroid Build Coastguard Worker m = stbi__get_marker(j);
3107*89c4ff92SAndroid Build Coastguard Worker }
3108*89c4ff92SAndroid Build Coastguard Worker if (j->progressive)
3109*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg_finish(j);
3110*89c4ff92SAndroid Build Coastguard Worker return 1;
3111*89c4ff92SAndroid Build Coastguard Worker }
3112*89c4ff92SAndroid Build Coastguard Worker
3113*89c4ff92SAndroid Build Coastguard Worker // static jfif-centered resampling (across block boundaries)
3114*89c4ff92SAndroid Build Coastguard Worker
3115*89c4ff92SAndroid Build Coastguard Worker typedef stbi_uc *(*resample_row_func)(stbi_uc *out, stbi_uc *in0, stbi_uc *in1,
3116*89c4ff92SAndroid Build Coastguard Worker int w, int hs);
3117*89c4ff92SAndroid Build Coastguard Worker
3118*89c4ff92SAndroid Build Coastguard Worker #define stbi__div4(x) ((stbi_uc) ((x) >> 2))
3119*89c4ff92SAndroid Build Coastguard Worker
resample_row_1(stbi_uc * out,stbi_uc * in_near,stbi_uc * in_far,int w,int hs)3120*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *resample_row_1(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far, int w, int hs)
3121*89c4ff92SAndroid Build Coastguard Worker {
3122*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(out);
3123*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(in_far);
3124*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(w);
3125*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(hs);
3126*89c4ff92SAndroid Build Coastguard Worker return in_near;
3127*89c4ff92SAndroid Build Coastguard Worker }
3128*89c4ff92SAndroid Build Coastguard Worker
stbi__resample_row_v_2(stbi_uc * out,stbi_uc * in_near,stbi_uc * in_far,int w,int hs)3129*89c4ff92SAndroid Build Coastguard Worker static stbi_uc* stbi__resample_row_v_2(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far, int w, int hs)
3130*89c4ff92SAndroid Build Coastguard Worker {
3131*89c4ff92SAndroid Build Coastguard Worker // need to generate two samples vertically for every one in input
3132*89c4ff92SAndroid Build Coastguard Worker int i;
3133*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(hs);
3134*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < w; ++i)
3135*89c4ff92SAndroid Build Coastguard Worker out[i] = stbi__div4(3*in_near[i] + in_far[i] + 2);
3136*89c4ff92SAndroid Build Coastguard Worker return out;
3137*89c4ff92SAndroid Build Coastguard Worker }
3138*89c4ff92SAndroid Build Coastguard Worker
stbi__resample_row_h_2(stbi_uc * out,stbi_uc * in_near,stbi_uc * in_far,int w,int hs)3139*89c4ff92SAndroid Build Coastguard Worker static stbi_uc* stbi__resample_row_h_2(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far, int w, int hs)
3140*89c4ff92SAndroid Build Coastguard Worker {
3141*89c4ff92SAndroid Build Coastguard Worker // need to generate two samples horizontally for every one in input
3142*89c4ff92SAndroid Build Coastguard Worker int i;
3143*89c4ff92SAndroid Build Coastguard Worker stbi_uc *input = in_near;
3144*89c4ff92SAndroid Build Coastguard Worker
3145*89c4ff92SAndroid Build Coastguard Worker if (w == 1) {
3146*89c4ff92SAndroid Build Coastguard Worker // if only one sample, can't do any interpolation
3147*89c4ff92SAndroid Build Coastguard Worker out[0] = out[1] = input[0];
3148*89c4ff92SAndroid Build Coastguard Worker return out;
3149*89c4ff92SAndroid Build Coastguard Worker }
3150*89c4ff92SAndroid Build Coastguard Worker
3151*89c4ff92SAndroid Build Coastguard Worker out[0] = input[0];
3152*89c4ff92SAndroid Build Coastguard Worker out[1] = stbi__div4(input[0]*3 + input[1] + 2);
3153*89c4ff92SAndroid Build Coastguard Worker for (i=1; i < w-1; ++i) {
3154*89c4ff92SAndroid Build Coastguard Worker int n = 3*input[i]+2;
3155*89c4ff92SAndroid Build Coastguard Worker out[i*2+0] = stbi__div4(n+input[i-1]);
3156*89c4ff92SAndroid Build Coastguard Worker out[i*2+1] = stbi__div4(n+input[i+1]);
3157*89c4ff92SAndroid Build Coastguard Worker }
3158*89c4ff92SAndroid Build Coastguard Worker out[i*2+0] = stbi__div4(input[w-2]*3 + input[w-1] + 2);
3159*89c4ff92SAndroid Build Coastguard Worker out[i*2+1] = input[w-1];
3160*89c4ff92SAndroid Build Coastguard Worker
3161*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(in_far);
3162*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(hs);
3163*89c4ff92SAndroid Build Coastguard Worker
3164*89c4ff92SAndroid Build Coastguard Worker return out;
3165*89c4ff92SAndroid Build Coastguard Worker }
3166*89c4ff92SAndroid Build Coastguard Worker
3167*89c4ff92SAndroid Build Coastguard Worker #define stbi__div16(x) ((stbi_uc) ((x) >> 4))
3168*89c4ff92SAndroid Build Coastguard Worker
stbi__resample_row_hv_2(stbi_uc * out,stbi_uc * in_near,stbi_uc * in_far,int w,int hs)3169*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__resample_row_hv_2(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far, int w, int hs)
3170*89c4ff92SAndroid Build Coastguard Worker {
3171*89c4ff92SAndroid Build Coastguard Worker // need to generate 2x2 samples for every one in input
3172*89c4ff92SAndroid Build Coastguard Worker int i,t0,t1;
3173*89c4ff92SAndroid Build Coastguard Worker if (w == 1) {
3174*89c4ff92SAndroid Build Coastguard Worker out[0] = out[1] = stbi__div4(3*in_near[0] + in_far[0] + 2);
3175*89c4ff92SAndroid Build Coastguard Worker return out;
3176*89c4ff92SAndroid Build Coastguard Worker }
3177*89c4ff92SAndroid Build Coastguard Worker
3178*89c4ff92SAndroid Build Coastguard Worker t1 = 3*in_near[0] + in_far[0];
3179*89c4ff92SAndroid Build Coastguard Worker out[0] = stbi__div4(t1+2);
3180*89c4ff92SAndroid Build Coastguard Worker for (i=1; i < w; ++i) {
3181*89c4ff92SAndroid Build Coastguard Worker t0 = t1;
3182*89c4ff92SAndroid Build Coastguard Worker t1 = 3*in_near[i]+in_far[i];
3183*89c4ff92SAndroid Build Coastguard Worker out[i*2-1] = stbi__div16(3*t0 + t1 + 8);
3184*89c4ff92SAndroid Build Coastguard Worker out[i*2 ] = stbi__div16(3*t1 + t0 + 8);
3185*89c4ff92SAndroid Build Coastguard Worker }
3186*89c4ff92SAndroid Build Coastguard Worker out[w*2-1] = stbi__div4(t1+2);
3187*89c4ff92SAndroid Build Coastguard Worker
3188*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(hs);
3189*89c4ff92SAndroid Build Coastguard Worker
3190*89c4ff92SAndroid Build Coastguard Worker return out;
3191*89c4ff92SAndroid Build Coastguard Worker }
3192*89c4ff92SAndroid Build Coastguard Worker
3193*89c4ff92SAndroid Build Coastguard Worker #if defined(STBI_SSE2) || defined(STBI_NEON)
stbi__resample_row_hv_2_simd(stbi_uc * out,stbi_uc * in_near,stbi_uc * in_far,int w,int hs)3194*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__resample_row_hv_2_simd(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far, int w, int hs)
3195*89c4ff92SAndroid Build Coastguard Worker {
3196*89c4ff92SAndroid Build Coastguard Worker // need to generate 2x2 samples for every one in input
3197*89c4ff92SAndroid Build Coastguard Worker int i=0,t0,t1;
3198*89c4ff92SAndroid Build Coastguard Worker
3199*89c4ff92SAndroid Build Coastguard Worker if (w == 1) {
3200*89c4ff92SAndroid Build Coastguard Worker out[0] = out[1] = stbi__div4(3*in_near[0] + in_far[0] + 2);
3201*89c4ff92SAndroid Build Coastguard Worker return out;
3202*89c4ff92SAndroid Build Coastguard Worker }
3203*89c4ff92SAndroid Build Coastguard Worker
3204*89c4ff92SAndroid Build Coastguard Worker t1 = 3*in_near[0] + in_far[0];
3205*89c4ff92SAndroid Build Coastguard Worker // process groups of 8 pixels for as long as we can.
3206*89c4ff92SAndroid Build Coastguard Worker // note we can't handle the last pixel in a row in this loop
3207*89c4ff92SAndroid Build Coastguard Worker // because we need to handle the filter boundary conditions.
3208*89c4ff92SAndroid Build Coastguard Worker for (; i < ((w-1) & ~7); i += 8) {
3209*89c4ff92SAndroid Build Coastguard Worker #if defined(STBI_SSE2)
3210*89c4ff92SAndroid Build Coastguard Worker // load and perform the vertical filtering pass
3211*89c4ff92SAndroid Build Coastguard Worker // this uses 3*x + y = 4*x + (y - x)
3212*89c4ff92SAndroid Build Coastguard Worker __m128i zero = _mm_setzero_si128();
3213*89c4ff92SAndroid Build Coastguard Worker __m128i farb = _mm_loadl_epi64((__m128i *) (in_far + i));
3214*89c4ff92SAndroid Build Coastguard Worker __m128i nearb = _mm_loadl_epi64((__m128i *) (in_near + i));
3215*89c4ff92SAndroid Build Coastguard Worker __m128i farw = _mm_unpacklo_epi8(farb, zero);
3216*89c4ff92SAndroid Build Coastguard Worker __m128i nearw = _mm_unpacklo_epi8(nearb, zero);
3217*89c4ff92SAndroid Build Coastguard Worker __m128i diff = _mm_sub_epi16(farw, nearw);
3218*89c4ff92SAndroid Build Coastguard Worker __m128i nears = _mm_slli_epi16(nearw, 2);
3219*89c4ff92SAndroid Build Coastguard Worker __m128i curr = _mm_add_epi16(nears, diff); // current row
3220*89c4ff92SAndroid Build Coastguard Worker
3221*89c4ff92SAndroid Build Coastguard Worker // horizontal filter works the same based on shifted vers of current
3222*89c4ff92SAndroid Build Coastguard Worker // row. "prev" is current row shifted right by 1 pixel; we need to
3223*89c4ff92SAndroid Build Coastguard Worker // insert the previous pixel value (from t1).
3224*89c4ff92SAndroid Build Coastguard Worker // "next" is current row shifted left by 1 pixel, with first pixel
3225*89c4ff92SAndroid Build Coastguard Worker // of next block of 8 pixels added in.
3226*89c4ff92SAndroid Build Coastguard Worker __m128i prv0 = _mm_slli_si128(curr, 2);
3227*89c4ff92SAndroid Build Coastguard Worker __m128i nxt0 = _mm_srli_si128(curr, 2);
3228*89c4ff92SAndroid Build Coastguard Worker __m128i prev = _mm_insert_epi16(prv0, t1, 0);
3229*89c4ff92SAndroid Build Coastguard Worker __m128i next = _mm_insert_epi16(nxt0, 3*in_near[i+8] + in_far[i+8], 7);
3230*89c4ff92SAndroid Build Coastguard Worker
3231*89c4ff92SAndroid Build Coastguard Worker // horizontal filter, polyphase implementation since it's convenient:
3232*89c4ff92SAndroid Build Coastguard Worker // even pixels = 3*cur + prev = cur*4 + (prev - cur)
3233*89c4ff92SAndroid Build Coastguard Worker // odd pixels = 3*cur + next = cur*4 + (next - cur)
3234*89c4ff92SAndroid Build Coastguard Worker // note the shared term.
3235*89c4ff92SAndroid Build Coastguard Worker __m128i bias = _mm_set1_epi16(8);
3236*89c4ff92SAndroid Build Coastguard Worker __m128i curs = _mm_slli_epi16(curr, 2);
3237*89c4ff92SAndroid Build Coastguard Worker __m128i prvd = _mm_sub_epi16(prev, curr);
3238*89c4ff92SAndroid Build Coastguard Worker __m128i nxtd = _mm_sub_epi16(next, curr);
3239*89c4ff92SAndroid Build Coastguard Worker __m128i curb = _mm_add_epi16(curs, bias);
3240*89c4ff92SAndroid Build Coastguard Worker __m128i even = _mm_add_epi16(prvd, curb);
3241*89c4ff92SAndroid Build Coastguard Worker __m128i odd = _mm_add_epi16(nxtd, curb);
3242*89c4ff92SAndroid Build Coastguard Worker
3243*89c4ff92SAndroid Build Coastguard Worker // interleave even and odd pixels, then undo scaling.
3244*89c4ff92SAndroid Build Coastguard Worker __m128i int0 = _mm_unpacklo_epi16(even, odd);
3245*89c4ff92SAndroid Build Coastguard Worker __m128i int1 = _mm_unpackhi_epi16(even, odd);
3246*89c4ff92SAndroid Build Coastguard Worker __m128i de0 = _mm_srli_epi16(int0, 4);
3247*89c4ff92SAndroid Build Coastguard Worker __m128i de1 = _mm_srli_epi16(int1, 4);
3248*89c4ff92SAndroid Build Coastguard Worker
3249*89c4ff92SAndroid Build Coastguard Worker // pack and write output
3250*89c4ff92SAndroid Build Coastguard Worker __m128i outv = _mm_packus_epi16(de0, de1);
3251*89c4ff92SAndroid Build Coastguard Worker _mm_storeu_si128((__m128i *) (out + i*2), outv);
3252*89c4ff92SAndroid Build Coastguard Worker #elif defined(STBI_NEON)
3253*89c4ff92SAndroid Build Coastguard Worker // load and perform the vertical filtering pass
3254*89c4ff92SAndroid Build Coastguard Worker // this uses 3*x + y = 4*x + (y - x)
3255*89c4ff92SAndroid Build Coastguard Worker uint8x8_t farb = vld1_u8(in_far + i);
3256*89c4ff92SAndroid Build Coastguard Worker uint8x8_t nearb = vld1_u8(in_near + i);
3257*89c4ff92SAndroid Build Coastguard Worker int16x8_t diff = vreinterpretq_s16_u16(vsubl_u8(farb, nearb));
3258*89c4ff92SAndroid Build Coastguard Worker int16x8_t nears = vreinterpretq_s16_u16(vshll_n_u8(nearb, 2));
3259*89c4ff92SAndroid Build Coastguard Worker int16x8_t curr = vaddq_s16(nears, diff); // current row
3260*89c4ff92SAndroid Build Coastguard Worker
3261*89c4ff92SAndroid Build Coastguard Worker // horizontal filter works the same based on shifted vers of current
3262*89c4ff92SAndroid Build Coastguard Worker // row. "prev" is current row shifted right by 1 pixel; we need to
3263*89c4ff92SAndroid Build Coastguard Worker // insert the previous pixel value (from t1).
3264*89c4ff92SAndroid Build Coastguard Worker // "next" is current row shifted left by 1 pixel, with first pixel
3265*89c4ff92SAndroid Build Coastguard Worker // of next block of 8 pixels added in.
3266*89c4ff92SAndroid Build Coastguard Worker int16x8_t prv0 = vextq_s16(curr, curr, 7);
3267*89c4ff92SAndroid Build Coastguard Worker int16x8_t nxt0 = vextq_s16(curr, curr, 1);
3268*89c4ff92SAndroid Build Coastguard Worker int16x8_t prev = vsetq_lane_s16(t1, prv0, 0);
3269*89c4ff92SAndroid Build Coastguard Worker int16x8_t next = vsetq_lane_s16(3*in_near[i+8] + in_far[i+8], nxt0, 7);
3270*89c4ff92SAndroid Build Coastguard Worker
3271*89c4ff92SAndroid Build Coastguard Worker // horizontal filter, polyphase implementation since it's convenient:
3272*89c4ff92SAndroid Build Coastguard Worker // even pixels = 3*cur + prev = cur*4 + (prev - cur)
3273*89c4ff92SAndroid Build Coastguard Worker // odd pixels = 3*cur + next = cur*4 + (next - cur)
3274*89c4ff92SAndroid Build Coastguard Worker // note the shared term.
3275*89c4ff92SAndroid Build Coastguard Worker int16x8_t curs = vshlq_n_s16(curr, 2);
3276*89c4ff92SAndroid Build Coastguard Worker int16x8_t prvd = vsubq_s16(prev, curr);
3277*89c4ff92SAndroid Build Coastguard Worker int16x8_t nxtd = vsubq_s16(next, curr);
3278*89c4ff92SAndroid Build Coastguard Worker int16x8_t even = vaddq_s16(curs, prvd);
3279*89c4ff92SAndroid Build Coastguard Worker int16x8_t odd = vaddq_s16(curs, nxtd);
3280*89c4ff92SAndroid Build Coastguard Worker
3281*89c4ff92SAndroid Build Coastguard Worker // undo scaling and round, then store with even/odd phases interleaved
3282*89c4ff92SAndroid Build Coastguard Worker uint8x8x2_t o;
3283*89c4ff92SAndroid Build Coastguard Worker o.val[0] = vqrshrun_n_s16(even, 4);
3284*89c4ff92SAndroid Build Coastguard Worker o.val[1] = vqrshrun_n_s16(odd, 4);
3285*89c4ff92SAndroid Build Coastguard Worker vst2_u8(out + i*2, o);
3286*89c4ff92SAndroid Build Coastguard Worker #endif
3287*89c4ff92SAndroid Build Coastguard Worker
3288*89c4ff92SAndroid Build Coastguard Worker // "previous" value for next iter
3289*89c4ff92SAndroid Build Coastguard Worker t1 = 3*in_near[i+7] + in_far[i+7];
3290*89c4ff92SAndroid Build Coastguard Worker }
3291*89c4ff92SAndroid Build Coastguard Worker
3292*89c4ff92SAndroid Build Coastguard Worker t0 = t1;
3293*89c4ff92SAndroid Build Coastguard Worker t1 = 3*in_near[i] + in_far[i];
3294*89c4ff92SAndroid Build Coastguard Worker out[i*2] = stbi__div16(3*t1 + t0 + 8);
3295*89c4ff92SAndroid Build Coastguard Worker
3296*89c4ff92SAndroid Build Coastguard Worker for (++i; i < w; ++i) {
3297*89c4ff92SAndroid Build Coastguard Worker t0 = t1;
3298*89c4ff92SAndroid Build Coastguard Worker t1 = 3*in_near[i]+in_far[i];
3299*89c4ff92SAndroid Build Coastguard Worker out[i*2-1] = stbi__div16(3*t0 + t1 + 8);
3300*89c4ff92SAndroid Build Coastguard Worker out[i*2 ] = stbi__div16(3*t1 + t0 + 8);
3301*89c4ff92SAndroid Build Coastguard Worker }
3302*89c4ff92SAndroid Build Coastguard Worker out[w*2-1] = stbi__div4(t1+2);
3303*89c4ff92SAndroid Build Coastguard Worker
3304*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(hs);
3305*89c4ff92SAndroid Build Coastguard Worker
3306*89c4ff92SAndroid Build Coastguard Worker return out;
3307*89c4ff92SAndroid Build Coastguard Worker }
3308*89c4ff92SAndroid Build Coastguard Worker #endif
3309*89c4ff92SAndroid Build Coastguard Worker
stbi__resample_row_generic(stbi_uc * out,stbi_uc * in_near,stbi_uc * in_far,int w,int hs)3310*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__resample_row_generic(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far, int w, int hs)
3311*89c4ff92SAndroid Build Coastguard Worker {
3312*89c4ff92SAndroid Build Coastguard Worker // resample with nearest-neighbor
3313*89c4ff92SAndroid Build Coastguard Worker int i,j;
3314*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(in_far);
3315*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < w; ++i)
3316*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < hs; ++j)
3317*89c4ff92SAndroid Build Coastguard Worker out[i*hs+j] = in_near[i];
3318*89c4ff92SAndroid Build Coastguard Worker return out;
3319*89c4ff92SAndroid Build Coastguard Worker }
3320*89c4ff92SAndroid Build Coastguard Worker
3321*89c4ff92SAndroid Build Coastguard Worker // this is a reduced-precision calculation of YCbCr-to-RGB introduced
3322*89c4ff92SAndroid Build Coastguard Worker // to make sure the code produces the same results in both SIMD and scalar
3323*89c4ff92SAndroid Build Coastguard Worker #define stbi__float2fixed(x) (((int) ((x) * 4096.0f + 0.5f)) << 8)
stbi__YCbCr_to_RGB_row(stbi_uc * out,const stbi_uc * y,const stbi_uc * pcb,const stbi_uc * pcr,int count,int step)3324*89c4ff92SAndroid Build Coastguard Worker static void stbi__YCbCr_to_RGB_row(stbi_uc *out, const stbi_uc *y, const stbi_uc *pcb, const stbi_uc *pcr, int count, int step)
3325*89c4ff92SAndroid Build Coastguard Worker {
3326*89c4ff92SAndroid Build Coastguard Worker int i;
3327*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < count; ++i) {
3328*89c4ff92SAndroid Build Coastguard Worker int y_fixed = (y[i] << 20) + (1<<19); // rounding
3329*89c4ff92SAndroid Build Coastguard Worker int r,g,b;
3330*89c4ff92SAndroid Build Coastguard Worker int cr = pcr[i] - 128;
3331*89c4ff92SAndroid Build Coastguard Worker int cb = pcb[i] - 128;
3332*89c4ff92SAndroid Build Coastguard Worker r = y_fixed + cr* stbi__float2fixed(1.40200f);
3333*89c4ff92SAndroid Build Coastguard Worker g = y_fixed + (cr*-stbi__float2fixed(0.71414f)) + ((cb*-stbi__float2fixed(0.34414f)) & 0xffff0000);
3334*89c4ff92SAndroid Build Coastguard Worker b = y_fixed + cb* stbi__float2fixed(1.77200f);
3335*89c4ff92SAndroid Build Coastguard Worker r >>= 20;
3336*89c4ff92SAndroid Build Coastguard Worker g >>= 20;
3337*89c4ff92SAndroid Build Coastguard Worker b >>= 20;
3338*89c4ff92SAndroid Build Coastguard Worker if ((unsigned) r > 255) { if (r < 0) r = 0; else r = 255; }
3339*89c4ff92SAndroid Build Coastguard Worker if ((unsigned) g > 255) { if (g < 0) g = 0; else g = 255; }
3340*89c4ff92SAndroid Build Coastguard Worker if ((unsigned) b > 255) { if (b < 0) b = 0; else b = 255; }
3341*89c4ff92SAndroid Build Coastguard Worker out[0] = (stbi_uc)r;
3342*89c4ff92SAndroid Build Coastguard Worker out[1] = (stbi_uc)g;
3343*89c4ff92SAndroid Build Coastguard Worker out[2] = (stbi_uc)b;
3344*89c4ff92SAndroid Build Coastguard Worker out[3] = 255;
3345*89c4ff92SAndroid Build Coastguard Worker out += step;
3346*89c4ff92SAndroid Build Coastguard Worker }
3347*89c4ff92SAndroid Build Coastguard Worker }
3348*89c4ff92SAndroid Build Coastguard Worker
3349*89c4ff92SAndroid Build Coastguard Worker #if defined(STBI_SSE2) || defined(STBI_NEON)
stbi__YCbCr_to_RGB_simd(stbi_uc * out,stbi_uc const * y,stbi_uc const * pcb,stbi_uc const * pcr,int count,int step)3350*89c4ff92SAndroid Build Coastguard Worker static void stbi__YCbCr_to_RGB_simd(stbi_uc *out, stbi_uc const *y, stbi_uc const *pcb, stbi_uc const *pcr, int count, int step)
3351*89c4ff92SAndroid Build Coastguard Worker {
3352*89c4ff92SAndroid Build Coastguard Worker int i = 0;
3353*89c4ff92SAndroid Build Coastguard Worker
3354*89c4ff92SAndroid Build Coastguard Worker #ifdef STBI_SSE2
3355*89c4ff92SAndroid Build Coastguard Worker // step == 3 is pretty ugly on the final interleave, and i'm not convinced
3356*89c4ff92SAndroid Build Coastguard Worker // it's useful in practice (you wouldn't use it for textures, for example).
3357*89c4ff92SAndroid Build Coastguard Worker // so just accelerate step == 4 case.
3358*89c4ff92SAndroid Build Coastguard Worker if (step == 4) {
3359*89c4ff92SAndroid Build Coastguard Worker // this is a fairly straightforward implementation and not super-optimized.
3360*89c4ff92SAndroid Build Coastguard Worker __m128i signflip = _mm_set1_epi8(-0x80);
3361*89c4ff92SAndroid Build Coastguard Worker __m128i cr_const0 = _mm_set1_epi16( (short) ( 1.40200f*4096.0f+0.5f));
3362*89c4ff92SAndroid Build Coastguard Worker __m128i cr_const1 = _mm_set1_epi16( - (short) ( 0.71414f*4096.0f+0.5f));
3363*89c4ff92SAndroid Build Coastguard Worker __m128i cb_const0 = _mm_set1_epi16( - (short) ( 0.34414f*4096.0f+0.5f));
3364*89c4ff92SAndroid Build Coastguard Worker __m128i cb_const1 = _mm_set1_epi16( (short) ( 1.77200f*4096.0f+0.5f));
3365*89c4ff92SAndroid Build Coastguard Worker __m128i y_bias = _mm_set1_epi8((char) (unsigned char) 128);
3366*89c4ff92SAndroid Build Coastguard Worker __m128i xw = _mm_set1_epi16(255); // alpha channel
3367*89c4ff92SAndroid Build Coastguard Worker
3368*89c4ff92SAndroid Build Coastguard Worker for (; i+7 < count; i += 8) {
3369*89c4ff92SAndroid Build Coastguard Worker // load
3370*89c4ff92SAndroid Build Coastguard Worker __m128i y_bytes = _mm_loadl_epi64((__m128i *) (y+i));
3371*89c4ff92SAndroid Build Coastguard Worker __m128i cr_bytes = _mm_loadl_epi64((__m128i *) (pcr+i));
3372*89c4ff92SAndroid Build Coastguard Worker __m128i cb_bytes = _mm_loadl_epi64((__m128i *) (pcb+i));
3373*89c4ff92SAndroid Build Coastguard Worker __m128i cr_biased = _mm_xor_si128(cr_bytes, signflip); // -128
3374*89c4ff92SAndroid Build Coastguard Worker __m128i cb_biased = _mm_xor_si128(cb_bytes, signflip); // -128
3375*89c4ff92SAndroid Build Coastguard Worker
3376*89c4ff92SAndroid Build Coastguard Worker // unpack to short (and left-shift cr, cb by 8)
3377*89c4ff92SAndroid Build Coastguard Worker __m128i yw = _mm_unpacklo_epi8(y_bias, y_bytes);
3378*89c4ff92SAndroid Build Coastguard Worker __m128i crw = _mm_unpacklo_epi8(_mm_setzero_si128(), cr_biased);
3379*89c4ff92SAndroid Build Coastguard Worker __m128i cbw = _mm_unpacklo_epi8(_mm_setzero_si128(), cb_biased);
3380*89c4ff92SAndroid Build Coastguard Worker
3381*89c4ff92SAndroid Build Coastguard Worker // color transform
3382*89c4ff92SAndroid Build Coastguard Worker __m128i yws = _mm_srli_epi16(yw, 4);
3383*89c4ff92SAndroid Build Coastguard Worker __m128i cr0 = _mm_mulhi_epi16(cr_const0, crw);
3384*89c4ff92SAndroid Build Coastguard Worker __m128i cb0 = _mm_mulhi_epi16(cb_const0, cbw);
3385*89c4ff92SAndroid Build Coastguard Worker __m128i cb1 = _mm_mulhi_epi16(cbw, cb_const1);
3386*89c4ff92SAndroid Build Coastguard Worker __m128i cr1 = _mm_mulhi_epi16(crw, cr_const1);
3387*89c4ff92SAndroid Build Coastguard Worker __m128i rws = _mm_add_epi16(cr0, yws);
3388*89c4ff92SAndroid Build Coastguard Worker __m128i gwt = _mm_add_epi16(cb0, yws);
3389*89c4ff92SAndroid Build Coastguard Worker __m128i bws = _mm_add_epi16(yws, cb1);
3390*89c4ff92SAndroid Build Coastguard Worker __m128i gws = _mm_add_epi16(gwt, cr1);
3391*89c4ff92SAndroid Build Coastguard Worker
3392*89c4ff92SAndroid Build Coastguard Worker // descale
3393*89c4ff92SAndroid Build Coastguard Worker __m128i rw = _mm_srai_epi16(rws, 4);
3394*89c4ff92SAndroid Build Coastguard Worker __m128i bw = _mm_srai_epi16(bws, 4);
3395*89c4ff92SAndroid Build Coastguard Worker __m128i gw = _mm_srai_epi16(gws, 4);
3396*89c4ff92SAndroid Build Coastguard Worker
3397*89c4ff92SAndroid Build Coastguard Worker // back to byte, set up for transpose
3398*89c4ff92SAndroid Build Coastguard Worker __m128i brb = _mm_packus_epi16(rw, bw);
3399*89c4ff92SAndroid Build Coastguard Worker __m128i gxb = _mm_packus_epi16(gw, xw);
3400*89c4ff92SAndroid Build Coastguard Worker
3401*89c4ff92SAndroid Build Coastguard Worker // transpose to interleave channels
3402*89c4ff92SAndroid Build Coastguard Worker __m128i t0 = _mm_unpacklo_epi8(brb, gxb);
3403*89c4ff92SAndroid Build Coastguard Worker __m128i t1 = _mm_unpackhi_epi8(brb, gxb);
3404*89c4ff92SAndroid Build Coastguard Worker __m128i o0 = _mm_unpacklo_epi16(t0, t1);
3405*89c4ff92SAndroid Build Coastguard Worker __m128i o1 = _mm_unpackhi_epi16(t0, t1);
3406*89c4ff92SAndroid Build Coastguard Worker
3407*89c4ff92SAndroid Build Coastguard Worker // store
3408*89c4ff92SAndroid Build Coastguard Worker _mm_storeu_si128((__m128i *) (out + 0), o0);
3409*89c4ff92SAndroid Build Coastguard Worker _mm_storeu_si128((__m128i *) (out + 16), o1);
3410*89c4ff92SAndroid Build Coastguard Worker out += 32;
3411*89c4ff92SAndroid Build Coastguard Worker }
3412*89c4ff92SAndroid Build Coastguard Worker }
3413*89c4ff92SAndroid Build Coastguard Worker #endif
3414*89c4ff92SAndroid Build Coastguard Worker
3415*89c4ff92SAndroid Build Coastguard Worker #ifdef STBI_NEON
3416*89c4ff92SAndroid Build Coastguard Worker // in this version, step=3 support would be easy to add. but is there demand?
3417*89c4ff92SAndroid Build Coastguard Worker if (step == 4) {
3418*89c4ff92SAndroid Build Coastguard Worker // this is a fairly straightforward implementation and not super-optimized.
3419*89c4ff92SAndroid Build Coastguard Worker uint8x8_t signflip = vdup_n_u8(0x80);
3420*89c4ff92SAndroid Build Coastguard Worker int16x8_t cr_const0 = vdupq_n_s16( (short) ( 1.40200f*4096.0f+0.5f));
3421*89c4ff92SAndroid Build Coastguard Worker int16x8_t cr_const1 = vdupq_n_s16( - (short) ( 0.71414f*4096.0f+0.5f));
3422*89c4ff92SAndroid Build Coastguard Worker int16x8_t cb_const0 = vdupq_n_s16( - (short) ( 0.34414f*4096.0f+0.5f));
3423*89c4ff92SAndroid Build Coastguard Worker int16x8_t cb_const1 = vdupq_n_s16( (short) ( 1.77200f*4096.0f+0.5f));
3424*89c4ff92SAndroid Build Coastguard Worker
3425*89c4ff92SAndroid Build Coastguard Worker for (; i+7 < count; i += 8) {
3426*89c4ff92SAndroid Build Coastguard Worker // load
3427*89c4ff92SAndroid Build Coastguard Worker uint8x8_t y_bytes = vld1_u8(y + i);
3428*89c4ff92SAndroid Build Coastguard Worker uint8x8_t cr_bytes = vld1_u8(pcr + i);
3429*89c4ff92SAndroid Build Coastguard Worker uint8x8_t cb_bytes = vld1_u8(pcb + i);
3430*89c4ff92SAndroid Build Coastguard Worker int8x8_t cr_biased = vreinterpret_s8_u8(vsub_u8(cr_bytes, signflip));
3431*89c4ff92SAndroid Build Coastguard Worker int8x8_t cb_biased = vreinterpret_s8_u8(vsub_u8(cb_bytes, signflip));
3432*89c4ff92SAndroid Build Coastguard Worker
3433*89c4ff92SAndroid Build Coastguard Worker // expand to s16
3434*89c4ff92SAndroid Build Coastguard Worker int16x8_t yws = vreinterpretq_s16_u16(vshll_n_u8(y_bytes, 4));
3435*89c4ff92SAndroid Build Coastguard Worker int16x8_t crw = vshll_n_s8(cr_biased, 7);
3436*89c4ff92SAndroid Build Coastguard Worker int16x8_t cbw = vshll_n_s8(cb_biased, 7);
3437*89c4ff92SAndroid Build Coastguard Worker
3438*89c4ff92SAndroid Build Coastguard Worker // color transform
3439*89c4ff92SAndroid Build Coastguard Worker int16x8_t cr0 = vqdmulhq_s16(crw, cr_const0);
3440*89c4ff92SAndroid Build Coastguard Worker int16x8_t cb0 = vqdmulhq_s16(cbw, cb_const0);
3441*89c4ff92SAndroid Build Coastguard Worker int16x8_t cr1 = vqdmulhq_s16(crw, cr_const1);
3442*89c4ff92SAndroid Build Coastguard Worker int16x8_t cb1 = vqdmulhq_s16(cbw, cb_const1);
3443*89c4ff92SAndroid Build Coastguard Worker int16x8_t rws = vaddq_s16(yws, cr0);
3444*89c4ff92SAndroid Build Coastguard Worker int16x8_t gws = vaddq_s16(vaddq_s16(yws, cb0), cr1);
3445*89c4ff92SAndroid Build Coastguard Worker int16x8_t bws = vaddq_s16(yws, cb1);
3446*89c4ff92SAndroid Build Coastguard Worker
3447*89c4ff92SAndroid Build Coastguard Worker // undo scaling, round, convert to byte
3448*89c4ff92SAndroid Build Coastguard Worker uint8x8x4_t o;
3449*89c4ff92SAndroid Build Coastguard Worker o.val[0] = vqrshrun_n_s16(rws, 4);
3450*89c4ff92SAndroid Build Coastguard Worker o.val[1] = vqrshrun_n_s16(gws, 4);
3451*89c4ff92SAndroid Build Coastguard Worker o.val[2] = vqrshrun_n_s16(bws, 4);
3452*89c4ff92SAndroid Build Coastguard Worker o.val[3] = vdup_n_u8(255);
3453*89c4ff92SAndroid Build Coastguard Worker
3454*89c4ff92SAndroid Build Coastguard Worker // store, interleaving r/g/b/a
3455*89c4ff92SAndroid Build Coastguard Worker vst4_u8(out, o);
3456*89c4ff92SAndroid Build Coastguard Worker out += 8*4;
3457*89c4ff92SAndroid Build Coastguard Worker }
3458*89c4ff92SAndroid Build Coastguard Worker }
3459*89c4ff92SAndroid Build Coastguard Worker #endif
3460*89c4ff92SAndroid Build Coastguard Worker
3461*89c4ff92SAndroid Build Coastguard Worker for (; i < count; ++i) {
3462*89c4ff92SAndroid Build Coastguard Worker int y_fixed = (y[i] << 20) + (1<<19); // rounding
3463*89c4ff92SAndroid Build Coastguard Worker int r,g,b;
3464*89c4ff92SAndroid Build Coastguard Worker int cr = pcr[i] - 128;
3465*89c4ff92SAndroid Build Coastguard Worker int cb = pcb[i] - 128;
3466*89c4ff92SAndroid Build Coastguard Worker r = y_fixed + cr* stbi__float2fixed(1.40200f);
3467*89c4ff92SAndroid Build Coastguard Worker g = y_fixed + cr*-stbi__float2fixed(0.71414f) + ((cb*-stbi__float2fixed(0.34414f)) & 0xffff0000);
3468*89c4ff92SAndroid Build Coastguard Worker b = y_fixed + cb* stbi__float2fixed(1.77200f);
3469*89c4ff92SAndroid Build Coastguard Worker r >>= 20;
3470*89c4ff92SAndroid Build Coastguard Worker g >>= 20;
3471*89c4ff92SAndroid Build Coastguard Worker b >>= 20;
3472*89c4ff92SAndroid Build Coastguard Worker if ((unsigned) r > 255) { if (r < 0) r = 0; else r = 255; }
3473*89c4ff92SAndroid Build Coastguard Worker if ((unsigned) g > 255) { if (g < 0) g = 0; else g = 255; }
3474*89c4ff92SAndroid Build Coastguard Worker if ((unsigned) b > 255) { if (b < 0) b = 0; else b = 255; }
3475*89c4ff92SAndroid Build Coastguard Worker out[0] = (stbi_uc)r;
3476*89c4ff92SAndroid Build Coastguard Worker out[1] = (stbi_uc)g;
3477*89c4ff92SAndroid Build Coastguard Worker out[2] = (stbi_uc)b;
3478*89c4ff92SAndroid Build Coastguard Worker out[3] = 255;
3479*89c4ff92SAndroid Build Coastguard Worker out += step;
3480*89c4ff92SAndroid Build Coastguard Worker }
3481*89c4ff92SAndroid Build Coastguard Worker }
3482*89c4ff92SAndroid Build Coastguard Worker #endif
3483*89c4ff92SAndroid Build Coastguard Worker
3484*89c4ff92SAndroid Build Coastguard Worker // set up the kernels
stbi__setup_jpeg(stbi__jpeg * j)3485*89c4ff92SAndroid Build Coastguard Worker static void stbi__setup_jpeg(stbi__jpeg *j)
3486*89c4ff92SAndroid Build Coastguard Worker {
3487*89c4ff92SAndroid Build Coastguard Worker j->idct_block_kernel = stbi__idct_block;
3488*89c4ff92SAndroid Build Coastguard Worker j->YCbCr_to_RGB_kernel = stbi__YCbCr_to_RGB_row;
3489*89c4ff92SAndroid Build Coastguard Worker j->resample_row_hv_2_kernel = stbi__resample_row_hv_2;
3490*89c4ff92SAndroid Build Coastguard Worker
3491*89c4ff92SAndroid Build Coastguard Worker #ifdef STBI_SSE2
3492*89c4ff92SAndroid Build Coastguard Worker if (stbi__sse2_available()) {
3493*89c4ff92SAndroid Build Coastguard Worker j->idct_block_kernel = stbi__idct_simd;
3494*89c4ff92SAndroid Build Coastguard Worker j->YCbCr_to_RGB_kernel = stbi__YCbCr_to_RGB_simd;
3495*89c4ff92SAndroid Build Coastguard Worker j->resample_row_hv_2_kernel = stbi__resample_row_hv_2_simd;
3496*89c4ff92SAndroid Build Coastguard Worker }
3497*89c4ff92SAndroid Build Coastguard Worker #endif
3498*89c4ff92SAndroid Build Coastguard Worker
3499*89c4ff92SAndroid Build Coastguard Worker #ifdef STBI_NEON
3500*89c4ff92SAndroid Build Coastguard Worker j->idct_block_kernel = stbi__idct_simd;
3501*89c4ff92SAndroid Build Coastguard Worker j->YCbCr_to_RGB_kernel = stbi__YCbCr_to_RGB_simd;
3502*89c4ff92SAndroid Build Coastguard Worker j->resample_row_hv_2_kernel = stbi__resample_row_hv_2_simd;
3503*89c4ff92SAndroid Build Coastguard Worker #endif
3504*89c4ff92SAndroid Build Coastguard Worker }
3505*89c4ff92SAndroid Build Coastguard Worker
3506*89c4ff92SAndroid Build Coastguard Worker // clean up the temporary component buffers
stbi__cleanup_jpeg(stbi__jpeg * j)3507*89c4ff92SAndroid Build Coastguard Worker static void stbi__cleanup_jpeg(stbi__jpeg *j)
3508*89c4ff92SAndroid Build Coastguard Worker {
3509*89c4ff92SAndroid Build Coastguard Worker stbi__free_jpeg_components(j, j->s->img_n, 0);
3510*89c4ff92SAndroid Build Coastguard Worker }
3511*89c4ff92SAndroid Build Coastguard Worker
3512*89c4ff92SAndroid Build Coastguard Worker typedef struct
3513*89c4ff92SAndroid Build Coastguard Worker {
3514*89c4ff92SAndroid Build Coastguard Worker resample_row_func resample;
3515*89c4ff92SAndroid Build Coastguard Worker stbi_uc *line0,*line1;
3516*89c4ff92SAndroid Build Coastguard Worker int hs,vs; // expansion factor in each axis
3517*89c4ff92SAndroid Build Coastguard Worker int w_lores; // horizontal pixels pre-expansion
3518*89c4ff92SAndroid Build Coastguard Worker int ystep; // how far through vertical expansion we are
3519*89c4ff92SAndroid Build Coastguard Worker int ypos; // which pre-expansion row we're on
3520*89c4ff92SAndroid Build Coastguard Worker } stbi__resample;
3521*89c4ff92SAndroid Build Coastguard Worker
3522*89c4ff92SAndroid Build Coastguard Worker // fast 0..255 * 0..255 => 0..255 rounded multiplication
stbi__blinn_8x8(stbi_uc x,stbi_uc y)3523*89c4ff92SAndroid Build Coastguard Worker static stbi_uc stbi__blinn_8x8(stbi_uc x, stbi_uc y)
3524*89c4ff92SAndroid Build Coastguard Worker {
3525*89c4ff92SAndroid Build Coastguard Worker unsigned int t = x*y + 128;
3526*89c4ff92SAndroid Build Coastguard Worker return (stbi_uc) ((t + (t >>8)) >> 8);
3527*89c4ff92SAndroid Build Coastguard Worker }
3528*89c4ff92SAndroid Build Coastguard Worker
load_jpeg_image(stbi__jpeg * z,int * out_x,int * out_y,int * comp,int req_comp)3529*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *load_jpeg_image(stbi__jpeg *z, int *out_x, int *out_y, int *comp, int req_comp)
3530*89c4ff92SAndroid Build Coastguard Worker {
3531*89c4ff92SAndroid Build Coastguard Worker int n, decode_n, is_rgb;
3532*89c4ff92SAndroid Build Coastguard Worker z->s->img_n = 0; // make stbi__cleanup_jpeg safe
3533*89c4ff92SAndroid Build Coastguard Worker
3534*89c4ff92SAndroid Build Coastguard Worker // validate req_comp
3535*89c4ff92SAndroid Build Coastguard Worker if (req_comp < 0 || req_comp > 4) return stbi__errpuc("bad req_comp", "Internal error");
3536*89c4ff92SAndroid Build Coastguard Worker
3537*89c4ff92SAndroid Build Coastguard Worker // load a jpeg image from whichever source, but leave in YCbCr format
3538*89c4ff92SAndroid Build Coastguard Worker if (!stbi__decode_jpeg_image(z)) { stbi__cleanup_jpeg(z); return NULL; }
3539*89c4ff92SAndroid Build Coastguard Worker
3540*89c4ff92SAndroid Build Coastguard Worker // determine actual number of components to generate
3541*89c4ff92SAndroid Build Coastguard Worker n = req_comp ? req_comp : z->s->img_n >= 3 ? 3 : 1;
3542*89c4ff92SAndroid Build Coastguard Worker
3543*89c4ff92SAndroid Build Coastguard Worker is_rgb = z->s->img_n == 3 && (z->rgb == 3 || (z->app14_color_transform == 0 && !z->jfif));
3544*89c4ff92SAndroid Build Coastguard Worker
3545*89c4ff92SAndroid Build Coastguard Worker if (z->s->img_n == 3 && n < 3 && !is_rgb)
3546*89c4ff92SAndroid Build Coastguard Worker decode_n = 1;
3547*89c4ff92SAndroid Build Coastguard Worker else
3548*89c4ff92SAndroid Build Coastguard Worker decode_n = z->s->img_n;
3549*89c4ff92SAndroid Build Coastguard Worker
3550*89c4ff92SAndroid Build Coastguard Worker // resample and color-convert
3551*89c4ff92SAndroid Build Coastguard Worker {
3552*89c4ff92SAndroid Build Coastguard Worker int k;
3553*89c4ff92SAndroid Build Coastguard Worker unsigned int i,j;
3554*89c4ff92SAndroid Build Coastguard Worker stbi_uc *output;
3555*89c4ff92SAndroid Build Coastguard Worker stbi_uc *coutput[4];
3556*89c4ff92SAndroid Build Coastguard Worker
3557*89c4ff92SAndroid Build Coastguard Worker stbi__resample res_comp[4];
3558*89c4ff92SAndroid Build Coastguard Worker
3559*89c4ff92SAndroid Build Coastguard Worker for (k=0; k < decode_n; ++k) {
3560*89c4ff92SAndroid Build Coastguard Worker stbi__resample *r = &res_comp[k];
3561*89c4ff92SAndroid Build Coastguard Worker
3562*89c4ff92SAndroid Build Coastguard Worker // allocate line buffer big enough for upsampling off the edges
3563*89c4ff92SAndroid Build Coastguard Worker // with upsample factor of 4
3564*89c4ff92SAndroid Build Coastguard Worker z->img_comp[k].linebuf = (stbi_uc *) stbi__malloc(z->s->img_x + 3);
3565*89c4ff92SAndroid Build Coastguard Worker if (!z->img_comp[k].linebuf) { stbi__cleanup_jpeg(z); return stbi__errpuc("outofmem", "Out of memory"); }
3566*89c4ff92SAndroid Build Coastguard Worker
3567*89c4ff92SAndroid Build Coastguard Worker r->hs = z->img_h_max / z->img_comp[k].h;
3568*89c4ff92SAndroid Build Coastguard Worker r->vs = z->img_v_max / z->img_comp[k].v;
3569*89c4ff92SAndroid Build Coastguard Worker r->ystep = r->vs >> 1;
3570*89c4ff92SAndroid Build Coastguard Worker r->w_lores = (z->s->img_x + r->hs-1) / r->hs;
3571*89c4ff92SAndroid Build Coastguard Worker r->ypos = 0;
3572*89c4ff92SAndroid Build Coastguard Worker r->line0 = r->line1 = z->img_comp[k].data;
3573*89c4ff92SAndroid Build Coastguard Worker
3574*89c4ff92SAndroid Build Coastguard Worker if (r->hs == 1 && r->vs == 1) r->resample = resample_row_1;
3575*89c4ff92SAndroid Build Coastguard Worker else if (r->hs == 1 && r->vs == 2) r->resample = stbi__resample_row_v_2;
3576*89c4ff92SAndroid Build Coastguard Worker else if (r->hs == 2 && r->vs == 1) r->resample = stbi__resample_row_h_2;
3577*89c4ff92SAndroid Build Coastguard Worker else if (r->hs == 2 && r->vs == 2) r->resample = z->resample_row_hv_2_kernel;
3578*89c4ff92SAndroid Build Coastguard Worker else r->resample = stbi__resample_row_generic;
3579*89c4ff92SAndroid Build Coastguard Worker }
3580*89c4ff92SAndroid Build Coastguard Worker
3581*89c4ff92SAndroid Build Coastguard Worker // can't error after this so, this is safe
3582*89c4ff92SAndroid Build Coastguard Worker output = (stbi_uc *) stbi__malloc_mad3(n, z->s->img_x, z->s->img_y, 1);
3583*89c4ff92SAndroid Build Coastguard Worker if (!output) { stbi__cleanup_jpeg(z); return stbi__errpuc("outofmem", "Out of memory"); }
3584*89c4ff92SAndroid Build Coastguard Worker
3585*89c4ff92SAndroid Build Coastguard Worker // now go ahead and resample
3586*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < z->s->img_y; ++j) {
3587*89c4ff92SAndroid Build Coastguard Worker stbi_uc *out = output + n * z->s->img_x * j;
3588*89c4ff92SAndroid Build Coastguard Worker for (k=0; k < decode_n; ++k) {
3589*89c4ff92SAndroid Build Coastguard Worker stbi__resample *r = &res_comp[k];
3590*89c4ff92SAndroid Build Coastguard Worker int y_bot = r->ystep >= (r->vs >> 1);
3591*89c4ff92SAndroid Build Coastguard Worker coutput[k] = r->resample(z->img_comp[k].linebuf,
3592*89c4ff92SAndroid Build Coastguard Worker y_bot ? r->line1 : r->line0,
3593*89c4ff92SAndroid Build Coastguard Worker y_bot ? r->line0 : r->line1,
3594*89c4ff92SAndroid Build Coastguard Worker r->w_lores, r->hs);
3595*89c4ff92SAndroid Build Coastguard Worker if (++r->ystep >= r->vs) {
3596*89c4ff92SAndroid Build Coastguard Worker r->ystep = 0;
3597*89c4ff92SAndroid Build Coastguard Worker r->line0 = r->line1;
3598*89c4ff92SAndroid Build Coastguard Worker if (++r->ypos < z->img_comp[k].y)
3599*89c4ff92SAndroid Build Coastguard Worker r->line1 += z->img_comp[k].w2;
3600*89c4ff92SAndroid Build Coastguard Worker }
3601*89c4ff92SAndroid Build Coastguard Worker }
3602*89c4ff92SAndroid Build Coastguard Worker if (n >= 3) {
3603*89c4ff92SAndroid Build Coastguard Worker stbi_uc *y = coutput[0];
3604*89c4ff92SAndroid Build Coastguard Worker if (z->s->img_n == 3) {
3605*89c4ff92SAndroid Build Coastguard Worker if (is_rgb) {
3606*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i) {
3607*89c4ff92SAndroid Build Coastguard Worker out[0] = y[i];
3608*89c4ff92SAndroid Build Coastguard Worker out[1] = coutput[1][i];
3609*89c4ff92SAndroid Build Coastguard Worker out[2] = coutput[2][i];
3610*89c4ff92SAndroid Build Coastguard Worker out[3] = 255;
3611*89c4ff92SAndroid Build Coastguard Worker out += n;
3612*89c4ff92SAndroid Build Coastguard Worker }
3613*89c4ff92SAndroid Build Coastguard Worker } else {
3614*89c4ff92SAndroid Build Coastguard Worker z->YCbCr_to_RGB_kernel(out, y, coutput[1], coutput[2], z->s->img_x, n);
3615*89c4ff92SAndroid Build Coastguard Worker }
3616*89c4ff92SAndroid Build Coastguard Worker } else if (z->s->img_n == 4) {
3617*89c4ff92SAndroid Build Coastguard Worker if (z->app14_color_transform == 0) { // CMYK
3618*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i) {
3619*89c4ff92SAndroid Build Coastguard Worker stbi_uc m = coutput[3][i];
3620*89c4ff92SAndroid Build Coastguard Worker out[0] = stbi__blinn_8x8(coutput[0][i], m);
3621*89c4ff92SAndroid Build Coastguard Worker out[1] = stbi__blinn_8x8(coutput[1][i], m);
3622*89c4ff92SAndroid Build Coastguard Worker out[2] = stbi__blinn_8x8(coutput[2][i], m);
3623*89c4ff92SAndroid Build Coastguard Worker out[3] = 255;
3624*89c4ff92SAndroid Build Coastguard Worker out += n;
3625*89c4ff92SAndroid Build Coastguard Worker }
3626*89c4ff92SAndroid Build Coastguard Worker } else if (z->app14_color_transform == 2) { // YCCK
3627*89c4ff92SAndroid Build Coastguard Worker z->YCbCr_to_RGB_kernel(out, y, coutput[1], coutput[2], z->s->img_x, n);
3628*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i) {
3629*89c4ff92SAndroid Build Coastguard Worker stbi_uc m = coutput[3][i];
3630*89c4ff92SAndroid Build Coastguard Worker out[0] = stbi__blinn_8x8(255 - out[0], m);
3631*89c4ff92SAndroid Build Coastguard Worker out[1] = stbi__blinn_8x8(255 - out[1], m);
3632*89c4ff92SAndroid Build Coastguard Worker out[2] = stbi__blinn_8x8(255 - out[2], m);
3633*89c4ff92SAndroid Build Coastguard Worker out += n;
3634*89c4ff92SAndroid Build Coastguard Worker }
3635*89c4ff92SAndroid Build Coastguard Worker } else { // YCbCr + alpha? Ignore the fourth channel for now
3636*89c4ff92SAndroid Build Coastguard Worker z->YCbCr_to_RGB_kernel(out, y, coutput[1], coutput[2], z->s->img_x, n);
3637*89c4ff92SAndroid Build Coastguard Worker }
3638*89c4ff92SAndroid Build Coastguard Worker } else
3639*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i) {
3640*89c4ff92SAndroid Build Coastguard Worker out[0] = out[1] = out[2] = y[i];
3641*89c4ff92SAndroid Build Coastguard Worker out[3] = 255; // not used if n==3
3642*89c4ff92SAndroid Build Coastguard Worker out += n;
3643*89c4ff92SAndroid Build Coastguard Worker }
3644*89c4ff92SAndroid Build Coastguard Worker } else {
3645*89c4ff92SAndroid Build Coastguard Worker if (is_rgb) {
3646*89c4ff92SAndroid Build Coastguard Worker if (n == 1)
3647*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i)
3648*89c4ff92SAndroid Build Coastguard Worker *out++ = stbi__compute_y(coutput[0][i], coutput[1][i], coutput[2][i]);
3649*89c4ff92SAndroid Build Coastguard Worker else {
3650*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i, out += 2) {
3651*89c4ff92SAndroid Build Coastguard Worker out[0] = stbi__compute_y(coutput[0][i], coutput[1][i], coutput[2][i]);
3652*89c4ff92SAndroid Build Coastguard Worker out[1] = 255;
3653*89c4ff92SAndroid Build Coastguard Worker }
3654*89c4ff92SAndroid Build Coastguard Worker }
3655*89c4ff92SAndroid Build Coastguard Worker } else if (z->s->img_n == 4 && z->app14_color_transform == 0) {
3656*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i) {
3657*89c4ff92SAndroid Build Coastguard Worker stbi_uc m = coutput[3][i];
3658*89c4ff92SAndroid Build Coastguard Worker stbi_uc r = stbi__blinn_8x8(coutput[0][i], m);
3659*89c4ff92SAndroid Build Coastguard Worker stbi_uc g = stbi__blinn_8x8(coutput[1][i], m);
3660*89c4ff92SAndroid Build Coastguard Worker stbi_uc b = stbi__blinn_8x8(coutput[2][i], m);
3661*89c4ff92SAndroid Build Coastguard Worker out[0] = stbi__compute_y(r, g, b);
3662*89c4ff92SAndroid Build Coastguard Worker out[1] = 255;
3663*89c4ff92SAndroid Build Coastguard Worker out += n;
3664*89c4ff92SAndroid Build Coastguard Worker }
3665*89c4ff92SAndroid Build Coastguard Worker } else if (z->s->img_n == 4 && z->app14_color_transform == 2) {
3666*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i) {
3667*89c4ff92SAndroid Build Coastguard Worker out[0] = stbi__blinn_8x8(255 - coutput[0][i], coutput[3][i]);
3668*89c4ff92SAndroid Build Coastguard Worker out[1] = 255;
3669*89c4ff92SAndroid Build Coastguard Worker out += n;
3670*89c4ff92SAndroid Build Coastguard Worker }
3671*89c4ff92SAndroid Build Coastguard Worker } else {
3672*89c4ff92SAndroid Build Coastguard Worker stbi_uc *y = coutput[0];
3673*89c4ff92SAndroid Build Coastguard Worker if (n == 1)
3674*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i) out[i] = y[i];
3675*89c4ff92SAndroid Build Coastguard Worker else
3676*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < z->s->img_x; ++i) *out++ = y[i], *out++ = 255;
3677*89c4ff92SAndroid Build Coastguard Worker }
3678*89c4ff92SAndroid Build Coastguard Worker }
3679*89c4ff92SAndroid Build Coastguard Worker }
3680*89c4ff92SAndroid Build Coastguard Worker stbi__cleanup_jpeg(z);
3681*89c4ff92SAndroid Build Coastguard Worker *out_x = z->s->img_x;
3682*89c4ff92SAndroid Build Coastguard Worker *out_y = z->s->img_y;
3683*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = z->s->img_n >= 3 ? 3 : 1; // report original components, not output
3684*89c4ff92SAndroid Build Coastguard Worker return output;
3685*89c4ff92SAndroid Build Coastguard Worker }
3686*89c4ff92SAndroid Build Coastguard Worker }
3687*89c4ff92SAndroid Build Coastguard Worker
stbi__jpeg_load(stbi__context * s,int * x,int * y,int * comp,int req_comp,stbi__result_info * ri)3688*89c4ff92SAndroid Build Coastguard Worker static void *stbi__jpeg_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri)
3689*89c4ff92SAndroid Build Coastguard Worker {
3690*89c4ff92SAndroid Build Coastguard Worker unsigned char* result;
3691*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg* j = (stbi__jpeg*) stbi__malloc(sizeof(stbi__jpeg));
3692*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(ri);
3693*89c4ff92SAndroid Build Coastguard Worker j->s = s;
3694*89c4ff92SAndroid Build Coastguard Worker stbi__setup_jpeg(j);
3695*89c4ff92SAndroid Build Coastguard Worker result = load_jpeg_image(j, x,y,comp,req_comp);
3696*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(j);
3697*89c4ff92SAndroid Build Coastguard Worker return result;
3698*89c4ff92SAndroid Build Coastguard Worker }
3699*89c4ff92SAndroid Build Coastguard Worker
stbi__jpeg_test(stbi__context * s)3700*89c4ff92SAndroid Build Coastguard Worker static int stbi__jpeg_test(stbi__context *s)
3701*89c4ff92SAndroid Build Coastguard Worker {
3702*89c4ff92SAndroid Build Coastguard Worker int r;
3703*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg* j = (stbi__jpeg*)stbi__malloc(sizeof(stbi__jpeg));
3704*89c4ff92SAndroid Build Coastguard Worker j->s = s;
3705*89c4ff92SAndroid Build Coastguard Worker stbi__setup_jpeg(j);
3706*89c4ff92SAndroid Build Coastguard Worker r = stbi__decode_jpeg_header(j, STBI__SCAN_type);
3707*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
3708*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(j);
3709*89c4ff92SAndroid Build Coastguard Worker return r;
3710*89c4ff92SAndroid Build Coastguard Worker }
3711*89c4ff92SAndroid Build Coastguard Worker
stbi__jpeg_info_raw(stbi__jpeg * j,int * x,int * y,int * comp)3712*89c4ff92SAndroid Build Coastguard Worker static int stbi__jpeg_info_raw(stbi__jpeg *j, int *x, int *y, int *comp)
3713*89c4ff92SAndroid Build Coastguard Worker {
3714*89c4ff92SAndroid Build Coastguard Worker if (!stbi__decode_jpeg_header(j, STBI__SCAN_header)) {
3715*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( j->s );
3716*89c4ff92SAndroid Build Coastguard Worker return 0;
3717*89c4ff92SAndroid Build Coastguard Worker }
3718*89c4ff92SAndroid Build Coastguard Worker if (x) *x = j->s->img_x;
3719*89c4ff92SAndroid Build Coastguard Worker if (y) *y = j->s->img_y;
3720*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = j->s->img_n >= 3 ? 3 : 1;
3721*89c4ff92SAndroid Build Coastguard Worker return 1;
3722*89c4ff92SAndroid Build Coastguard Worker }
3723*89c4ff92SAndroid Build Coastguard Worker
stbi__jpeg_info(stbi__context * s,int * x,int * y,int * comp)3724*89c4ff92SAndroid Build Coastguard Worker static int stbi__jpeg_info(stbi__context *s, int *x, int *y, int *comp)
3725*89c4ff92SAndroid Build Coastguard Worker {
3726*89c4ff92SAndroid Build Coastguard Worker int result;
3727*89c4ff92SAndroid Build Coastguard Worker stbi__jpeg* j = (stbi__jpeg*) (stbi__malloc(sizeof(stbi__jpeg)));
3728*89c4ff92SAndroid Build Coastguard Worker j->s = s;
3729*89c4ff92SAndroid Build Coastguard Worker result = stbi__jpeg_info_raw(j, x, y, comp);
3730*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(j);
3731*89c4ff92SAndroid Build Coastguard Worker return result;
3732*89c4ff92SAndroid Build Coastguard Worker }
3733*89c4ff92SAndroid Build Coastguard Worker #endif
3734*89c4ff92SAndroid Build Coastguard Worker
3735*89c4ff92SAndroid Build Coastguard Worker // public domain zlib decode v0.2 Sean Barrett 2006-11-18
3736*89c4ff92SAndroid Build Coastguard Worker // simple implementation
3737*89c4ff92SAndroid Build Coastguard Worker // - all input must be provided in an upfront buffer
3738*89c4ff92SAndroid Build Coastguard Worker // - all output is written to a single output buffer (can malloc/realloc)
3739*89c4ff92SAndroid Build Coastguard Worker // performance
3740*89c4ff92SAndroid Build Coastguard Worker // - fast huffman
3741*89c4ff92SAndroid Build Coastguard Worker
3742*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_ZLIB
3743*89c4ff92SAndroid Build Coastguard Worker
3744*89c4ff92SAndroid Build Coastguard Worker // fast-way is faster to check than jpeg huffman, but slow way is slower
3745*89c4ff92SAndroid Build Coastguard Worker #define STBI__ZFAST_BITS 9 // accelerate all cases in default tables
3746*89c4ff92SAndroid Build Coastguard Worker #define STBI__ZFAST_MASK ((1 << STBI__ZFAST_BITS) - 1)
3747*89c4ff92SAndroid Build Coastguard Worker
3748*89c4ff92SAndroid Build Coastguard Worker // zlib-style huffman encoding
3749*89c4ff92SAndroid Build Coastguard Worker // (jpegs packs from left, zlib from right, so can't share code)
3750*89c4ff92SAndroid Build Coastguard Worker typedef struct
3751*89c4ff92SAndroid Build Coastguard Worker {
3752*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 fast[1 << STBI__ZFAST_BITS];
3753*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 firstcode[16];
3754*89c4ff92SAndroid Build Coastguard Worker int maxcode[17];
3755*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 firstsymbol[16];
3756*89c4ff92SAndroid Build Coastguard Worker stbi_uc size[288];
3757*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 value[288];
3758*89c4ff92SAndroid Build Coastguard Worker } stbi__zhuffman;
3759*89c4ff92SAndroid Build Coastguard Worker
stbi__bitreverse16(int n)3760*89c4ff92SAndroid Build Coastguard Worker stbi_inline static int stbi__bitreverse16(int n)
3761*89c4ff92SAndroid Build Coastguard Worker {
3762*89c4ff92SAndroid Build Coastguard Worker n = ((n & 0xAAAA) >> 1) | ((n & 0x5555) << 1);
3763*89c4ff92SAndroid Build Coastguard Worker n = ((n & 0xCCCC) >> 2) | ((n & 0x3333) << 2);
3764*89c4ff92SAndroid Build Coastguard Worker n = ((n & 0xF0F0) >> 4) | ((n & 0x0F0F) << 4);
3765*89c4ff92SAndroid Build Coastguard Worker n = ((n & 0xFF00) >> 8) | ((n & 0x00FF) << 8);
3766*89c4ff92SAndroid Build Coastguard Worker return n;
3767*89c4ff92SAndroid Build Coastguard Worker }
3768*89c4ff92SAndroid Build Coastguard Worker
stbi__bit_reverse(int v,int bits)3769*89c4ff92SAndroid Build Coastguard Worker stbi_inline static int stbi__bit_reverse(int v, int bits)
3770*89c4ff92SAndroid Build Coastguard Worker {
3771*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(bits <= 16);
3772*89c4ff92SAndroid Build Coastguard Worker // to bit reverse n bits, reverse 16 and shift
3773*89c4ff92SAndroid Build Coastguard Worker // e.g. 11 bits, bit reverse and shift away 5
3774*89c4ff92SAndroid Build Coastguard Worker return stbi__bitreverse16(v) >> (16-bits);
3775*89c4ff92SAndroid Build Coastguard Worker }
3776*89c4ff92SAndroid Build Coastguard Worker
stbi__zbuild_huffman(stbi__zhuffman * z,const stbi_uc * sizelist,int num)3777*89c4ff92SAndroid Build Coastguard Worker static int stbi__zbuild_huffman(stbi__zhuffman *z, const stbi_uc *sizelist, int num)
3778*89c4ff92SAndroid Build Coastguard Worker {
3779*89c4ff92SAndroid Build Coastguard Worker int i,k=0;
3780*89c4ff92SAndroid Build Coastguard Worker int code, next_code[16], sizes[17];
3781*89c4ff92SAndroid Build Coastguard Worker
3782*89c4ff92SAndroid Build Coastguard Worker // DEFLATE spec for generating codes
3783*89c4ff92SAndroid Build Coastguard Worker memset(sizes, 0, sizeof(sizes));
3784*89c4ff92SAndroid Build Coastguard Worker memset(z->fast, 0, sizeof(z->fast));
3785*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < num; ++i)
3786*89c4ff92SAndroid Build Coastguard Worker ++sizes[sizelist[i]];
3787*89c4ff92SAndroid Build Coastguard Worker sizes[0] = 0;
3788*89c4ff92SAndroid Build Coastguard Worker for (i=1; i < 16; ++i)
3789*89c4ff92SAndroid Build Coastguard Worker if (sizes[i] > (1 << i))
3790*89c4ff92SAndroid Build Coastguard Worker return stbi__err("bad sizes", "Corrupt PNG");
3791*89c4ff92SAndroid Build Coastguard Worker code = 0;
3792*89c4ff92SAndroid Build Coastguard Worker for (i=1; i < 16; ++i) {
3793*89c4ff92SAndroid Build Coastguard Worker next_code[i] = code;
3794*89c4ff92SAndroid Build Coastguard Worker z->firstcode[i] = (stbi__uint16) code;
3795*89c4ff92SAndroid Build Coastguard Worker z->firstsymbol[i] = (stbi__uint16) k;
3796*89c4ff92SAndroid Build Coastguard Worker code = (code + sizes[i]);
3797*89c4ff92SAndroid Build Coastguard Worker if (sizes[i])
3798*89c4ff92SAndroid Build Coastguard Worker if (code-1 >= (1 << i)) return stbi__err("bad codelengths","Corrupt PNG");
3799*89c4ff92SAndroid Build Coastguard Worker z->maxcode[i] = code << (16-i); // preshift for inner loop
3800*89c4ff92SAndroid Build Coastguard Worker code <<= 1;
3801*89c4ff92SAndroid Build Coastguard Worker k += sizes[i];
3802*89c4ff92SAndroid Build Coastguard Worker }
3803*89c4ff92SAndroid Build Coastguard Worker z->maxcode[16] = 0x10000; // sentinel
3804*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < num; ++i) {
3805*89c4ff92SAndroid Build Coastguard Worker int s = sizelist[i];
3806*89c4ff92SAndroid Build Coastguard Worker if (s) {
3807*89c4ff92SAndroid Build Coastguard Worker int c = next_code[s] - z->firstcode[s] + z->firstsymbol[s];
3808*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 fastv = (stbi__uint16) ((s << 9) | i);
3809*89c4ff92SAndroid Build Coastguard Worker z->size [c] = (stbi_uc ) s;
3810*89c4ff92SAndroid Build Coastguard Worker z->value[c] = (stbi__uint16) i;
3811*89c4ff92SAndroid Build Coastguard Worker if (s <= STBI__ZFAST_BITS) {
3812*89c4ff92SAndroid Build Coastguard Worker int j = stbi__bit_reverse(next_code[s],s);
3813*89c4ff92SAndroid Build Coastguard Worker while (j < (1 << STBI__ZFAST_BITS)) {
3814*89c4ff92SAndroid Build Coastguard Worker z->fast[j] = fastv;
3815*89c4ff92SAndroid Build Coastguard Worker j += (1 << s);
3816*89c4ff92SAndroid Build Coastguard Worker }
3817*89c4ff92SAndroid Build Coastguard Worker }
3818*89c4ff92SAndroid Build Coastguard Worker ++next_code[s];
3819*89c4ff92SAndroid Build Coastguard Worker }
3820*89c4ff92SAndroid Build Coastguard Worker }
3821*89c4ff92SAndroid Build Coastguard Worker return 1;
3822*89c4ff92SAndroid Build Coastguard Worker }
3823*89c4ff92SAndroid Build Coastguard Worker
3824*89c4ff92SAndroid Build Coastguard Worker // zlib-from-memory implementation for PNG reading
3825*89c4ff92SAndroid Build Coastguard Worker // because PNG allows splitting the zlib stream arbitrarily,
3826*89c4ff92SAndroid Build Coastguard Worker // and it's annoying structurally to have PNG call ZLIB call PNG,
3827*89c4ff92SAndroid Build Coastguard Worker // we require PNG read all the IDATs and combine them into a single
3828*89c4ff92SAndroid Build Coastguard Worker // memory buffer
3829*89c4ff92SAndroid Build Coastguard Worker
3830*89c4ff92SAndroid Build Coastguard Worker typedef struct
3831*89c4ff92SAndroid Build Coastguard Worker {
3832*89c4ff92SAndroid Build Coastguard Worker stbi_uc *zbuffer, *zbuffer_end;
3833*89c4ff92SAndroid Build Coastguard Worker int num_bits;
3834*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 code_buffer;
3835*89c4ff92SAndroid Build Coastguard Worker
3836*89c4ff92SAndroid Build Coastguard Worker char *zout;
3837*89c4ff92SAndroid Build Coastguard Worker char *zout_start;
3838*89c4ff92SAndroid Build Coastguard Worker char *zout_end;
3839*89c4ff92SAndroid Build Coastguard Worker int z_expandable;
3840*89c4ff92SAndroid Build Coastguard Worker
3841*89c4ff92SAndroid Build Coastguard Worker stbi__zhuffman z_length, z_distance;
3842*89c4ff92SAndroid Build Coastguard Worker } stbi__zbuf;
3843*89c4ff92SAndroid Build Coastguard Worker
stbi__zget8(stbi__zbuf * z)3844*89c4ff92SAndroid Build Coastguard Worker stbi_inline static stbi_uc stbi__zget8(stbi__zbuf *z)
3845*89c4ff92SAndroid Build Coastguard Worker {
3846*89c4ff92SAndroid Build Coastguard Worker if (z->zbuffer >= z->zbuffer_end) return 0;
3847*89c4ff92SAndroid Build Coastguard Worker return *z->zbuffer++;
3848*89c4ff92SAndroid Build Coastguard Worker }
3849*89c4ff92SAndroid Build Coastguard Worker
stbi__fill_bits(stbi__zbuf * z)3850*89c4ff92SAndroid Build Coastguard Worker static void stbi__fill_bits(stbi__zbuf *z)
3851*89c4ff92SAndroid Build Coastguard Worker {
3852*89c4ff92SAndroid Build Coastguard Worker do {
3853*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(z->code_buffer < (1U << z->num_bits));
3854*89c4ff92SAndroid Build Coastguard Worker z->code_buffer |= (unsigned int) stbi__zget8(z) << z->num_bits;
3855*89c4ff92SAndroid Build Coastguard Worker z->num_bits += 8;
3856*89c4ff92SAndroid Build Coastguard Worker } while (z->num_bits <= 24);
3857*89c4ff92SAndroid Build Coastguard Worker }
3858*89c4ff92SAndroid Build Coastguard Worker
stbi__zreceive(stbi__zbuf * z,int n)3859*89c4ff92SAndroid Build Coastguard Worker stbi_inline static unsigned int stbi__zreceive(stbi__zbuf *z, int n)
3860*89c4ff92SAndroid Build Coastguard Worker {
3861*89c4ff92SAndroid Build Coastguard Worker unsigned int k;
3862*89c4ff92SAndroid Build Coastguard Worker if (z->num_bits < n) stbi__fill_bits(z);
3863*89c4ff92SAndroid Build Coastguard Worker k = z->code_buffer & ((1 << n) - 1);
3864*89c4ff92SAndroid Build Coastguard Worker z->code_buffer >>= n;
3865*89c4ff92SAndroid Build Coastguard Worker z->num_bits -= n;
3866*89c4ff92SAndroid Build Coastguard Worker return k;
3867*89c4ff92SAndroid Build Coastguard Worker }
3868*89c4ff92SAndroid Build Coastguard Worker
stbi__zhuffman_decode_slowpath(stbi__zbuf * a,stbi__zhuffman * z)3869*89c4ff92SAndroid Build Coastguard Worker static int stbi__zhuffman_decode_slowpath(stbi__zbuf *a, stbi__zhuffman *z)
3870*89c4ff92SAndroid Build Coastguard Worker {
3871*89c4ff92SAndroid Build Coastguard Worker int b,s,k;
3872*89c4ff92SAndroid Build Coastguard Worker // not resolved by fast table, so compute it the slow way
3873*89c4ff92SAndroid Build Coastguard Worker // use jpeg approach, which requires MSbits at top
3874*89c4ff92SAndroid Build Coastguard Worker k = stbi__bit_reverse(a->code_buffer, 16);
3875*89c4ff92SAndroid Build Coastguard Worker for (s=STBI__ZFAST_BITS+1; ; ++s)
3876*89c4ff92SAndroid Build Coastguard Worker if (k < z->maxcode[s])
3877*89c4ff92SAndroid Build Coastguard Worker break;
3878*89c4ff92SAndroid Build Coastguard Worker if (s == 16) return -1; // invalid code!
3879*89c4ff92SAndroid Build Coastguard Worker // code size is s, so:
3880*89c4ff92SAndroid Build Coastguard Worker b = (k >> (16-s)) - z->firstcode[s] + z->firstsymbol[s];
3881*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(z->size[b] == s);
3882*89c4ff92SAndroid Build Coastguard Worker a->code_buffer >>= s;
3883*89c4ff92SAndroid Build Coastguard Worker a->num_bits -= s;
3884*89c4ff92SAndroid Build Coastguard Worker return z->value[b];
3885*89c4ff92SAndroid Build Coastguard Worker }
3886*89c4ff92SAndroid Build Coastguard Worker
stbi__zhuffman_decode(stbi__zbuf * a,stbi__zhuffman * z)3887*89c4ff92SAndroid Build Coastguard Worker stbi_inline static int stbi__zhuffman_decode(stbi__zbuf *a, stbi__zhuffman *z)
3888*89c4ff92SAndroid Build Coastguard Worker {
3889*89c4ff92SAndroid Build Coastguard Worker int b,s;
3890*89c4ff92SAndroid Build Coastguard Worker if (a->num_bits < 16) stbi__fill_bits(a);
3891*89c4ff92SAndroid Build Coastguard Worker b = z->fast[a->code_buffer & STBI__ZFAST_MASK];
3892*89c4ff92SAndroid Build Coastguard Worker if (b) {
3893*89c4ff92SAndroid Build Coastguard Worker s = b >> 9;
3894*89c4ff92SAndroid Build Coastguard Worker a->code_buffer >>= s;
3895*89c4ff92SAndroid Build Coastguard Worker a->num_bits -= s;
3896*89c4ff92SAndroid Build Coastguard Worker return b & 511;
3897*89c4ff92SAndroid Build Coastguard Worker }
3898*89c4ff92SAndroid Build Coastguard Worker return stbi__zhuffman_decode_slowpath(a, z);
3899*89c4ff92SAndroid Build Coastguard Worker }
3900*89c4ff92SAndroid Build Coastguard Worker
stbi__zexpand(stbi__zbuf * z,char * zout,int n)3901*89c4ff92SAndroid Build Coastguard Worker static int stbi__zexpand(stbi__zbuf *z, char *zout, int n) // need to make room for n bytes
3902*89c4ff92SAndroid Build Coastguard Worker {
3903*89c4ff92SAndroid Build Coastguard Worker char *q;
3904*89c4ff92SAndroid Build Coastguard Worker int cur, limit, old_limit;
3905*89c4ff92SAndroid Build Coastguard Worker z->zout = zout;
3906*89c4ff92SAndroid Build Coastguard Worker if (!z->z_expandable) return stbi__err("output buffer limit","Corrupt PNG");
3907*89c4ff92SAndroid Build Coastguard Worker cur = (int) (z->zout - z->zout_start);
3908*89c4ff92SAndroid Build Coastguard Worker limit = old_limit = (int) (z->zout_end - z->zout_start);
3909*89c4ff92SAndroid Build Coastguard Worker while (cur + n > limit)
3910*89c4ff92SAndroid Build Coastguard Worker limit *= 2;
3911*89c4ff92SAndroid Build Coastguard Worker q = (char *) STBI_REALLOC_SIZED(z->zout_start, old_limit, limit);
3912*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(old_limit);
3913*89c4ff92SAndroid Build Coastguard Worker if (q == NULL) return stbi__err("outofmem", "Out of memory");
3914*89c4ff92SAndroid Build Coastguard Worker z->zout_start = q;
3915*89c4ff92SAndroid Build Coastguard Worker z->zout = q + cur;
3916*89c4ff92SAndroid Build Coastguard Worker z->zout_end = q + limit;
3917*89c4ff92SAndroid Build Coastguard Worker return 1;
3918*89c4ff92SAndroid Build Coastguard Worker }
3919*89c4ff92SAndroid Build Coastguard Worker
3920*89c4ff92SAndroid Build Coastguard Worker static int stbi__zlength_base[31] = {
3921*89c4ff92SAndroid Build Coastguard Worker 3,4,5,6,7,8,9,10,11,13,
3922*89c4ff92SAndroid Build Coastguard Worker 15,17,19,23,27,31,35,43,51,59,
3923*89c4ff92SAndroid Build Coastguard Worker 67,83,99,115,131,163,195,227,258,0,0 };
3924*89c4ff92SAndroid Build Coastguard Worker
3925*89c4ff92SAndroid Build Coastguard Worker static int stbi__zlength_extra[31]=
3926*89c4ff92SAndroid Build Coastguard Worker { 0,0,0,0,0,0,0,0,1,1,1,1,2,2,2,2,3,3,3,3,4,4,4,4,5,5,5,5,0,0,0 };
3927*89c4ff92SAndroid Build Coastguard Worker
3928*89c4ff92SAndroid Build Coastguard Worker static int stbi__zdist_base[32] = { 1,2,3,4,5,7,9,13,17,25,33,49,65,97,129,193,
3929*89c4ff92SAndroid Build Coastguard Worker 257,385,513,769,1025,1537,2049,3073,4097,6145,8193,12289,16385,24577,0,0};
3930*89c4ff92SAndroid Build Coastguard Worker
3931*89c4ff92SAndroid Build Coastguard Worker static int stbi__zdist_extra[32] =
3932*89c4ff92SAndroid Build Coastguard Worker { 0,0,0,0,1,1,2,2,3,3,4,4,5,5,6,6,7,7,8,8,9,9,10,10,11,11,12,12,13,13};
3933*89c4ff92SAndroid Build Coastguard Worker
stbi__parse_huffman_block(stbi__zbuf * a)3934*89c4ff92SAndroid Build Coastguard Worker static int stbi__parse_huffman_block(stbi__zbuf *a)
3935*89c4ff92SAndroid Build Coastguard Worker {
3936*89c4ff92SAndroid Build Coastguard Worker char *zout = a->zout;
3937*89c4ff92SAndroid Build Coastguard Worker for(;;) {
3938*89c4ff92SAndroid Build Coastguard Worker int z = stbi__zhuffman_decode(a, &a->z_length);
3939*89c4ff92SAndroid Build Coastguard Worker if (z < 256) {
3940*89c4ff92SAndroid Build Coastguard Worker if (z < 0) return stbi__err("bad huffman code","Corrupt PNG"); // error in huffman codes
3941*89c4ff92SAndroid Build Coastguard Worker if (zout >= a->zout_end) {
3942*89c4ff92SAndroid Build Coastguard Worker if (!stbi__zexpand(a, zout, 1)) return 0;
3943*89c4ff92SAndroid Build Coastguard Worker zout = a->zout;
3944*89c4ff92SAndroid Build Coastguard Worker }
3945*89c4ff92SAndroid Build Coastguard Worker *zout++ = (char) z;
3946*89c4ff92SAndroid Build Coastguard Worker } else {
3947*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p;
3948*89c4ff92SAndroid Build Coastguard Worker int len,dist;
3949*89c4ff92SAndroid Build Coastguard Worker if (z == 256) {
3950*89c4ff92SAndroid Build Coastguard Worker a->zout = zout;
3951*89c4ff92SAndroid Build Coastguard Worker return 1;
3952*89c4ff92SAndroid Build Coastguard Worker }
3953*89c4ff92SAndroid Build Coastguard Worker z -= 257;
3954*89c4ff92SAndroid Build Coastguard Worker len = stbi__zlength_base[z];
3955*89c4ff92SAndroid Build Coastguard Worker if (stbi__zlength_extra[z]) len += stbi__zreceive(a, stbi__zlength_extra[z]);
3956*89c4ff92SAndroid Build Coastguard Worker z = stbi__zhuffman_decode(a, &a->z_distance);
3957*89c4ff92SAndroid Build Coastguard Worker if (z < 0) return stbi__err("bad huffman code","Corrupt PNG");
3958*89c4ff92SAndroid Build Coastguard Worker dist = stbi__zdist_base[z];
3959*89c4ff92SAndroid Build Coastguard Worker if (stbi__zdist_extra[z]) dist += stbi__zreceive(a, stbi__zdist_extra[z]);
3960*89c4ff92SAndroid Build Coastguard Worker if (zout - a->zout_start < dist) return stbi__err("bad dist","Corrupt PNG");
3961*89c4ff92SAndroid Build Coastguard Worker if (zout + len > a->zout_end) {
3962*89c4ff92SAndroid Build Coastguard Worker if (!stbi__zexpand(a, zout, len)) return 0;
3963*89c4ff92SAndroid Build Coastguard Worker zout = a->zout;
3964*89c4ff92SAndroid Build Coastguard Worker }
3965*89c4ff92SAndroid Build Coastguard Worker p = (stbi_uc *) (zout - dist);
3966*89c4ff92SAndroid Build Coastguard Worker if (dist == 1) { // run of one byte; common in images.
3967*89c4ff92SAndroid Build Coastguard Worker stbi_uc v = *p;
3968*89c4ff92SAndroid Build Coastguard Worker if (len) { do *zout++ = v; while (--len); }
3969*89c4ff92SAndroid Build Coastguard Worker } else {
3970*89c4ff92SAndroid Build Coastguard Worker if (len) { do *zout++ = *p++; while (--len); }
3971*89c4ff92SAndroid Build Coastguard Worker }
3972*89c4ff92SAndroid Build Coastguard Worker }
3973*89c4ff92SAndroid Build Coastguard Worker }
3974*89c4ff92SAndroid Build Coastguard Worker }
3975*89c4ff92SAndroid Build Coastguard Worker
stbi__compute_huffman_codes(stbi__zbuf * a)3976*89c4ff92SAndroid Build Coastguard Worker static int stbi__compute_huffman_codes(stbi__zbuf *a)
3977*89c4ff92SAndroid Build Coastguard Worker {
3978*89c4ff92SAndroid Build Coastguard Worker static stbi_uc length_dezigzag[19] = { 16,17,18,0,8,7,9,6,10,5,11,4,12,3,13,2,14,1,15 };
3979*89c4ff92SAndroid Build Coastguard Worker stbi__zhuffman z_codelength;
3980*89c4ff92SAndroid Build Coastguard Worker stbi_uc lencodes[286+32+137];//padding for maximum single op
3981*89c4ff92SAndroid Build Coastguard Worker stbi_uc codelength_sizes[19];
3982*89c4ff92SAndroid Build Coastguard Worker int i,n;
3983*89c4ff92SAndroid Build Coastguard Worker
3984*89c4ff92SAndroid Build Coastguard Worker int hlit = stbi__zreceive(a,5) + 257;
3985*89c4ff92SAndroid Build Coastguard Worker int hdist = stbi__zreceive(a,5) + 1;
3986*89c4ff92SAndroid Build Coastguard Worker int hclen = stbi__zreceive(a,4) + 4;
3987*89c4ff92SAndroid Build Coastguard Worker int ntot = hlit + hdist;
3988*89c4ff92SAndroid Build Coastguard Worker
3989*89c4ff92SAndroid Build Coastguard Worker memset(codelength_sizes, 0, sizeof(codelength_sizes));
3990*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < hclen; ++i) {
3991*89c4ff92SAndroid Build Coastguard Worker int s = stbi__zreceive(a,3);
3992*89c4ff92SAndroid Build Coastguard Worker codelength_sizes[length_dezigzag[i]] = (stbi_uc) s;
3993*89c4ff92SAndroid Build Coastguard Worker }
3994*89c4ff92SAndroid Build Coastguard Worker if (!stbi__zbuild_huffman(&z_codelength, codelength_sizes, 19)) return 0;
3995*89c4ff92SAndroid Build Coastguard Worker
3996*89c4ff92SAndroid Build Coastguard Worker n = 0;
3997*89c4ff92SAndroid Build Coastguard Worker while (n < ntot) {
3998*89c4ff92SAndroid Build Coastguard Worker int c = stbi__zhuffman_decode(a, &z_codelength);
3999*89c4ff92SAndroid Build Coastguard Worker if (c < 0 || c >= 19) return stbi__err("bad codelengths", "Corrupt PNG");
4000*89c4ff92SAndroid Build Coastguard Worker if (c < 16)
4001*89c4ff92SAndroid Build Coastguard Worker lencodes[n++] = (stbi_uc) c;
4002*89c4ff92SAndroid Build Coastguard Worker else {
4003*89c4ff92SAndroid Build Coastguard Worker stbi_uc fill = 0;
4004*89c4ff92SAndroid Build Coastguard Worker if (c == 16) {
4005*89c4ff92SAndroid Build Coastguard Worker c = stbi__zreceive(a,2)+3;
4006*89c4ff92SAndroid Build Coastguard Worker if (n == 0) return stbi__err("bad codelengths", "Corrupt PNG");
4007*89c4ff92SAndroid Build Coastguard Worker fill = lencodes[n-1];
4008*89c4ff92SAndroid Build Coastguard Worker } else if (c == 17)
4009*89c4ff92SAndroid Build Coastguard Worker c = stbi__zreceive(a,3)+3;
4010*89c4ff92SAndroid Build Coastguard Worker else {
4011*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(c == 18);
4012*89c4ff92SAndroid Build Coastguard Worker c = stbi__zreceive(a,7)+11;
4013*89c4ff92SAndroid Build Coastguard Worker }
4014*89c4ff92SAndroid Build Coastguard Worker if (ntot - n < c) return stbi__err("bad codelengths", "Corrupt PNG");
4015*89c4ff92SAndroid Build Coastguard Worker memset(lencodes+n, fill, c);
4016*89c4ff92SAndroid Build Coastguard Worker n += c;
4017*89c4ff92SAndroid Build Coastguard Worker }
4018*89c4ff92SAndroid Build Coastguard Worker }
4019*89c4ff92SAndroid Build Coastguard Worker if (n != ntot) return stbi__err("bad codelengths","Corrupt PNG");
4020*89c4ff92SAndroid Build Coastguard Worker if (!stbi__zbuild_huffman(&a->z_length, lencodes, hlit)) return 0;
4021*89c4ff92SAndroid Build Coastguard Worker if (!stbi__zbuild_huffman(&a->z_distance, lencodes+hlit, hdist)) return 0;
4022*89c4ff92SAndroid Build Coastguard Worker return 1;
4023*89c4ff92SAndroid Build Coastguard Worker }
4024*89c4ff92SAndroid Build Coastguard Worker
stbi__parse_uncompressed_block(stbi__zbuf * a)4025*89c4ff92SAndroid Build Coastguard Worker static int stbi__parse_uncompressed_block(stbi__zbuf *a)
4026*89c4ff92SAndroid Build Coastguard Worker {
4027*89c4ff92SAndroid Build Coastguard Worker stbi_uc header[4];
4028*89c4ff92SAndroid Build Coastguard Worker int len,nlen,k;
4029*89c4ff92SAndroid Build Coastguard Worker if (a->num_bits & 7)
4030*89c4ff92SAndroid Build Coastguard Worker stbi__zreceive(a, a->num_bits & 7); // discard
4031*89c4ff92SAndroid Build Coastguard Worker // drain the bit-packed data into header
4032*89c4ff92SAndroid Build Coastguard Worker k = 0;
4033*89c4ff92SAndroid Build Coastguard Worker while (a->num_bits > 0) {
4034*89c4ff92SAndroid Build Coastguard Worker header[k++] = (stbi_uc) (a->code_buffer & 255); // suppress MSVC run-time check
4035*89c4ff92SAndroid Build Coastguard Worker a->code_buffer >>= 8;
4036*89c4ff92SAndroid Build Coastguard Worker a->num_bits -= 8;
4037*89c4ff92SAndroid Build Coastguard Worker }
4038*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(a->num_bits == 0);
4039*89c4ff92SAndroid Build Coastguard Worker // now fill header the normal way
4040*89c4ff92SAndroid Build Coastguard Worker while (k < 4)
4041*89c4ff92SAndroid Build Coastguard Worker header[k++] = stbi__zget8(a);
4042*89c4ff92SAndroid Build Coastguard Worker len = header[1] * 256 + header[0];
4043*89c4ff92SAndroid Build Coastguard Worker nlen = header[3] * 256 + header[2];
4044*89c4ff92SAndroid Build Coastguard Worker if (nlen != (len ^ 0xffff)) return stbi__err("zlib corrupt","Corrupt PNG");
4045*89c4ff92SAndroid Build Coastguard Worker if (a->zbuffer + len > a->zbuffer_end) return stbi__err("read past buffer","Corrupt PNG");
4046*89c4ff92SAndroid Build Coastguard Worker if (a->zout + len > a->zout_end)
4047*89c4ff92SAndroid Build Coastguard Worker if (!stbi__zexpand(a, a->zout, len)) return 0;
4048*89c4ff92SAndroid Build Coastguard Worker memcpy(a->zout, a->zbuffer, len);
4049*89c4ff92SAndroid Build Coastguard Worker a->zbuffer += len;
4050*89c4ff92SAndroid Build Coastguard Worker a->zout += len;
4051*89c4ff92SAndroid Build Coastguard Worker return 1;
4052*89c4ff92SAndroid Build Coastguard Worker }
4053*89c4ff92SAndroid Build Coastguard Worker
stbi__parse_zlib_header(stbi__zbuf * a)4054*89c4ff92SAndroid Build Coastguard Worker static int stbi__parse_zlib_header(stbi__zbuf *a)
4055*89c4ff92SAndroid Build Coastguard Worker {
4056*89c4ff92SAndroid Build Coastguard Worker int cmf = stbi__zget8(a);
4057*89c4ff92SAndroid Build Coastguard Worker int cm = cmf & 15;
4058*89c4ff92SAndroid Build Coastguard Worker /* int cinfo = cmf >> 4; */
4059*89c4ff92SAndroid Build Coastguard Worker int flg = stbi__zget8(a);
4060*89c4ff92SAndroid Build Coastguard Worker if ((cmf*256+flg) % 31 != 0) return stbi__err("bad zlib header","Corrupt PNG"); // zlib spec
4061*89c4ff92SAndroid Build Coastguard Worker if (flg & 32) return stbi__err("no preset dict","Corrupt PNG"); // preset dictionary not allowed in png
4062*89c4ff92SAndroid Build Coastguard Worker if (cm != 8) return stbi__err("bad compression","Corrupt PNG"); // DEFLATE required for png
4063*89c4ff92SAndroid Build Coastguard Worker // window = 1 << (8 + cinfo)... but who cares, we fully buffer output
4064*89c4ff92SAndroid Build Coastguard Worker return 1;
4065*89c4ff92SAndroid Build Coastguard Worker }
4066*89c4ff92SAndroid Build Coastguard Worker
4067*89c4ff92SAndroid Build Coastguard Worker static const stbi_uc stbi__zdefault_length[288] =
4068*89c4ff92SAndroid Build Coastguard Worker {
4069*89c4ff92SAndroid Build Coastguard Worker 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
4070*89c4ff92SAndroid Build Coastguard Worker 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
4071*89c4ff92SAndroid Build Coastguard Worker 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
4072*89c4ff92SAndroid Build Coastguard Worker 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
4073*89c4ff92SAndroid Build Coastguard Worker 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,
4074*89c4ff92SAndroid Build Coastguard Worker 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9, 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,
4075*89c4ff92SAndroid Build Coastguard Worker 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9, 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,
4076*89c4ff92SAndroid Build Coastguard Worker 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9, 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,
4077*89c4ff92SAndroid Build Coastguard Worker 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7, 7,7,7,7,7,7,7,7,8,8,8,8,8,8,8,8
4078*89c4ff92SAndroid Build Coastguard Worker };
4079*89c4ff92SAndroid Build Coastguard Worker static const stbi_uc stbi__zdefault_distance[32] =
4080*89c4ff92SAndroid Build Coastguard Worker {
4081*89c4ff92SAndroid Build Coastguard Worker 5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5
4082*89c4ff92SAndroid Build Coastguard Worker };
4083*89c4ff92SAndroid Build Coastguard Worker /*
4084*89c4ff92SAndroid Build Coastguard Worker Init algorithm:
4085*89c4ff92SAndroid Build Coastguard Worker {
4086*89c4ff92SAndroid Build Coastguard Worker int i; // use <= to match clearly with spec
4087*89c4ff92SAndroid Build Coastguard Worker for (i=0; i <= 143; ++i) stbi__zdefault_length[i] = 8;
4088*89c4ff92SAndroid Build Coastguard Worker for ( ; i <= 255; ++i) stbi__zdefault_length[i] = 9;
4089*89c4ff92SAndroid Build Coastguard Worker for ( ; i <= 279; ++i) stbi__zdefault_length[i] = 7;
4090*89c4ff92SAndroid Build Coastguard Worker for ( ; i <= 287; ++i) stbi__zdefault_length[i] = 8;
4091*89c4ff92SAndroid Build Coastguard Worker
4092*89c4ff92SAndroid Build Coastguard Worker for (i=0; i <= 31; ++i) stbi__zdefault_distance[i] = 5;
4093*89c4ff92SAndroid Build Coastguard Worker }
4094*89c4ff92SAndroid Build Coastguard Worker */
4095*89c4ff92SAndroid Build Coastguard Worker
stbi__parse_zlib(stbi__zbuf * a,int parse_header)4096*89c4ff92SAndroid Build Coastguard Worker static int stbi__parse_zlib(stbi__zbuf *a, int parse_header)
4097*89c4ff92SAndroid Build Coastguard Worker {
4098*89c4ff92SAndroid Build Coastguard Worker int final, type;
4099*89c4ff92SAndroid Build Coastguard Worker if (parse_header)
4100*89c4ff92SAndroid Build Coastguard Worker if (!stbi__parse_zlib_header(a)) return 0;
4101*89c4ff92SAndroid Build Coastguard Worker a->num_bits = 0;
4102*89c4ff92SAndroid Build Coastguard Worker a->code_buffer = 0;
4103*89c4ff92SAndroid Build Coastguard Worker do {
4104*89c4ff92SAndroid Build Coastguard Worker final = stbi__zreceive(a,1);
4105*89c4ff92SAndroid Build Coastguard Worker type = stbi__zreceive(a,2);
4106*89c4ff92SAndroid Build Coastguard Worker if (type == 0) {
4107*89c4ff92SAndroid Build Coastguard Worker if (!stbi__parse_uncompressed_block(a)) return 0;
4108*89c4ff92SAndroid Build Coastguard Worker } else if (type == 3) {
4109*89c4ff92SAndroid Build Coastguard Worker return 0;
4110*89c4ff92SAndroid Build Coastguard Worker } else {
4111*89c4ff92SAndroid Build Coastguard Worker if (type == 1) {
4112*89c4ff92SAndroid Build Coastguard Worker // use fixed code lengths
4113*89c4ff92SAndroid Build Coastguard Worker if (!stbi__zbuild_huffman(&a->z_length , stbi__zdefault_length , 288)) return 0;
4114*89c4ff92SAndroid Build Coastguard Worker if (!stbi__zbuild_huffman(&a->z_distance, stbi__zdefault_distance, 32)) return 0;
4115*89c4ff92SAndroid Build Coastguard Worker } else {
4116*89c4ff92SAndroid Build Coastguard Worker if (!stbi__compute_huffman_codes(a)) return 0;
4117*89c4ff92SAndroid Build Coastguard Worker }
4118*89c4ff92SAndroid Build Coastguard Worker if (!stbi__parse_huffman_block(a)) return 0;
4119*89c4ff92SAndroid Build Coastguard Worker }
4120*89c4ff92SAndroid Build Coastguard Worker } while (!final);
4121*89c4ff92SAndroid Build Coastguard Worker return 1;
4122*89c4ff92SAndroid Build Coastguard Worker }
4123*89c4ff92SAndroid Build Coastguard Worker
stbi__do_zlib(stbi__zbuf * a,char * obuf,int olen,int exp,int parse_header)4124*89c4ff92SAndroid Build Coastguard Worker static int stbi__do_zlib(stbi__zbuf *a, char *obuf, int olen, int exp, int parse_header)
4125*89c4ff92SAndroid Build Coastguard Worker {
4126*89c4ff92SAndroid Build Coastguard Worker a->zout_start = obuf;
4127*89c4ff92SAndroid Build Coastguard Worker a->zout = obuf;
4128*89c4ff92SAndroid Build Coastguard Worker a->zout_end = obuf + olen;
4129*89c4ff92SAndroid Build Coastguard Worker a->z_expandable = exp;
4130*89c4ff92SAndroid Build Coastguard Worker
4131*89c4ff92SAndroid Build Coastguard Worker return stbi__parse_zlib(a, parse_header);
4132*89c4ff92SAndroid Build Coastguard Worker }
4133*89c4ff92SAndroid Build Coastguard Worker
stbi_zlib_decode_malloc_guesssize(const char * buffer,int len,int initial_size,int * outlen)4134*89c4ff92SAndroid Build Coastguard Worker STBIDEF char *stbi_zlib_decode_malloc_guesssize(const char *buffer, int len, int initial_size, int *outlen)
4135*89c4ff92SAndroid Build Coastguard Worker {
4136*89c4ff92SAndroid Build Coastguard Worker stbi__zbuf a;
4137*89c4ff92SAndroid Build Coastguard Worker char *p = (char *) stbi__malloc(initial_size);
4138*89c4ff92SAndroid Build Coastguard Worker if (p == NULL) return NULL;
4139*89c4ff92SAndroid Build Coastguard Worker a.zbuffer = (stbi_uc *) buffer;
4140*89c4ff92SAndroid Build Coastguard Worker a.zbuffer_end = (stbi_uc *) buffer + len;
4141*89c4ff92SAndroid Build Coastguard Worker if (stbi__do_zlib(&a, p, initial_size, 1, 1)) {
4142*89c4ff92SAndroid Build Coastguard Worker if (outlen) *outlen = (int) (a.zout - a.zout_start);
4143*89c4ff92SAndroid Build Coastguard Worker return a.zout_start;
4144*89c4ff92SAndroid Build Coastguard Worker } else {
4145*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(a.zout_start);
4146*89c4ff92SAndroid Build Coastguard Worker return NULL;
4147*89c4ff92SAndroid Build Coastguard Worker }
4148*89c4ff92SAndroid Build Coastguard Worker }
4149*89c4ff92SAndroid Build Coastguard Worker
stbi_zlib_decode_malloc(char const * buffer,int len,int * outlen)4150*89c4ff92SAndroid Build Coastguard Worker STBIDEF char *stbi_zlib_decode_malloc(char const *buffer, int len, int *outlen)
4151*89c4ff92SAndroid Build Coastguard Worker {
4152*89c4ff92SAndroid Build Coastguard Worker return stbi_zlib_decode_malloc_guesssize(buffer, len, 16384, outlen);
4153*89c4ff92SAndroid Build Coastguard Worker }
4154*89c4ff92SAndroid Build Coastguard Worker
stbi_zlib_decode_malloc_guesssize_headerflag(const char * buffer,int len,int initial_size,int * outlen,int parse_header)4155*89c4ff92SAndroid Build Coastguard Worker STBIDEF char *stbi_zlib_decode_malloc_guesssize_headerflag(const char *buffer, int len, int initial_size, int *outlen, int parse_header)
4156*89c4ff92SAndroid Build Coastguard Worker {
4157*89c4ff92SAndroid Build Coastguard Worker stbi__zbuf a;
4158*89c4ff92SAndroid Build Coastguard Worker char *p = (char *) stbi__malloc(initial_size);
4159*89c4ff92SAndroid Build Coastguard Worker if (p == NULL) return NULL;
4160*89c4ff92SAndroid Build Coastguard Worker a.zbuffer = (stbi_uc *) buffer;
4161*89c4ff92SAndroid Build Coastguard Worker a.zbuffer_end = (stbi_uc *) buffer + len;
4162*89c4ff92SAndroid Build Coastguard Worker if (stbi__do_zlib(&a, p, initial_size, 1, parse_header)) {
4163*89c4ff92SAndroid Build Coastguard Worker if (outlen) *outlen = (int) (a.zout - a.zout_start);
4164*89c4ff92SAndroid Build Coastguard Worker return a.zout_start;
4165*89c4ff92SAndroid Build Coastguard Worker } else {
4166*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(a.zout_start);
4167*89c4ff92SAndroid Build Coastguard Worker return NULL;
4168*89c4ff92SAndroid Build Coastguard Worker }
4169*89c4ff92SAndroid Build Coastguard Worker }
4170*89c4ff92SAndroid Build Coastguard Worker
stbi_zlib_decode_buffer(char * obuffer,int olen,char const * ibuffer,int ilen)4171*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_zlib_decode_buffer(char *obuffer, int olen, char const *ibuffer, int ilen)
4172*89c4ff92SAndroid Build Coastguard Worker {
4173*89c4ff92SAndroid Build Coastguard Worker stbi__zbuf a;
4174*89c4ff92SAndroid Build Coastguard Worker a.zbuffer = (stbi_uc *) ibuffer;
4175*89c4ff92SAndroid Build Coastguard Worker a.zbuffer_end = (stbi_uc *) ibuffer + ilen;
4176*89c4ff92SAndroid Build Coastguard Worker if (stbi__do_zlib(&a, obuffer, olen, 0, 1))
4177*89c4ff92SAndroid Build Coastguard Worker return (int) (a.zout - a.zout_start);
4178*89c4ff92SAndroid Build Coastguard Worker else
4179*89c4ff92SAndroid Build Coastguard Worker return -1;
4180*89c4ff92SAndroid Build Coastguard Worker }
4181*89c4ff92SAndroid Build Coastguard Worker
stbi_zlib_decode_noheader_malloc(char const * buffer,int len,int * outlen)4182*89c4ff92SAndroid Build Coastguard Worker STBIDEF char *stbi_zlib_decode_noheader_malloc(char const *buffer, int len, int *outlen)
4183*89c4ff92SAndroid Build Coastguard Worker {
4184*89c4ff92SAndroid Build Coastguard Worker stbi__zbuf a;
4185*89c4ff92SAndroid Build Coastguard Worker char *p = (char *) stbi__malloc(16384);
4186*89c4ff92SAndroid Build Coastguard Worker if (p == NULL) return NULL;
4187*89c4ff92SAndroid Build Coastguard Worker a.zbuffer = (stbi_uc *) buffer;
4188*89c4ff92SAndroid Build Coastguard Worker a.zbuffer_end = (stbi_uc *) buffer+len;
4189*89c4ff92SAndroid Build Coastguard Worker if (stbi__do_zlib(&a, p, 16384, 1, 0)) {
4190*89c4ff92SAndroid Build Coastguard Worker if (outlen) *outlen = (int) (a.zout - a.zout_start);
4191*89c4ff92SAndroid Build Coastguard Worker return a.zout_start;
4192*89c4ff92SAndroid Build Coastguard Worker } else {
4193*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(a.zout_start);
4194*89c4ff92SAndroid Build Coastguard Worker return NULL;
4195*89c4ff92SAndroid Build Coastguard Worker }
4196*89c4ff92SAndroid Build Coastguard Worker }
4197*89c4ff92SAndroid Build Coastguard Worker
stbi_zlib_decode_noheader_buffer(char * obuffer,int olen,const char * ibuffer,int ilen)4198*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_zlib_decode_noheader_buffer(char *obuffer, int olen, const char *ibuffer, int ilen)
4199*89c4ff92SAndroid Build Coastguard Worker {
4200*89c4ff92SAndroid Build Coastguard Worker stbi__zbuf a;
4201*89c4ff92SAndroid Build Coastguard Worker a.zbuffer = (stbi_uc *) ibuffer;
4202*89c4ff92SAndroid Build Coastguard Worker a.zbuffer_end = (stbi_uc *) ibuffer + ilen;
4203*89c4ff92SAndroid Build Coastguard Worker if (stbi__do_zlib(&a, obuffer, olen, 0, 0))
4204*89c4ff92SAndroid Build Coastguard Worker return (int) (a.zout - a.zout_start);
4205*89c4ff92SAndroid Build Coastguard Worker else
4206*89c4ff92SAndroid Build Coastguard Worker return -1;
4207*89c4ff92SAndroid Build Coastguard Worker }
4208*89c4ff92SAndroid Build Coastguard Worker #endif
4209*89c4ff92SAndroid Build Coastguard Worker
4210*89c4ff92SAndroid Build Coastguard Worker // public domain "baseline" PNG decoder v0.10 Sean Barrett 2006-11-18
4211*89c4ff92SAndroid Build Coastguard Worker // simple implementation
4212*89c4ff92SAndroid Build Coastguard Worker // - only 8-bit samples
4213*89c4ff92SAndroid Build Coastguard Worker // - no CRC checking
4214*89c4ff92SAndroid Build Coastguard Worker // - allocates lots of intermediate memory
4215*89c4ff92SAndroid Build Coastguard Worker // - avoids problem of streaming data between subsystems
4216*89c4ff92SAndroid Build Coastguard Worker // - avoids explicit window management
4217*89c4ff92SAndroid Build Coastguard Worker // performance
4218*89c4ff92SAndroid Build Coastguard Worker // - uses stb_zlib, a PD zlib implementation with fast huffman decoding
4219*89c4ff92SAndroid Build Coastguard Worker
4220*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PNG
4221*89c4ff92SAndroid Build Coastguard Worker typedef struct
4222*89c4ff92SAndroid Build Coastguard Worker {
4223*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 length;
4224*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 type;
4225*89c4ff92SAndroid Build Coastguard Worker } stbi__pngchunk;
4226*89c4ff92SAndroid Build Coastguard Worker
stbi__get_chunk_header(stbi__context * s)4227*89c4ff92SAndroid Build Coastguard Worker static stbi__pngchunk stbi__get_chunk_header(stbi__context *s)
4228*89c4ff92SAndroid Build Coastguard Worker {
4229*89c4ff92SAndroid Build Coastguard Worker stbi__pngchunk c;
4230*89c4ff92SAndroid Build Coastguard Worker c.length = stbi__get32be(s);
4231*89c4ff92SAndroid Build Coastguard Worker c.type = stbi__get32be(s);
4232*89c4ff92SAndroid Build Coastguard Worker return c;
4233*89c4ff92SAndroid Build Coastguard Worker }
4234*89c4ff92SAndroid Build Coastguard Worker
stbi__check_png_header(stbi__context * s)4235*89c4ff92SAndroid Build Coastguard Worker static int stbi__check_png_header(stbi__context *s)
4236*89c4ff92SAndroid Build Coastguard Worker {
4237*89c4ff92SAndroid Build Coastguard Worker static stbi_uc png_sig[8] = { 137,80,78,71,13,10,26,10 };
4238*89c4ff92SAndroid Build Coastguard Worker int i;
4239*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < 8; ++i)
4240*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != png_sig[i]) return stbi__err("bad png sig","Not a PNG");
4241*89c4ff92SAndroid Build Coastguard Worker return 1;
4242*89c4ff92SAndroid Build Coastguard Worker }
4243*89c4ff92SAndroid Build Coastguard Worker
4244*89c4ff92SAndroid Build Coastguard Worker typedef struct
4245*89c4ff92SAndroid Build Coastguard Worker {
4246*89c4ff92SAndroid Build Coastguard Worker stbi__context *s;
4247*89c4ff92SAndroid Build Coastguard Worker stbi_uc *idata, *expanded, *out;
4248*89c4ff92SAndroid Build Coastguard Worker int depth;
4249*89c4ff92SAndroid Build Coastguard Worker } stbi__png;
4250*89c4ff92SAndroid Build Coastguard Worker
4251*89c4ff92SAndroid Build Coastguard Worker
4252*89c4ff92SAndroid Build Coastguard Worker enum {
4253*89c4ff92SAndroid Build Coastguard Worker STBI__F_none=0,
4254*89c4ff92SAndroid Build Coastguard Worker STBI__F_sub=1,
4255*89c4ff92SAndroid Build Coastguard Worker STBI__F_up=2,
4256*89c4ff92SAndroid Build Coastguard Worker STBI__F_avg=3,
4257*89c4ff92SAndroid Build Coastguard Worker STBI__F_paeth=4,
4258*89c4ff92SAndroid Build Coastguard Worker // synthetic filters used for first scanline to avoid needing a dummy row of 0s
4259*89c4ff92SAndroid Build Coastguard Worker STBI__F_avg_first,
4260*89c4ff92SAndroid Build Coastguard Worker STBI__F_paeth_first
4261*89c4ff92SAndroid Build Coastguard Worker };
4262*89c4ff92SAndroid Build Coastguard Worker
4263*89c4ff92SAndroid Build Coastguard Worker static stbi_uc first_row_filter[5] =
4264*89c4ff92SAndroid Build Coastguard Worker {
4265*89c4ff92SAndroid Build Coastguard Worker STBI__F_none,
4266*89c4ff92SAndroid Build Coastguard Worker STBI__F_sub,
4267*89c4ff92SAndroid Build Coastguard Worker STBI__F_none,
4268*89c4ff92SAndroid Build Coastguard Worker STBI__F_avg_first,
4269*89c4ff92SAndroid Build Coastguard Worker STBI__F_paeth_first
4270*89c4ff92SAndroid Build Coastguard Worker };
4271*89c4ff92SAndroid Build Coastguard Worker
stbi__paeth(int a,int b,int c)4272*89c4ff92SAndroid Build Coastguard Worker static int stbi__paeth(int a, int b, int c)
4273*89c4ff92SAndroid Build Coastguard Worker {
4274*89c4ff92SAndroid Build Coastguard Worker int p = a + b - c;
4275*89c4ff92SAndroid Build Coastguard Worker int pa = abs(p-a);
4276*89c4ff92SAndroid Build Coastguard Worker int pb = abs(p-b);
4277*89c4ff92SAndroid Build Coastguard Worker int pc = abs(p-c);
4278*89c4ff92SAndroid Build Coastguard Worker if (pa <= pb && pa <= pc) return a;
4279*89c4ff92SAndroid Build Coastguard Worker if (pb <= pc) return b;
4280*89c4ff92SAndroid Build Coastguard Worker return c;
4281*89c4ff92SAndroid Build Coastguard Worker }
4282*89c4ff92SAndroid Build Coastguard Worker
4283*89c4ff92SAndroid Build Coastguard Worker static stbi_uc stbi__depth_scale_table[9] = { 0, 0xff, 0x55, 0, 0x11, 0,0,0, 0x01 };
4284*89c4ff92SAndroid Build Coastguard Worker
4285*89c4ff92SAndroid Build Coastguard Worker // create the png data from post-deflated data
stbi__create_png_image_raw(stbi__png * a,stbi_uc * raw,stbi__uint32 raw_len,int out_n,stbi__uint32 x,stbi__uint32 y,int depth,int color)4286*89c4ff92SAndroid Build Coastguard Worker static int stbi__create_png_image_raw(stbi__png *a, stbi_uc *raw, stbi__uint32 raw_len, int out_n, stbi__uint32 x, stbi__uint32 y, int depth, int color)
4287*89c4ff92SAndroid Build Coastguard Worker {
4288*89c4ff92SAndroid Build Coastguard Worker int bytes = (depth == 16? 2 : 1);
4289*89c4ff92SAndroid Build Coastguard Worker stbi__context *s = a->s;
4290*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 i,j,stride = x*out_n*bytes;
4291*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 img_len, img_width_bytes;
4292*89c4ff92SAndroid Build Coastguard Worker int k;
4293*89c4ff92SAndroid Build Coastguard Worker int img_n = s->img_n; // copy it into a local for later
4294*89c4ff92SAndroid Build Coastguard Worker
4295*89c4ff92SAndroid Build Coastguard Worker int output_bytes = out_n*bytes;
4296*89c4ff92SAndroid Build Coastguard Worker int filter_bytes = img_n*bytes;
4297*89c4ff92SAndroid Build Coastguard Worker int width = x;
4298*89c4ff92SAndroid Build Coastguard Worker
4299*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(out_n == s->img_n || out_n == s->img_n+1);
4300*89c4ff92SAndroid Build Coastguard Worker a->out = (stbi_uc *) stbi__malloc_mad3(x, y, output_bytes, 0); // extra bytes to write off the end into
4301*89c4ff92SAndroid Build Coastguard Worker if (!a->out) return stbi__err("outofmem", "Out of memory");
4302*89c4ff92SAndroid Build Coastguard Worker
4303*89c4ff92SAndroid Build Coastguard Worker img_width_bytes = (((img_n * x * depth) + 7) >> 3);
4304*89c4ff92SAndroid Build Coastguard Worker img_len = (img_width_bytes + 1) * y;
4305*89c4ff92SAndroid Build Coastguard Worker // we used to check for exact match between raw_len and img_len on non-interlaced PNGs,
4306*89c4ff92SAndroid Build Coastguard Worker // but issue #276 reported a PNG in the wild that had extra data at the end (all zeros),
4307*89c4ff92SAndroid Build Coastguard Worker // so just check for raw_len < img_len always.
4308*89c4ff92SAndroid Build Coastguard Worker if (raw_len < img_len) return stbi__err("not enough pixels","Corrupt PNG");
4309*89c4ff92SAndroid Build Coastguard Worker
4310*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < y; ++j) {
4311*89c4ff92SAndroid Build Coastguard Worker stbi_uc *cur = a->out + stride*j;
4312*89c4ff92SAndroid Build Coastguard Worker stbi_uc *prior;
4313*89c4ff92SAndroid Build Coastguard Worker int filter = *raw++;
4314*89c4ff92SAndroid Build Coastguard Worker
4315*89c4ff92SAndroid Build Coastguard Worker if (filter > 4)
4316*89c4ff92SAndroid Build Coastguard Worker return stbi__err("invalid filter","Corrupt PNG");
4317*89c4ff92SAndroid Build Coastguard Worker
4318*89c4ff92SAndroid Build Coastguard Worker if (depth < 8) {
4319*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(img_width_bytes <= x);
4320*89c4ff92SAndroid Build Coastguard Worker cur += x*out_n - img_width_bytes; // store output to the rightmost img_len bytes, so we can decode in place
4321*89c4ff92SAndroid Build Coastguard Worker filter_bytes = 1;
4322*89c4ff92SAndroid Build Coastguard Worker width = img_width_bytes;
4323*89c4ff92SAndroid Build Coastguard Worker }
4324*89c4ff92SAndroid Build Coastguard Worker prior = cur - stride; // bugfix: need to compute this after 'cur +=' computation above
4325*89c4ff92SAndroid Build Coastguard Worker
4326*89c4ff92SAndroid Build Coastguard Worker // if first row, use special filter that doesn't sample previous row
4327*89c4ff92SAndroid Build Coastguard Worker if (j == 0) filter = first_row_filter[filter];
4328*89c4ff92SAndroid Build Coastguard Worker
4329*89c4ff92SAndroid Build Coastguard Worker // handle first byte explicitly
4330*89c4ff92SAndroid Build Coastguard Worker for (k=0; k < filter_bytes; ++k) {
4331*89c4ff92SAndroid Build Coastguard Worker switch (filter) {
4332*89c4ff92SAndroid Build Coastguard Worker case STBI__F_none : cur[k] = raw[k]; break;
4333*89c4ff92SAndroid Build Coastguard Worker case STBI__F_sub : cur[k] = raw[k]; break;
4334*89c4ff92SAndroid Build Coastguard Worker case STBI__F_up : cur[k] = STBI__BYTECAST(raw[k] + prior[k]); break;
4335*89c4ff92SAndroid Build Coastguard Worker case STBI__F_avg : cur[k] = STBI__BYTECAST(raw[k] + (prior[k]>>1)); break;
4336*89c4ff92SAndroid Build Coastguard Worker case STBI__F_paeth : cur[k] = STBI__BYTECAST(raw[k] + stbi__paeth(0,prior[k],0)); break;
4337*89c4ff92SAndroid Build Coastguard Worker case STBI__F_avg_first : cur[k] = raw[k]; break;
4338*89c4ff92SAndroid Build Coastguard Worker case STBI__F_paeth_first: cur[k] = raw[k]; break;
4339*89c4ff92SAndroid Build Coastguard Worker }
4340*89c4ff92SAndroid Build Coastguard Worker }
4341*89c4ff92SAndroid Build Coastguard Worker
4342*89c4ff92SAndroid Build Coastguard Worker if (depth == 8) {
4343*89c4ff92SAndroid Build Coastguard Worker if (img_n != out_n)
4344*89c4ff92SAndroid Build Coastguard Worker cur[img_n] = 255; // first pixel
4345*89c4ff92SAndroid Build Coastguard Worker raw += img_n;
4346*89c4ff92SAndroid Build Coastguard Worker cur += out_n;
4347*89c4ff92SAndroid Build Coastguard Worker prior += out_n;
4348*89c4ff92SAndroid Build Coastguard Worker } else if (depth == 16) {
4349*89c4ff92SAndroid Build Coastguard Worker if (img_n != out_n) {
4350*89c4ff92SAndroid Build Coastguard Worker cur[filter_bytes] = 255; // first pixel top byte
4351*89c4ff92SAndroid Build Coastguard Worker cur[filter_bytes+1] = 255; // first pixel bottom byte
4352*89c4ff92SAndroid Build Coastguard Worker }
4353*89c4ff92SAndroid Build Coastguard Worker raw += filter_bytes;
4354*89c4ff92SAndroid Build Coastguard Worker cur += output_bytes;
4355*89c4ff92SAndroid Build Coastguard Worker prior += output_bytes;
4356*89c4ff92SAndroid Build Coastguard Worker } else {
4357*89c4ff92SAndroid Build Coastguard Worker raw += 1;
4358*89c4ff92SAndroid Build Coastguard Worker cur += 1;
4359*89c4ff92SAndroid Build Coastguard Worker prior += 1;
4360*89c4ff92SAndroid Build Coastguard Worker }
4361*89c4ff92SAndroid Build Coastguard Worker
4362*89c4ff92SAndroid Build Coastguard Worker // this is a little gross, so that we don't switch per-pixel or per-component
4363*89c4ff92SAndroid Build Coastguard Worker if (depth < 8 || img_n == out_n) {
4364*89c4ff92SAndroid Build Coastguard Worker int nk = (width - 1)*filter_bytes;
4365*89c4ff92SAndroid Build Coastguard Worker #define STBI__CASE(f) \
4366*89c4ff92SAndroid Build Coastguard Worker case f: \
4367*89c4ff92SAndroid Build Coastguard Worker for (k=0; k < nk; ++k)
4368*89c4ff92SAndroid Build Coastguard Worker switch (filter) {
4369*89c4ff92SAndroid Build Coastguard Worker // "none" filter turns into a memcpy here; make that explicit.
4370*89c4ff92SAndroid Build Coastguard Worker case STBI__F_none: memcpy(cur, raw, nk); break;
4371*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_sub) { cur[k] = STBI__BYTECAST(raw[k] + cur[k-filter_bytes]); } break;
4372*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_up) { cur[k] = STBI__BYTECAST(raw[k] + prior[k]); } break;
4373*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_avg) { cur[k] = STBI__BYTECAST(raw[k] + ((prior[k] + cur[k-filter_bytes])>>1)); } break;
4374*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_paeth) { cur[k] = STBI__BYTECAST(raw[k] + stbi__paeth(cur[k-filter_bytes],prior[k],prior[k-filter_bytes])); } break;
4375*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_avg_first) { cur[k] = STBI__BYTECAST(raw[k] + (cur[k-filter_bytes] >> 1)); } break;
4376*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_paeth_first) { cur[k] = STBI__BYTECAST(raw[k] + stbi__paeth(cur[k-filter_bytes],0,0)); } break;
4377*89c4ff92SAndroid Build Coastguard Worker }
4378*89c4ff92SAndroid Build Coastguard Worker #undef STBI__CASE
4379*89c4ff92SAndroid Build Coastguard Worker raw += nk;
4380*89c4ff92SAndroid Build Coastguard Worker } else {
4381*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(img_n+1 == out_n);
4382*89c4ff92SAndroid Build Coastguard Worker #define STBI__CASE(f) \
4383*89c4ff92SAndroid Build Coastguard Worker case f: \
4384*89c4ff92SAndroid Build Coastguard Worker for (i=x-1; i >= 1; --i, cur[filter_bytes]=255,raw+=filter_bytes,cur+=output_bytes,prior+=output_bytes) \
4385*89c4ff92SAndroid Build Coastguard Worker for (k=0; k < filter_bytes; ++k)
4386*89c4ff92SAndroid Build Coastguard Worker switch (filter) {
4387*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_none) { cur[k] = raw[k]; } break;
4388*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_sub) { cur[k] = STBI__BYTECAST(raw[k] + cur[k- output_bytes]); } break;
4389*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_up) { cur[k] = STBI__BYTECAST(raw[k] + prior[k]); } break;
4390*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_avg) { cur[k] = STBI__BYTECAST(raw[k] + ((prior[k] + cur[k- output_bytes])>>1)); } break;
4391*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_paeth) { cur[k] = STBI__BYTECAST(raw[k] + stbi__paeth(cur[k- output_bytes],prior[k],prior[k- output_bytes])); } break;
4392*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_avg_first) { cur[k] = STBI__BYTECAST(raw[k] + (cur[k- output_bytes] >> 1)); } break;
4393*89c4ff92SAndroid Build Coastguard Worker STBI__CASE(STBI__F_paeth_first) { cur[k] = STBI__BYTECAST(raw[k] + stbi__paeth(cur[k- output_bytes],0,0)); } break;
4394*89c4ff92SAndroid Build Coastguard Worker }
4395*89c4ff92SAndroid Build Coastguard Worker #undef STBI__CASE
4396*89c4ff92SAndroid Build Coastguard Worker
4397*89c4ff92SAndroid Build Coastguard Worker // the loop above sets the high byte of the pixels' alpha, but for
4398*89c4ff92SAndroid Build Coastguard Worker // 16 bit png files we also need the low byte set. we'll do that here.
4399*89c4ff92SAndroid Build Coastguard Worker if (depth == 16) {
4400*89c4ff92SAndroid Build Coastguard Worker cur = a->out + stride*j; // start at the beginning of the row again
4401*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < x; ++i,cur+=output_bytes) {
4402*89c4ff92SAndroid Build Coastguard Worker cur[filter_bytes+1] = 255;
4403*89c4ff92SAndroid Build Coastguard Worker }
4404*89c4ff92SAndroid Build Coastguard Worker }
4405*89c4ff92SAndroid Build Coastguard Worker }
4406*89c4ff92SAndroid Build Coastguard Worker }
4407*89c4ff92SAndroid Build Coastguard Worker
4408*89c4ff92SAndroid Build Coastguard Worker // we make a separate pass to expand bits to pixels; for performance,
4409*89c4ff92SAndroid Build Coastguard Worker // this could run two scanlines behind the above code, so it won't
4410*89c4ff92SAndroid Build Coastguard Worker // intefere with filtering but will still be in the cache.
4411*89c4ff92SAndroid Build Coastguard Worker if (depth < 8) {
4412*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < y; ++j) {
4413*89c4ff92SAndroid Build Coastguard Worker stbi_uc *cur = a->out + stride*j;
4414*89c4ff92SAndroid Build Coastguard Worker stbi_uc *in = a->out + stride*j + x*out_n - img_width_bytes;
4415*89c4ff92SAndroid Build Coastguard Worker // unpack 1/2/4-bit into a 8-bit buffer. allows us to keep the common 8-bit path optimal at minimal cost for 1/2/4-bit
4416*89c4ff92SAndroid Build Coastguard Worker // png guarante byte alignment, if width is not multiple of 8/4/2 we'll decode dummy trailing data that will be skipped in the later loop
4417*89c4ff92SAndroid Build Coastguard Worker stbi_uc scale = (color == 0) ? stbi__depth_scale_table[depth] : 1; // scale grayscale values to 0..255 range
4418*89c4ff92SAndroid Build Coastguard Worker
4419*89c4ff92SAndroid Build Coastguard Worker // note that the final byte might overshoot and write more data than desired.
4420*89c4ff92SAndroid Build Coastguard Worker // we can allocate enough data that this never writes out of memory, but it
4421*89c4ff92SAndroid Build Coastguard Worker // could also overwrite the next scanline. can it overwrite non-empty data
4422*89c4ff92SAndroid Build Coastguard Worker // on the next scanline? yes, consider 1-pixel-wide scanlines with 1-bit-per-pixel.
4423*89c4ff92SAndroid Build Coastguard Worker // so we need to explicitly clamp the final ones
4424*89c4ff92SAndroid Build Coastguard Worker
4425*89c4ff92SAndroid Build Coastguard Worker if (depth == 4) {
4426*89c4ff92SAndroid Build Coastguard Worker for (k=x*img_n; k >= 2; k-=2, ++in) {
4427*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 4) );
4428*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in ) & 0x0f);
4429*89c4ff92SAndroid Build Coastguard Worker }
4430*89c4ff92SAndroid Build Coastguard Worker if (k > 0) *cur++ = scale * ((*in >> 4) );
4431*89c4ff92SAndroid Build Coastguard Worker } else if (depth == 2) {
4432*89c4ff92SAndroid Build Coastguard Worker for (k=x*img_n; k >= 4; k-=4, ++in) {
4433*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 6) );
4434*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 4) & 0x03);
4435*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 2) & 0x03);
4436*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in ) & 0x03);
4437*89c4ff92SAndroid Build Coastguard Worker }
4438*89c4ff92SAndroid Build Coastguard Worker if (k > 0) *cur++ = scale * ((*in >> 6) );
4439*89c4ff92SAndroid Build Coastguard Worker if (k > 1) *cur++ = scale * ((*in >> 4) & 0x03);
4440*89c4ff92SAndroid Build Coastguard Worker if (k > 2) *cur++ = scale * ((*in >> 2) & 0x03);
4441*89c4ff92SAndroid Build Coastguard Worker } else if (depth == 1) {
4442*89c4ff92SAndroid Build Coastguard Worker for (k=x*img_n; k >= 8; k-=8, ++in) {
4443*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 7) );
4444*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 6) & 0x01);
4445*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 5) & 0x01);
4446*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 4) & 0x01);
4447*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 3) & 0x01);
4448*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 2) & 0x01);
4449*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in >> 1) & 0x01);
4450*89c4ff92SAndroid Build Coastguard Worker *cur++ = scale * ((*in ) & 0x01);
4451*89c4ff92SAndroid Build Coastguard Worker }
4452*89c4ff92SAndroid Build Coastguard Worker if (k > 0) *cur++ = scale * ((*in >> 7) );
4453*89c4ff92SAndroid Build Coastguard Worker if (k > 1) *cur++ = scale * ((*in >> 6) & 0x01);
4454*89c4ff92SAndroid Build Coastguard Worker if (k > 2) *cur++ = scale * ((*in >> 5) & 0x01);
4455*89c4ff92SAndroid Build Coastguard Worker if (k > 3) *cur++ = scale * ((*in >> 4) & 0x01);
4456*89c4ff92SAndroid Build Coastguard Worker if (k > 4) *cur++ = scale * ((*in >> 3) & 0x01);
4457*89c4ff92SAndroid Build Coastguard Worker if (k > 5) *cur++ = scale * ((*in >> 2) & 0x01);
4458*89c4ff92SAndroid Build Coastguard Worker if (k > 6) *cur++ = scale * ((*in >> 1) & 0x01);
4459*89c4ff92SAndroid Build Coastguard Worker }
4460*89c4ff92SAndroid Build Coastguard Worker if (img_n != out_n) {
4461*89c4ff92SAndroid Build Coastguard Worker int q;
4462*89c4ff92SAndroid Build Coastguard Worker // insert alpha = 255
4463*89c4ff92SAndroid Build Coastguard Worker cur = a->out + stride*j;
4464*89c4ff92SAndroid Build Coastguard Worker if (img_n == 1) {
4465*89c4ff92SAndroid Build Coastguard Worker for (q=x-1; q >= 0; --q) {
4466*89c4ff92SAndroid Build Coastguard Worker cur[q*2+1] = 255;
4467*89c4ff92SAndroid Build Coastguard Worker cur[q*2+0] = cur[q];
4468*89c4ff92SAndroid Build Coastguard Worker }
4469*89c4ff92SAndroid Build Coastguard Worker } else {
4470*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(img_n == 3);
4471*89c4ff92SAndroid Build Coastguard Worker for (q=x-1; q >= 0; --q) {
4472*89c4ff92SAndroid Build Coastguard Worker cur[q*4+3] = 255;
4473*89c4ff92SAndroid Build Coastguard Worker cur[q*4+2] = cur[q*3+2];
4474*89c4ff92SAndroid Build Coastguard Worker cur[q*4+1] = cur[q*3+1];
4475*89c4ff92SAndroid Build Coastguard Worker cur[q*4+0] = cur[q*3+0];
4476*89c4ff92SAndroid Build Coastguard Worker }
4477*89c4ff92SAndroid Build Coastguard Worker }
4478*89c4ff92SAndroid Build Coastguard Worker }
4479*89c4ff92SAndroid Build Coastguard Worker }
4480*89c4ff92SAndroid Build Coastguard Worker } else if (depth == 16) {
4481*89c4ff92SAndroid Build Coastguard Worker // force the image data from big-endian to platform-native.
4482*89c4ff92SAndroid Build Coastguard Worker // this is done in a separate pass due to the decoding relying
4483*89c4ff92SAndroid Build Coastguard Worker // on the data being untouched, but could probably be done
4484*89c4ff92SAndroid Build Coastguard Worker // per-line during decode if care is taken.
4485*89c4ff92SAndroid Build Coastguard Worker stbi_uc *cur = a->out;
4486*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *cur16 = (stbi__uint16*)cur;
4487*89c4ff92SAndroid Build Coastguard Worker
4488*89c4ff92SAndroid Build Coastguard Worker for(i=0; i < x*y*out_n; ++i,cur16++,cur+=2) {
4489*89c4ff92SAndroid Build Coastguard Worker *cur16 = (cur[0] << 8) | cur[1];
4490*89c4ff92SAndroid Build Coastguard Worker }
4491*89c4ff92SAndroid Build Coastguard Worker }
4492*89c4ff92SAndroid Build Coastguard Worker
4493*89c4ff92SAndroid Build Coastguard Worker return 1;
4494*89c4ff92SAndroid Build Coastguard Worker }
4495*89c4ff92SAndroid Build Coastguard Worker
stbi__create_png_image(stbi__png * a,stbi_uc * image_data,stbi__uint32 image_data_len,int out_n,int depth,int color,int interlaced)4496*89c4ff92SAndroid Build Coastguard Worker static int stbi__create_png_image(stbi__png *a, stbi_uc *image_data, stbi__uint32 image_data_len, int out_n, int depth, int color, int interlaced)
4497*89c4ff92SAndroid Build Coastguard Worker {
4498*89c4ff92SAndroid Build Coastguard Worker int bytes = (depth == 16 ? 2 : 1);
4499*89c4ff92SAndroid Build Coastguard Worker int out_bytes = out_n * bytes;
4500*89c4ff92SAndroid Build Coastguard Worker stbi_uc *final;
4501*89c4ff92SAndroid Build Coastguard Worker int p;
4502*89c4ff92SAndroid Build Coastguard Worker if (!interlaced)
4503*89c4ff92SAndroid Build Coastguard Worker return stbi__create_png_image_raw(a, image_data, image_data_len, out_n, a->s->img_x, a->s->img_y, depth, color);
4504*89c4ff92SAndroid Build Coastguard Worker
4505*89c4ff92SAndroid Build Coastguard Worker // de-interlacing
4506*89c4ff92SAndroid Build Coastguard Worker final = (stbi_uc *) stbi__malloc_mad3(a->s->img_x, a->s->img_y, out_bytes, 0);
4507*89c4ff92SAndroid Build Coastguard Worker for (p=0; p < 7; ++p) {
4508*89c4ff92SAndroid Build Coastguard Worker int xorig[] = { 0,4,0,2,0,1,0 };
4509*89c4ff92SAndroid Build Coastguard Worker int yorig[] = { 0,0,4,0,2,0,1 };
4510*89c4ff92SAndroid Build Coastguard Worker int xspc[] = { 8,8,4,4,2,2,1 };
4511*89c4ff92SAndroid Build Coastguard Worker int yspc[] = { 8,8,8,4,4,2,2 };
4512*89c4ff92SAndroid Build Coastguard Worker int i,j,x,y;
4513*89c4ff92SAndroid Build Coastguard Worker // pass1_x[4] = 0, pass1_x[5] = 1, pass1_x[12] = 1
4514*89c4ff92SAndroid Build Coastguard Worker x = (a->s->img_x - xorig[p] + xspc[p]-1) / xspc[p];
4515*89c4ff92SAndroid Build Coastguard Worker y = (a->s->img_y - yorig[p] + yspc[p]-1) / yspc[p];
4516*89c4ff92SAndroid Build Coastguard Worker if (x && y) {
4517*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 img_len = ((((a->s->img_n * x * depth) + 7) >> 3) + 1) * y;
4518*89c4ff92SAndroid Build Coastguard Worker if (!stbi__create_png_image_raw(a, image_data, image_data_len, out_n, x, y, depth, color)) {
4519*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(final);
4520*89c4ff92SAndroid Build Coastguard Worker return 0;
4521*89c4ff92SAndroid Build Coastguard Worker }
4522*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < y; ++j) {
4523*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < x; ++i) {
4524*89c4ff92SAndroid Build Coastguard Worker int out_y = j*yspc[p]+yorig[p];
4525*89c4ff92SAndroid Build Coastguard Worker int out_x = i*xspc[p]+xorig[p];
4526*89c4ff92SAndroid Build Coastguard Worker memcpy(final + out_y*a->s->img_x*out_bytes + out_x*out_bytes,
4527*89c4ff92SAndroid Build Coastguard Worker a->out + (j*x+i)*out_bytes, out_bytes);
4528*89c4ff92SAndroid Build Coastguard Worker }
4529*89c4ff92SAndroid Build Coastguard Worker }
4530*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(a->out);
4531*89c4ff92SAndroid Build Coastguard Worker image_data += img_len;
4532*89c4ff92SAndroid Build Coastguard Worker image_data_len -= img_len;
4533*89c4ff92SAndroid Build Coastguard Worker }
4534*89c4ff92SAndroid Build Coastguard Worker }
4535*89c4ff92SAndroid Build Coastguard Worker a->out = final;
4536*89c4ff92SAndroid Build Coastguard Worker
4537*89c4ff92SAndroid Build Coastguard Worker return 1;
4538*89c4ff92SAndroid Build Coastguard Worker }
4539*89c4ff92SAndroid Build Coastguard Worker
stbi__compute_transparency(stbi__png * z,stbi_uc tc[3],int out_n)4540*89c4ff92SAndroid Build Coastguard Worker static int stbi__compute_transparency(stbi__png *z, stbi_uc tc[3], int out_n)
4541*89c4ff92SAndroid Build Coastguard Worker {
4542*89c4ff92SAndroid Build Coastguard Worker stbi__context *s = z->s;
4543*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 i, pixel_count = s->img_x * s->img_y;
4544*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p = z->out;
4545*89c4ff92SAndroid Build Coastguard Worker
4546*89c4ff92SAndroid Build Coastguard Worker // compute color-based transparency, assuming we've
4547*89c4ff92SAndroid Build Coastguard Worker // already got 255 as the alpha value in the output
4548*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(out_n == 2 || out_n == 4);
4549*89c4ff92SAndroid Build Coastguard Worker
4550*89c4ff92SAndroid Build Coastguard Worker if (out_n == 2) {
4551*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < pixel_count; ++i) {
4552*89c4ff92SAndroid Build Coastguard Worker p[1] = (p[0] == tc[0] ? 0 : 255);
4553*89c4ff92SAndroid Build Coastguard Worker p += 2;
4554*89c4ff92SAndroid Build Coastguard Worker }
4555*89c4ff92SAndroid Build Coastguard Worker } else {
4556*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < pixel_count; ++i) {
4557*89c4ff92SAndroid Build Coastguard Worker if (p[0] == tc[0] && p[1] == tc[1] && p[2] == tc[2])
4558*89c4ff92SAndroid Build Coastguard Worker p[3] = 0;
4559*89c4ff92SAndroid Build Coastguard Worker p += 4;
4560*89c4ff92SAndroid Build Coastguard Worker }
4561*89c4ff92SAndroid Build Coastguard Worker }
4562*89c4ff92SAndroid Build Coastguard Worker return 1;
4563*89c4ff92SAndroid Build Coastguard Worker }
4564*89c4ff92SAndroid Build Coastguard Worker
stbi__compute_transparency16(stbi__png * z,stbi__uint16 tc[3],int out_n)4565*89c4ff92SAndroid Build Coastguard Worker static int stbi__compute_transparency16(stbi__png *z, stbi__uint16 tc[3], int out_n)
4566*89c4ff92SAndroid Build Coastguard Worker {
4567*89c4ff92SAndroid Build Coastguard Worker stbi__context *s = z->s;
4568*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 i, pixel_count = s->img_x * s->img_y;
4569*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *p = (stbi__uint16*) z->out;
4570*89c4ff92SAndroid Build Coastguard Worker
4571*89c4ff92SAndroid Build Coastguard Worker // compute color-based transparency, assuming we've
4572*89c4ff92SAndroid Build Coastguard Worker // already got 65535 as the alpha value in the output
4573*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(out_n == 2 || out_n == 4);
4574*89c4ff92SAndroid Build Coastguard Worker
4575*89c4ff92SAndroid Build Coastguard Worker if (out_n == 2) {
4576*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < pixel_count; ++i) {
4577*89c4ff92SAndroid Build Coastguard Worker p[1] = (p[0] == tc[0] ? 0 : 65535);
4578*89c4ff92SAndroid Build Coastguard Worker p += 2;
4579*89c4ff92SAndroid Build Coastguard Worker }
4580*89c4ff92SAndroid Build Coastguard Worker } else {
4581*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < pixel_count; ++i) {
4582*89c4ff92SAndroid Build Coastguard Worker if (p[0] == tc[0] && p[1] == tc[1] && p[2] == tc[2])
4583*89c4ff92SAndroid Build Coastguard Worker p[3] = 0;
4584*89c4ff92SAndroid Build Coastguard Worker p += 4;
4585*89c4ff92SAndroid Build Coastguard Worker }
4586*89c4ff92SAndroid Build Coastguard Worker }
4587*89c4ff92SAndroid Build Coastguard Worker return 1;
4588*89c4ff92SAndroid Build Coastguard Worker }
4589*89c4ff92SAndroid Build Coastguard Worker
stbi__expand_png_palette(stbi__png * a,stbi_uc * palette,int len,int pal_img_n)4590*89c4ff92SAndroid Build Coastguard Worker static int stbi__expand_png_palette(stbi__png *a, stbi_uc *palette, int len, int pal_img_n)
4591*89c4ff92SAndroid Build Coastguard Worker {
4592*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 i, pixel_count = a->s->img_x * a->s->img_y;
4593*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p, *temp_out, *orig = a->out;
4594*89c4ff92SAndroid Build Coastguard Worker
4595*89c4ff92SAndroid Build Coastguard Worker p = (stbi_uc *) stbi__malloc_mad2(pixel_count, pal_img_n, 0);
4596*89c4ff92SAndroid Build Coastguard Worker if (p == NULL) return stbi__err("outofmem", "Out of memory");
4597*89c4ff92SAndroid Build Coastguard Worker
4598*89c4ff92SAndroid Build Coastguard Worker // between here and free(out) below, exitting would leak
4599*89c4ff92SAndroid Build Coastguard Worker temp_out = p;
4600*89c4ff92SAndroid Build Coastguard Worker
4601*89c4ff92SAndroid Build Coastguard Worker if (pal_img_n == 3) {
4602*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < pixel_count; ++i) {
4603*89c4ff92SAndroid Build Coastguard Worker int n = orig[i]*4;
4604*89c4ff92SAndroid Build Coastguard Worker p[0] = palette[n ];
4605*89c4ff92SAndroid Build Coastguard Worker p[1] = palette[n+1];
4606*89c4ff92SAndroid Build Coastguard Worker p[2] = palette[n+2];
4607*89c4ff92SAndroid Build Coastguard Worker p += 3;
4608*89c4ff92SAndroid Build Coastguard Worker }
4609*89c4ff92SAndroid Build Coastguard Worker } else {
4610*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < pixel_count; ++i) {
4611*89c4ff92SAndroid Build Coastguard Worker int n = orig[i]*4;
4612*89c4ff92SAndroid Build Coastguard Worker p[0] = palette[n ];
4613*89c4ff92SAndroid Build Coastguard Worker p[1] = palette[n+1];
4614*89c4ff92SAndroid Build Coastguard Worker p[2] = palette[n+2];
4615*89c4ff92SAndroid Build Coastguard Worker p[3] = palette[n+3];
4616*89c4ff92SAndroid Build Coastguard Worker p += 4;
4617*89c4ff92SAndroid Build Coastguard Worker }
4618*89c4ff92SAndroid Build Coastguard Worker }
4619*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(a->out);
4620*89c4ff92SAndroid Build Coastguard Worker a->out = temp_out;
4621*89c4ff92SAndroid Build Coastguard Worker
4622*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(len);
4623*89c4ff92SAndroid Build Coastguard Worker
4624*89c4ff92SAndroid Build Coastguard Worker return 1;
4625*89c4ff92SAndroid Build Coastguard Worker }
4626*89c4ff92SAndroid Build Coastguard Worker
4627*89c4ff92SAndroid Build Coastguard Worker static int stbi__unpremultiply_on_load = 0;
4628*89c4ff92SAndroid Build Coastguard Worker static int stbi__de_iphone_flag = 0;
4629*89c4ff92SAndroid Build Coastguard Worker
stbi_set_unpremultiply_on_load(int flag_true_if_should_unpremultiply)4630*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_set_unpremultiply_on_load(int flag_true_if_should_unpremultiply)
4631*89c4ff92SAndroid Build Coastguard Worker {
4632*89c4ff92SAndroid Build Coastguard Worker stbi__unpremultiply_on_load = flag_true_if_should_unpremultiply;
4633*89c4ff92SAndroid Build Coastguard Worker }
4634*89c4ff92SAndroid Build Coastguard Worker
stbi_convert_iphone_png_to_rgb(int flag_true_if_should_convert)4635*89c4ff92SAndroid Build Coastguard Worker STBIDEF void stbi_convert_iphone_png_to_rgb(int flag_true_if_should_convert)
4636*89c4ff92SAndroid Build Coastguard Worker {
4637*89c4ff92SAndroid Build Coastguard Worker stbi__de_iphone_flag = flag_true_if_should_convert;
4638*89c4ff92SAndroid Build Coastguard Worker }
4639*89c4ff92SAndroid Build Coastguard Worker
stbi__de_iphone(stbi__png * z)4640*89c4ff92SAndroid Build Coastguard Worker static void stbi__de_iphone(stbi__png *z)
4641*89c4ff92SAndroid Build Coastguard Worker {
4642*89c4ff92SAndroid Build Coastguard Worker stbi__context *s = z->s;
4643*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 i, pixel_count = s->img_x * s->img_y;
4644*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p = z->out;
4645*89c4ff92SAndroid Build Coastguard Worker
4646*89c4ff92SAndroid Build Coastguard Worker if (s->img_out_n == 3) { // convert bgr to rgb
4647*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < pixel_count; ++i) {
4648*89c4ff92SAndroid Build Coastguard Worker stbi_uc t = p[0];
4649*89c4ff92SAndroid Build Coastguard Worker p[0] = p[2];
4650*89c4ff92SAndroid Build Coastguard Worker p[2] = t;
4651*89c4ff92SAndroid Build Coastguard Worker p += 3;
4652*89c4ff92SAndroid Build Coastguard Worker }
4653*89c4ff92SAndroid Build Coastguard Worker } else {
4654*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(s->img_out_n == 4);
4655*89c4ff92SAndroid Build Coastguard Worker if (stbi__unpremultiply_on_load) {
4656*89c4ff92SAndroid Build Coastguard Worker // convert bgr to rgb and unpremultiply
4657*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < pixel_count; ++i) {
4658*89c4ff92SAndroid Build Coastguard Worker stbi_uc a = p[3];
4659*89c4ff92SAndroid Build Coastguard Worker stbi_uc t = p[0];
4660*89c4ff92SAndroid Build Coastguard Worker if (a) {
4661*89c4ff92SAndroid Build Coastguard Worker stbi_uc half = a / 2;
4662*89c4ff92SAndroid Build Coastguard Worker p[0] = (p[2] * 255 + half) / a;
4663*89c4ff92SAndroid Build Coastguard Worker p[1] = (p[1] * 255 + half) / a;
4664*89c4ff92SAndroid Build Coastguard Worker p[2] = ( t * 255 + half) / a;
4665*89c4ff92SAndroid Build Coastguard Worker } else {
4666*89c4ff92SAndroid Build Coastguard Worker p[0] = p[2];
4667*89c4ff92SAndroid Build Coastguard Worker p[2] = t;
4668*89c4ff92SAndroid Build Coastguard Worker }
4669*89c4ff92SAndroid Build Coastguard Worker p += 4;
4670*89c4ff92SAndroid Build Coastguard Worker }
4671*89c4ff92SAndroid Build Coastguard Worker } else {
4672*89c4ff92SAndroid Build Coastguard Worker // convert bgr to rgb
4673*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < pixel_count; ++i) {
4674*89c4ff92SAndroid Build Coastguard Worker stbi_uc t = p[0];
4675*89c4ff92SAndroid Build Coastguard Worker p[0] = p[2];
4676*89c4ff92SAndroid Build Coastguard Worker p[2] = t;
4677*89c4ff92SAndroid Build Coastguard Worker p += 4;
4678*89c4ff92SAndroid Build Coastguard Worker }
4679*89c4ff92SAndroid Build Coastguard Worker }
4680*89c4ff92SAndroid Build Coastguard Worker }
4681*89c4ff92SAndroid Build Coastguard Worker }
4682*89c4ff92SAndroid Build Coastguard Worker
4683*89c4ff92SAndroid Build Coastguard Worker #define STBI__PNG_TYPE(a,b,c,d) (((a) << 24) + ((b) << 16) + ((c) << 8) + (d))
4684*89c4ff92SAndroid Build Coastguard Worker
stbi__parse_png_file(stbi__png * z,int scan,int req_comp)4685*89c4ff92SAndroid Build Coastguard Worker static int stbi__parse_png_file(stbi__png *z, int scan, int req_comp)
4686*89c4ff92SAndroid Build Coastguard Worker {
4687*89c4ff92SAndroid Build Coastguard Worker stbi_uc palette[1024], pal_img_n=0;
4688*89c4ff92SAndroid Build Coastguard Worker stbi_uc has_trans=0, tc[3];
4689*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 tc16[3];
4690*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 ioff=0, idata_limit=0, i, pal_len=0;
4691*89c4ff92SAndroid Build Coastguard Worker int first=1,k,interlace=0, color=0, is_iphone=0;
4692*89c4ff92SAndroid Build Coastguard Worker stbi__context *s = z->s;
4693*89c4ff92SAndroid Build Coastguard Worker
4694*89c4ff92SAndroid Build Coastguard Worker z->expanded = NULL;
4695*89c4ff92SAndroid Build Coastguard Worker z->idata = NULL;
4696*89c4ff92SAndroid Build Coastguard Worker z->out = NULL;
4697*89c4ff92SAndroid Build Coastguard Worker
4698*89c4ff92SAndroid Build Coastguard Worker if (!stbi__check_png_header(s)) return 0;
4699*89c4ff92SAndroid Build Coastguard Worker
4700*89c4ff92SAndroid Build Coastguard Worker if (scan == STBI__SCAN_type) return 1;
4701*89c4ff92SAndroid Build Coastguard Worker
4702*89c4ff92SAndroid Build Coastguard Worker for (;;) {
4703*89c4ff92SAndroid Build Coastguard Worker stbi__pngchunk c = stbi__get_chunk_header(s);
4704*89c4ff92SAndroid Build Coastguard Worker switch (c.type) {
4705*89c4ff92SAndroid Build Coastguard Worker case STBI__PNG_TYPE('C','g','B','I'):
4706*89c4ff92SAndroid Build Coastguard Worker is_iphone = 1;
4707*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, c.length);
4708*89c4ff92SAndroid Build Coastguard Worker break;
4709*89c4ff92SAndroid Build Coastguard Worker case STBI__PNG_TYPE('I','H','D','R'): {
4710*89c4ff92SAndroid Build Coastguard Worker int comp,filter;
4711*89c4ff92SAndroid Build Coastguard Worker if (!first) return stbi__err("multiple IHDR","Corrupt PNG");
4712*89c4ff92SAndroid Build Coastguard Worker first = 0;
4713*89c4ff92SAndroid Build Coastguard Worker if (c.length != 13) return stbi__err("bad IHDR len","Corrupt PNG");
4714*89c4ff92SAndroid Build Coastguard Worker s->img_x = stbi__get32be(s); if (s->img_x > (1 << 24)) return stbi__err("too large","Very large image (corrupt?)");
4715*89c4ff92SAndroid Build Coastguard Worker s->img_y = stbi__get32be(s); if (s->img_y > (1 << 24)) return stbi__err("too large","Very large image (corrupt?)");
4716*89c4ff92SAndroid Build Coastguard Worker z->depth = stbi__get8(s); if (z->depth != 1 && z->depth != 2 && z->depth != 4 && z->depth != 8 && z->depth != 16) return stbi__err("1/2/4/8/16-bit only","PNG not supported: 1/2/4/8/16-bit only");
4717*89c4ff92SAndroid Build Coastguard Worker color = stbi__get8(s); if (color > 6) return stbi__err("bad ctype","Corrupt PNG");
4718*89c4ff92SAndroid Build Coastguard Worker if (color == 3 && z->depth == 16) return stbi__err("bad ctype","Corrupt PNG");
4719*89c4ff92SAndroid Build Coastguard Worker if (color == 3) pal_img_n = 3; else if (color & 1) return stbi__err("bad ctype","Corrupt PNG");
4720*89c4ff92SAndroid Build Coastguard Worker comp = stbi__get8(s); if (comp) return stbi__err("bad comp method","Corrupt PNG");
4721*89c4ff92SAndroid Build Coastguard Worker filter= stbi__get8(s); if (filter) return stbi__err("bad filter method","Corrupt PNG");
4722*89c4ff92SAndroid Build Coastguard Worker interlace = stbi__get8(s); if (interlace>1) return stbi__err("bad interlace method","Corrupt PNG");
4723*89c4ff92SAndroid Build Coastguard Worker if (!s->img_x || !s->img_y) return stbi__err("0-pixel image","Corrupt PNG");
4724*89c4ff92SAndroid Build Coastguard Worker if (!pal_img_n) {
4725*89c4ff92SAndroid Build Coastguard Worker s->img_n = (color & 2 ? 3 : 1) + (color & 4 ? 1 : 0);
4726*89c4ff92SAndroid Build Coastguard Worker if ((1 << 30) / s->img_x / s->img_n < s->img_y) return stbi__err("too large", "Image too large to decode");
4727*89c4ff92SAndroid Build Coastguard Worker if (scan == STBI__SCAN_header) return 1;
4728*89c4ff92SAndroid Build Coastguard Worker } else {
4729*89c4ff92SAndroid Build Coastguard Worker // if paletted, then pal_n is our final components, and
4730*89c4ff92SAndroid Build Coastguard Worker // img_n is # components to decompress/filter.
4731*89c4ff92SAndroid Build Coastguard Worker s->img_n = 1;
4732*89c4ff92SAndroid Build Coastguard Worker if ((1 << 30) / s->img_x / 4 < s->img_y) return stbi__err("too large","Corrupt PNG");
4733*89c4ff92SAndroid Build Coastguard Worker // if SCAN_header, have to scan to see if we have a tRNS
4734*89c4ff92SAndroid Build Coastguard Worker }
4735*89c4ff92SAndroid Build Coastguard Worker break;
4736*89c4ff92SAndroid Build Coastguard Worker }
4737*89c4ff92SAndroid Build Coastguard Worker
4738*89c4ff92SAndroid Build Coastguard Worker case STBI__PNG_TYPE('P','L','T','E'): {
4739*89c4ff92SAndroid Build Coastguard Worker if (first) return stbi__err("first not IHDR", "Corrupt PNG");
4740*89c4ff92SAndroid Build Coastguard Worker if (c.length > 256*3) return stbi__err("invalid PLTE","Corrupt PNG");
4741*89c4ff92SAndroid Build Coastguard Worker pal_len = c.length / 3;
4742*89c4ff92SAndroid Build Coastguard Worker if (pal_len * 3 != c.length) return stbi__err("invalid PLTE","Corrupt PNG");
4743*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < pal_len; ++i) {
4744*89c4ff92SAndroid Build Coastguard Worker palette[i*4+0] = stbi__get8(s);
4745*89c4ff92SAndroid Build Coastguard Worker palette[i*4+1] = stbi__get8(s);
4746*89c4ff92SAndroid Build Coastguard Worker palette[i*4+2] = stbi__get8(s);
4747*89c4ff92SAndroid Build Coastguard Worker palette[i*4+3] = 255;
4748*89c4ff92SAndroid Build Coastguard Worker }
4749*89c4ff92SAndroid Build Coastguard Worker break;
4750*89c4ff92SAndroid Build Coastguard Worker }
4751*89c4ff92SAndroid Build Coastguard Worker
4752*89c4ff92SAndroid Build Coastguard Worker case STBI__PNG_TYPE('t','R','N','S'): {
4753*89c4ff92SAndroid Build Coastguard Worker if (first) return stbi__err("first not IHDR", "Corrupt PNG");
4754*89c4ff92SAndroid Build Coastguard Worker if (z->idata) return stbi__err("tRNS after IDAT","Corrupt PNG");
4755*89c4ff92SAndroid Build Coastguard Worker if (pal_img_n) {
4756*89c4ff92SAndroid Build Coastguard Worker if (scan == STBI__SCAN_header) { s->img_n = 4; return 1; }
4757*89c4ff92SAndroid Build Coastguard Worker if (pal_len == 0) return stbi__err("tRNS before PLTE","Corrupt PNG");
4758*89c4ff92SAndroid Build Coastguard Worker if (c.length > pal_len) return stbi__err("bad tRNS len","Corrupt PNG");
4759*89c4ff92SAndroid Build Coastguard Worker pal_img_n = 4;
4760*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < c.length; ++i)
4761*89c4ff92SAndroid Build Coastguard Worker palette[i*4+3] = stbi__get8(s);
4762*89c4ff92SAndroid Build Coastguard Worker } else {
4763*89c4ff92SAndroid Build Coastguard Worker if (!(s->img_n & 1)) return stbi__err("tRNS with alpha","Corrupt PNG");
4764*89c4ff92SAndroid Build Coastguard Worker if (c.length != (stbi__uint32) s->img_n*2) return stbi__err("bad tRNS len","Corrupt PNG");
4765*89c4ff92SAndroid Build Coastguard Worker has_trans = 1;
4766*89c4ff92SAndroid Build Coastguard Worker if (z->depth == 16) {
4767*89c4ff92SAndroid Build Coastguard Worker for (k = 0; k < s->img_n; ++k) tc16[k] = (stbi__uint16)stbi__get16be(s); // copy the values as-is
4768*89c4ff92SAndroid Build Coastguard Worker } else {
4769*89c4ff92SAndroid Build Coastguard Worker for (k = 0; k < s->img_n; ++k) tc[k] = (stbi_uc)(stbi__get16be(s) & 255) * stbi__depth_scale_table[z->depth]; // non 8-bit images will be larger
4770*89c4ff92SAndroid Build Coastguard Worker }
4771*89c4ff92SAndroid Build Coastguard Worker }
4772*89c4ff92SAndroid Build Coastguard Worker break;
4773*89c4ff92SAndroid Build Coastguard Worker }
4774*89c4ff92SAndroid Build Coastguard Worker
4775*89c4ff92SAndroid Build Coastguard Worker case STBI__PNG_TYPE('I','D','A','T'): {
4776*89c4ff92SAndroid Build Coastguard Worker if (first) return stbi__err("first not IHDR", "Corrupt PNG");
4777*89c4ff92SAndroid Build Coastguard Worker if (pal_img_n && !pal_len) return stbi__err("no PLTE","Corrupt PNG");
4778*89c4ff92SAndroid Build Coastguard Worker if (scan == STBI__SCAN_header) { s->img_n = pal_img_n; return 1; }
4779*89c4ff92SAndroid Build Coastguard Worker if ((int)(ioff + c.length) < (int)ioff) return 0;
4780*89c4ff92SAndroid Build Coastguard Worker if (ioff + c.length > idata_limit) {
4781*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 idata_limit_old = idata_limit;
4782*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p;
4783*89c4ff92SAndroid Build Coastguard Worker if (idata_limit == 0) idata_limit = c.length > 4096 ? c.length : 4096;
4784*89c4ff92SAndroid Build Coastguard Worker while (ioff + c.length > idata_limit)
4785*89c4ff92SAndroid Build Coastguard Worker idata_limit *= 2;
4786*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(idata_limit_old);
4787*89c4ff92SAndroid Build Coastguard Worker p = (stbi_uc *) STBI_REALLOC_SIZED(z->idata, idata_limit_old, idata_limit); if (p == NULL) return stbi__err("outofmem", "Out of memory");
4788*89c4ff92SAndroid Build Coastguard Worker z->idata = p;
4789*89c4ff92SAndroid Build Coastguard Worker }
4790*89c4ff92SAndroid Build Coastguard Worker if (!stbi__getn(s, z->idata+ioff,c.length)) return stbi__err("outofdata","Corrupt PNG");
4791*89c4ff92SAndroid Build Coastguard Worker ioff += c.length;
4792*89c4ff92SAndroid Build Coastguard Worker break;
4793*89c4ff92SAndroid Build Coastguard Worker }
4794*89c4ff92SAndroid Build Coastguard Worker
4795*89c4ff92SAndroid Build Coastguard Worker case STBI__PNG_TYPE('I','E','N','D'): {
4796*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 raw_len, bpl;
4797*89c4ff92SAndroid Build Coastguard Worker if (first) return stbi__err("first not IHDR", "Corrupt PNG");
4798*89c4ff92SAndroid Build Coastguard Worker if (scan != STBI__SCAN_load) return 1;
4799*89c4ff92SAndroid Build Coastguard Worker if (z->idata == NULL) return stbi__err("no IDAT","Corrupt PNG");
4800*89c4ff92SAndroid Build Coastguard Worker // initial guess for decoded data size to avoid unnecessary reallocs
4801*89c4ff92SAndroid Build Coastguard Worker bpl = (s->img_x * z->depth + 7) / 8; // bytes per line, per component
4802*89c4ff92SAndroid Build Coastguard Worker raw_len = bpl * s->img_y * s->img_n /* pixels */ + s->img_y /* filter mode per row */;
4803*89c4ff92SAndroid Build Coastguard Worker z->expanded = (stbi_uc *) stbi_zlib_decode_malloc_guesssize_headerflag((char *) z->idata, ioff, raw_len, (int *) &raw_len, !is_iphone);
4804*89c4ff92SAndroid Build Coastguard Worker if (z->expanded == NULL) return 0; // zlib should set error
4805*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(z->idata); z->idata = NULL;
4806*89c4ff92SAndroid Build Coastguard Worker if ((req_comp == s->img_n+1 && req_comp != 3 && !pal_img_n) || has_trans)
4807*89c4ff92SAndroid Build Coastguard Worker s->img_out_n = s->img_n+1;
4808*89c4ff92SAndroid Build Coastguard Worker else
4809*89c4ff92SAndroid Build Coastguard Worker s->img_out_n = s->img_n;
4810*89c4ff92SAndroid Build Coastguard Worker if (!stbi__create_png_image(z, z->expanded, raw_len, s->img_out_n, z->depth, color, interlace)) return 0;
4811*89c4ff92SAndroid Build Coastguard Worker if (has_trans) {
4812*89c4ff92SAndroid Build Coastguard Worker if (z->depth == 16) {
4813*89c4ff92SAndroid Build Coastguard Worker if (!stbi__compute_transparency16(z, tc16, s->img_out_n)) return 0;
4814*89c4ff92SAndroid Build Coastguard Worker } else {
4815*89c4ff92SAndroid Build Coastguard Worker if (!stbi__compute_transparency(z, tc, s->img_out_n)) return 0;
4816*89c4ff92SAndroid Build Coastguard Worker }
4817*89c4ff92SAndroid Build Coastguard Worker }
4818*89c4ff92SAndroid Build Coastguard Worker if (is_iphone && stbi__de_iphone_flag && s->img_out_n > 2)
4819*89c4ff92SAndroid Build Coastguard Worker stbi__de_iphone(z);
4820*89c4ff92SAndroid Build Coastguard Worker if (pal_img_n) {
4821*89c4ff92SAndroid Build Coastguard Worker // pal_img_n == 3 or 4
4822*89c4ff92SAndroid Build Coastguard Worker s->img_n = pal_img_n; // record the actual colors we had
4823*89c4ff92SAndroid Build Coastguard Worker s->img_out_n = pal_img_n;
4824*89c4ff92SAndroid Build Coastguard Worker if (req_comp >= 3) s->img_out_n = req_comp;
4825*89c4ff92SAndroid Build Coastguard Worker if (!stbi__expand_png_palette(z, palette, pal_len, s->img_out_n))
4826*89c4ff92SAndroid Build Coastguard Worker return 0;
4827*89c4ff92SAndroid Build Coastguard Worker } else if (has_trans) {
4828*89c4ff92SAndroid Build Coastguard Worker // non-paletted image with tRNS -> source image has (constant) alpha
4829*89c4ff92SAndroid Build Coastguard Worker ++s->img_n;
4830*89c4ff92SAndroid Build Coastguard Worker }
4831*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(z->expanded); z->expanded = NULL;
4832*89c4ff92SAndroid Build Coastguard Worker return 1;
4833*89c4ff92SAndroid Build Coastguard Worker }
4834*89c4ff92SAndroid Build Coastguard Worker
4835*89c4ff92SAndroid Build Coastguard Worker default:
4836*89c4ff92SAndroid Build Coastguard Worker // if critical, fail
4837*89c4ff92SAndroid Build Coastguard Worker if (first) return stbi__err("first not IHDR", "Corrupt PNG");
4838*89c4ff92SAndroid Build Coastguard Worker if ((c.type & (1 << 29)) == 0) {
4839*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_FAILURE_STRINGS
4840*89c4ff92SAndroid Build Coastguard Worker // not threadsafe
4841*89c4ff92SAndroid Build Coastguard Worker static char invalid_chunk[] = "XXXX PNG chunk not known";
4842*89c4ff92SAndroid Build Coastguard Worker invalid_chunk[0] = STBI__BYTECAST(c.type >> 24);
4843*89c4ff92SAndroid Build Coastguard Worker invalid_chunk[1] = STBI__BYTECAST(c.type >> 16);
4844*89c4ff92SAndroid Build Coastguard Worker invalid_chunk[2] = STBI__BYTECAST(c.type >> 8);
4845*89c4ff92SAndroid Build Coastguard Worker invalid_chunk[3] = STBI__BYTECAST(c.type >> 0);
4846*89c4ff92SAndroid Build Coastguard Worker #endif
4847*89c4ff92SAndroid Build Coastguard Worker return stbi__err(invalid_chunk, "PNG not supported: unknown PNG chunk type");
4848*89c4ff92SAndroid Build Coastguard Worker }
4849*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, c.length);
4850*89c4ff92SAndroid Build Coastguard Worker break;
4851*89c4ff92SAndroid Build Coastguard Worker }
4852*89c4ff92SAndroid Build Coastguard Worker // end of PNG chunk, read and skip CRC
4853*89c4ff92SAndroid Build Coastguard Worker stbi__get32be(s);
4854*89c4ff92SAndroid Build Coastguard Worker }
4855*89c4ff92SAndroid Build Coastguard Worker }
4856*89c4ff92SAndroid Build Coastguard Worker
stbi__do_png(stbi__png * p,int * x,int * y,int * n,int req_comp,stbi__result_info * ri)4857*89c4ff92SAndroid Build Coastguard Worker static void *stbi__do_png(stbi__png *p, int *x, int *y, int *n, int req_comp, stbi__result_info *ri)
4858*89c4ff92SAndroid Build Coastguard Worker {
4859*89c4ff92SAndroid Build Coastguard Worker void *result=NULL;
4860*89c4ff92SAndroid Build Coastguard Worker if (req_comp < 0 || req_comp > 4) return stbi__errpuc("bad req_comp", "Internal error");
4861*89c4ff92SAndroid Build Coastguard Worker if (stbi__parse_png_file(p, STBI__SCAN_load, req_comp)) {
4862*89c4ff92SAndroid Build Coastguard Worker if (p->depth < 8)
4863*89c4ff92SAndroid Build Coastguard Worker ri->bits_per_channel = 8;
4864*89c4ff92SAndroid Build Coastguard Worker else
4865*89c4ff92SAndroid Build Coastguard Worker ri->bits_per_channel = p->depth;
4866*89c4ff92SAndroid Build Coastguard Worker result = p->out;
4867*89c4ff92SAndroid Build Coastguard Worker p->out = NULL;
4868*89c4ff92SAndroid Build Coastguard Worker if (req_comp && req_comp != p->s->img_out_n) {
4869*89c4ff92SAndroid Build Coastguard Worker if (ri->bits_per_channel == 8)
4870*89c4ff92SAndroid Build Coastguard Worker result = stbi__convert_format((unsigned char *) result, p->s->img_out_n, req_comp, p->s->img_x, p->s->img_y);
4871*89c4ff92SAndroid Build Coastguard Worker else
4872*89c4ff92SAndroid Build Coastguard Worker result = stbi__convert_format16((stbi__uint16 *) result, p->s->img_out_n, req_comp, p->s->img_x, p->s->img_y);
4873*89c4ff92SAndroid Build Coastguard Worker p->s->img_out_n = req_comp;
4874*89c4ff92SAndroid Build Coastguard Worker if (result == NULL) return result;
4875*89c4ff92SAndroid Build Coastguard Worker }
4876*89c4ff92SAndroid Build Coastguard Worker *x = p->s->img_x;
4877*89c4ff92SAndroid Build Coastguard Worker *y = p->s->img_y;
4878*89c4ff92SAndroid Build Coastguard Worker if (n) *n = p->s->img_n;
4879*89c4ff92SAndroid Build Coastguard Worker }
4880*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(p->out); p->out = NULL;
4881*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(p->expanded); p->expanded = NULL;
4882*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(p->idata); p->idata = NULL;
4883*89c4ff92SAndroid Build Coastguard Worker
4884*89c4ff92SAndroid Build Coastguard Worker return result;
4885*89c4ff92SAndroid Build Coastguard Worker }
4886*89c4ff92SAndroid Build Coastguard Worker
stbi__png_load(stbi__context * s,int * x,int * y,int * comp,int req_comp,stbi__result_info * ri)4887*89c4ff92SAndroid Build Coastguard Worker static void *stbi__png_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri)
4888*89c4ff92SAndroid Build Coastguard Worker {
4889*89c4ff92SAndroid Build Coastguard Worker stbi__png p;
4890*89c4ff92SAndroid Build Coastguard Worker p.s = s;
4891*89c4ff92SAndroid Build Coastguard Worker return stbi__do_png(&p, x,y,comp,req_comp, ri);
4892*89c4ff92SAndroid Build Coastguard Worker }
4893*89c4ff92SAndroid Build Coastguard Worker
stbi__png_test(stbi__context * s)4894*89c4ff92SAndroid Build Coastguard Worker static int stbi__png_test(stbi__context *s)
4895*89c4ff92SAndroid Build Coastguard Worker {
4896*89c4ff92SAndroid Build Coastguard Worker int r;
4897*89c4ff92SAndroid Build Coastguard Worker r = stbi__check_png_header(s);
4898*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
4899*89c4ff92SAndroid Build Coastguard Worker return r;
4900*89c4ff92SAndroid Build Coastguard Worker }
4901*89c4ff92SAndroid Build Coastguard Worker
stbi__png_info_raw(stbi__png * p,int * x,int * y,int * comp)4902*89c4ff92SAndroid Build Coastguard Worker static int stbi__png_info_raw(stbi__png *p, int *x, int *y, int *comp)
4903*89c4ff92SAndroid Build Coastguard Worker {
4904*89c4ff92SAndroid Build Coastguard Worker if (!stbi__parse_png_file(p, STBI__SCAN_header, 0)) {
4905*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( p->s );
4906*89c4ff92SAndroid Build Coastguard Worker return 0;
4907*89c4ff92SAndroid Build Coastguard Worker }
4908*89c4ff92SAndroid Build Coastguard Worker if (x) *x = p->s->img_x;
4909*89c4ff92SAndroid Build Coastguard Worker if (y) *y = p->s->img_y;
4910*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = p->s->img_n;
4911*89c4ff92SAndroid Build Coastguard Worker return 1;
4912*89c4ff92SAndroid Build Coastguard Worker }
4913*89c4ff92SAndroid Build Coastguard Worker
stbi__png_info(stbi__context * s,int * x,int * y,int * comp)4914*89c4ff92SAndroid Build Coastguard Worker static int stbi__png_info(stbi__context *s, int *x, int *y, int *comp)
4915*89c4ff92SAndroid Build Coastguard Worker {
4916*89c4ff92SAndroid Build Coastguard Worker stbi__png p;
4917*89c4ff92SAndroid Build Coastguard Worker p.s = s;
4918*89c4ff92SAndroid Build Coastguard Worker return stbi__png_info_raw(&p, x, y, comp);
4919*89c4ff92SAndroid Build Coastguard Worker }
4920*89c4ff92SAndroid Build Coastguard Worker #endif
4921*89c4ff92SAndroid Build Coastguard Worker
4922*89c4ff92SAndroid Build Coastguard Worker // Microsoft/Windows BMP image
4923*89c4ff92SAndroid Build Coastguard Worker
4924*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_BMP
stbi__bmp_test_raw(stbi__context * s)4925*89c4ff92SAndroid Build Coastguard Worker static int stbi__bmp_test_raw(stbi__context *s)
4926*89c4ff92SAndroid Build Coastguard Worker {
4927*89c4ff92SAndroid Build Coastguard Worker int r;
4928*89c4ff92SAndroid Build Coastguard Worker int sz;
4929*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != 'B') return 0;
4930*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != 'M') return 0;
4931*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard filesize
4932*89c4ff92SAndroid Build Coastguard Worker stbi__get16le(s); // discard reserved
4933*89c4ff92SAndroid Build Coastguard Worker stbi__get16le(s); // discard reserved
4934*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard data offset
4935*89c4ff92SAndroid Build Coastguard Worker sz = stbi__get32le(s);
4936*89c4ff92SAndroid Build Coastguard Worker r = (sz == 12 || sz == 40 || sz == 56 || sz == 108 || sz == 124);
4937*89c4ff92SAndroid Build Coastguard Worker return r;
4938*89c4ff92SAndroid Build Coastguard Worker }
4939*89c4ff92SAndroid Build Coastguard Worker
stbi__bmp_test(stbi__context * s)4940*89c4ff92SAndroid Build Coastguard Worker static int stbi__bmp_test(stbi__context *s)
4941*89c4ff92SAndroid Build Coastguard Worker {
4942*89c4ff92SAndroid Build Coastguard Worker int r = stbi__bmp_test_raw(s);
4943*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
4944*89c4ff92SAndroid Build Coastguard Worker return r;
4945*89c4ff92SAndroid Build Coastguard Worker }
4946*89c4ff92SAndroid Build Coastguard Worker
4947*89c4ff92SAndroid Build Coastguard Worker
4948*89c4ff92SAndroid Build Coastguard Worker // returns 0..31 for the highest set bit
stbi__high_bit(unsigned int z)4949*89c4ff92SAndroid Build Coastguard Worker static int stbi__high_bit(unsigned int z)
4950*89c4ff92SAndroid Build Coastguard Worker {
4951*89c4ff92SAndroid Build Coastguard Worker int n=0;
4952*89c4ff92SAndroid Build Coastguard Worker if (z == 0) return -1;
4953*89c4ff92SAndroid Build Coastguard Worker if (z >= 0x10000) n += 16, z >>= 16;
4954*89c4ff92SAndroid Build Coastguard Worker if (z >= 0x00100) n += 8, z >>= 8;
4955*89c4ff92SAndroid Build Coastguard Worker if (z >= 0x00010) n += 4, z >>= 4;
4956*89c4ff92SAndroid Build Coastguard Worker if (z >= 0x00004) n += 2, z >>= 2;
4957*89c4ff92SAndroid Build Coastguard Worker if (z >= 0x00002) n += 1, z >>= 1;
4958*89c4ff92SAndroid Build Coastguard Worker return n;
4959*89c4ff92SAndroid Build Coastguard Worker }
4960*89c4ff92SAndroid Build Coastguard Worker
stbi__bitcount(unsigned int a)4961*89c4ff92SAndroid Build Coastguard Worker static int stbi__bitcount(unsigned int a)
4962*89c4ff92SAndroid Build Coastguard Worker {
4963*89c4ff92SAndroid Build Coastguard Worker a = (a & 0x55555555) + ((a >> 1) & 0x55555555); // max 2
4964*89c4ff92SAndroid Build Coastguard Worker a = (a & 0x33333333) + ((a >> 2) & 0x33333333); // max 4
4965*89c4ff92SAndroid Build Coastguard Worker a = (a + (a >> 4)) & 0x0f0f0f0f; // max 8 per 4, now 8 bits
4966*89c4ff92SAndroid Build Coastguard Worker a = (a + (a >> 8)); // max 16 per 8 bits
4967*89c4ff92SAndroid Build Coastguard Worker a = (a + (a >> 16)); // max 32 per 8 bits
4968*89c4ff92SAndroid Build Coastguard Worker return a & 0xff;
4969*89c4ff92SAndroid Build Coastguard Worker }
4970*89c4ff92SAndroid Build Coastguard Worker
stbi__shiftsigned(int v,int shift,int bits)4971*89c4ff92SAndroid Build Coastguard Worker static int stbi__shiftsigned(int v, int shift, int bits)
4972*89c4ff92SAndroid Build Coastguard Worker {
4973*89c4ff92SAndroid Build Coastguard Worker int result;
4974*89c4ff92SAndroid Build Coastguard Worker int z=0;
4975*89c4ff92SAndroid Build Coastguard Worker
4976*89c4ff92SAndroid Build Coastguard Worker if (shift < 0) v <<= -shift;
4977*89c4ff92SAndroid Build Coastguard Worker else v >>= shift;
4978*89c4ff92SAndroid Build Coastguard Worker result = v;
4979*89c4ff92SAndroid Build Coastguard Worker
4980*89c4ff92SAndroid Build Coastguard Worker z = bits;
4981*89c4ff92SAndroid Build Coastguard Worker while (z < 8) {
4982*89c4ff92SAndroid Build Coastguard Worker result += v >> z;
4983*89c4ff92SAndroid Build Coastguard Worker z += bits;
4984*89c4ff92SAndroid Build Coastguard Worker }
4985*89c4ff92SAndroid Build Coastguard Worker return result;
4986*89c4ff92SAndroid Build Coastguard Worker }
4987*89c4ff92SAndroid Build Coastguard Worker
4988*89c4ff92SAndroid Build Coastguard Worker typedef struct
4989*89c4ff92SAndroid Build Coastguard Worker {
4990*89c4ff92SAndroid Build Coastguard Worker int bpp, offset, hsz;
4991*89c4ff92SAndroid Build Coastguard Worker unsigned int mr,mg,mb,ma, all_a;
4992*89c4ff92SAndroid Build Coastguard Worker } stbi__bmp_data;
4993*89c4ff92SAndroid Build Coastguard Worker
stbi__bmp_parse_header(stbi__context * s,stbi__bmp_data * info)4994*89c4ff92SAndroid Build Coastguard Worker static void *stbi__bmp_parse_header(stbi__context *s, stbi__bmp_data *info)
4995*89c4ff92SAndroid Build Coastguard Worker {
4996*89c4ff92SAndroid Build Coastguard Worker int hsz;
4997*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != 'B' || stbi__get8(s) != 'M') return stbi__errpuc("not BMP", "Corrupt BMP");
4998*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard filesize
4999*89c4ff92SAndroid Build Coastguard Worker stbi__get16le(s); // discard reserved
5000*89c4ff92SAndroid Build Coastguard Worker stbi__get16le(s); // discard reserved
5001*89c4ff92SAndroid Build Coastguard Worker info->offset = stbi__get32le(s);
5002*89c4ff92SAndroid Build Coastguard Worker info->hsz = hsz = stbi__get32le(s);
5003*89c4ff92SAndroid Build Coastguard Worker info->mr = info->mg = info->mb = info->ma = 0;
5004*89c4ff92SAndroid Build Coastguard Worker
5005*89c4ff92SAndroid Build Coastguard Worker if (hsz != 12 && hsz != 40 && hsz != 56 && hsz != 108 && hsz != 124) return stbi__errpuc("unknown BMP", "BMP type not supported: unknown");
5006*89c4ff92SAndroid Build Coastguard Worker if (hsz == 12) {
5007*89c4ff92SAndroid Build Coastguard Worker s->img_x = stbi__get16le(s);
5008*89c4ff92SAndroid Build Coastguard Worker s->img_y = stbi__get16le(s);
5009*89c4ff92SAndroid Build Coastguard Worker } else {
5010*89c4ff92SAndroid Build Coastguard Worker s->img_x = stbi__get32le(s);
5011*89c4ff92SAndroid Build Coastguard Worker s->img_y = stbi__get32le(s);
5012*89c4ff92SAndroid Build Coastguard Worker }
5013*89c4ff92SAndroid Build Coastguard Worker if (stbi__get16le(s) != 1) return stbi__errpuc("bad BMP", "bad BMP");
5014*89c4ff92SAndroid Build Coastguard Worker info->bpp = stbi__get16le(s);
5015*89c4ff92SAndroid Build Coastguard Worker if (info->bpp == 1) return stbi__errpuc("monochrome", "BMP type not supported: 1-bit");
5016*89c4ff92SAndroid Build Coastguard Worker if (hsz != 12) {
5017*89c4ff92SAndroid Build Coastguard Worker int compress = stbi__get32le(s);
5018*89c4ff92SAndroid Build Coastguard Worker if (compress == 1 || compress == 2) return stbi__errpuc("BMP RLE", "BMP type not supported: RLE");
5019*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard sizeof
5020*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard hres
5021*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard vres
5022*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard colorsused
5023*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard max important
5024*89c4ff92SAndroid Build Coastguard Worker if (hsz == 40 || hsz == 56) {
5025*89c4ff92SAndroid Build Coastguard Worker if (hsz == 56) {
5026*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s);
5027*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s);
5028*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s);
5029*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s);
5030*89c4ff92SAndroid Build Coastguard Worker }
5031*89c4ff92SAndroid Build Coastguard Worker if (info->bpp == 16 || info->bpp == 32) {
5032*89c4ff92SAndroid Build Coastguard Worker if (compress == 0) {
5033*89c4ff92SAndroid Build Coastguard Worker if (info->bpp == 32) {
5034*89c4ff92SAndroid Build Coastguard Worker info->mr = 0xffu << 16;
5035*89c4ff92SAndroid Build Coastguard Worker info->mg = 0xffu << 8;
5036*89c4ff92SAndroid Build Coastguard Worker info->mb = 0xffu << 0;
5037*89c4ff92SAndroid Build Coastguard Worker info->ma = 0xffu << 24;
5038*89c4ff92SAndroid Build Coastguard Worker info->all_a = 0; // if all_a is 0 at end, then we loaded alpha channel but it was all 0
5039*89c4ff92SAndroid Build Coastguard Worker } else {
5040*89c4ff92SAndroid Build Coastguard Worker info->mr = 31u << 10;
5041*89c4ff92SAndroid Build Coastguard Worker info->mg = 31u << 5;
5042*89c4ff92SAndroid Build Coastguard Worker info->mb = 31u << 0;
5043*89c4ff92SAndroid Build Coastguard Worker }
5044*89c4ff92SAndroid Build Coastguard Worker } else if (compress == 3) {
5045*89c4ff92SAndroid Build Coastguard Worker info->mr = stbi__get32le(s);
5046*89c4ff92SAndroid Build Coastguard Worker info->mg = stbi__get32le(s);
5047*89c4ff92SAndroid Build Coastguard Worker info->mb = stbi__get32le(s);
5048*89c4ff92SAndroid Build Coastguard Worker // not documented, but generated by photoshop and handled by mspaint
5049*89c4ff92SAndroid Build Coastguard Worker if (info->mr == info->mg && info->mg == info->mb) {
5050*89c4ff92SAndroid Build Coastguard Worker // ?!?!?
5051*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad BMP", "bad BMP");
5052*89c4ff92SAndroid Build Coastguard Worker }
5053*89c4ff92SAndroid Build Coastguard Worker } else
5054*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad BMP", "bad BMP");
5055*89c4ff92SAndroid Build Coastguard Worker }
5056*89c4ff92SAndroid Build Coastguard Worker } else {
5057*89c4ff92SAndroid Build Coastguard Worker int i;
5058*89c4ff92SAndroid Build Coastguard Worker if (hsz != 108 && hsz != 124)
5059*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad BMP", "bad BMP");
5060*89c4ff92SAndroid Build Coastguard Worker info->mr = stbi__get32le(s);
5061*89c4ff92SAndroid Build Coastguard Worker info->mg = stbi__get32le(s);
5062*89c4ff92SAndroid Build Coastguard Worker info->mb = stbi__get32le(s);
5063*89c4ff92SAndroid Build Coastguard Worker info->ma = stbi__get32le(s);
5064*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard color space
5065*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < 12; ++i)
5066*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard color space parameters
5067*89c4ff92SAndroid Build Coastguard Worker if (hsz == 124) {
5068*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard rendering intent
5069*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard offset of profile data
5070*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard size of profile data
5071*89c4ff92SAndroid Build Coastguard Worker stbi__get32le(s); // discard reserved
5072*89c4ff92SAndroid Build Coastguard Worker }
5073*89c4ff92SAndroid Build Coastguard Worker }
5074*89c4ff92SAndroid Build Coastguard Worker }
5075*89c4ff92SAndroid Build Coastguard Worker return (void *) 1;
5076*89c4ff92SAndroid Build Coastguard Worker }
5077*89c4ff92SAndroid Build Coastguard Worker
5078*89c4ff92SAndroid Build Coastguard Worker
stbi__bmp_load(stbi__context * s,int * x,int * y,int * comp,int req_comp,stbi__result_info * ri)5079*89c4ff92SAndroid Build Coastguard Worker static void *stbi__bmp_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri)
5080*89c4ff92SAndroid Build Coastguard Worker {
5081*89c4ff92SAndroid Build Coastguard Worker stbi_uc *out;
5082*89c4ff92SAndroid Build Coastguard Worker unsigned int mr=0,mg=0,mb=0,ma=0, all_a;
5083*89c4ff92SAndroid Build Coastguard Worker stbi_uc pal[256][4];
5084*89c4ff92SAndroid Build Coastguard Worker int psize=0,i,j,width;
5085*89c4ff92SAndroid Build Coastguard Worker int flip_vertically, pad, target;
5086*89c4ff92SAndroid Build Coastguard Worker stbi__bmp_data info;
5087*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(ri);
5088*89c4ff92SAndroid Build Coastguard Worker
5089*89c4ff92SAndroid Build Coastguard Worker info.all_a = 255;
5090*89c4ff92SAndroid Build Coastguard Worker if (stbi__bmp_parse_header(s, &info) == NULL)
5091*89c4ff92SAndroid Build Coastguard Worker return NULL; // error code already set
5092*89c4ff92SAndroid Build Coastguard Worker
5093*89c4ff92SAndroid Build Coastguard Worker flip_vertically = ((int) s->img_y) > 0;
5094*89c4ff92SAndroid Build Coastguard Worker s->img_y = abs((int) s->img_y);
5095*89c4ff92SAndroid Build Coastguard Worker
5096*89c4ff92SAndroid Build Coastguard Worker mr = info.mr;
5097*89c4ff92SAndroid Build Coastguard Worker mg = info.mg;
5098*89c4ff92SAndroid Build Coastguard Worker mb = info.mb;
5099*89c4ff92SAndroid Build Coastguard Worker ma = info.ma;
5100*89c4ff92SAndroid Build Coastguard Worker all_a = info.all_a;
5101*89c4ff92SAndroid Build Coastguard Worker
5102*89c4ff92SAndroid Build Coastguard Worker if (info.hsz == 12) {
5103*89c4ff92SAndroid Build Coastguard Worker if (info.bpp < 24)
5104*89c4ff92SAndroid Build Coastguard Worker psize = (info.offset - 14 - 24) / 3;
5105*89c4ff92SAndroid Build Coastguard Worker } else {
5106*89c4ff92SAndroid Build Coastguard Worker if (info.bpp < 16)
5107*89c4ff92SAndroid Build Coastguard Worker psize = (info.offset - 14 - info.hsz) >> 2;
5108*89c4ff92SAndroid Build Coastguard Worker }
5109*89c4ff92SAndroid Build Coastguard Worker
5110*89c4ff92SAndroid Build Coastguard Worker s->img_n = ma ? 4 : 3;
5111*89c4ff92SAndroid Build Coastguard Worker if (req_comp && req_comp >= 3) // we can directly decode 3 or 4
5112*89c4ff92SAndroid Build Coastguard Worker target = req_comp;
5113*89c4ff92SAndroid Build Coastguard Worker else
5114*89c4ff92SAndroid Build Coastguard Worker target = s->img_n; // if they want monochrome, we'll post-convert
5115*89c4ff92SAndroid Build Coastguard Worker
5116*89c4ff92SAndroid Build Coastguard Worker // sanity-check size
5117*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad3sizes_valid(target, s->img_x, s->img_y, 0))
5118*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("too large", "Corrupt BMP");
5119*89c4ff92SAndroid Build Coastguard Worker
5120*89c4ff92SAndroid Build Coastguard Worker out = (stbi_uc *) stbi__malloc_mad3(target, s->img_x, s->img_y, 0);
5121*89c4ff92SAndroid Build Coastguard Worker if (!out) return stbi__errpuc("outofmem", "Out of memory");
5122*89c4ff92SAndroid Build Coastguard Worker if (info.bpp < 16) {
5123*89c4ff92SAndroid Build Coastguard Worker int z=0;
5124*89c4ff92SAndroid Build Coastguard Worker if (psize == 0 || psize > 256) { STBI_FREE(out); return stbi__errpuc("invalid", "Corrupt BMP"); }
5125*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < psize; ++i) {
5126*89c4ff92SAndroid Build Coastguard Worker pal[i][2] = stbi__get8(s);
5127*89c4ff92SAndroid Build Coastguard Worker pal[i][1] = stbi__get8(s);
5128*89c4ff92SAndroid Build Coastguard Worker pal[i][0] = stbi__get8(s);
5129*89c4ff92SAndroid Build Coastguard Worker if (info.hsz != 12) stbi__get8(s);
5130*89c4ff92SAndroid Build Coastguard Worker pal[i][3] = 255;
5131*89c4ff92SAndroid Build Coastguard Worker }
5132*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, info.offset - 14 - info.hsz - psize * (info.hsz == 12 ? 3 : 4));
5133*89c4ff92SAndroid Build Coastguard Worker if (info.bpp == 4) width = (s->img_x + 1) >> 1;
5134*89c4ff92SAndroid Build Coastguard Worker else if (info.bpp == 8) width = s->img_x;
5135*89c4ff92SAndroid Build Coastguard Worker else { STBI_FREE(out); return stbi__errpuc("bad bpp", "Corrupt BMP"); }
5136*89c4ff92SAndroid Build Coastguard Worker pad = (-width)&3;
5137*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < (int) s->img_y; ++j) {
5138*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < (int) s->img_x; i += 2) {
5139*89c4ff92SAndroid Build Coastguard Worker int v=stbi__get8(s),v2=0;
5140*89c4ff92SAndroid Build Coastguard Worker if (info.bpp == 4) {
5141*89c4ff92SAndroid Build Coastguard Worker v2 = v & 15;
5142*89c4ff92SAndroid Build Coastguard Worker v >>= 4;
5143*89c4ff92SAndroid Build Coastguard Worker }
5144*89c4ff92SAndroid Build Coastguard Worker out[z++] = pal[v][0];
5145*89c4ff92SAndroid Build Coastguard Worker out[z++] = pal[v][1];
5146*89c4ff92SAndroid Build Coastguard Worker out[z++] = pal[v][2];
5147*89c4ff92SAndroid Build Coastguard Worker if (target == 4) out[z++] = 255;
5148*89c4ff92SAndroid Build Coastguard Worker if (i+1 == (int) s->img_x) break;
5149*89c4ff92SAndroid Build Coastguard Worker v = (info.bpp == 8) ? stbi__get8(s) : v2;
5150*89c4ff92SAndroid Build Coastguard Worker out[z++] = pal[v][0];
5151*89c4ff92SAndroid Build Coastguard Worker out[z++] = pal[v][1];
5152*89c4ff92SAndroid Build Coastguard Worker out[z++] = pal[v][2];
5153*89c4ff92SAndroid Build Coastguard Worker if (target == 4) out[z++] = 255;
5154*89c4ff92SAndroid Build Coastguard Worker }
5155*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, pad);
5156*89c4ff92SAndroid Build Coastguard Worker }
5157*89c4ff92SAndroid Build Coastguard Worker } else {
5158*89c4ff92SAndroid Build Coastguard Worker int rshift=0,gshift=0,bshift=0,ashift=0,rcount=0,gcount=0,bcount=0,acount=0;
5159*89c4ff92SAndroid Build Coastguard Worker int z = 0;
5160*89c4ff92SAndroid Build Coastguard Worker int easy=0;
5161*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, info.offset - 14 - info.hsz);
5162*89c4ff92SAndroid Build Coastguard Worker if (info.bpp == 24) width = 3 * s->img_x;
5163*89c4ff92SAndroid Build Coastguard Worker else if (info.bpp == 16) width = 2*s->img_x;
5164*89c4ff92SAndroid Build Coastguard Worker else /* bpp = 32 and pad = 0 */ width=0;
5165*89c4ff92SAndroid Build Coastguard Worker pad = (-width) & 3;
5166*89c4ff92SAndroid Build Coastguard Worker if (info.bpp == 24) {
5167*89c4ff92SAndroid Build Coastguard Worker easy = 1;
5168*89c4ff92SAndroid Build Coastguard Worker } else if (info.bpp == 32) {
5169*89c4ff92SAndroid Build Coastguard Worker if (mb == 0xff && mg == 0xff00 && mr == 0x00ff0000 && ma == 0xff000000)
5170*89c4ff92SAndroid Build Coastguard Worker easy = 2;
5171*89c4ff92SAndroid Build Coastguard Worker }
5172*89c4ff92SAndroid Build Coastguard Worker if (!easy) {
5173*89c4ff92SAndroid Build Coastguard Worker if (!mr || !mg || !mb) { STBI_FREE(out); return stbi__errpuc("bad masks", "Corrupt BMP"); }
5174*89c4ff92SAndroid Build Coastguard Worker // right shift amt to put high bit in position #7
5175*89c4ff92SAndroid Build Coastguard Worker rshift = stbi__high_bit(mr)-7; rcount = stbi__bitcount(mr);
5176*89c4ff92SAndroid Build Coastguard Worker gshift = stbi__high_bit(mg)-7; gcount = stbi__bitcount(mg);
5177*89c4ff92SAndroid Build Coastguard Worker bshift = stbi__high_bit(mb)-7; bcount = stbi__bitcount(mb);
5178*89c4ff92SAndroid Build Coastguard Worker ashift = stbi__high_bit(ma)-7; acount = stbi__bitcount(ma);
5179*89c4ff92SAndroid Build Coastguard Worker }
5180*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < (int) s->img_y; ++j) {
5181*89c4ff92SAndroid Build Coastguard Worker if (easy) {
5182*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < (int) s->img_x; ++i) {
5183*89c4ff92SAndroid Build Coastguard Worker unsigned char a;
5184*89c4ff92SAndroid Build Coastguard Worker out[z+2] = stbi__get8(s);
5185*89c4ff92SAndroid Build Coastguard Worker out[z+1] = stbi__get8(s);
5186*89c4ff92SAndroid Build Coastguard Worker out[z+0] = stbi__get8(s);
5187*89c4ff92SAndroid Build Coastguard Worker z += 3;
5188*89c4ff92SAndroid Build Coastguard Worker a = (easy == 2 ? stbi__get8(s) : 255);
5189*89c4ff92SAndroid Build Coastguard Worker all_a |= a;
5190*89c4ff92SAndroid Build Coastguard Worker if (target == 4) out[z++] = a;
5191*89c4ff92SAndroid Build Coastguard Worker }
5192*89c4ff92SAndroid Build Coastguard Worker } else {
5193*89c4ff92SAndroid Build Coastguard Worker int bpp = info.bpp;
5194*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < (int) s->img_x; ++i) {
5195*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 v = (bpp == 16 ? (stbi__uint32) stbi__get16le(s) : stbi__get32le(s));
5196*89c4ff92SAndroid Build Coastguard Worker int a;
5197*89c4ff92SAndroid Build Coastguard Worker out[z++] = STBI__BYTECAST(stbi__shiftsigned(v & mr, rshift, rcount));
5198*89c4ff92SAndroid Build Coastguard Worker out[z++] = STBI__BYTECAST(stbi__shiftsigned(v & mg, gshift, gcount));
5199*89c4ff92SAndroid Build Coastguard Worker out[z++] = STBI__BYTECAST(stbi__shiftsigned(v & mb, bshift, bcount));
5200*89c4ff92SAndroid Build Coastguard Worker a = (ma ? stbi__shiftsigned(v & ma, ashift, acount) : 255);
5201*89c4ff92SAndroid Build Coastguard Worker all_a |= a;
5202*89c4ff92SAndroid Build Coastguard Worker if (target == 4) out[z++] = STBI__BYTECAST(a);
5203*89c4ff92SAndroid Build Coastguard Worker }
5204*89c4ff92SAndroid Build Coastguard Worker }
5205*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, pad);
5206*89c4ff92SAndroid Build Coastguard Worker }
5207*89c4ff92SAndroid Build Coastguard Worker }
5208*89c4ff92SAndroid Build Coastguard Worker
5209*89c4ff92SAndroid Build Coastguard Worker // if alpha channel is all 0s, replace with all 255s
5210*89c4ff92SAndroid Build Coastguard Worker if (target == 4 && all_a == 0)
5211*89c4ff92SAndroid Build Coastguard Worker for (i=4*s->img_x*s->img_y-1; i >= 0; i -= 4)
5212*89c4ff92SAndroid Build Coastguard Worker out[i] = 255;
5213*89c4ff92SAndroid Build Coastguard Worker
5214*89c4ff92SAndroid Build Coastguard Worker if (flip_vertically) {
5215*89c4ff92SAndroid Build Coastguard Worker stbi_uc t;
5216*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < (int) s->img_y>>1; ++j) {
5217*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p1 = out + j *s->img_x*target;
5218*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p2 = out + (s->img_y-1-j)*s->img_x*target;
5219*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < (int) s->img_x*target; ++i) {
5220*89c4ff92SAndroid Build Coastguard Worker t = p1[i], p1[i] = p2[i], p2[i] = t;
5221*89c4ff92SAndroid Build Coastguard Worker }
5222*89c4ff92SAndroid Build Coastguard Worker }
5223*89c4ff92SAndroid Build Coastguard Worker }
5224*89c4ff92SAndroid Build Coastguard Worker
5225*89c4ff92SAndroid Build Coastguard Worker if (req_comp && req_comp != target) {
5226*89c4ff92SAndroid Build Coastguard Worker out = stbi__convert_format(out, target, req_comp, s->img_x, s->img_y);
5227*89c4ff92SAndroid Build Coastguard Worker if (out == NULL) return out; // stbi__convert_format frees input on failure
5228*89c4ff92SAndroid Build Coastguard Worker }
5229*89c4ff92SAndroid Build Coastguard Worker
5230*89c4ff92SAndroid Build Coastguard Worker *x = s->img_x;
5231*89c4ff92SAndroid Build Coastguard Worker *y = s->img_y;
5232*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = s->img_n;
5233*89c4ff92SAndroid Build Coastguard Worker return out;
5234*89c4ff92SAndroid Build Coastguard Worker }
5235*89c4ff92SAndroid Build Coastguard Worker #endif
5236*89c4ff92SAndroid Build Coastguard Worker
5237*89c4ff92SAndroid Build Coastguard Worker // Targa Truevision - TGA
5238*89c4ff92SAndroid Build Coastguard Worker // by Jonathan Dummer
5239*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_TGA
5240*89c4ff92SAndroid Build Coastguard Worker // returns STBI_rgb or whatever, 0 on error
stbi__tga_get_comp(int bits_per_pixel,int is_grey,int * is_rgb16)5241*89c4ff92SAndroid Build Coastguard Worker static int stbi__tga_get_comp(int bits_per_pixel, int is_grey, int* is_rgb16)
5242*89c4ff92SAndroid Build Coastguard Worker {
5243*89c4ff92SAndroid Build Coastguard Worker // only RGB or RGBA (incl. 16bit) or grey allowed
5244*89c4ff92SAndroid Build Coastguard Worker if(is_rgb16) *is_rgb16 = 0;
5245*89c4ff92SAndroid Build Coastguard Worker switch(bits_per_pixel) {
5246*89c4ff92SAndroid Build Coastguard Worker case 8: return STBI_grey;
5247*89c4ff92SAndroid Build Coastguard Worker case 16: if(is_grey) return STBI_grey_alpha;
5248*89c4ff92SAndroid Build Coastguard Worker // else: fall-through
5249*89c4ff92SAndroid Build Coastguard Worker case 15: if(is_rgb16) *is_rgb16 = 1;
5250*89c4ff92SAndroid Build Coastguard Worker return STBI_rgb;
5251*89c4ff92SAndroid Build Coastguard Worker case 24: // fall-through
5252*89c4ff92SAndroid Build Coastguard Worker case 32: return bits_per_pixel/8;
5253*89c4ff92SAndroid Build Coastguard Worker default: return 0;
5254*89c4ff92SAndroid Build Coastguard Worker }
5255*89c4ff92SAndroid Build Coastguard Worker }
5256*89c4ff92SAndroid Build Coastguard Worker
stbi__tga_info(stbi__context * s,int * x,int * y,int * comp)5257*89c4ff92SAndroid Build Coastguard Worker static int stbi__tga_info(stbi__context *s, int *x, int *y, int *comp)
5258*89c4ff92SAndroid Build Coastguard Worker {
5259*89c4ff92SAndroid Build Coastguard Worker int tga_w, tga_h, tga_comp, tga_image_type, tga_bits_per_pixel, tga_colormap_bpp;
5260*89c4ff92SAndroid Build Coastguard Worker int sz, tga_colormap_type;
5261*89c4ff92SAndroid Build Coastguard Worker stbi__get8(s); // discard Offset
5262*89c4ff92SAndroid Build Coastguard Worker tga_colormap_type = stbi__get8(s); // colormap type
5263*89c4ff92SAndroid Build Coastguard Worker if( tga_colormap_type > 1 ) {
5264*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5265*89c4ff92SAndroid Build Coastguard Worker return 0; // only RGB or indexed allowed
5266*89c4ff92SAndroid Build Coastguard Worker }
5267*89c4ff92SAndroid Build Coastguard Worker tga_image_type = stbi__get8(s); // image type
5268*89c4ff92SAndroid Build Coastguard Worker if ( tga_colormap_type == 1 ) { // colormapped (paletted) image
5269*89c4ff92SAndroid Build Coastguard Worker if (tga_image_type != 1 && tga_image_type != 9) {
5270*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5271*89c4ff92SAndroid Build Coastguard Worker return 0;
5272*89c4ff92SAndroid Build Coastguard Worker }
5273*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s,4); // skip index of first colormap entry and number of entries
5274*89c4ff92SAndroid Build Coastguard Worker sz = stbi__get8(s); // check bits per palette color entry
5275*89c4ff92SAndroid Build Coastguard Worker if ( (sz != 8) && (sz != 15) && (sz != 16) && (sz != 24) && (sz != 32) ) {
5276*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5277*89c4ff92SAndroid Build Coastguard Worker return 0;
5278*89c4ff92SAndroid Build Coastguard Worker }
5279*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s,4); // skip image x and y origin
5280*89c4ff92SAndroid Build Coastguard Worker tga_colormap_bpp = sz;
5281*89c4ff92SAndroid Build Coastguard Worker } else { // "normal" image w/o colormap - only RGB or grey allowed, +/- RLE
5282*89c4ff92SAndroid Build Coastguard Worker if ( (tga_image_type != 2) && (tga_image_type != 3) && (tga_image_type != 10) && (tga_image_type != 11) ) {
5283*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5284*89c4ff92SAndroid Build Coastguard Worker return 0; // only RGB or grey allowed, +/- RLE
5285*89c4ff92SAndroid Build Coastguard Worker }
5286*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s,9); // skip colormap specification and image x/y origin
5287*89c4ff92SAndroid Build Coastguard Worker tga_colormap_bpp = 0;
5288*89c4ff92SAndroid Build Coastguard Worker }
5289*89c4ff92SAndroid Build Coastguard Worker tga_w = stbi__get16le(s);
5290*89c4ff92SAndroid Build Coastguard Worker if( tga_w < 1 ) {
5291*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5292*89c4ff92SAndroid Build Coastguard Worker return 0; // test width
5293*89c4ff92SAndroid Build Coastguard Worker }
5294*89c4ff92SAndroid Build Coastguard Worker tga_h = stbi__get16le(s);
5295*89c4ff92SAndroid Build Coastguard Worker if( tga_h < 1 ) {
5296*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5297*89c4ff92SAndroid Build Coastguard Worker return 0; // test height
5298*89c4ff92SAndroid Build Coastguard Worker }
5299*89c4ff92SAndroid Build Coastguard Worker tga_bits_per_pixel = stbi__get8(s); // bits per pixel
5300*89c4ff92SAndroid Build Coastguard Worker stbi__get8(s); // ignore alpha bits
5301*89c4ff92SAndroid Build Coastguard Worker if (tga_colormap_bpp != 0) {
5302*89c4ff92SAndroid Build Coastguard Worker if((tga_bits_per_pixel != 8) && (tga_bits_per_pixel != 16)) {
5303*89c4ff92SAndroid Build Coastguard Worker // when using a colormap, tga_bits_per_pixel is the size of the indexes
5304*89c4ff92SAndroid Build Coastguard Worker // I don't think anything but 8 or 16bit indexes makes sense
5305*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5306*89c4ff92SAndroid Build Coastguard Worker return 0;
5307*89c4ff92SAndroid Build Coastguard Worker }
5308*89c4ff92SAndroid Build Coastguard Worker tga_comp = stbi__tga_get_comp(tga_colormap_bpp, 0, NULL);
5309*89c4ff92SAndroid Build Coastguard Worker } else {
5310*89c4ff92SAndroid Build Coastguard Worker tga_comp = stbi__tga_get_comp(tga_bits_per_pixel, (tga_image_type == 3) || (tga_image_type == 11), NULL);
5311*89c4ff92SAndroid Build Coastguard Worker }
5312*89c4ff92SAndroid Build Coastguard Worker if(!tga_comp) {
5313*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5314*89c4ff92SAndroid Build Coastguard Worker return 0;
5315*89c4ff92SAndroid Build Coastguard Worker }
5316*89c4ff92SAndroid Build Coastguard Worker if (x) *x = tga_w;
5317*89c4ff92SAndroid Build Coastguard Worker if (y) *y = tga_h;
5318*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = tga_comp;
5319*89c4ff92SAndroid Build Coastguard Worker return 1; // seems to have passed everything
5320*89c4ff92SAndroid Build Coastguard Worker }
5321*89c4ff92SAndroid Build Coastguard Worker
stbi__tga_test(stbi__context * s)5322*89c4ff92SAndroid Build Coastguard Worker static int stbi__tga_test(stbi__context *s)
5323*89c4ff92SAndroid Build Coastguard Worker {
5324*89c4ff92SAndroid Build Coastguard Worker int res = 0;
5325*89c4ff92SAndroid Build Coastguard Worker int sz, tga_color_type;
5326*89c4ff92SAndroid Build Coastguard Worker stbi__get8(s); // discard Offset
5327*89c4ff92SAndroid Build Coastguard Worker tga_color_type = stbi__get8(s); // color type
5328*89c4ff92SAndroid Build Coastguard Worker if ( tga_color_type > 1 ) goto errorEnd; // only RGB or indexed allowed
5329*89c4ff92SAndroid Build Coastguard Worker sz = stbi__get8(s); // image type
5330*89c4ff92SAndroid Build Coastguard Worker if ( tga_color_type == 1 ) { // colormapped (paletted) image
5331*89c4ff92SAndroid Build Coastguard Worker if (sz != 1 && sz != 9) goto errorEnd; // colortype 1 demands image type 1 or 9
5332*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s,4); // skip index of first colormap entry and number of entries
5333*89c4ff92SAndroid Build Coastguard Worker sz = stbi__get8(s); // check bits per palette color entry
5334*89c4ff92SAndroid Build Coastguard Worker if ( (sz != 8) && (sz != 15) && (sz != 16) && (sz != 24) && (sz != 32) ) goto errorEnd;
5335*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s,4); // skip image x and y origin
5336*89c4ff92SAndroid Build Coastguard Worker } else { // "normal" image w/o colormap
5337*89c4ff92SAndroid Build Coastguard Worker if ( (sz != 2) && (sz != 3) && (sz != 10) && (sz != 11) ) goto errorEnd; // only RGB or grey allowed, +/- RLE
5338*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s,9); // skip colormap specification and image x/y origin
5339*89c4ff92SAndroid Build Coastguard Worker }
5340*89c4ff92SAndroid Build Coastguard Worker if ( stbi__get16le(s) < 1 ) goto errorEnd; // test width
5341*89c4ff92SAndroid Build Coastguard Worker if ( stbi__get16le(s) < 1 ) goto errorEnd; // test height
5342*89c4ff92SAndroid Build Coastguard Worker sz = stbi__get8(s); // bits per pixel
5343*89c4ff92SAndroid Build Coastguard Worker if ( (tga_color_type == 1) && (sz != 8) && (sz != 16) ) goto errorEnd; // for colormapped images, bpp is size of an index
5344*89c4ff92SAndroid Build Coastguard Worker if ( (sz != 8) && (sz != 15) && (sz != 16) && (sz != 24) && (sz != 32) ) goto errorEnd;
5345*89c4ff92SAndroid Build Coastguard Worker
5346*89c4ff92SAndroid Build Coastguard Worker res = 1; // if we got this far, everything's good and we can return 1 instead of 0
5347*89c4ff92SAndroid Build Coastguard Worker
5348*89c4ff92SAndroid Build Coastguard Worker errorEnd:
5349*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5350*89c4ff92SAndroid Build Coastguard Worker return res;
5351*89c4ff92SAndroid Build Coastguard Worker }
5352*89c4ff92SAndroid Build Coastguard Worker
5353*89c4ff92SAndroid Build Coastguard Worker // read 16bit value and convert to 24bit RGB
stbi__tga_read_rgb16(stbi__context * s,stbi_uc * out)5354*89c4ff92SAndroid Build Coastguard Worker static void stbi__tga_read_rgb16(stbi__context *s, stbi_uc* out)
5355*89c4ff92SAndroid Build Coastguard Worker {
5356*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 px = (stbi__uint16)stbi__get16le(s);
5357*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 fiveBitMask = 31;
5358*89c4ff92SAndroid Build Coastguard Worker // we have 3 channels with 5bits each
5359*89c4ff92SAndroid Build Coastguard Worker int r = (px >> 10) & fiveBitMask;
5360*89c4ff92SAndroid Build Coastguard Worker int g = (px >> 5) & fiveBitMask;
5361*89c4ff92SAndroid Build Coastguard Worker int b = px & fiveBitMask;
5362*89c4ff92SAndroid Build Coastguard Worker // Note that this saves the data in RGB(A) order, so it doesn't need to be swapped later
5363*89c4ff92SAndroid Build Coastguard Worker out[0] = (stbi_uc)((r * 255)/31);
5364*89c4ff92SAndroid Build Coastguard Worker out[1] = (stbi_uc)((g * 255)/31);
5365*89c4ff92SAndroid Build Coastguard Worker out[2] = (stbi_uc)((b * 255)/31);
5366*89c4ff92SAndroid Build Coastguard Worker
5367*89c4ff92SAndroid Build Coastguard Worker // some people claim that the most significant bit might be used for alpha
5368*89c4ff92SAndroid Build Coastguard Worker // (possibly if an alpha-bit is set in the "image descriptor byte")
5369*89c4ff92SAndroid Build Coastguard Worker // but that only made 16bit test images completely translucent..
5370*89c4ff92SAndroid Build Coastguard Worker // so let's treat all 15 and 16bit TGAs as RGB with no alpha.
5371*89c4ff92SAndroid Build Coastguard Worker }
5372*89c4ff92SAndroid Build Coastguard Worker
stbi__tga_load(stbi__context * s,int * x,int * y,int * comp,int req_comp,stbi__result_info * ri)5373*89c4ff92SAndroid Build Coastguard Worker static void *stbi__tga_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri)
5374*89c4ff92SAndroid Build Coastguard Worker {
5375*89c4ff92SAndroid Build Coastguard Worker // read in the TGA header stuff
5376*89c4ff92SAndroid Build Coastguard Worker int tga_offset = stbi__get8(s);
5377*89c4ff92SAndroid Build Coastguard Worker int tga_indexed = stbi__get8(s);
5378*89c4ff92SAndroid Build Coastguard Worker int tga_image_type = stbi__get8(s);
5379*89c4ff92SAndroid Build Coastguard Worker int tga_is_RLE = 0;
5380*89c4ff92SAndroid Build Coastguard Worker int tga_palette_start = stbi__get16le(s);
5381*89c4ff92SAndroid Build Coastguard Worker int tga_palette_len = stbi__get16le(s);
5382*89c4ff92SAndroid Build Coastguard Worker int tga_palette_bits = stbi__get8(s);
5383*89c4ff92SAndroid Build Coastguard Worker int tga_x_origin = stbi__get16le(s);
5384*89c4ff92SAndroid Build Coastguard Worker int tga_y_origin = stbi__get16le(s);
5385*89c4ff92SAndroid Build Coastguard Worker int tga_width = stbi__get16le(s);
5386*89c4ff92SAndroid Build Coastguard Worker int tga_height = stbi__get16le(s);
5387*89c4ff92SAndroid Build Coastguard Worker int tga_bits_per_pixel = stbi__get8(s);
5388*89c4ff92SAndroid Build Coastguard Worker int tga_comp, tga_rgb16=0;
5389*89c4ff92SAndroid Build Coastguard Worker int tga_inverted = stbi__get8(s);
5390*89c4ff92SAndroid Build Coastguard Worker // int tga_alpha_bits = tga_inverted & 15; // the 4 lowest bits - unused (useless?)
5391*89c4ff92SAndroid Build Coastguard Worker // image data
5392*89c4ff92SAndroid Build Coastguard Worker unsigned char *tga_data;
5393*89c4ff92SAndroid Build Coastguard Worker unsigned char *tga_palette = NULL;
5394*89c4ff92SAndroid Build Coastguard Worker int i, j;
5395*89c4ff92SAndroid Build Coastguard Worker unsigned char raw_data[4] = {0};
5396*89c4ff92SAndroid Build Coastguard Worker int RLE_count = 0;
5397*89c4ff92SAndroid Build Coastguard Worker int RLE_repeating = 0;
5398*89c4ff92SAndroid Build Coastguard Worker int read_next_pixel = 1;
5399*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(ri);
5400*89c4ff92SAndroid Build Coastguard Worker
5401*89c4ff92SAndroid Build Coastguard Worker // do a tiny bit of precessing
5402*89c4ff92SAndroid Build Coastguard Worker if ( tga_image_type >= 8 )
5403*89c4ff92SAndroid Build Coastguard Worker {
5404*89c4ff92SAndroid Build Coastguard Worker tga_image_type -= 8;
5405*89c4ff92SAndroid Build Coastguard Worker tga_is_RLE = 1;
5406*89c4ff92SAndroid Build Coastguard Worker }
5407*89c4ff92SAndroid Build Coastguard Worker tga_inverted = 1 - ((tga_inverted >> 5) & 1);
5408*89c4ff92SAndroid Build Coastguard Worker
5409*89c4ff92SAndroid Build Coastguard Worker // If I'm paletted, then I'll use the number of bits from the palette
5410*89c4ff92SAndroid Build Coastguard Worker if ( tga_indexed ) tga_comp = stbi__tga_get_comp(tga_palette_bits, 0, &tga_rgb16);
5411*89c4ff92SAndroid Build Coastguard Worker else tga_comp = stbi__tga_get_comp(tga_bits_per_pixel, (tga_image_type == 3), &tga_rgb16);
5412*89c4ff92SAndroid Build Coastguard Worker
5413*89c4ff92SAndroid Build Coastguard Worker if(!tga_comp) // shouldn't really happen, stbi__tga_test() should have ensured basic consistency
5414*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad format", "Can't find out TGA pixelformat");
5415*89c4ff92SAndroid Build Coastguard Worker
5416*89c4ff92SAndroid Build Coastguard Worker // tga info
5417*89c4ff92SAndroid Build Coastguard Worker *x = tga_width;
5418*89c4ff92SAndroid Build Coastguard Worker *y = tga_height;
5419*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = tga_comp;
5420*89c4ff92SAndroid Build Coastguard Worker
5421*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad3sizes_valid(tga_width, tga_height, tga_comp, 0))
5422*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("too large", "Corrupt TGA");
5423*89c4ff92SAndroid Build Coastguard Worker
5424*89c4ff92SAndroid Build Coastguard Worker tga_data = (unsigned char*)stbi__malloc_mad3(tga_width, tga_height, tga_comp, 0);
5425*89c4ff92SAndroid Build Coastguard Worker if (!tga_data) return stbi__errpuc("outofmem", "Out of memory");
5426*89c4ff92SAndroid Build Coastguard Worker
5427*89c4ff92SAndroid Build Coastguard Worker // skip to the data's starting position (offset usually = 0)
5428*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, tga_offset );
5429*89c4ff92SAndroid Build Coastguard Worker
5430*89c4ff92SAndroid Build Coastguard Worker if ( !tga_indexed && !tga_is_RLE && !tga_rgb16 ) {
5431*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < tga_height; ++i) {
5432*89c4ff92SAndroid Build Coastguard Worker int row = tga_inverted ? tga_height -i - 1 : i;
5433*89c4ff92SAndroid Build Coastguard Worker stbi_uc *tga_row = tga_data + row*tga_width*tga_comp;
5434*89c4ff92SAndroid Build Coastguard Worker stbi__getn(s, tga_row, tga_width * tga_comp);
5435*89c4ff92SAndroid Build Coastguard Worker }
5436*89c4ff92SAndroid Build Coastguard Worker } else {
5437*89c4ff92SAndroid Build Coastguard Worker // do I need to load a palette?
5438*89c4ff92SAndroid Build Coastguard Worker if ( tga_indexed)
5439*89c4ff92SAndroid Build Coastguard Worker {
5440*89c4ff92SAndroid Build Coastguard Worker // any data to skip? (offset usually = 0)
5441*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, tga_palette_start );
5442*89c4ff92SAndroid Build Coastguard Worker // load the palette
5443*89c4ff92SAndroid Build Coastguard Worker tga_palette = (unsigned char*)stbi__malloc_mad2(tga_palette_len, tga_comp, 0);
5444*89c4ff92SAndroid Build Coastguard Worker if (!tga_palette) {
5445*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(tga_data);
5446*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("outofmem", "Out of memory");
5447*89c4ff92SAndroid Build Coastguard Worker }
5448*89c4ff92SAndroid Build Coastguard Worker if (tga_rgb16) {
5449*89c4ff92SAndroid Build Coastguard Worker stbi_uc *pal_entry = tga_palette;
5450*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(tga_comp == STBI_rgb);
5451*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < tga_palette_len; ++i) {
5452*89c4ff92SAndroid Build Coastguard Worker stbi__tga_read_rgb16(s, pal_entry);
5453*89c4ff92SAndroid Build Coastguard Worker pal_entry += tga_comp;
5454*89c4ff92SAndroid Build Coastguard Worker }
5455*89c4ff92SAndroid Build Coastguard Worker } else if (!stbi__getn(s, tga_palette, tga_palette_len * tga_comp)) {
5456*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(tga_data);
5457*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(tga_palette);
5458*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad palette", "Corrupt TGA");
5459*89c4ff92SAndroid Build Coastguard Worker }
5460*89c4ff92SAndroid Build Coastguard Worker }
5461*89c4ff92SAndroid Build Coastguard Worker // load the data
5462*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < tga_width * tga_height; ++i)
5463*89c4ff92SAndroid Build Coastguard Worker {
5464*89c4ff92SAndroid Build Coastguard Worker // if I'm in RLE mode, do I need to get a RLE stbi__pngchunk?
5465*89c4ff92SAndroid Build Coastguard Worker if ( tga_is_RLE )
5466*89c4ff92SAndroid Build Coastguard Worker {
5467*89c4ff92SAndroid Build Coastguard Worker if ( RLE_count == 0 )
5468*89c4ff92SAndroid Build Coastguard Worker {
5469*89c4ff92SAndroid Build Coastguard Worker // yep, get the next byte as a RLE command
5470*89c4ff92SAndroid Build Coastguard Worker int RLE_cmd = stbi__get8(s);
5471*89c4ff92SAndroid Build Coastguard Worker RLE_count = 1 + (RLE_cmd & 127);
5472*89c4ff92SAndroid Build Coastguard Worker RLE_repeating = RLE_cmd >> 7;
5473*89c4ff92SAndroid Build Coastguard Worker read_next_pixel = 1;
5474*89c4ff92SAndroid Build Coastguard Worker } else if ( !RLE_repeating )
5475*89c4ff92SAndroid Build Coastguard Worker {
5476*89c4ff92SAndroid Build Coastguard Worker read_next_pixel = 1;
5477*89c4ff92SAndroid Build Coastguard Worker }
5478*89c4ff92SAndroid Build Coastguard Worker } else
5479*89c4ff92SAndroid Build Coastguard Worker {
5480*89c4ff92SAndroid Build Coastguard Worker read_next_pixel = 1;
5481*89c4ff92SAndroid Build Coastguard Worker }
5482*89c4ff92SAndroid Build Coastguard Worker // OK, if I need to read a pixel, do it now
5483*89c4ff92SAndroid Build Coastguard Worker if ( read_next_pixel )
5484*89c4ff92SAndroid Build Coastguard Worker {
5485*89c4ff92SAndroid Build Coastguard Worker // load however much data we did have
5486*89c4ff92SAndroid Build Coastguard Worker if ( tga_indexed )
5487*89c4ff92SAndroid Build Coastguard Worker {
5488*89c4ff92SAndroid Build Coastguard Worker // read in index, then perform the lookup
5489*89c4ff92SAndroid Build Coastguard Worker int pal_idx = (tga_bits_per_pixel == 8) ? stbi__get8(s) : stbi__get16le(s);
5490*89c4ff92SAndroid Build Coastguard Worker if ( pal_idx >= tga_palette_len ) {
5491*89c4ff92SAndroid Build Coastguard Worker // invalid index
5492*89c4ff92SAndroid Build Coastguard Worker pal_idx = 0;
5493*89c4ff92SAndroid Build Coastguard Worker }
5494*89c4ff92SAndroid Build Coastguard Worker pal_idx *= tga_comp;
5495*89c4ff92SAndroid Build Coastguard Worker for (j = 0; j < tga_comp; ++j) {
5496*89c4ff92SAndroid Build Coastguard Worker raw_data[j] = tga_palette[pal_idx+j];
5497*89c4ff92SAndroid Build Coastguard Worker }
5498*89c4ff92SAndroid Build Coastguard Worker } else if(tga_rgb16) {
5499*89c4ff92SAndroid Build Coastguard Worker STBI_ASSERT(tga_comp == STBI_rgb);
5500*89c4ff92SAndroid Build Coastguard Worker stbi__tga_read_rgb16(s, raw_data);
5501*89c4ff92SAndroid Build Coastguard Worker } else {
5502*89c4ff92SAndroid Build Coastguard Worker // read in the data raw
5503*89c4ff92SAndroid Build Coastguard Worker for (j = 0; j < tga_comp; ++j) {
5504*89c4ff92SAndroid Build Coastguard Worker raw_data[j] = stbi__get8(s);
5505*89c4ff92SAndroid Build Coastguard Worker }
5506*89c4ff92SAndroid Build Coastguard Worker }
5507*89c4ff92SAndroid Build Coastguard Worker // clear the reading flag for the next pixel
5508*89c4ff92SAndroid Build Coastguard Worker read_next_pixel = 0;
5509*89c4ff92SAndroid Build Coastguard Worker } // end of reading a pixel
5510*89c4ff92SAndroid Build Coastguard Worker
5511*89c4ff92SAndroid Build Coastguard Worker // copy data
5512*89c4ff92SAndroid Build Coastguard Worker for (j = 0; j < tga_comp; ++j)
5513*89c4ff92SAndroid Build Coastguard Worker tga_data[i*tga_comp+j] = raw_data[j];
5514*89c4ff92SAndroid Build Coastguard Worker
5515*89c4ff92SAndroid Build Coastguard Worker // in case we're in RLE mode, keep counting down
5516*89c4ff92SAndroid Build Coastguard Worker --RLE_count;
5517*89c4ff92SAndroid Build Coastguard Worker }
5518*89c4ff92SAndroid Build Coastguard Worker // do I need to invert the image?
5519*89c4ff92SAndroid Build Coastguard Worker if ( tga_inverted )
5520*89c4ff92SAndroid Build Coastguard Worker {
5521*89c4ff92SAndroid Build Coastguard Worker for (j = 0; j*2 < tga_height; ++j)
5522*89c4ff92SAndroid Build Coastguard Worker {
5523*89c4ff92SAndroid Build Coastguard Worker int index1 = j * tga_width * tga_comp;
5524*89c4ff92SAndroid Build Coastguard Worker int index2 = (tga_height - 1 - j) * tga_width * tga_comp;
5525*89c4ff92SAndroid Build Coastguard Worker for (i = tga_width * tga_comp; i > 0; --i)
5526*89c4ff92SAndroid Build Coastguard Worker {
5527*89c4ff92SAndroid Build Coastguard Worker unsigned char temp = tga_data[index1];
5528*89c4ff92SAndroid Build Coastguard Worker tga_data[index1] = tga_data[index2];
5529*89c4ff92SAndroid Build Coastguard Worker tga_data[index2] = temp;
5530*89c4ff92SAndroid Build Coastguard Worker ++index1;
5531*89c4ff92SAndroid Build Coastguard Worker ++index2;
5532*89c4ff92SAndroid Build Coastguard Worker }
5533*89c4ff92SAndroid Build Coastguard Worker }
5534*89c4ff92SAndroid Build Coastguard Worker }
5535*89c4ff92SAndroid Build Coastguard Worker // clear my palette, if I had one
5536*89c4ff92SAndroid Build Coastguard Worker if ( tga_palette != NULL )
5537*89c4ff92SAndroid Build Coastguard Worker {
5538*89c4ff92SAndroid Build Coastguard Worker STBI_FREE( tga_palette );
5539*89c4ff92SAndroid Build Coastguard Worker }
5540*89c4ff92SAndroid Build Coastguard Worker }
5541*89c4ff92SAndroid Build Coastguard Worker
5542*89c4ff92SAndroid Build Coastguard Worker // swap RGB - if the source data was RGB16, it already is in the right order
5543*89c4ff92SAndroid Build Coastguard Worker if (tga_comp >= 3 && !tga_rgb16)
5544*89c4ff92SAndroid Build Coastguard Worker {
5545*89c4ff92SAndroid Build Coastguard Worker unsigned char* tga_pixel = tga_data;
5546*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < tga_width * tga_height; ++i)
5547*89c4ff92SAndroid Build Coastguard Worker {
5548*89c4ff92SAndroid Build Coastguard Worker unsigned char temp = tga_pixel[0];
5549*89c4ff92SAndroid Build Coastguard Worker tga_pixel[0] = tga_pixel[2];
5550*89c4ff92SAndroid Build Coastguard Worker tga_pixel[2] = temp;
5551*89c4ff92SAndroid Build Coastguard Worker tga_pixel += tga_comp;
5552*89c4ff92SAndroid Build Coastguard Worker }
5553*89c4ff92SAndroid Build Coastguard Worker }
5554*89c4ff92SAndroid Build Coastguard Worker
5555*89c4ff92SAndroid Build Coastguard Worker // convert to target component count
5556*89c4ff92SAndroid Build Coastguard Worker if (req_comp && req_comp != tga_comp)
5557*89c4ff92SAndroid Build Coastguard Worker tga_data = stbi__convert_format(tga_data, tga_comp, req_comp, tga_width, tga_height);
5558*89c4ff92SAndroid Build Coastguard Worker
5559*89c4ff92SAndroid Build Coastguard Worker // the things I do to get rid of an error message, and yet keep
5560*89c4ff92SAndroid Build Coastguard Worker // Microsoft's C compilers happy... [8^(
5561*89c4ff92SAndroid Build Coastguard Worker tga_palette_start = tga_palette_len = tga_palette_bits =
5562*89c4ff92SAndroid Build Coastguard Worker tga_x_origin = tga_y_origin = 0;
5563*89c4ff92SAndroid Build Coastguard Worker // OK, done
5564*89c4ff92SAndroid Build Coastguard Worker return tga_data;
5565*89c4ff92SAndroid Build Coastguard Worker }
5566*89c4ff92SAndroid Build Coastguard Worker #endif
5567*89c4ff92SAndroid Build Coastguard Worker
5568*89c4ff92SAndroid Build Coastguard Worker // *************************************************************************************************
5569*89c4ff92SAndroid Build Coastguard Worker // Photoshop PSD loader -- PD by Thatcher Ulrich, integration by Nicolas Schulz, tweaked by STB
5570*89c4ff92SAndroid Build Coastguard Worker
5571*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PSD
stbi__psd_test(stbi__context * s)5572*89c4ff92SAndroid Build Coastguard Worker static int stbi__psd_test(stbi__context *s)
5573*89c4ff92SAndroid Build Coastguard Worker {
5574*89c4ff92SAndroid Build Coastguard Worker int r = (stbi__get32be(s) == 0x38425053);
5575*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
5576*89c4ff92SAndroid Build Coastguard Worker return r;
5577*89c4ff92SAndroid Build Coastguard Worker }
5578*89c4ff92SAndroid Build Coastguard Worker
stbi__psd_decode_rle(stbi__context * s,stbi_uc * p,int pixelCount)5579*89c4ff92SAndroid Build Coastguard Worker static int stbi__psd_decode_rle(stbi__context *s, stbi_uc *p, int pixelCount)
5580*89c4ff92SAndroid Build Coastguard Worker {
5581*89c4ff92SAndroid Build Coastguard Worker int count, nleft, len;
5582*89c4ff92SAndroid Build Coastguard Worker
5583*89c4ff92SAndroid Build Coastguard Worker count = 0;
5584*89c4ff92SAndroid Build Coastguard Worker while ((nleft = pixelCount - count) > 0) {
5585*89c4ff92SAndroid Build Coastguard Worker len = stbi__get8(s);
5586*89c4ff92SAndroid Build Coastguard Worker if (len == 128) {
5587*89c4ff92SAndroid Build Coastguard Worker // No-op.
5588*89c4ff92SAndroid Build Coastguard Worker } else if (len < 128) {
5589*89c4ff92SAndroid Build Coastguard Worker // Copy next len+1 bytes literally.
5590*89c4ff92SAndroid Build Coastguard Worker len++;
5591*89c4ff92SAndroid Build Coastguard Worker if (len > nleft) return 0; // corrupt data
5592*89c4ff92SAndroid Build Coastguard Worker count += len;
5593*89c4ff92SAndroid Build Coastguard Worker while (len) {
5594*89c4ff92SAndroid Build Coastguard Worker *p = stbi__get8(s);
5595*89c4ff92SAndroid Build Coastguard Worker p += 4;
5596*89c4ff92SAndroid Build Coastguard Worker len--;
5597*89c4ff92SAndroid Build Coastguard Worker }
5598*89c4ff92SAndroid Build Coastguard Worker } else if (len > 128) {
5599*89c4ff92SAndroid Build Coastguard Worker stbi_uc val;
5600*89c4ff92SAndroid Build Coastguard Worker // Next -len+1 bytes in the dest are replicated from next source byte.
5601*89c4ff92SAndroid Build Coastguard Worker // (Interpret len as a negative 8-bit int.)
5602*89c4ff92SAndroid Build Coastguard Worker len = 257 - len;
5603*89c4ff92SAndroid Build Coastguard Worker if (len > nleft) return 0; // corrupt data
5604*89c4ff92SAndroid Build Coastguard Worker val = stbi__get8(s);
5605*89c4ff92SAndroid Build Coastguard Worker count += len;
5606*89c4ff92SAndroid Build Coastguard Worker while (len) {
5607*89c4ff92SAndroid Build Coastguard Worker *p = val;
5608*89c4ff92SAndroid Build Coastguard Worker p += 4;
5609*89c4ff92SAndroid Build Coastguard Worker len--;
5610*89c4ff92SAndroid Build Coastguard Worker }
5611*89c4ff92SAndroid Build Coastguard Worker }
5612*89c4ff92SAndroid Build Coastguard Worker }
5613*89c4ff92SAndroid Build Coastguard Worker
5614*89c4ff92SAndroid Build Coastguard Worker return 1;
5615*89c4ff92SAndroid Build Coastguard Worker }
5616*89c4ff92SAndroid Build Coastguard Worker
stbi__psd_load(stbi__context * s,int * x,int * y,int * comp,int req_comp,stbi__result_info * ri,int bpc)5617*89c4ff92SAndroid Build Coastguard Worker static void *stbi__psd_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri, int bpc)
5618*89c4ff92SAndroid Build Coastguard Worker {
5619*89c4ff92SAndroid Build Coastguard Worker int pixelCount;
5620*89c4ff92SAndroid Build Coastguard Worker int channelCount, compression;
5621*89c4ff92SAndroid Build Coastguard Worker int channel, i;
5622*89c4ff92SAndroid Build Coastguard Worker int bitdepth;
5623*89c4ff92SAndroid Build Coastguard Worker int w,h;
5624*89c4ff92SAndroid Build Coastguard Worker stbi_uc *out;
5625*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(ri);
5626*89c4ff92SAndroid Build Coastguard Worker
5627*89c4ff92SAndroid Build Coastguard Worker // Check identifier
5628*89c4ff92SAndroid Build Coastguard Worker if (stbi__get32be(s) != 0x38425053) // "8BPS"
5629*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("not PSD", "Corrupt PSD image");
5630*89c4ff92SAndroid Build Coastguard Worker
5631*89c4ff92SAndroid Build Coastguard Worker // Check file type version.
5632*89c4ff92SAndroid Build Coastguard Worker if (stbi__get16be(s) != 1)
5633*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("wrong version", "Unsupported version of PSD image");
5634*89c4ff92SAndroid Build Coastguard Worker
5635*89c4ff92SAndroid Build Coastguard Worker // Skip 6 reserved bytes.
5636*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, 6 );
5637*89c4ff92SAndroid Build Coastguard Worker
5638*89c4ff92SAndroid Build Coastguard Worker // Read the number of channels (R, G, B, A, etc).
5639*89c4ff92SAndroid Build Coastguard Worker channelCount = stbi__get16be(s);
5640*89c4ff92SAndroid Build Coastguard Worker if (channelCount < 0 || channelCount > 16)
5641*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("wrong channel count", "Unsupported number of channels in PSD image");
5642*89c4ff92SAndroid Build Coastguard Worker
5643*89c4ff92SAndroid Build Coastguard Worker // Read the rows and columns of the image.
5644*89c4ff92SAndroid Build Coastguard Worker h = stbi__get32be(s);
5645*89c4ff92SAndroid Build Coastguard Worker w = stbi__get32be(s);
5646*89c4ff92SAndroid Build Coastguard Worker
5647*89c4ff92SAndroid Build Coastguard Worker // Make sure the depth is 8 bits.
5648*89c4ff92SAndroid Build Coastguard Worker bitdepth = stbi__get16be(s);
5649*89c4ff92SAndroid Build Coastguard Worker if (bitdepth != 8 && bitdepth != 16)
5650*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("unsupported bit depth", "PSD bit depth is not 8 or 16 bit");
5651*89c4ff92SAndroid Build Coastguard Worker
5652*89c4ff92SAndroid Build Coastguard Worker // Make sure the color mode is RGB.
5653*89c4ff92SAndroid Build Coastguard Worker // Valid options are:
5654*89c4ff92SAndroid Build Coastguard Worker // 0: Bitmap
5655*89c4ff92SAndroid Build Coastguard Worker // 1: Grayscale
5656*89c4ff92SAndroid Build Coastguard Worker // 2: Indexed color
5657*89c4ff92SAndroid Build Coastguard Worker // 3: RGB color
5658*89c4ff92SAndroid Build Coastguard Worker // 4: CMYK color
5659*89c4ff92SAndroid Build Coastguard Worker // 7: Multichannel
5660*89c4ff92SAndroid Build Coastguard Worker // 8: Duotone
5661*89c4ff92SAndroid Build Coastguard Worker // 9: Lab color
5662*89c4ff92SAndroid Build Coastguard Worker if (stbi__get16be(s) != 3)
5663*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("wrong color format", "PSD is not in RGB color format");
5664*89c4ff92SAndroid Build Coastguard Worker
5665*89c4ff92SAndroid Build Coastguard Worker // Skip the Mode Data. (It's the palette for indexed color; other info for other modes.)
5666*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s,stbi__get32be(s) );
5667*89c4ff92SAndroid Build Coastguard Worker
5668*89c4ff92SAndroid Build Coastguard Worker // Skip the image resources. (resolution, pen tool paths, etc)
5669*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, stbi__get32be(s) );
5670*89c4ff92SAndroid Build Coastguard Worker
5671*89c4ff92SAndroid Build Coastguard Worker // Skip the reserved data.
5672*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, stbi__get32be(s) );
5673*89c4ff92SAndroid Build Coastguard Worker
5674*89c4ff92SAndroid Build Coastguard Worker // Find out if the data is compressed.
5675*89c4ff92SAndroid Build Coastguard Worker // Known values:
5676*89c4ff92SAndroid Build Coastguard Worker // 0: no compression
5677*89c4ff92SAndroid Build Coastguard Worker // 1: RLE compressed
5678*89c4ff92SAndroid Build Coastguard Worker compression = stbi__get16be(s);
5679*89c4ff92SAndroid Build Coastguard Worker if (compression > 1)
5680*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad compression", "PSD has an unknown compression format");
5681*89c4ff92SAndroid Build Coastguard Worker
5682*89c4ff92SAndroid Build Coastguard Worker // Check size
5683*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad3sizes_valid(4, w, h, 0))
5684*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("too large", "Corrupt PSD");
5685*89c4ff92SAndroid Build Coastguard Worker
5686*89c4ff92SAndroid Build Coastguard Worker // Create the destination image.
5687*89c4ff92SAndroid Build Coastguard Worker
5688*89c4ff92SAndroid Build Coastguard Worker if (!compression && bitdepth == 16 && bpc == 16) {
5689*89c4ff92SAndroid Build Coastguard Worker out = (stbi_uc *) stbi__malloc_mad3(8, w, h, 0);
5690*89c4ff92SAndroid Build Coastguard Worker ri->bits_per_channel = 16;
5691*89c4ff92SAndroid Build Coastguard Worker } else
5692*89c4ff92SAndroid Build Coastguard Worker out = (stbi_uc *) stbi__malloc(4 * w*h);
5693*89c4ff92SAndroid Build Coastguard Worker
5694*89c4ff92SAndroid Build Coastguard Worker if (!out) return stbi__errpuc("outofmem", "Out of memory");
5695*89c4ff92SAndroid Build Coastguard Worker pixelCount = w*h;
5696*89c4ff92SAndroid Build Coastguard Worker
5697*89c4ff92SAndroid Build Coastguard Worker // Initialize the data to zero.
5698*89c4ff92SAndroid Build Coastguard Worker //memset( out, 0, pixelCount * 4 );
5699*89c4ff92SAndroid Build Coastguard Worker
5700*89c4ff92SAndroid Build Coastguard Worker // Finally, the image data.
5701*89c4ff92SAndroid Build Coastguard Worker if (compression) {
5702*89c4ff92SAndroid Build Coastguard Worker // RLE as used by .PSD and .TIFF
5703*89c4ff92SAndroid Build Coastguard Worker // Loop until you get the number of unpacked bytes you are expecting:
5704*89c4ff92SAndroid Build Coastguard Worker // Read the next source byte into n.
5705*89c4ff92SAndroid Build Coastguard Worker // If n is between 0 and 127 inclusive, copy the next n+1 bytes literally.
5706*89c4ff92SAndroid Build Coastguard Worker // Else if n is between -127 and -1 inclusive, copy the next byte -n+1 times.
5707*89c4ff92SAndroid Build Coastguard Worker // Else if n is 128, noop.
5708*89c4ff92SAndroid Build Coastguard Worker // Endloop
5709*89c4ff92SAndroid Build Coastguard Worker
5710*89c4ff92SAndroid Build Coastguard Worker // The RLE-compressed data is preceeded by a 2-byte data count for each row in the data,
5711*89c4ff92SAndroid Build Coastguard Worker // which we're going to just skip.
5712*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, h * channelCount * 2 );
5713*89c4ff92SAndroid Build Coastguard Worker
5714*89c4ff92SAndroid Build Coastguard Worker // Read the RLE data by channel.
5715*89c4ff92SAndroid Build Coastguard Worker for (channel = 0; channel < 4; channel++) {
5716*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p;
5717*89c4ff92SAndroid Build Coastguard Worker
5718*89c4ff92SAndroid Build Coastguard Worker p = out+channel;
5719*89c4ff92SAndroid Build Coastguard Worker if (channel >= channelCount) {
5720*89c4ff92SAndroid Build Coastguard Worker // Fill this channel with default data.
5721*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < pixelCount; i++, p += 4)
5722*89c4ff92SAndroid Build Coastguard Worker *p = (channel == 3 ? 255 : 0);
5723*89c4ff92SAndroid Build Coastguard Worker } else {
5724*89c4ff92SAndroid Build Coastguard Worker // Read the RLE data.
5725*89c4ff92SAndroid Build Coastguard Worker if (!stbi__psd_decode_rle(s, p, pixelCount)) {
5726*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(out);
5727*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("corrupt", "bad RLE data");
5728*89c4ff92SAndroid Build Coastguard Worker }
5729*89c4ff92SAndroid Build Coastguard Worker }
5730*89c4ff92SAndroid Build Coastguard Worker }
5731*89c4ff92SAndroid Build Coastguard Worker
5732*89c4ff92SAndroid Build Coastguard Worker } else {
5733*89c4ff92SAndroid Build Coastguard Worker // We're at the raw image data. It's each channel in order (Red, Green, Blue, Alpha, ...)
5734*89c4ff92SAndroid Build Coastguard Worker // where each channel consists of an 8-bit (or 16-bit) value for each pixel in the image.
5735*89c4ff92SAndroid Build Coastguard Worker
5736*89c4ff92SAndroid Build Coastguard Worker // Read the data by channel.
5737*89c4ff92SAndroid Build Coastguard Worker for (channel = 0; channel < 4; channel++) {
5738*89c4ff92SAndroid Build Coastguard Worker if (channel >= channelCount) {
5739*89c4ff92SAndroid Build Coastguard Worker // Fill this channel with default data.
5740*89c4ff92SAndroid Build Coastguard Worker if (bitdepth == 16 && bpc == 16) {
5741*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *q = ((stbi__uint16 *) out) + channel;
5742*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 val = channel == 3 ? 65535 : 0;
5743*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < pixelCount; i++, q += 4)
5744*89c4ff92SAndroid Build Coastguard Worker *q = val;
5745*89c4ff92SAndroid Build Coastguard Worker } else {
5746*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p = out+channel;
5747*89c4ff92SAndroid Build Coastguard Worker stbi_uc val = channel == 3 ? 255 : 0;
5748*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < pixelCount; i++, p += 4)
5749*89c4ff92SAndroid Build Coastguard Worker *p = val;
5750*89c4ff92SAndroid Build Coastguard Worker }
5751*89c4ff92SAndroid Build Coastguard Worker } else {
5752*89c4ff92SAndroid Build Coastguard Worker if (ri->bits_per_channel == 16) { // output bpc
5753*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *q = ((stbi__uint16 *) out) + channel;
5754*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < pixelCount; i++, q += 4)
5755*89c4ff92SAndroid Build Coastguard Worker *q = (stbi__uint16) stbi__get16be(s);
5756*89c4ff92SAndroid Build Coastguard Worker } else {
5757*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p = out+channel;
5758*89c4ff92SAndroid Build Coastguard Worker if (bitdepth == 16) { // input bpc
5759*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < pixelCount; i++, p += 4)
5760*89c4ff92SAndroid Build Coastguard Worker *p = (stbi_uc) (stbi__get16be(s) >> 8);
5761*89c4ff92SAndroid Build Coastguard Worker } else {
5762*89c4ff92SAndroid Build Coastguard Worker for (i = 0; i < pixelCount; i++, p += 4)
5763*89c4ff92SAndroid Build Coastguard Worker *p = stbi__get8(s);
5764*89c4ff92SAndroid Build Coastguard Worker }
5765*89c4ff92SAndroid Build Coastguard Worker }
5766*89c4ff92SAndroid Build Coastguard Worker }
5767*89c4ff92SAndroid Build Coastguard Worker }
5768*89c4ff92SAndroid Build Coastguard Worker }
5769*89c4ff92SAndroid Build Coastguard Worker
5770*89c4ff92SAndroid Build Coastguard Worker // remove weird white matte from PSD
5771*89c4ff92SAndroid Build Coastguard Worker if (channelCount >= 4) {
5772*89c4ff92SAndroid Build Coastguard Worker if (ri->bits_per_channel == 16) {
5773*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < w*h; ++i) {
5774*89c4ff92SAndroid Build Coastguard Worker stbi__uint16 *pixel = (stbi__uint16 *) out + 4*i;
5775*89c4ff92SAndroid Build Coastguard Worker if (pixel[3] != 0 && pixel[3] != 65535) {
5776*89c4ff92SAndroid Build Coastguard Worker float a = pixel[3] / 65535.0f;
5777*89c4ff92SAndroid Build Coastguard Worker float ra = 1.0f / a;
5778*89c4ff92SAndroid Build Coastguard Worker float inv_a = 65535.0f * (1 - ra);
5779*89c4ff92SAndroid Build Coastguard Worker pixel[0] = (stbi__uint16) (pixel[0]*ra + inv_a);
5780*89c4ff92SAndroid Build Coastguard Worker pixel[1] = (stbi__uint16) (pixel[1]*ra + inv_a);
5781*89c4ff92SAndroid Build Coastguard Worker pixel[2] = (stbi__uint16) (pixel[2]*ra + inv_a);
5782*89c4ff92SAndroid Build Coastguard Worker }
5783*89c4ff92SAndroid Build Coastguard Worker }
5784*89c4ff92SAndroid Build Coastguard Worker } else {
5785*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < w*h; ++i) {
5786*89c4ff92SAndroid Build Coastguard Worker unsigned char *pixel = out + 4*i;
5787*89c4ff92SAndroid Build Coastguard Worker if (pixel[3] != 0 && pixel[3] != 255) {
5788*89c4ff92SAndroid Build Coastguard Worker float a = pixel[3] / 255.0f;
5789*89c4ff92SAndroid Build Coastguard Worker float ra = 1.0f / a;
5790*89c4ff92SAndroid Build Coastguard Worker float inv_a = 255.0f * (1 - ra);
5791*89c4ff92SAndroid Build Coastguard Worker pixel[0] = (unsigned char) (pixel[0]*ra + inv_a);
5792*89c4ff92SAndroid Build Coastguard Worker pixel[1] = (unsigned char) (pixel[1]*ra + inv_a);
5793*89c4ff92SAndroid Build Coastguard Worker pixel[2] = (unsigned char) (pixel[2]*ra + inv_a);
5794*89c4ff92SAndroid Build Coastguard Worker }
5795*89c4ff92SAndroid Build Coastguard Worker }
5796*89c4ff92SAndroid Build Coastguard Worker }
5797*89c4ff92SAndroid Build Coastguard Worker }
5798*89c4ff92SAndroid Build Coastguard Worker
5799*89c4ff92SAndroid Build Coastguard Worker // convert to desired output format
5800*89c4ff92SAndroid Build Coastguard Worker if (req_comp && req_comp != 4) {
5801*89c4ff92SAndroid Build Coastguard Worker if (ri->bits_per_channel == 16)
5802*89c4ff92SAndroid Build Coastguard Worker out = (stbi_uc *) stbi__convert_format16((stbi__uint16 *) out, 4, req_comp, w, h);
5803*89c4ff92SAndroid Build Coastguard Worker else
5804*89c4ff92SAndroid Build Coastguard Worker out = stbi__convert_format(out, 4, req_comp, w, h);
5805*89c4ff92SAndroid Build Coastguard Worker if (out == NULL) return out; // stbi__convert_format frees input on failure
5806*89c4ff92SAndroid Build Coastguard Worker }
5807*89c4ff92SAndroid Build Coastguard Worker
5808*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = 4;
5809*89c4ff92SAndroid Build Coastguard Worker *y = h;
5810*89c4ff92SAndroid Build Coastguard Worker *x = w;
5811*89c4ff92SAndroid Build Coastguard Worker
5812*89c4ff92SAndroid Build Coastguard Worker return out;
5813*89c4ff92SAndroid Build Coastguard Worker }
5814*89c4ff92SAndroid Build Coastguard Worker #endif
5815*89c4ff92SAndroid Build Coastguard Worker
5816*89c4ff92SAndroid Build Coastguard Worker // *************************************************************************************************
5817*89c4ff92SAndroid Build Coastguard Worker // Softimage PIC loader
5818*89c4ff92SAndroid Build Coastguard Worker // by Tom Seddon
5819*89c4ff92SAndroid Build Coastguard Worker //
5820*89c4ff92SAndroid Build Coastguard Worker // See http://softimage.wiki.softimage.com/index.php/INFO:_PIC_file_format
5821*89c4ff92SAndroid Build Coastguard Worker // See http://ozviz.wasp.uwa.edu.au/~pbourke/dataformats/softimagepic/
5822*89c4ff92SAndroid Build Coastguard Worker
5823*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PIC
stbi__pic_is4(stbi__context * s,const char * str)5824*89c4ff92SAndroid Build Coastguard Worker static int stbi__pic_is4(stbi__context *s,const char *str)
5825*89c4ff92SAndroid Build Coastguard Worker {
5826*89c4ff92SAndroid Build Coastguard Worker int i;
5827*89c4ff92SAndroid Build Coastguard Worker for (i=0; i<4; ++i)
5828*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != (stbi_uc)str[i])
5829*89c4ff92SAndroid Build Coastguard Worker return 0;
5830*89c4ff92SAndroid Build Coastguard Worker
5831*89c4ff92SAndroid Build Coastguard Worker return 1;
5832*89c4ff92SAndroid Build Coastguard Worker }
5833*89c4ff92SAndroid Build Coastguard Worker
stbi__pic_test_core(stbi__context * s)5834*89c4ff92SAndroid Build Coastguard Worker static int stbi__pic_test_core(stbi__context *s)
5835*89c4ff92SAndroid Build Coastguard Worker {
5836*89c4ff92SAndroid Build Coastguard Worker int i;
5837*89c4ff92SAndroid Build Coastguard Worker
5838*89c4ff92SAndroid Build Coastguard Worker if (!stbi__pic_is4(s,"\x53\x80\xF6\x34"))
5839*89c4ff92SAndroid Build Coastguard Worker return 0;
5840*89c4ff92SAndroid Build Coastguard Worker
5841*89c4ff92SAndroid Build Coastguard Worker for(i=0;i<84;++i)
5842*89c4ff92SAndroid Build Coastguard Worker stbi__get8(s);
5843*89c4ff92SAndroid Build Coastguard Worker
5844*89c4ff92SAndroid Build Coastguard Worker if (!stbi__pic_is4(s,"PICT"))
5845*89c4ff92SAndroid Build Coastguard Worker return 0;
5846*89c4ff92SAndroid Build Coastguard Worker
5847*89c4ff92SAndroid Build Coastguard Worker return 1;
5848*89c4ff92SAndroid Build Coastguard Worker }
5849*89c4ff92SAndroid Build Coastguard Worker
5850*89c4ff92SAndroid Build Coastguard Worker typedef struct
5851*89c4ff92SAndroid Build Coastguard Worker {
5852*89c4ff92SAndroid Build Coastguard Worker stbi_uc size,type,channel;
5853*89c4ff92SAndroid Build Coastguard Worker } stbi__pic_packet;
5854*89c4ff92SAndroid Build Coastguard Worker
stbi__readval(stbi__context * s,int channel,stbi_uc * dest)5855*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__readval(stbi__context *s, int channel, stbi_uc *dest)
5856*89c4ff92SAndroid Build Coastguard Worker {
5857*89c4ff92SAndroid Build Coastguard Worker int mask=0x80, i;
5858*89c4ff92SAndroid Build Coastguard Worker
5859*89c4ff92SAndroid Build Coastguard Worker for (i=0; i<4; ++i, mask>>=1) {
5860*89c4ff92SAndroid Build Coastguard Worker if (channel & mask) {
5861*89c4ff92SAndroid Build Coastguard Worker if (stbi__at_eof(s)) return stbi__errpuc("bad file","PIC file too short");
5862*89c4ff92SAndroid Build Coastguard Worker dest[i]=stbi__get8(s);
5863*89c4ff92SAndroid Build Coastguard Worker }
5864*89c4ff92SAndroid Build Coastguard Worker }
5865*89c4ff92SAndroid Build Coastguard Worker
5866*89c4ff92SAndroid Build Coastguard Worker return dest;
5867*89c4ff92SAndroid Build Coastguard Worker }
5868*89c4ff92SAndroid Build Coastguard Worker
stbi__copyval(int channel,stbi_uc * dest,const stbi_uc * src)5869*89c4ff92SAndroid Build Coastguard Worker static void stbi__copyval(int channel,stbi_uc *dest,const stbi_uc *src)
5870*89c4ff92SAndroid Build Coastguard Worker {
5871*89c4ff92SAndroid Build Coastguard Worker int mask=0x80,i;
5872*89c4ff92SAndroid Build Coastguard Worker
5873*89c4ff92SAndroid Build Coastguard Worker for (i=0;i<4; ++i, mask>>=1)
5874*89c4ff92SAndroid Build Coastguard Worker if (channel&mask)
5875*89c4ff92SAndroid Build Coastguard Worker dest[i]=src[i];
5876*89c4ff92SAndroid Build Coastguard Worker }
5877*89c4ff92SAndroid Build Coastguard Worker
stbi__pic_load_core(stbi__context * s,int width,int height,int * comp,stbi_uc * result)5878*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__pic_load_core(stbi__context *s,int width,int height,int *comp, stbi_uc *result)
5879*89c4ff92SAndroid Build Coastguard Worker {
5880*89c4ff92SAndroid Build Coastguard Worker int act_comp=0,num_packets=0,y,chained;
5881*89c4ff92SAndroid Build Coastguard Worker stbi__pic_packet packets[10];
5882*89c4ff92SAndroid Build Coastguard Worker
5883*89c4ff92SAndroid Build Coastguard Worker // this will (should...) cater for even some bizarre stuff like having data
5884*89c4ff92SAndroid Build Coastguard Worker // for the same channel in multiple packets.
5885*89c4ff92SAndroid Build Coastguard Worker do {
5886*89c4ff92SAndroid Build Coastguard Worker stbi__pic_packet *packet;
5887*89c4ff92SAndroid Build Coastguard Worker
5888*89c4ff92SAndroid Build Coastguard Worker if (num_packets==sizeof(packets)/sizeof(packets[0]))
5889*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad format","too many packets");
5890*89c4ff92SAndroid Build Coastguard Worker
5891*89c4ff92SAndroid Build Coastguard Worker packet = &packets[num_packets++];
5892*89c4ff92SAndroid Build Coastguard Worker
5893*89c4ff92SAndroid Build Coastguard Worker chained = stbi__get8(s);
5894*89c4ff92SAndroid Build Coastguard Worker packet->size = stbi__get8(s);
5895*89c4ff92SAndroid Build Coastguard Worker packet->type = stbi__get8(s);
5896*89c4ff92SAndroid Build Coastguard Worker packet->channel = stbi__get8(s);
5897*89c4ff92SAndroid Build Coastguard Worker
5898*89c4ff92SAndroid Build Coastguard Worker act_comp |= packet->channel;
5899*89c4ff92SAndroid Build Coastguard Worker
5900*89c4ff92SAndroid Build Coastguard Worker if (stbi__at_eof(s)) return stbi__errpuc("bad file","file too short (reading packets)");
5901*89c4ff92SAndroid Build Coastguard Worker if (packet->size != 8) return stbi__errpuc("bad format","packet isn't 8bpp");
5902*89c4ff92SAndroid Build Coastguard Worker } while (chained);
5903*89c4ff92SAndroid Build Coastguard Worker
5904*89c4ff92SAndroid Build Coastguard Worker *comp = (act_comp & 0x10 ? 4 : 3); // has alpha channel?
5905*89c4ff92SAndroid Build Coastguard Worker
5906*89c4ff92SAndroid Build Coastguard Worker for(y=0; y<height; ++y) {
5907*89c4ff92SAndroid Build Coastguard Worker int packet_idx;
5908*89c4ff92SAndroid Build Coastguard Worker
5909*89c4ff92SAndroid Build Coastguard Worker for(packet_idx=0; packet_idx < num_packets; ++packet_idx) {
5910*89c4ff92SAndroid Build Coastguard Worker stbi__pic_packet *packet = &packets[packet_idx];
5911*89c4ff92SAndroid Build Coastguard Worker stbi_uc *dest = result+y*width*4;
5912*89c4ff92SAndroid Build Coastguard Worker
5913*89c4ff92SAndroid Build Coastguard Worker switch (packet->type) {
5914*89c4ff92SAndroid Build Coastguard Worker default:
5915*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad format","packet has bad compression type");
5916*89c4ff92SAndroid Build Coastguard Worker
5917*89c4ff92SAndroid Build Coastguard Worker case 0: {//uncompressed
5918*89c4ff92SAndroid Build Coastguard Worker int x;
5919*89c4ff92SAndroid Build Coastguard Worker
5920*89c4ff92SAndroid Build Coastguard Worker for(x=0;x<width;++x, dest+=4)
5921*89c4ff92SAndroid Build Coastguard Worker if (!stbi__readval(s,packet->channel,dest))
5922*89c4ff92SAndroid Build Coastguard Worker return 0;
5923*89c4ff92SAndroid Build Coastguard Worker break;
5924*89c4ff92SAndroid Build Coastguard Worker }
5925*89c4ff92SAndroid Build Coastguard Worker
5926*89c4ff92SAndroid Build Coastguard Worker case 1://Pure RLE
5927*89c4ff92SAndroid Build Coastguard Worker {
5928*89c4ff92SAndroid Build Coastguard Worker int left=width, i;
5929*89c4ff92SAndroid Build Coastguard Worker
5930*89c4ff92SAndroid Build Coastguard Worker while (left>0) {
5931*89c4ff92SAndroid Build Coastguard Worker stbi_uc count,value[4];
5932*89c4ff92SAndroid Build Coastguard Worker
5933*89c4ff92SAndroid Build Coastguard Worker count=stbi__get8(s);
5934*89c4ff92SAndroid Build Coastguard Worker if (stbi__at_eof(s)) return stbi__errpuc("bad file","file too short (pure read count)");
5935*89c4ff92SAndroid Build Coastguard Worker
5936*89c4ff92SAndroid Build Coastguard Worker if (count > left)
5937*89c4ff92SAndroid Build Coastguard Worker count = (stbi_uc) left;
5938*89c4ff92SAndroid Build Coastguard Worker
5939*89c4ff92SAndroid Build Coastguard Worker if (!stbi__readval(s,packet->channel,value)) return 0;
5940*89c4ff92SAndroid Build Coastguard Worker
5941*89c4ff92SAndroid Build Coastguard Worker for(i=0; i<count; ++i,dest+=4)
5942*89c4ff92SAndroid Build Coastguard Worker stbi__copyval(packet->channel,dest,value);
5943*89c4ff92SAndroid Build Coastguard Worker left -= count;
5944*89c4ff92SAndroid Build Coastguard Worker }
5945*89c4ff92SAndroid Build Coastguard Worker }
5946*89c4ff92SAndroid Build Coastguard Worker break;
5947*89c4ff92SAndroid Build Coastguard Worker
5948*89c4ff92SAndroid Build Coastguard Worker case 2: {//Mixed RLE
5949*89c4ff92SAndroid Build Coastguard Worker int left=width;
5950*89c4ff92SAndroid Build Coastguard Worker while (left>0) {
5951*89c4ff92SAndroid Build Coastguard Worker int count = stbi__get8(s), i;
5952*89c4ff92SAndroid Build Coastguard Worker if (stbi__at_eof(s)) return stbi__errpuc("bad file","file too short (mixed read count)");
5953*89c4ff92SAndroid Build Coastguard Worker
5954*89c4ff92SAndroid Build Coastguard Worker if (count >= 128) { // Repeated
5955*89c4ff92SAndroid Build Coastguard Worker stbi_uc value[4];
5956*89c4ff92SAndroid Build Coastguard Worker
5957*89c4ff92SAndroid Build Coastguard Worker if (count==128)
5958*89c4ff92SAndroid Build Coastguard Worker count = stbi__get16be(s);
5959*89c4ff92SAndroid Build Coastguard Worker else
5960*89c4ff92SAndroid Build Coastguard Worker count -= 127;
5961*89c4ff92SAndroid Build Coastguard Worker if (count > left)
5962*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad file","scanline overrun");
5963*89c4ff92SAndroid Build Coastguard Worker
5964*89c4ff92SAndroid Build Coastguard Worker if (!stbi__readval(s,packet->channel,value))
5965*89c4ff92SAndroid Build Coastguard Worker return 0;
5966*89c4ff92SAndroid Build Coastguard Worker
5967*89c4ff92SAndroid Build Coastguard Worker for(i=0;i<count;++i, dest += 4)
5968*89c4ff92SAndroid Build Coastguard Worker stbi__copyval(packet->channel,dest,value);
5969*89c4ff92SAndroid Build Coastguard Worker } else { // Raw
5970*89c4ff92SAndroid Build Coastguard Worker ++count;
5971*89c4ff92SAndroid Build Coastguard Worker if (count>left) return stbi__errpuc("bad file","scanline overrun");
5972*89c4ff92SAndroid Build Coastguard Worker
5973*89c4ff92SAndroid Build Coastguard Worker for(i=0;i<count;++i, dest+=4)
5974*89c4ff92SAndroid Build Coastguard Worker if (!stbi__readval(s,packet->channel,dest))
5975*89c4ff92SAndroid Build Coastguard Worker return 0;
5976*89c4ff92SAndroid Build Coastguard Worker }
5977*89c4ff92SAndroid Build Coastguard Worker left-=count;
5978*89c4ff92SAndroid Build Coastguard Worker }
5979*89c4ff92SAndroid Build Coastguard Worker break;
5980*89c4ff92SAndroid Build Coastguard Worker }
5981*89c4ff92SAndroid Build Coastguard Worker }
5982*89c4ff92SAndroid Build Coastguard Worker }
5983*89c4ff92SAndroid Build Coastguard Worker }
5984*89c4ff92SAndroid Build Coastguard Worker
5985*89c4ff92SAndroid Build Coastguard Worker return result;
5986*89c4ff92SAndroid Build Coastguard Worker }
5987*89c4ff92SAndroid Build Coastguard Worker
stbi__pic_load(stbi__context * s,int * px,int * py,int * comp,int req_comp,stbi__result_info * ri)5988*89c4ff92SAndroid Build Coastguard Worker static void *stbi__pic_load(stbi__context *s,int *px,int *py,int *comp,int req_comp, stbi__result_info *ri)
5989*89c4ff92SAndroid Build Coastguard Worker {
5990*89c4ff92SAndroid Build Coastguard Worker stbi_uc *result;
5991*89c4ff92SAndroid Build Coastguard Worker int i, x,y, internal_comp;
5992*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(ri);
5993*89c4ff92SAndroid Build Coastguard Worker
5994*89c4ff92SAndroid Build Coastguard Worker if (!comp) comp = &internal_comp;
5995*89c4ff92SAndroid Build Coastguard Worker
5996*89c4ff92SAndroid Build Coastguard Worker for (i=0; i<92; ++i)
5997*89c4ff92SAndroid Build Coastguard Worker stbi__get8(s);
5998*89c4ff92SAndroid Build Coastguard Worker
5999*89c4ff92SAndroid Build Coastguard Worker x = stbi__get16be(s);
6000*89c4ff92SAndroid Build Coastguard Worker y = stbi__get16be(s);
6001*89c4ff92SAndroid Build Coastguard Worker if (stbi__at_eof(s)) return stbi__errpuc("bad file","file too short (pic header)");
6002*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad3sizes_valid(x, y, 4, 0)) return stbi__errpuc("too large", "PIC image too large to decode");
6003*89c4ff92SAndroid Build Coastguard Worker
6004*89c4ff92SAndroid Build Coastguard Worker stbi__get32be(s); //skip `ratio'
6005*89c4ff92SAndroid Build Coastguard Worker stbi__get16be(s); //skip `fields'
6006*89c4ff92SAndroid Build Coastguard Worker stbi__get16be(s); //skip `pad'
6007*89c4ff92SAndroid Build Coastguard Worker
6008*89c4ff92SAndroid Build Coastguard Worker // intermediate buffer is RGBA
6009*89c4ff92SAndroid Build Coastguard Worker result = (stbi_uc *) stbi__malloc_mad3(x, y, 4, 0);
6010*89c4ff92SAndroid Build Coastguard Worker memset(result, 0xff, x*y*4);
6011*89c4ff92SAndroid Build Coastguard Worker
6012*89c4ff92SAndroid Build Coastguard Worker if (!stbi__pic_load_core(s,x,y,comp, result)) {
6013*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(result);
6014*89c4ff92SAndroid Build Coastguard Worker result=0;
6015*89c4ff92SAndroid Build Coastguard Worker }
6016*89c4ff92SAndroid Build Coastguard Worker *px = x;
6017*89c4ff92SAndroid Build Coastguard Worker *py = y;
6018*89c4ff92SAndroid Build Coastguard Worker if (req_comp == 0) req_comp = *comp;
6019*89c4ff92SAndroid Build Coastguard Worker result=stbi__convert_format(result,4,req_comp,x,y);
6020*89c4ff92SAndroid Build Coastguard Worker
6021*89c4ff92SAndroid Build Coastguard Worker return result;
6022*89c4ff92SAndroid Build Coastguard Worker }
6023*89c4ff92SAndroid Build Coastguard Worker
stbi__pic_test(stbi__context * s)6024*89c4ff92SAndroid Build Coastguard Worker static int stbi__pic_test(stbi__context *s)
6025*89c4ff92SAndroid Build Coastguard Worker {
6026*89c4ff92SAndroid Build Coastguard Worker int r = stbi__pic_test_core(s);
6027*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
6028*89c4ff92SAndroid Build Coastguard Worker return r;
6029*89c4ff92SAndroid Build Coastguard Worker }
6030*89c4ff92SAndroid Build Coastguard Worker #endif
6031*89c4ff92SAndroid Build Coastguard Worker
6032*89c4ff92SAndroid Build Coastguard Worker // *************************************************************************************************
6033*89c4ff92SAndroid Build Coastguard Worker // GIF loader -- public domain by Jean-Marc Lienher -- simplified/shrunk by stb
6034*89c4ff92SAndroid Build Coastguard Worker
6035*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_GIF
6036*89c4ff92SAndroid Build Coastguard Worker typedef struct
6037*89c4ff92SAndroid Build Coastguard Worker {
6038*89c4ff92SAndroid Build Coastguard Worker stbi__int16 prefix;
6039*89c4ff92SAndroid Build Coastguard Worker stbi_uc first;
6040*89c4ff92SAndroid Build Coastguard Worker stbi_uc suffix;
6041*89c4ff92SAndroid Build Coastguard Worker } stbi__gif_lzw;
6042*89c4ff92SAndroid Build Coastguard Worker
6043*89c4ff92SAndroid Build Coastguard Worker typedef struct
6044*89c4ff92SAndroid Build Coastguard Worker {
6045*89c4ff92SAndroid Build Coastguard Worker int w,h;
6046*89c4ff92SAndroid Build Coastguard Worker stbi_uc *out, *old_out; // output buffer (always 4 components)
6047*89c4ff92SAndroid Build Coastguard Worker int flags, bgindex, ratio, transparent, eflags, delay;
6048*89c4ff92SAndroid Build Coastguard Worker stbi_uc pal[256][4];
6049*89c4ff92SAndroid Build Coastguard Worker stbi_uc lpal[256][4];
6050*89c4ff92SAndroid Build Coastguard Worker stbi__gif_lzw codes[4096];
6051*89c4ff92SAndroid Build Coastguard Worker stbi_uc *color_table;
6052*89c4ff92SAndroid Build Coastguard Worker int parse, step;
6053*89c4ff92SAndroid Build Coastguard Worker int lflags;
6054*89c4ff92SAndroid Build Coastguard Worker int start_x, start_y;
6055*89c4ff92SAndroid Build Coastguard Worker int max_x, max_y;
6056*89c4ff92SAndroid Build Coastguard Worker int cur_x, cur_y;
6057*89c4ff92SAndroid Build Coastguard Worker int line_size;
6058*89c4ff92SAndroid Build Coastguard Worker } stbi__gif;
6059*89c4ff92SAndroid Build Coastguard Worker
stbi__gif_test_raw(stbi__context * s)6060*89c4ff92SAndroid Build Coastguard Worker static int stbi__gif_test_raw(stbi__context *s)
6061*89c4ff92SAndroid Build Coastguard Worker {
6062*89c4ff92SAndroid Build Coastguard Worker int sz;
6063*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != 'G' || stbi__get8(s) != 'I' || stbi__get8(s) != 'F' || stbi__get8(s) != '8') return 0;
6064*89c4ff92SAndroid Build Coastguard Worker sz = stbi__get8(s);
6065*89c4ff92SAndroid Build Coastguard Worker if (sz != '9' && sz != '7') return 0;
6066*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != 'a') return 0;
6067*89c4ff92SAndroid Build Coastguard Worker return 1;
6068*89c4ff92SAndroid Build Coastguard Worker }
6069*89c4ff92SAndroid Build Coastguard Worker
stbi__gif_test(stbi__context * s)6070*89c4ff92SAndroid Build Coastguard Worker static int stbi__gif_test(stbi__context *s)
6071*89c4ff92SAndroid Build Coastguard Worker {
6072*89c4ff92SAndroid Build Coastguard Worker int r = stbi__gif_test_raw(s);
6073*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
6074*89c4ff92SAndroid Build Coastguard Worker return r;
6075*89c4ff92SAndroid Build Coastguard Worker }
6076*89c4ff92SAndroid Build Coastguard Worker
stbi__gif_parse_colortable(stbi__context * s,stbi_uc pal[256][4],int num_entries,int transp)6077*89c4ff92SAndroid Build Coastguard Worker static void stbi__gif_parse_colortable(stbi__context *s, stbi_uc pal[256][4], int num_entries, int transp)
6078*89c4ff92SAndroid Build Coastguard Worker {
6079*89c4ff92SAndroid Build Coastguard Worker int i;
6080*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < num_entries; ++i) {
6081*89c4ff92SAndroid Build Coastguard Worker pal[i][2] = stbi__get8(s);
6082*89c4ff92SAndroid Build Coastguard Worker pal[i][1] = stbi__get8(s);
6083*89c4ff92SAndroid Build Coastguard Worker pal[i][0] = stbi__get8(s);
6084*89c4ff92SAndroid Build Coastguard Worker pal[i][3] = transp == i ? 0 : 255;
6085*89c4ff92SAndroid Build Coastguard Worker }
6086*89c4ff92SAndroid Build Coastguard Worker }
6087*89c4ff92SAndroid Build Coastguard Worker
stbi__gif_header(stbi__context * s,stbi__gif * g,int * comp,int is_info)6088*89c4ff92SAndroid Build Coastguard Worker static int stbi__gif_header(stbi__context *s, stbi__gif *g, int *comp, int is_info)
6089*89c4ff92SAndroid Build Coastguard Worker {
6090*89c4ff92SAndroid Build Coastguard Worker stbi_uc version;
6091*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != 'G' || stbi__get8(s) != 'I' || stbi__get8(s) != 'F' || stbi__get8(s) != '8')
6092*89c4ff92SAndroid Build Coastguard Worker return stbi__err("not GIF", "Corrupt GIF");
6093*89c4ff92SAndroid Build Coastguard Worker
6094*89c4ff92SAndroid Build Coastguard Worker version = stbi__get8(s);
6095*89c4ff92SAndroid Build Coastguard Worker if (version != '7' && version != '9') return stbi__err("not GIF", "Corrupt GIF");
6096*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != 'a') return stbi__err("not GIF", "Corrupt GIF");
6097*89c4ff92SAndroid Build Coastguard Worker
6098*89c4ff92SAndroid Build Coastguard Worker stbi__g_failure_reason = "";
6099*89c4ff92SAndroid Build Coastguard Worker g->w = stbi__get16le(s);
6100*89c4ff92SAndroid Build Coastguard Worker g->h = stbi__get16le(s);
6101*89c4ff92SAndroid Build Coastguard Worker g->flags = stbi__get8(s);
6102*89c4ff92SAndroid Build Coastguard Worker g->bgindex = stbi__get8(s);
6103*89c4ff92SAndroid Build Coastguard Worker g->ratio = stbi__get8(s);
6104*89c4ff92SAndroid Build Coastguard Worker g->transparent = -1;
6105*89c4ff92SAndroid Build Coastguard Worker
6106*89c4ff92SAndroid Build Coastguard Worker if (comp != 0) *comp = 4; // can't actually tell whether it's 3 or 4 until we parse the comments
6107*89c4ff92SAndroid Build Coastguard Worker
6108*89c4ff92SAndroid Build Coastguard Worker if (is_info) return 1;
6109*89c4ff92SAndroid Build Coastguard Worker
6110*89c4ff92SAndroid Build Coastguard Worker if (g->flags & 0x80)
6111*89c4ff92SAndroid Build Coastguard Worker stbi__gif_parse_colortable(s,g->pal, 2 << (g->flags & 7), -1);
6112*89c4ff92SAndroid Build Coastguard Worker
6113*89c4ff92SAndroid Build Coastguard Worker return 1;
6114*89c4ff92SAndroid Build Coastguard Worker }
6115*89c4ff92SAndroid Build Coastguard Worker
stbi__gif_info_raw(stbi__context * s,int * x,int * y,int * comp)6116*89c4ff92SAndroid Build Coastguard Worker static int stbi__gif_info_raw(stbi__context *s, int *x, int *y, int *comp)
6117*89c4ff92SAndroid Build Coastguard Worker {
6118*89c4ff92SAndroid Build Coastguard Worker stbi__gif* g = (stbi__gif*) stbi__malloc(sizeof(stbi__gif));
6119*89c4ff92SAndroid Build Coastguard Worker if (!stbi__gif_header(s, g, comp, 1)) {
6120*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(g);
6121*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6122*89c4ff92SAndroid Build Coastguard Worker return 0;
6123*89c4ff92SAndroid Build Coastguard Worker }
6124*89c4ff92SAndroid Build Coastguard Worker if (x) *x = g->w;
6125*89c4ff92SAndroid Build Coastguard Worker if (y) *y = g->h;
6126*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(g);
6127*89c4ff92SAndroid Build Coastguard Worker return 1;
6128*89c4ff92SAndroid Build Coastguard Worker }
6129*89c4ff92SAndroid Build Coastguard Worker
stbi__out_gif_code(stbi__gif * g,stbi__uint16 code)6130*89c4ff92SAndroid Build Coastguard Worker static void stbi__out_gif_code(stbi__gif *g, stbi__uint16 code)
6131*89c4ff92SAndroid Build Coastguard Worker {
6132*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p, *c;
6133*89c4ff92SAndroid Build Coastguard Worker
6134*89c4ff92SAndroid Build Coastguard Worker // recurse to decode the prefixes, since the linked-list is backwards,
6135*89c4ff92SAndroid Build Coastguard Worker // and working backwards through an interleaved image would be nasty
6136*89c4ff92SAndroid Build Coastguard Worker if (g->codes[code].prefix >= 0)
6137*89c4ff92SAndroid Build Coastguard Worker stbi__out_gif_code(g, g->codes[code].prefix);
6138*89c4ff92SAndroid Build Coastguard Worker
6139*89c4ff92SAndroid Build Coastguard Worker if (g->cur_y >= g->max_y) return;
6140*89c4ff92SAndroid Build Coastguard Worker
6141*89c4ff92SAndroid Build Coastguard Worker p = &g->out[g->cur_x + g->cur_y];
6142*89c4ff92SAndroid Build Coastguard Worker c = &g->color_table[g->codes[code].suffix * 4];
6143*89c4ff92SAndroid Build Coastguard Worker
6144*89c4ff92SAndroid Build Coastguard Worker if (c[3] >= 128) {
6145*89c4ff92SAndroid Build Coastguard Worker p[0] = c[2];
6146*89c4ff92SAndroid Build Coastguard Worker p[1] = c[1];
6147*89c4ff92SAndroid Build Coastguard Worker p[2] = c[0];
6148*89c4ff92SAndroid Build Coastguard Worker p[3] = c[3];
6149*89c4ff92SAndroid Build Coastguard Worker }
6150*89c4ff92SAndroid Build Coastguard Worker g->cur_x += 4;
6151*89c4ff92SAndroid Build Coastguard Worker
6152*89c4ff92SAndroid Build Coastguard Worker if (g->cur_x >= g->max_x) {
6153*89c4ff92SAndroid Build Coastguard Worker g->cur_x = g->start_x;
6154*89c4ff92SAndroid Build Coastguard Worker g->cur_y += g->step;
6155*89c4ff92SAndroid Build Coastguard Worker
6156*89c4ff92SAndroid Build Coastguard Worker while (g->cur_y >= g->max_y && g->parse > 0) {
6157*89c4ff92SAndroid Build Coastguard Worker g->step = (1 << g->parse) * g->line_size;
6158*89c4ff92SAndroid Build Coastguard Worker g->cur_y = g->start_y + (g->step >> 1);
6159*89c4ff92SAndroid Build Coastguard Worker --g->parse;
6160*89c4ff92SAndroid Build Coastguard Worker }
6161*89c4ff92SAndroid Build Coastguard Worker }
6162*89c4ff92SAndroid Build Coastguard Worker }
6163*89c4ff92SAndroid Build Coastguard Worker
stbi__process_gif_raster(stbi__context * s,stbi__gif * g)6164*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__process_gif_raster(stbi__context *s, stbi__gif *g)
6165*89c4ff92SAndroid Build Coastguard Worker {
6166*89c4ff92SAndroid Build Coastguard Worker stbi_uc lzw_cs;
6167*89c4ff92SAndroid Build Coastguard Worker stbi__int32 len, init_code;
6168*89c4ff92SAndroid Build Coastguard Worker stbi__uint32 first;
6169*89c4ff92SAndroid Build Coastguard Worker stbi__int32 codesize, codemask, avail, oldcode, bits, valid_bits, clear;
6170*89c4ff92SAndroid Build Coastguard Worker stbi__gif_lzw *p;
6171*89c4ff92SAndroid Build Coastguard Worker
6172*89c4ff92SAndroid Build Coastguard Worker lzw_cs = stbi__get8(s);
6173*89c4ff92SAndroid Build Coastguard Worker if (lzw_cs > 12) return NULL;
6174*89c4ff92SAndroid Build Coastguard Worker clear = 1 << lzw_cs;
6175*89c4ff92SAndroid Build Coastguard Worker first = 1;
6176*89c4ff92SAndroid Build Coastguard Worker codesize = lzw_cs + 1;
6177*89c4ff92SAndroid Build Coastguard Worker codemask = (1 << codesize) - 1;
6178*89c4ff92SAndroid Build Coastguard Worker bits = 0;
6179*89c4ff92SAndroid Build Coastguard Worker valid_bits = 0;
6180*89c4ff92SAndroid Build Coastguard Worker for (init_code = 0; init_code < clear; init_code++) {
6181*89c4ff92SAndroid Build Coastguard Worker g->codes[init_code].prefix = -1;
6182*89c4ff92SAndroid Build Coastguard Worker g->codes[init_code].first = (stbi_uc) init_code;
6183*89c4ff92SAndroid Build Coastguard Worker g->codes[init_code].suffix = (stbi_uc) init_code;
6184*89c4ff92SAndroid Build Coastguard Worker }
6185*89c4ff92SAndroid Build Coastguard Worker
6186*89c4ff92SAndroid Build Coastguard Worker // support no starting clear code
6187*89c4ff92SAndroid Build Coastguard Worker avail = clear+2;
6188*89c4ff92SAndroid Build Coastguard Worker oldcode = -1;
6189*89c4ff92SAndroid Build Coastguard Worker
6190*89c4ff92SAndroid Build Coastguard Worker len = 0;
6191*89c4ff92SAndroid Build Coastguard Worker for(;;) {
6192*89c4ff92SAndroid Build Coastguard Worker if (valid_bits < codesize) {
6193*89c4ff92SAndroid Build Coastguard Worker if (len == 0) {
6194*89c4ff92SAndroid Build Coastguard Worker len = stbi__get8(s); // start new block
6195*89c4ff92SAndroid Build Coastguard Worker if (len == 0)
6196*89c4ff92SAndroid Build Coastguard Worker return g->out;
6197*89c4ff92SAndroid Build Coastguard Worker }
6198*89c4ff92SAndroid Build Coastguard Worker --len;
6199*89c4ff92SAndroid Build Coastguard Worker bits |= (stbi__int32) stbi__get8(s) << valid_bits;
6200*89c4ff92SAndroid Build Coastguard Worker valid_bits += 8;
6201*89c4ff92SAndroid Build Coastguard Worker } else {
6202*89c4ff92SAndroid Build Coastguard Worker stbi__int32 code = bits & codemask;
6203*89c4ff92SAndroid Build Coastguard Worker bits >>= codesize;
6204*89c4ff92SAndroid Build Coastguard Worker valid_bits -= codesize;
6205*89c4ff92SAndroid Build Coastguard Worker // @OPTIMIZE: is there some way we can accelerate the non-clear path?
6206*89c4ff92SAndroid Build Coastguard Worker if (code == clear) { // clear code
6207*89c4ff92SAndroid Build Coastguard Worker codesize = lzw_cs + 1;
6208*89c4ff92SAndroid Build Coastguard Worker codemask = (1 << codesize) - 1;
6209*89c4ff92SAndroid Build Coastguard Worker avail = clear + 2;
6210*89c4ff92SAndroid Build Coastguard Worker oldcode = -1;
6211*89c4ff92SAndroid Build Coastguard Worker first = 0;
6212*89c4ff92SAndroid Build Coastguard Worker } else if (code == clear + 1) { // end of stream code
6213*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, len);
6214*89c4ff92SAndroid Build Coastguard Worker while ((len = stbi__get8(s)) > 0)
6215*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s,len);
6216*89c4ff92SAndroid Build Coastguard Worker return g->out;
6217*89c4ff92SAndroid Build Coastguard Worker } else if (code <= avail) {
6218*89c4ff92SAndroid Build Coastguard Worker if (first) return stbi__errpuc("no clear code", "Corrupt GIF");
6219*89c4ff92SAndroid Build Coastguard Worker
6220*89c4ff92SAndroid Build Coastguard Worker if (oldcode >= 0) {
6221*89c4ff92SAndroid Build Coastguard Worker p = &g->codes[avail++];
6222*89c4ff92SAndroid Build Coastguard Worker if (avail > 4096) return stbi__errpuc("too many codes", "Corrupt GIF");
6223*89c4ff92SAndroid Build Coastguard Worker p->prefix = (stbi__int16) oldcode;
6224*89c4ff92SAndroid Build Coastguard Worker p->first = g->codes[oldcode].first;
6225*89c4ff92SAndroid Build Coastguard Worker p->suffix = (code == avail) ? p->first : g->codes[code].first;
6226*89c4ff92SAndroid Build Coastguard Worker } else if (code == avail)
6227*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("illegal code in raster", "Corrupt GIF");
6228*89c4ff92SAndroid Build Coastguard Worker
6229*89c4ff92SAndroid Build Coastguard Worker stbi__out_gif_code(g, (stbi__uint16) code);
6230*89c4ff92SAndroid Build Coastguard Worker
6231*89c4ff92SAndroid Build Coastguard Worker if ((avail & codemask) == 0 && avail <= 0x0FFF) {
6232*89c4ff92SAndroid Build Coastguard Worker codesize++;
6233*89c4ff92SAndroid Build Coastguard Worker codemask = (1 << codesize) - 1;
6234*89c4ff92SAndroid Build Coastguard Worker }
6235*89c4ff92SAndroid Build Coastguard Worker
6236*89c4ff92SAndroid Build Coastguard Worker oldcode = code;
6237*89c4ff92SAndroid Build Coastguard Worker } else {
6238*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("illegal code in raster", "Corrupt GIF");
6239*89c4ff92SAndroid Build Coastguard Worker }
6240*89c4ff92SAndroid Build Coastguard Worker }
6241*89c4ff92SAndroid Build Coastguard Worker }
6242*89c4ff92SAndroid Build Coastguard Worker }
6243*89c4ff92SAndroid Build Coastguard Worker
stbi__fill_gif_background(stbi__gif * g,int x0,int y0,int x1,int y1)6244*89c4ff92SAndroid Build Coastguard Worker static void stbi__fill_gif_background(stbi__gif *g, int x0, int y0, int x1, int y1)
6245*89c4ff92SAndroid Build Coastguard Worker {
6246*89c4ff92SAndroid Build Coastguard Worker int x, y;
6247*89c4ff92SAndroid Build Coastguard Worker stbi_uc *c = g->pal[g->bgindex];
6248*89c4ff92SAndroid Build Coastguard Worker for (y = y0; y < y1; y += 4 * g->w) {
6249*89c4ff92SAndroid Build Coastguard Worker for (x = x0; x < x1; x += 4) {
6250*89c4ff92SAndroid Build Coastguard Worker stbi_uc *p = &g->out[y + x];
6251*89c4ff92SAndroid Build Coastguard Worker p[0] = c[2];
6252*89c4ff92SAndroid Build Coastguard Worker p[1] = c[1];
6253*89c4ff92SAndroid Build Coastguard Worker p[2] = c[0];
6254*89c4ff92SAndroid Build Coastguard Worker p[3] = 0;
6255*89c4ff92SAndroid Build Coastguard Worker }
6256*89c4ff92SAndroid Build Coastguard Worker }
6257*89c4ff92SAndroid Build Coastguard Worker }
6258*89c4ff92SAndroid Build Coastguard Worker
6259*89c4ff92SAndroid Build Coastguard Worker // this function is designed to support animated gifs, although stb_image doesn't support it
stbi__gif_load_next(stbi__context * s,stbi__gif * g,int * comp,int req_comp)6260*89c4ff92SAndroid Build Coastguard Worker static stbi_uc *stbi__gif_load_next(stbi__context *s, stbi__gif *g, int *comp, int req_comp)
6261*89c4ff92SAndroid Build Coastguard Worker {
6262*89c4ff92SAndroid Build Coastguard Worker int i;
6263*89c4ff92SAndroid Build Coastguard Worker stbi_uc *prev_out = 0;
6264*89c4ff92SAndroid Build Coastguard Worker
6265*89c4ff92SAndroid Build Coastguard Worker if (g->out == 0 && !stbi__gif_header(s, g, comp,0))
6266*89c4ff92SAndroid Build Coastguard Worker return 0; // stbi__g_failure_reason set by stbi__gif_header
6267*89c4ff92SAndroid Build Coastguard Worker
6268*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad3sizes_valid(g->w, g->h, 4, 0))
6269*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("too large", "GIF too large");
6270*89c4ff92SAndroid Build Coastguard Worker
6271*89c4ff92SAndroid Build Coastguard Worker prev_out = g->out;
6272*89c4ff92SAndroid Build Coastguard Worker g->out = (stbi_uc *) stbi__malloc_mad3(4, g->w, g->h, 0);
6273*89c4ff92SAndroid Build Coastguard Worker if (g->out == 0) return stbi__errpuc("outofmem", "Out of memory");
6274*89c4ff92SAndroid Build Coastguard Worker
6275*89c4ff92SAndroid Build Coastguard Worker switch ((g->eflags & 0x1C) >> 2) {
6276*89c4ff92SAndroid Build Coastguard Worker case 0: // unspecified (also always used on 1st frame)
6277*89c4ff92SAndroid Build Coastguard Worker stbi__fill_gif_background(g, 0, 0, 4 * g->w, 4 * g->w * g->h);
6278*89c4ff92SAndroid Build Coastguard Worker break;
6279*89c4ff92SAndroid Build Coastguard Worker case 1: // do not dispose
6280*89c4ff92SAndroid Build Coastguard Worker if (prev_out) memcpy(g->out, prev_out, 4 * g->w * g->h);
6281*89c4ff92SAndroid Build Coastguard Worker g->old_out = prev_out;
6282*89c4ff92SAndroid Build Coastguard Worker break;
6283*89c4ff92SAndroid Build Coastguard Worker case 2: // dispose to background
6284*89c4ff92SAndroid Build Coastguard Worker if (prev_out) memcpy(g->out, prev_out, 4 * g->w * g->h);
6285*89c4ff92SAndroid Build Coastguard Worker stbi__fill_gif_background(g, g->start_x, g->start_y, g->max_x, g->max_y);
6286*89c4ff92SAndroid Build Coastguard Worker break;
6287*89c4ff92SAndroid Build Coastguard Worker case 3: // dispose to previous
6288*89c4ff92SAndroid Build Coastguard Worker if (g->old_out) {
6289*89c4ff92SAndroid Build Coastguard Worker for (i = g->start_y; i < g->max_y; i += 4 * g->w)
6290*89c4ff92SAndroid Build Coastguard Worker memcpy(&g->out[i + g->start_x], &g->old_out[i + g->start_x], g->max_x - g->start_x);
6291*89c4ff92SAndroid Build Coastguard Worker }
6292*89c4ff92SAndroid Build Coastguard Worker break;
6293*89c4ff92SAndroid Build Coastguard Worker }
6294*89c4ff92SAndroid Build Coastguard Worker
6295*89c4ff92SAndroid Build Coastguard Worker for (;;) {
6296*89c4ff92SAndroid Build Coastguard Worker switch (stbi__get8(s)) {
6297*89c4ff92SAndroid Build Coastguard Worker case 0x2C: /* Image Descriptor */
6298*89c4ff92SAndroid Build Coastguard Worker {
6299*89c4ff92SAndroid Build Coastguard Worker int prev_trans = -1;
6300*89c4ff92SAndroid Build Coastguard Worker stbi__int32 x, y, w, h;
6301*89c4ff92SAndroid Build Coastguard Worker stbi_uc *o;
6302*89c4ff92SAndroid Build Coastguard Worker
6303*89c4ff92SAndroid Build Coastguard Worker x = stbi__get16le(s);
6304*89c4ff92SAndroid Build Coastguard Worker y = stbi__get16le(s);
6305*89c4ff92SAndroid Build Coastguard Worker w = stbi__get16le(s);
6306*89c4ff92SAndroid Build Coastguard Worker h = stbi__get16le(s);
6307*89c4ff92SAndroid Build Coastguard Worker if (((x + w) > (g->w)) || ((y + h) > (g->h)))
6308*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("bad Image Descriptor", "Corrupt GIF");
6309*89c4ff92SAndroid Build Coastguard Worker
6310*89c4ff92SAndroid Build Coastguard Worker g->line_size = g->w * 4;
6311*89c4ff92SAndroid Build Coastguard Worker g->start_x = x * 4;
6312*89c4ff92SAndroid Build Coastguard Worker g->start_y = y * g->line_size;
6313*89c4ff92SAndroid Build Coastguard Worker g->max_x = g->start_x + w * 4;
6314*89c4ff92SAndroid Build Coastguard Worker g->max_y = g->start_y + h * g->line_size;
6315*89c4ff92SAndroid Build Coastguard Worker g->cur_x = g->start_x;
6316*89c4ff92SAndroid Build Coastguard Worker g->cur_y = g->start_y;
6317*89c4ff92SAndroid Build Coastguard Worker
6318*89c4ff92SAndroid Build Coastguard Worker g->lflags = stbi__get8(s);
6319*89c4ff92SAndroid Build Coastguard Worker
6320*89c4ff92SAndroid Build Coastguard Worker if (g->lflags & 0x40) {
6321*89c4ff92SAndroid Build Coastguard Worker g->step = 8 * g->line_size; // first interlaced spacing
6322*89c4ff92SAndroid Build Coastguard Worker g->parse = 3;
6323*89c4ff92SAndroid Build Coastguard Worker } else {
6324*89c4ff92SAndroid Build Coastguard Worker g->step = g->line_size;
6325*89c4ff92SAndroid Build Coastguard Worker g->parse = 0;
6326*89c4ff92SAndroid Build Coastguard Worker }
6327*89c4ff92SAndroid Build Coastguard Worker
6328*89c4ff92SAndroid Build Coastguard Worker if (g->lflags & 0x80) {
6329*89c4ff92SAndroid Build Coastguard Worker stbi__gif_parse_colortable(s,g->lpal, 2 << (g->lflags & 7), g->eflags & 0x01 ? g->transparent : -1);
6330*89c4ff92SAndroid Build Coastguard Worker g->color_table = (stbi_uc *) g->lpal;
6331*89c4ff92SAndroid Build Coastguard Worker } else if (g->flags & 0x80) {
6332*89c4ff92SAndroid Build Coastguard Worker if (g->transparent >= 0 && (g->eflags & 0x01)) {
6333*89c4ff92SAndroid Build Coastguard Worker prev_trans = g->pal[g->transparent][3];
6334*89c4ff92SAndroid Build Coastguard Worker g->pal[g->transparent][3] = 0;
6335*89c4ff92SAndroid Build Coastguard Worker }
6336*89c4ff92SAndroid Build Coastguard Worker g->color_table = (stbi_uc *) g->pal;
6337*89c4ff92SAndroid Build Coastguard Worker } else
6338*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("missing color table", "Corrupt GIF");
6339*89c4ff92SAndroid Build Coastguard Worker
6340*89c4ff92SAndroid Build Coastguard Worker o = stbi__process_gif_raster(s, g);
6341*89c4ff92SAndroid Build Coastguard Worker if (o == NULL) return NULL;
6342*89c4ff92SAndroid Build Coastguard Worker
6343*89c4ff92SAndroid Build Coastguard Worker if (prev_trans != -1)
6344*89c4ff92SAndroid Build Coastguard Worker g->pal[g->transparent][3] = (stbi_uc) prev_trans;
6345*89c4ff92SAndroid Build Coastguard Worker
6346*89c4ff92SAndroid Build Coastguard Worker return o;
6347*89c4ff92SAndroid Build Coastguard Worker }
6348*89c4ff92SAndroid Build Coastguard Worker
6349*89c4ff92SAndroid Build Coastguard Worker case 0x21: // Comment Extension.
6350*89c4ff92SAndroid Build Coastguard Worker {
6351*89c4ff92SAndroid Build Coastguard Worker int len;
6352*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) == 0xF9) { // Graphic Control Extension.
6353*89c4ff92SAndroid Build Coastguard Worker len = stbi__get8(s);
6354*89c4ff92SAndroid Build Coastguard Worker if (len == 4) {
6355*89c4ff92SAndroid Build Coastguard Worker g->eflags = stbi__get8(s);
6356*89c4ff92SAndroid Build Coastguard Worker g->delay = stbi__get16le(s);
6357*89c4ff92SAndroid Build Coastguard Worker g->transparent = stbi__get8(s);
6358*89c4ff92SAndroid Build Coastguard Worker } else {
6359*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, len);
6360*89c4ff92SAndroid Build Coastguard Worker break;
6361*89c4ff92SAndroid Build Coastguard Worker }
6362*89c4ff92SAndroid Build Coastguard Worker }
6363*89c4ff92SAndroid Build Coastguard Worker while ((len = stbi__get8(s)) != 0)
6364*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, len);
6365*89c4ff92SAndroid Build Coastguard Worker break;
6366*89c4ff92SAndroid Build Coastguard Worker }
6367*89c4ff92SAndroid Build Coastguard Worker
6368*89c4ff92SAndroid Build Coastguard Worker case 0x3B: // gif stream termination code
6369*89c4ff92SAndroid Build Coastguard Worker return (stbi_uc *) s; // using '1' causes warning on some compilers
6370*89c4ff92SAndroid Build Coastguard Worker
6371*89c4ff92SAndroid Build Coastguard Worker default:
6372*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("unknown code", "Corrupt GIF");
6373*89c4ff92SAndroid Build Coastguard Worker }
6374*89c4ff92SAndroid Build Coastguard Worker }
6375*89c4ff92SAndroid Build Coastguard Worker
6376*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(req_comp);
6377*89c4ff92SAndroid Build Coastguard Worker }
6378*89c4ff92SAndroid Build Coastguard Worker
stbi__gif_load(stbi__context * s,int * x,int * y,int * comp,int req_comp,stbi__result_info * ri)6379*89c4ff92SAndroid Build Coastguard Worker static void *stbi__gif_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri)
6380*89c4ff92SAndroid Build Coastguard Worker {
6381*89c4ff92SAndroid Build Coastguard Worker stbi_uc *u = 0;
6382*89c4ff92SAndroid Build Coastguard Worker stbi__gif* g = (stbi__gif*) stbi__malloc(sizeof(stbi__gif));
6383*89c4ff92SAndroid Build Coastguard Worker memset(g, 0, sizeof(*g));
6384*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(ri);
6385*89c4ff92SAndroid Build Coastguard Worker
6386*89c4ff92SAndroid Build Coastguard Worker u = stbi__gif_load_next(s, g, comp, req_comp);
6387*89c4ff92SAndroid Build Coastguard Worker if (u == (stbi_uc *) s) u = 0; // end of animated gif marker
6388*89c4ff92SAndroid Build Coastguard Worker if (u) {
6389*89c4ff92SAndroid Build Coastguard Worker *x = g->w;
6390*89c4ff92SAndroid Build Coastguard Worker *y = g->h;
6391*89c4ff92SAndroid Build Coastguard Worker if (req_comp && req_comp != 4)
6392*89c4ff92SAndroid Build Coastguard Worker u = stbi__convert_format(u, 4, req_comp, g->w, g->h);
6393*89c4ff92SAndroid Build Coastguard Worker }
6394*89c4ff92SAndroid Build Coastguard Worker else if (g->out)
6395*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(g->out);
6396*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(g);
6397*89c4ff92SAndroid Build Coastguard Worker return u;
6398*89c4ff92SAndroid Build Coastguard Worker }
6399*89c4ff92SAndroid Build Coastguard Worker
stbi__gif_info(stbi__context * s,int * x,int * y,int * comp)6400*89c4ff92SAndroid Build Coastguard Worker static int stbi__gif_info(stbi__context *s, int *x, int *y, int *comp)
6401*89c4ff92SAndroid Build Coastguard Worker {
6402*89c4ff92SAndroid Build Coastguard Worker return stbi__gif_info_raw(s,x,y,comp);
6403*89c4ff92SAndroid Build Coastguard Worker }
6404*89c4ff92SAndroid Build Coastguard Worker #endif
6405*89c4ff92SAndroid Build Coastguard Worker
6406*89c4ff92SAndroid Build Coastguard Worker // *************************************************************************************************
6407*89c4ff92SAndroid Build Coastguard Worker // Radiance RGBE HDR loader
6408*89c4ff92SAndroid Build Coastguard Worker // originally by Nicolas Schulz
6409*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
stbi__hdr_test_core(stbi__context * s,const char * signature)6410*89c4ff92SAndroid Build Coastguard Worker static int stbi__hdr_test_core(stbi__context *s, const char *signature)
6411*89c4ff92SAndroid Build Coastguard Worker {
6412*89c4ff92SAndroid Build Coastguard Worker int i;
6413*89c4ff92SAndroid Build Coastguard Worker for (i=0; signature[i]; ++i)
6414*89c4ff92SAndroid Build Coastguard Worker if (stbi__get8(s) != signature[i])
6415*89c4ff92SAndroid Build Coastguard Worker return 0;
6416*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
6417*89c4ff92SAndroid Build Coastguard Worker return 1;
6418*89c4ff92SAndroid Build Coastguard Worker }
6419*89c4ff92SAndroid Build Coastguard Worker
stbi__hdr_test(stbi__context * s)6420*89c4ff92SAndroid Build Coastguard Worker static int stbi__hdr_test(stbi__context* s)
6421*89c4ff92SAndroid Build Coastguard Worker {
6422*89c4ff92SAndroid Build Coastguard Worker int r = stbi__hdr_test_core(s, "#?RADIANCE\n");
6423*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
6424*89c4ff92SAndroid Build Coastguard Worker if(!r) {
6425*89c4ff92SAndroid Build Coastguard Worker r = stbi__hdr_test_core(s, "#?RGBE\n");
6426*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
6427*89c4ff92SAndroid Build Coastguard Worker }
6428*89c4ff92SAndroid Build Coastguard Worker return r;
6429*89c4ff92SAndroid Build Coastguard Worker }
6430*89c4ff92SAndroid Build Coastguard Worker
6431*89c4ff92SAndroid Build Coastguard Worker #define STBI__HDR_BUFLEN 1024
stbi__hdr_gettoken(stbi__context * z,char * buffer)6432*89c4ff92SAndroid Build Coastguard Worker static char *stbi__hdr_gettoken(stbi__context *z, char *buffer)
6433*89c4ff92SAndroid Build Coastguard Worker {
6434*89c4ff92SAndroid Build Coastguard Worker int len=0;
6435*89c4ff92SAndroid Build Coastguard Worker char c = '\0';
6436*89c4ff92SAndroid Build Coastguard Worker
6437*89c4ff92SAndroid Build Coastguard Worker c = (char) stbi__get8(z);
6438*89c4ff92SAndroid Build Coastguard Worker
6439*89c4ff92SAndroid Build Coastguard Worker while (!stbi__at_eof(z) && c != '\n') {
6440*89c4ff92SAndroid Build Coastguard Worker buffer[len++] = c;
6441*89c4ff92SAndroid Build Coastguard Worker if (len == STBI__HDR_BUFLEN-1) {
6442*89c4ff92SAndroid Build Coastguard Worker // flush to end of line
6443*89c4ff92SAndroid Build Coastguard Worker while (!stbi__at_eof(z) && stbi__get8(z) != '\n')
6444*89c4ff92SAndroid Build Coastguard Worker ;
6445*89c4ff92SAndroid Build Coastguard Worker break;
6446*89c4ff92SAndroid Build Coastguard Worker }
6447*89c4ff92SAndroid Build Coastguard Worker c = (char) stbi__get8(z);
6448*89c4ff92SAndroid Build Coastguard Worker }
6449*89c4ff92SAndroid Build Coastguard Worker
6450*89c4ff92SAndroid Build Coastguard Worker buffer[len] = 0;
6451*89c4ff92SAndroid Build Coastguard Worker return buffer;
6452*89c4ff92SAndroid Build Coastguard Worker }
6453*89c4ff92SAndroid Build Coastguard Worker
stbi__hdr_convert(float * output,stbi_uc * input,int req_comp)6454*89c4ff92SAndroid Build Coastguard Worker static void stbi__hdr_convert(float *output, stbi_uc *input, int req_comp)
6455*89c4ff92SAndroid Build Coastguard Worker {
6456*89c4ff92SAndroid Build Coastguard Worker if ( input[3] != 0 ) {
6457*89c4ff92SAndroid Build Coastguard Worker float f1;
6458*89c4ff92SAndroid Build Coastguard Worker // Exponent
6459*89c4ff92SAndroid Build Coastguard Worker f1 = (float) ldexp(1.0f, input[3] - (int)(128 + 8));
6460*89c4ff92SAndroid Build Coastguard Worker if (req_comp <= 2)
6461*89c4ff92SAndroid Build Coastguard Worker output[0] = (input[0] + input[1] + input[2]) * f1 / 3;
6462*89c4ff92SAndroid Build Coastguard Worker else {
6463*89c4ff92SAndroid Build Coastguard Worker output[0] = input[0] * f1;
6464*89c4ff92SAndroid Build Coastguard Worker output[1] = input[1] * f1;
6465*89c4ff92SAndroid Build Coastguard Worker output[2] = input[2] * f1;
6466*89c4ff92SAndroid Build Coastguard Worker }
6467*89c4ff92SAndroid Build Coastguard Worker if (req_comp == 2) output[1] = 1;
6468*89c4ff92SAndroid Build Coastguard Worker if (req_comp == 4) output[3] = 1;
6469*89c4ff92SAndroid Build Coastguard Worker } else {
6470*89c4ff92SAndroid Build Coastguard Worker switch (req_comp) {
6471*89c4ff92SAndroid Build Coastguard Worker case 4: output[3] = 1; /* fallthrough */
6472*89c4ff92SAndroid Build Coastguard Worker case 3: output[0] = output[1] = output[2] = 0;
6473*89c4ff92SAndroid Build Coastguard Worker break;
6474*89c4ff92SAndroid Build Coastguard Worker case 2: output[1] = 1; /* fallthrough */
6475*89c4ff92SAndroid Build Coastguard Worker case 1: output[0] = 0;
6476*89c4ff92SAndroid Build Coastguard Worker break;
6477*89c4ff92SAndroid Build Coastguard Worker }
6478*89c4ff92SAndroid Build Coastguard Worker }
6479*89c4ff92SAndroid Build Coastguard Worker }
6480*89c4ff92SAndroid Build Coastguard Worker
stbi__hdr_load(stbi__context * s,int * x,int * y,int * comp,int req_comp,stbi__result_info * ri)6481*89c4ff92SAndroid Build Coastguard Worker static float *stbi__hdr_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri)
6482*89c4ff92SAndroid Build Coastguard Worker {
6483*89c4ff92SAndroid Build Coastguard Worker char buffer[STBI__HDR_BUFLEN];
6484*89c4ff92SAndroid Build Coastguard Worker char *token;
6485*89c4ff92SAndroid Build Coastguard Worker int valid = 0;
6486*89c4ff92SAndroid Build Coastguard Worker int width, height;
6487*89c4ff92SAndroid Build Coastguard Worker stbi_uc *scanline;
6488*89c4ff92SAndroid Build Coastguard Worker float *hdr_data;
6489*89c4ff92SAndroid Build Coastguard Worker int len;
6490*89c4ff92SAndroid Build Coastguard Worker unsigned char count, value;
6491*89c4ff92SAndroid Build Coastguard Worker int i, j, k, c1,c2, z;
6492*89c4ff92SAndroid Build Coastguard Worker const char *headerToken;
6493*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(ri);
6494*89c4ff92SAndroid Build Coastguard Worker
6495*89c4ff92SAndroid Build Coastguard Worker // Check identifier
6496*89c4ff92SAndroid Build Coastguard Worker headerToken = stbi__hdr_gettoken(s,buffer);
6497*89c4ff92SAndroid Build Coastguard Worker if (strcmp(headerToken, "#?RADIANCE") != 0 && strcmp(headerToken, "#?RGBE") != 0)
6498*89c4ff92SAndroid Build Coastguard Worker return stbi__errpf("not HDR", "Corrupt HDR image");
6499*89c4ff92SAndroid Build Coastguard Worker
6500*89c4ff92SAndroid Build Coastguard Worker // Parse header
6501*89c4ff92SAndroid Build Coastguard Worker for(;;) {
6502*89c4ff92SAndroid Build Coastguard Worker token = stbi__hdr_gettoken(s,buffer);
6503*89c4ff92SAndroid Build Coastguard Worker if (token[0] == 0) break;
6504*89c4ff92SAndroid Build Coastguard Worker if (strcmp(token, "FORMAT=32-bit_rle_rgbe") == 0) valid = 1;
6505*89c4ff92SAndroid Build Coastguard Worker }
6506*89c4ff92SAndroid Build Coastguard Worker
6507*89c4ff92SAndroid Build Coastguard Worker if (!valid) return stbi__errpf("unsupported format", "Unsupported HDR format");
6508*89c4ff92SAndroid Build Coastguard Worker
6509*89c4ff92SAndroid Build Coastguard Worker // Parse width and height
6510*89c4ff92SAndroid Build Coastguard Worker // can't use sscanf() if we're not using stdio!
6511*89c4ff92SAndroid Build Coastguard Worker token = stbi__hdr_gettoken(s,buffer);
6512*89c4ff92SAndroid Build Coastguard Worker if (strncmp(token, "-Y ", 3)) return stbi__errpf("unsupported data layout", "Unsupported HDR format");
6513*89c4ff92SAndroid Build Coastguard Worker token += 3;
6514*89c4ff92SAndroid Build Coastguard Worker height = (int) strtol(token, &token, 10);
6515*89c4ff92SAndroid Build Coastguard Worker while (*token == ' ') ++token;
6516*89c4ff92SAndroid Build Coastguard Worker if (strncmp(token, "+X ", 3)) return stbi__errpf("unsupported data layout", "Unsupported HDR format");
6517*89c4ff92SAndroid Build Coastguard Worker token += 3;
6518*89c4ff92SAndroid Build Coastguard Worker width = (int) strtol(token, NULL, 10);
6519*89c4ff92SAndroid Build Coastguard Worker
6520*89c4ff92SAndroid Build Coastguard Worker *x = width;
6521*89c4ff92SAndroid Build Coastguard Worker *y = height;
6522*89c4ff92SAndroid Build Coastguard Worker
6523*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = 3;
6524*89c4ff92SAndroid Build Coastguard Worker if (req_comp == 0) req_comp = 3;
6525*89c4ff92SAndroid Build Coastguard Worker
6526*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad4sizes_valid(width, height, req_comp, sizeof(float), 0))
6527*89c4ff92SAndroid Build Coastguard Worker return stbi__errpf("too large", "HDR image is too large");
6528*89c4ff92SAndroid Build Coastguard Worker
6529*89c4ff92SAndroid Build Coastguard Worker // Read data
6530*89c4ff92SAndroid Build Coastguard Worker hdr_data = (float *) stbi__malloc_mad4(width, height, req_comp, sizeof(float), 0);
6531*89c4ff92SAndroid Build Coastguard Worker if (!hdr_data)
6532*89c4ff92SAndroid Build Coastguard Worker return stbi__errpf("outofmem", "Out of memory");
6533*89c4ff92SAndroid Build Coastguard Worker
6534*89c4ff92SAndroid Build Coastguard Worker // Load image data
6535*89c4ff92SAndroid Build Coastguard Worker // image data is stored as some number of sca
6536*89c4ff92SAndroid Build Coastguard Worker if ( width < 8 || width >= 32768) {
6537*89c4ff92SAndroid Build Coastguard Worker // Read flat data
6538*89c4ff92SAndroid Build Coastguard Worker for (j=0; j < height; ++j) {
6539*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < width; ++i) {
6540*89c4ff92SAndroid Build Coastguard Worker stbi_uc rgbe[4];
6541*89c4ff92SAndroid Build Coastguard Worker main_decode_loop:
6542*89c4ff92SAndroid Build Coastguard Worker stbi__getn(s, rgbe, 4);
6543*89c4ff92SAndroid Build Coastguard Worker stbi__hdr_convert(hdr_data + j * width * req_comp + i * req_comp, rgbe, req_comp);
6544*89c4ff92SAndroid Build Coastguard Worker }
6545*89c4ff92SAndroid Build Coastguard Worker }
6546*89c4ff92SAndroid Build Coastguard Worker } else {
6547*89c4ff92SAndroid Build Coastguard Worker // Read RLE-encoded data
6548*89c4ff92SAndroid Build Coastguard Worker scanline = NULL;
6549*89c4ff92SAndroid Build Coastguard Worker
6550*89c4ff92SAndroid Build Coastguard Worker for (j = 0; j < height; ++j) {
6551*89c4ff92SAndroid Build Coastguard Worker c1 = stbi__get8(s);
6552*89c4ff92SAndroid Build Coastguard Worker c2 = stbi__get8(s);
6553*89c4ff92SAndroid Build Coastguard Worker len = stbi__get8(s);
6554*89c4ff92SAndroid Build Coastguard Worker if (c1 != 2 || c2 != 2 || (len & 0x80)) {
6555*89c4ff92SAndroid Build Coastguard Worker // not run-length encoded, so we have to actually use THIS data as a decoded
6556*89c4ff92SAndroid Build Coastguard Worker // pixel (note this can't be a valid pixel--one of RGB must be >= 128)
6557*89c4ff92SAndroid Build Coastguard Worker stbi_uc rgbe[4];
6558*89c4ff92SAndroid Build Coastguard Worker rgbe[0] = (stbi_uc) c1;
6559*89c4ff92SAndroid Build Coastguard Worker rgbe[1] = (stbi_uc) c2;
6560*89c4ff92SAndroid Build Coastguard Worker rgbe[2] = (stbi_uc) len;
6561*89c4ff92SAndroid Build Coastguard Worker rgbe[3] = (stbi_uc) stbi__get8(s);
6562*89c4ff92SAndroid Build Coastguard Worker stbi__hdr_convert(hdr_data, rgbe, req_comp);
6563*89c4ff92SAndroid Build Coastguard Worker i = 1;
6564*89c4ff92SAndroid Build Coastguard Worker j = 0;
6565*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(scanline);
6566*89c4ff92SAndroid Build Coastguard Worker goto main_decode_loop; // yes, this makes no sense
6567*89c4ff92SAndroid Build Coastguard Worker }
6568*89c4ff92SAndroid Build Coastguard Worker len <<= 8;
6569*89c4ff92SAndroid Build Coastguard Worker len |= stbi__get8(s);
6570*89c4ff92SAndroid Build Coastguard Worker if (len != width) { STBI_FREE(hdr_data); STBI_FREE(scanline); return stbi__errpf("invalid decoded scanline length", "corrupt HDR"); }
6571*89c4ff92SAndroid Build Coastguard Worker if (scanline == NULL) {
6572*89c4ff92SAndroid Build Coastguard Worker scanline = (stbi_uc *) stbi__malloc_mad2(width, 4, 0);
6573*89c4ff92SAndroid Build Coastguard Worker if (!scanline) {
6574*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(hdr_data);
6575*89c4ff92SAndroid Build Coastguard Worker return stbi__errpf("outofmem", "Out of memory");
6576*89c4ff92SAndroid Build Coastguard Worker }
6577*89c4ff92SAndroid Build Coastguard Worker }
6578*89c4ff92SAndroid Build Coastguard Worker
6579*89c4ff92SAndroid Build Coastguard Worker for (k = 0; k < 4; ++k) {
6580*89c4ff92SAndroid Build Coastguard Worker int nleft;
6581*89c4ff92SAndroid Build Coastguard Worker i = 0;
6582*89c4ff92SAndroid Build Coastguard Worker while ((nleft = width - i) > 0) {
6583*89c4ff92SAndroid Build Coastguard Worker count = stbi__get8(s);
6584*89c4ff92SAndroid Build Coastguard Worker if (count > 128) {
6585*89c4ff92SAndroid Build Coastguard Worker // Run
6586*89c4ff92SAndroid Build Coastguard Worker value = stbi__get8(s);
6587*89c4ff92SAndroid Build Coastguard Worker count -= 128;
6588*89c4ff92SAndroid Build Coastguard Worker if (count > nleft) { STBI_FREE(hdr_data); STBI_FREE(scanline); return stbi__errpf("corrupt", "bad RLE data in HDR"); }
6589*89c4ff92SAndroid Build Coastguard Worker for (z = 0; z < count; ++z)
6590*89c4ff92SAndroid Build Coastguard Worker scanline[i++ * 4 + k] = value;
6591*89c4ff92SAndroid Build Coastguard Worker } else {
6592*89c4ff92SAndroid Build Coastguard Worker // Dump
6593*89c4ff92SAndroid Build Coastguard Worker if (count > nleft) { STBI_FREE(hdr_data); STBI_FREE(scanline); return stbi__errpf("corrupt", "bad RLE data in HDR"); }
6594*89c4ff92SAndroid Build Coastguard Worker for (z = 0; z < count; ++z)
6595*89c4ff92SAndroid Build Coastguard Worker scanline[i++ * 4 + k] = stbi__get8(s);
6596*89c4ff92SAndroid Build Coastguard Worker }
6597*89c4ff92SAndroid Build Coastguard Worker }
6598*89c4ff92SAndroid Build Coastguard Worker }
6599*89c4ff92SAndroid Build Coastguard Worker for (i=0; i < width; ++i)
6600*89c4ff92SAndroid Build Coastguard Worker stbi__hdr_convert(hdr_data+(j*width + i)*req_comp, scanline + i*4, req_comp);
6601*89c4ff92SAndroid Build Coastguard Worker }
6602*89c4ff92SAndroid Build Coastguard Worker if (scanline)
6603*89c4ff92SAndroid Build Coastguard Worker STBI_FREE(scanline);
6604*89c4ff92SAndroid Build Coastguard Worker }
6605*89c4ff92SAndroid Build Coastguard Worker
6606*89c4ff92SAndroid Build Coastguard Worker return hdr_data;
6607*89c4ff92SAndroid Build Coastguard Worker }
6608*89c4ff92SAndroid Build Coastguard Worker
stbi__hdr_info(stbi__context * s,int * x,int * y,int * comp)6609*89c4ff92SAndroid Build Coastguard Worker static int stbi__hdr_info(stbi__context *s, int *x, int *y, int *comp)
6610*89c4ff92SAndroid Build Coastguard Worker {
6611*89c4ff92SAndroid Build Coastguard Worker char buffer[STBI__HDR_BUFLEN];
6612*89c4ff92SAndroid Build Coastguard Worker char *token;
6613*89c4ff92SAndroid Build Coastguard Worker int valid = 0;
6614*89c4ff92SAndroid Build Coastguard Worker int dummy;
6615*89c4ff92SAndroid Build Coastguard Worker
6616*89c4ff92SAndroid Build Coastguard Worker if (!x) x = &dummy;
6617*89c4ff92SAndroid Build Coastguard Worker if (!y) y = &dummy;
6618*89c4ff92SAndroid Build Coastguard Worker if (!comp) comp = &dummy;
6619*89c4ff92SAndroid Build Coastguard Worker
6620*89c4ff92SAndroid Build Coastguard Worker if (stbi__hdr_test(s) == 0) {
6621*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6622*89c4ff92SAndroid Build Coastguard Worker return 0;
6623*89c4ff92SAndroid Build Coastguard Worker }
6624*89c4ff92SAndroid Build Coastguard Worker
6625*89c4ff92SAndroid Build Coastguard Worker for(;;) {
6626*89c4ff92SAndroid Build Coastguard Worker token = stbi__hdr_gettoken(s,buffer);
6627*89c4ff92SAndroid Build Coastguard Worker if (token[0] == 0) break;
6628*89c4ff92SAndroid Build Coastguard Worker if (strcmp(token, "FORMAT=32-bit_rle_rgbe") == 0) valid = 1;
6629*89c4ff92SAndroid Build Coastguard Worker }
6630*89c4ff92SAndroid Build Coastguard Worker
6631*89c4ff92SAndroid Build Coastguard Worker if (!valid) {
6632*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6633*89c4ff92SAndroid Build Coastguard Worker return 0;
6634*89c4ff92SAndroid Build Coastguard Worker }
6635*89c4ff92SAndroid Build Coastguard Worker token = stbi__hdr_gettoken(s,buffer);
6636*89c4ff92SAndroid Build Coastguard Worker if (strncmp(token, "-Y ", 3)) {
6637*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6638*89c4ff92SAndroid Build Coastguard Worker return 0;
6639*89c4ff92SAndroid Build Coastguard Worker }
6640*89c4ff92SAndroid Build Coastguard Worker token += 3;
6641*89c4ff92SAndroid Build Coastguard Worker *y = (int) strtol(token, &token, 10);
6642*89c4ff92SAndroid Build Coastguard Worker while (*token == ' ') ++token;
6643*89c4ff92SAndroid Build Coastguard Worker if (strncmp(token, "+X ", 3)) {
6644*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6645*89c4ff92SAndroid Build Coastguard Worker return 0;
6646*89c4ff92SAndroid Build Coastguard Worker }
6647*89c4ff92SAndroid Build Coastguard Worker token += 3;
6648*89c4ff92SAndroid Build Coastguard Worker *x = (int) strtol(token, NULL, 10);
6649*89c4ff92SAndroid Build Coastguard Worker *comp = 3;
6650*89c4ff92SAndroid Build Coastguard Worker return 1;
6651*89c4ff92SAndroid Build Coastguard Worker }
6652*89c4ff92SAndroid Build Coastguard Worker #endif // STBI_NO_HDR
6653*89c4ff92SAndroid Build Coastguard Worker
6654*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_BMP
stbi__bmp_info(stbi__context * s,int * x,int * y,int * comp)6655*89c4ff92SAndroid Build Coastguard Worker static int stbi__bmp_info(stbi__context *s, int *x, int *y, int *comp)
6656*89c4ff92SAndroid Build Coastguard Worker {
6657*89c4ff92SAndroid Build Coastguard Worker void *p;
6658*89c4ff92SAndroid Build Coastguard Worker stbi__bmp_data info;
6659*89c4ff92SAndroid Build Coastguard Worker
6660*89c4ff92SAndroid Build Coastguard Worker info.all_a = 255;
6661*89c4ff92SAndroid Build Coastguard Worker p = stbi__bmp_parse_header(s, &info);
6662*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6663*89c4ff92SAndroid Build Coastguard Worker if (p == NULL)
6664*89c4ff92SAndroid Build Coastguard Worker return 0;
6665*89c4ff92SAndroid Build Coastguard Worker if (x) *x = s->img_x;
6666*89c4ff92SAndroid Build Coastguard Worker if (y) *y = s->img_y;
6667*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = info.ma ? 4 : 3;
6668*89c4ff92SAndroid Build Coastguard Worker return 1;
6669*89c4ff92SAndroid Build Coastguard Worker }
6670*89c4ff92SAndroid Build Coastguard Worker #endif
6671*89c4ff92SAndroid Build Coastguard Worker
6672*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PSD
stbi__psd_info(stbi__context * s,int * x,int * y,int * comp)6673*89c4ff92SAndroid Build Coastguard Worker static int stbi__psd_info(stbi__context *s, int *x, int *y, int *comp)
6674*89c4ff92SAndroid Build Coastguard Worker {
6675*89c4ff92SAndroid Build Coastguard Worker int channelCount, dummy;
6676*89c4ff92SAndroid Build Coastguard Worker if (!x) x = &dummy;
6677*89c4ff92SAndroid Build Coastguard Worker if (!y) y = &dummy;
6678*89c4ff92SAndroid Build Coastguard Worker if (!comp) comp = &dummy;
6679*89c4ff92SAndroid Build Coastguard Worker if (stbi__get32be(s) != 0x38425053) {
6680*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6681*89c4ff92SAndroid Build Coastguard Worker return 0;
6682*89c4ff92SAndroid Build Coastguard Worker }
6683*89c4ff92SAndroid Build Coastguard Worker if (stbi__get16be(s) != 1) {
6684*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6685*89c4ff92SAndroid Build Coastguard Worker return 0;
6686*89c4ff92SAndroid Build Coastguard Worker }
6687*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, 6);
6688*89c4ff92SAndroid Build Coastguard Worker channelCount = stbi__get16be(s);
6689*89c4ff92SAndroid Build Coastguard Worker if (channelCount < 0 || channelCount > 16) {
6690*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6691*89c4ff92SAndroid Build Coastguard Worker return 0;
6692*89c4ff92SAndroid Build Coastguard Worker }
6693*89c4ff92SAndroid Build Coastguard Worker *y = stbi__get32be(s);
6694*89c4ff92SAndroid Build Coastguard Worker *x = stbi__get32be(s);
6695*89c4ff92SAndroid Build Coastguard Worker if (stbi__get16be(s) != 8) {
6696*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6697*89c4ff92SAndroid Build Coastguard Worker return 0;
6698*89c4ff92SAndroid Build Coastguard Worker }
6699*89c4ff92SAndroid Build Coastguard Worker if (stbi__get16be(s) != 3) {
6700*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6701*89c4ff92SAndroid Build Coastguard Worker return 0;
6702*89c4ff92SAndroid Build Coastguard Worker }
6703*89c4ff92SAndroid Build Coastguard Worker *comp = 4;
6704*89c4ff92SAndroid Build Coastguard Worker return 1;
6705*89c4ff92SAndroid Build Coastguard Worker }
6706*89c4ff92SAndroid Build Coastguard Worker #endif
6707*89c4ff92SAndroid Build Coastguard Worker
6708*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PIC
stbi__pic_info(stbi__context * s,int * x,int * y,int * comp)6709*89c4ff92SAndroid Build Coastguard Worker static int stbi__pic_info(stbi__context *s, int *x, int *y, int *comp)
6710*89c4ff92SAndroid Build Coastguard Worker {
6711*89c4ff92SAndroid Build Coastguard Worker int act_comp=0,num_packets=0,chained,dummy;
6712*89c4ff92SAndroid Build Coastguard Worker stbi__pic_packet packets[10];
6713*89c4ff92SAndroid Build Coastguard Worker
6714*89c4ff92SAndroid Build Coastguard Worker if (!x) x = &dummy;
6715*89c4ff92SAndroid Build Coastguard Worker if (!y) y = &dummy;
6716*89c4ff92SAndroid Build Coastguard Worker if (!comp) comp = &dummy;
6717*89c4ff92SAndroid Build Coastguard Worker
6718*89c4ff92SAndroid Build Coastguard Worker if (!stbi__pic_is4(s,"\x53\x80\xF6\x34")) {
6719*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
6720*89c4ff92SAndroid Build Coastguard Worker return 0;
6721*89c4ff92SAndroid Build Coastguard Worker }
6722*89c4ff92SAndroid Build Coastguard Worker
6723*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, 88);
6724*89c4ff92SAndroid Build Coastguard Worker
6725*89c4ff92SAndroid Build Coastguard Worker *x = stbi__get16be(s);
6726*89c4ff92SAndroid Build Coastguard Worker *y = stbi__get16be(s);
6727*89c4ff92SAndroid Build Coastguard Worker if (stbi__at_eof(s)) {
6728*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s);
6729*89c4ff92SAndroid Build Coastguard Worker return 0;
6730*89c4ff92SAndroid Build Coastguard Worker }
6731*89c4ff92SAndroid Build Coastguard Worker if ( (*x) != 0 && (1 << 28) / (*x) < (*y)) {
6732*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6733*89c4ff92SAndroid Build Coastguard Worker return 0;
6734*89c4ff92SAndroid Build Coastguard Worker }
6735*89c4ff92SAndroid Build Coastguard Worker
6736*89c4ff92SAndroid Build Coastguard Worker stbi__skip(s, 8);
6737*89c4ff92SAndroid Build Coastguard Worker
6738*89c4ff92SAndroid Build Coastguard Worker do {
6739*89c4ff92SAndroid Build Coastguard Worker stbi__pic_packet *packet;
6740*89c4ff92SAndroid Build Coastguard Worker
6741*89c4ff92SAndroid Build Coastguard Worker if (num_packets==sizeof(packets)/sizeof(packets[0]))
6742*89c4ff92SAndroid Build Coastguard Worker return 0;
6743*89c4ff92SAndroid Build Coastguard Worker
6744*89c4ff92SAndroid Build Coastguard Worker packet = &packets[num_packets++];
6745*89c4ff92SAndroid Build Coastguard Worker chained = stbi__get8(s);
6746*89c4ff92SAndroid Build Coastguard Worker packet->size = stbi__get8(s);
6747*89c4ff92SAndroid Build Coastguard Worker packet->type = stbi__get8(s);
6748*89c4ff92SAndroid Build Coastguard Worker packet->channel = stbi__get8(s);
6749*89c4ff92SAndroid Build Coastguard Worker act_comp |= packet->channel;
6750*89c4ff92SAndroid Build Coastguard Worker
6751*89c4ff92SAndroid Build Coastguard Worker if (stbi__at_eof(s)) {
6752*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6753*89c4ff92SAndroid Build Coastguard Worker return 0;
6754*89c4ff92SAndroid Build Coastguard Worker }
6755*89c4ff92SAndroid Build Coastguard Worker if (packet->size != 8) {
6756*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6757*89c4ff92SAndroid Build Coastguard Worker return 0;
6758*89c4ff92SAndroid Build Coastguard Worker }
6759*89c4ff92SAndroid Build Coastguard Worker } while (chained);
6760*89c4ff92SAndroid Build Coastguard Worker
6761*89c4ff92SAndroid Build Coastguard Worker *comp = (act_comp & 0x10 ? 4 : 3);
6762*89c4ff92SAndroid Build Coastguard Worker
6763*89c4ff92SAndroid Build Coastguard Worker return 1;
6764*89c4ff92SAndroid Build Coastguard Worker }
6765*89c4ff92SAndroid Build Coastguard Worker #endif
6766*89c4ff92SAndroid Build Coastguard Worker
6767*89c4ff92SAndroid Build Coastguard Worker // *************************************************************************************************
6768*89c4ff92SAndroid Build Coastguard Worker // Portable Gray Map and Portable Pixel Map loader
6769*89c4ff92SAndroid Build Coastguard Worker // by Ken Miller
6770*89c4ff92SAndroid Build Coastguard Worker //
6771*89c4ff92SAndroid Build Coastguard Worker // PGM: http://netpbm.sourceforge.net/doc/pgm.html
6772*89c4ff92SAndroid Build Coastguard Worker // PPM: http://netpbm.sourceforge.net/doc/ppm.html
6773*89c4ff92SAndroid Build Coastguard Worker //
6774*89c4ff92SAndroid Build Coastguard Worker // Known limitations:
6775*89c4ff92SAndroid Build Coastguard Worker // Does not support comments in the header section
6776*89c4ff92SAndroid Build Coastguard Worker // Does not support ASCII image data (formats P2 and P3)
6777*89c4ff92SAndroid Build Coastguard Worker // Does not support 16-bit-per-channel
6778*89c4ff92SAndroid Build Coastguard Worker
6779*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PNM
6780*89c4ff92SAndroid Build Coastguard Worker
stbi__pnm_test(stbi__context * s)6781*89c4ff92SAndroid Build Coastguard Worker static int stbi__pnm_test(stbi__context *s)
6782*89c4ff92SAndroid Build Coastguard Worker {
6783*89c4ff92SAndroid Build Coastguard Worker char p, t;
6784*89c4ff92SAndroid Build Coastguard Worker p = (char) stbi__get8(s);
6785*89c4ff92SAndroid Build Coastguard Worker t = (char) stbi__get8(s);
6786*89c4ff92SAndroid Build Coastguard Worker if (p != 'P' || (t != '5' && t != '6')) {
6787*89c4ff92SAndroid Build Coastguard Worker stbi__rewind( s );
6788*89c4ff92SAndroid Build Coastguard Worker return 0;
6789*89c4ff92SAndroid Build Coastguard Worker }
6790*89c4ff92SAndroid Build Coastguard Worker return 1;
6791*89c4ff92SAndroid Build Coastguard Worker }
6792*89c4ff92SAndroid Build Coastguard Worker
stbi__pnm_load(stbi__context * s,int * x,int * y,int * comp,int req_comp,stbi__result_info * ri)6793*89c4ff92SAndroid Build Coastguard Worker static void *stbi__pnm_load(stbi__context *s, int *x, int *y, int *comp, int req_comp, stbi__result_info *ri)
6794*89c4ff92SAndroid Build Coastguard Worker {
6795*89c4ff92SAndroid Build Coastguard Worker stbi_uc *out;
6796*89c4ff92SAndroid Build Coastguard Worker STBI_NOTUSED(ri);
6797*89c4ff92SAndroid Build Coastguard Worker
6798*89c4ff92SAndroid Build Coastguard Worker if (!stbi__pnm_info(s, (int *)&s->img_x, (int *)&s->img_y, (int *)&s->img_n))
6799*89c4ff92SAndroid Build Coastguard Worker return 0;
6800*89c4ff92SAndroid Build Coastguard Worker
6801*89c4ff92SAndroid Build Coastguard Worker *x = s->img_x;
6802*89c4ff92SAndroid Build Coastguard Worker *y = s->img_y;
6803*89c4ff92SAndroid Build Coastguard Worker if (comp) *comp = s->img_n;
6804*89c4ff92SAndroid Build Coastguard Worker
6805*89c4ff92SAndroid Build Coastguard Worker if (!stbi__mad3sizes_valid(s->img_n, s->img_x, s->img_y, 0))
6806*89c4ff92SAndroid Build Coastguard Worker return stbi__errpuc("too large", "PNM too large");
6807*89c4ff92SAndroid Build Coastguard Worker
6808*89c4ff92SAndroid Build Coastguard Worker out = (stbi_uc *) stbi__malloc_mad3(s->img_n, s->img_x, s->img_y, 0);
6809*89c4ff92SAndroid Build Coastguard Worker if (!out) return stbi__errpuc("outofmem", "Out of memory");
6810*89c4ff92SAndroid Build Coastguard Worker stbi__getn(s, out, s->img_n * s->img_x * s->img_y);
6811*89c4ff92SAndroid Build Coastguard Worker
6812*89c4ff92SAndroid Build Coastguard Worker if (req_comp && req_comp != s->img_n) {
6813*89c4ff92SAndroid Build Coastguard Worker out = stbi__convert_format(out, s->img_n, req_comp, s->img_x, s->img_y);
6814*89c4ff92SAndroid Build Coastguard Worker if (out == NULL) return out; // stbi__convert_format frees input on failure
6815*89c4ff92SAndroid Build Coastguard Worker }
6816*89c4ff92SAndroid Build Coastguard Worker return out;
6817*89c4ff92SAndroid Build Coastguard Worker }
6818*89c4ff92SAndroid Build Coastguard Worker
stbi__pnm_isspace(char c)6819*89c4ff92SAndroid Build Coastguard Worker static int stbi__pnm_isspace(char c)
6820*89c4ff92SAndroid Build Coastguard Worker {
6821*89c4ff92SAndroid Build Coastguard Worker return c == ' ' || c == '\t' || c == '\n' || c == '\v' || c == '\f' || c == '\r';
6822*89c4ff92SAndroid Build Coastguard Worker }
6823*89c4ff92SAndroid Build Coastguard Worker
stbi__pnm_skip_whitespace(stbi__context * s,char * c)6824*89c4ff92SAndroid Build Coastguard Worker static void stbi__pnm_skip_whitespace(stbi__context *s, char *c)
6825*89c4ff92SAndroid Build Coastguard Worker {
6826*89c4ff92SAndroid Build Coastguard Worker for (;;) {
6827*89c4ff92SAndroid Build Coastguard Worker while (!stbi__at_eof(s) && stbi__pnm_isspace(*c))
6828*89c4ff92SAndroid Build Coastguard Worker *c = (char) stbi__get8(s);
6829*89c4ff92SAndroid Build Coastguard Worker
6830*89c4ff92SAndroid Build Coastguard Worker if (stbi__at_eof(s) || *c != '#')
6831*89c4ff92SAndroid Build Coastguard Worker break;
6832*89c4ff92SAndroid Build Coastguard Worker
6833*89c4ff92SAndroid Build Coastguard Worker while (!stbi__at_eof(s) && *c != '\n' && *c != '\r' )
6834*89c4ff92SAndroid Build Coastguard Worker *c = (char) stbi__get8(s);
6835*89c4ff92SAndroid Build Coastguard Worker }
6836*89c4ff92SAndroid Build Coastguard Worker }
6837*89c4ff92SAndroid Build Coastguard Worker
stbi__pnm_isdigit(char c)6838*89c4ff92SAndroid Build Coastguard Worker static int stbi__pnm_isdigit(char c)
6839*89c4ff92SAndroid Build Coastguard Worker {
6840*89c4ff92SAndroid Build Coastguard Worker return c >= '0' && c <= '9';
6841*89c4ff92SAndroid Build Coastguard Worker }
6842*89c4ff92SAndroid Build Coastguard Worker
stbi__pnm_getinteger(stbi__context * s,char * c)6843*89c4ff92SAndroid Build Coastguard Worker static int stbi__pnm_getinteger(stbi__context *s, char *c)
6844*89c4ff92SAndroid Build Coastguard Worker {
6845*89c4ff92SAndroid Build Coastguard Worker int value = 0;
6846*89c4ff92SAndroid Build Coastguard Worker
6847*89c4ff92SAndroid Build Coastguard Worker while (!stbi__at_eof(s) && stbi__pnm_isdigit(*c)) {
6848*89c4ff92SAndroid Build Coastguard Worker value = value*10 + (*c - '0');
6849*89c4ff92SAndroid Build Coastguard Worker *c = (char) stbi__get8(s);
6850*89c4ff92SAndroid Build Coastguard Worker }
6851*89c4ff92SAndroid Build Coastguard Worker
6852*89c4ff92SAndroid Build Coastguard Worker return value;
6853*89c4ff92SAndroid Build Coastguard Worker }
6854*89c4ff92SAndroid Build Coastguard Worker
stbi__pnm_info(stbi__context * s,int * x,int * y,int * comp)6855*89c4ff92SAndroid Build Coastguard Worker static int stbi__pnm_info(stbi__context *s, int *x, int *y, int *comp)
6856*89c4ff92SAndroid Build Coastguard Worker {
6857*89c4ff92SAndroid Build Coastguard Worker int maxv, dummy;
6858*89c4ff92SAndroid Build Coastguard Worker char c, p, t;
6859*89c4ff92SAndroid Build Coastguard Worker
6860*89c4ff92SAndroid Build Coastguard Worker if (!x) x = &dummy;
6861*89c4ff92SAndroid Build Coastguard Worker if (!y) y = &dummy;
6862*89c4ff92SAndroid Build Coastguard Worker if (!comp) comp = &dummy;
6863*89c4ff92SAndroid Build Coastguard Worker
6864*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
6865*89c4ff92SAndroid Build Coastguard Worker
6866*89c4ff92SAndroid Build Coastguard Worker // Get identifier
6867*89c4ff92SAndroid Build Coastguard Worker p = (char) stbi__get8(s);
6868*89c4ff92SAndroid Build Coastguard Worker t = (char) stbi__get8(s);
6869*89c4ff92SAndroid Build Coastguard Worker if (p != 'P' || (t != '5' && t != '6')) {
6870*89c4ff92SAndroid Build Coastguard Worker stbi__rewind(s);
6871*89c4ff92SAndroid Build Coastguard Worker return 0;
6872*89c4ff92SAndroid Build Coastguard Worker }
6873*89c4ff92SAndroid Build Coastguard Worker
6874*89c4ff92SAndroid Build Coastguard Worker *comp = (t == '6') ? 3 : 1; // '5' is 1-component .pgm; '6' is 3-component .ppm
6875*89c4ff92SAndroid Build Coastguard Worker
6876*89c4ff92SAndroid Build Coastguard Worker c = (char) stbi__get8(s);
6877*89c4ff92SAndroid Build Coastguard Worker stbi__pnm_skip_whitespace(s, &c);
6878*89c4ff92SAndroid Build Coastguard Worker
6879*89c4ff92SAndroid Build Coastguard Worker *x = stbi__pnm_getinteger(s, &c); // read width
6880*89c4ff92SAndroid Build Coastguard Worker stbi__pnm_skip_whitespace(s, &c);
6881*89c4ff92SAndroid Build Coastguard Worker
6882*89c4ff92SAndroid Build Coastguard Worker *y = stbi__pnm_getinteger(s, &c); // read height
6883*89c4ff92SAndroid Build Coastguard Worker stbi__pnm_skip_whitespace(s, &c);
6884*89c4ff92SAndroid Build Coastguard Worker
6885*89c4ff92SAndroid Build Coastguard Worker maxv = stbi__pnm_getinteger(s, &c); // read max value
6886*89c4ff92SAndroid Build Coastguard Worker
6887*89c4ff92SAndroid Build Coastguard Worker if (maxv > 255)
6888*89c4ff92SAndroid Build Coastguard Worker return stbi__err("max value > 255", "PPM image not 8-bit");
6889*89c4ff92SAndroid Build Coastguard Worker else
6890*89c4ff92SAndroid Build Coastguard Worker return 1;
6891*89c4ff92SAndroid Build Coastguard Worker }
6892*89c4ff92SAndroid Build Coastguard Worker #endif
6893*89c4ff92SAndroid Build Coastguard Worker
stbi__info_main(stbi__context * s,int * x,int * y,int * comp)6894*89c4ff92SAndroid Build Coastguard Worker static int stbi__info_main(stbi__context *s, int *x, int *y, int *comp)
6895*89c4ff92SAndroid Build Coastguard Worker {
6896*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_JPEG
6897*89c4ff92SAndroid Build Coastguard Worker if (stbi__jpeg_info(s, x, y, comp)) return 1;
6898*89c4ff92SAndroid Build Coastguard Worker #endif
6899*89c4ff92SAndroid Build Coastguard Worker
6900*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PNG
6901*89c4ff92SAndroid Build Coastguard Worker if (stbi__png_info(s, x, y, comp)) return 1;
6902*89c4ff92SAndroid Build Coastguard Worker #endif
6903*89c4ff92SAndroid Build Coastguard Worker
6904*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_GIF
6905*89c4ff92SAndroid Build Coastguard Worker if (stbi__gif_info(s, x, y, comp)) return 1;
6906*89c4ff92SAndroid Build Coastguard Worker #endif
6907*89c4ff92SAndroid Build Coastguard Worker
6908*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_BMP
6909*89c4ff92SAndroid Build Coastguard Worker if (stbi__bmp_info(s, x, y, comp)) return 1;
6910*89c4ff92SAndroid Build Coastguard Worker #endif
6911*89c4ff92SAndroid Build Coastguard Worker
6912*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PSD
6913*89c4ff92SAndroid Build Coastguard Worker if (stbi__psd_info(s, x, y, comp)) return 1;
6914*89c4ff92SAndroid Build Coastguard Worker #endif
6915*89c4ff92SAndroid Build Coastguard Worker
6916*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PIC
6917*89c4ff92SAndroid Build Coastguard Worker if (stbi__pic_info(s, x, y, comp)) return 1;
6918*89c4ff92SAndroid Build Coastguard Worker #endif
6919*89c4ff92SAndroid Build Coastguard Worker
6920*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_PNM
6921*89c4ff92SAndroid Build Coastguard Worker if (stbi__pnm_info(s, x, y, comp)) return 1;
6922*89c4ff92SAndroid Build Coastguard Worker #endif
6923*89c4ff92SAndroid Build Coastguard Worker
6924*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_HDR
6925*89c4ff92SAndroid Build Coastguard Worker if (stbi__hdr_info(s, x, y, comp)) return 1;
6926*89c4ff92SAndroid Build Coastguard Worker #endif
6927*89c4ff92SAndroid Build Coastguard Worker
6928*89c4ff92SAndroid Build Coastguard Worker // test tga last because it's a crappy test!
6929*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_TGA
6930*89c4ff92SAndroid Build Coastguard Worker if (stbi__tga_info(s, x, y, comp))
6931*89c4ff92SAndroid Build Coastguard Worker return 1;
6932*89c4ff92SAndroid Build Coastguard Worker #endif
6933*89c4ff92SAndroid Build Coastguard Worker return stbi__err("unknown image type", "Image not of any known type, or corrupt");
6934*89c4ff92SAndroid Build Coastguard Worker }
6935*89c4ff92SAndroid Build Coastguard Worker
6936*89c4ff92SAndroid Build Coastguard Worker #ifndef STBI_NO_STDIO
stbi_info(char const * filename,int * x,int * y,int * comp)6937*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_info(char const *filename, int *x, int *y, int *comp)
6938*89c4ff92SAndroid Build Coastguard Worker {
6939*89c4ff92SAndroid Build Coastguard Worker FILE *f = stbi__fopen(filename, "rb");
6940*89c4ff92SAndroid Build Coastguard Worker int result;
6941*89c4ff92SAndroid Build Coastguard Worker if (!f) return stbi__err("can't fopen", "Unable to open file");
6942*89c4ff92SAndroid Build Coastguard Worker result = stbi_info_from_file(f, x, y, comp);
6943*89c4ff92SAndroid Build Coastguard Worker fclose(f);
6944*89c4ff92SAndroid Build Coastguard Worker return result;
6945*89c4ff92SAndroid Build Coastguard Worker }
6946*89c4ff92SAndroid Build Coastguard Worker
stbi_info_from_file(FILE * f,int * x,int * y,int * comp)6947*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_info_from_file(FILE *f, int *x, int *y, int *comp)
6948*89c4ff92SAndroid Build Coastguard Worker {
6949*89c4ff92SAndroid Build Coastguard Worker int r;
6950*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
6951*89c4ff92SAndroid Build Coastguard Worker long pos = ftell(f);
6952*89c4ff92SAndroid Build Coastguard Worker stbi__start_file(&s, f);
6953*89c4ff92SAndroid Build Coastguard Worker r = stbi__info_main(&s,x,y,comp);
6954*89c4ff92SAndroid Build Coastguard Worker fseek(f,pos,SEEK_SET);
6955*89c4ff92SAndroid Build Coastguard Worker return r;
6956*89c4ff92SAndroid Build Coastguard Worker }
6957*89c4ff92SAndroid Build Coastguard Worker #endif // !STBI_NO_STDIO
6958*89c4ff92SAndroid Build Coastguard Worker
stbi_info_from_memory(stbi_uc const * buffer,int len,int * x,int * y,int * comp)6959*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_info_from_memory(stbi_uc const *buffer, int len, int *x, int *y, int *comp)
6960*89c4ff92SAndroid Build Coastguard Worker {
6961*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
6962*89c4ff92SAndroid Build Coastguard Worker stbi__start_mem(&s,buffer,len);
6963*89c4ff92SAndroid Build Coastguard Worker return stbi__info_main(&s,x,y,comp);
6964*89c4ff92SAndroid Build Coastguard Worker }
6965*89c4ff92SAndroid Build Coastguard Worker
stbi_info_from_callbacks(stbi_io_callbacks const * c,void * user,int * x,int * y,int * comp)6966*89c4ff92SAndroid Build Coastguard Worker STBIDEF int stbi_info_from_callbacks(stbi_io_callbacks const *c, void *user, int *x, int *y, int *comp)
6967*89c4ff92SAndroid Build Coastguard Worker {
6968*89c4ff92SAndroid Build Coastguard Worker stbi__context s;
6969*89c4ff92SAndroid Build Coastguard Worker stbi__start_callbacks(&s, (stbi_io_callbacks *) c, user);
6970*89c4ff92SAndroid Build Coastguard Worker return stbi__info_main(&s,x,y,comp);
6971*89c4ff92SAndroid Build Coastguard Worker }
6972*89c4ff92SAndroid Build Coastguard Worker
6973*89c4ff92SAndroid Build Coastguard Worker #endif // STB_IMAGE_IMPLEMENTATION
6974*89c4ff92SAndroid Build Coastguard Worker
6975*89c4ff92SAndroid Build Coastguard Worker /*
6976*89c4ff92SAndroid Build Coastguard Worker revision history:
6977*89c4ff92SAndroid Build Coastguard Worker 2.16 (2017-07-23) all functions have 16-bit variants;
6978*89c4ff92SAndroid Build Coastguard Worker STBI_NO_STDIO works again;
6979*89c4ff92SAndroid Build Coastguard Worker compilation fixes;
6980*89c4ff92SAndroid Build Coastguard Worker fix rounding in unpremultiply;
6981*89c4ff92SAndroid Build Coastguard Worker optimize vertical flip;
6982*89c4ff92SAndroid Build Coastguard Worker disable raw_len validation;
6983*89c4ff92SAndroid Build Coastguard Worker documentation fixes
6984*89c4ff92SAndroid Build Coastguard Worker 2.15 (2017-03-18) fix png-1,2,4 bug; now all Imagenet JPGs decode;
6985*89c4ff92SAndroid Build Coastguard Worker warning fixes; disable run-time SSE detection on gcc;
6986*89c4ff92SAndroid Build Coastguard Worker uniform handling of optional "return" values;
6987*89c4ff92SAndroid Build Coastguard Worker thread-safe initialization of zlib tables
6988*89c4ff92SAndroid Build Coastguard Worker 2.14 (2017-03-03) remove deprecated STBI_JPEG_OLD; fixes for Imagenet JPGs
6989*89c4ff92SAndroid Build Coastguard Worker 2.13 (2016-11-29) add 16-bit API, only supported for PNG right now
6990*89c4ff92SAndroid Build Coastguard Worker 2.12 (2016-04-02) fix typo in 2.11 PSD fix that caused crashes
6991*89c4ff92SAndroid Build Coastguard Worker 2.11 (2016-04-02) allocate large structures on the stack
6992*89c4ff92SAndroid Build Coastguard Worker remove white matting for transparent PSD
6993*89c4ff92SAndroid Build Coastguard Worker fix reported channel count for PNG & BMP
6994*89c4ff92SAndroid Build Coastguard Worker re-enable SSE2 in non-gcc 64-bit
6995*89c4ff92SAndroid Build Coastguard Worker support RGB-formatted JPEG
6996*89c4ff92SAndroid Build Coastguard Worker read 16-bit PNGs (only as 8-bit)
6997*89c4ff92SAndroid Build Coastguard Worker 2.10 (2016-01-22) avoid warning introduced in 2.09 by STBI_REALLOC_SIZED
6998*89c4ff92SAndroid Build Coastguard Worker 2.09 (2016-01-16) allow comments in PNM files
6999*89c4ff92SAndroid Build Coastguard Worker 16-bit-per-pixel TGA (not bit-per-component)
7000*89c4ff92SAndroid Build Coastguard Worker info() for TGA could break due to .hdr handling
7001*89c4ff92SAndroid Build Coastguard Worker info() for BMP to shares code instead of sloppy parse
7002*89c4ff92SAndroid Build Coastguard Worker can use STBI_REALLOC_SIZED if allocator doesn't support realloc
7003*89c4ff92SAndroid Build Coastguard Worker code cleanup
7004*89c4ff92SAndroid Build Coastguard Worker 2.08 (2015-09-13) fix to 2.07 cleanup, reading RGB PSD as RGBA
7005*89c4ff92SAndroid Build Coastguard Worker 2.07 (2015-09-13) fix compiler warnings
7006*89c4ff92SAndroid Build Coastguard Worker partial animated GIF support
7007*89c4ff92SAndroid Build Coastguard Worker limited 16-bpc PSD support
7008*89c4ff92SAndroid Build Coastguard Worker #ifdef unused functions
7009*89c4ff92SAndroid Build Coastguard Worker bug with < 92 byte PIC,PNM,HDR,TGA
7010*89c4ff92SAndroid Build Coastguard Worker 2.06 (2015-04-19) fix bug where PSD returns wrong '*comp' value
7011*89c4ff92SAndroid Build Coastguard Worker 2.05 (2015-04-19) fix bug in progressive JPEG handling, fix warning
7012*89c4ff92SAndroid Build Coastguard Worker 2.04 (2015-04-15) try to re-enable SIMD on MinGW 64-bit
7013*89c4ff92SAndroid Build Coastguard Worker 2.03 (2015-04-12) extra corruption checking (mmozeiko)
7014*89c4ff92SAndroid Build Coastguard Worker stbi_set_flip_vertically_on_load (nguillemot)
7015*89c4ff92SAndroid Build Coastguard Worker fix NEON support; fix mingw support
7016*89c4ff92SAndroid Build Coastguard Worker 2.02 (2015-01-19) fix incorrect assert, fix warning
7017*89c4ff92SAndroid Build Coastguard Worker 2.01 (2015-01-17) fix various warnings; suppress SIMD on gcc 32-bit without -msse2
7018*89c4ff92SAndroid Build Coastguard Worker 2.00b (2014-12-25) fix STBI_MALLOC in progressive JPEG
7019*89c4ff92SAndroid Build Coastguard Worker 2.00 (2014-12-25) optimize JPG, including x86 SSE2 & NEON SIMD (ryg)
7020*89c4ff92SAndroid Build Coastguard Worker progressive JPEG (stb)
7021*89c4ff92SAndroid Build Coastguard Worker PGM/PPM support (Ken Miller)
7022*89c4ff92SAndroid Build Coastguard Worker STBI_MALLOC,STBI_REALLOC,STBI_FREE
7023*89c4ff92SAndroid Build Coastguard Worker GIF bugfix -- seemingly never worked
7024*89c4ff92SAndroid Build Coastguard Worker STBI_NO_*, STBI_ONLY_*
7025*89c4ff92SAndroid Build Coastguard Worker 1.48 (2014-12-14) fix incorrectly-named assert()
7026*89c4ff92SAndroid Build Coastguard Worker 1.47 (2014-12-14) 1/2/4-bit PNG support, both direct and paletted (Omar Cornut & stb)
7027*89c4ff92SAndroid Build Coastguard Worker optimize PNG (ryg)
7028*89c4ff92SAndroid Build Coastguard Worker fix bug in interlaced PNG with user-specified channel count (stb)
7029*89c4ff92SAndroid Build Coastguard Worker 1.46 (2014-08-26)
7030*89c4ff92SAndroid Build Coastguard Worker fix broken tRNS chunk (colorkey-style transparency) in non-paletted PNG
7031*89c4ff92SAndroid Build Coastguard Worker 1.45 (2014-08-16)
7032*89c4ff92SAndroid Build Coastguard Worker fix MSVC-ARM internal compiler error by wrapping malloc
7033*89c4ff92SAndroid Build Coastguard Worker 1.44 (2014-08-07)
7034*89c4ff92SAndroid Build Coastguard Worker various warning fixes from Ronny Chevalier
7035*89c4ff92SAndroid Build Coastguard Worker 1.43 (2014-07-15)
7036*89c4ff92SAndroid Build Coastguard Worker fix MSVC-only compiler problem in code changed in 1.42
7037*89c4ff92SAndroid Build Coastguard Worker 1.42 (2014-07-09)
7038*89c4ff92SAndroid Build Coastguard Worker don't define _CRT_SECURE_NO_WARNINGS (affects user code)
7039*89c4ff92SAndroid Build Coastguard Worker fixes to stbi__cleanup_jpeg path
7040*89c4ff92SAndroid Build Coastguard Worker added STBI_ASSERT to avoid requiring assert.h
7041*89c4ff92SAndroid Build Coastguard Worker 1.41 (2014-06-25)
7042*89c4ff92SAndroid Build Coastguard Worker fix search&replace from 1.36 that messed up comments/error messages
7043*89c4ff92SAndroid Build Coastguard Worker 1.40 (2014-06-22)
7044*89c4ff92SAndroid Build Coastguard Worker fix gcc struct-initialization warning
7045*89c4ff92SAndroid Build Coastguard Worker 1.39 (2014-06-15)
7046*89c4ff92SAndroid Build Coastguard Worker fix to TGA optimization when req_comp != number of components in TGA;
7047*89c4ff92SAndroid Build Coastguard Worker fix to GIF loading because BMP wasn't rewinding (whoops, no GIFs in my test suite)
7048*89c4ff92SAndroid Build Coastguard Worker add support for BMP version 5 (more ignored fields)
7049*89c4ff92SAndroid Build Coastguard Worker 1.38 (2014-06-06)
7050*89c4ff92SAndroid Build Coastguard Worker suppress MSVC warnings on integer casts truncating values
7051*89c4ff92SAndroid Build Coastguard Worker fix accidental rename of 'skip' field of I/O
7052*89c4ff92SAndroid Build Coastguard Worker 1.37 (2014-06-04)
7053*89c4ff92SAndroid Build Coastguard Worker remove duplicate typedef
7054*89c4ff92SAndroid Build Coastguard Worker 1.36 (2014-06-03)
7055*89c4ff92SAndroid Build Coastguard Worker convert to header file single-file library
7056*89c4ff92SAndroid Build Coastguard Worker if de-iphone isn't set, load iphone images color-swapped instead of returning NULL
7057*89c4ff92SAndroid Build Coastguard Worker 1.35 (2014-05-27)
7058*89c4ff92SAndroid Build Coastguard Worker various warnings
7059*89c4ff92SAndroid Build Coastguard Worker fix broken STBI_SIMD path
7060*89c4ff92SAndroid Build Coastguard Worker fix bug where stbi_load_from_file no longer left file pointer in correct place
7061*89c4ff92SAndroid Build Coastguard Worker fix broken non-easy path for 32-bit BMP (possibly never used)
7062*89c4ff92SAndroid Build Coastguard Worker TGA optimization by Arseny Kapoulkine
7063*89c4ff92SAndroid Build Coastguard Worker 1.34 (unknown)
7064*89c4ff92SAndroid Build Coastguard Worker use STBI_NOTUSED in stbi__resample_row_generic(), fix one more leak in tga failure case
7065*89c4ff92SAndroid Build Coastguard Worker 1.33 (2011-07-14)
7066*89c4ff92SAndroid Build Coastguard Worker make stbi_is_hdr work in STBI_NO_HDR (as specified), minor compiler-friendly improvements
7067*89c4ff92SAndroid Build Coastguard Worker 1.32 (2011-07-13)
7068*89c4ff92SAndroid Build Coastguard Worker support for "info" function for all supported filetypes (SpartanJ)
7069*89c4ff92SAndroid Build Coastguard Worker 1.31 (2011-06-20)
7070*89c4ff92SAndroid Build Coastguard Worker a few more leak fixes, bug in PNG handling (SpartanJ)
7071*89c4ff92SAndroid Build Coastguard Worker 1.30 (2011-06-11)
7072*89c4ff92SAndroid Build Coastguard Worker added ability to load files via callbacks to accomidate custom input streams (Ben Wenger)
7073*89c4ff92SAndroid Build Coastguard Worker removed deprecated format-specific test/load functions
7074*89c4ff92SAndroid Build Coastguard Worker removed support for installable file formats (stbi_loader) -- would have been broken for IO callbacks anyway
7075*89c4ff92SAndroid Build Coastguard Worker error cases in bmp and tga give messages and don't leak (Raymond Barbiero, grisha)
7076*89c4ff92SAndroid Build Coastguard Worker fix inefficiency in decoding 32-bit BMP (David Woo)
7077*89c4ff92SAndroid Build Coastguard Worker 1.29 (2010-08-16)
7078*89c4ff92SAndroid Build Coastguard Worker various warning fixes from Aurelien Pocheville
7079*89c4ff92SAndroid Build Coastguard Worker 1.28 (2010-08-01)
7080*89c4ff92SAndroid Build Coastguard Worker fix bug in GIF palette transparency (SpartanJ)
7081*89c4ff92SAndroid Build Coastguard Worker 1.27 (2010-08-01)
7082*89c4ff92SAndroid Build Coastguard Worker cast-to-stbi_uc to fix warnings
7083*89c4ff92SAndroid Build Coastguard Worker 1.26 (2010-07-24)
7084*89c4ff92SAndroid Build Coastguard Worker fix bug in file buffering for PNG reported by SpartanJ
7085*89c4ff92SAndroid Build Coastguard Worker 1.25 (2010-07-17)
7086*89c4ff92SAndroid Build Coastguard Worker refix trans_data warning (Won Chun)
7087*89c4ff92SAndroid Build Coastguard Worker 1.24 (2010-07-12)
7088*89c4ff92SAndroid Build Coastguard Worker perf improvements reading from files on platforms with lock-heavy fgetc()
7089*89c4ff92SAndroid Build Coastguard Worker minor perf improvements for jpeg
7090*89c4ff92SAndroid Build Coastguard Worker deprecated type-specific functions so we'll get feedback if they're needed
7091*89c4ff92SAndroid Build Coastguard Worker attempt to fix trans_data warning (Won Chun)
7092*89c4ff92SAndroid Build Coastguard Worker 1.23 fixed bug in iPhone support
7093*89c4ff92SAndroid Build Coastguard Worker 1.22 (2010-07-10)
7094*89c4ff92SAndroid Build Coastguard Worker removed image *writing* support
7095*89c4ff92SAndroid Build Coastguard Worker stbi_info support from Jetro Lauha
7096*89c4ff92SAndroid Build Coastguard Worker GIF support from Jean-Marc Lienher
7097*89c4ff92SAndroid Build Coastguard Worker iPhone PNG-extensions from James Brown
7098*89c4ff92SAndroid Build Coastguard Worker warning-fixes from Nicolas Schulz and Janez Zemva (i.stbi__err. Janez (U+017D)emva)
7099*89c4ff92SAndroid Build Coastguard Worker 1.21 fix use of 'stbi_uc' in header (reported by jon blow)
7100*89c4ff92SAndroid Build Coastguard Worker 1.20 added support for Softimage PIC, by Tom Seddon
7101*89c4ff92SAndroid Build Coastguard Worker 1.19 bug in interlaced PNG corruption check (found by ryg)
7102*89c4ff92SAndroid Build Coastguard Worker 1.18 (2008-08-02)
7103*89c4ff92SAndroid Build Coastguard Worker fix a threading bug (local mutable static)
7104*89c4ff92SAndroid Build Coastguard Worker 1.17 support interlaced PNG
7105*89c4ff92SAndroid Build Coastguard Worker 1.16 major bugfix - stbi__convert_format converted one too many pixels
7106*89c4ff92SAndroid Build Coastguard Worker 1.15 initialize some fields for thread safety
7107*89c4ff92SAndroid Build Coastguard Worker 1.14 fix threadsafe conversion bug
7108*89c4ff92SAndroid Build Coastguard Worker header-file-only version (#define STBI_HEADER_FILE_ONLY before including)
7109*89c4ff92SAndroid Build Coastguard Worker 1.13 threadsafe
7110*89c4ff92SAndroid Build Coastguard Worker 1.12 const qualifiers in the API
7111*89c4ff92SAndroid Build Coastguard Worker 1.11 Support installable IDCT, colorspace conversion routines
7112*89c4ff92SAndroid Build Coastguard Worker 1.10 Fixes for 64-bit (don't use "unsigned long")
7113*89c4ff92SAndroid Build Coastguard Worker optimized upsampling by Fabian "ryg" Giesen
7114*89c4ff92SAndroid Build Coastguard Worker 1.09 Fix format-conversion for PSD code (bad global variables!)
7115*89c4ff92SAndroid Build Coastguard Worker 1.08 Thatcher Ulrich's PSD code integrated by Nicolas Schulz
7116*89c4ff92SAndroid Build Coastguard Worker 1.07 attempt to fix C++ warning/errors again
7117*89c4ff92SAndroid Build Coastguard Worker 1.06 attempt to fix C++ warning/errors again
7118*89c4ff92SAndroid Build Coastguard Worker 1.05 fix TGA loading to return correct *comp and use good luminance calc
7119*89c4ff92SAndroid Build Coastguard Worker 1.04 default float alpha is 1, not 255; use 'void *' for stbi_image_free
7120*89c4ff92SAndroid Build Coastguard Worker 1.03 bugfixes to STBI_NO_STDIO, STBI_NO_HDR
7121*89c4ff92SAndroid Build Coastguard Worker 1.02 support for (subset of) HDR files, float interface for preferred access to them
7122*89c4ff92SAndroid Build Coastguard Worker 1.01 fix bug: possible bug in handling right-side up bmps... not sure
7123*89c4ff92SAndroid Build Coastguard Worker fix bug: the stbi__bmp_load() and stbi__tga_load() functions didn't work at all
7124*89c4ff92SAndroid Build Coastguard Worker 1.00 interface to zlib that skips zlib header
7125*89c4ff92SAndroid Build Coastguard Worker 0.99 correct handling of alpha in palette
7126*89c4ff92SAndroid Build Coastguard Worker 0.98 TGA loader by lonesock; dynamically add loaders (untested)
7127*89c4ff92SAndroid Build Coastguard Worker 0.97 jpeg errors on too large a file; also catch another malloc failure
7128*89c4ff92SAndroid Build Coastguard Worker 0.96 fix detection of invalid v value - particleman@mollyrocket forum
7129*89c4ff92SAndroid Build Coastguard Worker 0.95 during header scan, seek to markers in case of padding
7130*89c4ff92SAndroid Build Coastguard Worker 0.94 STBI_NO_STDIO to disable stdio usage; rename all #defines the same
7131*89c4ff92SAndroid Build Coastguard Worker 0.93 handle jpegtran output; verbose errors
7132*89c4ff92SAndroid Build Coastguard Worker 0.92 read 4,8,16,24,32-bit BMP files of several formats
7133*89c4ff92SAndroid Build Coastguard Worker 0.91 output 24-bit Windows 3.0 BMP files
7134*89c4ff92SAndroid Build Coastguard Worker 0.90 fix a few more warnings; bump version number to approach 1.0
7135*89c4ff92SAndroid Build Coastguard Worker 0.61 bugfixes due to Marc LeBlanc, Christopher Lloyd
7136*89c4ff92SAndroid Build Coastguard Worker 0.60 fix compiling as c++
7137*89c4ff92SAndroid Build Coastguard Worker 0.59 fix warnings: merge Dave Moore's -Wall fixes
7138*89c4ff92SAndroid Build Coastguard Worker 0.58 fix bug: zlib uncompressed mode len/nlen was wrong endian
7139*89c4ff92SAndroid Build Coastguard Worker 0.57 fix bug: jpg last huffman symbol before marker was >9 bits but less than 16 available
7140*89c4ff92SAndroid Build Coastguard Worker 0.56 fix bug: zlib uncompressed mode len vs. nlen
7141*89c4ff92SAndroid Build Coastguard Worker 0.55 fix bug: restart_interval not initialized to 0
7142*89c4ff92SAndroid Build Coastguard Worker 0.54 allow NULL for 'int *comp'
7143*89c4ff92SAndroid Build Coastguard Worker 0.53 fix bug in png 3->4; speedup png decoding
7144*89c4ff92SAndroid Build Coastguard Worker 0.52 png handles req_comp=3,4 directly; minor cleanup; jpeg comments
7145*89c4ff92SAndroid Build Coastguard Worker 0.51 obey req_comp requests, 1-component jpegs return as 1-component,
7146*89c4ff92SAndroid Build Coastguard Worker on 'test' only check type, not whether we support this variant
7147*89c4ff92SAndroid Build Coastguard Worker 0.50 (2006-11-19)
7148*89c4ff92SAndroid Build Coastguard Worker first released version
7149*89c4ff92SAndroid Build Coastguard Worker */
7150*89c4ff92SAndroid Build Coastguard Worker
7151*89c4ff92SAndroid Build Coastguard Worker
7152*89c4ff92SAndroid Build Coastguard Worker /*
7153*89c4ff92SAndroid Build Coastguard Worker ------------------------------------------------------------------------------
7154*89c4ff92SAndroid Build Coastguard Worker MIT License
7155*89c4ff92SAndroid Build Coastguard Worker Copyright (c) 2017 Sean Barrett
7156*89c4ff92SAndroid Build Coastguard Worker Permission is hereby granted, free of charge, to any person obtaining a copy of
7157*89c4ff92SAndroid Build Coastguard Worker this software and associated documentation files (the "Software"), to deal in
7158*89c4ff92SAndroid Build Coastguard Worker the Software without restriction, including without limitation the rights to
7159*89c4ff92SAndroid Build Coastguard Worker use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
7160*89c4ff92SAndroid Build Coastguard Worker of the Software, and to permit persons to whom the Software is furnished to do
7161*89c4ff92SAndroid Build Coastguard Worker so, subject to the following conditions:
7162*89c4ff92SAndroid Build Coastguard Worker The above copyright notice and this permission notice shall be included in all
7163*89c4ff92SAndroid Build Coastguard Worker copies or substantial portions of the Software.
7164*89c4ff92SAndroid Build Coastguard Worker THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
7165*89c4ff92SAndroid Build Coastguard Worker IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
7166*89c4ff92SAndroid Build Coastguard Worker FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
7167*89c4ff92SAndroid Build Coastguard Worker AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
7168*89c4ff92SAndroid Build Coastguard Worker LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
7169*89c4ff92SAndroid Build Coastguard Worker OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
7170*89c4ff92SAndroid Build Coastguard Worker SOFTWARE.
7171*89c4ff92SAndroid Build Coastguard Worker ------------------------------------------------------------------------------
7172*89c4ff92SAndroid Build Coastguard Worker */
7173