[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.14, Fri May 3 00:41:22 2002 UTC revision 1.37.2.6, Wed Oct 30 18:01:48 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);   *  02.05.2002  add B-frame decode support(have some problem);
47   *              MinChen <chenm001@163.com>   *              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>
# Line 51  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 69  Line 84 
84  #include "prediction/mbprediction.h"  #include "prediction/mbprediction.h"
85  #include "utils/timer.h"  #include "utils/timer.h"
86  #include "utils/emms.h"  #include "utils/emms.h"
87    #include "motion/motion.h"
88    
89  #include "image/image.h"  #include "image/image.h"
90  #include "image/colorspace.h"  #include "image/colorspace.h"
91  #include "utils/mem_align.h"  #include "utils/mem_align.h"
92    
93  int decoder_create(XVID_DEC_PARAM * param)  int
94    decoder_create(XVID_DEC_PARAM * param)
95  {  {
96          DECODER * dec;          DECODER * dec;
97    
98          dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);          dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);
99          if (dec == NULL)          if (dec == NULL) {
         {  
100                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
101          }          }
102          param->handle = dec;          param->handle = dec;
# Line 93  Line 109 
109    
110          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;
111          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;
112            dec->low_delay = 0;
113    
114          if (image_create(&dec->cur, dec->edged_width, dec->edged_height))          if (image_create(&dec->cur, dec->edged_width, dec->edged_height)) {
         {  
115                  xvid_free(dec);                  xvid_free(dec);
116                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
117          }          }
118    
119          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height)) {
         {  
120                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
121                  xvid_free(dec);                  xvid_free(dec);
122                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
123          }          }
124          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
125          // for support B-frame to reference last 2 frame          // for support B-frame to reference last 2 frame
126          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {
         {  
127                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
128                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
129                  xvid_free(dec);                  xvid_free(dec);
130                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
131          }          }
132          if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height)) {
         {  
133                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
134                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
135                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
# Line 124  Line 137 
137                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
138          }          }
139    
140          dec->mbs = xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height, CACHE_LINE);          if (image_create(&dec->refh, dec->edged_width, dec->edged_height)) {
141          if (dec->mbs == NULL)                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
142          {                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
143                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
144                    image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
145                    xvid_free(dec);
146                    return XVID_ERR_MEMORY;
147            }
148    
149            dec->mbs =
150                    xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
151                                            CACHE_LINE);
152            if (dec->mbs == NULL) {
153                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
154                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
155                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
156                  image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
157                    image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
158                  xvid_free(dec);                  xvid_free(dec);
159                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
160          }          }
161    
162            memset(dec->mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
163    
164          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
165          // for skip MB flag          // for skip MB flag
166          dec->last_mbs = xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height, CACHE_LINE);          dec->last_mbs =
167          if (dec->last_mbs == NULL)                  xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
168          {                                          CACHE_LINE);
169            if (dec->last_mbs == NULL) {
170                  xvid_free(dec->mbs);                  xvid_free(dec->mbs);
171                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
172                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
173                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
174                  image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
175                    image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
176                  xvid_free(dec);                  xvid_free(dec);
177                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
178          }          }
179    
180            memset(dec->last_mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
181    
182          init_timer();          init_timer();
183    
184          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
# Line 159  Line 190 
190  }  }
191    
192    
193  int decoder_destroy(DECODER * dec)  int
194    decoder_destroy(DECODER * dec)
195  {  {
196          xvid_free(dec->last_mbs);          xvid_free(dec->last_mbs);
197          xvid_free(dec->mbs);          xvid_free(dec->mbs);
198          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
199          image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);          image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
200          image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);          image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
201            image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
202          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
203          xvid_free(dec);          xvid_free(dec);
204    
# Line 175  Line 208 
208    
209    
210    
211  static const int32_t dquant_table[4] =  static const int32_t dquant_table[4] = {
 {  
212          -1, -2, 1, 2          -1, -2, 1, 2
213  };  };
214    
# Line 185  Line 217 
217    
218  // decode an intra macroblock  // decode an intra macroblock
219    
220  void decoder_mbintra(DECODER * dec,  void
221    decoder_mbintra(DECODER * dec,
222                       MACROBLOCK * pMB,                       MACROBLOCK * pMB,
223                       const uint32_t x_pos,                       const uint32_t x_pos,
224                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 193  Line 226 
226                       const uint32_t cbp,                       const uint32_t cbp,
227                       Bitstream * bs,                       Bitstream * bs,
228                       const uint32_t quant,                       const uint32_t quant,
229                       const uint32_t intra_dc_threshold)                                  const uint32_t intra_dc_threshold,
230                                    const unsigned int bound)
231  {  {
232    
233          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
# Line 212  Line 246 
246    
247          memset(block, 0, 6*64*sizeof(int16_t));         // clear          memset(block, 0, 6*64*sizeof(int16_t));         // clear
248    
249          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
         {  
250                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
251                  int16_t predictors[8];                  int16_t predictors[8];
252                  int start_coeff;                  int start_coeff;
253    
254                  start_timer();                  start_timer();
255                  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],
256                  if (!acpred_flag)                                           iQuant, iDcScaler, predictors, bound);
257                  {                  if (!acpred_flag) {
258                          pMB->acpred_directions[i] = 0;                          pMB->acpred_directions[i] = 0;
259                  }                  }
260                  stop_prediction_timer();                  stop_prediction_timer();
261    
262                  if (quant < intra_dc_threshold)                  if (quant < intra_dc_threshold) {
                 {  
263                          int dc_size;                          int dc_size;
264                          int dc_dif;                          int dc_dif;
265    
266                          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);
267                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;
268    
269                          if (dc_size > 8)                          if (dc_size > 8) {
                         {  
270                                  BitstreamSkip(bs, 1);           // marker                                  BitstreamSkip(bs, 1);           // marker
271                          }                          }
272    
273                          block[i*64 + 0] = dc_dif;                          block[i*64 + 0] = dc_dif;
274                          start_coeff = 1;                          start_coeff = 1;
275                  }  
276                  else                          DPRINTF(DPRINTF_COEFF,"block[0] %i", dc_dif);
277                  {                  } else {
278                          start_coeff = 0;                          start_coeff = 0;
279                  }                  }
280    
281                  start_timer();                  start_timer();
282                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
283                  {                  {
284                          get_intra_block(bs, &block[i*64], pMB->acpred_directions[i], start_coeff);                          int direction = dec->alternate_vertical_scan ?
285                                    2 : pMB->acpred_directions[i];
286    
287                            get_intra_block(bs, &block[i * 64], direction, start_coeff);
288                  }                  }
289                  stop_coding_timer();                  stop_coding_timer();
290    
# Line 259  Line 293 
293                  stop_prediction_timer();                  stop_prediction_timer();
294    
295                  start_timer();                  start_timer();
296                  if (dec->quant_type == 0)                  if (dec->quant_type == 0) {
                 {  
297                          dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);                          dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
298                  }                  } else {
                 else  
                 {  
299                          dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);                          dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
300                  }                  }
301                  stop_iquant_timer();                  stop_iquant_timer();
# Line 274  Line 305 
305                  stop_idct_timer();                  stop_idct_timer();
306          }          }
307    
308          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
309                  next_block = stride;                  next_block = stride;
310                  stride *= 2;                  stride *= 2;
311          }          }
# Line 296  Line 326 
326    
327  #define SIGN(X) (((X)>0)?1:-1)  #define SIGN(X) (((X)>0)?1:-1)
328  #define ABS(X) (((X)>0)?(X):-(X))  #define ABS(X) (((X)>0)?(X):-(X))
 static const uint32_t roundtab[16] =  
 { 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 };  
   
