42 |
|
|
43 |
|
|
44 |
#include "quant_mpeg4.h" |
#include "quant_mpeg4.h" |
45 |
|
#include "quant_matrix.h" |
|
|
|
46 |
|
|
47 |
// function pointers |
// function pointers |
48 |
quant_intraFuncPtr quant4_intra; |
quant_intraFuncPtr quant4_intra; |
63 |
#define SCALEBITS 17 |
#define SCALEBITS 17 |
64 |
#define FIX(X) ((1UL << SCALEBITS) / (X) + 1) |
#define FIX(X) ((1UL << SCALEBITS) / (X) + 1) |
65 |
|
|
66 |
static const uint32_t multipliers[32] = |
static const uint32_t multipliers[32] = { |
|
{ |
|
67 |
0, FIX(2), FIX(4), FIX(6), |
0, FIX(2), FIX(4), FIX(6), |
68 |
FIX(8), FIX(10), FIX(12), FIX(14), |
FIX(8), FIX(10), FIX(12), FIX(14), |
69 |
FIX(16), FIX(18), FIX(20), FIX(22), |
FIX(16), FIX(18), FIX(20), FIX(22), |
74 |
FIX(56), FIX(58), FIX(60), FIX(62) |
FIX(56), FIX(58), FIX(60), FIX(62) |
75 |
}; |
}; |
76 |
|
|
|
|
|
|
static const int16_t default_intra_matrix[64] = { |
|
|
8,17,18,19,21,23,25,27, |
|
|
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 |
|
|
}; |
|
|
|
|
|
|
|
77 |
/* quantize intra-block |
/* quantize intra-block |
78 |
|
|
79 |
// const int32_t quantd = DIV_DIV(VM18P*quant, VM18Q); |
// const int32_t quantd = DIV_DIV(VM18P*quant, VM18Q); |
82 |
// coeff[i] = (level + quantd) / quant2; |
// coeff[i] = (level + quantd) / quant2; |
83 |
*/ |
*/ |
84 |
|
|
85 |
void quant4_intra_c(int16_t * coeff, const int16_t * data, const uint32_t quant, const uint32_t dcscalar) |
void |
86 |
|
quant4_intra_c(int16_t * coeff, |
87 |
|
const int16_t * data, |
88 |
|
const uint32_t quant, |
89 |
|
const uint32_t dcscalar) |
90 |
{ |
{ |
91 |
const uint32_t quantd = ((VM18P*quant) + (VM18Q/2)) / VM18Q; |
const uint32_t quantd = ((VM18P*quant) + (VM18Q/2)) / VM18Q; |
92 |
const uint32_t mult = multipliers[quant]; |
const uint32_t mult = multipliers[quant]; |
93 |
uint32_t i; |
uint32_t i; |
94 |
|
int16_t *intra_matrix; |
95 |
|
|
96 |
|
intra_matrix = get_intra_matrix(); |
97 |
|
|
98 |
coeff[0] = DIV_DIV(data[0], (int32_t)dcscalar); |
coeff[0] = DIV_DIV(data[0], (int32_t)dcscalar); |
99 |
|
|
100 |
for (i = 1; i < 64; i++) |
for (i = 1; i < 64; i++) { |
101 |
{ |
if (data[i] < 0) { |
|
if (data[i] < 0) |
|
|
{ |
|
102 |
uint32_t level = -data[i]; |
uint32_t level = -data[i]; |
103 |
level = ((level<<4) + (default_intra_matrix[i]>>1)) / default_intra_matrix[i]; |
|
104 |
|
level = ((level << 4) + (intra_matrix[i] >> 1)) / intra_matrix[i]; |
105 |
level = ((level + quantd) * mult) >> 17; |
level = ((level + quantd) * mult) >> 17; |
106 |
coeff[i] = -(int16_t)level; |
coeff[i] = -(int16_t)level; |
107 |
} |
} else if (data[i] > 0) { |
|
else if (data[i] > 0) |
|
|
{ |
|
108 |
uint32_t level = data[i]; |
uint32_t level = data[i]; |
109 |
level = ((level<<4) + (default_intra_matrix[i]>>1)) / default_intra_matrix[i]; |
|
110 |
|
level = ((level << 4) + (intra_matrix[i] >> 1)) / intra_matrix[i]; |
111 |
level = ((level + quantd) * mult) >> 17; |
level = ((level + quantd) * mult) >> 17; |
112 |
coeff[i] = level; |
coeff[i] = level; |
113 |
} |
} else { |
|
else |
|
|
{ |
|
114 |
coeff[i] = 0; |
coeff[i] = 0; |
115 |
} |
} |
116 |
} |
} |
122 |
// data[i] = (coeff[i] * default_intra_matrix[i] * quant2) >> 4; |
// data[i] = (coeff[i] * default_intra_matrix[i] * quant2) >> 4; |
123 |
*/ |
*/ |
124 |
|
|
125 |
void dequant4_intra_c(int16_t *data, const int16_t *coeff, const uint32_t quant, const uint32_t dcscalar) |
void |
126 |
|
dequant4_intra_c(int16_t * data, |
127 |
|
const int16_t * coeff, |
128 |
|
const uint32_t quant, |
129 |
|
const uint32_t dcscalar) |
130 |
{ |
{ |
131 |
uint32_t i; |
uint32_t i; |
132 |
|
int16_t *intra_matrix; |
133 |
|
|
134 |
|
intra_matrix = get_intra_matrix(); |
135 |
|
|
136 |
data[0] = coeff[0] * dcscalar; |
data[0] = coeff[0] * dcscalar; |
137 |
if (data[0] < -2048) |
if (data[0] < -2048) { |
|
{ |
|
138 |
data[0] = -2048; |
data[0] = -2048; |
139 |
} |
} else if (data[0] > 2047) { |
|
else if (data[0] > 2047) |
|
|
{ |
|
140 |
data[0] = 2047; |
data[0] = 2047; |
141 |
} |
} |
142 |
|
|
143 |
for (i = 1; i < 64; i++) |
for (i = 1; i < 64; i++) { |
144 |
{ |
if (coeff[i] == 0) { |
|
if (coeff[i] == 0) |
|
|
{ |
|
145 |
data[i] = 0; |
data[i] = 0; |
146 |
} |
} else if (coeff[i] < 0) { |
|
else if (coeff[i] < 0) |
|
|
{ |
|
147 |
uint32_t level = -coeff[i]; |
uint32_t level = -coeff[i]; |
148 |
level = (level * default_intra_matrix[i] * quant) >> 3; |
|
149 |
|
level = (level * intra_matrix[i] * quant) >> 3; |
150 |
data[i] = (level <= 2048 ? -(int16_t)level : -2048); |
data[i] = (level <= 2048 ? -(int16_t)level : -2048); |
151 |
} |
} else // if (coeff[i] > 0) |
|
else // if (coeff[i] > 0) |
|
152 |
{ |
{ |
153 |
uint32_t level = coeff[i]; |
uint32_t level = coeff[i]; |
154 |
level = (level * default_intra_matrix[i] * quant) >> 3; |
|
155 |
|
level = (level * intra_matrix[i] * quant) >> 3; |
156 |
data[i] = (level <= 2047 ? level : 2047); |
data[i] = (level <= 2047 ? level : 2047); |
157 |
} |
} |
158 |
} |
} |
167 |
// sum += abs(level); |
// sum += abs(level); |
168 |
*/ |
*/ |
169 |
|
|
170 |
uint32_t quant4_inter_c(int16_t * coeff, const int16_t * data, const uint32_t quant) |
uint32_t |
171 |
|
quant4_inter_c(int16_t * coeff, |
172 |
|
const int16_t * data, |
173 |
|
const uint32_t quant) |
174 |
{ |
{ |
175 |
const uint32_t mult = multipliers[quant]; |
const uint32_t mult = multipliers[quant]; |
176 |
uint32_t sum = 0; |
uint32_t sum = 0; |
177 |
uint32_t i; |
uint32_t i; |
178 |
|
int16_t *inter_matrix; |
179 |
|
|
180 |
for (i = 0; i < 64; i++) |
inter_matrix = get_inter_matrix(); |
181 |
{ |
|
182 |
if (data[i] < 0) |
for (i = 0; i < 64; i++) { |
183 |
{ |
if (data[i] < 0) { |
184 |
uint32_t level = -data[i]; |
uint32_t level = -data[i]; |
185 |
level = ((level<<4) + (default_inter_matrix[i]>>1)) / default_inter_matrix[i]; |
|
186 |
|
level = ((level << 4) + (inter_matrix[i] >> 1)) / inter_matrix[i]; |
187 |
level = (level * mult) >> 17; |
level = (level * mult) >> 17; |
188 |
sum += level; |
sum += level; |
189 |
coeff[i] = -(int16_t)level; |
coeff[i] = -(int16_t)level; |
190 |
} |
} else if (data[i] > 0) { |
|
else if (data[i] > 0) |
|
|
{ |
|
191 |
uint32_t level = data[i]; |
uint32_t level = data[i]; |
192 |
level = ((level<<4) + (default_inter_matrix[i]>>1)) / default_inter_matrix[i]; |
|
193 |
|
level = ((level << 4) + (inter_matrix[i] >> 1)) / inter_matrix[i]; |
194 |
level = (level * mult) >> 17; |
level = (level * mult) >> 17; |
195 |
sum += level; |
sum += level; |
196 |
coeff[i] = level; |
coeff[i] = level; |
197 |
} |
} else { |
|
else |
|
|
{ |
|
198 |
coeff[i] = 0; |
coeff[i] = 0; |
199 |
} |
} |
200 |
} |
} |
207 |
data = ((2 * coeff + SIGN(coeff)) * inter_matrix[i] * quant) / 16 |
data = ((2 * coeff + SIGN(coeff)) * inter_matrix[i] * quant) / 16 |
208 |
*/ |
*/ |
209 |
|
|
210 |
void dequant4_inter_c(int16_t *data, const int16_t *coeff, const uint32_t quant) |
void |
211 |
|
dequant4_inter_c(int16_t * data, |
212 |
|
const int16_t * coeff, |
213 |
|
const uint32_t quant) |
214 |
{ |
{ |
215 |
uint32_t sum = 0; |
uint32_t sum = 0; |
216 |
uint32_t i; |
uint32_t i; |
217 |
|
int16_t *inter_matrix; |
218 |
|
|
219 |
for (i = 0; i < 64; i++) |
inter_matrix = get_inter_matrix(); |
220 |
{ |
|
221 |
if (coeff[i] == 0) |
for (i = 0; i < 64; i++) { |
222 |
{ |
if (coeff[i] == 0) { |
223 |
data[i] = 0; |
data[i] = 0; |
224 |
} |
} else if (coeff[i] < 0) { |
|
else if (coeff[i] < 0) |
|
|
{ |
|
225 |
int32_t level = -coeff[i]; |
int32_t level = -coeff[i]; |
226 |
level = ((2 * level + 1) * default_inter_matrix[i] * quant) >> 4; |
|
227 |
|
level = ((2 * level + 1) * inter_matrix[i] * quant) >> 4; |
228 |
data[i] = (level <= 2048 ? -level : -2048); |
data[i] = (level <= 2048 ? -level : -2048); |
229 |
} |
} else // if (coeff[i] > 0) |
|
else // if (coeff[i] > 0) |
|
230 |
{ |
{ |
231 |
uint32_t level = coeff[i]; |
uint32_t level = coeff[i]; |
232 |
level = ((2 * level + 1) * default_inter_matrix[i] * quant) >> 4; |
|
233 |
|
level = ((2 * level + 1) * inter_matrix[i] * quant) >> 4; |
234 |
data[i] = (level <= 2047 ? level : 2047); |
data[i] = (level <= 2047 ? level : 2047); |
235 |
} |
} |
236 |
|
|
239 |
|
|
240 |
// mismatch control |
// mismatch control |
241 |
|
|
242 |
if ((sum & 1) == 0) |
if ((sum & 1) == 0) { |
|
{ |
|
243 |
data[63] ^= 1; |
data[63] ^= 1; |
244 |
} |
} |
245 |
} |
} |