1 /*
2  * jddctmgr.c
3  *
4  * Copyright (C) 1994-1995, 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 the inverse-DCT management logic.
9  * This code selects a particular IDCT implementation to be used,
10  * and it performs related housekeeping chores.  No code in this file
11  * is executed per IDCT step, only during output pass setup.
12  *
13  * Note that the IDCT routines are responsible for performing coefficient
14  * dequantization as well as the IDCT proper.  This module sets up the
15  * dequantization multiplier table needed by the IDCT routine.
16  */
17 
18 #define JPEG_INTERNALS
19 #include "jinclude.h"
20 #include "jpeglib.h"
21 #include "jdct.h"		/* Private declarations for DCT subsystem */
22 
23 
24 /*
25  * The decompressor input side (jdinput.c) saves away the appropriate
26  * quantization table for each component at the start of the first scan
27  * involving that component.  (This is necessary in order to correctly
28  * decode files that reuse Q-table slots.)
29  * When we are ready to make an output pass, the saved Q-table is converted
30  * to a multiplier table that will actually be used by the IDCT routine.
31  * The multiplier table contents are IDCT-method-dependent.  To support
32  * application changes in IDCT method between scans, we can remake the
33  * multiplier tables if necessary.
34  * In buffered-image mode, the first output pass may occur before any data
35  * has been seen for some components, and thus before their Q-tables have
36  * been saved away.  To handle this case, multiplier tables are preset
37  * to zeroes; the result of the IDCT will be a neutral gray level.
38  */
39 
40 
41 /* Private subobject for this module */
42 
43 typedef struct {
44   struct jpeg_inverse_dct pub;	/* public fields */
45 
46   /* This array contains the IDCT method code that each multiplier table
47    * is currently set up for, or -1 if it's not yet set up.
48    * The actual multiplier tables are pointed to by dct_table in the
49    * per-component comp_info structures.
50    */
51   int cur_method[MAX_COMPONENTS];
52 } my_idct_controller;
53 
54 typedef my_idct_controller * my_idct_ptr;
55 
56 
57 /* Allocated multiplier tables: big enough for any supported variant */
58 
59 typedef union {
60   ISLOW_MULT_TYPE islow_array[DCTSIZE2];
61 #ifdef DCT_IFAST_SUPPORTED
62   IFAST_MULT_TYPE ifast_array[DCTSIZE2];
63 #endif
64 #ifdef DCT_FLOAT_SUPPORTED
65   FLOAT_MULT_TYPE float_array[DCTSIZE2];
66 #endif
67 } multiplier_table;
68 
69 
70 /* The current scaled-IDCT routines require ISLOW-style multiplier tables,
71  * so be sure to compile that code if either ISLOW or SCALING is requested.
72  */
73 #ifdef DCT_ISLOW_SUPPORTED
74 #define PROVIDE_ISLOW_TABLES
75 #else
76 #ifdef IDCT_SCALING_SUPPORTED
77 #define PROVIDE_ISLOW_TABLES
78 #endif
79 #endif
80 
81 
82 /*
83  * Prepare for an output pass.
84  * Here we select the proper IDCT routine for each component and build
85  * a matching multiplier table.
86  */
87 
88 METHODDEF void
start_pass(j_decompress_ptr cinfo)89 start_pass (j_decompress_ptr cinfo)
90 {
91   my_idct_ptr idct = (my_idct_ptr) cinfo->idct;
92   int ci, i;
93   jpeg_component_info *compptr;
94   int method = 0;
95   inverse_DCT_method_ptr method_ptr = NULL;
96   JQUANT_TBL * qtbl;
97 
98   for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
99        ci++, compptr++) {
100     /* Select the proper IDCT routine for this component's scaling */
101     switch (compptr->DCT_scaled_size) {
102 #ifdef IDCT_SCALING_SUPPORTED
103     case 1:
104       method_ptr = jpeg_idct_1x1;
105       method = JDCT_ISLOW;	/* jidctred uses islow-style table */
106       break;
107     case 2:
108       method_ptr = jpeg_idct_2x2;
109       method = JDCT_ISLOW;	/* jidctred uses islow-style table */
110       break;
111     case 4:
112       method_ptr = jpeg_idct_4x4;
113       method = JDCT_ISLOW;	/* jidctred uses islow-style table */
114       break;
115 #endif
116     case DCTSIZE:
117       switch (cinfo->dct_method) {
118 #ifdef DCT_ISLOW_SUPPORTED
119       case JDCT_ISLOW:
120 	method_ptr = jpeg_idct_islow;
121 	method = JDCT_ISLOW;
122 	break;
123 #endif
124 #ifdef DCT_IFAST_SUPPORTED
125       case JDCT_IFAST:
126 	method_ptr = jpeg_idct_ifast;
127 	method = JDCT_IFAST;
128 	break;
129 #endif
130 #ifdef DCT_FLOAT_SUPPORTED
131       case JDCT_FLOAT:
132 	method_ptr = jpeg_idct_float;
133 	method = JDCT_FLOAT;
134 	break;
135 #endif
136       default:
137 	ERREXIT(cinfo, JERR_NOT_COMPILED);
138 	break;
139       }
140       break;
141     default:
142       ERREXIT1(cinfo, JERR_BAD_DCTSIZE, compptr->DCT_scaled_size);
143       break;
144     }
145     idct->pub.inverse_DCT[ci] = method_ptr;
146     /* Create multiplier table from quant table.
147      * However, we can skip this if the component is uninteresting
148      * or if we already built the table.  Also, if no quant table
149      * has yet been saved for the component, we leave the
150      * multiplier table all-zero; we'll be reading zeroes from the
151      * coefficient controller's buffer anyway.
152      */
153     if (! compptr->component_needed || idct->cur_method[ci] == method)
154       continue;
155     qtbl = compptr->quant_table;
156     if (qtbl == NULL)		/* happens if no data yet for component */
157       continue;
158     idct->cur_method[ci] = method;
159     switch (method) {
160 #ifdef PROVIDE_ISLOW_TABLES
161     case JDCT_ISLOW:
162       {
163 	/* For LL&M IDCT method, multipliers are equal to raw quantization
164 	 * coefficients, but are stored in natural order as ints.
165 	 */
166 	ISLOW_MULT_TYPE * ismtbl = (ISLOW_MULT_TYPE *) compptr->dct_table;
167 	for (i = 0; i < DCTSIZE2; i++) {
168 	  ismtbl[i] = (ISLOW_MULT_TYPE) qtbl->quantval[jpeg_zigzag_order[i]];
169 	}
170       }
171       break;
172 #endif
173 #ifdef DCT_IFAST_SUPPORTED
174     case JDCT_IFAST:
175       {
176 	/* For AA&N IDCT method, multipliers are equal to quantization
177 	 * coefficients scaled by scalefactor[row]*scalefactor[col], where
178 	 *   scalefactor[0] = 1
179 	 *   scalefactor[k] = cos(k*PI/16) * sqrt(2)    for k=1..7
180 	 * For integer operation, the multiplier table is to be scaled by
181 	 * IFAST_SCALE_BITS.  The multipliers are stored in natural order.
182 	 */
183 	IFAST_MULT_TYPE * ifmtbl = (IFAST_MULT_TYPE *) compptr->dct_table;
184 #define CONST_BITS 14
185 	static const INT16 aanscales[DCTSIZE2] = {
186 	  /* precomputed values scaled up by 14 bits */
187 	  16384, 22725, 21407, 19266, 16384, 12873,  8867,  4520,
188 	  22725, 31521, 29692, 26722, 22725, 17855, 12299,  6270,
189 	  21407, 29692, 27969, 25172, 21407, 16819, 11585,  5906,
190 	  19266, 26722, 25172, 22654, 19266, 15137, 10426,  5315,
191 	  16384, 22725, 21407, 19266, 16384, 12873,  8867,  4520,
192 	  12873, 17855, 16819, 15137, 12873, 10114,  6967,  3552,
193 	   8867, 12299, 11585, 10426,  8867,  6967,  4799,  2446,
194 	   4520,  6270,  5906,  5315,  4520,  3552,  2446,  1247
195 	};
196 	SHIFT_TEMPS
197 
198 	for (i = 0; i < DCTSIZE2; i++) {
199 	  ifmtbl[i] = (IFAST_MULT_TYPE)
200 	    DESCALE(MULTIPLY16V16((INT32) qtbl->quantval[jpeg_zigzag_order[i]],
201 				  (INT32) aanscales[i]),
202 		    CONST_BITS-IFAST_SCALE_BITS);
203 	}
204       }
205       break;
206 #endif
207 #ifdef DCT_FLOAT_SUPPORTED
208     case JDCT_FLOAT:
209       {
210 	/* For float AA&N IDCT method, multipliers are equal to quantization
211 	 * coefficients scaled by scalefactor[row]*scalefactor[col], where
212 	 *   scalefactor[0] = 1
213 	 *   scalefactor[k] = cos(k*PI/16) * sqrt(2)    for k=1..7
214 	 * The multipliers are stored in natural order.
215 	 */
216 	FLOAT_MULT_TYPE * fmtbl = (FLOAT_MULT_TYPE *) compptr->dct_table;
217 	int row, col;
218 	static const double aanscalefactor[DCTSIZE] = {
219 	  1.0, 1.387039845, 1.306562965, 1.175875602,
220 	  1.0, 0.785694958, 0.541196100, 0.275899379
221 	};
222 
223 	i = 0;
224 	for (row = 0; row < DCTSIZE; row++) {
225 	  for (col = 0; col < DCTSIZE; col++) {
226 	    fmtbl[i] = (FLOAT_MULT_TYPE)
227 	      ((double) qtbl->quantval[jpeg_zigzag_order[i]] *
228 	       aanscalefactor[row] * aanscalefactor[col]);
229 	    i++;
230 	  }
231 	}
232       }
233       break;
234 #endif
235     default:
236       ERREXIT(cinfo, JERR_NOT_COMPILED);
237       break;
238     }
239   }
240 }
241 
242 
243 /*
244  * Initialize IDCT manager.
245  */
246 
247 GLOBAL void
jinit_inverse_dct(j_decompress_ptr cinfo)248 jinit_inverse_dct (j_decompress_ptr cinfo)
249 {
250   my_idct_ptr idct;
251   int ci;
252   jpeg_component_info *compptr;
253 
254   idct = (my_idct_ptr)
255     (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE,
256 				SIZEOF(my_idct_controller));
257   cinfo->idct = (struct jpeg_inverse_dct *) idct;
258   idct->pub.start_pass = start_pass;
259 
260   for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
261        ci++, compptr++) {
262     /* Allocate and pre-zero a multiplier table for each component */
263     compptr->dct_table =
264       (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE,
265 				  SIZEOF(multiplier_table));
266     MEMZERO(compptr->dct_table, SIZEOF(multiplier_table));
267     /* Mark multiplier table not yet set up for any method */
268     idct->cur_method[ci] = -1;
269   }
270 }
271