[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.13, Sun Apr 28 23:35:25 2002 UTC revision 1.37, Wed Sep 4 03:23:23 2002 UTC
# Line 32  Line 32 
32   *   *
33   *  History:   *  History:
34   *   *
35     *  15.07.2002  fix a bug in B-frame decode at DIRECT mode
36     *              MinChen <chenm001@163.com>
37     *  10.07.2002  added BFRAMES_DEC_DEBUG support
38     *              Fix a little bug for low_delay flage
39     *              MinChen <chenm001@163.com>
40     *  28.06.2002  added basic resync support to iframe/pframe_decode()
41     *  22.06.2002  added primative N_VOP support
42     *                              #define BFRAMES_DEC now enables Minchen's bframe decoder
43     *  08.05.2002  add low_delay support for B_VOP decode
44     *              MinChen <chenm001@163.com>
45     *  05.05.2002  fix some B-frame decode problem
46     *  02.05.2002  add B-frame decode support(have some problem);
47     *              MinChen <chenm001@163.com>
48   *  22.04.2002  add some B-frame decode support;  chenm001 <chenm001@163.com>   *  22.04.2002  add some B-frame decode support;  chenm001 <chenm001@163.com>
49   *  29.03.2002  interlacing fix - compensated block wasn't being used when   *  29.03.2002  interlacing fix - compensated block wasn't being used when
50   *              reconstructing blocks, thus artifacts   *              reconstructing blocks, thus artifacts
# Line 49  Line 62 
62  #include <stdlib.h>  #include <stdlib.h>
63  #include <string.h>  #include <string.h>
64    
65    #ifdef BFRAMES_DEC_DEBUG
66            #define BFRAMES_DEC
67    #endif
68    
69  #include "xvid.h"  #include "xvid.h"
70  #include "portab.h"  #include "portab.h"
71    
# Line 72  Line 89 
89  #include "image/colorspace.h"  #include "image/colorspace.h"
90  #include "utils/mem_align.h"  #include "utils/mem_align.h"
91    
92  int decoder_create(XVID_DEC_PARAM * param)  int
93    decoder_create(XVID_DEC_PARAM * param)
94  {  {
95          DECODER * dec;          DECODER * dec;
96    
97          dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);          dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);
98          if (dec == NULL)          if (dec == NULL) {
         {  
99                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
100          }          }
101          param->handle = dec;          param->handle = dec;
# Line 91  Line 108 
108    
109          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;
110          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;
111            dec->low_delay = 0;
112    
113          if (image_create(&dec->cur, dec->edged_width, dec->edged_height))          if (image_create(&dec->cur, dec->edged_width, dec->edged_height)) {
         {  
114                  xvid_free(dec);                  xvid_free(dec);
115                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
116          }          }
117    
118          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[0], 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                  xvid_free(dec);                  xvid_free(dec);
121                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
122          }          }
123          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
124          // for support B-frame to reference last 2 frame          // for support B-frame to reference last 2 frame
125          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {
         {  
126                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
127                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
128                  xvid_free(dec);                  xvid_free(dec);
129                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
130          }          }
131            if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height)) {
132                    image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
133                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
134                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
135                    xvid_free(dec);
136                    return XVID_ERR_MEMORY;
137            }
138    
139          dec->mbs = xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height, CACHE_LINE);          dec->mbs =
140          if (dec->mbs == NULL)                  xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
141          {                                          CACHE_LINE);
142            if (dec->mbs == NULL) {
143                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
144                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
145                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
146                    image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
147                  xvid_free(dec);                  xvid_free(dec);
148                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
149          }          }
150    
151            memset(dec->mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
152    
153            // add by chenm001 <chenm001@163.com>
154            // for skip MB flag
155            dec->last_mbs =
156                    xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
157                                            CACHE_LINE);
158            if (dec->last_mbs == NULL) {
159                    xvid_free(dec->mbs);
160                    image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
161                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
162                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
163                    image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
164                    xvid_free(dec);
165                    return XVID_ERR_MEMORY;
166            }
167    
168            memset(dec->last_mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
169    
170          init_timer();          init_timer();
171    
172            // add by chenm001 <chenm001@163.com>
173            // for support B-frame to save reference frame's time
174            dec->frames = -1;
175            dec->time = dec->time_base = dec->last_time_base = 0;
176    
177          return XVID_ERR_OK;          return XVID_ERR_OK;
178  }  }
179    
180    
181  int decoder_destroy(DECODER * dec)  int
182    decoder_destroy(DECODER * dec)
183  {  {
184            xvid_free(dec->last_mbs);
185          xvid_free(dec->mbs);          xvid_free(dec->mbs);
186          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
187            image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
188            image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
189          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
190          xvid_free(dec);          xvid_free(dec);
191    
# Line 141  Line 195 
195    
196    
197    
198  static const int32_t dquant_table[4] =  static const int32_t dquant_table[4] = {
 {  
199          -1, -2, 1, 2          -1, -2, 1, 2
200  };  };
201    
202    
203    
204    
205  // decode an intra macroblock  // decode an intra macroblock
206    
207  void decoder_mbintra(DECODER * dec,  void
208    decoder_mbintra(DECODER * dec,
209                       MACROBLOCK * pMB,                       MACROBLOCK * pMB,
210                       const uint32_t x_pos,                       const uint32_t x_pos,
211                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 157  Line 213 
213                       const uint32_t cbp,                       const uint32_t cbp,
214                       Bitstream * bs,                       Bitstream * bs,
215                       const uint32_t quant,                       const uint32_t quant,
216                       const uint32_t intra_dc_threshold)                                  const uint32_t intra_dc_threshold,
217                                    const unsigned int bound)
218  {  {
219    
220          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
# Line 176  Line 233 
233    
234          memset(block, 0, 6*64*sizeof(int16_t));         // clear          memset(block, 0, 6*64*sizeof(int16_t));         // clear
235    
236          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
         {  
237                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
238                  int16_t predictors[8];                  int16_t predictors[8];
239                  int start_coeff;                  int start_coeff;
240    
241                  start_timer();                  start_timer();
242                  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],
243                  if (!acpred_flag)                                           iQuant, iDcScaler, predictors, bound);
244                  {                  if (!acpred_flag) {
245                          pMB->acpred_directions[i] = 0;                          pMB->acpred_directions[i] = 0;
246                  }                  }
247                  stop_prediction_timer();                  stop_prediction_timer();
248    
249                  if (quant < intra_dc_threshold)                  if (quant < intra_dc_threshold) {
                 {  
250                          int dc_size;                          int dc_size;
251                          int dc_dif;                          int dc_dif;
252    
253                          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);
254                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;
255    
256                          if (dc_size > 8)                          if (dc_size > 8) {
                         {  
257                                  BitstreamSkip(bs, 1);           // marker                                  BitstreamSkip(bs, 1);           // marker
258                          }                          }
259    
260                          block[i*64 + 0] = dc_dif;                          block[i*64 + 0] = dc_dif;
261                          start_coeff = 1;                          start_coeff = 1;
262                  }  
263                  else                          DPRINTF(DPRINTF_COEFF,"block[0] %i", dc_dif);
264                  {                  } else {
265                          start_coeff = 0;                          start_coeff = 0;
266                  }                  }
267    
268                  start_timer();                  start_timer();
269                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
270                  {                  {
271                          get_intra_block(bs, &block[i*64], pMB->acpred_directions[i], start_coeff);                          get_intra_block(bs, &block[i * 64], pMB->acpred_directions[i],
272                                                            start_coeff);
273                  }                  }
274                  stop_coding_timer();                  stop_coding_timer();
275    
# Line 223  Line 278 
278                  stop_prediction_timer();                  stop_prediction_timer();
279    
280                  start_timer();                  start_timer();
281                  if (dec->quant_type == 0)                  if (dec->quant_type == 0) {
                 {  
282                          dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);                          dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
283                  }                  } else {
                 else  
                 {  
284                          dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);                          dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
285                  }                  }
286                  stop_iquant_timer();                  stop_iquant_timer();
# Line 238  Line 290 
290                  stop_idct_timer();                  stop_idct_timer();
291          }          }
292    
293          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
294                  next_block = stride;                  next_block = stride;
295                  stride *= 2;                  stride *= 2;
296          }          }
# Line 266  Line 317 
317    
318  // decode an inter macroblock  // decode an inter macroblock
319    
320  void decoder_mbinter(DECODER * dec,  void
321    decoder_mbinter(DECODER * dec,
322                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
323                       const uint32_t x_pos,                       const uint32_t x_pos,
324                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 292  Line 344 
344          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
345          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
346    
347          if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)          if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
         {  
348                  uv_dx = pMB->mvs[0].x;                  uv_dx = pMB->mvs[0].x;
349                  uv_dy = pMB->mvs[0].y;                  uv_dy = pMB->mvs[0].y;
350    
351                    if (dec->quarterpel)
352                    {
353                            uv_dx = (uv_dx >> 1) | (uv_dx & 1);
354                            uv_dy = (uv_dy >> 1) | (uv_dy & 1);
355                    }
356    
357                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
358                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
359          }          } else {
         else  
         {  
360                  int sum;                  int sum;
361                  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;
362    
363                    if (dec->quarterpel)
364                    {
365                            sum /= 2;
366                    }
367    
368                  uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
369    
370                  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;
371    
372                    if (dec->quarterpel)
373                    {
374                            sum /= 2;
375                    }
376    
377                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
378          }          }
379    
380          start_timer();          start_timer();
381          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);          if(dec->quarterpel) {
382          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);                  DEBUG("QUARTERPEL");
383          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_quarterpel(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
384          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);                                                                    pMB->mvs[0].x, pMB->mvs[0].y, stride,  rounding);
385          interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, rounding);                  interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos,
386          interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, rounding);                                                                    pMB->mvs[1].x, pMB->mvs[1].y, stride,  rounding);
387                    interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos + 8,
388                                                                      pMB->mvs[2].x, pMB->mvs[2].y, stride,  rounding);
389                    interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8,
390                                                                      pMB->mvs[3].x, pMB->mvs[3].y, stride,  rounding);
391            }
392            else {
393                    interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
394                                                              pMB->mvs[0].x, pMB->mvs[0].y, stride,  rounding);
395                    interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos,
396                                                              pMB->mvs[1].x, pMB->mvs[1].y, stride,  rounding);
397                    interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos + 8,
398                                                              pMB->mvs[2].x, pMB->mvs[2].y, stride,  rounding);
399                    interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8,
400                                                              pMB->mvs[3].x, pMB->mvs[3].y, stride,  rounding);
401            }
402    
403            interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
404                                                      uv_dx, uv_dy, stride2, rounding);
405            interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
406                                                      uv_dx, uv_dy, stride2, rounding);
407          stop_comp_timer();          stop_comp_timer();
408    
409          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
         {  
410                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
411                  {                  {
412                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear
# Line 330  Line 416 
416                          stop_coding_timer();                          stop_coding_timer();
417    
418                          start_timer();                          start_timer();
419                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
                         {  
420                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                                  dequant_inter(&data[i*64], &block[i*64], iQuant);
421                          }                          } else {
                         else  
                         {  
422                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);
423                          }                          }
424                          stop_iquant_timer();                          stop_iquant_timer();
# Line 346  Line 429 
429                  }                  }
430          }          }
431    
432          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
433                  next_block = stride;                  next_block = stride;
434                  stride *= 2;                  stride *= 2;
435          }          }
# Line 369  Line 451 
451  }  }
452    
453    
454  void decoder_iframe(DECODER * dec, Bitstream * bs, int quant, int intra_dc_threshold)  void
455    decoder_iframe(DECODER * dec,
456                               Bitstream * bs,
457                               int quant,
458                               int intra_dc_threshold)
459  {  {
460            uint32_t bound;
461          uint32_t x, y;          uint32_t x, y;
462    
463          for (y = 0; y < dec->mb_height; y++)          bound = 0;
         {  
                 for (x = 0; x < dec->mb_width; x++)  
                 {  
                         MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];  
464    
465            for (y = 0; y < dec->mb_height; y++) {
466                    for (x = 0; x < dec->mb_width; x++) {
467                            MACROBLOCK *mb;
468                          uint32_t mcbpc;                          uint32_t mcbpc;
469                          uint32_t cbpc;                          uint32_t cbpc;
470                          uint32_t acpred_flag;                          uint32_t acpred_flag;
471                          uint32_t cbpy;                          uint32_t cbpy;
472                          uint32_t cbp;                          uint32_t cbp;
473    
474                            while (BitstreamShowBits(bs, 9) == 1)
475                                    BitstreamSkip(bs, 9);
476    
477                            if (check_resync_marker(bs, 0))
478                            {
479                                    bound = read_video_packet_header(bs, 0, &quant);
480                                    x = bound % dec->mb_width;
481                                    y = bound / dec->mb_width;
482                            }
483                            mb = &dec->mbs[y * dec->mb_width + x];
484    
485                            DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
486    
487                          mcbpc = get_mcbpc_intra(bs);                          mcbpc = get_mcbpc_intra(bs);
488                          mb->mode = mcbpc & 7;                          mb->mode = mcbpc & 7;
489                          cbpc = (mcbpc >> 4);                          cbpc = (mcbpc >> 4);
490    
491                          acpred_flag = BitstreamGetBit(bs);                          acpred_flag = BitstreamGetBit(bs);
492    
                         if (mb->mode == MODE_STUFFING)  
                         {  
                                 DEBUG("-- STUFFING ?");  
                                 continue;  
                         }  
   
493                          cbpy = get_cbpy(bs, 1);                          cbpy = get_cbpy(bs, 1);
494                          cbp = (cbpy << 2) | cbpc;                          cbp = (cbpy << 2) | cbpc;
495    
496                          if (mb->mode == MODE_INTRA_Q)                          if (mb->mode == MODE_INTRA_Q) {
                         {  
497                                  quant += dquant_table[BitstreamGetBits(bs,2)];                                  quant += dquant_table[BitstreamGetBits(bs,2)];
498                                  if (quant > 31)                                  if (quant > 31) {
                                 {  
499                                          quant = 31;                                          quant = 31;
500                                  }                                  } else if (quant < 1) {
                                 else if (quant < 1)  
                                 {  
501                                          quant = 1;                                          quant = 1;
502                                  }                                  }
503                          }                          }
504                          mb->quant = quant;                          mb->quant = quant;
505                            mb->mvs[0].x = mb->mvs[0].y =
506                            mb->mvs[1].x = mb->mvs[1].y =
507                            mb->mvs[2].x = mb->mvs[2].y =
508                            mb->mvs[3].x = mb->mvs[3].y =0;
509    
510                          if (dec->interlacing)                          if (dec->interlacing) {
                         {  
511                                  mb->field_dct = BitstreamGetBit(bs);                                  mb->field_dct = BitstreamGetBit(bs);
512                                  DEBUG1("deci: field_dct: ", mb->field_dct);                                  DEBUG1("deci: field_dct: ", mb->field_dct);
513                          }                          }
514    
515                          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,
516                                                            intra_dc_threshold, bound);
517                  }                  }
518                    if(dec->out_frm)
519                      output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,0,y,dec->mb_width);
520    
521          }          }
522    
523  }  }
524    
525    
526  void get_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, int k, VECTOR * mv, int fcode)  void
527    get_motion_vector(DECODER * dec,
528                                      Bitstream * bs,
529                                      int x,
530                                      int y,
531                                      int k,
532                                      VECTOR * mv,
533                                      int fcode,
534                                      const int bound)
535  {  {
536    
537          int scale_fac = 1 << (fcode - 1);          int scale_fac = 1 << (fcode - 1);
# Line 436  Line 539 
539          int low = ((-32) * scale_fac);          int low = ((-32) * scale_fac);
540          int range = (64 * scale_fac);          int range = (64 * scale_fac);
541    
542          VECTOR pmv[4];          VECTOR pmv;
         int32_t psad[4];  
   
543          int mv_x, mv_y;          int mv_x, mv_y;
         int pmv_x, pmv_y;  
544    
545            pmv = get_pmv2(dec->mbs, dec->mb_width, bound, x, y, k);
         get_pmvdata(dec->mbs, x, y, dec->mb_width, k, pmv, psad);  
   
         pmv_x = pmv[0].x;  
         pmv_y = pmv[0].y;  
546    
547          mv_x = get_mv(bs, fcode);          mv_x = get_mv(bs, fcode);
548          mv_y = get_mv(bs, fcode);          mv_y = get_mv(bs, fcode);
549    
550          mv_x += pmv_x;          DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i)", mv_x, mv_y, pmv.x, pmv.y);
         mv_y += pmv_y;  
551    
552          if (mv_x < low)          mv_x += pmv.x;
553          {          mv_y += pmv.y;
554    
555            if (mv_x < low) {
556                  mv_x += range;                  mv_x += range;
557          }          } else if (mv_x > high) {
         else if (mv_x > high)  
         {  
558                  mv_x -= range;                  mv_x -= range;
559          }          }
560    
561          if (mv_y < low)          if (mv_y < low) {
         {  
562                  mv_y += range;                  mv_y += range;
563          }          } else if (mv_y > high) {
         else if (mv_y > high)  
         {  
564                  mv_y -= range;                  mv_y -= range;
565          }          }
566    
# Line 478  Line 570 
570  }  }
571    
572    
573  void decoder_pframe(DECODER * dec, Bitstream * bs, int rounding, int quant, int fcode, int intra_dc_threshold)  void
574    decoder_pframe(DECODER * dec,
575                               Bitstream * bs,
576                               int rounding,
577                               int quant,
578                               int fcode,
579                               int intra_dc_threshold)
580  {  {
581    
582          uint32_t x, y;          uint32_t x, y;
583            uint32_t bound;
584            int cp_mb, st_mb;
585    
586          start_timer();          start_timer();
587          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,
588                                       dec->width, dec->height, dec->interlacing);
589          stop_edges_timer();          stop_edges_timer();
590    
591          for (y = 0; y < dec->mb_height; y++)          bound = 0;
592          {  
593                  for (x = 0; x < dec->mb_width; x++)          for (y = 0; y < dec->mb_height; y++) {
594                    cp_mb = st_mb = 0;
595                    for (x = 0; x < dec->mb_width; x++) {
596                            MACROBLOCK *mb;
597    
598                            // skip stuffing
599                            while (BitstreamShowBits(bs, 10) == 1)
600                                    BitstreamSkip(bs, 10);
601    
602                            if (check_resync_marker(bs, fcode - 1))
603                  {                  {
604                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                                  bound = read_video_packet_header(bs, fcode - 1, &quant);
605                                    x = bound % dec->mb_width;
606                                    y = bound / dec->mb_width;
607                            }
608                            mb = &dec->mbs[y * dec->mb_width + x];
609    
610                            DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
611    
612                          if (!BitstreamGetBit(bs))                       // not_coded                          //if (!(dec->mb_skip[y*dec->mb_width + x]=BitstreamGetBit(bs)))         // not_coded
613                            if (!(BitstreamGetBit(bs)))     // not_coded
614                          {                          {
615                                  uint32_t mcbpc;                                  uint32_t mcbpc;
616                                  uint32_t cbpc;                                  uint32_t cbpc;
# Line 502  Line 619 
619                                  uint32_t cbp;                                  uint32_t cbp;
620                                  uint32_t intra;                                  uint32_t intra;
621    
622                                    cp_mb++;
623                                  mcbpc = get_mcbpc_inter(bs);                                  mcbpc = get_mcbpc_inter(bs);
624                                  mb->mode = mcbpc & 7;                                  mb->mode = mcbpc & 7;
625                                  cbpc = (mcbpc >> 4);                                  cbpc = (mcbpc >> 4);
626    
627                                    DPRINTF(DPRINTF_MB, "mode %i", mb->mode);
628                                    DPRINTF(DPRINTF_MB, "cbpc %i", cbpc);
629                                  acpred_flag = 0;                                  acpred_flag = 0;
630    
631                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);
632    
633                                  if (intra)                                  if (intra) {
                                 {  
634                                          acpred_flag = BitstreamGetBit(bs);                                          acpred_flag = BitstreamGetBit(bs);
635                                  }                                  }
636    
                                 if (mb->mode == MODE_STUFFING)  
                                 {  
                                         DEBUG("-- STUFFING ?");  
                                         continue;  
                                 }  
   
637                                  cbpy = get_cbpy(bs, intra);                                  cbpy = get_cbpy(bs, intra);
638                                    DPRINTF(DPRINTF_MB, "cbpy %i", cbpy);
639    
640                                  cbp = (cbpy << 2) | cbpc;                                  cbp = (cbpy << 2) | cbpc;
641    
642                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q)                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q) {
643                                  {                                          int dquant = dquant_table[BitstreamGetBits(bs, 2)];
644                                          quant += dquant_table[BitstreamGetBits(bs,2)];                                          DPRINTF(DPRINTF_MB, "dquant %i", dquant);
645                                          if (quant > 31)                                          quant += dquant;
646                                          {                                          if (quant > 31) {
647                                                  quant = 31;                                                  quant = 31;
648                                          }                                          } else if (quant < 1) {
                                         else if (mb->quant < 1)  
                                         {  
649                                                  quant = 1;                                                  quant = 1;
650                                          }                                          }
651                                            DPRINTF(DPRINTF_MB, "quant %i", quant);
652                                  }                                  }
653                                  mb->quant = quant;                                  mb->quant = quant;
654    
655                                  if (dec->interlacing)                                  if (dec->interlacing) {
656                                  {                                          if (cbp || intra) {
657                                          mb->field_dct = BitstreamGetBit(bs);                                          mb->field_dct = BitstreamGetBit(bs);
658                                          DEBUG1("decp: field_dct: ", mb->field_dct);                                          DEBUG1("decp: field_dct: ", mb->field_dct);
659                                            }
660    
661                                          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
                                         {  
662                                                  mb->field_pred = BitstreamGetBit(bs);                                                  mb->field_pred = BitstreamGetBit(bs);
663                                                  DEBUG1("decp: field_pred: ", mb->field_pred);                                                  DEBUG1("decp: field_pred: ", mb->field_pred);
664    
665                                                  if (mb->field_pred)                                                  if (mb->field_pred) {
                                                 {  
666                                                          mb->field_for_top = BitstreamGetBit(bs);                                                          mb->field_for_top = BitstreamGetBit(bs);
667                                                          DEBUG1("decp: field_for_top: ", mb->field_for_top);                                                          DEBUG1("decp: field_for_top: ", mb->field_for_top);
668                                                          mb->field_for_bot = BitstreamGetBit(bs);                                                          mb->field_for_bot = BitstreamGetBit(bs);
# Line 557  Line 671 
671                                          }                                          }
672                                  }                                  }
673    
674                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
675                                  {                                          if (dec->interlacing && mb->field_pred) {
676                                          if (dec->interlacing && mb->field_pred)                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
677                                                                                      fcode, bound);
678                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1],
679                                                                                      fcode, bound);
680                                            } else {
681                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
682                                                                                      fcode, bound);
683                                                    mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
684                                                            mb->mvs[0].x;
685                                                    mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
686                                                            mb->mvs[0].y;
687                                            }
688                                    } else if (mb->mode == MODE_INTER4V ) {
689    
690                                            get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode, bound);
691                                            get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode, bound);
692                                            get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode, bound);
693                                            get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode, bound);
694                                    } else                  // MODE_INTRA, MODE_INTRA_Q
695                                    {
696                                            mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
697                                                    0;
698                                            mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
699                                                    0;
700                                            decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
701                                                                            intra_dc_threshold, bound);
702                                            continue;
703                                    }
704    
705                                    decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant,
706                                                                    rounding);
707                            } else                          // not coded
708                                          {                                          {
709                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);                                  DEBUG2("P-frame MB at (X,Y)=",x,y);
710                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1], fcode);                                  mb->mode = MODE_NOT_CODED;
711                                    mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
712                                    mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;
713    
714                                    // copy macroblock directly from ref to cur
715    
716                                    start_timer();
717    
718                                    transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
719                                                                     (16 * x),
720                                                                     dec->refn[0].y + (16 * y) * dec->edged_width +
721                                                                     (16 * x), dec->edged_width);
722    
723                                    transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
724                                                                     (16 * x + 8),
725                                                                     dec->refn[0].y + (16 * y) * dec->edged_width +
726                                                                     (16 * x + 8), dec->edged_width);
727    
728                                    transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
729                                                                     (16 * x),
730                                                                     dec->refn[0].y + (16 * y +
731                                                                                                       8) * dec->edged_width +
732                                                                     (16 * x), dec->edged_width);
733    
734                                    transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
735                                                                     (16 * x + 8),
736                                                                     dec->refn[0].y + (16 * y +
737                                                                                                       8) * dec->edged_width +
738                                                                     (16 * x + 8), dec->edged_width);
739    
740                                    transfer8x8_copy(dec->cur.u + (8 * y) * dec->edged_width / 2 +
741                                                                     (8 * x),
742                                                                     dec->refn[0].u +
743                                                                     (8 * y) * dec->edged_width / 2 + (8 * x),
744                                                                     dec->edged_width / 2);
745    
746                                    transfer8x8_copy(dec->cur.v + (8 * y) * dec->edged_width / 2 +
747                                                                     (8 * x),
748                                                                     dec->refn[0].v +
749                                                                     (8 * y) * dec->edged_width / 2 + (8 * x),
750                                                                     dec->edged_width / 2);
751                                    stop_transfer_timer();
752                                    if(dec->out_frm && cp_mb > 0) {
753                                      output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
754                                      cp_mb = 0;
755                                          }                                          }
756                                          else                                  st_mb = x+1;
757                            }
758                    }
759                    if(dec->out_frm && cp_mb > 0)
760                      output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
761            }
762    }
763    
764    
765    // add by MinChen <chenm001@163.com>
766    // decode B-frame motion vector
767    void
768    get_b_motion_vector(DECODER * dec,
769                                            Bitstream * bs,
770                                            int x,
771                                            int y,
772                                            VECTOR * mv,
773                                            int fcode,
774                                            const VECTOR pmv)
775                                          {                                          {
776                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);          int scale_fac = 1 << (fcode - 1);
777                                                  mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;          int high = (32 * scale_fac) - 1;
778                                                  mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;          int low = ((-32) * scale_fac);
779            int range = (64 * scale_fac);
780    
781            int mv_x, mv_y;
782            int pmv_x, pmv_y;
783    
784            pmv_x = pmv.x;
785            pmv_y = pmv.y;
786    
787            mv_x = get_mv(bs, fcode);
788            mv_y = get_mv(bs, fcode);
789    
790            mv_x += pmv_x;
791            mv_y += pmv_y;
792    
793            if (mv_x < low) {
794                    mv_x += range;
795            } else if (mv_x > high) {
796                    mv_x -= range;
797                                          }                                          }
798    
799            if (mv_y < low) {
800                    mv_y += range;
801            } else if (mv_y > high) {
802                    mv_y -= range;
803            }
804    
805            mv->x = mv_x;
806            mv->y = mv_y;
807                                  }                                  }
808                                  else if (mb->mode == MODE_INTER4V /* || mb->mode == MODE_INTER4V_Q */)  
809    
810    // add by MinChen <chenm001@163.com>
811    // decode an B-frame forward & backward inter macroblock
812    void
813    decoder_bf_mbinter(DECODER * dec,
814                                       const MACROBLOCK * pMB,
815                                       const uint32_t x_pos,
816                                       const uint32_t y_pos,
817                                       const uint32_t cbp,
818                                       Bitstream * bs,
819                                       const uint32_t quant,
820                                       const uint8_t ref)
821                                  {                                  {
822                                          get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);  
823                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode);          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
824                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode);          DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
825                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode);  
826            uint32_t stride = dec->edged_width;
827            uint32_t stride2 = stride / 2;
828            uint32_t next_block = stride * 8;
829            uint32_t i;
830            uint32_t iQuant = pMB->quant;
831            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
832            int uv_dx, uv_dy;
833    
834            pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
835            pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
836            pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
837    
838    
839            if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
840                    uv_dx = pMB->mvs[0].x;
841                    uv_dy = pMB->mvs[0].y;
842    
843                    uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
844                    uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
845            } else {
846                    int sum;
847    
848                    sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
849                    uv_dx =
850                            (sum ==
851                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
852                                                                      (ABS(sum) / 16) * 2));
853    
854                    sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
855                    uv_dy =
856                            (sum ==
857                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
858                                                                      (ABS(sum) / 16) * 2));
859                                  }                                  }
860                                  else  // MODE_INTRA, MODE_INTRA_Q  
861            start_timer();
862            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos, 16 * y_pos,
863                                                      pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
864            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos + 8,
865                                                      16 * y_pos, pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
866            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos,
867                                                      16 * y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,
868                                                      0);
869            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos + 8,
870                                                      16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
871                                                      0);
872            interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8 * x_pos, 8 * y_pos,
873                                                      uv_dx, uv_dy, stride2, 0);
874            interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8 * x_pos, 8 * y_pos,
875                                                      uv_dx, uv_dy, stride2, 0);
876            stop_comp_timer();
877    
878            for (i = 0; i < 6; i++) {
879                    if (cbp & (1 << (5 - i)))       // coded
880                                  {                                  {
881                                          mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;                          memset(&block[i * 64], 0, 64 * sizeof(int16_t));        // clear
882                                          mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;  
883                                          decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant, intra_dc_threshold);                          start_timer();
884                                          continue;                          get_inter_block(bs, &block[i * 64]);
885                            stop_coding_timer();
886    
887                            start_timer();
888                            if (dec->quant_type == 0) {
889                                    dequant_inter(&data[i * 64], &block[i * 64], iQuant);
890                            } else {
891                                    dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
892                            }
893                            stop_iquant_timer();
894    
895                            start_timer();
896                            idct(&data[i * 64]);
897                            stop_idct_timer();
898                    }
899                                  }                                  }
900    
901                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant, rounding);          if (dec->interlacing && pMB->field_dct) {
902                    next_block = stride;
903                    stride *= 2;
904                          }                          }
905                          else    // not coded  
906            start_timer();
907            if (cbp & 32)
908                    transfer_16to8add(pY_Cur, &data[0 * 64], stride);
909            if (cbp & 16)
910                    transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
911            if (cbp & 8)
912                    transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
913            if (cbp & 4)
914                    transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
915            if (cbp & 2)
916                    transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
917            if (cbp & 1)
918                    transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
919            stop_transfer_timer();
920    }
921    
922    
923    // add by MinChen <chenm001@163.com>
924    // decode an B-frame direct &  inter macroblock
925    void
926    decoder_bf_interpolate_mbinter(DECODER * dec,
927                                                               IMAGE forward,
928                                                               IMAGE backward,
929                                                               const MACROBLOCK * pMB,
930                                                               const uint32_t x_pos,
931                                                               const uint32_t y_pos,
932                                                               Bitstream * bs)
933                          {                          {
934    
935                                  mb->mode = MODE_NOT_CODED;          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
936                                  mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;          DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
937                                  mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;  
938            uint32_t stride = dec->edged_width;
939            uint32_t stride2 = stride / 2;
940            uint32_t next_block = stride * 8;
941            uint32_t iQuant = pMB->quant;
942            int uv_dx, uv_dy;
943            int b_uv_dx, b_uv_dy;
944            uint32_t i;
945            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
946        const uint32_t cbp = pMB->cbp;
947    
948            pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
949            pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
950            pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
951    
952    
953            if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
954                    uv_dx = pMB->mvs[0].x;
955                    uv_dy = pMB->mvs[0].y;
956    
957                    uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
958                    uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
959    
960                    b_uv_dx = pMB->b_mvs[0].x;
961                    b_uv_dy = pMB->b_mvs[0].y;
962    
963                    b_uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
964                    b_uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
965            } else {
966                    int sum;
967    
968                    sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
969                    uv_dx =
970                            (sum ==
971                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
972                                                                      (ABS(sum) / 16) * 2));
973    
974                    sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
975                    uv_dy =
976                            (sum ==
977                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
978                                                                      (ABS(sum) / 16) * 2));
979    
980                    sum =
981                            pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x +
982                            pMB->b_mvs[3].x;
983                    b_uv_dx =
984                            (sum ==
985                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
986                                                                      (ABS(sum) / 16) * 2));
987    
988                    sum =
989                            pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y +
990                            pMB->b_mvs[3].y;
991                    b_uv_dy =
992                            (sum ==
993                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
994                                                                      (ABS(sum) / 16) * 2));
995            }
996    
                                 // copy macroblock directly from ref to cur  
