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