[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 101, Fri Apr 5 14:40:36 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_averaging_period <= 0)
90                    pParam->rc_averaging_period = 100;
91    
92          if (pParam->rc_buffersize <= 0)          if (pParam->rc_buffer <= 0)
93                  pParam->rc_buffersize = 16;                  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    
# Line 190  Line 205 
205          image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height);          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);          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    
213          return XVID_ERR_OK;          return XVID_ERR_OK;
# Line 201  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 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    
 static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)  
 {  
   
         DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);  
         DECLARE_ALIGNED_MATRIX(qcoeff,    6, 64, int16_t, CACHE_LINE);  
   
         uint16_t x, y;  
   
         pEnc->iFrameNum = 0;  
         pEnc->mbParam.rounding_type = 1;  
         pEnc->mbParam.coding_type = I_VOP;  
   
         BitstreamWriteVolHeader(bs, &pEnc->mbParam);  
         BitstreamWriteVopHeader(bs, &pEnc->mbParam);  
   
         *pBits = BitstreamPos(bs);  
   
         pEnc->sStat.iTextBits = 0;  
         pEnc->sStat.kblks = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height;  
         pEnc->sStat.mblks = pEnc->sStat.ublks = 0;  
   
         for (y = 0; y < pEnc->mbParam.mb_height; y++)  
                 for (x = 0; x < pEnc->mbParam.mb_width; x++)  
                 {  
                         MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];  
   
                         CodeIntraMB(pEnc, pMB);  
   
                         MBTransQuantIntra(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, &pEnc->sCurrent);  
   
                         start_timer();  
                         MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs);  
                         stop_prediction_timer();  
   
                         start_timer();  
                         MBCoding(&pEnc->mbParam, pMB, qcoeff, bs, &pEnc->sStat);  
                         stop_coding_timer();  
                 }  
   
         emms();  
   
         *pBits = BitstreamPos(bs) - *pBits;  
         pEnc->sStat.fMvPrevSigma = -1;  
         pEnc->sStat.iMvSum = 0;  
         pEnc->sStat.iMvCount = 0;  
         pEnc->mbParam.fixed_code = 2;  
   
         return 1;                                        // intra  
 }  
   
   
389  #define FCODEBITS       3  #define FCODEBITS       3
390  #define MODEBITS        5  #define MODEBITS        5
391    
# Line 604  Line 587 
587  }  }
588    
589    
590    static int FrameCodeI(Encoder * pEnc, Bitstream * bs, uint32_t *pBits)
591    {
592    
593            DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
594            DECLARE_ALIGNED_MATRIX(qcoeff,    6, 64, int16_t, CACHE_LINE);
595    
596            uint16_t x, y;
597    
598            pEnc->iFrameNum = 0;
599            pEnc->mbParam.rounding_type = 1;
600            pEnc->mbParam.coding_type = I_VOP;
601    
602            BitstreamWriteVolHeader(bs, &pEnc->mbParam);
603            BitstreamWriteVopHeader(bs, &pEnc->mbParam);
604    
605            *pBits = BitstreamPos(bs);
606    
607            pEnc->sStat.iTextBits = 0;
608            pEnc->sStat.kblks = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height;
609            pEnc->sStat.mblks = pEnc->sStat.ublks = 0;
610    
611            for (y = 0; y < pEnc->mbParam.mb_height; y++)
612                    for (x = 0; x < pEnc->mbParam.mb_width; x++)
613                    {
614                            MACROBLOCK *pMB = &pEnc->pMBs[x + y * pEnc->mbParam.mb_width];
615    
616                            CodeIntraMB(pEnc, pMB);
617    
618                            MBTransQuantIntra(&pEnc->mbParam, pMB, x, y, dct_codes, qcoeff, &pEnc->sCurrent);
619    
620                            start_timer();
621                            MBPrediction(&pEnc->mbParam, x, y, pEnc->mbParam.mb_width, qcoeff, pEnc->pMBs);
622                            stop_prediction_timer();
623    
624                            start_timer();
625                            MBCoding(&pEnc->mbParam, pMB, qcoeff, bs, &pEnc->sStat);
626                            stop_coding_timer();
627                    }
628    
629            emms();
630    
631            *pBits = BitstreamPos(bs) - *pBits;
632            pEnc->sStat.fMvPrevSigma = -1;
633            pEnc->sStat.iMvSum = 0;
634            pEnc->sStat.iMvCount = 0;
635            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
643    }
644    
645    
646  #define INTRA_THRESHOLD 0.5  #define INTRA_THRESHOLD 0.5
647    
648  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)
# Line 660  Line 699 
699    
700          if (bIntra == 1)          if (bIntra == 1)
701          {          {
                 if (pEnc->mbParam.global_flags & XVID_HINTEDME_GET)  
                 {  
                         HintedMEGet(pEnc, 1);  
                 }  
702                  return FrameCodeI(pEnc, bs, pBits);                  return FrameCodeI(pEnc, bs, pBits);
703          }          }
704    
# Line 788  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.101  
changed lines
  Added in v.121

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