[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.1, Fri Mar 8 02:44:29 2002 UTC revision 1.21, Sat Jun 22 07:23:09 2002 UTC
# Line 1  Line 1 
1  /**************************************************************************  /**************************************************************************
2   *   *
3   *      XVID MPEG-4 VIDEO CODEC   *      XVID MPEG-4 VIDEO CODEC
4   *      decoder main   *  -  Decoder main module  -
5   *   *
6   *      This program is an implementation of a part of one or more MPEG-4   *      This program is an implementation of a part of one or more MPEG-4
7   *      Video tools as specified in ISO/IEC 14496-2 standard.  Those intending   *      Video tools as specified in ISO/IEC 14496-2 standard.  Those intending
# Line 24  Line 24 
24   *   *
25   *      You should have received a copy of the GNU General Public License   *      You should have received a copy of the GNU General Public License
26   *      along with this program; if not, write to the Free Software   *      along with this program; if not, write to the Free Software
27   *      Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
28   *   *
29   *************************************************************************/   *************************************************************************/
30    
# Line 32  Line 32 
32   *   *
33   *      History:   *      History:
34   *   *
35     *      22.06.2002      added primative N_VOP support
36     *                              #define BFRAMES_DEC now enables Minchenm's bframe decoder
37     *  08.05.2002  add low_delay support for B_VOP decode
38     *              MinChen <chenm001@163.com>
39     *  05.05.2002  fix some B-frame decode problem
40     *  02.05.2002  add B-frame decode support(have some problem);
41     *              MinChen <chenm001@163.com>
42     *  22.04.2002  add some B-frame decode support;  chenm001 <chenm001@163.com>
43     *  29.03.2002  interlacing fix - compensated block wasn't being used when
44     *              reconstructing blocks, thus artifacts
45     *              interlacing speedup - used transfers to re-interlace
46     *              interlaced decoding should be as fast as progressive now
47     *  26.03.2002  interlacing support - moved transfers outside decode loop
48   *      26.12.2001      decoder_mbinter: dequant/idct moved within if(coded) block   *      26.12.2001      decoder_mbinter: dequant/idct moved within if(coded) block
49   *      22.12.2001      block based interpolation   *  22.12.2001  lock based interpolation
50   *      01.12.2001      inital version; (c)2001 peter ross <pross@cs.rmit.edu.au>   *      01.12.2001      inital version; (c)2001 peter ross <pross@cs.rmit.edu.au>
51   *   *
52     *  $Id$
53     *
54   *************************************************************************/   *************************************************************************/
55    
56  #include <stdlib.h>  #include <stdlib.h>
57  #include <string.h>  // memset  #include <string.h>
58    
59  #include "xvid.h"  #include "xvid.h"
60  #include "portab.h"  #include "portab.h"
# Line 62  Line 77 
77    
78  #include "image/image.h"  #include "image/image.h"
79  #include "image/colorspace.h"  #include "image/colorspace.h"
80    #include "utils/mem_align.h"
81    
82  int decoder_create(XVID_DEC_PARAM * param)  int
83    decoder_create(XVID_DEC_PARAM * param)
84  {  {
85          DECODER * dec;          DECODER * dec;
86    
87          dec = malloc(sizeof(DECODER));          dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);
88          if (dec == NULL)          if (dec == NULL) {
         {  
89                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
90          }          }
91          param->handle = dec;          param->handle = dec;
# Line 82  Line 98 
98    
99          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;          dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;
100          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;          dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;
101            dec->low_delay = 0;
102    
103          if (image_create(&dec->cur, dec->edged_width, dec->edged_height))          if (image_create(&dec->cur, dec->edged_width, dec->edged_height)) {
104          {                  xvid_free(dec);
                 free(dec);  
105                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
106          }          }
107    
108          if (image_create(&dec->refn, dec->edged_width, dec->edged_height))          if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height)) {
109          {                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
110                    xvid_free(dec);
111                    return XVID_ERR_MEMORY;
112            }
113            // add by chenm001 <chenm001@163.com>
114            // for support B-frame to reference last 2 frame
115            if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {
116                    image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
117                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
118                    xvid_free(dec);
119                    return XVID_ERR_MEMORY;
120            }
121            if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height)) {
122                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
123                  free(dec);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
124                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
125                    xvid_free(dec);
126                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
127          }          }
128    
129          dec->mbs = malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);          dec->mbs =
130          if (dec->mbs == NULL)                  xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
131          {                                          CACHE_LINE);
132            if (dec->mbs == NULL) {
133                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);                  image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
134                  free(dec);                  image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
135                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
136                    image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
137                    xvid_free(dec);
138                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
139          }          }
140    
141            memset(dec->mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
142    
143            // add by chenm001 <chenm001@163.com>
144            // for skip MB flag
145            dec->last_mbs =
146                    xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
147                                            CACHE_LINE);
148            if (dec->last_mbs == NULL) {
149                    xvid_free(dec->mbs);
150                    image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
151                    image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
152                    image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
153                    image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
154                    xvid_free(dec);
155                    return XVID_ERR_MEMORY;
156            }
157    
158            memset(dec->last_mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
159    
160          init_timer();          init_timer();
161          create_vlc_tables();  
162            // add by chenm001 <chenm001@163.com>
163            // for support B-frame to save reference frame's time
164            dec->frames = -1;
165            dec->time = dec->time_base = dec->last_time_base = 0;
166    
167          return XVID_ERR_OK;          return XVID_ERR_OK;
168  }  }
169    
170    
171  int decoder_destroy(DECODER * dec)  int
172    decoder_destroy(DECODER * dec)
173  {  {
174          free(dec->mbs);          xvid_free(dec->last_mbs);
175          image_destroy(&dec->refn, dec->edged_width, dec->edged_height);          xvid_free(dec->mbs);
176            image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
177            image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
178            image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
179          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);          image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
180          free(dec);          xvid_free(dec);
   
         destroy_vlc_tables();  
181    
182          write_timer();          write_timer();
183          return XVID_ERR_OK;          return XVID_ERR_OK;
# Line 126  Line 185 
185    
186    
187    
188  static const int32_t dquant_table[4] =  static const int32_t dquant_table[4] = {
 {  
189          -1, -2, 1, 2          -1, -2, 1, 2
190  };  };
191    
192    
193    
194    
195  // decode an intra macroblock  // decode an intra macroblock
196    
197  void decoder_mbintra(DECODER * dec, MACROBLOCK * mb, int x, int y, uint32_t acpred_flag, uint32_t cbp, Bitstream * bs, int quant, int intra_dc_threshold)  void
198  {  decoder_mbintra(DECODER * dec,
199          uint32_t k;                                  MACROBLOCK * pMB,
200                                    const uint32_t x_pos,
201                                    const uint32_t y_pos,
202                                    const uint32_t acpred_flag,
203                                    const uint32_t cbp,
204                                    Bitstream * bs,
205                                    const uint32_t quant,
206                                    const uint32_t intra_dc_threshold)
207    {
208    
209            DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
210            DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
211    
212            uint32_t stride = dec->edged_width;
213            uint32_t stride2 = stride / 2;
214            uint32_t next_block = stride * 8;
215            uint32_t i;
216            uint32_t iQuant = pMB->quant;
217            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
218    
219            pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
220            pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
221            pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
222    
223          for (k = 0; k < 6; k++)          memset(block, 0, 6 * 64 * sizeof(int16_t));     // clear
224          {  
225                  uint32_t dcscalar;          for (i = 0; i < 6; i++) {
226                  int16_t block[64];                  uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
                 int16_t data[64];  
227                  int16_t predictors[8];                  int16_t predictors[8];
228                  int start_coeff;                  int start_coeff;
229    
                 dcscalar = get_dc_scaler(mb->quant, k < 4);  
   
230                  start_timer();                  start_timer();
231                  predict_acdc(dec->mbs, x, y, dec->mb_width, k, block, mb->quant, dcscalar, predictors);                  predict_acdc(dec->mbs, x_pos, y_pos, dec->mb_width, i, &block[i * 64],
232                  if (!acpred_flag)                                           iQuant, iDcScaler, predictors);
233                  {                  if (!acpred_flag) {
234                          mb->acpred_directions[k] = 0;                          pMB->acpred_directions[i] = 0;
235                  }                  }
236                  stop_prediction_timer();                  stop_prediction_timer();
237    
238                  memset(block, 0, 64*sizeof(int16_t));           // clear                  if (quant < intra_dc_threshold) {
   
                 if (quant < intra_dc_threshold)  
                 {  
239                          int dc_size;                          int dc_size;
240                          int dc_dif;                          int dc_dif;
241    
242                          dc_size = k < 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);
243                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;                          dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;
244    
245                          if (dc_size > 8)                          if (dc_size > 8) {
                         {  
246                                  BitstreamSkip(bs, 1);           // marker                                  BitstreamSkip(bs, 1);           // marker
247                          }                          }
248    
249                          block[0] = dc_dif;                          block[i * 64 + 0] = dc_dif;
250                          start_coeff = 1;                          start_coeff = 1;
251                  }                  } else {
                 else  
                 {  
252                          start_coeff = 0;                          start_coeff = 0;
253                  }                  }
254    
255                  start_timer();                  start_timer();
256                  if (cbp & (1 << (5-k)))                 // coded                  if (cbp & (1 << (5 - i)))       // coded
257                  {                  {
258                          get_intra_block(bs, block, mb->acpred_directions[k], start_coeff);                          get_intra_block(bs, &block[i * 64], pMB->acpred_directions[i],
259                                                            start_coeff);
260                  }                  }
261                  stop_coding_timer();                  stop_coding_timer();
262    
263                  start_timer();                  start_timer();
264                  add_acdc(mb, k, block, dcscalar, predictors);                  add_acdc(pMB, i, &block[i * 64], iDcScaler, predictors);
265                  stop_prediction_timer();                  stop_prediction_timer();
266    
267                  start_timer();                  start_timer();
268                  if (dec->quant_type == 0)                  if (dec->quant_type == 0) {
269                  {                          dequant_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler);
270                          dequant_intra(data, block, mb->quant, dcscalar);                  } else {
271                  }                          dequant4_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler);
                 else  
                 {  
                         dequant4_intra(data, block, mb->quant, dcscalar);  
272                  }                  }
273                  stop_iquant_timer();                  stop_iquant_timer();
274    
275                  start_timer();                  start_timer();
276                  idct(data);                  idct(&data[i * 64]);
277                  stop_idct_timer();                  stop_idct_timer();
   
                 start_timer();  
                 if (k < 4)  
                 {  
                         transfer_16to8copy(dec->cur.y + (16*y*dec->edged_width) + 16*x + (4*(k&2)*dec->edged_width) + 8*(k&1), data, dec->edged_width);  
278                  }                  }
279                  else if (k == 4)  
280                  {          if (dec->interlacing && pMB->field_dct) {
281                          transfer_16to8copy(dec->cur.u+ 8*y*(dec->edged_width/2) + 8*x, data, (dec->edged_width/2));                  next_block = stride;
282                  }                  stride *= 2;
                 else    // if (k == 5)  
                 {  
                         transfer_16to8copy(dec->cur.v + 8*y*(dec->edged_width/2) + 8*x, data, (dec->edged_width/2));  
283                  }                  }
284    
285            start_timer();
286            transfer_16to8copy(pY_Cur, &data[0 * 64], stride);
287            transfer_16to8copy(pY_Cur + 8, &data[1 * 64], stride);
288            transfer_16to8copy(pY_Cur + next_block, &data[2 * 64], stride);
289            transfer_16to8copy(pY_Cur + 8 + next_block, &data[3 * 64], stride);
290            transfer_16to8copy(pU_Cur, &data[4 * 64], stride2);
291            transfer_16to8copy(pV_Cur, &data[5 * 64], stride2);
292                  stop_transfer_timer();                  stop_transfer_timer();
293          }          }
 }  