329    
330  // decode an inter macroblock  // decode an inter macroblock
331    
332  void decoder_mbinter(DECODER * dec,  void
333    decoder_mbinter(DECODER * dec,
334                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
335                       const uint32_t x_pos,                       const uint32_t x_pos,
336                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 328  Line 356 
356          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
357          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
358    
359          if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)          if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
         {  
360                  uv_dx = pMB->mvs[0].x;                  uv_dx = pMB->mvs[0].x;
361                  uv_dy = pMB->mvs[0].y;                  uv_dy = pMB->mvs[0].y;
362    
363                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;                  if (dec->quarterpel)
                 uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;  
         }  
         else  
364          {          {
365                            uv_dx = (uv_dx >> 2) + roundtab_78[uv_dx & 0x7];
366                            uv_dy = (uv_dy >> 2) + roundtab_78[uv_dy & 0x7];
367                    }
368                    else {
369                            uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
370                            uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
371                    }
372    
373                    start_timer();
374                    if(dec->quarterpel) {
375                            interpolate16x16_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
376                                                                                dec->refh.y + 128, 16*x_pos, 16*y_pos,
377                                                                                pMB->mvs[0].x, pMB->mvs[0].y, stride,  rounding);
378                    }
379                    else {
380                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
381                                                                      pMB->mvs[0].x, pMB->mvs[0].y, stride,  rounding);
382                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos,
383                                                                  pMB->mvs[1].x, pMB->mvs[1].y, stride,  rounding);
384                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos + 8,
385                                                                      pMB->mvs[2].x, pMB->mvs[2].y, stride,  rounding);
386                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8,
387                                                                      pMB->mvs[3].x, pMB->mvs[3].y, stride,  rounding);
388                    }
389    
390                    interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
391                                                              uv_dx, uv_dy, stride2, rounding);
392                    interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
393                                                              uv_dx, uv_dy, stride2, rounding);
394                    stop_comp_timer();
395    
396            } else {
397                  int sum;                  int sum;
398    
399                    if(dec->quarterpel)
400                            sum = (pMB->mvs[0].x / 2) + (pMB->mvs[1].x / 2) + (pMB->mvs[2].x / 2) + (pMB->mvs[3].x / 2);
401                    else
402                  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;
                 uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );  
