pngrtran.c 160.8 KB
Newer Older
G
Guy Schalnat 已提交
1

A
Andreas Dilger 已提交
2
/* pngrtran.c - transforms the data in a row for PNG readers
3
 *
4
 * Last changed in libpng 1.5.4 [(PENDING RELEASE)]
5
 * Copyright (c) 1998-2011 Glenn Randers-Pehrson
6 7
 * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
 * (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
 * This file contains functions optionally called by an application
14
 * in order to tell libpng how to handle data when reading a PNG.
15
 * Transformations that are used in both reading and writing are
16 17
 * in pngtrans.c.
 */
G
Guy Schalnat 已提交
18

19
#include "pngpriv.h"
20

21 22
#ifdef PNG_READ_SUPPORTED

23
/* Set the action on getting a CRC error for an ancillary or critical chunk. */
24
void PNGAPI
25 26
png_set_crc_action(png_structp png_ptr, int crit_action, int ancil_action)
{
27
   png_debug(1, "in png_set_crc_action");
28

29 30
   if (png_ptr == NULL)
      return;
31 32

   /* Tell libpng how we react to CRC errors in critical chunks */
33 34
   switch (crit_action)
   {
35
      case PNG_CRC_NO_CHANGE:                        /* Leave setting as is */
36
         break;
37

38
      case PNG_CRC_WARN_USE:                               /* Warn/use data */
39 40 41
         png_ptr->flags &= ~PNG_FLAG_CRC_CRITICAL_MASK;
         png_ptr->flags |= PNG_FLAG_CRC_CRITICAL_USE;
         break;
42

43
      case PNG_CRC_QUIET_USE:                             /* Quiet/use data */
44 45 46 47
         png_ptr->flags &= ~PNG_FLAG_CRC_CRITICAL_MASK;
         png_ptr->flags |= PNG_FLAG_CRC_CRITICAL_USE |
                           PNG_FLAG_CRC_CRITICAL_IGNORE;
         break;
48

49
      case PNG_CRC_WARN_DISCARD:    /* Not a valid action for critical data */
50 51
         png_warning(png_ptr,
            "Can't discard critical data on CRC error");
52
      case PNG_CRC_ERROR_QUIT:                                /* Error/quit */
53

54 55 56 57 58 59
      case PNG_CRC_DEFAULT:
      default:
         png_ptr->flags &= ~PNG_FLAG_CRC_CRITICAL_MASK;
         break;
   }

60
   /* Tell libpng how we react to CRC errors in ancillary chunks */
61 62
   switch (ancil_action)
   {
63
      case PNG_CRC_NO_CHANGE:                       /* Leave setting as is */
64
         break;
65

66
      case PNG_CRC_WARN_USE:                              /* Warn/use data */
67 68 69
         png_ptr->flags &= ~PNG_FLAG_CRC_ANCILLARY_MASK;
         png_ptr->flags |= PNG_FLAG_CRC_ANCILLARY_USE;
         break;
70

71
      case PNG_CRC_QUIET_USE:                            /* Quiet/use data */
72 73 74 75
         png_ptr->flags &= ~PNG_FLAG_CRC_ANCILLARY_MASK;
         png_ptr->flags |= PNG_FLAG_CRC_ANCILLARY_USE |
                           PNG_FLAG_CRC_ANCILLARY_NOWARN;
         break;
76

77
      case PNG_CRC_ERROR_QUIT:                               /* Error/quit */
78 79 80
         png_ptr->flags &= ~PNG_FLAG_CRC_ANCILLARY_MASK;
         png_ptr->flags |= PNG_FLAG_CRC_ANCILLARY_NOWARN;
         break;
81

82
      case PNG_CRC_WARN_DISCARD:                      /* Warn/discard data */
83

84 85 86 87 88 89 90
      case PNG_CRC_DEFAULT:
      default:
         png_ptr->flags &= ~PNG_FLAG_CRC_ANCILLARY_MASK;
         break;
   }
}

G
[devel]  
Glenn Randers-Pehrson 已提交
91
#ifdef PNG_READ_BACKGROUND_SUPPORTED
92
/* Handle alpha and tRNS via a background color */
G
[devel]  
Glenn Randers-Pehrson 已提交
93 94
void PNGFAPI
png_set_background_fixed(png_structp png_ptr,
95
    png_const_color_16p background_color, int background_gamma_code,
G
[devel]  
Glenn Randers-Pehrson 已提交
96
    int need_expand, png_fixed_point background_gamma)
G
Guy Schalnat 已提交
97
{
G
[devel]  
Glenn Randers-Pehrson 已提交
98
   png_debug(1, "in png_set_background_fixed");
99

100 101
   if (png_ptr == NULL)
      return;
102

A
Andreas Dilger 已提交
103 104 105 106 107 108
   if (background_gamma_code == PNG_BACKGROUND_GAMMA_UNKNOWN)
   {
      png_warning(png_ptr, "Application must supply a known background gamma");
      return;
   }

109 110 111 112
   png_ptr->transformations |= PNG_COMPOSE | PNG_STRIP_ALPHA;
   png_ptr->transformations &= ~PNG_ENCODE_ALPHA;
   png_ptr->flags &= ~PNG_FLAG_OPTIMIZE_ALPHA;

113 114
   png_memcpy(&(png_ptr->background), background_color,
      png_sizeof(png_color_16));
G
[devel]  
Glenn Randers-Pehrson 已提交
115
   png_ptr->background_gamma = background_gamma;
G
Guy Schalnat 已提交
116
   png_ptr->background_gamma_type = (png_byte)(background_gamma_code);
117 118 119 120
   if (need_expand)
      png_ptr->transformations |= PNG_BACKGROUND_EXPAND;
   else
      png_ptr->transformations &= ~PNG_BACKGROUND_EXPAND;
G
Guy Schalnat 已提交
121
}
G
[devel]  
Glenn Randers-Pehrson 已提交
122 123 124 125

#  ifdef PNG_FLOATING_POINT_SUPPORTED
void PNGAPI
png_set_background(png_structp png_ptr,
126
    png_const_color_16p background_color, int background_gamma_code,
G
[devel]  
Glenn Randers-Pehrson 已提交
127 128 129 130 131 132 133
    int need_expand, double background_gamma)
{
   png_set_background_fixed(png_ptr, background_color, background_gamma_code,
      need_expand, png_fixed(png_ptr, background_gamma, "png_set_background"));
}
#  endif  /* FLOATING_POINT */
#endif /* READ_BACKGROUND */
G
Guy Schalnat 已提交
134

135
#ifdef PNG_READ_16_TO_8_SUPPORTED
136
/* Scale 16-bit depth files to 8-bit depth */
137
#  ifdef PNG_READ_SCALE_16_TO_8_SUPPORTED
138
void PNGAPI
139
png_set_scale_16(png_structp png_ptr)
G
Guy Schalnat 已提交
140
{
141
   png_debug(1, "in png_set_scale_16");
142

143 144
   if (png_ptr == NULL)
      return;
145

146 147 148
   png_ptr->transformations |= PNG_SCALE_16_TO_8;
#    ifdef PNG_READ_STRIP_16_TO_8_SUPPORTED
   png_ptr->transformations &= ~PNG_16_TO_8;
149
#    endif
150
}
151
#  endif
152

153
#  ifdef PNG_READ_STRIP_16_TO_8_SUPPORTED
154 155
/* Chop 16-bit depth files to 8-bit depth */
void PNGAPI
156
png_set_strip_16(png_structp png_ptr)
157
{
158
   png_debug(1, "in png_set_strip_16");
159 160 161 162

   if (png_ptr == NULL)
      return;

163 164 165
   png_ptr->transformations |= PNG_16_TO_8;
#    ifdef PNG_READ_SCALE_16_TO_8_SUPPORTED
   png_ptr->transformations &= ~PNG_SCALE_16_TO_8;
166
#    endif
G
Guy Schalnat 已提交
167
}
168
#  endif
169
#endif /* PNG_READ_16_TO_8_SUPPORTED */
G
Guy Schalnat 已提交
170

171
#ifdef PNG_READ_STRIP_ALPHA_SUPPORTED
172
void PNGAPI
A
Andreas Dilger 已提交
173 174
png_set_strip_alpha(png_structp png_ptr)
{
175
   png_debug(1, "in png_set_strip_alpha");
176

177 178
   if (png_ptr == NULL)
      return;
179

180
   png_ptr->transformations |= PNG_STRIP_ALPHA;
A
Andreas Dilger 已提交
181 182 183
}
#endif

184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245
#if defined(PNG_READ_ALPHA_MODE_SUPPORTED) || defined(PNG_READ_GAMMA_SUPPORTED)
static png_fixed_point
translate_gamma_flags(png_structp png_ptr, png_fixed_point output_gamma,
   int is_screen)
{
   /* Check for flag values.  The main reason for having the old Mac value as a
    * flag is that it is pretty near impossible to work out what the correct
    * value is from Apple documentation - a working Mac system is needed to
    * discover the value!
    */
   if (output_gamma == PNG_DEFAULT_sRGB ||
      output_gamma == PNG_FP_1 / PNG_DEFAULT_sRGB)
   {
      /* If there is no sRGB support this just sets the gamma to the standard
       * sRGB value.  (This is a side effect of using this function!)
       */
#     ifdef PNG_READ_sRGB_SUPPORTED
         png_ptr->flags |= PNG_FLAG_ASSUME_sRGB;
#     endif
      if (is_screen)
         output_gamma = PNG_GAMMA_sRGB;
      else
         output_gamma = PNG_GAMMA_sRGB_INVERSE;
   }

   else if (output_gamma == PNG_GAMMA_MAC_18 ||
      output_gamma == PNG_FP_1 / PNG_GAMMA_MAC_18)
   {
      if (is_screen)
         output_gamma = PNG_GAMMA_MAC_OLD;
      else
         output_gamma = PNG_GAMMA_MAC_INVERSE;
   }

   return output_gamma;
}

#  ifdef PNG_FLOATING_POINT_SUPPORTED
static png_fixed_point
convert_gamma_value(png_structp png_ptr, double output_gamma)
{
   /* The following silently ignores cases where fixed point (times 100,000)
    * gamma values are passed to the floating point API.  This is safe and it
    * means the fixed point constants work just fine with the floating point
    * API.  The alternative would just lead to undetected errors and spurious
    * bug reports.  Negative values fail inside the _fixed API unless they
    * correspond to the flag values.
    */
   if (output_gamma > 0 && output_gamma < 128)
      output_gamma *= PNG_FP_1;

   /* This preserves -1 and -2 exactly: */
   output_gamma = floor(output_gamma + .5);

   if (output_gamma > PNG_FP_MAX || output_gamma < PNG_FP_MIN)
      png_fixed_error(png_ptr, "gamma value");

   return (png_fixed_point)output_gamma;
}
#  endif
#endif /* READ_ALPHA_MODE || READ_GAMMA */

246 247 248 249 250 251 252 253 254 255 256 257 258
#ifdef PNG_READ_ALPHA_MODE_SUPPORTED
void PNGFAPI
png_set_alpha_mode_fixed(png_structp png_ptr, int mode,
   png_fixed_point output_gamma)
{
   int compose = 0;
   png_fixed_point file_gamma;

   png_debug(1, "in png_set_alpha_mode");

   if (png_ptr == NULL)
      return;

259
   output_gamma = translate_gamma_flags(png_ptr, output_gamma, 1/*screen*/);
260

261
   /* Validate the value to ensure it is in a reasonable range. The value
262 263 264 265 266
    * is expected to be 1 or greater, but this range test allows for some
    * viewing correction values.  The intent is to weed out users of this API
    * who use the inverse of the gamma value accidentally!  Since some of these
    * values are reasonable this may have to be changed.
    */
267
   if (output_gamma < 70000 || output_gamma > 300000)
268
      png_error(png_ptr, "output gamma out of expected range");
269

270
   /* The default file gamma is the inverse of the output gamma; the output
271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333
    * gamma may be changed below so get the file value first:
    */
   file_gamma = png_reciprocal(output_gamma);

   /* There are really 8 possibilities here, composed of any combination
    * of:
    *
    *    premultiply the color channels
    *    do not encode non-opaque pixels
    *    encode the alpha as well as the color channels
    *
    * The differences disappear if the input/output ('screen') gamma is 1.0,
    * because then the encoding is a no-op and there is only the choice of
    * premultiplying the color channels or not.
    *
    * png_set_alpha_mode and png_set_background interact because both use
    * png_compose to do the work.  Calling both is only useful when
    * png_set_alpha_mode is used to set the default mode - PNG_ALPHA_PNG - along
    * with a default gamma value.  Otherwise PNG_COMPOSE must not be set.
    */
   switch (mode)
   {
      case PNG_ALPHA_PNG:        /* default: png standard */
         /* No compose, but it may be set by png_set_background! */
         png_ptr->transformations &= ~PNG_ENCODE_ALPHA;
         png_ptr->flags &= ~PNG_FLAG_OPTIMIZE_ALPHA;
         break;

      case PNG_ALPHA_ASSOCIATED: /* color channels premultiplied */
         compose = 1;
         png_ptr->transformations &= ~PNG_ENCODE_ALPHA;
         png_ptr->flags &= ~PNG_FLAG_OPTIMIZE_ALPHA;
         /* The output is linear: */
         output_gamma = PNG_FP_1;
         break;

      case PNG_ALPHA_OPTIMIZED:  /* associated, non-opaque pixels linear */
         compose = 1;
         png_ptr->transformations &= ~PNG_ENCODE_ALPHA;
         png_ptr->flags |= PNG_FLAG_OPTIMIZE_ALPHA;
         /* output_gamma records the encoding of opaque pixels! */
         break;

      case PNG_ALPHA_BROKEN:     /* associated, non-linear, alpha encoded */
         compose = 1;
         png_ptr->transformations |= PNG_ENCODE_ALPHA;
         png_ptr->flags &= ~PNG_FLAG_OPTIMIZE_ALPHA;
         break;

      default:
         png_error(png_ptr, "invalid alpha mode");
   }

   /* Only set the default gamma if the file gamma has not been set (this has
    * the side effect that the gamma in a second call to png_set_alpha_mode will
    * be ignored.)
    */
   if (png_ptr->gamma == 0)
      png_ptr->gamma = file_gamma;

   /* But always set the output gamma: */
   png_ptr->screen_gamma = output_gamma;

334
   /* Finally, if pre-multiplying, set the background fields to achieve the
335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359
    * desired result.
    */
   if (compose)
   {
      /* And obtain alpha pre-multiplication by composing on black: */
      png_memset(&png_ptr->background, 0, sizeof png_ptr->background);
      png_ptr->background_gamma = png_ptr->gamma; /* just in case */
      png_ptr->background_gamma_type = PNG_BACKGROUND_GAMMA_FILE;
      png_ptr->transformations &= ~PNG_BACKGROUND_EXPAND;

      if (png_ptr->transformations & PNG_COMPOSE)
         png_error(png_ptr,
            "conflicting calls to set alpha mode and background");

      png_ptr->transformations |= PNG_COMPOSE;
   }

   /* New API, make sure apps call the correct initializers: */
   png_ptr->flags |= PNG_FLAG_DETECT_UNINITIALIZED;
}

#  ifdef PNG_FLOATING_POINT_SUPPORTED
void PNGAPI
png_set_alpha_mode(png_structp png_ptr, int mode, double output_gamma)
{
360 361
   png_set_alpha_mode_fixed(png_ptr, mode, convert_gamma_value(png_ptr,
      output_gamma));
362 363 364 365
}
#  endif
#endif

366
#ifdef PNG_READ_QUANTIZE_SUPPORTED
367
/* Dither file to 8-bit.  Supply a palette, the current number
368 369 370
 * of elements in the palette, the maximum number of elements
 * allowed, and a histogram if possible.  If the current number
 * of colors is greater then the maximum number, the palette will be
371 372
 * modified to fit in the maximum number.  "full_quantize" indicates
 * whether we need a quantizing cube set up for RGB images, or if we
373 374
 * simply are reducing the number of colors in a paletted image.
 */
G
Guy Schalnat 已提交
375 376

typedef struct png_dsort_struct
G
Guy Schalnat 已提交
377
{
G
Guy Schalnat 已提交
378
   struct png_dsort_struct FAR * next;
G
Guy Schalnat 已提交
379 380
   png_byte left;
   png_byte right;
G
Guy Schalnat 已提交
381 382 383
} png_dsort;
typedef png_dsort FAR *       png_dsortp;
typedef png_dsort FAR * FAR * png_dsortpp;
G
Guy Schalnat 已提交
384

385
void PNGAPI
386
png_set_quantize(png_structp png_ptr, png_colorp palette,
387
    int num_palette, int maximum_colors, png_const_uint_16p histogram,
388
    int full_quantize)
G
Guy Schalnat 已提交
389
{
390
   png_debug(1, "in png_set_quantize");
391

392 393
   if (png_ptr == NULL)
      return;
394

395
   png_ptr->transformations |= PNG_QUANTIZE;
G
Guy Schalnat 已提交
396

397
   if (!full_quantize)
G
Guy Schalnat 已提交
398 399 400
   {
      int i;

401
      png_ptr->quantize_index = (png_bytep)png_malloc(png_ptr,
402
          (png_uint_32)(num_palette * png_sizeof(png_byte)));
G
Guy Schalnat 已提交
403
      for (i = 0; i < num_palette; i++)
404
         png_ptr->quantize_index[i] = (png_byte)i;
G
Guy Schalnat 已提交
405 406 407 408
   }

   if (num_palette > maximum_colors)
   {
A
Andreas Dilger 已提交
409
      if (histogram != NULL)
G
Guy Schalnat 已提交
410
      {
A
Andreas Dilger 已提交
411
         /* This is easy enough, just throw out the least used colors.
412 413
          * Perhaps not the best solution, but good enough.
          */
G
Guy Schalnat 已提交
414 415 416

         int i;

417
         /* Initialize an array to sort colors */
418
         png_ptr->quantize_sort = (png_bytep)png_malloc(png_ptr,
419
             (png_uint_32)(num_palette * png_sizeof(png_byte)));
G
Guy Schalnat 已提交
420

421
         /* Initialize the quantize_sort array */
G
Guy Schalnat 已提交
422
         for (i = 0; i < num_palette; i++)
423
            png_ptr->quantize_sort[i] = (png_byte)i;
G
Guy Schalnat 已提交
424

A
Andreas Dilger 已提交
425
         /* Find the least used palette entries by starting a
426 427 428 429 430
          * bubble sort, and running it until we have sorted
          * out enough colors.  Note that we don't care about
          * sorting all the colors, just finding which are
          * least used.
          */
G
Guy Schalnat 已提交
431 432 433

         for (i = num_palette - 1; i >= maximum_colors; i--)
         {
434
            int done; /* To stop early if the list is pre-sorted */
G
Guy Schalnat 已提交
435
            int j;
G
Guy Schalnat 已提交
436 437 438 439

            done = 1;
            for (j = 0; j < i; j++)
            {
440 441
               if (histogram[png_ptr->quantize_sort[j]]
                   < histogram[png_ptr->quantize_sort[j + 1]])
G
Guy Schalnat 已提交
442 443 444
               {
                  png_byte t;

445 446 447
                  t = png_ptr->quantize_sort[j];
                  png_ptr->quantize_sort[j] = png_ptr->quantize_sort[j + 1];
                  png_ptr->quantize_sort[j + 1] = t;
G
Guy Schalnat 已提交
448 449 450
                  done = 0;
               }
            }
451

G
Guy Schalnat 已提交
452 453 454 455
            if (done)
               break;
         }

456
         /* Swap the palette around, and set up a table, if necessary */
457
         if (full_quantize)
G
Guy Schalnat 已提交
458
         {
459
            int j = num_palette;
G
Guy Schalnat 已提交
460

461 462 463
            /* Put all the useful colors within the max, but don't
             * move the others.
             */
464
            for (i = 0; i < maximum_colors; i++)
G
Guy Schalnat 已提交
465
            {
466
               if ((int)png_ptr->quantize_sort[i] >= maximum_colors)
G
Guy Schalnat 已提交
467 468 469
               {
                  do
                     j--;
470
                  while ((int)png_ptr->quantize_sort[j] >= maximum_colors);
471

G
Guy Schalnat 已提交
472 473
                  palette[i] = palette[j];
               }
G
Guy Schalnat 已提交
474
            }
G
Guy Schalnat 已提交
475 476 477
         }
         else
         {
478
            int j = num_palette;
G
Guy Schalnat 已提交
479

480 481 482
            /* Move all the used colors inside the max limit, and
             * develop a translation table.
             */
483
            for (i = 0; i < maximum_colors; i++)
G
Guy Schalnat 已提交
484
            {
485
               /* Only move the colors we need to */
486
               if ((int)png_ptr->quantize_sort[i] >= maximum_colors)
G
Guy Schalnat 已提交
487 488 489 490 491
               {
                  png_color tmp_color;

                  do
                     j--;
492
                  while ((int)png_ptr->quantize_sort[j] >= maximum_colors);
G
Guy Schalnat 已提交
493 494 495 496

                  tmp_color = palette[j];
                  palette[j] = palette[i];
                  palette[i] = tmp_color;
497
                  /* Indicate where the color went */
498 499
                  png_ptr->quantize_index[j] = (png_byte)i;
                  png_ptr->quantize_index[i] = (png_byte)j;
G
Guy Schalnat 已提交
500 501
               }
            }
A
Andreas Dilger 已提交
502

503
            /* Find closest color for those colors we are not using */
G
Guy Schalnat 已提交
504 505
            for (i = 0; i < num_palette; i++)
            {
506
               if ((int)png_ptr->quantize_index[i] >= maximum_colors)
G
Guy Schalnat 已提交
507
               {
A
Andreas Dilger 已提交
508
                  int min_d, k, min_k, d_index;
G
Guy Schalnat 已提交
509

510
                  /* Find the closest color to one we threw out */
511
                  d_index = png_ptr->quantize_index[i];
A
Andreas Dilger 已提交
512 513
                  min_d = PNG_COLOR_DIST(palette[d_index], palette[0]);
                  for (k = 1, min_k = 0; k < maximum_colors; k++)
G
Guy Schalnat 已提交
514 515 516
                  {
                     int d;

A
Andreas Dilger 已提交
517
                     d = PNG_COLOR_DIST(palette[d_index], palette[k]);
G
Guy Schalnat 已提交
518 519 520 521

                     if (d < min_d)
                     {
                        min_d = d;
A
Andreas Dilger 已提交
522
                        min_k = k;
G
Guy Schalnat 已提交
523 524
                     }
                  }
525
                  /* Point to closest color */
526
                  png_ptr->quantize_index[i] = (png_byte)min_k;
G
Guy Schalnat 已提交
527 528 529
               }
            }
         }
530 531
         png_free(png_ptr, png_ptr->quantize_sort);
         png_ptr->quantize_sort = NULL;
G
Guy Schalnat 已提交
532 533 534
      }
      else
      {
A
Andreas Dilger 已提交
535
         /* This is much harder to do simply (and quickly).  Perhaps
536 537 538 539 540 541 542
          * we need to go through a median cut routine, but those
          * don't always behave themselves with only a few colors
          * as input.  So we will just find the closest two colors,
          * and throw out one of them (chosen somewhat randomly).
          * [We don't understand this at all, so if someone wants to
          *  work on improving it, be our guest - AED, GRP]
          */
G
Guy Schalnat 已提交
543 544 545
         int i;
         int max_d;
         int num_new_palette;
546
         png_dsortp t;
G
Guy Schalnat 已提交
547
         png_dsortpp hash;
548

549
         t = NULL;
G
Guy Schalnat 已提交
550

551
         /* Initialize palette index arrays */
552
         png_ptr->index_to_palette = (png_bytep)png_malloc(png_ptr,
553
             (png_uint_32)(num_palette * png_sizeof(png_byte)));
554
         png_ptr->palette_to_index = (png_bytep)png_malloc(png_ptr,
555
             (png_uint_32)(num_palette * png_sizeof(png_byte)));
G
Guy Schalnat 已提交
556

557
         /* Initialize the sort array */
G
Guy Schalnat 已提交
558 559
         for (i = 0; i < num_palette; i++)
         {
560 561
            png_ptr->index_to_palette[i] = (png_byte)i;
            png_ptr->palette_to_index[i] = (png_byte)i;
G
Guy Schalnat 已提交
562 563
         }

564
         hash = (png_dsortpp)png_calloc(png_ptr, (png_uint_32)(769 *
565
             png_sizeof(png_dsortp)));
G
Guy Schalnat 已提交
566 567 568

         num_new_palette = num_palette;

569 570 571 572 573 574 575 576
         /* Initial wild guess at how far apart the farthest pixel
          * pair we will be eliminating will be.  Larger
          * numbers mean more areas will be allocated, Smaller
          * numbers run the risk of not saving enough data, and
          * having to do this all over again.
          *
          * I have not done extensive checking on this number.
          */
G
Guy Schalnat 已提交
577 578
         max_d = 96;

G
Guy Schalnat 已提交
579
         while (num_new_palette > maximum_colors)
G
Guy Schalnat 已提交
580 581 582 583 584 585 586 587 588
         {
            for (i = 0; i < num_new_palette - 1; i++)
            {
               int j;

               for (j = i + 1; j < num_new_palette; j++)
               {
                  int d;

G
Guy Schalnat 已提交
589
                  d = PNG_COLOR_DIST(palette[i], palette[j]);
G
Guy Schalnat 已提交
590 591 592 593

                  if (d <= max_d)
                  {

594
                     t = (png_dsortp)png_malloc_warn(png_ptr,
595
                         (png_uint_32)(png_sizeof(png_dsort)));
596

597 598
                     if (t == NULL)
                         break;
599

G
Guy Schalnat 已提交
600
                     t->next = hash[d];
G
Guy Schalnat 已提交
601 602
                     t->left = (png_byte)i;
                     t->right = (png_byte)j;
G
Guy Schalnat 已提交
603 604 605
                     hash[d] = t;
                  }
               }
606 607
               if (t == NULL)
                  break;
G
Guy Schalnat 已提交
608 609
            }

610
            if (t != NULL)
G
Guy Schalnat 已提交
611 612
            for (i = 0; i <= max_d; i++)
            {
A
Andreas Dilger 已提交
613
               if (hash[i] != NULL)
G
Guy Schalnat 已提交
614
               {
G
Guy Schalnat 已提交
615
                  png_dsortp p;
G
Guy Schalnat 已提交
616 617 618

                  for (p = hash[i]; p; p = p->next)
                  {
619
                     if ((int)png_ptr->index_to_palette[p->left]
620 621 622
                         < num_new_palette &&
                         (int)png_ptr->index_to_palette[p->right]
                         < num_new_palette)
G
Guy Schalnat 已提交
623 624 625
                     {
                        int j, next_j;

626
                        if (num_new_palette & 0x01)
G
Guy Schalnat 已提交
627 628 629 630 631 632 633 634 635 636 637
                        {
                           j = p->left;
                           next_j = p->right;
                        }
                        else
                        {
                           j = p->right;
                           next_j = p->left;
                        }

                        num_new_palette--;
638
                        palette[png_ptr->index_to_palette[j]]
639
                            = palette[num_new_palette];
640
                        if (!full_quantize)
G
Guy Schalnat 已提交
641 642 643 644 645
                        {
                           int k;

                           for (k = 0; k < num_palette; k++)
                           {
646
                              if (png_ptr->quantize_index[k] ==
647
                                  png_ptr->index_to_palette[j])
648
                                 png_ptr->quantize_index[k] =
649
                                     png_ptr->index_to_palette[next_j];
650

651
                              if ((int)png_ptr->quantize_index[k] ==
652
                                  num_new_palette)
653
                                 png_ptr->quantize_index[k] =
654
                                     png_ptr->index_to_palette[j];
G
Guy Schalnat 已提交
655 656 657
                           }
                        }

658
                        png_ptr->index_to_palette[png_ptr->palette_to_index
659
                            [num_new_palette]] = png_ptr->index_to_palette[j];
660

661
                        png_ptr->palette_to_index[png_ptr->index_to_palette[j]]
662
                            = png_ptr->palette_to_index[num_new_palette];
G
Guy Schalnat 已提交
663

664 665
                        png_ptr->index_to_palette[j] =
                            (png_byte)num_new_palette;
666

667 668
                        png_ptr->palette_to_index[num_new_palette] =
                            (png_byte)j;
G
Guy Schalnat 已提交
669 670 671 672 673 674 675 676 677 678 679
                     }
                     if (num_new_palette <= maximum_colors)
                        break;
                  }
                  if (num_new_palette <= maximum_colors)
                     break;
               }
            }

            for (i = 0; i < 769; i++)
            {
A
Andreas Dilger 已提交
680
               if (hash[i] != NULL)
G
Guy Schalnat 已提交
681
               {
682
                  png_dsortp p = hash[i];
G
Guy Schalnat 已提交
683 684 685
                  while (p)
                  {
                     t = p->next;
A
Andreas Dilger 已提交
686
                     png_free(png_ptr, p);
G
Guy Schalnat 已提交
687 688 689 690 691 692 693
                     p = t;
                  }
               }
               hash[i] = 0;
            }
            max_d += 96;
         }
A
Andreas Dilger 已提交
694
         png_free(png_ptr, hash);
695 696
         png_free(png_ptr, png_ptr->palette_to_index);
         png_free(png_ptr, png_ptr->index_to_palette);
697 698
         png_ptr->palette_to_index = NULL;
         png_ptr->index_to_palette = NULL;
G
Guy Schalnat 已提交
699 700 701
      }
      num_palette = maximum_colors;
   }
A
Andreas Dilger 已提交
702
   if (png_ptr->palette == NULL)
G
Guy Schalnat 已提交
703
   {
G
Guy Schalnat 已提交
704 705
      png_ptr->palette = palette;
   }
G
Guy Schalnat 已提交
706
   png_ptr->num_palette = (png_uint_16)num_palette;
G
Guy Schalnat 已提交
707

708
   if (full_quantize)
G
Guy Schalnat 已提交
709 710
   {
      int i;
G
Guy Schalnat 已提交
711
      png_bytep distance;
712 713 714 715 716
      int total_bits = PNG_QUANTIZE_RED_BITS + PNG_QUANTIZE_GREEN_BITS +
          PNG_QUANTIZE_BLUE_BITS;
      int num_red = (1 << PNG_QUANTIZE_RED_BITS);
      int num_green = (1 << PNG_QUANTIZE_GREEN_BITS);
      int num_blue = (1 << PNG_QUANTIZE_BLUE_BITS);
717
      png_size_t num_entries = ((png_size_t)1 << total_bits);
G
Guy Schalnat 已提交
718

719
      png_ptr->palette_lookup = (png_bytep)png_calloc(png_ptr,
720
          (png_uint_32)(num_entries * png_sizeof(png_byte)));
721

722
      distance = (png_bytep)png_malloc(png_ptr, (png_uint_32)(num_entries *
723
          png_sizeof(png_byte)));
724

725
      png_memset(distance, 0xff, num_entries * png_sizeof(png_byte));
726 727 728

      for (i = 0; i < num_palette; i++)
      {
729
         int ir, ig, ib;
730 731 732
         int r = (palette[i].red >> (8 - PNG_QUANTIZE_RED_BITS));
         int g = (palette[i].green >> (8 - PNG_QUANTIZE_GREEN_BITS));
         int b = (palette[i].blue >> (8 - PNG_QUANTIZE_BLUE_BITS));
G
Guy Schalnat 已提交
733 734 735

         for (ir = 0; ir < num_red; ir++)
         {
736 737
            /* int dr = abs(ir - r); */
            int dr = ((ir > r) ? ir - r : r - ir);
738 739
            int index_r = (ir << (PNG_QUANTIZE_BLUE_BITS +
                PNG_QUANTIZE_GREEN_BITS));
G
Guy Schalnat 已提交
740 741

            for (ig = 0; ig < num_green; ig++)
G
Guy Schalnat 已提交
742
            {
743 744
               /* int dg = abs(ig - g); */
               int dg = ((ig > g) ? ig - g : g - ig);
745 746
               int dt = dr + dg;
               int dm = ((dr > dg) ? dr : dg);
747
               int index_g = index_r | (ig << PNG_QUANTIZE_BLUE_BITS);
G
Guy Schalnat 已提交
748 749

               for (ib = 0; ib < num_blue; ib++)
G
Guy Schalnat 已提交
750
               {
751
                  int d_index = index_g | ib;
752 753
                  /* int db = abs(ib - b); */
                  int db = ((ib > b) ? ib - b : b - ib);
754 755
                  int dmax = ((dm > db) ? dm : db);
                  int d = dmax + dt + db;
G
Guy Schalnat 已提交
756

757
                  if (d < (int)distance[d_index])
G
Guy Schalnat 已提交
758
                  {
A
Andreas Dilger 已提交
759 760
                     distance[d_index] = (png_byte)d;
                     png_ptr->palette_lookup[d_index] = (png_byte)i;
G
Guy Schalnat 已提交
761 762
                  }
               }
763 764 765
            }
         }
      }
G
Guy Schalnat 已提交
766

A
Andreas Dilger 已提交
767
      png_free(png_ptr, distance);
G
Guy Schalnat 已提交
768 769
   }
}
770
#endif /* PNG_READ_QUANTIZE_SUPPORTED */
G
Guy Schalnat 已提交
771

