Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
6417 | ashmew2 | 1 | /* |
2 | * rdcolmap.c |
||
3 | * |
||
4 | * Copyright (C) 1994-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 implements djpeg's "-map file" switch. It reads a source image |
||
9 | * and constructs a colormap to be supplied to the JPEG decompressor. |
||
10 | * |
||
11 | * Currently, these file formats are supported for the map file: |
||
12 | * GIF: the contents of the GIF's global colormap are used. |
||
13 | * PPM (either text or raw flavor): the entire file is read and |
||
14 | * each unique pixel value is entered in the map. |
||
15 | * Note that reading a large PPM file will be horrendously slow. |
||
16 | * Typically, a PPM-format map file should contain just one pixel |
||
17 | * of each desired color. Such a file can be extracted from an |
||
18 | * ordinary image PPM file with ppmtomap(1). |
||
19 | * |
||
20 | * Rescaling a PPM that has a maxval unequal to MAXJSAMPLE is not |
||
21 | * currently implemented. |
||
22 | */ |
||
23 | |||
24 | #include "cdjpeg.h" /* Common decls for cjpeg/djpeg applications */ |
||
25 | |||
26 | #ifdef QUANT_2PASS_SUPPORTED /* otherwise can't quantize to supplied map */ |
||
27 | |||
28 | /* Portions of this code are based on the PBMPLUS library, which is: |
||
29 | ** |
||
30 | ** Copyright (C) 1988 by Jef Poskanzer. |
||
31 | ** |
||
32 | ** Permission to use, copy, modify, and distribute this software and its |
||
33 | ** documentation for any purpose and without fee is hereby granted, provided |
||
34 | ** that the above copyright notice appear in all copies and that both that |
||
35 | ** copyright notice and this permission notice appear in supporting |
||
36 | ** documentation. This software is provided "as is" without express or |
||
37 | ** implied warranty. |
||
38 | */ |
||
39 | |||
40 | |||
41 | /* |
||
42 | * Add a (potentially) new color to the color map. |
||
43 | */ |
||
44 | |||
45 | LOCAL(void) |
||
46 | add_map_entry (j_decompress_ptr cinfo, int R, int G, int B) |
||
47 | { |
||
48 | JSAMPROW colormap0 = cinfo->colormap[0]; |
||
49 | JSAMPROW colormap1 = cinfo->colormap[1]; |
||
50 | JSAMPROW colormap2 = cinfo->colormap[2]; |
||
51 | int ncolors = cinfo->actual_number_of_colors; |
||
52 | int index; |
||
53 | |||
54 | /* Check for duplicate color. */ |
||
55 | for (index = 0; index < ncolors; index++) { |
||
56 | if (GETJSAMPLE(colormap0[index]) == R && |
||
57 | GETJSAMPLE(colormap1[index]) == G && |
||
58 | GETJSAMPLE(colormap2[index]) == B) |
||
59 | return; /* color is already in map */ |
||
60 | } |
||
61 | |||
62 | /* Check for map overflow. */ |
||
63 | if (ncolors >= (MAXJSAMPLE+1)) |
||
64 | ERREXIT1(cinfo, JERR_QUANT_MANY_COLORS, (MAXJSAMPLE+1)); |
||
65 | |||
66 | /* OK, add color to map. */ |
||
67 | colormap0[ncolors] = (JSAMPLE) R; |
||
68 | colormap1[ncolors] = (JSAMPLE) G; |
||
69 | colormap2[ncolors] = (JSAMPLE) B; |
||
70 | cinfo->actual_number_of_colors++; |
||
71 | } |
||
72 | |||
73 | |||
74 | /* |
||
75 | * Extract color map from a GIF file. |
||
76 | */ |
||
77 | |||
78 | LOCAL(void) |
||
79 | read_gif_map (j_decompress_ptr cinfo, FILE * infile) |
||
80 | { |
||
81 | int header[13]; |
||
82 | int i, colormaplen; |
||
83 | int R, G, B; |
||
84 | |||
85 | /* Initial 'G' has already been read by read_color_map */ |
||
86 | /* Read the rest of the GIF header and logical screen descriptor */ |
||
87 | for (i = 1; i < 13; i++) { |
||
88 | if ((header[i] = getc(infile)) == EOF) |
||
89 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
90 | } |
||
91 | |||
92 | /* Verify GIF Header */ |
||
93 | if (header[1] != 'I' || header[2] != 'F') |
||
94 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
95 | |||
96 | /* There must be a global color map. */ |
||
97 | if ((header[10] & 0x80) == 0) |
||
98 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
99 | |||
100 | /* OK, fetch it. */ |
||
101 | colormaplen = 2 << (header[10] & 0x07); |
||
102 | |||
103 | for (i = 0; i < colormaplen; i++) { |
||
104 | R = getc(infile); |
||
105 | G = getc(infile); |
||
106 | B = getc(infile); |
||
107 | if (R == EOF || G == EOF || B == EOF) |
||
108 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
109 | add_map_entry(cinfo, |
||
110 | R << (BITS_IN_JSAMPLE-8), |
||
111 | G << (BITS_IN_JSAMPLE-8), |
||
112 | B << (BITS_IN_JSAMPLE-8)); |
||
113 | } |
||
114 | } |
||
115 | |||
116 | |||
117 | /* Support routines for reading PPM */ |
||
118 | |||
119 | |||
120 | LOCAL(int) |
||
121 | pbm_getc (FILE * infile) |
||
122 | /* Read next char, skipping over any comments */ |
||
123 | /* A comment/newline sequence is returned as a newline */ |
||
124 | { |
||
125 | register int ch; |
||
126 | |||
127 | ch = getc(infile); |
||
128 | if (ch == '#') { |
||
129 | do { |
||
130 | ch = getc(infile); |
||
131 | } while (ch != '\n' && ch != EOF); |
||
132 | } |
||
133 | return ch; |
||
134 | } |
||
135 | |||
136 | |||
137 | LOCAL(unsigned int) |
||
138 | read_pbm_integer (j_decompress_ptr cinfo, FILE * infile) |
||
139 | /* Read an unsigned decimal integer from the PPM file */ |
||
140 | /* Swallows one trailing character after the integer */ |
||
141 | /* Note that on a 16-bit-int machine, only values up to 64k can be read. */ |
||
142 | /* This should not be a problem in practice. */ |
||
143 | { |
||
144 | register int ch; |
||
145 | register unsigned int val; |
||
146 | |||
147 | /* Skip any leading whitespace */ |
||
148 | do { |
||
149 | ch = pbm_getc(infile); |
||
150 | if (ch == EOF) |
||
151 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
152 | } while (ch == ' ' || ch == '\t' || ch == '\n' || ch == '\r'); |
||
153 | |||
154 | if (ch < '0' || ch > '9') |
||
155 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
156 | |||
157 | val = ch - '0'; |
||
158 | while ((ch = pbm_getc(infile)) >= '0' && ch <= '9') { |
||
159 | val *= 10; |
||
160 | val += ch - '0'; |
||
161 | } |
||
162 | return val; |
||
163 | } |
||
164 | |||
165 | |||
166 | /* |
||
167 | * Extract color map from a PPM file. |
||
168 | */ |
||
169 | |||
170 | LOCAL(void) |
||
171 | read_ppm_map (j_decompress_ptr cinfo, FILE * infile) |
||
172 | { |
||
173 | int c; |
||
174 | unsigned int w, h, maxval, row, col; |
||
175 | int R, G, B; |
||
176 | |||
177 | /* Initial 'P' has already been read by read_color_map */ |
||
178 | c = getc(infile); /* save format discriminator for a sec */ |
||
179 | |||
180 | /* while we fetch the remaining header info */ |
||
181 | w = read_pbm_integer(cinfo, infile); |
||
182 | h = read_pbm_integer(cinfo, infile); |
||
183 | maxval = read_pbm_integer(cinfo, infile); |
||
184 | |||
185 | if (w <= 0 || h <= 0 || maxval <= 0) /* error check */ |
||
186 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
187 | |||
188 | /* For now, we don't support rescaling from an unusual maxval. */ |
||
189 | if (maxval != (unsigned int) MAXJSAMPLE) |
||
190 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
191 | |||
192 | switch (c) { |
||
193 | case '3': /* it's a text-format PPM file */ |
||
194 | for (row = 0; row < h; row++) { |
||
195 | for (col = 0; col < w; col++) { |
||
196 | R = read_pbm_integer(cinfo, infile); |
||
197 | G = read_pbm_integer(cinfo, infile); |
||
198 | B = read_pbm_integer(cinfo, infile); |
||
199 | add_map_entry(cinfo, R, G, B); |
||
200 | } |
||
201 | } |
||
202 | break; |
||
203 | |||
204 | case '6': /* it's a raw-format PPM file */ |
||
205 | for (row = 0; row < h; row++) { |
||
206 | for (col = 0; col < w; col++) { |
||
207 | R = getc(infile); |
||
208 | G = getc(infile); |
||
209 | B = getc(infile); |
||
210 | if (R == EOF || G == EOF || B == EOF) |
||
211 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
212 | add_map_entry(cinfo, R, G, B); |
||
213 | } |
||
214 | } |
||
215 | break; |
||
216 | |||
217 | default: |
||
218 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
219 | break; |
||
220 | } |
||
221 | } |
||
222 | |||
223 | |||
224 | /* |
||
225 | * Main entry point from djpeg.c. |
||
226 | * Input: opened input file (from file name argument on command line). |
||
227 | * Output: colormap and actual_number_of_colors fields are set in cinfo. |
||
228 | */ |
||
229 | |||
230 | GLOBAL(void) |
||
231 | read_color_map (j_decompress_ptr cinfo, FILE * infile) |
||
232 | { |
||
233 | /* Allocate space for a color map of maximum supported size. */ |
||
234 | cinfo->colormap = (*cinfo->mem->alloc_sarray) |
||
235 | ((j_common_ptr) cinfo, JPOOL_IMAGE, |
||
236 | (JDIMENSION) (MAXJSAMPLE+1), (JDIMENSION) 3); |
||
237 | cinfo->actual_number_of_colors = 0; /* initialize map to empty */ |
||
238 | |||
239 | /* Read first byte to determine file format */ |
||
240 | switch (getc(infile)) { |
||
241 | case 'G': |
||
242 | read_gif_map(cinfo, infile); |
||
243 | break; |
||
244 | case 'P': |
||
245 | read_ppm_map(cinfo, infile); |
||
246 | break; |
||
247 | default: |
||
248 | ERREXIT(cinfo, JERR_BAD_CMAP_FILE); |
||
249 | break; |
||
250 | } |
||
251 | } |
||
252 | |||
253 | #endif /* QUANT_2PASS_SUPPORTED */>>>>=>=>=>=>>><>><>><>>><>>> |