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

Diff of /xvidcore/src/decoder.c

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

revision 1.13, Sun Apr 28 23:35:25 2002 UTC revision 1.65, Tue Aug 10 21:58:55 2004 UTC
# Line 1  Line 1 
1  /**************************************************************************  /*****************************************************************************
2   *   *
3   *  XVID MPEG-4 VIDEO CODEC   *  XVID MPEG-4 VIDEO CODEC
4   *  -  Decoder main module  -   *  - Decoder Module -
5   *   *
6   *  This program is an implementation of a part of one or more MPEG-4   *  Copyright(C) 2002      MinChen <chenm001@163.com>
7   *  Video tools as specified in ISO/IEC 14496-2 standard.  Those intending   *               2002-2004 Peter Ross <pross@xvid.org>
  *  to use this software module in hardware or software products are  
  *  advised that its use may infringe existing patents or copyrights, and  
  *  any such use would be at such party's own risk.  The original  
  *  developer of this software module and his/her company, and subsequent  
  *  editors and their companies, will have no liability for use of this  
  *  software or modifications or derivatives thereof.  
8   *   *
9   *  This program is free software; you can redistribute it and/or modify   *  This program is free software; you can redistribute it and/or modify
10   *  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
# Line 26  Line 20 
20   *  along with this program; if not, write to the Free Software   *  along with this program; if not, write to the Free Software
21   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
22   *   *
  *************************************************************************/  
   
 /**************************************************************************  
  *  
  *  History:  
  *  
  *  22.04.2002  add some B-frame decode support;  chenm001 <chenm001@163.com>  
  *  29.03.2002  interlacing fix - compensated block wasn't being used when  
  *              reconstructing blocks, thus artifacts  
  *              interlacing speedup - used transfers to re-interlace  
  *              interlaced decoding should be as fast as progressive now  
  *  26.03.2002  interlacing support - moved transfers outside decode loop  
  *  26.12.2001  decoder_mbinter: dequant/idct moved within if(coded) block  
  *  22.12.2001  lock based interpolation  
  *  01.12.2001  inital version; (c)2001 peter ross <pross@cs.rmit.edu.au>  
  *  
23   *  $Id$   *  $Id$
24   *   *
25   *************************************************************************/   ****************************************************************************/
26    
27    #include <stdio.h>
28  #include <stdlib.h>  #include <stdlib.h>
29  #include <string.h>  #include <string.h>
30    
31    #ifdef BFRAMES_DEC_DEBUG
32            #define BFRAMES_DEC
33    #endif
34    
35  #include "xvid.h"  #include "xvid.h"
36  #include "portab.h"  #include "portab.h"
37    #include "global.h"
38    
39  #include "decoder.h"  #include "decoder.h"
40  #include "bitstream/bitstream.h"  #include "bitstream/bitstream.h"
41  #include "bitstream/mbcoding.h"  #include "bitstream/mbcoding.h"
42    
43  #include "quant/quant_h263.h"  #include "quant/quant.h"
44  #include "quant/quant_mpeg4.h"  #include "quant/quant_matrix.h"
45  #include "dct/idct.h"  #include "dct/idct.h"
46  #include "dct/fdct.h"  #include "dct/fdct.h"
47  #include "utils/mem_transfer.h"  #include "utils/mem_transfer.h"
48  #include "image/interpolate8x8.h"  #include "image/interpolate8x8.h"
49    #include "image/reduced.h"
50    #include "image/font.h"
51    #include "image/qpel.h"
52    
53  #include "bitstream/mbcoding.h"  #include "bitstream/mbcoding.h"
54  #include "prediction/mbprediction.h"  #include "prediction/mbprediction.h"
55  #include "utils/timer.h"  #include "utils/timer.h"
56  #include "utils/emms.h"  #include "utils/emms.h"
57    #include "motion/motion.h"
58    #include "motion/gmc.h"
59    
60  #include "image/image.h"  #include "image/image.h"
61  #include "image/colorspace.h"  #include "image/colorspace.h"
62    #include "image/postprocessing.h"
63  #include "utils/mem_align.h"  #include "utils/mem_align.h"
64    
65  int decoder_create(XVID_DEC_PARAM * param)  static int
66  {  decoder_resize(DECODER * dec)
         DECODER * dec;  
   
         dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);  
         if (dec == NULL)  
67          {          {
68                  return XVID_ERR_MEMORY;          /* free existing */
69          }          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
70          param->handle = dec;          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
71            image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
72          dec->width = param->width;          image_destroy(&dec->tmp, dec->edged_width, dec->edged_height);
73          dec->height = param->height;          image_destroy(&dec->qtmp, dec->edged_width, dec->edged_height);
74    
75            image_destroy(&dec->gmc, dec->edged_width, dec->edged_height);
76    
77            if (dec->last_mbs)
78                    xvid_free(dec->last_mbs);
79            if (dec->mbs)
80                    xvid_free(dec->mbs);
81            if (dec->qscale)
82                    xvid_free(dec->qscale);
83    
84            /* realloc */
85          dec->mb_width = (dec->width + 15) / 16;          dec->mb_width = (dec->width + 15) / 16;
86          dec->mb_height = (dec->height + 15) / 16;          dec->mb_height = (dec->height + 15) / 16;
87    
88          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;
89          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;
90    
91          if (image_create(&dec->cur, dec->edged_width, dec->edged_height))          if (image_create(&dec->cur, dec->edged_width, dec->edged_height)) {
         {  
92                  xvid_free(dec);                  xvid_free(dec);
93                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
94          }          }
95    
96          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height)) {
         {  
97                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
98                  xvid_free(dec);                  xvid_free(dec);
99                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
100          }          }
101          // add by chenm001 <chenm001@163.com>  
102          // for support B-frame to reference last 2 frame          /* Support B-frame to reference last 2 frame */
103          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {
104          {                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
105                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
106                    xvid_free(dec);
107                    return XVID_ERR_MEMORY;
108            }
109            if (image_create(&dec->tmp, dec->edged_width, dec->edged_height)) {
110                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
111                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
112                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
113                  xvid_free(dec);                  xvid_free(dec);
114                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
115          }          }
116    
117          dec->mbs = xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height, CACHE_LINE);          if (image_create(&dec->qtmp, dec->edged_width, dec->edged_height)) {
         if (dec->mbs == NULL)  
         {  
118                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
119                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
120                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
121                    image_destroy(&dec->tmp, dec->edged_width, dec->edged_height);
122                  xvid_free(dec);                  xvid_free(dec);
123                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
124          }          }
125    
126            if (image_create(&dec->gmc, dec->edged_width, dec->edged_height)) {
127                    image_destroy(&dec->qtmp, dec->edged_width, dec->edged_height);
128                    image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
129                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
130                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
131                    image_destroy(&dec->tmp, dec->edged_width, dec->edged_height);
132                    xvid_free(dec);
133                    return XVID_ERR_MEMORY;
134            }
135    
136            dec->mbs =
137                    xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
138                                            CACHE_LINE);
139            if (dec->mbs == NULL) {
140                    image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
141                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
142                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
143                    image_destroy(&dec->tmp, dec->edged_width, dec->edged_height);
144                    image_destroy(&dec->qtmp, dec->edged_width, dec->edged_height);
145                    xvid_free(dec);
146                    return XVID_ERR_MEMORY;
147            }
148            memset(dec->mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
149    
150            /* For skip MB flag */
151            dec->last_mbs =
152                    xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
153                                            CACHE_LINE);
154            if (dec->last_mbs == NULL) {
155                    xvid_free(dec->mbs);
156                    image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
157                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
158                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
159                    image_destroy(&dec->tmp, dec->edged_width, dec->edged_height);
160                    image_destroy(&dec->qtmp, dec->edged_width, dec->edged_height);
161                    xvid_free(dec);
162                    return XVID_ERR_MEMORY;
163            }
164    
165            memset(dec->last_mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
166    
167            /* nothing happens if that fails */
168            dec->qscale =
169                    xvid_malloc(sizeof(int) * dec->mb_width * dec->mb_height, CACHE_LINE);
170    
171            if (dec->qscale)
172                    memset(dec->qscale, 0, sizeof(int) * dec->mb_width * dec->mb_height);
173    
174            return 0;
175    }
176    
177    
178    int
179    decoder_create(xvid_dec_create_t * create)
180    {
181            DECODER *dec;
182    
183            if (XVID_VERSION_MAJOR(create->version) != 1)   /* v1.x.x */
184                    return XVID_ERR_VERSION;
185    
186            dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);
187            if (dec == NULL) {
188                    return XVID_ERR_MEMORY;
189            }
190    
191            memset(dec, 0, sizeof(DECODER));
192    
193            dec->mpeg_quant_matrices = xvid_malloc(sizeof(uint16_t) * 64 * 8, CACHE_LINE);
194            if (dec->mpeg_quant_matrices == NULL) {
195                    xvid_free(dec);
196                    return XVID_ERR_MEMORY;
197            }
198    
199            create->handle = dec;
200    
201            dec->width = create->width;
202            dec->height = create->height;
203    
204            image_null(&dec->cur);
205            image_null(&dec->refn[0]);
206            image_null(&dec->refn[1]);
207            image_null(&dec->tmp);
208            image_null(&dec->qtmp);
209    
210            /* image based GMC */
211            image_null(&dec->gmc);
212    
213            dec->mbs = NULL;
214            dec->last_mbs = NULL;
215            dec->qscale = NULL;
216    
217          init_timer();          init_timer();
218            init_postproc(&dec->postproc);
219            init_mpeg_matrix(dec->mpeg_quant_matrices);
220    
221            /* For B-frame support (used to save reference frame's time */
222            dec->frames = 0;
223            dec->time = dec->time_base = dec->last_time_base = 0;
224            dec->low_delay = 0;
225            dec->packed_mode = 0;
226            dec->time_inc_resolution = 1; /* until VOL header says otherwise */
227    
228            dec->fixed_dimensions = (dec->width > 0 && dec->height > 0);
229    
230          return XVID_ERR_OK;          if (dec->fixed_dimensions)
231                    return decoder_resize(dec);
232            else
233                    return 0;
234  }  }
235    
236    
237  int decoder_destroy(DECODER * dec)  int
238    decoder_destroy(DECODER * dec)
239  {  {
240            xvid_free(dec->last_mbs);
241          xvid_free(dec->mbs);          xvid_free(dec->mbs);
242            xvid_free(dec->qscale);
243    
244            /* image based GMC */
245            image_destroy(&dec->gmc, dec->edged_width, dec->edged_height);
246    
247          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);          image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
248            image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
249            image_destroy(&dec->tmp, dec->edged_width, dec->edged_height);
250            image_destroy(&dec->qtmp, dec->edged_width, dec->edged_height);
251          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
252            xvid_free(dec->mpeg_quant_matrices);
253          xvid_free(dec);          xvid_free(dec);
254    
255          write_timer();          write_timer();
256          return XVID_ERR_OK;          return 0;
257  }  }
258    
259    static const int32_t dquant_table[4] = {
   
 static const int32_t dquant_table[4] =  
 {  
260          -1, -2, 1, 2          -1, -2, 1, 2
261  };  };
262    
263    /* decode an intra macroblock */
264  // decode an intra macroblock  static void
265    decoder_mbintra(DECODER * dec,
 void decoder_mbintra(DECODER * dec,  
266                       MACROBLOCK * pMB,                       MACROBLOCK * pMB,
267                       const uint32_t x_pos,                       const uint32_t x_pos,
268                       const uint32_t y_pos,                       const uint32_t y_pos,
# Line 157  Line 270 
270                       const uint32_t cbp,                       const uint32_t cbp,
271                       Bitstream * bs,                       Bitstream * bs,
272                       const uint32_t quant,                       const uint32_t quant,
273                       const uint32_t intra_dc_threshold)                                  const uint32_t intra_dc_threshold,
274                                    const unsigned int bound,
275                                    const int reduced_resolution)
276  {  {
277    
278          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);          DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
# Line 170  Line 285 
285          uint32_t iQuant = pMB->quant;          uint32_t iQuant = pMB->quant;
286          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
287    
288            if (reduced_resolution) {
289                    pY_Cur = dec->cur.y + (y_pos << 5) * stride + (x_pos << 5);
290                    pU_Cur = dec->cur.u + (y_pos << 4) * stride2 + (x_pos << 4);
291                    pV_Cur = dec->cur.v + (y_pos << 4) * stride2 + (x_pos << 4);
292            }else{
293          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
294          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
295          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
296            }
297    
298          memset(block, 0, 6*64*sizeof(int16_t));         // clear          memset(block, 0, 6 * 64 * sizeof(int16_t));     /* clear */
299    
300          for (i = 0; i < 6; i++)          for (i = 0; i < 6; i++) {
         {  
301                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
302                  int16_t predictors[8];                  int16_t predictors[8];
303                  int start_coeff;                  int start_coeff;
304    
305                  start_timer();                  start_timer();
306                  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],
307                  if (!acpred_flag)                                           iQuant, iDcScaler, predictors, bound);
308                  {                  if (!acpred_flag) {
309                          pMB->acpred_directions[i] = 0;                          pMB->acpred_directions[i] = 0;
310                  }                  }
311                  stop_prediction_timer();                  stop_prediction_timer();
312    
313                  if (quant < intra_dc_threshold)                  if (quant < intra_dc_threshold) {
                 {  
314                          int dc_size;                          int dc_size;
315                          int dc_dif;                          int dc_dif;
316    
317                          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);
318                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;
319    
320                          if (dc_size > 8)                          if (dc_size > 8) {
321                          {                                  BitstreamSkip(bs, 1);   /* marker */
                                 BitstreamSkip(bs, 1);           // marker  
322                          }                          }
323    
324                          block[i*64 + 0] = dc_dif;                          block[i*64 + 0] = dc_dif;
325                          start_coeff = 1;                          start_coeff = 1;
326                  }  
327                  else                          DPRINTF(XVID_DEBUG_COEFF,"block[0] %i\n", dc_dif);
328                  {                  } else {
329                          start_coeff = 0;                          start_coeff = 0;
330                  }                  }
331    
332                  start_timer();                  start_timer();
333                  if (cbp & (1 << (5-i)))                 // coded                  if (cbp & (1 << (5 - i)))       /* coded */
334                  {                  {
335                          get_intra_block(bs, &block[i*64], pMB->acpred_directions[i], start_coeff);                          int direction = dec->alternate_vertical_scan ?
336                                    2 : pMB->acpred_directions[i];
337    
338                            get_intra_block(bs, &block[i * 64], direction, start_coeff);
339                  }                  }
340                  stop_coding_timer();                  stop_coding_timer();
341    
342                  start_timer();                  start_timer();
343                  add_acdc(pMB, i, &block[i*64], iDcScaler, predictors);                  add_acdc(pMB, i, &block[i * 64], iDcScaler, predictors, dec->bs_version);
344                  stop_prediction_timer();                  stop_prediction_timer();
345    
346                  start_timer();                  start_timer();
347                  if (dec->quant_type == 0)                  if (dec->quant_type == 0) {
348                  {                          dequant_h263_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler, dec->mpeg_quant_matrices);
349                          dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);                  } else {
350                  }                          dequant_mpeg_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler, dec->mpeg_quant_matrices);
                 else  
                 {  
                         dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);  
