[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.95.2.60, Fri Dec 19 10:55:58 2003 UTC revision 1.102.2.5, Sun Apr 4 11:49:36 2004 UTC
# Line 57  Line 57 
57                                            Bitstream * bs);                                            Bitstream * bs);
58    
59  static int FrameCodeP(Encoder * pEnc,  static int FrameCodeP(Encoder * pEnc,
60                                            Bitstream * bs,                                            Bitstream * bs);
                                           bool force_inter,  
                                           bool vol_header);  
61    
62  static void FrameCodeB(Encoder * pEnc,  static void FrameCodeB(Encoder * pEnc,
63                                             FRAMEINFO * frame,                                             FRAMEINFO * frame,
# Line 123  Line 121 
121          if (create->width%2 || create->height%2)          if (create->width%2 || create->height%2)
122                  return XVID_ERR_FAIL;                  return XVID_ERR_FAIL;
123    
124            if (create->width<=0 || create->height<=0)
125                    return XVID_ERR_FAIL;
126    
127          /* allocate encoder struct */          /* allocate encoder struct */
128    
129          pEnc = (Encoder *) xvid_malloc(sizeof(Encoder), CACHE_LINE);          pEnc = (Encoder *) xvid_malloc(sizeof(Encoder), CACHE_LINE);
# Line 652  Line 653 
653                  data.max_quant[i] = pEnc->mbParam.max_quant[i];                  data.max_quant[i] = pEnc->mbParam.max_quant[i];
654          }          }
655    
656          data.reference.csp = XVID_CSP_USER;          data.reference.csp = XVID_CSP_PLANAR;
657          data.reference.plane[0] = pEnc->reference->image.y;          data.reference.plane[0] = pEnc->reference->image.y;
658          data.reference.plane[1] = pEnc->reference->image.u;          data.reference.plane[1] = pEnc->reference->image.u;
659          data.reference.plane[2] = pEnc->reference->image.v;          data.reference.plane[2] = pEnc->reference->image.v;
# Line 660  Line 661 
661          data.reference.stride[1] = pEnc->mbParam.edged_width/2;          data.reference.stride[1] = pEnc->mbParam.edged_width/2;
662          data.reference.stride[2] = pEnc->mbParam.edged_width/2;          data.reference.stride[2] = pEnc->mbParam.edged_width/2;
663    
664          data.current.csp = XVID_CSP_USER;          data.current.csp = XVID_CSP_PLANAR;
665          data.current.plane[0] = frame->image.y;          data.current.plane[0] = frame->image.y;
666          data.current.plane[1] = frame->image.u;          data.current.plane[1] = frame->image.u;
667          data.current.plane[2] = frame->image.v;          data.current.plane[2] = frame->image.v;
# Line 690  Line 691 
691    
692          } else { /* XVID_PLG_AFTER */          } else { /* XVID_PLG_AFTER */
693                  if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {                  if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
694                          data.original.csp = XVID_CSP_USER;                          data.original.csp = XVID_CSP_PLANAR;
695                          data.original.plane[0] = original->y;                          data.original.plane[0] = original->y;
696                          data.original.plane[1] = original->u;                          data.original.plane[1] = original->u;
697                          data.original.plane[2] = original->v;                          data.original.plane[2] = original->v;
# Line 1086  Line 1087 
1087                                  DPRINTF(XVID_DEBUG_DEBUG,"*** PFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i\n",                                  DPRINTF(XVID_DEBUG_DEBUG,"*** PFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i\n",
1088                                  pEnc->bframenum_head, pEnc->bframenum_tail,                                  pEnc->bframenum_head, pEnc->bframenum_tail,
1089                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
1090                                    pEnc->mbParam.frame_drop_ratio = -1; /* it must be a coded vop */
1091    
1092                                  FrameCodeP(pEnc, &bs, 1, 0);                                  FrameCodeP(pEnc, &bs);
1093    
1094    
1095                                  if ((pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED) && pEnc->bframenum_tail==0) {                                  if ((pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED) && pEnc->bframenum_tail==0) {
# Line 1121  Line 1123 
1123    
1124          pEnc->current->fincr = pEnc->mbParam.fincr>0 ? pEnc->mbParam.fincr : frame->fincr;          pEnc->current->fincr = pEnc->mbParam.fincr>0 ? pEnc->mbParam.fincr : frame->fincr;
1125          inc_frame_num(pEnc);          inc_frame_num(pEnc);
1126          pEnc->current->vol_flags = pEnc->mbParam.vol_flags;          pEnc->current->vol_flags = frame->vol_flags;
1127          pEnc->current->vop_flags = frame->vop_flags;          pEnc->current->vop_flags = frame->vop_flags;
1128          pEnc->current->motion_flags = frame->motion;          pEnc->current->motion_flags = frame->motion;
1129          pEnc->current->fcode = pEnc->mbParam.m_fcode;          pEnc->current->fcode = pEnc->mbParam.m_fcode;
# Line 1156  Line 1158 
1158                  }                  }
1159          }          }
1160    
1161            if (type != I_VOP)
1162                    pEnc->current->vol_flags = pEnc->mbParam.vol_flags; /* don't allow VOL changes here */
1163    
1164          /* bframes buffer overflow check */          /* bframes buffer overflow check */
1165          if (type == B_VOP && pEnc->bframenum_tail >= pEnc->mbParam.max_bframes) {          if (type == B_VOP && pEnc->bframenum_tail >= pEnc->mbParam.max_bframes) {
1166                  type = P_VOP;                  type = P_VOP;
# Line 1336  Line 1341 
1341                                     pEnc->mbParam.edged_width, pEnc->mbParam.height);                                     pEnc->mbParam.edged_width, pEnc->mbParam.height);
1342                  }                  }
1343    
1344                  FrameCodeP(pEnc, &bs, 1, 0);                  if ( FrameCodeP(pEnc, &bs) == 0 ) {
1345                            /* N-VOP, we mustn't code b-frames yet */
1346                            call_plugins(pEnc, pEnc->current, &pEnc->sOriginal, XVID_PLG_AFTER, 0, 0, stats);
1347                            goto done;
1348                    }
1349          }          }
1350    
1351    
# Line 1437  Line 1446 
1446                  start_timer();                  start_timer();
1447                  image_setedges(&pEnc->current->image,                  image_setedges(&pEnc->current->image,
1448                          pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,                          pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
1449                          pEnc->mbParam.width, pEnc->mbParam.height);                          pEnc->mbParam.width, pEnc->mbParam.height, 0);
1450                  stop_edges_timer();                  stop_edges_timer();
1451          }          }
1452    
# Line 1449  Line 1458 
1458    
1459          SetMacroblockQuants(&pEnc->mbParam, pEnc->current);          SetMacroblockQuants(&pEnc->mbParam, pEnc->current);
1460    
1461          BitstreamWriteVolHeader(bs, &pEnc->mbParam);          BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
1462    
1463          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);
1464    
# Line 1514  Line 1523 
1523  /* FrameCodeP also handles S(GMC)-VOPs */  /* FrameCodeP also handles S(GMC)-VOPs */
1524  static int  static int
1525  FrameCodeP(Encoder * pEnc,  FrameCodeP(Encoder * pEnc,
1526                     Bitstream * bs,                     Bitstream * bs)
                    bool force_inter,  
                    bool vol_header)  