403    
404                    uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
405    
406                    if(dec->quarterpel)
407                            sum = (pMB->mvs[0].y / 2) + (pMB->mvs[1].y / 2) + (pMB->mvs[2].y / 2) + (pMB->mvs[3].y / 2);
408                    else
409                  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;
410                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );  
411          }                  uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
412    
413          start_timer();          start_timer();
414          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) {
415          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);                          interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
416          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);                                                                            dec->refh.y + 128, 16*x_pos, 16*y_pos,
417          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);
418          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, dec->refh.y, dec->refh.y + 64,
419          interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, rounding);                                                                            dec->refh.y + 128, 16*x_pos + 8, 16*y_pos,
420                                                                              pMB->mvs[1].x, pMB->mvs[1].y, stride,  rounding);
421                            interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
422                                                                              dec->refh.y + 128, 16*x_pos, 16*y_pos + 8,
423                                                                              pMB->mvs[2].x, pMB->mvs[2].y, stride,  rounding);
424                            interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
425                                                                              dec->refh.y + 128, 16*x_pos + 8, 16*y_pos + 8,
426                                                                              pMB->mvs[3].x, pMB->mvs[3].y, stride,  rounding);
427                    }
428                    else {
429                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
430                                                                      pMB->mvs[0].x, pMB->mvs[0].y, stride,  rounding);
431                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos,
432                                                                      pMB->mvs[1].x, pMB->mvs[1].y, stride,  rounding);
433                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos + 8,
434                                                                      pMB->mvs[2].x, pMB->mvs[2].y, stride,  rounding);
435                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8,
436                                                                      pMB->mvs[3].x, pMB->mvs[3].y, stride,  rounding);
437                    }
438    
439                    interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
440                                                              uv_dx, uv_dy, stride2, rounding);
441                    interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
442                                                              uv_dx, uv_dy, stride2, rounding);
443          stop_comp_timer();          stop_comp_timer();
444            }
445    
446            for (i = 0; i < 6; i++) {
447                    int direction = dec->alternate_vertical_scan ? 2 : 0;
448    
         for (i = 0; i < 6; i++)  
         {  
449                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
450                  {                  {
451                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear
452    
453                          start_timer();                          start_timer();
454                          get_inter_block(bs, &block[i*64]);                          get_inter_block(bs, &block[i * 64], direction);
455                          stop_coding_timer();                          stop_coding_timer();
456    
457                          start_timer();                          start_timer();
458                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
                         {  
459                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                                  dequant_inter(&data[i*64], &block[i*64], iQuant);
460                          }                          } else {
                         else  
                         {  
461                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);
462                          }                          }
463                          stop_iquant_timer();                          stop_iquant_timer();
# Line 382  Line 468 
468                  }                  }
469          }          }
470    
471          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
472                  next_block = stride;                  next_block = stride;
473                  stride *= 2;                  stride *= 2;
474          }          }
# Line 405  Line 490 
490  }  }
491    
492    
493  void decoder_iframe(DECODER * dec, Bitstream * bs, int quant, int intra_dc_threshold)  void
494    decoder_iframe(DECODER * dec,
495                               Bitstream * bs,
496                               int quant,
497                               int intra_dc_threshold)
498  {  {
499            uint32_t bound;
500          uint32_t x, y;          uint32_t x, y;
501    
502          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];  
503    
504            for (y = 0; y < dec->mb_height; y++) {
505                    for (x = 0; x < dec->mb_width; x++) {
506                            MACROBLOCK *mb;
507                          uint32_t mcbpc;                          uint32_t mcbpc;
508                          uint32_t cbpc;                          uint32_t cbpc;
509                          uint32_t acpred_flag;                          uint32_t acpred_flag;
510                          uint32_t cbpy;                          uint32_t cbpy;
511                          uint32_t cbp;                          uint32_t cbp;
512    
513                            while (BitstreamShowBits(bs, 9) == 1)
514                                    BitstreamSkip(bs, 9);
515    
516                            if (check_resync_marker(bs, 0))
517                            {
518                                    bound = read_video_packet_header(bs, 0, &quant);
519                                    x = bound % dec->mb_width;
520                                    y = bound / dec->mb_width;
521                            }
522                            mb = &dec->mbs[y * dec->mb_width + x];
523    
524                            DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
525    
526                          mcbpc = get_mcbpc_intra(bs);                          mcbpc = get_mcbpc_intra(bs);
527                          mb->mode = mcbpc & 7;                          mb->mode = mcbpc & 7;
528                          cbpc = (mcbpc >> 4);                          cbpc = (mcbpc >> 4);
529    
530                          acpred_flag = BitstreamGetBit(bs);                          acpred_flag = BitstreamGetBit(bs);
531    
                         if (mb->mode == MODE_STUFFING)  
                         {  
                                 DEBUG("-- STUFFING ?");  
                                 continue;  
                         }  
   
532                          cbpy = get_cbpy(bs, 1);                          cbpy = get_cbpy(bs, 1);
533                          cbp = (cbpy << 2) | cbpc;                          cbp = (cbpy << 2) | cbpc;
534    
535                          if (mb->mode == MODE_INTRA_Q)                          if (mb->mode == MODE_INTRA_Q) {
                         {  
536                                  quant += dquant_table[BitstreamGetBits(bs,2)];                                  quant += dquant_table[BitstreamGetBits(bs,2)];
537                                  if (quant > 31)                                  if (quant > 31) {
                                 {  
538                                          quant = 31;                                          quant = 31;
539                                  }                                  } else if (quant < 1) {
                                 else if (quant < 1)  
                                 {  
540                                          quant = 1;                                          quant = 1;
541                                  }                                  }
542                          }                          }
543                          mb->quant = quant;                          mb->quant = quant;
544                            mb->mvs[0].x = mb->mvs[0].y =
545                            mb->mvs[1].x = mb->mvs[1].y =
546                            mb->mvs[2].x = mb->mvs[2].y =
547                            mb->mvs[3].x = mb->mvs[3].y =0;
548    
549                          if (dec->interlacing)                          if (dec->interlacing) {
                         {  
550                                  mb->field_dct = BitstreamGetBit(bs);                                  mb->field_dct = BitstreamGetBit(bs);
551                                  DEBUG1("deci: field_dct: ", mb->field_dct);                                  DEBUG1("deci: field_dct: ", mb->field_dct);
552                          }                          }
553    
554                          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,
555                                                            intra_dc_threshold, bound);
556                  }                  }
557                    if(dec->out_frm)
558                      output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,0,y,dec->mb_width);
559    
560          }          }
561    
562  }  }
563    
564    
565  void get_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, int k, VECTOR * mv, int fcode)  void
566    get_motion_vector(DECODER * dec,
567                                      Bitstream * bs,
568                                      int x,
569                                      int y,
570                                      int k,
571                                      VECTOR * mv,
572                                      int fcode,
573                                      const int bound)
574  {  {
575    
576          int scale_fac = 1 << (fcode - 1);          int scale_fac = 1 << (fcode - 1);
# Line 472  Line 578 
578          int low = ((-32) * scale_fac);          int low = ((-32) * scale_fac);
579          int range = (64 * scale_fac);          int range = (64 * scale_fac);
580    
581          VECTOR pmv[4];          VECTOR pmv;
         int32_t psad[4];  
   
582          int mv_x, mv_y;          int mv_x, mv_y;
         int pmv_x, pmv_y;  
   
   
         get_pmvdata(dec->mbs, x, y, dec->mb_width, k, pmv, psad);  
583    
584          pmv_x = pmv[0].x;          pmv = get_pmv2(dec->mbs, dec->mb_width, bound, x, y, k);
         pmv_y = pmv[0].y;  
585    
586          mv_x = get_mv(bs, fcode);          mv_x = get_mv(bs, fcode);
587          mv_y = get_mv(bs, fcode);          mv_y = get_mv(bs, fcode);
588    
589          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;  
590    
591          if (mv_x < low)          mv_x += pmv.x;
592          {          mv_y += pmv.y;
593    
594            if (mv_x < low) {
595                  mv_x += range;                  mv_x += range;
596          }          } else if (mv_x > high) {
         else if (mv_x > high)  
         {  
597                  mv_x -= range;                  mv_x -= range;
598          }          }
599    
600          if (mv_y < low)          if (mv_y < low) {
         {  
601                  mv_y += range;                  mv_y += range;
602          }          } else if (mv_y > high) {
         else if (mv_y > high)  
         {  
603                  mv_y -= range;                  mv_y -= range;
604          }          }
605    
# Line 514  Line 609 
609  }  }
610    
611    
612  void decoder_pframe(DECODER * dec, Bitstream * bs, int rounding, int quant, int fcode, int intra_dc_threshold)  void
613    decoder_pframe(DECODER * dec,
614                               Bitstream * bs,
615                               int rounding,
616                               int quant,
617                               int fcode,
618                               int intra_dc_threshold)
619  {  {
620    
621          uint32_t x, y;          uint32_t x, y;
622            uint32_t bound;
623            int cp_mb, st_mb;
624    
625          start_timer();          start_timer();
626          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,
627                                       dec->width, dec->height);
628          stop_edges_timer();          stop_edges_timer();
629    
630          for (y = 0; y < dec->mb_height; y++)          bound = 0;
631          {  
632                  for (x = 0; x < dec->mb_width; x++)          for (y = 0; y < dec->mb_height; y++) {
633                    cp_mb = st_mb = 0;
634                    for (x = 0; x < dec->mb_width; x++) {
635                            MACROBLOCK *mb;
636    
637                            // skip stuffing
638                            while (BitstreamShowBits(bs, 10) == 1)
639                                    BitstreamSkip(bs, 10);
640    
641                            if (check_resync_marker(bs, fcode - 1))
642                  {                  {
643                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                                  bound = read_video_packet_header(bs, fcode - 1, &quant);
644                                    x = bound % dec->mb_width;
645                                    y = bound / dec->mb_width;
646                            }
647                            mb = &dec->mbs[y * dec->mb_width + x];
648    
649                            DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
650    
651                          //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
652                          if (!(BitstreamGetBit(bs)))                     // not_coded                          if (!(BitstreamGetBit(bs)))                     // not_coded
# Line 539  Line 658 
658                                  uint32_t cbp;                                  uint32_t cbp;
659                                  uint32_t intra;                                  uint32_t intra;
660    
661                                    cp_mb++;
662                                  mcbpc = get_mcbpc_inter(bs);                                  mcbpc = get_mcbpc_inter(bs);
663                                  mb->mode = mcbpc & 7;                                  mb->mode = mcbpc & 7;
664                                  cbpc = (mcbpc >> 4);                                  cbpc = (mcbpc >> 4);
665    
666                                    DPRINTF(DPRINTF_MB, "mode %i", mb->mode);
667                                    DPRINTF(DPRINTF_MB, "cbpc %i", cbpc);
668                                  acpred_flag = 0;                                  acpred_flag = 0;
669    
670                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);
671    
672                                  if (intra)                                  if (intra) {
                                 {  
673                                          acpred_flag = BitstreamGetBit(bs);                                          acpred_flag = BitstreamGetBit(bs);
674                                  }                                  }
675    
                                 if (mb->mode == MODE_STUFFING)  
                                 {  
                                         DEBUG("-- STUFFING ?");  
                                         continue;  
                                 }  
   
676                                  cbpy = get_cbpy(bs, intra);                                  cbpy = get_cbpy(bs, intra);
677                                    DPRINTF(DPRINTF_MB, "cbpy %i", cbpy);
678    
679                                  cbp = (cbpy << 2) | cbpc;                                  cbp = (cbpy << 2) | cbpc;
680    
681                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q)                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q) {
682                                  {                                          int dquant = dquant_table[BitstreamGetBits(bs, 2)];
683                                          quant += dquant_table[BitstreamGetBits(bs,2)];                                          DPRINTF(DPRINTF_MB, "dquant %i", dquant);
684                                          if (quant > 31)                                          quant += dquant;
685                                          {                                          if (quant > 31) {
686                                                  quant = 31;                                                  quant = 31;
687                                          }                                          } else if (quant < 1) {
                                         else if (mb->quant < 1)  
                                         {  
688                                                  quant = 1;                                                  quant = 1;
689                                          }                                          }
690                                            DPRINTF(DPRINTF_MB, "quant %i", quant);
691                                  }                                  }
692                                  mb->quant = quant;                                  mb->quant = quant;
693    
694                                  if (dec->interlacing)                                  if (dec->interlacing) {
695                                  {                                          if (cbp || intra) {
696                                          mb->field_dct = BitstreamGetBit(bs);                                          mb->field_dct = BitstreamGetBit(bs);
697                                          DEBUG1("decp: field_dct: ", mb->field_dct);                                          DEBUG1("decp: field_dct: ", mb->field_dct);
698                                            }
699    
700                                          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
                                         {  
701                                                  mb->field_pred = BitstreamGetBit(bs);                                                  mb->field_pred = BitstreamGetBit(bs);
702                                                  DEBUG1("decp: field_pred: ", mb->field_pred);                                                  DEBUG1("decp: field_pred: ", mb->field_pred);
703    
704                                                  if (mb->field_pred)                                                  if (mb->field_pred) {
                                                 {  
705                                                          mb->field_for_top = BitstreamGetBit(bs);                                                          mb->field_for_top = BitstreamGetBit(bs);
706                                                          DEBUG1("decp: field_for_top: ", mb->field_for_top);                                                          DEBUG1("decp: field_for_top: ", mb->field_for_top);
707                                                          mb->field_for_bot = BitstreamGetBit(bs);                                                          mb->field_for_bot = BitstreamGetBit(bs);
# Line 594  Line 710 
710                                          }                                          }
711                                  }                                  }
712    
713                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
714                                  {                                          if (dec->interlacing && mb->field_pred) {
715                                          if (dec->interlacing && mb->field_pred)                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
716                                          {                                                                                    fcode, bound);
717                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1],
718                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1], fcode);                                                                                    fcode, bound);
719                                          }                                          } else {
720                                          else                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
721                                          {                                                                                    fcode, bound);
722                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);                                                  mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
723                                                  mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                          mb->mvs[0].x;
724                                                  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 =
725                                          }                                                          mb->mvs[0].y;
726                                  }                                          }
727                                  else if (mb->mode == MODE_INTER4V /* || mb->mode == MODE_INTER4V_Q */)                                  } else if (mb->mode == MODE_INTER4V ) {
728                                  {  
729                                          get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);                                          get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode, bound);
730                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode);                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode, bound);
731                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode);                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode, bound);
732                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode);                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode, bound);
733                                  }                                  } else                  // MODE_INTRA, MODE_INTRA_Q
734                                  else  // MODE_INTRA, MODE_INTRA_Q                                  {
735                                  {                                          mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
736                                          mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;                                                  0;
737                                          mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;                                          mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
738                                          decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant, intra_dc_threshold);                                                  0;
739                                            decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
740                                                                            intra_dc_threshold, bound);
741                                          continue;                                          continue;
742                                  }                                  }
743    
744                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant, rounding);                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant,
745                          }                                                                  rounding);
746                          else    // not coded                          } else                          // not coded
747                          {                          {
   
748                                  mb->mode = MODE_NOT_CODED;                                  mb->mode = MODE_NOT_CODED;
749                                  mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;                                  mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
750                                  mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;                                  mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;
# Line 636  Line 753 
753    
754                                  start_timer();                                  start_timer();
755    
756                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),                                  transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
757                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x),                                                                   (16 * x),
758                                                   dec->edged_width);                                                                   dec->refn[0].y + (16 * y) * dec->edged_width +
759                                                                     (16 * x), dec->edged_width);
760                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x+8),  
761                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x+8),                                  transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
762                                                   dec->edged_width);                                                                   (16 * x + 8),
763                                                                     dec->refn[0].y + (16 * y) * dec->edged_width +
764                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x),                                                                   (16 * x + 8), dec->edged_width);
765                                                   dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x),  
766                                                   dec->edged_width);                                  transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
767                                                                     (16 * x),
768                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x+8),                                                                   dec->refn[0].y + (16 * y +
769                                                   dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x+8),                                                                                                     8) * dec->edged_width +
770                                                   dec->edged_width);                                                                   (16 * x), dec->edged_width);
771    
772                                  transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),                                  transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
773                                                   dec->refn[0].u + (8*y)*dec->edged_width/2 + (8*x),                                                                   (16 * x + 8),
774                                                                     dec->refn[0].y + (16 * y +
775                                                                                                       8) * dec->edged_width +
776                                                                     (16 * x + 8), dec->edged_width);
777    
778                                    transfer8x8_copy(dec->cur.u + (8 * y) * dec->edged_width / 2 +
779                                                                     (8 * x),
780                                                                     dec->refn[0].u +
781                                                                     (8 * y) * dec->edged_width / 2 + (8 * x),
782                                                   dec->edged_width/2);                                                   dec->edged_width/2);
783    
784                                  transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),                                  transfer8x8_copy(dec->cur.v + (8 * y) * dec->edged_width / 2 +
785                                                   dec->refn[0].v + (8*y)*dec->edged_width/2 + (8*x),                                                                   (8 * x),
786                                                                     dec->refn[0].v +
787                                                                     (8 * y) * dec->edged_width / 2 + (8 * x),
788                                                   dec->edged_width/2);                                                   dec->edged_width/2);
   
