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

Diff of /xvidcore/src/encoder.c

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

revision 1.6, Sat Mar 16 15:55:12 2002 UTC revision 1.22, Sun Apr 7 11:57:47 2002 UTC
# Line 16  Line 16 
16  #include "bitstream/mbcoding.h"  #include "bitstream/mbcoding.h"
17  #include "quant/adapt_quant.h"  #include "quant/adapt_quant.h"
18  #include "quant/quant_matrix.h"  #include "quant/quant_matrix.h"
19    #include "utils/mem_align.h"
20    
21  #define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT  #define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT
22    
# Line 81  Line 82 
82                  pParam->bitrate = 900000;                  pParam->bitrate = 900000;
83    
84          if (pParam->rc_buffersize <= 0)          if (pParam->rc_buffersize <= 0)
85                  pParam->rc_buffersize = pParam->bitrate * pParam->fbase;                  pParam->rc_buffersize = 16;
86    
87          if ((pParam->min_quantizer <= 0) || (pParam->min_quantizer > 31))          if ((pParam->min_quantizer <= 0) || (pParam->min_quantizer > 31))
88                  pParam->min_quantizer = 1;                  pParam->min_quantizer = 1;
# Line 95  Line 96 
96          if (pParam->max_quantizer < pParam->min_quantizer)          if (pParam->max_quantizer < pParam->min_quantizer)
97                  pParam->max_quantizer = pParam->min_quantizer;                  pParam->max_quantizer = pParam->min_quantizer;
98    
99          if ((pEnc = (Encoder *) malloc(sizeof(Encoder))) == NULL)          if ((pEnc = (Encoder *) xvid_malloc(sizeof(Encoder), CACHE_LINE)) == NULL)
100                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
101    
102          /* Fill members of Encoder structure */          /* Fill members of Encoder structure */
# Line 120  Line 121 
121          pEnc->iFrameNum = 0;          pEnc->iFrameNum = 0;
122          pEnc->iMaxKeyInterval = pParam->max_key_interval;          pEnc->iMaxKeyInterval = pParam->max_key_interval;
123    
124          if (image_create(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)          /* try to allocate memory */
         {  
                 free(pEnc);  
                 return XVID_ERR_MEMORY;  
         }  
   
         if (image_create(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)  
         {  
                 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);  
                 free(pEnc);  
                 return XVID_ERR_MEMORY;  
         }  
   
         if (image_create(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)  
         {  
                 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);  
                 image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);  
                 free(pEnc);  
                 return XVID_ERR_MEMORY;  
         }  
   
         if (image_create(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)  
         {  
                 image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);  
                 image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);  
                 image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);  
                 free(pEnc);  
                 return XVID_ERR_MEMORY;  
         }  
