15 |
#include "utils/emms.h" |
#include "utils/emms.h" |
16 |
#include "bitstream/mbcoding.h" |
#include "bitstream/mbcoding.h" |
17 |
#include "quant/adapt_quant.h" |
#include "quant/adapt_quant.h" |
18 |
|
#include "quant/quant_matrix.h" |
19 |
|
|
20 |
#define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT |
#define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT |
21 |
|
|
216 |
uint16_t x, y; |
uint16_t x, y; |
217 |
Bitstream bs; |
Bitstream bs; |
218 |
uint32_t bits; |
uint32_t bits; |
219 |
uint16_t quant_type = 0; |
uint16_t write_vol_header = 0; |
|
uint16_t quant_change = 0; |
|
|
|
|
|
IMAGE *pCurrent = &(pEnc->sCurrent); |
|
220 |
|
|
221 |
start_global_timer(); |
start_global_timer(); |
222 |
|
|
264 |
free(temp_dquants); |
free(temp_dquants); |
265 |
} |
} |
266 |
|
|
267 |
if(pEnc->mbParam.global_flags & XVID_H263QUANT) |
if(pEnc->mbParam.global_flags & XVID_H263QUANT) { |
268 |
quant_type = H263_QUANT; |
if(pEnc->mbParam.quant_type != H263_QUANT) |
269 |
else if(pEnc->mbParam.global_flags & XVID_MPEGQUANT) |
write_vol_header = 1; |
270 |
quant_type = MPEG4_QUANT; |
pEnc->mbParam.quant_type = H263_QUANT; |
271 |
|
} |
272 |
if(pEnc->mbParam.quant_type != quant_type) { |
else if(pEnc->mbParam.global_flags & XVID_MPEGQUANT) { |
273 |
pEnc->mbParam.quant_type = quant_type; |
int ret1, ret2; |
274 |
quant_change = 1; |
|
275 |
|
if(pEnc->mbParam.quant_type != MPEG4_QUANT) |
276 |
|
write_vol_header = 1; |
277 |
|
|
278 |
|
pEnc->mbParam.quant_type = MPEG4_QUANT; |
279 |
|
|
280 |
|
if ((pEnc->mbParam.global_flags & XVID_CUSTOM_QMATRIX) > 0) { |
281 |
|
if(pFrame->quant_intra_matrix != NULL) |
282 |
|
ret1 = set_intra_matrix(pFrame->quant_intra_matrix); |
283 |
|
if(pFrame->quant_inter_matrix != NULL) |
284 |
|
ret2 = set_inter_matrix(pFrame->quant_inter_matrix); |
285 |
|
} |
286 |
|
else { |
287 |
|
ret1 = set_intra_matrix(get_default_intra_matrix()); |
288 |
|
ret2 = set_inter_matrix(get_default_inter_matrix()); |
289 |
|
} |
290 |
|
if(write_vol_header == 0) |
291 |
|
write_vol_header = ret1 | ret2; |
292 |
} |
} |
|
else |
|
|
quant_change = 0; |
|
|
|
|
293 |
|
|
294 |
if (pFrame->intra < 0) |
if (pFrame->intra < 0) |
295 |
{ |
{ |
298 |
|
|
299 |
pFrame->intra = FrameCodeI(pEnc, &bs, &bits); |
pFrame->intra = FrameCodeI(pEnc, &bs, &bits); |
300 |
else |
else |
301 |
pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 0, quant_change); |
pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 0, write_vol_header); |
302 |
} |
} |
303 |
else |
else |
304 |
{ |
{ |
305 |
if (pFrame->intra == 1) |
if (pFrame->intra == 1) |
306 |
pFrame->intra = FrameCodeI(pEnc, &bs, &bits); |
pFrame->intra = FrameCodeI(pEnc, &bs, &bits); |
307 |
else |
else |
308 |
pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 1, quant_change); |
pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 1, write_vol_header); |
309 |
} |
} |
310 |
|
|
311 |
BitstreamPutBits(&bs, 0xFFFF, 16); |
BitstreamPutBits(&bs, 0xFFFF, 16); |
361 |
int16_t dct_codes[6][64]; |
int16_t dct_codes[6][64]; |
362 |
int16_t qcoeff[6][64]; |
int16_t qcoeff[6][64]; |
363 |
uint16_t x, y; |
uint16_t x, y; |
|
IMAGE *pCurrent = &pEnc->sCurrent; |
|
364 |
|
|
365 |
pEnc->iFrameNum = 0; |
pEnc->iFrameNum = 0; |
366 |
pEnc->mbParam.rounding_type = 1; |
pEnc->mbParam.rounding_type = 1; |
384 |
|
|
385 |
CodeIntraMB(pEnc, pMB); |
CodeIntraMB(pEnc, pMB); |
386 |
|
|
387 |
MBTransQuantIntra(&pEnc->mbParam, x, y, dct_codes, qcoeff, pCurrent); |
MBTransQuantIntra(&pEnc->mbParam, x, y, dct_codes, qcoeff, &pEnc->sCurrent); |
388 |
|
|
389 |
start_timer(); |
start_timer(); |
390 |
MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs); |
MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs); |