1 2 /* pngwrite.c - general routines to write a PNG file 3 * 4 * Last changed in libpng 1.2.45 [July 7, 2011] 5 * Copyright (c) 1998-2011 Glenn Randers-Pehrson 6 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger) 7 * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.) 8 * 9 * This code is released under the libpng license. 10 * For conditions of distribution and use, see the disclaimer 11 * and license in png.h 12 */ 13 14 /* Get internal access to png.h */ 15 #define PNG_INTERNAL 16 #define PNG_NO_PEDANTIC_WARNINGS 17 #include "png.h" 18 #ifdef PNG_WRITE_SUPPORTED 19 20 /* Writes all the PNG information. This is the suggested way to use the 21 * library. If you have a new chunk to add, make a function to write it, 22 * and put it in the correct location here. If you want the chunk written 23 * after the image data, put it in png_write_end(). I strongly encourage 24 * you to supply a PNG_INFO_ flag, and check info_ptr->valid before writing 25 * the chunk, as that will keep the code from breaking if you want to just 26 * write a plain PNG file. If you have long comments, I suggest writing 27 * them in png_write_end(), and compressing them. 28 */ 29 void PNGAPI 30 png_write_info_before_PLTE(png_structp png_ptr, png_infop info_ptr) 31 { 32 png_debug(1, "in png_write_info_before_PLTE"); 33 34 if (png_ptr == NULL || info_ptr == NULL) 35 return; 36 if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE)) 37 { 38 /* Write PNG signature */ 39 png_write_sig(png_ptr); 40 #ifdef PNG_MNG_FEATURES_SUPPORTED 41 if ((png_ptr->mode&PNG_HAVE_PNG_SIGNATURE) && \ 42 (png_ptr->mng_features_permitted)) 43 { 44 png_warning(png_ptr, "MNG features are not allowed in a PNG datastream"); 45 png_ptr->mng_features_permitted = 0; 46 } 47 #endif 48 /* Write IHDR information. */ 49 png_write_IHDR(png_ptr, info_ptr->width, info_ptr->height, 50 info_ptr->bit_depth, info_ptr->color_type, info_ptr->compression_type, 51 info_ptr->filter_type, 52 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 53 info_ptr->interlace_type); 54 #else 55 0); 56 #endif 57 /* The rest of these check to see if the valid field has the appropriate 58 * flag set, and if it does, writes the chunk. 59 */ 60 #ifdef PNG_WRITE_gAMA_SUPPORTED 61 if (info_ptr->valid & PNG_INFO_gAMA) 62 { 63 # ifdef PNG_FLOATING_POINT_SUPPORTED 64 png_write_gAMA(png_ptr, info_ptr->gamma); 65 #else 66 #ifdef PNG_FIXED_POINT_SUPPORTED 67 png_write_gAMA_fixed(png_ptr, info_ptr->int_gamma); 68 # endif 69 #endif 70 } 71 #endif 72 #ifdef PNG_WRITE_sRGB_SUPPORTED 73 if (info_ptr->valid & PNG_INFO_sRGB) 74 png_write_sRGB(png_ptr, (int)info_ptr->srgb_intent); 75 #endif 76 #ifdef PNG_WRITE_iCCP_SUPPORTED 77 if (info_ptr->valid & PNG_INFO_iCCP) 78 png_write_iCCP(png_ptr, info_ptr->iccp_name, PNG_COMPRESSION_TYPE_BASE, 79 info_ptr->iccp_profile, (int)info_ptr->iccp_proflen); 80 #endif 81 #ifdef PNG_WRITE_sBIT_SUPPORTED 82 if (info_ptr->valid & PNG_INFO_sBIT) 83 png_write_sBIT(png_ptr, &(info_ptr->sig_bit), info_ptr->color_type); 84 #endif 85 #ifdef PNG_WRITE_cHRM_SUPPORTED 86 if (info_ptr->valid & PNG_INFO_cHRM) 87 { 88 #ifdef PNG_FLOATING_POINT_SUPPORTED 89 png_write_cHRM(png_ptr, 90 info_ptr->x_white, info_ptr->y_white, 91 info_ptr->x_red, info_ptr->y_red, 92 info_ptr->x_green, info_ptr->y_green, 93 info_ptr->x_blue, info_ptr->y_blue); 94 #else 95 # ifdef PNG_FIXED_POINT_SUPPORTED 96 png_write_cHRM_fixed(png_ptr, 97 info_ptr->int_x_white, info_ptr->int_y_white, 98 info_ptr->int_x_red, info_ptr->int_y_red, 99 info_ptr->int_x_green, info_ptr->int_y_green, 100 info_ptr->int_x_blue, info_ptr->int_y_blue); 101 # endif 102 #endif 103 } 104 #endif 105 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED 106 if (info_ptr->unknown_chunks_num) 107 { 108 png_unknown_chunk *up; 109 110 png_debug(5, "writing extra chunks"); 111 112 for (up = info_ptr->unknown_chunks; 113 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num; 114 up++) 115 { 116 int keep = png_handle_as_unknown(png_ptr, up->name); 117 if (keep != PNG_HANDLE_CHUNK_NEVER && 118 up->location && !(up->location & PNG_HAVE_PLTE) && 119 !(up->location & PNG_HAVE_IDAT) && 120 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS || 121 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS))) 122 { 123 if (up->size == 0) 124 png_warning(png_ptr, "Writing zero-length unknown chunk"); 125 png_write_chunk(png_ptr, up->name, up->data, up->size); 126 } 127 } 128 } 129 #endif 130 png_ptr->mode |= PNG_WROTE_INFO_BEFORE_PLTE; 131 } 132 } 133 134 void PNGAPI 135 png_write_info(png_structp png_ptr, png_infop info_ptr) 136 { 137 #if defined(PNG_WRITE_TEXT_SUPPORTED) || defined(PNG_WRITE_sPLT_SUPPORTED) 138 int i; 139 #endif 140 141 png_debug(1, "in png_write_info"); 142 143 if (png_ptr == NULL || info_ptr == NULL) 144 return; 145 146 png_write_info_before_PLTE(png_ptr, info_ptr); 147 148 if (info_ptr->valid & PNG_INFO_PLTE) 149 png_write_PLTE(png_ptr, info_ptr->palette, 150 (png_uint_32)info_ptr->num_palette); 151 else if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE) 152 png_error(png_ptr, "Valid palette required for paletted images"); 153 154 #ifdef PNG_WRITE_tRNS_SUPPORTED 155 if (info_ptr->valid & PNG_INFO_tRNS) 156 { 157 #ifdef PNG_WRITE_INVERT_ALPHA_SUPPORTED 158 /* Invert the alpha channel (in tRNS) */ 159 if ((png_ptr->transformations & PNG_INVERT_ALPHA) && 160 info_ptr->color_type == PNG_COLOR_TYPE_PALETTE) 161 { 162 int j; 163 for (j = 0; j<(int)info_ptr->num_trans; j++) 164 info_ptr->trans[j] = (png_byte)(255 - info_ptr->trans[j]); 165 } 166 #endif 167 png_write_tRNS(png_ptr, info_ptr->trans, &(info_ptr->trans_values), 168 info_ptr->num_trans, info_ptr->color_type); 169 } 170 #endif 171 #ifdef PNG_WRITE_bKGD_SUPPORTED 172 if (info_ptr->valid & PNG_INFO_bKGD) 173 png_write_bKGD(png_ptr, &(info_ptr->background), info_ptr->color_type); 174 #endif 175 #ifdef PNG_WRITE_hIST_SUPPORTED 176 if (info_ptr->valid & PNG_INFO_hIST) 177 png_write_hIST(png_ptr, info_ptr->hist, info_ptr->num_palette); 178 #endif 179 #ifdef PNG_WRITE_oFFs_SUPPORTED 180 if (info_ptr->valid & PNG_INFO_oFFs) 181 png_write_oFFs(png_ptr, info_ptr->x_offset, info_ptr->y_offset, 182 info_ptr->offset_unit_type); 183 #endif 184 #ifdef PNG_WRITE_pCAL_SUPPORTED 185 if (info_ptr->valid & PNG_INFO_pCAL) 186 png_write_pCAL(png_ptr, info_ptr->pcal_purpose, info_ptr->pcal_X0, 187 info_ptr->pcal_X1, info_ptr->pcal_type, info_ptr->pcal_nparams, 188 info_ptr->pcal_units, info_ptr->pcal_params); 189 #endif 190 191 #ifdef PNG_sCAL_SUPPORTED 192 if (info_ptr->valid & PNG_INFO_sCAL) 193 #ifdef PNG_WRITE_sCAL_SUPPORTED 194 #if defined(PNG_FLOATING_POINT_SUPPORTED) && defined(PNG_STDIO_SUPPORTED) 195 png_write_sCAL(png_ptr, (int)info_ptr->scal_unit, 196 info_ptr->scal_pixel_width, info_ptr->scal_pixel_height); 197 #else /* !FLOATING_POINT */ 198 #ifdef PNG_FIXED_POINT_SUPPORTED 199 png_write_sCAL_s(png_ptr, (int)info_ptr->scal_unit, 200 info_ptr->scal_s_width, info_ptr->scal_s_height); 201 #endif /* FIXED_POINT */ 202 #endif /* FLOATING_POINT */ 203 #else /* !WRITE_sCAL */ 204 png_warning(png_ptr, 205 "png_write_sCAL not supported; sCAL chunk not written."); 206 #endif /* WRITE_sCAL */ 207 #endif /* sCAL */ 208 209 #ifdef PNG_WRITE_pHYs_SUPPORTED 210 if (info_ptr->valid & PNG_INFO_pHYs) 211 png_write_pHYs(png_ptr, info_ptr->x_pixels_per_unit, 212 info_ptr->y_pixels_per_unit, info_ptr->phys_unit_type); 213 #endif /* pHYs */ 214 215 #ifdef PNG_WRITE_tIME_SUPPORTED 216 if (info_ptr->valid & PNG_INFO_tIME) 217 { 218 png_write_tIME(png_ptr, &(info_ptr->mod_time)); 219 png_ptr->mode |= PNG_WROTE_tIME; 220 } 221 #endif /* tIME */ 222 223 #ifdef PNG_WRITE_sPLT_SUPPORTED 224 if (info_ptr->valid & PNG_INFO_sPLT) 225 for (i = 0; i < (int)info_ptr->splt_palettes_num; i++) 226 png_write_sPLT(png_ptr, info_ptr->splt_palettes + i); 227 #endif /* sPLT */ 228 229 #ifdef PNG_WRITE_TEXT_SUPPORTED 230 /* Check to see if we need to write text chunks */ 231 for (i = 0; i < info_ptr->num_text; i++) 232 { 233 png_debug2(2, "Writing header text chunk %d, type %d", i, 234 info_ptr->text[i].compression); 235 /* An internationalized chunk? */ 236 if (info_ptr->text[i].compression > 0) 237 { 238 #ifdef PNG_WRITE_iTXt_SUPPORTED 239 /* Write international chunk */ 240 png_write_iTXt(png_ptr, 241 info_ptr->text[i].compression, 242 info_ptr->text[i].key, 243 info_ptr->text[i].lang, 244 info_ptr->text[i].lang_key, 245 info_ptr->text[i].text); 246 #else 247 png_warning(png_ptr, "Unable to write international text"); 248 #endif 249 /* Mark this chunk as written */ 250 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR; 251 } 252 /* If we want a compressed text chunk */ 253 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_zTXt) 254 { 255 #ifdef PNG_WRITE_zTXt_SUPPORTED 256 /* Write compressed chunk */ 257 png_write_zTXt(png_ptr, info_ptr->text[i].key, 258 info_ptr->text[i].text, 0, 259 info_ptr->text[i].compression); 260 #else 261 png_warning(png_ptr, "Unable to write compressed text"); 262 #endif 263 /* Mark this chunk as written */ 264 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR; 265 } 266 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE) 267 { 268 #ifdef PNG_WRITE_tEXt_SUPPORTED 269 /* Write uncompressed chunk */ 270 png_write_tEXt(png_ptr, info_ptr->text[i].key, 271 info_ptr->text[i].text, 272 0); 273 /* Mark this chunk as written */ 274 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR; 275 #else 276 /* Can't get here */ 277 png_warning(png_ptr, "Unable to write uncompressed text"); 278 #endif 279 } 280 } 281 #endif /* tEXt */ 282 283 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED 284 if (info_ptr->unknown_chunks_num) 285 { 286 png_unknown_chunk *up; 287 288 png_debug(5, "writing extra chunks"); 289 290 for (up = info_ptr->unknown_chunks; 291 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num; 292 up++) 293 { 294 int keep = png_handle_as_unknown(png_ptr, up->name); 295 if (keep != PNG_HANDLE_CHUNK_NEVER && 296 up->location && (up->location & PNG_HAVE_PLTE) && 297 !(up->location & PNG_HAVE_IDAT) && 298 !(up->location & PNG_AFTER_IDAT) && 299 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS || 300 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS))) 301 { 302 png_write_chunk(png_ptr, up->name, up->data, up->size); 303 } 304 } 305 } 306 #endif 307 } 308 309 /* Writes the end of the PNG file. If you don't want to write comments or 310 * time information, you can pass NULL for info. If you already wrote these 311 * in png_write_info(), do not write them again here. If you have long 312 * comments, I suggest writing them here, and compressing them. 313 */ 314 void PNGAPI 315 png_write_end(png_structp png_ptr, png_infop info_ptr) 316 { 317 png_debug(1, "in png_write_end"); 318 319 if (png_ptr == NULL) 320 return; 321 if (!(png_ptr->mode & PNG_HAVE_IDAT)) 322 png_error(png_ptr, "No IDATs written into file"); 323 324 /* See if user wants us to write information chunks */ 325 if (info_ptr != NULL) 326 { 327 #ifdef PNG_WRITE_TEXT_SUPPORTED 328 int i; /* local index variable */ 329 #endif 330 #ifdef PNG_WRITE_tIME_SUPPORTED 331 /* Check to see if user has supplied a time chunk */ 332 if ((info_ptr->valid & PNG_INFO_tIME) && 333 !(png_ptr->mode & PNG_WROTE_tIME)) 334 png_write_tIME(png_ptr, &(info_ptr->mod_time)); 335 #endif 336 #ifdef PNG_WRITE_TEXT_SUPPORTED 337 /* Loop through comment chunks */ 338 for (i = 0; i < info_ptr->num_text; i++) 339 { 340 png_debug2(2, "Writing trailer text chunk %d, type %d", i, 341 info_ptr->text[i].compression); 342 /* An internationalized chunk? */ 343 if (info_ptr->text[i].compression > 0) 344 { 345 #ifdef PNG_WRITE_iTXt_SUPPORTED 346 /* Write international chunk */ 347 png_write_iTXt(png_ptr, 348 info_ptr->text[i].compression, 349 info_ptr->text[i].key, 350 info_ptr->text[i].lang, 351 info_ptr->text[i].lang_key, 352 info_ptr->text[i].text); 353 #else 354 png_warning(png_ptr, "Unable to write international text"); 355 #endif 356 /* Mark this chunk as written */ 357 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR; 358 } 359 else if (info_ptr->text[i].compression >= PNG_TEXT_COMPRESSION_zTXt) 360 { 361 #ifdef PNG_WRITE_zTXt_SUPPORTED 362 /* Write compressed chunk */ 363 png_write_zTXt(png_ptr, info_ptr->text[i].key, 364 info_ptr->text[i].text, 0, 365 info_ptr->text[i].compression); 366 #else 367 png_warning(png_ptr, "Unable to write compressed text"); 368 #endif 369 /* Mark this chunk as written */ 370 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_zTXt_WR; 371 } 372 else if (info_ptr->text[i].compression == PNG_TEXT_COMPRESSION_NONE) 373 { 374 #ifdef PNG_WRITE_tEXt_SUPPORTED 375 /* Write uncompressed chunk */ 376 png_write_tEXt(png_ptr, info_ptr->text[i].key, 377 info_ptr->text[i].text, 0); 378 #else 379 png_warning(png_ptr, "Unable to write uncompressed text"); 380 #endif 381 382 /* Mark this chunk as written */ 383 info_ptr->text[i].compression = PNG_TEXT_COMPRESSION_NONE_WR; 384 } 385 } 386 #endif 387 #ifdef PNG_WRITE_UNKNOWN_CHUNKS_SUPPORTED 388 if (info_ptr->unknown_chunks_num) 389 { 390 png_unknown_chunk *up; 391 392 png_debug(5, "writing extra chunks"); 393 394 for (up = info_ptr->unknown_chunks; 395 up < info_ptr->unknown_chunks + info_ptr->unknown_chunks_num; 396 up++) 397 { 398 int keep = png_handle_as_unknown(png_ptr, up->name); 399 if (keep != PNG_HANDLE_CHUNK_NEVER && 400 up->location && (up->location & PNG_AFTER_IDAT) && 401 ((up->name[3] & 0x20) || keep == PNG_HANDLE_CHUNK_ALWAYS || 402 (png_ptr->flags & PNG_FLAG_KEEP_UNSAFE_CHUNKS))) 403 { 404 png_write_chunk(png_ptr, up->name, up->data, up->size); 405 } 406 } 407 } 408 #endif 409 } 410 411 png_ptr->mode |= PNG_AFTER_IDAT; 412 413 /* Write end of PNG file */ 414 png_write_IEND(png_ptr); 415 /* This flush, added in libpng-1.0.8, removed from libpng-1.0.9beta03, 416 * and restored again in libpng-1.2.30, may cause some applications that 417 * do not set png_ptr->output_flush_fn to crash. If your application 418 * experiences a problem, please try building libpng with 419 * PNG_WRITE_FLUSH_AFTER_IEND_SUPPORTED defined, and report the event to 420 * png-mng-implement at lists.sf.net . 421 */ 422 #ifdef PNG_WRITE_FLUSH_SUPPORTED 423 # ifdef PNG_WRITE_FLUSH_AFTER_IEND_SUPPORTED 424 png_flush(png_ptr); 425 # endif 426 #endif 427 } 428 429 #ifdef PNG_CONVERT_tIME_SUPPORTED 430 /* "tm" structure is not supported on WindowsCE */ 431 void PNGAPI 432 png_convert_from_struct_tm(png_timep ptime, struct tm FAR * ttime) 433 { 434 png_debug(1, "in png_convert_from_struct_tm"); 435 436 ptime->year = (png_uint_16)(1900 + ttime->tm_year); 437 ptime->month = (png_byte)(ttime->tm_mon + 1); 438 ptime->day = (png_byte)ttime->tm_mday; 439 ptime->hour = (png_byte)ttime->tm_hour; 440 ptime->minute = (png_byte)ttime->tm_min; 441 ptime->second = (png_byte)ttime->tm_sec; 442 } 443 444 void PNGAPI 445 png_convert_from_time_t(png_timep ptime, time_t ttime) 446 { 447 struct tm *tbuf; 448 449 png_debug(1, "in png_convert_from_time_t"); 450 451 tbuf = gmtime(&ttime); 452 png_convert_from_struct_tm(ptime, tbuf); 453 } 454 #endif 455 456 /* Initialize png_ptr structure, and allocate any memory needed */ 457 png_structp PNGAPI 458 png_create_write_struct(png_const_charp user_png_ver, png_voidp error_ptr, 459 png_error_ptr error_fn, png_error_ptr warn_fn) 460 { 461 #ifdef PNG_USER_MEM_SUPPORTED 462 return (png_create_write_struct_2(user_png_ver, error_ptr, error_fn, 463 warn_fn, png_voidp_NULL, png_malloc_ptr_NULL, png_free_ptr_NULL)); 464 } 465 466 /* Alternate initialize png_ptr structure, and allocate any memory needed */ 467 png_structp PNGAPI 468 png_create_write_struct_2(png_const_charp user_png_ver, png_voidp error_ptr, 469 png_error_ptr error_fn, png_error_ptr warn_fn, png_voidp mem_ptr, 470 png_malloc_ptr malloc_fn, png_free_ptr free_fn) 471 { 472 #endif /* PNG_USER_MEM_SUPPORTED */ 473 #ifdef PNG_SETJMP_SUPPORTED 474 volatile 475 #endif 476 png_structp png_ptr; 477 #ifdef PNG_SETJMP_SUPPORTED 478 #ifdef USE_FAR_KEYWORD 479 jmp_buf jmpbuf; 480 #endif 481 #endif 482 int i; 483 484 png_debug(1, "in png_create_write_struct"); 485 486 #ifdef PNG_USER_MEM_SUPPORTED 487 png_ptr = (png_structp)png_create_struct_2(PNG_STRUCT_PNG, 488 (png_malloc_ptr)malloc_fn, (png_voidp)mem_ptr); 489 #else 490 png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG); 491 #endif /* PNG_USER_MEM_SUPPORTED */ 492 if (png_ptr == NULL) 493 return (NULL); 494 495 /* Added at libpng-1.2.6 */ 496 #ifdef PNG_SET_USER_LIMITS_SUPPORTED 497 png_ptr->user_width_max = PNG_USER_WIDTH_MAX; 498 png_ptr->user_height_max = PNG_USER_HEIGHT_MAX; 499 #endif 500 501 #ifdef PNG_SETJMP_SUPPORTED 502 #ifdef USE_FAR_KEYWORD 503 if (setjmp(jmpbuf)) 504 #else 505 if (setjmp(png_ptr->jmpbuf)) 506 #endif 507 { 508 png_free(png_ptr, png_ptr->zbuf); 509 png_ptr->zbuf = NULL; 510 #ifdef PNG_USER_MEM_SUPPORTED 511 png_destroy_struct_2((png_voidp)png_ptr, 512 (png_free_ptr)free_fn, (png_voidp)mem_ptr); 513 #else 514 png_destroy_struct((png_voidp)png_ptr); 515 #endif 516 return (NULL); 517 } 518 #ifdef USE_FAR_KEYWORD 519 png_memcpy(png_ptr->jmpbuf, jmpbuf, png_sizeof(jmp_buf)); 520 #endif 521 #endif 522 523 #ifdef PNG_USER_MEM_SUPPORTED 524 png_set_mem_fn(png_ptr, mem_ptr, malloc_fn, free_fn); 525 #endif /* PNG_USER_MEM_SUPPORTED */ 526 png_set_error_fn(png_ptr, error_ptr, error_fn, warn_fn); 527 528 if (user_png_ver) 529 { 530 i = 0; 531 do 532 { 533 if (user_png_ver[i] != png_libpng_ver[i]) 534 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH; 535 } while (png_libpng_ver[i++]); 536 } 537 538 if (png_ptr->flags & PNG_FLAG_LIBRARY_MISMATCH) 539 { 540 /* Libpng 0.90 and later are binary incompatible with libpng 0.89, so 541 * we must recompile any applications that use any older library version. 542 * For versions after libpng 1.0, we will be compatible, so we need 543 * only check the first digit. 544 */ 545 if (user_png_ver == NULL || user_png_ver[0] != png_libpng_ver[0] || 546 (user_png_ver[0] == '1' && user_png_ver[2] != png_libpng_ver[2]) || 547 (user_png_ver[0] == '0' && user_png_ver[2] < '9')) 548 { 549 #if defined(PNG_STDIO_SUPPORTED) && !defined(_WIN32_WCE) 550 char msg[80]; 551 if (user_png_ver) 552 { 553 png_snprintf(msg, 80, 554 "Application was compiled with png.h from libpng-%.20s", 555 user_png_ver); 556 png_warning(png_ptr, msg); 557 } 558 png_snprintf(msg, 80, 559 "Application is running with png.c from libpng-%.20s", 560 png_libpng_ver); 561 png_warning(png_ptr, msg); 562 #endif 563 #ifdef PNG_ERROR_NUMBERS_SUPPORTED 564 png_ptr->flags = 0; 565 #endif 566 png_error(png_ptr, 567 "Incompatible libpng version in application and library"); 568 } 569 } 570 571 /* Initialize zbuf - compression buffer */ 572 png_ptr->zbuf_size = PNG_ZBUF_SIZE; 573 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr, 574 (png_uint_32)png_ptr->zbuf_size); 575 576 png_set_write_fn(png_ptr, png_voidp_NULL, png_rw_ptr_NULL, 577 png_flush_ptr_NULL); 578 579 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED 580 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT, 581 1, png_doublep_NULL, png_doublep_NULL); 582 #endif 583 584 #ifdef PNG_SETJMP_SUPPORTED 585 /* Applications that neglect to set up their own setjmp() and then 586 * encounter a png_error() will longjmp here. Since the jmpbuf is 587 * then meaningless we abort instead of returning. 588 */ 589 #ifdef USE_FAR_KEYWORD 590 if (setjmp(jmpbuf)) 591 PNG_ABORT(); 592 png_memcpy(png_ptr->jmpbuf, jmpbuf, png_sizeof(jmp_buf)); 593 #else 594 if (setjmp(png_ptr->jmpbuf)) 595 PNG_ABORT(); 596 #endif 597 #endif 598 return (png_ptr); 599 } 600 601 /* Initialize png_ptr structure, and allocate any memory needed */ 602 #if defined(PNG_1_0_X) || defined(PNG_1_2_X) 603 /* Deprecated. */ 604 #undef png_write_init 605 void PNGAPI 606 png_write_init(png_structp png_ptr) 607 { 608 /* We only come here via pre-1.0.7-compiled applications */ 609 png_write_init_2(png_ptr, "1.0.6 or earlier", 0, 0); 610 } 611 612 void PNGAPI 613 png_write_init_2(png_structp png_ptr, png_const_charp user_png_ver, 614 png_size_t png_struct_size, png_size_t png_info_size) 615 { 616 /* We only come here via pre-1.0.12-compiled applications */ 617 if (png_ptr == NULL) return; 618 #if defined(PNG_STDIO_SUPPORTED) && !defined(_WIN32_WCE) 619 if (png_sizeof(png_struct) > png_struct_size || 620 png_sizeof(png_info) > png_info_size) 621 { 622 char msg[80]; 623 png_ptr->warning_fn = NULL; 624 if (user_png_ver) 625 { 626 png_snprintf(msg, 80, 627 "Application was compiled with png.h from libpng-%.20s", 628 user_png_ver); 629 png_warning(png_ptr, msg); 630 } 631 png_snprintf(msg, 80, 632 "Application is running with png.c from libpng-%.20s", 633 png_libpng_ver); 634 png_warning(png_ptr, msg); 635 } 636 #endif 637 if (png_sizeof(png_struct) > png_struct_size) 638 { 639 png_ptr->error_fn = NULL; 640 #ifdef PNG_ERROR_NUMBERS_SUPPORTED 641 png_ptr->flags = 0; 642 #endif 643 png_error(png_ptr, 644 "The png struct allocated by the application for writing is" 645 " too small."); 646 } 647 if (png_sizeof(png_info) > png_info_size) 648 { 649 png_ptr->error_fn = NULL; 650 #ifdef PNG_ERROR_NUMBERS_SUPPORTED 651 png_ptr->flags = 0; 652 #endif 653 png_error(png_ptr, 654 "The info struct allocated by the application for writing is" 655 " too small."); 656 } 657 png_write_init_3(&png_ptr, user_png_ver, png_struct_size); 658 } 659 #endif /* PNG_1_0_X || PNG_1_2_X */ 660 661 662 void PNGAPI 663 png_write_init_3(png_structpp ptr_ptr, png_const_charp user_png_ver, 664 png_size_t png_struct_size) 665 { 666 png_structp png_ptr = *ptr_ptr; 667 #ifdef PNG_SETJMP_SUPPORTED 668 jmp_buf tmp_jmp; /* to save current jump buffer */ 669 #endif 670 671 int i = 0; 672 673 if (png_ptr == NULL) 674 return; 675 676 do 677 { 678 if (user_png_ver[i] != png_libpng_ver[i]) 679 { 680 #ifdef PNG_LEGACY_SUPPORTED 681 png_ptr->flags |= PNG_FLAG_LIBRARY_MISMATCH; 682 #else 683 png_ptr->warning_fn = NULL; 684 png_warning(png_ptr, 685 "Application uses deprecated png_write_init() and should be recompiled."); 686 #endif 687 } 688 } while (png_libpng_ver[i++]); 689 690 png_debug(1, "in png_write_init_3"); 691 692 #ifdef PNG_SETJMP_SUPPORTED 693 /* Save jump buffer and error functions */ 694 png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof(jmp_buf)); 695 #endif 696 697 if (png_sizeof(png_struct) > png_struct_size) 698 { 699 png_destroy_struct(png_ptr); 700 png_ptr = (png_structp)png_create_struct(PNG_STRUCT_PNG); 701 *ptr_ptr = png_ptr; 702 } 703 704 /* Reset all variables to 0 */ 705 png_memset(png_ptr, 0, png_sizeof(png_struct)); 706 707 /* Added at libpng-1.2.6 */ 708 #ifdef PNG_SET_USER_LIMITS_SUPPORTED 709 png_ptr->user_width_max = PNG_USER_WIDTH_MAX; 710 png_ptr->user_height_max = PNG_USER_HEIGHT_MAX; 711 #endif 712 713 #ifdef PNG_SETJMP_SUPPORTED 714 /* Restore jump buffer */ 715 png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof(jmp_buf)); 716 #endif 717 718 png_set_write_fn(png_ptr, png_voidp_NULL, png_rw_ptr_NULL, 719 png_flush_ptr_NULL); 720 721 /* Initialize zbuf - compression buffer */ 722 png_ptr->zbuf_size = PNG_ZBUF_SIZE; 723 png_ptr->zbuf = (png_bytep)png_malloc(png_ptr, 724 (png_uint_32)png_ptr->zbuf_size); 725 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED 726 png_set_filter_heuristics(png_ptr, PNG_FILTER_HEURISTIC_DEFAULT, 727 1, png_doublep_NULL, png_doublep_NULL); 728 #endif 729 } 730 731 /* Write a few rows of image data. If the image is interlaced, 732 * either you will have to write the 7 sub images, or, if you 733 * have called png_set_interlace_handling(), you will have to 734 * "write" the image seven times. 735 */ 736 void PNGAPI 737 png_write_rows(png_structp png_ptr, png_bytepp row, 738 png_uint_32 num_rows) 739 { 740 png_uint_32 i; /* row counter */ 741 png_bytepp rp; /* row pointer */ 742 743 png_debug(1, "in png_write_rows"); 744 745 if (png_ptr == NULL) 746 return; 747 748 /* Loop through the rows */ 749 for (i = 0, rp = row; i < num_rows; i++, rp++) 750 { 751 png_write_row(png_ptr, *rp); 752 } 753 } 754 755 /* Write the image. You only need to call this function once, even 756 * if you are writing an interlaced image. 757 */ 758 void PNGAPI 759 png_write_image(png_structp png_ptr, png_bytepp image) 760 { 761 png_uint_32 i; /* row index */ 762 int pass, num_pass; /* pass variables */ 763 png_bytepp rp; /* points to current row */ 764 765 if (png_ptr == NULL) 766 return; 767 768 png_debug(1, "in png_write_image"); 769 770 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 771 /* Initialize interlace handling. If image is not interlaced, 772 * this will set pass to 1 773 */ 774 num_pass = png_set_interlace_handling(png_ptr); 775 #else 776 num_pass = 1; 777 #endif 778 /* Loop through passes */ 779 for (pass = 0; pass < num_pass; pass++) 780 { 781 /* Loop through image */ 782 for (i = 0, rp = image; i < png_ptr->height; i++, rp++) 783 { 784 png_write_row(png_ptr, *rp); 785 } 786 } 787 } 788 789 /* Called by user to write a row of image data */ 790 void PNGAPI 791 png_write_row(png_structp png_ptr, png_bytep row) 792 { 793 if (png_ptr == NULL) 794 return; 795 796 png_debug2(1, "in png_write_row (row %ld, pass %d)", 797 png_ptr->row_number, png_ptr->pass); 798 799 /* Initialize transformations and other stuff if first time */ 800 if (png_ptr->row_number == 0 && png_ptr->pass == 0) 801 { 802 /* Make sure we wrote the header info */ 803 if (!(png_ptr->mode & PNG_WROTE_INFO_BEFORE_PLTE)) 804 png_error(png_ptr, 805 "png_write_info was never called before png_write_row."); 806 807 /* Check for transforms that have been set but were defined out */ 808 #if !defined(PNG_WRITE_INVERT_SUPPORTED) && defined(PNG_READ_INVERT_SUPPORTED) 809 if (png_ptr->transformations & PNG_INVERT_MONO) 810 png_warning(png_ptr, 811 "PNG_WRITE_INVERT_SUPPORTED is not defined."); 812 #endif 813 #if !defined(PNG_WRITE_FILLER_SUPPORTED) && defined(PNG_READ_FILLER_SUPPORTED) 814 if (png_ptr->transformations & PNG_FILLER) 815 png_warning(png_ptr, 816 "PNG_WRITE_FILLER_SUPPORTED is not defined."); 817 #endif 818 #if !defined(PNG_WRITE_PACKSWAP_SUPPORTED) && \ 819 defined(PNG_READ_PACKSWAP_SUPPORTED) 820 if (png_ptr->transformations & PNG_PACKSWAP) 821 png_warning(png_ptr, 822 "PNG_WRITE_PACKSWAP_SUPPORTED is not defined."); 823 #endif 824 #if !defined(PNG_WRITE_PACK_SUPPORTED) && defined(PNG_READ_PACK_SUPPORTED) 825 if (png_ptr->transformations & PNG_PACK) 826 png_warning(png_ptr, "PNG_WRITE_PACK_SUPPORTED is not defined."); 827 #endif 828 #if !defined(PNG_WRITE_SHIFT_SUPPORTED) && defined(PNG_READ_SHIFT_SUPPORTED) 829 if (png_ptr->transformations & PNG_SHIFT) 830 png_warning(png_ptr, "PNG_WRITE_SHIFT_SUPPORTED is not defined."); 831 #endif 832 #if !defined(PNG_WRITE_BGR_SUPPORTED) && defined(PNG_READ_BGR_SUPPORTED) 833 if (png_ptr->transformations & PNG_BGR) 834 png_warning(png_ptr, "PNG_WRITE_BGR_SUPPORTED is not defined."); 835 #endif 836 #if !defined(PNG_WRITE_SWAP_SUPPORTED) && defined(PNG_READ_SWAP_SUPPORTED) 837 if (png_ptr->transformations & PNG_SWAP_BYTES) 838 png_warning(png_ptr, "PNG_WRITE_SWAP_SUPPORTED is not defined."); 839 #endif 840 841 png_write_start_row(png_ptr); 842 } 843 844 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 845 /* If interlaced and not interested in row, return */ 846 if (png_ptr->interlaced && (png_ptr->transformations & PNG_INTERLACE)) 847 { 848 switch (png_ptr->pass) 849 { 850 case 0: 851 if (png_ptr->row_number & 0x07) 852 { 853 png_write_finish_row(png_ptr); 854 return; 855 } 856 break; 857 case 1: 858 if ((png_ptr->row_number & 0x07) || png_ptr->width < 5) 859 { 860 png_write_finish_row(png_ptr); 861 return; 862 } 863 break; 864 case 2: 865 if ((png_ptr->row_number & 0x07) != 4) 866 { 867 png_write_finish_row(png_ptr); 868 return; 869 } 870 break; 871 case 3: 872 if ((png_ptr->row_number & 0x03) || png_ptr->width < 3) 873 { 874 png_write_finish_row(png_ptr); 875 return; 876 } 877 break; 878 case 4: 879 if ((png_ptr->row_number & 0x03) != 2) 880 { 881 png_write_finish_row(png_ptr); 882 return; 883 } 884 break; 885 case 5: 886 if ((png_ptr->row_number & 0x01) || png_ptr->width < 2) 887 { 888 png_write_finish_row(png_ptr); 889 return; 890 } 891 break; 892 case 6: 893 if (!(png_ptr->row_number & 0x01)) 894 { 895 png_write_finish_row(png_ptr); 896 return; 897 } 898 break; 899 } 900 } 901 #endif 902 903 /* Set up row info for transformations */ 904 png_ptr->row_info.color_type = png_ptr->color_type; 905 png_ptr->row_info.width = png_ptr->usr_width; 906 png_ptr->row_info.channels = png_ptr->usr_channels; 907 png_ptr->row_info.bit_depth = png_ptr->usr_bit_depth; 908 png_ptr->row_info.pixel_depth = (png_byte)(png_ptr->row_info.bit_depth * 909 png_ptr->row_info.channels); 910 911 png_ptr->row_info.rowbytes = PNG_ROWBYTES(png_ptr->row_info.pixel_depth, 912 png_ptr->row_info.width); 913 914 png_debug1(3, "row_info->color_type = %d", png_ptr->row_info.color_type); 915 png_debug1(3, "row_info->width = %lu", png_ptr->row_info.width); 916 png_debug1(3, "row_info->channels = %d", png_ptr->row_info.channels); 917 png_debug1(3, "row_info->bit_depth = %d", png_ptr->row_info.bit_depth); 918 png_debug1(3, "row_info->pixel_depth = %d", png_ptr->row_info.pixel_depth); 919 png_debug1(3, "row_info->rowbytes = %lu", png_ptr->row_info.rowbytes); 920 921 /* Copy user's row into buffer, leaving room for filter byte. */ 922 png_memcpy_check(png_ptr, png_ptr->row_buf + 1, row, 923 png_ptr->row_info.rowbytes); 924 925 #ifdef PNG_WRITE_INTERLACING_SUPPORTED 926 /* Handle interlacing */ 927 if (png_ptr->interlaced && png_ptr->pass < 6 && 928 (png_ptr->transformations & PNG_INTERLACE)) 929 { 930 png_do_write_interlace(&(png_ptr->row_info), 931 png_ptr->row_buf + 1, png_ptr->pass); 932 /* This should always get caught above, but still ... */ 933 if (!(png_ptr->row_info.width)) 934 { 935 png_write_finish_row(png_ptr); 936 return; 937 } 938 } 939 #endif 940 941 /* Handle other transformations */ 942 if (png_ptr->transformations) 943 png_do_write_transformations(png_ptr); 944 945 #ifdef PNG_MNG_FEATURES_SUPPORTED 946 /* Write filter_method 64 (intrapixel differencing) only if 947 * 1. Libpng was compiled with PNG_MNG_FEATURES_SUPPORTED and 948 * 2. Libpng did not write a PNG signature (this filter_method is only 949 * used in PNG datastreams that are embedded in MNG datastreams) and 950 * 3. The application called png_permit_mng_features with a mask that 951 * included PNG_FLAG_MNG_FILTER_64 and 952 * 4. The filter_method is 64 and 953 * 5. The color_type is RGB or RGBA 954 */ 955 if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) && 956 (png_ptr->filter_type == PNG_INTRAPIXEL_DIFFERENCING)) 957 { 958 /* Intrapixel differencing */ 959 png_do_write_intrapixel(&(png_ptr->row_info), png_ptr->row_buf + 1); 960 } 961 #endif 962 963 /* Find a filter if necessary, filter the row and write it out. */ 964 png_write_find_filter(png_ptr, &(png_ptr->row_info)); 965 966 if (png_ptr->write_row_fn != NULL) 967 (*(png_ptr->write_row_fn))(png_ptr, png_ptr->row_number, png_ptr->pass); 968 } 969 970 #ifdef PNG_WRITE_FLUSH_SUPPORTED 971 /* Set the automatic flush interval or 0 to turn flushing off */ 972 void PNGAPI 973 png_set_flush(png_structp png_ptr, int nrows) 974 { 975 png_debug(1, "in png_set_flush"); 976 977 if (png_ptr == NULL) 978 return; 979 png_ptr->flush_dist = (nrows < 0 ? 0 : nrows); 980 } 981 982 /* Flush the current output buffers now */ 983 void PNGAPI 984 png_write_flush(png_structp png_ptr) 985 { 986 int wrote_IDAT; 987 988 png_debug(1, "in png_write_flush"); 989 990 if (png_ptr == NULL) 991 return; 992 /* We have already written out all of the data */ 993 if (png_ptr->row_number >= png_ptr->num_rows) 994 return; 995 996 do 997 { 998 int ret; 999 1000 /* Compress the data */ 1001 ret = deflate(&png_ptr->zstream, Z_SYNC_FLUSH); 1002 wrote_IDAT = 0; 1003 1004 /* Check for compression errors */ 1005 if (ret != Z_OK) 1006 { 1007 if (png_ptr->zstream.msg != NULL) 1008 png_error(png_ptr, png_ptr->zstream.msg); 1009 else 1010 png_error(png_ptr, "zlib error"); 1011 } 1012 1013 if (!(png_ptr->zstream.avail_out)) 1014 { 1015 /* Write the IDAT and reset the zlib output buffer */ 1016 png_write_IDAT(png_ptr, png_ptr->zbuf, 1017 png_ptr->zbuf_size); 1018 png_ptr->zstream.next_out = png_ptr->zbuf; 1019 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size; 1020 wrote_IDAT = 1; 1021 } 1022 } while(wrote_IDAT == 1); 1023 1024 /* If there is any data left to be output, write it into a new IDAT */ 1025 if (png_ptr->zbuf_size != png_ptr->zstream.avail_out) 1026 { 1027 /* Write the IDAT and reset the zlib output buffer */ 1028 png_write_IDAT(png_ptr, png_ptr->zbuf, 1029 png_ptr->zbuf_size - png_ptr->zstream.avail_out); 1030 png_ptr->zstream.next_out = png_ptr->zbuf; 1031 png_ptr->zstream.avail_out = (uInt)png_ptr->zbuf_size; 1032 } 1033 png_ptr->flush_rows = 0; 1034 png_flush(png_ptr); 1035 } 1036 #endif /* PNG_WRITE_FLUSH_SUPPORTED */ 1037 1038 /* Free all memory used by the write */ 1039 void PNGAPI 1040 png_destroy_write_struct(png_structpp png_ptr_ptr, png_infopp info_ptr_ptr) 1041 { 1042 png_structp png_ptr = NULL; 1043 png_infop info_ptr = NULL; 1044 #ifdef PNG_USER_MEM_SUPPORTED 1045 png_free_ptr free_fn = NULL; 1046 png_voidp mem_ptr = NULL; 1047 #endif 1048 1049 png_debug(1, "in png_destroy_write_struct"); 1050 1051 if (png_ptr_ptr != NULL) 1052 { 1053 png_ptr = *png_ptr_ptr; 1054 #ifdef PNG_USER_MEM_SUPPORTED 1055 free_fn = png_ptr->free_fn; 1056 mem_ptr = png_ptr->mem_ptr; 1057 #endif 1058 } 1059 1060 #ifdef PNG_USER_MEM_SUPPORTED 1061 if (png_ptr != NULL) 1062 { 1063 free_fn = png_ptr->free_fn; 1064 mem_ptr = png_ptr->mem_ptr; 1065 } 1066 #endif 1067 1068 if (info_ptr_ptr != NULL) 1069 info_ptr = *info_ptr_ptr; 1070 1071 if (info_ptr != NULL) 1072 { 1073 if (png_ptr != NULL) 1074 { 1075 png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1); 1076 1077 #ifdef PNG_HANDLE_AS_UNKNOWN_SUPPORTED 1078 if (png_ptr->num_chunk_list) 1079 { 1080 png_free(png_ptr, png_ptr->chunk_list); 1081 png_ptr->chunk_list = NULL; 1082 png_ptr->num_chunk_list = 0; 1083 } 1084 #endif 1085 } 1086 1087 #ifdef PNG_USER_MEM_SUPPORTED 1088 png_destroy_struct_2((png_voidp)info_ptr, (png_free_ptr)free_fn, 1089 (png_voidp)mem_ptr); 1090 #else 1091 png_destroy_struct((png_voidp)info_ptr); 1092 #endif 1093 *info_ptr_ptr = NULL; 1094 } 1095 1096 if (png_ptr != NULL) 1097 { 1098 png_write_destroy(png_ptr); 1099 #ifdef PNG_USER_MEM_SUPPORTED 1100 png_destroy_struct_2((png_voidp)png_ptr, (png_free_ptr)free_fn, 1101 (png_voidp)mem_ptr); 1102 #else 1103 png_destroy_struct((png_voidp)png_ptr); 1104 #endif 1105 *png_ptr_ptr = NULL; 1106 } 1107 } 1108 1109 1110 /* Free any memory used in png_ptr struct (old method) */ 1111 void /* PRIVATE */ 1112 png_write_destroy(png_structp png_ptr) 1113 { 1114 #ifdef PNG_SETJMP_SUPPORTED 1115 jmp_buf tmp_jmp; /* Save jump buffer */ 1116 #endif 1117 png_error_ptr error_fn; 1118 png_error_ptr warning_fn; 1119 png_voidp error_ptr; 1120 #ifdef PNG_USER_MEM_SUPPORTED 1121 png_free_ptr free_fn; 1122 #endif 1123 1124 png_debug(1, "in png_write_destroy"); 1125 1126 /* Free any memory zlib uses */ 1127 deflateEnd(&png_ptr->zstream); 1128 1129 /* Free our memory. png_free checks NULL for us. */ 1130 png_free(png_ptr, png_ptr->zbuf); 1131 png_free(png_ptr, png_ptr->row_buf); 1132 #ifdef PNG_WRITE_FILTER_SUPPORTED 1133 png_free(png_ptr, png_ptr->prev_row); 1134 png_free(png_ptr, png_ptr->sub_row); 1135 png_free(png_ptr, png_ptr->up_row); 1136 png_free(png_ptr, png_ptr->avg_row); 1137 png_free(png_ptr, png_ptr->paeth_row); 1138 #endif 1139 1140 #ifdef PNG_TIME_RFC1123_SUPPORTED 1141 png_free(png_ptr, png_ptr->time_buffer); 1142 #endif 1143 1144 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED 1145 png_free(png_ptr, png_ptr->prev_filters); 1146 png_free(png_ptr, png_ptr->filter_weights); 1147 png_free(png_ptr, png_ptr->inv_filter_weights); 1148 png_free(png_ptr, png_ptr->filter_costs); 1149 png_free(png_ptr, png_ptr->inv_filter_costs); 1150 #endif 1151 1152 #ifdef PNG_SETJMP_SUPPORTED 1153 /* Reset structure */ 1154 png_memcpy(tmp_jmp, png_ptr->jmpbuf, png_sizeof(jmp_buf)); 1155 #endif 1156 1157 error_fn = png_ptr->error_fn; 1158 warning_fn = png_ptr->warning_fn; 1159 error_ptr = png_ptr->error_ptr; 1160 #ifdef PNG_USER_MEM_SUPPORTED 1161 free_fn = png_ptr->free_fn; 1162 #endif 1163 1164 png_memset(png_ptr, 0, png_sizeof(png_struct)); 1165 1166 png_ptr->error_fn = error_fn; 1167 png_ptr->warning_fn = warning_fn; 1168 png_ptr->error_ptr = error_ptr; 1169 #ifdef PNG_USER_MEM_SUPPORTED 1170 png_ptr->free_fn = free_fn; 1171 #endif 1172 1173 #ifdef PNG_SETJMP_SUPPORTED 1174 png_memcpy(png_ptr->jmpbuf, tmp_jmp, png_sizeof(jmp_buf)); 1175 #endif 1176 } 1177 1178 /* Allow the application to select one or more row filters to use. */ 1179 void PNGAPI 1180 png_set_filter(png_structp png_ptr, int method, int filters) 1181 { 1182 png_debug(1, "in png_set_filter"); 1183 1184 if (png_ptr == NULL) 1185 return; 1186 #ifdef PNG_MNG_FEATURES_SUPPORTED 1187 if ((png_ptr->mng_features_permitted & PNG_FLAG_MNG_FILTER_64) && 1188 (method == PNG_INTRAPIXEL_DIFFERENCING)) 1189 method = PNG_FILTER_TYPE_BASE; 1190 #endif 1191 if (method == PNG_FILTER_TYPE_BASE) 1192 { 1193 switch (filters & (PNG_ALL_FILTERS | 0x07)) 1194 { 1195 #ifdef PNG_WRITE_FILTER_SUPPORTED 1196 case 5: 1197 case 6: 1198 case 7: png_warning(png_ptr, "Unknown row filter for method 0"); 1199 #endif /* PNG_WRITE_FILTER_SUPPORTED */ 1200 case PNG_FILTER_VALUE_NONE: 1201 png_ptr->do_filter = PNG_FILTER_NONE; break; 1202 #ifdef PNG_WRITE_FILTER_SUPPORTED 1203 case PNG_FILTER_VALUE_SUB: 1204 png_ptr->do_filter = PNG_FILTER_SUB; break; 1205 case PNG_FILTER_VALUE_UP: 1206 png_ptr->do_filter = PNG_FILTER_UP; break; 1207 case PNG_FILTER_VALUE_AVG: 1208 png_ptr->do_filter = PNG_FILTER_AVG; break; 1209 case PNG_FILTER_VALUE_PAETH: 1210 png_ptr->do_filter = PNG_FILTER_PAETH; break; 1211 default: png_ptr->do_filter = (png_byte)filters; break; 1212 #else 1213 default: png_warning(png_ptr, "Unknown row filter for method 0"); 1214 #endif /* PNG_WRITE_FILTER_SUPPORTED */ 1215 } 1216 1217 /* If we have allocated the row_buf, this means we have already started 1218 * with the image and we should have allocated all of the filter buffers 1219 * that have been selected. If prev_row isn't already allocated, then 1220 * it is too late to start using the filters that need it, since we 1221 * will be missing the data in the previous row. If an application 1222 * wants to start and stop using particular filters during compression, 1223 * it should start out with all of the filters, and then add and 1224 * remove them after the start of compression. 1225 */ 1226 if (png_ptr->row_buf != NULL) 1227 { 1228 #ifdef PNG_WRITE_FILTER_SUPPORTED 1229 if ((png_ptr->do_filter & PNG_FILTER_SUB) && png_ptr->sub_row == NULL) 1230 { 1231 png_ptr->sub_row = (png_bytep)png_malloc(png_ptr, 1232 (png_ptr->rowbytes + 1)); 1233 png_ptr->sub_row[0] = PNG_FILTER_VALUE_SUB; 1234 } 1235 1236 if ((png_ptr->do_filter & PNG_FILTER_UP) && png_ptr->up_row == NULL) 1237 { 1238 if (png_ptr->prev_row == NULL) 1239 { 1240 png_warning(png_ptr, "Can't add Up filter after starting"); 1241 png_ptr->do_filter &= ~PNG_FILTER_UP; 1242 } 1243 else 1244 { 1245 png_ptr->up_row = (png_bytep)png_malloc(png_ptr, 1246 (png_ptr->rowbytes + 1)); 1247 png_ptr->up_row[0] = PNG_FILTER_VALUE_UP; 1248 } 1249 } 1250 1251 if ((png_ptr->do_filter & PNG_FILTER_AVG) && png_ptr->avg_row == NULL) 1252 { 1253 if (png_ptr->prev_row == NULL) 1254 { 1255 png_warning(png_ptr, "Can't add Average filter after starting"); 1256 png_ptr->do_filter &= ~PNG_FILTER_AVG; 1257 } 1258 else 1259 { 1260 png_ptr->avg_row = (png_bytep)png_malloc(png_ptr, 1261 (png_ptr->rowbytes + 1)); 1262 png_ptr->avg_row[0] = PNG_FILTER_VALUE_AVG; 1263 } 1264 } 1265 1266 if ((png_ptr->do_filter & PNG_FILTER_PAETH) && 1267 png_ptr->paeth_row == NULL) 1268 { 1269 if (png_ptr->prev_row == NULL) 1270 { 1271 png_warning(png_ptr, "Can't add Paeth filter after starting"); 1272 png_ptr->do_filter &= (png_byte)(~PNG_FILTER_PAETH); 1273 } 1274 else 1275 { 1276 png_ptr->paeth_row = (png_bytep)png_malloc(png_ptr, 1277 (png_ptr->rowbytes + 1)); 1278 png_ptr->paeth_row[0] = PNG_FILTER_VALUE_PAETH; 1279 } 1280 } 1281 1282 if (png_ptr->do_filter == PNG_NO_FILTERS) 1283 #endif /* PNG_WRITE_FILTER_SUPPORTED */ 1284 png_ptr->do_filter = PNG_FILTER_NONE; 1285 } 1286 } 1287 else 1288 png_error(png_ptr, "Unknown custom filter method"); 1289 } 1290 1291 /* This allows us to influence the way in which libpng chooses the "best" 1292 * filter for the current scanline. While the "minimum-sum-of-absolute- 1293 * differences metric is relatively fast and effective, there is some 1294 * question as to whether it can be improved upon by trying to keep the 1295 * filtered data going to zlib more consistent, hopefully resulting in 1296 * better compression. 1297 */ 1298 #ifdef PNG_WRITE_WEIGHTED_FILTER_SUPPORTED /* GRR 970116 */ 1299 void PNGAPI 1300 png_set_filter_heuristics(png_structp png_ptr, int heuristic_method, 1301 int num_weights, png_doublep filter_weights, 1302 png_doublep filter_costs) 1303 { 1304 int i; 1305 1306 png_debug(1, "in png_set_filter_heuristics"); 1307 1308 if (png_ptr == NULL) 1309 return; 1310 if (heuristic_method >= PNG_FILTER_HEURISTIC_LAST) 1311 { 1312 png_warning(png_ptr, "Unknown filter heuristic method"); 1313 return; 1314 } 1315 1316 if (heuristic_method == PNG_FILTER_HEURISTIC_DEFAULT) 1317 { 1318 heuristic_method = PNG_FILTER_HEURISTIC_UNWEIGHTED; 1319 } 1320 1321 if (num_weights < 0 || filter_weights == NULL || 1322 heuristic_method == PNG_FILTER_HEURISTIC_UNWEIGHTED) 1323 { 1324 num_weights = 0; 1325 } 1326 1327 png_ptr->num_prev_filters = (png_byte)num_weights; 1328 png_ptr->heuristic_method = (png_byte)heuristic_method; 1329 1330 if (num_weights > 0) 1331 { 1332 if (png_ptr->prev_filters == NULL) 1333 { 1334 png_ptr->prev_filters = (png_bytep)png_malloc(png_ptr, 1335 (png_uint_32)(png_sizeof(png_byte) * num_weights)); 1336 1337 /* To make sure that the weighting starts out fairly */ 1338 for (i = 0; i < num_weights; i++) 1339 { 1340 png_ptr->prev_filters[i] = 255; 1341 } 1342 } 1343 1344 if (png_ptr->filter_weights == NULL) 1345 { 1346 png_ptr->filter_weights = (png_uint_16p)png_malloc(png_ptr, 1347 (png_uint_32)(png_sizeof(png_uint_16) * num_weights)); 1348 1349 png_ptr->inv_filter_weights = (png_uint_16p)png_malloc(png_ptr, 1350 (png_uint_32)(png_sizeof(png_uint_16) * num_weights)); 1351 for (i = 0; i < num_weights; i++) 1352 { 1353 png_ptr->inv_filter_weights[i] = 1354 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR; 1355 } 1356 } 1357 1358 for (i = 0; i < num_weights; i++) 1359 { 1360 if (filter_weights[i] < 0.0) 1361 { 1362 png_ptr->inv_filter_weights[i] = 1363 png_ptr->filter_weights[i] = PNG_WEIGHT_FACTOR; 1364 } 1365 else 1366 { 1367 png_ptr->inv_filter_weights[i] = 1368 (png_uint_16)((double)PNG_WEIGHT_FACTOR*filter_weights[i]+0.5); 1369 png_ptr->filter_weights[i] = 1370 (png_uint_16)((double)PNG_WEIGHT_FACTOR/filter_weights[i]+0.5); 1371 } 1372 } 1373 } 1374 1375 /* If, in the future, there are other filter methods, this would 1376 * need to be based on png_ptr->filter. 1377 */ 1378 if (png_ptr->filter_costs == NULL) 1379 { 1380 png_ptr->filter_costs = (png_uint_16p)png_malloc(png_ptr, 1381 (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST)); 1382 1383 png_ptr->inv_filter_costs = (png_uint_16p)png_malloc(png_ptr, 1384 (png_uint_32)(png_sizeof(png_uint_16) * PNG_FILTER_VALUE_LAST)); 1385 1386 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++) 1387 { 1388 png_ptr->inv_filter_costs[i] = 1389 png_ptr->filter_costs[i] = PNG_COST_FACTOR; 1390 } 1391 } 1392 1393 /* Here is where we set the relative costs of the different filters. We 1394 * should take the desired compression level into account when setting 1395 * the costs, so that Paeth, for instance, has a high relative cost at low 1396 * compression levels, while it has a lower relative cost at higher 1397 * compression settings. The filter types are in order of increasing 1398 * relative cost, so it would be possible to do this with an algorithm. 1399 */ 1400 for (i = 0; i < PNG_FILTER_VALUE_LAST; i++) 1401 { 1402 if (filter_costs == NULL || filter_costs[i] < 0.0) 1403 { 1404 png_ptr->inv_filter_costs[i] = 1405 png_ptr->filter_costs[i] = PNG_COST_FACTOR; 1406 } 1407 else if (filter_costs[i] >= 1.0) 1408 { 1409 png_ptr->inv_filter_costs[i] = 1410 (png_uint_16)((double)PNG_COST_FACTOR / filter_costs[i] + 0.5); 1411 png_ptr->filter_costs[i] = 1412 (png_uint_16)((double)PNG_COST_FACTOR * filter_costs[i] + 0.5); 1413 } 1414 } 1415 } 1416 #endif /* PNG_WRITE_WEIGHTED_FILTER_SUPPORTED */ 1417 1418 void PNGAPI 1419 png_set_compression_level(png_structp png_ptr, int level) 1420 { 1421 png_debug(1, "in png_set_compression_level"); 1422 1423 if (png_ptr == NULL) 1424 return; 1425 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_LEVEL; 1426 png_ptr->zlib_level = level; 1427 } 1428 1429 void PNGAPI 1430 png_set_compression_mem_level(png_structp png_ptr, int mem_level) 1431 { 1432 png_debug(1, "in png_set_compression_mem_level"); 1433 1434 if (png_ptr == NULL) 1435 return; 1436 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_MEM_LEVEL; 1437 png_ptr->zlib_mem_level = mem_level; 1438 } 1439 1440 void PNGAPI 1441 png_set_compression_strategy(png_structp png_ptr, int strategy) 1442 { 1443 png_debug(1, "in png_set_compression_strategy"); 1444 1445 if (png_ptr == NULL) 1446 return; 1447 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_STRATEGY; 1448 png_ptr->zlib_strategy = strategy; 1449 } 1450 1451 void PNGAPI 1452 png_set_compression_window_bits(png_structp png_ptr, int window_bits) 1453 { 1454 if (png_ptr == NULL) 1455 return; 1456 if (window_bits > 15) 1457 png_warning(png_ptr, "Only compression windows <= 32k supported by PNG"); 1458 else if (window_bits < 8) 1459 png_warning(png_ptr, "Only compression windows >= 256 supported by PNG"); 1460 #ifndef WBITS_8_OK 1461 /* Avoid libpng bug with 256-byte windows */ 1462 if (window_bits == 8) 1463 { 1464 png_warning(png_ptr, "Compression window is being reset to 512"); 1465 window_bits = 9; 1466 } 1467 #endif 1468 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_WINDOW_BITS; 1469 png_ptr->zlib_window_bits = window_bits; 1470 } 1471 1472 void PNGAPI 1473 png_set_compression_method(png_structp png_ptr, int method) 1474 { 1475 png_debug(1, "in png_set_compression_method"); 1476 1477 if (png_ptr == NULL) 1478 return; 1479 if (method != 8) 1480 png_warning(png_ptr, "Only compression method 8 is supported by PNG"); 1481 png_ptr->flags |= PNG_FLAG_ZLIB_CUSTOM_METHOD; 1482 png_ptr->zlib_method = method; 1483 } 1484 1485 void PNGAPI 1486 png_set_write_status_fn(png_structp png_ptr, png_write_status_ptr write_row_fn) 1487 { 1488 if (png_ptr == NULL) 1489 return; 1490 png_ptr->write_row_fn = write_row_fn; 1491 } 1492 1493 #ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED 1494 void PNGAPI 1495 png_set_write_user_transform_fn(png_structp png_ptr, png_user_transform_ptr 1496 write_user_transform_fn) 1497 { 1498 png_debug(1, "in png_set_write_user_transform_fn"); 1499 1500 if (png_ptr == NULL) 1501 return; 1502 png_ptr->transformations |= PNG_USER_TRANSFORM; 1503 png_ptr->write_user_transform_fn = write_user_transform_fn; 1504 } 1505 #endif 1506 1507 1508 #ifdef PNG_INFO_IMAGE_SUPPORTED 1509 void PNGAPI 1510 png_write_png(png_structp png_ptr, png_infop info_ptr, 1511 int transforms, voidp params) 1512 { 1513 if (png_ptr == NULL || info_ptr == NULL) 1514 return; 1515 1516 /* Write the file header information. */ 1517 png_write_info(png_ptr, info_ptr); 1518 1519 /* ------ these transformations don't touch the info structure ------- */ 1520 1521 #ifdef PNG_WRITE_INVERT_SUPPORTED 1522 /* Invert monochrome pixels */ 1523 if (transforms & PNG_TRANSFORM_INVERT_MONO) 1524 png_set_invert_mono(png_ptr); 1525 #endif 1526 1527 #ifdef PNG_WRITE_SHIFT_SUPPORTED 1528 /* Shift the pixels up to a legal bit depth and fill in 1529 * as appropriate to correctly scale the image. 1530 */ 1531 if ((transforms & PNG_TRANSFORM_SHIFT) 1532 && (info_ptr->valid & PNG_INFO_sBIT)) 1533 png_set_shift(png_ptr, &info_ptr->sig_bit); 1534 #endif 1535 1536 #ifdef PNG_WRITE_PACK_SUPPORTED 1537 /* Pack pixels into bytes */ 1538 if (transforms & PNG_TRANSFORM_PACKING) 1539 png_set_packing(png_ptr); 1540 #endif 1541 1542 #ifdef PNG_WRITE_SWAP_ALPHA_SUPPORTED 1543 /* Swap location of alpha bytes from ARGB to RGBA */ 1544 if (transforms & PNG_TRANSFORM_SWAP_ALPHA) 1545 png_set_swap_alpha(png_ptr); 1546 #endif 1547 1548 #ifdef PNG_WRITE_FILLER_SUPPORTED 1549 /* Pack XRGB/RGBX/ARGB/RGBA into * RGB (4 channels -> 3 channels) */ 1550 if (transforms & PNG_TRANSFORM_STRIP_FILLER_AFTER) 1551 png_set_filler(png_ptr, 0, PNG_FILLER_AFTER); 1552 else if (transforms & PNG_TRANSFORM_STRIP_FILLER_BEFORE) 1553 png_set_filler(png_ptr, 0, PNG_FILLER_BEFORE); 1554 #endif 1555 1556 #ifdef PNG_WRITE_BGR_SUPPORTED 1557 /* Flip BGR pixels to RGB */ 1558 if (transforms & PNG_TRANSFORM_BGR) 1559 png_set_bgr(png_ptr); 1560 #endif 1561 1562 #ifdef PNG_WRITE_SWAP_SUPPORTED 1563 /* Swap bytes of 16-bit files to most significant byte first */ 1564 if (transforms & PNG_TRANSFORM_SWAP_ENDIAN) 1565 png_set_swap(png_ptr); 1566 #endif 1567 1568 #ifdef PNG_WRITE_PACKSWAP_SUPPORTED 1569 /* Swap bits of 1, 2, 4 bit packed pixel formats */ 1570 if (transforms & PNG_TRANSFORM_PACKSWAP) 1571 png_set_packswap(png_ptr); 1572 #endif 1573 1574 #ifdef PNG_WRITE_INVERT_ALPHA_SUPPORTED 1575 /* Invert the alpha channel from opacity to transparency */ 1576 if (transforms & PNG_TRANSFORM_INVERT_ALPHA) 1577 png_set_invert_alpha(png_ptr); 1578 #endif 1579 1580 /* ----------------------- end of transformations ------------------- */ 1581 1582 /* Write the bits */ 1583 if (info_ptr->valid & PNG_INFO_IDAT) 1584 png_write_image(png_ptr, info_ptr->row_pointers); 1585 1586 /* It is REQUIRED to call this to finish writing the rest of the file */ 1587 png_write_end(png_ptr, info_ptr); 1588 1589 transforms = transforms; /* Quiet compiler warnings */ 1590 params = params; 1591 } 1592 #endif 1593 #endif /* PNG_WRITE_SUPPORTED */ 1594
