git.cworth.org Git - apitrace/blob - thirdparty/libpng/pngwtran.c

   1
   2 /* pngwtran.c - transforms the data in a row for PNG writers
   3  *
   4  * Last changed in libpng 1.5.6 [November 3, 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 #include "pngpriv.h"
  15
  16 #ifdef PNG_WRITE_SUPPORTED
  17
  18 #ifdef PNG_WRITE_TRANSFORMS_SUPPORTED
  19 /* Transform the data according to the user's wishes.  The order of
  20  * transformations is significant.
  21  */
  22 void /* PRIVATE */
  23 png_do_write_transformations(png_structp png_ptr, png_row_infop row_info)
  24 {
  25    png_debug(1, "in png_do_write_transformations");
  26
  27    if (png_ptr == NULL)
  28       return;
  29
  30 #ifdef PNG_WRITE_USER_TRANSFORM_SUPPORTED
  31    if (png_ptr->transformations & PNG_USER_TRANSFORM)
  32       if (png_ptr->write_user_transform_fn != NULL)
  33          (*(png_ptr->write_user_transform_fn)) /* User write transform
  34                                                  function */
  35              (png_ptr,  /* png_ptr */
  36              row_info,  /* row_info: */
  37                 /*  png_uint_32 width;       width of row */
  38                 /*  png_size_t rowbytes;     number of bytes in row */
  39                 /*  png_byte color_type;     color type of pixels */
  40                 /*  png_byte bit_depth;      bit depth of samples */
  41                 /*  png_byte channels;       number of channels (1-4) */
  42                 /*  png_byte pixel_depth;    bits per pixel (depth*channels) */
  43              png_ptr->row_buf + 1);      /* start of pixel data for row */
  44 #endif
  45
  46 #ifdef PNG_WRITE_FILLER_SUPPORTED
  47    if (png_ptr->transformations & PNG_FILLER)
  48       png_do_strip_channel(row_info, png_ptr->row_buf + 1,
  49          !(png_ptr->flags & PNG_FLAG_FILLER_AFTER));
  50 #endif
  51
  52 #ifdef PNG_WRITE_PACKSWAP_SUPPORTED
  53    if (png_ptr->transformations & PNG_PACKSWAP)
  54       png_do_packswap(row_info, png_ptr->row_buf + 1);
  55 #endif
  56
  57 #ifdef PNG_WRITE_PACK_SUPPORTED
  58    if (png_ptr->transformations & PNG_PACK)
  59       png_do_pack(row_info, png_ptr->row_buf + 1,
  60           (png_uint_32)png_ptr->bit_depth);
  61 #endif
  62
  63 #ifdef PNG_WRITE_SWAP_SUPPORTED
  64    if (png_ptr->transformations & PNG_SWAP_BYTES)
  65       png_do_swap(row_info, png_ptr->row_buf + 1);
  66 #endif
  67
  68 #ifdef PNG_WRITE_SHIFT_SUPPORTED
  69    if (png_ptr->transformations & PNG_SHIFT)
  70       png_do_shift(row_info, png_ptr->row_buf + 1,
  71           &(png_ptr->shift));
  72 #endif
  73
  74 #ifdef PNG_WRITE_SWAP_ALPHA_SUPPORTED
  75    if (png_ptr->transformations & PNG_SWAP_ALPHA)
  76       png_do_write_swap_alpha(row_info, png_ptr->row_buf + 1);
  77 #endif
  78
  79 #ifdef PNG_WRITE_INVERT_ALPHA_SUPPORTED
  80    if (png_ptr->transformations & PNG_INVERT_ALPHA)
  81       png_do_write_invert_alpha(row_info, png_ptr->row_buf + 1);
  82 #endif
  83
  84 #ifdef PNG_WRITE_BGR_SUPPORTED
  85    if (png_ptr->transformations & PNG_BGR)
  86       png_do_bgr(row_info, png_ptr->row_buf + 1);
  87 #endif
  88
  89 #ifdef PNG_WRITE_INVERT_SUPPORTED
  90    if (png_ptr->transformations & PNG_INVERT_MONO)
  91       png_do_invert(row_info, png_ptr->row_buf + 1);
  92 #endif
  93 }
  94
  95 #ifdef PNG_WRITE_PACK_SUPPORTED
  96 /* Pack pixels into bytes.  Pass the true bit depth in bit_depth.  The
  97  * row_info bit depth should be 8 (one pixel per byte).  The channels
  98  * should be 1 (this only happens on grayscale and paletted images).
  99  */
 100 void /* PRIVATE */
 101 png_do_pack(png_row_infop row_info, png_bytep row, png_uint_32 bit_depth)
 102 {
 103    png_debug(1, "in png_do_pack");
 104
 105    if (row_info->bit_depth == 8 &&
 106       row_info->channels == 1)
 107    {
 108       switch ((int)bit_depth)
 109       {
 110          case 1:
 111          {
 112             png_bytep sp, dp;
 113             int mask, v;
 114             png_uint_32 i;
 115             png_uint_32 row_width = row_info->width;
 116
 117             sp = row;
 118             dp = row;
 119             mask = 0x80;
 120             v = 0;
 121
 122             for (i = 0; i < row_width; i++)
 123             {
 124                if (*sp != 0)
 125                   v |= mask;
 126
 127                sp++;
 128
 129                if (mask > 1)
 130                   mask >>= 1;
 131
 132                else
 133                {
 134                   mask = 0x80;
 135                   *dp = (png_byte)v;
 136                   dp++;
 137                   v = 0;
 138                }
 139             }
 140
 141             if (mask != 0x80)
 142                *dp = (png_byte)v;
 143
 144             break;
 145          }
 146
 147          case 2:
 148          {
 149             png_bytep sp, dp;
 150             int shift, v;
 151             png_uint_32 i;
 152             png_uint_32 row_width = row_info->width;
 153
 154             sp = row;
 155             dp = row;
 156             shift = 6;
 157             v = 0;
 158
 159             for (i = 0; i < row_width; i++)
 160             {
 161                png_byte value;
 162
 163                value = (png_byte)(*sp & 0x03);
 164                v |= (value << shift);
 165
 166                if (shift == 0)
 167                {
 168                   shift = 6;
 169                   *dp = (png_byte)v;
 170                   dp++;
 171                   v = 0;
 172                }
 173
 174                else
 175                   shift -= 2;
 176
 177                sp++;
 178             }
 179
 180             if (shift != 6)
 181                *dp = (png_byte)v;
 182
 183             break;
 184          }
 185
 186          case 4:
 187          {
 188             png_bytep sp, dp;
 189             int shift, v;
 190             png_uint_32 i;
 191             png_uint_32 row_width = row_info->width;
 192
 193             sp = row;
 194             dp = row;
 195             shift = 4;
 196             v = 0;
 197
 198             for (i = 0; i < row_width; i++)
 199             {
 200                png_byte value;
 201
 202                value = (png_byte)(*sp & 0x0f);
 203                v |= (value << shift);
 204
 205                if (shift == 0)
 206                {
 207                   shift = 4;
 208                   *dp = (png_byte)v;
 209                   dp++;
 210                   v = 0;
 211                }
 212
 213                else
 214                   shift -= 4;
 215
 216                sp++;
 217             }
 218
 219             if (shift != 4)
 220                *dp = (png_byte)v;
 221
 222             break;
 223          }
 224
 225          default:
 226             break;
 227       }
 228
 229       row_info->bit_depth = (png_byte)bit_depth;
 230       row_info->pixel_depth = (png_byte)(bit_depth * row_info->channels);
 231       row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth,
 232           row_info->width);
 233    }
 234 }
 235 #endif
 236
 237 #ifdef PNG_WRITE_SHIFT_SUPPORTED
 238 /* Shift pixel values to take advantage of whole range.  Pass the
 239  * true number of bits in bit_depth.  The row should be packed
 240  * according to row_info->bit_depth.  Thus, if you had a row of
 241  * bit depth 4, but the pixels only had values from 0 to 7, you
 242  * would pass 3 as bit_depth, and this routine would translate the
 243  * data to 0 to 15.
 244  */
 245 void /* PRIVATE */
 246 png_do_shift(png_row_infop row_info, png_bytep row,
 247     png_const_color_8p bit_depth)
 248 {
 249    png_debug(1, "in png_do_shift");
 250
 251    if (row_info->color_type != PNG_COLOR_TYPE_PALETTE)
 252    {
 253       int shift_start[4], shift_dec[4];
 254       int channels = 0;
 255
 256       if (row_info->color_type & PNG_COLOR_MASK_COLOR)
 257       {
 258          shift_start[channels] = row_info->bit_depth - bit_depth->red;
 259          shift_dec[channels] = bit_depth->red;
 260          channels++;
 261
 262          shift_start[channels] = row_info->bit_depth - bit_depth->green;
 263          shift_dec[channels] = bit_depth->green;
 264          channels++;
 265
 266          shift_start[channels] = row_info->bit_depth - bit_depth->blue;
 267          shift_dec[channels] = bit_depth->blue;
 268          channels++;
 269       }
 270
 271       else
 272       {
 273          shift_start[channels] = row_info->bit_depth - bit_depth->gray;
 274          shift_dec[channels] = bit_depth->gray;
 275          channels++;
 276       }
 277
 278       if (row_info->color_type & PNG_COLOR_MASK_ALPHA)
 279       {
 280          shift_start[channels] = row_info->bit_depth - bit_depth->alpha;
 281          shift_dec[channels] = bit_depth->alpha;
 282          channels++;
 283       }
 284
 285       /* With low row depths, could only be grayscale, so one channel */
 286       if (row_info->bit_depth < 8)
 287       {
 288          png_bytep bp = row;
 289          png_size_t i;
 290          png_byte mask;
 291          png_size_t row_bytes = row_info->rowbytes;
 292
 293          if (bit_depth->gray == 1 && row_info->bit_depth == 2)
 294             mask = 0x55;
 295
 296          else if (row_info->bit_depth == 4 && bit_depth->gray == 3)
 297             mask = 0x11;
 298
 299          else
 300             mask = 0xff;
 301
 302          for (i = 0; i < row_bytes; i++, bp++)
 303          {
 304             png_uint_16 v;
 305             int j;
 306
 307             v = *bp;
 308             *bp = 0;
 309
 310             for (j = shift_start[0]; j > -shift_dec[0]; j -= shift_dec[0])
 311             {
 312                if (j > 0)
 313                   *bp |= (png_byte)((v << j) & 0xff);
 314
 315                else
 316                   *bp |= (png_byte)((v >> (-j)) & mask);
 317             }
 318          }
 319       }
 320
 321       else if (row_info->bit_depth == 8)
 322       {
 323          png_bytep bp = row;
 324          png_uint_32 i;
 325          png_uint_32 istop = channels * row_info->width;
 326
 327          for (i = 0; i < istop; i++, bp++)
 328          {
 329
 330             png_uint_16 v;
 331             int j;
 332             int c = (int)(i%channels);
 333
 334             v = *bp;
 335             *bp = 0;
 336
 337             for (j = shift_start[c]; j > -shift_dec[c]; j -= shift_dec[c])
 338             {
 339                if (j > 0)
 340                   *bp |= (png_byte)((v << j) & 0xff);
 341
 342                else
 343                   *bp |= (png_byte)((v >> (-j)) & 0xff);
 344             }
 345          }
 346       }
 347
 348       else
 349       {
 350          png_bytep bp;
 351          png_uint_32 i;
 352          png_uint_32 istop = channels * row_info->width;
 353
 354          for (bp = row, i = 0; i < istop; i++)
 355          {
 356             int c = (int)(i%channels);
 357             png_uint_16 value, v;
 358             int j;
 359
 360             v = (png_uint_16)(((png_uint_16)(*bp) << 8) + *(bp + 1));
 361             value = 0;
 362
 363             for (j = shift_start[c]; j > -shift_dec[c]; j -= shift_dec[c])
 364             {
 365                if (j > 0)
 366                   value |= (png_uint_16)((v << j) & (png_uint_16)0xffff);
 367
 368                else
 369                   value |= (png_uint_16)((v >> (-j)) & (png_uint_16)0xffff);
 370             }
 371             *bp++ = (png_byte)(value >> 8);
 372             *bp++ = (png_byte)(value & 0xff);
 373          }
 374       }
 375    }
 376 }
 377 #endif
 378
 379 #ifdef PNG_WRITE_SWAP_ALPHA_SUPPORTED
 380 void /* PRIVATE */
 381 png_do_write_swap_alpha(png_row_infop row_info, png_bytep row)
 382 {
 383    png_debug(1, "in png_do_write_swap_alpha");
 384
 385    {
 386       if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
 387       {
 388          if (row_info->bit_depth == 8)
 389          {
 390             /* This converts from ARGB to RGBA */
 391             png_bytep sp, dp;
 392             png_uint_32 i;
 393             png_uint_32 row_width = row_info->width;
 394
 395             for (i = 0, sp = dp = row; i < row_width; i++)
 396             {
 397                png_byte save = *(sp++);
 398                *(dp++) = *(sp++);
 399                *(dp++) = *(sp++);
 400                *(dp++) = *(sp++);
 401                *(dp++) = save;
 402             }
 403          }
 404
 405 #ifdef PNG_WRITE_16BIT_SUPPORTED
 406          else
 407          {
 408             /* This converts from AARRGGBB to RRGGBBAA */
 409             png_bytep sp, dp;
 410             png_uint_32 i;
 411             png_uint_32 row_width = row_info->width;
 412
 413             for (i = 0, sp = dp = row; i < row_width; i++)
 414             {
 415                png_byte save[2];
 416                save[0] = *(sp++);
 417                save[1] = *(sp++);
 418                *(dp++) = *(sp++);
 419                *(dp++) = *(sp++);
 420                *(dp++) = *(sp++);
 421                *(dp++) = *(sp++);
 422                *(dp++) = *(sp++);
 423                *(dp++) = *(sp++);
 424                *(dp++) = save[0];
 425                *(dp++) = save[1];
 426             }
 427          }
 428 #endif /* PNG_WRITE_16BIT_SUPPORTED */
 429       }
 430
 431       else if (row_info->color_type == PNG_COLOR_TYPE_GRAY_ALPHA)
 432       {
 433          if (row_info->bit_depth == 8)
 434          {
 435             /* This converts from AG to GA */
 436             png_bytep sp, dp;
 437             png_uint_32 i;
 438             png_uint_32 row_width = row_info->width;
 439
 440             for (i = 0, sp = dp = row; i < row_width; i++)
 441             {
 442                png_byte save = *(sp++);
 443                *(dp++) = *(sp++);
 444                *(dp++) = save;
 445             }
 446          }
 447
 448 #ifdef PNG_WRITE_16BIT_SUPPORTED
 449          else
 450          {
 451             /* This converts from AAGG to GGAA */
 452             png_bytep sp, dp;
 453             png_uint_32 i;
 454             png_uint_32 row_width = row_info->width;
 455
 456             for (i = 0, sp = dp = row; i < row_width; i++)
 457             {
 458                png_byte save[2];
 459                save[0] = *(sp++);
 460                save[1] = *(sp++);
 461                *(dp++) = *(sp++);
 462                *(dp++) = *(sp++);
 463                *(dp++) = save[0];
 464                *(dp++) = save[1];
 465             }
 466          }
 467 #endif /* PNG_WRITE_16BIT_SUPPORTED */
 468       }
 469    }
 470 }
 471 #endif
 472
 473 #ifdef PNG_WRITE_INVERT_ALPHA_SUPPORTED
 474 void /* PRIVATE */
 475 png_do_write_invert_alpha(png_row_infop row_info, png_bytep row)
 476 {
 477    png_debug(1, "in png_do_write_invert_alpha");
 478
 479    {
 480       if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
 481       {
 482          if (row_info->bit_depth == 8)
 483          {
 484             /* This inverts the alpha channel in RGBA */
 485             png_bytep sp, dp;
 486             png_uint_32 i;
 487             png_uint_32 row_width = row_info->width;
 488
 489             for (i = 0, sp = dp = row; i < row_width; i++)
 490             {
 491                /* Does nothing
 492                *(dp++) = *(sp++);
 493                *(dp++) = *(sp++);
 494                *(dp++) = *(sp++);
 495                */
 496                sp+=3; dp = sp;
 497                *(dp++) = (png_byte)(255 - *(sp++));
 498             }
 499          }
 500
 501 #ifdef PNG_WRITE_16BIT_SUPPORTED
 502          else
 503          {
 504             /* This inverts the alpha channel in RRGGBBAA */
 505             png_bytep sp, dp;
 506             png_uint_32 i;
 507             png_uint_32 row_width = row_info->width;
 508
 509             for (i = 0, sp = dp = row; i < row_width; i++)
 510             {
 511                /* Does nothing
 512                *(dp++) = *(sp++);
 513                *(dp++) = *(sp++);
 514                *(dp++) = *(sp++);
 515                *(dp++) = *(sp++);
 516                *(dp++) = *(sp++);
 517                *(dp++) = *(sp++);
 518                */
 519                sp+=6; dp = sp;
 520                *(dp++) = (png_byte)(255 - *(sp++));
 521                *(dp++) = (png_byte)(255 - *(sp++));
 522             }
 523          }
 524 #endif /* PNG_WRITE_16BIT_SUPPORTED */
 525       }
 526
 527       else if (row_info->color_type == PNG_COLOR_TYPE_GRAY_ALPHA)
 528       {
 529          if (row_info->bit_depth == 8)
 530          {
 531             /* This inverts the alpha channel in GA */
 532             png_bytep sp, dp;
 533             png_uint_32 i;
 534             png_uint_32 row_width = row_info->width;
 535
 536             for (i = 0, sp = dp = row; i < row_width; i++)
 537             {
 538                *(dp++) = *(sp++);
 539                *(dp++) = (png_byte)(255 - *(sp++));
 540             }
 541          }
 542
 543 #ifdef PNG_WRITE_16BIT_SUPPORTED
 544          else
 545          {
 546             /* This inverts the alpha channel in GGAA */
 547             png_bytep sp, dp;
 548             png_uint_32 i;
 549             png_uint_32 row_width = row_info->width;
 550
 551             for (i = 0, sp = dp = row; i < row_width; i++)
 552             {
 553                /* Does nothing
 554                *(dp++) = *(sp++);
 555                *(dp++) = *(sp++);
 556                */
 557                sp+=2; dp = sp;
 558                *(dp++) = (png_byte)(255 - *(sp++));
 559                *(dp++) = (png_byte)(255 - *(sp++));
 560             }
 561          }
 562 #endif /* PNG_WRITE_16BIT_SUPPORTED */
 563       }
 564    }
 565 }
 566 #endif
 567 #endif /* PNG_WRITE_TRANSFORMS_SUPPORTED */
 568
 569 #ifdef PNG_MNG_FEATURES_SUPPORTED
 570 /* Undoes intrapixel differencing  */
 571 void /* PRIVATE */
 572 png_do_write_intrapixel(png_row_infop row_info, png_bytep row)
 573 {
 574    png_debug(1, "in png_do_write_intrapixel");
 575
 576    if ((row_info->color_type & PNG_COLOR_MASK_COLOR))
 577    {
 578       int bytes_per_pixel;
 579       png_uint_32 row_width = row_info->width;
 580       if (row_info->bit_depth == 8)
 581       {
 582          png_bytep rp;
 583          png_uint_32 i;
 584
 585          if (row_info->color_type == PNG_COLOR_TYPE_RGB)
 586             bytes_per_pixel = 3;
 587
 588          else if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
 589             bytes_per_pixel = 4;
 590
 591          else
 592             return;
 593
 594          for (i = 0, rp = row; i < row_width; i++, rp += bytes_per_pixel)
 595          {
 596             *(rp)     = (png_byte)((*rp       - *(rp + 1)) & 0xff);
 597             *(rp + 2) = (png_byte)((*(rp + 2) - *(rp + 1)) & 0xff);
 598          }
 599       }
 600
 601 #ifdef PNG_WRITE_16BIT_SUPPORTED
 602       else if (row_info->bit_depth == 16)
 603       {
 604          png_bytep rp;
 605          png_uint_32 i;
 606
 607          if (row_info->color_type == PNG_COLOR_TYPE_RGB)
 608             bytes_per_pixel = 6;
 609
 610          else if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
 611             bytes_per_pixel = 8;
 612
 613          else
 614             return;
 615
 616          for (i = 0, rp = row; i < row_width; i++, rp += bytes_per_pixel)
 617          {
 618             png_uint_32 s0   = (*(rp    ) << 8) | *(rp + 1);
 619             png_uint_32 s1   = (*(rp + 2) << 8) | *(rp + 3);
 620             png_uint_32 s2   = (*(rp + 4) << 8) | *(rp + 5);
 621             png_uint_32 red  = (png_uint_32)((s0 - s1) & 0xffffL);
 622             png_uint_32 blue = (png_uint_32)((s2 - s1) & 0xffffL);
 623             *(rp    ) = (png_byte)((red >> 8) & 0xff);
 624             *(rp + 1) = (png_byte)(red & 0xff);
 625             *(rp + 4) = (png_byte)((blue >> 8) & 0xff);
 626             *(rp + 5) = (png_byte)(blue & 0xff);
 627          }
 628       }
 629 #endif /* PNG_WRITE_16BIT_SUPPORTED */
 630    }
 631 }
 632 #endif /* PNG_MNG_FEATURES_SUPPORTED */
 633 #endif /* PNG_WRITE_SUPPORTED */