789                                  stop_transfer_timer();                                  stop_transfer_timer();
790                                    if(dec->out_frm && cp_mb > 0) {
791                                      output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
792                                      cp_mb = 0;
793                                    }
794                                    st_mb = x+1;
795                          }                          }
796                  }                  }
797                    if(dec->out_frm && cp_mb > 0)
798                      output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
799          }          }
800  }  }
801    
802    
803  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
804  // decode B-frame motion vector  // decode B-frame motion vector
805  void get_b_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, VECTOR * mv, int fcode, const VECTOR pmv)  void
806    get_b_motion_vector(DECODER * dec,
807                                            Bitstream * bs,
808                                            int x,
809                                            int y,
810                                            VECTOR * mv,
811                                            int fcode,
812                                            const VECTOR pmv)
813  {  {
814          int scale_fac = 1 << (fcode - 1);          int scale_fac = 1 << (fcode - 1);
815          int high = (32 * scale_fac) - 1;          int high = (32 * scale_fac) - 1;
# Line 688  Line 828 
828          mv_x += pmv_x;          mv_x += pmv_x;
829          mv_y += pmv_y;          mv_y += pmv_y;
830    
831          if (mv_x < low)          if (mv_x < low) {
         {  
832                  mv_x += range;                  mv_x += range;
833          }          } else if (mv_x > high) {
         else if (mv_x > high)  
         {  
834                  mv_x -= range;                  mv_x -= range;
835          }          }
836    
837          if (mv_y < low)          if (mv_y < low) {
         {  
838                  mv_y += range;                  mv_y += range;
839          }          } else if (mv_y > high) {
         else if (mv_y > high)  
         {  
840                  mv_y -= range;                  mv_y -= range;
841          }          }
842    
843          mv->x = mv_x;          mv->x = mv_x;
844          mv->y = mv_y;          mv->y = mv_y;
   
845  }  }
846    
847    
848  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
849  // decode an B-frame forward & backward inter macroblock  // decode an B-frame forward & backward inter macroblock
850  void decoder_bf_mbinter(DECODER * dec,  void
851    decoder_bf_mbinter(DECODER * dec,
852                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
853                       const uint32_t x_pos,                       const uint32_t x_pos,
854                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 739  Line 873 
873          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
874          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
875    
876          if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))  
877          {          if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
878                  uv_dx = pMB->mvs[0].x;                  uv_dx = pMB->mvs[0].x;
879                  uv_dy = pMB->mvs[0].y;                  uv_dy = pMB->mvs[0].y;
880    
881                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
882                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
883          }          } else {
         else  
         {  
884                  int sum;                  int sum;
885    
886                  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;
887                  uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dx =
888                            (sum ==
889                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
890                                                                      (ABS(sum) / 16) * 2));
891    
892                  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;
893                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dy =
894                            (sum ==
895                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
896                                                                      (ABS(sum) / 16) * 2));
897          }          }
898    
899          start_timer();          start_timer();
900          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,
901          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);
902          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,
903          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);
904          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,
905          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,
906                                                      0);
907            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos + 8,
908                                                      16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
909                                                      0);
910            interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8 * x_pos, 8 * y_pos,
911                                                      uv_dx, uv_dy, stride2, 0);
912            interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8 * x_pos, 8 * y_pos,
913                                                      uv_dx, uv_dy, stride2, 0);
914          stop_comp_timer();          stop_comp_timer();
915    
916          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
917          {                  int direction = dec->alternate_vertical_scan ? 2 : 0;
918    
919                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
920                  {                  {
921                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear
922    
923                          start_timer();                          start_timer();
924                          get_inter_block(bs, &block[i*64]);                          get_inter_block(bs, &block[i * 64], direction);
925                          stop_coding_timer();                          stop_coding_timer();
926    
927                          start_timer();                          start_timer();
928                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
                         {  
929                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                                  dequant_inter(&data[i*64], &block[i*64], iQuant);
930                          }                          } else {
                         else  
                         {  
931                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);
932                          }                          }
933                          stop_iquant_timer();                          stop_iquant_timer();
# Line 793  Line 938 
938                  }                  }
939          }          }
940    
941          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
942                  next_block = stride;                  next_block = stride;
943                  stride *= 2;                  stride *= 2;
944          }          }
# Line 815  Line 959 
959          stop_transfer_timer();          stop_transfer_timer();
960  }  }
961    
   
