/src/libjpeg-turbo.2.1.x/rdppm.c
Line | Count | Source (jump to first uncovered line) |
1 | | /* |
2 | | * rdppm.c |
3 | | * |
4 | | * This file was part of the Independent JPEG Group's software: |
5 | | * Copyright (C) 1991-1997, Thomas G. Lane. |
6 | | * Modified 2009 by Bill Allombert, Guido Vollbeding. |
7 | | * libjpeg-turbo Modifications: |
8 | | * Copyright (C) 2015-2017, 2020-2023, D. R. Commander. |
9 | | * For conditions of distribution and use, see the accompanying README.ijg |
10 | | * file. |
11 | | * |
12 | | * This file contains routines to read input images in PPM/PGM format. |
13 | | * The extended 2-byte-per-sample raw PPM/PGM formats are supported. |
14 | | * The PBMPLUS library is NOT required to compile this software |
15 | | * (but it is highly useful as a set of PPM image manipulation programs). |
16 | | * |
17 | | * These routines may need modification for non-Unix environments or |
18 | | * specialized applications. As they stand, they assume input from |
19 | | * an ordinary stdio stream. They further assume that reading begins |
20 | | * at the start of the file; start_input may need work if the |
21 | | * user interface has already read some data (e.g., to determine that |
22 | | * the file is indeed PPM format). |
23 | | */ |
24 | | |
25 | | #include "cmyk.h" |
26 | | #include "cdjpeg.h" /* Common decls for cjpeg/djpeg applications */ |
27 | | |
28 | | #ifdef PPM_SUPPORTED |
29 | | |
30 | | |
31 | | /* Portions of this code are based on the PBMPLUS library, which is: |
32 | | ** |
33 | | ** Copyright (C) 1988 by Jef Poskanzer. |
34 | | ** |
35 | | ** Permission to use, copy, modify, and distribute this software and its |
36 | | ** documentation for any purpose and without fee is hereby granted, provided |
37 | | ** that the above copyright notice appear in all copies and that both that |
38 | | ** copyright notice and this permission notice appear in supporting |
39 | | ** documentation. This software is provided "as is" without express or |
40 | | ** implied warranty. |
41 | | */ |
42 | | |
43 | | |
44 | | /* Macros to deal with unsigned chars as efficiently as compiler allows */ |
45 | | |
46 | | typedef unsigned char U_CHAR; |
47 | 38.2k | #define UCH(x) ((int)(x)) |
48 | | |
49 | | |
50 | | #define ReadOK(file, buffer, len) \ |
51 | 3.43k | (fread(buffer, 1, len, file) == ((size_t)(len))) |
52 | | |
53 | | static int alpha_index[JPEG_NUMCS] = { |
54 | | -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 3, 3, 0, 0, -1 |
55 | | }; |
56 | | |
57 | | |
58 | | /* Private version of data source object */ |
59 | | |
60 | | typedef struct { |
61 | | struct cjpeg_source_struct pub; /* public fields */ |
62 | | |
63 | | /* Usually these two pointers point to the same place: */ |
64 | | U_CHAR *iobuffer; /* fread's I/O buffer */ |
65 | | JSAMPROW pixrow; /* compressor input buffer */ |
66 | | size_t buffer_width; /* width of I/O buffer */ |
67 | | JSAMPLE *rescale; /* => maxval-remapping array, or NULL */ |
68 | | unsigned int maxval; |
69 | | } ppm_source_struct; |
70 | | |
71 | | typedef ppm_source_struct *ppm_source_ptr; |
72 | | |
73 | | |
74 | | LOCAL(int) |
75 | | pbm_getc(FILE *infile) |
76 | | /* Read next char, skipping over any comments */ |
77 | | /* A comment/newline sequence is returned as a newline */ |
78 | 29.3k | { |
79 | 29.3k | register int ch; |
80 | | |
81 | 29.3k | ch = getc(infile); |
82 | 29.3k | if (ch == '#') { |
83 | 1.62k | do { |
84 | 1.62k | ch = getc(infile); |
85 | 1.62k | } while (ch != '\n' && ch != EOF); |
86 | 664 | } |
87 | 29.3k | return ch; |
88 | 29.3k | } |
89 | | |
90 | | |
91 | | LOCAL(unsigned int) |
92 | | read_pbm_integer(j_compress_ptr cinfo, FILE *infile, unsigned int maxval) |
93 | | /* Read an unsigned decimal integer from the PPM file */ |
94 | | /* Swallows one trailing character after the integer */ |
95 | | /* Note that on a 16-bit-int machine, only values up to 64k can be read. */ |
96 | | /* This should not be a problem in practice. */ |
97 | 11.0k | { |
98 | 11.0k | register int ch; |
99 | 11.0k | register unsigned int val; |
100 | | |
101 | | /* Skip any leading whitespace */ |
102 | 13.9k | do { |
103 | 13.9k | ch = pbm_getc(infile); |
104 | 13.9k | if (ch == EOF) |
105 | 570 | ERREXIT(cinfo, JERR_INPUT_EOF); |
106 | 13.9k | } while (ch == ' ' || ch == '\t' || ch == '\n' || ch == '\r'); |
107 | | |
108 | 11.0k | if (ch < '0' || ch > '9') |
109 | 59 | ERREXIT(cinfo, JERR_PPM_NONNUMERIC); |
110 | | |
111 | 11.0k | val = ch - '0'; |
112 | 16.0k | while ((ch = pbm_getc(infile)) >= '0' && ch <= '9') { |
113 | 5.02k | val *= 10; |
114 | 5.02k | val += ch - '0'; |
115 | 5.02k | if (val > maxval) |
116 | 41 | ERREXIT(cinfo, JERR_PPM_OUTOFRANGE); |
117 | 5.02k | } |
118 | | |
119 | 11.0k | return val; |
120 | 11.0k | } |
121 | | |
122 | | |
123 | | /* |
124 | | * Read one row of pixels. |
125 | | * |
126 | | * We provide several different versions depending on input file format. |
127 | | * In all cases, input is scaled to the size of JSAMPLE. |
128 | | * |
129 | | * A really fast path is provided for reading byte/sample raw files with |
130 | | * maxval = MAXJSAMPLE, which is the normal case for 8-bit data. |
131 | | */ |
132 | | |
133 | | |
134 | | METHODDEF(JDIMENSION) |
135 | | get_text_gray_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
136 | | /* This version is for reading text-format PGM files with any maxval */ |
137 | 370 | { |
138 | 370 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
139 | 370 | FILE *infile = source->pub.input_file; |
140 | 370 | register JSAMPROW ptr; |
141 | 370 | register JSAMPLE *rescale = source->rescale; |
142 | 370 | JDIMENSION col; |
143 | 370 | unsigned int maxval = source->maxval; |
144 | | |
145 | 370 | ptr = source->pub.buffer[0]; |
146 | 1.95k | for (col = cinfo->image_width; col > 0; col--) { |
147 | 1.58k | *ptr++ = rescale[read_pbm_integer(cinfo, infile, maxval)]; |
148 | 1.58k | } |
149 | 370 | return 1; |
150 | 370 | } |
151 | | |
152 | | |
153 | 0 | #define GRAY_RGB_READ_LOOP(read_op, alpha_set_op) { \ |
154 | 0 | for (col = cinfo->image_width; col > 0; col--) { \ |
155 | 0 | ptr[rindex] = ptr[gindex] = ptr[bindex] = read_op; \ |
156 | 0 | alpha_set_op \ |
157 | 0 | ptr += ps; \ |
158 | 0 | } \ |
159 | 0 | } |
160 | | |
161 | | METHODDEF(JDIMENSION) |
162 | | get_text_gray_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
163 | | /* This version is for reading text-format PGM files with any maxval and |
164 | | converting to extended RGB */ |
165 | 0 | { |
166 | 0 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
167 | 0 | FILE *infile = source->pub.input_file; |
168 | 0 | register JSAMPROW ptr; |
169 | 0 | register JSAMPLE *rescale = source->rescale; |
170 | 0 | JDIMENSION col; |
171 | 0 | unsigned int maxval = source->maxval; |
172 | 0 | register int rindex = rgb_red[cinfo->in_color_space]; |
173 | 0 | register int gindex = rgb_green[cinfo->in_color_space]; |
174 | 0 | register int bindex = rgb_blue[cinfo->in_color_space]; |
175 | 0 | register int aindex = alpha_index[cinfo->in_color_space]; |
176 | 0 | register int ps = rgb_pixelsize[cinfo->in_color_space]; |
177 | |
|
178 | 0 | ptr = source->pub.buffer[0]; |
179 | 0 | if (maxval == MAXJSAMPLE) { |
180 | 0 | if (aindex >= 0) |
181 | 0 | GRAY_RGB_READ_LOOP((JSAMPLE)read_pbm_integer(cinfo, infile, maxval), |
182 | 0 | ptr[aindex] = MAXJSAMPLE;) |
183 | 0 | else |
184 | 0 | GRAY_RGB_READ_LOOP((JSAMPLE)read_pbm_integer(cinfo, infile, maxval), {}) |
185 | 0 | } else { |
186 | 0 | if (aindex >= 0) |
187 | 0 | GRAY_RGB_READ_LOOP(rescale[read_pbm_integer(cinfo, infile, maxval)], |
188 | 0 | ptr[aindex] = MAXJSAMPLE;) |
189 | 0 | else |
190 | 0 | GRAY_RGB_READ_LOOP(rescale[read_pbm_integer(cinfo, infile, maxval)], {}) |
191 | 0 | } |
192 | 0 | return 1; |
193 | 0 | } |
194 | | |
195 | | |
196 | | METHODDEF(JDIMENSION) |
197 | | get_text_gray_cmyk_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
198 | | /* This version is for reading text-format PGM files with any maxval and |
199 | | converting to CMYK */ |
200 | 0 | { |
201 | 0 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
202 | 0 | FILE *infile = source->pub.input_file; |
203 | 0 | register JSAMPROW ptr; |
204 | 0 | register JSAMPLE *rescale = source->rescale; |
205 | 0 | JDIMENSION col; |
206 | 0 | unsigned int maxval = source->maxval; |
207 | |
|
208 | 0 | ptr = source->pub.buffer[0]; |
209 | 0 | if (maxval == MAXJSAMPLE) { |
210 | 0 | for (col = cinfo->image_width; col > 0; col--) { |
211 | 0 | JSAMPLE gray = (JSAMPLE)read_pbm_integer(cinfo, infile, maxval); |
212 | 0 | rgb_to_cmyk(gray, gray, gray, ptr, ptr + 1, ptr + 2, ptr + 3); |
213 | 0 | ptr += 4; |
214 | 0 | } |
215 | 0 | } else { |
216 | 0 | for (col = cinfo->image_width; col > 0; col--) { |
217 | 0 | JSAMPLE gray = rescale[read_pbm_integer(cinfo, infile, maxval)]; |
218 | 0 | rgb_to_cmyk(gray, gray, gray, ptr, ptr + 1, ptr + 2, ptr + 3); |
219 | 0 | ptr += 4; |
220 | 0 | } |
221 | 0 | } |
222 | 0 | return 1; |
223 | 0 | } |
224 | | |
225 | | |
226 | 2.11k | #define RGB_READ_LOOP(read_op, alpha_set_op) { \ |
227 | 5.21k | for (col = cinfo->image_width; col > 0; col--) { \ |
228 | 3.09k | ptr[rindex] = read_op; \ |
229 | 3.09k | ptr[gindex] = read_op; \ |
230 | 3.09k | ptr[bindex] = read_op; \ |
231 | 3.09k | alpha_set_op \ |
232 | 3.09k | ptr += ps; \ |
233 | 3.09k | } \ |
234 | 2.11k | } |
235 | | |
236 | | METHODDEF(JDIMENSION) |
237 | | get_text_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
238 | | /* This version is for reading text-format PPM files with any maxval */ |
239 | 800 | { |
240 | 800 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
241 | 800 | FILE *infile = source->pub.input_file; |
242 | 800 | register JSAMPROW ptr; |
243 | 800 | register JSAMPLE *rescale = source->rescale; |
244 | 800 | JDIMENSION col; |
245 | 800 | unsigned int maxval = source->maxval; |
246 | 800 | register int rindex = rgb_red[cinfo->in_color_space]; |
247 | 800 | register int gindex = rgb_green[cinfo->in_color_space]; |
248 | 800 | register int bindex = rgb_blue[cinfo->in_color_space]; |
249 | 800 | register int aindex = alpha_index[cinfo->in_color_space]; |
250 | 800 | register int ps = rgb_pixelsize[cinfo->in_color_space]; |
251 | | |
252 | 800 | ptr = source->pub.buffer[0]; |
253 | 800 | if (maxval == MAXJSAMPLE) { |
254 | 232 | if (aindex >= 0) |
255 | 0 | RGB_READ_LOOP((JSAMPLE)read_pbm_integer(cinfo, infile, maxval), |
256 | 232 | ptr[aindex] = MAXJSAMPLE;) |
257 | 232 | else |
258 | 232 | RGB_READ_LOOP((JSAMPLE)read_pbm_integer(cinfo, infile, maxval), {}) |
259 | 568 | } else { |
260 | 568 | if (aindex >= 0) |
261 | 0 | RGB_READ_LOOP(rescale[read_pbm_integer(cinfo, infile, maxval)], |
262 | 568 | ptr[aindex] = MAXJSAMPLE;) |
263 | 568 | else |
264 | 568 | RGB_READ_LOOP(rescale[read_pbm_integer(cinfo, infile, maxval)], {}) |
265 | 568 | } |
266 | 800 | return 1; |
267 | 800 | } |
268 | | |
269 | | |
270 | | METHODDEF(JDIMENSION) |
271 | | get_text_rgb_cmyk_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
272 | | /* This version is for reading text-format PPM files with any maxval and |
273 | | converting to CMYK */ |
274 | 0 | { |
275 | 0 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
276 | 0 | FILE *infile = source->pub.input_file; |
277 | 0 | register JSAMPROW ptr; |
278 | 0 | register JSAMPLE *rescale = source->rescale; |
279 | 0 | JDIMENSION col; |
280 | 0 | unsigned int maxval = source->maxval; |
281 | |
|
282 | 0 | ptr = source->pub.buffer[0]; |
283 | 0 | if (maxval == MAXJSAMPLE) { |
284 | 0 | for (col = cinfo->image_width; col > 0; col--) { |
285 | 0 | JSAMPLE r = (JSAMPLE)read_pbm_integer(cinfo, infile, maxval); |
286 | 0 | JSAMPLE g = (JSAMPLE)read_pbm_integer(cinfo, infile, maxval); |
287 | 0 | JSAMPLE b = (JSAMPLE)read_pbm_integer(cinfo, infile, maxval); |
288 | 0 | rgb_to_cmyk(r, g, b, ptr, ptr + 1, ptr + 2, ptr + 3); |
289 | 0 | ptr += 4; |
290 | 0 | } |
291 | 0 | } else { |
292 | 0 | for (col = cinfo->image_width; col > 0; col--) { |
293 | 0 | JSAMPLE r = rescale[read_pbm_integer(cinfo, infile, maxval)]; |
294 | 0 | JSAMPLE g = rescale[read_pbm_integer(cinfo, infile, maxval)]; |
295 | 0 | JSAMPLE b = rescale[read_pbm_integer(cinfo, infile, maxval)]; |
296 | 0 | rgb_to_cmyk(r, g, b, ptr, ptr + 1, ptr + 2, ptr + 3); |
297 | 0 | ptr += 4; |
298 | 0 | } |
299 | 0 | } |
300 | 0 | return 1; |
301 | 0 | } |
302 | | |
303 | | |
304 | | METHODDEF(JDIMENSION) |
305 | | get_scaled_gray_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
306 | | /* This version is for reading raw-byte-format PGM files with any maxval */ |
307 | 671 | { |
308 | 671 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
309 | 671 | register JSAMPROW ptr; |
310 | 671 | register U_CHAR *bufferptr; |
311 | 671 | register JSAMPLE *rescale = source->rescale; |
312 | 671 | JDIMENSION col; |
313 | | |
314 | 671 | if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width)) |
315 | 77 | ERREXIT(cinfo, JERR_INPUT_EOF); |
316 | 671 | ptr = source->pub.buffer[0]; |
317 | 671 | bufferptr = source->iobuffer; |
318 | 32.6k | for (col = cinfo->image_width; col > 0; col--) { |
319 | 32.0k | *ptr++ = rescale[UCH(*bufferptr++)]; |
320 | 32.0k | } |
321 | 671 | return 1; |
322 | 671 | } |
323 | | |
324 | | |
325 | | METHODDEF(JDIMENSION) |
326 | | get_gray_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
327 | | /* This version is for reading raw-byte-format PGM files with any maxval |
328 | | and converting to extended RGB */ |
329 | 0 | { |
330 | 0 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
331 | 0 | register JSAMPROW ptr; |
332 | 0 | register U_CHAR *bufferptr; |
333 | 0 | register JSAMPLE *rescale = source->rescale; |
334 | 0 | JDIMENSION col; |
335 | 0 | unsigned int maxval = source->maxval; |
336 | 0 | register int rindex = rgb_red[cinfo->in_color_space]; |
337 | 0 | register int gindex = rgb_green[cinfo->in_color_space]; |
338 | 0 | register int bindex = rgb_blue[cinfo->in_color_space]; |
339 | 0 | register int aindex = alpha_index[cinfo->in_color_space]; |
340 | 0 | register int ps = rgb_pixelsize[cinfo->in_color_space]; |
341 | |
|
342 | 0 | if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width)) |
343 | 0 | ERREXIT(cinfo, JERR_INPUT_EOF); |
344 | 0 | ptr = source->pub.buffer[0]; |
345 | 0 | bufferptr = source->iobuffer; |
346 | 0 | if (maxval == MAXJSAMPLE) { |
347 | 0 | if (aindex >= 0) |
348 | 0 | GRAY_RGB_READ_LOOP(*bufferptr++, ptr[aindex] = MAXJSAMPLE;) |
349 | 0 | else |
350 | 0 | GRAY_RGB_READ_LOOP(*bufferptr++, {}) |
351 | 0 | } else { |
352 | 0 | if (aindex >= 0) |
353 | 0 | GRAY_RGB_READ_LOOP(rescale[UCH(*bufferptr++)], ptr[aindex] = MAXJSAMPLE;) |
354 | 0 | else |
355 | 0 | GRAY_RGB_READ_LOOP(rescale[UCH(*bufferptr++)], {}) |
356 | 0 | } |
357 | 0 | return 1; |
358 | 0 | } |
359 | | |
360 | | |
361 | | METHODDEF(JDIMENSION) |
362 | | get_gray_cmyk_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
363 | | /* This version is for reading raw-byte-format PGM files with any maxval |
364 | | and converting to CMYK */ |
365 | 0 | { |
366 | 0 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
367 | 0 | register JSAMPROW ptr; |
368 | 0 | register U_CHAR *bufferptr; |
369 | 0 | register JSAMPLE *rescale = source->rescale; |
370 | 0 | JDIMENSION col; |
371 | 0 | unsigned int maxval = source->maxval; |
372 | |
|
373 | 0 | if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width)) |
374 | 0 | ERREXIT(cinfo, JERR_INPUT_EOF); |
375 | 0 | ptr = source->pub.buffer[0]; |
376 | 0 | bufferptr = source->iobuffer; |
377 | 0 | if (maxval == MAXJSAMPLE) { |
378 | 0 | for (col = cinfo->image_width; col > 0; col--) { |
379 | 0 | JSAMPLE gray = *bufferptr++; |
380 | 0 | rgb_to_cmyk(gray, gray, gray, ptr, ptr + 1, ptr + 2, ptr + 3); |
381 | 0 | ptr += 4; |
382 | 0 | } |
383 | 0 | } else { |
384 | 0 | for (col = cinfo->image_width; col > 0; col--) { |
385 | 0 | JSAMPLE gray = rescale[UCH(*bufferptr++)]; |
386 | 0 | rgb_to_cmyk(gray, gray, gray, ptr, ptr + 1, ptr + 2, ptr + 3); |
387 | 0 | ptr += 4; |
388 | 0 | } |
389 | 0 | } |
390 | 0 | return 1; |
391 | 0 | } |
392 | | |
393 | | |
394 | | METHODDEF(JDIMENSION) |
395 | | get_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
396 | | /* This version is for reading raw-byte-format PPM files with any maxval */ |
397 | 1.31k | { |
398 | 1.31k | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
399 | 1.31k | register JSAMPROW ptr; |
400 | 1.31k | register U_CHAR *bufferptr; |
401 | 1.31k | register JSAMPLE *rescale = source->rescale; |
402 | 1.31k | JDIMENSION col; |
403 | 1.31k | unsigned int maxval = source->maxval; |
404 | 1.31k | register int rindex = rgb_red[cinfo->in_color_space]; |
405 | 1.31k | register int gindex = rgb_green[cinfo->in_color_space]; |
406 | 1.31k | register int bindex = rgb_blue[cinfo->in_color_space]; |
407 | 1.31k | register int aindex = alpha_index[cinfo->in_color_space]; |
408 | 1.31k | register int ps = rgb_pixelsize[cinfo->in_color_space]; |
409 | | |
410 | 1.31k | if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width)) |
411 | 170 | ERREXIT(cinfo, JERR_INPUT_EOF); |
412 | 1.31k | ptr = source->pub.buffer[0]; |
413 | 1.31k | bufferptr = source->iobuffer; |
414 | 1.31k | if (maxval == MAXJSAMPLE) { |
415 | 0 | if (aindex >= 0) |
416 | 0 | RGB_READ_LOOP(*bufferptr++, ptr[aindex] = MAXJSAMPLE;) |
417 | 0 | else |
418 | 0 | RGB_READ_LOOP(*bufferptr++, {}) |
419 | 1.31k | } else { |
420 | 1.31k | if (aindex >= 0) |
421 | 0 | RGB_READ_LOOP(rescale[UCH(*bufferptr++)], ptr[aindex] = MAXJSAMPLE;) |
422 | 1.31k | else |
423 | 1.31k | RGB_READ_LOOP(rescale[UCH(*bufferptr++)], {}) |
424 | 1.31k | } |
425 | 1.31k | return 1; |
426 | 1.31k | } |
427 | | |
428 | | |
429 | | METHODDEF(JDIMENSION) |
430 | | get_rgb_cmyk_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
431 | | /* This version is for reading raw-byte-format PPM files with any maxval and |
432 | | converting to CMYK */ |
433 | 0 | { |
434 | 0 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
435 | 0 | register JSAMPROW ptr; |
436 | 0 | register U_CHAR *bufferptr; |
437 | 0 | register JSAMPLE *rescale = source->rescale; |
438 | 0 | JDIMENSION col; |
439 | 0 | unsigned int maxval = source->maxval; |
440 | |
|
441 | 0 | if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width)) |
442 | 0 | ERREXIT(cinfo, JERR_INPUT_EOF); |
443 | 0 | ptr = source->pub.buffer[0]; |
444 | 0 | bufferptr = source->iobuffer; |
445 | 0 | if (maxval == MAXJSAMPLE) { |
446 | 0 | for (col = cinfo->image_width; col > 0; col--) { |
447 | 0 | JSAMPLE r = *bufferptr++; |
448 | 0 | JSAMPLE g = *bufferptr++; |
449 | 0 | JSAMPLE b = *bufferptr++; |
450 | 0 | rgb_to_cmyk(r, g, b, ptr, ptr + 1, ptr + 2, ptr + 3); |
451 | 0 | ptr += 4; |
452 | 0 | } |
453 | 0 | } else { |
454 | 0 | for (col = cinfo->image_width; col > 0; col--) { |
455 | 0 | JSAMPLE r = rescale[UCH(*bufferptr++)]; |
456 | 0 | JSAMPLE g = rescale[UCH(*bufferptr++)]; |
457 | 0 | JSAMPLE b = rescale[UCH(*bufferptr++)]; |
458 | 0 | rgb_to_cmyk(r, g, b, ptr, ptr + 1, ptr + 2, ptr + 3); |
459 | 0 | ptr += 4; |
460 | 0 | } |
461 | 0 | } |
462 | 0 | return 1; |
463 | 0 | } |
464 | | |
465 | | |
466 | | METHODDEF(JDIMENSION) |
467 | | get_raw_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
468 | | /* This version is for reading raw-byte-format files with maxval = MAXJSAMPLE. |
469 | | * In this case we just read right into the JSAMPLE buffer! |
470 | | * Note that same code works for PPM and PGM files. |
471 | | */ |
472 | 396 | { |
473 | 396 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
474 | | |
475 | 396 | if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width)) |
476 | 22 | ERREXIT(cinfo, JERR_INPUT_EOF); |
477 | 396 | return 1; |
478 | 396 | } |
479 | | |
480 | | |
481 | | METHODDEF(JDIMENSION) |
482 | | get_word_gray_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
483 | | /* This version is for reading raw-word-format PGM files with any maxval */ |
484 | 425 | { |
485 | 425 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
486 | 425 | register JSAMPROW ptr; |
487 | 425 | register U_CHAR *bufferptr; |
488 | 425 | register JSAMPLE *rescale = source->rescale; |
489 | 425 | JDIMENSION col; |
490 | 425 | unsigned int maxval = source->maxval; |
491 | | |
492 | 425 | if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width)) |
493 | 49 | ERREXIT(cinfo, JERR_INPUT_EOF); |
494 | 425 | ptr = source->pub.buffer[0]; |
495 | 425 | bufferptr = source->iobuffer; |
496 | 1.00k | for (col = cinfo->image_width; col > 0; col--) { |
497 | 578 | register unsigned int temp; |
498 | 578 | temp = UCH(*bufferptr++) << 8; |
499 | 578 | temp |= UCH(*bufferptr++); |
500 | 578 | if (temp > maxval) |
501 | 13 | ERREXIT(cinfo, JERR_PPM_OUTOFRANGE); |
502 | 578 | *ptr++ = rescale[temp]; |
503 | 578 | } |
504 | 425 | return 1; |
505 | 425 | } |
506 | | |
507 | | |
508 | | METHODDEF(JDIMENSION) |
509 | | get_word_rgb_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
510 | | /* This version is for reading raw-word-format PPM files with any maxval */ |
511 | 626 | { |
512 | 626 | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
513 | 626 | register JSAMPROW ptr; |
514 | 626 | register U_CHAR *bufferptr; |
515 | 626 | register JSAMPLE *rescale = source->rescale; |
516 | 626 | JDIMENSION col; |
517 | 626 | unsigned int maxval = source->maxval; |
518 | 626 | register int rindex = rgb_red[cinfo->in_color_space]; |
519 | 626 | register int gindex = rgb_green[cinfo->in_color_space]; |
520 | 626 | register int bindex = rgb_blue[cinfo->in_color_space]; |
521 | 626 | register int aindex = alpha_index[cinfo->in_color_space]; |
522 | 626 | register int ps = rgb_pixelsize[cinfo->in_color_space]; |
523 | | |
524 | 626 | if (!ReadOK(source->pub.input_file, source->iobuffer, source->buffer_width)) |
525 | 96 | ERREXIT(cinfo, JERR_INPUT_EOF); |
526 | 626 | ptr = source->pub.buffer[0]; |
527 | 626 | bufferptr = source->iobuffer; |
528 | 1.46k | for (col = cinfo->image_width; col > 0; col--) { |
529 | 842 | register unsigned int temp; |
530 | 842 | temp = UCH(*bufferptr++) << 8; |
531 | 842 | temp |= UCH(*bufferptr++); |
532 | 842 | if (temp > maxval) |
533 | 44 | ERREXIT(cinfo, JERR_PPM_OUTOFRANGE); |
534 | 842 | ptr[rindex] = rescale[temp]; |
535 | 842 | temp = UCH(*bufferptr++) << 8; |
536 | 842 | temp |= UCH(*bufferptr++); |
537 | 842 | if (temp > maxval) |
538 | 40 | ERREXIT(cinfo, JERR_PPM_OUTOFRANGE); |
539 | 842 | ptr[gindex] = rescale[temp]; |
540 | 842 | temp = UCH(*bufferptr++) << 8; |
541 | 842 | temp |= UCH(*bufferptr++); |
542 | 842 | if (temp > maxval) |
543 | 32 | ERREXIT(cinfo, JERR_PPM_OUTOFRANGE); |
544 | 842 | ptr[bindex] = rescale[temp]; |
545 | 842 | if (aindex >= 0) |
546 | 0 | ptr[aindex] = MAXJSAMPLE; |
547 | 842 | ptr += ps; |
548 | 842 | } |
549 | 626 | return 1; |
550 | 626 | } |
551 | | |
552 | | |
553 | | /* |
554 | | * Read the file header; return image size and component count. |
555 | | */ |
556 | | |
557 | | METHODDEF(void) |
558 | | start_input_ppm(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
559 | 2.12k | { |
560 | 2.12k | ppm_source_ptr source = (ppm_source_ptr)sinfo; |
561 | 2.12k | int c; |
562 | 2.12k | unsigned int w, h, maxval; |
563 | 2.12k | boolean need_iobuffer, use_raw_buffer, need_rescale; |
564 | | |
565 | 2.12k | if (getc(source->pub.input_file) != 'P') |
566 | 0 | ERREXIT(cinfo, JERR_PPM_NOT); |
567 | | |
568 | 2.12k | c = getc(source->pub.input_file); /* subformat discriminator character */ |
569 | | |
570 | | /* detect unsupported variants (ie, PBM) before trying to read header */ |
571 | 2.12k | switch (c) { |
572 | 368 | case '2': /* it's a text-format PGM file */ |
573 | 680 | case '3': /* it's a text-format PPM file */ |
574 | 1.21k | case '5': /* it's a raw-format PGM file */ |
575 | 2.11k | case '6': /* it's a raw-format PPM file */ |
576 | 2.11k | break; |
577 | 4 | default: |
578 | 4 | ERREXIT(cinfo, JERR_PPM_NOT); |
579 | 4 | break; |
580 | 2.12k | } |
581 | | |
582 | | /* fetch the remaining header info */ |
583 | 2.11k | w = read_pbm_integer(cinfo, source->pub.input_file, 65535); |
584 | 2.11k | h = read_pbm_integer(cinfo, source->pub.input_file, 65535); |
585 | 2.11k | maxval = read_pbm_integer(cinfo, source->pub.input_file, 65535); |
586 | | |
587 | 2.11k | if (w <= 0 || h <= 0 || maxval <= 0) /* error check */ |
588 | 6 | ERREXIT(cinfo, JERR_PPM_NOT); |
589 | 2.11k | #ifdef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION |
590 | 2.11k | if (sinfo->max_pixels && (unsigned long long)w * h > sinfo->max_pixels) |
591 | 52 | ERREXIT(cinfo, JERR_WIDTH_OVERFLOW); |
592 | 2.11k | #endif |
593 | | |
594 | 2.11k | cinfo->data_precision = BITS_IN_JSAMPLE; /* we always rescale data to this */ |
595 | 2.11k | cinfo->image_width = (JDIMENSION)w; |
596 | 2.11k | cinfo->image_height = (JDIMENSION)h; |
597 | 2.11k | source->maxval = maxval; |
598 | | |
599 | | /* initialize flags to most common settings */ |
600 | 2.11k | need_iobuffer = TRUE; /* do we need an I/O buffer? */ |
601 | 2.11k | use_raw_buffer = FALSE; /* do we map input buffer onto I/O buffer? */ |
602 | 2.11k | need_rescale = TRUE; /* do we need a rescale array? */ |
603 | | |
604 | 2.11k | switch (c) { |
605 | 244 | case '2': /* it's a text-format PGM file */ |
606 | 244 | if (cinfo->in_color_space == JCS_UNKNOWN || |
607 | 244 | cinfo->in_color_space == JCS_RGB) |
608 | 244 | cinfo->in_color_space = JCS_GRAYSCALE; |
609 | 244 | TRACEMS2(cinfo, 1, JTRC_PGM_TEXT, w, h); |
610 | 244 | if (cinfo->in_color_space == JCS_GRAYSCALE) |
611 | 244 | source->pub.get_pixel_rows = get_text_gray_row; |
612 | 0 | else if (IsExtRGB(cinfo->in_color_space)) |
613 | 0 | source->pub.get_pixel_rows = get_text_gray_rgb_row; |
614 | 0 | else if (cinfo->in_color_space == JCS_CMYK) |
615 | 0 | source->pub.get_pixel_rows = get_text_gray_cmyk_row; |
616 | 0 | else |
617 | 0 | ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE); |
618 | 244 | need_iobuffer = FALSE; |
619 | 244 | break; |
620 | | |
621 | 218 | case '3': /* it's a text-format PPM file */ |
622 | 218 | if (cinfo->in_color_space == JCS_UNKNOWN) |
623 | 0 | cinfo->in_color_space = JCS_EXT_RGB; |
624 | 218 | TRACEMS2(cinfo, 1, JTRC_PPM_TEXT, w, h); |
625 | 218 | if (IsExtRGB(cinfo->in_color_space)) |
626 | 218 | source->pub.get_pixel_rows = get_text_rgb_row; |
627 | 0 | else if (cinfo->in_color_space == JCS_CMYK) |
628 | 0 | source->pub.get_pixel_rows = get_text_rgb_cmyk_row; |
629 | 0 | else |
630 | 0 | ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE); |
631 | 218 | need_iobuffer = FALSE; |
632 | 218 | break; |
633 | | |
634 | 438 | case '5': /* it's a raw-format PGM file */ |
635 | 438 | if (cinfo->in_color_space == JCS_UNKNOWN || |
636 | 438 | cinfo->in_color_space == JCS_RGB) |
637 | 438 | cinfo->in_color_space = JCS_GRAYSCALE; |
638 | 438 | TRACEMS2(cinfo, 1, JTRC_PGM, w, h); |
639 | 438 | if (maxval > 255) { |
640 | 130 | if (cinfo->in_color_space == JCS_GRAYSCALE) |
641 | 130 | source->pub.get_pixel_rows = get_word_gray_row; |
642 | 0 | else |
643 | 0 | ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE); |
644 | 308 | } else if (maxval == MAXJSAMPLE && sizeof(JSAMPLE) == sizeof(U_CHAR) && |
645 | 308 | cinfo->in_color_space == JCS_GRAYSCALE) { |
646 | 40 | source->pub.get_pixel_rows = get_raw_row; |
647 | 40 | use_raw_buffer = TRUE; |
648 | 40 | need_rescale = FALSE; |
649 | 268 | } else { |
650 | 268 | if (cinfo->in_color_space == JCS_GRAYSCALE) |
651 | 268 | source->pub.get_pixel_rows = get_scaled_gray_row; |
652 | 0 | else if (IsExtRGB(cinfo->in_color_space)) |
653 | 0 | source->pub.get_pixel_rows = get_gray_rgb_row; |
654 | 0 | else if (cinfo->in_color_space == JCS_CMYK) |
655 | 0 | source->pub.get_pixel_rows = get_gray_cmyk_row; |
656 | 0 | else |
657 | 0 | ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE); |
658 | 268 | } |
659 | 438 | break; |
660 | | |
661 | 810 | case '6': /* it's a raw-format PPM file */ |
662 | 810 | if (cinfo->in_color_space == JCS_UNKNOWN) |
663 | 0 | cinfo->in_color_space = JCS_EXT_RGB; |
664 | 810 | TRACEMS2(cinfo, 1, JTRC_PPM, w, h); |
665 | 810 | if (maxval > 255) { |
666 | 214 | if (IsExtRGB(cinfo->in_color_space)) |
667 | 214 | source->pub.get_pixel_rows = get_word_rgb_row; |
668 | 0 | else |
669 | 0 | ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE); |
670 | 596 | } else if (maxval == MAXJSAMPLE && sizeof(JSAMPLE) == sizeof(U_CHAR) && |
671 | 596 | #if RGB_RED == 0 && RGB_GREEN == 1 && RGB_BLUE == 2 && RGB_PIXELSIZE == 3 |
672 | 596 | (cinfo->in_color_space == JCS_EXT_RGB || |
673 | 4 | cinfo->in_color_space == JCS_RGB)) { |
674 | | #else |
675 | | cinfo->in_color_space == JCS_EXT_RGB) { |
676 | | #endif |
677 | 4 | source->pub.get_pixel_rows = get_raw_row; |
678 | 4 | use_raw_buffer = TRUE; |
679 | 4 | need_rescale = FALSE; |
680 | 592 | } else { |
681 | 592 | if (IsExtRGB(cinfo->in_color_space)) |
682 | 592 | source->pub.get_pixel_rows = get_rgb_row; |
683 | 0 | else if (cinfo->in_color_space == JCS_CMYK) |
684 | 0 | source->pub.get_pixel_rows = get_rgb_cmyk_row; |
685 | 0 | else |
686 | 0 | ERREXIT(cinfo, JERR_BAD_IN_COLORSPACE); |
687 | 592 | } |
688 | 810 | break; |
689 | 2.11k | } |
690 | | |
691 | 1.71k | if (IsExtRGB(cinfo->in_color_space)) |
692 | 1.02k | cinfo->input_components = rgb_pixelsize[cinfo->in_color_space]; |
693 | 682 | else if (cinfo->in_color_space == JCS_GRAYSCALE) |
694 | 682 | cinfo->input_components = 1; |
695 | 0 | else if (cinfo->in_color_space == JCS_CMYK) |
696 | 0 | cinfo->input_components = 4; |
697 | | |
698 | | /* Allocate space for I/O buffer: 1 or 3 bytes or words/pixel. */ |
699 | 1.71k | if (need_iobuffer) { |
700 | 1.24k | if (c == '6') |
701 | 810 | source->buffer_width = (size_t)w * 3 * |
702 | 810 | ((maxval <= 255) ? sizeof(U_CHAR) : (2 * sizeof(U_CHAR))); |
703 | 438 | else |
704 | 438 | source->buffer_width = (size_t)w * |
705 | 438 | ((maxval <= 255) ? sizeof(U_CHAR) : (2 * sizeof(U_CHAR))); |
706 | 1.24k | source->iobuffer = (U_CHAR *) |
707 | 1.24k | (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE, |
708 | 1.24k | source->buffer_width); |
709 | 1.24k | } |
710 | | |
711 | | /* Create compressor input buffer. */ |
712 | 1.71k | if (use_raw_buffer) { |
713 | | /* For unscaled raw-input case, we can just map it onto the I/O buffer. */ |
714 | | /* Synthesize a JSAMPARRAY pointer structure */ |
715 | 44 | source->pixrow = (JSAMPROW)source->iobuffer; |
716 | 44 | source->pub.buffer = &source->pixrow; |
717 | 44 | source->pub.buffer_height = 1; |
718 | 1.66k | } else { |
719 | | /* Need to translate anyway, so make a separate sample buffer. */ |
720 | 1.66k | source->pub.buffer = (*cinfo->mem->alloc_sarray) |
721 | 1.66k | ((j_common_ptr)cinfo, JPOOL_IMAGE, |
722 | 1.66k | (JDIMENSION)w * cinfo->input_components, (JDIMENSION)1); |
723 | 1.66k | source->pub.buffer_height = 1; |
724 | 1.66k | } |
725 | | |
726 | | /* Compute the rescaling array if required. */ |
727 | 1.71k | if (need_rescale) { |
728 | 1.66k | long val, half_maxval; |
729 | | |
730 | | /* On 16-bit-int machines we have to be careful of maxval = 65535 */ |
731 | 1.66k | source->rescale = (JSAMPLE *) |
732 | 1.66k | (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE, |
733 | 1.66k | (size_t)(((long)MAX(maxval, 255) + 1L) * |
734 | 1.66k | sizeof(JSAMPLE))); |
735 | 1.66k | memset(source->rescale, 0, (size_t)(((long)MAX(maxval, 255) + 1L) * |
736 | 1.66k | sizeof(JSAMPLE))); |
737 | 1.66k | half_maxval = maxval / 2; |
738 | 6.60M | for (val = 0; val <= (long)maxval; val++) { |
739 | | /* The multiplication here must be done in 32 bits to avoid overflow */ |
740 | 6.60M | source->rescale[val] = (JSAMPLE)((val * MAXJSAMPLE + half_maxval) / |
741 | 6.60M | maxval); |
742 | 6.60M | } |
743 | 1.66k | } |
744 | 1.71k | } |
745 | | |
746 | | |
747 | | /* |
748 | | * Finish up at the end of the file. |
749 | | */ |
750 | | |
751 | | METHODDEF(void) |
752 | | finish_input_ppm(j_compress_ptr cinfo, cjpeg_source_ptr sinfo) |
753 | 500 | { |
754 | | /* no work */ |
755 | 500 | } |
756 | | |
757 | | |
758 | | /* |
759 | | * The module selection routine for PPM format input. |
760 | | */ |
761 | | |
762 | | GLOBAL(cjpeg_source_ptr) |
763 | | jinit_read_ppm(j_compress_ptr cinfo) |
764 | 2.12k | { |
765 | 2.12k | ppm_source_ptr source; |
766 | | |
767 | | /* Create module interface object */ |
768 | 2.12k | source = (ppm_source_ptr) |
769 | 2.12k | (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE, |
770 | 2.12k | sizeof(ppm_source_struct)); |
771 | | /* Fill in method ptrs, except get_pixel_rows which start_input sets */ |
772 | 2.12k | source->pub.start_input = start_input_ppm; |
773 | 2.12k | source->pub.finish_input = finish_input_ppm; |
774 | 2.12k | #ifdef FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION |
775 | 2.12k | source->pub.max_pixels = 0; |
776 | 2.12k | #endif |
777 | | |
778 | 2.12k | return (cjpeg_source_ptr)source; |
779 | 2.12k | } |
780 | | |
781 | | #endif /* PPM_SUPPORTED */ |