351                  }                  }
352                  stop_iquant_timer();                  stop_iquant_timer();
353    
354                  start_timer();                  start_timer();
355                  idct(&data[i*64]);                  idct(&data[i*64]);
356                  stop_idct_timer();                  stop_idct_timer();
357    
358          }          }
359    
360          if (dec->interlacing && pMB->field_dct)          if (dec->interlacing && pMB->field_dct) {
         {  
361                  next_block = stride;                  next_block = stride;
362                  stride *= 2;                  stride *= 2;
363          }          }
364    
365          start_timer();          start_timer();
366    
367            if (reduced_resolution)
368            {
369                    next_block*=2;
370                    copy_upsampled_8x8_16to8(pY_Cur, &data[0 * 64], stride);
371                    copy_upsampled_8x8_16to8(pY_Cur + 16, &data[1 * 64], stride);
372                    copy_upsampled_8x8_16to8(pY_Cur + next_block, &data[2 * 64], stride);
373                    copy_upsampled_8x8_16to8(pY_Cur + 16 + next_block, &data[3 * 64], stride);
374                    copy_upsampled_8x8_16to8(pU_Cur, &data[4 * 64], stride2);
375                    copy_upsampled_8x8_16to8(pV_Cur, &data[5 * 64], stride2);
376            }else{
377          transfer_16to8copy(pY_Cur,                  &data[0*64], stride);          transfer_16to8copy(pY_Cur,                  &data[0*64], stride);
378          transfer_16to8copy(pY_Cur + 8,              &data[1*64], stride);          transfer_16to8copy(pY_Cur + 8,              &data[1*64], stride);
379          transfer_16to8copy(pY_Cur + next_block,     &data[2*64], stride);          transfer_16to8copy(pY_Cur + next_block,     &data[2*64], stride);
380          transfer_16to8copy(pY_Cur + 8 + next_block, &data[3*64], stride);          transfer_16to8copy(pY_Cur + 8 + next_block, &data[3*64], stride);
381          transfer_16to8copy(pU_Cur,                  &data[4*64], stride2);          transfer_16to8copy(pU_Cur,                  &data[4*64], stride2);
382          transfer_16to8copy(pV_Cur,                  &data[5*64], stride2);          transfer_16to8copy(pV_Cur,                  &data[5*64], stride2);
383            }
384          stop_transfer_timer();          stop_transfer_timer();
385  }  }
386    
387    static void
388    decoder_mb_decode(DECODER * dec,
389                                    const uint32_t cbp,
390                                    Bitstream * bs,
391                                    uint8_t * pY_Cur,
392                                    uint8_t * pU_Cur,
393                                    uint8_t * pV_Cur,
394                                    int reduced_resolution,
395                                    const MACROBLOCK * pMB)
396    {
397            DECLARE_ALIGNED_MATRIX(data, 1, 64, int16_t, CACHE_LINE);
398    
399            int stride = dec->edged_width;
400            int next_block = stride * (reduced_resolution ? 16 : 8);
401            int i;
402            const uint32_t iQuant = pMB->quant;
403            const int direction = dec->alternate_vertical_scan ? 2 : 0;
404            typedef void (*get_inter_block_function_t)(
405                            Bitstream * bs,
406                            int16_t * block,
407                            int direction,
408                            const int quant,
409                            const uint16_t *matrix);
410            typedef void (*add_residual_function_t)(
411                            uint8_t *predicted_block,
412                            const int16_t *residual,
413                            int stride);
414    
415            const get_inter_block_function_t get_inter_block = (dec->quant_type == 0)
416                    ? (get_inter_block_function_t)get_inter_block_h263
417                    : (get_inter_block_function_t)get_inter_block_mpeg;
418    
419            const add_residual_function_t add_residual = (reduced_resolution)
420                    ? (add_residual_function_t)add_upsampled_8x8_16to8
421                    : (add_residual_function_t)transfer_16to8add;
422    
423            uint8_t *dst[6];
424            int strides[6];
425    
426    
427            if (dec->interlacing && pMB->field_dct) {
428                    next_block = stride;
429                    stride *= 2;
430            }
431    
432  #define SIGN(X) (((X)>0)?1:-1)          reduced_resolution = !!reduced_resolution;
433  #define ABS(X) (((X)>0)?(X):-(X))          dst[0] = pY_Cur;
434  static const uint32_t roundtab[16] =          dst[2] = pY_Cur + next_block;
435  { 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 };          dst[1] = dst[0] + (8<<reduced_resolution);
436            dst[3] = dst[2] + (8<<reduced_resolution);
437            dst[4] = pU_Cur;
438            dst[5] = pV_Cur;
439            strides[0] = strides[1] = strides[2] = strides[3] = stride;
440            strides[4] = stride/2;
441            strides[5] = stride/2;
442    
443            for (i = 0; i < 6; i++) {
444                    /* Process only coded blocks */
445                    if (cbp & (1 << (5 - i))) {
446    
447                            /* Clear the block */
448                            memset(&data[0], 0, 64*sizeof(int16_t));
449    
450  // decode an inter macroblock                          /* Decode coeffs and dequantize on the fly */
451                            start_timer();
452                            get_inter_block(bs, &data[0], direction, iQuant, get_inter_matrix(dec->mpeg_quant_matrices));
453                            stop_coding_timer();
454    
455                            /* iDCT */
456                            start_timer();
457                            idct(&data[0]);
458                            stop_idct_timer();
459    
460  void decoder_mbinter(DECODER * dec,                          /* Add this residual to the predicted block */
461                            start_timer();
462                            add_residual(dst[i], &data[0], strides[i]);
463                            stop_transfer_timer();
464                    }
465            }
466    }
467    
468    /* decode an inter macroblock */
469    static void
470    decoder_mbinter(DECODER * dec,
471                       const MACROBLOCK * pMB,                       const MACROBLOCK * pMB,
472                       const uint32_t x_pos,                       const uint32_t x_pos,
473                       const uint32_t y_pos,                       const uint32_t y_pos,
                      const uint32_t acpred_flag,  
474                       const uint32_t cbp,                       const uint32_t cbp,
475                       Bitstream * bs,                       Bitstream * bs,
476                       const uint32_t quant,                                  const uint32_t rounding,
477                       const uint32_t rounding)                                  const int reduced_resolution,
478                                    const int ref)
479  {  {
   
         DECLARE_ALIGNED_MATRIX(block,6, 64, int16_t, CACHE_LINE);  
         DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);  
   
480          uint32_t stride = dec->edged_width;          uint32_t stride = dec->edged_width;
481          uint32_t stride2 = stride / 2;          uint32_t stride2 = stride / 2;
         uint32_t next_block = stride * 8;  
482          uint32_t i;          uint32_t i;
483          uint32_t iQuant = pMB->quant;  
484          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;          uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
485    
486          int uv_dx, uv_dy;          int uv_dx, uv_dy;
487            VECTOR mv[4];   /* local copy of mvs */
488    
489            if (reduced_resolution) {
490                    pY_Cur = dec->cur.y + (y_pos << 5) * stride + (x_pos << 5);
491                    pU_Cur = dec->cur.u + (y_pos << 4) * stride2 + (x_pos << 4);
492                    pV_Cur = dec->cur.v + (y_pos << 4) * stride2 + (x_pos << 4);
493                    for (i = 0; i < 4; i++) {
494                            mv[i].x = RRV_MV_SCALEUP(pMB->mvs[i].x);
495                            mv[i].y = RRV_MV_SCALEUP(pMB->mvs[i].y);
496                    }
497            } else {
498          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
499          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
500          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
501                    for (i = 0; i < 4; i++)
502                            mv[i] = pMB->mvs[i];
503            }
504    
505          if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)          for (i = 0; i < 4; i++) {
506          {                  /* clip to valid range */
507                  uv_dx = pMB->mvs[0].x;                  int border = (int)(dec->mb_width - x_pos) << (5 + dec->quarterpel);
508                  uv_dy = pMB->mvs[0].y;                  if (mv[i].x > border) {
509                            DPRINTF(XVID_DEBUG_MV, "mv.x > max -- %d > %d, MB %d, %d", mv[i].x, border, x_pos, y_pos);
510                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;                          mv[i].x = border;
511                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;                  } else {
512                            border = (-(int)x_pos-1) << (5 + dec->quarterpel);
513                            if (mv[i].x < border) {
514                                    DPRINTF(XVID_DEBUG_MV, "mv.x < min -- %d < %d, MB %d, %d", mv[i].x, border, x_pos, y_pos);
515                                    mv[i].x = border;
516                            }
517          }          }
         else  
         {  
                 int sum;  
                 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) );  
