[svn] / branches / dev-api-3 / xvidcore / src / encoder.c Repository:
ViewVC logotype

Diff of /branches/dev-api-3/xvidcore/src/encoder.c

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

revision 573, Mon Sep 30 14:16:02 2002 UTC revision 619, Sat Nov 2 15:52:31 2002 UTC
# Line 39  Line 39 
39   *             MinChen <chenm001@163.com>   *             MinChen <chenm001@163.com>
40   *  14.04.2002 added FrameCodeB()   *  14.04.2002 added FrameCodeB()
41   *   *
42   *  $Id: encoder.c,v 1.76.2.8 2002-09-30 14:16:02 chl Exp $   *  $Id: encoder.c,v 1.76.2.14 2002-11-02 15:52:30 chl Exp $
43   *   *
44   ****************************************************************************/   ****************************************************************************/
45    
# Line 54  Line 54 
54  #include "global.h"  #include "global.h"
55  #include "utils/timer.h"  #include "utils/timer.h"
56  #include "image/image.h"  #include "image/image.h"
 #ifdef BFRAMES  
57  #include "image/font.h"  #include "image/font.h"
58  #include "motion/sad.h"  #include "motion/sad.h"
 #endif  
59  #include "motion/motion.h"  #include "motion/motion.h"
60  #include "bitstream/cbp.h"  #include "bitstream/cbp.h"
61  #include "utils/mbfunctions.h"  #include "utils/mbfunctions.h"
# Line 744  Line 742 
742          if (pEnc->bframenum_head > 0) {          if (pEnc->bframenum_head > 0) {
743                  pEnc->bframenum_head = pEnc->bframenum_tail = 0;                  pEnc->bframenum_head = pEnc->bframenum_tail = 0;
744    
745                    /* write an empty marker to the bitstream.
746    
747                       for divx5 decoder compatibility, this marker must consist
748                       of a not-coded p-vop, with a time_base of zero, and time_increment
749                       indentical to the future-referece frame.
750                    */
751    
752                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {
753                            int tmp;
754    
755                          DPRINTF(DPRINTF_DEBUG,"*** 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",
756                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
757                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
758    
                         set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);  
                         BitstreamWriteVopHeader(&bs, &pEnc->mbParam, pEnc->current, 0);  
759                          BitstreamPad(&bs);                          BitstreamPad(&bs);
760                          BitstreamPutBits(&bs, 0x7f, 8);  
761                            tmp = pEnc->current->seconds;
762                            pEnc->current->seconds = 0; /* force time_base = 0 */
763                            BitstreamWriteVopHeader(&bs, &pEnc->mbParam, pEnc->current, 0);
764                            pEnc->current->seconds = tmp;
765    
766                          pFrame->length = BitstreamLength(&bs);                          pFrame->length = BitstreamLength(&bs);
767                          pFrame->intra = 0;                          pFrame->intra = 0;
# Line 815  Line 823 
823                  pEnc->queue_head =  (pEnc->queue_head + 1) % pEnc->mbParam.max_bframes;                  pEnc->queue_head =  (pEnc->queue_head + 1) % pEnc->mbParam.max_bframes;
824                  pEnc->queue_size--;                  pEnc->queue_size--;
825    
826          } else if (BitstreamPos(&bs) == 0) {          } else {
827    
828                    /* if nothing was encoded, write an 'ignore this frame' flag
829                       to the bitstream */
830    
831                    if (BitstreamPos(&bs) == 0) {
832    
833                  DPRINTF(DPRINTF_DEBUG,"*** 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",
834                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
835                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
836    
837                            BitstreamPutBits(&bs, 0x7f, 8);
838                  pFrame->intra = 0;                  pFrame->intra = 0;
839                    }
                 set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);  
                 BitstreamWriteVopHeader(&bs, &pEnc->mbParam, pEnc->current, 0); // write N_VOP  
                 BitstreamPad(&bs);  
                 pFrame->length = BitstreamLength(&bs);  
   
                 return XVID_ERR_OK;  
   
         } else {  
840    
841                  pFrame->length = BitstreamLength(&bs);                  pFrame->length = BitstreamLength(&bs);
842                  return XVID_ERR_OK;                  return XVID_ERR_OK;
# Line 872  Line 878 
878    
879                  if ((pEnc->global & XVID_GLOBAL_DEBUG)) {                  if ((pEnc->global & XVID_GLOBAL_DEBUG)) {
880                          image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 5,                          image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 5,
881                                  "%i  if:%i  st:%i:%i", pEnc->m_framenum++, pEnc->iFrameNum, pEnc->current->seconds, pEnc->current->ticks);                                  "%i  if:%i  st:%i", pEnc->m_framenum++, pEnc->iFrameNum, pEnc->current->stamp);
882                  }                  }
883    
884          /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%          /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
# Line 963  Line 969 
969                  pEnc->flush_bframes = 1;                  pEnc->flush_bframes = 1;
970    
971                  if ((pEnc->global & XVID_GLOBAL_PACKED) && pEnc->bframenum_tail > 0) {                  if ((pEnc->global & XVID_GLOBAL_PACKED) && pEnc->bframenum_tail > 0) {
972                          BitstreamPad(&bs);                          BitstreamPadAlways(&bs);
973                          input_valid = 0;                          input_valid = 0;
974                          goto ipvop_loop;                          goto ipvop_loop;
975                  }                  }
# Line 990  Line 996 
996                  pEnc->flush_bframes = 1;                  pEnc->flush_bframes = 1;
997    
998                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {
999                          BitstreamPad(&bs);                          BitstreamPadAlways(&bs);
1000                          input_valid = 0;                          input_valid = 0;
1001                          goto ipvop_loop;                          goto ipvop_loop;
1002                  }                  }
# Line 1143  Line 1149 
1149                  pEnc->current->quant = pFrame->quant;                  pEnc->current->quant = pFrame->quant;
1150          }          }
1151    
1152            if ((pEnc->current->global_flags & XVID_QUARTERPEL))
1153                    pEnc->mbParam.m_quarterpel = 1;
1154            else
1155                    pEnc->mbParam.m_quarterpel = 0;
1156    
1157          if ((pEnc->current->global_flags & XVID_LUMIMASKING)) {          if ((pEnc->current->global_flags & XVID_LUMIMASKING)) {
1158                  int *temp_dquants =                  int *temp_dquants =
1159                          (int *) xvid_malloc(pEnc->mbParam.mb_width *                          (int *) xvid_malloc(pEnc->mbParam.mb_width *
# Line 1496  Line 1507 
1507          pEnc->iFrameNum = 0;          pEnc->iFrameNum = 0;
1508          pEnc->mbParam.m_rounding_type = 1;          pEnc->mbParam.m_rounding_type = 1;
1509          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;
1510            pEnc->current->quarterpel =  pEnc->mbParam.m_quarterpel;
1511          pEnc->current->coding_type = I_VOP;          pEnc->current->coding_type = I_VOP;
1512    
1513          BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);          BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
# Line 1505  Line 1517 
1517                  BitstreamWriteUserData(bs, DIVX501B481P, strlen(DIVX501B481P));                  BitstreamWriteUserData(bs, DIVX501B481P, strlen(DIVX501B481P));
1518          }          }
1519    
1520    #define XVID_ID "XviD" XVID_BS_VERSION
1521            BitstreamWriteUserData(bs, XVID_ID, strlen(XVID_ID));
1522    
1523          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);
1524          BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);          BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);
1525    
# Line 1572  Line 1587 
1587          int iLimit;          int iLimit;
1588          int x, y, k;          int x, y, k;
1589          int iSearchRange;          int iSearchRange;
1590          int bIntra;          int bIntra, skip_possible;
1591    
1592          /* IMAGE *pCurrent = &pEnc->current->image; */          /* IMAGE *pCurrent = &pEnc->current->image; */
1593          IMAGE *pRef = &pEnc->reference->image;          IMAGE *pRef = &pEnc->reference->image;
# Line 1584  Line 1599 
1599    
1600          pEnc->mbParam.m_rounding_type = 1 - pEnc->mbParam.m_rounding_type;          pEnc->mbParam.m_rounding_type = 1 - pEnc->mbParam.m_rounding_type;
1601          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;
1602            pEnc->current->quarterpel =  pEnc->mbParam.m_quarterpel;
1603          pEnc->current->fcode = pEnc->mbParam.m_fcode;          pEnc->current->fcode = pEnc->mbParam.m_fcode;
1604    
1605          if (!force_inter)          if (!force_inter)
# Line 1598  Line 1614 
1614                  image_interpolate(pRef, &pEnc->vInterH, &pEnc->vInterV,                  image_interpolate(pRef, &pEnc->vInterH, &pEnc->vInterV,
1615                                                    &pEnc->vInterHV, pEnc->mbParam.edged_width,                                                    &pEnc->vInterHV, pEnc->mbParam.edged_width,
1616                                                    pEnc->mbParam.edged_height,                                                    pEnc->mbParam.edged_height,
1617                                                      pEnc->mbParam.m_quarterpel,
1618                                                    pEnc->current->rounding_type);                                                    pEnc->current->rounding_type);
1619                  stop_inter_timer();                  stop_inter_timer();
1620          }          }
1621    
1622            if (pEnc->current->global_flags & XVID_GMC) {
1623                    printf("Global Motion = %d %d quarterpel=%d\n", pEnc->current->GMC_MV.x, pEnc->current->GMC_MV.y,pEnc->current->quarterpel);
1624                    pEnc->current->coding_type = S_VOP;
1625            } else
1626                    pEnc->current->coding_type = P_VOP;
1627    
1628          start_timer();          start_timer();
1629          if (pEnc->current->global_flags & XVID_HINTEDME_SET) {          if (pEnc->current->global_flags & XVID_HINTEDME_SET) {
1630                  HintedMESet(pEnc, &bIntra);                  HintedMESet(pEnc, &bIntra);
1631                  if (bIntra == 0) MotionEstimationHinted(&pEnc->mbParam, pEnc->current, pEnc->reference,          if (bIntra == 0) {
1632                            pEnc->current->fcode = FindFcode(&pEnc->mbParam, pEnc->current);
1633                            MotionEstimationHinted(&pEnc->mbParam, pEnc->current, pEnc->reference,
1634                                                                                          &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV);                                                                                          &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV);
1635                    }
1636    
1637          } else {          } else {
1638    
# Line 1614  Line 1640 
1640                  MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,                  MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,
1641                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
1642                           iLimit);                           iLimit);
   