294    
295    
296    
# Line 234  Line 304 
304    
305  // decode an inter macroblock  // decode an inter macroblock
306    
307  void decoder_mbinter(DECODER * dec, MACROBLOCK * mb, int x, int y, uint32_t acpred_flag, uint32_t cbp, Bitstream * bs, int quant, int rounding)  void
308  {  decoder_mbinter(DECODER * dec,
309          const uint32_t stride = dec->edged_width;                                  const MACROBLOCK * pMB,
310          const uint32_t stride2 = dec->edged_width / 2;                                  const uint32_t x_pos,
311                                    const uint32_t y_pos,
312                                    const uint32_t acpred_flag,
313                                    const uint32_t cbp,
314                                    Bitstream * bs,
315                                    const uint32_t quant,
316                                    const uint32_t rounding)
317    {
318    
319            DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
320            DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
321    
322            uint32_t stride = dec->edged_width;
323            uint32_t stride2 = stride / 2;
324            uint32_t next_block = stride * 8;
325            uint32_t i;
326            uint32_t iQuant = pMB->quant;
327            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
328          int uv_dx, uv_dy;          int uv_dx, uv_dy;
         uint32_t k;  
329    
330          if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)          pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
331          {          pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
332                  uv_dx = mb->mvs[0].x;          pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
333                  uv_dy = mb->mvs[0].y;  
334            if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
335                    uv_dx = pMB->mvs[0].x;
336                    uv_dy = pMB->mvs[0].y;
337    
338                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;                  uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
339                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;                  uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
340          }          } else {
         else  
         {  
341                  int sum;                  int sum;
                 sum = mb->mvs[0].x + mb->mvs[1].x + mb->mvs[2].x + mb->mvs[3].x;  
                 uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );  
342    
343                  sum = mb->mvs[0].y + mb->mvs[1].y + mb->mvs[2].y + mb->mvs[3].y;                  sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
344                  uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );                  uv_dx =
345                            (sum ==
346                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
347                                                                      (ABS(sum) / 16) * 2));
348    
349                    sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
350                    uv_dy =
351                            (sum ==
352                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
353                                                                      (ABS(sum) / 16) * 2));
354          }          }
355    
356          start_timer();          start_timer();
357          interpolate8x8_switch(dec->cur.y, dec->refn.y, 16*x,     16*y    , mb->mvs[0].x, mb->mvs[0].y, stride,  rounding);          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16 * x_pos, 16 * y_pos,
358          interpolate8x8_switch(dec->cur.y, dec->refn.y, 16*x + 8, 16*y    , mb->mvs[1].x, mb->mvs[1].y, stride,  rounding);                                                    pMB->mvs[0].x, pMB->mvs[0].y, stride, rounding);
359          interpolate8x8_switch(dec->cur.y, dec->refn.y, 16*x,     16*y + 8, mb->mvs[2].x, mb->mvs[2].y, stride,  rounding);          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16 * x_pos + 8,
360          interpolate8x8_switch(dec->cur.y, dec->refn.y, 16*x + 8, 16*y + 8, mb->mvs[3].x, mb->mvs[3].y, stride,  rounding);                                                    16 * y_pos, pMB->mvs[1].x, pMB->mvs[1].y, stride,
361          interpolate8x8_switch(dec->cur.u, dec->refn.u, 8*x, 8*y, uv_dx, uv_dy, stride2, rounding);                                                    rounding);
362          interpolate8x8_switch(dec->cur.v, dec->refn.v, 8*x, 8*y, uv_dx, uv_dy, stride2, rounding);          interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16 * x_pos,
363                                                      16 * y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,
364                                                      rounding);
365            interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16 * x_pos + 8,
366                                                      16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
367                                                      rounding);
368            interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
369                                                      uv_dx, uv_dy, stride2, rounding);
370            interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
371                                                      uv_dx, uv_dy, stride2, rounding);
372          stop_comp_timer();          stop_comp_timer();
373    
374            for (i = 0; i < 6; i++) {
375          for (k = 0; k < 6; k++)                  if (cbp & (1 << (5 - i)))       // coded
376          {          {
377                  int16_t block[64];                          memset(&block[i * 64], 0, 64 * sizeof(int16_t));        // clear
                 int16_t data[64];  
   
                 if (cbp & (1 << (5-k)))                 // coded  
                 {  
                         memset(block, 0, 64 * sizeof(int16_t));         // clear  
378    
379                          start_timer();                          start_timer();
380                          get_inter_block(bs, block);                          get_inter_block(bs, &block[i * 64]);
381                          stop_coding_timer();                          stop_coding_timer();
382    
383                          start_timer();                          start_timer();
384                          if (dec->quant_type == 0)                          if (dec->quant_type == 0) {
385                          {                                  dequant_inter(&data[i * 64], &block[i * 64], iQuant);
386                                  dequant_inter(data, block, mb->quant);                          } else {
387                          }                                  dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
                         else  
                         {  
                                 dequant4_inter(data, block, mb->quant);  
388                          }                          }
389                          stop_iquant_timer();                          stop_iquant_timer();
390    
391                          start_timer();                          start_timer();
392                          idct(data);                          idct(&data[i * 64]);
393                          stop_idct_timer();                          stop_idct_timer();
   
                         start_timer();  
                         if (k < 4)  
                         {  
                                 transfer_16to8add(dec->cur.y + (16*y + 4*(k&2))*stride + 16*x + 8*(k&1), data, stride);  
394                          }                          }
                         else if (k == 4)  
                         {  
                                 transfer_16to8add(dec->cur.u + 8*y*stride2 + 8*x, data, stride2);  
395                          }                          }
396                          else // k == 5  
397                          {          if (dec->interlacing && pMB->field_dct) {
398                                  transfer_16to8add(dec->cur.v + 8*y*stride2 + 8*x, data, stride2);                  next_block = stride;
399                    stride *= 2;
400                          }                          }
401    
402            start_timer();
403            if (cbp & 32)
404                    transfer_16to8add(pY_Cur, &data[0 * 64], stride);
405            if (cbp & 16)
406                    transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
407            if (cbp & 8)
408                    transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
409            if (cbp & 4)
410                    transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
411            if (cbp & 2)
412                    transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
413            if (cbp & 1)
414                    transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
415                          stop_transfer_timer();                          stop_transfer_timer();
416                  }                  }
         }  
 }  
   