G
[devel]  
Glenn Randers-Pehrson 已提交
772 773 774 775 776 777
#ifdef PNG_READ_GAMMA_SUPPORTED
void PNGFAPI
png_set_gamma_fixed(png_structp png_ptr, png_fixed_point scrn_gamma,
   png_fixed_point file_gamma)
{
   png_debug(1, "in png_set_gamma_fixed");
778

779 780
   if (png_ptr == NULL)
      return;
781

782
   /* New in libpng-1.5.4 - reserve particular negative values as flags. */
783 784 785
   scrn_gamma = translate_gamma_flags(png_ptr, scrn_gamma, 1/*screen*/);
   file_gamma = translate_gamma_flags(png_ptr, file_gamma, 0/*file*/);

786
#if PNG_LIBPNG_VER >= 10600
787
   /* Checking the gamma values for being >0 was added in 1.5.4 along with the
788 789 790
    * premultiplied alpha support; this actually hides an undocumented feature
    * of the previous implementation which allowed gamma processing to be
    * disabled in background handling.  There is no evidence (so far) that this
791
    * was being used; however, png_set_background itself accepted and must still
792 793 794
    * accept '0' for the gamma value it takes, because it isn't always used.
    *
    * Since this is an API change (albeit a very minor one that removes an
795
    * undocumented API feature) it will not be made until libpng-1.6.0.
796 797
    */
   if (file_gamma <= 0)
798
      png_error(png_ptr, "invalid file gamma in png_set_gamma");
799 800

   if (scrn_gamma <= 0)
801
      png_error(png_ptr, "invalid screen gamma in png_set_gamma");
802 803 804 805 806 807
#endif

   /* Set the gamma values unconditionally - this overrides the value in the PNG
    * file if a gAMA chunk was present.  png_set_alpha_mode provides a
    * different, easier, way to default the file gamma.
    */
G
[devel]  
Glenn Randers-Pehrson 已提交
808 809
   png_ptr->gamma = file_gamma;
   png_ptr->screen_gamma = scrn_gamma;
G
Guy Schalnat 已提交
810
}
G
[devel]  
Glenn Randers-Pehrson 已提交
811 812 813 814 815

#  ifdef PNG_FLOATING_POINT_SUPPORTED
void PNGAPI
png_set_gamma(png_structp png_ptr, double scrn_gamma, double file_gamma)
{
816 817
   png_set_gamma_fixed(png_ptr, convert_gamma_value(png_ptr, scrn_gamma),
      convert_gamma_value(png_ptr, file_gamma));
G
[devel]  
Glenn Randers-Pehrson 已提交
818 819 820
}
#  endif /* FLOATING_POINT_SUPPORTED */
#endif /* READ_GAMMA */
G
Guy Schalnat 已提交
821

822
#ifdef PNG_READ_EXPAND_SUPPORTED
823
/* Expand paletted images to RGB, expand grayscale images of
824
 * less than 8-bit depth to 8-bit depth, and expand tRNS chunks
825 826
 * to alpha channels.
 */
827
void PNGAPI
G
Guy Schalnat 已提交
828
png_set_expand(png_structp png_ptr)
G
Guy Schalnat 已提交
829
{
830
   png_debug(1, "in png_set_expand");
831

832 833
   if (png_ptr == NULL)
      return;
834

835
   png_ptr->transformations |= (PNG_EXPAND | PNG_EXPAND_tRNS);
836
   png_ptr->flags &= ~PNG_FLAG_ROW_INIT;
G
Guy Schalnat 已提交
837
}
838 839 840 841 842 843 844 845 846 847 848 849 850

/* GRR 19990627:  the following three functions currently are identical
 *  to png_set_expand().  However, it is entirely reasonable that someone
 *  might wish to expand an indexed image to RGB but *not* expand a single,
 *  fully transparent palette entry to a full alpha channel--perhaps instead
 *  convert tRNS to the grayscale/RGB format (16-bit RGB value), or replace
 *  the transparent color with a particular RGB value, or drop tRNS entirely.
 *  IOW, a future version of the library may make the transformations flag
 *  a bit more fine-grained, with separate bits for each of these three
 *  functions.
 *
 *  More to the point, these functions make it obvious what libpng will be
 *  doing, whereas "expand" can (and does) mean any number of things.
851
 *
852 853 854
 *  GRP 20060307: In libpng-1.2.9, png_set_gray_1_2_4_to_8() was modified
 *  to expand only the sample depth but not to expand the tRNS to alpha
 *  and its name was changed to png_set_expand_gray_1_2_4_to_8().
855 856 857
 */

/* Expand paletted images to RGB. */
858
void PNGAPI
859 860
png_set_palette_to_rgb(png_structp png_ptr)
{
861
   png_debug(1, "in png_set_palette_to_rgb");
862

863 864
   if (png_ptr == NULL)
      return;
865

866
   png_ptr->transformations |= (PNG_EXPAND | PNG_EXPAND_tRNS);
867
   png_ptr->flags &= ~PNG_FLAG_ROW_INIT;
868 869 870 871 872 873
}

/* Expand grayscale images of less than 8-bit depth to 8 bits. */
void PNGAPI
png_set_expand_gray_1_2_4_to_8(png_structp png_ptr)
{
874
   png_debug(1, "in png_set_expand_gray_1_2_4_to_8");
875

876 877
   if (png_ptr == NULL)
      return;
878

879 880
   png_ptr->transformations |= PNG_EXPAND;
   png_ptr->flags &= ~PNG_FLAG_ROW_INIT;
881 882
}

883 884


885
/* Expand tRNS chunks to alpha channels. */
886
void PNGAPI
887 888
png_set_tRNS_to_alpha(png_structp png_ptr)
{
889
   png_debug(1, "in png_set_tRNS_to_alpha");
890

891
   png_ptr->transformations |= (PNG_EXPAND | PNG_EXPAND_tRNS);
892
   png_ptr->flags &= ~PNG_FLAG_ROW_INIT;
893 894
}
#endif /* defined(PNG_READ_EXPAND_SUPPORTED) */
G
Guy Schalnat 已提交
895

896
#ifdef PNG_READ_EXPAND_16_SUPPORTED
897
/* Expand to 16-bit channels, expand the tRNS chunk too (because otherwise
898 899 900 901 902 903 904 905 906 907 908 909
 * it may not work correctly.)
 */
void PNGAPI
png_set_expand_16(png_structp png_ptr)
{
   png_debug(1, "in png_set_expand_16");

   if (png_ptr == NULL)
      return;

   png_ptr->transformations |= (PNG_EXPAND_16 | PNG_EXPAND | PNG_EXPAND_tRNS);
   png_ptr->flags &= ~PNG_FLAG_ROW_INIT;
910 911 912

   /* New API, make sure apps call the correct initializers: */
   png_ptr->flags |= PNG_FLAG_DETECT_UNINITIALIZED;
913 914 915
}
#endif

916
#ifdef PNG_READ_GRAY_TO_RGB_SUPPORTED
917
void PNGAPI
G
Guy Schalnat 已提交
918
png_set_gray_to_rgb(png_structp png_ptr)
G
Guy Schalnat 已提交
919
{
920
   png_debug(1, "in png_set_gray_to_rgb");
921

922 923 924 925 926 927 928
   if (png_ptr != NULL)
   {
      /* Because rgb must be 8 bits or more: */
      png_set_expand_gray_1_2_4_to_8(png_ptr);
      png_ptr->transformations |= PNG_GRAY_TO_RGB;
      png_ptr->flags &= ~PNG_FLAG_ROW_INIT;
   }
G
Guy Schalnat 已提交
929
}
G
Guy Schalnat 已提交
930
#endif
G
Guy Schalnat 已提交
931

932
#ifdef PNG_READ_RGB_TO_GRAY_SUPPORTED
G
[devel]  
Glenn Randers-Pehrson 已提交
933
void PNGFAPI
934
png_set_rgb_to_gray_fixed(png_structp png_ptr, int error_action,
935
    png_fixed_point red, png_fixed_point green)
A
Andreas Dilger 已提交
936
{
937
   png_debug(1, "in png_set_rgb_to_gray");
938

939 940
   if (png_ptr == NULL)
      return;
941

942 943
   switch(error_action)
   {
944 945 946
      case 1:
         png_ptr->transformations |= PNG_RGB_TO_GRAY;
         break;
947

948 949 950
      case 2:
         png_ptr->transformations |= PNG_RGB_TO_GRAY_WARN;
         break;
951

952 953
      case 3:
         png_ptr->transformations |= PNG_RGB_TO_GRAY_ERR;
954 955 956
         break;

      default:
957 958
         png_error(png_ptr, "invalid error action to rgb_to_gray");
         break;
959 960
   }
   if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
961
#ifdef PNG_READ_EXPAND_SUPPORTED
962 963 964
      png_ptr->transformations |= PNG_EXPAND;
#else
   {
965 966
      png_warning(png_ptr,
        "Cannot do RGB_TO_GRAY without EXPAND_SUPPORTED");
967

968 969 970 971
      png_ptr->transformations &= ~PNG_RGB_TO_GRAY;
   }
#endif
   {
972
      if (red >= 0 && green >= 0 && red + green <= PNG_FP_1)
973
      {
974
         png_uint_16 red_int, green_int;
975

976 977
         red_int = (png_uint_16)(((png_uint_32)red*32768L)/100000L);
         green_int = (png_uint_16)(((png_uint_32)green*32768L)/100000L);
978 979 980 981 982

         png_ptr->rgb_to_gray_red_coeff   = red_int;
         png_ptr->rgb_to_gray_green_coeff = green_int;
         png_ptr->rgb_to_gray_blue_coeff  =
          (png_uint_16)(32768 - red_int - green_int);
983
      }
984

985
      else
986
      {
987 988 989
         if (red >= 0 && green >= 0)
            png_warning(png_ptr,
               "ignoring out of range rgb_to_gray coefficients");
990

991 992 993 994 995 996 997 998 999 1000 1001 1002 1003
         /* Use the defaults, from the cHRM chunk if set, else the built in Rec
          * 709 values (which correspond to sRGB, so we don't have to worry
          * about the sRGB chunk!)
          */
         if (png_ptr->rgb_to_gray_red_coeff == 0 &&
            png_ptr->rgb_to_gray_green_coeff == 0 &&
            png_ptr->rgb_to_gray_blue_coeff == 0)
         {
            png_ptr->rgb_to_gray_red_coeff   = 6968;  /* .212671 * 32768 + .5 */
            png_ptr->rgb_to_gray_green_coeff = 23434; /* .715160 * 32768 + .5 */
            png_ptr->rgb_to_gray_blue_coeff  = 2366;
         }
      }
1004
   }
A
Andreas Dilger 已提交
1005
}
1006 1007 1008 1009 1010 1011 1012 1013

#ifdef PNG_FLOATING_POINT_SUPPORTED
/* Convert a RGB image to a grayscale of the same width.  This allows us,
 * for example, to convert a 24 bpp RGB image into an 8 bpp grayscale image.
 */

void PNGAPI
png_set_rgb_to_gray(png_structp png_ptr, int error_action, double red,
G
[devel]  
Glenn Randers-Pehrson 已提交
1014
   double green)
1015 1016 1017 1018
{
   if (png_ptr == NULL)
      return;

G
[devel]  
Glenn Randers-Pehrson 已提交
1019 1020 1021
   png_set_rgb_to_gray_fixed(png_ptr, error_action,
      png_fixed(png_ptr, red, "rgb to gray red coefficient"),
      png_fixed(png_ptr, green, "rgb to gray green coefficient"));
1022 1023 1024
}
#endif /* FLOATING POINT */

A
Andreas Dilger 已提交
1025 1026
#endif

1027
#if defined(PNG_READ_USER_TRANSFORM_SUPPORTED) || \
1028
    defined(PNG_WRITE_USER_TRANSFORM_SUPPORTED)
1029
void PNGAPI
1030
png_set_read_user_transform_fn(png_structp png_ptr, png_user_transform_ptr
1031
    read_user_transform_fn)
1032
{
1033
   png_debug(1, "in png_set_read_user_transform_fn");
1034

1035 1036
   if (png_ptr == NULL)
      return;
1037

1038
#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
1039 1040
   png_ptr->transformations |= PNG_USER_TRANSFORM;
   png_ptr->read_user_transform_fn = read_user_transform_fn;
1041
#endif
1042 1043 1044
}
#endif

1045
#ifdef PNG_READ_TRANSFORMS_SUPPORTED
1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066
#ifdef PNG_READ_GAMMA_SUPPORTED
/* In the case of gamma transformations only do transformations on images where
 * the [file] gamma and screen_gamma are not close reciprocals, otherwise it
 * slows things down slightly, and also needlessly introduces small errors.
 */
static int /* PRIVATE */
png_gamma_threshold(png_fixed_point screen_gamma, png_fixed_point file_gamma)
{
   /* PNG_GAMMA_THRESHOLD is the threshold for performing gamma
    * correction as a difference of the overall transform from 1.0
    *
    * We want to compare the threshold with s*f - 1, if we get
    * overflow here it is because of wacky gamma values so we
    * turn on processing anyway.
    */
   png_fixed_point gtest;
   return !png_muldiv(&gtest, screen_gamma, file_gamma, PNG_FP_1) ||
       png_gamma_significant(gtest);
}
#endif

A
Andreas Dilger 已提交
1067
/* Initialize everything needed for the read.  This includes modifying
1068 1069
 * the palette.
 */
1070 1071 1072 1073 1074 1075 1076 1077

/*For the moment 'png_init_palette_transformations' and
 * 'png_init_rgb_transformations' only do some flag canceling optimizations.
 * The intent is that these two routines should have palette or rgb operations
 * extracted from 'png_init_read_transformations'.
 */
static void /* PRIVATE */
png_init_palette_transformations(png_structp png_ptr)
G
Guy Schalnat 已提交
1078
{
1079 1080 1081 1082 1083 1084 1085 1086 1087 1088
   /* Called to handle the (input) palette case.  In png_do_read_transformations
    * the first step is to expand the palette if requested, so this code must
    * take care to only make changes that are invariant with respect to the
    * palette expansion, or only do them if there is no expansion.
    *
    * STRIP_ALPHA has already been handled in the caller (by setting num_trans
    * to 0.)
    */
   int input_has_alpha = 0;
   int input_has_transparency = 0;
1089

1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116
   if (png_ptr->num_trans > 0)
   {
      int i;

      /* Ignore if all the entries are opaque (unlikely!) */
      for (i=0; i<png_ptr->num_trans; ++i)
         if (png_ptr->trans_alpha[i] == 255)
            continue;
         else if (png_ptr->trans_alpha[i] == 0)
            input_has_transparency = 1;
         else
            input_has_alpha = 1;
   }

   /* If no alpha we can optimize. */
   if (!input_has_alpha)
   {
      /* Any alpha means background and associative alpha processing is
       * required, however if the alpha is 0 or 1 throughout OPTIIMIZE_ALPHA
       * and ENCODE_ALPHA are irrelevant.
       */
      png_ptr->transformations &= ~PNG_ENCODE_ALPHA;
      png_ptr->flags &= ~PNG_FLAG_OPTIMIZE_ALPHA;

      if (!input_has_transparency)
         png_ptr->transformations &= ~(PNG_COMPOSE | PNG_BACKGROUND_EXPAND);
   }
G
Guy Schalnat 已提交
1117

G
Guy Schalnat 已提交
1118
#if defined(PNG_READ_EXPAND_SUPPORTED) && defined(PNG_READ_BACKGROUND_SUPPORTED)
1119 1120 1121 1122
   /* png_set_background handling - deals with the complexity of whether the
    * background color is in the file format or the screen format in the case
    * where an 'expand' will happen.
    */
1123

1124 1125
   /* The following code cannot be entered in the alpha pre-multiplication case
    * because PNG_BACKGROUND_EXPAND is cancelled below.
1126 1127
    */
   if ((png_ptr->transformations & PNG_BACKGROUND_EXPAND) &&
1128
       (png_ptr->transformations & PNG_EXPAND))
1129
   {
1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161
      {
         png_ptr->background.red   =
             png_ptr->palette[png_ptr->background.index].red;
         png_ptr->background.green =
             png_ptr->palette[png_ptr->background.index].green;
         png_ptr->background.blue  =
             png_ptr->palette[png_ptr->background.index].blue;

#ifdef PNG_READ_INVERT_ALPHA_SUPPORTED
        if (png_ptr->transformations & PNG_INVERT_ALPHA)
        {
           if (!(png_ptr->transformations & PNG_EXPAND_tRNS))
           {
              /* Invert the alpha channel (in tRNS) unless the pixels are
               * going to be expanded, in which case leave it for later
               */
              int i, istop = png_ptr->num_trans;

              for (i=0; i<istop; i++)
                 png_ptr->trans_alpha[i] = (png_byte)(255 -
                    png_ptr->trans_alpha[i]);
           }
        }
#endif /* PNG_READ_INVERT_ALPHA_SUPPORTED */
      }
   } /* background expand and (therefore) no alpha association. */
#endif /* PNG_READ_EXPAND_SUPPORTED && PNG_READ_BACKGROUND_SUPPORTED */
}

static void /* PRIVATE */
png_init_rgb_transformations(png_structp png_ptr)
{
1162
   /* Added to libpng-1.5.4: check the color type to determine whether there
1163 1164 1165 1166 1167
    * is any alpha or transparency in the image and simply cancel the
    * background and alpha mode stuff if there isn't.
    */
   int input_has_alpha = (png_ptr->color_type & PNG_COLOR_MASK_ALPHA) != 0;
   int input_has_transparency = png_ptr->num_trans > 0;
1168

1169 1170
   /* If no alpha we can optimize. */
   if (!input_has_alpha)
1171
   {
1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182
      /* Any alpha means background and associative alpha processing is
       * required, however if the alpha is 0 or 1 throughout OPTIIMIZE_ALPHA
       * and ENCODE_ALPHA are irrelevant.
       */
#     ifdef PNG_READ_ALPHA_MODE_SUPPORTED
         png_ptr->transformations &= ~PNG_ENCODE_ALPHA;
         png_ptr->flags &= ~PNG_FLAG_OPTIMIZE_ALPHA;
#     endif

      if (!input_has_transparency)
         png_ptr->transformations &= ~(PNG_COMPOSE | PNG_BACKGROUND_EXPAND);
1183 1184
   }

1185 1186 1187 1188 1189 1190 1191 1192 1193
#if defined(PNG_READ_EXPAND_SUPPORTED) && defined(PNG_READ_BACKGROUND_SUPPORTED)
   /* png_set_background handling - deals with the complexity of whether the
    * background color is in the file format or the screen format in the case
    * where an 'expand' will happen.
    */

   /* The following code cannot be entered in the alpha pre-multiplication case
    * because PNG_BACKGROUND_EXPAND is cancelled below.
    */
1194
   if ((png_ptr->transformations & PNG_BACKGROUND_EXPAND) &&
1195 1196 1197
       (png_ptr->transformations & PNG_EXPAND) &&
       !(png_ptr->color_type & PNG_COLOR_MASK_COLOR))
       /* i.e., GRAY or GRAY_ALPHA */
G
Guy Schalnat 已提交
1198 1199
   {
      {
1200
         /* Expand background and tRNS chunks */
G
Guy Schalnat 已提交
1201 1202 1203
         switch (png_ptr->bit_depth)
         {
            case 1:
G
Guy Schalnat 已提交
1204
               png_ptr->background.gray *= (png_uint_16)0xff;
1205
               png_ptr->background.red = png_ptr->background.green
1206
                   =  png_ptr->background.blue = png_ptr->background.gray;
1207 1208
               if (!(png_ptr->transformations & PNG_EXPAND_tRNS))
               {
1209 1210
                 png_ptr->trans_color.gray *= (png_uint_16)0xff;
                 png_ptr->trans_color.red = png_ptr->trans_color.green
1211
                     = png_ptr->trans_color.blue = png_ptr->trans_color.gray;
1212
               }
G
Guy Schalnat 已提交
1213
               break;
1214

G
Guy Schalnat 已提交
1215
            case 2:
G
Guy Schalnat 已提交
1216
               png_ptr->background.gray *= (png_uint_16)0x55;
1217
               png_ptr->background.red = png_ptr->background.green
1218
                   = png_ptr->background.blue = png_ptr->background.gray;
1219 1220
               if (!(png_ptr->transformations & PNG_EXPAND_tRNS))
               {
1221 1222 1223
                  png_ptr->trans_color.gray *= (png_uint_16)0x55;
                  png_ptr->trans_color.red = png_ptr->trans_color.green
                      = png_ptr->trans_color.blue = png_ptr->trans_color.gray;
1224
               }
G
Guy Schalnat 已提交
1225
               break;
1226

G
Guy Schalnat 已提交
1227
            case 4:
G
Guy Schalnat 已提交
1228
               png_ptr->background.gray *= (png_uint_16)0x11;
1229
               png_ptr->background.red = png_ptr->background.green
1230
                   = png_ptr->background.blue = png_ptr->background.gray;
1231 1232
               if (!(png_ptr->transformations & PNG_EXPAND_tRNS))
               {
1233 1234 1235
                  png_ptr->trans_color.gray *= (png_uint_16)0x11;
                  png_ptr->trans_color.red = png_ptr->trans_color.green
                      = png_ptr->trans_color.blue = png_ptr->trans_color.gray;
1236
               }
G
Guy Schalnat 已提交
1237
               break;
1238

1239 1240
            default:

G
Guy Schalnat 已提交
1241
            case 8:
1242

G
Guy Schalnat 已提交
1243
            case 16:
1244
               png_ptr->background.red = png_ptr->background.green
1245
                   = png_ptr->background.blue = png_ptr->background.gray;
G
Guy Schalnat 已提交
1246 1247 1248
               break;
         }
      }
1249 1250 1251
   } /* background expand and (therefore) no alpha association. */
#endif /* PNG_READ_EXPAND_SUPPORTED && PNG_READ_BACKGROUND_SUPPORTED */
}
1252

