[svn] / trunk / xvidcore / src / motion / estimation_bvop.c Repository:
ViewVC logotype

Diff of /trunk/xvidcore/src/motion/estimation_bvop.c

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

revision 1475, Sat Jul 3 08:46:39 2004 UTC revision 1478, Thu Jul 8 07:12:54 2004 UTC
# Line 21  Line 21 
21   *  along with this program ; if not, write to the Free Software   *  along with this program ; if not, write to the Free Software
22   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
23   *   *
24   * $Id: estimation_bvop.c,v 1.6 2004-07-03 08:46:39 syskin Exp $   * $Id: estimation_bvop.c,v 1.7 2004-07-08 07:12:54 syskin Exp $
25   *   *
26   ****************************************************************************/   ****************************************************************************/
27    
# Line 164  Line 164 
164          const uint8_t *ReferenceF;          const uint8_t *ReferenceF;
165          const uint8_t *ReferenceB;          const uint8_t *ReferenceB;
166          VECTOR mvs, b_mvs;          VECTOR mvs, b_mvs;
167            const int blocks[4] = {0, 8, 8*data->iEdgedWidth, 8*data->iEdgedWidth+8};
168    
169          if (( x > 31) || ( x < -32) || ( y > 31) || (y < -32)) return;          if (( x > 31) || ( x < -32) || ( y > 31) || (y < -32)) return;
170    
# Line 187  Line 188 
188                  if (data->qpel) {                  if (data->qpel) {
189                          xcf += mvs.x/2; ycf += mvs.y/2;                          xcf += mvs.x/2; ycf += mvs.y/2;
190                          xcb += b_mvs.x/2; ycb += b_mvs.y/2;                          xcb += b_mvs.x/2; ycb += b_mvs.y/2;
191                            if (data->qpel_precision) {
192                                    ReferenceF = xvid_me_interpolate8x8qpel(mvs.x, mvs.y, k, 0, data);
193                                    ReferenceB = xvid_me_interpolate8x8qpel(b_mvs.x, b_mvs.y, k, 1, data);
194                                    goto done;
195                            }
196                            mvs.x >>=1; mvs.y >>=1; b_mvs.x >>=1; b_mvs.y >>=1; // qpel->hpel
197                  } else {                  } else {
198                          xcf += mvs.x; ycf += mvs.y;                          xcf += mvs.x; ycf += mvs.y;
199                          xcb += b_mvs.x; ycb += b_mvs.y;                          xcb += b_mvs.x; ycb += b_mvs.y;
                         mvs.x *= 2; mvs.y *= 2; /* we move to qpel precision anyway */  
                         b_mvs.x *= 2; b_mvs.y *= 2;  
200                  }                  }
201                    ReferenceF = GetReference(mvs.x, mvs.y, data) + blocks[k];
202                  ReferenceF = xvid_me_interpolate8x8qpel(mvs.x, mvs.y, k, 0, data);                  ReferenceB = GetReferenceB(b_mvs.x, b_mvs.y, 1, data) + blocks[k];
203                  ReferenceB = xvid_me_interpolate8x8qpel(b_mvs.x, b_mvs.y, k, 1, data);  done:
   
204                  sad += data->iMinSAD[k+1] =                  sad += data->iMinSAD[k+1] =
205                          sad8bi(data->Cur + 8*(k&1) + 8*(k>>1)*(data->iEdgedWidth),                          sad8bi(data->Cur + blocks[k],
206                                          ReferenceF, ReferenceB, data->iEdgedWidth);                                          ReferenceF, ReferenceB, data->iEdgedWidth);
207                  if (sad > *(data->iMinSAD)) return;                  if (sad > *(data->iMinSAD)) return;
208          }          }
# Line 246  Line 250 
250          if (data->qpel) {          if (data->qpel) {
251                  xcf = 4*(mvs.x/2); ycf = 4*(mvs.y/2);                  xcf = 4*(mvs.x/2); ycf = 4*(mvs.y/2);
252                  xcb = 4*(b_mvs.x/2); ycb = 4*(b_mvs.y/2);                  xcb = 4*(b_mvs.x/2); ycb = 4*(b_mvs.y/2);
253                    if (data->qpel_precision) {
254                  ReferenceF = xvid_me_interpolate16x16qpel(mvs.x, mvs.y, 0, data);                  ReferenceF = xvid_me_interpolate16x16qpel(mvs.x, mvs.y, 0, data);
255                  ReferenceB = xvid_me_interpolate16x16qpel(b_mvs.x, b_mvs.y, 1, data);                  ReferenceB = xvid_me_interpolate16x16qpel(b_mvs.x, b_mvs.y, 1, data);
256                            goto done;
257                    }
258                    mvs.x >>=1; mvs.y >>=1; b_mvs.x >>=1; b_mvs.y >>=1; // qpel->hpel
259          } else {          } else {
260                  xcf = 4*mvs.x; ycf = 4*mvs.y;                  xcf = 4*mvs.x; ycf = 4*mvs.y;
261                  xcb = 4*b_mvs.x; ycb = 4*b_mvs.y;                  xcb = 4*b_mvs.x; ycb = 4*b_mvs.y;
262            }
263                  ReferenceF = GetReference(mvs.x, mvs.y, data);                  ReferenceF = GetReference(mvs.x, mvs.y, data);
264                  ReferenceB = GetReferenceB(b_mvs.x, b_mvs.y, 1, data);                  ReferenceB = GetReferenceB(b_mvs.x, b_mvs.y, 1, data);
         }  