417    
418    
419  void decoder_iframe(DECODER * dec, Bitstream * bs, int quant, int intra_dc_threshold)  void
420    decoder_iframe(DECODER * dec,
421                               Bitstream * bs,
422                               int quant,
423                               int intra_dc_threshold)
424  {  {
425    
426          uint32_t x, y;          uint32_t x, y;
427    
428          for (y = 0; y < dec->mb_height; y++)          for (y = 0; y < dec->mb_height; y++) {
429          {                  for (x = 0; x < dec->mb_width; x++) {
                 for (x = 0; x < dec->mb_width; x++)  
                 {  
430                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
431    
432                          uint32_t mcbpc;                          uint32_t mcbpc;
# Line 339  Line 441 
441    
442                          acpred_flag = BitstreamGetBit(bs);                          acpred_flag = BitstreamGetBit(bs);
443    
444                          if (mb->mode == MODE_STUFFING)                          if (mb->mode == MODE_STUFFING) {
                         {  
445                                  DEBUG("-- STUFFING ?");                                  DEBUG("-- STUFFING ?");
446                                  continue;                                  continue;
447                          }                          }
# Line 348  Line 449 
449                          cbpy = get_cbpy(bs, 1);                          cbpy = get_cbpy(bs, 1);
450                          cbp = (cbpy << 2) | cbpc;                          cbp = (cbpy << 2) | cbpc;
451    
452                          if (mb->mode == MODE_INTRA_Q)                          if (mb->mode == MODE_INTRA_Q) {
                         {  
453                                  quant += dquant_table[BitstreamGetBits(bs,2)];                                  quant += dquant_table[BitstreamGetBits(bs,2)];
454                                  if (quant > 31)                                  if (quant > 31) {
                                 {  
455                                          quant = 31;                                          quant = 31;
456                                  }                                  } else if (quant < 1) {
                                 else if (quant < 1)  
                                 {  
457                                          quant = 1;                                          quant = 1;
458                                  }                                  }
459                          }                          }
460                          mb->quant = quant;                          mb->quant = quant;
461    
462                            if (dec->interlacing) {
463                                    mb->field_dct = BitstreamGetBit(bs);
464                                    DEBUG1("deci: field_dct: ", mb->field_dct);
465                            }
466    
467                          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,
468                                                            intra_dc_threshold);
469                  }                  }
470          }          }
471    
472  }  }
473    
474    
475  void get_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, int k, VECTOR * mv, int fcode)  void
476    get_motion_vector(DECODER * dec,
477                                      Bitstream * bs,
478                                      int x,
479                                      int y,
480                                      int k,
481                                      VECTOR * mv,
482                                      int fcode)
483  {  {
484    
485          int scale_fac = 1 << (fcode - 1);          int scale_fac = 1 << (fcode - 1);
486          int high = (32 * scale_fac) - 1;          int high = (32 * scale_fac) - 1;
487          int low = ((-32) * scale_fac);          int low = ((-32) * scale_fac);
488          int range = (64 * scale_fac);          int range = (64 * scale_fac);
489    
490          VECTOR pmv[4];          VECTOR pmv[4];
491          uint32_t psad[4];          int32_t psad[4];
492    
493          int mv_x, mv_y;          int mv_x, mv_y;
494          int pmv_x, pmv_y;          int pmv_x, pmv_y;
# Line 394  Line 505 
505          mv_x += pmv_x;          mv_x += pmv_x;
506          mv_y += pmv_y;          mv_y += pmv_y;
507    
508          if (mv_x < low)          if (mv_x < low) {
         {  
509                  mv_x += range;                  mv_x += range;
510          }          } else if (mv_x > high) {
         else if (mv_x > high)  
         {  
511                  mv_x -= range;                  mv_x -= range;
512          }          }
513    
514          if (mv_y < low)          if (mv_y < low) {
         {  
515                  mv_y += range;                  mv_y += range;
516          }          } else if (mv_y > high) {
         else if (mv_y > high)  
         {  
517                  mv_y -= range;                  mv_y -= range;
518          }          }
519    
# Line 418  Line 523 
523  }  }
524    
525    
526  void decoder_pframe(DECODER * dec, Bitstream * bs, int rounding, int quant, int fcode, int intra_dc_threshold)  void
527    decoder_pframe(DECODER * dec,
528                               Bitstream * bs,
529                               int rounding,
530                               int quant,
531                               int fcode,
532                               int intra_dc_threshold)
533  {  {
         uint32_t x, y;  
534    
535          image_swap(&dec->cur, &dec->refn);          uint32_t x, y;
536    
537          start_timer();          start_timer();
538          image_setedges(&dec->refn, dec->edged_width, dec->edged_height, dec->width, dec->height);          image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
539                                       dec->width, dec->height, dec->interlacing);
540          stop_edges_timer();          stop_edges_timer();
541    
542          for (y = 0; y < dec->mb_height; y++)          for (y = 0; y < dec->mb_height; y++) {
543          {                  for (x = 0; x < dec->mb_width; x++) {
                 for (x = 0; x < dec->mb_width; x++)  
                 {  
544                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];                          MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
545    
546                          if (!BitstreamGetBit(bs))                       // not_coded                          //if (!(dec->mb_skip[y*dec->mb_width + x]=BitstreamGetBit(bs)))         // not_coded
547                            if (!(BitstreamGetBit(bs)))     // not_coded
548                          {                          {
549                                  uint32_t mcbpc;                                  uint32_t mcbpc;
550                                  uint32_t cbpc;                                  uint32_t cbpc;
# Line 446  Line 556 
556                                  mcbpc = get_mcbpc_inter(bs);                                  mcbpc = get_mcbpc_inter(bs);
557                                  mb->mode = mcbpc & 7;                                  mb->mode = mcbpc & 7;
558                                  cbpc = (mcbpc >> 4);                                  cbpc = (mcbpc >> 4);
559                                    acpred_flag = 0;
560    
561                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);                                  intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);
562    
563                                  if (intra)                                  if (intra) {
                                 {  
564                                          acpred_flag = BitstreamGetBit(bs);                                          acpred_flag = BitstreamGetBit(bs);
565                                  }                                  }
566    
567                                  if (mb->mode == MODE_STUFFING)                                  if (mb->mode == MODE_STUFFING) {
                                 {  
568                                          DEBUG("-- STUFFING ?");                                          DEBUG("-- STUFFING ?");
569                                          continue;                                          continue;
570                                  }                                  }
# Line 463  Line 572 
572                                  cbpy = get_cbpy(bs, intra);                                  cbpy = get_cbpy(bs, intra);
573                                  cbp = (cbpy << 2) | cbpc;                                  cbp = (cbpy << 2) | cbpc;
574    
575                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q)                                  if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q) {
                                 {  
576                                          quant += dquant_table[BitstreamGetBits(bs,2)];                                          quant += dquant_table[BitstreamGetBits(bs,2)];
577                                          if (quant > 31)                                          if (quant > 31) {
                                         {  
578                                                  quant = 31;                                                  quant = 31;
579                                          }                                          } else if (mb->quant < 1) {
                                         else if (mb->quant < 1)  
                                         {  
580                                                  quant = 1;                                                  quant = 1;
581                                          }                                          }
582                                  }                                  }
583                                  mb->quant = quant;                                  mb->quant = quant;
584    
585                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)                                  if (dec->interlacing) {
586                                  {                                          mb->field_dct = BitstreamGetBit(bs);
587                                            DEBUG1("decp: field_dct: ", mb->field_dct);
588    
589                                            if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
590                                                    mb->field_pred = BitstreamGetBit(bs);
591                                                    DEBUG1("decp: field_pred: ", mb->field_pred);
592    
593                                                    if (mb->field_pred) {
594                                                            mb->field_for_top = BitstreamGetBit(bs);
595                                                            DEBUG1("decp: field_for_top: ", mb->field_for_top);
596                                                            mb->field_for_bot = BitstreamGetBit(bs);
597                                                            DEBUG1("decp: field_for_bot: ", mb->field_for_bot);
598                                                    }
599                                            }
600                                    }
601    
602                                          get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);                                  if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
603                                          mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;                                          if (dec->interlacing && mb->field_pred) {
604                                          mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;                                                  get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
605                                                                                      fcode);
606                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1],
607                                                                                      fcode);
608                                            } else {
609                                                    get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
610                                                                                      fcode);
611                                                    mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
612                                                            mb->mvs[0].x;
613                                                    mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
614                                                            mb->mvs[0].y;
615                                  }                                  }
616                                  else if (mb->mode == MODE_INTER4V /* || mb->mode == MODE_INTER4V_Q */)                                  } else if (mb->mode ==
617                                  {                                                     MODE_INTER4V /* || mb->mode == MODE_INTER4V_Q */ ) {
618                                          get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);                                          get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);
619                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode);                                          get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode);
620                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode);                                          get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode);
621                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode);                                          get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode);
622                                  }                                  } else                  // MODE_INTRA, MODE_INTRA_Q
                                 else  // MODE_INTRA, MODE_INTRA_Q  
