41 |
#include "bitstream/mbcoding.h" |
#include "bitstream/mbcoding.h" |
42 |
|
|
43 |
#include "quant/quant.h" |
#include "quant/quant.h" |
44 |
|
#include "quant/quant_matrix.h" |
45 |
#include "dct/idct.h" |
#include "dct/idct.h" |
46 |
#include "dct/fdct.h" |
#include "dct/fdct.h" |
47 |
#include "utils/mem_transfer.h" |
#include "utils/mem_transfer.h" |
58 |
|
|
59 |
#include "image/image.h" |
#include "image/image.h" |
60 |
#include "image/colorspace.h" |
#include "image/colorspace.h" |
61 |
|
#include "image/postprocessing.h" |
62 |
#include "utils/mem_align.h" |
#include "utils/mem_align.h" |
63 |
|
|
64 |
static int |
static int |
177 |
if (dec == NULL) { |
if (dec == NULL) { |
178 |
return XVID_ERR_MEMORY; |
return XVID_ERR_MEMORY; |
179 |
} |
} |
180 |
|
|
181 |
memset(dec, 0, sizeof(DECODER)); |
memset(dec, 0, sizeof(DECODER)); |
182 |
|
|
183 |
|
dec->mpeg_quant_matrices = xvid_malloc(sizeof(uint16_t) * 64 * 8, CACHE_LINE); |
184 |
|
if (dec->mpeg_quant_matrices == NULL) { |
185 |
|
xvid_free(dec); |
186 |
|
return XVID_ERR_MEMORY; |
187 |
|
} |
188 |
|
|
189 |
create->handle = dec; |
create->handle = dec; |
190 |
|
|
191 |
dec->width = create->width; |
dec->width = create->width; |
205 |
dec->last_mbs = NULL; |
dec->last_mbs = NULL; |
206 |
|
|
207 |
init_timer(); |
init_timer(); |
208 |
|
init_mpeg_matrix(dec->mpeg_quant_matrices); |
209 |
|
|
210 |
/* For B-frame support (used to save reference frame's time */ |
/* For B-frame support (used to save reference frame's time */ |
211 |
dec->frames = 0; |
dec->frames = 0; |
236 |
image_destroy(&dec->tmp, dec->edged_width, dec->edged_height); |
image_destroy(&dec->tmp, dec->edged_width, dec->edged_height); |
237 |
image_destroy(&dec->qtmp, dec->edged_width, dec->edged_height); |
image_destroy(&dec->qtmp, dec->edged_width, dec->edged_height); |
238 |
image_destroy(&dec->cur, dec->edged_width, dec->edged_height); |
image_destroy(&dec->cur, dec->edged_width, dec->edged_height); |
239 |
|
xvid_free(dec->mpeg_quant_matrices); |
240 |
xvid_free(dec); |
xvid_free(dec); |
241 |
|
|
242 |
write_timer(); |
write_timer(); |
332 |
|
|
333 |
start_timer(); |
start_timer(); |
334 |
if (dec->quant_type == 0) { |
if (dec->quant_type == 0) { |
335 |
dequant_h263_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler); |
dequant_h263_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler, dec->mpeg_quant_matrices); |
336 |
} else { |
} else { |
337 |
dequant_mpeg_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler); |
dequant_mpeg_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler, dec->mpeg_quant_matrices); |
338 |
} |
} |
339 |
stop_iquant_timer(); |
stop_iquant_timer(); |
340 |
|
|
403 |
stop_coding_timer(); |
stop_coding_timer(); |
404 |
|
|
405 |
start_timer(); |
start_timer(); |
406 |
dequant(&data[i * 64], block, iQuant); |
dequant(&data[i * 64], block, iQuant, dec->mpeg_quant_matrices); |
407 |
stop_iquant_timer(); |
stop_iquant_timer(); |
408 |
|
|
409 |
start_timer(); |
start_timer(); |
1175 |
const int64_t TRB = dec->time_pp - dec->time_bp, TRD = dec->time_pp; |
const int64_t TRB = dec->time_pp - dec->time_bp, TRD = dec->time_pp; |
1176 |
int i; |
int i; |
1177 |
|
|
|
#ifdef BFRAMES_DEC_DEBUG |
|
|
FILE *fp; |
|
|
static char first=0; |
|
|
#define BFRAME_DEBUG |
|
|
if (!first && fp) { \ |
|
|
fprintf(fp,"Y=%3d X=%3d MB=%2d CBP=%02X\n",y,x,mb->mode,mb->cbp); \ |
|
|
} |
|
|
#endif |
|
|
|
|
1178 |
start_timer(); |
start_timer(); |
1179 |
image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height, |
image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height, |
1180 |
dec->width, dec->height); |
dec->width, dec->height); |
1182 |
dec->width, dec->height); |
dec->width, dec->height); |
1183 |
stop_edges_timer(); |
stop_edges_timer(); |
1184 |
|
|
|
#ifdef BFRAMES_DEC_DEBUG |
|
|
if (!first){ |
|
|
fp=fopen("C:\\XVIDDBG.TXT","w"); |
|
|
} |
|
|
#endif |
|
|
|
|
1185 |
for (y = 0; y < dec->mb_height; y++) { |
for (y = 0; y < dec->mb_height; y++) { |
1186 |
/* Initialize Pred Motion Vector */ |
/* Initialize Pred Motion Vector */ |
1187 |
dec->p_fmv = dec->p_bmv = zeromv; |
dec->p_fmv = dec->p_bmv = zeromv; |
1188 |
for (x = 0; x < dec->mb_width; x++) { |
for (x = 0; x < dec->mb_width; x++) { |
1189 |
MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x]; |
MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x]; |
1190 |
MACROBLOCK *last_mb = &dec->last_mbs[y * dec->mb_width + x]; |
MACROBLOCK *last_mb = &dec->last_mbs[y * dec->mb_width + x]; |
1191 |
|
const int fcode_max = (fcode_forward>fcode_backward) ? fcode_forward : fcode_backward; |
1192 |
|
uint32_t intra_dc_threshold; /* fake variable */ |
1193 |
|
|
1194 |
|
if (check_resync_marker(bs, fcode_max - 1)) { |
1195 |
|
int bound = read_video_packet_header(bs, dec, fcode_max - 1, &quant, |
1196 |
|
&fcode_forward, &fcode_backward, &intra_dc_threshold); |
1197 |
|
x = bound % dec->mb_width; |
1198 |
|
y = bound / dec->mb_width; |
1199 |
|
/* reset predicted macroblocks */ |
1200 |
|
dec->p_fmv = dec->p_bmv = zeromv; |
1201 |
|
} |
1202 |
|
|
1203 |
mv = |
mv = |
1204 |
mb->b_mvs[0] = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] = |
mb->b_mvs[0] = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] = |
1311 |
} |
} |
1312 |
} /* End of for */ |
} /* End of for */ |
1313 |
} |
} |
|
|
|
|
#ifdef BFRAMES_DEC_DEBUG |
|
|
if (!first){ |
|
|
first=1; |
|
|
if (fp) |
|
|
fclose(fp); |
|
|
} |
|
|
#endif |
|
1314 |
} |
} |
1315 |
|
|
1316 |
/* perform post processing if necessary, and output the image */ |
/* perform post processing if necessary, and output the image */ |
1317 |
void decoder_output(DECODER * dec, IMAGE * img, MACROBLOCK * mbs, |
void decoder_output(DECODER * dec, IMAGE * img, MACROBLOCK * mbs, |
1318 |
xvid_dec_frame_t * frame, xvid_dec_stats_t * stats, int coding_type) |
xvid_dec_frame_t * frame, xvid_dec_stats_t * stats, int coding_type) |
1319 |
{ |
{ |
1320 |
|
if (frame->general & (XVID_DEBLOCKY|XVID_DEBLOCKUV) && mbs != NULL) /* post process */ |
1321 |
|
{ |
1322 |
|
/* note: image is stored to tmp */ |
1323 |
|
image_copy(&dec->tmp, img, dec->edged_width, dec->height); |
1324 |
|
image_deblock(&dec->tmp, dec->edged_width, |
1325 |
|
mbs, dec->mb_width, dec->mb_height, dec->mb_width, |
1326 |
|
frame->general); |
1327 |
|
img = &dec->tmp; |
1328 |
|
} |
1329 |
|
|
1330 |
image_output(img, dec->width, dec->height, |
image_output(img, dec->width, dec->height, |
1331 |
dec->edged_width, (uint8_t**)frame->output.plane, frame->output.stride, |
dec->edged_width, (uint8_t**)frame->output.plane, frame->output.stride, |
1332 |
frame->output.csp, dec->interlacing); |
frame->output.csp, dec->interlacing); |
1461 |
/* XXX: not_coded vops are not used for forward prediction */ |
/* XXX: not_coded vops are not used for forward prediction */ |
1462 |
/* we should not swap(last_mbs,mbs) */ |
/* we should not swap(last_mbs,mbs) */ |
1463 |
image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height); |
image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height); |
1464 |
|
SWAP(MACROBLOCK *, dec->mbs, dec->last_mbs); /* it will be swapped back */ |
1465 |
break; |
break; |
1466 |
} |
} |
1467 |
|
|
1503 |
/* attemping to decode a bvop without atleast 2 reference frames */ |
/* attemping to decode a bvop without atleast 2 reference frames */ |
1504 |
image_printf(&dec->cur, dec->edged_width, dec->height, 16, 16, |
image_printf(&dec->cur, dec->edged_width, dec->height, 16, 16, |
1505 |
"broken b-frame, mising ref frames"); |
"broken b-frame, mising ref frames"); |
1506 |
|
stats->type = XVID_TYPE_NOTHING; |
1507 |
} else if (dec->time_pp <= dec->time_bp) { |
} else if (dec->time_pp <= dec->time_bp) { |
1508 |
/* this occurs when dx50_bvop_compatibility==0 sequences are |
/* this occurs when dx50_bvop_compatibility==0 sequences are |
1509 |
decoded in vfw. */ |
decoded in vfw. */ |
1510 |
image_printf(&dec->cur, dec->edged_width, dec->height, 16, 16, |
image_printf(&dec->cur, dec->edged_width, dec->height, 16, 16, |
1511 |
"broken b-frame, tpp=%i tbp=%i", dec->time_pp, dec->time_bp); |
"broken b-frame, tpp=%i tbp=%i", dec->time_pp, dec->time_bp); |
1512 |
|
stats->type = XVID_TYPE_NOTHING; |
1513 |
} else { |
} else { |
1514 |
decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward); |
decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward); |
1515 |
|
decoder_output(dec, &dec->cur, dec->mbs, frame, stats, coding_type); |
1516 |
} |
} |
1517 |
|
|
|
decoder_output(dec, &dec->cur, dec->mbs, frame, stats, coding_type); |
|
1518 |
output = 1; |
output = 1; |
1519 |
dec->frames++; |
dec->frames++; |
1520 |
} |
} |