265    
266    done:
267          sad = sad16bi(data->Cur, ReferenceF, ReferenceB, data->iEdgedWidth);          sad = sad16bi(data->Cur, ReferenceF, ReferenceB, data->iEdgedWidth);
268          sad += (data->lambda16 * d_mv_bits(x, y, zeroMV, 1, 0, 0) * sad)>>10;          sad += (data->lambda16 * d_mv_bits(x, y, zeroMV, 1, 0, 0) * sad)>>10;
269    
# Line 310  Line 319 
319          }          }
320  }  }
321    
322    
323    static void
324    initialize_searchData(SearchData * Data_d,
325                                              SearchData * Data_f,
326                                              SearchData * Data_b,
327                                              SearchData * Data_i,
328                                              int x, int y,
329                                              const IMAGE * const f_Ref,
330                                              const uint8_t * const f_RefH,
331                                              const uint8_t * const f_RefV,
332                                              const uint8_t * const f_RefHV,
333                                              const IMAGE * const b_Ref,
334                                              const uint8_t * const b_RefH,
335                                              const uint8_t * const b_RefV,
336                                              const uint8_t * const b_RefHV,
337                                              const IMAGE * const pCur,
338                                              const MACROBLOCK * const b_mb)
339    {
340    
341            /* per-macroblock SearchData initialization - too many things would be repeated 4 times */
342            const uint8_t * RefP[6], * b_RefP[6], * Cur[3];
343            const uint32_t iEdgedWidth = Data_d->iEdgedWidth;
344            unsigned int lambda;
345            int i;
346    
347            /* luma */
348            int offset = (x + iEdgedWidth*y) * 16;
349            RefP[0] = f_Ref->y + offset;
350            RefP[2] = f_RefH + offset;
351            RefP[1] = f_RefV + offset;
352            RefP[3] = f_RefHV + offset;
353            b_RefP[0] = b_Ref->y + offset;
354            b_RefP[2] = b_RefH + offset;
355            b_RefP[1] = b_RefV + offset;
356            b_RefP[3] = b_RefHV + offset;
357            Cur[0] = pCur->y + offset;
358    
359            /* chroma */
360            offset = (x + (iEdgedWidth/2)*y) * 8;
361            RefP[4] = f_Ref->u + offset;
362            RefP[5] = f_Ref->v + offset;
363            b_RefP[4] = b_Ref->u + offset;
364            b_RefP[5] = b_Ref->v + offset;
365            Cur[1] = pCur->u + offset;
366            Cur[2] = pCur->v + offset;
367    
368            lambda = xvid_me_lambda_vec16[b_mb->quant];
369    
370            for (i = 0; i < 6; i++) {
371                    Data_d->RefP[i] = Data_f->RefP[i] = Data_i->RefP[i] = RefP[i];
372                    Data_d->b_RefP[i] = Data_b->RefP[i] = Data_i->b_RefP[i] = b_RefP[i];
373            }
374            Data_d->Cur = Data_f->Cur = Data_b->Cur = Data_i->Cur = Cur[0];
375            Data_d->CurU = Data_f->CurU = Data_b->CurU = Data_i->CurU = Cur[1];
376            Data_d->CurV = Data_f->CurV = Data_b->CurV = Data_i->CurV = Cur[2];
377    
378            Data_d->lambda16 = lambda/4;
379            Data_f->lambda16 = Data_b->lambda16 = Data_i->lambda16 = lambda;
380    
381            /* reset chroma-sad cache */
382            Data_d->b_chromaX = Data_d->b_chromaY = Data_d->chromaX = Data_d->chromaY = Data_d->chromaSAD = 256*4096;
383            Data_i->b_chromaX = Data_i->b_chromaY = Data_i->chromaX = Data_i->chromaY = Data_i->chromaSAD = 256*4096;
384            Data_f->chromaX = Data_f->chromaY = Data_f->chromaSAD = 256*4096;
385            Data_b->chromaX = Data_b->chromaY = Data_b->chromaSAD = 256*4096;
386    
387            *Data_d->iMinSAD = *Data_b->iMinSAD = *Data_f->iMinSAD = *Data_i->iMinSAD = 4096*256;
388    }
389    
390  static __inline VECTOR  static __inline VECTOR
391  ChoosePred(const MACROBLOCK * const pMB, const uint32_t mode)  ChoosePred(const MACROBLOCK * const pMB, const uint32_t mode)
392  {  {
# Line 321  Line 398 
398  PreparePredictionsBF(VECTOR * const pmv, const int x, const int y,  PreparePredictionsBF(VECTOR * const pmv, const int x, const int y,
399                                                          const uint32_t iWcount,                                                          const uint32_t iWcount,
400                                                          const MACROBLOCK * const pMB,                                                          const MACROBLOCK * const pMB,
401                                                          const uint32_t mode_curr)                                                          const uint32_t mode_curr,
402                                                            const VECTOR hint)
403  {  {
404    
405          /* [0] is prediction */          /* [0] is prediction */
406          pmv[0].x = EVEN(pmv[0].x); pmv[0].y = EVEN(pmv[0].y);          pmv[0].x = (pmv[0].x); pmv[0].y = (pmv[0].y);
407    
408          pmv[1].x = pmv[1].y = 0; /* [1] is zero */          pmv[1].x = pmv[1].y = 0; /* [1] is zero */
409    
410          pmv[2] = ChoosePred(pMB, mode_curr);          pmv[2].x = hint.x; pmv[2].y = hint.y;
         pmv[2].x = EVEN(pmv[2].x); pmv[2].y = EVEN(pmv[2].y);  
411    
412          if ((y != 0)&&(x != (int)(iWcount+1))) {                        /* [3] top-right neighbour */          if ((y != 0)&&(x != (int)(iWcount+1))) {                        /* [3] top-right neighbour */
413                  pmv[3] = ChoosePred(pMB+1-iWcount, mode_curr);                  pmv[3] = ChoosePred(pMB+1-iWcount, mode_curr);
                 pmv[3].x = EVEN(pmv[3].x); pmv[3].y = EVEN(pmv[3].y);  
414          } else pmv[3].x = pmv[3].y = 0;          } else pmv[3].x = pmv[3].y = 0;
415    
416          if (y != 0) {          if (y != 0) {
417                  pmv[4] = ChoosePred(pMB-iWcount, mode_curr);                  pmv[4] = ChoosePred(pMB-iWcount, mode_curr);
                 pmv[4].x = EVEN(pmv[4].x); pmv[4].y = EVEN(pmv[4].y);  
418          } else pmv[4].x = pmv[4].y = 0;          } else pmv[4].x = pmv[4].y = 0;
419    
420          if (x != 0) {          if (x != 0) {
421                  pmv[5] = ChoosePred(pMB-1, mode_curr);                  pmv[5] = ChoosePred(pMB-1, mode_curr);
                 pmv[5].x = EVEN(pmv[5].x); pmv[5].y = EVEN(pmv[5].y);  
422          } else pmv[5].x = pmv[5].y = 0;          } else pmv[5].x = pmv[5].y = 0;
423    
424          if (x != 0 && y != 0) {          if (x != 0 && y != 0) {
425                  pmv[6] = ChoosePred(pMB-1-iWcount, mode_curr);                  pmv[6] = ChoosePred(pMB-1-iWcount, mode_curr);
                 pmv[6].x = EVEN(pmv[6].x); pmv[6].y = EVEN(pmv[6].y);  
426          } else pmv[6].x = pmv[6].y = 0;          } else pmv[6].x = pmv[6].y = 0;
427  }  }
428    
   
429  /* search backward or forward */  /* search backward or forward */
430  static void  static void
431  SearchBF(       const IMAGE * const pRef,  SearchBF_initial(const int x, const int y,
                         const uint8_t * const pRefH,  
                         const uint8_t * const pRefV,  
                         const uint8_t * const pRefHV,  
                         const int x, const int y,  
432                          const uint32_t MotionFlags,                          const uint32_t MotionFlags,
433                          const uint32_t iFcode,                          const uint32_t iFcode,
434                          const MBParam * const pParam,                          const MBParam * const pParam,
# Line 368  Line 436 
436                          const VECTOR * const predMV,                          const VECTOR * const predMV,
437                          int32_t * const best_sad,                          int32_t * const best_sad,
438                          const int32_t mode_current,                          const int32_t mode_current,
439                          SearchData * const Data)                          SearchData * const Data,
440                            VECTOR hint)
441  {  {
442    
443          int i;          int i;
444          VECTOR pmv[7];          VECTOR pmv[7];
         int threshA = (MotionFlags & XVID_ME_FASTREFINE16) ? 150 : 300;  
445          *Data->iMinSAD = MV_MAX_ERROR;          *Data->iMinSAD = MV_MAX_ERROR;
         Data->iFcode = iFcode;  
446          Data->qpel_precision = 0;          Data->qpel_precision = 0;
         Data->chromaX = Data->chromaY = Data->chromaSAD = 256*4096; /* reset chroma-sad cache */  
   
         Data->RefP[0] = pRef->y + (x + Data->iEdgedWidth*y) * 16;  
         Data->RefP[2] = pRefH + (x + Data->iEdgedWidth*y) * 16;  
         Data->RefP[1] = pRefV + (x + Data->iEdgedWidth*y) * 16;  
         Data->RefP[3] = pRefHV + (x + Data->iEdgedWidth*y) * 16;  
         Data->RefP[4] = pRef->u + (x + y * (Data->iEdgedWidth/2)) * 8;  
         Data->RefP[5] = pRef->v + (x + y * (Data->iEdgedWidth/2)) * 8;  
447    
448          Data->predMV = *predMV;          Data->predMV = *predMV;
449    
# Line 392  Line 451 
451                                  pParam->width, pParam->height, iFcode - Data->qpel, 1, 0);                                  pParam->width, pParam->height, iFcode - Data->qpel, 1, 0);
452    
453          pmv[0] = Data->predMV;          pmv[0] = Data->predMV;
454          if (Data->qpel) { pmv[0].x /= 2; pmv[0].y /= 2; }          if (Data->qpel) {
455                    pmv[0].x /= 2; pmv[0].y /= 2;
456                    hint.x /= 2; hint.y /= 2;
457            }
458    
459          PreparePredictionsBF(pmv, x, y, pParam->mb_width, pMB, mode_current);          PreparePredictionsBF(pmv, x, y, pParam->mb_width, pMB, mode_current, hint);
460    
461          Data->currentMV->x = Data->currentMV->y = 0;          Data->currentMV->x = Data->currentMV->y = 0;
462    
# Line 414  Line 476 
476                  MainSearchPtr(Data->currentMV->x, Data->currentMV->y, Data, mask, CheckCandidate16no4v);                  MainSearchPtr(Data->currentMV->x, Data->currentMV->y, Data, mask, CheckCandidate16no4v);
477          }          }
478    
479            if (Data->iMinSAD[0] < *best_sad) *best_sad = Data->iMinSAD[0];
480    }
481    
482    static void
483    SearchBF_final(const int x, const int y,
484                            const uint32_t MotionFlags,
485                            const MBParam * const pParam,
486                            int32_t * const best_sad,
487                            SearchData * const Data)
488    {
489          if(!Data->qpel) {          if(!Data->qpel) {
490                  /* halfpel mode */                  /* halfpel mode */
491                  if (MotionFlags & XVID_ME_HALFPELREFINE16)                  if (MotionFlags & XVID_ME_HALFPELREFINE16)
492                                  xvid_me_SubpelRefine(Data, CheckCandidate16no4v, 0);                                  xvid_me_SubpelRefine(Data->currentMV[0], Data, CheckCandidate16no4v, 0);
493          } else {          } else {
494                  /* qpel mode */                  /* qpel mode */
495                  if(MotionFlags & XVID_ME_FASTREFINE16) {                  if(MotionFlags & XVID_ME_FASTREFINE16) {
496                          /* fast */                          /* fast */
497                          get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 4,                          get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 4,
498                                                  pParam->width, pParam->height, Data->iFcode, 2, 0);                                                  pParam->width, pParam->height, Data->iFcode, 2, 0);
                         if (*Data->iMinSAD < *best_sad + 2*threshA)  
499                                  FullRefine_Fast(Data, CheckCandidate16no4v, 0);                                  FullRefine_Fast(Data, CheckCandidate16no4v, 0);
500    
501                  } else {                  } else {
502    
503                          Data->currentQMV->x = 2*Data->currentMV->x;                          Data->currentQMV->x = 2*Data->currentMV->x;
504                          Data->currentQMV->y = 2*Data->currentMV->y;                          Data->currentQMV->y = 2*Data->currentMV->y;
505                          if(MotionFlags & XVID_ME_QUARTERPELREFINE16) {                          if(MotionFlags & XVID_ME_QUARTERPELREFINE16) {
506                                  /* full */                                  /* full */
507                                  if (MotionFlags & XVID_ME_HALFPELREFINE16) {                                  if (MotionFlags & XVID_ME_HALFPELREFINE16) {
508                                          xvid_me_SubpelRefine(Data, CheckCandidate16no4v, 0); /* hpel part */                                          xvid_me_SubpelRefine(Data->currentMV[0], Data, CheckCandidate16no4v, 0); /* hpel part */
509                                          Data->currentQMV->x = 2*Data->currentMV->x;                                          Data->currentQMV->x = 2*Data->currentMV->x;
510                                          Data->currentQMV->y = 2*Data->currentMV->y;                                          Data->currentQMV->y = 2*Data->currentMV->y;
511                                  }                                  }
512                                  get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 4,                                  get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 4,
513                                                          pParam->width, pParam->height, Data->iFcode, 2, 0);                                                          pParam->width, pParam->height, Data->iFcode, 2, 0);
514                                  Data->qpel_precision = 1;                                  Data->qpel_precision = 1;
515                                  if (*Data->iMinSAD < *best_sad + threshA)                                  xvid_me_SubpelRefine(Data->currentQMV[0], Data, CheckCandidate16no4v, 0); /* qpel part */
                                         xvid_me_SubpelRefine(Data, CheckCandidate16no4v, 0); /* qpel part */  
                         }  
516                  }                  }
517          }          }
   
         /* three bits are needed to code backward mode. four for forward */  
   
         if (mode_current == MODE_FORWARD) *Data->iMinSAD += 4 * Data->lambda16;  
         else *Data->iMinSAD += 3 * Data->lambda16;  
   
         if (*Data->iMinSAD < *best_sad) {  
                 *best_sad = *Data->iMinSAD;  
                 pMB->mode = mode_current;  
                 if (Data->qpel) {  
                         pMB->pmvs[0].x = Data->currentQMV->x - predMV->x;  
                         pMB->pmvs[0].y = Data->currentQMV->y - predMV->y;  
                         if (mode_current == MODE_FORWARD)  
                                 pMB->qmvs[0] = *Data->currentQMV;  
                         else  
                                 pMB->b_qmvs[0] = *Data->currentQMV;  
                 } else {  
                         pMB->pmvs[0].x = Data->currentMV->x - predMV->x;  
                         pMB->pmvs[0].y = Data->currentMV->y - predMV->y;  
                 }  
                 if (mode_current == MODE_FORWARD) pMB->mvs[0] = *Data->currentMV;  
                 else pMB->b_mvs[0] = *Data->currentMV;  
518          }          }
519            if (Data->iMinSAD[0] < *best_sad) *best_sad = Data->iMinSAD[0];
520    
         if (mode_current == MODE_FORWARD) *(Data->currentMV+2) = *Data->currentMV;  
         else *(Data->currentMV+1) = *Data->currentMV; /* we store currmv for interpolate search */  
