1 |
27 |
unneback |
/*
|
2 |
|
|
* wrppm.c
|
3 |
|
|
*
|
4 |
|
|
* Copyright (C) 1991-1996, Thomas G. Lane.
|
5 |
|
|
* This file is part of the Independent JPEG Group's software.
|
6 |
|
|
* For conditions of distribution and use, see the accompanying README file.
|
7 |
|
|
*
|
8 |
|
|
* This file contains routines to write output images in PPM/PGM format.
|
9 |
|
|
* The extended 2-byte-per-sample raw PPM/PGM formats are supported.
|
10 |
|
|
* The PBMPLUS library is NOT required to compile this software
|
11 |
|
|
* (but it is highly useful as a set of PPM image manipulation programs).
|
12 |
|
|
*
|
13 |
|
|
* These routines may need modification for non-Unix environments or
|
14 |
|
|
* specialized applications. As they stand, they assume output to
|
15 |
|
|
* an ordinary stdio stream.
|
16 |
|
|
*/
|
17 |
|
|
|
18 |
|
|
#include "cdjpeg.h" /* Common decls for cjpeg/djpeg applications */
|
19 |
|
|
|
20 |
|
|
#ifdef PPM_SUPPORTED
|
21 |
|
|
|
22 |
|
|
|
23 |
|
|
/*
|
24 |
|
|
* For 12-bit JPEG data, we either downscale the values to 8 bits
|
25 |
|
|
* (to write standard byte-per-sample PPM/PGM files), or output
|
26 |
|
|
* nonstandard word-per-sample PPM/PGM files. Downscaling is done
|
27 |
|
|
* if PPM_NORAWWORD is defined (this can be done in the Makefile
|
28 |
|
|
* or in jconfig.h).
|
29 |
|
|
* (When the core library supports data precision reduction, a cleaner
|
30 |
|
|
* implementation will be to ask for that instead.)
|
31 |
|
|
*/
|
32 |
|
|
|
33 |
|
|
#if BITS_IN_JSAMPLE == 8
|
34 |
|
|
#define PUTPPMSAMPLE(ptr,v) *ptr++ = (char) (v)
|
35 |
|
|
#define BYTESPERSAMPLE 1
|
36 |
|
|
#define PPM_MAXVAL 255
|
37 |
|
|
#else
|
38 |
|
|
#ifdef PPM_NORAWWORD
|
39 |
|
|
#define PUTPPMSAMPLE(ptr,v) *ptr++ = (char) ((v) >> (BITS_IN_JSAMPLE-8))
|
40 |
|
|
#define BYTESPERSAMPLE 1
|
41 |
|
|
#define PPM_MAXVAL 255
|
42 |
|
|
#else
|
43 |
|
|
/* The word-per-sample format always puts the LSB first. */
|
44 |
|
|
#define PUTPPMSAMPLE(ptr,v) \
|
45 |
|
|
{ register int val_ = v; \
|
46 |
|
|
*ptr++ = (char) (val_ & 0xFF); \
|
47 |
|
|
*ptr++ = (char) ((val_ >> 8) & 0xFF); \
|
48 |
|
|
}
|
49 |
|
|
#define BYTESPERSAMPLE 2
|
50 |
|
|
#define PPM_MAXVAL ((1<<BITS_IN_JSAMPLE)-1)
|
51 |
|
|
#endif
|
52 |
|
|
#endif
|
53 |
|
|
|
54 |
|
|
|
55 |
|
|
/*
|
56 |
|
|
* When JSAMPLE is the same size as char, we can just fwrite() the
|
57 |
|
|
* decompressed data to the PPM or PGM file. On PCs, in order to make this
|
58 |
|
|
* work the output buffer must be allocated in near data space, because we are
|
59 |
|
|
* assuming small-data memory model wherein fwrite() can't reach far memory.
|
60 |
|
|
* If you need to process very wide images on a PC, you might have to compile
|
61 |
|
|
* in large-memory model, or else replace fwrite() with a putc() loop ---
|
62 |
|
|
* which will be much slower.
|
63 |
|
|
*/
|
64 |
|
|
|
65 |
|
|
|
66 |
|
|
/* Private version of data destination object */
|
67 |
|
|
|
68 |
|
|
typedef struct {
|
69 |
|
|
struct djpeg_dest_struct pub; /* public fields */
|
70 |
|
|
|
71 |
|
|
/* Usually these two pointers point to the same place: */
|
72 |
|
|
char *iobuffer; /* fwrite's I/O buffer */
|
73 |
|
|
JSAMPROW pixrow; /* decompressor output buffer */
|
74 |
|
|
size_t buffer_width; /* width of I/O buffer */
|
75 |
|
|
JDIMENSION samples_per_row; /* JSAMPLEs per output row */
|
76 |
|
|
} ppm_dest_struct;
|
77 |
|
|
|
78 |
|
|
typedef ppm_dest_struct * ppm_dest_ptr;
|
79 |
|
|
|
80 |
|
|
|
81 |
|
|
/*
|
82 |
|
|
* Write some pixel data.
|
83 |
|
|
* In this module rows_supplied will always be 1.
|
84 |
|
|
*
|
85 |
|
|
* put_pixel_rows handles the "normal" 8-bit case where the decompressor
|
86 |
|
|
* output buffer is physically the same as the fwrite buffer.
|
87 |
|
|
*/
|
88 |
|
|
|
89 |
|
|
METHODDEF(void)
|
90 |
|
|
put_pixel_rows (j_decompress_ptr cinfo, djpeg_dest_ptr dinfo,
|
91 |
|
|
JDIMENSION rows_supplied)
|
92 |
|
|
{
|
93 |
|
|
ppm_dest_ptr dest = (ppm_dest_ptr) dinfo;
|
94 |
|
|
|
95 |
|
|
(void) JFWRITE(dest->pub.output_file, dest->iobuffer, dest->buffer_width);
|
96 |
|
|
}
|
97 |
|
|
|
98 |
|
|
|
99 |
|
|
/*
|
100 |
|
|
* This code is used when we have to copy the data and apply a pixel
|
101 |
|
|
* format translation. Typically this only happens in 12-bit mode.
|
102 |
|
|
*/
|
103 |
|
|
|
104 |
|
|
METHODDEF(void)
|
105 |
|
|
copy_pixel_rows (j_decompress_ptr cinfo, djpeg_dest_ptr dinfo,
|
106 |
|
|
JDIMENSION rows_supplied)
|
107 |
|
|
{
|
108 |
|
|
ppm_dest_ptr dest = (ppm_dest_ptr) dinfo;
|
109 |
|
|
register char * bufferptr;
|
110 |
|
|
register JSAMPROW ptr;
|
111 |
|
|
register JDIMENSION col;
|
112 |
|
|
|
113 |
|
|
ptr = dest->pub.buffer[0];
|
114 |
|
|
bufferptr = dest->iobuffer;
|
115 |
|
|
for (col = dest->samples_per_row; col > 0; col--) {
|
116 |
|
|
PUTPPMSAMPLE(bufferptr, GETJSAMPLE(*ptr++));
|
117 |
|
|
}
|
118 |
|
|
(void) JFWRITE(dest->pub.output_file, dest->iobuffer, dest->buffer_width);
|
119 |
|
|
}
|
120 |
|
|
|
121 |
|
|
|
122 |
|
|
/*
|
123 |
|
|
* Write some pixel data when color quantization is in effect.
|
124 |
|
|
* We have to demap the color index values to straight data.
|
125 |
|
|
*/
|
126 |
|
|
|
127 |
|
|
METHODDEF(void)
|
128 |
|
|
put_demapped_rgb (j_decompress_ptr cinfo, djpeg_dest_ptr dinfo,
|
129 |
|
|
JDIMENSION rows_supplied)
|
130 |
|
|
{
|
131 |
|
|
ppm_dest_ptr dest = (ppm_dest_ptr) dinfo;
|
132 |
|
|
register char * bufferptr;
|
133 |
|
|
register int pixval;
|
134 |
|
|
register JSAMPROW ptr;
|
135 |
|
|
register JSAMPROW color_map0 = cinfo->colormap[0];
|
136 |
|
|
register JSAMPROW color_map1 = cinfo->colormap[1];
|
137 |
|
|
register JSAMPROW color_map2 = cinfo->colormap[2];
|
138 |
|
|
register JDIMENSION col;
|
139 |
|
|
|
140 |
|
|
ptr = dest->pub.buffer[0];
|
141 |
|
|
bufferptr = dest->iobuffer;
|
142 |
|
|
for (col = cinfo->output_width; col > 0; col--) {
|
143 |
|
|
pixval = GETJSAMPLE(*ptr++);
|
144 |
|
|
PUTPPMSAMPLE(bufferptr, GETJSAMPLE(color_map0[pixval]));
|
145 |
|
|
PUTPPMSAMPLE(bufferptr, GETJSAMPLE(color_map1[pixval]));
|
146 |
|
|
PUTPPMSAMPLE(bufferptr, GETJSAMPLE(color_map2[pixval]));
|
147 |
|
|
}
|
148 |
|
|
(void) JFWRITE(dest->pub.output_file, dest->iobuffer, dest->buffer_width);
|
149 |
|
|
}
|
150 |
|
|
|
151 |
|
|
|
152 |
|
|
METHODDEF(void)
|
153 |
|
|
put_demapped_gray (j_decompress_ptr cinfo, djpeg_dest_ptr dinfo,
|
154 |
|
|
JDIMENSION rows_supplied)
|
155 |
|
|
{
|
156 |
|
|
ppm_dest_ptr dest = (ppm_dest_ptr) dinfo;
|
157 |
|
|
register char * bufferptr;
|
158 |
|
|
register JSAMPROW ptr;
|
159 |
|
|
register JSAMPROW color_map = cinfo->colormap[0];
|
160 |
|
|
register JDIMENSION col;
|
161 |
|
|
|
162 |
|
|
ptr = dest->pub.buffer[0];
|
163 |
|
|
bufferptr = dest->iobuffer;
|
164 |
|
|
for (col = cinfo->output_width; col > 0; col--) {
|
165 |
|
|
PUTPPMSAMPLE(bufferptr, GETJSAMPLE(color_map[GETJSAMPLE(*ptr++)]));
|
166 |
|
|
}
|
167 |
|
|
(void) JFWRITE(dest->pub.output_file, dest->iobuffer, dest->buffer_width);
|
168 |
|
|
}
|
169 |
|
|
|
170 |
|
|
|
171 |
|
|
/*
|
172 |
|
|
* Startup: write the file header.
|
173 |
|
|
*/
|
174 |
|
|
|
175 |
|
|
METHODDEF(void)
|
176 |
|
|
start_output_ppm (j_decompress_ptr cinfo, djpeg_dest_ptr dinfo)
|
177 |
|
|
{
|
178 |
|
|
ppm_dest_ptr dest = (ppm_dest_ptr) dinfo;
|
179 |
|
|
|
180 |
|
|
/* Emit file header */
|
181 |
|
|
switch (cinfo->out_color_space) {
|
182 |
|
|
case JCS_GRAYSCALE:
|
183 |
|
|
/* emit header for raw PGM format */
|
184 |
|
|
fprintf(dest->pub.output_file, "P5\n%ld %ld\n%d\n",
|
185 |
|
|
(long) cinfo->output_width, (long) cinfo->output_height,
|
186 |
|
|
PPM_MAXVAL);
|
187 |
|
|
break;
|
188 |
|
|
case JCS_RGB:
|
189 |
|
|
/* emit header for raw PPM format */
|
190 |
|
|
fprintf(dest->pub.output_file, "P6\n%ld %ld\n%d\n",
|
191 |
|
|
(long) cinfo->output_width, (long) cinfo->output_height,
|
192 |
|
|
PPM_MAXVAL);
|
193 |
|
|
break;
|
194 |
|
|
default:
|
195 |
|
|
ERREXIT(cinfo, JERR_PPM_COLORSPACE);
|
196 |
|
|
}
|
197 |
|
|
}
|
198 |
|
|
|
199 |
|
|
|
200 |
|
|
/*
|
201 |
|
|
* Finish up at the end of the file.
|
202 |
|
|
*/
|
203 |
|
|
|
204 |
|
|
METHODDEF(void)
|
205 |
|
|
finish_output_ppm (j_decompress_ptr cinfo, djpeg_dest_ptr dinfo)
|
206 |
|
|
{
|
207 |
|
|
/* Make sure we wrote the output file OK */
|
208 |
|
|
fflush(dinfo->output_file);
|
209 |
|
|
if (ferror(dinfo->output_file))
|
210 |
|
|
ERREXIT(cinfo, JERR_FILE_WRITE);
|
211 |
|
|
}
|
212 |
|
|
|
213 |
|
|
|
214 |
|
|
/*
|
215 |
|
|
* The module selection routine for PPM format output.
|
216 |
|
|
*/
|
217 |
|
|
|
218 |
|
|
GLOBAL(djpeg_dest_ptr)
|
219 |
|
|
jinit_write_ppm (j_decompress_ptr cinfo)
|
220 |
|
|
{
|
221 |
|
|
ppm_dest_ptr dest;
|
222 |
|
|
|
223 |
|
|
/* Create module interface object, fill in method pointers */
|
224 |
|
|
dest = (ppm_dest_ptr)
|
225 |
|
|
(*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE,
|
226 |
|
|
SIZEOF(ppm_dest_struct));
|
227 |
|
|
dest->pub.start_output = start_output_ppm;
|
228 |
|
|
dest->pub.finish_output = finish_output_ppm;
|
229 |
|
|
|
230 |
|
|
/* Calculate output image dimensions so we can allocate space */
|
231 |
|
|
jpeg_calc_output_dimensions(cinfo);
|
232 |
|
|
|
233 |
|
|
/* Create physical I/O buffer. Note we make this near on a PC. */
|
234 |
|
|
dest->samples_per_row = cinfo->output_width * cinfo->out_color_components;
|
235 |
|
|
dest->buffer_width = dest->samples_per_row * (BYTESPERSAMPLE * SIZEOF(char));
|
236 |
|
|
dest->iobuffer = (char *) (*cinfo->mem->alloc_small)
|
237 |
|
|
((j_common_ptr) cinfo, JPOOL_IMAGE, dest->buffer_width);
|
238 |
|
|
|
239 |
|
|
if (cinfo->quantize_colors || BITS_IN_JSAMPLE != 8 ||
|
240 |
|
|
SIZEOF(JSAMPLE) != SIZEOF(char)) {
|
241 |
|
|
/* When quantizing, we need an output buffer for colormap indexes
|
242 |
|
|
* that's separate from the physical I/O buffer. We also need a
|
243 |
|
|
* separate buffer if pixel format translation must take place.
|
244 |
|
|
*/
|
245 |
|
|
dest->pub.buffer = (*cinfo->mem->alloc_sarray)
|
246 |
|
|
((j_common_ptr) cinfo, JPOOL_IMAGE,
|
247 |
|
|
cinfo->output_width * cinfo->output_components, (JDIMENSION) 1);
|
248 |
|
|
dest->pub.buffer_height = 1;
|
249 |
|
|
if (! cinfo->quantize_colors)
|
250 |
|
|
dest->pub.put_pixel_rows = copy_pixel_rows;
|
251 |
|
|
else if (cinfo->out_color_space == JCS_GRAYSCALE)
|
252 |
|
|
dest->pub.put_pixel_rows = put_demapped_gray;
|
253 |
|
|
else
|
254 |
|
|
dest->pub.put_pixel_rows = put_demapped_rgb;
|
255 |
|
|
} else {
|
256 |
|
|
/* We will fwrite() directly from decompressor output buffer. */
|
257 |
|
|
/* Synthesize a JSAMPARRAY pointer structure */
|
258 |
|
|
/* Cast here implies near->far pointer conversion on PCs */
|
259 |
|
|
dest->pixrow = (JSAMPROW) dest->iobuffer;
|
260 |
|
|
dest->pub.buffer = & dest->pixrow;
|
261 |
|
|
dest->pub.buffer_height = 1;
|
262 |
|
|
dest->pub.put_pixel_rows = put_pixel_rows;
|
263 |
|
|
}
|
264 |
|
|
|
265 |
|
|
return (djpeg_dest_ptr) dest;
|
266 |
|
|
}
|
267 |
|
|
|
268 |
|
|
#endif /* PPM_SUPPORTED */
|