32 |
* |
* |
33 |
* History: |
* History: |
34 |
* |
* |
35 |
|
* 15.07.2002 fix a bug in B-frame decode at DIRECT mode |
36 |
|
* MinChen <chenm001@163.com> |
37 |
* 10.07.2002 added BFRAMES_DEC_DEBUG support |
* 10.07.2002 added BFRAMES_DEC_DEBUG support |
38 |
* Fix a little bug for low_delay flage |
* Fix a little bug for low_delay flage |
39 |
* MinChen <chenm001@163.com> |
* MinChen <chenm001@163.com> |
55 |
* 22.12.2001 lock based interpolation |
* 22.12.2001 lock based interpolation |
56 |
* 01.12.2001 inital version; (c)2001 peter ross <pross@cs.rmit.edu.au> |
* 01.12.2001 inital version; (c)2001 peter ross <pross@cs.rmit.edu.au> |
57 |
* |
* |
58 |
* $Id: decoder.c,v 1.27 2002-07-11 00:15:59 chenm001 Exp $ |
* $Id: decoder.c,v 1.30 2002-07-15 23:50:31 chenm001 Exp $ |
59 |
* |
* |
60 |
*************************************************************************/ |
*************************************************************************/ |
61 |
|
|
480 |
} |
} |
481 |
} |
} |
482 |
mb->quant = quant; |
mb->quant = quant; |
483 |
|
mb->mvs[0].x = mb->mvs[0].y = |
484 |
|
mb->mvs[1].x = mb->mvs[1].y = |
485 |
|
mb->mvs[2].x = mb->mvs[2].y = |
486 |
|
mb->mvs[3].x = mb->mvs[3].y =0; |
487 |
|
|
488 |
if (dec->interlacing) { |
if (dec->interlacing) { |
489 |
mb->field_dct = BitstreamGetBit(bs); |
mb->field_dct = BitstreamGetBit(bs); |
799 |
pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3); |
pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3); |
800 |
pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3); |
pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3); |
801 |
|
|
802 |
|
|
803 |
if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) { |
if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) { |
804 |
uv_dx = pMB->mvs[0].x; |
uv_dx = pMB->mvs[0].x; |
805 |
uv_dy = pMB->mvs[0].y; |
uv_dy = pMB->mvs[0].y; |
893 |
const MACROBLOCK * pMB, |
const MACROBLOCK * pMB, |
894 |
const uint32_t x_pos, |
const uint32_t x_pos, |
895 |
const uint32_t y_pos, |
const uint32_t y_pos, |
|
const uint32_t cbp, |
|
896 |
Bitstream * bs) |
Bitstream * bs) |
897 |
{ |
{ |
898 |
|
|
907 |
int b_uv_dx, b_uv_dy; |
int b_uv_dx, b_uv_dy; |
908 |
uint32_t i; |
uint32_t i; |
909 |
uint8_t *pY_Cur, *pU_Cur, *pV_Cur; |
uint8_t *pY_Cur, *pU_Cur, *pV_Cur; |
910 |
|
const uint32_t cbp = pMB->cbp; |
911 |
|
|
912 |
pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4); |
pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4); |
913 |
pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3); |
pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3); |
914 |
pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3); |
pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3); |
915 |
|
|
916 |
|
|
917 |
if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) { |
if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) { |
918 |
uv_dx = pMB->mvs[0].x; |
uv_dx = pMB->mvs[0].x; |
919 |
uv_dy = pMB->mvs[0].y; |
uv_dy = pMB->mvs[0].y; |
1003 |
stride2); |
stride2); |
1004 |
interpolate8x8_c(dec->cur.v, dec->refn[2].v, 8 * x_pos, 8 * y_pos, |
interpolate8x8_c(dec->cur.v, dec->refn[2].v, 8 * x_pos, 8 * y_pos, |
1005 |
stride2); |
stride2); |
|
|
|
1006 |
stop_comp_timer(); |
stop_comp_timer(); |
1007 |
|
|
1008 |
for (i = 0; i < 6; i++) { |
for (i = 0; i < 6; i++) { |
1093 |
int fcode_forward, |
int fcode_forward, |
1094 |
int fcode_backward) |
int fcode_backward) |
1095 |
{ |
{ |
|
|
|
1096 |
uint32_t x, y; |
uint32_t x, y; |
1097 |
VECTOR mv, zeromv; |
VECTOR mv; |
1098 |
|
const VECTOR zeromv = {0,0}; |
1099 |
#ifdef BFRAMES_DEC_DEBUG |
#ifdef BFRAMES_DEC_DEBUG |
1100 |
FILE *fp; |
FILE *fp; |
1101 |
static char first=0; |
static char first=0; |
1107 |
start_timer(); |
start_timer(); |
1108 |
image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height, |
image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height, |
1109 |
dec->width, dec->height, dec->interlacing); |
dec->width, dec->height, dec->interlacing); |
1110 |
//image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing); |
image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height, |
1111 |
|
dec->width, dec->height, dec->interlacing); |
1112 |
stop_edges_timer(); |
stop_edges_timer(); |
1113 |
|
|
1114 |
#ifdef BFRAMES_DEC_DEBUG |
#ifdef BFRAMES_DEC_DEBUG |
1119 |
|
|
1120 |
for (y = 0; y < dec->mb_height; y++) { |
for (y = 0; y < dec->mb_height; y++) { |
1121 |
// Initialize Pred Motion Vector |
// Initialize Pred Motion Vector |
1122 |
dec->p_fmv.x = dec->p_fmv.y = dec->p_bmv.x = dec->p_bmv.y = 0; |
dec->p_fmv = dec->p_bmv = zeromv; |
1123 |
for (x = 0; x < dec->mb_width; x++) { |
for (x = 0; x < dec->mb_width; x++) { |
1124 |
MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x]; |
MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x]; |
1125 |
MACROBLOCK *last_mb = &dec->last_mbs[y * dec->mb_width + x]; |
MACROBLOCK *last_mb = &dec->last_mbs[y * dec->mb_width + x]; |
1126 |
|
|
1127 |
mb->mvs[0].x = mb->mvs[0].y = zeromv.x = zeromv.y = mv.x = mv.y = 0; |
mv = |
1128 |
|
mb->b_mvs[0] = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] = |
1129 |
|
mb->mvs[0] = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = zeromv; |
1130 |
|
|
1131 |
// the last P_VOP is skip macroblock ? |
// the last P_VOP is skip macroblock ? |
1132 |
if (last_mb->mode == MODE_NOT_CODED) { |
if (last_mb->mode == MODE_NOT_CODED) { |
1133 |
//DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y); |
//DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y); |
|
mb->mb_type = MODE_NOT_CODED; |
|
1134 |
mb->cbp = 0; |
mb->cbp = 0; |
1135 |
#ifdef BFRAMES_DEC_DEBUG |
#ifdef BFRAMES_DEC_DEBUG |
1136 |
|
mb->mb_type = MODE_NOT_CODED; |
1137 |
BFRAME_DEBUG |
BFRAME_DEBUG |
1138 |
#endif |
#endif |
1139 |
mb->mb_type = MODE_FORWARD; |
mb->mb_type = MODE_FORWARD; |
1140 |
mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x; |
mb->quant = last_mb->quant; |
1141 |
mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y; |
//mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x; |
1142 |
mb->quant = 8; |
//mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y; |
1143 |
|
|
1144 |
decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1); |
decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, mb->quant, 1); |
1145 |
continue; |
continue; |
1146 |
} |
} |
|
//t=BitstreamShowBits(bs,32); |
|
1147 |
|
|
1148 |
if (!BitstreamGetBit(bs)) { // modb=='0' |
if (!BitstreamGetBit(bs)) { // modb=='0' |
1149 |
const uint8_t modb2 = BitstreamGetBit(bs); |
const uint8_t modb2 = BitstreamGetBit(bs); |
1163 |
} else if (quant < 1) { |
} else if (quant < 1) { |
1164 |
quant = 1; |
quant = 1; |
1165 |
} |
} |
|
} else { |
|
|
quant = 8; |
|
1166 |
} |
} |
|
mb->quant = quant; |
|
1167 |
} else { |
} else { |
1168 |
mb->mb_type = MODE_DIRECT_NONE_MV; |
mb->mb_type = MODE_DIRECT_NONE_MV; |
1169 |
mb->cbp = 0; |
mb->cbp = 0; |
1170 |
} |
} |
1171 |
|
|
1172 |
mb->mode = MODE_INTER; |
mb->quant = quant; |
1173 |
|
mb->mode = MODE_INTER4V; |
1174 |
//DEBUG1("Switch bm_type=",mb->mb_type); |
//DEBUG1("Switch bm_type=",mb->mb_type); |
1175 |
|
|
1176 |
#ifdef BFRAMES_DEC_DEBUG |
#ifdef BFRAMES_DEC_DEBUG |
1178 |
#endif |
#endif |
1179 |
switch (mb->mb_type) { |
switch (mb->mb_type) { |
1180 |
case MODE_DIRECT: |
case MODE_DIRECT: |
1181 |
get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], 1, zeromv); |
get_b_motion_vector(dec, bs, x, y, &mv, 1, zeromv); |
1182 |
|
|
1183 |
case MODE_DIRECT_NONE_MV: |
case MODE_DIRECT_NONE_MV: |
1184 |
{ // Because this file is a C file not C++ so I use '{' to define var |
{ // Because this file is a C file not C++ so I use '{' to define var |
1185 |
const int64_t TRB = dec->time_pp - dec->time_bp, TRD = |
const int64_t TRB = dec->time_pp - dec->time_bp, TRD = dec->time_pp; |
|
dec->time_pp; |
|
1186 |
int i; |
int i; |
1187 |
|
|
1188 |
for (i = 0; i < 4; i++) { |
for (i = 0; i < 4; i++) { |
1189 |
mb->mvs[i].x = |
mb->mvs[i].x = (int32_t) ((TRB * last_mb->mvs[i].x) |
1190 |
(int32_t) ((TRB * last_mb->mvs[i].x) / TRD + |
/ TRD + mv.x); |
1191 |
mb->mvs[0].x); |
mb->b_mvs[i].x = (int32_t) ((mv.x == 0) |
1192 |
mb->b_mvs[i].x = |
? ((TRB - TRD) * last_mb->mvs[i].x) |
1193 |
(int32_t) ((mb->mvs[0].x == |
/ TRD |
1194 |
0) ? ((TRB - |
: mb->mvs[i].x - last_mb->mvs[i].x); |
1195 |
TRD) * last_mb->mvs[i].x) / |
mb->mvs[i].y = (int32_t) ((TRB * last_mb->mvs[i].y) |
1196 |
TRD : mb->mvs[i].x - last_mb->mvs[i].x); |
/ TRD + mv.y); |
1197 |
mb->mvs[i].y = |
mb->b_mvs[i].y = (int32_t) ((mv.y == 0) |
1198 |
(int32_t) ((TRB * last_mb->mvs[i].y) / TRD + |
? ((TRB - TRD) * last_mb->mvs[i].y) |
1199 |
mb->mvs[0].y); |
/ TRD |
1200 |
mb->b_mvs[i].y = |
: mb->mvs[i].y - last_mb->mvs[i].y); |
|
(int32_t) ((mb->mvs[0].y == |
|
|
0) ? ((TRB - |
|
|
TRD) * last_mb->mvs[i].y) / |
|
|
TRD : mb->mvs[i].y - last_mb->mvs[i].y); |
|
1201 |
} |
} |
1202 |
//DEBUG("B-frame Direct!\n"); |
//DEBUG("B-frame Direct!\n"); |
1203 |
} |
} |
|
mb->mode = MODE_INTER4V; |
|
1204 |
decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], |
decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], |
1205 |
mb, x, y, mb->cbp, bs); |
mb, x, y, bs); |
1206 |
break; |
break; |
1207 |
|
|
1208 |
case MODE_INTERPOLATE: |
case MODE_INTERPOLATE: |
1209 |
get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward, |
get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward, |
1210 |
dec->p_fmv); |
dec->p_fmv); |
1211 |
dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = |
dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0]; |
|
mb->mvs[0].x; |
|
|
dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = |
|
|
mb->mvs[0].y; |
|
1212 |
|
|
1213 |
get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0], |
get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0], |
1214 |
fcode_backward, dec->p_bmv); |
fcode_backward, dec->p_bmv); |
1215 |
dec->p_bmv.x = mb->b_mvs[1].x = mb->b_mvs[2].x = |
dec->p_bmv = mb->b_mvs[1] = mb->b_mvs[2] = |
1216 |
mb->b_mvs[3].x = mb->b_mvs[0].x; |
mb->b_mvs[3] = mb->b_mvs[0]; |
|
dec->p_bmv.y = mb->b_mvs[1].y = mb->b_mvs[2].y = |
|
|
mb->b_mvs[3].y = mb->b_mvs[0].y; |
|
1217 |
|
|
1218 |
decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], |
decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], |
1219 |
mb, x, y, mb->cbp, bs); |
mb, x, y, bs); |
1220 |
//DEBUG("B-frame Bidir!\n"); |
//DEBUG("B-frame Bidir!\n"); |
1221 |
break; |
break; |
1222 |
|
|
1223 |
case MODE_BACKWARD: |
case MODE_BACKWARD: |
1224 |
get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_backward, |
get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_backward, |
1225 |
dec->p_bmv); |
dec->p_bmv); |
1226 |
dec->p_bmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = |
dec->p_bmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0]; |
|
mb->mvs[0].x; |
|
|
dec->p_bmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = |
|
|
mb->mvs[0].y; |
|
1227 |
|
|
1228 |
|
mb->mode = MODE_INTER; |
1229 |
decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0); |
decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0); |
1230 |
//DEBUG("B-frame Backward!\n"); |
//DEBUG("B-frame Backward!\n"); |
1231 |
break; |
break; |
1233 |
case MODE_FORWARD: |
case MODE_FORWARD: |
1234 |
get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward, |
get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward, |
1235 |
dec->p_fmv); |
dec->p_fmv); |
1236 |
dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = |
dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0]; |
|
mb->mvs[0].x; |
|
|
dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = |
|
|
mb->mvs[0].y; |
|
1237 |
|
|
1238 |
|
mb->mode = MODE_INTER; |
1239 |
decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1); |
decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1); |
1240 |
//DEBUG("B-frame Forward!\n"); |
//DEBUG("B-frame Forward!\n"); |
1241 |
break; |
break; |
1331 |
return XVID_ERR_FAIL; |
return XVID_ERR_FAIL; |
1332 |
} |
} |
1333 |
|
|
1334 |
|
#ifdef BFRAMES_DEC_DEBUG |
1335 |
if (frame->length != BitstreamPos(&bs) / 8){ |
if (frame->length != BitstreamPos(&bs) / 8){ |
1336 |
DEBUG2("InLen/UseLen",frame->length, BitstreamPos(&bs) / 8); |
DEBUG2("InLen/UseLen",frame->length, BitstreamPos(&bs) / 8); |
1337 |
} |
} |
1338 |
|
#endif |
1339 |
frame->length = BitstreamPos(&bs) / 8; |
frame->length = BitstreamPos(&bs) / 8; |
1340 |
|
|
1341 |
|
|