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