1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266
void /* PRIVATE */
png_init_read_transformations(png_structp png_ptr)
{
   png_debug(1, "in png_init_read_transformations");

   /* This internal function is called from png_read_start_row in pngrutil.c
    * and it is called before the 'rowbytes' calculation is done, so the code
    * in here can change or update the transformations flags.
    *
    * First do updates that do not depend on the details of the PNG image data
    * being processed.
    */

#ifdef PNG_READ_GAMMA_SUPPORTED
1267
   /* Prior to 1.5.4 these tests were performed from png_set_gamma, 1.5.4 adds
1268
    * png_set_alpha_mode and this is another source for a default file gamma so
1269
    * the test needs to be performed later - here.  In addition prior to 1.5.4
1270 1271 1272 1273 1274 1275 1276 1277
    * the tests were repeated for the PALETTE color type here - this is no
    * longer necessary (and doesn't seem to have been necessary before.)
    */
   {
      /* The following temporary indicates if overall gamma correction is
       * required.
       */
      int gamma_correction = 0;
1278

1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289
      if (png_ptr->gamma != 0) /* has been set */
      {
         if (png_ptr->screen_gamma != 0) /* screen set too */
            gamma_correction = png_gamma_threshold(png_ptr->gamma,
               png_ptr->screen_gamma);

         else
            /* Assume the output matches the input; a long time default behavior
             * of libpng, although the standard has nothing to say about this.
             */
            png_ptr->screen_gamma = png_reciprocal(png_ptr->gamma);
G
Guy Schalnat 已提交
1290
      }
1291 1292 1293

      else if (png_ptr->screen_gamma != 0)
         /* The converse - assume the file matches the screen, note that this
1294
          * perhaps undesireable default can (from 1.5.4) be changed by calling
1295 1296 1297 1298 1299 1300 1301 1302 1303
          * png_set_alpha_mode (even if the alpha handling mode isn't required
          * or isn't changed from the default.)
          */
         png_ptr->gamma = png_reciprocal(png_ptr->screen_gamma);

      else /* neither are set */
         /* Just in case the following prevents any processing - file and screen
          * are both assumed to be linear and there is no way to introduce a
          * third gamma value other than png_set_background with 'UNIQUE', and,
1304
          * prior to 1.5.4
1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320
          */
         png_ptr->screen_gamma = png_ptr->gamma = PNG_FP_1;

      /* Now turn the gamma transformation on or off as appropriate.  Notice
       * that PNG_GAMMA just refers to the file->screen correction.  Alpha
       * composition may independently cause gamma correction because it needs
       * linear data (e.g. if the file has a gAMA chunk but the screen gamma
       * hasn't been specified.)  In any case this flag may get turned off in
       * the code immediately below if the transform can be handled outside the
       * row loop.
       */
      if (gamma_correction)
         png_ptr->transformations |= PNG_GAMMA;

      else
         png_ptr->transformations &= ~PNG_GAMMA;
G
Guy Schalnat 已提交
1321
   }
G
Guy Schalnat 已提交
1322
#endif
G
Guy Schalnat 已提交
1323

1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336
   /* Certain transformations have the effect of preventing other
    * transformations that happen afterward in png_do_read_transformations,
    * resolve the interdependencies here.  From the code of
    * png_do_read_transformations the order is:
    *
    *  1) PNG_EXPAND (including PNG_EXPAND_tRNS)
    *  2) PNG_STRIP_ALPHA (if no compose)
    *  3) PNG_RGB_TO_GRAY
    *  4) PNG_GRAY_TO_RGB iff !PNG_BACKGROUND_IS_GRAY
    *  5) PNG_COMPOSE
    *  6) PNG_GAMMA
    *  7) PNG_STRIP_ALPHA (if compose)
    *  8) PNG_ENCODE_ALPHA
1337
    *  9) PNG_16_TO_8 or PNG_SCALE_16_TO_8 (strip16/scale16)
1338
    * 10) PNG_QUANTIZE (converts to palette)
1339
    * 11) PNG_EXPAND_16
1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364
    * 12) PNG_GRAY_TO_RGB iff PNG_BACKGROUND_IS_GRAY
    * 13) PNG_INVERT_MONO
    * 14) PNG_SHIFT
    * 15) PNG_PACK
    * 16) PNG_BGR
    * 17) PNG_PACKSWAP
    * 18) PNG_FILLER (includes PNG_ADD_ALPHA)
    * 19) PNG_INVERT_ALPHA
    * 20) PNG_SWAP_ALPHA
    * 21) PNG_SWAP_BYTES
    * 22) PNG_USER_TRANSFORM [must be last]
    */
#ifdef PNG_READ_STRIP_ALPHA_SUPPORTED
   if ((png_ptr->transformations & PNG_STRIP_ALPHA) &&
      !(png_ptr->transformations & PNG_COMPOSE))
   {
      /* Stripping the alpha channel happens immediately after the 'expand'
       * transformations, before all other transformation, so it cancels out
       * the alpha handling.  It has the side effect negating the effect of
       * PNG_EXPAND_tRNS too:
       */
      png_ptr->transformations &= ~(PNG_BACKGROUND_EXPAND | PNG_ENCODE_ALPHA |
         PNG_EXPAND_tRNS);
      png_ptr->flags &= ~PNG_FLAG_OPTIMIZE_ALPHA;

1365
      /* Kill the tRNS chunk itself too.  Prior to 1.5.4 this did not happen
1366 1367 1368 1369
       * so transparency information would remain just so long as it wasn't
       * expanded.  This produces unexpected API changes if the set of things
       * that do PNG_EXPAND_tRNS changes (perfectly possible given the
       * documentation - which says ask for what you want, accept what you
1370
       * get.)  This makes the behavior consistent from 1.5.4:
1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384
       */
      png_ptr->num_trans = 0;
   }
#endif /* STRIP_ALPHA supported, no COMPOSE */

#ifdef PNG_READ_ALPHA_MODE_SUPPORTED
   /* If the screen gamma is about 1.0 then the OPTIMIZE_ALPHA and ENCODE_ALPHA
    * settings will have no effect.
    */
   if (!png_gamma_significant(png_ptr->screen_gamma))
   {
      png_ptr->transformations &= ~PNG_ENCODE_ALPHA;
      png_ptr->flags &= ~PNG_FLAG_OPTIMIZE_ALPHA;
   }
G
Guy Schalnat 已提交
1385
#endif
1386

1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399
#if defined(PNG_READ_EXPAND_SUPPORTED) && \
   defined(PNG_READ_BACKGROUND_SUPPORTED) && \
   defined(PNG_READ_GRAY_TO_RGB_SUPPORTED)
   /* Detect gray background and attempt to enable optimization for
    * gray --> RGB case.
    *
    * Note:  if PNG_BACKGROUND_EXPAND is set and color_type is either RGB or
    * RGB_ALPHA (in which case need_expand is superfluous anyway), the
    * background color might actually be gray yet not be flagged as such.
    * This is not a problem for the current code, which uses
    * PNG_BACKGROUND_IS_GRAY only to decide when to do the
    * png_do_gray_to_rgb() transformation.
    *
1400
    * TODO: this code needs to be revised to avoid the complexity and
1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414
    * interdependencies.  The color type of the background should be recorded in
    * png_set_background, along with the bit depth, then the code has a record
    * of exactly what color space the background is currently in.
    */
   if (png_ptr->transformations & PNG_BACKGROUND_EXPAND)
   {
      /* PNG_BACKGROUND_EXPAND: the background is in the file color space, so if
       * the file was greyscale the background value is gray.
       */
      if (!(png_ptr->color_type & PNG_COLOR_MASK_COLOR))
         png_ptr->mode |= PNG_BACKGROUND_IS_GRAY;
   }

   else if (png_ptr->transformations & PNG_COMPOSE)
1415
   {
1416 1417 1418 1419 1420 1421
      /* PNG_COMPOSE: png_set_background was called with need_expand false,
       * so the color is in the color space of the output or png_set_alpha_mode
       * was called and the color is black.  Ignore RGB_TO_GRAY because that
       * happens before GRAY_TO_RGB.
       */
      if (png_ptr->transformations & PNG_GRAY_TO_RGB)
1422
      {
1423 1424 1425 1426 1427 1428
         if (png_ptr->background.red == png_ptr->background.green &&
             png_ptr->background.red == png_ptr->background.blue)
         {
            png_ptr->mode |= PNG_BACKGROUND_IS_GRAY;
            png_ptr->background.gray = png_ptr->background.red;
         }
1429
      }
1430
   }
1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445
#endif /* PNG_READ_GRAY_TO_RGB_SUPPORTED (etc) */

   /* For indexed PNG data (PNG_COLOR_TYPE_PALETTE) many of the transformations
    * can be performed directly on the palette, and some (such as rgb to gray)
    * can be optimized inside the palette.  This is particularly true of the
    * composite (background and alpha) stuff, which can be pretty much all done
    * in the palette even if the result is expanded to RGB or gray afterward.
    *
    * NOTE: this is Not Yet Implemented, the code behaves as in 1.5.1 and
    * earlier and the palette stuff is actually handled on the first row.  This
    * leads to the reported bug that the palette returned by png_get_PLTE is not
    * updated.
    */
   if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
      png_init_palette_transformations(png_ptr);
1446

1447 1448 1449
   else
      png_init_rgb_transformations(png_ptr);

1450 1451 1452 1453 1454 1455 1456 1457 1458
#if defined(PNG_READ_BACKGROUND_SUPPORTED) && \
   defined(PNG_READ_EXPAND_16_SUPPORTED)
   if ((png_ptr->transformations & PNG_EXPAND_16) &&
      (png_ptr->transformations & PNG_COMPOSE) &&
      !(png_ptr->transformations & PNG_BACKGROUND_EXPAND) &&
      png_ptr->bit_depth != 16)
   {
      /* TODO: fix this.  Because the expand_16 operation is after the compose
       * handling the background color must be 8, not 16, bits deep, but the
1459
       * application will supply a 16-bit value so reduce it here.
1460 1461 1462
       *
       * The PNG_BACKGROUND_EXPAND code above does not expand to 16 bits at
       * present, so that case is ok (until do_expand_16 is moved.)
1463 1464 1465
       *
       * NOTE: this discards the low 16 bits of the user supplied background
       * color, but until expand_16 works properly there is no choice!
1466
       */
1467 1468 1469 1470 1471
#     define CHOP(x) (x)=((png_uint_16)(((png_uint_32)(x)*255+32895) >> 16))
      CHOP(png_ptr->background.red);
      CHOP(png_ptr->background.green);
      CHOP(png_ptr->background.blue);
      CHOP(png_ptr->background.gray);
1472 1473
#     undef CHOP
   }
1474
#endif /* PNG_READ_BACKGROUND_SUPPORTED && PNG_READ_EXPAND_16_SUPPORTED */
1475

1476 1477 1478 1479
   /* NOTE: below 'PNG_READ_ALPHA_MODE_SUPPORTED' is presumed to also enable the
    * background support (see the comments in scripts/pnglibconf.dfa), this
    * allows pre-multiplication of the alpha channel to be implemented as
    * compositing on black.  This is probably sub-optimal and has been done in
1480
    * 1.5.4 betas simply to enable external critique and testing (i.e. to
1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498
    * implement the new API quickly, without lots of internal changes.)
    */

#ifdef PNG_READ_GAMMA_SUPPORTED
#  ifdef PNG_READ_BACKGROUND_SUPPORTED
      /* Includes ALPHA_MODE */
      png_ptr->background_1 = png_ptr->background;
#  endif

   /* This needs to change - in the palette image case a whole set of tables are
    * built when it would be quicker to just calculate the correct value for
    * each palette entry directly.  Also, the test is too tricky - why check
    * PNG_RGB_TO_GRAY if PNG_GAMMA is not set?  The answer seems to be that
    * PNG_GAMMA is cancelled even if the gamma is known?  The test excludes the
    * PNG_COMPOSE case, so apparently if there is no *overall* gamma correction
    * the gamma tables will not be built even if composition is required on a
    * gamma encoded value.
    *
1499
    * In 1.5.4 this is addressed below by an additional check on the individual
1500 1501 1502 1503 1504 1505 1506 1507 1508 1509
    * file gamma - if it is not 1.0 both RGB_TO_GRAY and COMPOSE need the
    * tables.
    */
   if ((png_ptr->transformations & PNG_GAMMA)
      || ((png_ptr->transformations & PNG_RGB_TO_GRAY)
         && (png_gamma_significant(png_ptr->gamma) ||
            png_gamma_significant(png_ptr->screen_gamma)))
      || ((png_ptr->transformations & PNG_COMPOSE)
         && (png_gamma_significant(png_ptr->gamma)
            || png_gamma_significant(png_ptr->screen_gamma)
1510
#  ifdef PNG_READ_BACKGROUND_SUPPORTED
1511
            || (png_ptr->background_gamma_type == PNG_BACKGROUND_GAMMA_UNIQUE
1512 1513 1514
               && png_gamma_significant(png_ptr->background_gamma))
#  endif
      )) || ((png_ptr->transformations & PNG_ENCODE_ALPHA)
1515 1516
         && png_gamma_significant(png_ptr->screen_gamma))
      )
G
Guy Schalnat 已提交
1517
   {
1518
      png_build_gamma_table(png_ptr, png_ptr->bit_depth);
1519

1520
#ifdef PNG_READ_BACKGROUND_SUPPORTED
1521
      if (png_ptr->transformations & PNG_COMPOSE)
G
Guy Schalnat 已提交
1522
      {
1523
         if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
G
Guy Schalnat 已提交
1524
         {
1525 1526 1527
            /* We don't get to here unless there is a tRNS chunk with non-opaque
             * entries - see the checking code at the start of this function.
             */
G
Guy Schalnat 已提交
1528
            png_color back, back_1;
1529 1530 1531
            png_colorp palette = png_ptr->palette;
            int num_palette = png_ptr->num_palette;
            int i;
A
Andreas Dilger 已提交
1532 1533
            if (png_ptr->background_gamma_type == PNG_BACKGROUND_GAMMA_FILE)
            {
G
[devel]  
Glenn Randers-Pehrson 已提交
1534

A
Andreas Dilger 已提交
1535 1536 1537 1538 1539 1540 1541 1542 1543 1544
               back.red = png_ptr->gamma_table[png_ptr->background.red];
               back.green = png_ptr->gamma_table[png_ptr->background.green];
               back.blue = png_ptr->gamma_table[png_ptr->background.blue];

               back_1.red = png_ptr->gamma_to_1[png_ptr->background.red];
               back_1.green = png_ptr->gamma_to_1[png_ptr->background.green];
               back_1.blue = png_ptr->gamma_to_1[png_ptr->background.blue];
            }
            else
            {
G
[devel]  
Glenn Randers-Pehrson 已提交
1545
               png_fixed_point g, gs;
A
Andreas Dilger 已提交
1546

1547 1548 1549 1550
               switch (png_ptr->background_gamma_type)
               {
                  case PNG_BACKGROUND_GAMMA_SCREEN:
                     g = (png_ptr->screen_gamma);
G
[devel]  
Glenn Randers-Pehrson 已提交
1551
                     gs = PNG_FP_1;
1552
                     break;
1553

1554
                  case PNG_BACKGROUND_GAMMA_FILE:
G
[devel]  
Glenn Randers-Pehrson 已提交
1555 1556
                     g = png_reciprocal(png_ptr->gamma);
                     gs = png_reciprocal2(png_ptr->gamma,
1557
                        png_ptr->screen_gamma);
1558
                     break;
1559

1560
                  case PNG_BACKGROUND_GAMMA_UNIQUE:
G
[devel]  
Glenn Randers-Pehrson 已提交
1561 1562
                     g = png_reciprocal(png_ptr->background_gamma);
                     gs = png_reciprocal2(png_ptr->background_gamma,
1563
                        png_ptr->screen_gamma);
1564 1565
                     break;
                  default:
G
[devel]  
Glenn Randers-Pehrson 已提交
1566 1567
                     g = PNG_FP_1;    /* back_1 */
                     gs = PNG_FP_1;   /* back */
1568
                     break;
1569
               }
A
Andreas Dilger 已提交
1570

1571
               if (png_gamma_significant(gs))
1572 1573 1574 1575 1576 1577 1578 1579 1580 1581
               {
                  back.red = png_gamma_8bit_correct(png_ptr->background.red,
                      gs);
                  back.green = png_gamma_8bit_correct(png_ptr->background.green,
                      gs);
                  back.blue = png_gamma_8bit_correct(png_ptr->background.blue,
                      gs);
               }

               else
A
Andreas Dilger 已提交
1582
               {
1583
                  back.red   = (png_byte)png_ptr->background.red;
A
Andreas Dilger 已提交
1584
                  back.green = (png_byte)png_ptr->background.green;
1585
                  back.blue  = (png_byte)png_ptr->background.blue;
A
Andreas Dilger 已提交
1586
               }
1587

1588 1589 1590 1591 1592 1593 1594 1595 1596 1597
               if (png_gamma_significant(g))
               {
                  back_1.red = png_gamma_8bit_correct(png_ptr->background.red,
                     g);
                  back_1.green = png_gamma_8bit_correct(
                     png_ptr->background.green, g);
                  back_1.blue = png_gamma_8bit_correct(png_ptr->background.blue,
                     g);
               }

A
Andreas Dilger 已提交
1598 1599
               else
               {
1600 1601 1602
                  back_1.red   = (png_byte)png_ptr->background.red;
                  back_1.green = (png_byte)png_ptr->background.green;
                  back_1.blue  = (png_byte)png_ptr->background.blue;
A
Andreas Dilger 已提交
1603 1604
               }
            }
1605

G
Guy Schalnat 已提交
1606 1607
            for (i = 0; i < num_palette; i++)
            {
1608 1609
               if (i < (int)png_ptr->num_trans &&
                   png_ptr->trans_alpha[i] != 0xff)
G
Guy Schalnat 已提交
1610
               {
1611
                  if (png_ptr->trans_alpha[i] == 0)
G
Guy Schalnat 已提交
1612 1613 1614
                  {
                     palette[i] = back;
                  }
1615
                  else /* if (png_ptr->trans_alpha[i] != 0xff) */
G
Guy Schalnat 已提交
1616
                  {
A
Andreas Dilger 已提交
1617
                     png_byte v, w;
G
Guy Schalnat 已提交
1618 1619

                     v = png_ptr->gamma_to_1[palette[i].red];
1620
                     png_composite(w, v, png_ptr->trans_alpha[i], back_1.red);
G
Guy Schalnat 已提交
1621 1622 1623
                     palette[i].red = png_ptr->gamma_from_1[w];

                     v = png_ptr->gamma_to_1[palette[i].green];
1624
                     png_composite(w, v, png_ptr->trans_alpha[i], back_1.green);
G
Guy Schalnat 已提交
1625 1626 1627
                     palette[i].green = png_ptr->gamma_from_1[w];

                     v = png_ptr->gamma_to_1[palette[i].blue];
1628
                     png_composite(w, v, png_ptr->trans_alpha[i], back_1.blue);
G
Guy Schalnat 已提交
1629 1630 1631 1632 1633 1634 1635 1636 1637 1638
                     palette[i].blue = png_ptr->gamma_from_1[w];
                  }
               }
               else
               {
                  palette[i].red = png_ptr->gamma_table[palette[i].red];
                  palette[i].green = png_ptr->gamma_table[palette[i].green];
                  palette[i].blue = png_ptr->gamma_table[palette[i].blue];
               }
            }
1639 1640 1641

            /* Prevent the transformations being done again.
             *
1642
             * NOTE: this is highly dubious; it zaps the transformations in
1643 1644
             * place.  This seems inconsistent with the general treatment of the
             * transformations elsewhere.
1645
             */
1646 1647
            png_ptr->transformations &= ~(PNG_COMPOSE | PNG_GAMMA);
         } /* color_type == PNG_COLOR_TYPE_PALETTE */
1648

1649
         /* if (png_ptr->background_gamma_type!=PNG_BACKGROUND_GAMMA_UNKNOWN) */
1650
         else /* color_type != PNG_COLOR_TYPE_PALETTE */
G
Guy Schalnat 已提交
1651
         {
1652 1653
            png_fixed_point g = PNG_FP_1;
            png_fixed_point gs = PNG_FP_1;
G
Guy Schalnat 已提交
1654 1655

            switch (png_ptr->background_gamma_type)
G
Guy Schalnat 已提交
1656
            {
G
Guy Schalnat 已提交
1657
               case PNG_BACKGROUND_GAMMA_SCREEN:
G
[devel]  
Glenn Randers-Pehrson 已提交
1658 1659
                  g = png_ptr->screen_gamma;
                  /* gs = PNG_FP_1; */
G
Guy Schalnat 已提交
1660
                  break;
1661

G
Guy Schalnat 已提交
1662
               case PNG_BACKGROUND_GAMMA_FILE:
G
[devel]  
Glenn Randers-Pehrson 已提交
1663 1664
                  g = png_reciprocal(png_ptr->gamma);
                  gs = png_reciprocal2(png_ptr->gamma, png_ptr->screen_gamma);
G
Guy Schalnat 已提交
1665
                  break;
1666

G
Guy Schalnat 已提交
1667
               case PNG_BACKGROUND_GAMMA_UNIQUE:
G
[devel]  
Glenn Randers-Pehrson 已提交
1668 1669
                  g = png_reciprocal(png_ptr->background_gamma);
                  gs = png_reciprocal2(png_ptr->background_gamma,
1670
                      png_ptr->screen_gamma);
G
Guy Schalnat 已提交
1671
                  break;
1672 1673

               default:
1674
                  png_error(png_ptr, "invalid background gamma type");
G
Guy Schalnat 已提交
1675 1676
            }

G
[devel]  
Glenn Randers-Pehrson 已提交
1677
            png_ptr->background_1.gray = png_gamma_correct(png_ptr,
1678
                png_ptr->background.gray, g);
1679

G
[devel]  
Glenn Randers-Pehrson 已提交
1680 1681
            png_ptr->background.gray = png_gamma_correct(png_ptr,
                png_ptr->background.gray, gs);
1682

1683 1684 1685
            if ((png_ptr->background.red != png_ptr->background.green) ||
                (png_ptr->background.red != png_ptr->background.blue) ||
                (png_ptr->background.red != png_ptr->background.gray))
G
Guy Schalnat 已提交
1686
            {
1687
               /* RGB or RGBA with color background */
G
[devel]  
Glenn Randers-Pehrson 已提交
1688 1689
               png_ptr->background_1.red = png_gamma_correct(png_ptr,
                   png_ptr->background.red, g);
1690

G
[devel]  
Glenn Randers-Pehrson 已提交
1691 1692
               png_ptr->background_1.green = png_gamma_correct(png_ptr,
                   png_ptr->background.green, g);
1693

G
[devel]  
Glenn Randers-Pehrson 已提交
1694 1695
               png_ptr->background_1.blue = png_gamma_correct(png_ptr,
                   png_ptr->background.blue, g);
1696

G
[devel]  
Glenn Randers-Pehrson 已提交
1697 1698
               png_ptr->background.red = png_gamma_correct(png_ptr,
                   png_ptr->background.red, gs);
1699

G
[devel]  
Glenn Randers-Pehrson 已提交
1700 1701
               png_ptr->background.green = png_gamma_correct(png_ptr,
                   png_ptr->background.green, gs);
1702

G
[devel]  
Glenn Randers-Pehrson 已提交
1703 1704
               png_ptr->background.blue = png_gamma_correct(png_ptr,
                   png_ptr->background.blue, gs);
G
Guy Schalnat 已提交
1705
            }
1706

G
Guy Schalnat 已提交
1707 1708
            else
            {
1709 1710
               /* GRAY, GRAY ALPHA, RGB, or RGBA with gray background */
               png_ptr->background_1.red = png_ptr->background_1.green
1711
                   = png_ptr->background_1.blue = png_ptr->background_1.gray;
1712

1713
               png_ptr->background.red = png_ptr->background.green
1714
                   = png_ptr->background.blue = png_ptr->background.gray;
G
Guy Schalnat 已提交
1715
            }
1716 1717 1718
         } /* color_type != PNG_COLOR_TYPE_PALETTE */
      }/* png_ptr->transformations & PNG_BACKGROUND */

G
Guy Schalnat 已提交
1719
      else
1720
      /* Transformation does not include PNG_BACKGROUND */
1721
#endif /* PNG_READ_BACKGROUND_SUPPORTED */
1722
      if (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
G
Guy Schalnat 已提交
1723
      {
1724 1725 1726
         png_colorp palette = png_ptr->palette;
         int num_palette = png_ptr->num_palette;
         int i;
G
Guy Schalnat 已提交
1727

1728 1729 1730
         /*NOTE: there are other transformations that should probably be in here
          * too.
          */
G
Guy Schalnat 已提交
1731 1732 1733 1734 1735 1736
         for (i = 0; i < num_palette; i++)
         {
            palette[i].red = png_ptr->gamma_table[palette[i].red];
            palette[i].green = png_ptr->gamma_table[palette[i].green];
            palette[i].blue = png_ptr->gamma_table[palette[i].blue];
         }
1737

1738 1739
         /* Done the gamma correction. */
         png_ptr->transformations &= ~PNG_GAMMA;
1740
      } /* color_type == PALETTE && !PNG_BACKGROUND transformation */
G
Guy Schalnat 已提交
1741
   }
1742
#ifdef PNG_READ_BACKGROUND_SUPPORTED
G
Guy Schalnat 已提交
1743
   else
G
Guy Schalnat 已提交
1744
#endif
G
[devel]  
Glenn Randers-Pehrson 已提交
1745
#endif /* PNG_READ_GAMMA_SUPPORTED */
1746

1747
#ifdef PNG_READ_BACKGROUND_SUPPORTED
1748 1749 1750
   /* No GAMMA transformation (see the hanging else 4 lines above) */
   if ((png_ptr->transformations & PNG_COMPOSE) &&
       (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE))
G
Guy Schalnat 已提交
1751
   {
1752 1753
      int i;
      int istop = (int)png_ptr->num_trans;
G
Guy Schalnat 已提交
1754
      png_color back;
1755
      png_colorp palette = png_ptr->palette;
G
Guy Schalnat 已提交
1756 1757 1758 1759 1760

      back.red   = (png_byte)png_ptr->background.red;
      back.green = (png_byte)png_ptr->background.green;
      back.blue  = (png_byte)png_ptr->background.blue;

1761
      for (i = 0; i < istop; i++)
G
Guy Schalnat 已提交
1762
      {
1763
         if (png_ptr->trans_alpha[i] == 0)
G
Guy Schalnat 已提交
1764 1765 1766
         {
            palette[i] = back;
         }
1767

1768
         else if (png_ptr->trans_alpha[i] != 0xff)
G
Guy Schalnat 已提交
1769
         {
1770
            /* The png_composite() macro is defined in png.h */
A
Andreas Dilger 已提交
1771
            png_composite(palette[i].red, palette[i].red,
1772
                png_ptr->trans_alpha[i], back.red);
1773

A
Andreas Dilger 已提交
1774
            png_composite(palette[i].green, palette[i].green,
1775
                png_ptr->trans_alpha[i], back.green);
1776

A
Andreas Dilger 已提交
1777
            png_composite(palette[i].blue, palette[i].blue,
1778
                png_ptr->trans_alpha[i], back.blue);
G
Guy Schalnat 已提交
1779 1780
         }
      }
1781

1782
      png_ptr->transformations &= ~PNG_COMPOSE;
G
Guy Schalnat 已提交
1783
   }
1784
#endif /* PNG_READ_BACKGROUND_SUPPORTED */
G
Guy Schalnat 已提交
1785

1786
#ifdef PNG_READ_SHIFT_SUPPORTED
G
Guy Schalnat 已提交
1787
   if ((png_ptr->transformations & PNG_SHIFT) &&
1788
       (png_ptr->color_type == PNG_COLOR_TYPE_PALETTE))
G
Guy Schalnat 已提交
1789
   {
1790 1791 1792 1793 1794
      png_uint_16 i;
      png_uint_16 istop = png_ptr->num_palette;
      int sr = 8 - png_ptr->sig_bit.red;
      int sg = 8 - png_ptr->sig_bit.green;
      int sb = 8 - png_ptr->sig_bit.blue;
G
Guy Schalnat 已提交
1795 1796 1797

      if (sr < 0 || sr > 8)
         sr = 0;
1798

G
Guy Schalnat 已提交
1799 1800
      if (sg < 0 || sg > 8)
         sg = 0;
1801

G
Guy Schalnat 已提交
1802 1803
      if (sb < 0 || sb > 8)
         sb = 0;
1804

1805
      for (i = 0; i < istop; i++)
G
Guy Schalnat 已提交
1806 1807 1808 1809 1810 1811
      {
         png_ptr->palette[i].red >>= sr;
         png_ptr->palette[i].green >>= sg;
         png_ptr->palette[i].blue >>= sb;
      }
   }
1812
#endif  /* PNG_READ_SHIFT_SUPPORTED */
G
Guy Schalnat 已提交
1813 1814
}

A
Andreas Dilger 已提交
1815
/* Modify the info structure to reflect the transformations.  The
1816 1817 1818
 * info should be updated so a PNG file could be written with it,
 * assuming the transformations result in valid PNG data.
 */
1819
void /* PRIVATE */
G
Guy Schalnat 已提交
1820
png_read_transform_info(png_structp png_ptr, png_infop info_ptr)
G
Guy Schalnat 已提交
1821
{
1822
   png_debug(1, "in png_read_transform_info");
1823

1824
#ifdef PNG_READ_EXPAND_SUPPORTED
A
Andreas Dilger 已提交
1825
   if (png_ptr->transformations & PNG_EXPAND)
G
Guy Schalnat 已提交
1826
   {
A
Andreas Dilger 已提交
1827 1828
      if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
      {
1829
         /* This check must match what actually happens in
1830
          * png_do_expand_palette; if it ever checks the tRNS chunk to see if
1831 1832 1833
          * it is all opaque we must do the same (at present it does not.)
          */
         if (png_ptr->num_trans > 0)
A
Andreas Dilger 已提交
1834
            info_ptr->color_type = PNG_COLOR_TYPE_RGB_ALPHA;
1835

A
Andreas Dilger 已提交
1836 1837
         else
            info_ptr->color_type = PNG_COLOR_TYPE_RGB;
1838

G
Guy Schalnat 已提交
1839
         info_ptr->bit_depth = 8;
A
Andreas Dilger 已提交
1840 1841 1842 1843 1844
         info_ptr->num_trans = 0;
      }
      else
      {
         if (png_ptr->num_trans)
1845 1846
         {
            if (png_ptr->transformations & PNG_EXPAND_tRNS)
1847
               info_ptr->color_type |= PNG_COLOR_MASK_ALPHA;
1848
         }
A
Andreas Dilger 已提交
1849 1850
         if (info_ptr->bit_depth < 8)
            info_ptr->bit_depth = 8;
1851

A
Andreas Dilger 已提交
1852 1853
         info_ptr->num_trans = 0;
      }
G
Guy Schalnat 已提交
1854 1855 1856
   }
#endif

1857 1858 1859 1860 1861 1862
#if defined(PNG_READ_BACKGROUND_SUPPORTED) ||\
   defined(PNG_READ_ALPHA_MODE_SUPPORTED)
   /* The following is almost certainly wrong unless the background value is in
    * the screen space!
    */
   if (png_ptr->transformations & PNG_COMPOSE)
G
Guy Schalnat 已提交
1863 1864 1865
      info_ptr->background = png_ptr->background;
#endif

1866
#ifdef PNG_READ_GAMMA_SUPPORTED
1867
   /* The following used to be conditional on PNG_GAMMA (prior to 1.5.4),
1868 1869 1870 1871 1872
    * however it seems that the code in png_init_read_transformations, which has
    * been called before this from png_read_update_info->png_read_start_row
    * sometimes does the gamma transform and cancels the flag.
    */
   info_ptr->gamma = png_ptr->gamma;
1873
#endif
1874

1875
#ifdef PNG_READ_16_TO_8_SUPPORTED
1876
#  ifdef PNG_READ_16BIT_SUPPORTED
1877
   if ((png_ptr->transformations & (PNG_16_TO_8 | PNG_SCALE_16_TO_8)) &&
1878
       (info_ptr->bit_depth == 16))
G
Guy Schalnat 已提交
1879
      info_ptr->bit_depth = 8;
1880 1881
#  else

1882
   /* Force chopping 16-bit input down to 8 */
1883 1884
   if (info_ptr->bit_depth == 16)
   {
1885 1886
      if (!(png_ptr->transformations & PNG_16_TO_8))
#    if PNG_READ_STRIP_16_TO_8_SUPPORTED
1887
        png_ptr->transformations |=PNG_16_TO_8;
1888
#    else
1889
        png_ptr->transformations |=PNG_SCALE_16_TO_8;
1890
#    endif
1891 1892
      info_ptr->bit_depth = 8;
   }
1893
#  endif
G
Guy Schalnat 已提交
1894 1895
#endif

1896
#ifdef PNG_READ_GRAY_TO_RGB_SUPPORTED
1897 1898 1899 1900
   if (png_ptr->transformations & PNG_GRAY_TO_RGB)
      info_ptr->color_type |= PNG_COLOR_MASK_COLOR;
#endif

1901
#ifdef PNG_READ_RGB_TO_GRAY_SUPPORTED
1902 1903 1904 1905
   if (png_ptr->transformations & PNG_RGB_TO_GRAY)
      info_ptr->color_type &= ~PNG_COLOR_MASK_COLOR;
#endif

1906 1907
#ifdef PNG_READ_QUANTIZE_SUPPORTED
   if (png_ptr->transformations & PNG_QUANTIZE)
G
Guy Schalnat 已提交
1908 1909
   {
      if (((info_ptr->color_type == PNG_COLOR_TYPE_RGB) ||
1910 1911
          (info_ptr->color_type == PNG_COLOR_TYPE_RGB_ALPHA)) &&
          png_ptr->palette_lookup && info_ptr->bit_depth == 8)
G
Guy Schalnat 已提交
1912 1913 1914
      {
         info_ptr->color_type = PNG_COLOR_TYPE_PALETTE;
      }
G
Guy Schalnat 已提交
1915
   }
G
Guy Schalnat 已提交
1916 1917
#endif

1918 1919 1920 1921 1922 1923 1924 1925
#ifdef PNG_READ_EXPAND_16_SUPPORTED
   if (png_ptr->transformations & PNG_EXPAND_16 && info_ptr->bit_depth == 8 &&
      info_ptr->color_type != PNG_COLOR_TYPE_PALETTE)
   {
      info_ptr->bit_depth = 16;
   }
#endif

1926
#ifdef PNG_READ_PACK_SUPPORTED
1927
   if ((png_ptr->transformations & PNG_PACK) && (info_ptr->bit_depth < 8))
G
Guy Schalnat 已提交
1928 1929 1930
      info_ptr->bit_depth = 8;
#endif

G
Guy Schalnat 已提交
1931
   if (info_ptr->color_type == PNG_COLOR_TYPE_PALETTE)
G
Guy Schalnat 已提交
1932
      info_ptr->channels = 1;
1933

G
Guy Schalnat 已提交
1934 1935
   else if (info_ptr->color_type & PNG_COLOR_MASK_COLOR)
      info_ptr->channels = 3;
1936

G
Guy Schalnat 已提交
1937 1938
   else
      info_ptr->channels = 1;
A
Andreas Dilger 已提交
1939

1940
#ifdef PNG_READ_STRIP_ALPHA_SUPPORTED
1941
   if (png_ptr->transformations & PNG_STRIP_ALPHA)
1942
   {
A
Andreas Dilger 已提交
1943
      info_ptr->color_type &= ~PNG_COLOR_MASK_ALPHA;
1944 1945
      info_ptr->num_trans = 0;
   }
A
Andreas Dilger 已提交
1946 1947
#endif

1948 1949 1950
   if (info_ptr->color_type & PNG_COLOR_MASK_ALPHA)
      info_ptr->channels++;

1951
#ifdef PNG_READ_FILLER_SUPPORTED
1952
   /* STRIP_ALPHA and FILLER allowed:  MASK_ALPHA bit stripped above */
1953 1954 1955
   if ((png_ptr->transformations & PNG_FILLER) &&
       ((info_ptr->color_type == PNG_COLOR_TYPE_RGB) ||
       (info_ptr->color_type == PNG_COLOR_TYPE_GRAY)))
1956
   {
1957
      info_ptr->channels++;
1958
      /* If adding a true alpha channel not just filler */
1959
      if (png_ptr->transformations & PNG_ADD_ALPHA)
1960
         info_ptr->color_type |= PNG_COLOR_MASK_ALPHA;
1961
   }
A
Andreas Dilger 已提交
1962 1963
#endif

1964 1965
#if defined(PNG_USER_TRANSFORM_PTR_SUPPORTED) && \
defined(PNG_READ_USER_TRANSFORM_SUPPORTED)
1966
   if (png_ptr->transformations & PNG_USER_TRANSFORM)
1967 1968
   {
      if (info_ptr->bit_depth < png_ptr->user_transform_depth)
1969
         info_ptr->bit_depth = png_ptr->user_transform_depth;
1970

1971
      if (info_ptr->channels < png_ptr->user_transform_channels)
1972
         info_ptr->channels = png_ptr->user_transform_channels;
1973
   }
1974 1975
#endif

G
Guy Schalnat 已提交
1976
   info_ptr->pixel_depth = (png_byte)(info_ptr->channels *
1977
       info_ptr->bit_depth);
1978

1979
   info_ptr->rowbytes = PNG_ROWBYTES(info_ptr->pixel_depth, info_ptr->width);
1980

1981
   /* Adding in 1.5.4: cache the above value in png_struct so that we can later
1982 1983 1984 1985 1986 1987 1988
    * check in png_rowbytes that the user buffer won't get overwritten.  Note
    * that the field is not always set - if png_read_update_info isn't called
    * the application has to either not do any transforms or get the calculation
    * right itself.
    */
   png_ptr->info_rowbytes = info_ptr->rowbytes;

1989
#ifndef PNG_READ_EXPAND_SUPPORTED
1990
   if (png_ptr)
1991 1992
      return;
#endif
G
Guy Schalnat 已提交
1993 1994
}

1995 1996 1997 1998
/* Transform the row.  The order of transformations is significant,
 * and is very touchy.  If you add a transformation, take care to
 * decide how it fits in with the other transformations here.
 */
1999
void /* PRIVATE */
G
Guy Schalnat 已提交
2000
png_do_read_transformations(png_structp png_ptr)
G
Guy Schalnat 已提交
2001
{
2002
   png_debug(1, "in png_do_read_transformations");
2003

A
Andreas Dilger 已提交
2004
   if (png_ptr->row_buf == NULL)
G
Guy Schalnat 已提交
2005
   {
2006
      /* Prior to 1.5.4 this output row/pass where the NULL pointer is, but this
2007 2008 2009
       * error is incredibly rare and incredibly easy to debug without this
       * information.
       */
2010
      png_error(png_ptr, "NULL row buffer");
G
Guy Schalnat 已提交
2011
   }
2012

2013 2014
   /* The following is debugging; prior to 1.5.4 the code was never compiled in;
    * in 1.5.4 PNG_FLAG_DETECT_UNINITIALIZED was added and the macro
2015 2016 2017 2018 2019 2020 2021 2022
    * PNG_WARN_UNINITIALIZED_ROW removed.  In 1.5 the new flag is set only for
    * selected new APIs to ensure that there is no API change.
    */
   if ((png_ptr->flags & PNG_FLAG_DETECT_UNINITIALIZED) != 0 &&
      !(png_ptr->flags & PNG_FLAG_ROW_INIT))
   {
      /* Application has failed to call either png_read_start_image() or
       * png_read_update_info() after setting transforms that expand pixels.
2023
       * This check added to libpng-1.2.19 (but not enabled until 1.5.4).
2024
       */
2025
      png_error(png_ptr, "Uninitialized row");
2026
   }
A
Andreas Dilger 已提交
2027

2028
#ifdef PNG_READ_EXPAND_SUPPORTED
A
Andreas Dilger 已提交
2029
   if (png_ptr->transformations & PNG_EXPAND)
G
Guy Schalnat 已提交
2030
   {
A
Andreas Dilger 已提交
2031 2032 2033
      if (png_ptr->row_info.color_type == PNG_COLOR_TYPE_PALETTE)
      {
         png_do_expand_palette(&(png_ptr->row_info), png_ptr->row_buf + 1,
2034
             png_ptr->palette, png_ptr->trans_alpha, png_ptr->num_trans);
A
Andreas Dilger 已提交
2035
      }
2036

2037
      else
A
Andreas Dilger 已提交
2038
      {
2039 2040
         if (png_ptr->num_trans &&
             (png_ptr->transformations & PNG_EXPAND_tRNS))
A
Andreas Dilger 已提交
2041
            png_do_expand(&(png_ptr->row_info), png_ptr->row_buf + 1,
2042
                &(png_ptr->trans_color));
2043

2044
         else
A
Andreas Dilger 已提交
2045
            png_do_expand(&(png_ptr->row_info), png_ptr->row_buf + 1,
2046
                NULL);
A
Andreas Dilger 已提交
2047
      }
G
Guy Schalnat 已提交
2048
   }
G
Guy Schalnat 已提交
2049 2050
#endif

2051
#ifdef PNG_READ_STRIP_ALPHA_SUPPORTED
2052
   if ((png_ptr->transformations & PNG_STRIP_ALPHA) &&
2053
      !(png_ptr->transformations & PNG_COMPOSE) &&
2054 2055 2056
      (png_ptr->row_info.color_type == PNG_COLOR_TYPE_RGB_ALPHA ||
      png_ptr->row_info.color_type == PNG_COLOR_TYPE_GRAY_ALPHA))
      png_do_strip_channel(&(png_ptr->row_info), png_ptr->row_buf + 1,
2057
         0 /* at_start == false, because SWAP_ALPHA happens later */);
A
Andreas Dilger 已提交
2058 2059
#endif

2060
#ifdef PNG_READ_RGB_TO_GRAY_SUPPORTED
2061 2062 2063
   if (png_ptr->transformations & PNG_RGB_TO_GRAY)
   {
      int rgb_error =
2064 2065 2066
          png_do_rgb_to_gray(png_ptr, &(png_ptr->row_info),
              png_ptr->row_buf + 1);

2067
      if (rgb_error)
2068 2069
      {
         png_ptr->rgb_to_gray_status=1;
2070
         if ((png_ptr->transformations & PNG_RGB_TO_GRAY) ==
2071
             PNG_RGB_TO_GRAY_WARN)
2072
            png_warning(png_ptr, "png_do_rgb_to_gray found nongray pixel");
2073

2074 2075
         if ((png_ptr->transformations & PNG_RGB_TO_GRAY) ==
             PNG_RGB_TO_GRAY_ERR)
2076 2077 2078 2079 2080
            png_error(png_ptr, "png_do_rgb_to_gray found nongray pixel");
      }
   }
#endif

2081
/* From Andreas Dilger e-mail to png-implement, 26 March 1998:
2082
 *
2083 2084 2085 2086
 *   In most cases, the "simple transparency" should be done prior to doing
 *   gray-to-RGB, or you will have to test 3x as many bytes to check if a
 *   pixel is transparent.  You would also need to make sure that the
 *   transparency information is upgraded to RGB.
2087
 *
2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109
 *   To summarize, the current flow is:
 *   - Gray + simple transparency -> compare 1 or 2 gray bytes and composite
 *                                   with background "in place" if transparent,
 *                                   convert to RGB if necessary
 *   - Gray + alpha -> composite with gray background and remove alpha bytes,
 *                                   convert to RGB if necessary
 *
 *   To support RGB backgrounds for gray images we need:
 *   - Gray + simple transparency -> convert to RGB + simple transparency,
 *                                   compare 3 or 6 bytes and composite with
 *                                   background "in place" if transparent
 *                                   (3x compare/pixel compared to doing
 *                                   composite with gray bkgrnd)
 *   - Gray + alpha -> convert to RGB + alpha, composite with background and
 *                                   remove alpha bytes (3x float
 *                                   operations/pixel compared with composite
 *                                   on gray background)
 *
 *  Greg's change will do this.  The reason it wasn't done before is for
 *  performance, as this increases the per-pixel operations.  If we would check
 *  in advance if the background was gray or RGB, and position the gray-to-RGB
 *  transform appropriately, then it would save a lot of work/time.
2110 2111
 */

2112
#ifdef PNG_READ_GRAY_TO_RGB_SUPPORTED
2113 2114 2115
   /* If gray -> RGB, do so now only if background is non-gray; else do later
    * for performance reasons
    */
2116
   if ((png_ptr->transformations & PNG_GRAY_TO_RGB) &&
2117
       !(png_ptr->mode & PNG_BACKGROUND_IS_GRAY))
2118 2119 2120
      png_do_gray_to_rgb(&(png_ptr->row_info), png_ptr->row_buf + 1);
#endif

2121 2122 2123 2124
#if (defined PNG_READ_BACKGROUND_SUPPORTED) ||\
   (defined PNG_READ_ALPHA_MODE_SUPPORTED)
   if (png_ptr->transformations & PNG_COMPOSE)
      png_do_compose(&(png_ptr->row_info), png_ptr->row_buf + 1, png_ptr);
G
Guy Schalnat 已提交
2125 2126
#endif

2127
#ifdef PNG_READ_GAMMA_SUPPORTED
G
Guy Schalnat 已提交
2128
   if ((png_ptr->transformations & PNG_GAMMA) &&
2129 2130 2131
#if (defined PNG_READ_BACKGROUND_SUPPORTED) ||\
   (defined PNG_READ_ALPHA_MODE_SUPPORTED)
       !((png_ptr->transformations & PNG_COMPOSE) &&
2132 2133
       ((png_ptr->num_trans != 0) ||
       (png_ptr->color_type & PNG_COLOR_MASK_ALPHA))) &&
2134
#endif
2135
       (png_ptr->color_type != PNG_COLOR_TYPE_PALETTE))
2136
      png_do_gamma(&(png_ptr->row_info), png_ptr->row_buf + 1, png_ptr);
G
Guy Schalnat 已提交
2137 2138
#endif

2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153
#ifdef PNG_READ_STRIP_ALPHA_SUPPORTED
   if ((png_ptr->transformations & PNG_STRIP_ALPHA) &&
      (png_ptr->transformations & PNG_COMPOSE) &&
      (png_ptr->row_info.color_type == PNG_COLOR_TYPE_RGB_ALPHA ||
      png_ptr->row_info.color_type == PNG_COLOR_TYPE_GRAY_ALPHA))
      png_do_strip_channel(&(png_ptr->row_info), png_ptr->row_buf + 1,
         0 /* at_start == false, because SWAP_ALPHA happens later */);
#endif

#ifdef PNG_READ_ALPHA_MODE_SUPPORTED
   if ((png_ptr->transformations & PNG_ENCODE_ALPHA) &&
      (png_ptr->row_info.color_type & PNG_COLOR_MASK_ALPHA))
      png_do_encode_alpha(&(png_ptr->row_info), png_ptr->row_buf + 1, png_ptr);
#endif

2154
#ifdef PNG_READ_SCALE_16_TO_8_SUPPORTED
2155
   if (png_ptr->transformations & PNG_SCALE_16_TO_8)
2156
      png_do_scale_16_to_8(&(png_ptr->row_info), png_ptr->row_buf + 1);
2157 2158 2159
#endif
#ifdef PNG_READ_STRIP_16_TO_8_SUPPORTED
   if (png_ptr->transformations & PNG_16_TO_8)
2160 2161 2162
      png_do_chop(&(png_ptr->row_info), png_ptr->row_buf + 1);
#endif

2163 2164
#ifdef PNG_READ_QUANTIZE_SUPPORTED
   if (png_ptr->transformations & PNG_QUANTIZE)
G
Guy Schalnat 已提交
2165
   {
2166
      png_do_quantize(&(png_ptr->row_info), png_ptr->row_buf + 1,
2167
          png_ptr->palette_lookup, png_ptr->quantize_index);
2168

2169
      if (png_ptr->row_info.rowbytes == 0)
2170
         png_error(png_ptr, "png_do_quantize returned rowbytes=0");
A
Andreas Dilger 已提交
2171
   }
2172
#endif /* PNG_READ_QUANTIZE_SUPPORTED */
G
Guy Schalnat 已提交
2173

2174 2175 2176 2177 2178 2179 2180 2181 2182
#ifdef PNG_READ_EXPAND_16_SUPPORTED
   /* Do the expansion now, after all the arithmetic has been done.  Notice
    * that previous transformations can handle the PNG_EXPAND_16 flag if this
    * is efficient (particularly true in the case of gamma correction, where
    * better accuracy results faster!)
    */
   if (png_ptr->transformations & PNG_EXPAND_16)
      png_do_expand_16(&png_ptr->row_info, png_ptr->row_buf + 1);
#endif
2183 2184

#ifdef PNG_READ_GRAY_TO_RGB_SUPPORTED
2185
   /*NOTE: moved here in 1.5.4 (from much later in this list.) */
2186 2187 2188 2189
   if ((png_ptr->transformations & PNG_GRAY_TO_RGB) &&
       (png_ptr->mode & PNG_BACKGROUND_IS_GRAY))
      png_do_gray_to_rgb(&(png_ptr->row_info), png_ptr->row_buf + 1);
#endif
2190

2191
#ifdef PNG_READ_INVERT_SUPPORTED
G
Guy Schalnat 已提交
2192
   if (png_ptr->transformations & PNG_INVERT_MONO)
G
Guy Schalnat 已提交
2193
      png_do_invert(&(png_ptr->row_info), png_ptr->row_buf + 1);
G
Guy Schalnat 已提交
2194 2195
#endif

2196
#ifdef PNG_READ_SHIFT_SUPPORTED
G
Guy Schalnat 已提交
2197 2198
   if (png_ptr->transformations & PNG_SHIFT)
      png_do_unshift(&(png_ptr->row_info), png_ptr->row_buf + 1,
2199
          &(png_ptr->shift));
G
Guy Schalnat 已提交
2200 2201
#endif

2202
#ifdef PNG_READ_PACK_SUPPORTED
G
Guy Schalnat 已提交
2203
   if (png_ptr->transformations & PNG_PACK)
G
Guy Schalnat 已提交
2204
      png_do_unpack(&(png_ptr->row_info), png_ptr->row_buf + 1);
G
Guy Schalnat 已提交
2205 2206
#endif

2207
#ifdef PNG_READ_BGR_SUPPORTED
G
Guy Schalnat 已提交
2208 2209
   if (png_ptr->transformations & PNG_BGR)
      png_do_bgr(&(png_ptr->row_info), png_ptr->row_buf + 1);
G
Guy Schalnat 已提交
2210 2211
#endif

2212
#ifdef PNG_READ_PACKSWAP_SUPPORTED
A
Andreas Dilger 已提交
2213 2214 2215 2216
   if (png_ptr->transformations & PNG_PACKSWAP)
      png_do_packswap(&(png_ptr->row_info), png_ptr->row_buf + 1);
#endif

2217
#ifdef PNG_READ_FILLER_SUPPORTED
G
Guy Schalnat 已提交
2218 2219
   if (png_ptr->transformations & PNG_FILLER)
      png_do_read_filler(&(png_ptr->row_info), png_ptr->row_buf + 1,
2220
          (png_uint_32)png_ptr->filler, png_ptr->flags);
A
Andreas Dilger 已提交
2221 2222
#endif

2223
#ifdef PNG_READ_INVERT_ALPHA_SUPPORTED
2224 2225 2226 2227
   if (png_ptr->transformations & PNG_INVERT_ALPHA)
      png_do_read_invert_alpha(&(png_ptr->row_info), png_ptr->row_buf + 1);
#endif

2228
#ifdef PNG_READ_SWAP_ALPHA_SUPPORTED
2229 2230 2231 2232
   if (png_ptr->transformations & PNG_SWAP_ALPHA)
      png_do_read_swap_alpha(&(png_ptr->row_info), png_ptr->row_buf + 1);
#endif

2233
#ifdef PNG_READ_16BIT_SUPPORTED
2234
#ifdef PNG_READ_SWAP_SUPPORTED
A
Andreas Dilger 已提交
2235 2236
   if (png_ptr->transformations & PNG_SWAP_BYTES)
      png_do_swap(&(png_ptr->row_info), png_ptr->row_buf + 1);
G
Guy Schalnat 已提交
2237
#endif
2238
#endif
2239

2240
#ifdef PNG_READ_USER_TRANSFORM_SUPPORTED
2241
   if (png_ptr->transformations & PNG_USER_TRANSFORM)
2242
    {
2243
      if (png_ptr->read_user_transform_fn != NULL)
2244
         (*(png_ptr->read_user_transform_fn)) /* User read transform function */
2245 2246 2247
             (png_ptr,                    /* png_ptr */
             &(png_ptr->row_info),     /* row_info: */
                /*  png_uint_32 width;       width of row */
2248
                /*  png_size_t rowbytes;     number of bytes in row */
2249 2250 2251 2252 2253
                /*  png_byte color_type;     color type of pixels */
                /*  png_byte bit_depth;      bit depth of samples */
                /*  png_byte channels;       number of channels (1-4) */
                /*  png_byte pixel_depth;    bits per pixel (depth*channels) */
             png_ptr->row_buf + 1);    /* start of pixel data for row */
2254
#ifdef PNG_USER_TRANSFORM_PTR_SUPPORTED
2255
      if (png_ptr->user_transform_depth)
2256
         png_ptr->row_info.bit_depth = png_ptr->user_transform_depth;
2257

2258
      if (png_ptr->user_transform_channels)
2259
         png_ptr->row_info.channels = png_ptr->user_transform_channels;
2260
#endif
2261
      png_ptr->row_info.pixel_depth = (png_byte)(png_ptr->row_info.bit_depth *
2262
          png_ptr->row_info.channels);
2263

2264
      png_ptr->row_info.rowbytes = PNG_ROWBYTES(png_ptr->row_info.pixel_depth,
2265
          png_ptr->row_info.width);
2266
   }
2267
#endif
G
Guy Schalnat 已提交
2268 2269
}

2270
#ifdef PNG_READ_PACK_SUPPORTED
2271 2272 2273 2274 2275 2276
/* Unpack pixels of 1, 2, or 4 bits per pixel into 1 byte per pixel,
 * without changing the actual values.  Thus, if you had a row with
 * a bit depth of 1, you would end up with bytes that only contained
 * the numbers 0 or 1.  If you would rather they contain 0 and 255, use
 * png_do_shift() after this.
 */
2277
void /* PRIVATE */
G
Guy Schalnat 已提交
2278
png_do_unpack(png_row_infop row_info, png_bytep row)
G
Guy Schalnat 已提交
2279
{
2280
   png_debug(1, "in png_do_unpack");
2281

A
Andreas Dilger 已提交
2282
   if (row_info->bit_depth < 8)
G
Guy Schalnat 已提交
2283
   {
2284
      png_uint_32 i;
2285
      png_uint_32 row_width=row_info->width;
A
Andreas Dilger 已提交
2286

G
Guy Schalnat 已提交
2287 2288 2289 2290
      switch (row_info->bit_depth)
      {
         case 1:
         {
2291 2292
            png_bytep sp = row + (png_size_t)((row_width - 1) >> 3);
            png_bytep dp = row + (png_size_t)row_width - 1;
2293
            png_uint_32 shift = 7 - (int)((row_width + 7) & 0x07);
2294
            for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
2295
            {
2296
               *dp = (png_byte)((*sp >> shift) & 0x01);
2297

G
Guy Schalnat 已提交
2298 2299 2300 2301
               if (shift == 7)
               {
                  shift = 0;
                  sp--;
G
Guy Schalnat 已提交
2302
               }
2303

G
Guy Schalnat 已提交
2304 2305 2306 2307 2308 2309 2310
               else
                  shift++;

               dp--;
            }
            break;
         }
2311

G
Guy Schalnat 已提交
2312 2313 2314
         case 2:
         {

2315 2316
            png_bytep sp = row + (png_size_t)((row_width - 1) >> 2);
            png_bytep dp = row + (png_size_t)row_width - 1;
2317
            png_uint_32 shift = (int)((3 - ((row_width + 3) & 0x03)) << 1);
2318
            for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
2319
            {
2320
               *dp = (png_byte)((*sp >> shift) & 0x03);
2321

G
Guy Schalnat 已提交
2322 2323 2324 2325 2326
               if (shift == 6)
               {
                  shift = 0;
                  sp--;
               }
2327

G
Guy Schalnat 已提交
2328 2329 2330 2331 2332 2333 2334
               else
                  shift += 2;

               dp--;
            }
            break;
         }
2335

G
Guy Schalnat 已提交
2336 2337
         case 4:
         {
2338 2339
            png_bytep sp = row + (png_size_t)((row_width - 1) >> 1);
            png_bytep dp = row + (png_size_t)row_width - 1;
2340
            png_uint_32 shift = (int)((1 - ((row_width + 1) & 0x01)) << 2);
2341
            for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
2342
            {
2343
               *dp = (png_byte)((*sp >> shift) & 0x0f);
2344

G
Guy Schalnat 已提交
2345 2346 2347 2348 2349
               if (shift == 4)
               {
                  shift = 0;
                  sp--;
               }
2350

G
Guy Schalnat 已提交
2351 2352 2353 2354 2355 2356 2357
               else
                  shift = 4;

               dp--;
            }
            break;
         }
2358 2359 2360

         default:
            break;
G
Guy Schalnat 已提交
2361 2362
      }
      row_info->bit_depth = 8;
G
Guy Schalnat 已提交
2363
      row_info->pixel_depth = (png_byte)(8 * row_info->channels);
2364
      row_info->rowbytes = row_width * row_info->channels;
G
Guy Schalnat 已提交
2365 2366
   }
}
G
Guy Schalnat 已提交
2367
#endif
G
Guy Schalnat 已提交
2368

2369
#ifdef PNG_READ_SHIFT_SUPPORTED
2370 2371 2372 2373 2374
/* Reverse the effects of png_do_shift.  This routine merely shifts the
 * pixels back to their significant bits values.  Thus, if you have
 * a row of bit depth 8, but only 5 are significant, this will shift
 * the values back to 0 through 31.
 */
2375
void /* PRIVATE */
2376 2377
png_do_unshift(png_row_infop row_info, png_bytep row,
    png_const_color_8p sig_bits)
G
Guy Schalnat 已提交
2378
{
2379
   png_debug(1, "in png_do_unshift");
2380

A
Andreas Dilger 已提交
2381 2382
   if (
       row_info->color_type != PNG_COLOR_TYPE_PALETTE)
G
Guy Schalnat 已提交
2383 2384
   {
      int shift[4];
2385 2386 2387
      int channels = 0;
      int c;
      png_uint_16 value = 0;
2388
      png_uint_32 row_width = row_info->width;
G
Guy Schalnat 已提交
2389 2390 2391

      if (row_info->color_type & PNG_COLOR_MASK_COLOR)
      {
G
Guy Schalnat 已提交
2392 2393 2394
         shift[channels++] = row_info->bit_depth - sig_bits->red;
         shift[channels++] = row_info->bit_depth - sig_bits->green;
         shift[channels++] = row_info->bit_depth - sig_bits->blue;
G
Guy Schalnat 已提交
2395
      }
2396

G
Guy Schalnat 已提交
2397 2398
      else
      {
G
Guy Schalnat 已提交
2399
         shift[channels++] = row_info->bit_depth - sig_bits->gray;
G
Guy Schalnat 已提交
2400
      }
2401

G
Guy Schalnat 已提交
2402 2403
      if (row_info->color_type & PNG_COLOR_MASK_ALPHA)
      {
G
Guy Schalnat 已提交
2404
         shift[channels++] = row_info->bit_depth - sig_bits->alpha;
G
Guy Schalnat 已提交
2405
      }
G
Guy Schalnat 已提交
2406

A
Andreas Dilger 已提交
2407
      for (c = 0; c < channels; c++)
G
Guy Schalnat 已提交
2408
      {
A
Andreas Dilger 已提交
2409 2410
         if (shift[c] <= 0)
            shift[c] = 0;
2411

G
Guy Schalnat 已提交
2412 2413 2414
         else
            value = 1;
      }
G
Guy Schalnat 已提交
2415

G
Guy Schalnat 已提交
2416 2417
      if (!value)
         return;
G
Guy Schalnat 已提交
2418

G
Guy Schalnat 已提交
2419
      switch (row_info->bit_depth)
G
Guy Schalnat 已提交
2420
      {
2421
         default:
2422
            break;
2423

G
Guy Schalnat 已提交
2424 2425
         case 2:
         {
A
Andreas Dilger 已提交
2426
            png_bytep bp;
2427 2428
            png_size_t i;
            png_size_t istop = row_info->rowbytes;
A
Andreas Dilger 已提交
2429

2430
            for (bp = row, i = 0; i < istop; i++)
G
Guy Schalnat 已提交
2431 2432
            {
               *bp >>= 1;
2433
               *bp++ &= 0x55;
G
Guy Schalnat 已提交
2434 2435 2436
            }
            break;
         }
2437

G
Guy Schalnat 已提交
2438 2439
         case 4:
         {
2440
            png_bytep bp = row;
2441 2442
            png_size_t i;
            png_size_t istop = row_info->rowbytes;
2443
            png_byte mask = (png_byte)((((int)0xf0 >> shift[0]) & (int)0xf0) |
2444
                (png_byte)((int)0xf >> shift[0]));
2445

2446
            for (i = 0; i < istop; i++)
G
Guy Schalnat 已提交
2447 2448
            {
               *bp >>= shift[0];
2449
               *bp++ &= mask;
G
Guy Schalnat 已提交
2450 2451 2452
            }
            break;
         }
2453

G
Guy Schalnat 已提交
2454 2455
         case 8:
         {
2456
            png_bytep bp = row;
2457
            png_uint_32 i;
2458
            png_uint_32 istop = row_width * channels;
G
Guy Schalnat 已提交
2459

2460
            for (i = 0; i < istop; i++)
A
Andreas Dilger 已提交
2461
            {
2462
               *bp++ >>= shift[i%channels];
G
Guy Schalnat 已提交
2463 2464 2465
            }
            break;
         }
2466

2467
#ifdef PNG_READ_16BIT_SUPPORTED
G
Guy Schalnat 已提交
2468 2469
         case 16:
         {
2470 2471 2472
            png_bytep bp = row;
            png_uint_32 i;
            png_uint_32 istop = channels * row_width;
G
Guy Schalnat 已提交
2473

2474
            for (i = 0; i < istop; i++)
A
Andreas Dilger 已提交
2475
            {
2476 2477 2478 2479
               value = (png_uint_16)((*bp << 8) + *(bp + 1));
               value >>= shift[i%channels];
               *bp++ = (png_byte)(value >> 8);
               *bp++ = (png_byte)(value & 0xff);
G
Guy Schalnat 已提交
2480 2481 2482
            }
            break;
         }
2483
#endif
G
Guy Schalnat 已提交
2484 2485 2486
      }
   }
}
G
Guy Schalnat 已提交
2487
#endif
G
Guy Schalnat 已提交
2488

2489
#ifdef PNG_READ_SCALE_16_TO_8_SUPPORTED
2490
/* Scale rows of bit depth 16 down to 8 accurately */
2491
void /* PRIVATE */
2492
png_do_scale_16_to_8(png_row_infop row_info, png_bytep row)
G
Guy Schalnat 已提交
2493
{
2494
   png_debug(1, "in png_do_scale_16_to_8");
2495

A
Andreas Dilger 已提交
2496
   if (row_info->bit_depth == 16)
G
Guy Schalnat 已提交
2497
   {
2498 2499 2500
      png_bytep sp = row; /* source */
      png_bytep dp = row; /* destinaton */
      png_bytep ep = sp + row_info->rowbytes; /* end+1 */
A
Andreas Dilger 已提交
2501

2502
      while (sp < ep)
G
Guy Schalnat 已提交
2503
      {
2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534
         /* The input is an array of 16 bit components, these must be scaled to
          * 8 bits each.  For a 16 bit value V the required value (from the PNG
          * specification) is:
          *
          *    (V * 255) / 65535
          *
          * This reduces to round(V / 257), or floor((V + 128.5)/257)
          *
          * Represent V as the two byte value vhi.vlo.  Make a guess that the
          * result is the top byte of V, vhi, then the correction to this value
          * is:
          *
          *    error = floor(((V-vhi.vhi) + 128.5) / 257)
          *          = floor(((vlo-vhi) + 128.5) / 257)
          *
          * This can be approximated using integer arithmetic (and a signed
          * shift):
          *
          *    error = (vlo-vhi+128) >> 8;
          *
          * The approximate differs from the exact answer only when (vlo-vhi) is
          * 128; it then gives a correction of +1 when the exact correction is
          * 0.  This gives 128 errors.  The exact answer (correct for all 16 bit
          * input values) is:
          *
          *    error = (vlo-vhi+128)*65535 >> 24;
          *
          * An alternative arithmetic calculation which also gives no errors is:
          *
          *    (V * 255 + 32895) >> 16
          */
2535

2536
         png_int_32 tmp = *sp++; /* must be signed! */
2537
         tmp += (((int)*sp++ - tmp + 128) * 65535) >> 24;
2538
         *dp++ = (png_byte)tmp;
G
Guy Schalnat 已提交
2539
      }
2540

G
Guy Schalnat 已提交
2541
      row_info->bit_depth = 8;
G
Guy Schalnat 已提交
2542
      row_info->pixel_depth = (png_byte)(8 * row_info->channels);
2543
      row_info->rowbytes = row_info->width * row_info->channels;
A
Andreas Dilger 已提交
2544 2545
   }
}
2546
#endif
2547

2548
#ifdef PNG_READ_STRIP_16_TO_8_SUPPORTED
2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573
void /* PRIVATE */
/* Simply discard the low byte.  This was the default behavior prior
 * to libpng-1.5.4.
 */
png_do_chop(png_row_infop row_info, png_bytep row)
{
   png_debug(1, "in png_do_chop");

   if (row_info->bit_depth == 16)
   {
      png_bytep sp = row; /* source */
      png_bytep dp = row; /* destinaton */
      png_bytep ep = sp + row_info->rowbytes; /* end+1 */

      while (sp < ep)
      {
         *dp++ = *sp++;
         sp++;
      }

      row_info->bit_depth = 8;
      row_info->pixel_depth = (png_byte)(8 * row_info->channels);
      row_info->rowbytes = row_info->width * row_info->channels;
   }
}
2574
#endif
A
Andreas Dilger 已提交
2575

2576
#ifdef PNG_READ_SWAP_ALPHA_SUPPORTED
2577
void /* PRIVATE */
A
Andreas Dilger 已提交
2578 2579
png_do_read_swap_alpha(png_row_infop row_info, png_bytep row)
{
2580
   png_debug(1, "in png_do_read_swap_alpha");
2581

A
Andreas Dilger 已提交
2582
   {
2583
      png_uint_32 row_width = row_info->width;
A
Andreas Dilger 已提交
2584 2585 2586 2587 2588
      if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
      {
         /* This converts from RGBA to ARGB */
         if (row_info->bit_depth == 8)
         {
2589 2590
            png_bytep sp = row + row_info->rowbytes;
            png_bytep dp = sp;
A
Andreas Dilger 已提交
2591 2592
            png_byte save;
            png_uint_32 i;
2593

2594
            for (i = 0; i < row_width; i++)
A
Andreas Dilger 已提交
2595 2596 2597 2598 2599 2600 2601 2602
            {
               save = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = save;
            }
         }
2603 2604

#ifdef PNG_READ_16BIT_SUPPORTED
A
Andreas Dilger 已提交
2605 2606 2607
         /* This converts from RRGGBBAA to AARRGGBB */
         else
         {
2608 2609
            png_bytep sp = row + row_info->rowbytes;
            png_bytep dp = sp;
A
Andreas Dilger 已提交
2610 2611
            png_byte save[2];
            png_uint_32 i;
2612

2613
            for (i = 0; i < row_width; i++)
A
Andreas Dilger 已提交
2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626
            {
               save[0] = *(--sp);
               save[1] = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = save[0];
               *(--dp) = save[1];
            }
         }
2627
#endif
A
Andreas Dilger 已提交
2628
      }
2629

A
Andreas Dilger 已提交
2630 2631 2632 2633 2634
      else if (row_info->color_type == PNG_COLOR_TYPE_GRAY_ALPHA)
      {
         /* This converts from GA to AG */
         if (row_info->bit_depth == 8)
         {
2635 2636
            png_bytep sp = row + row_info->rowbytes;
            png_bytep dp = sp;
A
Andreas Dilger 已提交
2637 2638
            png_byte save;
            png_uint_32 i;
2639

2640
            for (i = 0; i < row_width; i++)
A
Andreas Dilger 已提交
2641 2642 2643 2644 2645 2646
            {
               save = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = save;
            }
         }
2647 2648

#ifdef PNG_READ_16BIT_SUPPORTED
A
Andreas Dilger 已提交
2649 2650 2651
         /* This converts from GGAA to AAGG */
         else
         {
2652 2653
            png_bytep sp = row + row_info->rowbytes;
            png_bytep dp = sp;
A
Andreas Dilger 已提交
2654 2655 2656
            png_byte save[2];
            png_uint_32 i;

2657
            for (i = 0; i < row_width; i++)
A
Andreas Dilger 已提交
2658 2659 2660 2661 2662 2663 2664 2665 2666
            {
               save[0] = *(--sp);
               save[1] = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = save[0];
               *(--dp) = save[1];
            }
         }
2667
#endif
A
Andreas Dilger 已提交
2668
      }
G
Guy Schalnat 已提交
2669 2670
   }
}
G
Guy Schalnat 已提交
2671
#endif
G
Guy Schalnat 已提交
2672

2673
#ifdef PNG_READ_INVERT_ALPHA_SUPPORTED
2674
void /* PRIVATE */
2675 2676
png_do_read_invert_alpha(png_row_infop row_info, png_bytep row)
{
2677
   png_uint_32 row_width;
2678
   png_debug(1, "in png_do_read_invert_alpha");
2679

2680 2681
   row_width = row_info->width;
   if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
2682
   {
2683
      if (row_info->bit_depth == 8)
2684 2685
      {
         /* This inverts the alpha channel in RGBA */
2686 2687 2688
         png_bytep sp = row + row_info->rowbytes;
         png_bytep dp = sp;
         png_uint_32 i;
2689

2690 2691 2692
         for (i = 0; i < row_width; i++)
         {
            *(--dp) = (png_byte)(255 - *(--sp));
2693

2694 2695 2696 2697 2698
/*          This does nothing:
            *(--dp) = *(--sp);
            *(--dp) = *(--sp);
            *(--dp) = *(--sp);
            We can replace it with:
2699
*/
2700 2701
            sp-=3;
            dp=sp;
2702
         }
2703
      }
2704 2705

#ifdef PNG_READ_16BIT_SUPPORTED
2706 2707 2708 2709 2710 2711
      /* This inverts the alpha channel in RRGGBBAA */
      else
      {
         png_bytep sp = row + row_info->rowbytes;
         png_bytep dp = sp;
         png_uint_32 i;
2712

2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725
         for (i = 0; i < row_width; i++)
         {
            *(--dp) = (png_byte)(255 - *(--sp));
            *(--dp) = (png_byte)(255 - *(--sp));

/*          This does nothing:
            *(--dp) = *(--sp);
            *(--dp) = *(--sp);
            *(--dp) = *(--sp);
            *(--dp) = *(--sp);
            *(--dp) = *(--sp);
            *(--dp) = *(--sp);
            We can replace it with:
2726
*/
2727 2728
            sp-=6;
            dp=sp;
2729 2730
         }
      }
2731
#endif
2732 2733 2734 2735
   }
   else if (row_info->color_type == PNG_COLOR_TYPE_GRAY_ALPHA)
   {
      if (row_info->bit_depth == 8)
2736
      {
2737
         /* This inverts the alpha channel in GA */
2738 2739 2740
         png_bytep sp = row + row_info->rowbytes;
         png_bytep dp = sp;
         png_uint_32 i;
2741

2742 2743 2744 2745
         for (i = 0; i < row_width; i++)
         {
            *(--dp) = (png_byte)(255 - *(--sp));
            *(--dp) = *(--sp);
2746
         }
2747
      }
2748 2749

#ifdef PNG_READ_16BIT_SUPPORTED
2750 2751
      else
      {
2752
         /* This inverts the alpha channel in GGAA */
2753 2754 2755
         png_bytep sp  = row + row_info->rowbytes;
         png_bytep dp = sp;
         png_uint_32 i;
2756

2757 2758 2759 2760
         for (i = 0; i < row_width; i++)
         {
            *(--dp) = (png_byte)(255 - *(--sp));
            *(--dp) = (png_byte)(255 - *(--sp));
2761
/*
2762 2763
            *(--dp) = *(--sp);
            *(--dp) = *(--sp);
2764
*/
2765 2766
            sp-=2;
            dp=sp;
2767 2768
         }
      }
2769
#endif
2770 2771 2772 2773
   }
}
#endif

2774
#ifdef PNG_READ_FILLER_SUPPORTED
A
Andreas Dilger 已提交
2775
/* Add filler channel if we have RGB color */
2776
void /* PRIVATE */
G
Guy Schalnat 已提交
2777
png_do_read_filler(png_row_infop row_info, png_bytep row,
2778
    png_uint_32 filler, png_uint_32 flags)
G
Guy Schalnat 已提交
2779
{
G
Guy Schalnat 已提交
2780
   png_uint_32 i;
2781 2782
   png_uint_32 row_width = row_info->width;

2783
#ifdef PNG_READ_16BIT_SUPPORTED
2784
   png_byte hi_filler = (png_byte)((filler>>8) & 0xff);
2785
#endif
2786
   png_byte lo_filler = (png_byte)(filler & 0xff);
A
Andreas Dilger 已提交
2787

2788
   png_debug(1, "in png_do_read_filler");
2789

A
Andreas Dilger 已提交
2790
   if (
2791
       row_info->color_type == PNG_COLOR_TYPE_GRAY)
G
Guy Schalnat 已提交
2792
   {
2793
      if (row_info->bit_depth == 8)
G
Guy Schalnat 已提交
2794
      {
2795
         if (flags & PNG_FLAG_FILLER_AFTER)
G
Guy Schalnat 已提交
2796
         {
2797
            /* This changes the data from G to GX */
2798 2799
            png_bytep sp = row + (png_size_t)row_width;
            png_bytep dp =  sp + (png_size_t)row_width;
2800 2801
            for (i = 1; i < row_width; i++)
            {
2802
               *(--dp) = lo_filler;
2803 2804
               *(--dp) = *(--sp);
            }
2805
            *(--dp) = lo_filler;
2806 2807 2808 2809
            row_info->channels = 2;
            row_info->pixel_depth = 16;
            row_info->rowbytes = row_width * 2;
         }
2810

2811 2812
         else
         {
2813
            /* This changes the data from G to XG */
2814 2815
            png_bytep sp = row + (png_size_t)row_width;
            png_bytep dp = sp  + (png_size_t)row_width;
2816 2817 2818
            for (i = 0; i < row_width; i++)
            {
               *(--dp) = *(--sp);
2819
               *(--dp) = lo_filler;
2820 2821 2822 2823 2824 2825
            }
            row_info->channels = 2;
            row_info->pixel_depth = 16;
            row_info->rowbytes = row_width * 2;
         }
      }
2826

2827
#ifdef PNG_READ_16BIT_SUPPORTED
2828
      else if (row_info->bit_depth == 16)
2829 2830 2831
      {
         if (flags & PNG_FLAG_FILLER_AFTER)
         {
2832
            /* This changes the data from GG to GGXX */
2833 2834
            png_bytep sp = row + (png_size_t)row_width * 2;
            png_bytep dp = sp  + (png_size_t)row_width * 2;
2835 2836 2837
            for (i = 1; i < row_width; i++)
            {
               *(--dp) = hi_filler;
2838
               *(--dp) = lo_filler;
2839 2840 2841
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
            }
2842 2843
            *(--dp) = hi_filler;
            *(--dp) = lo_filler;
2844 2845
            row_info->channels = 2;
            row_info->pixel_depth = 32;
2846
            row_info->rowbytes = row_width * 4;
2847
         }
2848

2849 2850
         else
         {
2851
            /* This changes the data from GG to XXGG */
2852 2853
            png_bytep sp = row + (png_size_t)row_width * 2;
            png_bytep dp = sp  + (png_size_t)row_width * 2;
2854 2855 2856 2857 2858
            for (i = 0; i < row_width; i++)
            {
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = hi_filler;
2859
               *(--dp) = lo_filler;
2860 2861
            }
            row_info->channels = 2;
2862 2863
            row_info->pixel_depth = 32;
            row_info->rowbytes = row_width * 4;
G
Guy Schalnat 已提交
2864
         }
G
Guy Schalnat 已提交
2865
      }
2866
#endif
2867 2868 2869
   } /* COLOR_TYPE == GRAY */
   else if (row_info->color_type == PNG_COLOR_TYPE_RGB)
   {
2870
      if (row_info->bit_depth == 8)
2871 2872 2873
      {
         if (flags & PNG_FLAG_FILLER_AFTER)
         {
2874
            /* This changes the data from RGB to RGBX */
2875 2876
            png_bytep sp = row + (png_size_t)row_width * 3;
            png_bytep dp = sp  + (png_size_t)row_width;
2877 2878
            for (i = 1; i < row_width; i++)
            {
2879
               *(--dp) = lo_filler;
2880 2881 2882 2883
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
            }
2884
            *(--dp) = lo_filler;
2885 2886 2887 2888
            row_info->channels = 4;
            row_info->pixel_depth = 32;
            row_info->rowbytes = row_width * 4;
         }
2889

2890 2891
         else
         {
2892
            /* This changes the data from RGB to XRGB */
2893 2894
            png_bytep sp = row + (png_size_t)row_width * 3;
            png_bytep dp = sp + (png_size_t)row_width;
2895 2896 2897 2898 2899
            for (i = 0; i < row_width; i++)
            {
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
2900
               *(--dp) = lo_filler;
2901 2902 2903 2904 2905 2906
            }
            row_info->channels = 4;
            row_info->pixel_depth = 32;
            row_info->rowbytes = row_width * 4;
         }
      }
2907

2908
#ifdef PNG_READ_16BIT_SUPPORTED
2909
      else if (row_info->bit_depth == 16)
G
Guy Schalnat 已提交
2910
      {
2911
         if (flags & PNG_FLAG_FILLER_AFTER)
G
Guy Schalnat 已提交
2912
         {
2913
            /* This changes the data from RRGGBB to RRGGBBXX */
2914 2915
            png_bytep sp = row + (png_size_t)row_width * 6;
            png_bytep dp = sp  + (png_size_t)row_width * 2;
2916 2917 2918
            for (i = 1; i < row_width; i++)
            {
               *(--dp) = hi_filler;
2919
               *(--dp) = lo_filler;
2920 2921 2922 2923 2924 2925 2926
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
            }
2927 2928
            *(--dp) = hi_filler;
            *(--dp) = lo_filler;
2929 2930
            row_info->channels = 4;
            row_info->pixel_depth = 64;
2931
            row_info->rowbytes = row_width * 8;
2932
         }
2933

2934 2935
         else
         {
2936
            /* This changes the data from RRGGBB to XXRRGGBB */
2937 2938
            png_bytep sp = row + (png_size_t)row_width * 6;
            png_bytep dp = sp  + (png_size_t)row_width * 2;
2939 2940 2941 2942 2943 2944 2945 2946 2947
            for (i = 0; i < row_width; i++)
            {
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = *(--sp);
               *(--dp) = hi_filler;
2948
               *(--dp) = lo_filler;
2949
            }
2950

2951 2952
            row_info->channels = 4;
            row_info->pixel_depth = 64;
2953
            row_info->rowbytes = row_width * 8;
G
Guy Schalnat 已提交
2954
         }
G
Guy Schalnat 已提交
2955
      }
2956
#endif
2957
   } /* COLOR_TYPE == RGB */
G
Guy Schalnat 已提交
2958
}
G
Guy Schalnat 已提交
2959
#endif
G
Guy Schalnat 已提交
2960

2961
#ifdef PNG_READ_GRAY_TO_RGB_SUPPORTED
2962
/* Expand grayscale files to RGB, with or without alpha */
2963
void /* PRIVATE */
G
Guy Schalnat 已提交
2964
png_do_gray_to_rgb(png_row_infop row_info, png_bytep row)
G
Guy Schalnat 已提交
2965
{
G
Guy Schalnat 已提交
2966
   png_uint_32 i;
2967
   png_uint_32 row_width = row_info->width;
G
Guy Schalnat 已提交
2968

2969
   png_debug(1, "in png_do_gray_to_rgb");
2970

A
Andreas Dilger 已提交
2971
   if (row_info->bit_depth >= 8 &&
2972
       !(row_info->color_type & PNG_COLOR_MASK_COLOR))
G
Guy Schalnat 已提交
2973 2974 2975 2976 2977
   {
      if (row_info->color_type == PNG_COLOR_TYPE_GRAY)
      {
         if (row_info->bit_depth == 8)
         {
2978
            /* This changes G to RGB */
2979 2980
            png_bytep sp = row + (png_size_t)row_width - 1;
            png_bytep dp = sp  + (png_size_t)row_width * 2;
2981
            for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
2982 2983 2984
            {
               *(dp--) = *sp;
               *(dp--) = *sp;
2985
               *(dp--) = *(sp--);
G
Guy Schalnat 已提交
2986 2987
            }
         }
2988

G
Guy Schalnat 已提交
2989 2990
         else
         {
2991
            /* This changes GG to RRGGBB */
2992 2993
            png_bytep sp = row + (png_size_t)row_width * 2 - 1;
            png_bytep dp = sp  + (png_size_t)row_width * 4;
2994
            for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
2995 2996 2997 2998 2999
            {
               *(dp--) = *sp;
               *(dp--) = *(sp - 1);
               *(dp--) = *sp;
               *(dp--) = *(sp - 1);
3000 3001
               *(dp--) = *(sp--);
               *(dp--) = *(sp--);
G
Guy Schalnat 已提交
3002 3003 3004
            }
         }
      }
3005

G
Guy Schalnat 已提交
3006 3007 3008 3009
      else if (row_info->color_type == PNG_COLOR_TYPE_GRAY_ALPHA)
      {
         if (row_info->bit_depth == 8)
         {
3010
            /* This changes GA to RGBA */
3011 3012
            png_bytep sp = row + (png_size_t)row_width * 2 - 1;
            png_bytep dp = sp  + (png_size_t)row_width * 2;
3013
            for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
3014 3015 3016 3017
            {
               *(dp--) = *(sp--);
               *(dp--) = *sp;
               *(dp--) = *sp;
3018
               *(dp--) = *(sp--);
G
Guy Schalnat 已提交
3019 3020
            }
         }
3021

G
Guy Schalnat 已提交
3022 3023
         else
         {
3024
            /* This changes GGAA to RRGGBBAA */
3025 3026
            png_bytep sp = row + (png_size_t)row_width * 4 - 1;
            png_bytep dp = sp  + (png_size_t)row_width * 4;
3027
            for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
3028 3029 3030 3031 3032 3033 3034
            {
               *(dp--) = *(sp--);
               *(dp--) = *(sp--);
               *(dp--) = *sp;
               *(dp--) = *(sp - 1);
               *(dp--) = *sp;
               *(dp--) = *(sp - 1);
3035 3036
               *(dp--) = *(sp--);
               *(dp--) = *(sp--);
G
Guy Schalnat 已提交
3037 3038 3039
            }
         }
      }
G
Guy Schalnat 已提交
3040
      row_info->channels += (png_byte)2;
G
Guy Schalnat 已提交
3041
      row_info->color_type |= PNG_COLOR_MASK_COLOR;
G
Guy Schalnat 已提交
3042
      row_info->pixel_depth = (png_byte)(row_info->channels *
3043
          row_info->bit_depth);
3044
      row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth, row_width);
G
Guy Schalnat 已提交
3045 3046
   }
}
G
Guy Schalnat 已提交
3047
#endif
G
Guy Schalnat 已提交
3048

