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

Diff of /trunk/xvidcore/src/encoder.c

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

revision 84, Fri Mar 29 07:03:24 2002 UTC revision 121, Mon Apr 15 08:03:50 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 78  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 = 16;                  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 116  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;
# Line 140  Line 148 
148                  image_create(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height) < 0 ||                  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 ||                  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 ||                  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)                  (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);
# Line 149  Line 160 
160                  image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  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                  image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);                  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)                  if (pEnc)
167                  {                  {
168                          xvid_free(pEnc);                          xvid_free(pEnc);
# Line 164  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 * 100 / pParam->fincr,                  RateControlInit(pParam->rc_bitrate, pParam->rc_reaction_delay_factor,
184                            pParam->rc_averaging_period, pParam->rc_buffer, pParam->fbase * 1000 / pParam->fincr,
185                                  pParam->max_quantizer, pParam->min_quantizer);                                  pParam->max_quantizer, pParam->min_quantizer);
186          }          }
187    
         create_vlc_tables();  
188          init_timer();          init_timer();
189    
190          return XVID_ERR_OK;          return XVID_ERR_OK;
# Line 188  Line 202 
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            image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
208    #ifdef _DEBUG
209                    image_destroy(&pEnc->sOriginal, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);
210    #endif
211          xvid_free(pEnc);          xvid_free(pEnc);
212    
         destroy_vlc_tables();  
   
