[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 643, Tue Nov 12 14:44:53 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.18 2002-11-12 14:44:53 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
# Line 957  Line 959 
959                                  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");
960                          }                          }
961                          FrameCodeP(pEnc, &bs, &bits, 1, 0);                          FrameCodeP(pEnc, &bs, &bits, 1, 0);
962                            bframes_count = 0;
963    
964                          pFrame->intra = 0;                          pFrame->intra = 0;
965    
966                  } else {                  } else {
967    
968                          FrameCodeI(pEnc, &bs, &bits);                          FrameCodeI(pEnc, &bs, &bits);
969                            bframes_count = 0;
970                          pFrame->intra = 1;                          pFrame->intra = 1;
971    
972                          pEnc->bframenum_dx50bvop = -1;                          pEnc->bframenum_dx50bvop = -1;
# Line 994  Line 998 
998                  }                  }
999    
1000                  FrameCodeP(pEnc, &bs, &bits, 1, 0);                  FrameCodeP(pEnc, &bs, &bits, 1, 0);
1001                    bframes_count = 0;
1002                  pFrame->intra = 0;                  pFrame->intra = 0;
1003                  pEnc->flush_bframes = 1;                  pEnc->flush_bframes = 1;
1004    
1005                  if ((pEnc->global & XVID_GLOBAL_PACKED)) {                  if ((pEnc->global & XVID_GLOBAL_PACKED) && (pEnc->bframenum_tail > 0)) {
1006                          BitstreamPadAlways(&bs);                          BitstreamPadAlways(&bs);
1007                          input_valid = 0;                          input_valid = 0;
1008                          goto ipvop_loop;                          goto ipvop_loop;
# Line 1078  Line 1083 
1083          stop_global_timer();          stop_global_timer();
1084          write_timer();          write_timer();
1085    
1086            emms();
1087          return XVID_ERR_OK;          return XVID_ERR_OK;
1088  }  }
1089    
# Line 1132  Line 1138 
1138          start_timer();          start_timer();
1139          if (image_input          if (image_input
1140                  (&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height,                  (&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height,
1141                   pEnc->mbParam.edged_width, pFrame->image, pFrame->colorspace) < 0)                   pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING) < 0)
1142                  return XVID_ERR_FORMAT;                  return XVID_ERR_FORMAT;
1143          stop_conv_timer();          stop_conv_timer();
1144    
# Line 1509  Line 1515 
1515          pEnc->iFrameNum = 0;          pEnc->iFrameNum = 0;
1516          pEnc->mbParam.m_rounding_type = 1;          pEnc->mbParam.m_rounding_type = 1;
1517          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;
1518            pEnc->current->quarterpel =  pEnc->mbParam.m_quarterpel;
1519          pEnc->current->coding_type = I_VOP;          pEnc->current->coding_type = I_VOP;
1520    
1521          BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);          BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
# Line 1518  Line 1525 
1525                  BitstreamWriteUserData(bs, DIVX501B481P, strlen(DIVX501B481P));                  BitstreamWriteUserData(bs, DIVX501B481P, strlen(DIVX501B481P));
1526          }          }
1527    
1528    #define XVID_ID "XviD" XVID_BS_VERSION
1529            BitstreamWriteUserData(bs, XVID_ID, strlen(XVID_ID));
1530    
1531          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);
1532          BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);          BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);
1533    
# Line 1597  Line 1607 
1607    
1608          pEnc->mbParam.m_rounding_type = 1 - pEnc->mbParam.m_rounding_type;          pEnc->mbParam.m_rounding_type = 1 - pEnc->mbParam.m_rounding_type;
1609          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;
1610            pEnc->current->quarterpel =  pEnc->mbParam.m_quarterpel;
1611          pEnc->current->fcode = pEnc->mbParam.m_fcode;          pEnc->current->fcode = pEnc->mbParam.m_fcode;
1612    
1613          if (!force_inter)          if (!force_inter)
# Line 1616  Line 1627 
1627                  stop_inter_timer();                  stop_inter_timer();
1628          }          }
1629    
1630            if (pEnc->current->global_flags & XVID_GMC) {
1631    //              printf("Global Motion = %d %d quarterpel=%d\n", pEnc->current->GMC_MV.x, pEnc->current->GMC_MV.y,pEnc->current->quarterpel);
1632                    DPRINTF(DPRINTF_HEADER, "Global Motion = %d %d quarterpel=%d\n", pEnc->current->GMC_MV.x, pEnc->current->GMC_MV.y,pEnc->current->quarterpel);
1633                    pEnc->current->coding_type = S_VOP;
1634            } else
1635                    pEnc->current->coding_type = P_VOP;
1636    
1637          start_timer();          start_timer();
1638          if (pEnc->current->global_flags & XVID_HINTEDME_SET) {          if (pEnc->current->global_flags & XVID_HINTEDME_SET) {
1639                  HintedMESet(pEnc, &bIntra);                  HintedMESet(pEnc, &bIntra);
# Line 1631  Line 1649 
1649                  MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,                  MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,
1650                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
1651                           iLimit);                           iLimit);
   
