[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.15, Mon May 6 03:51:43 2002 UTC revision 1.37.2.17, Mon Dec 9 10:47:05 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 64  Line 79 
79  #include "dct/fdct.h"  #include "dct/fdct.h"
80  #include "utils/mem_transfer.h"  #include "utils/mem_transfer.h"
81  #include "image/interpolate8x8.h"  #include "image/interpolate8x8.h"
82    #include "image/reduced.h"
83    
84  #include "bitstream/mbcoding.h"  #include "bitstream/mbcoding.h"
85  #include "prediction/mbprediction.h"  #include "prediction/mbprediction.h"
86  #include "utils/timer.h"  #include "utils/timer.h"
87  #include "utils/emms.h"  #include "utils/emms.h"
88    #include "motion/motion.h"
89    
90  #include "image/image.h"  #include "image/image.h"
91  #include "image/colorspace.h"  #include "image/colorspace.h"
92  #include "utils/mem_align.h"  #include "utils/mem_align.h"
93    
94  int decoder_create(XVID_DEC_PARAM * param)  int
95    decoder_resize(DECODER * dec)
96  {  {
97          DECODER * dec;          /* free existing */
98    
99          dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
100          if (dec == NULL)          image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
101          {          image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
102                  return XVID_ERR_MEMORY;          image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
103          }          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
         param->handle = dec;  
104    
105          dec->width = param->width;          if (dec->last_mbs)
106          dec->height = param->height;                  xvid_free(dec->last_mbs);
107            if (dec->mbs)
108                    xvid_free(dec->mbs);
109    
110            /* realloc */
111    
112          dec->mb_width = (dec->width + 15) / 16;          dec->mb_width = (dec->width + 15) / 16;
113          dec->mb_height = (dec->height + 15) / 16;          dec->mb_height = (dec->height + 15) / 16;
# Line 94  Line 115 
115          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;
116          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;
117    
118          if (image_create(&dec->cur, dec->edged_width, dec->edged_height))          if (image_create(&dec->cur, dec->edged_width, dec->edged_height)) {
         {  
119                  xvid_free(dec);                  xvid_free(dec);
120                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
121          }          }
122    
123          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height)) {
         {  
124                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
125                  xvid_free(dec);                  xvid_free(dec);
126                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
127          }          }
128    
129          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
130          // for support B-frame to reference last 2 frame          // for support B-frame to reference last 2 frame
131          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {
         {  
132                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
133                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
134                  xvid_free(dec);                  xvid_free(dec);
135                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
136          }          }
137          if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height)) {
         {  
138                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
139                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
140                  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 142 
142                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
143          }          }
144    
145          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)) {
         if (dec->mbs == NULL)  
         {  
146                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
147                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
148                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
# Line 134  Line 150 
150                  xvid_free(dec);                  xvid_free(dec);
151                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
152          }          }
153    
154            dec->mbs =
155                    xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
156                                            CACHE_LINE);
157            if (dec->mbs == NULL) {
158                    image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
159                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
160                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
161                    image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
162                    image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
163                    xvid_free(dec);
164                    return XVID_ERR_MEMORY;
165            }
166            memset(dec->mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
167    
168          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
169          // for skip MB flag          // for skip MB flag
170          dec->last_mbs = xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height, CACHE_LINE);          dec->last_mbs =
171          if (dec->last_mbs == NULL)                  xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
172          {                                          CACHE_LINE);
173            if (dec->last_mbs == NULL) {
174                  xvid_free(dec->mbs);                  xvid_free(dec->mbs);
175                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
176                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
177                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
178                  image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
179                    image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
180                  xvid_free(dec);                  xvid_free(dec);
181                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
182          }          }
183    
184            memset(dec->last_mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
185    
186            return XVID_ERR_OK;
187    }
188    
189    
190    int
191    decoder_create(XVID_DEC_PARAM * param)
192    {
193            DECODER *dec;
194    
195            dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);
196            if (dec == NULL) {
197                    return XVID_ERR_MEMORY;
198            }
199            memset(dec, 0, sizeof(DECODER));
200    
201            param->handle = dec;
202    
203            dec->width = param->width;
204            dec->height = param->height;
205    
206            image_null(&dec->cur);
207            image_null(&dec->refn[0]);
208            image_null(&dec->refn[1]);
209            image_null(&dec->refn[2]);
210            image_null(&dec->refh);
211    
212            dec->mbs = NULL;
213            dec->last_mbs = NULL;
214    
215          init_timer();          init_timer();
216    
217          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
218          // for support B-frame to save reference frame's time          // for support B-frame to save reference frame's time
219          dec->frames = -1;          dec->frames = -1;
220          dec->time = dec->time_base = dec->last_time_base = 0;          dec->time = dec->time_base = dec->last_time_base = 0;
221            dec->low_delay = 0;
222            dec->packed_mode = 0;
223    
224            dec->fixed_dimensions = (dec->width > 0 && dec->height > 0);
225    
226            if (dec->fixed_dimensions)
227                    return decoder_resize(dec);
228            else
229          return XVID_ERR_OK;          return XVID_ERR_OK;
230  }  }
231    
232    
233  int decoder_destroy(DECODER * dec)  int
234    decoder_destroy(DECODER * dec)
235  {  {
236          xvid_free(dec->last_mbs);          xvid_free(dec->last_mbs);
237          xvid_free(dec->mbs);          xvid_free(dec->mbs);
238          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
239          image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);          image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
240          image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);          image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
241            image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
242          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
243          xvid_free(dec);          xvid_free(dec);
244    
# Line 175  Line 248 
248    
249    
250    
251  static const int32_t dquant_table[4] =  static const int32_t dquant_table[4] = {
 {  
252          -1, -2, 1, 2          -1, -2, 1, 2
253  };  };
254    
# Line 185  Line 257 
257    
258  // decode an intra macroblock  // decode an intra macroblock
259    
260  void decoder_mbintra(DECODER * dec,  void
261    decoder_mbintra(DECODER * dec,
262                       MACROBLOCK * pMB,                       MACROBLOCK * pMB,
263                       const uint32_t x_pos,                       const uint32_t x_pos,
264                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 193  Line 266 
266                       const uint32_t cbp,                       const uint32_t cbp,
267                       Bitstream * bs,                       Bitstream * bs,
268                       const uint32_t quant,                       const uint32_t quant,
269                       const uint32_t intra_dc_threshold)                                  const uint32_t intra_dc_threshold,
270                                    const unsigned int bound,
271                                    const int reduced_resolution)
272  {  {
273    
274          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
# Line 206  Line 281 
281          uint32_t iQuant = pMB->quant;          uint32_t iQuant = pMB->quant;
282          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
283    
284            if (reduced_resolution) {
285                    pY_Cur = dec->cur.y + (y_pos << 5) * stride + (x_pos << 5);
286                    pU_Cur = dec->cur.u + (y_pos << 4) * stride2 + (x_pos << 4);
287                    pV_Cur = dec->cur.v + (y_pos << 4) * stride2 + (x_pos << 4);
288            }else{
289          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
290          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
291          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
292            }
293    
294          memset(block, 0, 6*64*sizeof(int16_t));         // clear          memset(block, 0, 6*64*sizeof(int16_t));         // clear
295    
296          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
         {  
297                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
298                  int16_t predictors[8];                  int16_t predictors[8];
299                  int start_coeff;                  int start_coeff;
300    
301                  start_timer();                  start_timer();
302                  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],
303                  if (!acpred_flag)                                           iQuant, iDcScaler, predictors, bound);
304                  {                  if (!acpred_flag) {
305                          pMB->acpred_directions[i] = 0;                          pMB->acpred_directions[i] = 0;
306                  }                  }
307                  stop_prediction_timer();                  stop_prediction_timer();
308    
309                  if (quant < intra_dc_threshold)                  if (quant < intra_dc_threshold) {
                 {  
310                          int dc_size;                          int dc_size;
311                          int dc_dif;                          int dc_dif;
312    
313                          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);
314                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;
315    
316                          if (dc_size > 8)                          if (dc_size > 8) {
                         {  
317                                  BitstreamSkip(bs, 1);           // marker                                  BitstreamSkip(bs, 1);           // marker
318                          }                          }
319    
320                          block[i*64 + 0] = dc_dif;                          block[i*64 + 0] = dc_dif;
321                          start_coeff = 1;                          start_coeff = 1;
322                  }  
323                  else                          DPRINTF(DPRINTF_COEFF,"block[0] %i", dc_dif);
324                  {                  } else {
325                          start_coeff = 0;                          start_coeff = 0;
326                  }                  }
327    
328                  start_timer();                  start_timer();
329                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
330                  {                  {
331                          get_intra_block(bs, &block[i*64], pMB->acpred_directions[i], start_coeff);                          int direction = dec->alternate_vertical_scan ?
332                                    2 : pMB->acpred_directions[i];
333    
334                            get_intra_block(bs, &block[i * 64], direction, start_coeff);
335                  }                  }
336                  stop_coding_timer();                  stop_coding_timer();
337    
# Line 259  Line 340 
340                  stop_prediction_timer();                  stop_prediction_timer();
341    
342                  start_timer();                  start_timer();
343                  if (dec->quant_type == 0)                  if (dec->quant_type == 0) {
                 {  
344                          dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);                          dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
345                  }                  } else {
                 else  
                 {  
346                          dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);                          dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
347                  }                  }
348                  stop_iquant_timer();                  stop_iquant_timer();
# Line 272  Line 350 
350                  start_timer();                  start_timer();
351                  idct(&data[i*64]);                  idct(&data[i*64]);
352                  stop_idct_timer();                  stop_idct_timer();
353    
354          }          }
355    
356          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
357                  next_block = stride;                  next_block = stride;
358                  stride *= 2;                  stride *= 2;
359          }          }
360    
361          start_timer();          start_timer();
362    
363            if (reduced_resolution)
364            {
365                    next_block*=2;
366                    copy_upsampled_8x8_16to8(pY_Cur, &data[0 * 64], stride);
367                    copy_upsampled_8x8_16to8(pY_Cur + 16, &data[1 * 64], stride);
368                    copy_upsampled_8x8_16to8(pY_Cur + next_block, &data[2 * 64], stride);
369                    copy_upsampled_8x8_16to8(pY_Cur + 16 + next_block, &data[3 * 64], stride);
370                    copy_upsampled_8x8_16to8(pU_Cur, &data[4 * 64], stride2);
371                    copy_upsampled_8x8_16to8(pV_Cur, &data[5 * 64], stride2);
372            }else{
373          transfer_16to8copy(pY_Cur,                  &data[0*64], stride);          transfer_16to8copy(pY_Cur,                  &data[0*64], stride);
374          transfer_16to8copy(pY_Cur + 8,              &data[1*64], stride);          transfer_16to8copy(pY_Cur + 8,              &data[1*64], stride);
375          transfer_16to8copy(pY_Cur + next_block,     &data[2*64], stride);          transfer_16to8copy(pY_Cur + next_block,     &data[2*64], stride);
376          transfer_16to8copy(pY_Cur + 8 + next_block, &data[3*64], stride);          transfer_16to8copy(pY_Cur + 8 + next_block, &data[3*64], stride);
377          transfer_16to8copy(pU_Cur,                  &data[4*64], stride2);          transfer_16to8copy(pU_Cur,                  &data[4*64], stride2);
378          transfer_16to8copy(pV_Cur,                  &data[5*64], stride2);          transfer_16to8copy(pV_Cur,                  &data[5*64], stride2);
379            }
380          stop_transfer_timer();          stop_transfer_timer();
381  }  }
382    
# Line 296  Line 386 
386    
387  #define SIGN(X) (((X)>0)?1:-1)  #define SIGN(X) (((X)>0)?1:-1)
388  #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 };  
   
