[cvs] / xvidcore / src / decoder.c Repository:
ViewVC logotype

Diff of /xvidcore/src/decoder.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.23, Sun Jun 30 10:46:29 2002 UTC revision 1.31, Thu Jul 18 00:07:04 2002 UTC
# Line 32  Line 32 
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
38     *              Fix a little bug for low_delay flage
39     *              MinChen <chenm001@163.com>
40   *  28.06.2002  added basic resync support to iframe/pframe_decode()   *  28.06.2002  added basic resync support to iframe/pframe_decode()
41   *      22.06.2002      added primative N_VOP support   *      22.06.2002      added primative N_VOP support
42   *                              #define BFRAMES_DEC now enables Minchenm's bframe decoder   *                              #define BFRAMES_DEC now enables Minchen's bframe decoder
43   *  08.05.2002  add low_delay support for B_VOP decode   *  08.05.2002  add low_delay support for B_VOP decode
44   *              MinChen <chenm001@163.com>   *              MinChen <chenm001@163.com>
45   *  05.05.2002  fix some B-frame decode problem   *  05.05.2002  fix some B-frame decode problem
# Line 57  Line 62 
62  #include <stdlib.h>  #include <stdlib.h>
63  #include <string.h>  #include <string.h>
64    
65    #ifdef BFRAMES_DEC_DEBUG
66            #define BFRAMES_DEC
67    #endif
68    
69  #include "xvid.h"  #include "xvid.h"
70  #include "portab.h"  #include "portab.h"
71    
# Line 471  Line 480 
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);
# Line 501  Line 514 
514          int low = ((-32) * scale_fac);          int low = ((-32) * scale_fac);
515          int range = (64 * scale_fac);          int range = (64 * scale_fac);
516    
517          VECTOR pmv[4];          VECTOR pmv;
         int32_t psad[4];  
   
518          int mv_x, mv_y;          int mv_x, mv_y;
         int pmv_x, pmv_y;  
519    
520            pmv = get_pmv2(dec->mbs, dec->mb_width, bound, x, y, k);
         get_pmvdata2(dec->mbs, x, y, dec->mb_width, k, pmv, psad, bound);  
   
         pmv_x = pmv[0].x;  
         pmv_y = pmv[0].y;  