1652          }          }
1653          stop_motion_timer();          stop_motion_timer();
1654    
1655          if (bIntra == 1) return FrameCodeI(pEnc, bs, pBits);          if (bIntra == 1) return FrameCodeI(pEnc, bs, pBits);
1656    
1657          pEnc->current->coding_type = P_VOP;          if ( (pEnc->current->GMC_MV.x == 0) && (pEnc->current->GMC_MV.y == 0) )
1658                            pEnc->current->coding_type = P_VOP;             /* no global motion -> no GMC */
1659    
1660    
1661          if (vol_header)          if (vol_header)
1662                  BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);                  BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
# Line 1716  Line 1735 
1735                                                          (pMB->dquant == NO_CHANGE);                                                          (pMB->dquant == NO_CHANGE);
1736    
1737                          if(pEnc->mbParam.m_quarterpel)                          if(pEnc->mbParam.m_quarterpel)
1738                                  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);
1739                            }
1740                          else                          else
1741                                  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);
1742                            }
1743    
1744                          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 */  
1745    
1746    /* This is a candidate for SKIPping, but for P-VOPs check intermediate B-frames first */
1747                                  int bSkip = 1;                                  int bSkip = 1;
                                 pMB->mode = MODE_NOT_CODED;  
1748    
1749                                    if (pEnc->current->coding_type == P_VOP)        /* special rule for P-VOP's SKIP */
1750                                  for (k=pEnc->bframenum_head; k< pEnc->bframenum_tail; k++)                                  for (k=pEnc->bframenum_head; k< pEnc->bframenum_tail; k++)
1751                                  {                                  {
1752                                          int iSAD;                                          int iSAD;
# Line 1737  Line 1758 
1758                                                  break;                                                  break;
1759                                          }                                          }
1760                                  }                                  }
1761    
1762                                  if (!bSkip)                                  if (!bSkip)
1763                                  {                                  {
1764                                          VECTOR predMV;                                          VECTOR predMV;
1765                                          if(pEnc->mbParam.m_quarterpel)                                          if(pEnc->mbParam.m_quarterpel) {
1766                                                  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);
1767                                          else                                                  pMB->pmvs[0].x = pMB->qmvs[0].x - predMV.x;  /* with GMC, qmvs doesn't have to be (0,0)! */
1768                                                    pMB->pmvs[0].y = pMB->qmvs[0].y - predMV.y;
1769                                            }
1770                                            else {
1771                                                  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);
1772                                          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)! */
1773                                                    pMB->pmvs[0].y = pMB->mvs[0].y - predMV.y;
1774                                            }
1775                                          pMB->mode = MODE_INTER;                                          pMB->mode = MODE_INTER;
1776                                          pMB->cbp = 0;                                          pMB->cbp = 0;
1777                                          MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->sStat);                                          MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->sStat);
1778                                  }                                  }
1779                                  else MBSkip(bs);                                  else
1780                                    {
1781                                            pMB->mode = MODE_NOT_CODED;
1782                                            MBSkip(bs);
1783                                    }
1784    
1785                          } else {                          } else {
1786                                  if (pEnc->current->global_flags & XVID_GREYSCALE)                                  if (pEnc->current->global_flags & XVID_GREYSCALE)
# Line 1811  Line 1842 
1842                  pEnc->current->quant = pEnc->reference->quant;                  pEnc->current->quant = pEnc->reference->quant;
1843                  pEnc->current->motion_flags = pEnc->reference->motion_flags;                  pEnc->current->motion_flags = pEnc->reference->motion_flags;
1844                  pEnc->current->rounding_type = pEnc->reference->rounding_type;                  pEnc->current->rounding_type = pEnc->reference->rounding_type;
1845                    pEnc->current->quarterpel =  pEnc->reference->quarterpel;
1846                  pEnc->current->fcode = pEnc->reference->fcode;                  pEnc->current->fcode = pEnc->reference->fcode;
1847                  pEnc->current->bcode = pEnc->reference->bcode;                  pEnc->current->bcode = pEnc->reference->bcode;
1848                  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 1882 
1882          }          }
1883  #endif  #endif
1884    
1885            frame->quarterpel =  pEnc->mbParam.m_quarterpel;
1886    
1887          // forward          // forward
1888          image_setedges(f_ref, pEnc->mbParam.edged_width,          image_setedges(f_ref, pEnc->mbParam.edged_width,
1889                                     pEnc->mbParam.edged_height, pEnc->mbParam.width,                                     pEnc->mbParam.edged_height, pEnc->mbParam.width,
# Line 1877  Line 1911 
1911                  ((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)),     // time_pp                  ((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)),     // time_pp
1912                          pEnc->reference->mbs, f_ref,                          pEnc->reference->mbs, f_ref,
1913                                                   &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,                                                   &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
1914                                                   pEnc->current->mbs, b_ref, &pEnc->vInterH,                                                   pEnc->current, b_ref, &pEnc->vInterH,
1915                                                   &pEnc->vInterV, &pEnc->vInterHV);                                                   &pEnc->vInterV, &pEnc->vInterHV);
1916    
1917    

Legend:
Removed from v.579  
changed lines
  Added in v.643

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