962  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
963  // decode an B-frame direct &  inter macroblock  // decode an B-frame direct &  inter macroblock
964  void decoder_bf_interpolate_mbinter(DECODER * dec,  void
965                           IMAGE forward, IMAGE backward,  decoder_bf_interpolate_mbinter(DECODER * dec,
966                                                               IMAGE forward,
967                                                               IMAGE backward,
968                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
969                       const uint32_t x_pos,                       const uint32_t x_pos,
970                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 835  Line 980 
980          uint32_t iQuant = pMB->quant;          uint32_t iQuant = pMB->quant;
981          int uv_dx, uv_dy;          int uv_dx, uv_dy;
982          int b_uv_dx, b_uv_dy;          int b_uv_dx, b_uv_dy;
983            uint32_t i;
984            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
985        const uint32_t cbp = pMB->cbp;
986    
987          if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
988          {          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
989            pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
990    
991    
992            if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
993                  uv_dx = pMB->mvs[0].x;                  uv_dx = pMB->mvs[0].x;
994                  uv_dy = pMB->mvs[0].y;                  uv_dy = pMB->mvs[0].y;
995    
# Line 849  Line 1001 
1001    
1002                  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;
1003                  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;
1004          }          } else {
         else  
         {  
1005                  int sum;                  int sum;
1006    
1007                  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;
1008                  uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dx =
1009                            (sum ==
1010                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
1011                                                                      (ABS(sum) / 16) * 2));
1012    
1013                  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;
1014                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dy =
1015                            (sum ==
1016                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
1017                                                                      (ABS(sum) / 16) * 2));
1018    
1019                    sum =
1020                            pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x +
1021                            pMB->b_mvs[3].x;
1022                    b_uv_dx =
1023                            (sum ==
1024                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
1025                                                                      (ABS(sum) / 16) * 2));
1026    
1027                    sum =
1028                            pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y +
1029                            pMB->b_mvs[3].y;
1030                    b_uv_dy =
1031                            (sum ==
1032                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
1033                                                                      (ABS(sum) / 16) * 2));
1034            }
1035    
1036    
1037            start_timer();
1038            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos,
1039                                                      pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1040            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8, 16 * y_pos,
1041                                                      pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1042            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos + 8,
1043                                                      pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1044            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8,
1045                                                      16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
1046                                                      0);
1047            interpolate8x8_switch(dec->cur.u, forward.u, 8 * x_pos, 8 * y_pos, uv_dx,
1048                                                      uv_dy, stride2, 0);
1049            interpolate8x8_switch(dec->cur.v, forward.v, 8 * x_pos, 8 * y_pos, uv_dx,
1050                                                      uv_dy, stride2, 0);
1051    
1052    
1053            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos, 16 * y_pos,
1054                                                      pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1055            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1056                                                      16 * y_pos, pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride,
1057                                                      0);
1058            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos,
1059                                                      16 * y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y,
1060                                                      stride, 0);
1061            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1062                                                      16 * y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y,
1063                                                      stride, 0);
1064            interpolate8x8_switch(dec->refn[2].u, backward.u, 8 * x_pos, 8 * y_pos,
1065                                                      b_uv_dx, b_uv_dy, stride2, 0);
1066            interpolate8x8_switch(dec->refn[2].v, backward.v, 8 * x_pos, 8 * y_pos,
1067                                                      b_uv_dx, b_uv_dy, stride2, 0);
1068    
1069            interpolate8x8_avg2(dec->cur.y + (16 * y_pos * stride) + 16 * x_pos,
1070                                                    dec->cur.y + (16 * y_pos * stride) + 16 * x_pos,
1071                                                    dec->refn[2].y + (16 * y_pos * stride) + 16 * x_pos,
1072                                                    stride, 0);
1073    
1074            interpolate8x8_avg2(dec->cur.y + (16 * y_pos * stride) + 16 * x_pos + 8,
1075                                                    dec->cur.y + (16 * y_pos * stride) + 16 * x_pos + 8,
1076                                                    dec->refn[2].y + (16 * y_pos * stride) + 16 * x_pos + 8,
1077                                                    stride, 0);
1078    
1079            interpolate8x8_avg2(dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1080                                                    dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1081                                                    dec->refn[2].y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1082                                                    stride, 0);
1083    
1084            interpolate8x8_avg2(dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1085                                                    dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1086                                                    dec->refn[2].y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1087                                                    stride, 0);
1088    
1089            interpolate8x8_avg2(dec->cur.u + (8 * y_pos * stride2) + 8 * x_pos,
1090                                                    dec->cur.u + (8 * y_pos * stride2) + 8 * x_pos,
1091                                                    dec->refn[2].u + (8 * y_pos * stride2) + 8 * x_pos,
1092                                                    stride2, 0);
1093    
1094            interpolate8x8_avg2(dec->cur.v + (8 * y_pos * stride2) + 8 * x_pos,
1095                                                    dec->cur.v + (8 * y_pos * stride2) + 8 * x_pos,
1096                                                    dec->refn[2].v + (8 * y_pos * stride2) + 8 * x_pos,
1097                                                    stride2, 0);
1098    
1099                  sum = pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x + pMB->b_mvs[3].x;          stop_comp_timer();
                 b_uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );  
1100    
1101                  sum = pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y + pMB->b_mvs[3].y;          for (i = 0; i < 6; i++) {
1102                  b_uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  int direction = dec->alternate_vertical_scan ? 2 : 0;
         }  
