open_image 0.0.1

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.
@@ -0,0 +1,1377 @@
1
+ #include "stb_image_write.h"
2
+
3
+ #ifdef _WIN32
4
+ #ifndef _CRT_SECURE_NO_WARNINGS
5
+ #define _CRT_SECURE_NO_WARNINGS
6
+ #endif
7
+ #ifndef _CRT_NONSTDC_NO_DEPRECATE
8
+ #define _CRT_NONSTDC_NO_DEPRECATE
9
+ #endif
10
+ #endif
11
+
12
+ #ifndef STBI_WRITE_NO_STDIO
13
+ #include <stdio.h>
14
+ #endif // STBI_WRITE_NO_STDIO
15
+
16
+ #include <stdarg.h>
17
+ #include <stdlib.h>
18
+ #include <string.h>
19
+ #include <math.h>
20
+
21
+ #if defined(STBIW_MALLOC) && defined(STBIW_FREE) && (defined(STBIW_REALLOC) || defined(STBIW_REALLOC_SIZED))
22
+ // ok
23
+ #elif !defined(STBIW_MALLOC) && !defined(STBIW_FREE) && !defined(STBIW_REALLOC) && !defined(STBIW_REALLOC_SIZED)
24
+ // ok
25
+ #else
26
+ #error "Must define all or none of STBIW_MALLOC, STBIW_FREE, and STBIW_REALLOC (or STBIW_REALLOC_SIZED)."
27
+ #endif
28
+
29
+ #ifndef STBIW_MALLOC
30
+ #define STBIW_MALLOC(sz) malloc(sz)
31
+ #define STBIW_REALLOC(p,newsz) realloc(p,newsz)
32
+ #define STBIW_FREE(p) free(p)
33
+ #endif
34
+
35
+ #ifndef STBIW_REALLOC_SIZED
36
+ #define STBIW_REALLOC_SIZED(p,oldsz,newsz) STBIW_REALLOC(p,newsz)
37
+ #endif
38
+
39
+
40
+ #ifndef STBIW_MEMMOVE
41
+ #define STBIW_MEMMOVE(a,b,sz) memmove(a,b,sz)
42
+ #endif
43
+
44
+
45
+ #ifndef STBIW_ASSERT
46
+ #include <assert.h>
47
+ #define STBIW_ASSERT(x) assert(x)
48
+ #endif
49
+
50
+ #define STBIW_UCHAR(x) (unsigned char) ((x) & 0xff)
51
+
52
+ #ifdef STB_IMAGE_WRITE_STATIC
53
+ static int stbi__flip_vertically_on_write=0;
54
+ static int stbi_write_png_compression_level = 8;
55
+ static int stbi_write_tga_with_rle = 1;
56
+ static int stbi_write_force_png_filter = -1;
57
+ #else
58
+ int stbi_write_png_compression_level = 8;
59
+ int stbi__flip_vertically_on_write=0;
60
+ int stbi_write_tga_with_rle = 1;
61
+ int stbi_write_force_png_filter = -1;
62
+ #endif
63
+
64
+ STBIWDEF void stbi_flip_vertically_on_write(int flag)
65
+ {
66
+ stbi__flip_vertically_on_write = flag;
67
+ }
68
+
69
+ typedef struct
70
+ {
71
+ stbi_write_func *func;
72
+ void *context;
73
+ } stbi__write_context;
74
+
75
+ // initialize a callback-based context
76
+ static void stbi__start_write_callbacks(stbi__write_context *s, stbi_write_func *c, void *context)
77
+ {
78
+ s->func = c;
79
+ s->context = context;
80
+ }
81
+
82
+ #ifndef STBI_WRITE_NO_STDIO
83
+
84
+ static void stbi__stdio_write(void *context, void *data, int size)
85
+ {
86
+ fwrite(data,1,size,(FILE*) context);
87
+ }
88
+
89
+ static int stbi__start_write_file(stbi__write_context *s, const char *filename)
90
+ {
91
+ FILE *f;
92
+ #ifdef STBI_MSC_SECURE_CRT
93
+ if (fopen_s(&f, filename, "wb"))
94
+ f = NULL;
95
+ #else
96
+ f = fopen(filename, "wb");
97
+ #endif
98
+ stbi__start_write_callbacks(s, stbi__stdio_write, (void *) f);
99
+ return f != NULL;
100
+ }
101
+
102
+ static void stbi__end_write_file(stbi__write_context *s)
103
+ {
104
+ fclose((FILE *)s->context);
105
+ }
106
+
107
+ #endif // !STBI_WRITE_NO_STDIO
108
+
109
+ typedef unsigned int stbiw_uint32;
110
+ typedef int stb_image_write_test[sizeof(stbiw_uint32)==4 ? 1 : -1];
111
+
112
+ static void stbiw__writefv(stbi__write_context *s, const char *fmt, va_list v)
113
+ {
114
+ while (*fmt) {
115
+ switch (*fmt++) {
116
+ case ' ': break;
117
+ case '1': { unsigned char x = STBIW_UCHAR(va_arg(v, int));
118
+ s->func(s->context,&x,1);
119
+ break; }
120
+ case '2': { int x = va_arg(v,int);
121
+ unsigned char b[2];
122
+ b[0] = STBIW_UCHAR(x);
123
+ b[1] = STBIW_UCHAR(x>>8);
124
+ s->func(s->context,b,2);
125
+ break; }
126
+ case '4': { stbiw_uint32 x = va_arg(v,int);
127
+ unsigned char b[4];
128
+ b[0]=STBIW_UCHAR(x);
129
+ b[1]=STBIW_UCHAR(x>>8);
130
+ b[2]=STBIW_UCHAR(x>>16);
131
+ b[3]=STBIW_UCHAR(x>>24);
132
+ s->func(s->context,b,4);
133
+ break; }
134
+ default:
135
+ STBIW_ASSERT(0);
136
+ return;
137
+ }
138
+ }
139
+ }
140
+
141
+ static void stbiw__writef(stbi__write_context *s, const char *fmt, ...)
142
+ {
143
+ va_list v;
144
+ va_start(v, fmt);
145
+ stbiw__writefv(s, fmt, v);
146
+ va_end(v);
147
+ }
148
+
149
+ static void stbiw__putc(stbi__write_context *s, unsigned char c)
150
+ {
151
+ s->func(s->context, &c, 1);
152
+ }
153
+
154
+ static void stbiw__write3(stbi__write_context *s, unsigned char a, unsigned char b, unsigned char c)
155
+ {
156
+ unsigned char arr[3];
157
+ arr[0] = a, arr[1] = b, arr[2] = c;
158
+ s->func(s->context, arr, 3);
159
+ }
160
+
161
+ static void stbiw__write_pixel(stbi__write_context *s, int rgb_dir, int comp, int write_alpha, int expand_mono, unsigned char *d)
162
+ {
163
+ unsigned char bg[3] = { 255, 0, 255}, px[3];
164
+ int k;
165
+
166
+ if (write_alpha < 0)
167
+ s->func(s->context, &d[comp - 1], 1);
168
+
169
+ switch (comp) {
170
+ case 2: // 2 pixels = mono + alpha, alpha is written separately, so same as 1-channel case
171
+ case 1:
172
+ if (expand_mono)
173
+ stbiw__write3(s, d[0], d[0], d[0]); // monochrome bmp
174
+ else
175
+ s->func(s->context, d, 1); // monochrome TGA
176
+ break;
177
+ case 4:
178
+ if (!write_alpha) {
179
+ // composite against pink background
180
+ for (k = 0; k < 3; ++k)
181
+ px[k] = bg[k] + ((d[k] - bg[k]) * d[3]) / 255;
182
+ stbiw__write3(s, px[1 - rgb_dir], px[1], px[1 + rgb_dir]);
183
+ break;
184
+ }
185
+ /* FALLTHROUGH */
186
+ case 3:
187
+ stbiw__write3(s, d[1 - rgb_dir], d[1], d[1 + rgb_dir]);
188
+ break;
189
+ }
190
+ if (write_alpha > 0)
191
+ s->func(s->context, &d[comp - 1], 1);
192
+ }
193
+
194
+ 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)
195
+ {
196
+ stbiw_uint32 zero = 0;
197
+ int i,j, j_end;
198
+
199
+ if (y <= 0)
200
+ return;
201
+
202
+ if (stbi__flip_vertically_on_write)
203
+ vdir *= -1;
204
+
205
+ if (vdir < 0)
206
+ j_end = -1, j = y-1;
207
+ else
208
+ j_end = y, j = 0;
209
+
210
+ for (; j != j_end; j += vdir) {
211
+ for (i=0; i < x; ++i) {
212
+ unsigned char *d = (unsigned char *) data + (j*x+i)*comp;
213
+ stbiw__write_pixel(s, rgb_dir, comp, write_alpha, expand_mono, d);
214
+ }
215
+ s->func(s->context, &zero, scanline_pad);
216
+ }
217
+ }
218
+
219
+ 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, ...)
220
+ {
221
+ if (y < 0 || x < 0) {
222
+ return 0;
223
+ } else {
224
+ va_list v;
225
+ va_start(v, fmt);
226
+ stbiw__writefv(s, fmt, v);
227
+ va_end(v);
228
+ stbiw__write_pixels(s,rgb_dir,vdir,x,y,comp,data,alpha,pad, expand_mono);
229
+ return 1;
230
+ }
231
+ }
232
+
233
+ static int stbi_write_bmp_core(stbi__write_context *s, int x, int y, int comp, const void *data)
234
+ {
235
+ int pad = (-x*3) & 3;
236
+ return stbiw__outfile(s,-1,-1,x,y,comp,1,(void *) data,0,pad,
237
+ "11 4 22 4" "4 44 22 444444",
238
+ 'B', 'M', 14+40+(x*3+pad)*y, 0,0, 14+40, // file header
239
+ 40, x,y, 1,24, 0,0,0,0,0,0); // bitmap header
240
+ }
241
+
242
+ STBIWDEF int stbi_write_bmp_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data)
243
+ {
244
+ stbi__write_context s;
245
+ stbi__start_write_callbacks(&s, func, context);
246
+ return stbi_write_bmp_core(&s, x, y, comp, data);
247
+ }
248
+
249
+ #ifndef STBI_WRITE_NO_STDIO
250
+ STBIWDEF int stbi_write_bmp(char const *filename, int x, int y, int comp, const void *data)
251
+ {
252
+ stbi__write_context s;
253
+ if (stbi__start_write_file(&s,filename)) {
254
+ int r = stbi_write_bmp_core(&s, x, y, comp, data);
255
+ stbi__end_write_file(&s);
256
+ return r;
257
+ } else
258
+ return 0;
259
+ }
260
+ #endif //!STBI_WRITE_NO_STDIO
261
+
262
+ static int stbi_write_tga_core(stbi__write_context *s, int x, int y, int comp, void *data)
263
+ {
264
+ int has_alpha = (comp == 2 || comp == 4);
265
+ int colorbytes = has_alpha ? comp-1 : comp;
266
+ int format = colorbytes < 2 ? 3 : 2; // 3 color channels (RGB/RGBA) = 2, 1 color channel (Y/YA) = 3
267
+
268
+ if (y < 0 || x < 0)
269
+ return 0;
270
+
271
+ if (!stbi_write_tga_with_rle) {
272
+ return stbiw__outfile(s, -1, -1, x, y, comp, 0, (void *) data, has_alpha, 0,
273
+ "111 221 2222 11", 0, 0, format, 0, 0, 0, 0, 0, x, y, (colorbytes + has_alpha) * 8, has_alpha * 8);
274
+ } else {
275
+ int i,j,k;
276
+ int jend, jdir;
277
+
278
+ 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);
279
+
280
+ if (stbi__flip_vertically_on_write) {
281
+ j = 0;
282
+ jend = y;
283
+ jdir = 1;
284
+ } else {
285
+ j = y-1;
286
+ jend = -1;
287
+ jdir = -1;
288
+ }
289
+ for (; j != jend; j += jdir) {
290
+ unsigned char *row = (unsigned char *) data + j * x * comp;
291
+ int len;
292
+
293
+ for (i = 0; i < x; i += len) {
294
+ unsigned char *begin = row + i * comp;
295
+ int diff = 1;
296
+ len = 1;
297
+
298
+ if (i < x - 1) {
299
+ ++len;
300
+ diff = memcmp(begin, row + (i + 1) * comp, comp);
301
+ if (diff) {
302
+ const unsigned char *prev = begin;
303
+ for (k = i + 2; k < x && len < 128; ++k) {
304
+ if (memcmp(prev, row + k * comp, comp)) {
305
+ prev += comp;
306
+ ++len;
307
+ } else {
308
+ --len;
309
+ break;
310
+ }
311
+ }
312
+ } else {
313
+ for (k = i + 2; k < x && len < 128; ++k) {
314
+ if (!memcmp(begin, row + k * comp, comp)) {
315
+ ++len;
316
+ } else {
317
+ break;
318
+ }
319
+ }
320
+ }
321
+ }
322
+
323
+ if (diff) {
324
+ unsigned char header = STBIW_UCHAR(len - 1);
325
+ s->func(s->context, &header, 1);
326
+ for (k = 0; k < len; ++k) {
327
+ stbiw__write_pixel(s, -1, comp, has_alpha, 0, begin + k * comp);
328
+ }
329
+ } else {
330
+ unsigned char header = STBIW_UCHAR(len - 129);
331
+ s->func(s->context, &header, 1);
332
+ stbiw__write_pixel(s, -1, comp, has_alpha, 0, begin);
333
+ }
334
+ }
335
+ }
336
+ }
337
+ return 1;
338
+ }
339
+
340
+ STBIWDEF int stbi_write_tga_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data)
341
+ {
342
+ stbi__write_context s;
343
+ stbi__start_write_callbacks(&s, func, context);
344
+ return stbi_write_tga_core(&s, x, y, comp, (void *) data);
345
+ }
346
+
347
+ #ifndef STBI_WRITE_NO_STDIO
348
+ STBIWDEF int stbi_write_tga(char const *filename, int x, int y, int comp, const void *data)
349
+ {
350
+ stbi__write_context s;
351
+ if (stbi__start_write_file(&s,filename)) {
352
+ int r = stbi_write_tga_core(&s, x, y, comp, (void *) data);
353
+ stbi__end_write_file(&s);
354
+ return r;
355
+ } else
356
+ return 0;
357
+ }
358
+ #endif
359
+
360
+ // *************************************************************************************************
361
+ // Radiance RGBE HDR writer
362
+ // by Baldur Karlsson
363
+
364
+ #define stbiw__max(a, b) ((a) > (b) ? (a) : (b))
365
+
366
+ void stbiw__linear_to_rgbe(unsigned char *rgbe, float *linear)
367
+ {
368
+ int exponent;
369
+ float maxcomp = stbiw__max(linear[0], stbiw__max(linear[1], linear[2]));
370
+
371
+ if (maxcomp < 1e-32f) {
372
+ rgbe[0] = rgbe[1] = rgbe[2] = rgbe[3] = 0;
373
+ } else {
374
+ float normalize = (float) frexp(maxcomp, &exponent) * 256.0f/maxcomp;
375
+
376
+ rgbe[0] = (unsigned char)(linear[0] * normalize);
377
+ rgbe[1] = (unsigned char)(linear[1] * normalize);
378
+ rgbe[2] = (unsigned char)(linear[2] * normalize);
379
+ rgbe[3] = (unsigned char)(exponent + 128);
380
+ }
381
+ }
382
+
383
+ void stbiw__write_run_data(stbi__write_context *s, int length, unsigned char databyte)
384
+ {
385
+ unsigned char lengthbyte = STBIW_UCHAR(length+128);
386
+ STBIW_ASSERT(length+128 <= 255);
387
+ s->func(s->context, &lengthbyte, 1);
388
+ s->func(s->context, &databyte, 1);
389
+ }
390
+
391
+ void stbiw__write_dump_data(stbi__write_context *s, int length, unsigned char *data)
392
+ {
393
+ unsigned char lengthbyte = STBIW_UCHAR(length);
394
+ STBIW_ASSERT(length <= 128); // inconsistent with spec but consistent with official code
395
+ s->func(s->context, &lengthbyte, 1);
396
+ s->func(s->context, data, length);
397
+ }
398
+
399
+ void stbiw__write_hdr_scanline(stbi__write_context *s, int width, int ncomp, unsigned char *scratch, float *scanline)
400
+ {
401
+ unsigned char scanlineheader[4] = { 2, 2, 0, 0 };
402
+ unsigned char rgbe[4];
403
+ float linear[3];
404
+ int x;
405
+
406
+ scanlineheader[2] = (width&0xff00)>>8;
407
+ scanlineheader[3] = (width&0x00ff);
408
+
409
+ /* skip RLE for images too small or large */
410
+ if (width < 8 || width >= 32768) {
411
+ for (x=0; x < width; x++) {
412
+ switch (ncomp) {
413
+ case 4: /* fallthrough */
414
+ case 3: linear[2] = scanline[x*ncomp + 2];
415
+ linear[1] = scanline[x*ncomp + 1];
416
+ linear[0] = scanline[x*ncomp + 0];
417
+ break;
418
+ default:
419
+ linear[0] = linear[1] = linear[2] = scanline[x*ncomp + 0];
420
+ break;
421
+ }
422
+ stbiw__linear_to_rgbe(rgbe, linear);
423
+ s->func(s->context, rgbe, 4);
424
+ }
425
+ } else {
426
+ int c,r;
427
+ /* encode into scratch buffer */
428
+ for (x=0; x < width; x++) {
429
+ switch(ncomp) {
430
+ case 4: /* fallthrough */
431
+ case 3: linear[2] = scanline[x*ncomp + 2];
432
+ linear[1] = scanline[x*ncomp + 1];
433
+ linear[0] = scanline[x*ncomp + 0];
434
+ break;
435
+ default:
436
+ linear[0] = linear[1] = linear[2] = scanline[x*ncomp + 0];
437
+ break;
438
+ }
439
+ stbiw__linear_to_rgbe(rgbe, linear);
440
+ scratch[x + width*0] = rgbe[0];
441
+ scratch[x + width*1] = rgbe[1];
442
+ scratch[x + width*2] = rgbe[2];
443
+ scratch[x + width*3] = rgbe[3];
444
+ }
445
+
446
+ s->func(s->context, scanlineheader, 4);
447
+
448
+ /* RLE each component separately */
449
+ for (c=0; c < 4; c++) {
450
+ unsigned char *comp = &scratch[width*c];
451
+
452
+ x = 0;
453
+ while (x < width) {
454
+ // find first run
455
+ r = x;
456
+ while (r+2 < width) {
457
+ if (comp[r] == comp[r+1] && comp[r] == comp[r+2])
458
+ break;
459
+ ++r;
460
+ }
461
+ if (r+2 >= width)
462
+ r = width;
463
+ // dump up to first run
464
+ while (x < r) {
465
+ int len = r-x;
466
+ if (len > 128) len = 128;
467
+ stbiw__write_dump_data(s, len, &comp[x]);
468
+ x += len;
469
+ }
470
+ // if there's a run, output it
471
+ if (r+2 < width) { // same test as what we break out of in search loop, so only true if we break'd
472
+ // find next byte after run
473
+ while (r < width && comp[r] == comp[x])
474
+ ++r;
475
+ // output run up to r
476
+ while (x < r) {
477
+ int len = r-x;
478
+ if (len > 127) len = 127;
479
+ stbiw__write_run_data(s, len, comp[x]);
480
+ x += len;
481
+ }
482
+ }
483
+ }
484
+ }
485
+ }
486
+ }
487
+
488
+ static int stbi_write_hdr_core(stbi__write_context *s, int x, int y, int comp, float *data)
489
+ {
490
+ if (y <= 0 || x <= 0 || data == NULL)
491
+ return 0;
492
+ else {
493
+ // Each component is stored separately. Allocate scratch space for full output scanline.
494
+ unsigned char *scratch = (unsigned char *) STBIW_MALLOC(x*4);
495
+ int i, len;
496
+ char buffer[128];
497
+ char header[] = "#?RADIANCE\n# Written by stb_image_write.h\nFORMAT=32-bit_rle_rgbe\n";
498
+ s->func(s->context, header, sizeof(header)-1);
499
+
500
+ #ifdef STBI_MSC_SECURE_CRT
501
+ len = sprintf_s(buffer, "EXPOSURE= 1.0000000000000\n\n-Y %d +X %d\n", y, x);
502
+ #else
503
+ len = sprintf(buffer, "EXPOSURE= 1.0000000000000\n\n-Y %d +X %d\n", y, x);
504
+ #endif
505
+ s->func(s->context, buffer, len);
506
+
507
+ for(i=0; i < y; i++)
508
+ stbiw__write_hdr_scanline(s, x, comp, scratch, data + comp*x*(stbi__flip_vertically_on_write ? y-1-i : i)*x);
509
+ STBIW_FREE(scratch);
510
+ return 1;
511
+ }
512
+ }
513
+
514
+ STBIWDEF int stbi_write_hdr_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const float *data)
515
+ {
516
+ stbi__write_context s;
517
+ stbi__start_write_callbacks(&s, func, context);
518
+ return stbi_write_hdr_core(&s, x, y, comp, (float *) data);
519
+ }
520
+
521
+ #ifndef STBI_WRITE_NO_STDIO
522
+ STBIWDEF int stbi_write_hdr(char const *filename, int x, int y, int comp, const float *data)
523
+ {
524
+ stbi__write_context s;
525
+ if (stbi__start_write_file(&s,filename)) {
526
+ int r = stbi_write_hdr_core(&s, x, y, comp, (float *) data);
527
+ stbi__end_write_file(&s);
528
+ return r;
529
+ } else
530
+ return 0;
531
+ }
532
+ #endif // STBI_WRITE_NO_STDIO
533
+
534
+
535
+ //////////////////////////////////////////////////////////////////////////////
536
+ //
537
+ // PNG writer
538
+ //
539
+
540
+ #ifndef STBIW_ZLIB_COMPRESS
541
+ // stretchy buffer; stbiw__sbpush() == vector<>::push_back() -- stbiw__sbcount() == vector<>::size()
542
+ #define stbiw__sbraw(a) ((int *) (a) - 2)
543
+ #define stbiw__sbm(a) stbiw__sbraw(a)[0]
544
+ #define stbiw__sbn(a) stbiw__sbraw(a)[1]
545
+
546
+ #define stbiw__sbneedgrow(a,n) ((a)==0 || stbiw__sbn(a)+n >= stbiw__sbm(a))
547
+ #define stbiw__sbmaybegrow(a,n) (stbiw__sbneedgrow(a,(n)) ? stbiw__sbgrow(a,n) : 0)
548
+ #define stbiw__sbgrow(a,n) stbiw__sbgrowf((void **) &(a), (n), sizeof(*(a)))
549
+
550
+ #define stbiw__sbpush(a, v) (stbiw__sbmaybegrow(a,1), (a)[stbiw__sbn(a)++] = (v))
551
+ #define stbiw__sbcount(a) ((a) ? stbiw__sbn(a) : 0)
552
+ #define stbiw__sbfree(a) ((a) ? STBIW_FREE(stbiw__sbraw(a)),0 : 0)
553
+
554
+ static void *stbiw__sbgrowf(void **arr, int increment, int itemsize)
555
+ {
556
+ int m = *arr ? 2*stbiw__sbm(*arr)+increment : increment+1;
557
+ void *p = STBIW_REALLOC_SIZED(*arr ? stbiw__sbraw(*arr) : 0, *arr ? (stbiw__sbm(*arr)*itemsize + sizeof(int)*2) : 0, itemsize * m + sizeof(int)*2);
558
+ STBIW_ASSERT(p);
559
+ if (p) {
560
+ if (!*arr) ((int *) p)[1] = 0;
561
+ *arr = (void *) ((int *) p + 2);
562
+ stbiw__sbm(*arr) = m;
563
+ }
564
+ return *arr;
565
+ }
566
+
567
+ static unsigned char *stbiw__zlib_flushf(unsigned char *data, unsigned int *bitbuffer, int *bitcount)
568
+ {
569
+ while (*bitcount >= 8) {
570
+ stbiw__sbpush(data, STBIW_UCHAR(*bitbuffer));
571
+ *bitbuffer >>= 8;
572
+ *bitcount -= 8;
573
+ }
574
+ return data;
575
+ }
576
+
577
+ static int stbiw__zlib_bitrev(int code, int codebits)
578
+ {
579
+ int res=0;
580
+ while (codebits--) {
581
+ res = (res << 1) | (code & 1);
582
+ code >>= 1;
583
+ }
584
+ return res;
585
+ }
586
+
587
+ static unsigned int stbiw__zlib_countm(unsigned char *a, unsigned char *b, int limit)
588
+ {
589
+ int i;
590
+ for (i=0; i < limit && i < 258; ++i)
591
+ if (a[i] != b[i]) break;
592
+ return i;
593
+ }
594
+
595
+ static unsigned int stbiw__zhash(unsigned char *data)
596
+ {
597
+ stbiw_uint32 hash = data[0] + (data[1] << 8) + (data[2] << 16);
598
+ hash ^= hash << 3;
599
+ hash += hash >> 5;
600
+ hash ^= hash << 4;
601
+ hash += hash >> 17;
602
+ hash ^= hash << 25;
603
+ hash += hash >> 6;
604
+ return hash;
605
+ }
606
+
607
+ #define stbiw__zlib_flush() (out = stbiw__zlib_flushf(out, &bitbuf, &bitcount))
608
+ #define stbiw__zlib_add(code,codebits) \
609
+ (bitbuf |= (code) << bitcount, bitcount += (codebits), stbiw__zlib_flush())
610
+ #define stbiw__zlib_huffa(b,c) stbiw__zlib_add(stbiw__zlib_bitrev(b,c),c)
611
+ // default huffman tables
612
+ #define stbiw__zlib_huff1(n) stbiw__zlib_huffa(0x30 + (n), 8)
613
+ #define stbiw__zlib_huff2(n) stbiw__zlib_huffa(0x190 + (n)-144, 9)
614
+ #define stbiw__zlib_huff3(n) stbiw__zlib_huffa(0 + (n)-256,7)
615
+ #define stbiw__zlib_huff4(n) stbiw__zlib_huffa(0xc0 + (n)-280,8)
616
+ #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))
617
+ #define stbiw__zlib_huffb(n) ((n) <= 143 ? stbiw__zlib_huff1(n) : stbiw__zlib_huff2(n))
618
+
619
+ #define stbiw__ZHASH 16384
620
+
621
+ #endif // STBIW_ZLIB_COMPRESS
622
+
623
+ unsigned char * stbi_zlib_compress(unsigned char *data, int data_len, int *out_len, int quality)
624
+ {
625
+ #ifdef STBIW_ZLIB_COMPRESS
626
+ // user provided a zlib compress implementation, use that
627
+ return STBIW_ZLIB_COMPRESS(data, data_len, out_len, quality);
628
+ #else // use builtin
629
+ 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 };
630
+ 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 };
631
+ 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 };
632
+ 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 };
633
+ unsigned int bitbuf=0;
634
+ int i,j, bitcount=0;
635
+ unsigned char *out = NULL;
636
+ unsigned char ***hash_table = (unsigned char***) STBIW_MALLOC(stbiw__ZHASH * sizeof(char**));
637
+ if (hash_table == NULL)
638
+ return NULL;
639
+ if (quality < 5) quality = 5;
640
+
641
+ stbiw__sbpush(out, 0x78); // DEFLATE 32K window
642
+ stbiw__sbpush(out, 0x5e); // FLEVEL = 1
643
+ stbiw__zlib_add(1,1); // BFINAL = 1
644
+ stbiw__zlib_add(1,2); // BTYPE = 1 -- fixed huffman
645
+
646
+ for (i=0; i < stbiw__ZHASH; ++i)
647
+ hash_table[i] = NULL;
648
+
649
+ i=0;
650
+ while (i < data_len-3) {
651
+ // hash next 3 bytes of data to be compressed
652
+ int h = stbiw__zhash(data+i)&(stbiw__ZHASH-1), best=3;
653
+ unsigned char *bestloc = 0;
654
+ unsigned char **hlist = hash_table[h];
655
+ int n = stbiw__sbcount(hlist);
656
+ for (j=0; j < n; ++j) {
657
+ if (hlist[j]-data > i-32768) { // if entry lies within window
658
+ int d = stbiw__zlib_countm(hlist[j], data+i, data_len-i);
659
+ if (d >= best) best=d,bestloc=hlist[j];
660
+ }
661
+ }
662
+ // when hash table entry is too long, delete half the entries
663
+ if (hash_table[h] && stbiw__sbn(hash_table[h]) == 2*quality) {
664
+ STBIW_MEMMOVE(hash_table[h], hash_table[h]+quality, sizeof(hash_table[h][0])*quality);
665
+ stbiw__sbn(hash_table[h]) = quality;
666
+ }
667
+ stbiw__sbpush(hash_table[h],data+i);
668
+
669
+ if (bestloc) {
670
+ // "lazy matching" - check match at *next* byte, and if it's better, do cur byte as literal
671
+ h = stbiw__zhash(data+i+1)&(stbiw__ZHASH-1);
672
+ hlist = hash_table[h];
673
+ n = stbiw__sbcount(hlist);
674
+ for (j=0; j < n; ++j) {
675
+ if (hlist[j]-data > i-32767) {
676
+ int e = stbiw__zlib_countm(hlist[j], data+i+1, data_len-i-1);
677
+ if (e > best) { // if next match is better, bail on current match
678
+ bestloc = NULL;
679
+ break;
680
+ }
681
+ }
682
+ }
683
+ }
684
+
685
+ if (bestloc) {
686
+ int d = (int) (data+i - bestloc); // distance back
687
+ STBIW_ASSERT(d <= 32767 && best <= 258);
688
+ for (j=0; best > lengthc[j+1]-1; ++j);
689
+ stbiw__zlib_huff(j+257);
690
+ if (lengtheb[j]) stbiw__zlib_add(best - lengthc[j], lengtheb[j]);
691
+ for (j=0; d > distc[j+1]-1; ++j);
692
+ stbiw__zlib_add(stbiw__zlib_bitrev(j,5),5);
693
+ if (disteb[j]) stbiw__zlib_add(d - distc[j], disteb[j]);
694
+ i += best;
695
+ } else {
696
+ stbiw__zlib_huffb(data[i]);
697
+ ++i;
698
+ }
699
+ }
700
+ // write out final bytes
701
+ for (;i < data_len; ++i)
702
+ stbiw__zlib_huffb(data[i]);
703
+ stbiw__zlib_huff(256); // end of block
704
+ // pad with 0 bits to byte boundary
705
+ while (bitcount)
706
+ stbiw__zlib_add(0,1);
707
+
708
+ for (i=0; i < stbiw__ZHASH; ++i)
709
+ (void) stbiw__sbfree(hash_table[i]);
710
+ STBIW_FREE(hash_table);
711
+
712
+ {
713
+ // compute adler32 on input
714
+ unsigned int s1=1, s2=0;
715
+ int blocklen = (int) (data_len % 5552);
716
+ j=0;
717
+ while (j < data_len) {
718
+ for (i=0; i < blocklen; ++i) s1 += data[j+i], s2 += s1;
719
+ s1 %= 65521, s2 %= 65521;
720
+ j += blocklen;
721
+ blocklen = 5552;
722
+ }
723
+ stbiw__sbpush(out, STBIW_UCHAR(s2 >> 8));
724
+ stbiw__sbpush(out, STBIW_UCHAR(s2));
725
+ stbiw__sbpush(out, STBIW_UCHAR(s1 >> 8));
726
+ stbiw__sbpush(out, STBIW_UCHAR(s1));
727
+ }
728
+ *out_len = stbiw__sbn(out);
729
+ // make returned pointer freeable
730
+ STBIW_MEMMOVE(stbiw__sbraw(out), out, *out_len);
731
+ return (unsigned char *) stbiw__sbraw(out);
732
+ #endif // STBIW_ZLIB_COMPRESS
733
+ }
734
+
735
+ static unsigned int stbiw__crc32(unsigned char *buffer, int len)
736
+ {
737
+ static unsigned int crc_table[256] =
738
+ {
739
+ 0x00000000, 0x77073096, 0xEE0E612C, 0x990951BA, 0x076DC419, 0x706AF48F, 0xE963A535, 0x9E6495A3,
740
+ 0x0eDB8832, 0x79DCB8A4, 0xE0D5E91E, 0x97D2D988, 0x09B64C2B, 0x7EB17CBD, 0xE7B82D07, 0x90BF1D91,
741
+ 0x1DB71064, 0x6AB020F2, 0xF3B97148, 0x84BE41DE, 0x1ADAD47D, 0x6DDDE4EB, 0xF4D4B551, 0x83D385C7,
742
+ 0x136C9856, 0x646BA8C0, 0xFD62F97A, 0x8A65C9EC, 0x14015C4F, 0x63066CD9, 0xFA0F3D63, 0x8D080DF5,
743
+ 0x3B6E20C8, 0x4C69105E, 0xD56041E4, 0xA2677172, 0x3C03E4D1, 0x4B04D447, 0xD20D85FD, 0xA50AB56B,
744
+ 0x35B5A8FA, 0x42B2986C, 0xDBBBC9D6, 0xACBCF940, 0x32D86CE3, 0x45DF5C75, 0xDCD60DCF, 0xABD13D59,
745
+ 0x26D930AC, 0x51DE003A, 0xC8D75180, 0xBFD06116, 0x21B4F4B5, 0x56B3C423, 0xCFBA9599, 0xB8BDA50F,
746
+ 0x2802B89E, 0x5F058808, 0xC60CD9B2, 0xB10BE924, 0x2F6F7C87, 0x58684C11, 0xC1611DAB, 0xB6662D3D,
747
+ 0x76DC4190, 0x01DB7106, 0x98D220BC, 0xEFD5102A, 0x71B18589, 0x06B6B51F, 0x9FBFE4A5, 0xE8B8D433,
748
+ 0x7807C9A2, 0x0F00F934, 0x9609A88E, 0xE10E9818, 0x7F6A0DBB, 0x086D3D2D, 0x91646C97, 0xE6635C01,
749
+ 0x6B6B51F4, 0x1C6C6162, 0x856530D8, 0xF262004E, 0x6C0695ED, 0x1B01A57B, 0x8208F4C1, 0xF50FC457,
750
+ 0x65B0D9C6, 0x12B7E950, 0x8BBEB8EA, 0xFCB9887C, 0x62DD1DDF, 0x15DA2D49, 0x8CD37CF3, 0xFBD44C65,
751
+ 0x4DB26158, 0x3AB551CE, 0xA3BC0074, 0xD4BB30E2, 0x4ADFA541, 0x3DD895D7, 0xA4D1C46D, 0xD3D6F4FB,
752
+ 0x4369E96A, 0x346ED9FC, 0xAD678846, 0xDA60B8D0, 0x44042D73, 0x33031DE5, 0xAA0A4C5F, 0xDD0D7CC9,
753
+ 0x5005713C, 0x270241AA, 0xBE0B1010, 0xC90C2086, 0x5768B525, 0x206F85B3, 0xB966D409, 0xCE61E49F,
754
+ 0x5EDEF90E, 0x29D9C998, 0xB0D09822, 0xC7D7A8B4, 0x59B33D17, 0x2EB40D81, 0xB7BD5C3B, 0xC0BA6CAD,
755
+ 0xEDB88320, 0x9ABFB3B6, 0x03B6E20C, 0x74B1D29A, 0xEAD54739, 0x9DD277AF, 0x04DB2615, 0x73DC1683,
756
+ 0xE3630B12, 0x94643B84, 0x0D6D6A3E, 0x7A6A5AA8, 0xE40ECF0B, 0x9309FF9D, 0x0A00AE27, 0x7D079EB1,
757
+ 0xF00F9344, 0x8708A3D2, 0x1E01F268, 0x6906C2FE, 0xF762575D, 0x806567CB, 0x196C3671, 0x6E6B06E7,
758
+ 0xFED41B76, 0x89D32BE0, 0x10DA7A5A, 0x67DD4ACC, 0xF9B9DF6F, 0x8EBEEFF9, 0x17B7BE43, 0x60B08ED5,
759
+ 0xD6D6A3E8, 0xA1D1937E, 0x38D8C2C4, 0x4FDFF252, 0xD1BB67F1, 0xA6BC5767, 0x3FB506DD, 0x48B2364B,
760
+ 0xD80D2BDA, 0xAF0A1B4C, 0x36034AF6, 0x41047A60, 0xDF60EFC3, 0xA867DF55, 0x316E8EEF, 0x4669BE79,
761
+ 0xCB61B38C, 0xBC66831A, 0x256FD2A0, 0x5268E236, 0xCC0C7795, 0xBB0B4703, 0x220216B9, 0x5505262F,
762
+ 0xC5BA3BBE, 0xB2BD0B28, 0x2BB45A92, 0x5CB36A04, 0xC2D7FFA7, 0xB5D0CF31, 0x2CD99E8B, 0x5BDEAE1D,
763
+ 0x9B64C2B0, 0xEC63F226, 0x756AA39C, 0x026D930A, 0x9C0906A9, 0xEB0E363F, 0x72076785, 0x05005713,
764
+ 0x95BF4A82, 0xE2B87A14, 0x7BB12BAE, 0x0CB61B38, 0x92D28E9B, 0xE5D5BE0D, 0x7CDCEFB7, 0x0BDBDF21,
765
+ 0x86D3D2D4, 0xF1D4E242, 0x68DDB3F8, 0x1FDA836E, 0x81BE16CD, 0xF6B9265B, 0x6FB077E1, 0x18B74777,
766
+ 0x88085AE6, 0xFF0F6A70, 0x66063BCA, 0x11010B5C, 0x8F659EFF, 0xF862AE69, 0x616BFFD3, 0x166CCF45,
767
+ 0xA00AE278, 0xD70DD2EE, 0x4E048354, 0x3903B3C2, 0xA7672661, 0xD06016F7, 0x4969474D, 0x3E6E77DB,
768
+ 0xAED16A4A, 0xD9D65ADC, 0x40DF0B66, 0x37D83BF0, 0xA9BCAE53, 0xDEBB9EC5, 0x47B2CF7F, 0x30B5FFE9,
769
+ 0xBDBDF21C, 0xCABAC28A, 0x53B39330, 0x24B4A3A6, 0xBAD03605, 0xCDD70693, 0x54DE5729, 0x23D967BF,
770
+ 0xB3667A2E, 0xC4614AB8, 0x5D681B02, 0x2A6F2B94, 0xB40BBE37, 0xC30C8EA1, 0x5A05DF1B, 0x2D02EF8D
771
+ };
772
+
773
+ unsigned int crc = ~0u;
774
+ int i;
775
+ for (i=0; i < len; ++i)
776
+ crc = (crc >> 8) ^ crc_table[buffer[i] ^ (crc & 0xff)];
777
+ return ~crc;
778
+ }
779
+
780
+ #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)
781
+ #define stbiw__wp32(data,v) stbiw__wpng4(data, (v)>>24,(v)>>16,(v)>>8,(v));
782
+ #define stbiw__wptag(data,s) stbiw__wpng4(data, s[0],s[1],s[2],s[3])
783
+
784
+ static void stbiw__wpcrc(unsigned char **data, int len)
785
+ {
786
+ unsigned int crc = stbiw__crc32(*data - len - 4, len+4);
787
+ stbiw__wp32(*data, crc);
788
+ }
789
+
790
+ static unsigned char stbiw__paeth(int a, int b, int c)
791
+ {
792
+ int p = a + b - c, pa = abs(p-a), pb = abs(p-b), pc = abs(p-c);
793
+ if (pa <= pb && pa <= pc) return STBIW_UCHAR(a);
794
+ if (pb <= pc) return STBIW_UCHAR(b);
795
+ return STBIW_UCHAR(c);
796
+ }
797
+
798
+ // @OPTIMIZE: provide an option that always forces left-predict or paeth predict
799
+ 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)
800
+ {
801
+ static int mapping[] = { 0,1,2,3,4 };
802
+ static int firstmap[] = { 0,1,0,5,6 };
803
+ int *mymap = (y != 0) ? mapping : firstmap;
804
+ int i;
805
+ int type = mymap[filter_type];
806
+ unsigned char *z = pixels + stride_bytes * (stbi__flip_vertically_on_write ? height-1-y : y);
807
+ int signed_stride = stbi__flip_vertically_on_write ? -stride_bytes : stride_bytes;
808
+ for (i = 0; i < n; ++i) {
809
+ switch (type) {
810
+ case 0: line_buffer[i] = z[i]; break;
811
+ case 1: line_buffer[i] = z[i]; break;
812
+ case 2: line_buffer[i] = z[i] - z[i-signed_stride]; break;
813
+ case 3: line_buffer[i] = z[i] - (z[i-signed_stride]>>1); break;
814
+ case 4: line_buffer[i] = (signed char) (z[i] - stbiw__paeth(0,z[i-signed_stride],0)); break;
815
+ case 5: line_buffer[i] = z[i]; break;
816
+ case 6: line_buffer[i] = z[i]; break;
817
+ }
818
+ }
819
+ for (i=n; i < width*n; ++i) {
820
+ switch (type) {
821
+ case 0: line_buffer[i] = z[i]; break;
822
+ case 1: line_buffer[i] = z[i] - z[i-n]; break;
823
+ case 2: line_buffer[i] = z[i] - z[i-signed_stride]; break;
824
+ case 3: line_buffer[i] = z[i] - ((z[i-n] + z[i-signed_stride])>>1); break;
825
+ case 4: line_buffer[i] = z[i] - stbiw__paeth(z[i-n], z[i-signed_stride], z[i-signed_stride-n]); break;
826
+ case 5: line_buffer[i] = z[i] - (z[i-n]>>1); break;
827
+ case 6: line_buffer[i] = z[i] - stbiw__paeth(z[i-n], 0,0); break;
828
+ }
829
+ }
830
+ }
831
+
832
+ unsigned char *stbi_write_png_to_mem(unsigned char *pixels, int stride_bytes, int x, int y, int n, int *out_len)
833
+ {
834
+ int force_filter = stbi_write_force_png_filter;
835
+ int ctype[5] = { -1, 0, 4, 2, 6 };
836
+ unsigned char sig[8] = { 137,80,78,71,13,10,26,10 };
837
+ unsigned char *out,*o, *filt, *zlib;
838
+ signed char *line_buffer;
839
+ int j,zlen;
840
+
841
+ if (stride_bytes == 0)
842
+ stride_bytes = x * n;
843
+
844
+ if (force_filter >= 5) {
845
+ force_filter = -1;
846
+ }
847
+
848
+ filt = (unsigned char *) STBIW_MALLOC((x*n+1) * y); if (!filt) return 0;
849
+ line_buffer = (signed char *) STBIW_MALLOC(x * n); if (!line_buffer) { STBIW_FREE(filt); return 0; }
850
+ for (j=0; j < y; ++j) {
851
+ int filter_type;
852
+ if (force_filter > -1) {
853
+ filter_type = force_filter;
854
+ stbiw__encode_png_line(pixels, stride_bytes, x, y, j, n, force_filter, line_buffer);
855
+ } else { // Estimate the best filter by running through all of them:
856
+ int best_filter = 0, best_filter_val = 0x7fffffff, est, i;
857
+ for (filter_type = 0; filter_type < 5; filter_type++) {
858
+ stbiw__encode_png_line(pixels, stride_bytes, x, y, j, n, filter_type, line_buffer);
859
+
860
+ // Estimate the entropy of the line using this filter; the less, the better.
861
+ est = 0;
862
+ for (i = 0; i < x*n; ++i) {
863
+ est += abs((signed char) line_buffer[i]);
864
+ }
865
+ if (est < best_filter_val) {
866
+ best_filter_val = est;
867
+ best_filter = filter_type;
868
+ }
869
+ }
870
+ if (filter_type != best_filter) { // If the last iteration already got us the best filter, don't redo it
871
+ stbiw__encode_png_line(pixels, stride_bytes, x, y, j, n, best_filter, line_buffer);
872
+ filter_type = best_filter;
873
+ }
874
+ }
875
+ // when we get here, filter_type contains the filter type, and line_buffer contains the data
876
+ filt[j*(x*n+1)] = (unsigned char) filter_type;
877
+ STBIW_MEMMOVE(filt+j*(x*n+1)+1, line_buffer, x*n);
878
+ }
879
+ STBIW_FREE(line_buffer);
880
+ zlib = stbi_zlib_compress(filt, y*( x*n+1), &zlen, stbi_write_png_compression_level);
881
+ STBIW_FREE(filt);
882
+ if (!zlib) return 0;
883
+
884
+ // each tag requires 12 bytes of overhead
885
+ out = (unsigned char *) STBIW_MALLOC(8 + 12+13 + 12+zlen + 12);
886
+ if (!out) return 0;
887
+ *out_len = 8 + 12+13 + 12+zlen + 12;
888
+
889
+ o=out;
890
+ STBIW_MEMMOVE(o,sig,8); o+= 8;
891
+ stbiw__wp32(o, 13); // header length
892
+ stbiw__wptag(o, "IHDR");
893
+ stbiw__wp32(o, x);
894
+ stbiw__wp32(o, y);
895
+ *o++ = 8;
896
+ *o++ = STBIW_UCHAR(ctype[n]);
897
+ *o++ = 0;
898
+ *o++ = 0;
899
+ *o++ = 0;
900
+ stbiw__wpcrc(&o,13);
901
+
902
+ stbiw__wp32(o, zlen);
903
+ stbiw__wptag(o, "IDAT");
904
+ STBIW_MEMMOVE(o, zlib, zlen);
905
+ o += zlen;
906
+ STBIW_FREE(zlib);
907
+ stbiw__wpcrc(&o, zlen);
908
+
909
+ stbiw__wp32(o,0);
910
+ stbiw__wptag(o, "IEND");
911
+ stbiw__wpcrc(&o,0);
912
+
913
+ STBIW_ASSERT(o == out + *out_len);
914
+
915
+ return out;
916
+ }
917
+
918
+ #ifndef STBI_WRITE_NO_STDIO
919
+ STBIWDEF int stbi_write_png(char const *filename, int x, int y, int comp, const void *data, int stride_bytes)
920
+ {
921
+ FILE *f;
922
+ int len;
923
+ unsigned char *png = stbi_write_png_to_mem((unsigned char *) data, stride_bytes, x, y, comp, &len);
924
+ if (png == NULL) return 0;
925
+ #ifdef STBI_MSC_SECURE_CRT
926
+ if (fopen_s(&f, filename, "wb"))
927
+ f = NULL;
928
+ #else
929
+ f = fopen(filename, "wb");
930
+ #endif
931
+ if (!f) { STBIW_FREE(png); return 0; }
932
+ fwrite(png, 1, len, f);
933
+ fclose(f);
934
+ STBIW_FREE(png);
935
+ return 1;
936
+ }
937
+ #endif
938
+
939
+ 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)
940
+ {
941
+ int len;
942
+ unsigned char *png = stbi_write_png_to_mem((unsigned char *) data, stride_bytes, x, y, comp, &len);
943
+ if (png == NULL) return 0;
944
+ func(context, png, len);
945
+ STBIW_FREE(png);
946
+ return 1;
947
+ }
948
+
949
+
950
+ /* ***************************************************************************
951
+ *
952
+ * JPEG writer
953
+ *
954
+ * This is based on Jon Olick's jo_jpeg.cpp:
955
+ * public domain Simple, Minimalistic JPEG writer - http://www.jonolick.com/code.html
956
+ */
957
+
958
+ 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,
959
+ 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 };
960
+
961
+ static void stbiw__jpg_writeBits(stbi__write_context *s, int *bitBufP, int *bitCntP, const unsigned short *bs) {
962
+ int bitBuf = *bitBufP, bitCnt = *bitCntP;
963
+ bitCnt += bs[1];
964
+ bitBuf |= bs[0] << (24 - bitCnt);
965
+ while(bitCnt >= 8) {
966
+ unsigned char c = (bitBuf >> 16) & 255;
967
+ stbiw__putc(s, c);
968
+ if(c == 255) {
969
+ stbiw__putc(s, 0);
970
+ }
971
+ bitBuf <<= 8;
972
+ bitCnt -= 8;
973
+ }
974
+ *bitBufP = bitBuf;
975
+ *bitCntP = bitCnt;
976
+ }
977
+
978
+ static void stbiw__jpg_DCT(float *d0p, float *d1p, float *d2p, float *d3p, float *d4p, float *d5p, float *d6p, float *d7p) {
979
+ float d0 = *d0p, d1 = *d1p, d2 = *d2p, d3 = *d3p, d4 = *d4p, d5 = *d5p, d6 = *d6p, d7 = *d7p;
980
+ float z1, z2, z3, z4, z5, z11, z13;
981
+
982
+ float tmp0 = d0 + d7;
983
+ float tmp7 = d0 - d7;
984
+ float tmp1 = d1 + d6;
985
+ float tmp6 = d1 - d6;
986
+ float tmp2 = d2 + d5;
987
+ float tmp5 = d2 - d5;
988
+ float tmp3 = d3 + d4;
989
+ float tmp4 = d3 - d4;
990
+
991
+ // Even part
992
+ float tmp10 = tmp0 + tmp3; // phase 2
993
+ float tmp13 = tmp0 - tmp3;
994
+ float tmp11 = tmp1 + tmp2;
995
+ float tmp12 = tmp1 - tmp2;
996
+
997
+ d0 = tmp10 + tmp11; // phase 3
998
+ d4 = tmp10 - tmp11;
999
+
1000
+ z1 = (tmp12 + tmp13) * 0.707106781f; // c4
1001
+ d2 = tmp13 + z1; // phase 5
1002
+ d6 = tmp13 - z1;
1003
+
1004
+ // Odd part
1005
+ tmp10 = tmp4 + tmp5; // phase 2
1006
+ tmp11 = tmp5 + tmp6;
1007
+ tmp12 = tmp6 + tmp7;
1008
+
1009
+ // The rotator is modified from fig 4-8 to avoid extra negations.
1010
+ z5 = (tmp10 - tmp12) * 0.382683433f; // c6
1011
+ z2 = tmp10 * 0.541196100f + z5; // c2-c6
1012
+ z4 = tmp12 * 1.306562965f + z5; // c2+c6
1013
+ z3 = tmp11 * 0.707106781f; // c4
1014
+
1015
+ z11 = tmp7 + z3; // phase 5
1016
+ z13 = tmp7 - z3;
1017
+
1018
+ *d5p = z13 + z2; // phase 6
1019
+ *d3p = z13 - z2;
1020
+ *d1p = z11 + z4;
1021
+ *d7p = z11 - z4;
1022
+
1023
+ *d0p = d0; *d2p = d2; *d4p = d4; *d6p = d6;
1024
+ }
1025
+
1026
+ static void stbiw__jpg_calcBits(int val, unsigned short bits[2]) {
1027
+ int tmp1 = val < 0 ? -val : val;
1028
+ val = val < 0 ? val-1 : val;
1029
+ bits[1] = 1;
1030
+ while(tmp1 >>= 1) {
1031
+ ++bits[1];
1032
+ }
1033
+ bits[0] = val & ((1<<bits[1])-1);
1034
+ }
1035
+
1036
+ 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]) {
1037
+ const unsigned short EOB[2] = { HTAC[0x00][0], HTAC[0x00][1] };
1038
+ const unsigned short M16zeroes[2] = { HTAC[0xF0][0], HTAC[0xF0][1] };
1039
+ int dataOff, i, diff, end0pos;
1040
+ int DU[64];
1041
+
1042
+ // DCT rows
1043
+ for(dataOff=0; dataOff<64; dataOff+=8) {
1044
+ 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]);
1045
+ }
1046
+ // DCT columns
1047
+ for(dataOff=0; dataOff<8; ++dataOff) {
1048
+ 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]);
1049
+ }
1050
+ // Quantize/descale/zigzag the coefficients
1051
+ for(i=0; i<64; ++i) {
1052
+ float v = CDU[i]*fdtbl[i];
1053
+ // DU[stbiw__jpg_ZigZag[i]] = (int)(v < 0 ? ceilf(v - 0.5f) : floorf(v + 0.5f));
1054
+ // ceilf() and floorf() are C99, not C89, but I /think/ they're not needed here anyway?
1055
+ DU[stbiw__jpg_ZigZag[i]] = (int)(v < 0 ? v - 0.5f : v + 0.5f);
1056
+ }
1057
+
1058
+ // Encode DC
1059
+ diff = DU[0] - DC;
1060
+ if (diff == 0) {
1061
+ stbiw__jpg_writeBits(s, bitBuf, bitCnt, HTDC[0]);
1062
+ } else {
1063
+ unsigned short bits[2];
1064
+ stbiw__jpg_calcBits(diff, bits);
1065
+ stbiw__jpg_writeBits(s, bitBuf, bitCnt, HTDC[bits[1]]);
1066
+ stbiw__jpg_writeBits(s, bitBuf, bitCnt, bits);
1067
+ }
1068
+ // Encode ACs
1069
+ end0pos = 63;
1070
+ for(; (end0pos>0)&&(DU[end0pos]==0); --end0pos) {
1071
+ }
1072
+ // end0pos = first element in reverse order !=0
1073
+ if(end0pos == 0) {
1074
+ stbiw__jpg_writeBits(s, bitBuf, bitCnt, EOB);
1075
+ return DU[0];
1076
+ }
1077
+ for(i = 1; i <= end0pos; ++i) {
1078
+ int startpos = i;
1079
+ int nrzeroes;
1080
+ unsigned short bits[2];
1081
+ for (; DU[i]==0 && i<=end0pos; ++i) {
1082
+ }
1083
+ nrzeroes = i-startpos;
1084
+ if ( nrzeroes >= 16 ) {
1085
+ int lng = nrzeroes>>4;
1086
+ int nrmarker;
1087
+ for (nrmarker=1; nrmarker <= lng; ++nrmarker)
1088
+ stbiw__jpg_writeBits(s, bitBuf, bitCnt, M16zeroes);
1089
+ nrzeroes &= 15;
1090
+ }
1091
+ stbiw__jpg_calcBits(DU[i], bits);
1092
+ stbiw__jpg_writeBits(s, bitBuf, bitCnt, HTAC[(nrzeroes<<4)+bits[1]]);
1093
+ stbiw__jpg_writeBits(s, bitBuf, bitCnt, bits);
1094
+ }
1095
+ if(end0pos != 63) {
1096
+ stbiw__jpg_writeBits(s, bitBuf, bitCnt, EOB);
1097
+ }
1098
+ return DU[0];
1099
+ }
1100
+
1101
+ static int stbi_write_jpg_core(stbi__write_context *s, int width, int height, int comp, const void* data, int quality) {
1102
+ // Constants that don't pollute global namespace
1103
+ 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};
1104
+ static const unsigned char std_dc_luminance_values[] = {0,1,2,3,4,5,6,7,8,9,10,11};
1105
+ 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};
1106
+ static const unsigned char std_ac_luminance_values[] = {
1107
+ 0x01,0x02,0x03,0x00,0x04,0x11,0x05,0x12,0x21,0x31,0x41,0x06,0x13,0x51,0x61,0x07,0x22,0x71,0x14,0x32,0x81,0x91,0xa1,0x08,
1108
+ 0x23,0x42,0xb1,0xc1,0x15,0x52,0xd1,0xf0,0x24,0x33,0x62,0x72,0x82,0x09,0x0a,0x16,0x17,0x18,0x19,0x1a,0x25,0x26,0x27,0x28,
1109
+ 0x29,0x2a,0x34,0x35,0x36,0x37,0x38,0x39,0x3a,0x43,0x44,0x45,0x46,0x47,0x48,0x49,0x4a,0x53,0x54,0x55,0x56,0x57,0x58,0x59,
1110
+ 0x5a,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6a,0x73,0x74,0x75,0x76,0x77,0x78,0x79,0x7a,0x83,0x84,0x85,0x86,0x87,0x88,0x89,
1111
+ 0x8a,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9a,0xa2,0xa3,0xa4,0xa5,0xa6,0xa7,0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,0xb5,0xb6,
1112
+ 0xb7,0xb8,0xb9,0xba,0xc2,0xc3,0xc4,0xc5,0xc6,0xc7,0xc8,0xc9,0xca,0xd2,0xd3,0xd4,0xd5,0xd6,0xd7,0xd8,0xd9,0xda,0xe1,0xe2,
1113
+ 0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,0xea,0xf1,0xf2,0xf3,0xf4,0xf5,0xf6,0xf7,0xf8,0xf9,0xfa
1114
+ };
1115
+ 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};
1116
+ static const unsigned char std_dc_chrominance_values[] = {0,1,2,3,4,5,6,7,8,9,10,11};
1117
+ 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};
1118
+ static const unsigned char std_ac_chrominance_values[] = {
1119
+ 0x00,0x01,0x02,0x03,0x11,0x04,0x05,0x21,0x31,0x06,0x12,0x41,0x51,0x07,0x61,0x71,0x13,0x22,0x32,0x81,0x08,0x14,0x42,0x91,
1120
+ 0xa1,0xb1,0xc1,0x09,0x23,0x33,0x52,0xf0,0x15,0x62,0x72,0xd1,0x0a,0x16,0x24,0x34,0xe1,0x25,0xf1,0x17,0x18,0x19,0x1a,0x26,
1121
+ 0x27,0x28,0x29,0x2a,0x35,0x36,0x37,0x38,0x39,0x3a,0x43,0x44,0x45,0x46,0x47,0x48,0x49,0x4a,0x53,0x54,0x55,0x56,0x57,0x58,
1122
+ 0x59,0x5a,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6a,0x73,0x74,0x75,0x76,0x77,0x78,0x79,0x7a,0x82,0x83,0x84,0x85,0x86,0x87,
1123
+ 0x88,0x89,0x8a,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9a,0xa2,0xa3,0xa4,0xa5,0xa6,0xa7,0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,
1124
+ 0xb5,0xb6,0xb7,0xb8,0xb9,0xba,0xc2,0xc3,0xc4,0xc5,0xc6,0xc7,0xc8,0xc9,0xca,0xd2,0xd3,0xd4,0xd5,0xd6,0xd7,0xd8,0xd9,0xda,
1125
+ 0xe2,0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,0xea,0xf2,0xf3,0xf4,0xf5,0xf6,0xf7,0xf8,0xf9,0xfa
1126
+ };
1127
+ // Huffman tables
1128
+ 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}};
1129
+ 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}};
1130
+ static const unsigned short YAC_HT[256][2] = {
1131
+ {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},
1132
+ {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},
1133
+ {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},
1134
+ {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},
1135
+ {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},
1136
+ {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},
1137
+ {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},
1138
+ {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},
1139
+ {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},
1140
+ {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},
1141
+ {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},
1142
+ {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},
1143
+ {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},
1144
+ {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},
1145
+ {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},
1146
+ {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}
1147
+ };
1148
+ static const unsigned short UVAC_HT[256][2] = {
1149
+ {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},
1150
+ {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},
1151
+ {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},
1152
+ {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},
1153
+ {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},
1154
+ {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},
1155
+ {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},
1156
+ {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},
1157
+ {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},
1158
+ {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},
1159
+ {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},
1160
+ {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},
1161
+ {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},
1162
+ {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},
1163
+ {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},
1164
+ {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}
1165
+ };
1166
+ 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,
1167
+ 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};
1168
+ 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,
1169
+ 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};
1170
+ static const float aasf[] = { 1.0f * 2.828427125f, 1.387039845f * 2.828427125f, 1.306562965f * 2.828427125f, 1.175875602f * 2.828427125f,
1171
+ 1.0f * 2.828427125f, 0.785694958f * 2.828427125f, 0.541196100f * 2.828427125f, 0.275899379f * 2.828427125f };
1172
+
1173
+ int row, col, i, k;
1174
+ float fdtbl_Y[64], fdtbl_UV[64];
1175
+ unsigned char YTable[64], UVTable[64];
1176
+
1177
+ if(!data || !width || !height || comp > 4 || comp < 1) {
1178
+ return 0;
1179
+ }
1180
+
1181
+ quality = quality ? quality : 90;
1182
+ quality = quality < 1 ? 1 : quality > 100 ? 100 : quality;
1183
+ quality = quality < 50 ? 5000 / quality : 200 - quality * 2;
1184
+
1185
+ for(i = 0; i < 64; ++i) {
1186
+ int uvti, yti = (YQT[i]*quality+50)/100;
1187
+ YTable[stbiw__jpg_ZigZag[i]] = (unsigned char) (yti < 1 ? 1 : yti > 255 ? 255 : yti);
1188
+ uvti = (UVQT[i]*quality+50)/100;
1189
+ UVTable[stbiw__jpg_ZigZag[i]] = (unsigned char) (uvti < 1 ? 1 : uvti > 255 ? 255 : uvti);
1190
+ }
1191
+
1192
+ for(row = 0, k = 0; row < 8; ++row) {
1193
+ for(col = 0; col < 8; ++col, ++k) {
1194
+ fdtbl_Y[k] = 1 / (YTable [stbiw__jpg_ZigZag[k]] * aasf[row] * aasf[col]);
1195
+ fdtbl_UV[k] = 1 / (UVTable[stbiw__jpg_ZigZag[k]] * aasf[row] * aasf[col]);
1196
+ }
1197
+ }
1198
+
1199
+ // Write Headers
1200
+ {
1201
+ 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 };
1202
+ static const unsigned char head2[] = { 0xFF,0xDA,0,0xC,3,1,0,2,0x11,3,0x11,0,0x3F,0 };
1203
+ const unsigned char head1[] = { 0xFF,0xC0,0,0x11,8,(unsigned char)(height>>8),STBIW_UCHAR(height),(unsigned char)(width>>8),STBIW_UCHAR(width),
1204
+ 3,1,0x11,0,2,0x11,1,3,0x11,1,0xFF,0xC4,0x01,0xA2,0 };
1205
+ s->func(s->context, (void*)head0, sizeof(head0));
1206
+ s->func(s->context, (void*)YTable, sizeof(YTable));
1207
+ stbiw__putc(s, 1);
1208
+ s->func(s->context, UVTable, sizeof(UVTable));
1209
+ s->func(s->context, (void*)head1, sizeof(head1));
1210
+ s->func(s->context, (void*)(std_dc_luminance_nrcodes+1), sizeof(std_dc_luminance_nrcodes)-1);
1211
+ s->func(s->context, (void*)std_dc_luminance_values, sizeof(std_dc_luminance_values));
1212
+ stbiw__putc(s, 0x10); // HTYACinfo
1213
+ s->func(s->context, (void*)(std_ac_luminance_nrcodes+1), sizeof(std_ac_luminance_nrcodes)-1);
1214
+ s->func(s->context, (void*)std_ac_luminance_values, sizeof(std_ac_luminance_values));
1215
+ stbiw__putc(s, 1); // HTUDCinfo
1216
+ s->func(s->context, (void*)(std_dc_chrominance_nrcodes+1), sizeof(std_dc_chrominance_nrcodes)-1);
1217
+ s->func(s->context, (void*)std_dc_chrominance_values, sizeof(std_dc_chrominance_values));
1218
+ stbiw__putc(s, 0x11); // HTUACinfo
1219
+ s->func(s->context, (void*)(std_ac_chrominance_nrcodes+1), sizeof(std_ac_chrominance_nrcodes)-1);
1220
+ s->func(s->context, (void*)std_ac_chrominance_values, sizeof(std_ac_chrominance_values));
1221
+ s->func(s->context, (void*)head2, sizeof(head2));
1222
+ }
1223
+
1224
+ // Encode 8x8 macroblocks
1225
+ {
1226
+ static const unsigned short fillBits[] = {0x7F, 7};
1227
+ const unsigned char *imageData = (const unsigned char *)data;
1228
+ int DCY=0, DCU=0, DCV=0;
1229
+ int bitBuf=0, bitCnt=0;
1230
+ // comp == 2 is grey+alpha (alpha is ignored)
1231
+ int ofsG = comp > 2 ? 1 : 0, ofsB = comp > 2 ? 2 : 0;
1232
+ int x, y, pos;
1233
+ for(y = 0; y < height; y += 8) {
1234
+ for(x = 0; x < width; x += 8) {
1235
+ float YDU[64], UDU[64], VDU[64];
1236
+ for(row = y, pos = 0; row < y+8; ++row) {
1237
+ for(col = x; col < x+8; ++col, ++pos) {
1238
+ int p = (stbi__flip_vertically_on_write ? height-1-row : row)*width*comp + col*comp;
1239
+ float r, g, b;
1240
+ if(row >= height) {
1241
+ p -= width*comp*(row+1 - height);
1242
+ }
1243
+ if(col >= width) {
1244
+ p -= comp*(col+1 - width);
1245
+ }
1246
+
1247
+ r = imageData[p+0];
1248
+ g = imageData[p+ofsG];
1249
+ b = imageData[p+ofsB];
1250
+ YDU[pos]=+0.29900f*r+0.58700f*g+0.11400f*b-128;
1251
+ UDU[pos]=-0.16874f*r-0.33126f*g+0.50000f*b;
1252
+ VDU[pos]=+0.50000f*r-0.41869f*g-0.08131f*b;
1253
+ }
1254
+ }
1255
+
1256
+ DCY = stbiw__jpg_processDU(s, &bitBuf, &bitCnt, YDU, fdtbl_Y, DCY, YDC_HT, YAC_HT);
1257
+ DCU = stbiw__jpg_processDU(s, &bitBuf, &bitCnt, UDU, fdtbl_UV, DCU, UVDC_HT, UVAC_HT);
1258
+ DCV = stbiw__jpg_processDU(s, &bitBuf, &bitCnt, VDU, fdtbl_UV, DCV, UVDC_HT, UVAC_HT);
1259
+ }
1260
+ }
1261
+
1262
+ // Do the bit alignment of the EOI marker
1263
+ stbiw__jpg_writeBits(s, &bitBuf, &bitCnt, fillBits);
1264
+ }
1265
+
1266
+ // EOI
1267
+ stbiw__putc(s, 0xFF);
1268
+ stbiw__putc(s, 0xD9);
1269
+
1270
+ return 1;
1271
+ }
1272
+
1273
+ STBIWDEF int stbi_write_jpg_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data, int quality)
1274
+ {
1275
+ stbi__write_context s;
1276
+ stbi__start_write_callbacks(&s, func, context);
1277
+ return stbi_write_jpg_core(&s, x, y, comp, (void *) data, quality);
1278
+ }
1279
+
1280
+
1281
+ #ifndef STBI_WRITE_NO_STDIO
1282
+ STBIWDEF int stbi_write_jpg(char const *filename, int x, int y, int comp, const void *data, int quality)
1283
+ {
1284
+ stbi__write_context s;
1285
+ if (stbi__start_write_file(&s,filename)) {
1286
+ int r = stbi_write_jpg_core(&s, x, y, comp, data, quality);
1287
+ stbi__end_write_file(&s);
1288
+ return r;
1289
+ } else
1290
+ return 0;
1291
+ }
1292
+ #endif
1293
+
1294
+ /* Revision history
1295
+ 1.09 (2018-02-11)
1296
+ fix typo in zlib quality API, improve STB_I_W_STATIC in C++
1297
+ 1.08 (2018-01-29)
1298
+ add stbi__flip_vertically_on_write, external zlib, zlib quality, choose PNG filter
1299
+ 1.07 (2017-07-24)
1300
+ doc fix
1301
+ 1.06 (2017-07-23)
1302
+ writing JPEG (using Jon Olick's code)
1303
+ 1.05 ???
1304
+ 1.04 (2017-03-03)
1305
+ monochrome BMP expansion
1306
+ 1.03 ???
1307
+ 1.02 (2016-04-02)
1308
+ avoid allocating large structures on the stack
1309
+ 1.01 (2016-01-16)
1310
+ STBIW_REALLOC_SIZED: support allocators with no realloc support
1311
+ avoid race-condition in crc initialization
1312
+ minor compile issues
1313
+ 1.00 (2015-09-14)
1314
+ installable file IO function
1315
+ 0.99 (2015-09-13)
1316
+ warning fixes; TGA rle support
1317
+ 0.98 (2015-04-08)
1318
+ added STBIW_MALLOC, STBIW_ASSERT etc
1319
+ 0.97 (2015-01-18)
1320
+ fixed HDR asserts, rewrote HDR rle logic
1321
+ 0.96 (2015-01-17)
1322
+ add HDR output
1323
+ fix monochrome BMP
1324
+ 0.95 (2014-08-17)
1325
+ add monochrome TGA output
1326
+ 0.94 (2014-05-31)
1327
+ rename private functions to avoid conflicts with stb_image.h
1328
+ 0.93 (2014-05-27)
1329
+ warning fixes
1330
+ 0.92 (2010-08-01)
1331
+ casts to unsigned char to fix warnings
1332
+ 0.91 (2010-07-17)
1333
+ first public release
1334
+ 0.90 first internal release
1335
+ */
1336
+
1337
+ /*
1338
+ ------------------------------------------------------------------------------
1339
+ This software is available under 2 licenses -- choose whichever you prefer.
1340
+ ------------------------------------------------------------------------------
1341
+ ALTERNATIVE A - MIT License
1342
+ Copyright (c) 2017 Sean Barrett
1343
+ Permission is hereby granted, free of charge, to any person obtaining a copy of
1344
+ this software and associated documentation files (the "Software"), to deal in
1345
+ the Software without restriction, including without limitation the rights to
1346
+ use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
1347
+ of the Software, and to permit persons to whom the Software is furnished to do
1348
+ so, subject to the following conditions:
1349
+ The above copyright notice and this permission notice shall be included in all
1350
+ copies or substantial portions of the Software.
1351
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
1352
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
1353
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
1354
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
1355
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
1356
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
1357
+ SOFTWARE.
1358
+ ------------------------------------------------------------------------------
1359
+ ALTERNATIVE B - Public Domain (www.unlicense.org)
1360
+ This is free and unencumbered software released into the public domain.
1361
+ Anyone is free to copy, modify, publish, use, compile, sell, or distribute this
1362
+ software, either in source code form or as a compiled binary, for any purpose,
1363
+ commercial or non-commercial, and by any means.
1364
+ In jurisdictions that recognize copyright laws, the author or authors of this
1365
+ software dedicate any and all copyright interest in the software to the public
1366
+ domain. We make this dedication for the benefit of the public at large and to
1367
+ the detriment of our heirs and successors. We intend this dedication to be an
1368
+ overt act of relinquishment in perpetuity of all present and future rights to
1369
+ this software under copyright law.
1370
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
1371
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
1372
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
1373
+ AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
1374
+ ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
1375
+ WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
1376
+ ------------------------------------------------------------------------------
1377
+ */