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

Diff of /trunk/xvidcore/src/decoder.c

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

revision 179, Mon May 20 17:12:53 2002 UTC revision 195, Wed Jun 12 20:38:41 2002 UTC
# Line 47  Line 47 
47   *  22.12.2001  lock based interpolation   *  22.12.2001  lock based interpolation
48   *  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>
49   *   *
50   *  $Id: decoder.c,v 1.18 2002-05-20 17:12:53 Isibaar Exp $   *  $Id: decoder.c,v 1.19 2002-06-12 20:38:39 edgomez Exp $
51   *   *
52   *************************************************************************/   *************************************************************************/
53    
# Line 77  Line 77 
77  #include "image/colorspace.h"  #include "image/colorspace.h"
78  #include "utils/mem_align.h"  #include "utils/mem_align.h"
79    
80  int decoder_create(XVID_DEC_PARAM * param)  int
81    decoder_create(XVID_DEC_PARAM * param)
82  {  {
83          DECODER * dec;          DECODER * dec;
84    
85          dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);          dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);
86          if (dec == NULL)          if (dec == NULL) {
         {  
87                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
88          }          }
89          param->handle = dec;          param->handle = dec;
# Line 97  Line 97 
97          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;
98          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;
99    
100          if (image_create(&dec->cur, dec->edged_width, dec->edged_height))          if (image_create(&dec->cur, dec->edged_width, dec->edged_height)) {
         {  
101                  xvid_free(dec);                  xvid_free(dec);
102                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
103          }          }
104    
105          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height)) {
         {  
106                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
107                  xvid_free(dec);                  xvid_free(dec);
108                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
109          }          }
110          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
111          // for support B-frame to reference last 2 frame          // for support B-frame to reference last 2 frame
112          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {
         {  
113                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
114                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
115                  xvid_free(dec);                  xvid_free(dec);
116                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
117          }          }
118          if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height)) {
         {  
119                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
120                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
121                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
# Line 127  Line 123 
123                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
124          }          }
125    
126          dec->mbs = xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height, CACHE_LINE);          dec->mbs =
127          if (dec->mbs == NULL)                  xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
128          {                                          CACHE_LINE);
129            if (dec->mbs == NULL) {
130                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
131                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
132                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
# Line 139  Line 136 
136          }          }
137          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
138          // for skip MB flag          // for skip MB flag
139          dec->last_mbs = xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height, CACHE_LINE);          dec->last_mbs =
140          if (dec->last_mbs == NULL)                  xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
141          {                                          CACHE_LINE);
142            if (dec->last_mbs == NULL) {
143                  xvid_free(dec->mbs);                  xvid_free(dec->mbs);
144                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
145                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
# Line 162  Line 160 
160  }  }
161    
162    
163  int decoder_destroy(DECODER * dec)  int
164    decoder_destroy(DECODER * dec)
165  {  {
166          xvid_free(dec->last_mbs);          xvid_free(dec->last_mbs);
167          xvid_free(dec->mbs);          xvid_free(dec->mbs);
# Line 178  Line 177 
177    
178    
179    
180  static const int32_t dquant_table[4] =  static const int32_t dquant_table[4] = {
 {  
181          -1, -2, 1, 2          -1, -2, 1, 2
182  };  };
183    
# Line 188  Line 186 
186    
187  // decode an intra macroblock  // decode an intra macroblock
188    
189  void decoder_mbintra(DECODER * dec,  void
190    decoder_mbintra(DECODER * dec,
191                       MACROBLOCK * pMB,                       MACROBLOCK * pMB,
192                       const uint32_t x_pos,                       const uint32_t x_pos,
193                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 215  Line 214 
214    
215          memset(block, 0, 6*64*sizeof(int16_t));         // clear          memset(block, 0, 6*64*sizeof(int16_t));         // clear
216    
217          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
         {  
218                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
219                  int16_t predictors[8];                  int16_t predictors[8];
220                  int start_coeff;                  int start_coeff;
221    
222                  start_timer();                  start_timer();
223                  predict_acdc(dec->mbs, x_pos, y_pos, dec->mb_width, i, &block[i*64], iQuant, iDcScaler, predictors);                  predict_acdc(dec->mbs, x_pos, y_pos, dec->mb_width, i, &block[i * 64],
224                  if (!acpred_flag)                                           iQuant, iDcScaler, predictors);
225                  {                  if (!acpred_flag) {
226                          pMB->acpred_directions[i] = 0;                          pMB->acpred_directions[i] = 0;
227                  }                  }
228                  stop_prediction_timer();                  stop_prediction_timer();
229    
230                  if (quant < intra_dc_threshold)                  if (quant < intra_dc_threshold) {
                 {  
231                          int dc_size;                          int dc_size;
232                          int dc_dif;                          int dc_dif;
233    
234                          dc_size = i < 4 ?  get_dc_size_lum(bs) : get_dc_size_chrom(bs);                          dc_size = i < 4 ?  get_dc_size_lum(bs) : get_dc_size_chrom(bs);
235                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;
236    
237                          if (dc_size > 8)                          if (dc_size > 8) {
                         {  
238                                  BitstreamSkip(bs, 1);           // marker                                  BitstreamSkip(bs, 1);           // marker
239                          }                          }
240    
241                          block[i*64 + 0] = dc_dif;                          block[i*64 + 0] = dc_dif;
242                          start_coeff = 1;                          start_coeff = 1;
243                  }                  } else {
                 else  
                 {  
244                          start_coeff = 0;                          start_coeff = 0;
245                  }                  }
246    
247                  start_timer();                  start_timer();
248                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
249                  {                  {
250                          get_intra_block(bs, &block[i*64], pMB->acpred_directions[i], start_coeff);                          get_intra_block(bs, &block[i * 64], pMB->acpred_directions[i],
251                                                            start_coeff);
252                  }                  }
253                  stop_coding_timer();                  stop_coding_timer();
254    
# Line 262  Line 257 
257                  stop_prediction_timer();                  stop_prediction_timer();
258    
259                  start_timer();                  start_timer();
260                  if (dec->quant_type == 0)                  if (dec->quant_type == 0) {
                 {  
261                          dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);                          dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
262                  }                  } else {
                 else  
                 {  
263                          dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);                          dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
264                  }                  }
265                  stop_iquant_timer();                  stop_iquant_timer();
# Line 277  Line 269 
269                  stop_idct_timer();                  stop_idct_timer();
270          }          }
271    
272          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
273                  next_block = stride;                  next_block = stride;
274                  stride *= 2;                  stride *= 2;
275          }          }
# Line 305  Line 296 
296    
297  // decode an inter macroblock  // decode an inter macroblock
298    
299  void decoder_mbinter(DECODER * dec,  void
300    decoder_mbinter(DECODER * dec,
301                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
302                       const uint32_t x_pos,                       const uint32_t x_pos,
303                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 331  Line 323 
323          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
324          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
325    
326          if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)          if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
         {  
327                  uv_dx = pMB->mvs[0].x;                  uv_dx = pMB->mvs[0].x;
328                  uv_dy = pMB->mvs[0].y;                  uv_dy = pMB->mvs[0].y;
329    
330                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
331                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
332          }          } else {
         else  
         {  
333                  int sum;                  int sum;
334    
335                  sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;                  sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
336                  uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dx =
337                            (sum ==
338                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
339                                                                      (ABS(sum) / 16) * 2));
340    
341                  sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;                  sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
342                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dy =
343                            (sum ==
344                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
345                                                                      (ABS(sum) / 16) * 2));
346          }          }
347    
348          start_timer();          start_timer();
349          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos,     16*y_pos    , pMB->mvs[0].x, pMB->mvs[0].y, stride,  rounding);          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16 * x_pos, 16 * y_pos,
350          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos    , pMB->mvs[1].x, pMB->mvs[1].y, stride,  rounding);                                                    pMB->mvs[0].x, pMB->mvs[0].y, stride, rounding);
351          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos,     16*y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,  rounding);          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16 * x_pos + 8,
352          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,  rounding);                                                    16 * y_pos, pMB->mvs[1].x, pMB->mvs[1].y, stride,
353          interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, rounding);                                                    rounding);
354          interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, rounding);          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16 * x_pos,
355                                                      16 * y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,
356                                                      rounding);
357            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16 * x_pos + 8,
358                                                      16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
359                                                      rounding);
360            interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
361                                                      uv_dx, uv_dy, stride2, rounding);
362            interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
363                                                      uv_dx, uv_dy, stride2, rounding);
364          stop_comp_timer();          stop_comp_timer();
365    
366          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
         {  
367                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
368                  {                  {
369                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear
# Line 369  Line 373 
373                          stop_coding_timer();                          stop_coding_timer();
374    
375                          start_timer();                          start_timer();
376                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
                         {  
377                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                                  dequant_inter(&data[i*64], &block[i*64], iQuant);
378                          }                          } else {
                         else  
                         {  
379                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);
380                          }                          }
381                          stop_iquant_timer();                          stop_iquant_timer();
# Line 385  Line 386 
386                  }                  }
387          }          }
388    
389          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
390                  next_block = stride;                  next_block = stride;
391                  stride *= 2;                  stride *= 2;
392          }          }
# Line 408  Line 408 
408  }  }
409    
410    
411  void decoder_iframe(DECODER * dec, Bitstream * bs, int quant, int intra_dc_threshold)  void
412    decoder_iframe(DECODER * dec,
413                               Bitstream * bs,
414                               int quant,
415                               int intra_dc_threshold)
416  {  {
417    
418          uint32_t x, y;          uint32_t x, y;
419    
420          for (y = 0; y < dec->mb_height; y++)          for (y = 0; y < dec->mb_height; y++) {
421          {                  for (x = 0; x < dec->mb_width; x++) {
                 for (x = 0; x < dec->mb_width; x++)  
                 {  
422                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
423    
424                          uint32_t mcbpc;                          uint32_t mcbpc;
# Line 431  Line 433 
433    
434                          acpred_flag = BitstreamGetBit(bs);                          acpred_flag = BitstreamGetBit(bs);
435    
436                          if (mb->mode == MODE_STUFFING)                          if (mb->mode == MODE_STUFFING) {
                         {  
437                                  DEBUG("-- STUFFING ?");                                  DEBUG("-- STUFFING ?");
438                                  continue;                                  continue;
439                          }                          }
# Line 440  Line 441 
441                          cbpy = get_cbpy(bs, 1);                          cbpy = get_cbpy(bs, 1);
442                          cbp = (cbpy << 2) | cbpc;                          cbp = (cbpy << 2) | cbpc;
443    
444                          if (mb->mode == MODE_INTRA_Q)                          if (mb->mode == MODE_INTRA_Q) {
                         {  
445                                  quant += dquant_table[BitstreamGetBits(bs,2)];                                  quant += dquant_table[BitstreamGetBits(bs,2)];
446                                  if (quant > 31)                                  if (quant > 31) {
                                 {  
447                                          quant = 31;                                          quant = 31;
448                                  }                                  } else if (quant < 1) {
                                 else if (quant < 1)  
                                 {  
449                                          quant = 1;                                          quant = 1;
450                                  }                                  }
451                          }                          }
452                          mb->quant = quant;                          mb->quant = quant;
453    
454                          if (dec->interlacing)                          if (dec->interlacing) {
                         {  
455                                  mb->field_dct = BitstreamGetBit(bs);                                  mb->field_dct = BitstreamGetBit(bs);
456                                  DEBUG1("deci: field_dct: ", mb->field_dct);                                  DEBUG1("deci: field_dct: ", mb->field_dct);
457                          }                          }
458    
459                          decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant, intra_dc_threshold);                          decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
460                                                            intra_dc_threshold);
461                  }                  }
462          }          }
463    
464  }  }
465    
466    
467  void get_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, int k, VECTOR * mv, int fcode)  void
468    get_motion_vector(DECODER * dec,
469                                      Bitstream * bs,
470                                      int x,
471                                      int y,
472                                      int k,
473                                      VECTOR * mv,
474                                      int fcode)
475  {  {
476    
477          int scale_fac = 1 << (fcode - 1);          int scale_fac = 1 << (fcode - 1);
# Line 493  Line 497 
497          mv_x += pmv_x;          mv_x += pmv_x;
498          mv_y += pmv_y;          mv_y += pmv_y;
499    
500          if (mv_x < low)          if (mv_x < low) {
         {  
501                  mv_x += range;                  mv_x += range;
502          }          } else if (mv_x > high) {
         else if (mv_x > high)  
         {  
503                  mv_x -= range;                  mv_x -= range;
504          }          }
505    
506          if (mv_y < low)          if (mv_y < low) {
         {  
507                  mv_y += range;                  mv_y += range;
508          }          } else if (mv_y > high) {
         else if (mv_y > high)  
         {  
509                  mv_y -= range;                  mv_y -= range;
510          }          }
511    
# Line 517  Line 515 
515  }  }
516    
517    
518  void decoder_pframe(DECODER * dec, Bitstream * bs, int rounding, int quant, int fcode, int intra_dc_threshold)  void
519    decoder_pframe(DECODER * dec,
520                               Bitstream * bs,
521                               int rounding,
522                               int quant,
523                               int fcode,
524                               int intra_dc_threshold)
525  {  {
526    
527          uint32_t x, y;          uint32_t x, y;
528    
529          start_timer();          start_timer();
530          image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing);          image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
531                                       dec->width, dec->height, dec->interlacing);
532          stop_edges_timer();          stop_edges_timer();
533    
534          for (y = 0; y < dec->mb_height; y++)          for (y = 0; y < dec->mb_height; y++) {
535          {                  for (x = 0; x < dec->mb_width; x++) {
                 for (x = 0; x < dec->mb_width; x++)  
                 {  
536                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
537    
538                          //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
# Line 549  Line 552 
552    
553                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);
554    
555                                  if (intra)                                  if (intra) {
                                 {  
556                                          acpred_flag = BitstreamGetBit(bs);                                          acpred_flag = BitstreamGetBit(bs);
557                                  }                                  }
558    
559                                  if (mb->mode == MODE_STUFFING)                                  if (mb->mode == MODE_STUFFING) {
                                 {  
560                                          DEBUG("-- STUFFING ?");                                          DEBUG("-- STUFFING ?");
561                                          continue;                                          continue;
562                                  }                                  }
# Line 563  Line 564 
564                                  cbpy = get_cbpy(bs, intra);                                  cbpy = get_cbpy(bs, intra);
565                                  cbp = (cbpy << 2) | cbpc;                                  cbp = (cbpy << 2) | cbpc;
566    
567                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q)                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q) {
                                 {  
568                                          quant += dquant_table[BitstreamGetBits(bs,2)];                                          quant += dquant_table[BitstreamGetBits(bs,2)];
569                                          if (quant > 31)                                          if (quant > 31) {
                                         {  
570                                                  quant = 31;                                                  quant = 31;
571                                          }                                          } else if (mb->quant < 1) {
                                         else if (mb->quant < 1)  
                                         {  
572                                                  quant = 1;                                                  quant = 1;
573                                          }                                          }
574                                  }                                  }
575                                  mb->quant = quant;                                  mb->quant = quant;
576    
577                                  if (dec->interlacing)                                  if (dec->interlacing) {
                                 {  
578                                          mb->field_dct = BitstreamGetBit(bs);                                          mb->field_dct = BitstreamGetBit(bs);
579                                          DEBUG1("decp: field_dct: ", mb->field_dct);                                          DEBUG1("decp: field_dct: ", mb->field_dct);
580    
581                                          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
                                         {  
582                                                  mb->field_pred = BitstreamGetBit(bs);                                                  mb->field_pred = BitstreamGetBit(bs);
583                                                  DEBUG1("decp: field_pred: ", mb->field_pred);                                                  DEBUG1("decp: field_pred: ", mb->field_pred);
584    
585                                                  if (mb->field_pred)                                                  if (mb->field_pred) {
                                                 {  
586                                                          mb->field_for_top = BitstreamGetBit(bs);                                                          mb->field_for_top = BitstreamGetBit(bs);
587                                                          DEBUG1("decp: field_for_top: ", mb->field_for_top);                                                          DEBUG1("decp: field_for_top: ", mb->field_for_top);
588                                                          mb->field_for_bot = BitstreamGetBit(bs);                                                          mb->field_for_bot = BitstreamGetBit(bs);
# Line 597  Line 591 
591                                          }                                          }
592                                  }                                  }
593    
594                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
595                                  {                                          if (dec->interlacing && mb->field_pred) {
596                                          if (dec->interlacing && mb->field_pred)                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
597                                          {                                                                                    fcode);
598                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1],
599                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1], fcode);                                                                                    fcode);
600                                          }                                          } else {
601                                          else                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
602                                          {                                                                                    fcode);
603                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);                                                  mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
604                                                  mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                          mb->mvs[0].x;
605                                                  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 =
606                                          }                                                          mb->mvs[0].y;
607                                  }                                  }
608                                  else if (mb->mode == MODE_INTER4V /* || mb->mode == MODE_INTER4V_Q */)                                  } else if (mb->mode ==
609                                  {                                                     MODE_INTER4V /* || mb->mode == MODE_INTER4V_Q */ ) {
610                                          get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);                                          get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);
611                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode);                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode);
612                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode);                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode);
613                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode);                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode);
614                                  }                                  } else                  // MODE_INTRA, MODE_INTRA_Q
                                 else  // MODE_INTRA, MODE_INTRA_Q  