213          return XVID_ERR_OK;          return XVID_ERR_OK;
214  }  }
215    
# Line 202  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 212  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 223  Line 245 
245    
246          EMMS();          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 323  Line 349 
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 352  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                            if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)
435                            {
436                                    tmp.x  = (hint->rawhints) ? bhint->mvs[0].x : BitstreamGetBits(&bs, length);
437                                    tmp.y  = (hint->rawhints) ? bhint->mvs[0].y : BitstreamGetBits(&bs, length);
438                                    tmp.x -= (tmp.x >= high) ? high*2 : 0;
439                                    tmp.y -= (tmp.y >= high) ? high*2 : 0;
440    
441                                    get_pmvdata(pEnc->pMBs, x, y, pEnc->mbParam.mb_width, 0, pred, dummy);
442    
443                                    for (vec=0 ; vec<4 ; ++vec)
444                                    {
445                                            pMB->mvs[vec].x  = tmp.x;
446                                            pMB->mvs[vec].y  = tmp.y;
447                                            pMB->pmvs[vec].x = pMB->mvs[0].x - pred[0].x;
448                                            pMB->pmvs[vec].y = pMB->mvs[0].y - pred[0].y;
449                                    }
450                            }
451                            else if (pMB->mode == MODE_INTER4V)
452                            {
453                                    for (vec=0 ; vec<4 ; ++vec)
454                                    {
455                                            tmp.x  = (hint->rawhints) ? bhint->mvs[vec].x : BitstreamGetBits(&bs, length);
456                                            tmp.y  = (hint->rawhints) ? bhint->mvs[vec].y : BitstreamGetBits(&bs, length);
457                                            tmp.x -= (tmp.x >= high) ? high*2 : 0;
458                                            tmp.y -= (tmp.y >= high) ? high*2 : 0;
459    
460                                            get_pmvdata(pEnc->pMBs, x, y, pEnc->mbParam.mb_width, vec, pred, dummy);
461    
462                                            pMB->mvs[vec].x  = tmp.x;
463                                            pMB->mvs[vec].y  = tmp.y;
464                                            pMB->pmvs[vec].x = pMB->mvs[vec].x - pred[0].x;
465                                            pMB->pmvs[vec].y = pMB->mvs[vec].y - pred[0].y;
466                                    }
467                            }
468                            else    // intra / intra_q / stuffing / not_coded
469                            {
470                                    for (vec=0 ; vec<4 ; ++vec)
471                                    {
472                                            pMB->mvs[vec].x  = pMB->mvs[vec].y  = 0;
473                                    }
474                            }
475                    }
476            }
477    }
478    
479    
480    void HintedMEGet(Encoder * pEnc, int intra)
481    {
482            HINTINFO * hint;
483            Bitstream bs;
484            uint32_t x, y;
485            int length, high;
486    
487            hint = pEnc->mbParam.hint;
488    
489            if (hint->rawhints)
490            {
491                    hint->mvhint.intra = intra;
492            }
493            else
494            {
495                    BitstreamInit(&bs, hint->hintstream, 0);
496                    BitstreamPutBit(&bs, intra);
497            }
498    
499            if (intra)
500            {
501                    if (!hint->rawhints)
502                    {
503                            BitstreamPad(&bs);
504                            hint->hintlength = BitstreamLength(&bs);
505                    }
506                    return;
507            }
508    
509            length  = pEnc->mbParam.fixed_code + 5;
510            high    = 1 << (length - 1);
511    
512            if (hint->rawhints)
513            {
514                    hint->mvhint.fcode = pEnc->mbParam.fixed_code;
515            }
516            else
517            {
518                    BitstreamPutBits(&bs, pEnc->mbParam.fixed_code, FCODEBITS);
519            }
520    
521            for (y=0 ; y<pEnc->mbParam.mb_height ; ++y)
522            {
523                    for (x=0 ; x<pEnc->mbParam.mb_width ; ++x)
524                    {
525                            MACROBLOCK * pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
526                            MVBLOCKHINT * bhint = &hint->mvhint.block[x + y * pEnc->mbParam.mb_width];
527                            VECTOR tmp;
528    
529                            if (hint->rawhints)
530                            {
531                                    bhint->mode = pMB->mode;
532                            }
533                            else
534                            {
535                                    BitstreamPutBits(&bs, pMB->mode, MODEBITS);
536                            }
537    
538                            if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)
539                            {
540                                    tmp.x  = pMB->mvs[0].x;
541                                    tmp.y  = pMB->mvs[0].y;
542                                    tmp.x += (tmp.x < 0) ? high*2 : 0;
543                                    tmp.y += (tmp.y < 0) ? high*2 : 0;
544    
545                                    if (hint->rawhints)
546                                    {
547                                            bhint->mvs[0].x = tmp.x;
548                                            bhint->mvs[0].y = tmp.y;
549                                    }
550                                    else
551                                    {
552                                            BitstreamPutBits(&bs, tmp.x, length);
553                                            BitstreamPutBits(&bs, tmp.y, length);
554                                    }
555                            }
556                            else if (pMB->mode == MODE_INTER4V)
557                            {
558                                    int vec;
559    
560                                    for (vec=0 ; vec<4 ; ++vec)
561                                    {
562                                            tmp.x  = pMB->mvs[vec].x;
563                                            tmp.y  = pMB->mvs[vec].y;
564                                            tmp.x += (tmp.x < 0) ? high*2 : 0;
565                                            tmp.y += (tmp.y < 0) ? high*2 : 0;
566    
567                                            if (hint->rawhints)
568                                            {
569                                                    bhint->mvs[vec].x = tmp.x;
570                                                    bhint->mvs[vec].y = tmp.y;
571                                            }
572                                            else
573                                            {
574                                                    BitstreamPutBits(&bs, tmp.x, length);
575                                                    BitstreamPutBits(&bs, tmp.y, length);
576                                            }
577                                    }
578                            }
579                    }
580            }
581    
582            if (!hint->rawhints)
583            {
584                    BitstreamPad(&bs);
585                    hint->hintlength = BitstreamLength(&bs);
586            }
587    }
588    
589    
590  static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)  static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)
591  {  {
592    
# Line 399  Line 634 
634          pEnc->sStat.iMvCount = 0;          pEnc->sStat.iMvCount = 0;
635          pEnc->mbParam.fixed_code = 2;          pEnc->mbParam.fixed_code = 2;
636    
637            if (pEnc->mbParam.global_flags & XVID_HINTEDME_GET)
638            {
639                    HintedMEGet(pEnc, 1);
640            }
641    
642          return 1;                                        // intra          return 1;                                        // intra
643  }  }
644    
# Line 436  Line 676 
676          else          else
677                  iLimit = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height + 1;                  iLimit = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height + 1;
678    
679          if ((pEnc->mbParam.global_flags & XVID_HALFPEL) > 0)          if ((pEnc->mbParam.global_flags & XVID_HALFPEL) > 0) {
         {  
                 IMAGE *vInterV = NULL;  
                 IMAGE *vInterVf = NULL;  
                 IMAGE *vInterHV = NULL;  
                 IMAGE *vInterHVf = NULL;  
   
                 // interpolate fields together if field ME is used  
                 if (pEnc->mbParam.global_flags & XVID_INTERLACING &&  
                         pEnc->mbParam.global_flags & XVID_FIELDME)  
                 {  
                         vInterVf = &pEnc->vInterVf;  
                         vInterHVf = &pEnc->vInterHVf;  
                 }  
   
                 // perform normal interpolation, unless only field-based ME is allowed  
                 if (!(pEnc->mbParam.global_flags & XVID_INTERLACING) ||  
                         !(pEnc->mbParam.global_flags & XVID_FIELDMEONLY))  
                 {  
                         vInterV = &pEnc->vInterV;  
                         vInterHV = &pEnc->vInterHV;  
                 }  
   
680                  start_timer();                  start_timer();
681                  image_interpolate(pRef,                  image_interpolate(pRef, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
682                                  &pEnc->vInterH,                                    pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
                                 vInterV, vInterVf,  
                                 vInterHV, vInterHVf,  
                                 pEnc->mbParam.edged_width,  
                                 pEnc->mbParam.edged_height,  
683                                  pEnc->mbParam.rounding_type);                                  pEnc->mbParam.rounding_type);
684                  stop_inter_timer();                  stop_inter_timer();
685          }          }
686    
687          start_timer();          start_timer();
688          bIntra = MotionEstimation(pEnc->pMBs,          if (pEnc->mbParam.global_flags & XVID_HINTEDME_SET)
689                                  &pEnc->mbParam,          {
690                                  &pEnc->sReference,                  HintedMESet(pEnc, &bIntra);
691                                  &pEnc->vInterH,          }
692                                  &pEnc->vInterV, &pEnc->vInterVf,          else
693                                  &pEnc->vInterHV, &pEnc->vInterHVf,          {
694                                  &pEnc->sCurrent,                  bIntra = MotionEstimation(pEnc->pMBs, &pEnc->mbParam, &pEnc->sReference,
695                                  iLimit);                                            &pEnc->vInterH, &pEnc->vInterV,
696                                              &pEnc->vInterHV, &pEnc->sCurrent, iLimit);
697            }
698          stop_motion_timer();          stop_motion_timer();
699    
700          if (bIntra == 1)          if (bIntra == 1)
701            {
702                  return FrameCodeI(pEnc, bs, pBits);                  return FrameCodeI(pEnc, bs, pBits);
703            }
704    
705          pEnc->mbParam.coding_type = P_VOP;          pEnc->mbParam.coding_type = P_VOP;
706    
# Line 513  Line 731 
731                                                       x, y,                                                       x, y,
732                                                       &pEnc->sReference,                                                       &pEnc->sReference,
733                                                       &pEnc->vInterH,                                                       &pEnc->vInterH,
734                                                       &pEnc->vInterV, &pEnc->vInterVf,                                                       &pEnc->vInterV,
735                                                       &pEnc->vInterHV, &pEnc->vInterHVf,                                                       &pEnc->vInterHV,
736                                                       &pEnc->sCurrent,                                                       &pEnc->sCurrent,
737                                                       dct_codes,                                                       dct_codes,
738                                                       pEnc->mbParam.width,                                                       pEnc->mbParam.width,
# Line 572  Line 790 
790    
791          emms();          emms();
792    
793            if (pEnc->mbParam.global_flags & XVID_HINTEDME_GET)
794            {
795                    HintedMEGet(pEnc, 0);
796            }
797    
798          if (pEnc->sStat.iMvCount == 0)          if (pEnc->sStat.iMvCount == 0)
799                  pEnc->sStat.iMvCount = 1;                  pEnc->sStat.iMvCount = 1;
800    
# Line 600  Line 823 
823    
824          return 0;                                        // inter          return 0;                                        // inter
825  }  }
826    
827    
828    
829    /*
830    static void FrameCodeB(Encoder * pEnc, FRAMEINFO * frame, Bitstream * bs, uint32_t *pBits)
831    {
832        int16_t dct_codes[6][64];
833        int16_t qcoeff[6][64];
834        uint32_t x, y;
835            VECTOR forward;
836            VECTOR backward;
837    
838        IMAGE *f_ref = &pEnc->reference->image;
839            IMAGE *b_ref = &pEnc->current->image;
840    
841            // forward
842            image_setedges(f_ref, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, pEnc->mbParam.width, pEnc->mbParam.height);
843            start_timer();
844            image_interpolate(f_ref, &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
845                    pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, 0);
846            stop_inter_timer();
847    
848            // backward
849            image_setedges(b_ref, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, pEnc->mbParam.width, pEnc->mbParam.height);
850        start_timer();
851            image_interpolate(b_ref, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
852                    pEnc->mbParam.edged_width, pEnc->mbParam.edged_height, 0);
853            stop_inter_timer();
854    
855            start_timer();
856            MotionEstimationBVOP(&pEnc->mbParam, frame,
857                    pEnc->reference->mbs, f_ref, &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
858                    pEnc->current->mbs, b_ref, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV);
859    
860            stop_motion_timer();
861    
862            if (test_quant_type(&pEnc->mbParam, pEnc->current))
863            {
864                    BitstreamWriteVolHeader(bs, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.quant_type);
865            }
866    
867        frame->coding_type = B_VOP;
868        BitstreamWriteVopHeader(bs, B_VOP, frame->tick, 0,
869                            frame->quant, frame->fcode, frame->bcode);
870    
871        *pBits = BitstreamPos(bs);
872    
873        pEnc->sStat.iTextBits = 0;
874        pEnc->sStat.iMvSum = 0;
875        pEnc->sStat.iMvCount = 0;
876            pEnc->sStat.kblks = pEnc->sStat.mblks = pEnc->sStat.ublks = 0;
877    
878    
879        for (y = 0; y < pEnc->mbParam.mb_height; y++)
880            {
881                    // reset prediction
882    
883                    forward.x = 0;
884                    forward.y = 0;
885                    backward.x = 0;
886                    backward.y = 0;
887    
888                    for (x = 0; x < pEnc->mbParam.mb_width; x++)
889                    {
890                            MACROBLOCK * f_mb = &pEnc->reference->mbs[x + y * pEnc->mbParam.mb_width];
891                            MACROBLOCK * b_mb = &pEnc->current->mbs[x + y * pEnc->mbParam.mb_width];
892                            MACROBLOCK * mb = &frame->mbs[x + y * pEnc->mbParam.mb_width];
893    
894                            // decoder ignores mb when refence block is INTER(0,0), CBP=0
895                            if (mb->mode == MODE_NOT_CODED)
896                            {
897                                    mb->mvs[0].x = 0;
898                                    mb->mvs[0].y = 0;
899                                    continue;
900                            }
901    
902                            MBMotionCompensationBVOP(&pEnc->mbParam, mb, x, y, &frame->image,
903                                            f_ref, &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
904                                            b_ref, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
905                                            dct_codes);
906    
907                            mb->quant = frame->quant;
908                            mb->cbp = MBTransQuantInter(&pEnc->mbParam, frame, x, y, dct_codes, qcoeff);
909                            //mb->cbp = MBTransQuantBVOP(&pEnc->mbParam, x, y, dct_codes, qcoeff, &frame->image, frame->quant);
910    
911    
912                            if ((mb->mode == MODE_INTERPOLATE || mb->mode == MODE_DIRECT) &&
913                                    mb->cbp == 0 &&
914                                    mb->mvs[0].x == 0 &&
915                                    mb->mvs[0].y == 0)
916                            {
917                                    mb->mode = 5;  // skipped
918                            }
919    
920                            if (mb->mode == MODE_INTERPOLATE || mb->mode == MODE_FORWARD)
921                            {
922                                    mb->pmvs[0].x = mb->mvs[0].x - forward.x;
923                                    mb->pmvs[0].y = mb->mvs[0].y - forward.y;
924                                    forward.x = mb->mvs[0].x;
925                                    forward.y = mb->mvs[0].y;
926                            }
927    
928                            if (mb->mode == MODE_INTERPOLATE || mb->mode == MODE_BACKWARD)
929                            {
930                                    mb->b_pmvs[0].x = mb->b_mvs[0].x - backward.x;
931                                    mb->b_pmvs[0].y = mb->b_mvs[0].y - backward.y;
932                                    backward.x = mb->b_mvs[0].x;
933                                    backward.y = mb->b_mvs[0].y;
934                            }
935    
936    //                      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);
937    
938                            start_timer();
939                            MBCodingBVOP(frame, mb, qcoeff, bs, &pEnc->sStat);
940                            stop_coding_timer();
941                    }
942            }
943    
944            emms();
945    
946            // TODO: dynamic fcode/bcode ???
947    
948            *pBits = BitstreamPos(bs) - *pBits;
949    
950    }
951    
952    */

Legend:
Removed from v.84  
changed lines
  Added in v.121

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