1 |
/************************************************************************** |
/***************************************************************************** |
2 |
* |
* |
3 |
* XVID MPEG-4 VIDEO CODEC |
* XVID MPEG-4 VIDEO CODEC |
4 |
* mpeg-4 quantization/dequantization |
* - Mpeg4 quantization/dequantization functions - |
5 |
* |
* |
6 |
* This program is an implementation of a part of one or more MPEG-4 |
* Copyright(C) 2002 Peter Ross <pross@xvid.org> |
|
* Video tools as specified in ISO/IEC 14496-2 standard. Those intending |
|
|
* to use this software module in hardware or software products are |
|
|
* advised that its use may infringe existing patents or copyrights, and |
|
|
* any such use would be at such party's own risk. The original |
|
|
* developer of this software module and his/her company, and subsequent |
|
|
* editors and their companies, will have no liability for use of this |
|
|
* software or modifications or derivatives thereof. |
|
7 |
* |
* |
8 |
* This program is free software; you can redistribute it and/or modify |
* This file is part of XviD, a free MPEG-4 video encoder/decoder |
9 |
* it under the terms of the GNU General Public License as published by |
* |
10 |
|
* XviD is free software; you can redistribute it and/or modify it |
11 |
|
* under the terms of the GNU General Public License as published by |
12 |
* the Free Software Foundation; either version 2 of the License, or |
* the Free Software Foundation; either version 2 of the License, or |
13 |
* (at your option) any later version. |
* (at your option) any later version. |
14 |
* |
* |
19 |
* |
* |
20 |
* You should have received a copy of the GNU General Public License |
* You should have received a copy of the GNU General Public License |
21 |
* along with this program; if not, write to the Free Software |
* along with this program; if not, write to the Free Software |
22 |
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
23 |
* |
* |
24 |
*************************************************************************/ |
* Under section 8 of the GNU General Public License, the copyright |
25 |
|
* holders of XVID explicitly forbid distribution in the following |
26 |
/************************************************************************** |
* countries: |
27 |
* |
* |
28 |
* History: |
* - Japan |
29 |
|
* - United States of America |
30 |
* |
* |
31 |
* 26.01.2002 fixed quant4_intra dcscalar signed/unsigned error |
* Linking XviD statically or dynamically with other modules is making a |
32 |
* 20.01.2002 increased accuracy of >> divide |
* combined work based on XviD. Thus, the terms and conditions of the |
33 |
* 26.12.2001 divide-by-multiplication optimization |
* GNU General Public License cover the whole combination. |
|
* 22.12.2001 [-127,127] clamping removed; minor tweaks |
|
|
* 19.11.2001 inital version <pross@cs.rmit.edu.au> |
|
34 |
* |
* |
35 |
*************************************************************************/ |
* As a special exception, the copyright holders of XviD give you |
36 |
|
* permission to link XviD with independent modules that communicate with |
37 |
|
* XviD solely through the VFW1.1 and DShow interfaces, regardless of the |
38 |
|
* license terms of these independent modules, and to copy and distribute |
39 |
|
* the resulting combined work under terms of your choice, provided that |
40 |
|
* every copy of the combined work is accompanied by a complete copy of |
41 |
|
* the source code of XviD (the version of XviD used to produce the |
42 |
|
* combined work), being distributed under the terms of the GNU General |
43 |
|
* Public License plus this exception. An independent module is a module |
44 |
|
* which is not derived from or based on XviD. |
45 |
|
* |
46 |
|
* Note that people who make modified versions of XviD are not obligated |
47 |
|
* to grant this special exception for their modified versions; it is |
48 |
|
* their choice whether to do so. The GNU General Public License gives |
49 |
|
* permission to release a modified version without this exception; this |
50 |
|
* exception also makes it possible to release a modified version which |
51 |
|
* carries forward this exception. |
52 |
|
* |
53 |
|
* $Id$ |
54 |
|
* |
55 |
|
****************************************************************************/ |
56 |
|
|
57 |
#include "quant_mpeg4.h" |
#include "quant_mpeg4.h" |
58 |
|
#include "quant_matrix.h" |
59 |
|
|
60 |
|
/***************************************************************************** |
61 |
|
* Function pointers |
62 |
|
****************************************************************************/ |
63 |
|
|
|
|
|
|
// function pointers |
|
64 |
quant_intraFuncPtr quant4_intra; |
quant_intraFuncPtr quant4_intra; |
65 |
quant_intraFuncPtr dequant4_intra; |
quant_intraFuncPtr dequant4_intra; |
66 |
dequant_interFuncPtr dequant4_inter; |
dequant_interFuncPtr dequant4_inter; |
67 |
quant_interFuncPtr quant4_inter; |
quant_interFuncPtr quant4_inter; |
68 |
|
|
69 |
|
|
70 |
|
/***************************************************************************** |
71 |
|
* Local data |
72 |
|
****************************************************************************/ |
73 |
|
|
74 |
#define DIV_DIV(A,B) ( (A) > 0 ? ((A)+((B)>>1))/(B) : ((A)-((B)>>1))/(B) ) |
#define DIV_DIV(A,B) ( (A) > 0 ? ((A)+((B)>>1))/(B) : ((A)-((B)>>1))/(B) ) |
75 |
#define SIGN(A) ((A)>0?1:-1) |
#define SIGN(A) ((A)>0?1:-1) |
76 |
#define VM18P 3 |
#define VM18P 3 |
77 |
#define VM18Q 4 |
#define VM18Q 4 |
78 |
|
|
79 |
|
|
80 |
// divide-by-multiply table |
/* |
81 |
// need 17 bit shift (16 causes slight errors when q > 19) |
* divide-by-multiply table |
82 |
|
* need 17 bit shift (16 causes slight errors when q > 19) |
83 |
|
*/ |
84 |
|
|
85 |
#define SCALEBITS 17 |
#define SCALEBITS 17 |
86 |
#define FIX(X) ((1UL << SCALEBITS) / (X) + 1) |
#define FIX(X) ((1UL << SCALEBITS) / (X) + 1) |
87 |
|
|
88 |
static const uint32_t multipliers[32] = |
static const uint32_t multipliers[32] = { |
|
{ |
|
89 |
0, FIX(2), FIX(4), FIX(6), |
0, FIX(2), FIX(4), FIX(6), |
90 |
FIX(8), FIX(10), FIX(12), FIX(14), |
FIX(8), FIX(10), FIX(12), FIX(14), |
91 |
FIX(16), FIX(18), FIX(20), FIX(22), |
FIX(16), FIX(18), FIX(20), FIX(22), |
96 |
FIX(56), FIX(58), FIX(60), FIX(62) |
FIX(56), FIX(58), FIX(60), FIX(62) |
97 |
}; |
}; |
98 |
|
|
99 |
|
/***************************************************************************** |
100 |
|
* Functions |
101 |
|
****************************************************************************/ |
102 |
|
|
103 |
static const int16_t default_intra_matrix[64] = { |
#if 0 |
104 |
8,17,18,19,21,23,25,27, |
/* quantize intra-block */ |
|
17,18,19,21,23,25,27,28, |
|
|
20,21,22,23,24,26,28,30, |
|
|
21,22,23,24,26,28,30,32, |
|
|
22,23,24,26,28,30,32,35, |
|
|
23,24,26,28,30,32,35,38, |
|
|
25,26,28,30,32,35,38,41, |
|
|
27,28,30,32,35,38,41,45 |
|
|
}; |
|
|
|
|
|
static const int16_t default_inter_matrix[64] = { |
|
|
16,17,18,19,20,21,22,23, |
|
|
17,18,19,20,21,22,23,24, |
|
|
18,19,20,21,22,23,24,25, |
|
|
19,20,21,22,23,24,26,27, |
|
|
20,21,22,23,25,26,27,28, |
|
|
21,22,23,24,26,27,28,30, |
|
|
22,23,24,26,27,28,30,31, |
|
|
23,24,25,27,28,30,31,33 |
|
|
}; |
|
|
|
|
|
|
|
|
/* quantize intra-block |
|
105 |
|
|
106 |
// const int32_t quantd = DIV_DIV(VM18P*quant, VM18Q); |
// const int32_t quantd = DIV_DIV(VM18P*quant, VM18Q); |
107 |
// |
// |
108 |
// level = DIV_DIV(16 * data[i], default_intra_matrix[i]); |
// level = DIV_DIV(16 * data[i], default_intra_matrix[i]); |
109 |
// coeff[i] = (level + quantd) / quant2; |
// coeff[i] = (level + quantd) / quant2; |
110 |
*/ |
#endif |
111 |
|
|
112 |
void quant4_intra_c(int16_t * coeff, const int16_t * data, const uint32_t quant, const uint32_t dcscalar) |
void |
113 |
|
quant4_intra_c(int16_t * coeff, |
114 |
|
const int16_t * data, |
115 |
|
const uint32_t quant, |
116 |
|
const uint32_t dcscalar) |
117 |
{ |
{ |
118 |
const uint32_t quantd = ((VM18P*quant) + (VM18Q/2)) / VM18Q; |
const uint32_t quantd = ((VM18P*quant) + (VM18Q/2)) / VM18Q; |
119 |
const uint32_t mult = multipliers[quant]; |
const uint32_t mult = multipliers[quant]; |
120 |
uint32_t i; |
uint32_t i; |
121 |
|
int16_t *intra_matrix; |
122 |
|
|
123 |
|
intra_matrix = get_intra_matrix(); |
124 |
|
|
125 |
coeff[0] = DIV_DIV(data[0], (int32_t)dcscalar); |
coeff[0] = DIV_DIV(data[0], (int32_t)dcscalar); |
126 |
|
|
127 |
for (i = 1; i < 64; i++) |
for (i = 1; i < 64; i++) { |
128 |
{ |
if (data[i] < 0) { |
|
if (data[i] < 0) |
|
|
{ |
|
129 |
uint32_t level = -data[i]; |
uint32_t level = -data[i]; |
130 |
level = ((level<<4) + (default_intra_matrix[i]>>1)) / default_intra_matrix[i]; |
|
131 |
|
level = ((level << 4) + (intra_matrix[i] >> 1)) / intra_matrix[i]; |
132 |
level = ((level + quantd) * mult) >> 17; |
level = ((level + quantd) * mult) >> 17; |
133 |
coeff[i] = -(int16_t)level; |
coeff[i] = -(int16_t)level; |
134 |
} |
} else if (data[i] > 0) { |
|
else if (data[i] > 0) |
|
|
{ |
|
135 |
uint32_t level = data[i]; |
uint32_t level = data[i]; |
136 |
level = ((level<<4) + (default_intra_matrix[i]>>1)) / default_intra_matrix[i]; |
|
137 |
|
level = ((level << 4) + (intra_matrix[i] >> 1)) / intra_matrix[i]; |
138 |
level = ((level + quantd) * mult) >> 17; |
level = ((level + quantd) * mult) >> 17; |
139 |
coeff[i] = level; |
coeff[i] = level; |
140 |
} |
} else { |
|
else |
|
|
{ |
|
141 |
coeff[i] = 0; |
coeff[i] = 0; |
142 |
} |
} |
143 |
} |
} |
145 |
|
|
146 |
|
|
147 |
|
|
148 |
/* dequantize intra-block & clamp to [-2048,2047] |
/* dequantize intra-block & clamp to [-2048,2047] */ |
149 |
// data[i] = (coeff[i] * default_intra_matrix[i] * quant2) >> 4; |
/* data[i] = (coeff[i] * default_intra_matrix[i] * quant2) >> 4; */ |
|
*/ |
|
150 |
|
|
151 |
void dequant4_intra_c(int16_t *data, const int16_t *coeff, const uint32_t quant, const uint32_t dcscalar) |
|
152 |
|
void |
153 |
|
dequant4_intra_c(int16_t * data, |
154 |
|
const int16_t * coeff, |
155 |
|
const uint32_t quant, |
156 |
|
const uint32_t dcscalar) |
157 |
{ |
{ |
158 |
uint32_t i; |
uint32_t i; |
159 |
|
int16_t *intra_matrix; |
160 |
|
|
161 |
|
intra_matrix = get_intra_matrix(); |
162 |
|
|
163 |
data[0] = coeff[0] * dcscalar; |
data[0] = coeff[0] * dcscalar; |
164 |
if (data[0] < -2048) |
if (data[0] < -2048) { |
|
{ |
|
165 |
data[0] = -2048; |
data[0] = -2048; |
166 |
} |
} else if (data[0] > 2047) { |
|
else if (data[0] > 2047) |
|
|
{ |
|
167 |
data[0] = 2047; |
data[0] = 2047; |
168 |
} |
} |
169 |
|
|
170 |
for (i = 1; i < 64; i++) |
for (i = 1; i < 64; i++) { |
171 |
{ |
if (coeff[i] == 0) { |
|
if (coeff[i] == 0) |
|
|
{ |
|
172 |
data[i] = 0; |
data[i] = 0; |
173 |
} |
} else if (coeff[i] < 0) { |
|
else if (coeff[i] < 0) |
|
|
{ |
|
174 |
uint32_t level = -coeff[i]; |
uint32_t level = -coeff[i]; |
175 |
level = (level * default_intra_matrix[i] * quant) >> 3; |
|
176 |
|
level = (level * intra_matrix[i] * quant) >> 3; |
177 |
data[i] = (level <= 2048 ? -(int16_t)level : -2048); |
data[i] = (level <= 2048 ? -(int16_t)level : -2048); |
178 |
} |
} else /* if (coeff[i] > 0) */ |
|
else // if (coeff[i] > 0) |
|
179 |
{ |
{ |
180 |
uint32_t level = coeff[i]; |
uint32_t level = coeff[i]; |
181 |
level = (level * default_intra_matrix[i] * quant) >> 3; |
|
182 |
|
level = (level * intra_matrix[i] * quant) >> 3; |
183 |
data[i] = (level <= 2047 ? level : 2047); |
data[i] = (level <= 2047 ? level : 2047); |
184 |
} |
} |
185 |
} |
} |
187 |
|
|
188 |
|
|
189 |
|
|
190 |
/* quantize inter-block |
#if 0 |
191 |
|
/* quantize inter-block */ |
192 |
|
|
193 |
// level = DIV_DIV(16 * data[i], default_intra_matrix[i]); |
// level = DIV_DIV(16 * data[i], default_intra_matrix[i]); |
194 |
// coeff[i] = (level + quantd) / quant2; |
// coeff[i] = (level + quantd) / quant2; |
195 |
// sum += abs(level); |
// sum += abs(level); |
196 |
*/ |
#endif |
197 |
|
|
198 |
uint32_t quant4_inter_c(int16_t * coeff, const int16_t * data, const uint32_t quant) |
uint32_t |
199 |
|
quant4_inter_c(int16_t * coeff, |
200 |
|
const int16_t * data, |
201 |
|
const uint32_t quant) |
202 |
{ |
{ |
203 |
const uint32_t mult = multipliers[quant]; |
const uint32_t mult = multipliers[quant]; |
204 |
uint32_t sum = 0; |
uint32_t sum = 0; |
205 |
uint32_t i; |
uint32_t i; |
206 |
|
int16_t *inter_matrix; |
207 |
|
|
208 |
for (i = 0; i < 64; i++) |
inter_matrix = get_inter_matrix(); |
209 |
{ |
|
210 |
if (data[i] < 0) |
for (i = 0; i < 64; i++) { |
211 |
{ |
if (data[i] < 0) { |
212 |
uint32_t level = -data[i]; |
uint32_t level = -data[i]; |
213 |
level = ((level<<4) + (default_inter_matrix[i]>>1)) / default_inter_matrix[i]; |
|
214 |
|
level = ((level << 4) + (inter_matrix[i] >> 1)) / inter_matrix[i]; |
215 |
level = (level * mult) >> 17; |
level = (level * mult) >> 17; |
216 |
sum += level; |
sum += level; |
217 |
coeff[i] = -(int16_t)level; |
coeff[i] = -(int16_t)level; |
218 |
} |
} else if (data[i] > 0) { |
|
else if (data[i] > 0) |
|
|
{ |
|
219 |
uint32_t level = data[i]; |
uint32_t level = data[i]; |
220 |
level = ((level<<4) + (default_inter_matrix[i]>>1)) / default_inter_matrix[i]; |
|
221 |
|
level = ((level << 4) + (inter_matrix[i] >> 1)) / inter_matrix[i]; |
222 |
level = (level * mult) >> 17; |
level = (level * mult) >> 17; |
223 |
sum += level; |
sum += level; |
224 |
coeff[i] = level; |
coeff[i] = level; |
225 |
} |
} else { |
|
else |
|
|
{ |
|
226 |
coeff[i] = 0; |
coeff[i] = 0; |
227 |
} |
} |
228 |
} |
} |
235 |
data = ((2 * coeff + SIGN(coeff)) * inter_matrix[i] * quant) / 16 |
data = ((2 * coeff + SIGN(coeff)) * inter_matrix[i] * quant) / 16 |
236 |
*/ |
*/ |
237 |
|
|
238 |
void dequant4_inter_c(int16_t *data, const int16_t *coeff, const uint32_t quant) |
void |
239 |
|
dequant4_inter_c(int16_t * data, |
240 |
|
const int16_t * coeff, |
241 |
|
const uint32_t quant) |
242 |
{ |
{ |
243 |
uint32_t sum = 0; |
uint32_t sum = 0; |
244 |
uint32_t i; |
uint32_t i; |
245 |
|
int16_t *inter_matrix; |
246 |
|
|
247 |
for (i = 0; i < 64; i++) |
inter_matrix = get_inter_matrix(); |
248 |
{ |
|
249 |
if (coeff[i] == 0) |
for (i = 0; i < 64; i++) { |
250 |
{ |
if (coeff[i] == 0) { |
251 |
data[i] = 0; |
data[i] = 0; |
252 |
} |
} else if (coeff[i] < 0) { |
|
else if (coeff[i] < 0) |
|
|
{ |
|
253 |
int32_t level = -coeff[i]; |
int32_t level = -coeff[i]; |
254 |
level = ((2 * level + 1) * default_inter_matrix[i] * quant) >> 4; |
|
255 |
|
level = ((2 * level + 1) * inter_matrix[i] * quant) >> 4; |
256 |
data[i] = (level <= 2048 ? -level : -2048); |
data[i] = (level <= 2048 ? -level : -2048); |
257 |
} |
} else /* if (coeff[i] > 0) */ |
|
else // if (coeff[i] > 0) |
|
258 |
{ |
{ |
259 |
uint32_t level = coeff[i]; |
uint32_t level = coeff[i]; |
260 |
level = ((2 * level + 1) * default_inter_matrix[i] * quant) >> 4; |
|
261 |
|
level = ((2 * level + 1) * inter_matrix[i] * quant) >> 4; |
262 |
data[i] = (level <= 2047 ? level : 2047); |
data[i] = (level <= 2047 ? level : 2047); |
263 |
} |
} |
264 |
|
|
265 |
sum ^= data[i]; |
sum ^= data[i]; |
266 |
} |
} |
267 |
|
|
268 |
// mismatch control |
/* mismatch control */ |
269 |
|
|
270 |
if ((sum & 1) == 0) |
if ((sum & 1) == 0) { |
|
{ |
|
271 |
data[63] ^= 1; |
data[63] ^= 1; |
272 |
} |
} |
273 |
} |
} |