1527  {  {
1528          float fSigma;          float fSigma;
1529          int bits = BitstreamPos(bs);          int bits = BitstreamPos(bs);
# Line 1524  Line 1531 
1531          DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);          DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
1532          DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);          DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);
1533    
         int iLimit;  
1534          int x, y, k;          int x, y, k;
1535          int iSearchRange;          int iSearchRange;
1536          int bIntra=0, skip_possible;          int skip_possible;
1537          FRAMEINFO *const current = pEnc->current;          FRAMEINFO *const current = pEnc->current;
1538          FRAMEINFO *const reference = pEnc->reference;          FRAMEINFO *const reference = pEnc->reference;
1539          MBParam * const pParam = &pEnc->mbParam;          MBParam * const pParam = &pEnc->mbParam;
1540          int mb_width = pParam->mb_width;          int mb_width = pParam->mb_width;
1541          int mb_height = pParam->mb_height;          int mb_height = pParam->mb_height;
1542            int coded = 1;
1543    
1544    
1545          /* IMAGE *pCurrent = &current->image; */          /* IMAGE *pCurrent = &current->image; */
# Line 1548  Line 1555 
1555          if (!reference->is_edged) {          if (!reference->is_edged) {
1556                  start_timer();                  start_timer();
1557                  image_setedges(pRef, pParam->edged_width, pParam->edged_height,                  image_setedges(pRef, pParam->edged_width, pParam->edged_height,
1558                                             pParam->width, pParam->height);                                             pParam->width, pParam->height, 0);
1559                  stop_edges_timer();                  stop_edges_timer();
1560                  reference->is_edged = 1;                  reference->is_edged = 1;
1561          }          }
# Line 1557  Line 1564 
1564          current->rounding_type = pParam->m_rounding_type;          current->rounding_type = pParam->m_rounding_type;
1565          current->fcode = pParam->m_fcode;          current->fcode = pParam->m_fcode;
1566    
         if (!force_inter)  
                 iLimit = (int)(mb_width * mb_height *  INTRA_THRESHOLD);  
         else  
                 iLimit = mb_width * mb_height + 1;  
   