389    
390  // decode an inter macroblock  // decode an inter macroblock
391    
392  void decoder_mbinter(DECODER * dec,  static void
393    rrv_mv_scaleup(VECTOR * mv)
394    {
395            if (mv->x > 0) {
396                    mv->x = 2*mv->x - 1;
397            } else if (mv->x < 0) {
398                    mv->x = 2*mv->x + 1;
399            }
400    
401            if (mv->y > 0) {
402                    mv->y = 2*mv->y - 1;
403            } else if (mv->y < 0) {
404                    mv->y = 2*mv->y + 1;
405            }
406    }
407    
408    
409    
410    void
411    decoder_mbinter(DECODER * dec,
412                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
413                       const uint32_t x_pos,                       const uint32_t x_pos,
414                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 310  Line 416 
416                       const uint32_t cbp,                       const uint32_t cbp,
417                       Bitstream * bs,                       Bitstream * bs,
418                       const uint32_t quant,                       const uint32_t quant,
419                       const uint32_t rounding)                                  const uint32_t rounding,
420                                    const int reduced_resolution)
421  {  {
422    
423          DECLARE_ALIGNED_MATRIX(block,6, 64, int16_t, CACHE_LINE);          DECLARE_ALIGNED_MATRIX(block,6, 64, int16_t, CACHE_LINE);
# Line 318  Line 425 
425    
426          uint32_t stride = dec->edged_width;          uint32_t stride = dec->edged_width;
427          uint32_t stride2 = stride / 2;          uint32_t stride2 = stride / 2;
428          uint32_t next_block = stride * 8;          uint32_t next_block = stride * (reduced_resolution ? 16 : 8);
429          uint32_t i;          uint32_t i;
430          uint32_t iQuant = pMB->quant;          uint32_t iQuant = pMB->quant;
431          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
432    
433          int uv_dx, uv_dy;          int uv_dx, uv_dy;
434            VECTOR mv[4];
435    
436            for (i = 0; i < 4; i++)
437            {
438                    mv[i] = pMB->mvs[i];
439                    //DPRINTF(DPRINTF_MB, "mv[%i]   orig=%i,%i   local=%i", i, pMB->mvs[i].x, pMB->mvs[i].y,                                                mv[i].x, mv[i].y);
440            }
441    
442            if (reduced_resolution) {
443                    pY_Cur = dec->cur.y + (y_pos << 5) * stride + (x_pos << 5);
444                    pU_Cur = dec->cur.u + (y_pos << 4) * stride2 + (x_pos << 4);
445                    pV_Cur = dec->cur.v + (y_pos << 4) * stride2 + (x_pos << 4);
446                    rrv_mv_scaleup(&mv[0]);
447                    rrv_mv_scaleup(&mv[1]);
448                    rrv_mv_scaleup(&mv[2]);
449                    rrv_mv_scaleup(&mv[3]);
450            }else{
451          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
452          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
453          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
454            }
455    
456          if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)          if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
457                    uv_dx = mv[0].x;
458                    uv_dy = mv[0].y;
459    
460                    if (dec->quarterpel)
461          {          {
462                  uv_dx = pMB->mvs[0].x;                          uv_dx /= 2;
463                  uv_dy = pMB->mvs[0].y;                          uv_dy /= 2;
464                    }
465    
466                    uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
467                    uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
468    
469                    start_timer();
470                    if (reduced_resolution)
471                    {
472                            interpolate32x32_switch(dec->cur.y, dec->refn[0].y, 32*x_pos, 32*y_pos,
473                                                                      mv[0].x, mv[0].y, stride,  rounding);
474                            interpolate16x16_switch(dec->cur.u, dec->refn[0].u, 16 * x_pos, 16 * y_pos,
475                                                                      uv_dx, uv_dy, stride2, rounding);
476                            interpolate16x16_switch(dec->cur.v, dec->refn[0].v, 16 * x_pos, 16 * y_pos,
477                                                                      uv_dx, uv_dy, stride2, rounding);
478    
                 uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;  
                 uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;  
479          }          }
480          else          else
481          {          {
482                  int sum;                          if(dec->quarterpel) {
483                  sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;                                  interpolate16x16_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
484                  uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                                                                                          dec->refh.y + 128, 16*x_pos, 16*y_pos,
485                                                                                            mv[0].x, mv[0].y, stride,  rounding);
486                            }
487                            else {
488                                    interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
489                                                                              mv[0].x, mv[0].y, stride,  rounding);
490                            }
491    
492                  sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;                          interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
493                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                                                                    uv_dx, uv_dy, stride2, rounding);
494                            interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
495                                                                      uv_dx, uv_dy, stride2, rounding);
496          }          }
497                    stop_comp_timer();
498    
499            } else {        /* MODE_INTER4V */
500                    int sum;
501    
502                    if(dec->quarterpel)
503                            sum = (mv[0].x / 2) + (mv[1].x / 2) + (mv[2].x / 2) + (mv[3].x / 2);
504                    else
505                            sum = mv[0].x + mv[1].x + mv[2].x + mv[3].x;
506    
507                    uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
508    
509                    if(dec->quarterpel)
510                            sum = (mv[0].y / 2) + (mv[1].y / 2) + (mv[2].y / 2) + (mv[3].y / 2);
511                    else
512                            sum = mv[0].y + mv[1].y + mv[2].y + mv[3].y;
513    
514                    uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
515    
516          start_timer();          start_timer();
517          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 (reduced_resolution)
518          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);                  {
519          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);                          interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 32*x_pos, 32*y_pos,
520          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);                                                                    mv[0].x, mv[0].y, stride,  rounding);
521          interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, rounding);                          interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 32*x_pos + 16, 32*y_pos,
522          interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, rounding);                                                                    mv[1].x, mv[1].y, stride,  rounding);
523          stop_comp_timer();                          interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 32*x_pos, 32*y_pos + 16,
524                                                                      mv[2].x, mv[2].y, stride,  rounding);
525                            interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 32*x_pos + 16, 32*y_pos + 16,
526                                                                      mv[3].x, mv[3].y, stride,  rounding);
527                            interpolate16x16_switch(dec->cur.u, dec->refn[0].u, 16 * x_pos, 16 * y_pos,
528                                                                      uv_dx, uv_dy, stride2, rounding);
529                            interpolate16x16_switch(dec->cur.v, dec->refn[0].v, 16 * x_pos, 16 * y_pos,
530                                                                      uv_dx, uv_dy, stride2, rounding);
531    
532          for (i = 0; i < 6; i++)                          // set_block(pY_Cur, stride, 32, 32, 127);
533                    }
534                    else
535          {          {
536                            if(dec->quarterpel) {
537                                    interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
538                                                                                      dec->refh.y + 128, 16*x_pos, 16*y_pos,
539                                                                                      mv[0].x, mv[0].y, stride,  rounding);
540                                    interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
541                                                                                      dec->refh.y + 128, 16*x_pos + 8, 16*y_pos,
542                                                                                      mv[1].x, mv[1].y, stride,  rounding);
543                                    interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
544                                                                                      dec->refh.y + 128, 16*x_pos, 16*y_pos + 8,
545                                                                                      mv[2].x, mv[2].y, stride,  rounding);
546                                    interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
547                                                                                      dec->refh.y + 128, 16*x_pos + 8, 16*y_pos + 8,
548                                                                                      mv[3].x, mv[3].y, stride,  rounding);
549                            }
550                            else {
551                                    interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
552                                                                              mv[0].x, mv[0].y, stride,  rounding);
553                                    interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos,
554                                                                              mv[1].x, mv[1].y, stride,  rounding);
555                                    interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos + 8,
556                                                                              mv[2].x, mv[2].y, stride,  rounding);
557                                    interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8,
558                                                                              mv[3].x, mv[3].y, stride,  rounding);
559                            }
560    
561                            interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
562                                                                      uv_dx, uv_dy, stride2, rounding);
563                            interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
564                                                                      uv_dx, uv_dy, stride2, rounding);
565                    }
566                    stop_comp_timer();
567            }
568    
569            for (i = 0; i < 6; i++) {
570                    int direction = dec->alternate_vertical_scan ? 2 : 0;
571    
572                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
573                  {                  {
574                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear
575    
576                          start_timer();                          start_timer();
577                          get_inter_block(bs, &block[i*64]);                          get_inter_block(bs, &block[i * 64], direction);
578                          stop_coding_timer();                          stop_coding_timer();
579    
580                          start_timer();                          start_timer();
581                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
                         {  
582                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                                  dequant_inter(&data[i*64], &block[i*64], iQuant);
583                          }                          } else {
                         else  
                         {  
584                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);
585                          }                          }
586                          stop_iquant_timer();                          stop_iquant_timer();
# Line 382  Line 591 
591                  }                  }
592          }          }
593    
594          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
595                  next_block = stride;                  next_block = stride;
596                  stride *= 2;                  stride *= 2;
597          }          }
598    
599          start_timer();          start_timer();
600            if (reduced_resolution)
601            {
602                    if (cbp & 32)
603                            add_upsampled_8x8_16to8(pY_Cur, &data[0 * 64], stride);
604                    if (cbp & 16)
605                            add_upsampled_8x8_16to8(pY_Cur + 16, &data[1 * 64], stride);
606                    if (cbp & 8)
607                            add_upsampled_8x8_16to8(pY_Cur + next_block, &data[2 * 64], stride);
608                    if (cbp & 4)
609                            add_upsampled_8x8_16to8(pY_Cur + 16 + next_block, &data[3 * 64], stride);
610                    if (cbp & 2)
611                            add_upsampled_8x8_16to8(pU_Cur, &data[4 * 64], stride2);
612                    if (cbp & 1)
613                            add_upsampled_8x8_16to8(pV_Cur, &data[5 * 64], stride2);
614            }
615            else
616            {
617          if (cbp & 32)          if (cbp & 32)
618                  transfer_16to8add(pY_Cur,                  &data[0*64], stride);                  transfer_16to8add(pY_Cur,                  &data[0*64], stride);
619          if (cbp & 16)          if (cbp & 16)
# Line 401  Line 626 
626                  transfer_16to8add(pU_Cur,                  &data[4*64], stride2);                  transfer_16to8add(pU_Cur,                  &data[4*64], stride2);
627          if (cbp & 1)          if (cbp & 1)
628                  transfer_16to8add(pV_Cur,                  &data[5*64], stride2);                  transfer_16to8add(pV_Cur,                  &data[5*64], stride2);
629            }
630          stop_transfer_timer();          stop_transfer_timer();
631  }  }
632    
633    
634  void decoder_iframe(DECODER * dec, Bitstream * bs, int quant, int intra_dc_threshold)  void
635    decoder_iframe(DECODER * dec,
636                               Bitstream * bs,
637                               int reduced_resolution,
638                               int quant,
639                               int intra_dc_threshold)
640  {  {
641            uint32_t bound;
642          uint32_t x, y;          uint32_t x, y;
643            int mb_width = dec->mb_width;
644            int mb_height = dec->mb_height;
645    
646          for (y = 0; y < dec->mb_height; y++)          if (reduced_resolution)
         {  
                 for (x = 0; x < dec->mb_width; x++)  
647                  {                  {
648                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                  mb_width = (dec->width + 31) / 32;
649                    mb_height = (dec->height + 31) / 32;
650            }
651    
652            bound = 0;
653    
654            for (y = 0; y < mb_height; y++) {
655                    for (x = 0; x < mb_width; x++) {
656                            MACROBLOCK *mb;
657                          uint32_t mcbpc;                          uint32_t mcbpc;
658                          uint32_t cbpc;                          uint32_t cbpc;
659                          uint32_t acpred_flag;                          uint32_t acpred_flag;
660                          uint32_t cbpy;                          uint32_t cbpy;
661                          uint32_t cbp;                          uint32_t cbp;
662    
663                            while (BitstreamShowBits(bs, 9) == 1)
664                                    BitstreamSkip(bs, 9);
665    
666                            if (check_resync_marker(bs, 0))
667                            {
668                                    bound = read_video_packet_header(bs, dec, 0,
669                                                            &quant, NULL, NULL, &intra_dc_threshold);
670                                    x = bound % mb_width;
671                                    y = bound / mb_width;
672                            }
673                            mb = &dec->mbs[y * dec->mb_width + x];
674    
675                            DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
676    
677                          mcbpc = get_mcbpc_intra(bs);                          mcbpc = get_mcbpc_intra(bs);
678                          mb->mode = mcbpc & 7;                          mb->mode = mcbpc & 7;
679                          cbpc = (mcbpc >> 4);                          cbpc = (mcbpc >> 4);
680    
681                          acpred_flag = BitstreamGetBit(bs);                          acpred_flag = BitstreamGetBit(bs);
682    
                         if (mb->mode == MODE_STUFFING)  
                         {  
                                 DEBUG("-- STUFFING ?");  
                                 continue;  
                         }  
   
683                          cbpy = get_cbpy(bs, 1);                          cbpy = get_cbpy(bs, 1);
684                          cbp = (cbpy << 2) | cbpc;                          cbp = (cbpy << 2) | cbpc;
685    
686                          if (mb->mode == MODE_INTRA_Q)                          if (mb->mode == MODE_INTRA_Q) {
                         {  
687                                  quant += dquant_table[BitstreamGetBits(bs,2)];                                  quant += dquant_table[BitstreamGetBits(bs,2)];
688                                  if (quant > 31)                                  if (quant > 31) {
                                 {  
689                                          quant = 31;                                          quant = 31;
690                                  }                                  } else if (quant < 1) {
                                 else if (quant < 1)  
                                 {  
691                                          quant = 1;                                          quant = 1;
692                                  }                                  }
693                          }                          }
694                          mb->quant = quant;                          mb->quant = quant;
695                            mb->mvs[0].x = mb->mvs[0].y =
696                            mb->mvs[1].x = mb->mvs[1].y =
697                            mb->mvs[2].x = mb->mvs[2].y =
698                            mb->mvs[3].x = mb->mvs[3].y =0;
699    
700                          if (dec->interlacing)                          if (dec->interlacing) {
                         {  
701                                  mb->field_dct = BitstreamGetBit(bs);                                  mb->field_dct = BitstreamGetBit(bs);
702                                  DEBUG1("deci: field_dct: ", mb->field_dct);                                  DEBUG1("deci: field_dct: ", mb->field_dct);
703                          }                          }
704    
705                          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,
706                                                            intra_dc_threshold, bound, reduced_resolution);
707    
708                  }                  }
709                    if(dec->out_frm)
710                      output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,0,y,mb_width);
711          }          }
712    
713  }  }
714    
715    
716  void get_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, int k, VECTOR * mv, int fcode)  void
717    get_motion_vector(DECODER * dec,
718                                      Bitstream * bs,
719                                      int x,
720                                      int y,
721                                      int k,
722                                      VECTOR * ret_mv,
723                                      int fcode,
724                                      const int bound)
725  {  {
726    
727          int scale_fac = 1 << (fcode - 1);          int scale_fac = 1 << (fcode - 1);
# Line 472  Line 729 
729          int low = ((-32) * scale_fac);          int low = ((-32) * scale_fac);
730          int range = (64 * scale_fac);          int range = (64 * scale_fac);
731    
732          VECTOR pmv[4];          VECTOR pmv;
733          int32_t psad[4];          VECTOR mv;
734    
735          int mv_x, mv_y;          pmv = get_pmv2(dec->mbs, dec->mb_width, bound, x, y, k);
         int pmv_x, pmv_y;  
736    
737            mv.x = get_mv(bs, fcode);
738            mv.y = get_mv(bs, fcode);
739    
740          get_pmvdata(dec->mbs, x, y, dec->mb_width, k, pmv, psad);          DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i)", mv.x, mv.y, pmv.x, pmv.y);
741    
742          pmv_x = pmv[0].x;          mv.x += pmv.x;
743          pmv_y = pmv[0].y;          mv.y += pmv.y;
744    
745          mv_x = get_mv(bs, fcode);          if (mv.x < low) {
746          mv_y = get_mv(bs, fcode);                  mv.x += range;
747            } else if (mv.x > high) {
748          mv_x += pmv_x;                  mv.x -= range;
         mv_y += pmv_y;  
   
         if (mv_x < low)  
         {  
                 mv_x += range;  
749          }          }
750          else if (mv_x > high)  
751          {          if (mv.y < low) {
752                  mv_x -= range;                  mv.y += range;
753            } else if (mv.y > high) {
754                    mv.y -= range;
755          }          }
756    
757          if (mv_y < low)          ret_mv->x = mv.x;
758          {          ret_mv->y = mv.y;
                 mv_y += range;  
759          }          }
760          else if (mv_y > high)  
761    
762    
763    static __inline int gmc_sanitize(int value, int quarterpel, int fcode)
764          {          {
765                  mv_y -= range;          int length = 1 << (fcode+4);
         }  
766    
767          mv->x = mv_x;          if (quarterpel) value *= 2;
         mv->y = mv_y;  
768    
769            if (value < -length)
770                    return -length;
771            else if (value >= length)
772                    return length-1;
773            else return value;
774  }  }
775    
776    
777  void decoder_pframe(DECODER * dec, Bitstream * bs, int rounding, int quant, int fcode, int intra_dc_threshold)  /* for P_VOP set gmc_mv to NULL */
778    void
779    decoder_pframe(DECODER * dec,
780                               Bitstream * bs,
781                               int rounding,
782                               int reduced_resolution,
783                               int quant,
784                               int fcode,
785                               int intra_dc_threshold,
786                               VECTOR * gmc_mv)
787  {  {
788    
789          uint32_t x, y;          uint32_t x, y;
790            uint32_t bound;
791            int cp_mb, st_mb;
792            int mb_width = dec->mb_width;
793            int mb_height = dec->mb_height;
794    
795            if (reduced_resolution)
796            {
797                    mb_width = (dec->width + 31) / 32;
798                    mb_height = (dec->height + 31) / 32;
799            }
800    
801          start_timer();          start_timer();
802          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,
803                                       dec->width, dec->height);
804          stop_edges_timer();          stop_edges_timer();
805    
806          for (y = 0; y < dec->mb_height; y++)          bound = 0;
807          {  
808                  for (x = 0; x < dec->mb_width; x++)          for (y = 0; y < mb_height; y++) {
809                    cp_mb = st_mb = 0;
810                    for (x = 0; x < mb_width; x++) {
811                            MACROBLOCK *mb;
812    
813                            // skip stuffing
814                            while (BitstreamShowBits(bs, 10) == 1)
815                                    BitstreamSkip(bs, 10);
816    
817                            if (check_resync_marker(bs, fcode - 1))
818                  {                  {
819                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                                  bound = read_video_packet_header(bs, dec, fcode - 1,
820                                            &quant, &fcode, NULL, &intra_dc_threshold);
821                                    x = bound % mb_width;
822                                    y = bound / mb_width;
823                            }
824                            mb = &dec->mbs[y * dec->mb_width + x];
825    
826                            DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
827    
828                          //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
829                          if (!(BitstreamGetBit(bs)))                     // not_coded                          if (!(BitstreamGetBit(bs)))                     // not_coded
# Line 538  Line 834 
834                                  uint32_t cbpy;                                  uint32_t cbpy;
835                                  uint32_t cbp;                                  uint32_t cbp;
836                                  uint32_t intra;                                  uint32_t intra;
837                                    int mcsel = 0;          // mcsel: '0'=local motion, '1'=GMC
838    
839                                    cp_mb++;
840                                  mcbpc = get_mcbpc_inter(bs);                                  mcbpc = get_mcbpc_inter(bs);
841                                  mb->mode = mcbpc & 7;                                  mb->mode = mcbpc & 7;
842                                  cbpc = (mcbpc >> 4);                                  cbpc = (mcbpc >> 4);
843    
844                                    DPRINTF(DPRINTF_MB, "mode %i", mb->mode);
845                                    DPRINTF(DPRINTF_MB, "cbpc %i", cbpc);
846                                  acpred_flag = 0;                                  acpred_flag = 0;
847    
848                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);
849    
850                                  if (intra)                                  if (intra) {
                                 {  
851                                          acpred_flag = BitstreamGetBit(bs);                                          acpred_flag = BitstreamGetBit(bs);
852                                  }                                  }
853    
854                                  if (mb->mode == MODE_STUFFING)                                  if (gmc_mv && (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q))
855                                  {                                  {
856                                          DEBUG("-- STUFFING ?");                                          mcsel = BitstreamGetBit(bs);
                                         continue;  
857                                  }                                  }
858    
859                                  cbpy = get_cbpy(bs, intra);                                  cbpy = get_cbpy(bs, intra);
860                                    DPRINTF(DPRINTF_MB, "cbpy %i", cbpy);
861    
862                                  cbp = (cbpy << 2) | cbpc;                                  cbp = (cbpy << 2) | cbpc;
863    
864                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q)                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q) {
865                                  {                                          int dquant = dquant_table[BitstreamGetBits(bs, 2)];
866                                          quant += dquant_table[BitstreamGetBits(bs,2)];                                          DPRINTF(DPRINTF_MB, "dquant %i", dquant);
867                                          if (quant > 31)                                          quant += dquant;
868                                          {                                          if (quant > 31) {
869                                                  quant = 31;                                                  quant = 31;
870                                          }                                          } else if (quant < 1) {
                                         else if (mb->quant < 1)  
                                         {  
871                                                  quant = 1;                                                  quant = 1;
872                                          }                                          }
873                                            DPRINTF(DPRINTF_MB, "quant %i", quant);
874                                  }                                  }
875                                  mb->quant = quant;                                  mb->quant = quant;
876    
877                                  if (dec->interlacing)                                  if (dec->interlacing) {
878                                  {                                          if (cbp || intra) {
879                                          mb->field_dct = BitstreamGetBit(bs);                                          mb->field_dct = BitstreamGetBit(bs);
880                                          DEBUG1("decp: field_dct: ", mb->field_dct);                                          DEBUG1("decp: field_dct: ", mb->field_dct);
881                                            }
882    
883                                          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
                                         {  
884                                                  mb->field_pred = BitstreamGetBit(bs);                                                  mb->field_pred = BitstreamGetBit(bs);
885                                                  DEBUG1("decp: field_pred: ", mb->field_pred);                                                  DEBUG1("decp: field_pred: ", mb->field_pred);
886    
887                                                  if (mb->field_pred)                                                  if (mb->field_pred) {
                                                 {  
888                                                          mb->field_for_top = BitstreamGetBit(bs);                                                          mb->field_for_top = BitstreamGetBit(bs);
889                                                          DEBUG1("decp: field_for_top: ", mb->field_for_top);                                                          DEBUG1("decp: field_for_top: ", mb->field_for_top);
890                                                          mb->field_for_bot = BitstreamGetBit(bs);                                                          mb->field_for_bot = BitstreamGetBit(bs);
# Line 594  Line 893 
893                                          }                                          }
894                                  }                                  }
895    
896                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
897                                  {  
898                                          if (dec->interlacing && mb->field_pred)                                          if (mcsel)
                                         {  
                                                 get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);  
                                                 get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1], fcode);  
                                         }  
                                         else  
                                         {  
                                                 get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);  
                                                 mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;  
                                                 mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;  
                                         }  
                                 }  
                                 else if (mb->mode == MODE_INTER4V /* || mb->mode == MODE_INTER4V_Q */)  
                                 {  
                                         get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);  
                                         get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode);  
                                         get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode);  
                                         get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode);  
                                 }  
                                 else  // MODE_INTRA, MODE_INTRA_Q  
