[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.76.2.26, Mon Dec 9 10:47:05 2002 UTC revision 1.76.2.32, Fri Jan 3 16:25:14 2003 UTC
# Line 73  Line 73 
73   ****************************************************************************/   ****************************************************************************/
74    
75  #define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT  #define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT
76  #define SWAP(A,B)    { void * tmp = A; A = B; B = tmp; }  #define SWAP(_T_,A,B)    { _T_ tmp = A; A = B; B = tmp; }
77    
78  /*****************************************************************************  /*****************************************************************************
79   * Local function prototypes   * Local function prototypes
# Line 646  Line 646 
646    
647    
648    
649    /* convert pFrame->intra to coding_type */
650    static int intra2coding_type(int intra)
651    {
652            if (intra < 0)          return -1;
653            if (intra == 1)         return I_VOP;
654            if (intra == 2)         return B_VOP;
655    
656            return P_VOP;
657    }
658    
659    
660    
661  /*****************************************************************************  /*****************************************************************************
# Line 703  Line 713 
713                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
714    
715                          pEnc->bframenum_tail--;                          pEnc->bframenum_tail--;
716                          SWAP(pEnc->current, pEnc->reference);                          SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
717    
718                          SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]);                          SWAP(FRAMEINFO *, pEnc->current, pEnc->bframes[pEnc->bframenum_tail]);
719    
720                          FrameCodeP(pEnc, &bs, &bits, 1, 0);                          FrameCodeP(pEnc, &bs, &bits, 1, 0);
721                          bframes_count = 0;                          bframes_count = 0;
# Line 781  Line 791 
791          if (pEnc->bframenum_dx50bvop != -1)          if (pEnc->bframenum_dx50bvop != -1)
792          {          {
793    
794                  SWAP(pEnc->current, pEnc->reference);                  SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
795                  SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_dx50bvop]);                  SWAP(FRAMEINFO *, pEnc->current, pEnc->bframes[pEnc->bframenum_dx50bvop]);
796    
797                  if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {                  if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {
798                          image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 IVOP");                          image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 IVOP");
# Line 796  Line 806 
806    
807          } else if (input_valid) {          } else if (input_valid) {
808    
809                  SWAP(pEnc->current, pEnc->reference);                  SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
810    
811                  start_timer();                  start_timer();
812                  if (image_input                  if (image_input
# Line 822  Line 832 
832    
833          } else if (pEnc->queue_size > 0) {          } else if (pEnc->queue_size > 0) {
834    
835                  SWAP(pEnc->current, pEnc->reference);                  SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
836    
837                  image_swap(&pEnc->current->image, &pEnc->queue[pEnc->queue_head]);                  image_swap(&pEnc->current->image, &pEnc->queue[pEnc->queue_head]);
838                  pEnc->queue_head =  (pEnc->queue_head + 1) % pEnc->mbParam.max_bframes;                  pEnc->queue_head =  (pEnc->queue_head + 1) % pEnc->mbParam.max_bframes;
# Line 855  Line 865 
865          // only inc frame num, adapt quant, etc. if we havent seen it before          // only inc frame num, adapt quant, etc. if we havent seen it before
866          if (pEnc->bframenum_dx50bvop < 0 )          if (pEnc->bframenum_dx50bvop < 0 )
867          {          {
868                    mode = intra2coding_type(pFrame->intra);
869                  if (pFrame->quant == 0)                  if (pFrame->quant == 0)
870                          pEnc->current->quant = RateControlGetQ(&pEnc->rate_control, 0);                          pEnc->current->quant = RateControlGetQ(&pEnc->rate_control, 0);
871                  else                  else
# Line 928  Line 939 
939           * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */           * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
940          pEnc->iFrameNum++;          pEnc->iFrameNum++;
941    
942          if (pEnc->iFrameNum == 0 || pFrame->intra == 1 || pEnc->bframenum_dx50bvop >= 0 ||          if (pEnc->iFrameNum == 0 || pEnc->bframenum_dx50bvop >= 0 ||
943                  (pFrame->intra < 0 && pEnc->mbParam.iMaxKeyInterval > 0 &&                  (mode < 0 && pEnc->mbParam.iMaxKeyInterval > 0 &&
944                   pEnc->iFrameNum >= pEnc->mbParam.iMaxKeyInterval)                          pEnc->iFrameNum >= pEnc->mbParam.iMaxKeyInterval))
945                  || 2 == (mode = MEanalysis(&pEnc->reference->image, pEnc->current,          {
946                    mode = I_VOP;
947            }else{
948                    mode = MEanalysis(&pEnc->reference->image, pEnc->current,
949                                                                          &pEnc->mbParam, pEnc->mbParam.iMaxKeyInterval,                                                                          &pEnc->mbParam, pEnc->mbParam.iMaxKeyInterval,
950                                                                          (pFrame->intra < 0) ? pEnc->iFrameNum : 0,                                          (mode < 0) ? pEnc->iFrameNum : 0,
951                                                                          bframes_count++))) {                                          bframes_count++);
952            }
953    
954            if (mode == I_VOP) {
955                  /*                  /*
956                   * This will be coded as an Intra Frame                   * This will be coded as an Intra Frame
957                   */                   */
# Line 969  Line 985 
985                          pEnc->bframenum_tail--;                          pEnc->bframenum_tail--;
986                          pEnc->bframenum_dx50bvop = pEnc->bframenum_tail;                          pEnc->bframenum_dx50bvop = pEnc->bframenum_tail;
987    
988                          SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_dx50bvop]);                          SWAP(FRAMEINFO *, pEnc->current, pEnc->bframes[pEnc->bframenum_dx50bvop]);
989                          if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {                          if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {
990                                  image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 BVOP->PVOP");                                  image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 BVOP->PVOP");
991                          }                          }
# Line 999  Line 1015 
1015                   * NB : sequences like "IIBB" decode fine with msfdam but,                   * NB : sequences like "IIBB" decode fine with msfdam but,
1016                   *      go screwy with divx 5.00                   *      go screwy with divx 5.00
1017                   */                   */
1018          } else if (pEnc->bframenum_tail >= pEnc->mbParam.max_bframes || mode != 0) {          } else if (mode == P_VOP || pEnc->bframenum_tail >= pEnc->mbParam.max_bframes) {
 //      } else if (pFrame->intra == 0 || pEnc->bframenum_tail >= pEnc->mbParam.max_bframes || mode != 0) {  
1019                  /*                  /*
1020                   * This will be coded as a Predicted Frame                   * This will be coded as a Predicted Frame
1021                   */                   */
# Line 1024  Line 1039 
1039                          goto ipvop_loop;                          goto ipvop_loop;
1040                  }                  }
1041    
1042          } else {          } else {        /* mode == B_VOP */
1043                  /*                  /*
1044                   * This will be coded as a Bidirectional Frame                   * This will be coded as a Bidirectional Frame
1045                   */                   */
# Line 1051  Line 1066 
1066                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size,pEnc->current->quant);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size,pEnc->current->quant);
1067    
1068                  /* store frame into bframe buffer & swap ref back to current */                  /* store frame into bframe buffer & swap ref back to current */
1069                  SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]);                  SWAP(FRAMEINFO *, pEnc->current, pEnc->bframes[pEnc->bframenum_tail]);
1070                  SWAP(pEnc->current, pEnc->reference);                  SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
1071    
1072                  pEnc->bframenum_tail++;                  pEnc->bframenum_tail++;
1073    
# Line 1132  Line 1147 
1147          ENC_CHECK(pFrame->bitstream);          ENC_CHECK(pFrame->bitstream);
1148          ENC_CHECK(pFrame->image);          ENC_CHECK(pFrame->image);
1149    
1150          SWAP(pEnc->current, pEnc->reference);          SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
1151    
1152          pEnc->current->global_flags = pFrame->general;          pEnc->current->global_flags = pFrame->general;
1153          pEnc->current->motion_flags = pFrame->motion;          pEnc->current->motion_flags = pFrame->motion;
# Line 1548  Line 1563 
1563    
1564          BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);          BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
1565    
         /* XXX: move this stuff to BitstreamWriteVolHeader */  
 #define DIVX501B481P "DivX501b481p"  
         if ((pEnc->mbParam.global & XVID_GLOBAL_PACKED)) {  
                 BitstreamWriteUserData(bs, DIVX501B481P, strlen(DIVX501B481P));  
         }  
   
 #define XVID_ID "XviD" XVID_BS_VERSION  
         BitstreamWriteUserData(bs, XVID_ID, strlen(XVID_ID));  
   
