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

Legend:
Removed from v.1.8  
changed lines
  Added in v.1.37.2.7

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