[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 579, Sat Oct 5 21:37:44 2002 UTC revision 658, Tue Nov 19 13:04:35 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.12 2002-10-05 21:36:35 Isibaar Exp $   *  $Id: encoder.c,v 1.76.2.19 2002-11-19 13:03:57 syskin 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 109  Line 107 
107  };  };
108    
109    
 static void __inline  
 image_null(IMAGE * image)  
 {  
         image->y = image->u = image->v = NULL;  
 }  
   
   
110  /*****************************************************************************  /*****************************************************************************
111   * Encoder creation   * Encoder creation
112   *   *
# Line 625  Line 616 
616          start_timer();          start_timer();
617          if (image_input          if (image_input
618                  (&pEnc->queue[pEnc->queue_tail], pEnc->mbParam.width, pEnc->mbParam.height,                  (&pEnc->queue[pEnc->queue_tail], pEnc->mbParam.width, pEnc->mbParam.height,
619                   pEnc->mbParam.edged_width, pFrame->image, pFrame->colorspace))                   pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING))
620                  return;                  return;
621          stop_conv_timer();          stop_conv_timer();
622    
# Line 675  Line 666 
666          uint32_t bits, mode;          uint32_t bits, mode;
667    
668          int input_valid = 1;          int input_valid = 1;
669            int bframes_count = 0;
670    
671  #ifdef _DEBUG_PSNR  #ifdef _DEBUG_PSNR
672          float psnr;          float psnr;
# Line 715  Line 707 
707                          SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]);                          SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]);
708    
709                          FrameCodeP(pEnc, &bs, &bits, 1, 0);                          FrameCodeP(pEnc, &bs, &bits, 1, 0);
710                            bframes_count = 0;
711    
712                          BitstreamPad(&bs);                          BitstreamPad(&bs);
713                          pFrame->length = BitstreamLength(&bs);                          pFrame->length = BitstreamLength(&bs);
714                          pFrame->intra = 0;                          pFrame->intra = 0;
715    
716                            emms();
717    
718                          return XVID_ERR_OK;                          return XVID_ERR_OK;
719                  }                  }
720    
# Line 738  Line 733 
733                  if (input_valid)                  if (input_valid)
734                          queue_image(pEnc, pFrame);                          queue_image(pEnc, pFrame);
735    
736                    emms();
737    
738                  return XVID_ERR_OK;                  return XVID_ERR_OK;
739          }          }
740    
# Line 771  Line 768 
768                          if (input_valid)                          if (input_valid)
769                                  queue_image(pEnc, pFrame);                                  queue_image(pEnc, pFrame);
770    
771                            emms();
772    
773                          return XVID_ERR_OK;                          return XVID_ERR_OK;
774                  }                  }
775          }          }
# Line 801  Line 800 
800                  start_timer();                  start_timer();
801                  if (image_input                  if (image_input
802                          (&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height,                          (&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height,
803                          pEnc->mbParam.edged_width, pFrame->image, pFrame->colorspace))                          pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING))
804                    {
805                            emms();
806                          return XVID_ERR_FORMAT;                          return XVID_ERR_FORMAT;
807                    }
808                  stop_conv_timer();                  stop_conv_timer();
809    
810                  // queue input frame, and dequue next image                  // queue input frame, and dequue next image
# Line 841  Line 843 
843                  }                  }
844    
845                  pFrame->length = BitstreamLength(&bs);                  pFrame->length = BitstreamLength(&bs);
846                    emms();
847                  return XVID_ERR_OK;                  return XVID_ERR_OK;
848          }          }
849    
# Line 927  Line 930 
930          if (pEnc->iFrameNum == 0 || pFrame->intra == 1 || pEnc->bframenum_dx50bvop >= 0 ||          if (pEnc->iFrameNum == 0 || pFrame->intra == 1 || pEnc->bframenum_dx50bvop >= 0 ||
931                  (pFrame->intra < 0 && pEnc->iMaxKeyInterval > 0 &&                  (pFrame->intra < 0 && pEnc->iMaxKeyInterval > 0 &&
932                   pEnc->iFrameNum >= pEnc->iMaxKeyInterval)                   pEnc->iFrameNum >= pEnc->iMaxKeyInterval)
933                  || /*image_mad(&pEnc->reference->image, &pEnc->current->image,                  || 2 == (mode = MEanalysis(&pEnc->reference->image, pEnc->current,
934                                           pEnc->mbParam.edged_width, pEnc->mbParam.width,                                                                          &pEnc->mbParam, pEnc->iMaxKeyInterval,
935                                           pEnc->mbParam.height) > 30) {*/                                                                          (pFrame->intra < 0) ? pEnc->iFrameNum : 0,
936                          2 == (mode = MEanalysis(&pEnc->reference->image, &pEnc->current->image,                                                                          bframes_count++))) {
                                          &pEnc->mbParam, pEnc->current->mbs, pEnc->current->fcode))) {  
937    
938                  /*                  /*
939                   * This will be coded as an Intra Frame                   * This will be coded as an Intra Frame
940                   */                   */
941                    if ((pEnc->current->global_flags & XVID_QUARTERPEL))
942                            pEnc->mbParam.m_quarterpel = 1;
943                    else
944                            pEnc->mbParam.m_quarterpel = 0;
945    
946                  DPRINTF(DPRINTF_DEBUG,"*** 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",
947                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
# Line 957  Line 963 
963                                  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");
964                          }                          }
965                          FrameCodeP(pEnc, &bs, &bits, 1, 0);                          FrameCodeP(pEnc, &bs, &bits, 1, 0);
966                            bframes_count = 0;
967    
968                          pFrame->intra = 0;                          pFrame->intra = 0;
969    
970                  } else {                  } else {
971    
972                          FrameCodeI(pEnc, &bs, &bits);                          FrameCodeI(pEnc, &bs, &bits);
973                            bframes_count = 0;
974                          pFrame->intra = 1;                          pFrame->intra = 1;
975    
976                          pEnc->bframenum_dx50bvop = -1;                          pEnc->bframenum_dx50bvop = -1;
# Line 994  Line 1002 
1002                  }                  }
1003    
1004                  FrameCodeP(pEnc, &bs, &bits, 1, 0);                  FrameCodeP(pEnc, &bs, &bits, 1, 0);
1005                    bframes_count = 0;
1006                  pFrame->intra = 0;                  pFrame->intra = 0;
1007                  pEnc->flush_bframes = 1;                  pEnc->flush_bframes = 1;
1008    
1009                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {                  if ((pEnc->global & XVID_GLOBAL_PACKED) && (pEnc->bframenum_tail > 0)) {
1010                          BitstreamPadAlways(&bs);                          BitstreamPadAlways(&bs);
1011                          input_valid = 0;                          input_valid = 0;
1012                          goto ipvop_loop;                          goto ipvop_loop;
# Line 1078  Line 1087 
1087          stop_global_timer();          stop_global_timer();
1088          write_timer();          write_timer();
1089    
1090            emms();
1091          return XVID_ERR_OK;          return XVID_ERR_OK;
1092  }  }
1093    
# Line 1132  Line 1142 
1142          start_timer();          start_timer();
1143          if (image_input          if (image_input
1144                  (&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height,                  (&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height,
1145                   pEnc->mbParam.edged_width, pFrame->image, pFrame->colorspace) < 0)                   pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING) < 0)
1146                  return XVID_ERR_FORMAT;                  return XVID_ERR_FORMAT;
1147          stop_conv_timer();          stop_conv_timer();
1148    
# Line 1509  Line 1519 
1519          pEnc->iFrameNum = 0;          pEnc->iFrameNum = 0;
1520          pEnc->mbParam.m_rounding_type = 1;          pEnc->mbParam.m_rounding_type = 1;
1521          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;
1522            pEnc->current->quarterpel =  pEnc->mbParam.m_quarterpel;
1523          pEnc->current->coding_type = I_VOP;          pEnc->current->coding_type = I_VOP;
1524    
1525          BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);          BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
# Line 1518  Line 1529 
1529                  BitstreamWriteUserData(bs, DIVX501B481P, strlen(DIVX501B481P));                  BitstreamWriteUserData(bs, DIVX501B481P, strlen(DIVX501B481P));
1530          }          }
1531    
1532    #define XVID_ID "XviD" XVID_BS_VERSION
1533            BitstreamWriteUserData(bs, XVID_ID, strlen(XVID_ID));
1534    
1535          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);
1536          BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);          BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);
1537    
# Line 1597  Line 1611 
1611    
1612          pEnc->mbParam.m_rounding_type = 1 - pEnc->mbParam.m_rounding_type;          pEnc->mbParam.m_rounding_type = 1 - pEnc->mbParam.m_rounding_type;
1613          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;
1614            pEnc->current->quarterpel =  pEnc->mbParam.m_quarterpel;
1615          pEnc->current->fcode = pEnc->mbParam.m_fcode;          pEnc->current->fcode = pEnc->mbParam.m_fcode;
1616    
1617          if (!force_inter)          if (!force_inter)
# Line 1616  Line 1631 
1631                  stop_inter_timer();                  stop_inter_timer();
1632          }          }
1633    
1634            if (pEnc->current->global_flags & XVID_GMC) {
1635    //              printf("Global Motion = %d %d quarterpel=%d\n", pEnc->current->GMC_MV.x, pEnc->current->GMC_MV.y,pEnc->current->quarterpel);
1636                    DPRINTF(DPRINTF_HEADER, "Global Motion = %d %d quarterpel=%d\n", pEnc->current->GMC_MV.x, pEnc->current->GMC_MV.y,pEnc->current->quarterpel);
1637                    pEnc->current->coding_type = S_VOP;
1638            } else
1639                    pEnc->current->coding_type = P_VOP;
1640    
1641          start_timer();          start_timer();
1642          if (pEnc->current->global_flags & XVID_HINTEDME_SET) {          if (pEnc->current->global_flags & XVID_HINTEDME_SET) {
1643                  HintedMESet(pEnc, &bIntra);                  HintedMESet(pEnc, &bIntra);
# Line 1631  Line 1653 
1653                  MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,                  MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,
1654                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
1655                           iLimit);                           iLimit);
   