899                                  {                                  {
900                                          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 = gmc_sanitize(gmc_mv[0].x, dec->quarterpel, fcode);
901                                          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 = gmc_sanitize(gmc_mv[0].y, dec->quarterpel, fcode);
902                                          decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant, intra_dc_threshold);  
903                                            } else if (dec->interlacing && mb->field_pred) {
904                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
905                                                                                      fcode, bound);
906                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1],
907                                                                                      fcode, bound);
908                                            } else {
909                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
910                                                                                      fcode, bound);
911                                                    mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
912                                                            mb->mvs[0].x;
913                                                    mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
914                                                            mb->mvs[0].y;
915                                            }
916                                    } else if (mb->mode == MODE_INTER4V ) {
917    
918                                            get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode, bound);
919                                            get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode, bound);
920                                            get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode, bound);
921                                            get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode, bound);
922                                    } else                  // MODE_INTRA, MODE_INTRA_Q
923                                    {
924                                            mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
925                                                    0;
926                                            mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
927                                                    0;
928                                            decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
929                                                                            intra_dc_threshold, bound, reduced_resolution);
930                                          continue;                                          continue;
931                                  }                                  }
932    
933                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant, rounding);                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant,
934                                                                    rounding, reduced_resolution);
935    
936                          }                          }
937                          else    // not coded                          else if (gmc_mv)        /* not coded S_VOP macroblock */
938                          {                          {
                                 //DEBUG2("P-frame MB at (X,Y)=",x,y);  
939                                  mb->mode = MODE_NOT_CODED;                                  mb->mode = MODE_NOT_CODED;
940                                    mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = gmc_sanitize(gmc_mv[0].x, dec->quarterpel, fcode);
941                                    mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = gmc_sanitize(gmc_mv[0].y, dec->quarterpel, fcode);
942                                    decoder_mbinter(dec, mb, x, y, 0, 0, bs, quant, rounding, reduced_resolution);
943                            }
944                            else    /* not coded P_VOP macroblock */
945                            {
946                                    mb->mode = MODE_NOT_CODED;
947    
948                                  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;
949                                  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;
   
950                                  // copy macroblock directly from ref to cur                                  // copy macroblock directly from ref to cur
951    
952                                  start_timer();                                  start_timer();
953    
954                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),                                  if (reduced_resolution)
955                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x),                                  {
956                                                   dec->edged_width);                                          transfer32x32_copy(dec->cur.y + (32*y)*dec->edged_width + (32*x),
957                                                                             dec->refn[0].y + (32*y)*dec->edged_width + (32*x),
                                 transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x+8),  
                                                  dec->refn[0].y + (16*y)*dec->edged_width + (16*x+8),  