1103    
1104                    if (cbp & (1 << (5 - i)))       // coded
1105                    {
1106                            memset(&block[i * 64], 0, 64 * sizeof(int16_t));        // clear
1107    
1108          start_timer();          start_timer();
1109                            get_inter_block(bs, &block[i * 64], direction);
1110                            stop_coding_timer();
1111    
1112          interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos,     16*y_pos    , pMB->mvs[0].x, pMB->mvs[0].y, stride,  0);                          start_timer();
1113          interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos + 8, 16*y_pos    , pMB->mvs[1].x, pMB->mvs[1].y, stride,  0);                          if (dec->quant_type == 0) {
1114          interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos,     16*y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,  0);                                  dequant_inter(&data[i * 64], &block[i * 64], iQuant);
1115          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);                          } else {
1116          interpolate8x8_switch(dec->cur.u, forward.u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);                                  dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
1117          interpolate8x8_switch(dec->cur.v, forward.v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);                          }
1118                            stop_iquant_timer();
         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);  
         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);  
         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);  
         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);  
         interpolate8x8_switch(dec->refn[2].u, backward.u, 8*x_pos,      8*y_pos,      b_uv_dx,         b_uv_dy,         stride2, 0);  
         interpolate8x8_switch(dec->refn[2].v, backward.v, 8*x_pos,      8*y_pos,      b_uv_dx,         b_uv_dy,         stride2, 0);  
   
         interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos,     16*y_pos    , stride);  
         interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos + 8, 16*y_pos    , stride);  
         interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos,     16*y_pos + 8, stride);  
         interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos + 8, 16*y_pos + 8, stride);  
         interpolate8x8_c(dec->cur.u, dec->refn[2].u, 8*x_pos,      8*y_pos,      stride2);  
         interpolate8x8_c(dec->cur.v, dec->refn[2].v, 8*x_pos,      8*y_pos,      stride2);  
   
 /*  
         interpolate8x8_c(dec->cur.y, backward.y, 16*x_pos,     16*y_pos    , pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride);  
         interpolate8x8_c(dec->cur.y, backward.y, 16*x_pos + 8, 16*y_pos    , pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride);  
         interpolate8x8_c(dec->cur.y, backward.y, 16*x_pos,     16*y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y, stride);  
         interpolate8x8_c(dec->cur.y, backward.y, 16*x_pos + 8, 16*y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y, stride);  
         interpolate8x8_c(dec->cur.u, backward.u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2);  
         interpolate8x8_c(dec->cur.v, backward.v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2);  
 */  
         stop_comp_timer();  
1119    
1120                            start_timer();
1121                            idct(&data[i * 64]);
1122                            stop_idct_timer();
1123                    }
1124            }
1125    
1126            if (dec->interlacing && pMB->field_dct) {
1127                    next_block = stride;
1128                    stride *= 2;
1129            }
1130    
1131            start_timer();
1132            if (cbp & 32)
1133                    transfer_16to8add(pY_Cur, &data[0 * 64], stride);
1134            if (cbp & 16)
1135                    transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
1136            if (cbp & 8)
1137                    transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
1138            if (cbp & 4)
1139                    transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
1140            if (cbp & 2)
1141                    transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
1142            if (cbp & 1)
1143                    transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
1144            stop_transfer_timer();
1145  }  }
1146    
1147    
1148  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
1149  // for decode B-frame dbquant  // for decode B-frame dbquant
1150  int32_t __inline get_dbquant(Bitstream * bs)  int32_t __inline
1151    get_dbquant(Bitstream * bs)
1152  {  {
1153          if (!BitstreamGetBit(bs))               // '0'          if (!BitstreamGetBit(bs))               // '0'
1154                  return(0);                  return(0);
# Line 923  Line 1165 
1165  // 01       1  // 01       1
1166  // 001      2  // 001      2
1167  // 0001     3  // 0001     3
1168  int32_t __inline get_mbtype(Bitstream * bs)  int32_t __inline
1169    get_mbtype(Bitstream * bs)
1170  {  {
1171          int32_t mb_type=0;          int32_t mb_type;
1172          for(;mb_type<=3 && (!BitstreamGetBit(bs));mb_type++);  
1173            for (mb_type = 0; mb_type <= 3; mb_type++) {
1174                    if (BitstreamGetBit(bs))
1175                            break;
1176            }
1177    
1178          if (mb_type<=3)          if (mb_type<=3)
1179                  return(mb_type);                  return(mb_type);
1180          else          else
1181                  return(-1);                  return(-1);
1182  }  }
1183    
1184  void decoder_bframe(DECODER * dec, Bitstream * bs, int quant, int fcode_forward, int fcode_backward)  void
1185    decoder_bframe(DECODER * dec,
1186                               Bitstream * bs,
1187                               int quant,
1188                               int fcode_forward,
1189                               int fcode_backward)
1190  {  {
   
1191          uint32_t        x, y;          uint32_t        x, y;
1192          VECTOR          mv, zeromv;          VECTOR mv;
1193            const VECTOR zeromv = {0,0};
1194    #ifdef BFRAMES_DEC_DEBUG
1195            FILE *fp;
1196            static char first=0;
1197    #define BFRAME_DEBUG    if (!first && fp){ \
1198                    fprintf(fp,"Y=%3d   X=%3d   MB=%2d   CBP=%02X\n",y,x,mb->mb_type,mb->cbp); \
1199            }
1200    #endif
1201    
1202          start_timer();          start_timer();
1203          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,
1204          //image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing);                                     dec->width, dec->height);
1205            image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height,
1206                                       dec->width, dec->height);
1207          stop_edges_timer();          stop_edges_timer();
1208    
1209          for (y = 0; y < dec->mb_height; y++)  #ifdef BFRAMES_DEC_DEBUG
1210          {          if (!first){
1211                  for (x = 0; x < dec->mb_width; x++)                  fp=fopen("C:\\XVIDDBG.TXT","w");
1212                  {          }
1213    #endif
1214    
1215            for (y = 0; y < dec->mb_height; y++) {
1216                    // Initialize Pred Motion Vector
1217                    dec->p_fmv = dec->p_bmv = zeromv;
1218                    for (x = 0; x < dec->mb_width; x++) {
1219                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
1220                          MACROBLOCK * last_mb = &dec->last_mbs[y*dec->mb_width + x];                          MACROBLOCK * last_mb = &dec->last_mbs[y*dec->mb_width + x];
1221    
1222                          // the last P_VOP is skip macroblock ?                          mv =
1223                            mb->b_mvs[0] = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] =
1224                            mb->mvs[0] = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = zeromv;
1225    
1226                            // the last P_VOP is skip macroblock ?
1227                          if (last_mb->mode == MODE_NOT_CODED){                          if (last_mb->mode == MODE_NOT_CODED){
1228                                  DEBUG("Skip MB in B-frame!");                                  //DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y);
1229                                    mb->cbp = 0;
1230                                  // copy macroblock directly from ref to cur  #ifdef BFRAMES_DEC_DEBUG
1231                                  start_timer();                                  mb->mb_type = MODE_NOT_CODED;
1232                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),          BFRAME_DEBUG
1233                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x),  #endif
1234                                                   dec->edged_width);                                  mb->mb_type = MODE_FORWARD;
1235                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x+8),                                  mb->quant = last_mb->quant;
1236                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x+8),                                  //mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1237                                                   dec->edged_width);                                  //mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
                                 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);  
                                 transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),  
                                                  dec->refn[0].u + (8*y)*dec->edged_width/2 + (8*x),  
                                                  dec->edged_width/2);  
                                 transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),  
                                                  dec->refn[0].v + (8*y)*dec->edged_width/2 + (8*x),  
                                                  dec->edged_width/2);  
                                 stop_transfer_timer();  