997    
998                                  start_timer();                                  start_timer();
999            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos,
1000                                                      pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1001            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8, 16 * y_pos,
1002                                                      pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1003            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos + 8,
1004                                                      pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1005            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8,
1006                                                      16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
1007                                                      0);
1008            interpolate8x8_switch(dec->cur.u, forward.u, 8 * x_pos, 8 * y_pos, uv_dx,
1009                                                      uv_dy, stride2, 0);
1010            interpolate8x8_switch(dec->cur.v, forward.v, 8 * x_pos, 8 * y_pos, uv_dx,
1011                                                      uv_dy, stride2, 0);
1012    
1013    
1014            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos, 16 * y_pos,
1015                                                      pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1016            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1017                                                      16 * y_pos, pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride,
1018                                                      0);
1019            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos,
1020                                                      16 * y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y,
1021                                                      stride, 0);
1022            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1023                                                      16 * y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y,
1024                                                      stride, 0);
1025            interpolate8x8_switch(dec->refn[2].u, backward.u, 8 * x_pos, 8 * y_pos,
1026                                                      b_uv_dx, b_uv_dy, stride2, 0);
1027            interpolate8x8_switch(dec->refn[2].v, backward.v, 8 * x_pos, 8 * y_pos,
1028                                                      b_uv_dx, b_uv_dy, stride2, 0);
1029    
1030            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos, 16 * y_pos,
1031                                             stride);
1032            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos + 8, 16 * y_pos,
1033                                             stride);
1034            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos, 16 * y_pos + 8,
1035                                             stride);
1036            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos + 8,
1037                                             16 * y_pos + 8, stride);
1038            interpolate8x8_c(dec->cur.u, dec->refn[2].u, 8 * x_pos, 8 * y_pos,
1039                                             stride2);
1040            interpolate8x8_c(dec->cur.v, dec->refn[2].v, 8 * x_pos, 8 * y_pos,
1041                                             stride2);
1042            stop_comp_timer();
1043    
1044            for (i = 0; i < 6; i++) {
1045                    if (cbp & (1 << (5 - i)))       // coded
1046                    {
1047                            memset(&block[i * 64], 0, 64 * sizeof(int16_t));        // clear
1048    
1049                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),                          start_timer();
1050                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x),                          get_inter_block(bs, &block[i * 64]);
1051                                                   dec->edged_width);                          stop_coding_timer();
   
                                 transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x+8),  
                                                  dec->refn[0].y + (16*y)*dec->edged_width + (16*x+8),  
                                                  dec->edged_width);  
   
                                 transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x),  
                                                  dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x),  
                                                  dec->edged_width);  
   
                                 transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x+8),  
                                                  dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x+8),  
                                                  dec->edged_width);  