958                                                   dec->edged_width);                                                   dec->edged_width);
959    
960                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x),                                          transfer16x16_copy(dec->cur.u + (16*y)*dec->edged_width/2 + (16*x),
961                                                   dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x),                                                                          dec->refn[0].u + (16*y)*dec->edged_width/2 + (16*x),
962                                                   dec->edged_width);                                                                          dec->edged_width/2);
963    
964                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x+8),                                          transfer16x16_copy(dec->cur.v + (16*y)*dec->edged_width/2 + (16*x),
965                                                   dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x+8),                                                                           dec->refn[0].v + (16*y)*dec->edged_width/2 + (16*x),
966                                                                             dec->edged_width/2);
967                                    }
968                                    else
969                                    {
970                                            transfer16x16_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),
971                                                                             dec->refn[0].y + (16*y)*dec->edged_width + (16*x),
972                                                   dec->edged_width);                                                   dec->edged_width);
973    
974                                  transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),                                  transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),
# Line 659  Line 978 
978                                  transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),                                  transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),
979                                                   dec->refn[0].v + (8*y)*dec->edged_width/2 + (8*x),                                                   dec->refn[0].v + (8*y)*dec->edged_width/2 + (8*x),
980                                                   dec->edged_width/2);                                                   dec->edged_width/2);
981                                    }
982    
983                                  stop_transfer_timer();                                  stop_transfer_timer();
984    
985                                    if(dec->out_frm && cp_mb > 0) {
986                                      output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
987                                      cp_mb = 0;
988                          }                          }
989                                    st_mb = x+1;
990                  }                  }
991          }          }
992                    if(dec->out_frm && cp_mb > 0)
993                      output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
994            }
995  }  }
996    
997    
998  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
999  // decode B-frame motion vector  // decode B-frame motion vector
1000  void get_b_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, VECTOR * mv, int fcode, const VECTOR pmv)  void
1001    get_b_motion_vector(DECODER * dec,
1002                                            Bitstream * bs,
1003                                            int x,
1004                                            int y,
1005                                            VECTOR * mv,
1006                                            int fcode,
1007                                            const VECTOR pmv)
1008  {  {
1009          int scale_fac = 1 << (fcode - 1);          int scale_fac = 1 << (fcode - 1);
1010          int high = (32 * scale_fac) - 1;          int high = (32 * scale_fac) - 1;
# Line 688  Line 1023 
1023          mv_x += pmv_x;          mv_x += pmv_x;
1024          mv_y += pmv_y;          mv_y += pmv_y;
1025    
1026          if (mv_x < low)          if (mv_x < low) {
         {  
1027                  mv_x += range;                  mv_x += range;
1028          }          } else if (mv_x > high) {
         else if (mv_x > high)  
         {  
1029                  mv_x -= range;                  mv_x -= range;
1030          }          }
1031    
1032          if (mv_y < low)          if (mv_y < low) {
         {  
1033                  mv_y += range;                  mv_y += range;
1034          }          } else if (mv_y > high) {
         else if (mv_y > high)  
         {  
1035                  mv_y -= range;                  mv_y -= range;
1036          }          }
1037    
# Line 713  Line 1042 
1042    
1043  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
1044  // decode an B-frame forward & backward inter macroblock  // decode an B-frame forward & backward inter macroblock
1045  void decoder_bf_mbinter(DECODER * dec,  void
1046    decoder_bf_mbinter(DECODER * dec,
1047                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
1048                       const uint32_t x_pos,                       const uint32_t x_pos,
1049                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 738  Line 1068 
1068          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
1069          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
1070    
1071          if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))  
1072          {          if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
1073                  uv_dx = pMB->mvs[0].x;                  uv_dx = pMB->mvs[0].x;
1074                  uv_dy = pMB->mvs[0].y;                  uv_dy = pMB->mvs[0].y;
1075    
1076                  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  
1077          {          {
1078                            uv_dx /= 2;
1079                            uv_dy /= 2;
1080                    }
1081    
1082                    uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
1083                    uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
1084            } else {
1085                  int sum;                  int sum;
1086    
1087                    if(dec->quarterpel)
1088                            sum = (pMB->mvs[0].x / 2) + (pMB->mvs[1].x / 2) + (pMB->mvs[2].x / 2) + (pMB->mvs[3].x / 2);
1089                    else
1090                  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) );  
1091    
1092                    uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
1093    
1094                    if(dec->quarterpel)
1095                            sum = (pMB->mvs[0].y / 2) + (pMB->mvs[1].y / 2) + (pMB->mvs[2].y / 2) + (pMB->mvs[3].y / 2);
1096                    else
1097                  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;
1098                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );  
1099                    uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
1100          }          }
1101    
1102          start_timer();          start_timer();
1103          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);          if(dec->quarterpel) {
1104          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);                  interpolate16x16_quarterpel(dec->cur.y, dec->refn[ref].y, dec->refh.y, dec->refh.y + 64,
1105          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);                                                                      dec->refh.y + 128, 16*x_pos, 16*y_pos,
1106          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);                                                                      pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1107          interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);          }
1108          interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);          else {
1109                    interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos, 16*y_pos,
1110                                                              pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1111                    interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos + 8, 16*y_pos,
1112                                                          pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1113                    interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos, 16*y_pos + 8,
1114                                                              pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1115                    interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos + 8, 16*y_pos + 8,
1116                                                              pMB->mvs[3].x, pMB->mvs[3].y, stride, 0);
1117            }
1118    
1119            interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8 * x_pos, 8 * y_pos,
1120                                                      uv_dx, uv_dy, stride2, 0);
1121            interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8 * x_pos, 8 * y_pos,
1122                                                      uv_dx, uv_dy, stride2, 0);
1123          stop_comp_timer();          stop_comp_timer();
1124    
1125          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
1126          {                  int direction = dec->alternate_vertical_scan ? 2 : 0;
1127    
1128                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
1129                  {                  {
1130                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear
1131    
1132                          start_timer();                          start_timer();
1133                          get_inter_block(bs, &block[i*64]);                          get_inter_block(bs, &block[i * 64], direction);
1134                          stop_coding_timer();                          stop_coding_timer();
1135    
1136                          start_timer();                          start_timer();
1137                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
                         {  
1138                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                                  dequant_inter(&data[i*64], &block[i*64], iQuant);
1139                          }                          } else {
                         else  
                         {  
1140                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);
1141                          }                          }
1142                          stop_iquant_timer();                          stop_iquant_timer();
1143    
                         data[0]=10;  
1144                          start_timer();                          start_timer();
1145                          idct(&data[i*64]);                          idct(&data[i*64]);
1146                          stop_idct_timer();                          stop_idct_timer();
1147                  }                  }
1148          }          }
1149    
1150          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
1151                  next_block = stride;                  next_block = stride;
1152                  stride *= 2;                  stride *= 2;
1153          }          }
# Line 815  Line 1168 
1168          stop_transfer_timer();          stop_transfer_timer();
1169  }  }
1170    
   