1656          }          }
1657          stop_motion_timer();          stop_motion_timer();
1658    
1659          if (bIntra == 1) return FrameCodeI(pEnc, bs, pBits);          if (bIntra == 1) return FrameCodeI(pEnc, bs, pBits);
1660    
1661          pEnc->current->coding_type = P_VOP;          if ( (pEnc->current->GMC_MV.x == 0) && (pEnc->current->GMC_MV.y == 0) )
1662                            pEnc->current->coding_type = P_VOP;             /* no global motion -> no GMC */
1663    
1664    
1665          if (vol_header)          if (vol_header)
1666                  BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);                  BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
# Line 1716  Line 1739 
1739                                                          (pMB->dquant == NO_CHANGE);                                                          (pMB->dquant == NO_CHANGE);
1740    
1741                          if(pEnc->mbParam.m_quarterpel)                          if(pEnc->mbParam.m_quarterpel)
1742                                  skip_possible &= (pMB->qmvs[0].x == 0) & (pMB->qmvs[0].y == 0);                          {       skip_possible &= (pMB->qmvs[0].x == pEnc->current->GMC_MV.x) & (pMB->qmvs[0].y == pEnc->current->GMC_MV.y);
1743                            }
1744                          else                          else
1745                                  skip_possible &= (pMB->mvs[0].x == 0) & (pMB->mvs[0].y == 0);                          {       skip_possible &= (pMB->mvs[0].x == pEnc->current->GMC_MV.x) & (pMB->mvs[0].y == pEnc->current->GMC_MV.y);
1746                            }
1747    
1748                          if ((pMB->mode == MODE_NOT_CODED) || (skip_possible)) {                          if ((pMB->mode == MODE_NOT_CODED) || (skip_possible)) {
 /* This is a candidate for SKIPping, but check intermediate B-frames first */  
1749    
1750    /* This is a candidate for SKIPping, but for P-VOPs check intermediate B-frames first */
1751                                  int bSkip = 1;                                  int bSkip = 1;
                                 pMB->mode = MODE_NOT_CODED;  
1752    
1753                                    if (pEnc->current->coding_type == P_VOP)        /* special rule for P-VOP's SKIP */
1754                                  for (k=pEnc->bframenum_head; k< pEnc->bframenum_tail; k++)                                  for (k=pEnc->bframenum_head; k< pEnc->bframenum_tail; k++)
1755                                  {                                  {
1756                                          int iSAD;                                          int iSAD;
# Line 1737  Line 1762 
1762                                                  break;                                                  break;
1763                                          }                                          }
1764                                  }                                  }
1765    
1766                                  if (!bSkip)                                  if (!bSkip)
1767                                  {                                  {
1768                                          VECTOR predMV;                                          VECTOR predMV;
1769                                          if(pEnc->mbParam.m_quarterpel)                                          if(pEnc->mbParam.m_quarterpel) {
1770                                                  predMV = get_qpmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, 0);                                                  predMV = get_qpmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, 0);
1771                                          else                                                  pMB->pmvs[0].x = pMB->qmvs[0].x - predMV.x;  /* with GMC, qmvs doesn't have to be (0,0)! */
1772                                                    pMB->pmvs[0].y = pMB->qmvs[0].y - predMV.y;
1773                                            }
1774                                            else {
1775                                                  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);
1776                                          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)! */
1777                                                    pMB->pmvs[0].y = pMB->mvs[0].y - predMV.y;
1778                                            }
1779                                          pMB->mode = MODE_INTER;                                          pMB->mode = MODE_INTER;
1780                                          pMB->cbp = 0;                                          pMB->cbp = 0;
1781                                          MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->sStat);                                          MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->sStat);
1782                                  }                                  }
1783                                  else MBSkip(bs);                                  else
1784                                    {
1785                                            pMB->mode = MODE_NOT_CODED;
1786                                            MBSkip(bs);
1787                                    }
1788    
1789                          } else {                          } else {
1790                                  if (pEnc->current->global_flags & XVID_GREYSCALE)                                  if (pEnc->current->global_flags & XVID_GREYSCALE)
# Line 1811  Line 1846 
1846                  pEnc->current->quant = pEnc->reference->quant;                  pEnc->current->quant = pEnc->reference->quant;
1847                  pEnc->current->motion_flags = pEnc->reference->motion_flags;                  pEnc->current->motion_flags = pEnc->reference->motion_flags;
1848                  pEnc->current->rounding_type = pEnc->reference->rounding_type;                  pEnc->current->rounding_type = pEnc->reference->rounding_type;
1849                    pEnc->current->quarterpel =  pEnc->reference->quarterpel;
1850                  pEnc->current->fcode = pEnc->reference->fcode;                  pEnc->current->fcode = pEnc->reference->fcode;
1851                  pEnc->current->bcode = pEnc->reference->bcode;                  pEnc->current->bcode = pEnc->reference->bcode;
1852                  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 1850  Line 1886 
1886          }          }
1887  #endif  #endif
1888    
1889            frame->quarterpel =  pEnc->mbParam.m_quarterpel;
1890    
1891          // forward          // forward
1892          image_setedges(f_ref, pEnc->mbParam.edged_width,          image_setedges(f_ref, pEnc->mbParam.edged_width,
1893                                     pEnc->mbParam.edged_height, pEnc->mbParam.width,                                     pEnc->mbParam.edged_height, pEnc->mbParam.width,
# Line 1877  Line 1915 
1915                  ((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)),     // time_pp                  ((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)),     // time_pp
1916                          pEnc->reference->mbs, f_ref,                          pEnc->reference->mbs, f_ref,
1917                                                   &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,                                                   &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
1918                                                   pEnc->current->mbs, b_ref, &pEnc->vInterH,                                                   pEnc->current, b_ref, &pEnc->vInterH,
1919                                                   &pEnc->vInterV, &pEnc->vInterHV);                                                   &pEnc->vInterV, &pEnc->vInterHV);
1920    
1921    
Legend:
Removed from v.579  
changed lines
  Added in v.658
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4