1238    
1239                                  DEBUG("skip MB in B-frame!");                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, mb->quant, 1);
1240                                  continue;                                  continue;
1241                          }                          }
1242    
1243                          if (!BitstreamGetBit(bs)){      // modb=='0'                          if (!BitstreamGetBit(bs)){      // modb=='0'
1244                                  uint8_t modb2=BitstreamGetBit(bs);                                  const uint8_t modb2 = BitstreamGetBit(bs);
1245    
1246                                  mb->mb_type = get_mbtype(bs);                                  mb->mb_type = get_mbtype(bs);
1247    
# Line 995  Line 1253 
1253                                  if (mb->mb_type && mb->cbp){                                  if (mb->mb_type && mb->cbp){
1254                                          quant += get_dbquant(bs);                                          quant += get_dbquant(bs);
1255    
1256                                          if (quant > 31)                                          if (quant > 31) {
                                         {  
1257                                                  quant = 31;                                                  quant = 31;
1258                                          }                                          } else if (quant < 1) {
                                         else if (mb->quant < 1)  
                                         {  
1259                                                  quant = 1;                                                  quant = 1;
1260                                          }                                          }
1261                                  }                                  }
                                 mb->quant = quant;  
1262                          } else {                          } else {
1263                                  mb->mb_type = MODE_DIRECT_NONE_MV;                                  mb->mb_type = MODE_DIRECT_NONE_MV;
1264                                  mb->cbp = 0;                                  mb->cbp = 0;
1265                          }                          }
1266    
1267                          mb->mb_type = MODE_INTER;                          mb->quant = quant;
1268                          mb->mvs[0].x=mb->mvs[0].y=zeromv.x = zeromv.y = mv.x = mv.y = 0;                          mb->mode = MODE_INTER4V;
1269                          //DEBUG1("Switch bm_type=",mb->mb_type);                          //DEBUG1("Switch bm_type=",mb->mb_type);
1270    
1271                          switch(mb->mb_type)  #ifdef BFRAMES_DEC_DEBUG
1272                          {          BFRAME_DEBUG
1273    #endif
1274    
1275                            switch (mb->mb_type) {
1276                          case MODE_DIRECT:                          case MODE_DIRECT:
1277                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], 1, zeromv);                                  get_b_motion_vector(dec, bs, x, y, &mv, 1, zeromv);
1278    
1279                          case MODE_DIRECT_NONE_MV:                          case MODE_DIRECT_NONE_MV:
1280                                  {       // Because this file is a C file not C++ so I use '{' to define var                                  {
1281                                          const int32_t   TRB=dec->time_pp-dec->time_bp,                                          const int64_t TRB = dec->time_pp - dec->time_bp, TRD = dec->time_pp;
                                                                         TRD=dec->time_pp;  
1282                                          int i;                                          int i;
                                         for(i=0;i<4;i++){  
                                                 mb->mvs[i].x    = (TRB * last_mb->mvs[i].x)/TRD+mb->mvs[0].x;  
                                                 mb->b_mvs[i].x  = (mb->mvs[0].x==0)?((TRB-TRD)*last_mb->mvs[i].x)/TRD:mb->mvs[i].x-last_mb->mvs[i].x;  
                                                 mb->mvs[i].y    = (TRB * last_mb->mvs[i].y)/TRD+mb->mvs[0].y;  
                                                 mb->b_mvs[i].y  = (mb->mvs[0].y==0)?((TRB-TRD)*last_mb->mvs[i].y)/TRD:mb->mvs[i].y-last_mb->mvs[i].y;  
1283    
1284                                                  //DEBUG2("B-direct MVF=",mb->mvs[i].x,mb->mvs[i].y);                                          for (i = 0; i < 4; i++) {
1285                                                  //DEBUG2("B-direct MVB=",mb->b_mvs[i].x,mb->b_mvs[i].y);                                                  mb->mvs[i].x = (int32_t) ((TRB * last_mb->mvs[i].x)
1286                                                                          / TRD + mv.x);
1287                                                    mb->b_mvs[i].x = (int32_t) ((mv.x == 0)
1288                                                                                    ? ((TRB - TRD) * last_mb->mvs[i].x)
1289                                                                                      / TRD
1290                                                                                    : mb->mvs[i].x - last_mb->mvs[i].x);
1291                                                    mb->mvs[i].y = (int32_t) ((TRB * last_mb->mvs[i].y)
1292                                                                          / TRD + mv.y);
1293                                                    mb->b_mvs[i].y = (int32_t) ((mv.y == 0)
1294                                                                                    ? ((TRB - TRD) * last_mb->mvs[i].y)
1295                                                                                      / TRD
1296                                                                                : mb->mvs[i].y - last_mb->mvs[i].y);
1297                                          }                                          }
1298                                          //DEBUG("B-frame Direct!\n");                                          //DEBUG("B-frame Direct!\n");
1299                                  }                                  }
1300                                  mb->mode = MODE_INTER4V;                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1301                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], mb, x, y, bs);                                                                                             mb, x, y, bs);
1302                                  break;                                  break;
1303    
1304                          case MODE_INTERPOLATE:                          case MODE_INTERPOLATE:
1305                                  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,
1306                                  dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                                          dec->p_fmv);
1307                                  dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                  dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] =     mb->mvs[0];
1308    
1309                                  get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0], fcode_backward, dec->p_bmv);                                  get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0],
1310                                  dec->p_bmv.x = mb->b_mvs[1].x = mb->b_mvs[2].x = mb->b_mvs[3].x = mb->b_mvs[0].x;                                                                          fcode_backward, dec->p_bmv);
1311                                  dec->p_bmv.y = mb->b_mvs[1].y = mb->b_mvs[2].y = mb->b_mvs[3].y = mb->b_mvs[0].y;                                  dec->p_bmv = mb->b_mvs[1] = mb->b_mvs[2] =
1312                                            mb->b_mvs[3] = mb->b_mvs[0];
1313    
1314                                  mb->mode = MODE_INTER4V;                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1315                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], mb, x, y, bs);                                                                                             mb, x, y, bs);
1316                                  //DEBUG("B-frame Bidir!\n");                                  //DEBUG("B-frame Bidir!\n");
1317                                  break;                                  break;
1318    
1319                          case MODE_BACKWARD:                          case MODE_BACKWARD:
1320                                  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,
1321                                  dec->p_bmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                                          dec->p_bmv);
1322                                  dec->p_bmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                  dec->p_bmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] =     mb->mvs[0];
1323    
1324                                    mb->mode = MODE_INTER;
1325                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);
1326                                  //DEBUG("B-frame Backward!\n");                                  //DEBUG("B-frame Backward!\n");
1327                                  break;                                  break;
1328    
1329                          case MODE_FORWARD:                          case MODE_FORWARD:
1330                                  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,
1331                                  dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                                          dec->p_fmv);
1332                                  dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                  dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] =     mb->mvs[0];
1333    
1334                                    mb->mode = MODE_INTER;
1335                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1336                                  DEBUG("B-frame Forward!\n");                                  //DEBUG("B-frame Forward!\n");
1337                                  break;                                  break;
1338    
1339                          default:                          default:
# Line 1075  Line 1342 
1342    
1343                  }       // end of FOR                  }       // end of FOR
1344          }          }
1345    #ifdef BFRAMES_DEC_DEBUG
1346            if (!first){
1347                    first=1;
1348                    if (fp)
1349                            fclose(fp);
1350            }
1351    #endif
1352  }  }
1353    
1354  // swap two MACROBLOCK array  // swap two MACROBLOCK array
1355  void mb_swap(MACROBLOCK *mb1, MACROBLOCK *mb2)  void
1356    mb_swap(MACROBLOCK ** mb1,
1357                    MACROBLOCK ** mb2)
1358  {  {
1359          MACROBLOCK *temp=mb1;          MACROBLOCK *temp = *mb1;
1360          mb1=mb2;  
1361          mb2=temp;          *mb1 = *mb2;
1362            *mb2 = temp;
1363  }  }
1364    
1365  int decoder_decode(DECODER * dec, XVID_DEC_FRAME * frame)  int
1366    decoder_decode(DECODER * dec,
1367                               XVID_DEC_FRAME * frame)
1368  {  {
1369    
1370          Bitstream bs;          Bitstream bs;
# Line 1098  Line 1377 
1377    
1378          start_global_timer();          start_global_timer();
1379    
1380            dec->out_frm = (frame->colorspace == XVID_CSP_EXTERN) ? frame->image : NULL;
1381    
1382          BitstreamInit(&bs, frame->bitstream, frame->length);          BitstreamInit(&bs, frame->bitstream, frame->length);
1383    
1384            if(BitstreamShowBits(&bs, 8) == 0x7f)
1385                    return XVID_ERR_OK;
1386    
1387          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
1388          // for support B-frame to reference last 2 frame          // for support B-frame to reference last 2 frame
1389          dec->frames ++;          dec->frames ++;
1390          vop_type=BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward, &fcode_backward, &intra_dc_threshold);          vop_type =
1391                    BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward,
1392                                                             &fcode_backward, &intra_dc_threshold);
1393    
1394          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
1395          switch (vop_type)  
1396          {          switch (vop_type) {
1397          case P_VOP :          case P_VOP :
1398                  decoder_pframe(dec, &bs, rounding, quant, fcode_forward, intra_dc_threshold);                  decoder_pframe(dec, &bs, rounding, quant, fcode_forward,
1399                                               intra_dc_threshold);
1400    #ifdef BFRAMES_DEC
1401                  DEBUG1("P_VOP  Time=",dec->time);                  DEBUG1("P_VOP  Time=",dec->time);
1402    #endif
1403                  break;                  break;
1404    
1405          case I_VOP :          case I_VOP :
1406                  decoder_iframe(dec, &bs, quant, intra_dc_threshold);                  decoder_iframe(dec, &bs, quant, intra_dc_threshold);
1407    #ifdef BFRAMES_DEC
1408                  DEBUG1("I_VOP  Time=",dec->time);                  DEBUG1("I_VOP  Time=",dec->time);
1409    #endif
1410                  break;                  break;
1411    
1412          case B_VOP :          case B_VOP :
1413    #ifdef BFRAMES_DEC
1414                  if (dec->time_pp > dec->time_bp){                  if (dec->time_pp > dec->time_bp){
                         decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);  
1415                          DEBUG1("B_VOP  Time=",dec->time);                          DEBUG1("B_VOP  Time=",dec->time);
1416                            decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);
1417                  } else {                  } else {
1418                          DEBUG("broken B-frame!");                          DEBUG("broken B-frame!");
1419                  }                  }
1420    #else
1421                    image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1422    #endif
1423                  break;                  break;
1424    
1425          case N_VOP :    // vop not coded          case N_VOP :    // vop not coded
1426                    // when low_delay==0, N_VOP's should interpolate between the past and future frames
1427                    image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1428    #ifdef BFRAMES_DEC
1429                    DEBUG1("N_VOP  Time=", dec->time);
1430    #endif
1431                  break;                  break;
1432    
1433          default :          default :
1434                  return XVID_ERR_FAIL;                  return XVID_ERR_FAIL;
1435          }          }
1436    
1437    #ifdef BFRAMES_DEC_DEBUG
1438            if (frame->length != BitstreamPos(&bs) / 8){
1439                    DEBUG2("InLen/UseLen",frame->length, BitstreamPos(&bs) / 8);
1440            }
1441    #endif
1442          frame->length = BitstreamPos(&bs) / 8;          frame->length = BitstreamPos(&bs) / 8;
1443    
1444    
1445    #ifdef BFRAMES_DEC
1446            // test if no B_VOP
1447            if (dec->low_delay || dec->frames == 0) {
1448    #endif
1449            image_output(&dec->cur, dec->width, dec->height, dec->edged_width,
1450                                             frame->image, frame->stride, frame->colorspace);
1451    
1452    #ifdef BFRAMES_DEC
1453            } else {
1454          if (dec->frames >= 1){          if (dec->frames >= 1){
1455                  start_timer();                  start_timer();
1456                  if ((vop_type == I_VOP || vop_type == P_VOP))                          if ((vop_type == I_VOP || vop_type == P_VOP)) {
1457                  {                                  image_output(&dec->refn[0], dec->width, dec->height,
1458                          image_output(&dec->refn[1], dec->width, dec->height, dec->edged_width,                                                           dec->edged_width, frame->image, frame->stride,
1459                                       frame->image, frame->stride, frame->colorspace);                                                           frame->colorspace);
1460                  } else if (vop_type == B_VOP) {                  } else if (vop_type == B_VOP) {
1461                          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,                                  image_output(&dec->cur, dec->width, dec->height,
1462                                       frame->image, frame->stride, frame->colorspace);                                                           dec->edged_width, frame->image, frame->stride,
1463                                                             frame->colorspace);
1464                  }                  }
1465                  stop_conv_timer();                  stop_conv_timer();
1466          }          }
1467            }
1468    #endif
1469    
1470          if (vop_type==I_VOP || vop_type==P_VOP){          if (vop_type==I_VOP || vop_type==P_VOP){
1471                  image_swap(&dec->refn[0], &dec->refn[1]);                  image_swap(&dec->refn[0], &dec->refn[1]);
1472                  image_swap(&dec->cur, &dec->refn[0]);                  image_swap(&dec->cur, &dec->refn[0]);
1473    
1474                  // swap MACROBLOCK                  // swap MACROBLOCK
1475                  mb_swap(dec->mbs, dec->last_mbs);                  // the Divx will not set the low_delay flage some times
1476                    // so follow code will wrong to not swap at that time
1477                    // this will broken bitstream! so I'm change it,
1478                    // But that is not the best way! can anyone tell me how
1479                    // to do another way?
1480                    // 18-07-2002   MinChen<chenm001@163.com>
1481                    //if (!dec->low_delay && vop_type == P_VOP)
1482                    if (vop_type == P_VOP)
1483                            mb_swap(&dec->mbs, &dec->last_mbs);
1484          }          }
1485    
1486          emms();          emms();
# Line 1160  Line 1488 
1488          stop_global_timer();          stop_global_timer();
1489    
1490          return XVID_ERR_OK;          return XVID_ERR_OK;
   
1491  }  }

Legend:
Removed from v.1.14  
changed lines
  Added in v.1.37.2.6

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