1171  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
1172  // decode an B-frame direct &  inter macroblock  // decode an B-frame direct &  inter macroblock
1173  void decoder_bf_interpolate_mbinter(DECODER * dec,  void
1174                           IMAGE forward, IMAGE backward,  decoder_bf_interpolate_mbinter(DECODER * dec,
1175                                                               IMAGE forward,
1176                                                               IMAGE backward,
1177                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
1178                       const uint32_t x_pos,                       const uint32_t x_pos,
1179                       const uint32_t y_pos,                       const uint32_t y_pos,
                      const uint32_t cbp,  
1180                       Bitstream * bs )                       Bitstream * bs )
1181  {  {
1182    
# Line 838  Line 1191 
1191          int                     b_uv_dx, b_uv_dy;          int                     b_uv_dx, b_uv_dy;
1192          uint32_t        i;          uint32_t        i;
1193          uint8_t         *pY_Cur, *pU_Cur, *pV_Cur;          uint8_t         *pY_Cur, *pU_Cur, *pV_Cur;
1194        const uint32_t cbp = pMB->cbp;
1195    
1196          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
1197          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
1198          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
1199    
1200          if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))  
1201          {          if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
1202                  uv_dx = pMB->mvs[0].x;                  uv_dx = pMB->mvs[0].x;
1203                  uv_dy = pMB->mvs[0].y;                  uv_dy = pMB->mvs[0].y;
1204    
                 uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;  
                 uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;  
   
1205                  b_uv_dx = pMB->b_mvs[0].x;                  b_uv_dx = pMB->b_mvs[0].x;
1206                  b_uv_dy = pMB->b_mvs[0].y;                  b_uv_dy = pMB->b_mvs[0].y;
1207    
1208                  b_uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;                  if (dec->quarterpel)
                 b_uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;  
         }  
         else  
1209          {          {
1210                            uv_dx /= 2;
1211                            uv_dy /= 2;
1212    
1213                            b_uv_dx /= 2;
1214                            b_uv_dy /= 2;
1215                    }
1216    
1217                    uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
1218                    uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
1219    
1220                    b_uv_dx = (b_uv_dx >> 1) + roundtab_79[b_uv_dx & 0x3];
1221                    b_uv_dy = (b_uv_dy >> 1) + roundtab_79[b_uv_dy & 0x3];
1222            } else {
1223                  int sum;                  int sum;
1224    
1225                    if(dec->quarterpel)
1226                            sum = (pMB->mvs[0].x / 2) + (pMB->mvs[1].x / 2) + (pMB->mvs[2].x / 2) + (pMB->mvs[3].x / 2);
1227                    else
1228                  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) );  
1229    
1230                    uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
1231    
1232                    if(dec->quarterpel)
1233                            sum = (pMB->mvs[0].y / 2) + (pMB->mvs[1].y / 2) + (pMB->mvs[2].y / 2) + (pMB->mvs[3].y / 2);
1234                    else
1235                  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;
                 uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );  
1236    
1237                    uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
1238    
1239    
1240                    if(dec->quarterpel)
1241                            sum = (pMB->b_mvs[0].x / 2) + (pMB->b_mvs[1].x / 2) + (pMB->b_mvs[2].x / 2) + (pMB->b_mvs[3].x / 2);
1242                    else
1243                  sum = pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x + pMB->b_mvs[3].x;                  sum = pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x + pMB->b_mvs[3].x;
                 b_uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );  
1244    
1245                    b_uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
1246    
1247                    if(dec->quarterpel)
1248                            sum = (pMB->b_mvs[0].y / 2) + (pMB->b_mvs[1].y / 2) + (pMB->b_mvs[2].y / 2) + (pMB->b_mvs[3].y / 2);
1249                    else
1250                  sum = pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y + pMB->b_mvs[3].y;                  sum = pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y + pMB->b_mvs[3].y;
1251                  b_uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );  
1252                    b_uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
1253          }          }
1254    
1255    
1256          start_timer();          start_timer();
1257          interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos,     16*y_pos    , pMB->mvs[0].x, pMB->mvs[0].y, stride,  0);          if(dec->quarterpel) {
1258          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((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))
1259          interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos,     16*y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,  0);                          interpolate16x16_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1260          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);                                                                              dec->refh.y + 128, 16*x_pos, 16*y_pos,
1261          interpolate8x8_switch(dec->cur.u, forward.u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);                                                                              pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1262          interpolate8x8_switch(dec->cur.v, forward.v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, 0);                  else {
1263                            interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1264                                                                                dec->refh.y + 128, 16*x_pos, 16*y_pos,
1265                                                                                pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1266                            interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1267                                                                                dec->refh.y + 128, 16*x_pos + 8, 16*y_pos,
1268                                                                                pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1269                            interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1270                                                                                dec->refh.y + 128, 16*x_pos, 16*y_pos + 8,
1271                                                                                pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1272                            interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1273                                                                                dec->refh.y + 128, 16*x_pos + 8, 16*y_pos + 8,
1274                                                                                pMB->mvs[3].x, pMB->mvs[3].y, stride, 0);
1275                    }
1276            }
1277            else {
1278                    interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos,
1279                                                              pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1280                    interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8, 16 * y_pos,
1281                                                              pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1282                    interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos + 8,
1283                                                              pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1284                    interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8,
1285                                                              16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
1286                                                              0);
1287            }
1288    
1289            interpolate8x8_switch(dec->cur.u, forward.u, 8 * x_pos, 8 * y_pos, uv_dx,
1290                                                      uv_dy, stride2, 0);
1291            interpolate8x8_switch(dec->cur.v, forward.v, 8 * x_pos, 8 * y_pos, uv_dx,
1292                                                      uv_dy, stride2, 0);
1293    
         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);  