521  }  }
522    
523  static void  static void
524  SkipDecisionB(const IMAGE * const pCur,  SkipDecisionB(MACROBLOCK * const pMB, const SearchData * const Data)
                                 const IMAGE * const f_Ref,  
                                 const IMAGE * const b_Ref,  
                                 MACROBLOCK * const pMB,  
                                 const uint32_t x, const uint32_t y,  
                                 const SearchData * const Data)  
525  {  {
526          int k;          int k;
527    
# Line 503  Line 543 
543                  b_dy = (b_dy >> 3) + roundtab_76[b_dy & 0xf];                  b_dy = (b_dy >> 3) + roundtab_76[b_dy & 0xf];
544                  b_dx = (b_dx >> 3) + roundtab_76[b_dx & 0xf];                  b_dx = (b_dx >> 3) + roundtab_76[b_dx & 0xf];
545    
546                  sum = sad8bi(pCur->u + 8 * x + 8 * y * stride,                  sum = sad8bi(Data->CurU,
547                                                  f_Ref->u + (y*8 + dy/2) * stride + x*8 + dx/2,                                                  Data->RefP[4] + (dy/2) * stride + dx/2,
548                                                  b_Ref->u + (y*8 + b_dy/2) * stride + x*8 + b_dx/2,                                                  Data->b_RefP[4] + (b_dy/2) * stride + b_dx/2,
549                                                  stride);                                                  stride);
550    
551                  if (sum >= MAX_CHROMA_SAD_FOR_SKIP * (int)Data->iQuant) return; /* no skip */                  if (sum >= MAX_CHROMA_SAD_FOR_SKIP * (int)Data->iQuant) return; /* no skip */
552    
553                  sum += sad8bi(pCur->v + 8*x + 8 * y * stride,                  sum += sad8bi(Data->CurV,
554                                                  f_Ref->v + (y*8 + dy/2) * stride + x*8 + dx/2,                                                  Data->RefP[5] + (dy/2) * stride + dx/2,
555                                                  b_Ref->v + (y*8 + b_dy/2) * stride + x*8 + b_dx/2,                                                  Data->b_RefP[5] + (b_dy/2) * stride + b_dx/2,
556                                                  stride);                                                  stride);
557    
558                  if (sum >= MAX_CHROMA_SAD_FOR_SKIP * (int)Data->iQuant) return; /* no skip */                  if (sum >= MAX_CHROMA_SAD_FOR_SKIP * (int)Data->iQuant) return; /* no skip */
# Line 527  Line 567 
567  }  }
568    
569  static uint32_t  static uint32_t
570  SearchDirect(const IMAGE * const f_Ref,  SearchDirect_initial(const int x, const int y,
                                 const uint8_t * const f_RefH,  
                                 const uint8_t * const f_RefV,  
                                 const uint8_t * const f_RefHV,  
                                 const IMAGE * const b_Ref,  
                                 const uint8_t * const b_RefH,  
                                 const uint8_t * const b_RefV,  
                                 const uint8_t * const b_RefHV,  
                                 const IMAGE * const pCur,  
                                 const int x, const int y,  
571                                  const uint32_t MotionFlags,                                  const uint32_t MotionFlags,
572                                  const int32_t TRB, const int32_t TRD,                                  const int32_t TRB, const int32_t TRD,
573                                  const MBParam * const pParam,                                  const MBParam * const pParam,
# Line 548  Line 579 
579  {  {
580          int32_t skip_sad;          int32_t skip_sad;
581          int k = (x + Data->iEdgedWidth*y) * 16;          int k = (x + Data->iEdgedWidth*y) * 16;
         MainSearchFunc *MainSearchPtr;  
         CheckFunc * CheckCandidate;  
   
         *Data->iMinSAD = 256*4096;  
         Data->RefP[0] = f_Ref->y + k;  
         Data->RefP[2] = f_RefH + k;  
         Data->RefP[1] = f_RefV + k;  
         Data->RefP[3] = f_RefHV + k;  
         Data->b_RefP[0] = b_Ref->y + k;  
         Data->b_RefP[2] = b_RefH + k;  
         Data->b_RefP[1] = b_RefV + k;  
         Data->b_RefP[3] = b_RefHV + k;  
         Data->RefP[4] = f_Ref->u + (x + (Data->iEdgedWidth/2) * y) * 8;  
         Data->RefP[5] = f_Ref->v + (x + (Data->iEdgedWidth/2) * y) * 8;  
         Data->b_RefP[4] = b_Ref->u + (x + (Data->iEdgedWidth/2) * y) * 8;  
         Data->b_RefP[5] = b_Ref->v + (x + (Data->iEdgedWidth/2) * y) * 8;  
582    
583          k = Data->qpel ? 4 : 2;          k = Data->qpel ? 4 : 2;
584          Data->max_dx = k * (pParam->width - x * 16);          Data->max_dx = k * (pParam->width - x * 16);
585          Data->max_dy = k * (pParam->height - y * 16);          Data->max_dy = k * (pParam->height - y * 16);
586          Data->min_dx = -k * (16 + x * 16);          Data->min_dx = -k * (16 + x * 16);
587          Data->min_dy = -k * (16 + y * 16);          Data->min_dy = -k * (16 + y * 16);
         Data->chromaX = Data->chromaY = Data->b_chromaX = Data->b_chromaY = 256*4096;  
588    
589          Data->referencemv = Data->qpel ? b_mb->qmvs : b_mb->mvs;          Data->referencemv = Data->qpel ? b_mb->qmvs : b_mb->mvs;
         Data->qpel_precision = 0;  
590    
591          for (k = 0; k < 4; k++) {          for (k = 0; k < 4; k++) {
592                  pMB->mvs[k].x = Data->directmvF[k].x = ((TRB * Data->referencemv[k].x) / TRD);                  Data->directmvF[k].x = ((TRB * Data->referencemv[k].x) / TRD);
593                  pMB->b_mvs[k].x = Data->directmvB[k].x = ((TRB - TRD) * Data->referencemv[k].x) / TRD;                  Data->directmvB[k].x = ((TRB - TRD) * Data->referencemv[k].x) / TRD;
594                  pMB->mvs[k].y = Data->directmvF[k].y = ((TRB * Data->referencemv[k].y) / TRD);                  Data->directmvF[k].y = ((TRB * Data->referencemv[k].y) / TRD);
595                  pMB->b_mvs[k].y = Data->directmvB[k].y = ((TRB - TRD) * Data->referencemv[k].y) / TRD;                  Data->directmvB[k].y = ((TRB - TRD) * Data->referencemv[k].y) / TRD;
596    
597                  if ( (pMB->b_mvs[k].x > Data->max_dx) | (pMB->b_mvs[k].x < Data->min_dx)                  if ( (Data->directmvB[k].x > Data->max_dx) | (Data->directmvB[k].x < Data->min_dx)
598                          | (pMB->b_mvs[k].y > Data->max_dy) | (pMB->b_mvs[k].y < Data->min_dy) ) {                          | (Data->directmvB[k].y > Data->max_dy) | (Data->directmvB[k].y < Data->min_dy) ) {
599    
600                          *best_sad = 256*4096; /* in that case, we won't use direct mode */                          Data->iMinSAD[0] = *best_sad = 256*4096; /* in that case, we won't use direct mode */
                         pMB->mode = MODE_DIRECT; /* just to make sure it doesn't say "MODE_DIRECT_NONE_MV" */  
                         pMB->b_mvs[0].x = pMB->b_mvs[0].y = 0;  
601                          return 256*4096;                          return 256*4096;
602                  }                  }
603                  if (b_mb->mode != MODE_INTER4V) {                  if (b_mb->mode != MODE_INTER4V) {
                         pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = pMB->mvs[0];  
                         pMB->b_mvs[1] = pMB->b_mvs[2] = pMB->b_mvs[3] = pMB->b_mvs[0];  
604                          Data->directmvF[1] = Data->directmvF[2] = Data->directmvF[3] = Data->directmvF[0];                          Data->directmvF[1] = Data->directmvF[2] = Data->directmvF[3] = Data->directmvF[0];
605                          Data->directmvB[1] = Data->directmvB[2] = Data->directmvB[3] = Data->directmvB[0];                          Data->directmvB[1] = Data->directmvB[2] = Data->directmvB[3] = Data->directmvB[0];
606                          break;                          break;
607                  }                  }
608          }          }
609            Data->qpel_precision = Data->qpel; /* this initial check is done with full precision, to find real
610                                                                                            SKIP sad */
611    
612          CheckCandidateDirect(0, 0, Data, 255);  /* will also fill iMinSAD[1..4] with 8x8 SADs */          CheckCandidateDirect(0, 0, Data, 255);  /* will also fill iMinSAD[1..4] with 8x8 SADs */
613    
614    
615          /* initial (fast) skip decision */          /* initial (fast) skip decision */
616          if (Data->iMinSAD[1] < (int)Data->iQuant * INITIAL_SKIP_THRESH          if (Data->iMinSAD[1] < (int)Data->iQuant * INITIAL_SKIP_THRESH
617                  && Data->iMinSAD[2] < (int)Data->iQuant * INITIAL_SKIP_THRESH                  && Data->iMinSAD[2] < (int)Data->iQuant * INITIAL_SKIP_THRESH
618                  && Data->iMinSAD[3] < (int)Data->iQuant * INITIAL_SKIP_THRESH                  && Data->iMinSAD[3] < (int)Data->iQuant * INITIAL_SKIP_THRESH
619                  && Data->iMinSAD[4] < (int)Data->iQuant * INITIAL_SKIP_THRESH) {                  && Data->iMinSAD[4] < (int)Data->iQuant * INITIAL_SKIP_THRESH) {
620                  /* possible skip */                  /* possible skip */
621                  SkipDecisionB(pCur, f_Ref, b_Ref, pMB, x, y, Data);                  SkipDecisionB(pMB, Data);
622                  if (pMB->mode == MODE_DIRECT_NONE_MV)                  if (pMB->mode == MODE_DIRECT_NONE_MV)
623                          return *Data->iMinSAD; /* skipped */                          return *Data->iMinSAD; /* skipped */
624          }          }
# Line 614  Line 626 
626          skip_sad = 4*MAX(MAX(Data->iMinSAD[1],Data->iMinSAD[2]), MAX(Data->iMinSAD[3],Data->iMinSAD[4]));          skip_sad = 4*MAX(MAX(Data->iMinSAD[1],Data->iMinSAD[2]), MAX(Data->iMinSAD[3],Data->iMinSAD[4]));
627          if (Data->chroma) skip_sad += Data->chromaSAD;          if (Data->chroma) skip_sad += Data->chromaSAD;
628    
629          CheckCandidate = b_mb->mode == MODE_INTER4V ? CheckCandidateDirect : CheckCandidateDirectno4v;          Data->currentMV[1].x = Data->directmvF[0].x + Data->currentMV->x; /* hints for forward and backward searches */
630            Data->currentMV[1].y = Data->directmvF[0].y + Data->currentMV->y;
631    
632            Data->currentMV[2].x = ((Data->currentMV->x == 0) ?
633                            Data->directmvB[0].x
634                            : Data->currentMV[1].x - Data->referencemv[0].x);
635    
636            Data->currentMV[2].y = ((Data->currentMV->y == 0) ?
637                            Data->directmvB[0].y
638                            : Data->currentMV[1].y - Data->referencemv[0].y);
639    
640            return skip_sad;
641    }
642    
643    static void
644    SearchDirect_final( const uint32_t MotionFlags,
645                                            const MACROBLOCK * const b_mb,
646                                            int32_t * const best_sad,
647                                            SearchData * const Data)
648    
649    {
650            CheckFunc * CheckCandidate = b_mb->mode == MODE_INTER4V ?
651                                                                            CheckCandidateDirect : CheckCandidateDirectno4v;
652            MainSearchFunc *MainSearchPtr;
653    
         if (!(MotionFlags & XVID_ME_SKIP_DELTASEARCH)) {  
654                  if (MotionFlags & XVID_ME_USESQUARES16) MainSearchPtr = xvid_me_SquareSearch;                  if (MotionFlags & XVID_ME_USESQUARES16) MainSearchPtr = xvid_me_SquareSearch;
655                          else if (MotionFlags & XVID_ME_ADVANCEDDIAMOND16) MainSearchPtr = xvid_me_AdvDiamondSearch;                          else if (MotionFlags & XVID_ME_ADVANCEDDIAMOND16) MainSearchPtr = xvid_me_AdvDiamondSearch;
656                                  else MainSearchPtr = xvid_me_DiamondSearch;                                  else MainSearchPtr = xvid_me_DiamondSearch;
657    
658            Data->qpel_precision = 0;
659                  MainSearchPtr(0, 0, Data, 255, CheckCandidate);                  MainSearchPtr(0, 0, Data, 255, CheckCandidate);
660    
661                  xvid_me_SubpelRefine(Data, CheckCandidate, 0);          Data->qpel_precision = Data->qpel;
662            if(Data->qpel) {
663                    *Data->iMinSAD = 256*4096; /* this old SAD was not real, it was in hpel precision */
664                    CheckCandidate(Data->currentMV->x, Data->currentMV->y, Data, 255);
665          }          }
666    
667          *Data->iMinSAD += Data->lambda16;          xvid_me_SubpelRefine(Data->currentMV[0], Data, CheckCandidate, 0);
         *best_sad = *Data->iMinSAD;  
   
         if (Data->qpel || b_mb->mode == MODE_INTER4V) pMB->mode = MODE_DIRECT;  
         else pMB->mode = MODE_DIRECT_NO4V; /* for faster compensation */  
668    
669          pMB->pmvs[3] = *Data->currentMV;          if (Data->iMinSAD[0] < *best_sad) {
670                    *best_sad = Data->iMinSAD[0];
         for (k = 0; k < 4; k++) {  
                 pMB->mvs[k].x = Data->directmvF[k].x + Data->currentMV->x;  
                 pMB->b_mvs[k].x = (     (Data->currentMV->x == 0)  
                                                         ? Data->directmvB[k].x  
                                                         :pMB->mvs[k].x - Data->referencemv[k].x);  
                 pMB->mvs[k].y = (Data->directmvF[k].y + Data->currentMV->y);  
                 pMB->b_mvs[k].y = ((Data->currentMV->y == 0)  
                                                         ? Data->directmvB[k].y  
                                                         : pMB->mvs[k].y - Data->referencemv[k].y);  
                 if (Data->qpel) {  
                         pMB->qmvs[k].x = pMB->mvs[k].x; pMB->mvs[k].x /= 2;  
                         pMB->b_qmvs[k].x = pMB->b_mvs[k].x; pMB->b_mvs[k].x /= 2;  
                         pMB->qmvs[k].y = pMB->mvs[k].y; pMB->mvs[k].y /= 2;  
                         pMB->b_qmvs[k].y = pMB->b_mvs[k].y; pMB->b_mvs[k].y /= 2;  
671                  }                  }
672    
                 if (b_mb->mode != MODE_INTER4V) {  
                         pMB->mvs[3] = pMB->mvs[2] = pMB->mvs[1] = pMB->mvs[0];  
                         pMB->b_mvs[3] = pMB->b_mvs[2] = pMB->b_mvs[1] = pMB->b_mvs[0];  
                         pMB->qmvs[3] = pMB->qmvs[2] = pMB->qmvs[1] = pMB->qmvs[0];  
                         pMB->b_qmvs[3] = pMB->b_qmvs[2] = pMB->b_qmvs[1] = pMB->b_qmvs[0];  
                         break;  
                 }  
         }  
         return skip_sad;  
673  }  }
674    
675    
676  static void set_range(int * range, SearchData * Data)  static __inline void
677    set_range(int * range, SearchData * Data)
678  {  {
679          Data->min_dx = range[0];          Data->min_dx = range[0];
680          Data->max_dx = range[1];          Data->max_dx = range[1];
# Line 671  Line 683 
683  }  }
684    
685  static void  static void
686  SearchInterpolate(const IMAGE * const f_Ref,  SearchInterpolate_initial(
                                 const uint8_t * const f_RefH,  
                                 const uint8_t * const f_RefV,  
                                 const uint8_t * const f_RefHV,  
                                 const IMAGE * const b_Ref,  
                                 const uint8_t * const b_RefH,  
                                 const uint8_t * const b_RefV,  
                                 const uint8_t * const b_RefHV,  
687                                  const int x, const int y,                                  const int x, const int y,
                                 const uint32_t fcode,  
                                 const uint32_t bcode,  
688                                  const uint32_t MotionFlags,                                  const uint32_t MotionFlags,
689                                  const MBParam * const pParam,                                  const MBParam * const pParam,
690                                  const VECTOR * const f_predMV,                                  const VECTOR * const f_predMV,
691                                  const VECTOR * const b_predMV,                                  const VECTOR * const b_predMV,
                                 MACROBLOCK * const pMB,  
692                                  int32_t * const best_sad,                                  int32_t * const best_sad,
693                                  SearchData * const Data)                                  SearchData * const Data,
694                                    const VECTOR startF,
695                                    const VECTOR startB)
696    
697  {  {
         int i, j;  
698          int b_range[4], f_range[4];          int b_range[4], f_range[4];
         int threshA = (MotionFlags & XVID_ME_FAST_MODEINTERPOLATE) ? 250 : 500;  
         int threshB = (MotionFlags & XVID_ME_FAST_MODEINTERPOLATE) ? 150 : 300;  
699    
700          Data->qpel_precision = 0;          Data->qpel_precision = 0;
         *Data->iMinSAD = 4096*256;  
         Data->iFcode = fcode; Data->bFcode = bcode;  
   
         i = (x + y * Data->iEdgedWidth) * 16;  
   
         Data->RefP[0] = f_Ref->y + i;  
         Data->RefP[2] = f_RefH + i;  
         Data->RefP[1] = f_RefV + i;  
         Data->RefP[3] = f_RefHV + i;  
         Data->b_RefP[0] = b_Ref->y + i;  
         Data->b_RefP[2] = b_RefH + i;  
         Data->b_RefP[1] = b_RefV + i;  
         Data->b_RefP[3] = b_RefHV + i;  
         Data->RefP[4] = f_Ref->u + (x + (Data->iEdgedWidth/2) * y) * 8;  
         Data->RefP[5] = f_Ref->v + (x + (Data->iEdgedWidth/2) * y) * 8;  
         Data->b_RefP[4] = b_Ref->u + (x + (Data->iEdgedWidth/2) * y) * 8;  
         Data->b_RefP[5] = b_Ref->v + (x + (Data->iEdgedWidth/2) * y) * 8;  
   
         Data->chromaX = Data->chromaY = 256*4096;  
701    
702          Data->predMV = *f_predMV;          Data->predMV = *f_predMV;
703          Data->bpredMV = *b_predMV;          Data->bpredMV = *b_predMV;
704    
705          Data->currentMV[0] = Data->currentMV[2]; /* forward search left its vector here */          Data->currentMV[0] = startF;
706            Data->currentMV[1] = startB;
707    
708          get_range(f_range, f_range+1, f_range+2, f_range+3, x, y, 4, pParam->width, pParam->height, fcode - Data->qpel, 1, 0);          get_range(f_range, f_range+1, f_range+2, f_range+3, x, y, 4, pParam->width, pParam->height, Data->iFcode - Data->qpel, 1, 0);
709          get_range(b_range, b_range+1, b_range+2, b_range+3, x, y, 4, pParam->width, pParam->height, bcode - Data->qpel, 1, 0);          get_range(b_range, b_range+1, b_range+2, b_range+3, x, y, 4, pParam->width, pParam->height, Data->bFcode - Data->qpel, 1, 0);
710    
711          if (Data->currentMV[0].x > f_range[1]) Data->currentMV[0].x = f_range[1];          if (Data->currentMV[0].x > f_range[1]) Data->currentMV[0].x = f_range[1];
712          if (Data->currentMV[0].x < f_range[0]) Data->currentMV[0].x = f_range[0];          if (Data->currentMV[0].x < f_range[0]) Data->currentMV[0].x = f_range[0];
# Line 739  Line 722 
722    
723          CheckCandidateInt(Data->currentMV[0].x, Data->currentMV[0].y, Data, 1);          CheckCandidateInt(Data->currentMV[0].x, Data->currentMV[0].y, Data, 1);
724    
725            if (Data->iMinSAD[0] < *best_sad) *best_sad = Data->iMinSAD[0];
726    }
727    
728    static void
729    SearchInterpolate_final(const int x, const int y,
730                                                    const uint32_t MotionFlags,
731                                                    const MBParam * const pParam,
732                                                    int32_t * const best_sad,
733                                                    SearchData * const Data)
734    {
735            int i, j;
736            int b_range[4], f_range[4];
737    
738            get_range(f_range, f_range+1, f_range+2, f_range+3, x, y, 4, pParam->width, pParam->height, Data->iFcode - Data->qpel, 1, 0);
739            get_range(b_range, b_range+1, b_range+2, b_range+3, x, y, 4, pParam->width, pParam->height, Data->bFcode - Data->qpel, 1, 0);
740    
741          /* diamond */          /* diamond */
742          do {          do {
743                  Data->dir = 0;                  Data->dir = 0;
# Line 765  Line 764 
764    
765          /* qpel refinement */          /* qpel refinement */
766          if (Data->qpel) {          if (Data->qpel) {
                 if (*Data->iMinSAD > *best_sad + threshA) return;  
767                  Data->qpel_precision = 1;                  Data->qpel_precision = 1;
768                  get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 4, pParam->width, pParam->height, fcode, 2, 0);                  get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy,
769                            x, y, 4, pParam->width, pParam->height, Data->iFcode, 2, 0);
770    
771                  Data->currentQMV[0].x = 2 * Data->currentMV[0].x;                  Data->currentQMV[0].x = 2 * Data->currentMV[0].x;
772                  Data->currentQMV[0].y = 2 * Data->currentMV[0].y;                  Data->currentQMV[0].y = 2 * Data->currentMV[0].y;
773                  Data->currentQMV[1].x = 2 * Data->currentMV[1].x;                  Data->currentQMV[1].x = 2 * Data->currentMV[1].x;
774                  Data->currentQMV[1].y = 2 * Data->currentMV[1].y;                  Data->currentQMV[1].y = 2 * Data->currentMV[1].y;
775    
776                  if (MotionFlags & XVID_ME_QUARTERPELREFINE16)                  if (MotionFlags & XVID_ME_QUARTERPELREFINE16) {
777                          xvid_me_SubpelRefine(Data, CheckCandidateInt, 1);                          xvid_me_SubpelRefine(Data->currentQMV[0], Data, CheckCandidateInt, 1);
778    
779                  if (*Data->iMinSAD > *best_sad + threshB) return;                          get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy,
780                  get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 4, pParam->width, pParam->height, bcode, 2, 0);                                  x, y, 4, pParam->width, pParam->height, Data->bFcode, 2, 0);
781    
782                  if (MotionFlags & XVID_ME_QUARTERPELREFINE16)                          xvid_me_SubpelRefine(Data->currentQMV[1], Data, CheckCandidateInt, 2);
783                          xvid_me_SubpelRefine(Data, CheckCandidateInt, 2);                  }
784          }          }
785    
786          *Data->iMinSAD += 2 * Data->lambda16; /* two bits are needed to code interpolate mode. */          if (Data->iMinSAD[0] < *best_sad) *best_sad = Data->iMinSAD[0];
787    }
788    
789          if (*Data->iMinSAD < *best_sad) {  static void
790                  *best_sad = *Data->iMinSAD;  ModeDecision_BVOP_SAD(const SearchData * const Data_d,
791                  pMB->mvs[0] = Data->currentMV[0];                                            const SearchData * const Data_b,
792                  pMB->b_mvs[0] = Data->currentMV[1];                                            const SearchData * const Data_f,
793                  pMB->mode = MODE_INTERPOLATE;                                            const SearchData * const Data_i,
794                  if (Data->qpel) {                                            MACROBLOCK * const pMB,
795                          pMB->qmvs[0] = Data->currentQMV[0];                                            const MACROBLOCK * const b_mb,
796                          pMB->b_qmvs[0] = Data->currentQMV[1];                                            VECTOR * f_predMV,
797                                              VECTOR * b_predMV)
798    {
799            int mode = MODE_DIRECT, k;
800            int best_sad, f_sad, b_sad, i_sad;
801            const int qpel = Data_d->qpel;
802    
803            /* evaluate cost of all modes - quite simple in SAD */
804            best_sad = Data_d->iMinSAD[0] + 1*Data_d->lambda16;
805            b_sad = Data_b->iMinSAD[0] + 3*Data_d->lambda16;
806            f_sad = Data_f->iMinSAD[0] + 4*Data_d->lambda16;
807            i_sad = Data_i->iMinSAD[0] + 2*Data_d->lambda16;
808    
809            if (b_sad < best_sad) {
810                    mode = MODE_BACKWARD;
811                    best_sad = b_sad;
812            }
813    
814            if (f_sad < best_sad) {
815                    mode = MODE_FORWARD;
816                    best_sad = f_sad;
817            }
818    
819            if (i_sad < best_sad) {
820                    mode = MODE_INTERPOLATE;
821                    best_sad = i_sad;
822            }
823    
824            pMB->sad16 = best_sad;
825            pMB->mode = mode;
826    
827            switch (mode) {
828    
829            case MODE_DIRECT:
830                    if (!qpel && b_mb->mode != MODE_INTER4V) pMB->mode = MODE_DIRECT_NO4V; /* for faster compensation */
831    
832                    pMB->pmvs[3] = Data_d->currentMV[0];
833    
834                    for (k = 0; k < 4; k++) {
835                            pMB->mvs[k].x = Data_d->directmvF[k].x + Data_d->currentMV->x;
836                            pMB->b_mvs[k].x = (     (Data_d->currentMV->x == 0)
837                                                                    ? Data_d->directmvB[k].x
838                                                                    :pMB->mvs[k].x - Data_d->referencemv[k].x);
839                            pMB->mvs[k].y = (Data_d->directmvF[k].y + Data_d->currentMV->y);
840                            pMB->b_mvs[k].y = ((Data_d->currentMV->y == 0)
841                                                                    ? Data_d->directmvB[k].y
842                                                                    : pMB->mvs[k].y - Data_d->referencemv[k].y);
843                            if (qpel) {
844                                    pMB->qmvs[k].x = pMB->mvs[k].x; pMB->mvs[k].x /= 2;
845                                    pMB->b_qmvs[k].x = pMB->b_mvs[k].x; pMB->b_mvs[k].x /= 2;
846                                    pMB->qmvs[k].y = pMB->mvs[k].y; pMB->mvs[k].y /= 2;
847                                    pMB->b_qmvs[k].y = pMB->b_mvs[k].y; pMB->b_mvs[k].y /= 2;
848                            }
849    
850                            if (b_mb->mode != MODE_INTER4V) {
851                                    pMB->mvs[3] = pMB->mvs[2] = pMB->mvs[1] = pMB->mvs[0];
852                                    pMB->b_mvs[3] = pMB->b_mvs[2] = pMB->b_mvs[1] = pMB->b_mvs[0];
853                                    pMB->qmvs[3] = pMB->qmvs[2] = pMB->qmvs[1] = pMB->qmvs[0];
854                                    pMB->b_qmvs[3] = pMB->b_qmvs[2] = pMB->b_qmvs[1] = pMB->b_qmvs[0];
855                                    break;
856                            }
857                    }
858                    break;
859    
860            case MODE_FORWARD:
861                    if (qpel) {
862                            pMB->pmvs[0].x = Data_f->currentQMV->x - f_predMV->x;
863                            pMB->pmvs[0].y = Data_f->currentQMV->y - f_predMV->y;
864                            pMB->qmvs[0] = *Data_f->currentQMV;
865                            *f_predMV = Data_f->currentQMV[0];
866                    } else {
867                            pMB->pmvs[0].x = Data_f->currentMV->x - f_predMV->x;
868                            pMB->pmvs[0].y = Data_f->currentMV->y - f_predMV->y;
869                            *f_predMV = Data_f->currentMV[0];
870                    }
871                    pMB->mvs[0] = *Data_f->currentMV;
872                    break;
873    
874            case MODE_BACKWARD:
875                    if (qpel) {
876                            pMB->pmvs[0].x = Data_b->currentQMV->x - b_predMV->x;
877                            pMB->pmvs[0].y = Data_b->currentQMV->y - b_predMV->y;
878                            pMB->b_qmvs[0] = *Data_b->currentQMV;
879                            *b_predMV = Data_b->currentQMV[0];
880                    } else {
881                            pMB->pmvs[0].x = Data_b->currentMV->x - b_predMV->x;
882                            pMB->pmvs[0].y = Data_b->currentMV->y - b_predMV->y;
883                            *b_predMV = Data_b->currentMV[0];
884                    }
885                    pMB->b_mvs[0] = *Data_b->currentMV;
886                    break;
887    
888    
889            case MODE_INTERPOLATE:
890                    pMB->mvs[0] = Data_i->currentMV[0];
891                    pMB->b_mvs[0] = Data_i->currentMV[1];
892                    if (qpel) {
893                            pMB->qmvs[0] = Data_i->currentQMV[0];
894                            pMB->b_qmvs[0] = Data_i->currentQMV[1];
895                          pMB->pmvs[1].x = pMB->qmvs[0].x - f_predMV->x;                          pMB->pmvs[1].x = pMB->qmvs[0].x - f_predMV->x;
896                          pMB->pmvs[1].y = pMB->qmvs[0].y - f_predMV->y;                          pMB->pmvs[1].y = pMB->qmvs[0].y - f_predMV->y;
897                          pMB->pmvs[0].x = pMB->b_qmvs[0].x - b_predMV->x;                          pMB->pmvs[0].x = pMB->b_qmvs[0].x - b_predMV->x;
898                          pMB->pmvs[0].y = pMB->b_qmvs[0].y - b_predMV->y;                          pMB->pmvs[0].y = pMB->b_qmvs[0].y - b_predMV->y;
899                            *f_predMV = Data_i->currentQMV[0];
900                            *b_predMV = Data_i->currentQMV[1];
901                  } else {                  } else {
902                          pMB->pmvs[1].x = pMB->mvs[0].x - f_predMV->x;                          pMB->pmvs[1].x = pMB->mvs[0].x - f_predMV->x;
903                          pMB->pmvs[1].y = pMB->mvs[0].y - f_predMV->y;                          pMB->pmvs[1].y = pMB->mvs[0].y - f_predMV->y;
904                          pMB->pmvs[0].x = pMB->b_mvs[0].x - b_predMV->x;                          pMB->pmvs[0].x = pMB->b_mvs[0].x - b_predMV->x;
905                          pMB->pmvs[0].y = pMB->b_mvs[0].y - b_predMV->y;                          pMB->pmvs[0].y = pMB->b_mvs[0].y - b_predMV->y;
906                            *f_predMV = Data_i->currentMV[0];
907                            *b_predMV = Data_i->currentMV[1];
908                  }                  }
909                    break;
910          }          }
911  }  }
912    
# Line 826  Line 929 
929                                           const IMAGE * const b_refHV)                                           const IMAGE * const b_refHV)
930  {  {
931          uint32_t i, j;          uint32_t i, j;
932          int32_t best_sad;          int32_t best_sad, sad2;
933          uint32_t skip_sad;          uint32_t skip_sad;
934    
935          const MACROBLOCK * const b_mbs = b_reference->mbs;          const MACROBLOCK * const b_mbs = b_reference->mbs;
         MACROBLOCK *const pMBs = frame->mbs;  
936    
937          VECTOR f_predMV, b_predMV;          VECTOR f_predMV, b_predMV;
938    
# Line 838  Line 940 
940          const int32_t TRD = time_pp;          const int32_t TRD = time_pp;
941    
942          /* some pre-inintialized data for the rest of the search */          /* some pre-inintialized data for the rest of the search */
943            SearchData Data_d, Data_f, Data_b, Data_i;
944            memset(&Data_d, 0, sizeof(SearchData));
945    
946          SearchData Data;          Data_d.iEdgedWidth = pParam->edged_width;
947          memset(&Data, 0, sizeof(SearchData));          Data_d.qpel = pParam->vol_flags & XVID_VOL_QUARTERPEL ? 1 : 0;
948            Data_d.rounding = 0;
949            Data_d.chroma = frame->motion_flags & XVID_ME_CHROMA_BVOP;
950            Data_d.iQuant = frame->quant;
951    
952            Data_d.RefQ = f_refV->u; /* a good place, also used in MC (for similar purpose) */
953    
954            memcpy(&Data_f, &Data_d, sizeof(SearchData));
955            memcpy(&Data_b, &Data_d, sizeof(SearchData));
956            memcpy(&Data_i, &Data_d, sizeof(SearchData));
957    
958          Data.iEdgedWidth = pParam->edged_width;          Data_f.iFcode = Data_i.iFcode = frame->fcode;
959          Data.qpel = pParam->vol_flags & XVID_VOL_QUARTERPEL ? 1 : 0;          Data_b.iFcode = Data_i.bFcode = frame->bcode;
         Data.rounding = 0;  
         Data.chroma = frame->motion_flags & XVID_ME_CHROMA_BVOP;  
         Data.iQuant = frame->quant;  
960    
         Data.RefQ = f_refV->u; /* a good place, also used in MC (for similar purpose) */  
961    
         /* note: i==horizontal, j==vertical */  
962          for (j = 0; j < pParam->mb_height; j++) {          for (j = 0; j < pParam->mb_height; j++) {
963    
964                  f_predMV = b_predMV = zeroMV;   /* prediction is reset at left boundary */                  f_predMV = b_predMV = zeroMV;   /* prediction is reset at left boundary */
# Line 861  Line 969 
969                          int interpol_search = 0;                          int interpol_search = 0;
970                          int bf_search = 0;                          int bf_search = 0;
971                          int bf_thresh = 0;                          int bf_thresh = 0;
972                            pMB->mode = -1;
973    
974                            initialize_searchData(&Data_d, &Data_f, &Data_b, &Data_i,
975                                              i, j, f_ref, f_refH->y, f_refV->y, f_refHV->y,
976                                              b_ref, b_refH->y, b_refV->y, b_refHV->y,
977                                              &frame->image, b_mb);
978    
979  /* special case, if collocated block is SKIPed in P-VOP: encoding is forward (0,0), cpb=0 without further ado */  /* special case, if collocated block is SKIPed in P-VOP: encoding is forward (0,0), cpb=0 without further ado */
980                          if (b_reference->coding_type != S_VOP)                          if (b_reference->coding_type != S_VOP)
# Line 871  Line 985 
985                                          continue;                                          continue;
986                                  }                                  }
987    
                         Data.lambda16 = xvid_me_lambda_vec16[b_mb->quant];  
                         Data.Cur = frame->image.y + (j * Data.iEdgedWidth + i) * 16;  
                         Data.CurU = frame->image.u + (j * Data.iEdgedWidth/2 + i) * 8;  
                         Data.CurV = frame->image.v + (j * Data.iEdgedWidth/2 + i) * 8;  
   
988  /* direct search comes first, because it (1) checks for SKIP-mode  /* direct search comes first, because it (1) checks for SKIP-mode
989          and (2) sets very good predictions for forward and backward search */          and (2) sets very good predictions for forward and backward search */
990                          skip_sad = SearchDirect(f_ref, f_refH->y, f_refV->y, f_refHV->y,                          skip_sad = SearchDirect_initial(i, j, frame->motion_flags, TRB, TRD, pParam, pMB,
991                                                                          b_ref, b_refH->y, b_refV->y, b_refHV->y,                                                                                          b_mb, &best_sad, &Data_d);
                                                                         &frame->image,  
                                                                         i, j,  
                                                                         frame->motion_flags,  
                                                                         TRB, TRD,  
                                                                         pParam,  
                                                                         pMB, b_mb,  
                                                                         &best_sad,  
                                                                         &Data);  
992    
993                          if (pMB->mode == MODE_DIRECT_NONE_MV) {                          if (pMB->mode == MODE_DIRECT_NONE_MV) {
994                                  pMB->sad16 = best_sad;                                  pMB->sad16 = best_sad;
995                                  continue;                                  continue;
996                          }                          }
997    
998                          if (frame->motion_flags & XVID_ME_BFRAME_EARLYSTOP) {                          SearchBF_initial(i, j, frame->motion_flags, frame->fcode, pParam, pMB,
999                                  if(i > 0 && j > 0 && i < pParam->mb_width) {                                                  &f_predMV, &best_sad, MODE_FORWARD, &Data_f, Data_d.currentMV[1]);
                                         bf_thresh = MIN((&pMBs[(i-1) + j * pParam->mb_width])->sad16,  
                                                                      MIN((&pMBs[i + (j-1) * pParam->mb_width])->sad16,  
                                                                      (&pMBs[(i+1) + (j-1) * pParam->mb_width])->sad16));  
   
                                         if (((&pMBs[(i-1) + j * pParam->mb_width])->mode != MODE_FORWARD) &&  
                                                 ((&pMBs[(i-1) + j * pParam->mb_width])->mode != MODE_BACKWARD) &&  
                                                 ((&pMBs[(i-1) + j * pParam->mb_width])->mode != MODE_INTERPOLATE))  
                                                         bf_search++;  
   
                                         if (((&pMBs[i + (j - 1) * pParam->mb_width])->mode != MODE_FORWARD) &&  
                                                 ((&pMBs[i + (j - 1) * pParam->mb_width])->mode != MODE_BACKWARD) &&  
                                                 ((&pMBs[i + (j - 1) * pParam->mb_width])->mode != MODE_INTERPOLATE))  
                                                         bf_search++;  
   
                                         if (((&pMBs[(i + 1) + (j - 1) * pParam->mb_width])->mode != MODE_FORWARD) &&  
                                                 ((&pMBs[(i + 1) + (j - 1) * pParam->mb_width])->mode != MODE_BACKWARD) &&  
                                                 ((&pMBs[(i + 1) + (j - 1) * pParam->mb_width])->mode != MODE_INTERPOLATE))  
                                                 bf_search++;  
                                 }  
1000    
1001                                  if ((best_sad < bf_thresh) && (bf_search == 3))                          SearchBF_initial(i, j, frame->motion_flags, frame->bcode, pParam, pMB,
1002                                          continue;                                                  &b_predMV, &best_sad, MODE_BACKWARD, &Data_b, Data_d.currentMV[2]);
                         }  
1003    
1004                          /* forward search */                          sad2 = best_sad;
                         SearchBF(f_ref, f_refH->y, f_refV->y, f_refHV->y,  
                                                 i, j,  
                                                 frame->motion_flags,  
                                                 frame->fcode, pParam,  
                                                 pMB, &f_predMV, &best_sad,  
                                                 MODE_FORWARD, &Data);  
   
                         /* backward search */  
                         SearchBF(b_ref, b_refH->y, b_refV->y, b_refHV->y,  
                                                 i, j,  
                                                 frame->motion_flags,  
                                                 frame->bcode, pParam,  
                                                 pMB, &b_predMV, &best_sad,  
                                                 MODE_BACKWARD, &Data);  
   
                         /* interpolate search comes last, because it uses data from forward and backward as prediction */  
                         if (frame->motion_flags & XVID_ME_FAST_MODEINTERPOLATE) {  
   
                                 if(i > 0 && j > 0 && i < pParam->mb_width) {  
                                         if ((&pMBs[(i-1) + j * pParam->mb_width])->mode == MODE_INTERPOLATE)  
                                                 interpol_search++;  
                                         if ((&pMBs[i + (j - 1) * pParam->mb_width])->mode == MODE_INTERPOLATE)  
                                                 interpol_search++;  
                                         if ((&pMBs[(i + 1) + (j - 1) * pParam->mb_width])->mode == MODE_INTERPOLATE)  
                                                 interpol_search++;  
                                 }  
                                 else  
                                         interpol_search = 1;  
1005    
1006                                  interpol_search |= !(best_sad < 3 * Data.iQuant * MAX_SAD00_FOR_SKIP * (Data.chroma ? 3:2));                          if (Data_f.iMinSAD[0] < 2*sad2+1500)
1007                          }                                  SearchBF_final(i, j, frame->motion_flags, pParam, &best_sad, &Data_f);
                         else  
                                 interpol_search = 1;  
1008    
1009                          if (interpol_search) {                          if (Data_b.iMinSAD[0] < 2*sad2+1500)
1010                                  SearchInterpolate(f_ref, f_refH->y, f_refV->y, f_refHV->y,                                  SearchBF_final(i, j, frame->motion_flags, pParam, &best_sad, &Data_b);
                                                                   b_ref, b_refH->y, b_refV->y, b_refHV->y,  
                                                                   i, j,  
                                                                   frame->fcode, frame->bcode,  
                                                                   frame->motion_flags,  
                                                                   pParam,  
                                                                   &f_predMV, &b_predMV,  
                                                                   pMB, &best_sad,  
                                                                   &Data);  
                         }  
1011    
1012                          /* final skip decision */                          SearchInterpolate_initial(i, j, frame->motion_flags, pParam, &f_predMV, &b_predMV, &best_sad,
1013                          if ( (skip_sad < Data.iQuant * MAX_SAD00_FOR_SKIP )                                                                    &Data_i, Data_f.currentMV[0], Data_b.currentMV[0]);
                                 && ((100*best_sad)/(skip_sad+1) > FINAL_SKIP_THRESH) )  
1014    
1015                                  SkipDecisionB(&frame->image, f_ref, b_ref, pMB, i, j, &Data);                          if (((Data_i.iMinSAD[0] < 2*best_sad+2000) && (!(frame->motion_flags&XVID_ME_FAST_MODEINTERPOLATE))
1016                                    || Data_i.iMinSAD[0] <= best_sad))
1017    
1018                          switch (pMB->mode) {                                  SearchInterpolate_final(i, j, frame->motion_flags, pParam, &best_sad, &Data_i);
1019                                  case MODE_FORWARD:  
1020                                          f_predMV = Data.qpel ? pMB->qmvs[0] : pMB->mvs[0];  
1021                                          pMB->sad16 = best_sad;                          if ((Data_d.iMinSAD[0] <= best_sad) && (!(frame->motion_flags&XVID_ME_SKIP_DELTASEARCH)))
1022                                          break;                                  SearchDirect_final(frame->motion_flags, b_mb, &best_sad, &Data_d);
1023                                  case MODE_BACKWARD:  
1024                                          b_predMV = Data.qpel ? pMB->b_qmvs[0] : pMB->b_mvs[0];  
1025                                          pMB->sad16 = best_sad;                          /* final skip decision */
1026                                          break;                          if ( (skip_sad < Data_d.iQuant * MAX_SAD00_FOR_SKIP )
1027                                  case MODE_INTERPOLATE:                                  && ((100*best_sad)/(skip_sad+1) > FINAL_SKIP_THRESH) ) {
1028                                          f_predMV = Data.qpel ? pMB->qmvs[0] : pMB->mvs[0];  
1029                                          b_predMV = Data.qpel ? pMB->b_qmvs[0] : pMB->b_mvs[0];                                  SkipDecisionB(pMB, &Data_d);
1030                                          pMB->sad16 = best_sad;                                  if (pMB->mode == MODE_DIRECT_NONE_MV) { /* skipped? */
                                         break;  
                                 default:  
1031                                          pMB->sad16 = best_sad;                                          pMB->sad16 = best_sad;
1032                                          break;                                          continue;
1033                          }                          }
1034                  }                  }
1035    
1036                            ModeDecision_BVOP_SAD(&Data_d, &Data_b, &Data_f, &Data_i, pMB, b_mb, &f_predMV, &b_predMV);
1037    
1038                    }
1039          }          }
1040  }  }

Legend:
Removed from v.1475  
changed lines
  Added in v.1478

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