[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.5, Sat Mar 16 09:55:19 2002 UTC revision 1.26, Wed Apr 17 14:05:54 2002 UTC
# Line 1  Line 1 
1    // 14.04.2002   added FrameCodeB()
2    
3  #include <stdlib.h>  #include <stdlib.h>
4  #include <stdio.h>  #include <stdio.h>
5  #include <math.h>  #include <math.h>
# Line 16  Line 18 
18  #include "bitstream/mbcoding.h"  #include "bitstream/mbcoding.h"
19  #include "quant/adapt_quant.h"  #include "quant/adapt_quant.h"
20  #include "quant/quant_matrix.h"  #include "quant/quant_matrix.h"
21    #include "utils/mem_align.h"
22    
23  #define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT  #define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT
24    
# Line 77  Line 80 
80                  pParam->fincr = (int)(pParam->fincr / div);                  pParam->fincr = (int)(pParam->fincr / div);
81          }          }
82    
83          if (pParam->bitrate <= 0)          if (pParam->rc_bitrate <= 0)
84                  pParam->bitrate = 900000;                  pParam->rc_bitrate = 900000;
85    
86            if (pParam->rc_reaction_delay_factor <= 0)
87                    pParam->rc_reaction_delay_factor = 16;
88    
89          if (pParam->rc_buffersize <= 0)          if (pParam->rc_averaging_period <= 0)
90                  pParam->rc_buffersize = pParam->bitrate * pParam->fbase;                  pParam->rc_averaging_period = 100;
91    
92            if (pParam->rc_buffer <= 0)
93                    pParam->rc_buffer = 100;
94    
95          if ((pParam->min_quantizer <= 0) || (pParam->min_quantizer > 31))          if ((pParam->min_quantizer <= 0) || (pParam->min_quantizer > 31))
96                  pParam->min_quantizer = 1;                  pParam->min_quantizer = 1;
# Line 95  Line 104 
104          if (pParam->max_quantizer < pParam->min_quantizer)          if (pParam->max_quantizer < pParam->min_quantizer)
105                  pParam->max_quantizer = pParam->min_quantizer;                  pParam->max_quantizer = pParam->min_quantizer;
106    
107          if ((pEnc = (Encoder *) malloc(sizeof(Encoder))) == NULL)          if ((pEnc = (Encoder *) xvid_malloc(sizeof(Encoder), CACHE_LINE)) == NULL)
108                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
109    
110          /* Fill members of Encoder structure */          /* Fill members of Encoder structure */
# Line 115  Line 124 
124    
125          pEnc->mbParam.quant = 4;          pEnc->mbParam.quant = 4;
126    
127          pEnc->bitrate = pParam->bitrate;          pEnc->bitrate = pParam->rc_bitrate;
128    
129          pEnc->iFrameNum = 0;          pEnc->iFrameNum = 0;
130          pEnc->iMaxKeyInterval = pParam->max_key_interval;          pEnc->iMaxKeyInterval = pParam->max_key_interval;
131    
132          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;  
         }  
   
         if (image_create(&pEnc->vInterHV, 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);  
                 image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);  
                 free(pEnc);  
                 return XVID_ERR_MEMORY;  
         }  