1294    
1295          interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos,     16*y_pos    , stride);          if(dec->quarterpel) {
1296          interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos + 8, 16*y_pos    , stride);                  if((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))
1297          interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos,     16*y_pos + 8, stride);                          interpolate16x16_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1298          interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos + 8, 16*y_pos + 8, stride);                                                                              dec->refh.y + 128, 16*x_pos, 16*y_pos,
1299          interpolate8x8_c(dec->cur.u, dec->refn[2].u, 8*x_pos,      8*y_pos,      stride2);                                                                              pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1300          interpolate8x8_c(dec->cur.v, dec->refn[2].v, 8*x_pos,      8*y_pos,      stride2);                  else {
1301                            interpolate8x8_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1302                                                                                dec->refh.y + 128, 16*x_pos, 16*y_pos,
1303                                                                                pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1304                            interpolate8x8_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1305                                                                                dec->refh.y + 128, 16*x_pos + 8, 16*y_pos,
1306                                                                                pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride, 0);
1307                            interpolate8x8_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1308                                                                                dec->refh.y + 128, 16*x_pos, 16*y_pos + 8,
1309                                                                                pMB->b_mvs[2].x, pMB->b_mvs[2].y, stride, 0);
1310                            interpolate8x8_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1311                                                                                dec->refh.y + 128, 16*x_pos + 8, 16*y_pos + 8,
1312                                                                                pMB->b_mvs[3].x, pMB->b_mvs[3].y, stride, 0);
1313                    }
1314            }
1315            else {
1316                    interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos, 16 * y_pos,
1317                                                              pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1318                    interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1319                                                              16 * y_pos, pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride,
1320                                                              0);
1321                    interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos,
1322                                                              16 * y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y,
1323                                                              stride, 0);
1324                    interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1325                                                              16 * y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y,
1326                                                              stride, 0);
1327            }
1328    
1329            interpolate8x8_switch(dec->refn[2].u, backward.u, 8 * x_pos, 8 * y_pos,
1330                                                      b_uv_dx, b_uv_dy, stride2, 0);
1331            interpolate8x8_switch(dec->refn[2].v, backward.v, 8 * x_pos, 8 * y_pos,
1332                                                      b_uv_dx, b_uv_dy, stride2, 0);
1333    
1334            interpolate8x8_avg2(dec->cur.y + (16 * y_pos * stride) + 16 * x_pos,
1335                                                    dec->cur.y + (16 * y_pos * stride) + 16 * x_pos,
1336                                                    dec->refn[2].y + (16 * y_pos * stride) + 16 * x_pos,
1337                                                    stride, 1, 8);
1338    
1339            interpolate8x8_avg2(dec->cur.y + (16 * y_pos * stride) + 16 * x_pos + 8,
1340                                                    dec->cur.y + (16 * y_pos * stride) + 16 * x_pos + 8,
1341                                                    dec->refn[2].y + (16 * y_pos * stride) + 16 * x_pos + 8,
1342                                                    stride, 1, 8);
1343    
1344            interpolate8x8_avg2(dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1345                                                    dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1346                                                    dec->refn[2].y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1347                                                    stride, 1, 8);
1348    
1349            interpolate8x8_avg2(dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1350                                                    dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1351                                                    dec->refn[2].y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1352                                                    stride, 1, 8);
1353    
1354            interpolate8x8_avg2(dec->cur.u + (8 * y_pos * stride2) + 8 * x_pos,
1355                                                    dec->cur.u + (8 * y_pos * stride2) + 8 * x_pos,
1356                                                    dec->refn[2].u + (8 * y_pos * stride2) + 8 * x_pos,
1357                                                    stride2, 1, 8);
1358    
1359            interpolate8x8_avg2(dec->cur.v + (8 * y_pos * stride2) + 8 * x_pos,
1360                                                    dec->cur.v + (8 * y_pos * stride2) + 8 * x_pos,
1361                                                    dec->refn[2].v + (8 * y_pos * stride2) + 8 * x_pos,
1362                                                    stride2, 1, 8);
1363    
1364          stop_comp_timer();          stop_comp_timer();
1365    
1366          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
1367          {                  int direction = dec->alternate_vertical_scan ? 2 : 0;
1368    
1369                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5-i)))                 // coded
1370                  {                  {
1371                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear                          memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear
1372    
1373                          start_timer();                          start_timer();
1374                          get_inter_block(bs, &block[i*64]);                          get_inter_block(bs, &block[i * 64], direction);
1375                          stop_coding_timer();                          stop_coding_timer();
1376    
1377                          start_timer();                          start_timer();
1378                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
                         {  
1379                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                                  dequant_inter(&data[i*64], &block[i*64], iQuant);
1380                          }                          } else {
                         else  
                         {  
1381                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);
1382                          }                          }
1383                          stop_iquant_timer();                          stop_iquant_timer();
# Line 926  Line 1388 
1388                  }                  }
1389          }          }
1390    
1391          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
1392                  next_block = stride;                  next_block = stride;
1393                  stride *= 2;                  stride *= 2;
1394          }          }
# Line 951  Line 1412 
1412    
1413  // add by MinChen <chenm001@163.com>  // add by MinChen <chenm001@163.com>
1414  // for decode B-frame dbquant  // for decode B-frame dbquant
1415  int32_t __inline get_dbquant(Bitstream * bs)  int32_t __inline
1416    get_dbquant(Bitstream * bs)
1417  {  {
1418          if (!BitstreamGetBit(bs))               // '0'          if (!BitstreamGetBit(bs))               // '0'
1419                  return(0);                  return(0);
# Line 968  Line 1430 
1430  // 01       1  // 01       1
1431  // 001      2  // 001      2
1432  // 0001     3  // 0001     3
1433  int32_t __inline get_mbtype(Bitstream * bs)  int32_t __inline
1434    get_mbtype(Bitstream * bs)
1435  {  {
1436          int32_t mb_type;          int32_t mb_type;
1437    
1438          for(mb_type=0;mb_type<=3;mb_type++){          for(mb_type=0;mb_type<=3;mb_type++){
1439                  if  (BitstreamGetBit(bs))                  if  (BitstreamGetBit(bs))
1440                          break;                          break;
# Line 982  Line 1446 
1446                  return(-1);                  return(-1);
1447  }  }
1448    
1449  void decoder_bframe(DECODER * dec, Bitstream * bs, int quant, int fcode_forward, int fcode_backward)  void
1450    decoder_bframe(DECODER * dec,
1451                               Bitstream * bs,
1452                               int quant,
1453                               int fcode_forward,
1454                               int fcode_backward)
1455  {  {
   
1456          uint32_t        x, y;          uint32_t        x, y;
1457          VECTOR          mv, zeromv;          VECTOR mv;
1458            const VECTOR zeromv = {0,0};
1459    #ifdef BFRAMES_DEC_DEBUG
1460            FILE *fp;
1461            static char first=0;
1462    #define BFRAME_DEBUG    if (!first && fp){ \
1463                    fprintf(fp,"Y=%3d   X=%3d   MB=%2d   CBP=%02X\n",y,x,mb->mb_type,mb->cbp); \
1464            }
1465    #endif
1466    
1467          start_timer();          start_timer();
1468          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,
1469          //image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing);                                     dec->width, dec->height);
1470            image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height,
1471                                       dec->width, dec->height);
1472          stop_edges_timer();          stop_edges_timer();
1473    
1474    #ifdef BFRAMES_DEC_DEBUG
1475            if (!first){
1476                    fp=fopen("C:\\XVIDDBG.TXT","w");
1477            }
1478    #endif
1479    
1480          for (y = 0; y < dec->mb_height; y++)          for (y = 0; y < dec->mb_height; y++) {
         {  
1481                  // Initialize Pred Motion Vector                  // Initialize Pred Motion Vector
1482                  dec->p_fmv.x = dec->p_fmv.y = dec->p_bmv.x = dec->p_bmv.y = 0;                  dec->p_fmv = dec->p_bmv = zeromv;
1483                  for (x = 0; x < dec->mb_width; x++)                  for (x = 0; x < dec->mb_width; x++) {
                 {  
1484                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
1485                          MACROBLOCK * last_mb = &dec->last_mbs[y*dec->mb_width + x];                          MACROBLOCK * last_mb = &dec->last_mbs[y*dec->mb_width + x];
1486    
1487                          mb->mvs[0].x=mb->mvs[0].y=zeromv.x = zeromv.y = mv.x = mv.y = 0;                          mv =
1488                            mb->b_mvs[0] = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] =
1489                            mb->mvs[0] = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = zeromv;
1490    
1491                          // the last P_VOP is skip macroblock ?                          // the last P_VOP is skip macroblock ?
1492                          if (last_mb->mode == MODE_NOT_CODED){                          if (last_mb->mode == MODE_NOT_CODED){
1493                                  //DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y);                                  //DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y);
                                 mb->mb_type = MODE_FORWARD;  
1494                                  mb->cbp = 0;                                  mb->cbp = 0;
1495                                  mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;  #ifdef BFRAMES_DEC_DEBUG
1496                                  mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                  mb->mb_type = MODE_NOT_CODED;
1497                                  mb->quant = 8;          BFRAME_DEBUG
1498    #endif
1499                                    mb->mb_type = MODE_FORWARD;
1500                                    mb->quant = last_mb->quant;
1501                                    //mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1502                                    //mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
1503    
1504                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, mb->quant, 1);
1505                                  continue;                                  continue;
1506                          }                          }
1507    
                         //t=BitstreamShowBits(bs,32);  
   
1508                          if (!BitstreamGetBit(bs)){      // modb=='0'                          if (!BitstreamGetBit(bs)){      // modb=='0'
1509                                  const uint8_t modb2=BitstreamGetBit(bs);                                  const uint8_t modb2=BitstreamGetBit(bs);
1510    
# Line 1033  Line 1518 
1518                                  if (mb->mb_type && mb->cbp){                                  if (mb->mb_type && mb->cbp){
1519                                          quant += get_dbquant(bs);                                          quant += get_dbquant(bs);
1520    
1521                                          if (quant > 31)                                          if (quant > 31) {
                                         {  
1522                                                  quant = 31;                                                  quant = 31;
1523                                          }                                          } else if (quant < 1) {
                                         else if (mb->quant < 1)  
                                         {  
1524                                                  quant = 1;                                                  quant = 1;
1525                                          }                                          }
                                 } else {  
                                         quant = 8;  
1526                                  }                                  }
                                 mb->quant = quant;  
1527                          } else {                          } else {
1528                                  mb->mb_type = MODE_DIRECT_NONE_MV;                                  mb->mb_type = MODE_DIRECT_NONE_MV;
1529                                  mb->cbp = 0;                                  mb->cbp = 0;
1530                          }                          }
1531    
1532                          mb->mode = MODE_INTER;                          mb->quant = quant;
1533                            mb->mode = MODE_INTER4V;
1534                          //DEBUG1("Switch bm_type=",mb->mb_type);                          //DEBUG1("Switch bm_type=",mb->mb_type);
1535    
1536                          switch(mb->mb_type)  #ifdef BFRAMES_DEC_DEBUG
1537                          {          BFRAME_DEBUG
1538    #endif
1539    
1540                            switch (mb->mb_type) {
1541                          case MODE_DIRECT:                          case MODE_DIRECT:
1542                                  get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], 1, zeromv);                                  get_b_motion_vector(dec, bs, x, y, &mv, 1, zeromv);
1543    
1544                          case MODE_DIRECT_NONE_MV:                          case MODE_DIRECT_NONE_MV:
1545                                  {       // Because this file is a C file not C++ so I use '{' to define var                                  {
1546                                          const int64_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;  
1547                                          int i;                                          int i;
1548    
1549                                          for(i=0;i<4;i++){                                          for(i=0;i<4;i++){
1550                                                  mb->mvs[i].x    = (int32_t)((TRB * last_mb->mvs[i].x)/TRD+mb->mvs[0].x);                                                  mb->mvs[i].x = (int32_t) ((TRB * last_mb->mvs[i].x)
1551                                                  mb->b_mvs[i].x  = (int32_t)((mb->mvs[0].x==0)?((TRB-TRD)*last_mb->mvs[i].x)/TRD:mb->mvs[i].x-last_mb->mvs[i].x);                                                                        / TRD + mv.x);
1552                                                  mb->mvs[i].y    = (int32_t)((TRB * last_mb->mvs[i].y)/TRD+mb->mvs[0].y);                                                  mb->b_mvs[i].x = (int32_t) ((mv.x == 0)
1553                                                  mb->b_mvs[i].y  = (int32_t)((mb->mvs[0].y==0)?((TRB-TRD)*last_mb->mvs[i].y)/TRD:mb->mvs[i].y-last_mb->mvs[i].y);                                                                                  ? ((TRB - TRD) * last_mb->mvs[i].x)
1554                                                                                      / TRD
1555                                                                                    : mb->mvs[i].x - last_mb->mvs[i].x);
1556                                                    mb->mvs[i].y = (int32_t) ((TRB * last_mb->mvs[i].y)
1557                                                                          / TRD + mv.y);
1558                                                    mb->b_mvs[i].y = (int32_t) ((mv.y == 0)
1559                                                                                    ? ((TRB - TRD) * last_mb->mvs[i].y)
1560                                                                                      / TRD
1561                                                                                : mb->mvs[i].y - last_mb->mvs[i].y);
1562                                          }                                          }
1563                                          //DEBUG("B-frame Direct!\n");                                          //DEBUG("B-frame Direct!\n");
1564                                  }                                  }
1565                                  mb->mode = MODE_INTER4V;                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1566                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], mb, x, y, mb->cbp, bs);                                                                                             mb, x, y, bs);
1567                                  break;                                  break;
1568    
1569                          case MODE_INTERPOLATE:                          case MODE_INTERPOLATE:
1570                                  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,
1571                                  dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                                          dec->p_fmv);
1572                                  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];
1573    
1574                                  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],
1575                                  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);
1576                                  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] =
1577                                            mb->b_mvs[3] = mb->b_mvs[0];
1578    
1579                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], mb, x, y, mb->cbp, bs);                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1580                                                                                               mb, x, y, bs);
1581                                  //DEBUG("B-frame Bidir!\n");                                  //DEBUG("B-frame Bidir!\n");
1582                                  break;                                  break;
1583    
1584                          case MODE_BACKWARD:                          case MODE_BACKWARD:
1585                                  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,
1586                                  dec->p_bmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                                          dec->p_bmv);
1587                                  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];
1588    
1589                                    mb->mode = MODE_INTER;
1590                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);
1591                                  //DEBUG("B-frame Backward!\n");                                  //DEBUG("B-frame Backward!\n");
1592                                  break;                                  break;
1593    
1594                          case MODE_FORWARD:                          case MODE_FORWARD:
1595                                  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,
1596                                  dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                                                          dec->p_fmv);
1597                                  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];
1598    
1599                                    mb->mode = MODE_INTER;
1600                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);                                  decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1601                                  //DEBUG("B-frame Forward!\n");                                  //DEBUG("B-frame Forward!\n");
1602                                  break;                                  break;
# Line 1112  Line 1607 
1607    
1608                  }       // end of FOR                  }       // end of FOR
1609          }          }
1610    #ifdef BFRAMES_DEC_DEBUG
1611            if (!first){
1612                    first=1;
1613                    if (fp)
1614                            fclose(fp);
1615            }
1616    #endif
1617  }  }
1618    
1619  // swap two MACROBLOCK array  // swap two MACROBLOCK array
1620  void mb_swap(MACROBLOCK **mb1, MACROBLOCK **mb2)  void
1621    mb_swap(MACROBLOCK ** mb1,
1622                    MACROBLOCK ** mb2)
1623  {  {
1624          MACROBLOCK *temp=*mb1;          MACROBLOCK *temp=*mb1;
1625    
1626          *mb1=*mb2;          *mb1=*mb2;
1627          *mb2=temp;          *mb2=temp;
1628  }  }
1629    
1630  int decoder_decode(DECODER * dec, XVID_DEC_FRAME * frame)  int
1631    decoder_decode(DECODER * dec,
1632                               XVID_DEC_FRAME * frame, XVID_DEC_STATS * stats)
1633  {  {
1634    
1635          Bitstream bs;          Bitstream bs;
1636          uint32_t rounding;          uint32_t rounding;
1637            uint32_t reduced_resolution;
1638          uint32_t quant;          uint32_t quant;
1639          uint32_t fcode_forward;          uint32_t fcode_forward;
1640          uint32_t fcode_backward;          uint32_t fcode_backward;
1641          uint32_t intra_dc_threshold;          uint32_t intra_dc_threshold;
1642            VECTOR gmc_mv[5];
1643          uint32_t vop_type;          uint32_t vop_type;
1644            int success = 0;
1645    
1646          start_global_timer();          start_global_timer();
1647    
1648            dec->out_frm = (frame->colorspace == XVID_CSP_EXTERN) ? frame->image : NULL;
1649    
1650          BitstreamInit(&bs, frame->bitstream, frame->length);          BitstreamInit(&bs, frame->bitstream, frame->length);
1651    
1652            // XXX: 0x7f is only valid whilst decoding vfw xvid/divx5 avi's
1653            if(frame->length == 1 && BitstreamShowBits(&bs, 8) == 0x7f)
1654            {
1655                    if (stats)
1656                            stats->notify = XVID_DEC_VOP;
1657                    frame->length = 1;
1658                    image_output(&dec->refn[0], dec->width, dec->height, dec->edged_width,
1659                                             frame->image, frame->stride, frame->colorspace, dec->interlacing);
1660                    emms();
1661                    return XVID_ERR_OK;
1662            }
1663    
1664    start:
1665          // add by chenm001 <chenm001@163.com>          // add by chenm001 <chenm001@163.com>
1666          // for support B-frame to reference last 2 frame          // for support B-frame to reference last 2 frame
1667          dec->frames ++;          dec->frames ++;
         vop_type=BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward, &fcode_backward, &intra_dc_threshold);  
1668    
1669          dec->p_bmv.x=dec->p_bmv.y=dec->p_fmv.y=dec->p_fmv.y=0;          // init pred vector to 0  xxx:
1670          switch (vop_type)          vop_type =
1671                    BitstreamReadHeaders(&bs, dec, &rounding, &reduced_resolution,
1672                            &quant, &fcode_forward, &fcode_backward, &intra_dc_threshold, gmc_mv);
1673    
1674            //DPRINTF(DPRINTF_HEADER, "vop_type=%i", vop_type);
1675    
1676            if (vop_type == -1 && success)
1677                    goto done;
1678    
1679            if (vop_type == -2 || vop_type == -3)
1680            {
1681                    if (vop_type == -3)
1682                            decoder_resize(dec);
1683    
1684                    if (stats)
1685          {          {
1686                            stats->notify = XVID_DEC_VOL;
1687                            stats->data.vol.general = 0;
1688                            if (dec->interlacing)
1689                                    stats->data.vol.general |= XVID_INTERLACING;
1690                            stats->data.vol.width = dec->width;
1691                            stats->data.vol.height = dec->height;
1692                            stats->data.vol.aspect_ratio = dec->aspect_ratio;
1693                            stats->data.vol.par_width = dec->par_width;
1694                            stats->data.vol.par_height = dec->par_height;
1695                            frame->length = BitstreamPos(&bs) / 8;
1696                            return XVID_ERR_OK;
1697                    }
1698                    goto xxx;
1699            }
1700    
1701            dec->p_bmv.x = dec->p_bmv.y = dec->p_fmv.y = dec->p_fmv.y = 0;  // init pred vector to 0
1702    
1703            switch (vop_type) {
1704          case P_VOP :          case P_VOP :
1705                  decoder_pframe(dec, &bs, rounding, quant, fcode_forward, intra_dc_threshold);                  decoder_pframe(dec, &bs, rounding, reduced_resolution, quant,
1706                                                    fcode_forward, intra_dc_threshold, NULL);
1707    #ifdef BFRAMES_DEC
1708                  DEBUG1("P_VOP  Time=",dec->time);                  DEBUG1("P_VOP  Time=",dec->time);
1709    #endif
1710                  break;                  break;
1711    
1712          case I_VOP :          case I_VOP :
1713                  decoder_iframe(dec, &bs, quant, intra_dc_threshold);                  decoder_iframe(dec, &bs, reduced_resolution, quant, intra_dc_threshold);
1714    #ifdef BFRAMES_DEC
1715                  DEBUG1("I_VOP  Time=",dec->time);                  DEBUG1("I_VOP  Time=",dec->time);
1716    #endif
1717                  break;                  break;
1718    
1719          case B_VOP :          case B_VOP :
1720    #ifdef BFRAMES_DEC
1721                  if (dec->time_pp > dec->time_bp){                  if (dec->time_pp > dec->time_bp){
1722                          DEBUG1("B_VOP  Time=",dec->time);                          DEBUG1("B_VOP  Time=",dec->time);
1723                          decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);                          decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);
1724                  } else {                  } else {
1725                          DEBUG("broken B-frame!");                          DEBUG("broken B-frame!");
1726                  }                  }
1727    #else
1728                    image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1729    #endif
1730                    break;
1731    
1732            case S_VOP :
1733                    decoder_pframe(dec, &bs, rounding, reduced_resolution, quant,
1734                                                    fcode_forward, intra_dc_threshold, gmc_mv);
1735                  break;                  break;
1736    
1737          case N_VOP :    // vop not coded          case N_VOP :    // vop not coded
1738                    // when low_delay==0, N_VOP's should interpolate between the past and future frames
1739                    image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1740    #ifdef BFRAMES_DEC
1741                    DEBUG1("N_VOP  Time=", dec->time);
1742    #endif
1743                  break;                  break;
1744    
1745          default :          default :
1746                    if (stats)
1747                            stats->notify = 0;
1748    
1749                    emms();
1750                  return XVID_ERR_FAIL;                  return XVID_ERR_FAIL;
1751          }          }
1752    
         frame->length = BitstreamPos(&bs) / 8;  
