[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.43, Sat Nov 16 23:38:15 2002 UTC revision 1.44, Tue Nov 26 23:44:09 2002 UTC
# Line 116  Line 116 
116                  xvid_free(dec);                  xvid_free(dec);
117                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
118          }          }
119          // add by chenm001 <chenm001@163.com>          /* add by chenm001 <chenm001@163.com> */
120          // for support B-frame to reference last 2 frame          /* for support B-frame to reference last 2 frame */
121          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {
122                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
123                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
# Line 146  Line 146 
146    
147          memset(dec->mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);          memset(dec->mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
148    
149          // add by chenm001 <chenm001@163.com>          /* add by chenm001 <chenm001@163.com> */
150          // for skip MB flag          /* for skip MB flag */
151          dec->last_mbs =          dec->last_mbs =
152                  xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,                  xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
153                                          CACHE_LINE);                                          CACHE_LINE);
# Line 165  Line 165 
165    
166          init_timer();          init_timer();
167    
168          // add by chenm001 <chenm001@163.com>          /* add by chenm001 <chenm001@163.com> */
169          // for support B-frame to save reference frame's time          /* for support B-frame to save reference frame's time */
170          dec->frames = -1;          dec->frames = -1;
171          dec->time = dec->time_base = dec->last_time_base = 0;          dec->time = dec->time_base = dec->last_time_base = 0;
172    
# Line 198  Line 198 
198    
199    
200    
201  // decode an intra macroblock  /* decode an intra macroblock */
202    
203  void  void
204  decoder_mbintra(DECODER * dec,  decoder_mbintra(DECODER * dec,
# Line 227  Line 227 
227          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
228          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
229    
230          memset(block, 0, 6 * 64 * sizeof(int16_t));     // clear          memset(block, 0, 6 * 64 * sizeof(int16_t));     /* clear */
231    
232          for (i = 0; i < 6; i++) {          for (i = 0; i < 6; i++) {
233                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
# Line 250  Line 250 
250                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0;                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0;
251    
252                          if (dc_size > 8) {                          if (dc_size > 8) {
253                                  BitstreamSkip(bs, 1);   // marker                                  BitstreamSkip(bs, 1);   /* marker */
254                          }                          }
255    
256                          block[i * 64 + 0] = dc_dif;                          block[i * 64 + 0] = dc_dif;
# Line 262  Line 262 
262                  }                  }
263    
264                  start_timer();                  start_timer();
265                  if (cbp & (1 << (5 - i)))       // coded                  if (cbp & (1 << (5 - i)))       /* coded */
266                  {                  {
267                          get_intra_block(bs, &block[i * 64], pMB->acpred_directions[i],                          get_intra_block(bs, &block[i * 64], pMB->acpred_directions[i],
268                                                          start_coeff);                                                          start_coeff);
# Line 311  Line 311 
311          { 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 };          { 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 };
312    
313    
314  // decode an inter macroblock  /* decode an inter macroblock */
315    
316  void  void
317  decoder_mbinter(DECODER * dec,  decoder_mbinter(DECODER * dec,
# Line 403  Line 403 
403          stop_comp_timer();          stop_comp_timer();
404    
405          for (i = 0; i < 6; i++) {          for (i = 0; i < 6; i++) {
406                  if (cbp & (1 << (5 - i)))       // coded                  if (cbp & (1 << (5 - i)))       /* coded */
407                  {                  {
408                          memset(&block[i * 64], 0, 64 * sizeof(int16_t));        // clear                          memset(&block[i * 64], 0, 64 * sizeof(int16_t));        /* clear */
409    
410                          start_timer();                          start_timer();
411                          get_inter_block(bs, &block[i * 64]);                          get_inter_block(bs, &block[i * 64]);
# Line 591  Line 591 
591                  for (x = 0; x < dec->mb_width; x++) {                  for (x = 0; x < dec->mb_width; x++) {
592                          MACROBLOCK *mb;                          MACROBLOCK *mb;
593    
594                          // skip stuffing                          /* skip stuffing */
595                          while (BitstreamShowBits(bs, 10) == 1)                          while (BitstreamShowBits(bs, 10) == 1)
596                                  BitstreamSkip(bs, 10);                                  BitstreamSkip(bs, 10);
597    
# Line 605  Line 605 
605    
606                          DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));                          DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
607    
608                          //if (!(dec->mb_skip[y*dec->mb_width + x]=BitstreamGetBit(bs)))         // not_coded                          /*if (!(dec->mb_skip[y*dec->mb_width + x]=BitstreamGetBit(bs)))          not_coded */
609                          if (!(BitstreamGetBit(bs)))     // not_coded                          if (!(BitstreamGetBit(bs)))     /* not_coded */
610                          {                          {
611                                  uint32_t mcbpc;                                  uint32_t mcbpc;
612                                  uint32_t cbpc;                                  uint32_t cbpc;
# Line 687  Line 687 
687                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode, bound);                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode, bound);
688                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode, bound);                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode, bound);
689                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode, bound);                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode, bound);
690                                  } else                  // MODE_INTRA, MODE_INTRA_Q                                  } else                  /* MODE_INTRA, MODE_INTRA_Q */
691                                  {                                  {
692                                          mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =                                          mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
693                                                  0;                                                  0;
# Line 700  Line 700 
700    
701                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant,                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant,
702                                                                  rounding);                                                                  rounding);
703                          } else                          // not coded                          } else                          /* not coded */
704                          {                          {
705                                  DPRINTF(DPRINTF_DEBUG, "P-frame MB at (X,Y)=(%d,%d)", x, y);                                  DPRINTF(DPRINTF_DEBUG, "P-frame MB at (X,Y)=(%d,%d)", x, y);
706    
# Line 708  Line 708 
708                                  mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;                                  mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
709                                  mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;                                  mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;
710    
711                                  // copy macroblock directly from ref to cur                                  /* copy macroblock directly from ref to cur */
712    
713                                  start_timer();                                  start_timer();
714    
# Line 759  Line 759 
759  }  }
760    
761    
762  // add by MinChen <chenm001@163.com>  /* add by MinChen <chenm001@163.com> */
763  // decode B-frame motion vector  /* decode B-frame motion vector */
764  void  void
765  get_b_motion_vector(DECODER * dec,  get_b_motion_vector(DECODER * dec,
766                                          Bitstream * bs,                                          Bitstream * bs,
# Line 804  Line 804 
804  }  }
805    
806    
807  // add by MinChen <chenm001@163.com>  /* add by MinChen <chenm001@163.com> */
808  // decode an B-frame forward & backward inter macroblock  /* decode an B-frame forward & backward inter macroblock */
809  void  void
810  decoder_bf_mbinter(DECODER * dec,  decoder_bf_mbinter(DECODER * dec,
811                                     const MACROBLOCK * pMB,                                     const MACROBLOCK * pMB,
# Line 873  Line 873 
873          stop_comp_timer();          stop_comp_timer();
874    
875          for (i = 0; i < 6; i++) {          for (i = 0; i < 6; i++) {
876                  if (cbp & (1 << (5 - i)))       // coded                  if (cbp & (1 << (5 - i)))       /* coded */
877                  {                  {
878                          memset(&block[i * 64], 0, 64 * sizeof(int16_t));        // clear                          memset(&block[i * 64], 0, 64 * sizeof(int16_t));        /* clear */
879    
880                          start_timer();                          start_timer();
881                          get_inter_block(bs, &block[i * 64]);                          get_inter_block(bs, &block[i * 64]);
# Line 917  Line 917 
917  }  }
918    
919    
920  // add by MinChen <chenm001@163.com>  /* add by MinChen <chenm001@163.com> */
921  // decode an B-frame direct &  inter macroblock  /* decode an B-frame direct &  inter macroblock */
922  void  void
923  decoder_bf_interpolate_mbinter(DECODER * dec,  decoder_bf_interpolate_mbinter(DECODER * dec,
924                                                             IMAGE forward,                                                             IMAGE forward,
# Line 1039  Line 1039 
1039          stop_comp_timer();          stop_comp_timer();
1040    
1041          for (i = 0; i < 6; i++) {          for (i = 0; i < 6; i++) {
1042                  if (cbp & (1 << (5 - i)))       // coded                  if (cbp & (1 << (5 - i)))       /* coded */
1043                  {                  {
1044                          memset(&block[i * 64], 0, 64 * sizeof(int16_t));        // clear                          memset(&block[i * 64], 0, 64 * sizeof(int16_t));        /* clear */
1045    
1046                          start_timer();                          start_timer();
1047                          get_inter_block(bs, &block[i * 64]);                          get_inter_block(bs, &block[i * 64]);
# Line 1083  Line 1083 
1083  }  }
1084    
1085    
1086  // add by MinChen <chenm001@163.com>  /* add by MinChen <chenm001@163.com> */
1087  // for decode B-frame dbquant  /* for decode B-frame dbquant */
1088  int32_t __inline  int32_t __inline
1089  get_dbquant(Bitstream * bs)  get_dbquant(Bitstream * bs)
1090  {  {
1091          if (!BitstreamGetBit(bs))       // '0'          if (!BitstreamGetBit(bs))       /* '0' */
1092                  return (0);                  return (0);
1093          else if (!BitstreamGetBit(bs))  // '10'          else if (!BitstreamGetBit(bs))  /* '10' */
1094                  return (-2);                  return (-2);
1095          else          else
1096                  return (2);                             // '11'                  return (2);                             /* '11' */
1097  }  }
1098    
1099  // add by MinChen <chenm001@163.com>  /* add by MinChen <chenm001@163.com> */
1100  // for decode B-frame mb_type  /* for decode B-frame mb_type */
1101  // bit   ret_value  /* bit   ret_value */
1102  // 1        0  /* 1        0 */
1103  // 01       1  /* 01       1 */
1104  // 001      2  /* 001      2 */
1105  // 0001     3  /* 0001     3 */
1106  int32_t __inline  int32_t __inline
1107  get_mbtype(Bitstream * bs)  get_mbtype(Bitstream * bs)
1108  {  {
# Line 1151  Line 1151 
1151  #endif  #endif
1152    
1153          for (y = 0; y < dec->mb_height; y++) {          for (y = 0; y < dec->mb_height; y++) {
1154                  // Initialize Pred Motion Vector                  /* Initialize Pred Motion Vector */
1155                  dec->p_fmv = dec->p_bmv = zeromv;                  dec->p_fmv = dec->p_bmv = zeromv;
1156                  for (x = 0; x < dec->mb_width; x++) {                  for (x = 0; x < dec->mb_width; x++) {
1157                          MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x];                          MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x];
# Line 1161  Line 1161 
1161                          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] =
1162                          mb->mvs[0] = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = zeromv;                          mb->mvs[0] = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = zeromv;
1163    
1164                          // the last P_VOP is skip macroblock ?                          /* the last P_VOP is skip macroblock ? */
1165                          if (last_mb->mode == MODE_NOT_CODED) {                          if (last_mb->mode == MODE_NOT_CODED) {
1166                                  //DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y);                                  /*DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y); */
1167                                  mb->cbp = 0;                                  mb->cbp = 0;
1168  #ifdef BFRAMES_DEC_DEBUG  #ifdef BFRAMES_DEC_DEBUG
1169                                  mb->mb_type = MODE_NOT_CODED;                                  mb->mb_type = MODE_NOT_CODED;
# Line 1171  Line 1171 
1171  #endif  #endif
1172                                  mb->mb_type = MODE_FORWARD;                                  mb->mb_type = MODE_FORWARD;
1173                                  mb->quant = last_mb->quant;                                  mb->quant = last_mb->quant;
1174                                  //mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                  /*mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x; */
1175                                  //mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                  /*mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y; */
1176    
1177                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, mb->quant, 1);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, mb->quant, 1);
1178                                  continue;                                  continue;
1179                          }                          }
1180    
1181                          if (!BitstreamGetBit(bs)) {     // modb=='0'                          if (!BitstreamGetBit(bs)) {     /* modb=='0' */
1182                                  const uint8_t modb2 = BitstreamGetBit(bs);                                  const uint8_t modb2 = BitstreamGetBit(bs);
1183    
1184                                  mb->mb_type = get_mbtype(bs);                                  mb->mb_type = get_mbtype(bs);
1185    
1186                                  if (!modb2) {   // modb=='00'                                  if (!modb2) {   /* modb=='00' */
1187                                          mb->cbp = BitstreamGetBits(bs, 6);                                          mb->cbp = BitstreamGetBits(bs, 6);
1188                                  } else {                                  } else {
1189                                          mb->cbp = 0;                                          mb->cbp = 0;
# Line 1204  Line 1204 
1204    
1205                          mb->quant = quant;                          mb->quant = quant;
1206                          mb->mode = MODE_INTER4V;                          mb->mode = MODE_INTER4V;
1207                          //DEBUG1("Switch bm_type=",mb->mb_type);                          /*DEBUG1("Switch bm_type=",mb->mb_type); */
1208    
1209  #ifdef BFRAMES_DEC_DEBUG  #ifdef BFRAMES_DEC_DEBUG
1210          BFRAME_DEBUG          BFRAME_DEBUG
# Line 1233  Line 1233 
1233                                                                                    / TRD                                                                                    / TRD
1234                                                                              : mb->mvs[i].y - last_mb->mvs[i].y);                                                                              : mb->mvs[i].y - last_mb->mvs[i].y);
1235                                          }                                          }
1236                                          //DEBUG("B-frame Direct!\n");                                          /*DEBUG("B-frame Direct!\n"); */
1237                                  }                                  }
1238                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1239                                                                                             mb, x, y, bs);                                                                                             mb, x, y, bs);
# Line 1251  Line 1251 
1251    
1252                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1253                                                                                             mb, x, y, bs);                                                                                             mb, x, y, bs);
1254                                  //DEBUG("B-frame Bidir!\n");                                  /*DEBUG("B-frame Bidir!\n"); */
1255                                  break;                                  break;
1256    
1257                          case MODE_BACKWARD:                          case MODE_BACKWARD:
# Line 1261  Line 1261 
1261    
1262                                  mb->mode = MODE_INTER;                                  mb->mode = MODE_INTER;
1263                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);
1264                                  //DEBUG("B-frame Backward!\n");                                  /*DEBUG("B-frame Backward!\n"); */
1265                                  break;                                  break;
1266    
1267                          case MODE_FORWARD:                          case MODE_FORWARD:
# Line 1271  Line 1271 
1271    
1272                                  mb->mode = MODE_INTER;                                  mb->mode = MODE_INTER;
1273                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1274                                  //DEBUG("B-frame Forward!\n");                                  /*DEBUG("B-frame Forward!\n"); */
1275                                  break;                                  break;
1276    
1277                          default:                          default:
1278                                  DPRINTF(DPRINTF_ERROR, "Not support B-frame mb_type = %d", mb->mb_type);                                  DPRINTF(DPRINTF_ERROR, "Not support B-frame mb_type = %d", mb->mb_type);
1279                          }                          }
1280    
1281                  }                                               // end of FOR                  }                                               /* end of FOR */
1282          }          }
1283  #ifdef BFRAMES_DEC_DEBUG  #ifdef BFRAMES_DEC_DEBUG
1284          if (!first){          if (!first){
# Line 1289  Line 1289 
1289  #endif  #endif
1290  }  }
1291    
1292  // swap two MACROBLOCK array  /* swap two MACROBLOCK array */
1293  void  void
1294  mb_swap(MACROBLOCK ** mb1,  mb_swap(MACROBLOCK ** mb1,
1295                  MACROBLOCK ** mb2)                  MACROBLOCK ** mb2)
# Line 1319  Line 1319 
1319    
1320          BitstreamInit(&bs, frame->bitstream, frame->length);          BitstreamInit(&bs, frame->bitstream, frame->length);
1321    
1322          // add by chenm001 <chenm001@163.com>          /* add by chenm001 <chenm001@163.com> */
1323          // for support B-frame to reference last 2 frame          /* for support B-frame to reference last 2 frame */
1324          dec->frames++;          dec->frames++;
1325          vop_type =          vop_type =
1326                  BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward,                  BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward,
1327                                                           &fcode_backward, &intra_dc_threshold);                                                           &fcode_backward, &intra_dc_threshold);
1328    
1329          dec->p_bmv.x = dec->p_bmv.y = dec->p_fmv.y = dec->p_fmv.y = 0;  // init pred vector to 0          dec->p_bmv.x = dec->p_bmv.y = dec->p_fmv.y = dec->p_fmv.y = 0;  /* init pred vector to 0 */
1330    
1331          switch (vop_type) {          switch (vop_type) {
1332          case P_VOP:          case P_VOP:
# Line 1357  Line 1357 
1357  #endif  #endif
1358                  break;                  break;
1359    
1360          case N_VOP:                             // vop not coded          case N_VOP:                             /* vop not coded */
1361                  // when low_delay==0, N_VOP's should interpolate between the past and future frames                  /* when low_delay==0, N_VOP's should interpolate between the past and future frames */
1362                  image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);                  image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1363                  break;                  break;
1364    
# Line 1375  Line 1375 
1375    
1376    
1377  #ifdef BFRAMES_DEC  #ifdef BFRAMES_DEC
1378          // test if no B_VOP          /* test if no B_VOP */
1379          if (dec->low_delay || dec->frames == 0) {          if (dec->low_delay || dec->frames == 0) {
1380  #endif  #endif
1381          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,
# Line 1403  Line 1403 
1403                  image_swap(&dec->refn[0], &dec->refn[1]);                  image_swap(&dec->refn[0], &dec->refn[1]);
1404                  image_swap(&dec->cur, &dec->refn[0]);                  image_swap(&dec->cur, &dec->refn[0]);
1405    
1406                  // swap MACROBLOCK                  /* swap MACROBLOCK */
1407                  // the Divx will not set the low_delay flage some times                  /* the Divx will not set the low_delay flage some times */
1408                  // so follow code will wrong to not swap at that time                  /* so follow code will wrong to not swap at that time */
1409                  // this will broken bitstream! so I'm change it,                  /* this will broken bitstream! so I'm change it, */
1410                  // But that is not the best way! can anyone tell me how                  /* But that is not the best way! can anyone tell me how */
1411                  // to do another way?                  /* to do another way? */
1412                  // 18-07-2002   MinChen<chenm001@163.com>                  /* 18-07-2002   MinChen<chenm001@163.com> */
1413                  //if (!dec->low_delay && vop_type == P_VOP)                  /*if (!dec->low_delay && vop_type == P_VOP) */
1414                  if (vop_type == P_VOP)                  if (vop_type == P_VOP)
1415                          mb_swap(&dec->mbs, &dec->last_mbs);                          mb_swap(&dec->mbs, &dec->last_mbs);
1416          }          }

Legend:
Removed from v.1.43  
changed lines
  Added in v.1.44

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