133    
134          pEnc->pMBs = malloc(sizeof(MACROBLOCK) * pEnc->mbParam.mb_width * pEnc->mbParam.mb_height);          pEnc->sCurrent.y        =       pEnc->sCurrent.u        =       pEnc->sCurrent.v        = NULL;
135          if (pEnc->pMBs == NULL)          pEnc->sReference.y      =       pEnc->sReference.u      =       pEnc->sReference.v      = NULL;
136            pEnc->vInterH.y         =       pEnc->vInterH.u         =       pEnc->vInterH.v         = NULL;
137            pEnc->vInterV.y         =       pEnc->vInterV.u         =       pEnc->vInterV.v         = NULL;
138            pEnc->vInterVf.y        =       pEnc->vInterVf.u        =       pEnc->vInterVf.v        = NULL;
139            pEnc->vInterHV.y        =       pEnc->vInterHV.u        =       pEnc->vInterHV.v        = NULL;
140            pEnc->vInterHVf.y       =       pEnc->vInterHVf.u       =       pEnc->vInterHVf.v       = NULL;
141    
142            pEnc->pMBs = NULL;
143    
144            if (image_create(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
145                    image_create(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
146                    image_create(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
147                    image_create(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
148                    image_create(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
149                    image_create(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
150                    image_create(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
151    #ifdef _DEBUG
152                    image_create(&pEnc->sOriginal, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||
153    #endif
154                    (pEnc->pMBs = xvid_malloc(sizeof(MACROBLOCK) * pEnc->mbParam.mb_width * pEnc->mbParam.mb_height, CACHE_LINE)) == NULL)
155          {          {
156                  image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
157                  image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
158                  image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
159                  image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
160                    image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
161                  image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
162                  free(pEnc);                  image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
163    #ifdef _DEBUG
164                    image_destroy(&pEnc->sOriginal, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
165    #endif
166                    if (pEnc)
167                    {
168                            xvid_free(pEnc);
169                    }
170                  return XVID_ERR_MEMORY;                  return XVID_ERR_MEMORY;
171          }          }
172    
# Line 180  Line 178 
178    
179          pParam->handle = (void *)pEnc;          pParam->handle = (void *)pEnc;
180    
181          if (pParam->bitrate)          if (pParam->rc_bitrate)
182          {          {
183                  RateControlInit(pParam->bitrate, pParam->rc_buffersize, pParam->fbase, pParam->width,                  RateControlInit(pParam->rc_bitrate, pParam->rc_reaction_delay_factor,
184                                  pParam->height, pParam->max_quantizer, pParam->min_quantizer);                          pParam->rc_averaging_period, pParam->rc_buffer, pParam->fbase * 1000 / pParam->fincr,
185                            pParam->max_quantizer, pParam->min_quantizer);
186          }          }
187    
188          create_vlc_tables();          init_timer();
189    
190          return XVID_ERR_OK;          return XVID_ERR_OK;
191  }  }
# Line 198  Line 197 
197          ENC_CHECK(pEnc->sCurrent.y);          ENC_CHECK(pEnc->sCurrent.y);
198          ENC_CHECK(pEnc->sReference.y);          ENC_CHECK(pEnc->sReference.y);
199    
200          free(pEnc->pMBs);          xvid_free(pEnc->pMBs);
201          image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
202          image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->sReference, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
203          image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->vInterH, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
204          image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
205            image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
206          image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
207          free(pEnc);          image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
208    #ifdef _DEBUG
209          destroy_vlc_tables();                  image_destroy(&pEnc->sOriginal, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
210    #endif
211            xvid_free(pEnc);
212    
213          return XVID_ERR_OK;          return XVID_ERR_OK;
214  }  }
# Line 217  Line 219 
219          Bitstream bs;          Bitstream bs;
220          uint32_t bits;          uint32_t bits;
221          uint16_t write_vol_header = 0;          uint16_t write_vol_header = 0;
222    #ifdef _DEBUG
223            float psnr;
224            uint8_t temp[100];
225    #endif
226    
227          start_global_timer();          start_global_timer();
228    
# Line 227  Line 233 
233    
234          pEnc->mbParam.global_flags = pFrame->general;          pEnc->mbParam.global_flags = pFrame->general;
235          pEnc->mbParam.motion_flags = pFrame->motion;          pEnc->mbParam.motion_flags = pFrame->motion;
236            pEnc->mbParam.hint = &pFrame->hint;
237    
238          start_timer();          start_timer();
239          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 236  Line 243 
243          }          }
244          stop_conv_timer();          stop_conv_timer();
245    
246            EMMS();
247    
248    #ifdef _DEBUG
249            image_copy(&pEnc->sOriginal, &pEnc->sCurrent, pEnc->mbParam.edged_width, pEnc->mbParam.height);
250    #endif
251    
252          BitstreamInit(&bs, pFrame->bitstream, 0);          BitstreamInit(&bs, pFrame->bitstream, 0);
253    
254          if (pFrame->quant == 0)          if (pFrame->quant == 0)
# Line 249  Line 262 
262    
263          if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0)          if ((pEnc->mbParam.global_flags & XVID_LUMIMASKING) > 0)
264          {          {
265                  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);
266    
267                  pEnc->mbParam.quant = adaptive_quantization(pEnc->sCurrent.y, pEnc->mbParam.width,                  pEnc->mbParam.quant = adaptive_quantization(pEnc->sCurrent.y,
268                                                              temp_dquants, pFrame->quant, pFrame->quant,                                                              pEnc->mbParam.width,
269                                                              2*pFrame->quant, pEnc->mbParam.mb_width, pEnc->mbParam.mb_height);                                                              temp_dquants,
270                                                                pEnc->mbParam.quant,
271                                                                pEnc->mbParam.quant,
272                                                                2*pEnc->mbParam.quant,
273                                                                pEnc->mbParam.mb_width,
274                                                                pEnc->mbParam.mb_height);
275    
276                  for (y = 0; y < pEnc->mbParam.mb_height; y++)                  for (y = 0; y < pEnc->mbParam.mb_height; y++)
277                          for (x = 0; x < pEnc->mbParam.mb_width; x++)                          for (x = 0; x < pEnc->mbParam.mb_width; x++)
# Line 261  Line 279 
279                                  MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];                                  MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
280                                  pMB->dquant = iDQtab[(temp_dquants[y * pEnc->mbParam.mb_width + x] + 2)];                                  pMB->dquant = iDQtab[(temp_dquants[y * pEnc->mbParam.mb_width + x] + 2)];
281                          }                          }
282                  free(temp_dquants);                  xvid_free(temp_dquants);
283          }          }
284    
285          if(pEnc->mbParam.global_flags & XVID_H263QUANT) {          if(pEnc->mbParam.global_flags & XVID_H263QUANT) {
# Line 272  Line 290 
290          else if(pEnc->mbParam.global_flags & XVID_MPEGQUANT) {          else if(pEnc->mbParam.global_flags & XVID_MPEGQUANT) {
291                  int ret1, ret2;                  int ret1, ret2;
292    
293                    ret1 = ret2 = 0;
294    
295                  if(pEnc->mbParam.quant_type != MPEG4_QUANT)                  if(pEnc->mbParam.quant_type != MPEG4_QUANT)
296                          write_vol_header = 1;                          write_vol_header = 1;
297    
# Line 322  Line 342 
342                  pResult->ublks = pEnc->sStat.ublks;                  pResult->ublks = pEnc->sStat.ublks;
343          }          }
344    
345            EMMS();
346    
347          if (pFrame->quant == 0)          if (pFrame->quant == 0)
348          {          {
349                  RateControlUpdate(pEnc->mbParam.quant, pFrame->length, pFrame->intra);                  RateControlUpdate(pEnc->mbParam.quant, pFrame->length, pFrame->intra);
350          }          }
351    
352    #ifdef _DEBUG
353            psnr = image_psnr(&pEnc->sOriginal, &pEnc->sCurrent, pEnc->mbParam.edged_width,
354                                    pEnc->mbParam.width, pEnc->mbParam.height);
355    
356            sprintf(temp, "PSNR: %f\n", psnr);
357            DEBUG(temp);
358    #endif
359    
360          pEnc->iFrameNum++;          pEnc->iFrameNum++;
361          image_swap(&pEnc->sCurrent, &pEnc->sReference);          image_swap(&pEnc->sCurrent, &pEnc->sReference);
362    
# Line 356  Line 386 
386  }  }
387    
388    
389    #define FCODEBITS       3
390    #define MODEBITS        5
391    
392    void HintedMESet(Encoder * pEnc, int * intra)
393    {
394            HINTINFO * hint;
395            Bitstream bs;
396            int length, high;
397            uint32_t x, y;
398    
399            hint = pEnc->mbParam.hint;
400    
401            if (hint->rawhints)
402            {
403                    *intra = hint->mvhint.intra;
404            }
405            else
406            {
407                    BitstreamInit(&bs, hint->hintstream, hint->hintlength);
408                    *intra = BitstreamGetBit(&bs);
409            }
410    
411            if (*intra)
412            {
413                    return;
414            }
415    
416            pEnc->mbParam.fixed_code = (hint->rawhints) ? hint->mvhint.fcode : BitstreamGetBits(&bs, FCODEBITS);
417    
418            length  = pEnc->mbParam.fixed_code + 5;
419            high    = 1 << (length - 1);
420    
421            for (y=0 ; y<pEnc->mbParam.mb_height ; ++y)
422            {
423                    for (x=0 ; x<pEnc->mbParam.mb_width ; ++x)
424                    {
425                            MACROBLOCK * pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
426                            MVBLOCKHINT * bhint = &hint->mvhint.block[x + y * pEnc->mbParam.mb_width];
427                            VECTOR pred[4];
428                            VECTOR tmp;
429                            int dummy[4];
430                            int vec;
431    
432                            pMB->mode = (hint->rawhints) ? bhint->mode : BitstreamGetBits(&bs, MODEBITS);
433    
434                            pMB->mode = (pMB->mode == MODE_INTER_Q) ? MODE_INTER : pMB->mode;
435                            pMB->mode = (pMB->mode == MODE_INTRA_Q) ? MODE_INTRA : pMB->mode;
436    
437                            if (pMB->mode == MODE_INTER)
438                            {
439                                    tmp.x  = (hint->rawhints) ? bhint->mvs[0].x : BitstreamGetBits(&bs, length);
440                                    tmp.y  = (hint->rawhints) ? bhint->mvs[0].y : BitstreamGetBits(&bs, length);
441                                    tmp.x -= (tmp.x >= high) ? high*2 : 0;
442                                    tmp.y -= (tmp.y >= high) ? high*2 : 0;
443    
444                                    get_pmvdata(pEnc->pMBs, x, y, pEnc->mbParam.mb_width, 0, pred, dummy);
445    
446                                    for (vec=0 ; vec<4 ; ++vec)
447                                    {
448                                            pMB->mvs[vec].x  = tmp.x;
449                                            pMB->mvs[vec].y  = tmp.y;
450                                            pMB->pmvs[vec].x = pMB->mvs[0].x - pred[0].x;
451                                            pMB->pmvs[vec].y = pMB->mvs[0].y - pred[0].y;
452                                    }
453                            }
454                            else if (pMB->mode == MODE_INTER4V)
455                            {
456                                    for (vec=0 ; vec<4 ; ++vec)
457                                    {
458                                            tmp.x  = (hint->rawhints) ? bhint->mvs[vec].x : BitstreamGetBits(&bs, length);
459                                            tmp.y  = (hint->rawhints) ? bhint->mvs[vec].y : BitstreamGetBits(&bs, length);
460                                            tmp.x -= (tmp.x >= high) ? high*2 : 0;
461                                            tmp.y -= (tmp.y >= high) ? high*2 : 0;
462    
463                                            get_pmvdata(pEnc->pMBs, x, y, pEnc->mbParam.mb_width, vec, pred, dummy);
464    
465                                            pMB->mvs[vec].x  = tmp.x;
466                                            pMB->mvs[vec].y  = tmp.y;
467                                            pMB->pmvs[vec].x = pMB->mvs[vec].x - pred[0].x;
468                                            pMB->pmvs[vec].y = pMB->mvs[vec].y - pred[0].y;
469                                    }
470                            }
471                            else    // intra / stuffing / not_coded
472                            {
473                                    for (vec=0 ; vec<4 ; ++vec)
474                                    {
475                                            pMB->mvs[vec].x  = pMB->mvs[vec].y  = 0;
476                                    }
477                            }
478    
479                            if (pMB->dquant != NO_CHANGE && pMB->mode == MODE_INTER4V)
480                            {
481                                    pMB->mode = MODE_INTRA;
482    
483                                    for (vec=0 ; vec<4 ; ++vec)
484                                    {
485                                            pMB->mvs[vec].x = pMB->mvs[vec].y = 0;
486                                    }
487                            }
488                    }
489            }
490    }
491    
492    
493    void HintedMEGet(Encoder * pEnc, int intra)
494    {
495            HINTINFO * hint;
496            Bitstream bs;
497            uint32_t x, y;
498            int length, high;
499    
500            hint = pEnc->mbParam.hint;
501    
502            if (hint->rawhints)
503            {
504                    hint->mvhint.intra = intra;
505            }
506            else
507            {
508                    BitstreamInit(&bs, hint->hintstream, 0);
509                    BitstreamPutBit(&bs, intra);
510            }
511    
512            if (intra)
513            {
514                    if (!hint->rawhints)
515                    {
516                            BitstreamPad(&bs);
517                            hint->hintlength = BitstreamLength(&bs);
518                    }
519                    return;
520            }
521    
522            length  = pEnc->mbParam.fixed_code + 5;
523            high    = 1 << (length - 1);
524    
525            if (hint->rawhints)
526            {
527                    hint->mvhint.fcode = pEnc->mbParam.fixed_code;
528            }
529            else
530            {
531                    BitstreamPutBits(&bs, pEnc->mbParam.fixed_code, FCODEBITS);
532            }
533    
534            for (y=0 ; y<pEnc->mbParam.mb_height ; ++y)
535            {
536                    for (x=0 ; x<pEnc->mbParam.mb_width ; ++x)
537                    {
538                            MACROBLOCK * pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
539                            MVBLOCKHINT * bhint = &hint->mvhint.block[x + y * pEnc->mbParam.mb_width];
540                            VECTOR tmp;
541    
542                            if (hint->rawhints)
543                            {
544                                    bhint->mode = pMB->mode;
545                            }
546                            else
547                            {
548                                    BitstreamPutBits(&bs, pMB->mode, MODEBITS);
549                            }
550    
551                            if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)
552                            {
553                                    tmp.x  = pMB->mvs[0].x;
554                                    tmp.y  = pMB->mvs[0].y;
555                                    tmp.x += (tmp.x < 0) ? high*2 : 0;
556                                    tmp.y += (tmp.y < 0) ? high*2 : 0;
557    
558                                    if (hint->rawhints)
559                                    {
560                                            bhint->mvs[0].x = tmp.x;
561                                            bhint->mvs[0].y = tmp.y;
562                                    }
563                                    else
564                                    {
565                                            BitstreamPutBits(&bs, tmp.x, length);
566                                            BitstreamPutBits(&bs, tmp.y, length);
567                                    }
568                            }
569                            else if (pMB->mode == MODE_INTER4V)
570                            {
571                                    int vec;
572    
573                                    for (vec=0 ; vec<4 ; ++vec)
574                                    {
575                                            tmp.x  = pMB->mvs[vec].x;
576                                            tmp.y  = pMB->mvs[vec].y;
577                                            tmp.x += (tmp.x < 0) ? high*2 : 0;
578                                            tmp.y += (tmp.y < 0) ? high*2 : 0;
579    
580                                            if (hint->rawhints)
581                                            {
582                                                    bhint->mvs[vec].x = tmp.x;
583                                                    bhint->mvs[vec].y = tmp.y;
584                                            }
585                                            else
586                                            {
587                                                    BitstreamPutBits(&bs, tmp.x, length);
588                                                    BitstreamPutBits(&bs, tmp.y, length);
589                                            }
590                                    }
591                            }
592                    }
593            }
594    
595            if (!hint->rawhints)
596            {
597                    BitstreamPad(&bs);
598                    hint->hintlength = BitstreamLength(&bs);
599            }
600    }
601    
602    
603  static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)  static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)
604  {  {
605          int16_t dct_codes[6][64];  
606          int16_t qcoeff[6][64];          DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
607            DECLARE_ALIGNED_MATRIX(qcoeff,    6, 64, int16_t, CACHE_LINE);
608    
609          uint16_t x, y;          uint16_t x, y;
610    
611          pEnc->iFrameNum = 0;          pEnc->iFrameNum = 0;
612          pEnc->mbParam.rounding_type = 1;          pEnc->mbParam.rounding_type = 1;
613          pEnc->mbParam.coding_type = I_VOP;          pEnc->mbParam.coding_type = I_VOP;
614    
615          BitstreamWriteVolHeader(bs, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.quant_type);          BitstreamWriteVolHeader(bs, &pEnc->mbParam);
616          BitstreamWriteVopHeader(bs, I_VOP, pEnc->mbParam.rounding_type,          BitstreamWriteVopHeader(bs, &pEnc->mbParam);
                                 pEnc->mbParam.quant,  
                                 pEnc->mbParam.fixed_code);  
617    
618          *pBits = BitstreamPos(bs);          *pBits = BitstreamPos(bs);
619    
# Line 384  Line 628 
628    
629                          CodeIntraMB(pEnc, pMB);                          CodeIntraMB(pEnc, pMB);
630    
631                          MBTransQuantIntra(&pEnc->mbParam, x, y, dct_codes, qcoeff, &pEnc->sCurrent);                          MBTransQuantIntra(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, &pEnc->sCurrent);
632    
633                          start_timer();                          start_timer();
634                          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 403  Line 647 
647          pEnc->sStat.iMvCount = 0;          pEnc->sStat.iMvCount = 0;
648          pEnc->mbParam.fixed_code = 2;          pEnc->mbParam.fixed_code = 2;
649    
650            if (pEnc->mbParam.global_flags & XVID_HINTEDME_GET)
651            {
652                    HintedMEGet(pEnc, 1);
653            }
654    
655          return 1;                                        // intra          return 1;                                        // intra
656  }  }
657    
# Line 412  Line 661 
661  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)
662  {  {
663          float fSigma;          float fSigma;
664          int16_t dct_codes[6][64];  
665          int16_t qcoeff[6][64];          DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
666            DECLARE_ALIGNED_MATRIX(qcoeff,    6, 64, int16_t, CACHE_LINE);
667    
668          int iLimit;          int iLimit;
669          uint32_t x, y;          uint32_t x, y;
670          int iSearchRange;          int iSearchRange;
# Line 422  Line 673 
673          IMAGE *pCurrent = &pEnc->sCurrent;          IMAGE *pCurrent = &pEnc->sCurrent;
674          IMAGE *pRef = &pEnc->sReference;          IMAGE *pRef = &pEnc->sReference;
675    
676          image_setedges(pRef,pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, pEnc->mbParam.width, pEnc->mbParam.height);          start_timer();
677            image_setedges(pRef,
678                           pEnc->mbParam.edged_width,
679                           pEnc->mbParam.edged_height,
680                           pEnc->mbParam.width,
681                           pEnc->mbParam.height,
682                           pEnc->mbParam.global_flags & XVID_INTERLACING);
683            stop_edges_timer();
684    
685          pEnc->mbParam.rounding_type = 1 - pEnc->mbParam.rounding_type;          pEnc->mbParam.rounding_type = 1 - pEnc->mbParam.rounding_type;
686    
# Line 440  Line 698 
698          }          }
699    
700          start_timer();          start_timer();
701            if (pEnc->mbParam.global_flags & XVID_HINTEDME_SET)
702            {
703                    HintedMESet(pEnc, &bIntra);
704            }
705            else
706            {
707          bIntra = MotionEstimation(pEnc->pMBs, &pEnc->mbParam, &pEnc->sReference,          bIntra = MotionEstimation(pEnc->pMBs, &pEnc->mbParam, &pEnc->sReference,
708                                    &pEnc->vInterH, &pEnc->vInterV,                                    &pEnc->vInterH, &pEnc->vInterV,
709                                    &pEnc->vInterHV, &pEnc->sCurrent, iLimit);                                    &pEnc->vInterHV, &pEnc->sCurrent, iLimit);
710            }
711          stop_motion_timer();          stop_motion_timer();
712    
713          if (bIntra == 1)          if (bIntra == 1)
714            {
715                  return FrameCodeI(pEnc, bs, pBits);                  return FrameCodeI(pEnc, bs, pBits);
716            }
717    
718          pEnc->mbParam.coding_type = P_VOP;          pEnc->mbParam.coding_type = P_VOP;
719    
720          if(vol_header)          if(vol_header)
721                  BitstreamWriteVolHeader(bs, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.quant_type);                  BitstreamWriteVolHeader(bs, &pEnc->mbParam);
722    
723          BitstreamWriteVopHeader(bs, P_VOP, pEnc->mbParam.rounding_type,          BitstreamWriteVopHeader(bs, &pEnc->mbParam);
                                 pEnc->mbParam.quant,  
                                 pEnc->mbParam.fixed_code);  