1753    
1754          if (dec->frames >= 1){          /* reduced resolution deblocking filter */
1755                  start_timer();  
1756                  if ((vop_type == I_VOP || vop_type == P_VOP))          if (reduced_resolution)
1757                  {                  {
1758                          image_output(&dec->refn[0], dec->width, dec->height, dec->edged_width,                  const int rmb_width = (dec->width + 31) / 32;
1759                                       frame->image, frame->stride, frame->colorspace);                  const int rmb_height = (dec->height + 31) / 32;
1760                  } else if (vop_type == B_VOP) {                  const int edged_width2 = dec->edged_width /2;
1761                    int i,j;
1762    
1763                    /* horizontal deblocking */
1764    
1765                    for (j = 1; j < rmb_height*2; j++)      // luma: j,i in block units
1766                    for (i = 0; i < rmb_width*2; i++)
1767                    {
1768                            if (dec->mbs[(j-1)/2*dec->mb_width + (i/2)].mode != MODE_NOT_CODED ||
1769                                    dec->mbs[(j+0)/2*dec->mb_width + (i/2)].mode != MODE_NOT_CODED)
1770                            {
1771                                    xvid_HFilter_31_C(dec->cur.y + (j*16 - 1)*dec->edged_width + i*16,
1772                                                                  dec->cur.y + (j*16 + 0)*dec->edged_width + i*16, 2);
1773                            }
1774                    }
1775    
1776                    for (j = 1; j < rmb_height; j++)        // chroma
1777                    for (i = 0; i < rmb_width; i++)
1778                    {
1779                            if (dec->mbs[(j-1)*dec->mb_width + i].mode != MODE_NOT_CODED ||
1780                                    dec->mbs[(j+0)*dec->mb_width + i].mode != MODE_NOT_CODED)
1781                            {
1782                                    hfilter_31(dec->cur.u + (j*16 - 1)*edged_width2 + i*16,
1783                                                                      dec->cur.u + (j*16 + 0)*edged_width2 + i*16, 2);
1784                                    hfilter_31(dec->cur.v + (j*16 - 1)*edged_width2 + i*16,
1785                                                                      dec->cur.v + (j*16 + 0)*edged_width2 + i*16, 2);
1786                            }
1787                    }
1788    
1789                    /* vertical deblocking */
1790    
1791                    for (j = 0; j < rmb_height*2; j++)              // luma: i,j in block units
1792                    for (i = 1; i < rmb_width*2; i++)
1793                    {
1794                            if (dec->mbs[(j/2)*dec->mb_width + (i-1)/2].mode != MODE_NOT_CODED ||
1795                                    dec->mbs[(j/2)*dec->mb_width + (i+0)/2].mode != MODE_NOT_CODED)
1796                            {
1797                                    vfilter_31(dec->cur.y + (j*16)*dec->edged_width + i*16 - 1,
1798                                                                  dec->cur.y + (j*16)*dec->edged_width + i*16 + 0,
1799                                                                      dec->edged_width, 2);
1800                            }
1801                    }
1802    
1803                    for (j = 0; j < rmb_height; j++)        // chroma
1804                    for (i = 1; i < rmb_width; i++)
1805                    {
1806                            if (dec->mbs[j*dec->mb_width + i - 1].mode != MODE_NOT_CODED ||
1807                                    dec->mbs[j*dec->mb_width + i + 0].mode != MODE_NOT_CODED)
1808                            {
1809                                    vfilter_31(dec->cur.u + (j*16)*edged_width2 + i*16 - 1,
1810                                                                      dec->cur.u + (j*16)*edged_width2 + i*16 + 0,
1811                                                                      edged_width2, 2);
1812                                    vfilter_31(dec->cur.v + (j*16)*edged_width2 + i*16 - 1,
1813                                                                      dec->cur.v + (j*16)*edged_width2 + i*16 + 0,
1814                                                                      edged_width2, 2);
1815                            }
1816                    }
1817            }
1818    
1819            BitstreamByteAlign(&bs);
1820    
1821    #ifdef BFRAMES_DEC
1822            // test if no B_VOP
1823            if (dec->low_delay || dec->frames == 0 || ((dec->packed_mode) && !(frame->length > BitstreamPos(&bs) / 8))) {
1824    #endif
1825                          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,                          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,
1826                                       frame->image, frame->stride, frame->colorspace);                                           frame->image, frame->stride, frame->colorspace, dec->interlacing);
1827    
1828    #ifdef BFRAMES_DEC
1829            } else {
1830                    if (dec->frames >= 1 && !(dec->packed_mode)) {
1831                            start_timer();
1832                            if ((vop_type == I_VOP || vop_type == P_VOP || vop_type == S_VOP)) {
1833                                    image_output(&dec->refn[0], dec->width, dec->height,
1834                                                             dec->edged_width, frame->image, frame->stride,
1835                                                             frame->colorspace, dec->interlacing);
1836                            } else if (vop_type == B_VOP) {
1837                                    image_output(&dec->cur, dec->width, dec->height,
1838                                                             dec->edged_width, frame->image, frame->stride,
1839                                                             frame->colorspace, dec->interlacing);
1840                  }                  }
1841                  stop_conv_timer();                  stop_conv_timer();
1842          }          }
1843          if (vop_type==I_VOP || vop_type==P_VOP){          }
1844    #endif
1845    
1846            if (vop_type == I_VOP || vop_type == P_VOP || vop_type == S_VOP) {
1847                  image_swap(&dec->refn[0], &dec->refn[1]);                  image_swap(&dec->refn[0], &dec->refn[1]);
1848                  image_swap(&dec->cur, &dec->refn[0]);                  image_swap(&dec->cur, &dec->refn[0]);
1849    
1850                  // swap MACROBLOCK                  // swap MACROBLOCK
1851                    // the Divx will not set the low_delay flage some times
1852                    // so follow code will wrong to not swap at that time
1853                    // this will broken bitstream! so I'm change it,
1854                    // But that is not the best way! can anyone tell me how
1855                    // to do another way?
1856                    // 18-07-2002   MinChen<chenm001@163.com>
1857                    //if (!dec->low_delay && vop_type == P_VOP)
1858                  if (vop_type==P_VOP)                  if (vop_type==P_VOP)
1859                          mb_swap(&dec->mbs, &dec->last_mbs);                          mb_swap(&dec->mbs, &dec->last_mbs);
1860          }          }
1861    
1862    
1863            if (success == 0 && dec->packed_mode)
1864            {
1865                    success = 1;
1866            //      if (frame->length > BitstreamPos(&bs) / 8)      // multiple vops packed together
1867                    goto start;
1868            }
1869    
1870    done :
1871    
1872            frame->length = BitstreamPos(&bs) / 8;
1873    
1874            if (stats)
1875            {
1876                    stats->notify = XVID_DEC_VOP;
1877                    stats->data.vop.time_base = (int)dec->time_base;
1878                    stats->data.vop.time_increment = 0;     //XXX: todo
1879            }
1880    
1881          emms();          emms();
1882    
1883          stop_global_timer();          stop_global_timer();

Legend:
Removed from v.1.15  
changed lines
  Added in v.1.37.2.17

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