tiny_gltf 0.1.1 → 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/.gitmodules +3 -0
- data/Gemfile.lock +3 -3
- data/bin/setup +2 -0
- data/ext/tiny_gltf/rb_tiny_gltf.h +18 -7
- data/ext/tiny_gltf/rb_tiny_gltf_accessor.cpp +4 -12
- data/ext/tiny_gltf/rb_tiny_gltf_animation.cpp +6 -5
- data/ext/tiny_gltf/rb_tiny_gltf_animation_channel.cpp +6 -5
- data/ext/tiny_gltf/rb_tiny_gltf_animation_sampler.cpp +6 -5
- data/ext/tiny_gltf/rb_tiny_gltf_asset.cpp +4 -3
- data/ext/tiny_gltf/rb_tiny_gltf_buffer.cpp +37 -8
- data/ext/tiny_gltf/rb_tiny_gltf_buffer_view.cpp +4 -3
- data/ext/tiny_gltf/rb_tiny_gltf_camera.cpp +4 -3
- data/ext/tiny_gltf/rb_tiny_gltf_extension_map.cpp +2 -2
- data/ext/tiny_gltf/rb_tiny_gltf_image.cpp +72 -12
- data/ext/tiny_gltf/rb_tiny_gltf_init.c +66 -1
- data/ext/tiny_gltf/rb_tiny_gltf_light.cpp +2 -1
- data/ext/tiny_gltf/rb_tiny_gltf_material.cpp +6 -5
- data/ext/tiny_gltf/rb_tiny_gltf_mesh.cpp +5 -17
- data/ext/tiny_gltf/rb_tiny_gltf_model.cpp +10 -10
- data/ext/tiny_gltf/rb_tiny_gltf_node.cpp +8 -7
- data/ext/tiny_gltf/rb_tiny_gltf_parameter_map.cpp +2 -2
- data/ext/tiny_gltf/rb_tiny_gltf_primitive.cpp +8 -7
- data/ext/tiny_gltf/rb_tiny_gltf_sampler.cpp +3 -2
- data/ext/tiny_gltf/rb_tiny_gltf_scene.cpp +7 -6
- data/ext/tiny_gltf/rb_tiny_gltf_skin.cpp +6 -5
- data/ext/tiny_gltf/rb_tiny_gltf_texture.cpp +7 -6
- data/ext/tiny_gltf/rb_tiny_gltf_types.cpp +7 -7
- data/ext/tiny_gltf/rb_tiny_gltf_value.cpp +5 -4
- data/ext/tiny_gltf/stb_image.h +1890 -869
- data/ext/tiny_gltf/stb_image_write.h +1241 -1451
- data/ext/tiny_gltf/tiny_gltf.h +3671 -1082
- data/lib/tiny_gltf.rb +385 -126
- data/lib/tiny_gltf/version.rb +1 -1
- metadata +4 -4
@@ -1,6 +1,6 @@
|
|
1
|
-
/* stb_image_write - v1.
|
2
|
-
|
3
|
-
|
1
|
+
/* stb_image_write - v1.11 - public domain - http://nothings.org/stb/stb_image_write.h
|
2
|
+
writes out PNG/BMP/TGA/JPEG/HDR images to C stdio - Sean Barrett 2010-2015
|
3
|
+
no warranty implied; use at your own risk
|
4
4
|
|
5
5
|
Before #including,
|
6
6
|
|
@@ -10,16 +10,14 @@ to C stdio - Sean Barrett 2010-2015 no warranty implied; use at your own risk
|
|
10
10
|
|
11
11
|
Will probably not work correctly with strict-aliasing optimizations.
|
12
12
|
|
13
|
-
If using a modern Microsoft Compiler, non-safe versions of CRT calls may
|
14
|
-
|
15
|
-
#including,
|
13
|
+
If using a modern Microsoft Compiler, non-safe versions of CRT calls may cause
|
14
|
+
compilation warnings or even errors. To avoid this, also before #including,
|
16
15
|
|
17
16
|
#define STBI_MSC_SECURE_CRT
|
18
17
|
|
19
18
|
ABOUT:
|
20
19
|
|
21
|
-
This header file is a library for writing images to C stdio
|
22
|
-
adapted to write to memory or a general streaming interface; let me know.
|
20
|
+
This header file is a library for writing images to C stdio or a callback.
|
23
21
|
|
24
22
|
The PNG output is not optimal; it is 20-50% larger than the file
|
25
23
|
written by a decent optimizing implementation; though providing a custom
|
@@ -33,48 +31,48 @@ BUILDING:
|
|
33
31
|
You can #define STBIW_MALLOC(), STBIW_REALLOC(), and STBIW_FREE() to replace
|
34
32
|
malloc,realloc,free.
|
35
33
|
You can #define STBIW_MEMMOVE() to replace memmove()
|
36
|
-
You can #define STBIW_ZLIB_COMPRESS to use a custom zlib-style compress
|
37
|
-
|
38
|
-
|
39
|
-
|
40
|
-
|
41
|
-
|
34
|
+
You can #define STBIW_ZLIB_COMPRESS to use a custom zlib-style compress function
|
35
|
+
for PNG compression (instead of the builtin one), it must have the following signature:
|
36
|
+
unsigned char * my_compress(unsigned char *data, int data_len, int *out_len, int quality);
|
37
|
+
The returned data will be freed with STBIW_FREE() (free() by default),
|
38
|
+
so it must be heap allocated with STBIW_MALLOC() (malloc() by default),
|
39
|
+
|
40
|
+
UNICODE:
|
41
|
+
|
42
|
+
If compiling for Windows and you wish to use Unicode filenames, compile
|
43
|
+
with
|
44
|
+
#define STBIW_WINDOWS_UTF8
|
45
|
+
and pass utf8-encoded filenames. Call stbiw_convert_wchar_to_utf8 to convert
|
46
|
+
Windows wchar_t filenames to utf8.
|
42
47
|
|
43
48
|
USAGE:
|
44
49
|
|
45
50
|
There are five functions, one for each image file format:
|
46
51
|
|
47
|
-
int stbi_write_png(char const *filename, int w, int h, int comp, const void
|
48
|
-
|
49
|
-
h, int comp, const void *data);
|
50
|
-
int h, int comp, const void *data
|
51
|
-
w, int h, int comp, const
|
52
|
-
|
53
|
-
|
54
|
-
|
55
|
-
|
56
|
-
|
57
|
-
|
58
|
-
|
59
|
-
|
60
|
-
|
61
|
-
int
|
62
|
-
|
63
|
-
stbi_write_bmp_to_func(stbi_write_func *func, void *context, int w, int h, int
|
64
|
-
comp, const void *data); int stbi_write_tga_to_func(stbi_write_func *func, void
|
65
|
-
*context, int w, int h, int comp, const void *data); int
|
66
|
-
stbi_write_hdr_to_func(stbi_write_func *func, void *context, int w, int h, int
|
67
|
-
comp, const float *data); int stbi_write_jpg_to_func(stbi_write_func *func, void
|
68
|
-
*context, int x, int y, int comp, const void *data, int quality);
|
52
|
+
int stbi_write_png(char const *filename, int w, int h, int comp, const void *data, int stride_in_bytes);
|
53
|
+
int stbi_write_bmp(char const *filename, int w, int h, int comp, const void *data);
|
54
|
+
int stbi_write_tga(char const *filename, int w, int h, int comp, const void *data);
|
55
|
+
int stbi_write_jpg(char const *filename, int w, int h, int comp, const void *data, int quality);
|
56
|
+
int stbi_write_hdr(char const *filename, int w, int h, int comp, const float *data);
|
57
|
+
|
58
|
+
void stbi_flip_vertically_on_write(int flag); // flag is non-zero to flip data vertically
|
59
|
+
|
60
|
+
There are also five equivalent functions that use an arbitrary write function. You are
|
61
|
+
expected to open/close your file-equivalent before and after calling these:
|
62
|
+
|
63
|
+
int stbi_write_png_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data, int stride_in_bytes);
|
64
|
+
int stbi_write_bmp_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data);
|
65
|
+
int stbi_write_tga_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data);
|
66
|
+
int stbi_write_hdr_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const float *data);
|
67
|
+
int stbi_write_jpg_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data, int quality);
|
69
68
|
|
70
69
|
where the callback is:
|
71
70
|
void stbi_write_func(void *context, void *data, int size);
|
72
71
|
|
73
72
|
You can configure it with these global variables:
|
74
|
-
int stbi_write_tga_with_rle; // defaults to true; set to 0 to
|
75
|
-
|
76
|
-
|
77
|
-
to -1; set to 0..5 to force a filter mode
|
73
|
+
int stbi_write_tga_with_rle; // defaults to true; set to 0 to disable RLE
|
74
|
+
int stbi_write_png_compression_level; // defaults to 8; set to higher for more compression
|
75
|
+
int stbi_write_force_png_filter; // defaults to -1; set to 0..5 to force a filter mode
|
78
76
|
|
79
77
|
|
80
78
|
You can define STBI_WRITE_NO_STDIO to disable the file variant of these
|
@@ -113,7 +111,6 @@ to -1; set to 0..5 to force a filter mode
|
|
113
111
|
TGA supports RLE or non-RLE compressed data. To use non-RLE-compressed
|
114
112
|
data, set the global variable 'stbi_write_tga_with_rle' to 0.
|
115
113
|
|
116
|
-
|
117
114
|
JPEG does ignore alpha channels in input data; quality is between 1 and 100.
|
118
115
|
Higher quality looks better but results in a bigger image.
|
119
116
|
JPEG baseline (no JPEG progressive).
|
@@ -121,7 +118,7 @@ to -1; set to 0..5 to force a filter mode
|
|
121
118
|
CREDITS:
|
122
119
|
|
123
120
|
|
124
|
-
Sean Barrett - PNG/BMP/TGA
|
121
|
+
Sean Barrett - PNG/BMP/TGA
|
125
122
|
Baldur Karlsson - HDR
|
126
123
|
Jean-Sebastien Guay - TGA monochrome
|
127
124
|
Tim Kelsey - misc enhancements
|
@@ -158,16 +155,17 @@ LICENSE
|
|
158
155
|
#ifndef INCLUDE_STB_IMAGE_WRITE_H
|
159
156
|
#define INCLUDE_STB_IMAGE_WRITE_H
|
160
157
|
|
161
|
-
|
162
|
-
|
158
|
+
#include <stdlib.h>
|
159
|
+
|
160
|
+
// if STB_IMAGE_WRITE_STATIC causes problems, try defining STBIWDEF to 'inline' or 'static inline'
|
163
161
|
#ifndef STBIWDEF
|
164
162
|
#ifdef STB_IMAGE_WRITE_STATIC
|
165
|
-
#define STBIWDEF
|
163
|
+
#define STBIWDEF static
|
166
164
|
#else
|
167
165
|
#ifdef __cplusplus
|
168
|
-
#define STBIWDEF
|
166
|
+
#define STBIWDEF extern "C"
|
169
167
|
#else
|
170
|
-
#define STBIWDEF
|
168
|
+
#define STBIWDEF extern
|
171
169
|
#endif
|
172
170
|
#endif
|
173
171
|
#endif
|
@@ -179,400 +177,422 @@ extern int stbi_write_force_png_filter;
|
|
179
177
|
#endif
|
180
178
|
|
181
179
|
#ifndef STBI_WRITE_NO_STDIO
|
182
|
-
STBIWDEF int stbi_write_png(char const *filename, int w, int h, int comp,
|
183
|
-
|
184
|
-
STBIWDEF int
|
185
|
-
|
186
|
-
STBIWDEF int
|
187
|
-
|
188
|
-
|
189
|
-
|
190
|
-
|
191
|
-
const void *data, int quality);
|
180
|
+
STBIWDEF int stbi_write_png(char const *filename, int w, int h, int comp, const void *data, int stride_in_bytes);
|
181
|
+
STBIWDEF int stbi_write_bmp(char const *filename, int w, int h, int comp, const void *data);
|
182
|
+
STBIWDEF int stbi_write_tga(char const *filename, int w, int h, int comp, const void *data);
|
183
|
+
STBIWDEF int stbi_write_hdr(char const *filename, int w, int h, int comp, const float *data);
|
184
|
+
STBIWDEF int stbi_write_jpg(char const *filename, int x, int y, int comp, const void *data, int quality);
|
185
|
+
|
186
|
+
#ifdef STBI_WINDOWS_UTF8
|
187
|
+
STBIWDEF int stbiw_convert_wchar_to_utf8(char *buffer, size_t bufferlen, const wchar_t* input);
|
188
|
+
#endif
|
192
189
|
#endif
|
193
190
|
|
194
191
|
typedef void stbi_write_func(void *context, void *data, int size);
|
195
192
|
|
196
|
-
STBIWDEF int stbi_write_png_to_func(stbi_write_func *func, void *context, int w,
|
197
|
-
|
198
|
-
|
199
|
-
STBIWDEF int
|
200
|
-
|
201
|
-
STBIWDEF int stbi_write_tga_to_func(stbi_write_func *func, void *context, int w,
|
202
|
-
int h, int comp, const void *data);
|
203
|
-
STBIWDEF int stbi_write_hdr_to_func(stbi_write_func *func, void *context, int w,
|
204
|
-
int h, int comp, const float *data);
|
205
|
-
STBIWDEF int stbi_write_jpg_to_func(stbi_write_func *func, void *context, int x,
|
206
|
-
int y, int comp, const void *data,
|
207
|
-
int quality);
|
193
|
+
STBIWDEF int stbi_write_png_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data, int stride_in_bytes);
|
194
|
+
STBIWDEF int stbi_write_bmp_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data);
|
195
|
+
STBIWDEF int stbi_write_tga_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data);
|
196
|
+
STBIWDEF int stbi_write_hdr_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const float *data);
|
197
|
+
STBIWDEF int stbi_write_jpg_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data, int quality);
|
208
198
|
|
209
199
|
STBIWDEF void stbi_flip_vertically_on_write(int flip_boolean);
|
210
200
|
|
211
|
-
#endif
|
201
|
+
#endif//INCLUDE_STB_IMAGE_WRITE_H
|
212
202
|
|
213
203
|
#ifdef STB_IMAGE_WRITE_IMPLEMENTATION
|
214
204
|
|
215
205
|
#ifdef _WIN32
|
216
|
-
#ifndef _CRT_SECURE_NO_WARNINGS
|
217
|
-
#define _CRT_SECURE_NO_WARNINGS
|
218
|
-
#endif
|
219
|
-
#ifndef _CRT_NONSTDC_NO_DEPRECATE
|
220
|
-
#define _CRT_NONSTDC_NO_DEPRECATE
|
221
|
-
#endif
|
206
|
+
#ifndef _CRT_SECURE_NO_WARNINGS
|
207
|
+
#define _CRT_SECURE_NO_WARNINGS
|
208
|
+
#endif
|
209
|
+
#ifndef _CRT_NONSTDC_NO_DEPRECATE
|
210
|
+
#define _CRT_NONSTDC_NO_DEPRECATE
|
211
|
+
#endif
|
222
212
|
#endif
|
223
213
|
|
224
214
|
#ifndef STBI_WRITE_NO_STDIO
|
225
215
|
#include <stdio.h>
|
226
|
-
#endif
|
216
|
+
#endif // STBI_WRITE_NO_STDIO
|
227
217
|
|
228
|
-
#include <math.h>
|
229
218
|
#include <stdarg.h>
|
230
219
|
#include <stdlib.h>
|
231
220
|
#include <string.h>
|
221
|
+
#include <math.h>
|
232
222
|
|
233
|
-
#if defined(STBIW_MALLOC) && defined(STBIW_FREE) &&
|
234
|
-
(defined(STBIW_REALLOC) || defined(STBIW_REALLOC_SIZED))
|
223
|
+
#if defined(STBIW_MALLOC) && defined(STBIW_FREE) && (defined(STBIW_REALLOC) || defined(STBIW_REALLOC_SIZED))
|
235
224
|
// ok
|
236
|
-
#elif !defined(STBIW_MALLOC) && !defined(STBIW_FREE) &&
|
237
|
-
!defined(STBIW_REALLOC) && !defined(STBIW_REALLOC_SIZED)
|
225
|
+
#elif !defined(STBIW_MALLOC) && !defined(STBIW_FREE) && !defined(STBIW_REALLOC) && !defined(STBIW_REALLOC_SIZED)
|
238
226
|
// ok
|
239
227
|
#else
|
240
|
-
#error
|
241
|
-
"Must define all or none of STBIW_MALLOC, STBIW_FREE, and STBIW_REALLOC (or STBIW_REALLOC_SIZED)."
|
228
|
+
#error "Must define all or none of STBIW_MALLOC, STBIW_FREE, and STBIW_REALLOC (or STBIW_REALLOC_SIZED)."
|
242
229
|
#endif
|
243
230
|
|
244
231
|
#ifndef STBIW_MALLOC
|
245
|
-
#define STBIW_MALLOC(sz)
|
246
|
-
#define STBIW_REALLOC(p,
|
247
|
-
#define STBIW_FREE(p)
|
232
|
+
#define STBIW_MALLOC(sz) malloc(sz)
|
233
|
+
#define STBIW_REALLOC(p,newsz) realloc(p,newsz)
|
234
|
+
#define STBIW_FREE(p) free(p)
|
248
235
|
#endif
|
249
236
|
|
250
237
|
#ifndef STBIW_REALLOC_SIZED
|
251
|
-
#define STBIW_REALLOC_SIZED(p,
|
238
|
+
#define STBIW_REALLOC_SIZED(p,oldsz,newsz) STBIW_REALLOC(p,newsz)
|
252
239
|
#endif
|
253
240
|
|
241
|
+
|
254
242
|
#ifndef STBIW_MEMMOVE
|
255
|
-
#define STBIW_MEMMOVE(a,
|
243
|
+
#define STBIW_MEMMOVE(a,b,sz) memmove(a,b,sz)
|
256
244
|
#endif
|
257
245
|
|
246
|
+
|
258
247
|
#ifndef STBIW_ASSERT
|
259
248
|
#include <assert.h>
|
260
249
|
#define STBIW_ASSERT(x) assert(x)
|
261
250
|
#endif
|
262
251
|
|
263
|
-
#define STBIW_UCHAR(x) (unsigned char)((x)&0xff)
|
252
|
+
#define STBIW_UCHAR(x) (unsigned char) ((x) & 0xff)
|
264
253
|
|
265
254
|
#ifdef STB_IMAGE_WRITE_STATIC
|
266
|
-
static int stbi__flip_vertically_on_write
|
255
|
+
static int stbi__flip_vertically_on_write=0;
|
267
256
|
static int stbi_write_png_compression_level = 8;
|
268
257
|
static int stbi_write_tga_with_rle = 1;
|
269
258
|
static int stbi_write_force_png_filter = -1;
|
270
259
|
#else
|
271
260
|
int stbi_write_png_compression_level = 8;
|
272
|
-
int stbi__flip_vertically_on_write
|
261
|
+
int stbi__flip_vertically_on_write=0;
|
273
262
|
int stbi_write_tga_with_rle = 1;
|
274
263
|
int stbi_write_force_png_filter = -1;
|
275
264
|
#endif
|
276
265
|
|
277
|
-
STBIWDEF void stbi_flip_vertically_on_write(int flag)
|
278
|
-
|
266
|
+
STBIWDEF void stbi_flip_vertically_on_write(int flag)
|
267
|
+
{
|
268
|
+
stbi__flip_vertically_on_write = flag;
|
279
269
|
}
|
280
270
|
|
281
|
-
typedef struct
|
282
|
-
|
283
|
-
|
271
|
+
typedef struct
|
272
|
+
{
|
273
|
+
stbi_write_func *func;
|
274
|
+
void *context;
|
284
275
|
} stbi__write_context;
|
285
276
|
|
286
277
|
// initialize a callback-based context
|
287
|
-
static void stbi__start_write_callbacks(stbi__write_context *s,
|
288
|
-
|
289
|
-
|
290
|
-
|
278
|
+
static void stbi__start_write_callbacks(stbi__write_context *s, stbi_write_func *c, void *context)
|
279
|
+
{
|
280
|
+
s->func = c;
|
281
|
+
s->context = context;
|
291
282
|
}
|
292
283
|
|
293
284
|
#ifndef STBI_WRITE_NO_STDIO
|
294
285
|
|
295
|
-
static void stbi__stdio_write(void *context, void *data, int size)
|
296
|
-
|
286
|
+
static void stbi__stdio_write(void *context, void *data, int size)
|
287
|
+
{
|
288
|
+
fwrite(data,1,size,(FILE*) context);
|
297
289
|
}
|
298
290
|
|
299
|
-
|
300
|
-
|
301
|
-
|
302
|
-
#
|
303
|
-
|
291
|
+
#if defined(_MSC_VER) && defined(STBI_WINDOWS_UTF8)
|
292
|
+
#ifdef __cplusplus
|
293
|
+
#define STBIW_EXTERN extern "C"
|
294
|
+
#else
|
295
|
+
#define STBIW_EXTERN extern
|
296
|
+
#endif
|
297
|
+
STBIW_EXTERN __declspec(dllimport) int __stdcall MultiByteToWideChar(unsigned int cp, unsigned long flags, const char *str, int cbmb, wchar_t *widestr, int cchwide);
|
298
|
+
STBIW_EXTERN __declspec(dllimport) int __stdcall WideCharToMultiByte(unsigned int cp, unsigned long flags, const wchar_t *widestr, int cchwide, char *str, int cbmb, const char *defchar, int *used_default);
|
299
|
+
|
300
|
+
STBIWDEF int stbiw_convert_wchar_to_utf8(char *buffer, size_t bufferlen, const wchar_t* input)
|
301
|
+
{
|
302
|
+
return WideCharToMultiByte(65001 /* UTF8 */, 0, input, -1, buffer, bufferlen, NULL, NULL);
|
303
|
+
}
|
304
|
+
#endif
|
305
|
+
|
306
|
+
static FILE *stbiw__fopen(char const *filename, char const *mode)
|
307
|
+
{
|
308
|
+
FILE *f;
|
309
|
+
#if defined(_MSC_VER) && defined(STBI_WINDOWS_UTF8)
|
310
|
+
wchar_t wMode[64];
|
311
|
+
wchar_t wFilename[1024];
|
312
|
+
if (0 == MultiByteToWideChar(65001 /* UTF8 */, 0, filename, -1, wFilename, sizeof(wFilename)))
|
313
|
+
return 0;
|
314
|
+
|
315
|
+
if (0 == MultiByteToWideChar(65001 /* UTF8 */, 0, mode, -1, wMode, sizeof(wMode)))
|
316
|
+
return 0;
|
317
|
+
|
318
|
+
#if _MSC_VER >= 1400
|
319
|
+
if (0 != _wfopen_s(&f, wFilename, wMode))
|
320
|
+
f = 0;
|
321
|
+
#else
|
322
|
+
f = _wfopen(wFilename, wMode);
|
323
|
+
#endif
|
324
|
+
|
325
|
+
#elif defined(_MSC_VER) && _MSC_VER >= 1400
|
326
|
+
if (0 != fopen_s(&f, filename, mode))
|
327
|
+
f=0;
|
304
328
|
#else
|
305
|
-
|
329
|
+
f = fopen(filename, mode);
|
306
330
|
#endif
|
307
|
-
|
308
|
-
return f != NULL;
|
331
|
+
return f;
|
309
332
|
}
|
310
333
|
|
311
|
-
static
|
312
|
-
|
334
|
+
static int stbi__start_write_file(stbi__write_context *s, const char *filename)
|
335
|
+
{
|
336
|
+
FILE *f = stbiw__fopen(filename, "wb");
|
337
|
+
stbi__start_write_callbacks(s, stbi__stdio_write, (void *) f);
|
338
|
+
return f != NULL;
|
313
339
|
}
|
314
340
|
|
315
|
-
|
341
|
+
static void stbi__end_write_file(stbi__write_context *s)
|
342
|
+
{
|
343
|
+
fclose((FILE *)s->context);
|
344
|
+
}
|
345
|
+
|
346
|
+
#endif // !STBI_WRITE_NO_STDIO
|
316
347
|
|
317
348
|
typedef unsigned int stbiw_uint32;
|
318
|
-
typedef int stb_image_write_test[sizeof(stbiw_uint32)
|
319
|
-
|
320
|
-
static void stbiw__writefv(stbi__write_context *s, const char *fmt, va_list v)
|
321
|
-
|
322
|
-
|
323
|
-
|
324
|
-
|
325
|
-
|
326
|
-
|
327
|
-
|
328
|
-
|
329
|
-
|
330
|
-
|
331
|
-
|
332
|
-
|
333
|
-
|
334
|
-
|
335
|
-
|
336
|
-
|
337
|
-
|
338
|
-
|
339
|
-
|
340
|
-
|
341
|
-
|
342
|
-
|
343
|
-
|
344
|
-
|
345
|
-
s->func(s->context, b, 4);
|
346
|
-
break;
|
349
|
+
typedef int stb_image_write_test[sizeof(stbiw_uint32)==4 ? 1 : -1];
|
350
|
+
|
351
|
+
static void stbiw__writefv(stbi__write_context *s, const char *fmt, va_list v)
|
352
|
+
{
|
353
|
+
while (*fmt) {
|
354
|
+
switch (*fmt++) {
|
355
|
+
case ' ': break;
|
356
|
+
case '1': { unsigned char x = STBIW_UCHAR(va_arg(v, int));
|
357
|
+
s->func(s->context,&x,1);
|
358
|
+
break; }
|
359
|
+
case '2': { int x = va_arg(v,int);
|
360
|
+
unsigned char b[2];
|
361
|
+
b[0] = STBIW_UCHAR(x);
|
362
|
+
b[1] = STBIW_UCHAR(x>>8);
|
363
|
+
s->func(s->context,b,2);
|
364
|
+
break; }
|
365
|
+
case '4': { stbiw_uint32 x = va_arg(v,int);
|
366
|
+
unsigned char b[4];
|
367
|
+
b[0]=STBIW_UCHAR(x);
|
368
|
+
b[1]=STBIW_UCHAR(x>>8);
|
369
|
+
b[2]=STBIW_UCHAR(x>>16);
|
370
|
+
b[3]=STBIW_UCHAR(x>>24);
|
371
|
+
s->func(s->context,b,4);
|
372
|
+
break; }
|
373
|
+
default:
|
374
|
+
STBIW_ASSERT(0);
|
375
|
+
return;
|
347
376
|
}
|
348
|
-
|
349
|
-
STBIW_ASSERT(0);
|
350
|
-
return;
|
351
|
-
}
|
352
|
-
}
|
377
|
+
}
|
353
378
|
}
|
354
379
|
|
355
|
-
static void stbiw__writef(stbi__write_context *s, const char *fmt, ...)
|
356
|
-
|
357
|
-
|
358
|
-
|
359
|
-
|
380
|
+
static void stbiw__writef(stbi__write_context *s, const char *fmt, ...)
|
381
|
+
{
|
382
|
+
va_list v;
|
383
|
+
va_start(v, fmt);
|
384
|
+
stbiw__writefv(s, fmt, v);
|
385
|
+
va_end(v);
|
360
386
|
}
|
361
387
|
|
362
|
-
static void stbiw__putc(stbi__write_context *s, unsigned char c)
|
363
|
-
|
388
|
+
static void stbiw__putc(stbi__write_context *s, unsigned char c)
|
389
|
+
{
|
390
|
+
s->func(s->context, &c, 1);
|
364
391
|
}
|
365
392
|
|
366
|
-
static void stbiw__write3(stbi__write_context *s, unsigned char a,
|
367
|
-
|
368
|
-
|
369
|
-
|
370
|
-
|
393
|
+
static void stbiw__write3(stbi__write_context *s, unsigned char a, unsigned char b, unsigned char c)
|
394
|
+
{
|
395
|
+
unsigned char arr[3];
|
396
|
+
arr[0] = a, arr[1] = b, arr[2] = c;
|
397
|
+
s->func(s->context, arr, 3);
|
371
398
|
}
|
372
399
|
|
373
|
-
static void stbiw__write_pixel(stbi__write_context *s, int rgb_dir, int comp,
|
374
|
-
|
375
|
-
|
376
|
-
|
377
|
-
|
378
|
-
|
379
|
-
|
380
|
-
|
381
|
-
|
382
|
-
|
383
|
-
|
384
|
-
|
385
|
-
|
386
|
-
|
387
|
-
|
388
|
-
|
389
|
-
|
390
|
-
|
391
|
-
|
392
|
-
|
393
|
-
|
394
|
-
|
395
|
-
|
396
|
-
|
397
|
-
|
398
|
-
|
399
|
-
|
400
|
-
|
401
|
-
|
402
|
-
|
400
|
+
static void stbiw__write_pixel(stbi__write_context *s, int rgb_dir, int comp, int write_alpha, int expand_mono, unsigned char *d)
|
401
|
+
{
|
402
|
+
unsigned char bg[3] = { 255, 0, 255}, px[3];
|
403
|
+
int k;
|
404
|
+
|
405
|
+
if (write_alpha < 0)
|
406
|
+
s->func(s->context, &d[comp - 1], 1);
|
407
|
+
|
408
|
+
switch (comp) {
|
409
|
+
case 2: // 2 pixels = mono + alpha, alpha is written separately, so same as 1-channel case
|
410
|
+
case 1:
|
411
|
+
if (expand_mono)
|
412
|
+
stbiw__write3(s, d[0], d[0], d[0]); // monochrome bmp
|
413
|
+
else
|
414
|
+
s->func(s->context, d, 1); // monochrome TGA
|
415
|
+
break;
|
416
|
+
case 4:
|
417
|
+
if (!write_alpha) {
|
418
|
+
// composite against pink background
|
419
|
+
for (k = 0; k < 3; ++k)
|
420
|
+
px[k] = bg[k] + ((d[k] - bg[k]) * d[3]) / 255;
|
421
|
+
stbiw__write3(s, px[1 - rgb_dir], px[1], px[1 + rgb_dir]);
|
422
|
+
break;
|
423
|
+
}
|
424
|
+
/* FALLTHROUGH */
|
425
|
+
case 3:
|
426
|
+
stbiw__write3(s, d[1 - rgb_dir], d[1], d[1 + rgb_dir]);
|
427
|
+
break;
|
428
|
+
}
|
429
|
+
if (write_alpha > 0)
|
430
|
+
s->func(s->context, &d[comp - 1], 1);
|
403
431
|
}
|
404
432
|
|
405
|
-
static void stbiw__write_pixels(stbi__write_context *s, int rgb_dir, int vdir,
|
406
|
-
|
407
|
-
|
408
|
-
|
409
|
-
|
410
|
-
|
411
|
-
|
412
|
-
|
413
|
-
|
414
|
-
|
415
|
-
|
416
|
-
|
417
|
-
|
418
|
-
|
419
|
-
|
420
|
-
|
421
|
-
|
422
|
-
|
423
|
-
|
424
|
-
|
425
|
-
|
426
|
-
|
427
|
-
|
433
|
+
static void stbiw__write_pixels(stbi__write_context *s, int rgb_dir, int vdir, int x, int y, int comp, void *data, int write_alpha, int scanline_pad, int expand_mono)
|
434
|
+
{
|
435
|
+
stbiw_uint32 zero = 0;
|
436
|
+
int i,j, j_end;
|
437
|
+
|
438
|
+
if (y <= 0)
|
439
|
+
return;
|
440
|
+
|
441
|
+
if (stbi__flip_vertically_on_write)
|
442
|
+
vdir *= -1;
|
443
|
+
|
444
|
+
if (vdir < 0)
|
445
|
+
j_end = -1, j = y-1;
|
446
|
+
else
|
447
|
+
j_end = y, j = 0;
|
448
|
+
|
449
|
+
for (; j != j_end; j += vdir) {
|
450
|
+
for (i=0; i < x; ++i) {
|
451
|
+
unsigned char *d = (unsigned char *) data + (j*x+i)*comp;
|
452
|
+
stbiw__write_pixel(s, rgb_dir, comp, write_alpha, expand_mono, d);
|
453
|
+
}
|
454
|
+
s->func(s->context, &zero, scanline_pad);
|
455
|
+
}
|
428
456
|
}
|
429
457
|
|
430
|
-
static int stbiw__outfile(stbi__write_context *s, int rgb_dir, int vdir, int x,
|
431
|
-
|
432
|
-
|
433
|
-
|
434
|
-
|
435
|
-
|
436
|
-
|
437
|
-
|
438
|
-
|
439
|
-
|
440
|
-
|
441
|
-
|
442
|
-
return 1;
|
443
|
-
}
|
458
|
+
static int stbiw__outfile(stbi__write_context *s, int rgb_dir, int vdir, int x, int y, int comp, int expand_mono, void *data, int alpha, int pad, const char *fmt, ...)
|
459
|
+
{
|
460
|
+
if (y < 0 || x < 0) {
|
461
|
+
return 0;
|
462
|
+
} else {
|
463
|
+
va_list v;
|
464
|
+
va_start(v, fmt);
|
465
|
+
stbiw__writefv(s, fmt, v);
|
466
|
+
va_end(v);
|
467
|
+
stbiw__write_pixels(s,rgb_dir,vdir,x,y,comp,data,alpha,pad, expand_mono);
|
468
|
+
return 1;
|
469
|
+
}
|
444
470
|
}
|
445
471
|
|
446
|
-
static int stbi_write_bmp_core(stbi__write_context *s, int x, int y, int comp,
|
447
|
-
|
448
|
-
|
449
|
-
|
450
|
-
|
451
|
-
|
452
|
-
|
453
|
-
14 + 40, // file header
|
454
|
-
40, x, y, 1, 24, 0, 0, 0, 0, 0, 0); // bitmap header
|
472
|
+
static int stbi_write_bmp_core(stbi__write_context *s, int x, int y, int comp, const void *data)
|
473
|
+
{
|
474
|
+
int pad = (-x*3) & 3;
|
475
|
+
return stbiw__outfile(s,-1,-1,x,y,comp,1,(void *) data,0,pad,
|
476
|
+
"11 4 22 4" "4 44 22 444444",
|
477
|
+
'B', 'M', 14+40+(x*3+pad)*y, 0,0, 14+40, // file header
|
478
|
+
40, x,y, 1,24, 0,0,0,0,0,0); // bitmap header
|
455
479
|
}
|
456
480
|
|
457
|
-
STBIWDEF int stbi_write_bmp_to_func(stbi_write_func *func, void *context, int x,
|
458
|
-
|
459
|
-
|
460
|
-
|
461
|
-
|
481
|
+
STBIWDEF int stbi_write_bmp_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data)
|
482
|
+
{
|
483
|
+
stbi__write_context s;
|
484
|
+
stbi__start_write_callbacks(&s, func, context);
|
485
|
+
return stbi_write_bmp_core(&s, x, y, comp, data);
|
462
486
|
}
|
463
487
|
|
464
488
|
#ifndef STBI_WRITE_NO_STDIO
|
465
|
-
STBIWDEF int stbi_write_bmp(char const *filename, int x, int y, int comp,
|
466
|
-
|
467
|
-
|
468
|
-
|
469
|
-
|
470
|
-
|
471
|
-
|
472
|
-
|
473
|
-
|
489
|
+
STBIWDEF int stbi_write_bmp(char const *filename, int x, int y, int comp, const void *data)
|
490
|
+
{
|
491
|
+
stbi__write_context s;
|
492
|
+
if (stbi__start_write_file(&s,filename)) {
|
493
|
+
int r = stbi_write_bmp_core(&s, x, y, comp, data);
|
494
|
+
stbi__end_write_file(&s);
|
495
|
+
return r;
|
496
|
+
} else
|
497
|
+
return 0;
|
474
498
|
}
|
475
|
-
#endif
|
476
|
-
|
477
|
-
static int stbi_write_tga_core(stbi__write_context *s, int x, int y, int comp,
|
478
|
-
|
479
|
-
|
480
|
-
|
481
|
-
|
482
|
-
|
483
|
-
|
484
|
-
|
485
|
-
|
486
|
-
|
487
|
-
|
488
|
-
|
489
|
-
|
490
|
-
|
491
|
-
|
492
|
-
|
493
|
-
|
494
|
-
|
495
|
-
|
496
|
-
|
497
|
-
|
498
|
-
|
499
|
-
|
500
|
-
|
501
|
-
|
502
|
-
|
503
|
-
|
504
|
-
j
|
505
|
-
|
506
|
-
|
507
|
-
|
508
|
-
|
509
|
-
|
510
|
-
|
511
|
-
|
512
|
-
|
513
|
-
|
514
|
-
|
515
|
-
|
516
|
-
|
517
|
-
|
518
|
-
|
519
|
-
|
520
|
-
|
521
|
-
|
522
|
-
|
523
|
-
|
524
|
-
|
525
|
-
|
526
|
-
|
527
|
-
|
528
|
-
|
529
|
-
|
499
|
+
#endif //!STBI_WRITE_NO_STDIO
|
500
|
+
|
501
|
+
static int stbi_write_tga_core(stbi__write_context *s, int x, int y, int comp, void *data)
|
502
|
+
{
|
503
|
+
int has_alpha = (comp == 2 || comp == 4);
|
504
|
+
int colorbytes = has_alpha ? comp-1 : comp;
|
505
|
+
int format = colorbytes < 2 ? 3 : 2; // 3 color channels (RGB/RGBA) = 2, 1 color channel (Y/YA) = 3
|
506
|
+
|
507
|
+
if (y < 0 || x < 0)
|
508
|
+
return 0;
|
509
|
+
|
510
|
+
if (!stbi_write_tga_with_rle) {
|
511
|
+
return stbiw__outfile(s, -1, -1, x, y, comp, 0, (void *) data, has_alpha, 0,
|
512
|
+
"111 221 2222 11", 0, 0, format, 0, 0, 0, 0, 0, x, y, (colorbytes + has_alpha) * 8, has_alpha * 8);
|
513
|
+
} else {
|
514
|
+
int i,j,k;
|
515
|
+
int jend, jdir;
|
516
|
+
|
517
|
+
stbiw__writef(s, "111 221 2222 11", 0,0,format+8, 0,0,0, 0,0,x,y, (colorbytes + has_alpha) * 8, has_alpha * 8);
|
518
|
+
|
519
|
+
if (stbi__flip_vertically_on_write) {
|
520
|
+
j = 0;
|
521
|
+
jend = y;
|
522
|
+
jdir = 1;
|
523
|
+
} else {
|
524
|
+
j = y-1;
|
525
|
+
jend = -1;
|
526
|
+
jdir = -1;
|
527
|
+
}
|
528
|
+
for (; j != jend; j += jdir) {
|
529
|
+
unsigned char *row = (unsigned char *) data + j * x * comp;
|
530
|
+
int len;
|
531
|
+
|
532
|
+
for (i = 0; i < x; i += len) {
|
533
|
+
unsigned char *begin = row + i * comp;
|
534
|
+
int diff = 1;
|
535
|
+
len = 1;
|
536
|
+
|
537
|
+
if (i < x - 1) {
|
538
|
+
++len;
|
539
|
+
diff = memcmp(begin, row + (i + 1) * comp, comp);
|
540
|
+
if (diff) {
|
541
|
+
const unsigned char *prev = begin;
|
542
|
+
for (k = i + 2; k < x && len < 128; ++k) {
|
543
|
+
if (memcmp(prev, row + k * comp, comp)) {
|
544
|
+
prev += comp;
|
545
|
+
++len;
|
546
|
+
} else {
|
547
|
+
--len;
|
548
|
+
break;
|
549
|
+
}
|
550
|
+
}
|
551
|
+
} else {
|
552
|
+
for (k = i + 2; k < x && len < 128; ++k) {
|
553
|
+
if (!memcmp(begin, row + k * comp, comp)) {
|
554
|
+
++len;
|
555
|
+
} else {
|
556
|
+
break;
|
557
|
+
}
|
558
|
+
}
|
559
|
+
}
|
530
560
|
}
|
531
|
-
|
532
|
-
|
533
|
-
|
534
|
-
|
535
|
-
|
536
|
-
|
537
|
-
|
561
|
+
|
562
|
+
if (diff) {
|
563
|
+
unsigned char header = STBIW_UCHAR(len - 1);
|
564
|
+
s->func(s->context, &header, 1);
|
565
|
+
for (k = 0; k < len; ++k) {
|
566
|
+
stbiw__write_pixel(s, -1, comp, has_alpha, 0, begin + k * comp);
|
567
|
+
}
|
568
|
+
} else {
|
569
|
+
unsigned char header = STBIW_UCHAR(len - 129);
|
570
|
+
s->func(s->context, &header, 1);
|
571
|
+
stbiw__write_pixel(s, -1, comp, has_alpha, 0, begin);
|
538
572
|
}
|
539
|
-
|
540
|
-
}
|
541
|
-
|
542
|
-
if (diff) {
|
543
|
-
unsigned char header = STBIW_UCHAR(len - 1);
|
544
|
-
s->func(s->context, &header, 1);
|
545
|
-
for (k = 0; k < len; ++k) {
|
546
|
-
stbiw__write_pixel(s, -1, comp, has_alpha, 0, begin + k * comp);
|
547
|
-
}
|
548
|
-
} else {
|
549
|
-
unsigned char header = STBIW_UCHAR(len - 129);
|
550
|
-
s->func(s->context, &header, 1);
|
551
|
-
stbiw__write_pixel(s, -1, comp, has_alpha, 0, begin);
|
552
|
-
}
|
573
|
+
}
|
553
574
|
}
|
554
|
-
|
555
|
-
|
556
|
-
return 1;
|
575
|
+
}
|
576
|
+
return 1;
|
557
577
|
}
|
558
578
|
|
559
|
-
STBIWDEF int stbi_write_tga_to_func(stbi_write_func *func, void *context, int x,
|
560
|
-
|
561
|
-
|
562
|
-
|
563
|
-
|
579
|
+
STBIWDEF int stbi_write_tga_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data)
|
580
|
+
{
|
581
|
+
stbi__write_context s;
|
582
|
+
stbi__start_write_callbacks(&s, func, context);
|
583
|
+
return stbi_write_tga_core(&s, x, y, comp, (void *) data);
|
564
584
|
}
|
565
585
|
|
566
586
|
#ifndef STBI_WRITE_NO_STDIO
|
567
|
-
STBIWDEF int stbi_write_tga(char const *filename, int x, int y, int comp,
|
568
|
-
|
569
|
-
|
570
|
-
|
571
|
-
|
572
|
-
|
573
|
-
|
574
|
-
|
575
|
-
|
587
|
+
STBIWDEF int stbi_write_tga(char const *filename, int x, int y, int comp, const void *data)
|
588
|
+
{
|
589
|
+
stbi__write_context s;
|
590
|
+
if (stbi__start_write_file(&s,filename)) {
|
591
|
+
int r = stbi_write_tga_core(&s, x, y, comp, (void *) data);
|
592
|
+
stbi__end_write_file(&s);
|
593
|
+
return r;
|
594
|
+
} else
|
595
|
+
return 0;
|
576
596
|
}
|
577
597
|
#endif
|
578
598
|
|
@@ -580,183 +600,176 @@ STBIWDEF int stbi_write_tga(char const *filename, int x, int y, int comp,
|
|
580
600
|
// Radiance RGBE HDR writer
|
581
601
|
// by Baldur Karlsson
|
582
602
|
|
583
|
-
#define stbiw__max(a, b)
|
603
|
+
#define stbiw__max(a, b) ((a) > (b) ? (a) : (b))
|
584
604
|
|
585
|
-
void stbiw__linear_to_rgbe(unsigned char *rgbe, float *linear)
|
586
|
-
|
587
|
-
|
605
|
+
static void stbiw__linear_to_rgbe(unsigned char *rgbe, float *linear)
|
606
|
+
{
|
607
|
+
int exponent;
|
608
|
+
float maxcomp = stbiw__max(linear[0], stbiw__max(linear[1], linear[2]));
|
588
609
|
|
589
|
-
|
590
|
-
|
591
|
-
|
592
|
-
|
610
|
+
if (maxcomp < 1e-32f) {
|
611
|
+
rgbe[0] = rgbe[1] = rgbe[2] = rgbe[3] = 0;
|
612
|
+
} else {
|
613
|
+
float normalize = (float) frexp(maxcomp, &exponent) * 256.0f/maxcomp;
|
593
614
|
|
594
|
-
|
595
|
-
|
596
|
-
|
597
|
-
|
598
|
-
|
615
|
+
rgbe[0] = (unsigned char)(linear[0] * normalize);
|
616
|
+
rgbe[1] = (unsigned char)(linear[1] * normalize);
|
617
|
+
rgbe[2] = (unsigned char)(linear[2] * normalize);
|
618
|
+
rgbe[3] = (unsigned char)(exponent + 128);
|
619
|
+
}
|
599
620
|
}
|
600
621
|
|
601
|
-
void stbiw__write_run_data(stbi__write_context *s, int length,
|
602
|
-
|
603
|
-
|
604
|
-
|
605
|
-
|
606
|
-
|
622
|
+
static void stbiw__write_run_data(stbi__write_context *s, int length, unsigned char databyte)
|
623
|
+
{
|
624
|
+
unsigned char lengthbyte = STBIW_UCHAR(length+128);
|
625
|
+
STBIW_ASSERT(length+128 <= 255);
|
626
|
+
s->func(s->context, &lengthbyte, 1);
|
627
|
+
s->func(s->context, &databyte, 1);
|
607
628
|
}
|
608
629
|
|
609
|
-
void stbiw__write_dump_data(stbi__write_context *s, int length,
|
610
|
-
|
611
|
-
|
612
|
-
|
613
|
-
|
614
|
-
|
615
|
-
s->func(s->context, &lengthbyte, 1);
|
616
|
-
s->func(s->context, data, length);
|
630
|
+
static void stbiw__write_dump_data(stbi__write_context *s, int length, unsigned char *data)
|
631
|
+
{
|
632
|
+
unsigned char lengthbyte = STBIW_UCHAR(length);
|
633
|
+
STBIW_ASSERT(length <= 128); // inconsistent with spec but consistent with official code
|
634
|
+
s->func(s->context, &lengthbyte, 1);
|
635
|
+
s->func(s->context, data, length);
|
617
636
|
}
|
618
637
|
|
619
|
-
void stbiw__write_hdr_scanline(stbi__write_context *s, int width, int ncomp,
|
620
|
-
|
621
|
-
|
622
|
-
|
623
|
-
|
624
|
-
|
625
|
-
|
626
|
-
|
627
|
-
|
628
|
-
|
629
|
-
|
630
|
-
|
631
|
-
|
632
|
-
|
633
|
-
|
634
|
-
|
635
|
-
|
636
|
-
|
637
|
-
|
638
|
-
|
639
|
-
|
640
|
-
|
641
|
-
|
638
|
+
static void stbiw__write_hdr_scanline(stbi__write_context *s, int width, int ncomp, unsigned char *scratch, float *scanline)
|
639
|
+
{
|
640
|
+
unsigned char scanlineheader[4] = { 2, 2, 0, 0 };
|
641
|
+
unsigned char rgbe[4];
|
642
|
+
float linear[3];
|
643
|
+
int x;
|
644
|
+
|
645
|
+
scanlineheader[2] = (width&0xff00)>>8;
|
646
|
+
scanlineheader[3] = (width&0x00ff);
|
647
|
+
|
648
|
+
/* skip RLE for images too small or large */
|
649
|
+
if (width < 8 || width >= 32768) {
|
650
|
+
for (x=0; x < width; x++) {
|
651
|
+
switch (ncomp) {
|
652
|
+
case 4: /* fallthrough */
|
653
|
+
case 3: linear[2] = scanline[x*ncomp + 2];
|
654
|
+
linear[1] = scanline[x*ncomp + 1];
|
655
|
+
linear[0] = scanline[x*ncomp + 0];
|
656
|
+
break;
|
657
|
+
default:
|
658
|
+
linear[0] = linear[1] = linear[2] = scanline[x*ncomp + 0];
|
659
|
+
break;
|
660
|
+
}
|
661
|
+
stbiw__linear_to_rgbe(rgbe, linear);
|
662
|
+
s->func(s->context, rgbe, 4);
|
642
663
|
}
|
643
|
-
|
644
|
-
|
645
|
-
|
646
|
-
|
647
|
-
|
648
|
-
|
649
|
-
|
650
|
-
|
651
|
-
|
652
|
-
|
653
|
-
|
654
|
-
|
655
|
-
|
656
|
-
|
657
|
-
|
658
|
-
|
659
|
-
|
664
|
+
} else {
|
665
|
+
int c,r;
|
666
|
+
/* encode into scratch buffer */
|
667
|
+
for (x=0; x < width; x++) {
|
668
|
+
switch(ncomp) {
|
669
|
+
case 4: /* fallthrough */
|
670
|
+
case 3: linear[2] = scanline[x*ncomp + 2];
|
671
|
+
linear[1] = scanline[x*ncomp + 1];
|
672
|
+
linear[0] = scanline[x*ncomp + 0];
|
673
|
+
break;
|
674
|
+
default:
|
675
|
+
linear[0] = linear[1] = linear[2] = scanline[x*ncomp + 0];
|
676
|
+
break;
|
677
|
+
}
|
678
|
+
stbiw__linear_to_rgbe(rgbe, linear);
|
679
|
+
scratch[x + width*0] = rgbe[0];
|
680
|
+
scratch[x + width*1] = rgbe[1];
|
681
|
+
scratch[x + width*2] = rgbe[2];
|
682
|
+
scratch[x + width*3] = rgbe[3];
|
660
683
|
}
|
661
|
-
|
662
|
-
|
663
|
-
|
664
|
-
|
665
|
-
|
666
|
-
|
667
|
-
|
668
|
-
|
669
|
-
|
670
|
-
|
671
|
-
|
672
|
-
|
673
|
-
|
674
|
-
|
675
|
-
|
676
|
-
|
677
|
-
|
678
|
-
|
679
|
-
|
680
|
-
|
681
|
-
|
682
|
-
|
683
|
-
|
684
|
-
|
685
|
-
|
686
|
-
|
687
|
-
|
688
|
-
|
689
|
-
|
690
|
-
|
691
|
-
|
692
|
-
|
693
|
-
|
694
|
-
|
695
|
-
|
696
|
-
|
697
|
-
|
698
|
-
|
699
|
-
|
700
|
-
x += len;
|
701
|
-
}
|
702
|
-
}
|
684
|
+
|
685
|
+
s->func(s->context, scanlineheader, 4);
|
686
|
+
|
687
|
+
/* RLE each component separately */
|
688
|
+
for (c=0; c < 4; c++) {
|
689
|
+
unsigned char *comp = &scratch[width*c];
|
690
|
+
|
691
|
+
x = 0;
|
692
|
+
while (x < width) {
|
693
|
+
// find first run
|
694
|
+
r = x;
|
695
|
+
while (r+2 < width) {
|
696
|
+
if (comp[r] == comp[r+1] && comp[r] == comp[r+2])
|
697
|
+
break;
|
698
|
+
++r;
|
699
|
+
}
|
700
|
+
if (r+2 >= width)
|
701
|
+
r = width;
|
702
|
+
// dump up to first run
|
703
|
+
while (x < r) {
|
704
|
+
int len = r-x;
|
705
|
+
if (len > 128) len = 128;
|
706
|
+
stbiw__write_dump_data(s, len, &comp[x]);
|
707
|
+
x += len;
|
708
|
+
}
|
709
|
+
// if there's a run, output it
|
710
|
+
if (r+2 < width) { // same test as what we break out of in search loop, so only true if we break'd
|
711
|
+
// find next byte after run
|
712
|
+
while (r < width && comp[r] == comp[x])
|
713
|
+
++r;
|
714
|
+
// output run up to r
|
715
|
+
while (x < r) {
|
716
|
+
int len = r-x;
|
717
|
+
if (len > 127) len = 127;
|
718
|
+
stbiw__write_run_data(s, len, comp[x]);
|
719
|
+
x += len;
|
720
|
+
}
|
721
|
+
}
|
722
|
+
}
|
703
723
|
}
|
704
|
-
|
705
|
-
}
|
724
|
+
}
|
706
725
|
}
|
707
726
|
|
708
|
-
static int stbi_write_hdr_core(stbi__write_context *s, int x, int y, int comp,
|
709
|
-
|
710
|
-
|
711
|
-
|
712
|
-
|
713
|
-
|
714
|
-
|
715
|
-
|
716
|
-
|
717
|
-
|
718
|
-
|
719
|
-
"#?RADIANCE\n# Written by stb_image_write.h\nFORMAT=32-bit_rle_rgbe\n";
|
720
|
-
s->func(s->context, header, sizeof(header) - 1);
|
727
|
+
static int stbi_write_hdr_core(stbi__write_context *s, int x, int y, int comp, float *data)
|
728
|
+
{
|
729
|
+
if (y <= 0 || x <= 0 || data == NULL)
|
730
|
+
return 0;
|
731
|
+
else {
|
732
|
+
// Each component is stored separately. Allocate scratch space for full output scanline.
|
733
|
+
unsigned char *scratch = (unsigned char *) STBIW_MALLOC(x*4);
|
734
|
+
int i, len;
|
735
|
+
char buffer[128];
|
736
|
+
char header[] = "#?RADIANCE\n# Written by stb_image_write.h\nFORMAT=32-bit_rle_rgbe\n";
|
737
|
+
s->func(s->context, header, sizeof(header)-1);
|
721
738
|
|
722
739
|
#ifdef STBI_MSC_SECURE_CRT
|
723
|
-
|
724
|
-
buffer, "EXPOSURE= 1.0000000000000\n\n-Y %d +X %d\n", y, x);
|
740
|
+
len = sprintf_s(buffer, "EXPOSURE= 1.0000000000000\n\n-Y %d +X %d\n", y, x);
|
725
741
|
#else
|
726
|
-
|
727
|
-
y, x);
|
742
|
+
len = sprintf(buffer, "EXPOSURE= 1.0000000000000\n\n-Y %d +X %d\n", y, x);
|
728
743
|
#endif
|
729
|
-
|
730
|
-
|
731
|
-
|
732
|
-
|
733
|
-
|
734
|
-
|
735
|
-
|
736
|
-
STBIW_FREE(scratch);
|
737
|
-
return 1;
|
738
|
-
}
|
744
|
+
s->func(s->context, buffer, len);
|
745
|
+
|
746
|
+
for(i=0; i < y; i++)
|
747
|
+
stbiw__write_hdr_scanline(s, x, comp, scratch, data + comp*x*(stbi__flip_vertically_on_write ? y-1-i : i));
|
748
|
+
STBIW_FREE(scratch);
|
749
|
+
return 1;
|
750
|
+
}
|
739
751
|
}
|
740
752
|
|
741
|
-
STBIWDEF int stbi_write_hdr_to_func(stbi_write_func *func, void *context, int x,
|
742
|
-
|
743
|
-
|
744
|
-
|
745
|
-
|
753
|
+
STBIWDEF int stbi_write_hdr_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const float *data)
|
754
|
+
{
|
755
|
+
stbi__write_context s;
|
756
|
+
stbi__start_write_callbacks(&s, func, context);
|
757
|
+
return stbi_write_hdr_core(&s, x, y, comp, (float *) data);
|
746
758
|
}
|
747
759
|
|
748
760
|
#ifndef STBI_WRITE_NO_STDIO
|
749
|
-
STBIWDEF int stbi_write_hdr(char const *filename, int x, int y, int comp,
|
750
|
-
|
751
|
-
|
752
|
-
|
753
|
-
|
754
|
-
|
755
|
-
|
756
|
-
|
757
|
-
|
761
|
+
STBIWDEF int stbi_write_hdr(char const *filename, int x, int y, int comp, const float *data)
|
762
|
+
{
|
763
|
+
stbi__write_context s;
|
764
|
+
if (stbi__start_write_file(&s,filename)) {
|
765
|
+
int r = stbi_write_hdr_core(&s, x, y, comp, (float *) data);
|
766
|
+
stbi__end_write_file(&s);
|
767
|
+
return r;
|
768
|
+
} else
|
769
|
+
return 0;
|
758
770
|
}
|
759
|
-
#endif
|
771
|
+
#endif // STBI_WRITE_NO_STDIO
|
772
|
+
|
760
773
|
|
761
774
|
//////////////////////////////////////////////////////////////////////////////
|
762
775
|
//
|
@@ -764,1009 +777,786 @@ STBIWDEF int stbi_write_hdr(char const *filename, int x, int y, int comp,
|
|
764
777
|
//
|
765
778
|
|
766
779
|
#ifndef STBIW_ZLIB_COMPRESS
|
767
|
-
// stretchy buffer; stbiw__sbpush() == vector<>::push_back() -- stbiw__sbcount()
|
768
|
-
|
769
|
-
#define
|
770
|
-
#define
|
771
|
-
|
772
|
-
|
773
|
-
#define
|
774
|
-
#define
|
775
|
-
|
776
|
-
#define
|
777
|
-
|
778
|
-
#define
|
779
|
-
|
780
|
-
|
781
|
-
|
782
|
-
|
783
|
-
|
784
|
-
|
785
|
-
|
786
|
-
|
787
|
-
*arr
|
788
|
-
|
789
|
-
|
790
|
-
|
791
|
-
if (!*arr) ((int *)p)[1] = 0;
|
792
|
-
*arr = (void *)((int *)p + 2);
|
793
|
-
stbiw__sbm(*arr) = m;
|
794
|
-
}
|
795
|
-
return *arr;
|
780
|
+
// stretchy buffer; stbiw__sbpush() == vector<>::push_back() -- stbiw__sbcount() == vector<>::size()
|
781
|
+
#define stbiw__sbraw(a) ((int *) (a) - 2)
|
782
|
+
#define stbiw__sbm(a) stbiw__sbraw(a)[0]
|
783
|
+
#define stbiw__sbn(a) stbiw__sbraw(a)[1]
|
784
|
+
|
785
|
+
#define stbiw__sbneedgrow(a,n) ((a)==0 || stbiw__sbn(a)+n >= stbiw__sbm(a))
|
786
|
+
#define stbiw__sbmaybegrow(a,n) (stbiw__sbneedgrow(a,(n)) ? stbiw__sbgrow(a,n) : 0)
|
787
|
+
#define stbiw__sbgrow(a,n) stbiw__sbgrowf((void **) &(a), (n), sizeof(*(a)))
|
788
|
+
|
789
|
+
#define stbiw__sbpush(a, v) (stbiw__sbmaybegrow(a,1), (a)[stbiw__sbn(a)++] = (v))
|
790
|
+
#define stbiw__sbcount(a) ((a) ? stbiw__sbn(a) : 0)
|
791
|
+
#define stbiw__sbfree(a) ((a) ? STBIW_FREE(stbiw__sbraw(a)),0 : 0)
|
792
|
+
|
793
|
+
static void *stbiw__sbgrowf(void **arr, int increment, int itemsize)
|
794
|
+
{
|
795
|
+
int m = *arr ? 2*stbiw__sbm(*arr)+increment : increment+1;
|
796
|
+
void *p = STBIW_REALLOC_SIZED(*arr ? stbiw__sbraw(*arr) : 0, *arr ? (stbiw__sbm(*arr)*itemsize + sizeof(int)*2) : 0, itemsize * m + sizeof(int)*2);
|
797
|
+
STBIW_ASSERT(p);
|
798
|
+
if (p) {
|
799
|
+
if (!*arr) ((int *) p)[1] = 0;
|
800
|
+
*arr = (void *) ((int *) p + 2);
|
801
|
+
stbiw__sbm(*arr) = m;
|
802
|
+
}
|
803
|
+
return *arr;
|
796
804
|
}
|
797
805
|
|
798
|
-
static unsigned char *stbiw__zlib_flushf(unsigned char *data,
|
799
|
-
|
800
|
-
|
801
|
-
|
802
|
-
|
803
|
-
|
804
|
-
|
805
|
-
|
806
|
-
return data;
|
806
|
+
static unsigned char *stbiw__zlib_flushf(unsigned char *data, unsigned int *bitbuffer, int *bitcount)
|
807
|
+
{
|
808
|
+
while (*bitcount >= 8) {
|
809
|
+
stbiw__sbpush(data, STBIW_UCHAR(*bitbuffer));
|
810
|
+
*bitbuffer >>= 8;
|
811
|
+
*bitcount -= 8;
|
812
|
+
}
|
813
|
+
return data;
|
807
814
|
}
|
808
815
|
|
809
|
-
static int stbiw__zlib_bitrev(int code, int codebits)
|
810
|
-
|
811
|
-
|
812
|
-
|
813
|
-
|
814
|
-
|
815
|
-
|
816
|
+
static int stbiw__zlib_bitrev(int code, int codebits)
|
817
|
+
{
|
818
|
+
int res=0;
|
819
|
+
while (codebits--) {
|
820
|
+
res = (res << 1) | (code & 1);
|
821
|
+
code >>= 1;
|
822
|
+
}
|
823
|
+
return res;
|
816
824
|
}
|
817
825
|
|
818
|
-
static unsigned int stbiw__zlib_countm(unsigned char *a, unsigned char *b,
|
819
|
-
|
820
|
-
|
821
|
-
|
822
|
-
|
823
|
-
|
826
|
+
static unsigned int stbiw__zlib_countm(unsigned char *a, unsigned char *b, int limit)
|
827
|
+
{
|
828
|
+
int i;
|
829
|
+
for (i=0; i < limit && i < 258; ++i)
|
830
|
+
if (a[i] != b[i]) break;
|
831
|
+
return i;
|
824
832
|
}
|
825
833
|
|
826
|
-
static unsigned int stbiw__zhash(unsigned char *data)
|
827
|
-
|
828
|
-
|
829
|
-
|
830
|
-
|
831
|
-
|
832
|
-
|
833
|
-
|
834
|
-
|
834
|
+
static unsigned int stbiw__zhash(unsigned char *data)
|
835
|
+
{
|
836
|
+
stbiw_uint32 hash = data[0] + (data[1] << 8) + (data[2] << 16);
|
837
|
+
hash ^= hash << 3;
|
838
|
+
hash += hash >> 5;
|
839
|
+
hash ^= hash << 4;
|
840
|
+
hash += hash >> 17;
|
841
|
+
hash ^= hash << 25;
|
842
|
+
hash += hash >> 6;
|
843
|
+
return hash;
|
835
844
|
}
|
836
845
|
|
837
846
|
#define stbiw__zlib_flush() (out = stbiw__zlib_flushf(out, &bitbuf, &bitcount))
|
838
|
-
#define stbiw__zlib_add(code,
|
839
|
-
|
840
|
-
#define stbiw__zlib_huffa(b,
|
847
|
+
#define stbiw__zlib_add(code,codebits) \
|
848
|
+
(bitbuf |= (code) << bitcount, bitcount += (codebits), stbiw__zlib_flush())
|
849
|
+
#define stbiw__zlib_huffa(b,c) stbiw__zlib_add(stbiw__zlib_bitrev(b,c),c)
|
841
850
|
// default huffman tables
|
842
|
-
#define stbiw__zlib_huff1(n)
|
843
|
-
#define stbiw__zlib_huff2(n)
|
844
|
-
#define stbiw__zlib_huff3(n)
|
845
|
-
#define stbiw__zlib_huff4(n)
|
846
|
-
#define stbiw__zlib_huff(n)
|
847
|
-
|
848
|
-
|
849
|
-
|
850
|
-
|
851
|
-
#
|
852
|
-
|
853
|
-
|
854
|
-
|
855
|
-
|
856
|
-
#endif // STBIW_ZLIB_COMPRESS
|
857
|
-
|
858
|
-
unsigned char *stbi_zlib_compress(unsigned char *data, int data_len,
|
859
|
-
int *out_len, int quality) {
|
851
|
+
#define stbiw__zlib_huff1(n) stbiw__zlib_huffa(0x30 + (n), 8)
|
852
|
+
#define stbiw__zlib_huff2(n) stbiw__zlib_huffa(0x190 + (n)-144, 9)
|
853
|
+
#define stbiw__zlib_huff3(n) stbiw__zlib_huffa(0 + (n)-256,7)
|
854
|
+
#define stbiw__zlib_huff4(n) stbiw__zlib_huffa(0xc0 + (n)-280,8)
|
855
|
+
#define stbiw__zlib_huff(n) ((n) <= 143 ? stbiw__zlib_huff1(n) : (n) <= 255 ? stbiw__zlib_huff2(n) : (n) <= 279 ? stbiw__zlib_huff3(n) : stbiw__zlib_huff4(n))
|
856
|
+
#define stbiw__zlib_huffb(n) ((n) <= 143 ? stbiw__zlib_huff1(n) : stbiw__zlib_huff2(n))
|
857
|
+
|
858
|
+
#define stbiw__ZHASH 16384
|
859
|
+
|
860
|
+
#endif // STBIW_ZLIB_COMPRESS
|
861
|
+
|
862
|
+
STBIWDEF unsigned char * stbi_zlib_compress(unsigned char *data, int data_len, int *out_len, int quality)
|
863
|
+
{
|
860
864
|
#ifdef STBIW_ZLIB_COMPRESS
|
861
|
-
|
862
|
-
|
863
|
-
#else
|
864
|
-
|
865
|
-
|
866
|
-
|
867
|
-
|
868
|
-
|
869
|
-
|
870
|
-
|
871
|
-
|
872
|
-
|
873
|
-
|
874
|
-
|
875
|
-
|
876
|
-
|
877
|
-
|
878
|
-
|
879
|
-
|
880
|
-
|
881
|
-
|
882
|
-
|
883
|
-
|
884
|
-
|
885
|
-
|
886
|
-
|
887
|
-
|
888
|
-
|
889
|
-
|
890
|
-
|
891
|
-
|
892
|
-
|
893
|
-
|
894
|
-
|
895
|
-
|
896
|
-
unsigned char *bestloc = 0;
|
897
|
-
unsigned char **hlist = hash_table[h];
|
898
|
-
int n = stbiw__sbcount(hlist);
|
899
|
-
for (j = 0; j < n; ++j) {
|
900
|
-
if (hlist[j] - data > i - 32768) { // if entry lies within window
|
901
|
-
int d = stbiw__zlib_countm(hlist[j], data + i, data_len - i);
|
902
|
-
if (d >= best) best = d, bestloc = hlist[j];
|
865
|
+
// user provided a zlib compress implementation, use that
|
866
|
+
return STBIW_ZLIB_COMPRESS(data, data_len, out_len, quality);
|
867
|
+
#else // use builtin
|
868
|
+
static unsigned short lengthc[] = { 3,4,5,6,7,8,9,10,11,13,15,17,19,23,27,31,35,43,51,59,67,83,99,115,131,163,195,227,258, 259 };
|
869
|
+
static unsigned char lengtheb[]= { 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 };
|
870
|
+
static unsigned short distc[] = { 1,2,3,4,5,7,9,13,17,25,33,49,65,97,129,193,257,385,513,769,1025,1537,2049,3073,4097,6145,8193,12289,16385,24577, 32768 };
|
871
|
+
static unsigned char disteb[] = { 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 };
|
872
|
+
unsigned int bitbuf=0;
|
873
|
+
int i,j, bitcount=0;
|
874
|
+
unsigned char *out = NULL;
|
875
|
+
unsigned char ***hash_table = (unsigned char***) STBIW_MALLOC(stbiw__ZHASH * sizeof(char**));
|
876
|
+
if (hash_table == NULL)
|
877
|
+
return NULL;
|
878
|
+
if (quality < 5) quality = 5;
|
879
|
+
|
880
|
+
stbiw__sbpush(out, 0x78); // DEFLATE 32K window
|
881
|
+
stbiw__sbpush(out, 0x5e); // FLEVEL = 1
|
882
|
+
stbiw__zlib_add(1,1); // BFINAL = 1
|
883
|
+
stbiw__zlib_add(1,2); // BTYPE = 1 -- fixed huffman
|
884
|
+
|
885
|
+
for (i=0; i < stbiw__ZHASH; ++i)
|
886
|
+
hash_table[i] = NULL;
|
887
|
+
|
888
|
+
i=0;
|
889
|
+
while (i < data_len-3) {
|
890
|
+
// hash next 3 bytes of data to be compressed
|
891
|
+
int h = stbiw__zhash(data+i)&(stbiw__ZHASH-1), best=3;
|
892
|
+
unsigned char *bestloc = 0;
|
893
|
+
unsigned char **hlist = hash_table[h];
|
894
|
+
int n = stbiw__sbcount(hlist);
|
895
|
+
for (j=0; j < n; ++j) {
|
896
|
+
if (hlist[j]-data > i-32768) { // if entry lies within window
|
897
|
+
int d = stbiw__zlib_countm(hlist[j], data+i, data_len-i);
|
898
|
+
if (d >= best) best=d,bestloc=hlist[j];
|
899
|
+
}
|
903
900
|
}
|
904
|
-
|
905
|
-
|
906
|
-
|
907
|
-
|
908
|
-
|
909
|
-
|
910
|
-
|
911
|
-
|
912
|
-
|
913
|
-
|
914
|
-
|
915
|
-
|
916
|
-
|
917
|
-
|
918
|
-
|
919
|
-
|
920
|
-
|
921
|
-
|
922
|
-
|
923
|
-
|
924
|
-
|
925
|
-
}
|
926
|
-
}
|
901
|
+
// when hash table entry is too long, delete half the entries
|
902
|
+
if (hash_table[h] && stbiw__sbn(hash_table[h]) == 2*quality) {
|
903
|
+
STBIW_MEMMOVE(hash_table[h], hash_table[h]+quality, sizeof(hash_table[h][0])*quality);
|
904
|
+
stbiw__sbn(hash_table[h]) = quality;
|
905
|
+
}
|
906
|
+
stbiw__sbpush(hash_table[h],data+i);
|
907
|
+
|
908
|
+
if (bestloc) {
|
909
|
+
// "lazy matching" - check match at *next* byte, and if it's better, do cur byte as literal
|
910
|
+
h = stbiw__zhash(data+i+1)&(stbiw__ZHASH-1);
|
911
|
+
hlist = hash_table[h];
|
912
|
+
n = stbiw__sbcount(hlist);
|
913
|
+
for (j=0; j < n; ++j) {
|
914
|
+
if (hlist[j]-data > i-32767) {
|
915
|
+
int e = stbiw__zlib_countm(hlist[j], data+i+1, data_len-i-1);
|
916
|
+
if (e > best) { // if next match is better, bail on current match
|
917
|
+
bestloc = NULL;
|
918
|
+
break;
|
919
|
+
}
|
920
|
+
}
|
921
|
+
}
|
927
922
|
}
|
928
|
-
|
929
|
-
|
930
|
-
|
931
|
-
|
932
|
-
|
933
|
-
|
934
|
-
|
935
|
-
|
936
|
-
|
937
|
-
|
938
|
-
|
939
|
-
|
940
|
-
|
941
|
-
|
942
|
-
|
923
|
+
|
924
|
+
if (bestloc) {
|
925
|
+
int d = (int) (data+i - bestloc); // distance back
|
926
|
+
STBIW_ASSERT(d <= 32767 && best <= 258);
|
927
|
+
for (j=0; best > lengthc[j+1]-1; ++j);
|
928
|
+
stbiw__zlib_huff(j+257);
|
929
|
+
if (lengtheb[j]) stbiw__zlib_add(best - lengthc[j], lengtheb[j]);
|
930
|
+
for (j=0; d > distc[j+1]-1; ++j);
|
931
|
+
stbiw__zlib_add(stbiw__zlib_bitrev(j,5),5);
|
932
|
+
if (disteb[j]) stbiw__zlib_add(d - distc[j], disteb[j]);
|
933
|
+
i += best;
|
934
|
+
} else {
|
935
|
+
stbiw__zlib_huffb(data[i]);
|
936
|
+
++i;
|
937
|
+
}
|
938
|
+
}
|
939
|
+
// write out final bytes
|
940
|
+
for (;i < data_len; ++i)
|
943
941
|
stbiw__zlib_huffb(data[i]);
|
944
|
-
|
945
|
-
|
946
|
-
|
947
|
-
|
948
|
-
|
949
|
-
|
950
|
-
|
951
|
-
|
952
|
-
|
953
|
-
|
954
|
-
|
955
|
-
|
956
|
-
|
957
|
-
|
958
|
-
|
959
|
-
|
960
|
-
|
961
|
-
|
962
|
-
|
963
|
-
|
964
|
-
|
965
|
-
|
966
|
-
|
967
|
-
|
968
|
-
|
969
|
-
|
970
|
-
|
971
|
-
|
972
|
-
|
973
|
-
|
974
|
-
STBIW_MEMMOVE(stbiw__sbraw(out), out, *out_len);
|
975
|
-
return (unsigned char *)stbiw__sbraw(out);
|
976
|
-
#endif // STBIW_ZLIB_COMPRESS
|
942
|
+
stbiw__zlib_huff(256); // end of block
|
943
|
+
// pad with 0 bits to byte boundary
|
944
|
+
while (bitcount)
|
945
|
+
stbiw__zlib_add(0,1);
|
946
|
+
|
947
|
+
for (i=0; i < stbiw__ZHASH; ++i)
|
948
|
+
(void) stbiw__sbfree(hash_table[i]);
|
949
|
+
STBIW_FREE(hash_table);
|
950
|
+
|
951
|
+
{
|
952
|
+
// compute adler32 on input
|
953
|
+
unsigned int s1=1, s2=0;
|
954
|
+
int blocklen = (int) (data_len % 5552);
|
955
|
+
j=0;
|
956
|
+
while (j < data_len) {
|
957
|
+
for (i=0; i < blocklen; ++i) s1 += data[j+i], s2 += s1;
|
958
|
+
s1 %= 65521, s2 %= 65521;
|
959
|
+
j += blocklen;
|
960
|
+
blocklen = 5552;
|
961
|
+
}
|
962
|
+
stbiw__sbpush(out, STBIW_UCHAR(s2 >> 8));
|
963
|
+
stbiw__sbpush(out, STBIW_UCHAR(s2));
|
964
|
+
stbiw__sbpush(out, STBIW_UCHAR(s1 >> 8));
|
965
|
+
stbiw__sbpush(out, STBIW_UCHAR(s1));
|
966
|
+
}
|
967
|
+
*out_len = stbiw__sbn(out);
|
968
|
+
// make returned pointer freeable
|
969
|
+
STBIW_MEMMOVE(stbiw__sbraw(out), out, *out_len);
|
970
|
+
return (unsigned char *) stbiw__sbraw(out);
|
971
|
+
#endif // STBIW_ZLIB_COMPRESS
|
977
972
|
}
|
978
973
|
|
979
|
-
static unsigned int stbiw__crc32(unsigned char *buffer, int len)
|
980
|
-
|
981
|
-
|
982
|
-
|
983
|
-
|
984
|
-
|
985
|
-
|
986
|
-
|
987
|
-
|
988
|
-
|
989
|
-
|
990
|
-
|
991
|
-
|
992
|
-
|
993
|
-
|
994
|
-
|
995
|
-
|
996
|
-
|
997
|
-
|
998
|
-
|
999
|
-
0x44042D73, 0x33031DE5, 0xAA0A4C5F, 0xDD0D7CC9,
|
1000
|
-
0xBE0B1010, 0xC90C2086, 0x5768B525, 0x206F85B3, 0xB966D409, 0xCE61E49F,
|
1001
|
-
0x5EDEF90E, 0x29D9C998, 0xB0D09822, 0xC7D7A8B4, 0x59B33D17, 0x2EB40D81,
|
1002
|
-
|
1003
|
-
|
1004
|
-
|
1005
|
-
|
1006
|
-
|
1007
|
-
|
1008
|
-
|
1009
|
-
|
1010
|
-
|
1011
|
-
|
1012
|
-
|
1013
|
-
|
1014
|
-
|
1015
|
-
|
1016
|
-
|
1017
|
-
|
1018
|
-
|
1019
|
-
|
1020
|
-
|
1021
|
-
|
1022
|
-
|
1023
|
-
|
1024
|
-
|
1025
|
-
|
1026
|
-
int i;
|
1027
|
-
for (i = 0; i < len; ++i)
|
1028
|
-
crc = (crc >> 8) ^ crc_table[buffer[i] ^ (crc & 0xff)];
|
1029
|
-
return ~crc;
|
974
|
+
static unsigned int stbiw__crc32(unsigned char *buffer, int len)
|
975
|
+
{
|
976
|
+
#ifdef STBIW_CRC32
|
977
|
+
return STBIW_CRC32(buffer, len);
|
978
|
+
#else
|
979
|
+
static unsigned int crc_table[256] =
|
980
|
+
{
|
981
|
+
0x00000000, 0x77073096, 0xEE0E612C, 0x990951BA, 0x076DC419, 0x706AF48F, 0xE963A535, 0x9E6495A3,
|
982
|
+
0x0eDB8832, 0x79DCB8A4, 0xE0D5E91E, 0x97D2D988, 0x09B64C2B, 0x7EB17CBD, 0xE7B82D07, 0x90BF1D91,
|
983
|
+
0x1DB71064, 0x6AB020F2, 0xF3B97148, 0x84BE41DE, 0x1ADAD47D, 0x6DDDE4EB, 0xF4D4B551, 0x83D385C7,
|
984
|
+
0x136C9856, 0x646BA8C0, 0xFD62F97A, 0x8A65C9EC, 0x14015C4F, 0x63066CD9, 0xFA0F3D63, 0x8D080DF5,
|
985
|
+
0x3B6E20C8, 0x4C69105E, 0xD56041E4, 0xA2677172, 0x3C03E4D1, 0x4B04D447, 0xD20D85FD, 0xA50AB56B,
|
986
|
+
0x35B5A8FA, 0x42B2986C, 0xDBBBC9D6, 0xACBCF940, 0x32D86CE3, 0x45DF5C75, 0xDCD60DCF, 0xABD13D59,
|
987
|
+
0x26D930AC, 0x51DE003A, 0xC8D75180, 0xBFD06116, 0x21B4F4B5, 0x56B3C423, 0xCFBA9599, 0xB8BDA50F,
|
988
|
+
0x2802B89E, 0x5F058808, 0xC60CD9B2, 0xB10BE924, 0x2F6F7C87, 0x58684C11, 0xC1611DAB, 0xB6662D3D,
|
989
|
+
0x76DC4190, 0x01DB7106, 0x98D220BC, 0xEFD5102A, 0x71B18589, 0x06B6B51F, 0x9FBFE4A5, 0xE8B8D433,
|
990
|
+
0x7807C9A2, 0x0F00F934, 0x9609A88E, 0xE10E9818, 0x7F6A0DBB, 0x086D3D2D, 0x91646C97, 0xE6635C01,
|
991
|
+
0x6B6B51F4, 0x1C6C6162, 0x856530D8, 0xF262004E, 0x6C0695ED, 0x1B01A57B, 0x8208F4C1, 0xF50FC457,
|
992
|
+
0x65B0D9C6, 0x12B7E950, 0x8BBEB8EA, 0xFCB9887C, 0x62DD1DDF, 0x15DA2D49, 0x8CD37CF3, 0xFBD44C65,
|
993
|
+
0x4DB26158, 0x3AB551CE, 0xA3BC0074, 0xD4BB30E2, 0x4ADFA541, 0x3DD895D7, 0xA4D1C46D, 0xD3D6F4FB,
|
994
|
+
0x4369E96A, 0x346ED9FC, 0xAD678846, 0xDA60B8D0, 0x44042D73, 0x33031DE5, 0xAA0A4C5F, 0xDD0D7CC9,
|
995
|
+
0x5005713C, 0x270241AA, 0xBE0B1010, 0xC90C2086, 0x5768B525, 0x206F85B3, 0xB966D409, 0xCE61E49F,
|
996
|
+
0x5EDEF90E, 0x29D9C998, 0xB0D09822, 0xC7D7A8B4, 0x59B33D17, 0x2EB40D81, 0xB7BD5C3B, 0xC0BA6CAD,
|
997
|
+
0xEDB88320, 0x9ABFB3B6, 0x03B6E20C, 0x74B1D29A, 0xEAD54739, 0x9DD277AF, 0x04DB2615, 0x73DC1683,
|
998
|
+
0xE3630B12, 0x94643B84, 0x0D6D6A3E, 0x7A6A5AA8, 0xE40ECF0B, 0x9309FF9D, 0x0A00AE27, 0x7D079EB1,
|
999
|
+
0xF00F9344, 0x8708A3D2, 0x1E01F268, 0x6906C2FE, 0xF762575D, 0x806567CB, 0x196C3671, 0x6E6B06E7,
|
1000
|
+
0xFED41B76, 0x89D32BE0, 0x10DA7A5A, 0x67DD4ACC, 0xF9B9DF6F, 0x8EBEEFF9, 0x17B7BE43, 0x60B08ED5,
|
1001
|
+
0xD6D6A3E8, 0xA1D1937E, 0x38D8C2C4, 0x4FDFF252, 0xD1BB67F1, 0xA6BC5767, 0x3FB506DD, 0x48B2364B,
|
1002
|
+
0xD80D2BDA, 0xAF0A1B4C, 0x36034AF6, 0x41047A60, 0xDF60EFC3, 0xA867DF55, 0x316E8EEF, 0x4669BE79,
|
1003
|
+
0xCB61B38C, 0xBC66831A, 0x256FD2A0, 0x5268E236, 0xCC0C7795, 0xBB0B4703, 0x220216B9, 0x5505262F,
|
1004
|
+
0xC5BA3BBE, 0xB2BD0B28, 0x2BB45A92, 0x5CB36A04, 0xC2D7FFA7, 0xB5D0CF31, 0x2CD99E8B, 0x5BDEAE1D,
|
1005
|
+
0x9B64C2B0, 0xEC63F226, 0x756AA39C, 0x026D930A, 0x9C0906A9, 0xEB0E363F, 0x72076785, 0x05005713,
|
1006
|
+
0x95BF4A82, 0xE2B87A14, 0x7BB12BAE, 0x0CB61B38, 0x92D28E9B, 0xE5D5BE0D, 0x7CDCEFB7, 0x0BDBDF21,
|
1007
|
+
0x86D3D2D4, 0xF1D4E242, 0x68DDB3F8, 0x1FDA836E, 0x81BE16CD, 0xF6B9265B, 0x6FB077E1, 0x18B74777,
|
1008
|
+
0x88085AE6, 0xFF0F6A70, 0x66063BCA, 0x11010B5C, 0x8F659EFF, 0xF862AE69, 0x616BFFD3, 0x166CCF45,
|
1009
|
+
0xA00AE278, 0xD70DD2EE, 0x4E048354, 0x3903B3C2, 0xA7672661, 0xD06016F7, 0x4969474D, 0x3E6E77DB,
|
1010
|
+
0xAED16A4A, 0xD9D65ADC, 0x40DF0B66, 0x37D83BF0, 0xA9BCAE53, 0xDEBB9EC5, 0x47B2CF7F, 0x30B5FFE9,
|
1011
|
+
0xBDBDF21C, 0xCABAC28A, 0x53B39330, 0x24B4A3A6, 0xBAD03605, 0xCDD70693, 0x54DE5729, 0x23D967BF,
|
1012
|
+
0xB3667A2E, 0xC4614AB8, 0x5D681B02, 0x2A6F2B94, 0xB40BBE37, 0xC30C8EA1, 0x5A05DF1B, 0x2D02EF8D
|
1013
|
+
};
|
1014
|
+
|
1015
|
+
unsigned int crc = ~0u;
|
1016
|
+
int i;
|
1017
|
+
for (i=0; i < len; ++i)
|
1018
|
+
crc = (crc >> 8) ^ crc_table[buffer[i] ^ (crc & 0xff)];
|
1019
|
+
return ~crc;
|
1020
|
+
#endif
|
1030
1021
|
}
|
1031
1022
|
|
1032
|
-
#define stbiw__wpng4(o, a,
|
1033
|
-
|
1034
|
-
|
1035
|
-
#define stbiw__wp32(data, v) \
|
1036
|
-
stbiw__wpng4(data, (v) >> 24, (v) >> 16, (v) >> 8, (v));
|
1037
|
-
#define stbiw__wptag(data, s) stbiw__wpng4(data, s[0], s[1], s[2], s[3])
|
1023
|
+
#define stbiw__wpng4(o,a,b,c,d) ((o)[0]=STBIW_UCHAR(a),(o)[1]=STBIW_UCHAR(b),(o)[2]=STBIW_UCHAR(c),(o)[3]=STBIW_UCHAR(d),(o)+=4)
|
1024
|
+
#define stbiw__wp32(data,v) stbiw__wpng4(data, (v)>>24,(v)>>16,(v)>>8,(v));
|
1025
|
+
#define stbiw__wptag(data,s) stbiw__wpng4(data, s[0],s[1],s[2],s[3])
|
1038
1026
|
|
1039
|
-
static void stbiw__wpcrc(unsigned char **data, int len)
|
1040
|
-
|
1041
|
-
|
1027
|
+
static void stbiw__wpcrc(unsigned char **data, int len)
|
1028
|
+
{
|
1029
|
+
unsigned int crc = stbiw__crc32(*data - len - 4, len+4);
|
1030
|
+
stbiw__wp32(*data, crc);
|
1042
1031
|
}
|
1043
1032
|
|
1044
|
-
static unsigned char stbiw__paeth(int a, int b, int c)
|
1045
|
-
|
1046
|
-
|
1047
|
-
|
1048
|
-
|
1033
|
+
static unsigned char stbiw__paeth(int a, int b, int c)
|
1034
|
+
{
|
1035
|
+
int p = a + b - c, pa = abs(p-a), pb = abs(p-b), pc = abs(p-c);
|
1036
|
+
if (pa <= pb && pa <= pc) return STBIW_UCHAR(a);
|
1037
|
+
if (pb <= pc) return STBIW_UCHAR(b);
|
1038
|
+
return STBIW_UCHAR(c);
|
1049
1039
|
}
|
1050
1040
|
|
1051
1041
|
// @OPTIMIZE: provide an option that always forces left-predict or paeth predict
|
1052
|
-
static void stbiw__encode_png_line(unsigned char *pixels, int stride_bytes,
|
1053
|
-
|
1054
|
-
|
1055
|
-
|
1056
|
-
|
1057
|
-
|
1058
|
-
|
1059
|
-
|
1060
|
-
|
1061
|
-
|
1062
|
-
|
1063
|
-
|
1064
|
-
|
1065
|
-
|
1066
|
-
|
1067
|
-
|
1068
|
-
|
1069
|
-
|
1070
|
-
|
1071
|
-
|
1072
|
-
|
1073
|
-
|
1074
|
-
|
1075
|
-
|
1076
|
-
|
1077
|
-
|
1078
|
-
|
1079
|
-
case
|
1080
|
-
|
1081
|
-
|
1082
|
-
|
1083
|
-
case 5:
|
1084
|
-
|
1085
|
-
|
1086
|
-
case 6:
|
1087
|
-
line_buffer[i] = z[i];
|
1088
|
-
break;
|
1089
|
-
}
|
1090
|
-
}
|
1091
|
-
for (i = n; i < width * n; ++i) {
|
1092
|
-
switch (type) {
|
1093
|
-
case 0:
|
1094
|
-
line_buffer[i] = z[i];
|
1095
|
-
break;
|
1096
|
-
case 1:
|
1097
|
-
line_buffer[i] = z[i] - z[i - n];
|
1098
|
-
break;
|
1099
|
-
case 2:
|
1100
|
-
line_buffer[i] = z[i] - z[i - signed_stride];
|
1101
|
-
break;
|
1102
|
-
case 3:
|
1103
|
-
line_buffer[i] = z[i] - ((z[i - n] + z[i - signed_stride]) >> 1);
|
1104
|
-
break;
|
1105
|
-
case 4:
|
1106
|
-
line_buffer[i] = z[i] - stbiw__paeth(z[i - n], z[i - signed_stride],
|
1107
|
-
z[i - signed_stride - n]);
|
1108
|
-
break;
|
1109
|
-
case 5:
|
1110
|
-
line_buffer[i] = z[i] - (z[i - n] >> 1);
|
1111
|
-
break;
|
1112
|
-
case 6:
|
1113
|
-
line_buffer[i] = z[i] - stbiw__paeth(z[i - n], 0, 0);
|
1114
|
-
break;
|
1115
|
-
}
|
1116
|
-
}
|
1042
|
+
static void stbiw__encode_png_line(unsigned char *pixels, int stride_bytes, int width, int height, int y, int n, int filter_type, signed char *line_buffer)
|
1043
|
+
{
|
1044
|
+
static int mapping[] = { 0,1,2,3,4 };
|
1045
|
+
static int firstmap[] = { 0,1,0,5,6 };
|
1046
|
+
int *mymap = (y != 0) ? mapping : firstmap;
|
1047
|
+
int i;
|
1048
|
+
int type = mymap[filter_type];
|
1049
|
+
unsigned char *z = pixels + stride_bytes * (stbi__flip_vertically_on_write ? height-1-y : y);
|
1050
|
+
int signed_stride = stbi__flip_vertically_on_write ? -stride_bytes : stride_bytes;
|
1051
|
+
|
1052
|
+
if (type==0) {
|
1053
|
+
memcpy(line_buffer, z, width*n);
|
1054
|
+
return;
|
1055
|
+
}
|
1056
|
+
|
1057
|
+
// first loop isn't optimized since it's just one pixel
|
1058
|
+
for (i = 0; i < n; ++i) {
|
1059
|
+
switch (type) {
|
1060
|
+
case 1: line_buffer[i] = z[i]; break;
|
1061
|
+
case 2: line_buffer[i] = z[i] - z[i-signed_stride]; break;
|
1062
|
+
case 3: line_buffer[i] = z[i] - (z[i-signed_stride]>>1); break;
|
1063
|
+
case 4: line_buffer[i] = (signed char) (z[i] - stbiw__paeth(0,z[i-signed_stride],0)); break;
|
1064
|
+
case 5: line_buffer[i] = z[i]; break;
|
1065
|
+
case 6: line_buffer[i] = z[i]; break;
|
1066
|
+
}
|
1067
|
+
}
|
1068
|
+
switch (type) {
|
1069
|
+
case 1: for (i=n; i < width*n; ++i) line_buffer[i] = z[i] - z[i-n]; break;
|
1070
|
+
case 2: for (i=n; i < width*n; ++i) line_buffer[i] = z[i] - z[i-signed_stride]; break;
|
1071
|
+
case 3: for (i=n; i < width*n; ++i) line_buffer[i] = z[i] - ((z[i-n] + z[i-signed_stride])>>1); break;
|
1072
|
+
case 4: for (i=n; i < width*n; ++i) line_buffer[i] = z[i] - stbiw__paeth(z[i-n], z[i-signed_stride], z[i-signed_stride-n]); break;
|
1073
|
+
case 5: for (i=n; i < width*n; ++i) line_buffer[i] = z[i] - (z[i-n]>>1); break;
|
1074
|
+
case 6: for (i=n; i < width*n; ++i) line_buffer[i] = z[i] - stbiw__paeth(z[i-n], 0,0); break;
|
1075
|
+
}
|
1117
1076
|
}
|
1118
1077
|
|
1119
|
-
unsigned char *stbi_write_png_to_mem(unsigned char *pixels, int stride_bytes,
|
1120
|
-
|
1121
|
-
|
1122
|
-
|
1123
|
-
|
1124
|
-
|
1125
|
-
|
1126
|
-
|
1127
|
-
|
1128
|
-
|
1129
|
-
|
1130
|
-
|
1131
|
-
|
1132
|
-
|
1133
|
-
|
1134
|
-
|
1135
|
-
|
1136
|
-
|
1137
|
-
|
1138
|
-
|
1139
|
-
|
1140
|
-
|
1141
|
-
|
1142
|
-
|
1143
|
-
|
1144
|
-
|
1145
|
-
|
1146
|
-
|
1147
|
-
|
1148
|
-
|
1149
|
-
|
1150
|
-
|
1151
|
-
|
1152
|
-
|
1153
|
-
|
1154
|
-
|
1155
|
-
|
1156
|
-
|
1157
|
-
|
1158
|
-
|
1159
|
-
|
1160
|
-
|
1161
|
-
best_filter = filter_type;
|
1162
|
-
}
|
1163
|
-
}
|
1164
|
-
if (filter_type != best_filter) { // If the last iteration already got us
|
1165
|
-
// the best filter, don't redo it
|
1166
|
-
stbiw__encode_png_line(pixels, stride_bytes, x, y, j, n, best_filter,
|
1167
|
-
line_buffer);
|
1168
|
-
filter_type = best_filter;
|
1078
|
+
STBIWDEF unsigned char *stbi_write_png_to_mem(const unsigned char *pixels, int stride_bytes, int x, int y, int n, int *out_len)
|
1079
|
+
{
|
1080
|
+
int force_filter = stbi_write_force_png_filter;
|
1081
|
+
int ctype[5] = { -1, 0, 4, 2, 6 };
|
1082
|
+
unsigned char sig[8] = { 137,80,78,71,13,10,26,10 };
|
1083
|
+
unsigned char *out,*o, *filt, *zlib;
|
1084
|
+
signed char *line_buffer;
|
1085
|
+
int j,zlen;
|
1086
|
+
|
1087
|
+
if (stride_bytes == 0)
|
1088
|
+
stride_bytes = x * n;
|
1089
|
+
|
1090
|
+
if (force_filter >= 5) {
|
1091
|
+
force_filter = -1;
|
1092
|
+
}
|
1093
|
+
|
1094
|
+
filt = (unsigned char *) STBIW_MALLOC((x*n+1) * y); if (!filt) return 0;
|
1095
|
+
line_buffer = (signed char *) STBIW_MALLOC(x * n); if (!line_buffer) { STBIW_FREE(filt); return 0; }
|
1096
|
+
for (j=0; j < y; ++j) {
|
1097
|
+
int filter_type;
|
1098
|
+
if (force_filter > -1) {
|
1099
|
+
filter_type = force_filter;
|
1100
|
+
stbiw__encode_png_line((unsigned char*)(pixels), stride_bytes, x, y, j, n, force_filter, line_buffer);
|
1101
|
+
} else { // Estimate the best filter by running through all of them:
|
1102
|
+
int best_filter = 0, best_filter_val = 0x7fffffff, est, i;
|
1103
|
+
for (filter_type = 0; filter_type < 5; filter_type++) {
|
1104
|
+
stbiw__encode_png_line((unsigned char*)(pixels), stride_bytes, x, y, j, n, filter_type, line_buffer);
|
1105
|
+
|
1106
|
+
// Estimate the entropy of the line using this filter; the less, the better.
|
1107
|
+
est = 0;
|
1108
|
+
for (i = 0; i < x*n; ++i) {
|
1109
|
+
est += abs((signed char) line_buffer[i]);
|
1110
|
+
}
|
1111
|
+
if (est < best_filter_val) {
|
1112
|
+
best_filter_val = est;
|
1113
|
+
best_filter = filter_type;
|
1114
|
+
}
|
1115
|
+
}
|
1116
|
+
if (filter_type != best_filter) { // If the last iteration already got us the best filter, don't redo it
|
1117
|
+
stbiw__encode_png_line((unsigned char*)(pixels), stride_bytes, x, y, j, n, best_filter, line_buffer);
|
1118
|
+
filter_type = best_filter;
|
1119
|
+
}
|
1169
1120
|
}
|
1170
|
-
|
1171
|
-
|
1172
|
-
|
1173
|
-
|
1174
|
-
|
1175
|
-
|
1176
|
-
|
1177
|
-
|
1178
|
-
|
1179
|
-
|
1180
|
-
|
1181
|
-
|
1182
|
-
|
1183
|
-
|
1184
|
-
|
1185
|
-
|
1186
|
-
|
1187
|
-
|
1188
|
-
|
1189
|
-
|
1190
|
-
|
1191
|
-
|
1192
|
-
|
1193
|
-
|
1194
|
-
|
1195
|
-
|
1196
|
-
|
1197
|
-
|
1198
|
-
|
1199
|
-
|
1200
|
-
|
1201
|
-
|
1202
|
-
|
1203
|
-
|
1204
|
-
|
1205
|
-
|
1206
|
-
|
1207
|
-
|
1208
|
-
|
1209
|
-
|
1210
|
-
|
1211
|
-
|
1212
|
-
STBIW_ASSERT(o == out + *out_len);
|
1213
|
-
|
1214
|
-
return out;
|
1121
|
+
// when we get here, filter_type contains the filter type, and line_buffer contains the data
|
1122
|
+
filt[j*(x*n+1)] = (unsigned char) filter_type;
|
1123
|
+
STBIW_MEMMOVE(filt+j*(x*n+1)+1, line_buffer, x*n);
|
1124
|
+
}
|
1125
|
+
STBIW_FREE(line_buffer);
|
1126
|
+
zlib = stbi_zlib_compress(filt, y*( x*n+1), &zlen, stbi_write_png_compression_level);
|
1127
|
+
STBIW_FREE(filt);
|
1128
|
+
if (!zlib) return 0;
|
1129
|
+
|
1130
|
+
// each tag requires 12 bytes of overhead
|
1131
|
+
out = (unsigned char *) STBIW_MALLOC(8 + 12+13 + 12+zlen + 12);
|
1132
|
+
if (!out) return 0;
|
1133
|
+
*out_len = 8 + 12+13 + 12+zlen + 12;
|
1134
|
+
|
1135
|
+
o=out;
|
1136
|
+
STBIW_MEMMOVE(o,sig,8); o+= 8;
|
1137
|
+
stbiw__wp32(o, 13); // header length
|
1138
|
+
stbiw__wptag(o, "IHDR");
|
1139
|
+
stbiw__wp32(o, x);
|
1140
|
+
stbiw__wp32(o, y);
|
1141
|
+
*o++ = 8;
|
1142
|
+
*o++ = STBIW_UCHAR(ctype[n]);
|
1143
|
+
*o++ = 0;
|
1144
|
+
*o++ = 0;
|
1145
|
+
*o++ = 0;
|
1146
|
+
stbiw__wpcrc(&o,13);
|
1147
|
+
|
1148
|
+
stbiw__wp32(o, zlen);
|
1149
|
+
stbiw__wptag(o, "IDAT");
|
1150
|
+
STBIW_MEMMOVE(o, zlib, zlen);
|
1151
|
+
o += zlen;
|
1152
|
+
STBIW_FREE(zlib);
|
1153
|
+
stbiw__wpcrc(&o, zlen);
|
1154
|
+
|
1155
|
+
stbiw__wp32(o,0);
|
1156
|
+
stbiw__wptag(o, "IEND");
|
1157
|
+
stbiw__wpcrc(&o,0);
|
1158
|
+
|
1159
|
+
STBIW_ASSERT(o == out + *out_len);
|
1160
|
+
|
1161
|
+
return out;
|
1215
1162
|
}
|
1216
1163
|
|
1217
1164
|
#ifndef STBI_WRITE_NO_STDIO
|
1218
|
-
STBIWDEF int stbi_write_png(char const *filename, int x, int y, int comp,
|
1219
|
-
|
1220
|
-
|
1221
|
-
|
1222
|
-
|
1223
|
-
|
1224
|
-
|
1225
|
-
|
1226
|
-
|
1227
|
-
|
1228
|
-
|
1229
|
-
|
1230
|
-
|
1231
|
-
STBIW_FREE(png);
|
1232
|
-
return 0;
|
1233
|
-
}
|
1234
|
-
fwrite(png, 1, len, f);
|
1235
|
-
fclose(f);
|
1236
|
-
STBIW_FREE(png);
|
1237
|
-
return 1;
|
1165
|
+
STBIWDEF int stbi_write_png(char const *filename, int x, int y, int comp, const void *data, int stride_bytes)
|
1166
|
+
{
|
1167
|
+
FILE *f;
|
1168
|
+
int len;
|
1169
|
+
unsigned char *png = stbi_write_png_to_mem((const unsigned char *) data, stride_bytes, x, y, comp, &len);
|
1170
|
+
if (png == NULL) return 0;
|
1171
|
+
|
1172
|
+
f = stbiw__fopen(filename, "wb");
|
1173
|
+
if (!f) { STBIW_FREE(png); return 0; }
|
1174
|
+
fwrite(png, 1, len, f);
|
1175
|
+
fclose(f);
|
1176
|
+
STBIW_FREE(png);
|
1177
|
+
return 1;
|
1238
1178
|
}
|
1239
1179
|
#endif
|
1240
1180
|
|
1241
|
-
STBIWDEF int stbi_write_png_to_func(stbi_write_func *func, void *context, int x,
|
1242
|
-
|
1243
|
-
|
1244
|
-
|
1245
|
-
|
1246
|
-
|
1247
|
-
|
1248
|
-
|
1249
|
-
STBIW_FREE(png);
|
1250
|
-
return 1;
|
1181
|
+
STBIWDEF int stbi_write_png_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data, int stride_bytes)
|
1182
|
+
{
|
1183
|
+
int len;
|
1184
|
+
unsigned char *png = stbi_write_png_to_mem((const unsigned char *) data, stride_bytes, x, y, comp, &len);
|
1185
|
+
if (png == NULL) return 0;
|
1186
|
+
func(context, png, len);
|
1187
|
+
STBIW_FREE(png);
|
1188
|
+
return 1;
|
1251
1189
|
}
|
1252
1190
|
|
1191
|
+
|
1253
1192
|
/* ***************************************************************************
|
1254
1193
|
*
|
1255
1194
|
* JPEG writer
|
1256
1195
|
*
|
1257
1196
|
* This is based on Jon Olick's jo_jpeg.cpp:
|
1258
|
-
* public domain Simple, Minimalistic JPEG writer -
|
1259
|
-
* http://www.jonolick.com/code.html
|
1197
|
+
* public domain Simple, Minimalistic JPEG writer - http://www.jonolick.com/code.html
|
1260
1198
|
*/
|
1261
1199
|
|
1262
|
-
static const unsigned char stbiw__jpg_ZigZag[] = {
|
1263
|
-
|
1264
|
-
|
1265
|
-
|
1266
|
-
|
1267
|
-
|
1268
|
-
|
1269
|
-
|
1270
|
-
|
1271
|
-
|
1272
|
-
|
1273
|
-
|
1274
|
-
|
1275
|
-
|
1276
|
-
|
1277
|
-
|
1278
|
-
|
1279
|
-
|
1280
|
-
bitCnt -= 8;
|
1281
|
-
}
|
1282
|
-
*bitBufP = bitBuf;
|
1283
|
-
*bitCntP = bitCnt;
|
1200
|
+
static const unsigned char stbiw__jpg_ZigZag[] = { 0,1,5,6,14,15,27,28,2,4,7,13,16,26,29,42,3,8,12,17,25,30,41,43,9,11,18,
|
1201
|
+
24,31,40,44,53,10,19,23,32,39,45,52,54,20,22,33,38,46,51,55,60,21,34,37,47,50,56,59,61,35,36,48,49,57,58,62,63 };
|
1202
|
+
|
1203
|
+
static void stbiw__jpg_writeBits(stbi__write_context *s, int *bitBufP, int *bitCntP, const unsigned short *bs) {
|
1204
|
+
int bitBuf = *bitBufP, bitCnt = *bitCntP;
|
1205
|
+
bitCnt += bs[1];
|
1206
|
+
bitBuf |= bs[0] << (24 - bitCnt);
|
1207
|
+
while(bitCnt >= 8) {
|
1208
|
+
unsigned char c = (bitBuf >> 16) & 255;
|
1209
|
+
stbiw__putc(s, c);
|
1210
|
+
if(c == 255) {
|
1211
|
+
stbiw__putc(s, 0);
|
1212
|
+
}
|
1213
|
+
bitBuf <<= 8;
|
1214
|
+
bitCnt -= 8;
|
1215
|
+
}
|
1216
|
+
*bitBufP = bitBuf;
|
1217
|
+
*bitCntP = bitCnt;
|
1284
1218
|
}
|
1285
1219
|
|
1286
|
-
static void stbiw__jpg_DCT(float *d0p, float *d1p, float *d2p, float *d3p,
|
1287
|
-
|
1288
|
-
|
1289
|
-
|
1290
|
-
|
1291
|
-
|
1292
|
-
|
1293
|
-
|
1294
|
-
|
1295
|
-
|
1296
|
-
|
1297
|
-
|
1298
|
-
|
1299
|
-
|
1300
|
-
|
1301
|
-
|
1302
|
-
|
1303
|
-
|
1304
|
-
|
1305
|
-
|
1306
|
-
|
1307
|
-
|
1308
|
-
|
1309
|
-
|
1310
|
-
|
1311
|
-
|
1312
|
-
|
1313
|
-
|
1314
|
-
|
1315
|
-
|
1316
|
-
|
1317
|
-
|
1318
|
-
|
1319
|
-
|
1320
|
-
|
1321
|
-
|
1322
|
-
|
1323
|
-
|
1324
|
-
|
1325
|
-
|
1326
|
-
|
1327
|
-
|
1328
|
-
|
1329
|
-
|
1330
|
-
|
1331
|
-
*
|
1332
|
-
|
1333
|
-
*d0p = d0;
|
1334
|
-
*d2p = d2;
|
1335
|
-
*d4p = d4;
|
1336
|
-
*d6p = d6;
|
1220
|
+
static void stbiw__jpg_DCT(float *d0p, float *d1p, float *d2p, float *d3p, float *d4p, float *d5p, float *d6p, float *d7p) {
|
1221
|
+
float d0 = *d0p, d1 = *d1p, d2 = *d2p, d3 = *d3p, d4 = *d4p, d5 = *d5p, d6 = *d6p, d7 = *d7p;
|
1222
|
+
float z1, z2, z3, z4, z5, z11, z13;
|
1223
|
+
|
1224
|
+
float tmp0 = d0 + d7;
|
1225
|
+
float tmp7 = d0 - d7;
|
1226
|
+
float tmp1 = d1 + d6;
|
1227
|
+
float tmp6 = d1 - d6;
|
1228
|
+
float tmp2 = d2 + d5;
|
1229
|
+
float tmp5 = d2 - d5;
|
1230
|
+
float tmp3 = d3 + d4;
|
1231
|
+
float tmp4 = d3 - d4;
|
1232
|
+
|
1233
|
+
// Even part
|
1234
|
+
float tmp10 = tmp0 + tmp3; // phase 2
|
1235
|
+
float tmp13 = tmp0 - tmp3;
|
1236
|
+
float tmp11 = tmp1 + tmp2;
|
1237
|
+
float tmp12 = tmp1 - tmp2;
|
1238
|
+
|
1239
|
+
d0 = tmp10 + tmp11; // phase 3
|
1240
|
+
d4 = tmp10 - tmp11;
|
1241
|
+
|
1242
|
+
z1 = (tmp12 + tmp13) * 0.707106781f; // c4
|
1243
|
+
d2 = tmp13 + z1; // phase 5
|
1244
|
+
d6 = tmp13 - z1;
|
1245
|
+
|
1246
|
+
// Odd part
|
1247
|
+
tmp10 = tmp4 + tmp5; // phase 2
|
1248
|
+
tmp11 = tmp5 + tmp6;
|
1249
|
+
tmp12 = tmp6 + tmp7;
|
1250
|
+
|
1251
|
+
// The rotator is modified from fig 4-8 to avoid extra negations.
|
1252
|
+
z5 = (tmp10 - tmp12) * 0.382683433f; // c6
|
1253
|
+
z2 = tmp10 * 0.541196100f + z5; // c2-c6
|
1254
|
+
z4 = tmp12 * 1.306562965f + z5; // c2+c6
|
1255
|
+
z3 = tmp11 * 0.707106781f; // c4
|
1256
|
+
|
1257
|
+
z11 = tmp7 + z3; // phase 5
|
1258
|
+
z13 = tmp7 - z3;
|
1259
|
+
|
1260
|
+
*d5p = z13 + z2; // phase 6
|
1261
|
+
*d3p = z13 - z2;
|
1262
|
+
*d1p = z11 + z4;
|
1263
|
+
*d7p = z11 - z4;
|
1264
|
+
|
1265
|
+
*d0p = d0; *d2p = d2; *d4p = d4; *d6p = d6;
|
1337
1266
|
}
|
1338
1267
|
|
1339
1268
|
static void stbiw__jpg_calcBits(int val, unsigned short bits[2]) {
|
1340
|
-
|
1341
|
-
|
1342
|
-
|
1343
|
-
|
1344
|
-
|
1345
|
-
|
1346
|
-
|
1269
|
+
int tmp1 = val < 0 ? -val : val;
|
1270
|
+
val = val < 0 ? val-1 : val;
|
1271
|
+
bits[1] = 1;
|
1272
|
+
while(tmp1 >>= 1) {
|
1273
|
+
++bits[1];
|
1274
|
+
}
|
1275
|
+
bits[0] = val & ((1<<bits[1])-1);
|
1347
1276
|
}
|
1348
1277
|
|
1349
|
-
static int stbiw__jpg_processDU(stbi__write_context *s, int *bitBuf,
|
1350
|
-
|
1351
|
-
|
1352
|
-
|
1353
|
-
|
1354
|
-
|
1355
|
-
|
1356
|
-
|
1357
|
-
|
1358
|
-
|
1359
|
-
|
1360
|
-
|
1361
|
-
|
1362
|
-
|
1363
|
-
|
1364
|
-
|
1365
|
-
|
1366
|
-
|
1367
|
-
|
1368
|
-
|
1369
|
-
|
1370
|
-
|
1371
|
-
|
1372
|
-
|
1373
|
-
|
1374
|
-
|
1375
|
-
|
1376
|
-
|
1377
|
-
|
1378
|
-
|
1379
|
-
|
1380
|
-
|
1381
|
-
|
1382
|
-
|
1383
|
-
|
1384
|
-
|
1385
|
-
|
1386
|
-
|
1387
|
-
|
1388
|
-
|
1389
|
-
|
1390
|
-
|
1391
|
-
|
1392
|
-
|
1393
|
-
|
1394
|
-
|
1395
|
-
|
1396
|
-
|
1397
|
-
|
1398
|
-
|
1399
|
-
|
1400
|
-
|
1401
|
-
|
1402
|
-
|
1403
|
-
|
1404
|
-
|
1405
|
-
|
1406
|
-
|
1407
|
-
|
1408
|
-
|
1409
|
-
|
1410
|
-
|
1411
|
-
|
1412
|
-
stbiw__jpg_calcBits(DU[i], bits);
|
1413
|
-
stbiw__jpg_writeBits(s, bitBuf, bitCnt, HTAC[(nrzeroes << 4) + bits[1]]);
|
1414
|
-
stbiw__jpg_writeBits(s, bitBuf, bitCnt, bits);
|
1415
|
-
}
|
1416
|
-
if (end0pos != 63) {
|
1417
|
-
stbiw__jpg_writeBits(s, bitBuf, bitCnt, EOB);
|
1418
|
-
}
|
1419
|
-
return DU[0];
|
1278
|
+
static int stbiw__jpg_processDU(stbi__write_context *s, int *bitBuf, int *bitCnt, float *CDU, float *fdtbl, int DC, const unsigned short HTDC[256][2], const unsigned short HTAC[256][2]) {
|
1279
|
+
const unsigned short EOB[2] = { HTAC[0x00][0], HTAC[0x00][1] };
|
1280
|
+
const unsigned short M16zeroes[2] = { HTAC[0xF0][0], HTAC[0xF0][1] };
|
1281
|
+
int dataOff, i, diff, end0pos;
|
1282
|
+
int DU[64];
|
1283
|
+
|
1284
|
+
// DCT rows
|
1285
|
+
for(dataOff=0; dataOff<64; dataOff+=8) {
|
1286
|
+
stbiw__jpg_DCT(&CDU[dataOff], &CDU[dataOff+1], &CDU[dataOff+2], &CDU[dataOff+3], &CDU[dataOff+4], &CDU[dataOff+5], &CDU[dataOff+6], &CDU[dataOff+7]);
|
1287
|
+
}
|
1288
|
+
// DCT columns
|
1289
|
+
for(dataOff=0; dataOff<8; ++dataOff) {
|
1290
|
+
stbiw__jpg_DCT(&CDU[dataOff], &CDU[dataOff+8], &CDU[dataOff+16], &CDU[dataOff+24], &CDU[dataOff+32], &CDU[dataOff+40], &CDU[dataOff+48], &CDU[dataOff+56]);
|
1291
|
+
}
|
1292
|
+
// Quantize/descale/zigzag the coefficients
|
1293
|
+
for(i=0; i<64; ++i) {
|
1294
|
+
float v = CDU[i]*fdtbl[i];
|
1295
|
+
// DU[stbiw__jpg_ZigZag[i]] = (int)(v < 0 ? ceilf(v - 0.5f) : floorf(v + 0.5f));
|
1296
|
+
// ceilf() and floorf() are C99, not C89, but I /think/ they're not needed here anyway?
|
1297
|
+
DU[stbiw__jpg_ZigZag[i]] = (int)(v < 0 ? v - 0.5f : v + 0.5f);
|
1298
|
+
}
|
1299
|
+
|
1300
|
+
// Encode DC
|
1301
|
+
diff = DU[0] - DC;
|
1302
|
+
if (diff == 0) {
|
1303
|
+
stbiw__jpg_writeBits(s, bitBuf, bitCnt, HTDC[0]);
|
1304
|
+
} else {
|
1305
|
+
unsigned short bits[2];
|
1306
|
+
stbiw__jpg_calcBits(diff, bits);
|
1307
|
+
stbiw__jpg_writeBits(s, bitBuf, bitCnt, HTDC[bits[1]]);
|
1308
|
+
stbiw__jpg_writeBits(s, bitBuf, bitCnt, bits);
|
1309
|
+
}
|
1310
|
+
// Encode ACs
|
1311
|
+
end0pos = 63;
|
1312
|
+
for(; (end0pos>0)&&(DU[end0pos]==0); --end0pos) {
|
1313
|
+
}
|
1314
|
+
// end0pos = first element in reverse order !=0
|
1315
|
+
if(end0pos == 0) {
|
1316
|
+
stbiw__jpg_writeBits(s, bitBuf, bitCnt, EOB);
|
1317
|
+
return DU[0];
|
1318
|
+
}
|
1319
|
+
for(i = 1; i <= end0pos; ++i) {
|
1320
|
+
int startpos = i;
|
1321
|
+
int nrzeroes;
|
1322
|
+
unsigned short bits[2];
|
1323
|
+
for (; DU[i]==0 && i<=end0pos; ++i) {
|
1324
|
+
}
|
1325
|
+
nrzeroes = i-startpos;
|
1326
|
+
if ( nrzeroes >= 16 ) {
|
1327
|
+
int lng = nrzeroes>>4;
|
1328
|
+
int nrmarker;
|
1329
|
+
for (nrmarker=1; nrmarker <= lng; ++nrmarker)
|
1330
|
+
stbiw__jpg_writeBits(s, bitBuf, bitCnt, M16zeroes);
|
1331
|
+
nrzeroes &= 15;
|
1332
|
+
}
|
1333
|
+
stbiw__jpg_calcBits(DU[i], bits);
|
1334
|
+
stbiw__jpg_writeBits(s, bitBuf, bitCnt, HTAC[(nrzeroes<<4)+bits[1]]);
|
1335
|
+
stbiw__jpg_writeBits(s, bitBuf, bitCnt, bits);
|
1336
|
+
}
|
1337
|
+
if(end0pos != 63) {
|
1338
|
+
stbiw__jpg_writeBits(s, bitBuf, bitCnt, EOB);
|
1339
|
+
}
|
1340
|
+
return DU[0];
|
1420
1341
|
}
|
1421
1342
|
|
1422
|
-
static int stbi_write_jpg_core(stbi__write_context *s, int width, int height,
|
1423
|
-
|
1424
|
-
|
1425
|
-
|
1426
|
-
|
1427
|
-
|
1428
|
-
|
1429
|
-
|
1430
|
-
|
1431
|
-
|
1432
|
-
|
1433
|
-
|
1434
|
-
|
1435
|
-
|
1436
|
-
|
1437
|
-
|
1438
|
-
|
1439
|
-
|
1440
|
-
|
1441
|
-
|
1442
|
-
|
1443
|
-
|
1444
|
-
|
1445
|
-
|
1446
|
-
|
1447
|
-
|
1448
|
-
|
1449
|
-
|
1450
|
-
|
1451
|
-
|
1452
|
-
|
1453
|
-
|
1454
|
-
|
1455
|
-
|
1456
|
-
|
1457
|
-
|
1458
|
-
|
1459
|
-
|
1460
|
-
|
1461
|
-
|
1462
|
-
|
1463
|
-
|
1464
|
-
|
1465
|
-
|
1466
|
-
|
1467
|
-
|
1468
|
-
|
1469
|
-
|
1470
|
-
{
|
1471
|
-
|
1472
|
-
{0,
|
1473
|
-
{
|
1474
|
-
|
1475
|
-
{10,
|
1476
|
-
{
|
1477
|
-
{
|
1478
|
-
{0,
|
1479
|
-
{
|
1480
|
-
{
|
1481
|
-
{0,
|
1482
|
-
{
|
1483
|
-
{
|
1484
|
-
{0,
|
1485
|
-
{
|
1486
|
-
|
1487
|
-
|
1488
|
-
|
1489
|
-
|
1490
|
-
|
1491
|
-
|
1492
|
-
|
1493
|
-
|
1494
|
-
|
1495
|
-
|
1496
|
-
|
1497
|
-
|
1498
|
-
|
1499
|
-
|
1500
|
-
|
1501
|
-
|
1502
|
-
|
1503
|
-
|
1504
|
-
|
1505
|
-
|
1506
|
-
|
1507
|
-
|
1508
|
-
|
1509
|
-
|
1510
|
-
|
1511
|
-
|
1512
|
-
|
1513
|
-
|
1514
|
-
|
1515
|
-
|
1516
|
-
|
1517
|
-
|
1518
|
-
|
1519
|
-
|
1520
|
-
|
1521
|
-
|
1522
|
-
{0,
|
1523
|
-
{
|
1524
|
-
|
1525
|
-
|
1526
|
-
|
1527
|
-
|
1528
|
-
|
1529
|
-
|
1530
|
-
|
1531
|
-
|
1532
|
-
|
1533
|
-
|
1534
|
-
|
1535
|
-
|
1536
|
-
|
1537
|
-
|
1538
|
-
|
1539
|
-
|
1540
|
-
|
1541
|
-
|
1542
|
-
|
1543
|
-
|
1544
|
-
|
1545
|
-
|
1546
|
-
|
1547
|
-
|
1548
|
-
|
1549
|
-
|
1550
|
-
|
1551
|
-
|
1552
|
-
|
1553
|
-
|
1554
|
-
|
1555
|
-
|
1556
|
-
|
1557
|
-
|
1558
|
-
|
1559
|
-
|
1560
|
-
|
1561
|
-
|
1562
|
-
|
1563
|
-
|
1564
|
-
|
1565
|
-
|
1566
|
-
|
1567
|
-
|
1568
|
-
|
1569
|
-
|
1570
|
-
|
1571
|
-
|
1572
|
-
|
1573
|
-
{16352, 14}, {65517, 16}, {65518, 16}, {65519, 16}, {65520, 16},
|
1574
|
-
{65521, 16}, {65522, 16}, {65523, 16}, {65524, 16}, {65525, 16},
|
1575
|
-
{0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0},
|
1576
|
-
{1018, 10}, {32707, 15}, {65526, 16}, {65527, 16}, {65528, 16},
|
1577
|
-
{65529, 16}, {65530, 16}, {65531, 16}, {65532, 16}, {65533, 16},
|
1578
|
-
{65534, 16}, {0, 0}, {0, 0}, {0, 0}, {0, 0},
|
1579
|
-
{0, 0}};
|
1580
|
-
static const int YQT[] = {
|
1581
|
-
16, 11, 10, 16, 24, 40, 51, 61, 12, 12, 14, 19, 26, 58, 60, 55,
|
1582
|
-
14, 13, 16, 24, 40, 57, 69, 56, 14, 17, 22, 29, 51, 87, 80, 62,
|
1583
|
-
18, 22, 37, 56, 68, 109, 103, 77, 24, 35, 55, 64, 81, 104, 113, 92,
|
1584
|
-
49, 64, 78, 87, 103, 121, 120, 101, 72, 92, 95, 98, 112, 100, 103, 99};
|
1585
|
-
static const int UVQT[] = {17, 18, 24, 47, 99, 99, 99, 99, 18, 21, 26, 66, 99,
|
1586
|
-
99, 99, 99, 24, 26, 56, 99, 99, 99, 99, 99, 47, 66,
|
1587
|
-
99, 99, 99, 99, 99, 99, 99, 99, 99, 99, 99, 99, 99,
|
1588
|
-
99, 99, 99, 99, 99, 99, 99, 99, 99, 99, 99, 99, 99,
|
1589
|
-
99, 99, 99, 99, 99, 99, 99, 99, 99, 99, 99, 99};
|
1590
|
-
static const float aasf[] = {
|
1591
|
-
1.0f * 2.828427125f, 1.387039845f * 2.828427125f,
|
1592
|
-
1.306562965f * 2.828427125f, 1.175875602f * 2.828427125f,
|
1593
|
-
1.0f * 2.828427125f, 0.785694958f * 2.828427125f,
|
1594
|
-
0.541196100f * 2.828427125f, 0.275899379f * 2.828427125f};
|
1595
|
-
|
1596
|
-
int row, col, i, k;
|
1597
|
-
float fdtbl_Y[64], fdtbl_UV[64];
|
1598
|
-
unsigned char YTable[64], UVTable[64];
|
1599
|
-
|
1600
|
-
if (!data || !width || !height || comp > 4 || comp < 1) {
|
1601
|
-
return 0;
|
1602
|
-
}
|
1603
|
-
|
1604
|
-
quality = quality ? quality : 90;
|
1605
|
-
quality = quality < 1 ? 1 : quality > 100 ? 100 : quality;
|
1606
|
-
quality = quality < 50 ? 5000 / quality : 200 - quality * 2;
|
1607
|
-
|
1608
|
-
for (i = 0; i < 64; ++i) {
|
1609
|
-
int uvti, yti = (YQT[i] * quality + 50) / 100;
|
1610
|
-
YTable[stbiw__jpg_ZigZag[i]] =
|
1611
|
-
(unsigned char)(yti < 1 ? 1 : yti > 255 ? 255 : yti);
|
1612
|
-
uvti = (UVQT[i] * quality + 50) / 100;
|
1613
|
-
UVTable[stbiw__jpg_ZigZag[i]] =
|
1614
|
-
(unsigned char)(uvti < 1 ? 1 : uvti > 255 ? 255 : uvti);
|
1615
|
-
}
|
1616
|
-
|
1617
|
-
for (row = 0, k = 0; row < 8; ++row) {
|
1618
|
-
for (col = 0; col < 8; ++col, ++k) {
|
1619
|
-
fdtbl_Y[k] = 1 / (YTable[stbiw__jpg_ZigZag[k]] * aasf[row] * aasf[col]);
|
1620
|
-
fdtbl_UV[k] = 1 / (UVTable[stbiw__jpg_ZigZag[k]] * aasf[row] * aasf[col]);
|
1621
|
-
}
|
1622
|
-
}
|
1623
|
-
|
1624
|
-
// Write Headers
|
1625
|
-
{
|
1626
|
-
static const unsigned char head0[] = {
|
1627
|
-
0xFF, 0xD8, 0xFF, 0xE0, 0, 0x10, 'J', 'F', 'I', 'F', 0, 1, 1,
|
1628
|
-
0, 0, 1, 0, 1, 0, 0, 0xFF, 0xDB, 0, 0x84, 0};
|
1629
|
-
static const unsigned char head2[] = {0xFF, 0xDA, 0, 0xC, 3, 1, 0,
|
1630
|
-
2, 0x11, 3, 0x11, 0, 0x3F, 0};
|
1631
|
-
const unsigned char head1[] = {0xFF,
|
1632
|
-
0xC0,
|
1633
|
-
0,
|
1634
|
-
0x11,
|
1635
|
-
8,
|
1636
|
-
(unsigned char)(height >> 8),
|
1637
|
-
STBIW_UCHAR(height),
|
1638
|
-
(unsigned char)(width >> 8),
|
1639
|
-
STBIW_UCHAR(width),
|
1640
|
-
3,
|
1641
|
-
1,
|
1642
|
-
0x11,
|
1643
|
-
0,
|
1644
|
-
2,
|
1645
|
-
0x11,
|
1646
|
-
1,
|
1647
|
-
3,
|
1648
|
-
0x11,
|
1649
|
-
1,
|
1650
|
-
0xFF,
|
1651
|
-
0xC4,
|
1652
|
-
0x01,
|
1653
|
-
0xA2,
|
1654
|
-
0};
|
1655
|
-
s->func(s->context, (void *)head0, sizeof(head0));
|
1656
|
-
s->func(s->context, (void *)YTable, sizeof(YTable));
|
1657
|
-
stbiw__putc(s, 1);
|
1658
|
-
s->func(s->context, UVTable, sizeof(UVTable));
|
1659
|
-
s->func(s->context, (void *)head1, sizeof(head1));
|
1660
|
-
s->func(s->context, (void *)(std_dc_luminance_nrcodes + 1),
|
1661
|
-
sizeof(std_dc_luminance_nrcodes) - 1);
|
1662
|
-
s->func(s->context, (void *)std_dc_luminance_values,
|
1663
|
-
sizeof(std_dc_luminance_values));
|
1664
|
-
stbiw__putc(s, 0x10); // HTYACinfo
|
1665
|
-
s->func(s->context, (void *)(std_ac_luminance_nrcodes + 1),
|
1666
|
-
sizeof(std_ac_luminance_nrcodes) - 1);
|
1667
|
-
s->func(s->context, (void *)std_ac_luminance_values,
|
1668
|
-
sizeof(std_ac_luminance_values));
|
1669
|
-
stbiw__putc(s, 1); // HTUDCinfo
|
1670
|
-
s->func(s->context, (void *)(std_dc_chrominance_nrcodes + 1),
|
1671
|
-
sizeof(std_dc_chrominance_nrcodes) - 1);
|
1672
|
-
s->func(s->context, (void *)std_dc_chrominance_values,
|
1673
|
-
sizeof(std_dc_chrominance_values));
|
1674
|
-
stbiw__putc(s, 0x11); // HTUACinfo
|
1675
|
-
s->func(s->context, (void *)(std_ac_chrominance_nrcodes + 1),
|
1676
|
-
sizeof(std_ac_chrominance_nrcodes) - 1);
|
1677
|
-
s->func(s->context, (void *)std_ac_chrominance_values,
|
1678
|
-
sizeof(std_ac_chrominance_values));
|
1679
|
-
s->func(s->context, (void *)head2, sizeof(head2));
|
1680
|
-
}
|
1681
|
-
|
1682
|
-
// Encode 8x8 macroblocks
|
1683
|
-
{
|
1684
|
-
static const unsigned short fillBits[] = {0x7F, 7};
|
1685
|
-
const unsigned char *imageData = (const unsigned char *)data;
|
1686
|
-
int DCY = 0, DCU = 0, DCV = 0;
|
1687
|
-
int bitBuf = 0, bitCnt = 0;
|
1688
|
-
// comp == 2 is grey+alpha (alpha is ignored)
|
1689
|
-
int ofsG = comp > 2 ? 1 : 0, ofsB = comp > 2 ? 2 : 0;
|
1690
|
-
int x, y, pos;
|
1691
|
-
for (y = 0; y < height; y += 8) {
|
1692
|
-
for (x = 0; x < width; x += 8) {
|
1693
|
-
float YDU[64], UDU[64], VDU[64];
|
1694
|
-
for (row = y, pos = 0; row < y + 8; ++row) {
|
1695
|
-
for (col = x; col < x + 8; ++col, ++pos) {
|
1696
|
-
int p = (stbi__flip_vertically_on_write ? height - 1 - row : row) *
|
1697
|
-
width * comp +
|
1698
|
-
col * comp;
|
1699
|
-
float r, g, b;
|
1700
|
-
if (row >= height) {
|
1701
|
-
p -= width * comp * (row + 1 - height);
|
1702
|
-
}
|
1703
|
-
if (col >= width) {
|
1704
|
-
p -= comp * (col + 1 - width);
|
1343
|
+
static int stbi_write_jpg_core(stbi__write_context *s, int width, int height, int comp, const void* data, int quality) {
|
1344
|
+
// Constants that don't pollute global namespace
|
1345
|
+
static const unsigned char std_dc_luminance_nrcodes[] = {0,0,1,5,1,1,1,1,1,1,0,0,0,0,0,0,0};
|
1346
|
+
static const unsigned char std_dc_luminance_values[] = {0,1,2,3,4,5,6,7,8,9,10,11};
|
1347
|
+
static const unsigned char std_ac_luminance_nrcodes[] = {0,0,2,1,3,3,2,4,3,5,5,4,4,0,0,1,0x7d};
|
1348
|
+
static const unsigned char std_ac_luminance_values[] = {
|
1349
|
+
0x01,0x02,0x03,0x00,0x04,0x11,0x05,0x12,0x21,0x31,0x41,0x06,0x13,0x51,0x61,0x07,0x22,0x71,0x14,0x32,0x81,0x91,0xa1,0x08,
|
1350
|
+
0x23,0x42,0xb1,0xc1,0x15,0x52,0xd1,0xf0,0x24,0x33,0x62,0x72,0x82,0x09,0x0a,0x16,0x17,0x18,0x19,0x1a,0x25,0x26,0x27,0x28,
|
1351
|
+
0x29,0x2a,0x34,0x35,0x36,0x37,0x38,0x39,0x3a,0x43,0x44,0x45,0x46,0x47,0x48,0x49,0x4a,0x53,0x54,0x55,0x56,0x57,0x58,0x59,
|
1352
|
+
0x5a,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6a,0x73,0x74,0x75,0x76,0x77,0x78,0x79,0x7a,0x83,0x84,0x85,0x86,0x87,0x88,0x89,
|
1353
|
+
0x8a,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9a,0xa2,0xa3,0xa4,0xa5,0xa6,0xa7,0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,0xb5,0xb6,
|
1354
|
+
0xb7,0xb8,0xb9,0xba,0xc2,0xc3,0xc4,0xc5,0xc6,0xc7,0xc8,0xc9,0xca,0xd2,0xd3,0xd4,0xd5,0xd6,0xd7,0xd8,0xd9,0xda,0xe1,0xe2,
|
1355
|
+
0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,0xea,0xf1,0xf2,0xf3,0xf4,0xf5,0xf6,0xf7,0xf8,0xf9,0xfa
|
1356
|
+
};
|
1357
|
+
static const unsigned char std_dc_chrominance_nrcodes[] = {0,0,3,1,1,1,1,1,1,1,1,1,0,0,0,0,0};
|
1358
|
+
static const unsigned char std_dc_chrominance_values[] = {0,1,2,3,4,5,6,7,8,9,10,11};
|
1359
|
+
static const unsigned char std_ac_chrominance_nrcodes[] = {0,0,2,1,2,4,4,3,4,7,5,4,4,0,1,2,0x77};
|
1360
|
+
static const unsigned char std_ac_chrominance_values[] = {
|
1361
|
+
0x00,0x01,0x02,0x03,0x11,0x04,0x05,0x21,0x31,0x06,0x12,0x41,0x51,0x07,0x61,0x71,0x13,0x22,0x32,0x81,0x08,0x14,0x42,0x91,
|
1362
|
+
0xa1,0xb1,0xc1,0x09,0x23,0x33,0x52,0xf0,0x15,0x62,0x72,0xd1,0x0a,0x16,0x24,0x34,0xe1,0x25,0xf1,0x17,0x18,0x19,0x1a,0x26,
|
1363
|
+
0x27,0x28,0x29,0x2a,0x35,0x36,0x37,0x38,0x39,0x3a,0x43,0x44,0x45,0x46,0x47,0x48,0x49,0x4a,0x53,0x54,0x55,0x56,0x57,0x58,
|
1364
|
+
0x59,0x5a,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6a,0x73,0x74,0x75,0x76,0x77,0x78,0x79,0x7a,0x82,0x83,0x84,0x85,0x86,0x87,
|
1365
|
+
0x88,0x89,0x8a,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9a,0xa2,0xa3,0xa4,0xa5,0xa6,0xa7,0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,
|
1366
|
+
0xb5,0xb6,0xb7,0xb8,0xb9,0xba,0xc2,0xc3,0xc4,0xc5,0xc6,0xc7,0xc8,0xc9,0xca,0xd2,0xd3,0xd4,0xd5,0xd6,0xd7,0xd8,0xd9,0xda,
|
1367
|
+
0xe2,0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,0xea,0xf2,0xf3,0xf4,0xf5,0xf6,0xf7,0xf8,0xf9,0xfa
|
1368
|
+
};
|
1369
|
+
// Huffman tables
|
1370
|
+
static const unsigned short YDC_HT[256][2] = { {0,2},{2,3},{3,3},{4,3},{5,3},{6,3},{14,4},{30,5},{62,6},{126,7},{254,8},{510,9}};
|
1371
|
+
static const unsigned short UVDC_HT[256][2] = { {0,2},{1,2},{2,2},{6,3},{14,4},{30,5},{62,6},{126,7},{254,8},{510,9},{1022,10},{2046,11}};
|
1372
|
+
static const unsigned short YAC_HT[256][2] = {
|
1373
|
+
{10,4},{0,2},{1,2},{4,3},{11,4},{26,5},{120,7},{248,8},{1014,10},{65410,16},{65411,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1374
|
+
{12,4},{27,5},{121,7},{502,9},{2038,11},{65412,16},{65413,16},{65414,16},{65415,16},{65416,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1375
|
+
{28,5},{249,8},{1015,10},{4084,12},{65417,16},{65418,16},{65419,16},{65420,16},{65421,16},{65422,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1376
|
+
{58,6},{503,9},{4085,12},{65423,16},{65424,16},{65425,16},{65426,16},{65427,16},{65428,16},{65429,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1377
|
+
{59,6},{1016,10},{65430,16},{65431,16},{65432,16},{65433,16},{65434,16},{65435,16},{65436,16},{65437,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1378
|
+
{122,7},{2039,11},{65438,16},{65439,16},{65440,16},{65441,16},{65442,16},{65443,16},{65444,16},{65445,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1379
|
+
{123,7},{4086,12},{65446,16},{65447,16},{65448,16},{65449,16},{65450,16},{65451,16},{65452,16},{65453,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1380
|
+
{250,8},{4087,12},{65454,16},{65455,16},{65456,16},{65457,16},{65458,16},{65459,16},{65460,16},{65461,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1381
|
+
{504,9},{32704,15},{65462,16},{65463,16},{65464,16},{65465,16},{65466,16},{65467,16},{65468,16},{65469,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1382
|
+
{505,9},{65470,16},{65471,16},{65472,16},{65473,16},{65474,16},{65475,16},{65476,16},{65477,16},{65478,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1383
|
+
{506,9},{65479,16},{65480,16},{65481,16},{65482,16},{65483,16},{65484,16},{65485,16},{65486,16},{65487,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1384
|
+
{1017,10},{65488,16},{65489,16},{65490,16},{65491,16},{65492,16},{65493,16},{65494,16},{65495,16},{65496,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1385
|
+
{1018,10},{65497,16},{65498,16},{65499,16},{65500,16},{65501,16},{65502,16},{65503,16},{65504,16},{65505,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1386
|
+
{2040,11},{65506,16},{65507,16},{65508,16},{65509,16},{65510,16},{65511,16},{65512,16},{65513,16},{65514,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1387
|
+
{65515,16},{65516,16},{65517,16},{65518,16},{65519,16},{65520,16},{65521,16},{65522,16},{65523,16},{65524,16},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1388
|
+
{2041,11},{65525,16},{65526,16},{65527,16},{65528,16},{65529,16},{65530,16},{65531,16},{65532,16},{65533,16},{65534,16},{0,0},{0,0},{0,0},{0,0},{0,0}
|
1389
|
+
};
|
1390
|
+
static const unsigned short UVAC_HT[256][2] = {
|
1391
|
+
{0,2},{1,2},{4,3},{10,4},{24,5},{25,5},{56,6},{120,7},{500,9},{1014,10},{4084,12},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1392
|
+
{11,4},{57,6},{246,8},{501,9},{2038,11},{4085,12},{65416,16},{65417,16},{65418,16},{65419,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1393
|
+
{26,5},{247,8},{1015,10},{4086,12},{32706,15},{65420,16},{65421,16},{65422,16},{65423,16},{65424,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1394
|
+
{27,5},{248,8},{1016,10},{4087,12},{65425,16},{65426,16},{65427,16},{65428,16},{65429,16},{65430,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1395
|
+
{58,6},{502,9},{65431,16},{65432,16},{65433,16},{65434,16},{65435,16},{65436,16},{65437,16},{65438,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1396
|
+
{59,6},{1017,10},{65439,16},{65440,16},{65441,16},{65442,16},{65443,16},{65444,16},{65445,16},{65446,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1397
|
+
{121,7},{2039,11},{65447,16},{65448,16},{65449,16},{65450,16},{65451,16},{65452,16},{65453,16},{65454,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1398
|
+
{122,7},{2040,11},{65455,16},{65456,16},{65457,16},{65458,16},{65459,16},{65460,16},{65461,16},{65462,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1399
|
+
{249,8},{65463,16},{65464,16},{65465,16},{65466,16},{65467,16},{65468,16},{65469,16},{65470,16},{65471,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1400
|
+
{503,9},{65472,16},{65473,16},{65474,16},{65475,16},{65476,16},{65477,16},{65478,16},{65479,16},{65480,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1401
|
+
{504,9},{65481,16},{65482,16},{65483,16},{65484,16},{65485,16},{65486,16},{65487,16},{65488,16},{65489,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1402
|
+
{505,9},{65490,16},{65491,16},{65492,16},{65493,16},{65494,16},{65495,16},{65496,16},{65497,16},{65498,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1403
|
+
{506,9},{65499,16},{65500,16},{65501,16},{65502,16},{65503,16},{65504,16},{65505,16},{65506,16},{65507,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1404
|
+
{2041,11},{65508,16},{65509,16},{65510,16},{65511,16},{65512,16},{65513,16},{65514,16},{65515,16},{65516,16},{0,0},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1405
|
+
{16352,14},{65517,16},{65518,16},{65519,16},{65520,16},{65521,16},{65522,16},{65523,16},{65524,16},{65525,16},{0,0},{0,0},{0,0},{0,0},{0,0},
|
1406
|
+
{1018,10},{32707,15},{65526,16},{65527,16},{65528,16},{65529,16},{65530,16},{65531,16},{65532,16},{65533,16},{65534,16},{0,0},{0,0},{0,0},{0,0},{0,0}
|
1407
|
+
};
|
1408
|
+
static const int YQT[] = {16,11,10,16,24,40,51,61,12,12,14,19,26,58,60,55,14,13,16,24,40,57,69,56,14,17,22,29,51,87,80,62,18,22,
|
1409
|
+
37,56,68,109,103,77,24,35,55,64,81,104,113,92,49,64,78,87,103,121,120,101,72,92,95,98,112,100,103,99};
|
1410
|
+
static const int UVQT[] = {17,18,24,47,99,99,99,99,18,21,26,66,99,99,99,99,24,26,56,99,99,99,99,99,47,66,99,99,99,99,99,99,
|
1411
|
+
99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99,99};
|
1412
|
+
static const float aasf[] = { 1.0f * 2.828427125f, 1.387039845f * 2.828427125f, 1.306562965f * 2.828427125f, 1.175875602f * 2.828427125f,
|
1413
|
+
1.0f * 2.828427125f, 0.785694958f * 2.828427125f, 0.541196100f * 2.828427125f, 0.275899379f * 2.828427125f };
|
1414
|
+
|
1415
|
+
int row, col, i, k;
|
1416
|
+
float fdtbl_Y[64], fdtbl_UV[64];
|
1417
|
+
unsigned char YTable[64], UVTable[64];
|
1418
|
+
|
1419
|
+
if(!data || !width || !height || comp > 4 || comp < 1) {
|
1420
|
+
return 0;
|
1421
|
+
}
|
1422
|
+
|
1423
|
+
quality = quality ? quality : 90;
|
1424
|
+
quality = quality < 1 ? 1 : quality > 100 ? 100 : quality;
|
1425
|
+
quality = quality < 50 ? 5000 / quality : 200 - quality * 2;
|
1426
|
+
|
1427
|
+
for(i = 0; i < 64; ++i) {
|
1428
|
+
int uvti, yti = (YQT[i]*quality+50)/100;
|
1429
|
+
YTable[stbiw__jpg_ZigZag[i]] = (unsigned char) (yti < 1 ? 1 : yti > 255 ? 255 : yti);
|
1430
|
+
uvti = (UVQT[i]*quality+50)/100;
|
1431
|
+
UVTable[stbiw__jpg_ZigZag[i]] = (unsigned char) (uvti < 1 ? 1 : uvti > 255 ? 255 : uvti);
|
1432
|
+
}
|
1433
|
+
|
1434
|
+
for(row = 0, k = 0; row < 8; ++row) {
|
1435
|
+
for(col = 0; col < 8; ++col, ++k) {
|
1436
|
+
fdtbl_Y[k] = 1 / (YTable [stbiw__jpg_ZigZag[k]] * aasf[row] * aasf[col]);
|
1437
|
+
fdtbl_UV[k] = 1 / (UVTable[stbiw__jpg_ZigZag[k]] * aasf[row] * aasf[col]);
|
1438
|
+
}
|
1439
|
+
}
|
1440
|
+
|
1441
|
+
// Write Headers
|
1442
|
+
{
|
1443
|
+
static const unsigned char head0[] = { 0xFF,0xD8,0xFF,0xE0,0,0x10,'J','F','I','F',0,1,1,0,0,1,0,1,0,0,0xFF,0xDB,0,0x84,0 };
|
1444
|
+
static const unsigned char head2[] = { 0xFF,0xDA,0,0xC,3,1,0,2,0x11,3,0x11,0,0x3F,0 };
|
1445
|
+
const unsigned char head1[] = { 0xFF,0xC0,0,0x11,8,(unsigned char)(height>>8),STBIW_UCHAR(height),(unsigned char)(width>>8),STBIW_UCHAR(width),
|
1446
|
+
3,1,0x11,0,2,0x11,1,3,0x11,1,0xFF,0xC4,0x01,0xA2,0 };
|
1447
|
+
s->func(s->context, (void*)head0, sizeof(head0));
|
1448
|
+
s->func(s->context, (void*)YTable, sizeof(YTable));
|
1449
|
+
stbiw__putc(s, 1);
|
1450
|
+
s->func(s->context, UVTable, sizeof(UVTable));
|
1451
|
+
s->func(s->context, (void*)head1, sizeof(head1));
|
1452
|
+
s->func(s->context, (void*)(std_dc_luminance_nrcodes+1), sizeof(std_dc_luminance_nrcodes)-1);
|
1453
|
+
s->func(s->context, (void*)std_dc_luminance_values, sizeof(std_dc_luminance_values));
|
1454
|
+
stbiw__putc(s, 0x10); // HTYACinfo
|
1455
|
+
s->func(s->context, (void*)(std_ac_luminance_nrcodes+1), sizeof(std_ac_luminance_nrcodes)-1);
|
1456
|
+
s->func(s->context, (void*)std_ac_luminance_values, sizeof(std_ac_luminance_values));
|
1457
|
+
stbiw__putc(s, 1); // HTUDCinfo
|
1458
|
+
s->func(s->context, (void*)(std_dc_chrominance_nrcodes+1), sizeof(std_dc_chrominance_nrcodes)-1);
|
1459
|
+
s->func(s->context, (void*)std_dc_chrominance_values, sizeof(std_dc_chrominance_values));
|
1460
|
+
stbiw__putc(s, 0x11); // HTUACinfo
|
1461
|
+
s->func(s->context, (void*)(std_ac_chrominance_nrcodes+1), sizeof(std_ac_chrominance_nrcodes)-1);
|
1462
|
+
s->func(s->context, (void*)std_ac_chrominance_values, sizeof(std_ac_chrominance_values));
|
1463
|
+
s->func(s->context, (void*)head2, sizeof(head2));
|
1464
|
+
}
|
1465
|
+
|
1466
|
+
// Encode 8x8 macroblocks
|
1467
|
+
{
|
1468
|
+
static const unsigned short fillBits[] = {0x7F, 7};
|
1469
|
+
const unsigned char *imageData = (const unsigned char *)data;
|
1470
|
+
int DCY=0, DCU=0, DCV=0;
|
1471
|
+
int bitBuf=0, bitCnt=0;
|
1472
|
+
// comp == 2 is grey+alpha (alpha is ignored)
|
1473
|
+
int ofsG = comp > 2 ? 1 : 0, ofsB = comp > 2 ? 2 : 0;
|
1474
|
+
int x, y, pos;
|
1475
|
+
for(y = 0; y < height; y += 8) {
|
1476
|
+
for(x = 0; x < width; x += 8) {
|
1477
|
+
float YDU[64], UDU[64], VDU[64];
|
1478
|
+
for(row = y, pos = 0; row < y+8; ++row) {
|
1479
|
+
// row >= height => use last input row
|
1480
|
+
int clamped_row = (row < height) ? row : height - 1;
|
1481
|
+
int base_p = (stbi__flip_vertically_on_write ? (height-1-clamped_row) : clamped_row)*width*comp;
|
1482
|
+
for(col = x; col < x+8; ++col, ++pos) {
|
1483
|
+
float r, g, b;
|
1484
|
+
// if col >= width => use pixel from last input column
|
1485
|
+
int p = base_p + ((col < width) ? col : (width-1))*comp;
|
1486
|
+
|
1487
|
+
r = imageData[p+0];
|
1488
|
+
g = imageData[p+ofsG];
|
1489
|
+
b = imageData[p+ofsB];
|
1490
|
+
YDU[pos]=+0.29900f*r+0.58700f*g+0.11400f*b-128;
|
1491
|
+
UDU[pos]=-0.16874f*r-0.33126f*g+0.50000f*b;
|
1492
|
+
VDU[pos]=+0.50000f*r-0.41869f*g-0.08131f*b;
|
1493
|
+
}
|
1705
1494
|
}
|
1706
1495
|
|
1707
|
-
|
1708
|
-
|
1709
|
-
|
1710
|
-
|
1711
|
-
UDU[pos] = -0.16874f * r - 0.33126f * g + 0.50000f * b;
|
1712
|
-
VDU[pos] = +0.50000f * r - 0.41869f * g - 0.08131f * b;
|
1713
|
-
}
|
1714
|
-
}
|
1715
|
-
|
1716
|
-
DCY = stbiw__jpg_processDU(s, &bitBuf, &bitCnt, YDU, fdtbl_Y, DCY,
|
1717
|
-
YDC_HT, YAC_HT);
|
1718
|
-
DCU = stbiw__jpg_processDU(s, &bitBuf, &bitCnt, UDU, fdtbl_UV, DCU,
|
1719
|
-
UVDC_HT, UVAC_HT);
|
1720
|
-
DCV = stbiw__jpg_processDU(s, &bitBuf, &bitCnt, VDU, fdtbl_UV, DCV,
|
1721
|
-
UVDC_HT, UVAC_HT);
|
1496
|
+
DCY = stbiw__jpg_processDU(s, &bitBuf, &bitCnt, YDU, fdtbl_Y, DCY, YDC_HT, YAC_HT);
|
1497
|
+
DCU = stbiw__jpg_processDU(s, &bitBuf, &bitCnt, UDU, fdtbl_UV, DCU, UVDC_HT, UVAC_HT);
|
1498
|
+
DCV = stbiw__jpg_processDU(s, &bitBuf, &bitCnt, VDU, fdtbl_UV, DCV, UVDC_HT, UVAC_HT);
|
1499
|
+
}
|
1722
1500
|
}
|
1723
|
-
}
|
1724
1501
|
|
1725
|
-
|
1726
|
-
|
1727
|
-
|
1502
|
+
// Do the bit alignment of the EOI marker
|
1503
|
+
stbiw__jpg_writeBits(s, &bitBuf, &bitCnt, fillBits);
|
1504
|
+
}
|
1728
1505
|
|
1729
|
-
|
1730
|
-
|
1731
|
-
|
1506
|
+
// EOI
|
1507
|
+
stbiw__putc(s, 0xFF);
|
1508
|
+
stbiw__putc(s, 0xD9);
|
1732
1509
|
|
1733
|
-
|
1510
|
+
return 1;
|
1734
1511
|
}
|
1735
1512
|
|
1736
|
-
STBIWDEF int stbi_write_jpg_to_func(stbi_write_func *func, void *context, int x,
|
1737
|
-
|
1738
|
-
|
1739
|
-
|
1740
|
-
|
1741
|
-
return stbi_write_jpg_core(&s, x, y, comp, (void *)data, quality);
|
1513
|
+
STBIWDEF int stbi_write_jpg_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data, int quality)
|
1514
|
+
{
|
1515
|
+
stbi__write_context s;
|
1516
|
+
stbi__start_write_callbacks(&s, func, context);
|
1517
|
+
return stbi_write_jpg_core(&s, x, y, comp, (void *) data, quality);
|
1742
1518
|
}
|
1743
1519
|
|
1520
|
+
|
1744
1521
|
#ifndef STBI_WRITE_NO_STDIO
|
1745
|
-
STBIWDEF int stbi_write_jpg(char const *filename, int x, int y, int comp,
|
1746
|
-
|
1747
|
-
|
1748
|
-
|
1749
|
-
|
1750
|
-
|
1751
|
-
|
1752
|
-
|
1753
|
-
|
1522
|
+
STBIWDEF int stbi_write_jpg(char const *filename, int x, int y, int comp, const void *data, int quality)
|
1523
|
+
{
|
1524
|
+
stbi__write_context s;
|
1525
|
+
if (stbi__start_write_file(&s,filename)) {
|
1526
|
+
int r = stbi_write_jpg_core(&s, x, y, comp, data, quality);
|
1527
|
+
stbi__end_write_file(&s);
|
1528
|
+
return r;
|
1529
|
+
} else
|
1530
|
+
return 0;
|
1754
1531
|
}
|
1755
1532
|
#endif
|
1756
1533
|
|
1757
|
-
#endif
|
1534
|
+
#endif // STB_IMAGE_WRITE_IMPLEMENTATION
|
1758
1535
|
|
1759
1536
|
/* Revision history
|
1537
|
+
1.10 (2019-02-07)
|
1538
|
+
support utf8 filenames in Windows; fix warnings and platform ifdefs
|
1760
1539
|
1.09 (2018-02-11)
|
1761
1540
|
fix typo in zlib quality API, improve STB_I_W_STATIC in C++
|
1762
1541
|
1.08 (2018-01-29)
|
1763
|
-
add stbi__flip_vertically_on_write, external zlib, zlib quality,
|
1764
|
-
|
1765
|
-
|
1766
|
-
|
1767
|
-
|
1768
|
-
|
1769
|
-
|
1542
|
+
add stbi__flip_vertically_on_write, external zlib, zlib quality, choose PNG filter
|
1543
|
+
1.07 (2017-07-24)
|
1544
|
+
doc fix
|
1545
|
+
1.06 (2017-07-23)
|
1546
|
+
writing JPEG (using Jon Olick's code)
|
1547
|
+
1.05 ???
|
1548
|
+
1.04 (2017-03-03)
|
1549
|
+
monochrome BMP expansion
|
1550
|
+
1.03 ???
|
1551
|
+
1.02 (2016-04-02)
|
1552
|
+
avoid allocating large structures on the stack
|
1553
|
+
1.01 (2016-01-16)
|
1554
|
+
STBIW_REALLOC_SIZED: support allocators with no realloc support
|
1555
|
+
avoid race-condition in crc initialization
|
1556
|
+
minor compile issues
|
1557
|
+
1.00 (2015-09-14)
|
1558
|
+
installable file IO function
|
1559
|
+
0.99 (2015-09-13)
|
1770
1560
|
warning fixes; TGA rle support
|
1771
1561
|
0.98 (2015-04-08)
|
1772
1562
|
added STBIW_MALLOC, STBIW_ASSERT etc
|
@@ -1794,38 +1584,38 @@ This software is available under 2 licenses -- choose whichever you prefer.
|
|
1794
1584
|
------------------------------------------------------------------------------
|
1795
1585
|
ALTERNATIVE A - MIT License
|
1796
1586
|
Copyright (c) 2017 Sean Barrett
|
1797
|
-
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
1798
|
-
this software and associated documentation files (the "Software"), to deal in
|
1799
|
-
the Software without restriction, including without limitation the rights to
|
1800
|
-
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
|
1801
|
-
of the Software, and to permit persons to whom the Software is furnished to do
|
1587
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
1588
|
+
this software and associated documentation files (the "Software"), to deal in
|
1589
|
+
the Software without restriction, including without limitation the rights to
|
1590
|
+
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
|
1591
|
+
of the Software, and to permit persons to whom the Software is furnished to do
|
1802
1592
|
so, subject to the following conditions:
|
1803
|
-
The above copyright notice and this permission notice shall be included in all
|
1593
|
+
The above copyright notice and this permission notice shall be included in all
|
1804
1594
|
copies or substantial portions of the Software.
|
1805
|
-
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
1806
|
-
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
1807
|
-
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
1808
|
-
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
1809
|
-
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
1810
|
-
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
1595
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
1596
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
1597
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
1598
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
1599
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
1600
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
1811
1601
|
SOFTWARE.
|
1812
1602
|
------------------------------------------------------------------------------
|
1813
1603
|
ALTERNATIVE B - Public Domain (www.unlicense.org)
|
1814
1604
|
This is free and unencumbered software released into the public domain.
|
1815
|
-
Anyone is free to copy, modify, publish, use, compile, sell, or distribute this
|
1816
|
-
software, either in source code form or as a compiled binary, for any purpose,
|
1605
|
+
Anyone is free to copy, modify, publish, use, compile, sell, or distribute this
|
1606
|
+
software, either in source code form or as a compiled binary, for any purpose,
|
1817
1607
|
commercial or non-commercial, and by any means.
|
1818
|
-
In jurisdictions that recognize copyright laws, the author or authors of this
|
1819
|
-
software dedicate any and all copyright interest in the software to the public
|
1820
|
-
domain. We make this dedication for the benefit of the public at large and to
|
1821
|
-
the detriment of our heirs and successors. We intend this dedication to be an
|
1822
|
-
overt act of relinquishment in perpetuity of all present and future rights to
|
1608
|
+
In jurisdictions that recognize copyright laws, the author or authors of this
|
1609
|
+
software dedicate any and all copyright interest in the software to the public
|
1610
|
+
domain. We make this dedication for the benefit of the public at large and to
|
1611
|
+
the detriment of our heirs and successors. We intend this dedication to be an
|
1612
|
+
overt act of relinquishment in perpetuity of all present and future rights to
|
1823
1613
|
this software under copyright law.
|
1824
|
-
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
1825
|
-
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
1826
|
-
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
1827
|
-
AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
1828
|
-
ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
1614
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
1615
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
1616
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
1617
|
+
AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
1618
|
+
ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
1829
1619
|
WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
1830
1620
|
------------------------------------------------------------------------------
|
1831
1621
|
*/
|