724    
725          *pBits = BitstreamPos(bs);          *pBits = BitstreamPos(bs);
726    
# Line 475  Line 740 
740                          if (!bIntra)                          if (!bIntra)
741                          {                          {
742                                  start_timer();                                  start_timer();
743                                  MBMotionCompensation(pMB, x, y, &pEnc->sReference,                                  MBMotionCompensation(pMB,
744                                                       &pEnc->vInterH, &pEnc->vInterV,                                                       x, y,
745                                                       &pEnc->vInterHV, &pEnc->sCurrent, dct_codes,                                                       &pEnc->sReference,
746                                                         &pEnc->vInterH,
747                                                         &pEnc->vInterV,
748                                                         &pEnc->vInterHV,
749                                                         &pEnc->sCurrent,
750                                                         dct_codes,
751                                                       pEnc->mbParam.width,                                                       pEnc->mbParam.width,
752                                                       pEnc->mbParam.height,                                                       pEnc->mbParam.height,
753                                                       pEnc->mbParam.edged_width,                                                       pEnc->mbParam.edged_width,
# Line 494  Line 764 
764                                  }                                  }
765                                  pMB->quant = pEnc->mbParam.quant;                                  pMB->quant = pEnc->mbParam.quant;
766    
767                                  pMB->cbp = MBTransQuantInter(&pEnc->mbParam, x, y, dct_codes, qcoeff, pCurrent);                                  pMB->field_pred = 0;
768    
769                                    pMB->cbp = MBTransQuantInter(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, pCurrent);
770                          }                          }
771                          else                          else
772                          {                          {
773                                  CodeIntraMB(pEnc, pMB);                                  CodeIntraMB(pEnc, pMB);
774                                  MBTransQuantIntra(&pEnc->mbParam, x, y, dct_codes, qcoeff, pCurrent);                                  MBTransQuantIntra(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, pCurrent);
775                          }                          }
776    
777                          start_timer();                          start_timer();
# Line 531  Line 803 
803    
804          emms();          emms();
805    
806            if (pEnc->mbParam.global_flags & XVID_HINTEDME_GET)
807            {
808                    HintedMEGet(pEnc, 0);
809            }
810    
811          if (pEnc->sStat.iMvCount == 0)          if (pEnc->sStat.iMvCount == 0)
812                  pEnc->sStat.iMvCount = 1;                  pEnc->sStat.iMvCount = 1;
813    
# Line 559  Line 836 
836    
837          return 0;                                        // inter          return 0;                                        // inter
838  }  }
839    
840    
841    
842    /*
843    static void FrameCodeB(Encoder * pEnc, FRAMEINFO * frame, Bitstream * bs, uint32_t *pBits)
844    {
845        int16_t dct_codes[6][64];
846        int16_t qcoeff[6][64];
847        uint32_t x, y;
848            VECTOR forward;
849            VECTOR backward;
850    
851        IMAGE *f_ref = &pEnc->reference->image;
852            IMAGE *b_ref = &pEnc->current->image;
853    
854            // forward
855            image_setedges(f_ref, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, pEnc->mbParam.width, pEnc->mbParam.height);
856            start_timer();
857            image_interpolate(f_ref, &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
858                    pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, 0);
859            stop_inter_timer();
860    
861            // backward
862            image_setedges(b_ref, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, pEnc->mbParam.width, pEnc->mbParam.height);
863        start_timer();
864            image_interpolate(b_ref, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
865                    pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, 0);
866            stop_inter_timer();
867    
868            start_timer();
869            MotionEstimationBVOP(&pEnc->mbParam, frame,
870                    pEnc->reference->mbs, f_ref, &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
871                    pEnc->current->mbs, b_ref, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV);
872    
873            stop_motion_timer();
874    
875            if (test_quant_type(&pEnc->mbParam, pEnc->current))
876            {
877                    BitstreamWriteVolHeader(bs, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.quant_type);
878            }
879    
880        frame->coding_type = B_VOP;
881        BitstreamWriteVopHeader(bs, B_VOP, frame->tick, 0,
882                            frame->quant, frame->fcode, frame->bcode);
883    
884        *pBits = BitstreamPos(bs);
885    
886        pEnc->sStat.iTextBits = 0;
887        pEnc->sStat.iMvSum = 0;
888        pEnc->sStat.iMvCount = 0;
889            pEnc->sStat.kblks = pEnc->sStat.mblks = pEnc->sStat.ublks = 0;
890    
891    
892        for (y = 0; y < pEnc->mbParam.mb_height; y++)
893            {
894                    // reset prediction
895    
896                    forward.x = 0;
897                    forward.y = 0;
898                    backward.x = 0;
899                    backward.y = 0;
900    
901                    for (x = 0; x < pEnc->mbParam.mb_width; x++)
902                    {
903                            MACROBLOCK * f_mb = &pEnc->reference->mbs[x + y * pEnc->mbParam.mb_width];
904                            MACROBLOCK * b_mb = &pEnc->current->mbs[x + y * pEnc->mbParam.mb_width];
905                            MACROBLOCK * mb = &frame->mbs[x + y * pEnc->mbParam.mb_width];
906    
907                            // decoder ignores mb when refence block is INTER(0,0), CBP=0
908                            if (mb->mode == MODE_NOT_CODED)
909                            {
910                                    mb->mvs[0].x = 0;
911                                    mb->mvs[0].y = 0;
912                                    continue;
913                            }
914    
915                            MBMotionCompensationBVOP(&pEnc->mbParam, mb, x, y, &frame->image,
916                                            f_ref, &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
917                                            b_ref, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
918                                            dct_codes);
919    
920                            mb->quant = frame->quant;
921                            mb->cbp = MBTransQuantInter(&pEnc->mbParam, frame, x, y, dct_codes, qcoeff);
922                            //mb->cbp = MBTransQuantBVOP(&pEnc->mbParam, x, y, dct_codes, qcoeff, &frame->image, frame->quant);
923    
924    
925                            if ((mb->mode == MODE_INTERPOLATE || mb->mode == MODE_DIRECT) &&
926                                    mb->cbp == 0 &&
927                                    mb->mvs[0].x == 0 &&
928                                    mb->mvs[0].y == 0)
929                            {
930                                    mb->mode = 5;  // skipped
931                            }
932    
933                            if (mb->mode == MODE_INTERPOLATE || mb->mode == MODE_FORWARD)
934                            {
935                                    mb->pmvs[0].x = mb->mvs[0].x - forward.x;
936                                    mb->pmvs[0].y = mb->mvs[0].y - forward.y;
937                                    forward.x = mb->mvs[0].x;
938                                    forward.y = mb->mvs[0].y;
939                            }
940    
941                            if (mb->mode == MODE_INTERPOLATE || mb->mode == MODE_BACKWARD)
942                            {
943                                    mb->b_pmvs[0].x = mb->b_mvs[0].x - backward.x;
944                                    mb->b_pmvs[0].y = mb->b_mvs[0].y - backward.y;
945                                    backward.x = mb->b_mvs[0].x;
946                                    backward.y = mb->b_mvs[0].y;
947                            }
948    
949    //                      printf("[%i %i] M=%i CBP=%i MVX=%i MVY=%i %i,%i  %i,%i\n", x, y, pMB->mode, pMB->cbp, pMB->mvs[0].x, bmb->pmvs[0].x, bmb->pmvs[0].y, forward.x, forward.y);
950    
951                            start_timer();
952                            MBCodingBVOP(frame, mb, qcoeff, bs, &pEnc->sStat);
953                            stop_coding_timer();
954                    }
955            }
956    
957            emms();
958    
959            // TODO: dynamic fcode/bcode ???
960    
961            *pBits = BitstreamPos(bs) - *pBits;
962    
963    }
964    
965    */

Legend:
Removed from v.1.5  
changed lines
  Added in v.1.26

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