125    
126          if (image_create(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0)          pEnc->sCurrent.y        =       pEnc->sCurrent.u        =       pEnc->sCurrent.v        = NULL;
127          {          pEnc->sReference.y      =       pEnc->sReference.u      =       pEnc->sReference.v      = NULL;
128                  image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          pEnc->vInterH.y         =       pEnc->vInterH.u         =       pEnc->vInterH.v         = NULL;
129                  image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          pEnc->vInterV.y         =       pEnc->vInterV.u         =       pEnc->vInterV.v         = NULL;
130                  image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          pEnc->vInterVf.y        =       pEnc->vInterVf.u        =       pEnc->vInterVf.v        = NULL;
131                  image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          pEnc->vInterHV.y        =       pEnc->vInterHV.u        =       pEnc->vInterHV.v        = NULL;
132                  free(pEnc);          pEnc->vInterHVf.y       =       pEnc->vInterHVf.u       =       pEnc->vInterHVf.v       = NULL;
133                  return XVID_ERR_MEMORY;  
134          }          pEnc->pMBs = NULL;
135    
136          pEnc->pMBs = malloc(sizeof(MACROBLOCK) * pEnc->mbParam.mb_width * pEnc->mbParam.mb_height);          if (image_create(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
137          if (pEnc->pMBs == NULL)                  image_create(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
138                    image_create(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
139                    image_create(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
140                    image_create(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
141                    image_create(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
142                    image_create(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
143                    (pEnc->pMBs = xvid_malloc(sizeof(MACROBLOCK) * pEnc->mbParam.mb_width * pEnc->mbParam.mb_height, CACHE_LINE)) == NULL)
144          {          {
145                  image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
146                  image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
147                  image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
148                  image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
149                    image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
150                  image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
151                  free(pEnc);                  image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
152                    if (pEnc)
153                    {
154                            xvid_free(pEnc);
155                    }
156                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
157          }          }
158    
# Line 182  Line 166 
166    
167          if (pParam->bitrate)          if (pParam->bitrate)
168          {          {
169                  RateControlInit(pParam->bitrate, pParam->rc_buffersize, pParam->fbase, pParam->width,                  RateControlInit(pParam->bitrate, pParam->rc_buffersize, pParam->fbase * 1000 / pParam->fincr,
170                                  pParam->height, pParam->max_quantizer, pParam->min_quantizer);                                  pParam->max_quantizer, pParam->min_quantizer);
171          }          }
172    
         create_vlc_tables();  
173          init_timer();          init_timer();
174    
175          return XVID_ERR_OK;          return XVID_ERR_OK;
# Line 199  Line 182 
182          ENC_CHECK(pEnc->sCurrent.y);          ENC_CHECK(pEnc->sCurrent.y);
183          ENC_CHECK(pEnc->sReference.y);          ENC_CHECK(pEnc->sReference.y);
184    
185          free(pEnc->pMBs);          xvid_free(pEnc->pMBs);
186          image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
187          image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
188          image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
189          image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
190            image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
191          image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
192          free(pEnc);          image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
193            xvid_free(pEnc);
         destroy_vlc_tables();  
194    
195          return XVID_ERR_OK;          return XVID_ERR_OK;
196  }  }
# Line 228  Line 211 
211    
212          pEnc->mbParam.global_flags = pFrame->general;          pEnc->mbParam.global_flags = pFrame->general;
213          pEnc->mbParam.motion_flags = pFrame->motion;          pEnc->mbParam.motion_flags = pFrame->motion;
214            pEnc->mbParam.hint = &pFrame->hint;
215    
216          start_timer();          start_timer();
217          if (image_input(&pEnc->sCurrent, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.edged_width,          if (image_input(&pEnc->sCurrent, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.edged_width,
# Line 237  Line 221 
221          }          }
222          stop_conv_timer();          stop_conv_timer();
223    
224            EMMS();
225    
226          BitstreamInit(&bs, pFrame->bitstream, 0);          BitstreamInit(&bs, pFrame->bitstream, 0);
227    
228          if (pFrame->quant == 0)          if (pFrame->quant == 0)
# Line 250  Line 236 
236    
237          if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0)          if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0)
238          {          {
239                  int * temp_dquants = (int *) malloc(pEnc->mbParam.mb_width * pEnc->mbParam.mb_height * sizeof(int));                  int * temp_dquants = (int *) xvid_malloc(pEnc->mbParam.mb_width * pEnc->mbParam.mb_height * sizeof(int), CACHE_LINE);
240    
241                  pEnc->mbParam.quant = adaptive_quantization(pEnc->sCurrent.y, pEnc->mbParam.width,                  pEnc->mbParam.quant = adaptive_quantization(pEnc->sCurrent.y,
242                                                              temp_dquants, pFrame->quant, pFrame->quant,                                                              pEnc->mbParam.width,
243                                                              2*pFrame->quant, pEnc->mbParam.mb_width, pEnc->mbParam.mb_height);                                                              temp_dquants,
244                                                                pEnc->mbParam.quant,
245                                                                pEnc->mbParam.quant,
246                                                                2*pEnc->mbParam.quant,
247                                                                pEnc->mbParam.mb_width,
248                                                                pEnc->mbParam.mb_height);
249    
250                  for (y = 0; y < pEnc->mbParam.mb_height; y++)                  for (y = 0; y < pEnc->mbParam.mb_height; y++)
251                          for (x = 0; x < pEnc->mbParam.mb_width; x++)                          for (x = 0; x < pEnc->mbParam.mb_width; x++)
# Line 262  Line 253 
253                                  MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];                                  MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
254                                  pMB->dquant = iDQtab[(temp_dquants[y * pEnc->mbParam.mb_width + x] + 2)];                                  pMB->dquant = iDQtab[(temp_dquants[y * pEnc->mbParam.mb_width + x] + 2)];
255                          }                          }
256                  free(temp_dquants);                  xvid_free(temp_dquants);
257          }          }
258    
259          if(pEnc->mbParam.global_flags & XVID_H263QUANT) {          if(pEnc->mbParam.global_flags & XVID_H263QUANT) {
# Line 273  Line 264 
264          else if(pEnc->mbParam.global_flags & XVID_MPEGQUANT) {          else if(pEnc->mbParam.global_flags & XVID_MPEGQUANT) {
265                  int ret1, ret2;                  int ret1, ret2;
266    
267                    ret1 = ret2 = 0;
268    
269                  if(pEnc->mbParam.quant_type != MPEG4_QUANT)                  if(pEnc->mbParam.quant_type != MPEG4_QUANT)
270                          write_vol_header = 1;                          write_vol_header = 1;
271    
# Line 323  Line 316 
316                  pResult->ublks = pEnc->sStat.ublks;                  pResult->ublks = pEnc->sStat.ublks;
317          }          }
318    
319            EMMS();
320    
321          if (pFrame->quant == 0)          if (pFrame->quant == 0)
322          {          {
323                  RateControlUpdate(pEnc->mbParam.quant, pFrame->length, pFrame->intra);                  RateControlUpdate(pEnc->mbParam.quant, pFrame->length, pFrame->intra);
# Line 357  Line 352 
352  }  }
353    
354    
355    #define FCODEBITS       3
356    #define MODEBITS        5
357    
358    void HintedMESet(Encoder * pEnc, int * intra)
359    {
360            HINTINFO * hint;
361            Bitstream bs;
362            int length, high;
363            uint32_t x, y;
364    
365            hint = pEnc->mbParam.hint;
366    
367            if (hint->rawhints)
368            {
369                    *intra = hint->mvhint.intra;
370            }
371            else
372            {
373                    BitstreamInit(&bs, hint->hintstream, hint->hintlength);
374                    *intra = BitstreamGetBit(&bs);
375            }
376    
377            if (*intra)
378            {
379                    return;
380            }
381    
382            pEnc->mbParam.fixed_code = (hint->rawhints) ? hint->mvhint.fcode : BitstreamGetBits(&bs, FCODEBITS);
383    
384            length  = pEnc->mbParam.fixed_code + 5;
385            high    = 1 << (length - 1);
386    
387            for (y=0 ; y<pEnc->mbParam.mb_height ; ++y)
388            {
389                    for (x=0 ; x<pEnc->mbParam.mb_width ; ++x)
390                    {
391                            MACROBLOCK * pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
392                            MVBLOCKHINT * bhint = &hint->mvhint.block[x + y * pEnc->mbParam.mb_width];
393                            VECTOR pred[4];
394                            VECTOR tmp;
395                            int dummy[4];
396                            int vec;
397    
398                            pMB->mode = (hint->rawhints) ? bhint->mode : BitstreamGetBits(&bs, MODEBITS);
399    
400                            if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)
401                            {
402                                    tmp.x  = (hint->rawhints) ? bhint->mvs[0].x : BitstreamGetBits(&bs, length);
403                                    tmp.y  = (hint->rawhints) ? bhint->mvs[0].y : BitstreamGetBits(&bs, length);
404                                    tmp.x -= (tmp.x >= high) ? high*2 : 0;
405                                    tmp.y -= (tmp.y >= high) ? high*2 : 0;
406    
407                                    get_pmvdata(pEnc->pMBs, x, y, pEnc->mbParam.mb_width, 0, pred, dummy);
408    
409                                    for (vec=0 ; vec<4 ; ++vec)
410                                    {
411                                            pMB->mvs[vec].x  = tmp.x;
412                                            pMB->mvs[vec].y  = tmp.y;
413                                            pMB->pmvs[vec].x = pMB->mvs[0].x - pred[0].x;
414                                            pMB->pmvs[vec].y = pMB->mvs[0].y - pred[0].y;
415                                    }
416                            }
417                            else if (pMB->mode == MODE_INTER4V)
418                            {
419                                    for (vec=0 ; vec<4 ; ++vec)
420                                    {
421                                            tmp.x  = (hint->rawhints) ? bhint->mvs[vec].x : BitstreamGetBits(&bs, length);
422                                            tmp.y  = (hint->rawhints) ? bhint->mvs[vec].y : BitstreamGetBits(&bs, length);
423                                            tmp.x -= (tmp.x >= high) ? high*2 : 0;
424                                            tmp.y -= (tmp.y >= high) ? high*2 : 0;
425    
426                                            get_pmvdata(pEnc->pMBs, x, y, pEnc->mbParam.mb_width, vec, pred, dummy);
427    
428                                            pMB->mvs[vec].x  = tmp.x;
429                                            pMB->mvs[vec].y  = tmp.y;
430                                            pMB->pmvs[vec].x = pMB->mvs[vec].x - pred[0].x;
431                                            pMB->pmvs[vec].y = pMB->mvs[vec].y - pred[0].y;
432                                    }
433                            }
434                            else    // intra / intra_q / stuffing / not_coded
435                            {
436                                    for (vec=0 ; vec<4 ; ++vec)
437                                    {
438                                            pMB->mvs[vec].x  = pMB->mvs[vec].y  = 0;
439                                    }
440                            }
441                    }
442            }
443    }
444    
445    
446    void HintedMEGet(Encoder * pEnc, int intra)
447    {
448            HINTINFO * hint;
449            Bitstream bs;
450            uint32_t x, y;
451            int length, high;
452    
453            hint = pEnc->mbParam.hint;
454    
455            if (hint->rawhints)
456            {
457                    hint->mvhint.intra = intra;
458            }
459            else
460            {
461                    BitstreamInit(&bs, hint->hintstream, 0);
462                    BitstreamPutBit(&bs, intra);
463            }
464    
465            if (intra)
466            {
467                    if (!hint->rawhints)
468                    {
469                            BitstreamPad(&bs);
470                            hint->hintlength = BitstreamLength(&bs);
471                    }
472                    return;
473            }
474    
475            length  = pEnc->mbParam.fixed_code + 5;
476            high    = 1 << (length - 1);
477    
478            if (hint->rawhints)
479            {
480                    hint->mvhint.fcode = pEnc->mbParam.fixed_code;
481            }
482            else
483            {
484                    BitstreamPutBits(&bs, pEnc->mbParam.fixed_code, FCODEBITS);
485            }
486    
487            for (y=0 ; y<pEnc->mbParam.mb_height ; ++y)
488            {
489                    for (x=0 ; x<pEnc->mbParam.mb_width ; ++x)
490                    {
491                            MACROBLOCK * pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
492                            MVBLOCKHINT * bhint = &hint->mvhint.block[x + y * pEnc->mbParam.mb_width];
493                            VECTOR tmp;
494    
495                            if (hint->rawhints)
496                            {
497                                    bhint->mode = pMB->mode;
498                            }
499                            else
500                            {
501                                    BitstreamPutBits(&bs, pMB->mode, MODEBITS);
502                            }
503    
504                            if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)
505                            {
506                                    tmp.x  = pMB->mvs[0].x;
507                                    tmp.y  = pMB->mvs[0].y;
508                                    tmp.x += (tmp.x < 0) ? high*2 : 0;
509                                    tmp.y += (tmp.y < 0) ? high*2 : 0;
510    
511                                    if (hint->rawhints)
512                                    {
513                                            bhint->mvs[0].x = tmp.x;
514                                            bhint->mvs[0].y = tmp.y;
515                                    }
516                                    else
517                                    {
518                                            BitstreamPutBits(&bs, tmp.x, length);
519                                            BitstreamPutBits(&bs, tmp.y, length);
520                                    }
521                            }
522                            else if (pMB->mode == MODE_INTER4V)
523                            {
524                                    int vec;
525    
526                                    for (vec=0 ; vec<4 ; ++vec)
527                                    {
528                                            tmp.x  = pMB->mvs[vec].x;
529                                            tmp.y  = pMB->mvs[vec].y;
530                                            tmp.x += (tmp.x < 0) ? high*2 : 0;
531                                            tmp.y += (tmp.y < 0) ? high*2 : 0;
532    
533                                            if (hint->rawhints)
534                                            {
535                                                    bhint->mvs[vec].x = tmp.x;
536                                                    bhint->mvs[vec].y = tmp.y;
537                                            }
538                                            else
539                                            {
540                                                    BitstreamPutBits(&bs, tmp.x, length);
541                                                    BitstreamPutBits(&bs, tmp.y, length);
542                                            }
543                                    }
544                            }
545                    }
546            }
547    
548            if (!hint->rawhints)
549            {
550                    BitstreamPad(&bs);
551                    hint->hintlength = BitstreamLength(&bs);
552            }
553    }
554    
555    
556  static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)  static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)
557  {  {
558          int16_t dct_codes[6][64];  
559          int16_t qcoeff[6][64];          DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
560            DECLARE_ALIGNED_MATRIX(qcoeff,    6, 64, int16_t, CACHE_LINE);
561    
562          uint16_t x, y;          uint16_t x, y;
563    
564          pEnc->iFrameNum = 0;          pEnc->iFrameNum = 0;
565          pEnc->mbParam.rounding_type = 1;          pEnc->mbParam.rounding_type = 1;
566          pEnc->mbParam.coding_type = I_VOP;          pEnc->mbParam.coding_type = I_VOP;
567    
568          BitstreamWriteVolHeader(bs, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.quant_type);          BitstreamWriteVolHeader(bs, &pEnc->mbParam);
569          BitstreamWriteVopHeader(bs, I_VOP, pEnc->mbParam.rounding_type,          BitstreamWriteVopHeader(bs, &pEnc->mbParam);
                                 pEnc->mbParam.quant,  
                                 pEnc->mbParam.fixed_code);  
570    
571          *pBits = BitstreamPos(bs);          *pBits = BitstreamPos(bs);
572    
# Line 385  Line 581 
581    
582                          CodeIntraMB(pEnc, pMB);                          CodeIntraMB(pEnc, pMB);
583    
584                          MBTransQuantIntra(&pEnc->mbParam, x, y, dct_codes, qcoeff, &pEnc->sCurrent);                          MBTransQuantIntra(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, &pEnc->sCurrent);
585    
586                          start_timer();                          start_timer();
587                          MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs);                          MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs);
# Line 404  Line 600 
600          pEnc->sStat.iMvCount = 0;          pEnc->sStat.iMvCount = 0;
601          pEnc->mbParam.fixed_code = 2;          pEnc->mbParam.fixed_code = 2;
602    
603            if (pEnc->mbParam.global_flags & XVID_HINTEDME_GET)
604            {
605                    HintedMEGet(pEnc, 1);
606            }
607    
608          return 1;                                        // intra          return 1;                                        // intra
609  }  }
610    
# Line 413  Line 614 
614  static int FrameCodeP(Encoder * pEnc, Bitstream * bs, uint32_t *pBits, bool force_inter, bool vol_header)  static int FrameCodeP(Encoder * pEnc, Bitstream * bs, uint32_t *pBits, bool force_inter, bool vol_header)
615  {  {
616          float fSigma;          float fSigma;
617          int16_t dct_codes[6][64];  
618          int16_t qcoeff[6][64];          DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
619            DECLARE_ALIGNED_MATRIX(qcoeff,    6, 64, int16_t, CACHE_LINE);
620    
621          int iLimit;          int iLimit;
622          uint32_t x, y;          uint32_t x, y;
623          int iSearchRange;          int iSearchRange;
# Line 423  Line 626 
626          IMAGE *pCurrent = &pEnc->sCurrent;          IMAGE *pCurrent = &pEnc->sCurrent;
627          IMAGE *pRef = &pEnc->sReference;          IMAGE *pRef = &pEnc->sReference;
628    
629          image_setedges(pRef,pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, pEnc->mbParam.width, pEnc->mbParam.height);          start_timer();
630            image_setedges(pRef,
631                           pEnc->mbParam.edged_width,
632                           pEnc->mbParam.edged_height,
633                           pEnc->mbParam.width,
634                           pEnc->mbParam.height,
635                           pEnc->mbParam.global_flags & XVID_INTERLACING);
636            stop_edges_timer();
637    
638          pEnc->mbParam.rounding_type = 1 - pEnc->mbParam.rounding_type;          pEnc->mbParam.rounding_type = 1 - pEnc->mbParam.rounding_type;
639    
# Line 441  Line 651 
651          }          }
652    
653          start_timer();          start_timer();
654            if (pEnc->mbParam.global_flags & XVID_HINTEDME_SET)
655            {
656                    HintedMESet(pEnc, &bIntra);
657            }
658            else
659            {
660          bIntra = MotionEstimation(pEnc->pMBs, &pEnc->mbParam, &pEnc->sReference,          bIntra = MotionEstimation(pEnc->pMBs, &pEnc->mbParam, &pEnc->sReference,
661                                    &pEnc->vInterH, &pEnc->vInterV,                                    &pEnc->vInterH, &pEnc->vInterV,
662                                    &pEnc->vInterHV, &pEnc->sCurrent, iLimit);                                    &pEnc->vInterHV, &pEnc->sCurrent, iLimit);
663            }
664          stop_motion_timer();          stop_motion_timer();
665    
666          if (bIntra == 1)          if (bIntra == 1)
667            {
668                  return FrameCodeI(pEnc, bs, pBits);                  return FrameCodeI(pEnc, bs, pBits);
669            }
670    
671          pEnc->mbParam.coding_type = P_VOP;          pEnc->mbParam.coding_type = P_VOP;
672    
673          if(vol_header)          if(vol_header)
674                  BitstreamWriteVolHeader(bs, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.quant_type);                  BitstreamWriteVolHeader(bs, &pEnc->mbParam);
675    
676          BitstreamWriteVopHeader(bs, P_VOP, pEnc->mbParam.rounding_type,          BitstreamWriteVopHeader(bs, &pEnc->mbParam);
                                 pEnc->mbParam.quant,  
                                 pEnc->mbParam.fixed_code);  
