The JUCE cross-platform C++ framework, with DISTRHO/KXStudio specific changes
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.

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