521    
522          mv_x = get_mv(bs, fcode);          mv_x = get_mv(bs, fcode);
523          mv_y = get_mv(bs, fcode);          mv_y = get_mv(bs, fcode);
524    
525          DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i)", mv_x, mv_y, pmv_x, pmv_y);          DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i)", mv_x, mv_y, pmv.x, pmv.y);
526    
527          mv_x += pmv_x;          mv_x += pmv.x;
528          mv_y += pmv_y;          mv_y += pmv.y;
529    
530          if (mv_x < low) {          if (mv_x < low) {
531                  mv_x += range;                  mv_x += range;
# Line 793  Line 799 
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;
# Line 886  Line 893 
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    
# Line 901  Line 907 
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;
# Line 995  Line 1003 
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++) {
# Line 1086  Line 1093 
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
1100            FILE *fp;
1101            static char first=0;
1102    #define BFRAME_DEBUG    if (!first && fp){ \
1103                    fprintf(fp,"Y=%3d   X=%3d   MB=%2d   CBP=%02X\n",y,x,mb->mb_type,mb->cbp); \
1104            }
1105    #endif
1106    
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
1115            if (!first){
1116                    fp=fopen("C:\\XVIDDBG.TXT","w");
1117            }
1118    #endif
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 =                          mv =
1128                                  0;                          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_FORWARD;  
1134                                  mb->cbp = 0;                                  mb->cbp = 0;
1135                                  mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;  #ifdef BFRAMES_DEC_DEBUG
1136                                  mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                  mb->mb_type = MODE_NOT_CODED;
1137                                  mb->quant = 8;          BFRAME_DEBUG
1138    #endif
1139                                    mb->mb_type = MODE_FORWARD;
1140                                    mb->quant = last_mb->quant;
1141                                    //mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1142                                    //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);
# Line 1139  Line 1163 
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
1177            BFRAME_DEBUG
1178    #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;
# Line 1221  Line 1233 
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;
1242    
1243                          default:                          default:
1244                                  DEBUG1("Not support B-frame mb_type =", mb->mb_type);                                  //DEBUG1("Not support B-frame mb_type =", mb->mb_type);
1245                                    ;
1246                          }                          }
1247    
1248                  }                                               // end of FOR                  }                                               // end of FOR
1249          }          }
1250    #ifdef BFRAMES_DEC_DEBUG
1251            if (!first){
1252                    first=1;
1253                    if (fp)
1254                            fclose(fp);
1255            }
1256    #endif
1257  }  }
1258    
1259  // swap two MACROBLOCK array  // swap two MACROBLOCK array
# Line 1313  Line 1331 
1331                  return XVID_ERR_FAIL;                  return XVID_ERR_FAIL;
1332          }          }
1333    
1334    #ifdef BFRAMES_DEC_DEBUG
1335            if (frame->length != BitstreamPos(&bs) / 8){
1336                    DEBUG2("InLen/UseLen",frame->length, BitstreamPos(&bs) / 8);
1337            }
1338    #endif
1339          frame->length = BitstreamPos(&bs) / 8;          frame->length = BitstreamPos(&bs) / 8;
1340    
1341    
1342  #ifdef BFRAMES_DEC  #ifdef BFRAMES_DEC
1343          // test if no B_VOP          // test if no B_VOP
1344          if (dec->low_delay) {          if (dec->low_delay) {
1345  #endif  #endif
1346                  image_output(&dec->cur, dec->width, dec->height, dec->edged_width,                  image_output(&dec->cur, dec->width, dec->height, dec->edged_width,
1347                                           frame->image, frame->stride, frame->colorspace);                                           frame->image, frame->stride, frame->colorspace);
1348    
1349  #ifdef BFRAMES_DEC  #ifdef BFRAMES_DEC
1350          } else {          } else {
1351                  if (dec->frames >= 1) {                  if (dec->frames >= 0) {
1352                          start_timer();                          start_timer();
1353                          if ((vop_type == I_VOP || vop_type == P_VOP)) {                          if ((vop_type == I_VOP || vop_type == P_VOP)) {
1354                                  image_output(&dec->refn[0], dec->width, dec->height,                                  image_output(&dec->refn[0], dec->width, dec->height,
# Line 1342  Line 1367 
1367          if (vop_type == I_VOP || vop_type == P_VOP) {          if (vop_type == I_VOP || vop_type == P_VOP) {
1368                  image_swap(&dec->refn[0], &dec->refn[1]);                  image_swap(&dec->refn[0], &dec->refn[1]);
1369                  image_swap(&dec->cur, &dec->refn[0]);                  image_swap(&dec->cur, &dec->refn[0]);
1370    
1371                  // swap MACROBLOCK                  // swap MACROBLOCK
1372                  if (dec->low_delay && vop_type == P_VOP)                  // the Divx will not set the low_delay flage some times
1373                    // so follow code will wrong to not swap at that time
1374                    // this will broken bitstream! so I'm change it,
1375                    // But that is not the best way! can anyone tell me how
1376                    // to do another way?
1377                    // 18-07-2002   MinChen<chenm001@163.com>
1378                    //if (!dec->low_delay && vop_type == P_VOP)
1379                    if (vop_type == P_VOP)
1380                          mb_swap(&dec->mbs, &dec->last_mbs);                          mb_swap(&dec->mbs, &dec->last_mbs);
1381          }          }
1382    

Legend:
Removed from v.1.23  
changed lines
  Added in v.1.31

No admin address has been configured
ViewVC Help
Powered by ViewVC 1.0.4