677    
678          *pBits = BitstreamPos(bs);          *pBits = BitstreamPos(bs);
679    
# Line 476  Line 693 
693                          if (!bIntra)                          if (!bIntra)
694                          {                          {
695                                  start_timer();                                  start_timer();
696                                  MBMotionCompensation(pMB, x, y, &pEnc->sReference,                                  MBMotionCompensation(pMB,
697                                                       &pEnc->vInterH, &pEnc->vInterV,                                                       x, y,
698                                                       &pEnc->vInterHV, &pEnc->sCurrent, dct_codes,                                                       &pEnc->sReference,
699                                                         &pEnc->vInterH,
700                                                         &pEnc->vInterV,
701                                                         &pEnc->vInterHV,
702                                                         &pEnc->sCurrent,
703                                                         dct_codes,
704                                                       pEnc->mbParam.width,                                                       pEnc->mbParam.width,
705                                                       pEnc->mbParam.height,                                                       pEnc->mbParam.height,
706                                                       pEnc->mbParam.edged_width,                                                       pEnc->mbParam.edged_width,
# Line 495  Line 717 
717                                  }                                  }
718                                  pMB->quant = pEnc->mbParam.quant;                                  pMB->quant = pEnc->mbParam.quant;
719    
720                                  pMB->cbp = MBTransQuantInter(&pEnc->mbParam, x, y, dct_codes, qcoeff, pCurrent);                                  pMB->field_pred = 0;
721    
722                                    pMB->cbp = MBTransQuantInter(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, pCurrent);
723                          }                          }
724                          else                          else
725                          {                          {
726                                  CodeIntraMB(pEnc, pMB);                                  CodeIntraMB(pEnc, pMB);
727                                  MBTransQuantIntra(&pEnc->mbParam, x, y, dct_codes, qcoeff, pCurrent);                                  MBTransQuantIntra(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, pCurrent);
728                          }                          }
729    
730                          start_timer();                          start_timer();
# Line 532  Line 756 
756    
757          emms();          emms();
758    
759            if (pEnc->mbParam.global_flags & XVID_HINTEDME_GET)
760            {
761                    HintedMEGet(pEnc, 0);
762            }
763    
764          if (pEnc->sStat.iMvCount == 0)          if (pEnc->sStat.iMvCount == 0)
765                  pEnc->sStat.iMvCount = 1;                  pEnc->sStat.iMvCount = 1;
766    

Legend:
Removed from v.1.6  
changed lines
  Added in v.1.22

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