xref: /aosp_15_r20/external/armnn/third-party/stb/stb_image.h (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
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