stb_image_write.h 40 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106
  1. /* stb_image_write - v1.05 - public domain - http://nothings.org/stb/stb_image_write.h
  2. writes out PNG/BMP/TGA 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. ABOUT:
  9. This header file is a library for writing images to C stdio. It could be
  10. adapted to write to memory or a general streaming interface; let me know.
  11. The PNG output is not optimal; it is 20-50% larger than the file
  12. written by a decent optimizing implementation. This library is designed
  13. for source code compactness and simplicity, not optimal image file size
  14. or run-time performance.
  15. BUILDING:
  16. You can #define STBIW_ASSERT(x) before the #include to avoid using assert.h.
  17. You can #define STBIW_MALLOC(), STBIW_REALLOC(), and STBIW_FREE() to replace
  18. malloc,realloc,free.
  19. You can define STBIW_MEMMOVE() to replace memmove()
  20. USAGE:
  21. There are four functions, one for each image file format:
  22. int stbi_write_png(char const *filename, int w, int h, int comp, const void *data, int stride_in_bytes);
  23. int stbi_write_bmp(char const *filename, int w, int h, int comp, const void *data);
  24. int stbi_write_tga(char const *filename, int w, int h, int comp, const void *data);
  25. int stbi_write_hdr(char const *filename, int w, int h, int comp, const float *data);
  26. There are also four equivalent functions that use an arbitrary write function. You are
  27. expected to open/close your file-equivalent before and after calling these:
  28. 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);
  29. int stbi_write_bmp_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data);
  30. int stbi_write_tga_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data);
  31. int stbi_write_hdr_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const float *data);
  32. where the callback is:
  33. void stbi_write_func(void *context, void *data, int size);
  34. You can define STBI_WRITE_NO_STDIO to disable the file variant of these
  35. functions, so the library will not use stdio.h at all. However, this will
  36. also disable HDR writing, because it requires stdio for formatted output.
  37. Each function returns 0 on failure and non-0 on success.
  38. The functions create an image file defined by the parameters. The image
  39. is a rectangle of pixels stored from left-to-right, top-to-bottom.
  40. Each pixel contains 'comp' channels of data stored interleaved with 8-bits
  41. per channel, in the following order: 1=Y, 2=YA, 3=RGB, 4=RGBA. (Y is
  42. monochrome color.) The rectangle is 'w' pixels wide and 'h' pixels tall.
  43. The *data pointer points to the first byte of the top-left-most pixel.
  44. For PNG, "stride_in_bytes" is the distance in bytes from the first byte of
  45. a row of pixels to the first byte of the next row of pixels.
  46. PNG creates output files with the same number of components as the input.
  47. The BMP format expands Y to RGB in the file format and does not
  48. output alpha.
  49. PNG supports writing rectangles of data even when the bytes storing rows of
  50. data are not consecutive in memory (e.g. sub-rectangles of a larger image),
  51. by supplying the stride between the beginning of adjacent rows. The other
  52. formats do not. (Thus you cannot write a native-format BMP through the BMP
  53. writer, both because it is in BGR order and because it may have padding
  54. at the end of the line.)
  55. HDR expects linear float data. Since the format is always 32-bit rgb(e)
  56. data, alpha (if provided) is discarded, and for monochrome data it is
  57. replicated across all three channels.
  58. TGA supports RLE or non-RLE compressed data. To use non-RLE-compressed
  59. data, set the global variable 'stbi_write_tga_with_rle' to 0.
  60. CREDITS:
  61. PNG/BMP/TGA
  62. Sean Barrett
  63. HDR
  64. Baldur Karlsson
  65. TGA monochrome:
  66. Jean-Sebastien Guay
  67. misc enhancements:
  68. Tim Kelsey
  69. TGA RLE
  70. Alan Hickman
  71. initial file IO callback implementation
  72. Emmanuel Julien
  73. bugfixes:
  74. github:Chribba
  75. Guillaume Chereau
  76. github:jry2
  77. github:romigrou
  78. Sergio Gonzalez
  79. Jonas Karlsson
  80. Filip Wasil
  81. Thatcher Ulrich
  82. github:poppolopoppo
  83. Patrick Boettcher
  84. LICENSE
  85. See end of file for license information.
  86. */
  87. #ifndef INCLUDE_STB_IMAGE_WRITE_H
  88. #define INCLUDE_STB_IMAGE_WRITE_H
  89. // ATOMIC BEGIN
  90. #include "../../Atomic/Container/Str.h"
  91. // ATOMIC END
  92. #ifdef __cplusplus
  93. extern "C" {
  94. #endif
  95. #ifdef STB_IMAGE_WRITE_STATIC
  96. #define STBIWDEF static
  97. #else
  98. #define STBIWDEF extern
  99. extern int stbi_write_tga_with_rle;
  100. #endif
  101. #ifndef STBI_WRITE_NO_STDIO
  102. STBIWDEF int stbi_write_png(char const *filename, int w, int h, int comp, const void *data, int stride_in_bytes);
  103. STBIWDEF int stbi_write_bmp(char const *filename, int w, int h, int comp, const void *data);
  104. STBIWDEF int stbi_write_tga(char const *filename, int w, int h, int comp, const void *data);
  105. STBIWDEF int stbi_write_hdr(char const *filename, int w, int h, int comp, const float *data);
  106. #endif
  107. typedef void stbi_write_func(void *context, void *data, int size);
  108. 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);
  109. STBIWDEF int stbi_write_bmp_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data);
  110. STBIWDEF int stbi_write_tga_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const void *data);
  111. STBIWDEF int stbi_write_hdr_to_func(stbi_write_func *func, void *context, int w, int h, int comp, const float *data);
  112. // ATOMIC BEGIN
  113. unsigned char *stbi_write_png_to_mem(unsigned char *pixels, int stride_bytes, int x, int y, int n, int *out_len);
  114. // ATOMIC END
  115. #ifdef __cplusplus
  116. }
  117. #endif
  118. #endif//INCLUDE_STB_IMAGE_WRITE_H
  119. #ifdef STB_IMAGE_WRITE_IMPLEMENTATION
  120. #ifdef _WIN32
  121. #ifndef _CRT_SECURE_NO_WARNINGS
  122. #define _CRT_SECURE_NO_WARNINGS
  123. #endif
  124. #ifndef _CRT_NONSTDC_NO_DEPRECATE
  125. #define _CRT_NONSTDC_NO_DEPRECATE
  126. #endif
  127. #endif
  128. #ifndef STBI_WRITE_NO_STDIO
  129. #include <stdio.h>
  130. #endif // STBI_WRITE_NO_STDIO
  131. #include <stdarg.h>
  132. #include <stdlib.h>
  133. #include <string.h>
  134. #include <math.h>
  135. #if defined(STBIW_MALLOC) && defined(STBIW_FREE) && (defined(STBIW_REALLOC) || defined(STBIW_REALLOC_SIZED))
  136. // ok
  137. #elif !defined(STBIW_MALLOC) && !defined(STBIW_FREE) && !defined(STBIW_REALLOC) && !defined(STBIW_REALLOC_SIZED)
  138. // ok
  139. #else
  140. #error "Must define all or none of STBIW_MALLOC, STBIW_FREE, and STBIW_REALLOC (or STBIW_REALLOC_SIZED)."
  141. #endif
  142. #ifndef STBIW_MALLOC
  143. #define STBIW_MALLOC(sz) malloc(sz)
  144. #define STBIW_REALLOC(p,newsz) realloc(p,newsz)
  145. #define STBIW_FREE(p) free(p)
  146. #endif
  147. #ifndef STBIW_REALLOC_SIZED
  148. #define STBIW_REALLOC_SIZED(p,oldsz,newsz) STBIW_REALLOC(p,newsz)
  149. #endif
  150. #ifndef STBIW_MEMMOVE
  151. #define STBIW_MEMMOVE(a,b,sz) memmove(a,b,sz)
  152. #endif
  153. #ifndef STBIW_ASSERT
  154. #include <assert.h>
  155. #define STBIW_ASSERT(x) assert(x)
  156. #endif
  157. #define STBIW_UCHAR(x) (unsigned char) ((x) & 0xff)
  158. typedef struct
  159. {
  160. stbi_write_func *func;
  161. void *context;
  162. } stbi__write_context;
  163. // initialize a callback-based context
  164. static void stbi__start_write_callbacks(stbi__write_context *s, stbi_write_func *c, void *context)
  165. {
  166. s->func = c;
  167. s->context = context;
  168. }
  169. #ifndef STBI_WRITE_NO_STDIO
  170. static void stbi__stdio_write(void *context, void *data, int size)
  171. {
  172. fwrite(data,1,size,(FILE*) context);
  173. }
  174. static int stbi__start_write_file(stbi__write_context *s, const char *filename)
  175. {
  176. // Urho3D: proper UTF8 handling for Windows, requires Urho3D WString class
  177. #ifndef _WIN32
  178. FILE *f = fopen(filename, "wb");
  179. #else
  180. Atomic::WString wstr(filename);
  181. FILE *f = _wfopen(wstr.CString(), L"wb");
  182. #endif
  183. stbi__start_write_callbacks(s, stbi__stdio_write, (void *) f);
  184. return f != NULL;
  185. }
  186. static void stbi__end_write_file(stbi__write_context *s)
  187. {
  188. fclose((FILE *)s->context);
  189. }
  190. #endif // !STBI_WRITE_NO_STDIO
  191. typedef unsigned int stbiw_uint32;
  192. typedef int stb_image_write_test[sizeof(stbiw_uint32)==4 ? 1 : -1];
  193. #ifdef STB_IMAGE_WRITE_STATIC
  194. static int stbi_write_tga_with_rle = 1;
  195. #else
  196. int stbi_write_tga_with_rle = 1;
  197. #endif
  198. static void stbiw__writefv(stbi__write_context *s, const char *fmt, va_list v)
  199. {
  200. while (*fmt) {
  201. switch (*fmt++) {
  202. case ' ': break;
  203. case '1': { unsigned char x = STBIW_UCHAR(va_arg(v, int));
  204. s->func(s->context,&x,1);
  205. break; }
  206. case '2': { int x = va_arg(v,int);
  207. unsigned char b[2];
  208. b[0] = STBIW_UCHAR(x);
  209. b[1] = STBIW_UCHAR(x>>8);
  210. s->func(s->context,b,2);
  211. break; }
  212. case '4': { stbiw_uint32 x = va_arg(v,int);
  213. unsigned char b[4];
  214. b[0]=STBIW_UCHAR(x);
  215. b[1]=STBIW_UCHAR(x>>8);
  216. b[2]=STBIW_UCHAR(x>>16);
  217. b[3]=STBIW_UCHAR(x>>24);
  218. s->func(s->context,b,4);
  219. break; }
  220. default:
  221. STBIW_ASSERT(0);
  222. return;
  223. }
  224. }
  225. }
  226. static void stbiw__writef(stbi__write_context *s, const char *fmt, ...)
  227. {
  228. va_list v;
  229. va_start(v, fmt);
  230. stbiw__writefv(s, fmt, v);
  231. va_end(v);
  232. }
  233. static void stbiw__write3(stbi__write_context *s, unsigned char a, unsigned char b, unsigned char c)
  234. {
  235. unsigned char arr[3];
  236. arr[0] = a, arr[1] = b, arr[2] = c;
  237. s->func(s->context, arr, 3);
  238. }
  239. static void stbiw__write_pixel(stbi__write_context *s, int rgb_dir, int comp, int write_alpha, int expand_mono, unsigned char *d)
  240. {
  241. unsigned char bg[3] = { 255, 0, 255}, px[3];
  242. int k;
  243. if (write_alpha < 0)
  244. s->func(s->context, &d[comp - 1], 1);
  245. switch (comp) {
  246. case 2: // 2 pixels = mono + alpha, alpha is written separately, so same as 1-channel case
  247. case 1:
  248. if (expand_mono)
  249. stbiw__write3(s, d[0], d[0], d[0]); // monochrome bmp
  250. else
  251. s->func(s->context, d, 1); // monochrome TGA
  252. break;
  253. case 4:
  254. if (!write_alpha) {
  255. // composite against pink background
  256. for (k = 0; k < 3; ++k)
  257. px[k] = bg[k] + ((d[k] - bg[k]) * d[3]) / 255;
  258. stbiw__write3(s, px[1 - rgb_dir], px[1], px[1 + rgb_dir]);
  259. break;
  260. }
  261. /* FALLTHROUGH */
  262. case 3:
  263. stbiw__write3(s, d[1 - rgb_dir], d[1], d[1 + rgb_dir]);
  264. break;
  265. }
  266. if (write_alpha > 0)
  267. s->func(s->context, &d[comp - 1], 1);
  268. }
  269. 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)
  270. {
  271. stbiw_uint32 zero = 0;
  272. int i,j, j_end;
  273. if (y <= 0)
  274. return;
  275. if (vdir < 0)
  276. j_end = -1, j = y-1;
  277. else
  278. j_end = y, j = 0;
  279. for (; j != j_end; j += vdir) {
  280. for (i=0; i < x; ++i) {
  281. unsigned char *d = (unsigned char *) data + (j*x+i)*comp;
  282. stbiw__write_pixel(s, rgb_dir, comp, write_alpha, expand_mono, d);
  283. }
  284. s->func(s->context, &zero, scanline_pad);
  285. }
  286. }
  287. 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, ...)
  288. {
  289. if (y < 0 || x < 0) {
  290. return 0;
  291. } else {
  292. va_list v;
  293. va_start(v, fmt);
  294. stbiw__writefv(s, fmt, v);
  295. va_end(v);
  296. stbiw__write_pixels(s,rgb_dir,vdir,x,y,comp,data,alpha,pad, expand_mono);
  297. return 1;
  298. }
  299. }
  300. static int stbi_write_bmp_core(stbi__write_context *s, int x, int y, int comp, const void *data)
  301. {
  302. int pad = (-x*3) & 3;
  303. return stbiw__outfile(s,-1,-1,x,y,comp,1,(void *) data,0,pad,
  304. "11 4 22 4" "4 44 22 444444",
  305. 'B', 'M', 14+40+(x*3+pad)*y, 0,0, 14+40, // file header
  306. 40, x,y, 1,24, 0,0,0,0,0,0); // bitmap header
  307. }
  308. STBIWDEF int stbi_write_bmp_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data)
  309. {
  310. stbi__write_context s;
  311. stbi__start_write_callbacks(&s, func, context);
  312. return stbi_write_bmp_core(&s, x, y, comp, data);
  313. }
  314. #ifndef STBI_WRITE_NO_STDIO
  315. STBIWDEF int stbi_write_bmp(char const *filename, int x, int y, int comp, const void *data)
  316. {
  317. stbi__write_context s;
  318. if (stbi__start_write_file(&s,filename)) {
  319. int r = stbi_write_bmp_core(&s, x, y, comp, data);
  320. stbi__end_write_file(&s);
  321. return r;
  322. } else
  323. return 0;
  324. }
  325. #endif //!STBI_WRITE_NO_STDIO
  326. static int stbi_write_tga_core(stbi__write_context *s, int x, int y, int comp, void *data)
  327. {
  328. int has_alpha = (comp == 2 || comp == 4);
  329. int colorbytes = has_alpha ? comp-1 : comp;
  330. int format = colorbytes < 2 ? 3 : 2; // 3 color channels (RGB/RGBA) = 2, 1 color channel (Y/YA) = 3
  331. if (y < 0 || x < 0)
  332. return 0;
  333. if (!stbi_write_tga_with_rle) {
  334. return stbiw__outfile(s, -1, -1, x, y, comp, 0, (void *) data, has_alpha, 0,
  335. "111 221 2222 11", 0, 0, format, 0, 0, 0, 0, 0, x, y, (colorbytes + has_alpha) * 8, has_alpha * 8);
  336. } else {
  337. int i,j,k;
  338. 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);
  339. for (j = y - 1; j >= 0; --j) {
  340. unsigned char *row = (unsigned char *) data + j * x * comp;
  341. int len;
  342. for (i = 0; i < x; i += len) {
  343. unsigned char *begin = row + i * comp;
  344. int diff = 1;
  345. len = 1;
  346. if (i < x - 1) {
  347. ++len;
  348. diff = memcmp(begin, row + (i + 1) * comp, comp);
  349. if (diff) {
  350. const unsigned char *prev = begin;
  351. for (k = i + 2; k < x && len < 128; ++k) {
  352. if (memcmp(prev, row + k * comp, comp)) {
  353. prev += comp;
  354. ++len;
  355. } else {
  356. --len;
  357. break;
  358. }
  359. }
  360. } else {
  361. for (k = i + 2; k < x && len < 128; ++k) {
  362. if (!memcmp(begin, row + k * comp, comp)) {
  363. ++len;
  364. } else {
  365. break;
  366. }
  367. }
  368. }
  369. }
  370. if (diff) {
  371. unsigned char header = STBIW_UCHAR(len - 1);
  372. s->func(s->context, &header, 1);
  373. for (k = 0; k < len; ++k) {
  374. stbiw__write_pixel(s, -1, comp, has_alpha, 0, begin + k * comp);
  375. }
  376. } else {
  377. unsigned char header = STBIW_UCHAR(len - 129);
  378. s->func(s->context, &header, 1);
  379. stbiw__write_pixel(s, -1, comp, has_alpha, 0, begin);
  380. }
  381. }
  382. }
  383. }
  384. return 1;
  385. }
  386. int stbi_write_tga_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const void *data)
  387. {
  388. stbi__write_context s;
  389. stbi__start_write_callbacks(&s, func, context);
  390. return stbi_write_tga_core(&s, x, y, comp, (void *) data);
  391. }
  392. #ifndef STBI_WRITE_NO_STDIO
  393. int stbi_write_tga(char const *filename, int x, int y, int comp, const void *data)
  394. {
  395. stbi__write_context s;
  396. if (stbi__start_write_file(&s,filename)) {
  397. int r = stbi_write_tga_core(&s, x, y, comp, (void *) data);
  398. stbi__end_write_file(&s);
  399. return r;
  400. } else
  401. return 0;
  402. }
  403. #endif
  404. // *************************************************************************************************
  405. // Radiance RGBE HDR writer
  406. // by Baldur Karlsson
  407. #define stbiw__max(a, b) ((a) > (b) ? (a) : (b))
  408. void stbiw__linear_to_rgbe(unsigned char *rgbe, float *linear)
  409. {
  410. int exponent;
  411. float maxcomp = stbiw__max(linear[0], stbiw__max(linear[1], linear[2]));
  412. if (maxcomp < 1e-32f) {
  413. rgbe[0] = rgbe[1] = rgbe[2] = rgbe[3] = 0;
  414. } else {
  415. float normalize = (float) frexp(maxcomp, &exponent) * 256.0f/maxcomp;
  416. rgbe[0] = (unsigned char)(linear[0] * normalize);
  417. rgbe[1] = (unsigned char)(linear[1] * normalize);
  418. rgbe[2] = (unsigned char)(linear[2] * normalize);
  419. rgbe[3] = (unsigned char)(exponent + 128);
  420. }
  421. }
  422. void stbiw__write_run_data(stbi__write_context *s, int length, unsigned char databyte)
  423. {
  424. unsigned char lengthbyte = STBIW_UCHAR(length+128);
  425. STBIW_ASSERT(length+128 <= 255);
  426. s->func(s->context, &lengthbyte, 1);
  427. s->func(s->context, &databyte, 1);
  428. }
  429. void stbiw__write_dump_data(stbi__write_context *s, int length, unsigned char *data)
  430. {
  431. unsigned char lengthbyte = STBIW_UCHAR(length);
  432. STBIW_ASSERT(length <= 128); // inconsistent with spec but consistent with official code
  433. s->func(s->context, &lengthbyte, 1);
  434. s->func(s->context, data, length);
  435. }
  436. void stbiw__write_hdr_scanline(stbi__write_context *s, int width, int ncomp, unsigned char *scratch, float *scanline)
  437. {
  438. unsigned char scanlineheader[4] = { 2, 2, 0, 0 };
  439. unsigned char rgbe[4];
  440. float linear[3];
  441. int x;
  442. scanlineheader[2] = (width&0xff00)>>8;
  443. scanlineheader[3] = (width&0x00ff);
  444. /* skip RLE for images too small or large */
  445. if (width < 8 || width >= 32768) {
  446. for (x=0; x < width; x++) {
  447. switch (ncomp) {
  448. case 4: /* fallthrough */
  449. case 3: linear[2] = scanline[x*ncomp + 2];
  450. linear[1] = scanline[x*ncomp + 1];
  451. linear[0] = scanline[x*ncomp + 0];
  452. break;
  453. default:
  454. linear[0] = linear[1] = linear[2] = scanline[x*ncomp + 0];
  455. break;
  456. }
  457. stbiw__linear_to_rgbe(rgbe, linear);
  458. s->func(s->context, rgbe, 4);
  459. }
  460. } else {
  461. int c,r;
  462. /* encode into scratch buffer */
  463. for (x=0; x < width; x++) {
  464. switch(ncomp) {
  465. case 4: /* fallthrough */
  466. case 3: linear[2] = scanline[x*ncomp + 2];
  467. linear[1] = scanline[x*ncomp + 1];
  468. linear[0] = scanline[x*ncomp + 0];
  469. break;
  470. default:
  471. linear[0] = linear[1] = linear[2] = scanline[x*ncomp + 0];
  472. break;
  473. }
  474. stbiw__linear_to_rgbe(rgbe, linear);
  475. scratch[x + width*0] = rgbe[0];
  476. scratch[x + width*1] = rgbe[1];
  477. scratch[x + width*2] = rgbe[2];
  478. scratch[x + width*3] = rgbe[3];
  479. }
  480. s->func(s->context, scanlineheader, 4);
  481. /* RLE each component separately */
  482. for (c=0; c < 4; c++) {
  483. unsigned char *comp = &scratch[width*c];
  484. x = 0;
  485. while (x < width) {
  486. // find first run
  487. r = x;
  488. while (r+2 < width) {
  489. if (comp[r] == comp[r+1] && comp[r] == comp[r+2])
  490. break;
  491. ++r;
  492. }
  493. if (r+2 >= width)
  494. r = width;
  495. // dump up to first run
  496. while (x < r) {
  497. int len = r-x;
  498. if (len > 128) len = 128;
  499. stbiw__write_dump_data(s, len, &comp[x]);
  500. x += len;
  501. }
  502. // if there's a run, output it
  503. if (r+2 < width) { // same test as what we break out of in search loop, so only true if we break'd
  504. // find next byte after run
  505. while (r < width && comp[r] == comp[x])
  506. ++r;
  507. // output run up to r
  508. while (x < r) {
  509. int len = r-x;
  510. if (len > 127) len = 127;
  511. stbiw__write_run_data(s, len, comp[x]);
  512. x += len;
  513. }
  514. }
  515. }
  516. }
  517. }
  518. }
  519. static int stbi_write_hdr_core(stbi__write_context *s, int x, int y, int comp, float *data)
  520. {
  521. if (y <= 0 || x <= 0 || data == NULL)
  522. return 0;
  523. else {
  524. // Each component is stored separately. Allocate scratch space for full output scanline.
  525. unsigned char *scratch = (unsigned char *) STBIW_MALLOC(x*4);
  526. int i, len;
  527. char buffer[128];
  528. char header[] = "#?RADIANCE\n# Written by stb_image_write.h\nFORMAT=32-bit_rle_rgbe\n";
  529. s->func(s->context, header, sizeof(header)-1);
  530. len = sprintf(buffer, "EXPOSURE= 1.0000000000000\n\n-Y %d +X %d\n", y, x);
  531. s->func(s->context, buffer, len);
  532. for(i=0; i < y; i++)
  533. stbiw__write_hdr_scanline(s, x, comp, scratch, data + comp*i*x);
  534. STBIW_FREE(scratch);
  535. return 1;
  536. }
  537. }
  538. int stbi_write_hdr_to_func(stbi_write_func *func, void *context, int x, int y, int comp, const float *data)
  539. {
  540. stbi__write_context s;
  541. stbi__start_write_callbacks(&s, func, context);
  542. return stbi_write_hdr_core(&s, x, y, comp, (float *) data);
  543. }
  544. #ifndef STBI_WRITE_NO_STDIO
  545. int stbi_write_hdr(char const *filename, int x, int y, int comp, const float *data)
  546. {
  547. stbi__write_context s;
  548. if (stbi__start_write_file(&s,filename)) {
  549. int r = stbi_write_hdr_core(&s, x, y, comp, (float *) data);
  550. stbi__end_write_file(&s);
  551. return r;
  552. } else
  553. return 0;
  554. }
  555. #endif // STBI_WRITE_NO_STDIO
  556. //////////////////////////////////////////////////////////////////////////////
  557. //
  558. // PNG writer
  559. //
  560. // stretchy buffer; stbiw__sbpush() == vector<>::push_back() -- stbiw__sbcount() == vector<>::size()
  561. #define stbiw__sbraw(a) ((int *) (a) - 2)
  562. #define stbiw__sbm(a) stbiw__sbraw(a)[0]
  563. #define stbiw__sbn(a) stbiw__sbraw(a)[1]
  564. #define stbiw__sbneedgrow(a,n) ((a)==0 || stbiw__sbn(a)+n >= stbiw__sbm(a))
  565. #define stbiw__sbmaybegrow(a,n) (stbiw__sbneedgrow(a,(n)) ? stbiw__sbgrow(a,n) : 0)
  566. #define stbiw__sbgrow(a,n) stbiw__sbgrowf((void **) &(a), (n), sizeof(*(a)))
  567. #define stbiw__sbpush(a, v) (stbiw__sbmaybegrow(a,1), (a)[stbiw__sbn(a)++] = (v))
  568. #define stbiw__sbcount(a) ((a) ? stbiw__sbn(a) : 0)
  569. #define stbiw__sbfree(a) ((a) ? STBIW_FREE(stbiw__sbraw(a)),0 : 0)
  570. static void *stbiw__sbgrowf(void **arr, int increment, int itemsize)
  571. {
  572. int m = *arr ? 2*stbiw__sbm(*arr)+increment : increment+1;
  573. void *p = STBIW_REALLOC_SIZED(*arr ? stbiw__sbraw(*arr) : 0, *arr ? (stbiw__sbm(*arr)*itemsize + sizeof(int)*2) : 0, itemsize * m + sizeof(int)*2);
  574. STBIW_ASSERT(p);
  575. if (p) {
  576. if (!*arr) ((int *) p)[1] = 0;
  577. *arr = (void *) ((int *) p + 2);
  578. stbiw__sbm(*arr) = m;
  579. }
  580. return *arr;
  581. }
  582. static unsigned char *stbiw__zlib_flushf(unsigned char *data, unsigned int *bitbuffer, int *bitcount)
  583. {
  584. while (*bitcount >= 8) {
  585. stbiw__sbpush(data, STBIW_UCHAR(*bitbuffer));
  586. *bitbuffer >>= 8;
  587. *bitcount -= 8;
  588. }
  589. return data;
  590. }
  591. static int stbiw__zlib_bitrev(int code, int codebits)
  592. {
  593. int res=0;
  594. while (codebits--) {
  595. res = (res << 1) | (code & 1);
  596. code >>= 1;
  597. }
  598. return res;
  599. }
  600. static unsigned int stbiw__zlib_countm(unsigned char *a, unsigned char *b, int limit)
  601. {
  602. int i;
  603. for (i=0; i < limit && i < 258; ++i)
  604. if (a[i] != b[i]) break;
  605. return i;
  606. }
  607. static unsigned int stbiw__zhash(unsigned char *data)
  608. {
  609. stbiw_uint32 hash = data[0] + (data[1] << 8) + (data[2] << 16);
  610. hash ^= hash << 3;
  611. hash += hash >> 5;
  612. hash ^= hash << 4;
  613. hash += hash >> 17;
  614. hash ^= hash << 25;
  615. hash += hash >> 6;
  616. return hash;
  617. }
  618. #define stbiw__zlib_flush() (out = stbiw__zlib_flushf(out, &bitbuf, &bitcount))
  619. #define stbiw__zlib_add(code,codebits) \
  620. (bitbuf |= (code) << bitcount, bitcount += (codebits), stbiw__zlib_flush())
  621. #define stbiw__zlib_huffa(b,c) stbiw__zlib_add(stbiw__zlib_bitrev(b,c),c)
  622. // default huffman tables
  623. #define stbiw__zlib_huff1(n) stbiw__zlib_huffa(0x30 + (n), 8)
  624. #define stbiw__zlib_huff2(n) stbiw__zlib_huffa(0x190 + (n)-144, 9)
  625. #define stbiw__zlib_huff3(n) stbiw__zlib_huffa(0 + (n)-256,7)
  626. #define stbiw__zlib_huff4(n) stbiw__zlib_huffa(0xc0 + (n)-280,8)
  627. #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))
  628. #define stbiw__zlib_huffb(n) ((n) <= 143 ? stbiw__zlib_huff1(n) : stbiw__zlib_huff2(n))
  629. #define stbiw__ZHASH 16384
  630. unsigned char * stbi_zlib_compress(unsigned char *data, int data_len, int *out_len, int quality)
  631. {
  632. 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 };
  633. 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 };
  634. 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 };
  635. 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 };
  636. unsigned int bitbuf=0;
  637. int i,j, bitcount=0;
  638. unsigned char *out = NULL;
  639. unsigned char ***hash_table = (unsigned char***) STBIW_MALLOC(stbiw__ZHASH * sizeof(char**));
  640. if (quality < 5) quality = 5;
  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. for (i=0; i < stbiw__ZHASH; ++i)
  646. hash_table[i] = NULL;
  647. i=0;
  648. while (i < data_len-3) {
  649. // hash next 3 bytes of data to be compressed
  650. int h = stbiw__zhash(data+i)&(stbiw__ZHASH-1), best=3;
  651. unsigned char *bestloc = 0;
  652. unsigned char **hlist = hash_table[h];
  653. int n = stbiw__sbcount(hlist);
  654. for (j=0; j < n; ++j) {
  655. if (hlist[j]-data > i-32768) { // if entry lies within window
  656. int d = stbiw__zlib_countm(hlist[j], data+i, data_len-i);
  657. if (d >= best) best=d,bestloc=hlist[j];
  658. }
  659. }
  660. // when hash table entry is too long, delete half the entries
  661. if (hash_table[h] && stbiw__sbn(hash_table[h]) == 2*quality) {
  662. STBIW_MEMMOVE(hash_table[h], hash_table[h]+quality, sizeof(hash_table[h][0])*quality);
  663. stbiw__sbn(hash_table[h]) = quality;
  664. }
  665. stbiw__sbpush(hash_table[h],data+i);
  666. if (bestloc) {
  667. // "lazy matching" - check match at *next* byte, and if it's better, do cur byte as literal
  668. h = stbiw__zhash(data+i+1)&(stbiw__ZHASH-1);
  669. hlist = hash_table[h];
  670. n = stbiw__sbcount(hlist);
  671. for (j=0; j < n; ++j) {
  672. if (hlist[j]-data > i-32767) {
  673. int e = stbiw__zlib_countm(hlist[j], data+i+1, data_len-i-1);
  674. if (e > best) { // if next match is better, bail on current match
  675. bestloc = NULL;
  676. break;
  677. }
  678. }
  679. }
  680. }
  681. if (bestloc) {
  682. int d = (int) (data+i - bestloc); // distance back
  683. STBIW_ASSERT(d <= 32767 && best <= 258);
  684. for (j=0; best > lengthc[j+1]-1; ++j);
  685. stbiw__zlib_huff(j+257);
  686. if (lengtheb[j]) stbiw__zlib_add(best - lengthc[j], lengtheb[j]);
  687. for (j=0; d > distc[j+1]-1; ++j);
  688. stbiw__zlib_add(stbiw__zlib_bitrev(j,5),5);
  689. if (disteb[j]) stbiw__zlib_add(d - distc[j], disteb[j]);
  690. i += best;
  691. } else {
  692. stbiw__zlib_huffb(data[i]);
  693. ++i;
  694. }
  695. }
  696. // write out final bytes
  697. for (;i < data_len; ++i)
  698. stbiw__zlib_huffb(data[i]);
  699. stbiw__zlib_huff(256); // end of block
  700. // pad with 0 bits to byte boundary
  701. while (bitcount)
  702. stbiw__zlib_add(0,1);
  703. for (i=0; i < stbiw__ZHASH; ++i)
  704. (void) stbiw__sbfree(hash_table[i]);
  705. STBIW_FREE(hash_table);
  706. {
  707. // compute adler32 on input
  708. unsigned int s1=1, s2=0;
  709. int blocklen = (int) (data_len % 5552);
  710. j=0;
  711. while (j < data_len) {
  712. for (i=0; i < blocklen; ++i) s1 += data[j+i], s2 += s1;
  713. s1 %= 65521, s2 %= 65521;
  714. j += blocklen;
  715. blocklen = 5552;
  716. }
  717. stbiw__sbpush(out, STBIW_UCHAR(s2 >> 8));
  718. stbiw__sbpush(out, STBIW_UCHAR(s2));
  719. stbiw__sbpush(out, STBIW_UCHAR(s1 >> 8));
  720. stbiw__sbpush(out, STBIW_UCHAR(s1));
  721. }
  722. *out_len = stbiw__sbn(out);
  723. // make returned pointer freeable
  724. STBIW_MEMMOVE(stbiw__sbraw(out), out, *out_len);
  725. return (unsigned char *) stbiw__sbraw(out);
  726. }
  727. static unsigned int stbiw__crc32(unsigned char *buffer, int len)
  728. {
  729. static unsigned int crc_table[256] =
  730. {
  731. 0x00000000, 0x77073096, 0xEE0E612C, 0x990951BA, 0x076DC419, 0x706AF48F, 0xE963A535, 0x9E6495A3,
  732. 0x0eDB8832, 0x79DCB8A4, 0xE0D5E91E, 0x97D2D988, 0x09B64C2B, 0x7EB17CBD, 0xE7B82D07, 0x90BF1D91,
  733. 0x1DB71064, 0x6AB020F2, 0xF3B97148, 0x84BE41DE, 0x1ADAD47D, 0x6DDDE4EB, 0xF4D4B551, 0x83D385C7,
  734. 0x136C9856, 0x646BA8C0, 0xFD62F97A, 0x8A65C9EC, 0x14015C4F, 0x63066CD9, 0xFA0F3D63, 0x8D080DF5,
  735. 0x3B6E20C8, 0x4C69105E, 0xD56041E4, 0xA2677172, 0x3C03E4D1, 0x4B04D447, 0xD20D85FD, 0xA50AB56B,
  736. 0x35B5A8FA, 0x42B2986C, 0xDBBBC9D6, 0xACBCF940, 0x32D86CE3, 0x45DF5C75, 0xDCD60DCF, 0xABD13D59,
  737. 0x26D930AC, 0x51DE003A, 0xC8D75180, 0xBFD06116, 0x21B4F4B5, 0x56B3C423, 0xCFBA9599, 0xB8BDA50F,
  738. 0x2802B89E, 0x5F058808, 0xC60CD9B2, 0xB10BE924, 0x2F6F7C87, 0x58684C11, 0xC1611DAB, 0xB6662D3D,
  739. 0x76DC4190, 0x01DB7106, 0x98D220BC, 0xEFD5102A, 0x71B18589, 0x06B6B51F, 0x9FBFE4A5, 0xE8B8D433,
  740. 0x7807C9A2, 0x0F00F934, 0x9609A88E, 0xE10E9818, 0x7F6A0DBB, 0x086D3D2D, 0x91646C97, 0xE6635C01,
  741. 0x6B6B51F4, 0x1C6C6162, 0x856530D8, 0xF262004E, 0x6C0695ED, 0x1B01A57B, 0x8208F4C1, 0xF50FC457,
  742. 0x65B0D9C6, 0x12B7E950, 0x8BBEB8EA, 0xFCB9887C, 0x62DD1DDF, 0x15DA2D49, 0x8CD37CF3, 0xFBD44C65,
  743. 0x4DB26158, 0x3AB551CE, 0xA3BC0074, 0xD4BB30E2, 0x4ADFA541, 0x3DD895D7, 0xA4D1C46D, 0xD3D6F4FB,
  744. 0x4369E96A, 0x346ED9FC, 0xAD678846, 0xDA60B8D0, 0x44042D73, 0x33031DE5, 0xAA0A4C5F, 0xDD0D7CC9,
  745. 0x5005713C, 0x270241AA, 0xBE0B1010, 0xC90C2086, 0x5768B525, 0x206F85B3, 0xB966D409, 0xCE61E49F,
  746. 0x5EDEF90E, 0x29D9C998, 0xB0D09822, 0xC7D7A8B4, 0x59B33D17, 0x2EB40D81, 0xB7BD5C3B, 0xC0BA6CAD,
  747. 0xEDB88320, 0x9ABFB3B6, 0x03B6E20C, 0x74B1D29A, 0xEAD54739, 0x9DD277AF, 0x04DB2615, 0x73DC1683,
  748. 0xE3630B12, 0x94643B84, 0x0D6D6A3E, 0x7A6A5AA8, 0xE40ECF0B, 0x9309FF9D, 0x0A00AE27, 0x7D079EB1,
  749. 0xF00F9344, 0x8708A3D2, 0x1E01F268, 0x6906C2FE, 0xF762575D, 0x806567CB, 0x196C3671, 0x6E6B06E7,
  750. 0xFED41B76, 0x89D32BE0, 0x10DA7A5A, 0x67DD4ACC, 0xF9B9DF6F, 0x8EBEEFF9, 0x17B7BE43, 0x60B08ED5,
  751. 0xD6D6A3E8, 0xA1D1937E, 0x38D8C2C4, 0x4FDFF252, 0xD1BB67F1, 0xA6BC5767, 0x3FB506DD, 0x48B2364B,
  752. 0xD80D2BDA, 0xAF0A1B4C, 0x36034AF6, 0x41047A60, 0xDF60EFC3, 0xA867DF55, 0x316E8EEF, 0x4669BE79,
  753. 0xCB61B38C, 0xBC66831A, 0x256FD2A0, 0x5268E236, 0xCC0C7795, 0xBB0B4703, 0x220216B9, 0x5505262F,
  754. 0xC5BA3BBE, 0xB2BD0B28, 0x2BB45A92, 0x5CB36A04, 0xC2D7FFA7, 0xB5D0CF31, 0x2CD99E8B, 0x5BDEAE1D,
  755. 0x9B64C2B0, 0xEC63F226, 0x756AA39C, 0x026D930A, 0x9C0906A9, 0xEB0E363F, 0x72076785, 0x05005713,
  756. 0x95BF4A82, 0xE2B87A14, 0x7BB12BAE, 0x0CB61B38, 0x92D28E9B, 0xE5D5BE0D, 0x7CDCEFB7, 0x0BDBDF21,
  757. 0x86D3D2D4, 0xF1D4E242, 0x68DDB3F8, 0x1FDA836E, 0x81BE16CD, 0xF6B9265B, 0x6FB077E1, 0x18B74777,
  758. 0x88085AE6, 0xFF0F6A70, 0x66063BCA, 0x11010B5C, 0x8F659EFF, 0xF862AE69, 0x616BFFD3, 0x166CCF45,
  759. 0xA00AE278, 0xD70DD2EE, 0x4E048354, 0x3903B3C2, 0xA7672661, 0xD06016F7, 0x4969474D, 0x3E6E77DB,
  760. 0xAED16A4A, 0xD9D65ADC, 0x40DF0B66, 0x37D83BF0, 0xA9BCAE53, 0xDEBB9EC5, 0x47B2CF7F, 0x30B5FFE9,
  761. 0xBDBDF21C, 0xCABAC28A, 0x53B39330, 0x24B4A3A6, 0xBAD03605, 0xCDD70693, 0x54DE5729, 0x23D967BF,
  762. 0xB3667A2E, 0xC4614AB8, 0x5D681B02, 0x2A6F2B94, 0xB40BBE37, 0xC30C8EA1, 0x5A05DF1B, 0x2D02EF8D
  763. };
  764. unsigned int crc = ~0u;
  765. int i;
  766. for (i=0; i < len; ++i)
  767. crc = (crc >> 8) ^ crc_table[buffer[i] ^ (crc & 0xff)];
  768. return ~crc;
  769. }
  770. #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)
  771. #define stbiw__wp32(data,v) stbiw__wpng4(data, (v)>>24,(v)>>16,(v)>>8,(v));
  772. #define stbiw__wptag(data,s) stbiw__wpng4(data, s[0],s[1],s[2],s[3])
  773. static void stbiw__wpcrc(unsigned char **data, int len)
  774. {
  775. unsigned int crc = stbiw__crc32(*data - len - 4, len+4);
  776. stbiw__wp32(*data, crc);
  777. }
  778. static unsigned char stbiw__paeth(int a, int b, int c)
  779. {
  780. int p = a + b - c, pa = abs(p-a), pb = abs(p-b), pc = abs(p-c);
  781. if (pa <= pb && pa <= pc) return STBIW_UCHAR(a);
  782. if (pb <= pc) return STBIW_UCHAR(b);
  783. return STBIW_UCHAR(c);
  784. }
  785. // @OPTIMIZE: provide an option that always forces left-predict or paeth predict
  786. unsigned char *stbi_write_png_to_mem(unsigned char *pixels, int stride_bytes, int x, int y, int n, int *out_len)
  787. {
  788. int ctype[5] = { -1, 0, 4, 2, 6 };
  789. unsigned char sig[8] = { 137,80,78,71,13,10,26,10 };
  790. unsigned char *out,*o, *filt, *zlib;
  791. signed char *line_buffer;
  792. int i,j,k,p,zlen;
  793. if (stride_bytes == 0)
  794. stride_bytes = x * n;
  795. filt = (unsigned char *) STBIW_MALLOC((x*n+1) * y); if (!filt) return 0;
  796. line_buffer = (signed char *) STBIW_MALLOC(x * n); if (!line_buffer) { STBIW_FREE(filt); return 0; }
  797. for (j=0; j < y; ++j) {
  798. static int mapping[] = { 0,1,2,3,4 };
  799. static int firstmap[] = { 0,1,0,5,6 };
  800. int *mymap = (j != 0) ? mapping : firstmap;
  801. int best = 0, bestval = 0x7fffffff;
  802. for (p=0; p < 2; ++p) {
  803. for (k= p?best:0; k < 5; ++k) { // @TODO: clarity: rewrite this to go 0..5, and 'continue' the unwanted ones during 2nd pass
  804. int type = mymap[k],est=0;
  805. unsigned char *z = pixels + stride_bytes*j;
  806. for (i=0; i < n; ++i)
  807. switch (type) {
  808. case 0: line_buffer[i] = z[i]; break;
  809. case 1: line_buffer[i] = z[i]; break;
  810. case 2: line_buffer[i] = z[i] - z[i-stride_bytes]; break;
  811. case 3: line_buffer[i] = z[i] - (z[i-stride_bytes]>>1); break;
  812. case 4: line_buffer[i] = (signed char) (z[i] - stbiw__paeth(0,z[i-stride_bytes],0)); break;
  813. case 5: line_buffer[i] = z[i]; break;
  814. case 6: line_buffer[i] = z[i]; break;
  815. }
  816. for (i=n; i < x*n; ++i) {
  817. switch (type) {
  818. case 0: line_buffer[i] = z[i]; break;
  819. case 1: line_buffer[i] = z[i] - z[i-n]; break;
  820. case 2: line_buffer[i] = z[i] - z[i-stride_bytes]; break;
  821. case 3: line_buffer[i] = z[i] - ((z[i-n] + z[i-stride_bytes])>>1); break;
  822. case 4: line_buffer[i] = z[i] - stbiw__paeth(z[i-n], z[i-stride_bytes], z[i-stride_bytes-n]); break;
  823. case 5: line_buffer[i] = z[i] - (z[i-n]>>1); break;
  824. case 6: line_buffer[i] = z[i] - stbiw__paeth(z[i-n], 0,0); break;
  825. }
  826. }
  827. if (p) break;
  828. for (i=0; i < x*n; ++i)
  829. est += abs((signed char) line_buffer[i]);
  830. if (est < bestval) { bestval = est; best = k; }
  831. }
  832. }
  833. // when we get here, best contains the filter type, and line_buffer contains the data
  834. filt[j*(x*n+1)] = (unsigned char) best;
  835. STBIW_MEMMOVE(filt+j*(x*n+1)+1, line_buffer, x*n);
  836. }
  837. STBIW_FREE(line_buffer);
  838. zlib = stbi_zlib_compress(filt, y*( x*n+1), &zlen, 8); // increase 8 to get smaller but use more memory
  839. STBIW_FREE(filt);
  840. if (!zlib) return 0;
  841. // each tag requires 12 bytes of overhead
  842. out = (unsigned char *) STBIW_MALLOC(8 + 12+13 + 12+zlen + 12);
  843. if (!out) return 0;
  844. *out_len = 8 + 12+13 + 12+zlen + 12;
  845. o=out;
  846. STBIW_MEMMOVE(o,sig,8); o+= 8;
  847. stbiw__wp32(o, 13); // header length
  848. stbiw__wptag(o, "IHDR");
  849. stbiw__wp32(o, x);
  850. stbiw__wp32(o, y);
  851. *o++ = 8;
  852. *o++ = STBIW_UCHAR(ctype[n]);
  853. *o++ = 0;
  854. *o++ = 0;
  855. *o++ = 0;
  856. stbiw__wpcrc(&o,13);
  857. stbiw__wp32(o, zlen);
  858. stbiw__wptag(o, "IDAT");
  859. STBIW_MEMMOVE(o, zlib, zlen);
  860. o += zlen;
  861. STBIW_FREE(zlib);
  862. stbiw__wpcrc(&o, zlen);
  863. stbiw__wp32(o,0);
  864. stbiw__wptag(o, "IEND");
  865. stbiw__wpcrc(&o,0);
  866. STBIW_ASSERT(o == out + *out_len);
  867. return out;
  868. }
  869. #ifndef STBI_WRITE_NO_STDIO
  870. STBIWDEF int stbi_write_png(char const *filename, int x, int y, int comp, const void *data, int stride_bytes)
  871. {
  872. FILE *f;
  873. int len;
  874. unsigned char *png = stbi_write_png_to_mem((unsigned char *) data, stride_bytes, x, y, comp, &len);
  875. if (png == NULL) return 0;
  876. f = fopen(filename, "wb");
  877. if (!f) { STBIW_FREE(png); return 0; }
  878. fwrite(png, 1, len, f);
  879. fclose(f);
  880. STBIW_FREE(png);
  881. return 1;
  882. }
  883. #endif
  884. 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)
  885. {
  886. int len;
  887. unsigned char *png = stbi_write_png_to_mem((unsigned char *) data, stride_bytes, x, y, comp, &len);
  888. if (png == NULL) return 0;
  889. func(context, png, len);
  890. STBIW_FREE(png);
  891. return 1;
  892. }
  893. #endif // STB_IMAGE_WRITE_IMPLEMENTATION
  894. /* Revision history
  895. 1.04 (2017-03-03)
  896. monochrome BMP expansion
  897. 1.03 ???
  898. 1.02 (2016-04-02)
  899. avoid allocating large structures on the stack
  900. 1.01 (2016-01-16)
  901. STBIW_REALLOC_SIZED: support allocators with no realloc support
  902. avoid race-condition in crc initialization
  903. minor compile issues
  904. 1.00 (2015-09-14)
  905. installable file IO function
  906. 0.99 (2015-09-13)
  907. warning fixes; TGA rle support
  908. 0.98 (2015-04-08)
  909. added STBIW_MALLOC, STBIW_ASSERT etc
  910. 0.97 (2015-01-18)
  911. fixed HDR asserts, rewrote HDR rle logic
  912. 0.96 (2015-01-17)
  913. add HDR output
  914. fix monochrome BMP
  915. 0.95 (2014-08-17)
  916. add monochrome TGA output
  917. 0.94 (2014-05-31)
  918. rename private functions to avoid conflicts with stb_image.h
  919. 0.93 (2014-05-27)
  920. warning fixes
  921. 0.92 (2010-08-01)
  922. casts to unsigned char to fix warnings
  923. 0.91 (2010-07-17)
  924. first public release
  925. 0.90 first internal release
  926. */
  927. /*
  928. ------------------------------------------------------------------------------
  929. This software is available under 2 licenses -- choose whichever you prefer.
  930. ------------------------------------------------------------------------------
  931. ALTERNATIVE A - MIT License
  932. Copyright (c) 2017 Sean Barrett
  933. Permission is hereby granted, free of charge, to any person obtaining a copy of
  934. this software and associated documentation files (the "Software"), to deal in
  935. the Software without restriction, including without limitation the rights to
  936. use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
  937. of the Software, and to permit persons to whom the Software is furnished to do
  938. so, subject to the following conditions:
  939. The above copyright notice and this permission notice shall be included in all
  940. copies or substantial portions of the Software.
  941. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  942. IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  943. FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  944. AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  945. LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  946. OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  947. SOFTWARE.
  948. ------------------------------------------------------------------------------
  949. ALTERNATIVE B - Public Domain (www.unlicense.org)
  950. This is free and unencumbered software released into the public domain.
  951. Anyone is free to copy, modify, publish, use, compile, sell, or distribute this
  952. software, either in source code form or as a compiled binary, for any purpose,
  953. commercial or non-commercial, and by any means.
  954. In jurisdictions that recognize copyright laws, the author or authors of this
  955. software dedicate any and all copyright interest in the software to the public
  956. domain. We make this dedication for the benefit of the public at large and to
  957. the detriment of our heirs and successors. We intend this dedication to be an
  958. overt act of relinquishment in perpetuity of all present and future rights to
  959. this software under copyright law.
  960. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  961. IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  962. FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  963. AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  964. ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  965. WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  966. ------------------------------------------------------------------------------
  967. */