1052    
1053                                  transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),                          start_timer();
1054                                                   dec->refn[0].u + (8*y)*dec->edged_width/2 + (8*x),                          if (dec->quant_type == 0) {
1055                                                   dec->edged_width/2);                                  dequant_inter(&data[i * 64], &block[i * 64], iQuant);
1056                            } else {
1057                                    dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
1058                            }
1059                            stop_iquant_timer();
1060    
1061                                  transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),                          start_timer();
1062                                                   dec->refn[0].v + (8*y)*dec->edged_width/2 + (8*x),                          idct(&data[i * 64]);
1063                                                   dec->edged_width/2);                          stop_idct_timer();
1064                    }
1065            }
1066    
1067            if (dec->interlacing && pMB->field_dct) {
1068                    next_block = stride;
1069                    stride *= 2;
1070            }
1071    
1072            start_timer();
1073            if (cbp & 32)
1074                    transfer_16to8add(pY_Cur, &data[0 * 64], stride);
1075            if (cbp & 16)
1076                    transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
1077            if (cbp & 8)
1078                    transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
1079            if (cbp & 4)
1080                    transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
1081            if (cbp & 2)
1082                    transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
1083            if (cbp & 1)
1084                    transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
1085                                  stop_transfer_timer();                                  stop_transfer_timer();
1086                          }                          }
1087    
1088    
1089    // add by MinChen <chenm001@163.com>
1090    // for decode B-frame dbquant
1091    int32_t __inline
1092    get_dbquant(Bitstream * bs)
1093    {
1094            if (!BitstreamGetBit(bs))       // '0'
1095                    return (0);
1096            else if (!BitstreamGetBit(bs))  // '10'
1097                    return (-2);
1098            else
1099                    return (2);                             // '11'
1100                  }                  }
1101    
1102    // add by MinChen <chenm001@163.com>
1103    // for decode B-frame mb_type
1104    // bit   ret_value
1105    // 1        0
1106    // 01       1
1107    // 001      2
1108    // 0001     3
1109    int32_t __inline
1110    get_mbtype(Bitstream * bs)
1111    {
1112            int32_t mb_type;
1113    
1114            for (mb_type = 0; mb_type <= 3; mb_type++) {
1115                    if (BitstreamGetBit(bs))
1116                            break;
1117          }          }
1118    
1119            if (mb_type <= 3)
1120                    return (mb_type);
1121            else
1122                    return (-1);
1123  }  }
1124    
1125  int decoder_decode(DECODER * dec, XVID_DEC_FRAME * frame)  void
1126    decoder_bframe(DECODER * dec,
1127                               Bitstream * bs,
1128                               int quant,
1129                               int fcode_forward,
1130                               int fcode_backward)
1131    {
1132            uint32_t x, y;
1133            VECTOR mv;
1134            const VECTOR zeromv = {0,0};
1135    #ifdef BFRAMES_DEC_DEBUG
1136            FILE *fp;
1137            static char first=0;
1138    #define BFRAME_DEBUG    if (!first && fp){ \
1139                    fprintf(fp,"Y=%3d   X=%3d   MB=%2d   CBP=%02X\n",y,x,mb->mb_type,mb->cbp); \
1140            }
1141    #endif
1142    
1143            start_timer();
1144            image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
1145                                       dec->width, dec->height, dec->interlacing);
1146            image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height,
1147                                       dec->width, dec->height, dec->interlacing);
1148            stop_edges_timer();
1149    
1150    #ifdef BFRAMES_DEC_DEBUG
1151            if (!first){
1152                    fp=fopen("C:\\XVIDDBG.TXT","w");
1153            }
1154    #endif
1155    
1156            for (y = 0; y < dec->mb_height; y++) {
1157                    // Initialize Pred Motion Vector
1158                    dec->p_fmv = dec->p_bmv = zeromv;
1159                    for (x = 0; x < dec->mb_width; x++) {
1160                            MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x];
1161                            MACROBLOCK *last_mb = &dec->last_mbs[y * dec->mb_width + x];
1162    
1163                            mv =
1164                            mb->b_mvs[0] = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] =
1165                            mb->mvs[0] = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = zeromv;
1166    
1167                            // the last P_VOP is skip macroblock ?
1168                            if (last_mb->mode == MODE_NOT_CODED) {
1169                                    //DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y);
1170                                    mb->cbp = 0;
1171    #ifdef BFRAMES_DEC_DEBUG
1172                                    mb->mb_type = MODE_NOT_CODED;
1173            BFRAME_DEBUG
1174    #endif
1175                                    mb->mb_type = MODE_FORWARD;
1176                                    mb->quant = last_mb->quant;
1177                                    //mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1178                                    //mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
1179    
1180                                    decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, mb->quant, 1);
1181                                    continue;
1182                            }
1183    
1184                            if (!BitstreamGetBit(bs)) {     // modb=='0'
1185                                    const uint8_t modb2 = BitstreamGetBit(bs);
1186    
1187                                    mb->mb_type = get_mbtype(bs);
1188    
1189                                    if (!modb2) {   // modb=='00'
1190                                            mb->cbp = BitstreamGetBits(bs, 6);
1191                                    } else {
1192                                            mb->cbp = 0;
1193                                    }
1194                                    if (mb->mb_type && mb->cbp) {
1195                                            quant += get_dbquant(bs);
1196    
1197                                            if (quant > 31) {
1198                                                    quant = 31;
1199                                            } else if (quant < 1) {
1200                                                    quant = 1;
1201                                            }
1202                                    }
1203                            } else {
1204                                    mb->mb_type = MODE_DIRECT_NONE_MV;
1205                                    mb->cbp = 0;
1206                            }
1207    
1208                            mb->quant = quant;
1209                            mb->mode = MODE_INTER4V;
1210                            //DEBUG1("Switch bm_type=",mb->mb_type);
1211    
1212    #ifdef BFRAMES_DEC_DEBUG
1213            BFRAME_DEBUG
1214    #endif
1215    
1216                            switch (mb->mb_type) {
1217                            case MODE_DIRECT:
1218                                    get_b_motion_vector(dec, bs, x, y, &mv, 1, zeromv);
1219    
1220                            case MODE_DIRECT_NONE_MV:
1221                                    {
1222                                            const int64_t TRB = dec->time_pp - dec->time_bp, TRD = dec->time_pp;
1223                                            int i;
1224    
1225                                            for (i = 0; i < 4; i++) {
1226                                                    mb->mvs[i].x = (int32_t) ((TRB * last_mb->mvs[i].x)
1227                                                                          / TRD + mv.x);
1228                                                    mb->b_mvs[i].x = (int32_t) ((mv.x == 0)
1229                                                                                    ? ((TRB - TRD) * last_mb->mvs[i].x)
1230                                                                                      / TRD
1231                                                                                    : mb->mvs[i].x - last_mb->mvs[i].x);
1232                                                    mb->mvs[i].y = (int32_t) ((TRB * last_mb->mvs[i].y)
1233                                                                          / TRD + mv.y);
1234                                                    mb->b_mvs[i].y = (int32_t) ((mv.y == 0)
1235                                                                                    ? ((TRB - TRD) * last_mb->mvs[i].y)
1236                                                                                      / TRD
1237                                                                                : mb->mvs[i].y - last_mb->mvs[i].y);
1238                                            }
1239                                            //DEBUG("B-frame Direct!\n");
1240                                    }
1241                                    decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1242                                                                                               mb, x, y, bs);
1243                                    break;
1244    
1245                            case MODE_INTERPOLATE:
1246                                    get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1247                                                                            dec->p_fmv);
1248                                    dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] =     mb->mvs[0];
1249    
1250                                    get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0],
1251                                                                            fcode_backward, dec->p_bmv);
1252                                    dec->p_bmv = mb->b_mvs[1] = mb->b_mvs[2] =
1253                                            mb->b_mvs[3] = mb->b_mvs[0];
1254    
1255                                    decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1256                                                                                               mb, x, y, bs);
1257                                    //DEBUG("B-frame Bidir!\n");
1258                                    break;
1259    
1260                            case MODE_BACKWARD:
1261                                    get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_backward,
1262                                                                            dec->p_bmv);
1263                                    dec->p_bmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] =     mb->mvs[0];
1264    
1265                                    mb->mode = MODE_INTER;
1266                                    decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);
1267                                    //DEBUG("B-frame Backward!\n");
1268                                    break;
1269    
1270                            case MODE_FORWARD:
1271                                    get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1272                                                                            dec->p_fmv);
1273                                    dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] =     mb->mvs[0];
1274    
1275                                    mb->mode = MODE_INTER;
1276                                    decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1277                                    //DEBUG("B-frame Forward!\n");
1278                                    break;
1279    
1280                            default:
1281                                    DEBUG1("Not support B-frame mb_type =", mb->mb_type);
1282                            }
1283    
1284                    }                                               // end of FOR
1285            }
1286    #ifdef BFRAMES_DEC_DEBUG
1287            if (!first){
1288                    first=1;
1289                    if (fp)
1290                            fclose(fp);
1291            }
1292    #endif
1293    }
1294    
1295    // swap two MACROBLOCK array
1296    void
1297    mb_swap(MACROBLOCK ** mb1,
1298                    MACROBLOCK ** mb2)
1299    {
1300            MACROBLOCK *temp = *mb1;
1301    
1302            *mb1 = *mb2;
1303            *mb2 = temp;
1304    }
1305    
1306    int
1307    decoder_decode(DECODER * dec,
1308                               XVID_DEC_FRAME * frame)
1309  {  {
1310    
1311          Bitstream bs;          Bitstream bs;
1312          uint32_t rounding;          uint32_t rounding;
1313          uint32_t quant;          uint32_t quant;
1314          uint32_t fcode;          uint32_t fcode_forward;
1315            uint32_t fcode_backward;
1316          uint32_t intra_dc_threshold;          uint32_t intra_dc_threshold;
1317          uint32_t vop_type;          uint32_t vop_type;
1318    
1319          start_global_timer();          start_global_timer();
1320    
1321            dec->out_frm = (frame->colorspace == XVID_CSP_EXTERN) ? frame->image : NULL;
1322    
1323          BitstreamInit(&bs, frame->bitstream, frame->length);          BitstreamInit(&bs, frame->bitstream, frame->length);
1324    
1325          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
1326          // for support B-frame to reference last 2 frame          // for support B-frame to reference last 2 frame
1327          vop_type=BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode, &intra_dc_threshold);          dec->frames++;
1328            vop_type =
1329                    BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward,
1330                                                             &fcode_backward, &intra_dc_threshold);
1331    
1332          if (vop_type==I_VOP || vop_type==P_VOP){          dec->p_bmv.x = dec->p_bmv.y = dec->p_fmv.y = dec->p_fmv.y = 0;  // init pred vector to 0
                 image_swap(&dec->refn[0], &dec->refn[1]);  
                 image_swap(&dec->cur, &dec->refn[0]);  
         }  