1643          }          }
1644          stop_motion_timer();          stop_motion_timer();
1645    
1646          if (bIntra == 1) return FrameCodeI(pEnc, bs, pBits);          if (bIntra == 1) return FrameCodeI(pEnc, bs, pBits);
1647    
1648          pEnc->current->coding_type = P_VOP;          if ( (pEnc->current->GMC_MV.x == 0) && (pEnc->current->GMC_MV.y == 0) )
1649                            pEnc->current->coding_type = P_VOP;             /* no global motion -> no GMC */
1650    
1651    
1652          if (vol_header)          if (vol_header)
1653                  BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);                  BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
# Line 1649  Line 1676 
1676                                                                           dct_codes, pEnc->mbParam.width,                                                                           dct_codes, pEnc->mbParam.width,
1677                                                                           pEnc->mbParam.height,                                                                           pEnc->mbParam.height,
1678                                                                           pEnc->mbParam.edged_width,                                                                           pEnc->mbParam.edged_width,
1679                                                                             pEnc->mbParam.m_quarterpel,
1680                                                                           pEnc->current->rounding_type);                                                                           pEnc->current->rounding_type);
1681                                  stop_comp_timer();                                  stop_comp_timer();
1682    
# Line 1694  Line 1722 
1722    
1723                          /* Finished processing the MB, now check if to CODE or SKIP */                          /* Finished processing the MB, now check if to CODE or SKIP */
1724    
1725                          if ((pMB->mode == MODE_NOT_CODED) ||                          skip_possible = (pMB->cbp == 0) & (pMB->mode == MODE_INTER) &
1726                                  (pMB->cbp == 0 && pMB->mode == MODE_INTER && pMB->mvs[0].x == 0 &&                                                          (pMB->dquant == NO_CHANGE);
                                 pMB->mvs[0].y == 0 && pMB->dquant == NO_CHANGE)) {  
1727    
1728  /* This is a candidate for SKIPping, but check intermediate B-frames first */                          if(pEnc->mbParam.m_quarterpel)
1729                            {       skip_possible &= (pMB->qmvs[0].x == pEnc->current->GMC_MV.x) & (pMB->qmvs[0].y == pEnc->current->GMC_MV.y);
1730                            }
1731                            else
1732                            {       skip_possible &= (pMB->mvs[0].x == pEnc->current->GMC_MV.x) & (pMB->mvs[0].y == pEnc->current->GMC_MV.y);
1733                            }
1734    
1735                            if ( (pMB->mode == MODE_NOT_CODED) || (skip_possible)) {
1736    
1737    /* This is a candidate for SKIPping, but for P-VOPs check intermediate B-frames first */
1738                                  int bSkip = 1;                                  int bSkip = 1;
                                 pMB->mode = MODE_NOT_CODED;  
1739    
1740                                    if (pEnc->current->coding_type == P_VOP)        /* special rule for P-VOP's SKIP */
1741                                  for (k=pEnc->bframenum_head; k< pEnc->bframenum_tail; k++)                                  for (k=pEnc->bframenum_head; k< pEnc->bframenum_tail; k++)
1742                                  {                                  {
1743                                          int iSAD;                                          int iSAD;
# Line 1714  Line 1749 
1749                                                  break;                                                  break;
1750                                          }                                          }
1751                                  }                                  }
1752    
1753                                  if (!bSkip)                                  if (!bSkip)
1754                                  {                                  {
1755                                          VECTOR predMV;                                          VECTOR predMV;
1756                                            if(pEnc->mbParam.m_quarterpel) {
1757                                                    predMV = get_qpmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, 0);
1758                                                    pMB->pmvs[0].x = pMB->qmvs[0].x - predMV.x;  /* with GMC, qmvs doesn't have to be (0,0)! */
1759                                                    pMB->pmvs[0].y = pMB->qmvs[0].y - predMV.y;
1760                                            }
1761                                            else {
1762                                          predMV = get_pmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, 0);                                          predMV = get_pmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, 0);
1763                                          pMB->pmvs[0].x = -predMV.x; pMB->pmvs[0].y = -predMV.y;                                                  pMB->pmvs[0].x = pMB->mvs[0].x - predMV.x; /* with GMC, mvs doesn't have to be (0,0)! */
1764                                                    pMB->pmvs[0].y = pMB->mvs[0].y - predMV.y;
1765                                            }
1766                                          pMB->mode = MODE_INTER;                                          pMB->mode = MODE_INTER;
1767                                          pMB->cbp = 0;                                          pMB->cbp = 0;
1768                                          MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->sStat);                                          MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->sStat);
1769                                  }                                  }
1770                                  else MBSkip(bs);                                  else
1771                                    {
1772                                            pMB->mode = MODE_NOT_CODED;
1773                                            MBSkip(bs);
1774                                    }
1775    
1776                          } else {                          } else {
1777                                  if (pEnc->current->global_flags & XVID_GREYSCALE)                                  if (pEnc->current->global_flags & XVID_GREYSCALE)
# Line 1752  Line 1800 
1800          iSearchRange = 1 << (3 + pEnc->mbParam.m_fcode);          iSearchRange = 1 << (3 + pEnc->mbParam.m_fcode);
1801    
1802          if ((fSigma > iSearchRange / 3)          if ((fSigma > iSearchRange / 3)
1803                  && (pEnc->mbParam.m_fcode <= 3))        // maximum search range 128                  && (pEnc->mbParam.m_fcode <= (3 + pEnc->mbParam.m_quarterpel))) // maximum search range 128
1804          {          {
1805                  pEnc->mbParam.m_fcode++;                  pEnc->mbParam.m_fcode++;
1806                  iSearchRange *= 2;                  iSearchRange *= 2;
1807          } else if ((fSigma < iSearchRange / 6)          } else if ((fSigma < iSearchRange / 6)
1808                             && (pEnc->sStat.fMvPrevSigma >= 0)                             && (pEnc->sStat.fMvPrevSigma >= 0)
1809                             && (pEnc->sStat.fMvPrevSigma < iSearchRange / 6)                             && (pEnc->sStat.fMvPrevSigma < iSearchRange / 6)
1810                             && (pEnc->mbParam.m_fcode >= 2))     // minimum search range 16                          && (pEnc->mbParam.m_fcode >= (2 + pEnc->mbParam.m_quarterpel))) // minimum search range 16
1811          {          {
1812                  pEnc->mbParam.m_fcode--;                  pEnc->mbParam.m_fcode--;
1813                  iSearchRange /= 2;                  iSearchRange /= 2;
# Line 1785  Line 1833 
1833                  pEnc->current->quant = pEnc->reference->quant;                  pEnc->current->quant = pEnc->reference->quant;
1834                  pEnc->current->motion_flags = pEnc->reference->motion_flags;                  pEnc->current->motion_flags = pEnc->reference->motion_flags;
1835                  pEnc->current->rounding_type = pEnc->reference->rounding_type;                  pEnc->current->rounding_type = pEnc->reference->rounding_type;
1836                    pEnc->current->quarterpel =  pEnc->reference->quarterpel;
1837                  pEnc->current->fcode = pEnc->reference->fcode;                  pEnc->current->fcode = pEnc->reference->fcode;
1838                  pEnc->current->bcode = pEnc->reference->bcode;                  pEnc->current->bcode = pEnc->reference->bcode;
1839                  image_copy(&pEnc->current->image, &pEnc->reference->image, pEnc->mbParam.edged_width, pEnc->mbParam.height);                  image_copy(&pEnc->current->image, &pEnc->reference->image, pEnc->mbParam.edged_width, pEnc->mbParam.height);
# Line 1824  Line 1873 
1873          }          }
1874  #endif  #endif
1875    
1876            frame->quarterpel =  pEnc->mbParam.m_quarterpel;
1877    
1878          // forward          // forward
1879          image_setedges(f_ref, pEnc->mbParam.edged_width,          image_setedges(f_ref, pEnc->mbParam.edged_width,
1880                                     pEnc->mbParam.edged_height, pEnc->mbParam.width,                                     pEnc->mbParam.edged_height, pEnc->mbParam.width,
# Line 1831  Line 1882 
1882          start_timer();          start_timer();
1883          image_interpolate(f_ref, &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,          image_interpolate(f_ref, &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
1884                                            pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,                                            pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
1885                                            0);                                            pEnc->mbParam.m_quarterpel, 0);
1886          stop_inter_timer();          stop_inter_timer();
1887    
1888          // backward          // backward
# Line 1841  Line 1892 
1892          start_timer();          start_timer();
1893          image_interpolate(b_ref, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,          image_interpolate(b_ref, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
1894                                            pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,                                            pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
1895                                            0);                                            pEnc->mbParam.m_quarterpel, 0);
1896          stop_inter_timer();          stop_inter_timer();
1897    
1898          start_timer();          start_timer();

Legend:
Removed from v.573  
changed lines
  Added in v.619

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