3049
#ifdef PNG_READ_RGB_TO_GRAY_SUPPORTED
3050
/* Reduce RGB files to grayscale, with or without alpha
3051
 * using the equation given in Poynton's ColorFAQ at
3052
 * <http://www.inforamp.net/~poynton/>  (THIS LINK IS DEAD June 2008)
3053
 * New link:
3054
 * <http://www.poynton.com/notes/colour_and_gamma/>
3055
 * Charles Poynton poynton at poynton.com
3056 3057 3058 3059
 *
 *     Y = 0.212671 * R + 0.715160 * G + 0.072169 * B
 *
 *  We approximate this with
3060
 *
3061
 *     Y = 0.21268 * R    + 0.7151 * G    + 0.07217 * B
3062 3063 3064
 *
 *  which can be expressed with integers as
 *
3065
 *     Y = (6969 * R + 23434 * G + 2365 * B)/32768
3066 3067 3068 3069 3070
 *
 *  The calculation is to be done in a linear colorspace.
 *
 *  Other integer coefficents can be used via png_set_rgb_to_gray().
 */
3071
int /* PRIVATE */
3072 3073 3074 3075 3076 3077 3078 3079
png_do_rgb_to_gray(png_structp png_ptr, png_row_infop row_info, png_bytep row)

{
   png_uint_32 i;

   png_uint_32 row_width = row_info->width;
   int rgb_error = 0;

3080
   png_debug(1, "in png_do_rgb_to_gray");
3081

3082
   if (!(row_info->color_type & PNG_COLOR_MASK_PALETTE) &&
3083
       (row_info->color_type & PNG_COLOR_MASK_COLOR))
3084
   {
3085 3086 3087
      png_uint_32 rc = png_ptr->rgb_to_gray_red_coeff;
      png_uint_32 gc = png_ptr->rgb_to_gray_green_coeff;
      png_uint_32 bc = png_ptr->rgb_to_gray_blue_coeff;
3088 3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103

      if (row_info->color_type == PNG_COLOR_TYPE_RGB)
      {
         if (row_info->bit_depth == 8)
         {
#if defined(PNG_READ_GAMMA_SUPPORTED) || defined(PNG_READ_BACKGROUND_SUPPORTED)
            if (png_ptr->gamma_from_1 != NULL && png_ptr->gamma_to_1 != NULL)
            {
               png_bytep sp = row;
               png_bytep dp = row;

               for (i = 0; i < row_width; i++)
               {
                  png_byte red   = png_ptr->gamma_to_1[*(sp++)];
                  png_byte green = png_ptr->gamma_to_1[*(sp++)];
                  png_byte blue  = png_ptr->gamma_to_1[*(sp++)];
3104

3105
                  if (red != green || red != blue)
3106 3107 3108
                  {
                     rgb_error |= 1;
                     *(dp++) = png_ptr->gamma_from_1[
3109
                         (rc*red + gc*green + bc*blue)>>15];
3110
                  }
3111

3112
                  else
3113
                     *(dp++) = *(sp - 1);
3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125
               }
            }
            else
#endif
            {
               png_bytep sp = row;
               png_bytep dp = row;
               for (i = 0; i < row_width; i++)
               {
                  png_byte red   = *(sp++);
                  png_byte green = *(sp++);
                  png_byte blue  = *(sp++);
3126

3127
                  if (red != green || red != blue)
3128 3129
                  {
                     rgb_error |= 1;
3130
                     *(dp++) = (png_byte)((rc*red + gc*green + bc*blue)>>15);
3131
                  }
3132

3133
                  else
3134
                     *(dp++) = *(sp - 1);
3135 3136 3137
               }
            }
         }
3138

3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150
         else /* RGB bit_depth == 16 */
         {
#if defined(PNG_READ_GAMMA_SUPPORTED) || defined(PNG_READ_BACKGROUND_SUPPORTED)
            if (png_ptr->gamma_16_to_1 != NULL &&
                png_ptr->gamma_16_from_1 != NULL)
            {
               png_bytep sp = row;
               png_bytep dp = row;
               for (i = 0; i < row_width; i++)
               {
                  png_uint_16 red, green, blue, w;

3151 3152 3153
                  red   = (png_uint_16)(((*(sp))<<8) | *(sp + 1)); sp += 2;
                  green = (png_uint_16)(((*(sp))<<8) | *(sp + 1)); sp += 2;
                  blue  = (png_uint_16)(((*(sp))<<8) | *(sp + 1)); sp += 2;
3154

3155
                  if (red == green && red == blue)
3156
                     w = red;
3157

3158 3159
                  else
                  {
3160 3161
                     png_uint_16 red_1   = png_ptr->gamma_16_to_1[(red&0xff)
                         >> png_ptr->gamma_shift][red>>8];
3162 3163
                     png_uint_16 green_1 =
                         png_ptr->gamma_16_to_1[(green&0xff) >>
3164
                         png_ptr->gamma_shift][green>>8];
3165 3166
                     png_uint_16 blue_1  = png_ptr->gamma_16_to_1[(blue&0xff)
                         >> png_ptr->gamma_shift][blue>>8];
3167
                     png_uint_16 gray16  = (png_uint_16)((rc*red_1 + gc*green_1
3168
                         + bc*blue_1)>>15);
3169 3170 3171 3172
                     w = png_ptr->gamma_16_from_1[(gray16&0xff) >>
                         png_ptr->gamma_shift][gray16 >> 8];
                     rgb_error |= 1;
                  }
3173

3174 3175
                  *(dp++) = (png_byte)((w>>8) & 0xff);
                  *(dp++) = (png_byte)(w & 0xff);
3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186
               }
            }
            else
#endif
            {
               png_bytep sp = row;
               png_bytep dp = row;
               for (i = 0; i < row_width; i++)
               {
                  png_uint_16 red, green, blue, gray16;

3187 3188 3189
                  red   = (png_uint_16)(((*(sp))<<8) | *(sp + 1)); sp += 2;
                  green = (png_uint_16)(((*(sp))<<8) | *(sp + 1)); sp += 2;
                  blue  = (png_uint_16)(((*(sp))<<8) | *(sp + 1)); sp += 2;
3190

3191
                  if (red != green || red != blue)
3192
                     rgb_error |= 1;
3193

3194
                  gray16  = (png_uint_16)((rc*red + gc*green + bc*blue)>>15);
3195 3196
                  *(dp++) = (png_byte)((gray16>>8) & 0xff);
                  *(dp++) = (png_byte)(gray16 & 0xff);
3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214
               }
            }
         }
      }
      if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
      {
         if (row_info->bit_depth == 8)
         {
#if defined(PNG_READ_GAMMA_SUPPORTED) || defined(PNG_READ_BACKGROUND_SUPPORTED)
            if (png_ptr->gamma_from_1 != NULL && png_ptr->gamma_to_1 != NULL)
            {
               png_bytep sp = row;
               png_bytep dp = row;
               for (i = 0; i < row_width; i++)
               {
                  png_byte red   = png_ptr->gamma_to_1[*(sp++)];
                  png_byte green = png_ptr->gamma_to_1[*(sp++)];
                  png_byte blue  = png_ptr->gamma_to_1[*(sp++)];
3215

3216
                  if (red != green || red != blue)
3217
                     rgb_error |= 1;
3218

3219
                  *(dp++) =  png_ptr->gamma_from_1
3220
                      [(rc*red + gc*green + bc*blue)>>15];
3221

3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234
                  *(dp++) = *(sp++);  /* alpha */
               }
            }
            else
#endif
            {
               png_bytep sp = row;
               png_bytep dp = row;
               for (i = 0; i < row_width; i++)
               {
                  png_byte red   = *(sp++);
                  png_byte green = *(sp++);
                  png_byte blue  = *(sp++);
3235
                  if (red != green || red != blue)
3236
                     rgb_error |= 1;
3237

3238
                  *(dp++) =  (png_byte)((rc*red + gc*green + bc*blue)>>15);
3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254
                  *(dp++) = *(sp++);  /* alpha */
               }
            }
         }
         else /* RGBA bit_depth == 16 */
         {
#if defined(PNG_READ_GAMMA_SUPPORTED) || defined(PNG_READ_BACKGROUND_SUPPORTED)
            if (png_ptr->gamma_16_to_1 != NULL &&
                png_ptr->gamma_16_from_1 != NULL)
            {
               png_bytep sp = row;
               png_bytep dp = row;
               for (i = 0; i < row_width; i++)
               {
                  png_uint_16 red, green, blue, w;

3255 3256 3257
                  red   = (png_uint_16)(((*(sp))<<8) | *(sp + 1)); sp += 2;
                  green = (png_uint_16)(((*(sp))<<8) | *(sp + 1)); sp += 2;
                  blue  = (png_uint_16)(((*(sp))<<8) | *(sp + 1)); sp += 2;
3258

3259
                  if (red == green && red == blue)
3260
                     w = red;
3261

3262 3263 3264
                  else
                  {
                     png_uint_16 red_1   = png_ptr->gamma_16_to_1[(red&0xff) >>
3265
                         png_ptr->gamma_shift][red>>8];
3266

3267 3268 3269
                     png_uint_16 green_1 =
                         png_ptr->gamma_16_to_1[(green&0xff) >>
                         png_ptr->gamma_shift][green>>8];
3270

3271
                     png_uint_16 blue_1  = png_ptr->gamma_16_to_1[(blue&0xff) >>
3272
                         png_ptr->gamma_shift][blue>>8];
3273

3274
                     png_uint_16 gray16  = (png_uint_16)((rc * red_1
3275
                         + gc * green_1 + bc * blue_1)>>15);
3276

3277 3278
                     w = png_ptr->gamma_16_from_1[(gray16&0xff) >>
                         png_ptr->gamma_shift][gray16 >> 8];
3279

3280 3281
                     rgb_error |= 1;
                  }
3282

3283 3284
                  *(dp++) = (png_byte)((w>>8) & 0xff);
                  *(dp++) = (png_byte)(w & 0xff);
3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296
                  *(dp++) = *(sp++);  /* alpha */
                  *(dp++) = *(sp++);
               }
            }
            else
#endif
            {
               png_bytep sp = row;
               png_bytep dp = row;
               for (i = 0; i < row_width; i++)
               {
                  png_uint_16 red, green, blue, gray16;
3297 3298 3299
                  red   = (png_uint_16)((*(sp)<<8) | *(sp + 1)); sp += 2;
                  green = (png_uint_16)((*(sp)<<8) | *(sp + 1)); sp += 2;
                  blue  = (png_uint_16)((*(sp)<<8) | *(sp + 1)); sp += 2;
3300

3301
                  if (red != green || red != blue)
3302
                     rgb_error |= 1;
3303

3304
                  gray16  = (png_uint_16)((rc*red + gc*green + bc*blue)>>15);
3305 3306
                  *(dp++) = (png_byte)((gray16>>8) & 0xff);
                  *(dp++) = (png_byte)(gray16 & 0xff);
3307 3308 3309 3310 3311 3312
                  *(dp++) = *(sp++);  /* alpha */
                  *(dp++) = *(sp++);
               }
            }
         }
      }
3313 3314 3315
      row_info->channels -= 2;
      row_info->color_type = (png_byte)(row_info->color_type &
          ~PNG_COLOR_MASK_COLOR);
3316
      row_info->pixel_depth = (png_byte)(row_info->channels *
3317
          row_info->bit_depth);
3318
      row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth, row_width);
3319 3320 3321 3322
   }
   return rgb_error;
}
#endif
3323
#endif /* PNG_READ_TRANSFORMS_SUPPORTED */
3324

