Audio plugin host https://kx.studio/carla
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

601 lines
19KB

  1. /*
  2. ==============================================================================
  3. This file is part of the JUCE library.
  4. Copyright (c) 2017 - ROLI Ltd.
  5. JUCE is an open source library subject to commercial or open-source
  6. licensing.
  7. By using JUCE, you agree to the terms of both the JUCE 5 End-User License
  8. Agreement and JUCE 5 Privacy Policy (both updated and effective as of the
  9. 27th April 2017).
  10. End User License Agreement: www.juce.com/juce-5-licence
  11. Privacy Policy: www.juce.com/juce-5-privacy-policy
  12. Or: You may also use this code under the terms of the GPL v3 (see
  13. www.gnu.org/licenses).
  14. JUCE IS PROVIDED "AS IS" WITHOUT ANY WARRANTY, AND ALL WARRANTIES, WHETHER
  15. EXPRESSED OR IMPLIED, INCLUDING MERCHANTABILITY AND FITNESS FOR PURPOSE, ARE
  16. DISCLAIMED.
  17. ==============================================================================
  18. */
  19. #if JUCE_MSVC
  20. #pragma warning (push)
  21. #pragma warning (disable: 4390 4611 4365 4267)
  22. #ifdef __INTEL_COMPILER
  23. #pragma warning (disable: 2544 2545)
  24. #endif
  25. #endif
  26. namespace zlibNamespace
  27. {
  28. #if JUCE_INCLUDE_ZLIB_CODE
  29. #undef OS_CODE
  30. #undef fdopen
  31. #include "../../juce_core/zip/zlib/zlib.h"
  32. #undef OS_CODE
  33. #else
  34. #include JUCE_ZLIB_INCLUDE_PATH
  35. #endif
  36. }
  37. #if ! defined (jmp_buf) || ! defined (longjmp)
  38. #include <setjmp.h>
  39. #endif
  40. namespace pnglibNamespace
  41. {
  42. using namespace zlibNamespace;
  43. #if JUCE_INCLUDE_PNGLIB_CODE || ! defined (JUCE_INCLUDE_PNGLIB_CODE)
  44. #if _MSC_VER != 1310
  45. using std::calloc; // (causes conflict in VS.NET 2003)
  46. using std::malloc;
  47. using std::free;
  48. #endif
  49. #if JUCE_CLANG
  50. #pragma clang diagnostic push
  51. #pragma clang diagnostic ignored "-Wsign-conversion"
  52. #if __has_warning("-Wcomma")
  53. #pragma clang diagnostic ignored "-Wcomma"
  54. #endif
  55. #endif
  56. #undef check
  57. using std::abs;
  58. #define NO_DUMMY_DECL
  59. #define PNGLCONF_H 1
  60. #if JUCE_ANDROID
  61. #define PNG_ARM_NEON_SUPPORTED
  62. #endif
  63. #define PNG_16BIT_SUPPORTED
  64. #define PNG_ALIGNED_MEMORY_SUPPORTED
  65. #define PNG_BENIGN_ERRORS_SUPPORTED
  66. #define PNG_BENIGN_READ_ERRORS_SUPPORTED
  67. #define PNG_BUILD_GRAYSCALE_PALETTE_SUPPORTED
  68. #define PNG_CHECK_FOR_INVALID_INDEX_SUPPORTED
  69. #define PNG_COLORSPACE_SUPPORTED
  70. #define PNG_CONSOLE_IO_SUPPORTED
  71. #define PNG_EASY_ACCESS_SUPPORTED
  72. #define PNG_FIXED_POINT_SUPPORTED
  73. #define PNG_FLOATING_ARITHMETIC_SUPPORTED
  74. #define PNG_FLOATING_POINT_SUPPORTED
  75. #define PNG_FORMAT_AFIRST_SUPPORTED
  76. #define PNG_FORMAT_BGR_SUPPORTED
  77. #define PNG_GAMMA_SUPPORTED
  78. #define PNG_GET_PALETTE_MAX_SUPPORTED
  79. #define PNG_HANDLE_AS_UNKNOWN_SUPPORTED
  80. #define PNG_INCH_CONVERSIONS_SUPPORTED
  81. #define PNG_INFO_IMAGE_SUPPORTED
  82. #define PNG_IO_STATE_SUPPORTED
  83. #define PNG_MNG_FEATURES_SUPPORTED
  84. #define PNG_POINTER_INDEXING_SUPPORTED
  85. #define PNG_PROGRESSIVE_READ_SUPPORTED
  86. #define PNG_READ_16BIT_SUPPORTED
  87. #define PNG_READ_ALPHA_MODE_SUPPORTED
  88. #define PNG_READ_ANCILLARY_CHUNKS_SUPPORTED
  89. #define PNG_READ_BACKGROUND_SUPPORTED
  90. #define PNG_READ_BGR_SUPPORTED
  91. #define PNG_READ_CHECK_FOR_INVALID_INDEX_SUPPORTED
  92. #define PNG_READ_COMPOSITE_NODIV_SUPPORTED
  93. #define PNG_READ_COMPRESSED_TEXT_SUPPORTED
  94. #define PNG_READ_EXPAND_16_SUPPORTED
  95. #define PNG_READ_EXPAND_SUPPORTED
  96. #define PNG_READ_FILLER_SUPPORTED
  97. #define PNG_READ_GAMMA_SUPPORTED
  98. #define PNG_READ_GET_PALETTE_MAX_SUPPORTED
  99. #define PNG_READ_GRAY_TO_RGB_SUPPORTED
  100. #define PNG_READ_INTERLACING_SUPPORTED
  101. #define PNG_READ_INT_FUNCTIONS_SUPPORTED
  102. #define PNG_READ_INVERT_ALPHA_SUPPORTED
  103. #define PNG_READ_INVERT_SUPPORTED
  104. #define PNG_READ_OPT_PLTE_SUPPORTED
  105. #define PNG_READ_PACKSWAP_SUPPORTED
  106. #define PNG_READ_PACK_SUPPORTED
  107. #define PNG_READ_QUANTIZE_SUPPORTED
  108. #define PNG_READ_RGB_TO_GRAY_SUPPORTED
  109. #define PNG_READ_SCALE_16_TO_8_SUPPORTED
  110. #define PNG_READ_SHIFT_SUPPORTED
  111. #define PNG_READ_STRIP_16_TO_8_SUPPORTED
  112. #define PNG_READ_STRIP_ALPHA_SUPPORTED
  113. #define PNG_READ_SUPPORTED
  114. #define PNG_READ_SWAP_ALPHA_SUPPORTED
  115. #define PNG_READ_SWAP_SUPPORTED
  116. #define PNG_READ_TEXT_SUPPORTED
  117. #define PNG_READ_TRANSFORMS_SUPPORTED
  118. #define PNG_READ_UNKNOWN_CHUNKS_SUPPORTED
  119. #define PNG_READ_USER_CHUNKS_SUPPORTED
  120. #define PNG_READ_USER_TRANSFORM_SUPPORTED
  121. #define PNG_READ_bKGD_SUPPORTED
  122. #define PNG_READ_cHRM_SUPPORTED
  123. #define PNG_READ_gAMA_SUPPORTED
  124. #define PNG_READ_hIST_SUPPORTED
  125. #define PNG_READ_iCCP_SUPPORTED
  126. #define PNG_READ_iTXt_SUPPORTED
  127. #define PNG_READ_oFFs_SUPPORTED
  128. #define PNG_READ_pCAL_SUPPORTED
  129. #define PNG_READ_pHYs_SUPPORTED
  130. #define PNG_READ_sBIT_SUPPORTED
  131. #define PNG_READ_sCAL_SUPPORTED
  132. #define PNG_READ_sPLT_SUPPORTED
  133. #define PNG_READ_sRGB_SUPPORTED
  134. #define PNG_READ_tEXt_SUPPORTED
  135. #define PNG_READ_tIME_SUPPORTED
  136. #define PNG_READ_tRNS_SUPPORTED
  137. #define PNG_READ_zTXt_SUPPORTED
  138. #define PNG_SAVE_INT_32_SUPPORTED
  139. #define PNG_SAVE_UNKNOWN_CHUNKS_SUPPORTED
  140. #define PNG_SEQUENTIAL_READ_SUPPORTED
  141. #define PNG_SET_CHUNK_CACHE_LIMIT_SUPPORTED
  142. #define PNG_SET_CHUNK_MALLOC_LIMIT_SUPPORTED
  143. #define PNG_SET_UNKNOWN_CHUNKS_SUPPORTED
  144. #define PNG_SET_USER_LIMITS_SUPPORTED
  145. #define PNG_SIMPLIFIED_READ_AFIRST_SUPPORTED
  146. #define PNG_SIMPLIFIED_READ_BGR_SUPPORTED
  147. #define PNG_SIMPLIFIED_WRITE_AFIRST_SUPPORTED
  148. #define PNG_SIMPLIFIED_WRITE_BGR_SUPPORTED
  149. #define PNG_STDIO_SUPPORTED
  150. #define PNG_STORE_UNKNOWN_CHUNKS_SUPPORTED
  151. #define PNG_TEXT_SUPPORTED
  152. #define PNG_TIME_RFC1123_SUPPORTED
  153. #define PNG_UNKNOWN_CHUNKS_SUPPORTED
  154. #define PNG_USER_CHUNKS_SUPPORTED
  155. #define PNG_USER_LIMITS_SUPPORTED
  156. #define PNG_USER_TRANSFORM_INFO_SUPPORTED
  157. #define PNG_USER_TRANSFORM_PTR_SUPPORTED
  158. #define PNG_WARNINGS_SUPPORTED
  159. #define PNG_WRITE_16BIT_SUPPORTED
  160. #define PNG_WRITE_ANCILLARY_CHUNKS_SUPPORTED
  161. #define PNG_WRITE_BGR_SUPPORTED
  162. #define PNG_WRITE_CHECK_FOR_INVALID_INDEX_SUPPORTED
  163. #define PNG_WRITE_COMPRESSED_TEXT_SUPPORTED
  164. #define PNG_WRITE_CUSTOMIZE_ZTXT_COMPRESSION_SUPPORTED
  165. #define PNG_WRITE_FILLER_SUPPORTED
  166. #define PNG_WRITE_FILTER_SUPPORTED
  167. #define PNG_WRITE_FLUSH_SUPPORTED
  168. #define PNG_WRITE_GET_PALETTE_MAX_SUPPORTED
  169. #define PNG_WRITE_INTERLACING_SUPPORTED
  170. #define PNG_WRITE_INT_FUNCTIONS_SUPPORTED
  171. #define PNG_WRITE_INVERT_ALPHA_SUPPORTED
  172. #define PNG_WRITE_INVERT_SUPPORTED
  173. #define PNG_WRITE_OPTIMIZE_CMF_SUPPORTED
  174. #define PNG_WRITE_PACKSWAP_SUPPORTED
  175. #define PNG_WRITE_PACK_SUPPORTED
  176. #define PNG_WRITE_SHIFT_SUPPORTED
  177. #define PNG_WRITE_SUPPORTED
  178. #define PNG_WRITE_SWAP_ALPHA_SUPPORTED
  179. #define PNG_WRITE_SWAP_SUPPORTED
  180. #define PNG_WRITE_TEXT_SUPPORTED
  181. #define PNG_WRITE_TRANSFORMS_SUPPORTED
  182. #define PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED
  183. #define PNG_WRITE_USER_TRANSFORM_SUPPORTED
  184. #define PNG_WRITE_WEIGHTED_FILTER_SUPPORTED
  185. #define PNG_WRITE_bKGD_SUPPORTED
  186. #define PNG_WRITE_cHRM_SUPPORTED
  187. #define PNG_WRITE_gAMA_SUPPORTED
  188. #define PNG_WRITE_hIST_SUPPORTED
  189. #define PNG_WRITE_iCCP_SUPPORTED
  190. #define PNG_WRITE_iTXt_SUPPORTED
  191. #define PNG_WRITE_oFFs_SUPPORTED
  192. #define PNG_WRITE_pCAL_SUPPORTED
  193. #define PNG_WRITE_pHYs_SUPPORTED
  194. #define PNG_WRITE_sBIT_SUPPORTED
  195. #define PNG_WRITE_sCAL_SUPPORTED
  196. #define PNG_WRITE_sPLT_SUPPORTED
  197. #define PNG_WRITE_sRGB_SUPPORTED
  198. #define PNG_WRITE_tEXt_SUPPORTED
  199. #define PNG_WRITE_tIME_SUPPORTED
  200. #define PNG_WRITE_tRNS_SUPPORTED
  201. #define PNG_WRITE_zTXt_SUPPORTED
  202. #define PNG_bKGD_SUPPORTED
  203. #define PNG_cHRM_SUPPORTED
  204. #define PNG_gAMA_SUPPORTED
  205. #define PNG_hIST_SUPPORTED
  206. #define PNG_iCCP_SUPPORTED
  207. #define PNG_iTXt_SUPPORTED
  208. #define PNG_oFFs_SUPPORTED
  209. #define PNG_pCAL_SUPPORTED
  210. #define PNG_pHYs_SUPPORTED
  211. #define PNG_sBIT_SUPPORTED
  212. #define PNG_sCAL_SUPPORTED
  213. #define PNG_sPLT_SUPPORTED
  214. #define PNG_sRGB_SUPPORTED
  215. #define PNG_tEXt_SUPPORTED
  216. #define PNG_tIME_SUPPORTED
  217. #define PNG_tRNS_SUPPORTED
  218. #define PNG_zTXt_SUPPORTED
  219. #define PNG_STRING_COPYRIGHT "";
  220. #define PNG_STRING_NEWLINE "\n"
  221. #define PNG_LITERAL_SHARP 0x23
  222. #define PNG_LITERAL_LEFT_SQUARE_BRACKET 0x5b
  223. #define PNG_LITERAL_RIGHT_SQUARE_BRACKET 0x5d
  224. #define PNG_API_RULE 0
  225. #define PNG_CALLOC_SUPPORTED
  226. #define PNG_COST_SHIFT 3
  227. #define PNG_DEFAULT_READ_MACROS 1
  228. #define PNG_GAMMA_THRESHOLD_FIXED 5000
  229. #define PNG_IDAT_READ_SIZE PNG_ZBUF_SIZE
  230. #define PNG_INFLATE_BUF_SIZE 1024
  231. #define PNG_MAX_GAMMA_8 11
  232. #define PNG_QUANTIZE_BLUE_BITS 5
  233. #define PNG_QUANTIZE_GREEN_BITS 5
  234. #define PNG_QUANTIZE_RED_BITS 5
  235. #define PNG_TEXT_Z_DEFAULT_COMPRESSION (-1)
  236. #define PNG_TEXT_Z_DEFAULT_STRATEGY 0
  237. #define PNG_WEIGHT_SHIFT 8
  238. #define PNG_ZBUF_SIZE 8192
  239. #define PNG_Z_DEFAULT_COMPRESSION (-1)
  240. #define PNG_Z_DEFAULT_NOFILTER_STRATEGY 0
  241. #define PNG_Z_DEFAULT_STRATEGY 1
  242. #define PNG_sCAL_PRECISION 5
  243. #define PNG_sRGB_PROFILE_CHECKS 2
  244. #define png_debug(a, b)
  245. #define png_debug1(a, b, c)
  246. #define png_debug2(a, b, c, d)
  247. #include "pnglib/png.h"
  248. #include "pnglib/pngconf.h"
  249. #define PNG_NO_EXTERN
  250. #include "pnglib/png.c"
  251. #include "pnglib/pngerror.c"
  252. #include "pnglib/pngget.c"
  253. #include "pnglib/pngmem.c"
  254. #include "pnglib/pngread.c"
  255. #include "pnglib/pngpread.c"
  256. #include "pnglib/pngrio.c"
  257. #include "pnglib/pngrtran.c"
  258. #include "pnglib/pngrutil.c"
  259. #include "pnglib/pngset.c"
  260. #include "pnglib/pngtrans.c"
  261. #include "pnglib/pngwio.c"
  262. #include "pnglib/pngwrite.c"
  263. #include "pnglib/pngwtran.c"
  264. #include "pnglib/pngwutil.c"
  265. #if JUCE_CLANG
  266. #pragma clang diagnostic pop
  267. #endif
  268. #else
  269. extern "C"
  270. {
  271. #include <png.h>
  272. #include <pngconf.h>
  273. }
  274. #endif
  275. }
  276. #undef max
  277. #undef min
  278. #undef fdopen
  279. #if JUCE_MSVC
  280. #pragma warning (pop)
  281. #endif
  282. //==============================================================================
  283. namespace PNGHelpers
  284. {
  285. using namespace pnglibNamespace;
  286. static void JUCE_CDECL writeDataCallback (png_structp png, png_bytep data, png_size_t length)
  287. {
  288. static_cast<OutputStream*> (png_get_io_ptr (png))->write (data, length);
  289. }
  290. #if ! JUCE_USING_COREIMAGE_LOADER
  291. static void JUCE_CDECL readCallback (png_structp png, png_bytep data, png_size_t length)
  292. {
  293. static_cast<InputStream*> (png_get_io_ptr (png))->read (data, (int) length);
  294. }
  295. struct PNGErrorStruct {};
  296. static void JUCE_CDECL errorCallback (png_structp p, png_const_charp)
  297. {
  298. #ifdef PNG_SETJMP_SUPPORTED
  299. setjmp(png_jmpbuf(p));
  300. #else
  301. longjmp (*(jmp_buf*) p->error_ptr, 1);
  302. #endif
  303. }
  304. static void JUCE_CDECL warningCallback (png_structp, png_const_charp) {}
  305. #if JUCE_MSVC
  306. #pragma warning (push)
  307. #pragma warning (disable: 4611) // (warning about setjmp)
  308. #endif
  309. static bool readHeader (InputStream& in, png_structp pngReadStruct, png_infop pngInfoStruct, jmp_buf& errorJumpBuf,
  310. png_uint_32& width, png_uint_32& height, int& bitDepth, int& colorType, int& interlaceType) noexcept
  311. {
  312. if (setjmp (errorJumpBuf) == 0)
  313. {
  314. // read the header..
  315. png_set_read_fn (pngReadStruct, &in, readCallback);
  316. png_read_info (pngReadStruct, pngInfoStruct);
  317. png_get_IHDR (pngReadStruct, pngInfoStruct,
  318. &width, &height,
  319. &bitDepth, &colorType,
  320. &interlaceType, 0, 0);
  321. if (bitDepth == 16)
  322. png_set_strip_16 (pngReadStruct);
  323. if (colorType == PNG_COLOR_TYPE_PALETTE)
  324. png_set_expand (pngReadStruct);
  325. if (bitDepth < 8)
  326. png_set_expand (pngReadStruct);
  327. if (colorType == PNG_COLOR_TYPE_GRAY || colorType == PNG_COLOR_TYPE_GRAY_ALPHA)
  328. png_set_gray_to_rgb (pngReadStruct);
  329. return true;
  330. }
  331. return false;
  332. }
  333. static bool readImageData (png_structp pngReadStruct, png_infop pngInfoStruct, jmp_buf& errorJumpBuf, png_bytepp rows) noexcept
  334. {
  335. if (setjmp (errorJumpBuf) == 0)
  336. {
  337. if (png_get_valid (pngReadStruct, pngInfoStruct, PNG_INFO_tRNS))
  338. png_set_expand (pngReadStruct);
  339. png_set_add_alpha (pngReadStruct, 0xff, PNG_FILLER_AFTER);
  340. png_read_image (pngReadStruct, rows);
  341. png_read_end (pngReadStruct, pngInfoStruct);
  342. return true;
  343. }
  344. return false;
  345. }
  346. #if JUCE_MSVC
  347. #pragma warning (pop)
  348. #endif
  349. static Image createImageFromData (bool hasAlphaChan, int width, int height, png_bytepp rows)
  350. {
  351. // now convert the data to a juce image format..
  352. Image image (hasAlphaChan ? Image::ARGB : Image::RGB, width, height, hasAlphaChan);
  353. image.getProperties()->set ("originalImageHadAlpha", image.hasAlphaChannel());
  354. hasAlphaChan = image.hasAlphaChannel(); // (the native image creator may not give back what we expect)
  355. const Image::BitmapData destData (image, Image::BitmapData::writeOnly);
  356. for (int y = 0; y < (int) height; ++y)
  357. {
  358. const uint8* src = rows[y];
  359. uint8* dest = destData.getLinePointer (y);
  360. if (hasAlphaChan)
  361. {
  362. for (int i = (int) width; --i >= 0;)
  363. {
  364. ((PixelARGB*) dest)->setARGB (src[3], src[0], src[1], src[2]);
  365. ((PixelARGB*) dest)->premultiply();
  366. dest += destData.pixelStride;
  367. src += 4;
  368. }
  369. }
  370. else
  371. {
  372. for (int i = (int) width; --i >= 0;)
  373. {
  374. ((PixelRGB*) dest)->setARGB (0, src[0], src[1], src[2]);
  375. dest += destData.pixelStride;
  376. src += 4;
  377. }
  378. }
  379. }
  380. return image;
  381. }
  382. static Image readImage (InputStream& in, png_structp pngReadStruct, png_infop pngInfoStruct)
  383. {
  384. jmp_buf errorJumpBuf;
  385. png_set_error_fn (pngReadStruct, &errorJumpBuf, errorCallback, warningCallback);
  386. png_uint_32 width = 0, height = 0;
  387. int bitDepth = 0, colorType = 0, interlaceType = 0;
  388. if (readHeader (in, pngReadStruct, pngInfoStruct, errorJumpBuf,
  389. width, height, bitDepth, colorType, interlaceType))
  390. {
  391. // Load the image into a temp buffer..
  392. const size_t lineStride = width * 4;
  393. HeapBlock<uint8> tempBuffer (height * lineStride);
  394. HeapBlock<png_bytep> rows (height);
  395. for (size_t y = 0; y < height; ++y)
  396. rows[y] = (png_bytep) (tempBuffer + lineStride * y);
  397. png_bytep trans_alpha = nullptr;
  398. png_color_16p trans_color = nullptr;
  399. int num_trans = 0;
  400. png_get_tRNS (pngReadStruct, pngInfoStruct, &trans_alpha, &num_trans, &trans_color);
  401. if (readImageData (pngReadStruct, pngInfoStruct, errorJumpBuf, rows))
  402. return createImageFromData ((colorType & PNG_COLOR_MASK_ALPHA) != 0 || num_trans != 0,
  403. (int) width, (int) height, rows);
  404. }
  405. return Image();
  406. }
  407. static Image readImage (InputStream& in)
  408. {
  409. if (png_structp pngReadStruct = png_create_read_struct (PNG_LIBPNG_VER_STRING, 0, 0, 0))
  410. {
  411. if (png_infop pngInfoStruct = png_create_info_struct (pngReadStruct))
  412. {
  413. Image image (readImage (in, pngReadStruct, pngInfoStruct));
  414. png_destroy_read_struct (&pngReadStruct, &pngInfoStruct, 0);
  415. return image;
  416. }
  417. png_destroy_read_struct (&pngReadStruct, 0, 0);
  418. }
  419. return Image();
  420. }
  421. #endif
  422. }
  423. //==============================================================================
  424. PNGImageFormat::PNGImageFormat() {}
  425. PNGImageFormat::~PNGImageFormat() {}
  426. String PNGImageFormat::getFormatName() { return "PNG"; }
  427. bool PNGImageFormat::usesFileExtension (const File& f) { return f.hasFileExtension ("png"); }
  428. bool PNGImageFormat::canUnderstand (InputStream& in)
  429. {
  430. const int bytesNeeded = 4;
  431. char header [bytesNeeded];
  432. return in.read (header, bytesNeeded) == bytesNeeded
  433. && header[1] == 'P'
  434. && header[2] == 'N'
  435. && header[3] == 'G';
  436. }
  437. #if JUCE_USING_COREIMAGE_LOADER
  438. Image juce_loadWithCoreImage (InputStream&);
  439. #endif
  440. Image PNGImageFormat::decodeImage (InputStream& in)
  441. {
  442. #if JUCE_USING_COREIMAGE_LOADER
  443. return juce_loadWithCoreImage (in);
  444. #else
  445. return PNGHelpers::readImage (in);
  446. #endif
  447. }
  448. bool PNGImageFormat::writeImageToStream (const Image& image, OutputStream& out)
  449. {
  450. using namespace pnglibNamespace;
  451. const int width = image.getWidth();
  452. const int height = image.getHeight();
  453. png_structp pngWriteStruct = png_create_write_struct (PNG_LIBPNG_VER_STRING, 0, 0, 0);
  454. if (pngWriteStruct == nullptr)
  455. return false;
  456. png_infop pngInfoStruct = png_create_info_struct (pngWriteStruct);
  457. if (pngInfoStruct == nullptr)
  458. {
  459. png_destroy_write_struct (&pngWriteStruct, (png_infopp) nullptr);
  460. return false;
  461. }
  462. png_set_write_fn (pngWriteStruct, &out, PNGHelpers::writeDataCallback, 0);
  463. png_set_IHDR (pngWriteStruct, pngInfoStruct, (png_uint_32) width, (png_uint_32) height, 8,
  464. image.hasAlphaChannel() ? PNG_COLOR_TYPE_RGB_ALPHA
  465. : PNG_COLOR_TYPE_RGB,
  466. PNG_INTERLACE_NONE,
  467. PNG_COMPRESSION_TYPE_BASE,
  468. PNG_FILTER_TYPE_BASE);
  469. HeapBlock<uint8> rowData ((size_t) width * 4);
  470. png_color_8 sig_bit;
  471. sig_bit.red = 8;
  472. sig_bit.green = 8;
  473. sig_bit.blue = 8;
  474. sig_bit.gray = 0;
  475. sig_bit.alpha = 8;
  476. png_set_sBIT (pngWriteStruct, pngInfoStruct, &sig_bit);
  477. png_write_info (pngWriteStruct, pngInfoStruct);
  478. png_set_shift (pngWriteStruct, &sig_bit);
  479. png_set_packing (pngWriteStruct);
  480. const Image::BitmapData srcData (image, Image::BitmapData::readOnly);
  481. for (int y = 0; y < height; ++y)
  482. {
  483. uint8* dst = rowData;
  484. const uint8* src = srcData.getLinePointer (y);
  485. if (image.hasAlphaChannel())
  486. {
  487. for (int i = width; --i >= 0;)
  488. {
  489. PixelARGB p (*(const PixelARGB*) src);
  490. p.unpremultiply();
  491. *dst++ = p.getRed();
  492. *dst++ = p.getGreen();
  493. *dst++ = p.getBlue();
  494. *dst++ = p.getAlpha();
  495. src += srcData.pixelStride;
  496. }
  497. }
  498. else
  499. {
  500. for (int i = width; --i >= 0;)
  501. {
  502. *dst++ = ((const PixelRGB*) src)->getRed();
  503. *dst++ = ((const PixelRGB*) src)->getGreen();
  504. *dst++ = ((const PixelRGB*) src)->getBlue();
  505. src += srcData.pixelStride;
  506. }
  507. }
  508. png_bytep rowPtr = rowData;
  509. png_write_rows (pngWriteStruct, &rowPtr, 1);
  510. }
  511. png_write_end (pngWriteStruct, pngInfoStruct);
  512. png_destroy_write_struct (&pngWriteStruct, &pngInfoStruct);
  513. return true;
  514. }