518    
519                  sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;                  border = (int)(dec->mb_height - y_pos) << (5 + dec->quarterpel);
520                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  if (mv[i].y >  border) {
521                            DPRINTF(XVID_DEBUG_MV, "mv.y > max -- %d > %d, MB %d, %d", mv[i].y, border, x_pos, y_pos);
522                            mv[i].y = border;
523                    } else {
524                            border = (-(int)y_pos-1) << (5 + dec->quarterpel);
525                            if (mv[i].y < border) {
526                                    DPRINTF(XVID_DEBUG_MV, "mv.y < min -- %d < %d, MB %d, %d", mv[i].y, border, x_pos, y_pos);
527                                    mv[i].y = border;
528                            }
529                    }
530          }          }
531    
532          start_timer();          start_timer();
         interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos,     16*y_pos    , pMB->mvs[0].x, pMB->mvs[0].y, stride,  rounding);  
         interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos    , pMB->mvs[1].x, pMB->mvs[1].y, stride,  rounding);  
         interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos,     16*y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,  rounding);  
         interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,  rounding);  
         interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, rounding);  
         interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8*x_pos,      8*y_pos,      uv_dx,         uv_dy,         stride2, rounding);  
         stop_comp_timer();  
533    
534          for (i = 0; i < 6; i++)          if (pMB->mode != MODE_INTER4V) { /* INTER, INTER_Q, NOT_CODED, FORWARD, BACKWARD */
         {  
                 if (cbp & (1 << (5-i)))                 // coded  
                 {  
                         memset(&block[i*64], 0, 64 * sizeof(int16_t));          // clear  
535    
536                          start_timer();                  uv_dx = mv[0].x;
537                          get_inter_block(bs, &block[i*64]);                  uv_dy = mv[0].y;
538                          stop_coding_timer();                  if (dec->quarterpel) {
539                            uv_dx /= 2;
540                          start_timer();                          uv_dy /= 2;
541                          if (dec->quant_type == 0)                  }
542                          {                  uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
543                                  dequant_inter(&data[i*64], &block[i*64], iQuant);                  uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
544                          }  
545                    if (reduced_resolution)
546                            interpolate32x32_switch(dec->cur.y, dec->refn[0].y, 32*x_pos, 32*y_pos,
547                                                                            mv[0].x, mv[0].y, stride, rounding);
548                    else if (dec->quarterpel)
549                            interpolate16x16_quarterpel(dec->cur.y, dec->refn[ref].y, dec->qtmp.y, dec->qtmp.y + 64,
550                                                                            dec->qtmp.y + 128, 16*x_pos, 16*y_pos,
551                                                                                            mv[0].x, mv[0].y, stride, rounding);
552                          else                          else
553                          {                          interpolate16x16_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos, 16*y_pos,
554                                  dequant4_inter(&data[i*64], &block[i*64], iQuant);                                                                          mv[0].x, mv[0].y, stride, rounding);
                         }  
                         stop_iquant_timer();  
555    
556                          start_timer();          } else {        /* MODE_INTER4V */
557                          idct(&data[i*64]);  
558                          stop_idct_timer();                  if(dec->quarterpel) {
559                            uv_dx = (mv[0].x / 2) + (mv[1].x / 2) + (mv[2].x / 2) + (mv[3].x / 2);
560                            uv_dy = (mv[0].y / 2) + (mv[1].y / 2) + (mv[2].y / 2) + (mv[3].y / 2);
561                    } else {
562                            uv_dx = mv[0].x + mv[1].x + mv[2].x + mv[3].x;
563                            uv_dy = mv[0].y + mv[1].y + mv[2].y + mv[3].y;
564                    }
565    
566                    uv_dx = (uv_dx >> 3) + roundtab_76[uv_dx & 0xf];
567                    uv_dy = (uv_dy >> 3) + roundtab_76[uv_dy & 0xf];
568    
569                    if (reduced_resolution) {
570                            interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 32*x_pos, 32*y_pos,
571                                                                    mv[0].x, mv[0].y, stride, rounding);
572                            interpolate16x16_switch(dec->cur.y, dec->refn[0].y , 32*x_pos + 16, 32*y_pos,
573                                                                    mv[1].x, mv[1].y, stride, rounding);
574                            interpolate16x16_switch(dec->cur.y, dec->refn[0].y , 32*x_pos, 32*y_pos + 16,
575                                                                    mv[2].x, mv[2].y, stride, rounding);
576                            interpolate16x16_switch(dec->cur.y, dec->refn[0].y , 32*x_pos + 16, 32*y_pos + 16,
577                                                                    mv[3].x, mv[3].y, stride, rounding);
578                            interpolate16x16_switch(dec->cur.u, dec->refn[0].u , 16 * x_pos, 16 * y_pos,
579                                                                    uv_dx, uv_dy, stride2, rounding);
580                            interpolate16x16_switch(dec->cur.v, dec->refn[0].v , 16 * x_pos, 16 * y_pos,
581                                                                    uv_dx, uv_dy, stride2, rounding);
582    
583                    } else if (dec->quarterpel) {
584                            interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y , dec->qtmp.y, dec->qtmp.y + 64,
585                                                                            dec->qtmp.y + 128, 16*x_pos, 16*y_pos,
586                                                                            mv[0].x, mv[0].y, stride, rounding);
587                            interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y , dec->qtmp.y, dec->qtmp.y + 64,
588                                                                            dec->qtmp.y + 128, 16*x_pos + 8, 16*y_pos,
589                                                                            mv[1].x, mv[1].y, stride, rounding);
590                            interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y , dec->qtmp.y, dec->qtmp.y + 64,
591                                                                            dec->qtmp.y + 128, 16*x_pos, 16*y_pos + 8,
592                                                                            mv[2].x, mv[2].y, stride, rounding);
593                            interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y , dec->qtmp.y, dec->qtmp.y + 64,
594                                                                            dec->qtmp.y + 128, 16*x_pos + 8, 16*y_pos + 8,
595                                                                            mv[3].x, mv[3].y, stride, rounding);
596                    } else {
597                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y , 16*x_pos, 16*y_pos,
598                                                                    mv[0].x, mv[0].y, stride, rounding);
599                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y , 16*x_pos + 8, 16*y_pos,
600                                                                    mv[1].x, mv[1].y, stride, rounding);
601                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y , 16*x_pos, 16*y_pos + 8,
602                                                                    mv[2].x, mv[2].y, stride, rounding);
603                            interpolate8x8_switch(dec->cur.y, dec->refn[0].y , 16*x_pos + 8, 16*y_pos + 8,
604                                                                    mv[3].x, mv[3].y, stride, rounding);
605                    }
606            }
607    
608            /* chroma */
609            if (reduced_resolution) {
610                    interpolate16x16_switch(dec->cur.u, dec->refn[0].u, 16 * x_pos, 16 * y_pos,
611                                                                    uv_dx, uv_dy, stride2, rounding);
612                    interpolate16x16_switch(dec->cur.v, dec->refn[0].v, 16 * x_pos, 16 * y_pos,
613                                                                    uv_dx, uv_dy, stride2, rounding);
614            } else {
615                    interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8 * x_pos, 8 * y_pos,
616                                                                    uv_dx, uv_dy, stride2, rounding);
617                    interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8 * x_pos, 8 * y_pos,
618                                                                    uv_dx, uv_dy, stride2, rounding);
619                  }                  }
620    
621            stop_comp_timer();
622    
623            if (cbp)
624                    decoder_mb_decode(dec, cbp, bs, pY_Cur, pU_Cur, pV_Cur,
625                                                            reduced_resolution, pMB);
626          }          }
627    
628          if (dec->interlacing && pMB->field_dct)  static void
629    decoder_mbgmc(DECODER * dec,
630                                    MACROBLOCK * const pMB,
631                                    const uint32_t x_pos,
632                                    const uint32_t y_pos,
633                                    const uint32_t fcode,
634                                    const uint32_t cbp,
635                                    Bitstream * bs,
636                                    const uint32_t rounding)
637          {          {
638                  next_block = stride;          const uint32_t stride = dec->edged_width;
639                  stride *= 2;          const uint32_t stride2 = stride / 2;
640          }  
641            uint8_t *const pY_Cur=dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
642            uint8_t *const pU_Cur=dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
643            uint8_t *const pV_Cur=dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
644    
645            NEW_GMC_DATA * gmc_data = &dec->new_gmc_data;
646    
647            pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = pMB->amv;
648    
649          start_timer();          start_timer();
650          if (cbp & 32)  
651                  transfer_16to8add(pY_Cur,                  &data[0*64], stride);  /* this is where the calculations are done */
652          if (cbp & 16)  
653                  transfer_16to8add(pY_Cur + 8,              &data[1*64], stride);          gmc_data->predict_16x16(gmc_data,
654          if (cbp & 8)                          dec->cur.y + y_pos*16*stride + x_pos*16, dec->refn[0].y,
655                  transfer_16to8add(pY_Cur + next_block,     &data[2*64], stride);                          stride, stride, x_pos, y_pos, rounding);
656          if (cbp & 4)  
657                  transfer_16to8add(pY_Cur + 8 + next_block, &data[3*64], stride);          gmc_data->predict_8x8(gmc_data,
658          if (cbp & 2)                          dec->cur.u + y_pos*8*stride2 + x_pos*8, dec->refn[0].u,
659                  transfer_16to8add(pU_Cur,                  &data[4*64], stride2);                          dec->cur.v + y_pos*8*stride2 + x_pos*8, dec->refn[0].v,
660          if (cbp & 1)                          stride2, stride2, x_pos, y_pos, rounding);
661                  transfer_16to8add(pV_Cur,                  &data[5*64], stride2);  
662            gmc_data->get_average_mv(gmc_data, &pMB->amv, x_pos, y_pos, dec->quarterpel);
663    
664            pMB->amv.x = gmc_sanitize(pMB->amv.x, dec->quarterpel, fcode);
665            pMB->amv.y = gmc_sanitize(pMB->amv.y, dec->quarterpel, fcode);
666    
667            pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = pMB->amv;
668    
669          stop_transfer_timer();          stop_transfer_timer();
670    
671            if (cbp)
672                    decoder_mb_decode(dec, cbp, bs, pY_Cur, pU_Cur, pV_Cur, 0, pMB);
673    
674  }  }
675    
676    
677  void decoder_iframe(DECODER * dec, Bitstream * bs, int quant, int intra_dc_threshold)  static void
678    decoder_iframe(DECODER * dec,
679                                    Bitstream * bs,
680                                    int reduced_resolution,
681                                    int quant,
682                                    int intra_dc_threshold)
683  {  {
684            uint32_t bound;
685          uint32_t x, y;          uint32_t x, y;
686            uint32_t mb_width = dec->mb_width;
687            uint32_t mb_height = dec->mb_height;
688    
689          for (y = 0; y < dec->mb_height; y++)          if (reduced_resolution) {
690          {                  mb_width = (dec->width + 31) / 32;
691                  for (x = 0; x < dec->mb_width; x++)                  mb_height = (dec->height + 31) / 32;
692                  {          }
693                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];  
694            bound = 0;
695    
696            for (y = 0; y < mb_height; y++) {
697                    for (x = 0; x < mb_width; x++) {
698                            MACROBLOCK *mb;
699                          uint32_t mcbpc;                          uint32_t mcbpc;
700                          uint32_t cbpc;                          uint32_t cbpc;
701                          uint32_t acpred_flag;                          uint32_t acpred_flag;
702                          uint32_t cbpy;                          uint32_t cbpy;
703                          uint32_t cbp;                          uint32_t cbp;
704    
705                            while (BitstreamShowBits(bs, 9) == 1)
706                                    BitstreamSkip(bs, 9);
707    
708                            if (check_resync_marker(bs, 0))
709                            {
710                                    bound = read_video_packet_header(bs, dec, 0,
711                                                            &quant, NULL, NULL, &intra_dc_threshold);
712                                    x = bound % mb_width;
713                                    y = bound / mb_width;
714                            }
715                            mb = &dec->mbs[y * dec->mb_width + x];
716    
717                            DPRINTF(XVID_DEBUG_MB, "macroblock (%i,%i) %08x\n", x, y, BitstreamShowBits(bs, 32));
718    
719                          mcbpc = get_mcbpc_intra(bs);                          mcbpc = get_mcbpc_intra(bs);
720                          mb->mode = mcbpc & 7;                          mb->mode = mcbpc & 7;
721                          cbpc = (mcbpc >> 4);                          cbpc = (mcbpc >> 4);
722    
723                          acpred_flag = BitstreamGetBit(bs);                          acpred_flag = BitstreamGetBit(bs);
724    
                         if (mb->mode == MODE_STUFFING)  
                         {  
                                 DEBUG("-- STUFFING ?");  
                                 continue;  
                         }  
   
725                          cbpy = get_cbpy(bs, 1);                          cbpy = get_cbpy(bs, 1);
726                          cbp = (cbpy << 2) | cbpc;                          cbp = (cbpy << 2) | cbpc;
727    
728                          if (mb->mode == MODE_INTRA_Q)                          if (mb->mode == MODE_INTRA_Q) {
                         {  
729                                  quant += dquant_table[BitstreamGetBits(bs,2)];                                  quant += dquant_table[BitstreamGetBits(bs,2)];
730                                  if (quant > 31)                                  if (quant > 31) {
                                 {  
731                                          quant = 31;                                          quant = 31;
732                                  }                                  } else if (quant < 1) {
                                 else if (quant < 1)  
                                 {  
733                                          quant = 1;                                          quant = 1;
734                                  }                                  }
735                          }                          }
736                          mb->quant = quant;                          mb->quant = quant;
737                            mb->mvs[0].x = mb->mvs[0].y =
738                            mb->mvs[1].x = mb->mvs[1].y =
739                            mb->mvs[2].x = mb->mvs[2].y =
740                            mb->mvs[3].x = mb->mvs[3].y =0;
741    
742                          if (dec->interlacing)                          if (dec->interlacing) {
                         {  
743                                  mb->field_dct = BitstreamGetBit(bs);                                  mb->field_dct = BitstreamGetBit(bs);
744                                  DEBUG1("deci: field_dct: ", mb->field_dct);                                  DPRINTF(XVID_DEBUG_MB,"deci: field_dct: %i\n", mb->field_dct);
745                          }                          }
746    
747                          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,
748                                                            intra_dc_threshold, bound, reduced_resolution);
749    
750                  }                  }
751                    if(dec->out_frm)
752                            output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,0,y,mb_width);
753          }          }
754    
755  }  }
756    
757    
758  void get_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, int k, VECTOR * mv, int fcode)  static void
759    get_motion_vector(DECODER * dec,
760                                    Bitstream * bs,
761                                    int x,
762                                    int y,
763                                    int k,
764                                    VECTOR * ret_mv,
765                                    int fcode,
766                                    const int bound)
767  {  {
768    
769          int scale_fac = 1 << (fcode - 1);          const int scale_fac = 1 << (fcode - 1);
770          int high = (32 * scale_fac) - 1;          const int high = (32 * scale_fac) - 1;
771          int low = ((-32) * scale_fac);          const int low = ((-32) * scale_fac);
772          int range = (64 * scale_fac);          const int range = (64 * scale_fac);
   
         VECTOR pmv[4];  
         int32_t psad[4];  
   
         int mv_x, mv_y;  
         int pmv_x, pmv_y;  
773    
774            const VECTOR pmv = get_pmv2(dec->mbs, dec->mb_width, bound, x, y, k);
775            VECTOR mv;
776    
777          get_pmvdata(dec->mbs, x, y, dec->mb_width, k, pmv, psad);          mv.x = get_mv(bs, fcode);
778            mv.y = get_mv(bs, fcode);
779    
780          pmv_x = pmv[0].x;          DPRINTF(XVID_DEBUG_MV,"mv_diff (%i,%i) pred (%i,%i) result (%i,%i)\n", mv.x, mv.y, pmv.x, pmv.y, mv.x+pmv.x, mv.y+pmv.y);
         pmv_y = pmv[0].y;  
781    
782          mv_x = get_mv(bs, fcode);          mv.x += pmv.x;
783          mv_y = get_mv(bs, fcode);          mv.y += pmv.y;
784    
785          mv_x += pmv_x;          if (mv.x < low) {
786          mv_y += pmv_y;                  mv.x += range;
787            } else if (mv.x > high) {
788                    mv.x -= range;
789            }
790    
791          if (mv_x < low)          if (mv.y < low) {
792          {                  mv.y += range;
793                  mv_x += range;          } else if (mv.y > high) {
794                    mv.y -= range;
795          }          }
796          else if (mv_x > high)  
797          {          ret_mv->x = mv.x;
798                  mv_x -= range;          ret_mv->y = mv.y;
799          }          }
800    
801          if (mv_y < low)  /* for P_VOP set gmc_warp to NULL */
802    static void
803    decoder_pframe(DECODER * dec,
804                                    Bitstream * bs,
805                                    int rounding,
806                                    int reduced_resolution,
807                                    int quant,
808                                    int fcode,
809                                    int intra_dc_threshold,
810                                    const WARPPOINTS *const gmc_warp)
811          {          {
812                  mv_y += range;          uint32_t x, y;
813            uint32_t bound;
814            int cp_mb, st_mb;
815            uint32_t mb_width = dec->mb_width;
816            uint32_t mb_height = dec->mb_height;
817    
818            if (reduced_resolution) {
819                    mb_width = (dec->width + 31) / 32;
820                    mb_height = (dec->height + 31) / 32;
821          }          }
822          else if (mv_y > high)  
823          {          if (!dec->is_edged[0]) {
824                  mv_y -= range;                  start_timer();
825                    image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
826                                                    dec->width, dec->height, dec->bs_version);
827                    dec->is_edged[0] = 1;
828                    stop_edges_timer();
829          }          }
830    
831          mv->x = mv_x;          if (gmc_warp) {
832          mv->y = mv_y;                  /* accuracy: 0==1/2, 1=1/4, 2=1/8, 3=1/16 */
833                    generate_GMCparameters( dec->sprite_warping_points,
834                                    dec->sprite_warping_accuracy, gmc_warp,
835                                    dec->width, dec->height, &dec->new_gmc_data);
836    
837                    /* image warping is done block-based in decoder_mbgmc(), now */
838  }  }
839    
840            bound = 0;
841    
842  void decoder_pframe(DECODER * dec, Bitstream * bs, int rounding, int quant, int fcode, int intra_dc_threshold)          for (y = 0; y < mb_height; y++) {
843  {                  cp_mb = st_mb = 0;
844                    for (x = 0; x < mb_width; x++) {
845                            MACROBLOCK *mb;
846    
847          uint32_t x, y;                          /* skip stuffing */
848                            while (BitstreamShowBits(bs, 10) == 1)
849                                    BitstreamSkip(bs, 10);
850    
851          start_timer();                          if (check_resync_marker(bs, fcode - 1)) {
852          image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing);                                  bound = read_video_packet_header(bs, dec, fcode - 1,
853          stop_edges_timer();                                          &quant, &fcode, NULL, &intra_dc_threshold);
854                                    x = bound % mb_width;
855                                    y = bound / mb_width;
856                            }
857                            mb = &dec->mbs[y * dec->mb_width + x];
858    
859          for (y = 0; y < dec->mb_height; y++)                          DPRINTF(XVID_DEBUG_MB, "macroblock (%i,%i) %08x\n", x, y, BitstreamShowBits(bs, 32));
         {  
                 for (x = 0; x < dec->mb_width; x++)  
                 {  
                         MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];  
860    
861                          if (!BitstreamGetBit(bs))                       // not_coded                          if (!(BitstreamGetBit(bs)))     { /* block _is_ coded */
862                          {                                  uint32_t mcbpc, cbpc, cbpy, cbp;
863                                  uint32_t mcbpc;                                  uint32_t intra, acpred_flag = 0;
864                                  uint32_t cbpc;                                  int mcsel = 0;          /* mcsel: '0'=local motion, '1'=GMC */
                                 uint32_t acpred_flag;  
                                 uint32_t cbpy;  
                                 uint32_t cbp;  
                                 uint32_t intra;  
865    
866                                    cp_mb++;
867                                  mcbpc = get_mcbpc_inter(bs);                                  mcbpc = get_mcbpc_inter(bs);
868                                  mb->mode = mcbpc & 7;                                  mb->mode = mcbpc & 7;
869                                  cbpc = (mcbpc >> 4);                                  cbpc = (mcbpc >> 4);
870                                  acpred_flag = 0;  
871                                    DPRINTF(XVID_DEBUG_MB, "mode %i\n", mb->mode);
872                                    DPRINTF(XVID_DEBUG_MB, "cbpc %i\n", cbpc);
873    
874                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);
875    
876                                  if (intra)                                  if (gmc_warp && (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q))
877                                  {                                          mcsel = BitstreamGetBit(bs);
878                                    else if (intra)
879                                          acpred_flag = BitstreamGetBit(bs);                                          acpred_flag = BitstreamGetBit(bs);
                                 }  
   
                                 if (mb->mode == MODE_STUFFING)  
                                 {  
                                         DEBUG("-- STUFFING ?");  
                                         continue;  
                                 }  
880    
881                                  cbpy = get_cbpy(bs, intra);                                  cbpy = get_cbpy(bs, intra);
882                                    DPRINTF(XVID_DEBUG_MB, "cbpy %i mcsel %i \n", cbpy,mcsel);
883    
884                                  cbp = (cbpy << 2) | cbpc;                                  cbp = (cbpy << 2) | cbpc;
885    
886                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q)                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q) {
887                                  {                                          int dquant = dquant_table[BitstreamGetBits(bs, 2)];
888                                          quant += dquant_table[BitstreamGetBits(bs,2)];                                          DPRINTF(XVID_DEBUG_MB, "dquant %i\n", dquant);
889                                          if (quant > 31)                                          quant += dquant;
890                                          {                                          if (quant > 31) {
891                                                  quant = 31;                                                  quant = 31;
892                                          }                                          } else if (quant < 1) {
                                         else if (mb->quant < 1)  
                                         {  
893                                                  quant = 1;                                                  quant = 1;
894                                          }                                          }
895                                            DPRINTF(XVID_DEBUG_MB, "quant %i\n", quant);
896                                  }                                  }
897                                  mb->quant = quant;                                  mb->quant = quant;
898    
899                                  if (dec->interlacing)                                  if (dec->interlacing) {
900                                  {                                          if (cbp || intra) {
901                                          mb->field_dct = BitstreamGetBit(bs);                                          mb->field_dct = BitstreamGetBit(bs);
902                                          DEBUG1("decp: field_dct: ", mb->field_dct);                                                  DPRINTF(XVID_DEBUG_MB,"decp: field_dct: %i\n", mb->field_dct);
903                                            }
904    
905                                          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                          if ((mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) && !mcsel) {
                                         {  
906                                                  mb->field_pred = BitstreamGetBit(bs);                                                  mb->field_pred = BitstreamGetBit(bs);
907                                                  DEBUG1("decp: field_pred: ", mb->field_pred);                                                  DPRINTF(XVID_DEBUG_MB, "decp: field_pred: %i\n", mb->field_pred);
908    
909                                                  if (mb->field_pred)                                                  if (mb->field_pred) {
                                                 {  
910                                                          mb->field_for_top = BitstreamGetBit(bs);                                                          mb->field_for_top = BitstreamGetBit(bs);
911                                                          DEBUG1("decp: field_for_top: ", mb->field_for_top);                                                          DPRINTF(XVID_DEBUG_MB,"decp: field_for_top: %i\n", mb->field_for_top);
912                                                          mb->field_for_bot = BitstreamGetBit(bs);                                                          mb->field_for_bot = BitstreamGetBit(bs);
913                                                          DEBUG1("decp: field_for_bot: ", mb->field_for_bot);                                                          DPRINTF(XVID_DEBUG_MB,"decp: field_for_bot: %i\n", mb->field_for_bot);
914                                                  }                                                  }
915                                          }                                          }
916                                  }                                  }
917    
918                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                  if (mcsel) {
919                                  {                                          decoder_mbgmc(dec, mb, x, y, fcode, cbp, bs, rounding);
920                                          if (dec->interlacing && mb->field_pred)                                          continue;
921    
922                                    } else if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
923    
924                                            if (dec->interlacing && mb->field_pred) {
925                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode, bound);
926                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1], fcode, bound);
927                                            } else {
928                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode, bound);
929                                                    mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0];
930                                            }
931                                    } else if (mb->mode == MODE_INTER4V ) {
932                                            get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode, bound);
933                                            get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode, bound);
934                                            get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode, bound);
935                                            get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode, bound);
936                                    } else {                /* MODE_INTRA, MODE_INTRA_Q */
937                                            mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
938                                            mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =     0;
939                                            decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
940                                                                            intra_dc_threshold, bound, reduced_resolution);
941                                            continue;
942                                    }
943    
944                                    decoder_mbinter(dec, mb, x, y, cbp, bs,
945                                                                    rounding, reduced_resolution, 0);
946    
947                            } else if (gmc_warp) {  /* a not coded S(GMC)-VOP macroblock */
948                                    mb->mode = MODE_NOT_CODED_GMC;
949                                    mb->quant = quant;
950                                    decoder_mbgmc(dec, mb, x, y, fcode, 0x00, bs, rounding);
951    
952                                    if(dec->out_frm && cp_mb > 0) {
953                                            output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
954                                            cp_mb = 0;
955                                    }
956                                    st_mb = x+1;
957                            } else {        /* not coded P_VOP macroblock */
958                                    mb->mode = MODE_NOT_CODED;
959                                    mb->quant = quant;
960    
961                                    mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
962                                    mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;
963    
964                                    decoder_mbinter(dec, mb, x, y, 0, bs,
965                                                                    rounding, reduced_resolution, 0);
966    
967                                    if(dec->out_frm && cp_mb > 0) {
968                                            output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
969                                            cp_mb = 0;
970                                    }
971                                    st_mb = x+1;
972                            }
973                    }
974    
975                    if(dec->out_frm && cp_mb > 0)
976                            output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
977            }
978    }
979    
980    
981    /* decode B-frame motion vector */
982    static void
983    get_b_motion_vector(Bitstream * bs,
984                                            VECTOR * mv,
985                                            int fcode,
986                                            const VECTOR pmv,
987                                            const DECODER * const dec,
988                                            const int x, const int y)
989                                          {                                          {
990                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);          const int scale_fac = 1 << (fcode - 1);
991                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1], fcode);          const int high = (32 * scale_fac) - 1;
992            const int low = ((-32) * scale_fac);
993            const int range = (64 * scale_fac);
994    
995            int mv_x = get_mv(bs, fcode);
996            int mv_y = get_mv(bs, fcode);
997    
998            mv_x += pmv.x;
999            mv_y += pmv.y;
1000    
1001            if (mv_x < low)
1002                    mv_x += range;
1003            else if (mv_x > high)
1004                    mv_x -= range;
1005    
1006            if (mv_y < low)
1007                    mv_y += range;
1008            else if (mv_y > high)
1009                    mv_y -= range;
1010    
1011            mv->x = mv_x;
1012            mv->y = mv_y;
1013                                          }                                          }
1014                                          else  
1015    /* decode an B-frame direct & interpolate macroblock */
1016    static void
1017    decoder_bf_interpolate_mbinter(DECODER * dec,
1018                                                                    IMAGE forward,
1019                                                                    IMAGE backward,
1020                                                                    const MACROBLOCK * pMB,
1021                                                                    const uint32_t x_pos,
1022                                                                    const uint32_t y_pos,
1023                                                                    Bitstream * bs,
1024                                                                    const int direct)
1025                                          {                                          {
1026                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);          uint32_t stride = dec->edged_width;
1027                                                  mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;          uint32_t stride2 = stride / 2;
1028                                                  mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;          int uv_dx, uv_dy;
1029            int b_uv_dx, b_uv_dy;
1030            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
1031            const uint32_t cbp = pMB->cbp;
1032    
1033            pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
1034            pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
1035            pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
1036    
1037            if (!direct) {
1038                    uv_dx = pMB->mvs[0].x;
1039                    uv_dy = pMB->mvs[0].y;
1040                    b_uv_dx = pMB->b_mvs[0].x;
1041                    b_uv_dy = pMB->b_mvs[0].y;
1042    
1043                    if (dec->quarterpel) {
1044                            uv_dx /= 2;
1045                            uv_dy /= 2;
1046                            b_uv_dx /= 2;
1047                            b_uv_dy /= 2;
1048                    }
1049    
1050                    uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
1051                    uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
1052                    b_uv_dx = (b_uv_dx >> 1) + roundtab_79[b_uv_dx & 0x3];
1053                    b_uv_dy = (b_uv_dy >> 1) + roundtab_79[b_uv_dy & 0x3];
1054    
1055            } else {
1056                    uv_dx = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
1057                    uv_dy = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
1058                    b_uv_dx = pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x + pMB->b_mvs[3].x;
1059                    b_uv_dy = pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y + pMB->b_mvs[3].y;
1060    
1061                    if (dec->quarterpel) {
1062                            uv_dx /= 2;
1063                            uv_dy /= 2;
1064                            b_uv_dx /= 2;
1065                            b_uv_dy /= 2;
1066                    }
1067    
1068                    uv_dx = (uv_dx >> 3) + roundtab_76[uv_dx & 0xf];
1069                    uv_dy = (uv_dy >> 3) + roundtab_76[uv_dy & 0xf];
1070                    b_uv_dx = (b_uv_dx >> 3) + roundtab_76[b_uv_dx & 0xf];
1071                    b_uv_dy = (b_uv_dy >> 3) + roundtab_76[b_uv_dy & 0xf];
1072                                          }                                          }
1073    
1074            start_timer();
1075            if(dec->quarterpel) {
1076                    if(!direct) {
1077                            interpolate16x16_quarterpel(dec->cur.y, forward.y, dec->qtmp.y, dec->qtmp.y + 64,
1078                                                                                    dec->qtmp.y + 128, 16*x_pos, 16*y_pos,
1079                                                                                    pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1080                    } else {
1081                            interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->qtmp.y, dec->qtmp.y + 64,
1082                                                                                    dec->qtmp.y + 128, 16*x_pos, 16*y_pos,
1083                                                                                    pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1084                            interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->qtmp.y, dec->qtmp.y + 64,
1085                                                                                    dec->qtmp.y + 128, 16*x_pos + 8, 16*y_pos,
1086                                                                                    pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1087                            interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->qtmp.y, dec->qtmp.y + 64,
1088                                                                                    dec->qtmp.y + 128, 16*x_pos, 16*y_pos + 8,
1089                                                                                    pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1090                            interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->qtmp.y, dec->qtmp.y + 64,
1091                                                                                    dec->qtmp.y + 128, 16*x_pos + 8, 16*y_pos + 8,
1092                                                                                    pMB->mvs[3].x, pMB->mvs[3].y, stride, 0);
1093                    }
1094            } else {
1095                    interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos,
1096                                                            pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1097                    interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8, 16 * y_pos,
1098                                                            pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1099                    interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos + 8,
1100                                                            pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1101                    interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8, 16 * y_pos + 8,
1102                                                            pMB->mvs[3].x, pMB->mvs[3].y, stride, 0);
1103            }
1104    
1105            interpolate8x8_switch(dec->cur.u, forward.u, 8 * x_pos, 8 * y_pos, uv_dx,
1106                                                    uv_dy, stride2, 0);
1107            interpolate8x8_switch(dec->cur.v, forward.v, 8 * x_pos, 8 * y_pos, uv_dx,
1108                                                    uv_dy, stride2, 0);
1109    
1110    
1111            if(dec->quarterpel) {
1112                    if(!direct) {
1113                            interpolate16x16_add_quarterpel(dec->cur.y, backward.y, dec->qtmp.y, dec->qtmp.y + 64,
1114                                            dec->qtmp.y + 128, 16*x_pos, 16*y_pos,
1115                                            pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1116                    } else {
1117                            interpolate8x8_add_quarterpel(dec->cur.y, backward.y, dec->qtmp.y, dec->qtmp.y + 64,
1118                                            dec->qtmp.y + 128, 16*x_pos, 16*y_pos,
1119                                            pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1120                            interpolate8x8_add_quarterpel(dec->cur.y, backward.y, dec->qtmp.y, dec->qtmp.y + 64,
1121                                            dec->qtmp.y + 128, 16*x_pos + 8, 16*y_pos,
1122                                            pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride, 0);
1123                            interpolate8x8_add_quarterpel(dec->cur.y, backward.y, dec->qtmp.y, dec->qtmp.y + 64,
1124                                            dec->qtmp.y + 128, 16*x_pos, 16*y_pos + 8,
1125                                            pMB->b_mvs[2].x, pMB->b_mvs[2].y, stride, 0);
1126                            interpolate8x8_add_quarterpel(dec->cur.y, backward.y, dec->qtmp.y, dec->qtmp.y + 64,
1127                                            dec->qtmp.y + 128, 16*x_pos + 8, 16*y_pos + 8,
1128                                            pMB->b_mvs[3].x, pMB->b_mvs[3].y, stride, 0);
1129                    }
1130            } else {
1131                    interpolate8x8_add_switch(dec->cur.y, backward.y, 16 * x_pos, 16 * y_pos,
1132                                    pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1133                    interpolate8x8_add_switch(dec->cur.y, backward.y, 16 * x_pos + 8,
1134                                    16 * y_pos, pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride, 0);
1135                    interpolate8x8_add_switch(dec->cur.y, backward.y, 16 * x_pos,
1136                                    16 * y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y, stride, 0);
1137                    interpolate8x8_add_switch(dec->cur.y, backward.y, 16 * x_pos + 8,
1138                                    16 * y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y, stride, 0);
1139            }
1140    
1141            interpolate8x8_add_switch(dec->cur.u, backward.u, 8 * x_pos, 8 * y_pos,
1142                            b_uv_dx, b_uv_dy, stride2, 0);
1143            interpolate8x8_add_switch(dec->cur.v, backward.v, 8 * x_pos, 8 * y_pos,
1144                            b_uv_dx, b_uv_dy, stride2, 0);
1145    
1146            stop_comp_timer();
1147    
1148            if (cbp)
1149                    decoder_mb_decode(dec, cbp, bs, pY_Cur, pU_Cur, pV_Cur, 0, pMB);
1150                                  }                                  }
1151                                  else if (mb->mode == MODE_INTER4V /* || mb->mode == MODE_INTER4V_Q */)  
1152    /* for decode B-frame dbquant */
1153    static __inline int32_t
1154    get_dbquant(Bitstream * bs)
1155                                  {                                  {
1156                                          get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);          if (!BitstreamGetBit(bs))               /*  '0' */
1157                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode);                  return (0);
1158                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode);          else if (!BitstreamGetBit(bs))  /* '10' */
1159                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode);                  return (-2);
1160            else                                                    /* '11' */
1161                    return (2);
1162                                  }                                  }
1163                                  else  // MODE_INTRA, MODE_INTRA_Q  
1164    /*
1165     * decode B-frame mb_type
1166     * bit          ret_value
1167     * 1            0
1168     * 01           1
1169     * 001          2
1170     * 0001         3
1171     */
1172    static int32_t __inline
1173    get_mbtype(Bitstream * bs)
1174                                  {                                  {
1175                                          mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;          int32_t mb_type;
1176                                          mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;  
1177                                          decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant, intra_dc_threshold);          for (mb_type = 0; mb_type <= 3; mb_type++)
1178                    if (BitstreamGetBit(bs))
1179                            return (mb_type);
1180    
1181            return -1;
1182    }
1183    
1184    static void
1185    decoder_bframe(DECODER * dec,
1186                                    Bitstream * bs,
1187                                    int quant,
1188                                    int fcode_forward,
1189                                    int fcode_backward)
1190    {
1191            uint32_t x, y;
1192            VECTOR mv;
1193            const VECTOR zeromv = {0,0};
1194            int i;
1195    
1196            if (!dec->is_edged[0]) {
1197                    start_timer();
1198                    image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
1199                                                    dec->width, dec->height, dec->bs_version);
1200                    dec->is_edged[0] = 1;
1201                    stop_edges_timer();
1202            }
1203    
1204            if (!dec->is_edged[1]) {
1205                    start_timer();
1206                    image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height,
1207                                                    dec->width, dec->height, dec->bs_version);
1208                    dec->is_edged[1] = 1;
1209                    stop_edges_timer();
1210            }
1211    
1212            for (y = 0; y < dec->mb_height; y++) {
1213                    /* Initialize Pred Motion Vector */
1214                    dec->p_fmv = dec->p_bmv = zeromv;
1215                    for (x = 0; x < dec->mb_width; x++) {
1216                            MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x];
1217                            MACROBLOCK *last_mb = &dec->last_mbs[y * dec->mb_width + x];
1218                            const int fcode_max = (fcode_forward>fcode_backward) ? fcode_forward : fcode_backward;
1219                            uint32_t intra_dc_threshold; /* fake variable */
1220    
1221                            if (check_resync_marker(bs, fcode_max  - 1)) {
1222                                    int bound = read_video_packet_header(bs, dec, fcode_max - 1, &quant,
1223                                                                                                             &fcode_forward, &fcode_backward, &intra_dc_threshold);
1224                                    x = bound % dec->mb_width;
1225                                    y = bound / dec->mb_width;
1226                                    /* reset predicted macroblocks */
1227                                    dec->p_fmv = dec->p_bmv = zeromv;
1228                            }
1229    
1230                            mv =
1231                            mb->b_mvs[0] = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] =
1232                            mb->mvs[0] = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = zeromv;
1233                            mb->quant = quant;
1234    
1235                            /*
1236                             * skip if the co-located P_VOP macroblock is not coded
1237                             * if not codec in co-located S_VOP macroblock is _not_
1238                             * automatically skipped
1239                             */
1240    
1241                            if (last_mb->mode == MODE_NOT_CODED) {
1242                                    mb->cbp = 0;
1243                                    mb->mode = MODE_FORWARD;
1244                                    decoder_mbinter(dec, mb, x, y, mb->cbp, bs, 0, 0, 1);
1245                                          continue;                                          continue;
1246                                  }                                  }
1247    
1248                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant, rounding);                          if (!BitstreamGetBit(bs)) {     /* modb=='0' */
1249                                    const uint8_t modb2 = BitstreamGetBit(bs);
1250    
1251                                    mb->mode = get_mbtype(bs);
1252    
1253                                    if (!modb2)             /* modb=='00' */
1254                                            mb->cbp = BitstreamGetBits(bs, 6);
1255                                    else
1256                                            mb->cbp = 0;
1257    
1258                                    if (mb->mode && mb->cbp) {
1259                                            quant += get_dbquant(bs);
1260                                            if (quant > 31)
1261                                                    quant = 31;
1262                                            else if (quant < 1)
1263                                                    quant = 1;
1264                          }                          }
1265                          else    // not coded                                  mb->quant = quant;
                         {  
1266    
1267                                  mb->mode = MODE_NOT_CODED;                                  if (dec->interlacing) {
1268                                  mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;                                          if (mb->cbp) {
1269                                  mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;                                                  mb->field_dct = BitstreamGetBit(bs);
1270                                                    DPRINTF(XVID_DEBUG_MB,"decp: field_dct: %i\n", mb->field_dct);
1271                                            }
1272    
1273                                  // copy macroblock directly from ref to cur                                          if (mb->mode) {
1274                                                    mb->field_pred = BitstreamGetBit(bs);
1275                                                    DPRINTF(XVID_DEBUG_MB, "decp: field_pred: %i\n", mb->field_pred);
1276    
1277                                  start_timer();                                                  if (mb->field_pred) {
1278                                                            mb->field_for_top = BitstreamGetBit(bs);
1279                                                            DPRINTF(XVID_DEBUG_MB,"decp: field_for_top: %i\n", mb->field_for_top);
1280                                                            mb->field_for_bot = BitstreamGetBit(bs);
1281                                                            DPRINTF(XVID_DEBUG_MB,"decp: field_for_bot: %i\n", mb->field_for_bot);
1282                                                    }
1283                                            }
1284                                    }
1285    
1286                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),                          } else {
1287                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x),                                  mb->mode = MODE_DIRECT_NONE_MV;
1288                                                   dec->edged_width);                                  mb->cbp = 0;
1289                            }
1290    
1291                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x+8),                          switch (mb->mode) {
1292                                                   dec->refn[0].y + (16*y)*dec->edged_width + (16*x+8),                          case MODE_DIRECT:
1293                                                   dec->edged_width);                                  get_b_motion_vector(bs, &mv, 1, zeromv, dec, x, y);
1294    
1295                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x),                          case MODE_DIRECT_NONE_MV:
1296                                                   dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x),                                  for (i = 0; i < 4; i++) {
1297                                                   dec->edged_width);                                          mb->mvs[i].x = last_mb->mvs[i].x*dec->time_bp/dec->time_pp + mv.x;
1298                                            mb->mvs[i].y = last_mb->mvs[i].y*dec->time_bp/dec->time_pp + mv.y;
1299    
1300                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x+8),                                          mb->b_mvs[i].x = (mv.x)
1301                                                   dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x+8),                                                  ?  mb->mvs[i].x - last_mb->mvs[i].x
1302                                                   dec->edged_width);                                                  : last_mb->mvs[i].x*(dec->time_bp - dec->time_pp)/dec->time_pp;
1303                                            mb->b_mvs[i].y = (mv.y)
1304                                                    ? mb->mvs[i].y - last_mb->mvs[i].y
1305                                                    : last_mb->mvs[i].y*(dec->time_bp - dec->time_pp)/dec->time_pp;
1306                                    }
1307    
1308                                  transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),                                  decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1309                                                   dec->refn[0].u + (8*y)*dec->edged_width/2 + (8*x),                                                                                                  mb, x, y, bs, 1);
1310                                                   dec->edged_width/2);                                  break;
1311    
1312                                  transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),                          case MODE_INTERPOLATE:
1313                                                   dec->refn[0].v + (8*y)*dec->edged_width/2 + (8*x),                                  get_b_motion_vector(bs, &mb->mvs[0], fcode_forward, dec->p_fmv, dec, x, y);
1314                                                   dec->edged_width/2);                                  dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] =     mb->mvs[0];
1315    
1316                                  stop_transfer_timer();                                  get_b_motion_vector(bs, &mb->b_mvs[0], fcode_backward, dec->p_bmv, dec, x, y);
1317                                    dec->p_bmv = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] = mb->b_mvs[0];
1318    
1319                                    decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1320                                                                                            mb, x, y, bs, 0);
1321                                    break;
1322    
1323                            case MODE_BACKWARD:
1324                                    get_b_motion_vector(bs, &mb->mvs[0], fcode_backward, dec->p_bmv, dec, x, y);
1325                                    dec->p_bmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] =     mb->mvs[0];
1326    
1327                                    decoder_mbinter(dec, mb, x, y, mb->cbp, bs, 0, 0, 0);
1328                                    break;
1329    
1330                            case MODE_FORWARD:
1331                                    get_b_motion_vector(bs, &mb->mvs[0], fcode_forward, dec->p_fmv, dec, x, y);
1332                                    dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] =     mb->mvs[0];
1333    
1334                                    decoder_mbinter(dec, mb, x, y, mb->cbp, bs, 0, 0, 1);
1335                                    break;
1336    
1337                            default:
1338                                    DPRINTF(XVID_DEBUG_ERROR,"Not supported B-frame mb_type = %i\n", mb->mode);
1339                            }
1340                    } /* End of for */
1341                          }                          }
1342                  }                  }
1343    
1344    /* perform post processing if necessary, and output the image */
1345    void decoder_output(DECODER * dec, IMAGE * img, MACROBLOCK * mbs,
1346                                            xvid_dec_frame_t * frame, xvid_dec_stats_t * stats,
1347                                            int coding_type, int quant)
1348    {
1349            const int brightness = XVID_VERSION_MINOR(frame->version) >= 1 ? frame->brightness : 0;
1350    
1351            if (dec->cartoon_mode)
1352                    frame->general &= ~XVID_FILMEFFECT;
1353    
1354            if ((frame->general & (XVID_DEBLOCKY|XVID_DEBLOCKUV|XVID_FILMEFFECT) || brightness!=0)
1355                    && mbs != NULL) /* post process */
1356            {
1357                    /* note: image is stored to tmp */
1358                    image_copy(&dec->tmp, img, dec->edged_width, dec->height);
1359                    image_postproc(&dec->postproc, &dec->tmp, dec->edged_width,
1360                                               mbs, dec->mb_width, dec->mb_height, dec->mb_width,
1361                                               frame->general, brightness, dec->frames, (coding_type == B_VOP));
1362                    img = &dec->tmp;
1363            }
1364    
1365            image_output(img, dec->width, dec->height,
1366                                     dec->edged_width, (uint8_t**)frame->output.plane, frame->output.stride,
1367                                     frame->output.csp, dec->interlacing);
1368    
1369            if (stats) {
1370                    stats->type = coding2type(coding_type);
1371                    stats->data.vop.time_base = (int)dec->time_base;
1372                    stats->data.vop.time_increment = 0;     /* XXX: todo */
1373                    stats->data.vop.qscale_stride = dec->mb_width;
1374                    stats->data.vop.qscale = dec->qscale;
1375                    if (stats->data.vop.qscale != NULL && mbs != NULL) {
1376                            int i;
1377                            for (i = 0; i < dec->mb_width*dec->mb_height; i++)
1378                                    stats->data.vop.qscale[i] = mbs[i].quant;
1379                    } else
1380                            stats->data.vop.qscale = NULL;
1381          }          }
1382  }  }
1383    
1384  int decoder_decode(DECODER * dec, XVID_DEC_FRAME * frame)  
1385    int
1386    decoder_decode(DECODER * dec,
1387                                    xvid_dec_frame_t * frame, xvid_dec_stats_t * stats)
1388  {  {
1389    
1390          Bitstream bs;          Bitstream bs;
1391          uint32_t rounding;          uint32_t rounding;
1392          uint32_t quant;          uint32_t reduced_resolution;
1393          uint32_t fcode;          uint32_t quant = 2;
1394            uint32_t fcode_forward;
1395            uint32_t fcode_backward;
1396          uint32_t intra_dc_threshold;          uint32_t intra_dc_threshold;
1397          uint32_t vop_type;          WARPPOINTS gmc_warp;
1398            int coding_type;
1399            int success, output, seen_something;
1400    
1401            if (XVID_VERSION_MAJOR(frame->version) != 1 || (stats && XVID_VERSION_MAJOR(stats->version) != 1))      /* v1.x.x */
1402                    return XVID_ERR_VERSION;
1403    
1404          start_global_timer();          start_global_timer();
1405    
1406            dec->low_delay_default = (frame->general & XVID_LOWDELAY);
1407            if ((frame->general & XVID_DISCONTINUITY))
1408                    dec->frames = 0;
1409            dec->out_frm = (frame->output.csp == XVID_CSP_SLICE) ? &frame->output : NULL;
1410    
1411            if (frame->length < 0) {        /* decoder flush */
1412                    int ret;
1413                    /* if not decoding "low_delay/packed", and this isn't low_delay and
1414                            we have a reference frame, then outout the reference frame */
1415                    if (!(dec->low_delay_default && dec->packed_mode) && !dec->low_delay && dec->frames>0) {
1416                            decoder_output(dec, &dec->refn[0], dec->last_mbs, frame, stats, dec->last_coding_type, quant);
1417                            dec->frames = 0;
1418                            ret = 0;
1419                    } else {
1420                            if (stats) stats->type = XVID_TYPE_NOTHING;
1421                            ret = XVID_ERR_END;
1422                    }
1423    
1424                    emms();
1425                    stop_global_timer();
1426                    return ret;
1427            }
1428    
1429          BitstreamInit(&bs, frame->bitstream, frame->length);          BitstreamInit(&bs, frame->bitstream, frame->length);
1430    
1431          // add by chenm001 <chenm001@163.com>          /* XXX: 0x7f is only valid whilst decoding vfw xvid/divx5 avi's */
1432          // for support B-frame to reference last 2 frame          if(dec->low_delay_default && frame->length == 1 && BitstreamShowBits(&bs, 8) == 0x7f)
1433          vop_type=BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode, &intra_dc_threshold);          {
1434                    image_output(&dec->refn[0], dec->width, dec->height, dec->edged_width,
1435                                             (uint8_t**)frame->output.plane, frame->output.stride, frame->output.csp, dec->interlacing);
1436                    if (stats) stats->type = XVID_TYPE_NOTHING;
1437                    emms();
1438                    return 1;       /* one byte consumed */
1439            }
1440    
1441          if (vop_type==I_VOP || vop_type==P_VOP){          success = 0;
1442                  image_swap(&dec->refn[0], &dec->refn[1]);          output = 0;
1443                  image_swap(&dec->cur, &dec->refn[0]);          seen_something = 0;
1444    
1445    repeat:
1446    
1447            coding_type = BitstreamReadHeaders(&bs, dec, &rounding, &reduced_resolution,
1448                            &quant, &fcode_forward, &fcode_backward, &intra_dc_threshold, &gmc_warp);
1449    
1450            DPRINTF(XVID_DEBUG_HEADER, "coding_type=%i,  packed=%i,  time=%lli,  time_pp=%i,  time_bp=%i\n",
1451                                                            coding_type,    dec->packed_mode, dec->time, dec->time_pp, dec->time_bp);
1452    
1453            if (coding_type == -1) { /* nothing */
1454                    if (success) goto done;
1455                    if (stats) stats->type = XVID_TYPE_NOTHING;
1456                    emms();
1457                    return BitstreamPos(&bs)/8;
1458          }          }
1459    
1460          switch (vop_type)          if (coding_type == -2 || coding_type == -3) {   /* vol and/or resize */
1461          {  
1462          case P_VOP :                  if (coding_type == -3)
1463                  decoder_pframe(dec, &bs, rounding, quant, fcode, intra_dc_threshold);                          decoder_resize(dec);
1464                  break;  
1465                    if (stats) {
1466                            stats->type = XVID_TYPE_VOL;
1467                            stats->data.vol.general = 0;
1468                            /*XXX: if (dec->interlacing)
1469                                    stats->data.vol.general |= ++INTERLACING; */
1470                            stats->data.vol.width = dec->width;
1471                            stats->data.vol.height = dec->height;
1472                            stats->data.vol.par = dec->aspect_ratio;
1473                            stats->data.vol.par_width = dec->par_width;
1474                            stats->data.vol.par_height = dec->par_height;
1475                            emms();
1476                            return BitstreamPos(&bs)/8;     /* number of bytes consumed */
1477                    }
1478                    goto repeat;
1479            }
1480    
1481            if(dec->frames == 0 && coding_type != I_VOP) {
1482                    /* 1st frame is not an i-vop */
1483                    goto repeat;
1484            }
1485    
1486            dec->p_bmv.x = dec->p_bmv.y = dec->p_fmv.y = dec->p_fmv.y = 0;  /* init pred vector to 0 */
1487    
1488            /* packed_mode: special-N_VOP treament */
1489            if (dec->packed_mode && coding_type == N_VOP) {
1490                    if (dec->low_delay_default && dec->frames > 0) {
1491                            decoder_output(dec, &dec->refn[0], dec->last_mbs, frame, stats, dec->last_coding_type, quant);
1492                            output = 1;
1493                    }
1494                    /* ignore otherwise */
1495            } else if (coding_type != B_VOP) {
1496                    switch(coding_type) {
1497          case I_VOP :          case I_VOP :
1498                  //DEBUG1("",intra_dc_threshold);                          decoder_iframe(dec, &bs, reduced_resolution, quant, intra_dc_threshold);
                 decoder_iframe(dec, &bs, quant, intra_dc_threshold);  
1499                  break;                  break;
1500                    case P_VOP :
1501          case B_VOP :    // ignore                          decoder_pframe(dec, &bs, rounding, reduced_resolution, quant,
1502                                                    fcode_forward, intra_dc_threshold, NULL);
1503                  break;                  break;
1504                    case S_VOP :
1505          case N_VOP :    // vop not coded                          decoder_pframe(dec, &bs, rounding, reduced_resolution, quant,
1506                                                    fcode_forward, intra_dc_threshold, &gmc_warp);
1507                  break;                  break;
1508                    case N_VOP :
1509                            /* XXX: not_coded vops are not used for forward prediction */
1510                            /* we should not swap(last_mbs,mbs) */
1511                            image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1512                            SWAP(MACROBLOCK *, dec->mbs, dec->last_mbs); /* it will be swapped back */
1513                            break;
1514                    }
1515    
1516          default :                  if (reduced_resolution) {
1517                  return XVID_ERR_FAIL;                          image_deblock_rrv(&dec->cur, dec->edged_width, dec->mbs,
1518                                    (dec->width + 31) / 32, (dec->height + 31) / 32, dec->mb_width,
1519                                    16, 0);
1520          }          }
1521    
1522          frame->length = BitstreamPos(&bs) / 8;                  /* note: for packed_mode, output is performed when the special-N_VOP is decoded */
1523                    if (!(dec->low_delay_default && dec->packed_mode)) {
1524                            if (dec->low_delay) {
1525                                    decoder_output(dec, &dec->cur, dec->mbs, frame, stats, coding_type, quant);
1526                                    output = 1;
1527                            } else if (dec->frames > 0)     { /* is the reference frame valid? */
1528                                    /* output the reference frame */
1529                                    decoder_output(dec, &dec->refn[0], dec->last_mbs, frame, stats, dec->last_coding_type, quant);
1530                                    output = 1;
1531                            }
1532                    }
1533    
1534          start_timer();                  image_swap(&dec->refn[0], &dec->refn[1]);
1535          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,                  dec->is_edged[1] = dec->is_edged[0];
1536                       frame->image, frame->stride, frame->colorspace);                  image_swap(&dec->cur, &dec->refn[0]);
1537          stop_conv_timer();                  dec->is_edged[0] = 0;
1538                    SWAP(MACROBLOCK *, dec->mbs, dec->last_mbs);
1539                    dec->last_reduced_resolution = reduced_resolution;
1540                    dec->last_coding_type = coding_type;
1541    
1542                    dec->frames++;
1543                    seen_something = 1;
1544    
1545            } else {        /* B_VOP */
1546    
1547                    if (dec->low_delay) {
1548                            DPRINTF(XVID_DEBUG_ERROR, "warning: bvop found in low_delay==1 stream\n");
1549                            dec->low_delay = 0;
1550                    }
1551    
1552                    if (dec->frames < 2) {
1553                            /* attemping to decode a bvop without atleast 2 reference frames */
1554                            image_printf(&dec->cur, dec->edged_width, dec->height, 16, 16,
1555                                                    "broken b-frame, mising ref frames");
1556                            if (stats) stats->type = XVID_TYPE_NOTHING;
1557                    } else if (dec->time_pp <= dec->time_bp) {
1558                            /* this occurs when dx50_bvop_compatibility==0 sequences are
1559                            decoded in vfw. */
1560                            image_printf(&dec->cur, dec->edged_width, dec->height, 16, 16,
1561                                                    "broken b-frame, tpp=%i tbp=%i", dec->time_pp, dec->time_bp);
1562                            if (stats) stats->type = XVID_TYPE_NOTHING;
1563                    } else {
1564                            decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);
1565                            decoder_output(dec, &dec->cur, dec->mbs, frame, stats, coding_type, quant);
1566                    }
1567    
1568                    output = 1;
1569                    dec->frames++;
1570            }
1571    
1572    #if 0 /* Avoids to read to much data because of 32bit reads in our BS functions */
1573             BitstreamByteAlign(&bs);
1574    #endif
1575    
1576            /* low_delay_default mode: repeat in packed_mode */
1577            if (dec->low_delay_default && dec->packed_mode && output == 0 && success == 0) {
1578                    success = 1;
1579                    goto repeat;
1580            }
1581    
1582    done :
1583    
1584            /* low_delay_default mode: if we've gotten here without outputting anything,
1585               then output the recently decoded frame, or print an error message  */
1586            if (dec->low_delay_default && output == 0) {
1587                    if (dec->packed_mode && seen_something) {
1588                            /* output the recently decoded frame */
1589                            decoder_output(dec, &dec->refn[0], dec->last_mbs, frame, stats, dec->last_coding_type, quant);
1590                    } else {
1591                            image_clear(&dec->cur, dec->width, dec->height, dec->edged_width, 0, 128, 128);
1592                            image_printf(&dec->cur, dec->edged_width, dec->height, 16, 16,
1593                                    "warning: nothing to output");
1594                            image_printf(&dec->cur, dec->edged_width, dec->height, 16, 64,
1595                                    "bframe decoder lag");
1596    
1597          emms();                          decoder_output(dec, &dec->cur, NULL, frame, stats, P_VOP, quant);
1598                            if (stats) stats->type = XVID_TYPE_NOTHING;
1599                    }
1600            }
1601    
1602            emms();
1603          stop_global_timer();          stop_global_timer();
1604    
1605          return XVID_ERR_OK;          return (BitstreamPos(&bs) + 7) / 8;     /* number of bytes consumed */
   
1606  }  }

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

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