1567          if ((current->vop_flags & XVID_VOP_HALFPEL)) {          if ((current->vop_flags & XVID_VOP_HALFPEL)) {
1568                  if (reference->is_interpolated != current->rounding_type) {                  if (reference->is_interpolated != current->rounding_type) {
1569                          start_timer();                          start_timer();
# Line 1632  Line 1634 
1634                  }                  }
1635          }          }
1636    
         bIntra =  
1637                  MotionEstimation(&pEnc->mbParam, current, reference,                  MotionEstimation(&pEnc->mbParam, current, reference,
1638                                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,                                           &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
1639                                           &pEnc->vGMC, iLimit);                                           &pEnc->vGMC, 256*4096);
1640    
1641    
1642          stop_motion_timer();          stop_motion_timer();
1643    
         if (bIntra == 1) return FrameCodeI(pEnc, bs);  
   
1644          set_timecodes(current,reference,pParam->fbase);          set_timecodes(current,reference,pParam->fbase);
         if (vol_header)  
         {       BitstreamWriteVolHeader(bs, &pEnc->mbParam);  
                 BitstreamPad(bs);  
         }  
1645    
1646          BitstreamWriteVopHeader(bs, &pEnc->mbParam, current, 1, current->mbs[0].quant);          BitstreamWriteVopHeader(bs, &pEnc->mbParam, current, 1, current->mbs[0].quant);
1647    
# Line 1659  Line 1654 
1654                          MACROBLOCK *pMB =                          MACROBLOCK *pMB =
1655                                  &current->mbs[x + y * pParam->mb_width];                                  &current->mbs[x + y * pParam->mb_width];
1656    
1657                          bIntra = (pMB->mode == MODE_INTRA) || (pMB->mode == MODE_INTRA_Q);                          int bIntra = (pMB->mode == MODE_INTRA) || (pMB->mode == MODE_INTRA_Q);
1658    
1659                          if (bIntra) {                          if (bIntra) {
1660                                  CodeIntraMB(pEnc, pMB);                                  CodeIntraMB(pEnc, pMB);
# Line 1860  Line 1855 
1855  #if 0  #if 0
1856          DPRINTF(XVID_DEBUG_DEBUG, "kmu %i %i %i\n", current->sStat.kblks, current->sStat.mblks, current->sStat.ublks);          DPRINTF(XVID_DEBUG_DEBUG, "kmu %i %i %i\n", current->sStat.kblks, current->sStat.mblks, current->sStat.ublks);
1857  #endif  #endif
1858          if (current->sStat.kblks + current->sStat.mblks <          if (current->sStat.kblks + current->sStat.mblks <=
1859                  (pParam->frame_drop_ratio * mb_width * mb_height) / 100)                  (pParam->frame_drop_ratio * mb_width * mb_height) / 100 &&
1860                    pEnc->mbParam.max_bframes == 0)
1861          {          {
1862                  current->sStat.kblks = current->sStat.mblks = 0;                  current->sStat.kblks = current->sStat.mblks = 0;
1863                  current->sStat.ublks = mb_width * mb_height;                  current->sStat.ublks = mb_width * mb_height;
# Line 1877  Line 1873 
1873                  current->rounding_type = reference->rounding_type;                  current->rounding_type = reference->rounding_type;
1874                  current->fcode = reference->fcode;                  current->fcode = reference->fcode;
1875                  current->bcode = reference->bcode;                  current->bcode = reference->bcode;
1876                    current->stamp = reference->stamp;
1877                  image_copy(&current->image, &reference->image, pParam->edged_width, pParam->height);                  image_copy(&current->image, &reference->image, pParam->edged_width, pParam->height);
1878                  memcpy(current->mbs, reference->mbs, sizeof(MACROBLOCK) * mb_width * mb_height);                  memcpy(current->mbs, reference->mbs, sizeof(MACROBLOCK) * mb_width * mb_height);
1879          }                  coded = 0;
1880    
1881            } else {
1882    
1883          pEnc->current->is_edged = 0; /* not edged */          pEnc->current->is_edged = 0; /* not edged */
1884          pEnc->current->is_interpolated = -1; /* not interpolated (fake rounding -1) */          pEnc->current->is_interpolated = -1; /* not interpolated (fake rounding -1) */
# Line 1890  Line 1889 
1889          image_swap(&pEnc->vInterH, &pEnc->f_refh);          image_swap(&pEnc->vInterH, &pEnc->f_refh);
1890          image_swap(&pEnc->vInterV, &pEnc->f_refv);          image_swap(&pEnc->vInterV, &pEnc->f_refv);
1891          image_swap(&pEnc->vInterHV, &pEnc->f_refhv);          image_swap(&pEnc->vInterHV, &pEnc->f_refhv);
1892            }
1893    
1894          /* XXX: debug          /* XXX: debug
1895          {          {
# Line 1911  Line 1910 
1910    
1911          current->length = (BitstreamPos(bs) - bits) / 8;          current->length = (BitstreamPos(bs) - bits) / 8;
1912    
1913          return 0;                                       /* inter */          return coded;
1914  }  }
1915    
1916    
# Line 1946  Line 1945 
1945          if (!pEnc->reference->is_edged) {          if (!pEnc->reference->is_edged) {
1946                  image_setedges(f_ref, pEnc->mbParam.edged_width,                  image_setedges(f_ref, pEnc->mbParam.edged_width,
1947                                             pEnc->mbParam.edged_height, pEnc->mbParam.width,                                             pEnc->mbParam.edged_height, pEnc->mbParam.width,
1948                                             pEnc->mbParam.height);                                             pEnc->mbParam.height, 0);
1949                  pEnc->current->is_edged = 1;                  pEnc->current->is_edged = 1;
1950          }          }
1951    
# Line 1963  Line 1962 
1962          if (!pEnc->current->is_edged) {          if (!pEnc->current->is_edged) {
1963                  image_setedges(b_ref, pEnc->mbParam.edged_width,                  image_setedges(b_ref, pEnc->mbParam.edged_width,
1964                                             pEnc->mbParam.edged_height, pEnc->mbParam.width,                                             pEnc->mbParam.edged_height, pEnc->mbParam.width,
1965                                             pEnc->mbParam.height);                                             pEnc->mbParam.height, 0);
1966                  pEnc->current->is_edged = 1;                  pEnc->current->is_edged = 1;
1967          }          }
1968    
Legend:
Removed from v.1.95.2.60  
changed lines
  Added in v.1.102.2.5
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4