1 : /*****************************************************************************
2 : * $Id: $
3 : *
4 : * Project: Creates a jpeg header
5 : * Purpose: Abreviated JPEG support
6 : * Author: Ivan Lucena, ivan.lucena@pmldnet.com
7 : *
8 : ******************************************************************************
9 : * Copyright (c) 2007, Ivan Lucena
10 : *
11 : * Permission is hereby granted, free of charge, to any person obtaining a
12 : * copy of this software and associated documentation files ( the "Software" ),
13 : * to deal in the Software without restriction, including without limitation
14 : * the rights to use, copy, modify, merge, publish, distribute, sublicense,
15 : * and/or sell copies of the Software, and to permit persons to whom the
16 : * Software is furnished to do so, subject to the following conditions:
17 : *
18 : * The above copyright notice and this permission notice shall be included
19 : * in all copies or substantial portions of the Software.
20 : *
21 : * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
22 : * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
23 : * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
24 : * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
25 : * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
26 : * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
27 : * DEALINGS IN THE SOFTWARE.
28 : *
29 : *****************************************************************************/
30 :
31 : #include "JpegHelper.h"
32 :
33 : static const GByte JPGHLP_1DC_Codes[] = {
34 : 0, 1, 5, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0,
35 : };
36 :
37 : static const GByte JPGHLP_1AC_Codes[] = {
38 : 0, 2, 1, 3, 3, 2, 4, 3, 5, 5, 4, 4, 0, 0, 1, 125,
39 : };
40 :
41 : static const GByte JPGHLP_1DC_Symbols[] = {
42 : 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11,
43 : };
44 :
45 : static const GByte JPGHLP_1AC_Symbols[] = {
46 : 0x01, 0x02, 0x03, 0x00, 0x04, 0x11, 0x05, 0x12,
47 : 0x21, 0x31, 0x41, 0x06, 0x13, 0x51, 0x61, 0x07,
48 : 0x22, 0x71, 0x14, 0x32, 0x81, 0x91, 0xa1, 0x08,
49 : 0x23, 0x42, 0xb1, 0xc1, 0x15, 0x52, 0xd1, 0xf0,
50 : 0x24, 0x33, 0x62, 0x72, 0x82, 0x09, 0x0a, 0x16,
51 : 0x17, 0x18, 0x19, 0x1a, 0x25, 0x26, 0x27, 0x28,
52 : 0x29, 0x2a, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39,
53 : 0x3a, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, 0x49,
54 : 0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59,
55 : 0x5a, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69,
56 : 0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79,
57 : 0x7a, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88, 0x89,
58 : 0x8a, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98,
59 : 0x99, 0x9a, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7,
60 : 0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6,
61 : 0xb7, 0xb8, 0xb9, 0xba, 0xc2, 0xc3, 0xc4, 0xc5,
62 : 0xc6, 0xc7, 0xc8, 0xc9, 0xCA, 0xd2, 0xd3, 0xd4,
63 : 0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda, 0xe1, 0xe2,
64 : 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea,
65 : 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8,
66 : 0xf9, 0xfa,
67 : };
68 :
69 : static const GByte JPGHLP_2AC_Codes[] = {
70 : 0, 2, 1, 2, 4, 4, 3, 4, 7, 5, 4, 4, 0, 1, 2, 119,
71 : };
72 :
73 : static const GByte JPGHLP_2DC_Codes[] = {
74 : 0, 3, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0,
75 : };
76 :
77 : static const GByte JPGHLP_2DC_Symbols[] = {
78 : 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11,
79 : };
80 :
81 : static const GByte JPGHLP_2AC_Symbols[] = {
82 : 0x00, 0x01, 0x02, 0x03, 0x11, 0x04, 0x05, 0x21,
83 : 0x31, 0x06, 0x12, 0x41, 0x51, 0x07, 0x61, 0x71,
84 : 0x13, 0x22, 0x32, 0x81, 0x08, 0x14, 0x42, 0x91,
85 : 0xa1, 0xb1, 0xc1, 0x09, 0x23, 0x33, 0x52, 0xf0,
86 : 0x15, 0x62, 0x72, 0xd1, 0x0a, 0x16, 0x24, 0x34,
87 : 0xe1, 0x25, 0xf1, 0x17, 0x18, 0x19, 0x1a, 0x26,
88 : 0x27, 0x28, 0x29, 0x2a, 0x35, 0x36, 0x37, 0x38,
89 : 0x39, 0x3a, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48,
90 : 0x49, 0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58,
91 : 0x59, 0x5a, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68,
92 : 0x69, 0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78,
93 : 0x79, 0x7a, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87,
94 : 0x88, 0x89, 0x8a, 0x92, 0x93, 0x94, 0x95, 0x96,
95 : 0x97, 0x98, 0x99, 0x9a, 0xa2, 0xa3, 0xa4, 0xa5,
96 : 0xa6, 0xa7, 0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4,
97 : 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, 0xc2, 0xc3,
98 : 0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xCA, 0xd2,
99 : 0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda,
100 : 0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9,
101 : 0xea, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8,
102 : 0xf9, 0xfa,
103 : };
104 :
105 : static const GByte JPGHLP_DQT_luminace[64] = {
106 : 10, 7, 6, 10, 14, 24, 31, 37,
107 : 7, 7, 8, 11, 16, 35, 36, 33,
108 : 8, 8, 10, 14, 24, 34, 41, 34,
109 : 8, 10, 13, 17, 31, 52, 48, 37,
110 : 11, 13, 22, 34, 41, 65, 62, 46,
111 : 14, 21, 33, 38, 49, 62, 68, 55,
112 : 29, 38, 47, 52, 62, 73, 72, 61,
113 : 43, 55, 57, 59, 67, 60, 62, 59
114 : };
115 :
116 : static const GByte JPGHLP_DQT_chrominance[64] = {
117 : 10, 11, 14, 28, 59, 59, 59, 59,
118 : 11, 13, 16, 40, 59, 59, 59, 59,
119 : 14, 16, 34, 59, 59, 59, 59, 59,
120 : 28, 40, 59, 59, 59, 59, 59, 59,
121 : 59, 59, 59, 59, 59, 59, 59, 59,
122 : 59, 59, 59, 59, 59, 59, 59, 59,
123 : 59, 59, 59, 59, 59, 59, 59, 59,
124 : 59, 59, 59, 59, 59, 59, 59, 59
125 : };
126 :
127 : static const GByte ZIGZAG[64] = {
128 : 0, 1, 5, 6, 14, 15, 27, 28,
129 : 2, 4, 7, 13, 16, 26, 29, 42,
130 : 3, 8, 12, 17, 25, 30, 41, 43,
131 : 9, 11, 18, 24, 31, 40, 44, 53,
132 : 10, 19, 23, 32, 39, 45, 52, 54,
133 : 20, 22, 33, 38, 46, 51, 55, 60,
134 : 21, 34, 37, 47, 50, 56, 59, 61,
135 : 35, 36, 48, 49, 57, 58, 62, 63
136 : };
137 :
138 : #define ZIGZAGCPY(ou, in) \
139 : { int i; for( i = 0; i < 64; i++ ) ou[ZIGZAG[i]] = in[i]; }
140 :
141 : #define ADJUST(tb, op, vl) \
142 : { int i; for( i = 0; i < 64; i++ ) tb[i] = (GByte) (tb[i] op vl); }
143 :
144 8 : int JPGHLP_HeaderMaker( GByte *pabyBuffer,
145 : const int nCols,
146 : const int nRows,
147 : const int nComponents,
148 : const int nRestart,
149 : const int nQuality )
150 : {
151 : int i;
152 :
153 8 : GByte *pabNext = pabyBuffer;
154 :
155 : // ------------------------------------------------------------------------
156 : // Start of Image
157 : // ------------------------------------------------------------------------
158 :
159 8 : *( pabNext++ ) = 0xFF; // Tag Mark
160 8 : *( pabNext++ ) = 0xD8; // SOI
161 :
162 : // ------------------------------------------------------------------------
163 : // Application Segment
164 : // ------------------------------------------------------------------------
165 :
166 8 : *( pabNext++ ) = 0xFF; // Tag Mark
167 8 : *( pabNext++ ) = 0xE0; // APP0
168 8 : *( pabNext++ ) = 0x00; // Segment Length (msb)
169 8 : *( pabNext++ ) = 0x10; // Segment Length (lsb)
170 8 : *( pabNext++ ) = 0x4a; // 'J'
171 8 : *( pabNext++ ) = 0x46; // 'F'
172 8 : *( pabNext++ ) = 0x49; // 'I'
173 8 : *( pabNext++ ) = 0x46; // 'F'
174 8 : *( pabNext++ ) = 0x00; // '\0'
175 8 : *( pabNext++ ) = 0x01; // Version 1
176 8 : *( pabNext++ ) = 0x01; // Sub Version 1
177 8 : *( pabNext++ ) = 0x00; // Pixels per inch, 4 Bits for X, 4 Bits for Y
178 8 : *( pabNext++ ) = 0x00; // Horizontal Pixel Density (msb)
179 8 : *( pabNext++ ) = 0x01; // Horizontal Pixel Density (lsb)
180 8 : *( pabNext++ ) = 0x00; // Vertical Pixel Density (msb)
181 8 : *( pabNext++ ) = 0x01; // Vertical Pixel Density (lsb)
182 8 : *( pabNext++ ) = 0x00; // Thumbnail Width
183 8 : *( pabNext++ ) = 0x00; // Thumbnail Height
184 :
185 : // ------------------------------------------------------------------------
186 : // Quantization Table Segment
187 : // ------------------------------------------------------------------------
188 :
189 : GByte abQuantTables[2][64];
190 8 : ZIGZAGCPY( abQuantTables[0], JPGHLP_DQT_luminace );
191 8 : ZIGZAGCPY( abQuantTables[1], JPGHLP_DQT_chrominance );
192 :
193 8 : if( nQuality == 30 )
194 : {
195 8 : ADJUST( abQuantTables[0], *, 0.5 );
196 8 : ADJUST( abQuantTables[1], *, 0.5 );
197 : }
198 :
199 20 : for( i = 0; i < 2 && i < nComponents; i++ )
200 : {
201 12 : *( pabNext++ ) = 0xFF; // Tag Mark
202 12 : *( pabNext++ ) = 0xDB; // DQT
203 12 : *( pabNext++ ) = 0; // Segment Length (msb)
204 12 : *( pabNext++ ) = 67; // Length (msb)
205 12 : *( pabNext++ ) = (GByte) i; // Table ID
206 12 : memcpy( pabNext, abQuantTables[i], 64 );
207 12 : pabNext += 64;
208 : }
209 :
210 : // ------------------------------------------------------------------------
211 : // Start Of Frame Segment
212 : // ------------------------------------------------------------------------
213 :
214 8 : *( pabNext++ ) = 0xFF;
215 8 : *( pabNext++ ) = 0xC0; // SOF
216 8 : *( pabNext++ ) = 0; // Segment Length (msb)
217 8 : if ( nComponents > 1 )
218 4 : *( pabNext++ ) = 17; // Segment Length (lsb)
219 : else
220 4 : *( pabNext++ ) = 11; // Segment Length (lsb)
221 8 : *( pabNext++ ) = 8; // 8-bit Precision
222 8 : *( pabNext++ ) = (GByte) (nRows >> 8); // Height in rows (msb)
223 8 : *( pabNext++ ) = (GByte) nRows;// Height in rows (lsb)
224 8 : *( pabNext++ ) = (GByte) (nCols >> 8); // Width in columns (msb)
225 8 : *( pabNext++ ) = (GByte) nCols;// Width in columns (lsb)
226 8 : *( pabNext++ ) = (GByte) nComponents;// Number of components
227 8 : *( pabNext++ ) = 0; // Component ID
228 8 : *( pabNext++ ) = 0x21; // Hozontal/Vertical Sampling
229 8 : *( pabNext++ ) = 0; // Quantization table ID
230 8 : if ( nComponents > 1 )
231 : {
232 4 : *( pabNext++ ) = 1; // Component ID
233 4 : *( pabNext++ ) = 0x11; // Hozontal/Vertical Sampling
234 4 : *( pabNext++ ) = 1; // Quantization table ID
235 4 : *( pabNext++ ) = 2; // Component ID
236 4 : *( pabNext++ ) = 0x11; // Hozontal/Vertical Sampling
237 4 : *( pabNext++ ) = 1; // Quantization table ID
238 : }
239 :
240 : // ------------------------------------------------------------------------
241 : // Huffman Table Segments
242 : // ------------------------------------------------------------------------
243 :
244 : const GByte *pabHuffTab[2][4];
245 8 : pabHuffTab[0][0] = JPGHLP_1DC_Codes;
246 8 : pabHuffTab[0][1] = JPGHLP_1AC_Codes;
247 8 : pabHuffTab[0][2] = JPGHLP_1DC_Symbols;
248 8 : pabHuffTab[0][3] = JPGHLP_1AC_Symbols;
249 :
250 8 : pabHuffTab[1][0] = JPGHLP_2DC_Codes;
251 8 : pabHuffTab[1][1] = JPGHLP_2AC_Codes;
252 8 : pabHuffTab[1][2] = JPGHLP_2DC_Symbols;
253 8 : pabHuffTab[1][3] = JPGHLP_2AC_Symbols;
254 :
255 : int pnHTs[2][4];
256 8 : pnHTs[0][0] = sizeof(JPGHLP_1DC_Codes);
257 8 : pnHTs[0][1] = sizeof(JPGHLP_1AC_Codes);
258 8 : pnHTs[0][2] = sizeof(JPGHLP_1DC_Symbols);
259 8 : pnHTs[0][3] = sizeof(JPGHLP_1AC_Symbols);
260 :
261 8 : pnHTs[1][0] = sizeof(JPGHLP_2DC_Codes);
262 8 : pnHTs[1][1] = sizeof(JPGHLP_2AC_Codes);
263 8 : pnHTs[1][2] = sizeof(JPGHLP_2DC_Symbols);
264 8 : pnHTs[1][3] = sizeof(JPGHLP_2AC_Symbols);
265 :
266 : int j, k;
267 20 : for( i = 0; i < 2 && i < nComponents; i++ )
268 : {
269 36 : for( j = 0; j < 2; j++ )
270 : {
271 24 : k = j + 2;
272 24 : int nCodes = pnHTs[i][j];
273 24 : int nSymbols = pnHTs[i][k];
274 24 : *( pabNext++ ) = 0xFF; // Tag Mark
275 24 : *( pabNext++ ) = 0xc4; // DHT
276 24 : *( pabNext++ ) = 0; // Segment Length (msb)
277 24 : *( pabNext++ ) = (GByte) (3 + nCodes + nSymbols); // Segment Length (lsb)
278 24 : *( pabNext++ ) = (GByte) ((j << 4) | i); // Table ID
279 24 : memcpy( pabNext, pabHuffTab[i][j], nCodes );
280 24 : pabNext += nCodes;
281 24 : memcpy( pabNext, pabHuffTab[i][k], nSymbols );
282 24 : pabNext += nSymbols;
283 : }
284 : }
285 :
286 : // ------------------------------------------------------------------------
287 : // Start Of Scan Segment
288 : // ------------------------------------------------------------------------
289 :
290 8 : *( pabNext++ ) = 0xFF; // Tag Mark
291 8 : *( pabNext++ ) = 0xDA; // SOS
292 8 : if (nComponents > 1 )
293 : {
294 4 : *( pabNext++ ) = 0; // Segment Length (msb)
295 4 : *( pabNext++ ) = 12; // Segment Length (lsb)
296 4 : *( pabNext++ ) = 3; // Number of components
297 4 : *( pabNext++ ) = 0; // Components 0
298 4 : *( pabNext++ ) = 0; // Huffman table ID
299 4 : *( pabNext++ ) = 1; // Components 1
300 4 : *( pabNext++ ) = 0x11; // Huffman table ID
301 4 : *( pabNext++ ) = 2; // Components 2
302 4 : *( pabNext++ ) = 0x11; // Huffman table ID
303 : }
304 : else
305 : {
306 4 : *( pabNext++ ) = 0; // Segment Length (msb)
307 4 : *( pabNext++ ) = 8; // Segment Length (lsb)
308 4 : *( pabNext++ ) = 1; // Number of components
309 4 : *( pabNext++ ) = 0; // Components 0
310 4 : *( pabNext++ ) = 0; // Huffman table ID
311 : }
312 8 : *( pabNext++ ) = 0; // First DCT coefficient
313 8 : *( pabNext++ ) = 63; // Last DCT coefficient
314 8 : *( pabNext++ ) = 0; // Spectral selection
315 :
316 8 : return pabNext - pabyBuffer;
317 : }
|