stb_image_write.h 63 KB

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