3325
#ifdef PNG_BUILD_GRAYSCALE_PALETTE_SUPPORTED
3326 3327 3328
/* Build a grayscale palette.  Palette is assumed to be 1 << bit_depth
 * large of png_color.  This lets grayscale images be treated as
 * paletted.  Most useful for gamma correction and simplification
3329
 * of code.  This API is not used internally.
3330
 */
3331
void PNGAPI
G
Guy Schalnat 已提交
3332
png_build_grayscale_palette(int bit_depth, png_colorp palette)
G
Guy Schalnat 已提交
3333 3334 3335 3336 3337 3338
{
   int num_palette;
   int color_inc;
   int i;
   int v;

3339
   png_debug(1, "in png_do_build_grayscale_palette");
3340

A
Andreas Dilger 已提交
3341
   if (palette == NULL)
G
Guy Schalnat 已提交
3342 3343 3344 3345 3346 3347 3348 3349
      return;

   switch (bit_depth)
   {
      case 1:
         num_palette = 2;
         color_inc = 0xff;
         break;
3350

G
Guy Schalnat 已提交
3351 3352 3353 3354
      case 2:
         num_palette = 4;
         color_inc = 0x55;
         break;
3355

G
Guy Schalnat 已提交
3356 3357 3358 3359
      case 4:
         num_palette = 16;
         color_inc = 0x11;
         break;
3360

G
Guy Schalnat 已提交
3361 3362 3363 3364
      case 8:
         num_palette = 256;
         color_inc = 1;
         break;
3365

G
Guy Schalnat 已提交
3366
      default:
G
Guy Schalnat 已提交
3367
         num_palette = 0;
G
Guy Schalnat 已提交
3368
         color_inc = 0;
G
Guy Schalnat 已提交
3369 3370 3371 3372 3373
         break;
   }

   for (i = 0, v = 0; i < num_palette; i++, v += color_inc)
   {
G
Guy Schalnat 已提交
3374 3375 3376
      palette[i].red = (png_byte)v;
      palette[i].green = (png_byte)v;
      palette[i].blue = (png_byte)v;
G
Guy Schalnat 已提交
3377 3378
   }
}
3379
#endif
G
Guy Schalnat 已提交
3380 3381