623                                  {                                  {
624                                          mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;                                          mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
625                                          mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;                                                  0;
626                                          decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant, intra_dc_threshold);                                          mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
627                                                    0;
628                                            decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
629                                                                            intra_dc_threshold);
630                                          continue;                                          continue;
631                                  }                                  }
632    
633                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant, rounding);                                  decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant,
634                          }                                                                  rounding);
635                          else    // not coded                          } else                          // not coded
636                          {                          {
637                                    //DEBUG2("P-frame MB at (X,Y)=",x,y);
638                                  mb->mode = MODE_NOT_CODED;                                  mb->mode = MODE_NOT_CODED;
639                                  mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;                                  mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
640                                  mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;                                  mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;
# Line 512  Line 643 
643    
644                                  start_timer();                                  start_timer();
645    
646                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),                                  transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
647                                                                  dec->refn.y + (16*y)*dec->edged_width + (16*x),                                                                   (16 * x),
648                                                                  dec->edged_width);                                                                   dec->refn[0].y + (16 * y) * dec->edged_width +
649                                                                     (16 * x), dec->edged_width);
650                                  transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x+8),  
651                                                                  dec->refn.y + (16*y)*dec->edged_width + (16*x+8),                                  transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
652                                                                  dec->edged_width);                                                                   (16 * x + 8),
653                                                                     dec->refn[0].y + (16 * y) * dec->edged_width +
654                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x),                                                                   (16 * x + 8), dec->edged_width);
655                                                                  dec->refn.y + (16*y+8)*dec->edged_width + (16*x),  
656                                                                  dec->edged_width);                                  transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
657                                                                     (16 * x),
658                                  transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x+8),                                                                   dec->refn[0].y + (16 * y +
659                                                                  dec->refn.y + (16*y+8)*dec->edged_width + (16*x+8),                                                                                                     8) * dec->edged_width +
660                                                                  dec->edged_width);                                                                   (16 * x), dec->edged_width);
661    
662                                  transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),                                  transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
663                                                                  dec->refn.u + (8*y)*dec->edged_width/2 + (8*x),                                                                   (16 * x + 8),
664                                                                     dec->refn[0].y + (16 * y +
665                                                                                                       8) * dec->edged_width +
666                                                                     (16 * x + 8), dec->edged_width);
667    
668                                    transfer8x8_copy(dec->cur.u + (8 * y) * dec->edged_width / 2 +
669                                                                     (8 * x),
670                                                                     dec->refn[0].u +
671                                                                     (8 * y) * dec->edged_width / 2 + (8 * x),
672                                                                  dec->edged_width/2);                                                                  dec->edged_width/2);
673    
674                                  transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),                                  transfer8x8_copy(dec->cur.v + (8 * y) * dec->edged_width / 2 +
675                                                                  dec->refn.v + (8*y)*dec->edged_width/2 + (8*x),                                                                   (8 * x),
676                                                                     dec->refn[0].v +
677                                                                     (8 * y) * dec->edged_width / 2 + (8 * x),
678                                                                  dec->edged_width/2);                                                                  dec->edged_width/2);
679    
680                                  stop_transfer_timer();                                  stop_transfer_timer();
# Line 542  Line 683 
683          }          }
684  }  }
685    
686  int decoder_decode(DECODER * dec, XVID_DEC_FRAME * frame)  
687    // add by MinChen <chenm001@163.com>
688    // decode B-frame motion vector
689    void
690    get_b_motion_vector(DECODER * dec,
691                                            Bitstream * bs,
692                                            int x,
693                                            int y,
694                                            VECTOR * mv,
695                                            int fcode,
696                                            const VECTOR pmv)
697    {
698            int scale_fac = 1 << (fcode - 1);
699            int high = (32 * scale_fac) - 1;
700            int low = ((-32) * scale_fac);
701            int range = (64 * scale_fac);
702    
703            int mv_x, mv_y;
704            int pmv_x, pmv_y;
705    
706            pmv_x = pmv.x;
707            pmv_y = pmv.y;
708    
709            mv_x = get_mv(bs, fcode);
710            mv_y = get_mv(bs, fcode);
711    
712            mv_x += pmv_x;
713            mv_y += pmv_y;
714    
715            if (mv_x < low) {
716                    mv_x += range;
717            } else if (mv_x > high) {
718                    mv_x -= range;
719            }
720    
721            if (mv_y < low) {
722                    mv_y += range;
723            } else if (mv_y > high) {
724                    mv_y -= range;
725            }
726    
727            mv->x = mv_x;
728            mv->y = mv_y;
729    }
730    
731    
732    // add by MinChen <chenm001@163.com>
733    // decode an B-frame forward & backward inter macroblock
734    void
735    decoder_bf_mbinter(DECODER * dec,
736                                       const MACROBLOCK * pMB,
737                                       const uint32_t x_pos,
738                                       const uint32_t y_pos,
739                                       const uint32_t cbp,
740                                       Bitstream * bs,
741                                       const uint32_t quant,
742                                       const uint8_t ref)
743    {
744    
745            DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
746            DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
747    
748            uint32_t stride = dec->edged_width;
749            uint32_t stride2 = stride / 2;
750            uint32_t next_block = stride * 8;
751            uint32_t i;
752            uint32_t iQuant = pMB->quant;
753            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
754            int uv_dx, uv_dy;
755    
756            pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
757            pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
758            pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
759    
760            if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
761                    uv_dx = pMB->mvs[0].x;
762                    uv_dy = pMB->mvs[0].y;
763    
764                    uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
765                    uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
766            } else {
767                    int sum;
768    
769                    sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
770                    uv_dx =
771                            (sum ==
772                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
773                                                                      (ABS(sum) / 16) * 2));
774    
775                    sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
776                    uv_dy =
777                            (sum ==
778                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
779                                                                      (ABS(sum) / 16) * 2));
780            }
781    
782            start_timer();
783            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos, 16 * y_pos,
784                                                      pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
785            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos + 8,
786                                                      16 * y_pos, pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
787            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos,
788                                                      16 * y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,
789                                                      0);
790            interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos + 8,
791                                                      16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
792                                                      0);
793            interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8 * x_pos, 8 * y_pos,
794                                                      uv_dx, uv_dy, stride2, 0);
795            interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8 * x_pos, 8 * y_pos,
796                                                      uv_dx, uv_dy, stride2, 0);
797            stop_comp_timer();
798    
799            for (i = 0; i < 6; i++) {
800                    if (cbp & (1 << (5 - i)))       // coded
801                    {
802                            memset(&block[i * 64], 0, 64 * sizeof(int16_t));        // clear
803    
804                            start_timer();
805                            get_inter_block(bs, &block[i * 64]);
806                            stop_coding_timer();
807    
808                            start_timer();
809                            if (dec->quant_type == 0) {
810                                    dequant_inter(&data[i * 64], &block[i * 64], iQuant);
811                            } else {
812                                    dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
813                            }
814                            stop_iquant_timer();
815    
816                            start_timer();
817                            idct(&data[i * 64]);
818                            stop_idct_timer();
819                    }
820            }
821    
822            if (dec->interlacing && pMB->field_dct) {
823                    next_block = stride;
824                    stride *= 2;
825            }
826    
827            start_timer();
828            if (cbp & 32)
829                    transfer_16to8add(pY_Cur, &data[0 * 64], stride);
830            if (cbp & 16)
831                    transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
832            if (cbp & 8)
833                    transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
834            if (cbp & 4)
835                    transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
836            if (cbp & 2)
837                    transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
838            if (cbp & 1)
839                    transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
840            stop_transfer_timer();
841    }
842    
843    
844    // add by MinChen <chenm001@163.com>
845    // decode an B-frame direct &  inter macroblock
846    void
847    decoder_bf_interpolate_mbinter(DECODER * dec,
848                                                               IMAGE forward,
849                                                               IMAGE backward,
850                                                               const MACROBLOCK * pMB,
851                                                               const uint32_t x_pos,
852                                                               const uint32_t y_pos,
853                                                               const uint32_t cbp,
854                                                               Bitstream * bs)
855    {
856    
857            DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
858            DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
859    
860            uint32_t stride = dec->edged_width;
861            uint32_t stride2 = stride / 2;
862            uint32_t next_block = stride * 8;
863            uint32_t iQuant = pMB->quant;
864            int uv_dx, uv_dy;
865            int b_uv_dx, b_uv_dy;
866            uint32_t i;
867            uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
868    
869            pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
870            pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
871            pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
872    
873            if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
874                    uv_dx = pMB->mvs[0].x;
875                    uv_dy = pMB->mvs[0].y;
876    
877                    uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
878                    uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
879    
880                    b_uv_dx = pMB->b_mvs[0].x;
881                    b_uv_dy = pMB->b_mvs[0].y;
882    
883                    b_uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
884                    b_uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
885            } else {
886                    int sum;
887    
888                    sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
889                    uv_dx =
890                            (sum ==
891                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
892                                                                      (ABS(sum) / 16) * 2));
893    
894                    sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
895                    uv_dy =
896                            (sum ==
897                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
898                                                                      (ABS(sum) / 16) * 2));
899    
900                    sum =
901                            pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x +
902                            pMB->b_mvs[3].x;
903                    b_uv_dx =
904                            (sum ==
905                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
906                                                                      (ABS(sum) / 16) * 2));
907    
908                    sum =
909                            pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y +
910                            pMB->b_mvs[3].y;
911                    b_uv_dy =
912                            (sum ==
913                             0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
914                                                                      (ABS(sum) / 16) * 2));
915            }
916    
917    
918            start_timer();
919            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos,
920                                                      pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
921            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8, 16 * y_pos,
922                                                      pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
923            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos + 8,
924                                                      pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
925            interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8,
926                                                      16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
927                                                      0);
928            interpolate8x8_switch(dec->cur.u, forward.u, 8 * x_pos, 8 * y_pos, uv_dx,
929                                                      uv_dy, stride2, 0);
930            interpolate8x8_switch(dec->cur.v, forward.v, 8 * x_pos, 8 * y_pos, uv_dx,
931                                                      uv_dy, stride2, 0);
932    
933    
934            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos, 16 * y_pos,
935                                                      pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
936            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
937                                                      16 * y_pos, pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride,
938                                                      0);
939            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos,
940                                                      16 * y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y,
941                                                      stride, 0);
942            interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
943                                                      16 * y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y,
944                                                      stride, 0);
945            interpolate8x8_switch(dec->refn[2].u, backward.u, 8 * x_pos, 8 * y_pos,
946                                                      b_uv_dx, b_uv_dy, stride2, 0);
947            interpolate8x8_switch(dec->refn[2].v, backward.v, 8 * x_pos, 8 * y_pos,
948                                                      b_uv_dx, b_uv_dy, stride2, 0);
949    
950            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos, 16 * y_pos,
951                                             stride);
952            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos + 8, 16 * y_pos,
953                                             stride);
954            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos, 16 * y_pos + 8,
955                                             stride);
956            interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16 * x_pos + 8,
957                                             16 * y_pos + 8, stride);
958            interpolate8x8_c(dec->cur.u, dec->refn[2].u, 8 * x_pos, 8 * y_pos,
959                                             stride2);
960            interpolate8x8_c(dec->cur.v, dec->refn[2].v, 8 * x_pos, 8 * y_pos,
961                                             stride2);
962    
963            stop_comp_timer();
964    
965            for (i = 0; i < 6; i++) {
966                    if (cbp & (1 << (5 - i)))       // coded
967                    {
968                            memset(&block[i * 64], 0, 64 * sizeof(int16_t));        // clear
969    
970                            start_timer();
971                            get_inter_block(bs, &block[i * 64]);
972                            stop_coding_timer();
973    
974                            start_timer();
975                            if (dec->quant_type == 0) {
976                                    dequant_inter(&data[i * 64], &block[i * 64], iQuant);
977                            } else {
978                                    dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
979                            }
980                            stop_iquant_timer();
981    
982                            start_timer();
983                            idct(&data[i * 64]);
984                            stop_idct_timer();
985                    }
986            }
987    
988            if (dec->interlacing && pMB->field_dct) {
989                    next_block = stride;
990                    stride *= 2;
991            }
992    
993            start_timer();
994            if (cbp & 32)
995                    transfer_16to8add(pY_Cur, &data[0 * 64], stride);
996            if (cbp & 16)
997                    transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
998            if (cbp & 8)
999                    transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
1000            if (cbp & 4)
1001                    transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
1002            if (cbp & 2)
1003                    transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
1004            if (cbp & 1)
1005                    transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
1006            stop_transfer_timer();
1007    }
1008    
1009    
1010    // add by MinChen <chenm001@163.com>
1011    // for decode B-frame dbquant
1012    int32_t __inline
1013    get_dbquant(Bitstream * bs)
1014    {
1015            if (!BitstreamGetBit(bs))       // '0'
1016                    return (0);
1017            else if (!BitstreamGetBit(bs))  // '10'
1018                    return (-2);
1019            else
1020                    return (2);                             // '11'
1021    }
1022    
1023    // add by MinChen <chenm001@163.com>
1024    // for decode B-frame mb_type
1025    // bit   ret_value
1026    // 1        0
1027    // 01       1
1028    // 001      2
1029    // 0001     3
1030    int32_t __inline
1031    get_mbtype(Bitstream * bs)
1032    {
1033            int32_t mb_type;
1034    
1035            for (mb_type = 0; mb_type <= 3; mb_type++) {
1036                    if (BitstreamGetBit(bs))
1037                            break;
1038            }
1039    
1040            if (mb_type <= 3)
1041                    return (mb_type);
1042            else
1043                    return (-1);
1044    }
1045    
1046    void
1047    decoder_bframe(DECODER * dec,
1048                               Bitstream * bs,
1049                               int quant,
1050                               int fcode_forward,
1051                               int fcode_backward)
1052  {  {
1053    
1054            uint32_t x, y;
1055            VECTOR mv, zeromv;
1056    
1057            start_timer();
1058            image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
1059                                       dec->width, dec->height, dec->interlacing);
1060            //image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing);
1061            stop_edges_timer();
1062    
1063    
1064            for (y = 0; y < dec->mb_height; y++) {
1065                    // Initialize Pred Motion Vector
1066                    dec->p_fmv.x = dec->p_fmv.y = dec->p_bmv.x = dec->p_bmv.y = 0;
1067                    for (x = 0; x < dec->mb_width; x++) {
1068                            MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x];
1069                            MACROBLOCK *last_mb = &dec->last_mbs[y * dec->mb_width + x];
1070    
1071                            mb->mvs[0].x = mb->mvs[0].y = zeromv.x = zeromv.y = mv.x = mv.y =
1072                                    0;
1073    
1074                            // the last P_VOP is skip macroblock ?
1075                            if (last_mb->mode == MODE_NOT_CODED) {
1076                                    //DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y);
1077                                    mb->mb_type = MODE_FORWARD;
1078                                    mb->cbp = 0;
1079                                    mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1080                                    mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
1081                                    mb->quant = 8;
1082    
1083                                    decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1084                                    continue;
1085                            }
1086                            //t=BitstreamShowBits(bs,32);
1087    
1088                            if (!BitstreamGetBit(bs)) {     // modb=='0'
1089                                    const uint8_t modb2 = BitstreamGetBit(bs);
1090    
1091                                    mb->mb_type = get_mbtype(bs);
1092    
1093                                    if (!modb2) {   // modb=='00'
1094                                            mb->cbp = BitstreamGetBits(bs, 6);
1095                                    } else {
1096                                            mb->cbp = 0;
1097                                    }
1098                                    if (mb->mb_type && mb->cbp) {
1099                                            quant += get_dbquant(bs);
1100    
1101                                            if (quant > 31) {
1102                                                    quant = 31;
1103                                            } else if (mb->quant < 1) {
1104                                                    quant = 1;
1105                                            }
1106                                    } else {
1107                                            quant = 8;
1108                                    }
1109                                    mb->quant = quant;
1110                            } else {
1111                                    mb->mb_type = MODE_DIRECT_NONE_MV;
1112                                    mb->cbp = 0;
1113                            }
1114    
1115                            mb->mode = MODE_INTER;
1116                            //DEBUG1("Switch bm_type=",mb->mb_type);
1117    
1118                            switch (mb->mb_type) {
1119                            case MODE_DIRECT:
1120                                    get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], 1, zeromv);
1121    
1122                            case MODE_DIRECT_NONE_MV:
1123                                    {                               // Because this file is a C file not C++ so I use '{' to define var
1124                                            const int64_t TRB = dec->time_pp - dec->time_bp, TRD =
1125                                                    dec->time_pp;
1126                                            int i;
1127    
1128                                            for (i = 0; i < 4; i++) {
1129                                                    mb->mvs[i].x =
1130                                                            (int32_t) ((TRB * last_mb->mvs[i].x) / TRD +
1131                                                                               mb->mvs[0].x);
1132                                                    mb->b_mvs[i].x =
1133                                                            (int32_t) ((mb->mvs[0].x ==
1134                                                                                    0) ? ((TRB -
1135                                                                                               TRD) * last_mb->mvs[i].x) /
1136                                                                               TRD : mb->mvs[i].x - last_mb->mvs[i].x);
1137                                                    mb->mvs[i].y =
1138                                                            (int32_t) ((TRB * last_mb->mvs[i].y) / TRD +
1139                                                                               mb->mvs[0].y);
1140                                                    mb->b_mvs[i].y =
1141                                                            (int32_t) ((mb->mvs[0].y ==
1142                                                                                    0) ? ((TRB -
1143                                                                                               TRD) * last_mb->mvs[i].y) /
1144                                                                               TRD : mb->mvs[i].y - last_mb->mvs[i].y);
1145                                            }
1146                                            //DEBUG("B-frame Direct!\n");
1147                                    }
1148                                    mb->mode = MODE_INTER4V;
1149                                    decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1150                                                                                               mb, x, y, mb->cbp, bs);
1151                                    break;
1152    
1153                            case MODE_INTERPOLATE:
1154                                    get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1155                                                                            dec->p_fmv);
1156                                    dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
1157                                            mb->mvs[0].x;
1158                                    dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
1159                                            mb->mvs[0].y;
1160    
1161                                    get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0],
1162                                                                            fcode_backward, dec->p_bmv);
1163                                    dec->p_bmv.x = mb->b_mvs[1].x = mb->b_mvs[2].x =
1164                                            mb->b_mvs[3].x = mb->b_mvs[0].x;
1165                                    dec->p_bmv.y = mb->b_mvs[1].y = mb->b_mvs[2].y =
1166                                            mb->b_mvs[3].y = mb->b_mvs[0].y;
1167    
1168                                    decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1169                                                                                               mb, x, y, mb->cbp, bs);
1170                                    //DEBUG("B-frame Bidir!\n");
1171                                    break;
1172    
1173                            case MODE_BACKWARD:
1174                                    get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_backward,
1175                                                                            dec->p_bmv);
1176                                    dec->p_bmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
1177                                            mb->mvs[0].x;
1178                                    dec->p_bmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
1179                                            mb->mvs[0].y;
1180    
1181                                    decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);
1182                                    //DEBUG("B-frame Backward!\n");
1183                                    break;
1184    
1185                            case MODE_FORWARD:
1186                                    get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1187                                                                            dec->p_fmv);
1188                                    dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
1189                                            mb->mvs[0].x;
1190                                    dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
1191                                            mb->mvs[0].y;
1192    
1193                                    decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1194                                    //DEBUG("B-frame Forward!\n");
1195                                    break;
1196    
1197                            default:
1198                                    DEBUG1("Not support B-frame mb_type =", mb->mb_type);
1199                            }
1200    
1201                    }                                               // end of FOR
1202            }
1203    }
1204    
1205    // swap two MACROBLOCK array
1206    void
1207    mb_swap(MACROBLOCK ** mb1,
1208                    MACROBLOCK ** mb2)
1209    {
1210            MACROBLOCK *temp = *mb1;
1211    
1212            *mb1 = *mb2;
1213            *mb2 = temp;
1214    }
1215    
1216    int
1217    decoder_decode(DECODER * dec,
1218                               XVID_DEC_FRAME * frame)
1219    {
1220    
1221          Bitstream bs;          Bitstream bs;
1222          uint32_t rounding;          uint32_t rounding;
1223          uint32_t quant;          uint32_t quant;
1224          uint32_t fcode;          uint32_t fcode_forward;
1225            uint32_t fcode_backward;
1226          uint32_t intra_dc_threshold;          uint32_t intra_dc_threshold;
1227            uint32_t vop_type;
1228    
1229          start_global_timer();          start_global_timer();
1230    
1231          BitstreamInit(&bs, frame->bitstream, frame->length);          BitstreamInit(&bs, frame->bitstream, frame->length);
1232    
1233          switch (BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode, &intra_dc_threshold))          // add by chenm001 <chenm001@163.com>
1234          {          // for support B-frame to reference last 2 frame
1235            dec->frames++;
1236            vop_type =
1237                    BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward,
1238                                                             &fcode_backward, &intra_dc_threshold);
1239    
1240            dec->p_bmv.x = dec->p_bmv.y = dec->p_fmv.y = dec->p_fmv.y = 0;  // init pred vector to 0
1241    
1242            switch (vop_type) {
1243          case P_VOP :          case P_VOP :
1244                  decoder_pframe(dec, &bs, rounding, quant, fcode, intra_dc_threshold);                  decoder_pframe(dec, &bs, rounding, quant, fcode_forward,
1245                                               intra_dc_threshold);
1246                    DEBUG1("P_VOP  Time=", dec->time);
1247                  break;                  break;
1248    
1249          case I_VOP :          case I_VOP :
                 //DEBUG1("",intra_dc_threshold);  
1250                  decoder_iframe(dec, &bs, quant, intra_dc_threshold);                  decoder_iframe(dec, &bs, quant, intra_dc_threshold);
1251                    DEBUG1("I_VOP  Time=", dec->time);
1252                  break;                  break;
1253    
1254          case B_VOP :    // ignore          case B_VOP:
1255    #ifdef BFRAMES_DEC
1256                    if (dec->time_pp > dec->time_bp) {
1257                            DEBUG1("B_VOP  Time=", dec->time);
1258                            decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);
1259                    } else {
1260                            DEBUG("broken B-frame!");
1261                    }
1262    #else
1263                    image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1264    #endif
1265                  break;                  break;
1266    
1267          case N_VOP :    // vop not coded          case N_VOP :    // vop not coded
1268                    // when low_delay==0, N_VOP's should interpolate between the past and future frames
1269                    image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1270                  break;                  break;
1271    
1272          default :          default :
# Line 577  Line 1275 
1275    
1276          frame->length = BitstreamPos(&bs) / 8;          frame->length = BitstreamPos(&bs) / 8;
1277    
1278          start_timer();  #ifdef BFRAMES_DEC
1279            // test if no B_VOP
1280            if (dec->low_delay) {
1281    #endif
1282          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,          image_output(&dec->cur, dec->width, dec->height, dec->edged_width,
1283                                  frame->image, frame->stride, frame->colorspace);                                  frame->image, frame->stride, frame->colorspace);
1284    #ifdef BFRAMES_DEC
1285            } else {
1286                    if (dec->frames >= 1) {
1287                            start_timer();
1288                            if ((vop_type == I_VOP || vop_type == P_VOP)) {
1289                                    image_output(&dec->refn[0], dec->width, dec->height,
1290                                                             dec->edged_width, frame->image, frame->stride,
1291                                                             frame->colorspace);
1292                            } else if (vop_type == B_VOP) {
1293                                    image_output(&dec->cur, dec->width, dec->height,
1294                                                             dec->edged_width, frame->image, frame->stride,
1295                                                             frame->colorspace);
1296                            }
1297          stop_conv_timer();          stop_conv_timer();
1298                    }
1299            }
1300    #endif
1301    
1302            if (vop_type == I_VOP || vop_type == P_VOP) {
1303                    image_swap(&dec->refn[0], &dec->refn[1]);
1304                    image_swap(&dec->cur, &dec->refn[0]);
1305                    // swap MACROBLOCK
1306                    if (dec->low_delay && vop_type == P_VOP)
1307                            mb_swap(&dec->mbs, &dec->last_mbs);
1308            }
1309    
1310          emms();          emms();
1311    

Legend:
Removed from v.1.1  
changed lines
  Added in v.1.21

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