1566          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);
1567          BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);          BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);
1568    
# Line 1590  Line 1596 
1596                          stop_coding_timer();                          stop_coding_timer();
1597                  }                  }
1598    
1599            if ((pEnc->current->global_flags & XVID_REDUCED))
1600            {
1601                    image_deblock_rrv(&pEnc->current->image, pEnc->mbParam.edged_width,
1602                            pEnc->current->mbs, mb_width, mb_height, pEnc->mbParam.mb_width,
1603                            16, XVID_DEC_DEBLOCKY|XVID_DEC_DEBLOCKUV);
1604            }
1605          emms();          emms();
1606    
1607          *pBits = BitstreamPos(bs) - *pBits;          *pBits = BitstreamPos(bs) - *pBits;
# Line 1632  Line 1644 
1644    
1645          if ((pEnc->current->global_flags & XVID_REDUCED))          if ((pEnc->current->global_flags & XVID_REDUCED))
1646          {          {
1647                  // mb_width = (pEnc->mbParam.width + 31) / 32;                  mb_width = (pEnc->mbParam.width + 31) / 32;
1648                  // mb_height = (pEnc->mbParam.height + 31) / 32;                  mb_height = (pEnc->mbParam.height + 31) / 32;
   
                 /* XXX: reduced resoltion not yet supported */  
                 pEnc->current->global_flags &= ~XVID_REDUCED;  
1649          }          }
1650    
1651    
# Line 1673  Line 1682 
1682                  pEnc->current->coding_type = P_VOP;                  pEnc->current->coding_type = P_VOP;
1683    
1684          start_timer();          start_timer();
1685          if (pEnc->current->global_flags & XVID_HINTEDME_SET) {          if (pEnc->current->global_flags & XVID_HINTEDME_SET)
1686                  HintedMESet(pEnc, &bIntra);                  HintedMESet(pEnc, &bIntra);
1687          if (bIntra == 0) {          else
                         pEnc->current->fcode = FindFcode(&pEnc->mbParam, pEnc->current);  
                         MotionEstimationHinted(&pEnc->mbParam, pEnc->current, pEnc->reference,  
                                                                                         &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV);  
                 }  
   
         } else {  
1688    
1689                  bIntra =                  bIntra =
1690                          MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,                          MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,
1691                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
1692                           iLimit);                           iLimit);
1693          }  
1694          stop_motion_timer();          stop_motion_timer();
1695    
1696          if (bIntra == 1) return FrameCodeI(pEnc, bs, pBits);          if (bIntra == 1) return FrameCodeI(pEnc, bs, pBits);
# Line 1834  Line 1837 
1837                  }                  }
1838          }          }
1839    
1840            if ((pEnc->current->global_flags & XVID_REDUCED))
1841            {
1842                    image_deblock_rrv(&pEnc->current->image, pEnc->mbParam.edged_width,
1843                            pEnc->current->mbs, mb_width, mb_height, pEnc->mbParam.mb_width,
1844                            16, XVID_DEC_DEBLOCKY|XVID_DEC_DEBLOCKUV);
1845            }
1846    
1847          emms();          emms();
1848    
1849          if (pEnc->current->global_flags & XVID_HINTEDME_GET) {          if (pEnc->current->global_flags & XVID_HINTEDME_GET) {
# Line 1902  Line 1912 
1912          }          }
1913          */          */
1914    
1915    
1916          *pBits = BitstreamPos(bs) - *pBits;          *pBits = BitstreamPos(bs) - *pBits;
1917    
1918          return 0;                                       // inter          return 0;                                       // inter
Legend:
Removed from v.1.76.2.26  
changed lines
  Added in v.1.76.2.32
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4