3382
#ifdef PNG_READ_TRANSFORMS_SUPPORTED
3383
#ifdef PNG_READ_BACKGROUND_SUPPORTED
A
Andreas Dilger 已提交
3384
/* Replace any alpha or transparency with the supplied background color.
3385 3386 3387
 * "background" is already in the screen gamma, while "background_1" is
 * at a gamma of 1.0.  Paletted files have already been taken care of.
 */
3388
void /* PRIVATE */
3389
png_do_compose(png_row_infop row_info, png_bytep row, png_structp png_ptr)
3390
{
3391
#ifdef PNG_READ_GAMMA_SUPPORTED
3392 3393 3394 3395 3396 3397 3398
   png_const_bytep gamma_table = png_ptr->gamma_table;
   png_const_bytep gamma_from_1 = png_ptr->gamma_from_1;
   png_const_bytep gamma_to_1 = png_ptr->gamma_to_1;
   png_const_uint_16pp gamma_16 = png_ptr->gamma_16_table;
   png_const_uint_16pp gamma_16_from_1 = png_ptr->gamma_16_from_1;
   png_const_uint_16pp gamma_16_to_1 = png_ptr->gamma_16_to_1;
   int gamma_shift = png_ptr->gamma_shift;
3399
#endif
3400

3401
   png_bytep sp;
G
Guy Schalnat 已提交
3402
   png_uint_32 i;
3403
   png_uint_32 row_width = row_info->width;
3404
   int optimize = (png_ptr->flags & PNG_FLAG_OPTIMIZE_ALPHA) != 0;
G
Guy Schalnat 已提交
3405
   int shift;
G
Guy Schalnat 已提交
3406

3407
   png_debug(1, "in png_do_compose");
3408

G
Guy Schalnat 已提交
3409 3410 3411 3412 3413 3414 3415 3416 3417 3418 3419
   {
      switch (row_info->color_type)
      {
         case PNG_COLOR_TYPE_GRAY:
         {
            switch (row_info->bit_depth)
            {
               case 1:
               {
                  sp = row;
                  shift = 7;
3420
                  for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
3421
                  {
3422
                     if ((png_uint_16)((*sp >> shift) & 0x01)
3423
                        == png_ptr->trans_color.gray)
G
Guy Schalnat 已提交
3424
                     {
G
Guy Schalnat 已提交
3425
                        *sp &= (png_byte)((0x7f7f >> (7 - shift)) & 0xff);
3426
                        *sp |= (png_byte)(png_ptr->background.gray << shift);
G
Guy Schalnat 已提交
3427
                     }
3428

G
Guy Schalnat 已提交
3429 3430 3431 3432 3433
                     if (!shift)
                     {
                        shift = 7;
                        sp++;
                     }
3434

G
Guy Schalnat 已提交
3435 3436 3437 3438 3439
                     else
                        shift--;
                  }
                  break;
               }
3440

G
Guy Schalnat 已提交
3441 3442
               case 2:
               {
3443
#ifdef PNG_READ_GAMMA_SUPPORTED
3444
                  if (gamma_table != NULL)
G
Guy Schalnat 已提交
3445
                  {
3446 3447 3448
                     sp = row;
                     shift = 6;
                     for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
3449
                     {
3450
                        if ((png_uint_16)((*sp >> shift) & 0x03)
3451
                            == png_ptr->trans_color.gray)
3452 3453
                        {
                           *sp &= (png_byte)((0x3f3f >> (6 - shift)) & 0xff);
3454
                           *sp |= (png_byte)(png_ptr->background.gray << shift);
3455
                        }
3456

3457 3458
                        else
                        {
3459
                           png_byte p = (png_byte)((*sp >> shift) & 0x03);
3460
                           png_byte g = (png_byte)((gamma_table [p | (p << 2) |
3461
                               (p << 4) | (p << 6)] >> 6) & 0x03);
3462 3463 3464
                           *sp &= (png_byte)((0x3f3f >> (6 - shift)) & 0xff);
                           *sp |= (png_byte)(g << shift);
                        }
3465

3466 3467 3468 3469 3470
                        if (!shift)
                        {
                           shift = 6;
                           sp++;
                        }
3471

3472 3473
                        else
                           shift -= 2;
G
Guy Schalnat 已提交
3474
                     }
3475
                  }
3476

3477 3478 3479 3480 3481 3482
                  else
#endif
                  {
                     sp = row;
                     shift = 6;
                     for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
3483
                     {
3484
                        if ((png_uint_16)((*sp >> shift) & 0x03)
3485
                            == png_ptr->trans_color.gray)
3486 3487
                        {
                           *sp &= (png_byte)((0x3f3f >> (6 - shift)) & 0xff);
3488
                           *sp |= (png_byte)(png_ptr->background.gray << shift);
3489
                        }
3490

3491 3492 3493 3494 3495
                        if (!shift)
                        {
                           shift = 6;
                           sp++;
                        }
3496

3497 3498
                        else
                           shift -= 2;
G
Guy Schalnat 已提交
3499 3500 3501 3502
                     }
                  }
                  break;
               }
3503

G
Guy Schalnat 已提交
3504 3505
               case 4:
               {
3506
#ifdef PNG_READ_GAMMA_SUPPORTED
3507
                  if (gamma_table != NULL)
G
Guy Schalnat 已提交
3508
                  {
3509 3510 3511
                     sp = row;
                     shift = 4;
                     for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
3512
                     {
3513
                        if ((png_uint_16)((*sp >> shift) & 0x0f)
3514
                            == png_ptr->trans_color.gray)
3515 3516
                        {
                           *sp &= (png_byte)((0xf0f >> (4 - shift)) & 0xff);
3517
                           *sp |= (png_byte)(png_ptr->background.gray << shift);
3518
                        }
3519

3520 3521
                        else
                        {
3522
                           png_byte p = (png_byte)((*sp >> shift) & 0x0f);
3523
                           png_byte g = (png_byte)((gamma_table[p |
3524
                               (p << 4)] >> 4) & 0x0f);
3525 3526 3527
                           *sp &= (png_byte)((0xf0f >> (4 - shift)) & 0xff);
                           *sp |= (png_byte)(g << shift);
                        }
3528

3529 3530 3531 3532 3533
                        if (!shift)
                        {
                           shift = 4;
                           sp++;
                        }
3534

3535 3536
                        else
                           shift -= 4;
G
Guy Schalnat 已提交
3537
                     }
3538
                  }
3539

3540 3541 3542 3543 3544 3545
                  else
#endif
                  {
                     sp = row;
                     shift = 4;
                     for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
3546
                     {
3547
                        if ((png_uint_16)((*sp >> shift) & 0x0f)
3548
                            == png_ptr->trans_color.gray)
3549 3550
                        {
                           *sp &= (png_byte)((0xf0f >> (4 - shift)) & 0xff);
3551
                           *sp |= (png_byte)(png_ptr->background.gray << shift);
3552
                        }
3553

3554 3555 3556 3557 3558
                        if (!shift)
                        {
                           shift = 4;
                           sp++;
                        }
3559

3560 3561
                        else
                           shift -= 4;
G
Guy Schalnat 已提交
3562 3563 3564 3565
                     }
                  }
                  break;
               }
3566

G
Guy Schalnat 已提交
3567 3568
               case 8:
               {
3569
#ifdef PNG_READ_GAMMA_SUPPORTED
A
Andreas Dilger 已提交
3570
                  if (gamma_table != NULL)
G
Guy Schalnat 已提交
3571
                  {
3572 3573
                     sp = row;
                     for (i = 0; i < row_width; i++, sp++)
G
Guy Schalnat 已提交
3574
                     {
3575 3576
                        if (*sp == png_ptr->trans_color.gray)
                           *sp = (png_byte)png_ptr->background.gray;
3577

G
Guy Schalnat 已提交
3578 3579 3580 3581 3582
                        else
                           *sp = gamma_table[*sp];
                     }
                  }
                  else
G
Guy Schalnat 已提交
3583
#endif
G
Guy Schalnat 已提交
3584
                  {
3585 3586
                     sp = row;
                     for (i = 0; i < row_width; i++, sp++)
G
Guy Schalnat 已提交
3587
                     {
3588 3589
                        if (*sp == png_ptr->trans_color.gray)
                           *sp = (png_byte)png_ptr->background.gray;
G
Guy Schalnat 已提交
3590 3591 3592 3593
                     }
                  }
                  break;
               }
3594

G
Guy Schalnat 已提交
3595 3596
               case 16:
               {
3597
#ifdef PNG_READ_GAMMA_SUPPORTED
A
Andreas Dilger 已提交
3598
                  if (gamma_16 != NULL)
G
Guy Schalnat 已提交
3599
                  {
3600 3601
                     sp = row;
                     for (i = 0; i < row_width; i++, sp += 2)
G
Guy Schalnat 已提交
3602 3603 3604
                     {
                        png_uint_16 v;

3605
                        v = (png_uint_16)(((*sp) << 8) + *(sp + 1));
3606

3607
                        if (v == png_ptr->trans_color.gray)
G
Guy Schalnat 已提交
3608
                        {
3609
                           /* Background is already in screen gamma */
3610 3611
                           *sp = (png_byte)((png_ptr->background.gray >> 8) & 0xff);
                           *(sp + 1) = (png_byte)(png_ptr->background.gray & 0xff);
G
Guy Schalnat 已提交
3612
                        }
3613

G
Guy Schalnat 已提交
3614 3615
                        else
                        {
A
Andreas Dilger 已提交
3616
                           v = gamma_16[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
3617
                           *sp = (png_byte)((v >> 8) & 0xff);
G
Guy Schalnat 已提交
3618
                           *(sp + 1) = (png_byte)(v & 0xff);
G
Guy Schalnat 已提交
3619
                        }
G
Guy Schalnat 已提交
3620
                     }
G
Guy Schalnat 已提交
3621 3622
                  }
                  else
G
Guy Schalnat 已提交
3623
#endif
G
Guy Schalnat 已提交
3624
                  {
3625 3626
                     sp = row;
                     for (i = 0; i < row_width; i++, sp += 2)
G
Guy Schalnat 已提交
3627 3628 3629
                     {
                        png_uint_16 v;

3630
                        v = (png_uint_16)(((*sp) << 8) + *(sp + 1));
3631

3632
                        if (v == png_ptr->trans_color.gray)
G
Guy Schalnat 已提交
3633
                        {
3634 3635
                           *sp = (png_byte)((png_ptr->background.gray >> 8) & 0xff);
                           *(sp + 1) = (png_byte)(png_ptr->background.gray & 0xff);
G
Guy Schalnat 已提交
3636 3637 3638 3639 3640
                        }
                     }
                  }
                  break;
               }
3641 3642 3643

               default:
                  break;
G
Guy Schalnat 已提交
3644 3645 3646
            }
            break;
         }
3647

G
Guy Schalnat 已提交
3648 3649 3650 3651
         case PNG_COLOR_TYPE_RGB:
         {
            if (row_info->bit_depth == 8)
            {
3652
#ifdef PNG_READ_GAMMA_SUPPORTED
A
Andreas Dilger 已提交
3653
               if (gamma_table != NULL)
G
Guy Schalnat 已提交
3654
               {
3655 3656
                  sp = row;
                  for (i = 0; i < row_width; i++, sp += 3)
G
Guy Schalnat 已提交
3657
                  {
3658 3659 3660
                     if (*sp == png_ptr->trans_color.red &&
                         *(sp + 1) == png_ptr->trans_color.green &&
                         *(sp + 2) == png_ptr->trans_color.blue)
G
Guy Schalnat 已提交
3661
                     {
3662 3663 3664
                        *sp = (png_byte)png_ptr->background.red;
                        *(sp + 1) = (png_byte)png_ptr->background.green;
                        *(sp + 2) = (png_byte)png_ptr->background.blue;
G
Guy Schalnat 已提交
3665
                     }
3666

G
Guy Schalnat 已提交
3667 3668 3669 3670 3671 3672 3673 3674 3675
                     else
                     {
                        *sp = gamma_table[*sp];
                        *(sp + 1) = gamma_table[*(sp + 1)];
                        *(sp + 2) = gamma_table[*(sp + 2)];
                     }
                  }
               }
               else
G
Guy Schalnat 已提交
3676
#endif
G
Guy Schalnat 已提交
3677
               {
3678 3679
                  sp = row;
                  for (i = 0; i < row_width; i++, sp += 3)
G
Guy Schalnat 已提交
3680
                  {
3681 3682 3683
                     if (*sp == png_ptr->trans_color.red &&
                         *(sp + 1) == png_ptr->trans_color.green &&
                         *(sp + 2) == png_ptr->trans_color.blue)
G
Guy Schalnat 已提交
3684
                     {
3685 3686 3687
                        *sp = (png_byte)png_ptr->background.red;
                        *(sp + 1) = (png_byte)png_ptr->background.green;
                        *(sp + 2) = (png_byte)png_ptr->background.blue;
G
Guy Schalnat 已提交
3688 3689 3690 3691
                     }
                  }
               }
            }
A
Andreas Dilger 已提交
3692
            else /* if (row_info->bit_depth == 16) */
G
Guy Schalnat 已提交
3693
            {
3694
#ifdef PNG_READ_GAMMA_SUPPORTED
A
Andreas Dilger 已提交
3695
               if (gamma_16 != NULL)
G
Guy Schalnat 已提交
3696
               {
3697 3698
                  sp = row;
                  for (i = 0; i < row_width; i++, sp += 6)
G
Guy Schalnat 已提交
3699
                  {
3700
                     png_uint_16 r = (png_uint_16)(((*sp) << 8) + *(sp + 1));
3701

3702 3703
                     png_uint_16 g = (png_uint_16)(((*(sp + 2)) << 8)
                         + *(sp + 3));
3704

3705 3706
                     png_uint_16 b = (png_uint_16)(((*(sp + 4)) << 8)
                         + *(sp + 5));
3707

3708 3709 3710
                     if (r == png_ptr->trans_color.red &&
                         g == png_ptr->trans_color.green &&
                         b == png_ptr->trans_color.blue)
G
Guy Schalnat 已提交
3711
                     {
3712
                        /* Background is already in screen gamma */
3713 3714 3715 3716 3717 3718
                        *sp = (png_byte)((png_ptr->background.red >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(png_ptr->background.red & 0xff);
                        *(sp + 2) = (png_byte)((png_ptr->background.green >> 8) & 0xff);
                        *(sp + 3) = (png_byte)(png_ptr->background.green & 0xff);
                        *(sp + 4) = (png_byte)((png_ptr->background.blue >> 8) & 0xff);
                        *(sp + 5) = (png_byte)(png_ptr->background.blue & 0xff);
G
Guy Schalnat 已提交
3719
                     }
3720

G
Guy Schalnat 已提交
3721 3722
                     else
                     {
3723
                        png_uint_16 v = gamma_16[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
3724
                        *sp = (png_byte)((v >> 8) & 0xff);
G
Guy Schalnat 已提交
3725
                        *(sp + 1) = (png_byte)(v & 0xff);
3726

A
Andreas Dilger 已提交
3727
                        v = gamma_16[*(sp + 3) >> gamma_shift][*(sp + 2)];
G
Guy Schalnat 已提交
3728 3729
                        *(sp + 2) = (png_byte)((v >> 8) & 0xff);
                        *(sp + 3) = (png_byte)(v & 0xff);
3730

A
Andreas Dilger 已提交
3731
                        v = gamma_16[*(sp + 5) >> gamma_shift][*(sp + 4)];
G
Guy Schalnat 已提交
3732 3733
                        *(sp + 4) = (png_byte)((v >> 8) & 0xff);
                        *(sp + 5) = (png_byte)(v & 0xff);
G
Guy Schalnat 已提交
3734 3735 3736
                     }
                  }
               }
3737

G
Guy Schalnat 已提交
3738
               else
G
Guy Schalnat 已提交
3739
#endif
G
Guy Schalnat 已提交
3740
               {
3741 3742
                  sp = row;
                  for (i = 0; i < row_width; i++, sp += 6)
G
Guy Schalnat 已提交
3743
                  {
3744
                     png_uint_16 r = (png_uint_16)(((*sp) << 8) + *(sp + 1));
3745

3746 3747
                     png_uint_16 g = (png_uint_16)(((*(sp + 2)) << 8)
                         + *(sp + 3));
3748

3749 3750
                     png_uint_16 b = (png_uint_16)(((*(sp + 4)) << 8)
                         + *(sp + 5));
G
Guy Schalnat 已提交
3751

3752 3753 3754
                     if (r == png_ptr->trans_color.red &&
                         g == png_ptr->trans_color.green &&
                         b == png_ptr->trans_color.blue)
G
Guy Schalnat 已提交
3755
                     {
3756 3757 3758 3759 3760 3761
                        *sp = (png_byte)((png_ptr->background.red >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(png_ptr->background.red & 0xff);
                        *(sp + 2) = (png_byte)((png_ptr->background.green >> 8) & 0xff);
                        *(sp + 3) = (png_byte)(png_ptr->background.green & 0xff);
                        *(sp + 4) = (png_byte)((png_ptr->background.blue >> 8) & 0xff);
                        *(sp + 5) = (png_byte)(png_ptr->background.blue & 0xff);
G
Guy Schalnat 已提交
3762 3763 3764 3765 3766 3767
                     }
                  }
               }
            }
            break;
         }
3768

G
Guy Schalnat 已提交
3769
         case PNG_COLOR_TYPE_GRAY_ALPHA:
G
Guy Schalnat 已提交
3770
         {
A
Andreas Dilger 已提交
3771
            if (row_info->bit_depth == 8)
G
Guy Schalnat 已提交
3772
            {
3773
#ifdef PNG_READ_GAMMA_SUPPORTED
A
Andreas Dilger 已提交
3774 3775 3776
               if (gamma_to_1 != NULL && gamma_from_1 != NULL &&
                   gamma_table != NULL)
               {
3777
                  sp = row;
3778
                  for (i = 0; i < row_width; i++, sp += 2)
G
Guy Schalnat 已提交
3779
                  {
3780
                     png_uint_16 a = *(sp + 1);
G
Guy Schalnat 已提交
3781

A
Andreas Dilger 已提交
3782
                     if (a == 0xff)
3783
                        *sp = gamma_table[*sp];
3784

A
Andreas Dilger 已提交
3785 3786
                     else if (a == 0)
                     {
3787
                        /* Background is already in screen gamma */
3788
                        *sp = (png_byte)png_ptr->background.gray;
A
Andreas Dilger 已提交
3789
                     }
3790

A
Andreas Dilger 已提交
3791 3792 3793
                     else
                     {
                        png_byte v, w;
G
Guy Schalnat 已提交
3794

A
Andreas Dilger 已提交
3795
                        v = gamma_to_1[*sp];
3796
                        png_composite(w, v, a, png_ptr->background_1.gray);
3797 3798 3799
                        if (!optimize)
                           w = gamma_from_1[w];
                        *sp = w;
G
Guy Schalnat 已提交
3800 3801
                     }
                  }
A
Andreas Dilger 已提交
3802 3803
               }
               else
G
Guy Schalnat 已提交
3804
#endif
A
Andreas Dilger 已提交
3805
               {
3806
                  sp = row;
3807
                  for (i = 0; i < row_width; i++, sp += 2)
G
Guy Schalnat 已提交
3808
                  {
3809
                     png_byte a = *(sp + 1);
G
Guy Schalnat 已提交
3810

3811 3812
                     if (a == 0)
                        *sp = (png_byte)png_ptr->background.gray;
3813

3814 3815
                     else if (a < 0xff)
                        png_composite(*sp, *sp, a, png_ptr->background_1.gray);
G
Guy Schalnat 已提交
3816 3817
                  }
               }
A
Andreas Dilger 已提交
3818 3819 3820
            }
            else /* if (png_ptr->bit_depth == 16) */
            {
3821
#ifdef PNG_READ_GAMMA_SUPPORTED
A
Andreas Dilger 已提交
3822 3823 3824
               if (gamma_16 != NULL && gamma_16_from_1 != NULL &&
                   gamma_16_to_1 != NULL)
               {
3825
                  sp = row;
3826
                  for (i = 0; i < row_width; i++, sp += 4)
G
Guy Schalnat 已提交
3827
                  {
3828 3829
                     png_uint_16 a = (png_uint_16)(((*(sp + 2)) << 8)
                         + *(sp + 3));
A
Andreas Dilger 已提交
3830 3831

                     if (a == (png_uint_16)0xffff)
G
Guy Schalnat 已提交
3832
                     {
A
Andreas Dilger 已提交
3833
                        png_uint_16 v;
G
Guy Schalnat 已提交
3834

A
Andreas Dilger 已提交
3835
                        v = gamma_16[*(sp + 1) >> gamma_shift][*sp];
3836 3837
                        *sp = (png_byte)((v >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(v & 0xff);
A
Andreas Dilger 已提交
3838
                     }
3839

A
Andreas Dilger 已提交
3840 3841
                     else if (a == 0)
                     {
3842
                        /* Background is already in screen gamma */
3843 3844
                        *sp = (png_byte)((png_ptr->background.gray >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(png_ptr->background.gray & 0xff);
A
Andreas Dilger 已提交
3845
                     }
3846

A
Andreas Dilger 已提交
3847 3848 3849
                     else
                     {
                        png_uint_16 g, v, w;
G
Guy Schalnat 已提交
3850

A
Andreas Dilger 已提交
3851
                        g = gamma_16_to_1[*(sp + 1) >> gamma_shift][*sp];
3852
                        png_composite_16(v, g, a, png_ptr->background_1.gray);
3853 3854 3855 3856 3857 3858
                        if (optimize)
                           w = v;
                        else
                           w = gamma_16_from_1[(v&0xff) >> gamma_shift][v >> 8];
                        *sp = (png_byte)((w >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(w & 0xff);
G
Guy Schalnat 已提交
3859 3860
                     }
                  }
A
Andreas Dilger 已提交
3861 3862
               }
               else
G
Guy Schalnat 已提交
3863
#endif
A
Andreas Dilger 已提交
3864
               {
3865
                  sp = row;
3866
                  for (i = 0; i < row_width; i++, sp += 4)
G
Guy Schalnat 已提交
3867
                  {
3868 3869
                     png_uint_16 a = (png_uint_16)(((*(sp + 2)) << 8)
                         + *(sp + 3));
3870

3871
                     if (a == 0)
A
Andreas Dilger 已提交
3872
                     {
3873 3874
                        *sp = (png_byte)((png_ptr->background.gray >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(png_ptr->background.gray & 0xff);
A
Andreas Dilger 已提交
3875
                     }
3876

3877
                     else if (a < 0xffff)
A
Andreas Dilger 已提交
3878 3879
                     {
                        png_uint_16 g, v;
G
Guy Schalnat 已提交
3880

3881
                        g = (png_uint_16)(((*sp) << 8) + *(sp + 1));
3882
                        png_composite_16(v, g, a, png_ptr->background_1.gray);
3883 3884
                        *sp = (png_byte)((v >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(v & 0xff);
G
Guy Schalnat 已提交
3885 3886 3887 3888 3889 3890
                     }
                  }
               }
            }
            break;
         }
3891

G
Guy Schalnat 已提交
3892 3893 3894 3895
         case PNG_COLOR_TYPE_RGB_ALPHA:
         {
            if (row_info->bit_depth == 8)
            {
3896
#ifdef PNG_READ_GAMMA_SUPPORTED
A
Andreas Dilger 已提交
3897 3898
               if (gamma_to_1 != NULL && gamma_from_1 != NULL &&
                   gamma_table != NULL)
G
Guy Schalnat 已提交
3899
               {
3900
                  sp = row;
3901
                  for (i = 0; i < row_width; i++, sp += 4)
G
Guy Schalnat 已提交
3902
                  {
3903
                     png_byte a = *(sp + 3);
G
Guy Schalnat 已提交
3904 3905 3906

                     if (a == 0xff)
                     {
3907 3908 3909
                        *sp = gamma_table[*sp];
                        *(sp + 1) = gamma_table[*(sp + 1)];
                        *(sp + 2) = gamma_table[*(sp + 2)];
G
Guy Schalnat 已提交
3910
                     }
3911

G
Guy Schalnat 已提交
3912 3913
                     else if (a == 0)
                     {
3914
                        /* Background is already in screen gamma */
3915 3916 3917
                        *sp = (png_byte)png_ptr->background.red;
                        *(sp + 1) = (png_byte)png_ptr->background.green;
                        *(sp + 2) = (png_byte)png_ptr->background.blue;
G
Guy Schalnat 已提交
3918
                     }
3919

G
Guy Schalnat 已提交
3920 3921
                     else
                     {
A
Andreas Dilger 已提交
3922
                        png_byte v, w;
G
Guy Schalnat 已提交
3923 3924

                        v = gamma_to_1[*sp];
3925
                        png_composite(w, v, a, png_ptr->background_1.red);
3926 3927
                        if (!optimize) w = gamma_from_1[w];
                        *sp = w;
3928

G
Guy Schalnat 已提交
3929
                        v = gamma_to_1[*(sp + 1)];
3930
                        png_composite(w, v, a, png_ptr->background_1.green);
3931 3932
                        if (!optimize) w = gamma_from_1[w];
                        *(sp + 1) = w;
3933

G
Guy Schalnat 已提交
3934
                        v = gamma_to_1[*(sp + 2)];
3935
                        png_composite(w, v, a, png_ptr->background_1.blue);
3936 3937
                        if (!optimize) w = gamma_from_1[w];
                        *(sp + 2) = w;
G
Guy Schalnat 已提交
3938 3939 3940 3941
                     }
                  }
               }
               else
G
Guy Schalnat 已提交
3942
#endif
G
Guy Schalnat 已提交
3943
               {
3944
                  sp = row;
3945
                  for (i = 0; i < row_width; i++, sp += 4)
G
Guy Schalnat 已提交
3946
                  {
3947
                     png_byte a = *(sp + 3);
G
Guy Schalnat 已提交
3948

3949
                     if (a == 0)
G
Guy Schalnat 已提交
3950
                     {
3951 3952 3953
                        *sp = (png_byte)png_ptr->background.red;
                        *(sp + 1) = (png_byte)png_ptr->background.green;
                        *(sp + 2) = (png_byte)png_ptr->background.blue;
G
Guy Schalnat 已提交
3954
                     }
3955

3956
                     else if (a < 0xff)
G
Guy Schalnat 已提交
3957
                     {
3958
                        png_composite(*sp, *sp, a, png_ptr->background.red);
3959

3960
                        png_composite(*(sp + 1), *(sp + 1), a,
3961
                            png_ptr->background.green);
3962

3963
                        png_composite(*(sp + 2), *(sp + 2), a,
3964
                            png_ptr->background.blue);
G
Guy Schalnat 已提交
3965 3966 3967 3968
                     }
                  }
               }
            }
A
Andreas Dilger 已提交
3969
            else /* if (row_info->bit_depth == 16) */
G
Guy Schalnat 已提交
3970
            {
3971
#ifdef PNG_READ_GAMMA_SUPPORTED
A
Andreas Dilger 已提交
3972 3973
               if (gamma_16 != NULL && gamma_16_from_1 != NULL &&
                   gamma_16_to_1 != NULL)
G
Guy Schalnat 已提交
3974
               {
3975
                  sp = row;
3976
                  for (i = 0; i < row_width; i++, sp += 8)
G
Guy Schalnat 已提交
3977
                  {
3978 3979
                     png_uint_16 a = (png_uint_16)(((png_uint_16)(*(sp + 6))
                         << 8) + (png_uint_16)(*(sp + 7)));
3980

G
Guy Schalnat 已提交
3981 3982 3983 3984
                     if (a == (png_uint_16)0xffff)
                     {
                        png_uint_16 v;

A
Andreas Dilger 已提交
3985
                        v = gamma_16[*(sp + 1) >> gamma_shift][*sp];
3986 3987
                        *sp = (png_byte)((v >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(v & 0xff);
3988

A
Andreas Dilger 已提交
3989
                        v = gamma_16[*(sp + 3) >> gamma_shift][*(sp + 2)];
3990 3991
                        *(sp + 2) = (png_byte)((v >> 8) & 0xff);
                        *(sp + 3) = (png_byte)(v & 0xff);
3992

A
Andreas Dilger 已提交
3993
                        v = gamma_16[*(sp + 5) >> gamma_shift][*(sp + 4)];
3994 3995
                        *(sp + 4) = (png_byte)((v >> 8) & 0xff);
                        *(sp + 5) = (png_byte)(v & 0xff);
G
Guy Schalnat 已提交
3996
                     }
3997

G
Guy Schalnat 已提交
3998 3999
                     else if (a == 0)
                     {
4000
                        /* Background is already in screen gamma */
4001 4002 4003 4004 4005 4006
                        *sp = (png_byte)((png_ptr->background.red >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(png_ptr->background.red & 0xff);
                        *(sp + 2) = (png_byte)((png_ptr->background.green >> 8) & 0xff);
                        *(sp + 3) = (png_byte)(png_ptr->background.green & 0xff);
                        *(sp + 4) = (png_byte)((png_ptr->background.blue >> 8) & 0xff);
                        *(sp + 5) = (png_byte)(png_ptr->background.blue & 0xff);
G
Guy Schalnat 已提交
4007
                     }
4008

G
Guy Schalnat 已提交
4009 4010
                     else
                     {
4011
                        png_uint_16 v, w;
A
Andreas Dilger 已提交
4012 4013

                        v = gamma_16_to_1[*(sp + 1) >> gamma_shift][*sp];
4014
                        png_composite_16(w, v, a, png_ptr->background_1.red);
4015 4016 4017 4018
                        if (!optimize)
                           w = gamma_16_from_1[((w&0xff) >> gamma_shift)][w >> 8];
                        *sp = (png_byte)((w >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(w & 0xff);
4019

A
Andreas Dilger 已提交
4020
                        v = gamma_16_to_1[*(sp + 3) >> gamma_shift][*(sp + 2)];
4021
                        png_composite_16(w, v, a, png_ptr->background_1.green);
4022 4023
                        if (!optimize)
                           w = gamma_16_from_1[((w&0xff) >> gamma_shift)][w >> 8];
4024

4025 4026
                        *(sp + 2) = (png_byte)((w >> 8) & 0xff);
                        *(sp + 3) = (png_byte)(w & 0xff);
4027

A
Andreas Dilger 已提交
4028
                        v = gamma_16_to_1[*(sp + 5) >> gamma_shift][*(sp + 4)];
4029
                        png_composite_16(w, v, a, png_ptr->background_1.blue);
4030 4031
                        if (!optimize)
                           w = gamma_16_from_1[((w&0xff) >> gamma_shift)][w >> 8];
4032

4033 4034
                        *(sp + 4) = (png_byte)((w >> 8) & 0xff);
                        *(sp + 5) = (png_byte)(w & 0xff);
G
Guy Schalnat 已提交
4035 4036 4037
                     }
                  }
               }
4038

G
Guy Schalnat 已提交
4039
               else
G
Guy Schalnat 已提交
4040
#endif
G
Guy Schalnat 已提交
4041
               {
4042
                  sp = row;
4043
                  for (i = 0; i < row_width; i++, sp += 8)
G
Guy Schalnat 已提交
4044
                  {
4045
                     png_uint_16 a = (png_uint_16)(((png_uint_16)(*(sp + 6))
4046
                         << 8) + (png_uint_16)(*(sp + 7)));
4047

4048
                     if (a == 0)
G
Guy Schalnat 已提交
4049
                     {
4050 4051 4052 4053 4054 4055
                        *sp = (png_byte)((png_ptr->background.red >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(png_ptr->background.red & 0xff);
                        *(sp + 2) = (png_byte)((png_ptr->background.green >> 8) & 0xff);
                        *(sp + 3) = (png_byte)(png_ptr->background.green & 0xff);
                        *(sp + 4) = (png_byte)((png_ptr->background.blue >> 8) & 0xff);
                        *(sp + 5) = (png_byte)(png_ptr->background.blue & 0xff);
G
Guy Schalnat 已提交
4056
                     }
4057

4058
                     else if (a < 0xffff)
G
Guy Schalnat 已提交
4059
                     {
4060
                        png_uint_16 v;
A
Andreas Dilger 已提交
4061

4062 4063 4064 4065 4066
                        png_uint_16 r = (png_uint_16)(((*sp) << 8) + *(sp + 1));
                        png_uint_16 g = (png_uint_16)(((*(sp + 2)) << 8)
                            + *(sp + 3));
                        png_uint_16 b = (png_uint_16)(((*(sp + 4)) << 8)
                            + *(sp + 5));
A
Andreas Dilger 已提交
4067

4068
                        png_composite_16(v, r, a, png_ptr->background.red);
4069 4070
                        *sp = (png_byte)((v >> 8) & 0xff);
                        *(sp + 1) = (png_byte)(v & 0xff);
4071

4072
                        png_composite_16(v, g, a, png_ptr->background.green);
4073 4074
                        *(sp + 2) = (png_byte)((v >> 8) & 0xff);
                        *(sp + 3) = (png_byte)(v & 0xff);
4075

4076
                        png_composite_16(v, b, a, png_ptr->background.blue);
4077 4078
                        *(sp + 4) = (png_byte)((v >> 8) & 0xff);
                        *(sp + 5) = (png_byte)(v & 0xff);
G
Guy Schalnat 已提交
4079 4080 4081 4082 4083 4084
                     }
                  }
               }
            }
            break;
         }
4085 4086 4087

         default:
            break;
G
Guy Schalnat 已提交
4088 4089 4090
      }
   }
}
G
Guy Schalnat 已提交
4091
#endif
G
Guy Schalnat 已提交
4092

4093
#ifdef PNG_READ_GAMMA_SUPPORTED
A
Andreas Dilger 已提交
4094
/* Gamma correct the image, avoiding the alpha channel.  Make sure
4095
 * you do this after you deal with the transparency issue on grayscale
4096
 * or RGB images. If your bit depth is 8, use gamma_table, if it
4097 4098 4099
 * is 16, use gamma_16_table and gamma_shift.  Build these with
 * build_gamma_table().
 */
4100
void /* PRIVATE */
4101
png_do_gamma(png_row_infop row_info, png_bytep row, png_structp png_ptr)
G
Guy Schalnat 已提交
4102
{
4103 4104 4105 4106
   png_const_bytep gamma_table = png_ptr->gamma_table;
   png_const_uint_16pp gamma_16_table = png_ptr->gamma_16_table;
   int gamma_shift = png_ptr->gamma_shift;

G
Guy Schalnat 已提交
4107
   png_bytep sp;
G
Guy Schalnat 已提交
4108
   png_uint_32 i;
4109
   png_uint_32 row_width=row_info->width;
G
Guy Schalnat 已提交
4110

4111
   png_debug(1, "in png_do_gamma");
4112

4113
   if (((row_info->bit_depth <= 8 && gamma_table != NULL) ||
4114
       (row_info->bit_depth == 16 && gamma_16_table != NULL)))
G
Guy Schalnat 已提交
4115 4116 4117 4118 4119 4120 4121
   {
      switch (row_info->color_type)
      {
         case PNG_COLOR_TYPE_RGB:
         {
            if (row_info->bit_depth == 8)
            {
4122 4123
               sp = row;
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4124 4125 4126 4127 4128 4129 4130 4131 4132
               {
                  *sp = gamma_table[*sp];
                  sp++;
                  *sp = gamma_table[*sp];
                  sp++;
                  *sp = gamma_table[*sp];
                  sp++;
               }
            }
4133

A
Andreas Dilger 已提交
4134
            else /* if (row_info->bit_depth == 16) */
G
Guy Schalnat 已提交
4135
            {
4136 4137
               sp = row;
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4138 4139 4140
               {
                  png_uint_16 v;

A
Andreas Dilger 已提交
4141
                  v = gamma_16_table[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
4142 4143 4144
                  *sp = (png_byte)((v >> 8) & 0xff);
                  *(sp + 1) = (png_byte)(v & 0xff);
                  sp += 2;
4145

A
Andreas Dilger 已提交
4146
                  v = gamma_16_table[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
4147 4148
                  *sp = (png_byte)((v >> 8) & 0xff);
                  *(sp + 1) = (png_byte)(v & 0xff);
G
Guy Schalnat 已提交
4149
                  sp += 2;
4150

A
Andreas Dilger 已提交
4151
                  v = gamma_16_table[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
4152 4153 4154 4155 4156 4157 4158
                  *sp = (png_byte)((v >> 8) & 0xff);
                  *(sp + 1) = (png_byte)(v & 0xff);
                  sp += 2;
               }
            }
            break;
         }
4159

G
Guy Schalnat 已提交
4160 4161 4162 4163
         case PNG_COLOR_TYPE_RGB_ALPHA:
         {
            if (row_info->bit_depth == 8)
            {
4164 4165
               sp = row;
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4166
               {
G
Guy Schalnat 已提交
4167 4168
                  *sp = gamma_table[*sp];
                  sp++;
4169

G
Guy Schalnat 已提交
4170 4171
                  *sp = gamma_table[*sp];
                  sp++;
4172

G
Guy Schalnat 已提交
4173 4174
                  *sp = gamma_table[*sp];
                  sp++;
4175

G
Guy Schalnat 已提交
4176 4177 4178
                  sp++;
               }
            }
4179

A
Andreas Dilger 已提交
4180
            else /* if (row_info->bit_depth == 16) */
G
Guy Schalnat 已提交
4181
            {
4182 4183
               sp = row;
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4184
               {
4185
                  png_uint_16 v = gamma_16_table[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
4186 4187
                  *sp = (png_byte)((v >> 8) & 0xff);
                  *(sp + 1) = (png_byte)(v & 0xff);
G
Guy Schalnat 已提交
4188
                  sp += 2;
4189

A
Andreas Dilger 已提交
4190
                  v = gamma_16_table[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
4191 4192 4193
                  *sp = (png_byte)((v >> 8) & 0xff);
                  *(sp + 1) = (png_byte)(v & 0xff);
                  sp += 2;
4194

A
Andreas Dilger 已提交
4195
                  v = gamma_16_table[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
4196 4197 4198 4199 4200
                  *sp = (png_byte)((v >> 8) & 0xff);
                  *(sp + 1) = (png_byte)(v & 0xff);
                  sp += 4;
               }
            }
G
Guy Schalnat 已提交
4201 4202
            break;
         }
4203

G
Guy Schalnat 已提交
4204 4205 4206 4207
         case PNG_COLOR_TYPE_GRAY_ALPHA:
         {
            if (row_info->bit_depth == 8)
            {
4208 4209
               sp = row;
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4210 4211
               {
                  *sp = gamma_table[*sp];
A
Andreas Dilger 已提交
4212
                  sp += 2;
G
Guy Schalnat 已提交
4213 4214
               }
            }
4215

A
Andreas Dilger 已提交
4216
            else /* if (row_info->bit_depth == 16) */
G
Guy Schalnat 已提交
4217
            {
4218 4219
               sp = row;
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4220
               {
4221
                  png_uint_16 v = gamma_16_table[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
4222 4223
                  *sp = (png_byte)((v >> 8) & 0xff);
                  *(sp + 1) = (png_byte)(v & 0xff);
G
Guy Schalnat 已提交
4224 4225 4226 4227 4228
                  sp += 4;
               }
            }
            break;
         }
4229

G
Guy Schalnat 已提交
4230 4231
         case PNG_COLOR_TYPE_GRAY:
         {
4232 4233
            if (row_info->bit_depth == 2)
            {
4234 4235
               sp = row;
               for (i = 0; i < row_width; i += 4)
4236 4237 4238 4239 4240 4241
               {
                  int a = *sp & 0xc0;
                  int b = *sp & 0x30;
                  int c = *sp & 0x0c;
                  int d = *sp & 0x03;

4242
                  *sp = (png_byte)(
4243 4244 4245 4246
                      ((((int)gamma_table[a|(a>>2)|(a>>4)|(a>>6)])   ) & 0xc0)|
                      ((((int)gamma_table[(b<<2)|b|(b>>2)|(b>>4)])>>2) & 0x30)|
                      ((((int)gamma_table[(c<<4)|(c<<2)|c|(c>>2)])>>4) & 0x0c)|
                      ((((int)gamma_table[(d<<6)|(d<<4)|(d<<2)|d])>>6) ));
4247 4248 4249
                  sp++;
               }
            }
4250

A
Andreas Dilger 已提交
4251
            if (row_info->bit_depth == 4)
G
Guy Schalnat 已提交
4252
            {
4253 4254
               sp = row;
               for (i = 0; i < row_width; i += 2)
A
Andreas Dilger 已提交
4255 4256 4257 4258
               {
                  int msb = *sp & 0xf0;
                  int lsb = *sp & 0x0f;

4259
                  *sp = (png_byte)((((int)gamma_table[msb | (msb >> 4)]) & 0xf0)
4260
                      | (((int)gamma_table[(lsb << 4) | lsb]) >> 4));
A
Andreas Dilger 已提交
4261 4262 4263
                  sp++;
               }
            }
4264

A
Andreas Dilger 已提交
4265 4266
            else if (row_info->bit_depth == 8)
            {
4267 4268
               sp = row;
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4269 4270 4271 4272 4273
               {
                  *sp = gamma_table[*sp];
                  sp++;
               }
            }
4274

G
Guy Schalnat 已提交
4275 4276
            else if (row_info->bit_depth == 16)
            {
4277 4278
               sp = row;
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4279
               {
4280
                  png_uint_16 v = gamma_16_table[*(sp + 1) >> gamma_shift][*sp];
G
Guy Schalnat 已提交
4281 4282
                  *sp = (png_byte)((v >> 8) & 0xff);
                  *(sp + 1) = (png_byte)(v & 0xff);
G
Guy Schalnat 已提交
4283 4284 4285 4286 4287
                  sp += 2;
               }
            }
            break;
         }
4288 4289 4290

         default:
            break;
G
Guy Schalnat 已提交
4291 4292 4293
      }
   }
}
G
Guy Schalnat 已提交
4294
#endif
G
Guy Schalnat 已提交
4295

4296 4297 4298 4299 4300 4301 4302 4303 4304 4305 4306 4307 4308 4309 4310 4311 4312 4313 4314 4315 4316 4317 4318 4319 4320 4321 4322 4323 4324 4325 4326 4327 4328 4329 4330 4331 4332 4333 4334 4335 4336 4337 4338 4339 4340 4341 4342 4343 4344 4345 4346 4347 4348 4349 4350 4351 4352 4353 4354 4355 4356 4357 4358 4359 4360 4361 4362
#ifdef PNG_READ_ALPHA_MODE_SUPPORTED
/* Encode the alpha channel to the output gamma (the input channel is always
 * linear.)  Called only with color types that have an alpha channel.  Needs the
 * from_1 tables.
 */
void /* PRIVATE */
png_do_encode_alpha(png_row_infop row_info, png_bytep row, png_structp png_ptr)
{
   png_uint_32 row_width = row_info->width;

   png_debug(1, "in png_do_encode_alpha");

   if (row_info->color_type & PNG_COLOR_MASK_ALPHA)
   {
      if (row_info->bit_depth == 8)
      {
         PNG_CONST png_bytep table = png_ptr->gamma_from_1;

         if (table != NULL)
         {
            PNG_CONST int step =
               (row_info->color_type & PNG_COLOR_MASK_COLOR) ? 4 : 2;

            /* The alpha channel is the last component: */
            row += step - 1;

            for (; row_width > 0; --row_width, row += step)
               *row = table[*row];

            return;
         }
      }

      else if (row_info->bit_depth == 16)
      {
         PNG_CONST png_uint_16pp table = png_ptr->gamma_16_from_1;
         PNG_CONST int gamma_shift = png_ptr->gamma_shift;

         if (table != NULL)
         {
            PNG_CONST int step =
               (row_info->color_type & PNG_COLOR_MASK_COLOR) ? 8 : 4;

            /* The alpha channel is the last component: */
            row += step - 2;

            for (; row_width > 0; --row_width, row += step)
            {
               png_uint_16 v;

               v = table[*(row + 1) >> gamma_shift][*row];
               *row = (png_byte)((v >> 8) & 0xff);
               *(row + 1) = (png_byte)(v & 0xff);
            }

            return;
         }
      }
   }

   /* Only get to here if called with a weird row_info; no harm has been done,
    * so just issue a warning.
    */
   png_warning(png_ptr, "png_do_encode_alpha: unexpected call");
}
#endif

4363
#ifdef PNG_READ_EXPAND_SUPPORTED
4364
/* Expands a palette row to an RGB or RGBA row depending
4365 4366
 * upon whether you supply trans and num_trans.
 */
4367
void /* PRIVATE */
G
Guy Schalnat 已提交
4368
png_do_expand_palette(png_row_infop row_info, png_bytep row,
4369
   png_const_colorp palette, png_const_bytep trans_alpha, int num_trans)
G
Guy Schalnat 已提交
4370
{
G
Guy Schalnat 已提交
4371
   int shift, value;
G
Guy Schalnat 已提交
4372
   png_bytep sp, dp;
G
Guy Schalnat 已提交
4373
   png_uint_32 i;
4374
   png_uint_32 row_width=row_info->width;
G
Guy Schalnat 已提交
4375

4376
   png_debug(1, "in png_do_expand_palette");
4377

4378
   if (row_info->color_type == PNG_COLOR_TYPE_PALETTE)
G
Guy Schalnat 已提交
4379 4380 4381 4382 4383 4384 4385
   {
      if (row_info->bit_depth < 8)
      {
         switch (row_info->bit_depth)
         {
            case 1:
            {
4386 4387
               sp = row + (png_size_t)((row_width - 1) >> 3);
               dp = row + (png_size_t)row_width - 1;
4388
               shift = 7 - (int)((row_width + 7) & 0x07);
4389
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4390
               {
4391
                  if ((*sp >> shift) & 0x01)
G
Guy Schalnat 已提交
4392
                     *dp = 1;
4393

G
Guy Schalnat 已提交
4394 4395
                  else
                     *dp = 0;
4396

G
Guy Schalnat 已提交
4397 4398 4399 4400 4401
                  if (shift == 7)
                  {
                     shift = 0;
                     sp--;
                  }
4402

G
Guy Schalnat 已提交
4403 4404 4405 4406 4407 4408 4409
                  else
                     shift++;

                  dp--;
               }
               break;
            }
4410

G
Guy Schalnat 已提交
4411 4412
            case 2:
            {
4413 4414
               sp = row + (png_size_t)((row_width - 1) >> 2);
               dp = row + (png_size_t)row_width - 1;
4415
               shift = (int)((3 - ((row_width + 3) & 0x03)) << 1);
4416
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4417
               {
4418
                  value = (*sp >> shift) & 0x03;
G
Guy Schalnat 已提交
4419
                  *dp = (png_byte)value;
G
Guy Schalnat 已提交
4420 4421 4422 4423 4424
                  if (shift == 6)
                  {
                     shift = 0;
                     sp--;
                  }
4425

G
Guy Schalnat 已提交
4426 4427 4428 4429 4430 4431 4432
                  else
                     shift += 2;

                  dp--;
               }
               break;
            }
4433

G
Guy Schalnat 已提交
4434 4435
            case 4:
            {
4436 4437
               sp = row + (png_size_t)((row_width - 1) >> 1);
               dp = row + (png_size_t)row_width - 1;
4438
               shift = (int)((row_width & 0x01) << 2);
4439
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4440
               {
4441
                  value = (*sp >> shift) & 0x0f;
G
Guy Schalnat 已提交
4442
                  *dp = (png_byte)value;
G
Guy Schalnat 已提交
4443 4444 4445 4446 4447
                  if (shift == 4)
                  {
                     shift = 0;
                     sp--;
                  }
4448

G
Guy Schalnat 已提交
4449 4450 4451 4452 4453 4454 4455
                  else
                     shift += 4;

                  dp--;
               }
               break;
            }
4456 4457 4458

            default:
               break;
G
Guy Schalnat 已提交
4459 4460 4461
         }
         row_info->bit_depth = 8;
         row_info->pixel_depth = 8;
4462
         row_info->rowbytes = row_width;
G
Guy Schalnat 已提交
4463
      }
4464

4465
      if (row_info->bit_depth == 8)
G
Guy Schalnat 已提交
4466 4467
      {
         {
4468
            if (num_trans > 0)
G
Guy Schalnat 已提交
4469
            {
4470 4471
               sp = row + (png_size_t)row_width - 1;
               dp = row + (png_size_t)(row_width << 2) - 1;
G
Guy Schalnat 已提交
4472

4473
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4474
               {
A
Andreas Dilger 已提交
4475
                  if ((int)(*sp) >= num_trans)
G
Guy Schalnat 已提交
4476
                     *dp-- = 0xff;
4477

G
Guy Schalnat 已提交
4478
                  else
4479
                     *dp-- = trans_alpha[*sp];
4480

G
Guy Schalnat 已提交
4481 4482 4483 4484 4485 4486 4487
                  *dp-- = palette[*sp].blue;
                  *dp-- = palette[*sp].green;
                  *dp-- = palette[*sp].red;
                  sp--;
               }
               row_info->bit_depth = 8;
               row_info->pixel_depth = 32;
4488
               row_info->rowbytes = row_width * 4;
G
Guy Schalnat 已提交
4489 4490 4491
               row_info->color_type = 6;
               row_info->channels = 4;
            }
4492

G
Guy Schalnat 已提交
4493 4494
            else
            {
4495 4496
               sp = row + (png_size_t)row_width - 1;
               dp = row + (png_size_t)(row_width * 3) - 1;
G
Guy Schalnat 已提交
4497

4498
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4499 4500 4501 4502 4503 4504
               {
                  *dp-- = palette[*sp].blue;
                  *dp-- = palette[*sp].green;
                  *dp-- = palette[*sp].red;
                  sp--;
               }
4505

G
Guy Schalnat 已提交
4506 4507
               row_info->bit_depth = 8;
               row_info->pixel_depth = 24;
4508
               row_info->rowbytes = row_width * 3;
G
Guy Schalnat 已提交
4509 4510 4511 4512 4513 4514 4515 4516
               row_info->color_type = 2;
               row_info->channels = 3;
            }
         }
      }
   }
}

4517 4518
/* If the bit depth < 8, it is expanded to 8.  Also, if the already
 * expanded transparency value is supplied, an alpha channel is built.
4519
 */
4520
void /* PRIVATE */
G
Guy Schalnat 已提交
4521
png_do_expand(png_row_infop row_info, png_bytep row,
4522
    png_const_color_16p trans_color)
G
Guy Schalnat 已提交
4523
{
G
Guy Schalnat 已提交
4524
   int shift, value;
G
Guy Schalnat 已提交
4525
   png_bytep sp, dp;
G
Guy Schalnat 已提交
4526
   png_uint_32 i;
4527
   png_uint_32 row_width=row_info->width;
G
Guy Schalnat 已提交
4528

4529
   png_debug(1, "in png_do_expand");
4530

G
Guy Schalnat 已提交
4531
   {
A
Andreas Dilger 已提交
4532
      if (row_info->color_type == PNG_COLOR_TYPE_GRAY)
G
Guy Schalnat 已提交
4533
      {
4534
         png_uint_16 gray = (png_uint_16)(trans_color ? trans_color->gray : 0);
A
Andreas Dilger 已提交
4535 4536

         if (row_info->bit_depth < 8)
G
Guy Schalnat 已提交
4537
         {
A
Andreas Dilger 已提交
4538
            switch (row_info->bit_depth)
G
Guy Schalnat 已提交
4539
            {
A
Andreas Dilger 已提交
4540
               case 1:
G
Guy Schalnat 已提交
4541
               {
4542
                  gray = (png_uint_16)((gray & 0x01) * 0xff);
4543 4544
                  sp = row + (png_size_t)((row_width - 1) >> 3);
                  dp = row + (png_size_t)row_width - 1;
4545
                  shift = 7 - (int)((row_width + 7) & 0x07);
4546
                  for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4547
                  {
4548
                     if ((*sp >> shift) & 0x01)
A
Andreas Dilger 已提交
4549
                        *dp = 0xff;
4550

A
Andreas Dilger 已提交
4551 4552
                     else
                        *dp = 0;
4553

A
Andreas Dilger 已提交
4554 4555 4556 4557 4558
                     if (shift == 7)
                     {
                        shift = 0;
                        sp--;
                     }
4559

A
Andreas Dilger 已提交
4560 4561
                     else
                        shift++;
G
Guy Schalnat 已提交
4562

A
Andreas Dilger 已提交
4563 4564 4565
                     dp--;
                  }
                  break;
G
Guy Schalnat 已提交
4566
               }
4567

A
Andreas Dilger 已提交
4568
               case 2:
G
Guy Schalnat 已提交
4569
               {
4570
                  gray = (png_uint_16)((gray & 0x03) * 0x55);
4571 4572
                  sp = row + (png_size_t)((row_width - 1) >> 2);
                  dp = row + (png_size_t)row_width - 1;
4573
                  shift = (int)((3 - ((row_width + 3) & 0x03)) << 1);
4574
                  for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4575
                  {
4576
                     value = (*sp >> shift) & 0x03;
A
Andreas Dilger 已提交
4577 4578 4579 4580 4581 4582 4583
                     *dp = (png_byte)(value | (value << 2) | (value << 4) |
                        (value << 6));
                     if (shift == 6)
                     {
                        shift = 0;
                        sp--;
                     }
4584

A
Andreas Dilger 已提交
4585 4586
                     else
                        shift += 2;
G
Guy Schalnat 已提交
4587

A
Andreas Dilger 已提交
4588 4589 4590
                     dp--;
                  }
                  break;
G
Guy Schalnat 已提交
4591
               }
4592

A
Andreas Dilger 已提交
4593
               case 4:
G
Guy Schalnat 已提交
4594
               {
4595
                  gray = (png_uint_16)((gray & 0x0f) * 0x11);
4596 4597
                  sp = row + (png_size_t)((row_width - 1) >> 1);
                  dp = row + (png_size_t)row_width - 1;
4598
                  shift = (int)((1 - ((row_width + 1) & 0x01)) << 2);
4599
                  for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4600
                  {
4601
                     value = (*sp >> shift) & 0x0f;
A
Andreas Dilger 已提交
4602 4603 4604 4605 4606 4607
                     *dp = (png_byte)(value | (value << 4));
                     if (shift == 4)
                     {
                        shift = 0;
                        sp--;
                     }
4608

A
Andreas Dilger 已提交
4609 4610
                     else
                        shift = 4;
G
Guy Schalnat 已提交
4611

A
Andreas Dilger 已提交
4612 4613 4614
                     dp--;
                  }
                  break;
G
Guy Schalnat 已提交
4615
               }
4616 4617 4618

               default:
                  break;
G
Guy Schalnat 已提交
4619
            }
4620

A
Andreas Dilger 已提交
4621 4622
            row_info->bit_depth = 8;
            row_info->pixel_depth = 8;
4623
            row_info->rowbytes = row_width;
G
Guy Schalnat 已提交
4624
         }
A
Andreas Dilger 已提交
4625

4626
         if (trans_color != NULL)
G
Guy Schalnat 已提交
4627
         {
A
Andreas Dilger 已提交
4628
            if (row_info->bit_depth == 8)
G
Guy Schalnat 已提交
4629
            {
4630
               gray = gray & 0xff;
4631 4632
               sp = row + (png_size_t)row_width - 1;
               dp = row + (png_size_t)(row_width << 1) - 1;
4633

4634
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4635
               {
A
Andreas Dilger 已提交
4636 4637
                  if (*sp == gray)
                     *dp-- = 0;
4638

A
Andreas Dilger 已提交
4639 4640
                  else
                     *dp-- = 0xff;
4641

A
Andreas Dilger 已提交
4642
                  *dp-- = *sp--;
G
Guy Schalnat 已提交
4643
               }
A
Andreas Dilger 已提交
4644
            }
4645

A
Andreas Dilger 已提交
4646 4647
            else if (row_info->bit_depth == 16)
            {
4648 4649
               png_byte gray_high = (png_byte)((gray >> 8) & 0xff);
               png_byte gray_low = (png_byte)(gray & 0xff);
A
Andreas Dilger 已提交
4650 4651
               sp = row + row_info->rowbytes - 1;
               dp = row + (row_info->rowbytes << 1) - 1;
4652
               for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4653
               {
4654
                  if (*(sp - 1) == gray_high && *(sp) == gray_low)
A
Andreas Dilger 已提交
4655 4656 4657 4658
                  {
                     *dp-- = 0;
                     *dp-- = 0;
                  }
4659

A
Andreas Dilger 已提交
4660 4661 4662 4663 4664
                  else
                  {
                     *dp-- = 0xff;
                     *dp-- = 0xff;
                  }
4665

A
Andreas Dilger 已提交
4666 4667
                  *dp-- = *sp--;
                  *dp-- = *sp--;
G
Guy Schalnat 已提交
4668 4669
               }
            }
4670

A
Andreas Dilger 已提交
4671 4672 4673
            row_info->color_type = PNG_COLOR_TYPE_GRAY_ALPHA;
            row_info->channels = 2;
            row_info->pixel_depth = (png_byte)(row_info->bit_depth << 1);
4674 4675
            row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth,
               row_width);
G
Guy Schalnat 已提交
4676 4677
         }
      }
4678
      else if (row_info->color_type == PNG_COLOR_TYPE_RGB && trans_color)
G
Guy Schalnat 已提交
4679 4680 4681
      {
         if (row_info->bit_depth == 8)
         {
4682 4683 4684
            png_byte red = (png_byte)(trans_color->red & 0xff);
            png_byte green = (png_byte)(trans_color->green & 0xff);
            png_byte blue = (png_byte)(trans_color->blue & 0xff);
G
Guy Schalnat 已提交
4685
            sp = row + (png_size_t)row_info->rowbytes - 1;
4686 4687
            dp = row + (png_size_t)(row_width << 2) - 1;
            for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4688
            {
4689
               if (*(sp - 2) == red && *(sp - 1) == green && *(sp) == blue)
G
Guy Schalnat 已提交
4690
                  *dp-- = 0;
4691

G
Guy Schalnat 已提交
4692 4693
               else
                  *dp-- = 0xff;
4694

G
Guy Schalnat 已提交
4695 4696 4697 4698 4699 4700 4701
               *dp-- = *sp--;
               *dp-- = *sp--;
               *dp-- = *sp--;
            }
         }
         else if (row_info->bit_depth == 16)
         {
4702 4703 4704 4705 4706 4707
            png_byte red_high = (png_byte)((trans_color->red >> 8) & 0xff);
            png_byte green_high = (png_byte)((trans_color->green >> 8) & 0xff);
            png_byte blue_high = (png_byte)((trans_color->blue >> 8) & 0xff);
            png_byte red_low = (png_byte)(trans_color->red & 0xff);
            png_byte green_low = (png_byte)(trans_color->green & 0xff);
            png_byte blue_low = (png_byte)(trans_color->blue & 0xff);
A
Andreas Dilger 已提交
4708
            sp = row + row_info->rowbytes - 1;
4709 4710
            dp = row + (png_size_t)(row_width << 3) - 1;
            for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4711
            {
4712
               if (*(sp - 5) == red_high &&
4713 4714 4715 4716 4717
                   *(sp - 4) == red_low &&
                   *(sp - 3) == green_high &&
                   *(sp - 2) == green_low &&
                   *(sp - 1) == blue_high &&
                   *(sp    ) == blue_low)
G
Guy Schalnat 已提交
4718 4719 4720 4721
               {
                  *dp-- = 0;
                  *dp-- = 0;
               }
4722

G
Guy Schalnat 已提交
4723 4724 4725 4726 4727
               else
               {
                  *dp-- = 0xff;
                  *dp-- = 0xff;
               }
4728

G
Guy Schalnat 已提交
4729 4730 4731 4732
               *dp-- = *sp--;
               *dp-- = *sp--;
               *dp-- = *sp--;
               *dp-- = *sp--;
G
Guy Schalnat 已提交
4733
               *dp-- = *sp--;
G
Guy Schalnat 已提交
4734 4735 4736 4737 4738
               *dp-- = *sp--;
            }
         }
         row_info->color_type = PNG_COLOR_TYPE_RGB_ALPHA;
         row_info->channels = 4;
G
Guy Schalnat 已提交
4739
         row_info->pixel_depth = (png_byte)(row_info->bit_depth << 2);
4740
         row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth, row_width);
G
Guy Schalnat 已提交
4741 4742 4743
      }
   }
}
G
Guy Schalnat 已提交
4744
#endif
G
Guy Schalnat 已提交
4745

4746 4747 4748 4749 4750 4751 4752 4753 4754 4755 4756 4757 4758 4759 4760 4761 4762 4763 4764 4765 4766 4767 4768 4769 4770 4771 4772 4773 4774 4775 4776
#ifdef PNG_READ_EXPAND_16_SUPPORTED
/* If the bit depth is 8 and the colour type is not a palette type expand the
 * whole row to 16 bits.  Has no effect otherwise.
 */
void /* PRIVATE */
png_do_expand_16(png_row_infop row_info, png_bytep row)
{
   if (row_info->bit_depth == 8 &&
      row_info->color_type != PNG_COLOR_TYPE_PALETTE)
   {
      /* The row have a sequence of bytes containing [0..255] and we need
       * to turn it into another row containing [0..65535], to do this we
       * calculate:
       *
       *  (input / 255) * 65535
       *
       *  Which happens to be exactly input * 257 and this can be achieved
       *  simply by byte replication in place (copying backwards).
       */
      png_byte *sp = row + row_info->rowbytes; /* source, last byte + 1 */
      png_byte *dp = sp + row_info->rowbytes;  /* destination, end + 1 */
      while (dp > sp)
         dp[-2] = dp[-1] = *--sp, dp -= 2;

      row_info->rowbytes *= 2;
      row_info->bit_depth = 16;
      row_info->pixel_depth = (png_byte)(row_info->channels * 16);
   }
}
#endif

4777
#ifdef PNG_READ_QUANTIZE_SUPPORTED
4778
void /* PRIVATE */
4779
png_do_quantize(png_row_infop row_info, png_bytep row,
4780
    png_const_bytep palette_lookup, png_const_bytep quantize_lookup)
G
Guy Schalnat 已提交
4781
{
G
Guy Schalnat 已提交
4782
   png_bytep sp, dp;
G
Guy Schalnat 已提交
4783
   png_uint_32 i;
4784
   png_uint_32 row_width=row_info->width;
G
Guy Schalnat 已提交
4785

4786
   png_debug(1, "in png_do_quantize");
4787

4788
   if (row_info->bit_depth == 8)
G
Guy Schalnat 已提交
4789
   {
4790
      if (row_info->color_type == PNG_COLOR_TYPE_RGB && palette_lookup)
G
Guy Schalnat 已提交
4791 4792 4793 4794
      {
         int r, g, b, p;
         sp = row;
         dp = row;
4795
         for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4796 4797 4798 4799 4800
         {
            r = *sp++;
            g = *sp++;
            b = *sp++;

4801 4802 4803 4804 4805 4806 4807
            /* This looks real messy, but the compiler will reduce
             * it down to a reasonable formula.  For example, with
             * 5 bits per color, we get:
             * p = (((r >> 3) & 0x1f) << 10) |
             *    (((g >> 3) & 0x1f) << 5) |
             *    ((b >> 3) & 0x1f);
             */
4808 4809 4810 4811 4812 4813 4814 4815
            p = (((r >> (8 - PNG_QUANTIZE_RED_BITS)) &
                ((1 << PNG_QUANTIZE_RED_BITS) - 1)) <<
                (PNG_QUANTIZE_GREEN_BITS + PNG_QUANTIZE_BLUE_BITS)) |
                (((g >> (8 - PNG_QUANTIZE_GREEN_BITS)) &
                ((1 << PNG_QUANTIZE_GREEN_BITS) - 1)) <<
                (PNG_QUANTIZE_BLUE_BITS)) |
                ((b >> (8 - PNG_QUANTIZE_BLUE_BITS)) &
                ((1 << PNG_QUANTIZE_BLUE_BITS) - 1));
G
Guy Schalnat 已提交
4816 4817 4818

            *dp++ = palette_lookup[p];
         }
4819

G
Guy Schalnat 已提交
4820 4821 4822
         row_info->color_type = PNG_COLOR_TYPE_PALETTE;
         row_info->channels = 1;
         row_info->pixel_depth = row_info->bit_depth;
4823
         row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth, row_width);
G
Guy Schalnat 已提交
4824
      }
4825

G
Guy Schalnat 已提交
4826
      else if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA &&
4827
         palette_lookup != NULL)
G
Guy Schalnat 已提交
4828 4829 4830 4831
      {
         int r, g, b, p;
         sp = row;
         dp = row;
4832
         for (i = 0; i < row_width; i++)
G
Guy Schalnat 已提交
4833 4834 4835 4836 4837 4838
         {
            r = *sp++;
            g = *sp++;
            b = *sp++;
            sp++;

4839 4840 4841 4842 4843 4844 4845 4846
            p = (((r >> (8 - PNG_QUANTIZE_RED_BITS)) &
                ((1 << PNG_QUANTIZE_RED_BITS) - 1)) <<
                (PNG_QUANTIZE_GREEN_BITS + PNG_QUANTIZE_BLUE_BITS)) |
                (((g >> (8 - PNG_QUANTIZE_GREEN_BITS)) &
                ((1 << PNG_QUANTIZE_GREEN_BITS) - 1)) <<
                (PNG_QUANTIZE_BLUE_BITS)) |
                ((b >> (8 - PNG_QUANTIZE_BLUE_BITS)) &
                ((1 << PNG_QUANTIZE_BLUE_BITS) - 1));
G
Guy Schalnat 已提交
4847 4848 4849

            *dp++ = palette_lookup[p];
         }
4850

G
Guy Schalnat 已提交
4851 4852 4853
         row_info->color_type = PNG_COLOR_TYPE_PALETTE;
         row_info->channels = 1;
         row_info->pixel_depth = row_info->bit_depth;
4854
         row_info->rowbytes = PNG_ROWBYTES(row_info->pixel_depth, row_width);
G
Guy Schalnat 已提交
4855
      }
4856

G
Guy Schalnat 已提交
4857
      else if (row_info->color_type == PNG_COLOR_TYPE_PALETTE &&
4858
         quantize_lookup)
G
Guy Schalnat 已提交
4859
      {
G
Guy Schalnat 已提交
4860
         sp = row;
4861

4862
         for (i = 0; i < row_width; i++, sp++)
G
Guy Schalnat 已提交
4863
         {
4864
            *sp = quantize_lookup[*sp];
G
Guy Schalnat 已提交
4865 4866 4867 4868
         }
      }
   }
}
4869
#endif /* PNG_READ_QUANTIZE_SUPPORTED */
4870
#endif /* PNG_READ_TRANSFORMS_SUPPORTED */
G
Guy Schalnat 已提交
4871

4872
#ifdef PNG_MNG_FEATURES_SUPPORTED
4873
/* Undoes intrapixel differencing  */
4874 4875 4876
void /* PRIVATE */
png_do_read_intrapixel(png_row_infop row_info, png_bytep row)
{
4877
   png_debug(1, "in png_do_read_intrapixel");
4878

4879 4880 4881 4882 4883
   if (
       (row_info->color_type & PNG_COLOR_MASK_COLOR))
   {
      int bytes_per_pixel;
      png_uint_32 row_width = row_info->width;
4884

4885 4886 4887 4888 4889 4890 4891
      if (row_info->bit_depth == 8)
      {
         png_bytep rp;
         png_uint_32 i;

         if (row_info->color_type == PNG_COLOR_TYPE_RGB)
            bytes_per_pixel = 3;
4892

4893 4894
         else if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
            bytes_per_pixel = 4;
4895

4896 4897 4898 4899 4900
         else
            return;

         for (i = 0, rp = row; i < row_width; i++, rp += bytes_per_pixel)
         {
4901 4902
            *(rp) = (png_byte)((256 + *rp + *(rp + 1)) & 0xff);
            *(rp+2) = (png_byte)((256 + *(rp + 2) + *(rp + 1)) & 0xff);
4903 4904 4905 4906 4907 4908 4909 4910 4911
         }
      }
      else if (row_info->bit_depth == 16)
      {
         png_bytep rp;
         png_uint_32 i;

         if (row_info->color_type == PNG_COLOR_TYPE_RGB)
            bytes_per_pixel = 6;
4912

4913 4914
         else if (row_info->color_type == PNG_COLOR_TYPE_RGB_ALPHA)
            bytes_per_pixel = 8;
4915

4916 4917 4918 4919 4920
         else
            return;

         for (i = 0, rp = row; i < row_width; i++, rp += bytes_per_pixel)
         {
4921 4922 4923 4924 4925
            png_uint_32 s0   = (*(rp    ) << 8) | *(rp + 1);
            png_uint_32 s1   = (*(rp + 2) << 8) | *(rp + 3);
            png_uint_32 s2   = (*(rp + 4) << 8) | *(rp + 5);
            png_uint_32 red  = (png_uint_32)((s0 + s1 + 65536L) & 0xffffL);
            png_uint_32 blue = (png_uint_32)((s2 + s1 + 65536L) & 0xffffL);
4926
            *(rp    ) = (png_byte)((red >> 8) & 0xff);
4927 4928 4929
            *(rp + 1) = (png_byte)(red & 0xff);
            *(rp + 4) = (png_byte)((blue >> 8) & 0xff);
            *(rp + 5) = (png_byte)(blue & 0xff);
4930 4931 4932 4933 4934
         }
      }
   }
}
#endif /* PNG_MNG_FEATURES_SUPPORTED */
4935
#endif /* PNG_READ_SUPPORTED */