1333    
1334          switch (vop_type)          switch (vop_type) {
         {  
1335          case P_VOP :          case P_VOP :
1336                  decoder_pframe(dec, &bs, rounding, quant, fcode, intra_dc_threshold);                  decoder_pframe(dec, &bs, rounding, quant, fcode_forward,
1337                                               intra_dc_threshold);
1338    #ifdef BFRAMES_DEC
1339                    DEBUG1("P_VOP  Time=", dec->time);
1340    #endif
1341                  break;                  break;
1342    
1343          case I_VOP :          case I_VOP :
                 //DEBUG1("",intra_dc_threshold);  
1344                  decoder_iframe(dec, &bs, quant, intra_dc_threshold);                  decoder_iframe(dec, &bs, quant, intra_dc_threshold);
1345    #ifdef BFRAMES_DEC
1346                    DEBUG1("I_VOP  Time=", dec->time);
1347    #endif
1348                  break;                  break;
1349    
1350          case B_VOP :    // ignore          case B_VOP:
1351    #ifdef BFRAMES_DEC
1352                    if (dec->time_pp > dec->time_bp) {
1353                            DEBUG1("B_VOP  Time=", dec->time);
1354                            decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);
1355                    } else {
1356                            DEBUG("broken B-frame!");
1357                    }
1358    #else
1359                    image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1360    #endif
1361                  break;                  break;
1362    
1363          case N_VOP :    // vop not coded          case N_VOP :    // vop not coded
1364                    // when low_delay==0, N_VOP's should interpolate between the past and future frames
1365                    image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1366                  break;                  break;
1367    
1368          default :          default :
1369                  return XVID_ERR_FAIL;                  return XVID_ERR_FAIL;
1370          }          }
1371    
1372    #ifdef BFRAMES_DEC_DEBUG
1373            if (frame->length != BitstreamPos(&bs) / 8){
1374                    DEBUG2("InLen/UseLen",frame->length, BitstreamPos(&bs) / 8);
1375            }
1376    #endif
1377          frame->length = BitstreamPos(&bs) / 8;          frame->length = BitstreamPos(&bs) / 8;
1378    
1379          start_timer();  
1380    #ifdef BFRAMES_DEC
1381            // test if no B_VOP
1382            if (dec->low_delay || dec->frames == 0) {
1383    #endif
1384          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,
1385                       frame->image, frame->stride, frame->colorspace);                       frame->image, frame->stride, frame->colorspace);
1386    
1387    #ifdef BFRAMES_DEC
1388            } else {
1389                    if (dec->frames >= 1) {
1390                            start_timer();
1391                            if ((vop_type == I_VOP || vop_type == P_VOP)) {
1392                                    image_output(&dec->refn[0], dec->width, dec->height,
1393                                                             dec->edged_width, frame->image, frame->stride,
1394                                                             frame->colorspace);
1395                            } else if (vop_type == B_VOP) {
1396                                    image_output(&dec->cur, dec->width, dec->height,
1397                                                             dec->edged_width, frame->image, frame->stride,
1398                                                             frame->colorspace);
1399                            }
1400          stop_conv_timer();          stop_conv_timer();
1401                    }
1402            }
1403    #endif
1404    
1405            if (vop_type == I_VOP || vop_type == P_VOP) {
1406                    image_swap(&dec->refn[0], &dec->refn[1]);
1407                    image_swap(&dec->cur, &dec->refn[0]);
1408    
1409                    // swap MACROBLOCK
1410                    // the Divx will not set the low_delay flage some times
1411                    // so follow code will wrong to not swap at that time
1412                    // this will broken bitstream! so I'm change it,
1413                    // But that is not the best way! can anyone tell me how
1414                    // to do another way?
1415                    // 18-07-2002   MinChen<chenm001@163.com>
1416                    //if (!dec->low_delay && vop_type == P_VOP)
1417                    if (vop_type == P_VOP)
1418                            mb_swap(&dec->mbs, &dec->last_mbs);
1419            }
1420    
1421          emms();          emms();
1422    
1423          stop_global_timer();          stop_global_timer();
1424    
1425          return XVID_ERR_OK;          return XVID_ERR_OK;
   
1426  }  }

Legend:
Removed from v.1.13  
changed lines
  Added in v.1.37

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