[svn] / tags / branch-release-1-0 / xvidcore / src / encoder.c Repository:
ViewVC logotype

Diff of /tags/branch-release-1-0/xvidcore/src/encoder.c

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

revision 236, Sun Jun 23 19:48:06 2002 UTC revision 259, Sat Jul 6 17:04:57 2002 UTC
# Line 37  Line 37 
37   *             MinChen <chenm001@163.com>   *             MinChen <chenm001@163.com>
38   *  14.04.2002 added FrameCodeB()   *  14.04.2002 added FrameCodeB()
39   *   *
40   *  $Id: encoder.c,v 1.47 2002-06-23 19:48:06 edgomez Exp $   *  $Id: encoder.c,v 1.52 2002-07-06 17:04:57 chl Exp $
41   *   *
42   ****************************************************************************/   ****************************************************************************/
43    
# Line 51  Line 51 
51  #include "global.h"  #include "global.h"
52  #include "utils/timer.h"  #include "utils/timer.h"
53  #include "image/image.h"  #include "image/image.h"
54    #ifdef BFRAMES
55    #include "image/font.h"
56    #endif
57  #include "motion/motion.h"  #include "motion/motion.h"
58  #include "bitstream/cbp.h"  #include "bitstream/cbp.h"
59  #include "utils/mbfunctions.h"  #include "utils/mbfunctions.h"
# Line 63  Line 66 
66  #include "quant/quant_matrix.h"  #include "quant/quant_matrix.h"
67  #include "utils/mem_align.h"  #include "utils/mem_align.h"
68    
69    #ifdef _SMP
70    #include "motion/smp_motion_est.h"
71    #endif
72  /*****************************************************************************  /*****************************************************************************
73   * Local macros   * Local macros
74   ****************************************************************************/   ****************************************************************************/
# Line 375  Line 381 
381          pEnc->bframenum_head = 0;          pEnc->bframenum_head = 0;
382          pEnc->bframenum_tail = 0;          pEnc->bframenum_tail = 0;
383          pEnc->flush_bframes = 0;          pEnc->flush_bframes = 0;
384            pEnc->bframenum_dx50bvop = -1;
385    
386          pEnc->queue = NULL;          pEnc->queue = NULL;
387    
# Line 621  Line 628 
628  {  {
629          if (pEnc->queue_size >= pEnc->mbParam.max_bframes)          if (pEnc->queue_size >= pEnc->mbParam.max_bframes)
630          {          {
631                  DPRINTF("FATAL: QUEUE FULL");                  DPRINTF(DPRINTF_DEBUG,"FATAL: QUEUE FULL");
632                  return;                  return;
633          }          }
634    
635          DPRINTF("*** QUEUE bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",          DPRINTF(DPRINTF_DEBUG,"*** QUEUE bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
636                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
637                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
638    
# Line 693  Line 700 
700                           * frame as a pframe                           * frame as a pframe
701                           */                           */
702    
703                          DPRINTF("*** BFRAME (final frame) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",                          DPRINTF(DPRINTF_DEBUG,"*** BFRAME (final frame) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
704                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
705                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
706    
# Line 712  Line 719 
719                  }                  }
720    
721    
722                  DPRINTF("*** BFRAME (flush) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",                  DPRINTF(DPRINTF_DEBUG,"*** BFRAME (flush) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
723                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
724                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
725    
# Line 734  Line 741 
741    
742                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {
743    
744                          DPRINTF("*** EMPTY bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",                          DPRINTF(DPRINTF_DEBUG,"*** EMPTY bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
745                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
746                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
747    
   
748                          BitstreamWriteVopHeader(&bs, &pEnc->mbParam, pEnc->current, 0);                          BitstreamWriteVopHeader(&bs, &pEnc->mbParam, pEnc->current, 0);
749                          BitstreamPad(&bs);                          BitstreamPad(&bs);
750                          BitstreamPutBits(&bs, 0x7f, 8);                          BitstreamPutBits(&bs, 0x7f, 8);
# Line 756  Line 762 
762    
763  bvop_loop:  bvop_loop:
764    
765          if (input_valid) {          if (pEnc->bframenum_dx50bvop != -1)
766            {
767    
768                    SWAP(pEnc->current, pEnc->reference);
769                    SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_dx50bvop]);
770    
771                    if ((pEnc->global & XVID_GLOBAL_DEBUG)) {
772                            image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 IVOP");
773                    }
774    
775                    if (input_valid)
776                    {
777                            queue_image(pEnc, pFrame);
778                            input_valid = 0;
779                    }
780    
781            } else if (input_valid) {
782    
783                  SWAP(pEnc->current, pEnc->reference);                  SWAP(pEnc->current, pEnc->reference);
784    
# Line 789  Line 811 
811    
812          } else if (BitstreamPos(&bs) == 0) {          } else if (BitstreamPos(&bs) == 0) {
813    
814                  DPRINTF("*** SKIP bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",                  DPRINTF(DPRINTF_DEBUG,"*** SKIP bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
815                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
816                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
817    
   
818                  pFrame->intra = 0;                  pFrame->intra = 0;
819    
820                  BitstreamPutBits(&bs, 0x7f, 8);                  BitstreamPutBits(&bs, 0x7f, 8);
# Line 815  Line 836 
836           * comment style :-)           * comment style :-)
837           * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */           * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
838    
 //$$    SWAP(pEnc->current, pEnc->reference);  
   
839          emms();          emms();
840    
841            // only inc frame num, adapt quant, etc. if we havent seen it before
842            if (pEnc->bframenum_dx50bvop < 0 )
843            {
844          if (pFrame->quant == 0)          if (pFrame->quant == 0)
845                  pEnc->current->quant = RateControlGetQ(&pEnc->rate_control, 0);                  pEnc->current->quant = RateControlGetQ(&pEnc->rate_control, 0);
846          else          else
# Line 832  Line 854 
854    
855          pEnc->current->global_flags = pFrame->general;          pEnc->current->global_flags = pFrame->general;
856          pEnc->current->motion_flags = pFrame->motion;          pEnc->current->motion_flags = pFrame->motion;
857          pEnc->current->seconds = pEnc->mbParam.m_seconds;  
         pEnc->current->ticks = pEnc->mbParam.m_ticks;  
858          /* ToDo : dynamic fcode (in both directions) */          /* ToDo : dynamic fcode (in both directions) */
859          pEnc->current->fcode = pEnc->mbParam.m_fcode;          pEnc->current->fcode = pEnc->mbParam.m_fcode;
860          pEnc->current->bcode = pEnc->mbParam.m_fcode;          pEnc->current->bcode = pEnc->mbParam.m_fcode;
861    
862  //$$$   start_timer();                  pEnc->current->seconds = pEnc->mbParam.m_seconds;
863  //$$$   if (image_input                  pEnc->current->ticks = pEnc->mbParam.m_ticks;
864  //$$$           (&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height,  
865  //$$$            pEnc->mbParam.edged_width, pFrame->image, pFrame->colorspace))                  inc_frame_num(pEnc);
 //$$$           return XVID_ERR_FORMAT;  
 //$$$   stop_conv_timer();  
866    
867  #ifdef _DEBUG_PSNR  #ifdef _DEBUG_PSNR
868          image_copy(&pEnc->sOriginal, &pEnc->current->image,          image_copy(&pEnc->sOriginal, &pEnc->current->image,
# Line 857  Line 876 
876                          "%i  if:%i  st:%i:%i", pEnc->m_framenum++, pEnc->iFrameNum, pEnc->current->seconds, pEnc->current->ticks);                          "%i  if:%i  st:%i:%i", pEnc->m_framenum++, pEnc->iFrameNum, pEnc->current->seconds, pEnc->current->ticks);
877          }          }
878    
   
879          /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%          /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
880           * Luminance masking           * Luminance masking
881           * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */           * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
# Line 887  Line 905 
905                          }                          }
906    
907  #undef OFFSET  #undef OFFSET
   
908                  }                  }
909    
910                  xvid_free(temp_dquants);                  xvid_free(temp_dquants);
911          }          }
912    
913            }
914    
915          /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%          /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
916           * ivop/pvop/bvop selection           * ivop/pvop/bvop selection
917           * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */           * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
918    
919    
920          if (pEnc->iFrameNum == 0 || pFrame->intra == 1 ||          if (pEnc->iFrameNum == 0 || pFrame->intra == 1 || pEnc->bframenum_dx50bvop >= 0 ||
921                  (pFrame->intra < 0 && pEnc->iMaxKeyInterval > 0 &&                  (pFrame->intra < 0 && pEnc->iMaxKeyInterval > 0 &&
922                   pEnc->iFrameNum >= pEnc->iMaxKeyInterval)                   pEnc->iFrameNum >= pEnc->iMaxKeyInterval)
923                  || image_mad(&pEnc->reference->image, &pEnc->current->image,                  || image_mad(&pEnc->reference->image, &pEnc->current->image,
# Line 908  Line 927 
927                   * This will be coded as an Intra Frame                   * This will be coded as an Intra Frame
928                   */                   */
929    
930                  DPRINTF("*** IFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",                  DPRINTF(DPRINTF_DEBUG,"*** IFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
931                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
932                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
933    
# Line 916  Line 935 
935                          image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 200, "IVOP");                          image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 200, "IVOP");
936                  }                  }
937    
938                  FrameCodeI(pEnc, &bs, &bits);                  // when we reach an iframe in DX50BVOP mode, encode the last bframe as a pframe
939    
940                    if ((pEnc->global & XVID_GLOBAL_DX50BVOP) && pEnc->bframenum_tail > 0) {
941    
942                            pEnc->bframenum_tail--;
943                            pEnc->bframenum_dx50bvop = pEnc->bframenum_tail;
944    
945                            SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_dx50bvop]);
946                            if ((pEnc->global & XVID_GLOBAL_DEBUG)) {
947                                    image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 BVOP->PVOP");
948                            }
949                            FrameCodeP(pEnc, &bs, &bits, 1, 0);
950    
951                            pFrame->intra = 0;
952    
953                    } else {
954    
955                            FrameCodeI(pEnc, &bs, &bits);
956                  pFrame->intra = 1;                  pFrame->intra = 1;
                 pEnc->flush_bframes = 1;  
957    
958                  inc_frame_num(pEnc);                          pEnc->bframenum_dx50bvop = -1;
959                    }
960    
961                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {                  pEnc->flush_bframes = 1;
962    
963                    if ((pEnc->global & XVID_GLOBAL_PACKED) && pEnc->bframenum_tail > 0) {
964                          BitstreamPad(&bs);                          BitstreamPad(&bs);
965                          input_valid = 0;                          input_valid = 0;
966                          goto ipvop_loop;                          goto ipvop_loop;
# Line 938  Line 975 
975                   * This will be coded as a Predicted Frame                   * This will be coded as a Predicted Frame
976                   */                   */
977    
978                  DPRINTF("*** PFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",                  DPRINTF(DPRINTF_DEBUG,"*** PFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
979                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
980                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
981    
# Line 950  Line 987 
987                  pFrame->intra = 0;                  pFrame->intra = 0;
988                  pEnc->flush_bframes = 1;                  pEnc->flush_bframes = 1;
989    
                 inc_frame_num(pEnc);  
   
990                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {
991                          BitstreamPad(&bs);                          BitstreamPad(&bs);
992                          input_valid = 0;                          input_valid = 0;
# Line 963  Line 998 
998                   * This will be coded as a Bidirectional Frame                   * This will be coded as a Bidirectional Frame
999                   */                   */
1000    
1001                  DPRINTF("*** BFRAME (store) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",                  DPRINTF(DPRINTF_DEBUG,"*** BFRAME (store) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
1002                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
1003                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
1004    
# Line 988  Line 1023 
1023                  pFrame->intra = 0;                  pFrame->intra = 0;
1024                  pFrame->length = 0;                  pFrame->length = 0;
1025    
                 inc_frame_num(pEnc);  
   
1026                  input_valid = 0;                  input_valid = 0;
1027                  goto bvop_loop;                  goto bvop_loop;
1028          }          }
# Line 1561  Line 1594 
1594          if (pEnc->current->global_flags & XVID_HINTEDME_SET) {          if (pEnc->current->global_flags & XVID_HINTEDME_SET) {
1595                  HintedMESet(pEnc, &bIntra);                  HintedMESet(pEnc, &bIntra);
1596          } else {          } else {
1597    
1598    #ifdef _SMP
1599                    if (NUMTHREADS > 1)
1600                            bIntra =
1601                                    SMP_MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,
1602                                                             &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
1603                                                             iLimit);
1604                    else
1605    #endif
1606    
1607                  bIntra =                  bIntra =
1608                          MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,                          MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,
1609                                                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,                                                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
1610                                                           iLimit);                                                           iLimit);
1611    
1612    
1613          }          }
1614          stop_motion_timer();          stop_motion_timer();
1615    

Legend:
Removed from v.236  
changed lines
  Added in v.259

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