615                                  {                                  {
616                                          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 =
617                                          mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;                                                  0;
618                                          decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant, intra_dc_threshold);                                          mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
619                                                    0;
620                                            decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
621                                                                            intra_dc_threshold);
622                                          continue;                                          continue;
623                                  }                                  }
624    
625                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant, rounding);                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant,
626                          }                                                                  rounding);
627                          else    // not coded                          } else                          // not coded
628                          {                          {
629                                  //DEBUG2("P-frame MB at (X,Y)=",x,y);                                  //DEBUG2("P-frame MB at (X,Y)=",x,y);
630                                  mb->mode = MODE_NOT_CODED;                                  mb->mode = MODE_NOT_CODED;
# Line 639  Line 635 
635    
636                                  start_timer();                                  start_timer();
637    
638                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),                                  transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
639                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x),                                                                   (16 * x),
640                                                   dec->edged_width);                                                                   dec->refn[0].y + (16 * y) * dec->edged_width +
641                                                                     (16 * x), dec->edged_width);
642                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x+8),  
643                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x+8),                                  transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
644                                                   dec->edged_width);                                                                   (16 * x + 8),
645                                                                     dec->refn[0].y + (16 * y) * dec->edged_width +
646                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x),                                                                   (16 * x + 8), dec->edged_width);
647                                                   dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x),  
648                                                   dec->edged_width);                                  transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
649                                                                     (16 * x),
650                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x+8),                                                                   dec->refn[0].y + (16 * y +
651                                                   dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x+8),                                                                                                     8) * dec->edged_width +
652                                                   dec->edged_width);                                                                   (16 * x), dec->edged_width);
653    
654                                  transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),                                  transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
655                                                   dec->refn[0].u + (8*y)*dec->edged_width/2 + (8*x),                                                                   (16 * x + 8),
656                                                                     dec->refn[0].y + (16 * y +
657                                                                                                       8) * dec->edged_width +
658                                                                     (16 * x + 8), dec->edged_width);
659    
660                                    transfer8x8_copy(dec->cur.u + (8 * y) * dec->edged_width / 2 +
661                                                                     (8 * x),
662                                                                     dec->refn[0].u +
663                                                                     (8 * y) * dec->edged_width / 2 + (8 * x),
664                                                   dec->edged_width/2);                                                   dec->edged_width/2);
665    
666                                  transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),                                  transfer8x8_copy(dec->cur.v + (8 * y) * dec->edged_width / 2 +
667                                                   dec->refn[0].v + (8*y)*dec->edged_width/2 + (8*x),                                                                   (8 * x),
668                                                                     dec->refn[0].v +
669                                                                     (8 * y) * dec->edged_width / 2 + (8 * x),
670                                                   dec->edged_width/2);                                                   dec->edged_width/2);
671    
672                                  stop_transfer_timer();                                  stop_transfer_timer();
# Line 672  Line 678 
678    
679  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
680  // decode B-frame motion vector  // decode B-frame motion vector
681  void get_b_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, VECTOR * mv, int fcode, const VECTOR pmv)  void
682    get_b_motion_vector(DECODER * dec,
683                                            Bitstream * bs,
684                                            int x,
685                                            int y,
686                                            VECTOR * mv,
687                                            int fcode,
688                                            const VECTOR pmv)
689  {  {
690          int scale_fac = 1 << (fcode - 1);          int scale_fac = 1 << (fcode - 1);
691          int high = (32 * scale_fac) - 1;          int high = (32 * scale_fac) - 1;
# Line 691  Line 704 
704          mv_x += pmv_x;          mv_x += pmv_x;
705          mv_y += pmv_y;          mv_y += pmv_y;
706    
707          if (mv_x < low)          if (mv_x < low) {
         {  
708                  mv_x += range;                  mv_x += range;
709          }          } else if (mv_x > high) {
         else if (mv_x > high)  
         {  
710                  mv_x -= range;                  mv_x -= range;
711          }          }
712    
713          if (mv_y < low)          if (mv_y < low) {
         {  
714                  mv_y += range;                  mv_y += range;
715          }          } else if (mv_y > high) {
         else if (mv_y > high)  
         {  
716                  mv_y -= range;                  mv_y -= range;
717          }          }
718    
# Line 716  Line 723 
723    
724  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
725  // decode an B-frame forward & backward inter macroblock  // decode an B-frame forward & backward inter macroblock
726  void decoder_bf_mbinter(DECODER * dec,  void
727    decoder_bf_mbinter(DECODER * dec,
728                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
729                       const uint32_t x_pos,                       const uint32_t x_pos,
730                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 741  Line 749 
749          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
750          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
751    
752          if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))          if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
         {  
753                  uv_dx = pMB->mvs[0].x;                  uv_dx = pMB->mvs[0].x;
754                  uv_dy = pMB->mvs[0].y;                  uv_dy = pMB->mvs[0].y;
755    
756                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
757                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
758          }          } else {
         else  
         {  
759                  int sum;                  int sum;
760    
761                  sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;                  sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
762                  uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dx =
763                            (sum ==
764                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
765                                                                      (ABS(sum) / 16) * 2));
766    
767                  sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;                  sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
768                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dy =
769                            (sum ==
770                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
771                                                                      (ABS(sum) / 16) * 2));
772          }          }
773    
774          start_timer();          start_timer();
775          interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos,     16*y_pos    , pMB->mvs[0].x, pMB->mvs[0].y, stride,  0);          interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos, 16 * y_pos,
776          interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos + 8, 16*y_pos    , pMB->mvs[1].x, pMB->mvs[1].y, stride,  0);                                                    pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
777          interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos,     16*y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,  0);          interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos + 8,
778          interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos + 8, 16*y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,  0);                                                    16 * y_pos, pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
779          interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);          interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos,
780          interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);                                                    16 * y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,
781                                                      0);
782            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos + 8,
783                                                      16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
784                                                      0);
785            interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8 * x_pos, 8 * y_pos,
786                                                      uv_dx, uv_dy, stride2, 0);
787            interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8 * x_pos, 8 * y_pos,
788                                                      uv_dx, uv_dy, stride2, 0);
789          stop_comp_timer();          stop_comp_timer();
790    
791          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
         {  
792                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
793                  {                  {
794                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear
# Line 779  Line 798 
798                          stop_coding_timer();                          stop_coding_timer();
799    
800                          start_timer();                          start_timer();
801                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
                         {  
802                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                                  dequant_inter(&data[i*64], &block[i*64], iQuant);
803                          }                          } else {
                         else  
                         {  
804                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);
805                          }                          }
806                          stop_iquant_timer();                          stop_iquant_timer();
# Line 795  Line 811 
811                  }                  }
812          }          }
813    
814          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
815                  next_block = stride;                  next_block = stride;
816                  stride *= 2;                  stride *= 2;
817          }          }
# Line 820  Line 835 
835    
836  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
837  // decode an B-frame direct &  inter macroblock  // decode an B-frame direct &  inter macroblock
838  void decoder_bf_interpolate_mbinter(DECODER * dec,  void
839                           IMAGE forward, IMAGE backward,  decoder_bf_interpolate_mbinter(DECODER * dec,
840                                                               IMAGE forward,
841                                                               IMAGE backward,
842                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
843                       const uint32_t x_pos,                       const uint32_t x_pos,
844                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 845  Line 862 
862          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
863          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
864    
865          if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))          if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
         {  
866                  uv_dx = pMB->mvs[0].x;                  uv_dx = pMB->mvs[0].x;
867                  uv_dy = pMB->mvs[0].y;                  uv_dy = pMB->mvs[0].y;
868    
# Line 858  Line 874 
874    
875                  b_uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;                  b_uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
876                  b_uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;                  b_uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
877          }          } else {
         else  
         {  
878                  int sum;                  int sum;
879    
880                  sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;                  sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
881                  uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dx =
882                            (sum ==
883                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
884                                                                      (ABS(sum) / 16) * 2));
885    
886                  sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;                  sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
887                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dy =
888                            (sum ==
889                  sum = pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x + pMB->b_mvs[3].x;                           0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
890                  b_uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                                                                    (ABS(sum) / 16) * 2));
891    
892                  sum = pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y + pMB->b_mvs[3].y;                  sum =
893                  b_uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                          pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x +
894          }                          pMB->b_mvs[3].x;
895                    b_uv_dx =
896                            (sum ==
897          start_timer();                           0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
898          interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos,     16*y_pos    , pMB->mvs[0].x, pMB->mvs[0].y, stride,  0);                                                                    (ABS(sum) / 16) * 2));
899          interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos + 8, 16*y_pos    , pMB->mvs[1].x, pMB->mvs[1].y, stride,  0);  
900          interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos,     16*y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,  0);                  sum =
901          interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos + 8, 16*y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,  0);                          pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y +
902          interpolate8x8_switch(dec->cur.u, forward.u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);                          pMB->b_mvs[3].y;
903          interpolate8x8_switch(dec->cur.v, forward.v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);                  b_uv_dy =
904                            (sum ==
905                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
906          interpolate8x8_switch(dec->refn[2].y, backward.y, 16*x_pos,     16*y_pos    , pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride,  0);                                                                    (ABS(sum) / 16) * 2));
907          interpolate8x8_switch(dec->refn[2].y, backward.y, 16*x_pos + 8, 16*y_pos    , pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride,  0);          }
908          interpolate8x8_switch(dec->refn[2].y, backward.y, 16*x_pos,     16*y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y, stride,  0);  
909          interpolate8x8_switch(dec->refn[2].y, backward.y, 16*x_pos + 8, 16*y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y, stride,  0);  
910          interpolate8x8_switch(dec->refn[2].u, backward.u, 8*x_pos,      8*y_pos,      b_uv_dx,         b_uv_dy,         stride2, 0);          start_timer();
911          interpolate8x8_switch(dec->refn[2].v, backward.v, 8*x_pos,      8*y_pos,      b_uv_dx,         b_uv_dy,         stride2, 0);          interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos,
912                                                      pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
913          interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos,     16*y_pos    , stride);          interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8, 16 * y_pos,
914          interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos + 8, 16*y_pos    , stride);                                                    pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
915          interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos,     16*y_pos + 8, stride);          interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos + 8,
916          interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos + 8, 16*y_pos + 8, stride);                                                    pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
917          interpolate8x8_c(dec->cur.u, dec->refn[2].u, 8*x_pos,      8*y_pos,      stride2);          interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8,
918          interpolate8x8_c(dec->cur.v, dec->refn[2].v, 8*x_pos,      8*y_pos,      stride2);                                                    16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
919                                                      0);
920            interpolate8x8_switch(dec->cur.u, forward.u, 8 * x_pos, 8 * y_pos, uv_dx,
921                                                      uv_dy, stride2, 0);
922            interpolate8x8_switch(dec->cur.v, forward.v, 8 * x_pos, 8 * y_pos, uv_dx,
923                                                      uv_dy, stride2, 0);
924    
925    
926            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos, 16 * y_pos,
927                                                      pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
928            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
929                                                      16 * y_pos, pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride,
930                                                      0);
931            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos,
932                                                      16 * y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y,
933                                                      stride, 0);
934            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
935                                                      16 * y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y,
936                                                      stride, 0);
937            interpolate8x8_switch(dec->refn[2].u, backward.u, 8 * x_pos, 8 * y_pos,
938                                                      b_uv_dx, b_uv_dy, stride2, 0);
939            interpolate8x8_switch(dec->refn[2].v, backward.v, 8 * x_pos, 8 * y_pos,
940                                                      b_uv_dx, b_uv_dy, stride2, 0);
941    
942            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos, 16 * y_pos,
943                                             stride);
944            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos + 8, 16 * y_pos,
945                                             stride);
946            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos, 16 * y_pos + 8,
947                                             stride);
948            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos + 8,
949                                             16 * y_pos + 8, stride);
950            interpolate8x8_c(dec->cur.u, dec->refn[2].u, 8 * x_pos, 8 * y_pos,
951                                             stride2);
952            interpolate8x8_c(dec->cur.v, dec->refn[2].v, 8 * x_pos, 8 * y_pos,
953                                             stride2);
954    
955          stop_comp_timer();          stop_comp_timer();
956    
957          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
         {  
958                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
959                  {                  {
960                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear
# Line 912  Line 964 
964                          stop_coding_timer();                          stop_coding_timer();
965    
966                          start_timer();                          start_timer();
967                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
                         {  
968                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                                  dequant_inter(&data[i*64], &block[i*64], iQuant);
969                          }                          } else {
                         else  
                         {  
970                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);
971                          }                          }
972                          stop_iquant_timer();                          stop_iquant_timer();
# Line 928  Line 977 
977                  }                  }
978          }          }
979    
980          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
981                  next_block = stride;                  next_block = stride;
982                  stride *= 2;                  stride *= 2;
983          }          }
# Line 953  Line 1001 
1001    
1002  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
1003  // for decode B-frame dbquant  // for decode B-frame dbquant
1004  int32_t __inline get_dbquant(Bitstream * bs)  int32_t __inline
1005    get_dbquant(Bitstream * bs)
1006  {  {
1007          if (!BitstreamGetBit(bs))               // '0'          if (!BitstreamGetBit(bs))               // '0'
1008                  return(0);                  return(0);
# Line 970  Line 1019 
1019  // 01       1  // 01       1
1020  // 001      2  // 001      2
1021  // 0001     3  // 0001     3
1022  int32_t __inline get_mbtype(Bitstream * bs)  int32_t __inline
1023    get_mbtype(Bitstream * bs)
1024  {  {
1025          int32_t mb_type;          int32_t mb_type;
1026    
1027          for(mb_type=0;mb_type<=3;mb_type++){          for(mb_type=0;mb_type<=3;mb_type++){
1028                  if  (BitstreamGetBit(bs))                  if  (BitstreamGetBit(bs))
1029                          break;                          break;
# Line 984  Line 1035 
1035                  return(-1);                  return(-1);
1036  }  }
1037    
1038  void decoder_bframe(DECODER * dec, Bitstream * bs, int quant, int fcode_forward, int fcode_backward)  void
1039    decoder_bframe(DECODER * dec,
1040                               Bitstream * bs,
1041                               int quant,
1042                               int fcode_forward,
1043                               int fcode_backward)
1044  {  {
1045    
1046          uint32_t        x, y;          uint32_t        x, y;
1047          VECTOR          mv, zeromv;          VECTOR          mv, zeromv;
1048    
1049          start_timer();          start_timer();
1050          image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing);          image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
1051                                       dec->width, dec->height, dec->interlacing);
1052          //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, dec->width, dec->height, dec->interlacing);
1053          stop_edges_timer();          stop_edges_timer();
1054    
1055    
1056          for (y = 0; y < dec->mb_height; y++)          for (y = 0; y < dec->mb_height; y++) {
         {  
1057                  // Initialize Pred Motion Vector                  // Initialize Pred Motion Vector
1058                  dec->p_fmv.x = dec->p_fmv.y = dec->p_bmv.x = dec->p_bmv.y = 0;                  dec->p_fmv.x = dec->p_fmv.y = dec->p_bmv.x = dec->p_bmv.y = 0;
1059                  for (x = 0; x < dec->mb_width; x++)                  for (x = 0; x < dec->mb_width; x++) {
                 {  
1060                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
1061                          MACROBLOCK * last_mb = &dec->last_mbs[y*dec->mb_width + x];                          MACROBLOCK * last_mb = &dec->last_mbs[y*dec->mb_width + x];
1062    
1063                          mb->mvs[0].x=mb->mvs[0].y=zeromv.x = zeromv.y = mv.x = mv.y = 0;                          mb->mvs[0].x = mb->mvs[0].y = zeromv.x = zeromv.y = mv.x = mv.y =
1064                                    0;
1065    
1066                          // the last P_VOP is skip macroblock ?                          // the last P_VOP is skip macroblock ?
1067                          if (last_mb->mode == MODE_NOT_CODED){                          if (last_mb->mode == MODE_NOT_CODED){
# Line 1019  Line 1075 
1075                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1076                                  continue;                                  continue;
1077                          }                          }
   
1078                          //t=BitstreamShowBits(bs,32);                          //t=BitstreamShowBits(bs,32);
1079    
1080                          if (!BitstreamGetBit(bs)){      // modb=='0'                          if (!BitstreamGetBit(bs)){      // modb=='0'
# Line 1035  Line 1090 
1090                                  if (mb->mb_type && mb->cbp){                                  if (mb->mb_type && mb->cbp){
1091                                          quant += get_dbquant(bs);                                          quant += get_dbquant(bs);
1092    
1093                                          if (quant > 31)                                          if (quant > 31) {
                                         {  
1094                                                  quant = 31;                                                  quant = 31;
1095                                          }                                          } else if (mb->quant < 1) {
                                         else if (mb->quant < 1)  
                                         {  
1096                                                  quant = 1;                                                  quant = 1;
1097                                          }                                          }
1098                                  } else {                                  } else {
# Line 1055  Line 1107 
1107                          mb->mode = MODE_INTER;                          mb->mode = MODE_INTER;
1108                          //DEBUG1("Switch bm_type=",mb->mb_type);                          //DEBUG1("Switch bm_type=",mb->mb_type);
1109    
1110                          switch(mb->mb_type)                          switch (mb->mb_type) {
                         {  
1111                          case MODE_DIRECT:                          case MODE_DIRECT:
1112                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], 1, zeromv);                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], 1, zeromv);
1113    
1114                          case MODE_DIRECT_NONE_MV:                          case MODE_DIRECT_NONE_MV:
1115                                  {       // 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
1116                                          const int64_t   TRB=dec->time_pp-dec->time_bp,                                          const int64_t TRB = dec->time_pp - dec->time_bp, TRD =
1117                                                                          TRD=dec->time_pp;                                                  dec->time_pp;
1118                                          int i;                                          int i;
1119    
1120                                          for(i=0;i<4;i++){                                          for(i=0;i<4;i++){
1121                                                  mb->mvs[i].x    = (int32_t)((TRB * last_mb->mvs[i].x)/TRD+mb->mvs[0].x);                                                  mb->mvs[i].x =
1122                                                  mb->b_mvs[i].x  = (int32_t)((mb->mvs[0].x==0)?((TRB-TRD)*last_mb->mvs[i].x)/TRD:mb->mvs[i].x-last_mb->mvs[i].x);                                                          (int32_t) ((TRB * last_mb->mvs[i].x) / TRD +
1123                                                  mb->mvs[i].y    = (int32_t)((TRB * last_mb->mvs[i].y)/TRD+mb->mvs[0].y);                                                                             mb->mvs[0].x);
1124                                                  mb->b_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);                                                  mb->b_mvs[i].x =
1125                                                            (int32_t) ((mb->mvs[0].x ==
1126                                                                                    0) ? ((TRB -
1127                                                                                               TRD) * last_mb->mvs[i].x) /
1128                                                                               TRD : mb->mvs[i].x - last_mb->mvs[i].x);
1129                                                    mb->mvs[i].y =
1130                                                            (int32_t) ((TRB * last_mb->mvs[i].y) / TRD +
1131                                                                               mb->mvs[0].y);
1132                                                    mb->b_mvs[i].y =
1133                                                            (int32_t) ((mb->mvs[0].y ==
1134                                                                                    0) ? ((TRB -
1135                                                                                               TRD) * last_mb->mvs[i].y) /
1136                                                                               TRD : mb->mvs[i].y - last_mb->mvs[i].y);
1137                                          }                                          }
1138                                          //DEBUG("B-frame Direct!\n");                                          //DEBUG("B-frame Direct!\n");
1139                                  }                                  }
1140                                  mb->mode = MODE_INTER4V;                                  mb->mode = MODE_INTER4V;
1141                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], mb, x, y, mb->cbp, bs);                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1142                                                                                               mb, x, y, mb->cbp, bs);
1143                                  break;                                  break;
1144    
1145                          case MODE_INTERPOLATE:                          case MODE_INTERPOLATE:
1146                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward, dec->p_fmv);                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1147                                  dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                                          dec->p_fmv);
1148                                  dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                  dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
1149                                            mb->mvs[0].x;
1150                                  get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0], fcode_backward, dec->p_bmv);                                  dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
1151                                  dec->p_bmv.x = mb->b_mvs[1].x = mb->b_mvs[2].x = mb->b_mvs[3].x = mb->b_mvs[0].x;                                          mb->mvs[0].y;
1152                                  dec->p_bmv.y = mb->b_mvs[1].y = mb->b_mvs[2].y = mb->b_mvs[3].y = mb->b_mvs[0].y;  
1153                                    get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0],
1154                                                                            fcode_backward, dec->p_bmv);
1155                                    dec->p_bmv.x = mb->b_mvs[1].x = mb->b_mvs[2].x =
1156                                            mb->b_mvs[3].x = mb->b_mvs[0].x;
1157                                    dec->p_bmv.y = mb->b_mvs[1].y = mb->b_mvs[2].y =
1158                                            mb->b_mvs[3].y = mb->b_mvs[0].y;
1159    
1160                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], mb, x, y, mb->cbp, bs);                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1161                                                                                               mb, x, y, mb->cbp, bs);
1162                                  //DEBUG("B-frame Bidir!\n");                                  //DEBUG("B-frame Bidir!\n");
1163                                  break;                                  break;
1164    
1165                          case MODE_BACKWARD:                          case MODE_BACKWARD:
1166                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_backward, dec->p_bmv);                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_backward,
1167                                  dec->p_bmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                                          dec->p_bmv);
1168                                  dec->p_bmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                  dec->p_bmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
1169                                            mb->mvs[0].x;
1170                                    dec->p_bmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
1171                                            mb->mvs[0].y;
1172    
1173                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);
1174                                  //DEBUG("B-frame Backward!\n");                                  //DEBUG("B-frame Backward!\n");
1175                                  break;                                  break;
1176    
1177                          case MODE_FORWARD:                          case MODE_FORWARD:
1178                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward, dec->p_fmv);                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1179                                  dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                                          dec->p_fmv);
1180                                  dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                  dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
1181                                            mb->mvs[0].x;
1182                                    dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
1183                                            mb->mvs[0].y;
1184    
1185                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1186                                  //DEBUG("B-frame Forward!\n");                                  //DEBUG("B-frame Forward!\n");
# Line 1117  Line 1195 
1195  }  }
1196    
1197  // swap two MACROBLOCK array  // swap two MACROBLOCK array
1198  void mb_swap(MACROBLOCK **mb1, MACROBLOCK **mb2)  void
1199    mb_swap(MACROBLOCK ** mb1,
1200                    MACROBLOCK ** mb2)
1201  {  {
1202          MACROBLOCK *temp=*mb1;          MACROBLOCK *temp=*mb1;
1203    
1204          *mb1=*mb2;          *mb1=*mb2;
1205          *mb2=temp;          *mb2=temp;
1206  }  }
1207    
1208  int decoder_decode(DECODER * dec, XVID_DEC_FRAME * frame)  int
1209    decoder_decode(DECODER * dec,
1210                               XVID_DEC_FRAME * frame)
1211  {  {
1212    
1213          Bitstream bs;          Bitstream bs;
# Line 1142  Line 1225 
1225          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
1226          // for support B-frame to reference last 2 frame          // for support B-frame to reference last 2 frame
1227          dec->frames++;          dec->frames++;
1228          vop_type=BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward, &fcode_backward, &intra_dc_threshold);          vop_type =
1229                    BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward,
1230                                                             &fcode_backward, &intra_dc_threshold);
1231    
1232          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
1233    
1234          switch (vop_type)          switch (vop_type) {
         {  
1235          case P_VOP :          case P_VOP :
1236                  decoder_pframe(dec, &bs, rounding, quant, fcode_forward, intra_dc_threshold);                  decoder_pframe(dec, &bs, rounding, quant, fcode_forward,
1237                                               intra_dc_threshold);
1238                  DEBUG1("P_VOP  Time=",dec->time);                  DEBUG1("P_VOP  Time=",dec->time);
1239                  break;                  break;
1240    
# Line 1188  Line 1273 
1273          } else {          } else {
1274                  if (dec->frames >= 1){                  if (dec->frames >= 1){
1275                          start_timer();                          start_timer();
1276                          if ((vop_type == I_VOP || vop_type == P_VOP))                          if ((vop_type == I_VOP || vop_type == P_VOP)) {
1277                          {                                  image_output(&dec->refn[0], dec->width, dec->height,
1278                                  image_output(&dec->refn[0], dec->width, dec->height, dec->edged_width,                                                           dec->edged_width, frame->image, frame->stride,
1279                                                   frame->image, frame->stride, frame->colorspace);                                                           frame->colorspace);
1280                          } else if (vop_type == B_VOP) {                          } else if (vop_type == B_VOP) {
1281                                  image_output(&dec->cur, dec->width, dec->height, dec->edged_width,                                  image_output(&dec->cur, dec->width, dec->height,
1282                                                   frame->image, frame->stride, frame->colorspace);                                                           dec->edged_width, frame->image, frame->stride,
1283                                                             frame->colorspace);
1284                          }                          }
1285                          stop_conv_timer();                          stop_conv_timer();
1286                  }                  }

Legend:
Removed from v.179  
changed lines
  Added in v.195

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