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

-revision 767, Wed Jan  8 14:37:52 2003 UTC
+revision 819, Sat Feb  8 03:49:47 2003 UTC
 Line 32
  #include <stdio.h>
  #include <stdlib.h>
  #include <string.h>     // memcpy
+ #include <math.h>       // lrint
  #include "../encoder.h"
  #include "../utils/mbfunctions.h"
-Line 49
+Line 50
  #define MAX_SAD00_FOR_SKIP      (20)
  #define MAX_CHROMA_SAD_FOR_SKIP (22)
- #define CHECK_CANDIDATE(X,Y,D) { \
+ #define CHECK_CANDIDATE(X,Y,D) { CheckCandidate((X),(Y), (D), &iDirection, data ); }
- (*CheckCandidate)((const int)(X),(const int)(Y), (D), &iDirection, data ); }
- static __inline int
+ static __inline uint32_t
  d_mv_bits(int x, int y, const VECTOR pred, const uint32_t iFcode, const int qpel, const int rrv)
  {
          int xb, yb;
-         if (qpel) { x *= 2; y *= 2;}
+         x = qpel ? x<<1 : x;
-         else if (rrv) { x = RRV_MV_SCALEDOWN(x); y = RRV_MV_SCALEDOWN(y); }
+         y = qpel ? y<<1 : y;
+         if (rrv) { x = RRV_MV_SCALEDOWN(x); y = RRV_MV_SCALEDOWN(y); }
          x -= pred.x;
          y -= pred.y;
          if (x) {
-                 if (x < 0) x = -x;
+                 x = ABS(x);
                  x += (1 << (iFcode - 1)) - 1;
                  x >>= (iFcode - 1);
                  if (x > 32) x = 32;
-Line 70
+Line 72
          } else xb = 1;
          if (y) {
-                 if (y < 0) y = -y;
+                 y = ABS(y);
                  y += (1 << (iFcode - 1)) - 1;
                  y >>= (iFcode - 1);
                  if (y > 32) y = 32;
-Line 79
+Line 81
          return xb + yb;
  }
+ static int32_t ChromaSAD2(int fx, int fy, int bx, int by, const SearchData * const data)
+ {
+         int sad;
+         const uint32_t stride = data->iEdgedWidth/2;
+         uint8_t * f_refu = data->RefQ,
+                 * f_refv = data->RefQ + 8,
+                 * b_refu = data->RefQ + 16,
+                 * b_refv = data->RefQ + 24;
+         switch (((fx & 1) << 1) | (fy & 1))     {
+                 case 0:
+                         fx = fx / 2; fy = fy / 2;
+                         f_refu = (uint8_t*)data->RefCU + fy * stride + fx, stride;
+                         f_refv = (uint8_t*)data->RefCV + fy * stride + fx, stride;
+                         break;
+                 case 1:
+                         fx = fx / 2; fy = (fy - 1) / 2;
+                         interpolate8x8_halfpel_v(f_refu, data->RefCU + fy * stride + fx, stride, data->rounding);
+                         interpolate8x8_halfpel_v(f_refv, data->RefCV + fy * stride + fx, stride, data->rounding);
+                         break;
+                 case 2:
+                         fx = (fx - 1) / 2; fy = fy / 2;
+                         interpolate8x8_halfpel_h(f_refu, data->RefCU + fy * stride + fx, stride, data->rounding);
+                         interpolate8x8_halfpel_h(f_refv, data->RefCV + fy * stride + fx, stride, data->rounding);
+                         break;
+                 default:
+                         fx = (fx - 1) / 2; fy = (fy - 1) / 2;
+                         interpolate8x8_halfpel_hv(f_refu, data->RefCU + fy * stride + fx, stride, data->rounding);
+                         interpolate8x8_halfpel_hv(f_refv, data->RefCV + fy * stride + fx, stride, data->rounding);
+                         break;
+         }
+         switch (((bx & 1) << 1) | (by & 1))     {
+                 case 0:
+                         bx = bx / 2; by = by / 2;
+                         b_refu = (uint8_t*)data->b_RefCU + by * stride + bx, stride;
+                         b_refv = (uint8_t*)data->b_RefCV + by * stride + bx, stride;
+                         break;
+                 case 1:
+                         bx = bx / 2; by = (by - 1) / 2;
+                         interpolate8x8_halfpel_v(b_refu, data->b_RefCU + by * stride + bx, stride, data->rounding);
+                         interpolate8x8_halfpel_v(b_refv, data->b_RefCV + by * stride + bx, stride, data->rounding);
+                         break;
+                 case 2:
+                         bx = (bx - 1) / 2; by = by / 2;
+                         interpolate8x8_halfpel_h(b_refu, data->b_RefCU + by * stride + bx, stride, data->rounding);
+                         interpolate8x8_halfpel_h(b_refv, data->b_RefCV + by * stride + bx, stride, data->rounding);
+                         break;
+                 default:
+                         bx = (bx - 1) / 2; by = (by - 1) / 2;
+                         interpolate8x8_halfpel_hv(b_refu, data->b_RefCU + by * stride + bx, stride, data->rounding);
+                         interpolate8x8_halfpel_hv(b_refv, data->b_RefCV + by * stride + bx, stride, data->rounding);
+                         break;
+         }
+         sad = sad8bi(data->CurU, b_refu, f_refu, stride);
+         sad += sad8bi(data->CurV, b_refv, f_refv, stride);
+         return sad;
+ }
  static int32_t
  ChromaSAD(int dx, int dy, const SearchData * const data)
  {
-Line 118
+Line 182
  }
  static __inline const uint8_t *
- GetReferenceB(const int x, const int y, const int dir, const SearchData * const data)
+ GetReferenceB(const int x, const int y, const uint32_t dir, const SearchData * const data)
  {
  //      dir : 0 = forward, 1 = backward
          switch ( (dir << 2) | ((x&1)<<1) | (y&1) ) {
-Line 139
+Line 203
  {
          switch ( ((x&1)<<1) | (y&1) ) {
                  case 0 : return data->Ref + x/2 + (y/2)*(data->iEdgedWidth);
+                 case 3 : return data->RefHV + (x-1)/2 + ((y-1)/2)*(data->iEdgedWidth);
                  case 1 : return data->RefV + x/2 + ((y-1)/2)*(data->iEdgedWidth);
-                 case 2 : return data->RefH + (x-1)/2 + (y/2)*(data->iEdgedWidth);
+                 default : return data->RefH + (x-1)/2 + (y/2)*(data->iEdgedWidth);      //case 2
-                 default : return data->RefHV + (x-1)/2 + ((y-1)/2)*(data->iEdgedWidth);
          }
  }
  static uint8_t *
- Interpolate8x8qpel(const int x, const int y, const int block, const int dir, const SearchData * const data)
+ Interpolate8x8qpel(const int x, const int y, const uint32_t block, const uint32_t dir, const SearchData * const data)
  {
  // create or find a qpel-precision reference picture; return pointer to it
-         uint8_t * Reference = (uint8_t *)data->RefQ + 16*dir;
+         uint8_t * Reference = data->RefQ + 16*dir;
-         const int32_t iEdgedWidth = data->iEdgedWidth;
+         const uint32_t iEdgedWidth = data->iEdgedWidth;
          const uint32_t rounding = data->rounding;
          const int halfpel_x = x/2;
          const int halfpel_y = y/2;
-Line 190
+Line 254
  }
  static uint8_t *
- Interpolate16x16qpel(const int x, const int y, const int dir, const SearchData * const data)
+ Interpolate16x16qpel(const int x, const int y, const uint32_t dir, const SearchData * const data)
  {
  // create or find a qpel-precision reference picture; return pointer to it
-         uint8_t * Reference = (uint8_t *)data->RefQ + 16*dir;
+         uint8_t * Reference = data->RefQ + 16*dir;
-         const int32_t iEdgedWidth = data->iEdgedWidth;
+         const uint32_t iEdgedWidth = data->iEdgedWidth;
          const uint32_t rounding = data->rounding;
          const int halfpel_x = x/2;
          const int halfpel_y = y/2;
-Line 202
+Line 266
          ref1 = GetReferenceB(halfpel_x, halfpel_y, dir, data);
          switch( ((x&1)<<1) + (y&1) ) {
-         case 0: // pure halfpel position
+         case 3: // x and y in qpel resolution - the "corners" (top left/right and
-                 return (uint8_t *) ref1;
+                          // bottom left/right) during qpel refinement
+                 ref2 = GetReferenceB(halfpel_x, y - halfpel_y, dir, data);
+                 ref3 = GetReferenceB(x - halfpel_x, halfpel_y, dir, data);
+                 ref4 = GetReferenceB(x - halfpel_x, y - halfpel_y, dir, data);
+                 interpolate8x8_avg4(Reference, ref1, ref2, ref3, ref4, iEdgedWidth, rounding);
+                 interpolate8x8_avg4(Reference+8, ref1+8, ref2+8, ref3+8, ref4+8, iEdgedWidth, rounding);
+                 interpolate8x8_avg4(Reference+8*iEdgedWidth, ref1+8*iEdgedWidth, ref2+8*iEdgedWidth, ref3+8*iEdgedWidth, ref4+8*iEdgedWidth, iEdgedWidth, rounding);
+                 interpolate8x8_avg4(Reference+8*iEdgedWidth+8, ref1+8*iEdgedWidth+8, ref2+8*iEdgedWidth+8, ref3+8*iEdgedWidth+8, ref4+8*iEdgedWidth+8, iEdgedWidth, rounding);
+                 break;
          case 1: // x halfpel, y qpel - top or bottom during qpel refinement
                  ref2 = GetReferenceB(halfpel_x, y - halfpel_y, dir, data);
                  interpolate8x8_avg2(Reference, ref1, ref2, iEdgedWidth, rounding, 8);
-Line 220
+Line 293
                  interpolate8x8_avg2(Reference+8*iEdgedWidth+8, ref1+8*iEdgedWidth+8, ref2+8*iEdgedWidth+8, iEdgedWidth, rounding, 8);
                  break;
-         default: // x and y in qpel resolution - the "corners" (top left/right and
+         case 0: // pure halfpel position
-                          // bottom left/right) during qpel refinement
+                 return (uint8_t *) ref1;
-                 ref2 = GetReferenceB(halfpel_x, y - halfpel_y, dir, data);
-                 ref3 = GetReferenceB(x - halfpel_x, halfpel_y, dir, data);
-                 ref4 = GetReferenceB(x - halfpel_x, y - halfpel_y, dir, data);
-                 interpolate8x8_avg4(Reference, ref1, ref2, ref3, ref4, iEdgedWidth, rounding);
-                 interpolate8x8_avg4(Reference+8, ref1+8, ref2+8, ref3+8, ref4+8, iEdgedWidth, rounding);
-                 interpolate8x8_avg4(Reference+8*iEdgedWidth, ref1+8*iEdgedWidth, ref2+8*iEdgedWidth, ref3+8*iEdgedWidth, ref4+8*iEdgedWidth, iEdgedWidth, rounding);
-                 interpolate8x8_avg4(Reference+8*iEdgedWidth+8, ref1+8*iEdgedWidth+8, ref2+8*iEdgedWidth+8, ref3+8*iEdgedWidth+8, ref4+8*iEdgedWidth+8, iEdgedWidth, rounding);
-                 break;
          }
          return Reference;
  }
-Line 239
+Line 304
  static void
  CheckCandidate16(const int x, const int y, const int Direction, int * const dir, const SearchData * const data)
  {
-         int t, xc, yc;
+         int xc, yc;
          const uint8_t * Reference;
          VECTOR * current;
+         int32_t sad; uint32_t t;
          if (( x > data->max_dx) || ( x < data->min_dx)
                  || ( y > data->max_dy) || (y < data->min_dy)) return;
-         if (data->qpel_precision) { // x and y are in 1/4 precision
+         if (!data->qpel_precision) {
-                 Reference = Interpolate16x16qpel(x, y, 0, data);
-                 xc = x/2; yc = y/2; //for chroma sad
-                 current = data->currentQMV;
-         } else {
                  Reference = GetReference(x, y, data);
                  current = data->currentMV;
                  xc = x; yc = y;
+         } else { // x and y are in 1/4 precision
+                 Reference = Interpolate16x16qpel(x, y, 0, data);
+                 xc = x/2; yc = y/2; //for chroma sad
+                 current = data->currentQMV;
          }
-         t = d_mv_bits(x, y, data->predMV, data->iFcode, data->qpel && !data->qpel_precision, 0);
-         data->temp[0] = sad16v(data->Cur, Reference, data->iEdgedWidth, data->temp + 1);
+         sad = sad16v(data->Cur, Reference, data->iEdgedWidth, data->temp + 1);
+         t = d_mv_bits(x, y, data->predMV, data->iFcode, data->qpel^data->qpel_precision, 0);
-         data->temp[0] += (data->lambda16 * t * data->temp[0])/1000;
+         sad += (data->lambda16 * t * sad)>>10;
-         data->temp[1] += (data->lambda8 * t * (data->temp[1] + NEIGH_8X8_BIAS))/100;
+         data->temp[1] += (data->lambda8 * t * (data->temp[1] + NEIGH_8X8_BIAS))>>10;
-         if (data->chroma) data->temp[0] += ChromaSAD((xc >> 1) + roundtab_79[xc & 0x3],
+         if (data->chroma) sad += ChromaSAD((xc >> 1) + roundtab_79[xc & 0x3],
                                                                                                          (yc >> 1) + roundtab_79[yc & 0x3], data);
-         if (data->temp[0] < data->iMinSAD[0]) {
+         if (sad < data->iMinSAD[0]) {
-                 data->iMinSAD[0] = data->temp[0];
+                 data->iMinSAD[0] = sad;
                  current[0].x = x; current[0].y = y;
-                 *dir = Direction; }
+                 *dir = Direction;
+         }
          if (data->temp[1] < data->iMinSAD[1]) {
                  data->iMinSAD[1] = data->temp[1]; current[1].x = x; current[1].y= y; }
-Line 282
+Line 349
  }
  static void
+ CheckCandidate8(const int x, const int y, const int Direction, int * const dir, const SearchData * const data)
+ {
+         int32_t sad; uint32_t t;
+         const uint8_t * Reference;
+         if ( (x > data->max_dx) || (x < data->min_dx)
+                 || (y > data->max_dy) || (y < data->min_dy) ) return;
+         if (!data->qpel_precision) Reference = GetReference(x, y, data);
+         else Reference = Interpolate8x8qpel(x, y, 0, 0, data);
+         sad = sad8(data->Cur, Reference, data->iEdgedWidth);
+         t = d_mv_bits(x, y, data->predMV, data->iFcode, data->qpel^data->qpel_precision, 0);
+         sad += (data->lambda8 * t * (sad+NEIGH_8X8_BIAS))>>10;
+         if (sad < *(data->iMinSAD)) {
+                 *(data->iMinSAD) = sad;
+                 data->currentMV->x = x; data->currentMV->y = y;
+                 *dir = Direction;
+         }
+ }
+ static void
  CheckCandidate32(const int x, const int y, const int Direction, int * const dir, const SearchData * const data)
  {
-         int t;
+         uint32_t t;
          const uint8_t * Reference;
          if ( (!(x&1) && x !=0) || (!(y&1) && y !=0) || //non-zero integer value
-Line 296
+Line 388
          data->temp[0] = sad32v_c(data->Cur, Reference, data->iEdgedWidth, data->temp + 1);
-         data->temp[0] += (data->lambda16 * t * data->temp[0])/1000;
+         data->temp[0] += (data->lambda16 * t * data->temp[0]) >> 10;
-         data->temp[1] += (data->lambda8 * t * (data->temp[1] + NEIGH_8X8_BIAS))/100;
+         data->temp[1] += (data->lambda8 * t * (data->temp[1] + NEIGH_8X8_BIAS))>>10;
          if (data->temp[0] < data->iMinSAD[0]) {
                  data->iMinSAD[0] = data->temp[0];
-Line 317
+Line 409
  static void
  CheckCandidate16no4v(const int x, const int y, const int Direction, int * const dir, const SearchData * const data)
  {
-         int32_t sad;
+         int32_t sad, xc, yc;
          const uint8_t * Reference;
-         int t;
+         uint32_t t;
          VECTOR * current;
-         if (( x > data->max_dx) || ( x < data->min_dx)
+         if ( (x > data->max_dx) | ( x < data->min_dx)
-                 || ( y > data->max_dy) || (y < data->min_dy)) return;
+                 | (y > data->max_dy) | (y < data->min_dy) ) return;
-         if (data->rrv)
+         if (data->rrv && (!(x&1) && x !=0) | (!(y&1) && y !=0) ) return; //non-zero even value
-                 if ( (!(x&1) && x !=0) || (!(y&1) && y !=0) ) return; //non-zero integer value
          if (data->qpel_precision) { // x and y are in 1/4 precision
                  Reference = Interpolate16x16qpel(x, y, 0, data);
                  current = data->currentQMV;
+                 xc = x/2; yc = y/2;
          } else {
                  Reference = GetReference(x, y, data);
                  current = data->currentMV;
+                 xc = x; yc = y;
          }
          t = d_mv_bits(x, y, data->predMV, data->iFcode,
-                                         data->qpel && !data->qpel_precision, data->rrv);
+                                         data->qpel^data->qpel_precision, data->rrv);
          sad = sad16(data->Cur, Reference, data->iEdgedWidth, 256*4096);
-         sad += (data->lambda16 * t * sad)/1000;
+         sad += (data->lambda16 * t * sad)>>10;
+         if (data->chroma) sad += ChromaSAD((xc >> 1) + roundtab_79[xc & 0x3],
+                                                                                 (yc >> 1) + roundtab_79[yc & 0x3], data);
          if (sad < *(data->iMinSAD)) {
                  *(data->iMinSAD) = sad;
                  current->x = x; current->y = y;
-                 *dir = Direction; }
+                 *dir = Direction;
+         }
  }
  static void
  CheckCandidate32I(const int x, const int y, const int Direction, int * const dir, const SearchData * const data)
  {
  // maximum speed - for P/B/I decision
+         int32_t sad;
          if (( x > data->max_dx) || ( x < data->min_dx)
                  || ( y > data->max_dy) || (y < data->min_dy)) return;
-         data->temp[0] = sad32v_c(data->Cur, data->Ref + x/2 + (y/2)*(data->iEdgedWidth),
+         sad = sad32v_c(data->Cur, data->Ref + x/2 + (y/2)*(data->iEdgedWidth),
                                                          data->iEdgedWidth, data->temp+1);
-         if (data->temp[0] < *(data->iMinSAD)) {
-                 *(data->iMinSAD) = data->temp[0];
+         if (sad < *(data->iMinSAD)) {
+                 *(data->iMinSAD) = sad;
                  data->currentMV[0].x = x; data->currentMV[0].y = y;
-                 *dir = Direction; }
+                 *dir = Direction;
+         }
          if (data->temp[1] < data->iMinSAD[1]) {
                  data->iMinSAD[1] = data->temp[1]; data->currentMV[1].x = x; data->currentMV[1].y = y; }
          if (data->temp[2] < data->iMinSAD[2]) {
-Line 375
+Line 476
  static void
  CheckCandidateInt(const int xf, const int yf, const int Direction, int * const dir, const SearchData * const data)
  {
-         int32_t sad;
+         int32_t sad, xb, yb, xcf, ycf, xcb, ycb;
-         int xb, yb, t;
+         uint32_t t;
          const uint8_t *ReferenceF, *ReferenceB;
          VECTOR *current;
-         if (( xf > data->max_dx) || ( xf < data->min_dx)
+         if ( (xf > data->max_dx) | (xf < data->min_dx)
-                 || ( yf > data->max_dy) || (yf < data->min_dy)) return;
+                 | (yf > data->max_dy) | (yf < data->min_dy) ) return;
-         if (data->qpel_precision) {
+         if (!data->qpel_precision) {
-                 ReferenceF = Interpolate16x16qpel(xf, yf, 0, data);
-                 xb = data->currentQMV[1].x; yb = data->currentQMV[1].y;
-                 current = data->currentQMV;
-                 ReferenceB = Interpolate16x16qpel(xb, yb, 1, data);
-         } else {
                  ReferenceF = GetReference(xf, yf, data);
                  xb = data->currentMV[1].x; yb = data->currentMV[1].y;
                  ReferenceB = GetReferenceB(xb, yb, 1, data);
                  current = data->currentMV;
+                 xcf = xf; ycf = yf;
+                 xcb = xb; ycb = yb;
+         } else {
+                 ReferenceF = Interpolate16x16qpel(xf, yf, 0, data);
+                 xb = data->currentQMV[1].x; yb = data->currentQMV[1].y;
+                 current = data->currentQMV;
+                 ReferenceB = Interpolate16x16qpel(xb, yb, 1, data);
+                 xcf = xf/2; ycf = yf/2;
+                 xcb = xb/2; ycb = yb/2;
          }
-         t = d_mv_bits(xf, yf, data->predMV, data->iFcode, data->qpel && !data->qpel_precision, 0)
+         t = d_mv_bits(xf, yf, data->predMV, data->iFcode, data->qpel^data->qpel_precision, 0)
-                  + d_mv_bits(xb, yb, data->bpredMV, data->iFcode, data->qpel && !data->qpel_precision, 0);
+                  + d_mv_bits(xb, yb, data->bpredMV, data->iFcode, data->qpel^data->qpel_precision, 0);
          sad = sad16bi(data->Cur, ReferenceF, ReferenceB, data->iEdgedWidth);
-         sad += (data->lambda16 * t * sad)/1000;
+         sad += (data->lambda16 * t * sad)>>10;
+         if (data->chroma) sad += ChromaSAD2((xcf >> 1) + roundtab_79[xcf & 0x3],
+                                                                                 (ycf >> 1) + roundtab_79[ycf & 0x3],
+                                                                                 (xcb >> 1) + roundtab_79[xcb & 0x3],
+                                                                                 (ycb >> 1) + roundtab_79[ycb & 0x3], data);
          if (sad < *(data->iMinSAD)) {
                  *(data->iMinSAD) = sad;
                  current->x = xf; current->y = yf;
-                 *dir = Direction; }
+                 *dir = Direction;
+         }
  }
  static void
  CheckCandidateDirect(const int x, const int y, const int Direction, int * const dir, const SearchData * const data)
  {
-         int32_t sad = 0;
+         int32_t sad = 0, xcf = 0, ycf = 0, xcb = 0, ycb = 0;
-         int k;
+         uint32_t k;
          const uint8_t *ReferenceF;
          const uint8_t *ReferenceB;
          VECTOR mvs, b_mvs;
-         const VECTOR zeroMV={0,0};
-         if (( x > 31) || ( x < -32) || ( y > 31) || (y < -32)) return;
+         if (( x > 31) | ( x < -32) | ( y > 31) | (y < -32)) return;
          for (k = 0; k < 4; k++) {
                  mvs.x = data->directmvF[k].x + x;
-Line 430
+Line 540
                          data->directmvB[k].y
                          : mvs.y - data->referencemv[k].y);
-                 if (( mvs.x > data->max_dx ) || ( mvs.x < data->min_dx )
+                 if ( (mvs.x > data->max_dx) | (mvs.x < data->min_dx)
-                         || ( mvs.y > data->max_dy ) || ( mvs.y < data->min_dy )
+                         | (mvs.y > data->max_dy) | (mvs.y < data->min_dy)
-                         || ( b_mvs.x > data->max_dx ) || ( b_mvs.x < data->min_dx )
+                         | (b_mvs.x > data->max_dx) | (b_mvs.x < data->min_dx)
-                         || ( b_mvs.y > data->max_dy ) || ( b_mvs.y < data->min_dy )) return;
+                         | (b_mvs.y > data->max_dy) | (b_mvs.y < data->min_dy) ) return;
-                 if (!data->qpel) {
+                 if (data->qpel) {
-                         mvs.x *= 2; mvs.y *= 2;
+                         xcf += mvs.x/2; ycf += mvs.y/2;
-                         b_mvs.x *= 2; b_mvs.y *= 2; //we move to qpel precision anyway
+                         xcb += b_mvs.x/2; ycb += b_mvs.y/2;
+                 } else {
+                         xcf += mvs.x; ycf += 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;
                  }
                  ReferenceF = Interpolate8x8qpel(mvs.x, mvs.y, k, 0, data);
                  ReferenceB = Interpolate8x8qpel(b_mvs.x, b_mvs.y, k, 1, data);
                  sad += sad8bi(data->Cur + 8*(k&1) + 8*(k>>1)*(data->iEdgedWidth),
-                                                 ReferenceF, ReferenceB,
+                                                 ReferenceF, ReferenceB, data->iEdgedWidth);
-                                                 data->iEdgedWidth);
                  if (sad > *(data->iMinSAD)) return;
          }
-         sad += (data->lambda16 * d_mv_bits(x, y, zeroMV, 1, 0, 0) * sad)/1000;
+         sad += (data->lambda16 * d_mv_bits(x, y, zeroMV, 1, 0, 0) * sad)>>10;
+         if (data->chroma) sad += ChromaSAD2((xcf >> 3) + roundtab_76[xcf & 0xf],
+                                                                                 (ycf >> 3) + roundtab_76[ycf & 0xf],
+                                                                                 (xcb >> 3) + roundtab_76[xcb & 0xf],
+                                                                                 (ycb >> 3) + roundtab_76[ycb & 0xf], data);
          if (sad < *(data->iMinSAD)) {
                  *(data->iMinSAD) = sad;
                  data->currentMV->x = x; data->currentMV->y = y;
-                 *dir = Direction; }
+                 *dir = Direction;
+         }
  }
  static void
  CheckCandidateDirectno4v(const int x, const int y, const int Direction, int * const dir, const SearchData * const data)
  {
-         int32_t sad;
+         int32_t sad, xcf, ycf, xcb, ycb;
          const uint8_t *ReferenceF;
          const uint8_t *ReferenceB;
          VECTOR mvs, b_mvs;
-         const VECTOR zeroMV = {0,0};
-         if (( x > 31) || ( x < -32) || ( y > 31) || (y < -32)) return;
+         if (( x > 31) | ( x < -32) | ( y > 31) | (y < -32)) return;
          mvs.x = data->directmvF[0].x + x;
          b_mvs.x = ((x == 0) ?
-Line 477
+Line 597
                  data->directmvB[0].y
                  : mvs.y - data->referencemv[0].y);
-         if (( mvs.x > data->max_dx ) || ( mvs.x < data->min_dx )
+         if ( (mvs.x > data->max_dx) | (mvs.x < data->min_dx)
-                 || ( mvs.y > data->max_dy ) || ( mvs.y < data->min_dy )
+                 | (mvs.y > data->max_dy) | (mvs.y < data->min_dy)
-                 || ( b_mvs.x > data->max_dx ) || ( b_mvs.x < data->min_dx )
+                 | (b_mvs.x > data->max_dx) | (b_mvs.x < data->min_dx)
-                 || ( b_mvs.y > data->max_dy ) || ( b_mvs.y < data->min_dy )) return;
+                 | (b_mvs.y > data->max_dy) | (b_mvs.y < data->min_dy) ) return;
-         if (!data->qpel) {
+         if (data->qpel) {
-                         mvs.x *= 2; mvs.y *= 2;
+                 xcf = 4*(mvs.x/2); ycf = 4*(mvs.y/2);
-                         b_mvs.x *= 2; b_mvs.y *= 2; //we move to qpel precision anyway
+                 xcb = 4*(b_mvs.x/2); ycb = 4*(b_mvs.y/2);
-         }
          ReferenceF = Interpolate16x16qpel(mvs.x, mvs.y, 0, data);
          ReferenceB = Interpolate16x16qpel(b_mvs.x, b_mvs.y, 1, data);
+         } else {
-         sad = sad16bi(data->Cur, ReferenceF, ReferenceB, data->iEdgedWidth);
+                 xcf = 4*mvs.x; ycf = 4*mvs.y;
-         sad += (data->lambda16 * d_mv_bits(x, y, zeroMV, 1, 0, 0) * sad)/1000;
+                 xcb = 4*b_mvs.x; ycb = 4*b_mvs.y;
+                 ReferenceF = GetReference(mvs.x, mvs.y, data);
-         if (sad < *(data->iMinSAD)) {
+                 ReferenceB = GetReferenceB(b_mvs.x, b_mvs.y, 1, data);
-                 *(data->iMinSAD) = sad;
-                 data->currentMV->x = x; data->currentMV->y = y;
-                 *dir = Direction; }
  }
- static void
+         sad = sad16bi(data->Cur, ReferenceF, ReferenceB, data->iEdgedWidth);
- CheckCandidate8(const int x, const int y, const int Direction, int * const dir, const SearchData * const data)
+         sad += (data->lambda16 * d_mv_bits(x, y, zeroMV, 1, 0, 0) * sad)>>10;
- {
-         int32_t sad; int t;
-         const uint8_t * Reference;
-         if (( x > data->max_dx) || ( x < data->min_dx)
-                 || ( y > data->max_dy) || (y < data->min_dy)) return;
-         if (data->qpel) Reference = Interpolate16x16qpel(x, y, 0, data);
-         else Reference =  GetReference(x, y, data);
-         sad = sad8(data->Cur, Reference, data->iEdgedWidth);
-         t = d_mv_bits(x, y, data->predMV, data->iFcode, data->qpel && !data->qpel_precision, 0);
-         sad += (data->lambda8 * t * (sad+NEIGH_8X8_BIAS))/100;
+         if (data->chroma) sad += ChromaSAD2((xcf >> 3) + roundtab_76[xcf & 0xf],
+                                                                                 (ycf >> 3) + roundtab_76[ycf & 0xf],
+                                                                                 (xcb >> 3) + roundtab_76[xcb & 0xf],
+                                                                                 (ycb >> 3) + roundtab_76[ycb & 0xf], data);
          if (sad < *(data->iMinSAD)) {
                  *(data->iMinSAD) = sad;
                  data->currentMV->x = x; data->currentMV->y = y;
-                 *dir = Direction; }
+                 *dir = Direction;
+         }
  }
  /* CHECK_CANDIATE FUNCTIONS END */
-Line 542
+Line 650
                  /* now we're doing diagonal checks near our candidate */
-                 if (iDirection) {               //checking if anything found
+                 if (iDirection) {               //if anything found
                          bDirection = iDirection;
                          iDirection = 0;
                          x = data->currentMV->x; y = data->currentMV->y;
-Line 668
+Line 776
  /* MAINSEARCH FUNCTIONS END */
- /* HALFPELREFINE COULD BE A MAINSEARCH FUNCTION, BUT THERE IS NO NEED FOR IT */
  static void
  SubpelRefine(const SearchData * const data)
  {
  /* Do a half-pel or q-pel refinement */
-         VECTOR backupMV;
+         const VECTOR centerMV = data->qpel_precision ? *data->currentQMV : *data->currentMV;
-         int iDirection; //not needed
+         int iDirection; //only needed because macro expects it
-         if (data->qpel_precision)
+         CHECK_CANDIDATE(centerMV.x, centerMV.y - 1, 0);
-                 backupMV = *(data->currentQMV);
+         CHECK_CANDIDATE(centerMV.x + 1, centerMV.y - 1, 0);
-         else backupMV = *(data->currentMV);
+         CHECK_CANDIDATE(centerMV.x + 1, centerMV.y, 0);
+         CHECK_CANDIDATE(centerMV.x + 1, centerMV.y + 1, 0);
-         CHECK_CANDIDATE(backupMV.x, backupMV.y - 1, 0);
+         CHECK_CANDIDATE(centerMV.x, centerMV.y + 1, 0);
-         CHECK_CANDIDATE(backupMV.x + 1, backupMV.y - 1, 0);
+         CHECK_CANDIDATE(centerMV.x - 1, centerMV.y + 1, 0);
-         CHECK_CANDIDATE(backupMV.x + 1, backupMV.y, 0);
+         CHECK_CANDIDATE(centerMV.x - 1, centerMV.y, 0);
-         CHECK_CANDIDATE(backupMV.x + 1, backupMV.y + 1, 0);
+         CHECK_CANDIDATE(centerMV.x - 1, centerMV.y - 1, 0);
-         CHECK_CANDIDATE(backupMV.x, backupMV.y + 1, 0);
-         CHECK_CANDIDATE(backupMV.x - 1, backupMV.y + 1, 0);
-         CHECK_CANDIDATE(backupMV.x - 1, backupMV.y, 0);
-         CHECK_CANDIDATE(backupMV.x - 1, backupMV.y - 1, 0);
  }
  static __inline int
  SkipDecisionP(const IMAGE * current, const IMAGE * reference,
                                                          const int x, const int y,
-                                                         const uint32_t iEdgedWidth, const uint32_t iQuant, int rrv)
+                                                         const uint32_t stride, const uint32_t iQuant, int rrv)
  {
- /*      keep repeating checks for all b-frames before this P frame,
+         if(!rrv) {
-         to make sure that SKIP is possible (todo)
+                 uint32_t sadC = sad8(current->u + x*8 + y*stride*8,
-         how: if skip is not possible set sad00 to a very high value */
+                                                 reference->u + x*8 + y*stride*8, stride);
-         if(rrv) {
-                 uint32_t sadC = sad16(current->u + x*16 + y*(iEdgedWidth/2)*16,
-                                                 reference->u + x*16 + y*(iEdgedWidth/2)*16, iEdgedWidth/2, 256*4096);
-                 if (sadC > iQuant * MAX_CHROMA_SAD_FOR_SKIP*4) return 0;
-                 sadC += sad16(current->v + (x + y*(iEdgedWidth/2))*16,
-                                                 reference->v + (x + y*(iEdgedWidth/2))*16, iEdgedWidth/2, 256*4096);
-                 if (sadC > iQuant * MAX_CHROMA_SAD_FOR_SKIP*4) return 0;
-                 return 1;
-         } else {
-                 uint32_t sadC = sad8(current->u + x*8 + y*(iEdgedWidth/2)*8,
-                                                 reference->u + x*8 + y*(iEdgedWidth/2)*8, iEdgedWidth/2);
                  if (sadC > iQuant * MAX_CHROMA_SAD_FOR_SKIP) return 0;
-                 sadC += sad8(current->v + (x + y*(iEdgedWidth/2))*8,
+                 sadC += sad8(current->v + (x + y*stride)*8,
-                                                 reference->v + (x + y*(iEdgedWidth/2))*8, iEdgedWidth/2);
+                                                 reference->v + (x + y*stride)*8, stride);
                  if (sadC > iQuant * MAX_CHROMA_SAD_FOR_SKIP) return 0;
                  return 1;
+         } else {
+                 uint32_t sadC = sad16(current->u + x*16 + y*stride*16,
+                                                 reference->u + x*16 + y*stride*16, stride, 256*4096);
+                 if (sadC > iQuant * MAX_CHROMA_SAD_FOR_SKIP*4) return 0;
+                 sadC += sad16(current->v + (x + y*stride)*16,
+                                                 reference->v + (x + y*stride)*16, stride, 256*4096);
+                 if (sadC > iQuant * MAX_CHROMA_SAD_FOR_SKIP*4) return 0;
+                 return 1;
          }
  }
-Line 723
+Line 823
  SkipMacroblockP(MACROBLOCK *pMB, const int32_t sad)
  {
          pMB->mode = MODE_NOT_CODED;
-         pMB->mvs[0].x = pMB->mvs[1].x = pMB->mvs[2].x = pMB->mvs[3].x = 0;
+         pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = zeroMV;
-         pMB->mvs[0].y = pMB->mvs[1].y = pMB->mvs[2].y = pMB->mvs[3].y = 0;
+         pMB->qmvs[0] = pMB->qmvs[1] = pMB->qmvs[2] = pMB->qmvs[3] = zeroMV;
-         pMB->qmvs[0].x = pMB->qmvs[1].x = pMB->qmvs[2].x = pMB->qmvs[3].x = 0;
-         pMB->qmvs[0].y = pMB->qmvs[1].y = pMB->qmvs[2].y = pMB->qmvs[3].y = 0;
          pMB->sad16 = pMB->sad8[0] = pMB->sad8[1] = pMB->sad8[2] = pMB->sad8[3] = sad;
  }
-Line 745
+Line 841
          const IMAGE *const pCurrent = &current->image;
          const IMAGE *const pRef = &reference->image;
-         const VECTOR zeroMV = { 0, 0 };
          uint32_t mb_width = pParam->mb_width;
          uint32_t mb_height = pParam->mb_height;
+         const uint32_t iEdgedWidth = pParam->edged_width;
          uint32_t x, y;
          uint32_t iIntra = 0;
-Line 761
+Line 856
          int32_t iMinSAD[5];
          SearchData Data;
          memset(&Data, 0, sizeof(SearchData));
-         Data.iEdgedWidth = pParam->edged_width;
+         Data.iEdgedWidth = iEdgedWidth;
          Data.currentMV = currentMV;
          Data.currentQMV = currentQMV;
          Data.iMinSAD = iMinSAD;
-Line 769
+Line 864
          Data.iFcode = current->fcode;
          Data.rounding = pParam->m_rounding_type;
          Data.qpel = pParam->m_quarterpel;
-         Data.chroma = current->global_flags & XVID_ME_COLOUR;
+         Data.chroma = current->motion_flags & PMV_CHROMA16;
          Data.rrv = current->global_flags & XVID_REDUCED;
          if ((current->global_flags & XVID_REDUCED)) {
-Line 785
+Line 880
                  for (x = 0; x < mb_width; x++)  {
                          MACROBLOCK *pMB = &pMBs[x + y * pParam->mb_width];
-                         if (Data.rrv) pMB->sad16 =
+                         if (!Data.rrv) pMB->sad16 =
-                                 sad32v_c(pCurrent->y + (x + y * pParam->edged_width) * 32,
+                                 sad16v(pCurrent->y + (x + y * iEdgedWidth) * 16,
-                                                         pRef->y + (x + y * pParam->edged_width) * 32,
+                                                         pRef->y + (x + y * iEdgedWidth) * 16,
                                                          pParam->edged_width, pMB->sad8 );
                          else pMB->sad16 =
-                                 sad16v(pCurrent->y + (x + y * pParam->edged_width) * 16,
+                                 sad32v_c(pCurrent->y + (x + y * iEdgedWidth) * 32,
-                                                         pRef->y + (x + y * pParam->edged_width) * 16,
+                                                         pRef->y + (x + y * iEdgedWidth) * 32,
                                                          pParam->edged_width, pMB->sad8 );
                          if (Data.chroma) {
-                                 pMB->sad16 += sad8(pCurrent->u + x*8 + y*(pParam->edged_width/2)*8,
+                                 Data.temp[7] = sad8(pCurrent->u + x*8 + y*(iEdgedWidth/2)*8,
-                                                                 pRef->u + x*8 + y*(pParam->edged_width/2)*8, pParam->edged_width/2);
+                                                                         pRef->u + x*8 + y*(iEdgedWidth/2)*8, iEdgedWidth/2)
+                                                                 + sad8(pCurrent->v + (x + y*(iEdgedWidth/2))*8,
-                                 pMB->sad16 += sad8(pCurrent->v + (x + y*(pParam->edged_width/2))*8,
+                                                                         pRef->v + (x + y*(iEdgedWidth/2))*8, iEdgedWidth/2);
-                                                                 pRef->v + (x + y*(pParam->edged_width/2))*8, pParam->edged_width/2);
+                                 pMB->sad16 += Data.temp[7];
                          }
-                         sad00 = pMB->sad16; //if no gmc; else sad00 = (..)
+                         sad00 = pMB->sad16;
                          if (!(current->global_flags & XVID_LUMIMASKING)) {
                                  pMB->dquant = NO_CHANGE;
-                                 pMB->quant = current->quant;
                          } else {
                                  if (pMB->dquant != NO_CHANGE) {
                                          quant += DQtab[pMB->dquant];
                                          if (quant > 31) quant = 31;
                                          else if (quant < 1) quant = 1;
                                  }
-                                 pMB->quant = quant;
                          }
+                         pMB->quant = current->quant;
  //initial skip decision
  /* no early skip for GMC (global vector = skip vector is unknown!)  */
-                         if (current->coding_type == P_VOP)      { /* no fast SKIP for S(GMC)-VOPs */
+                         if (!(current->global_flags & XVID_GMC))        { /* no fast SKIP for S(GMC)-VOPs */
                                  if (pMB->dquant == NO_CHANGE && sad00 < pMB->quant * INITIAL_SKIP_THRESH  * (Data.rrv ? 4:1) )
-                                         if (Data.chroma || SkipDecisionP(pCurrent, pRef, x, y, pParam->edged_width, pMB->quant, Data.rrv)) {
+                                         if (Data.chroma || SkipDecisionP(pCurrent, pRef, x, y, iEdgedWidth/2, pMB->quant, Data.rrv)) {
                                                  SkipMacroblockP(pMB, sad00);
                                                  continue;
                                          }
-Line 833
+Line 927
                                                  current->global_flags & XVID_INTER4V, pMB);
  /* final skip decision, a.k.a. "the vector you found, really that good?" */
-                         if (current->coding_type == P_VOP)      {
+                         if (!(current->global_flags & XVID_GMC))        {
                                  if ( (pMB->dquant == NO_CHANGE) && (sad00 < pMB->quant * MAX_SAD00_FOR_SKIP)
                                          && ((100*pMB->sad16)/(sad00+1) > FINAL_SKIP_THRESH * (Data.rrv ? 4:1)) )
-                                         if (Data.chroma || SkipDecisionP(pCurrent, pRef, x, y, pParam->edged_width, pMB->quant, Data.rrv)) {
+                                         if (Data.chroma || SkipDecisionP(pCurrent, pRef, x, y, iEdgedWidth/2, pMB->quant, Data.rrv)) {
                                                  SkipMacroblockP(pMB, sad00);
                                                  continue;
                                          }
-Line 852
+Line 946
                                  if ((pMB - 1)->mode == MODE_INTRA ) InterBias -= 80;
                          if (Data.chroma) InterBias += 50; // to compensate bigger SAD
-                         if (Data.rrv) InterBias *= 4; //??
+                         if (Data.rrv) InterBias *= 4;
                          if (InterBias < pMB->sad16)  {
                                  int32_t deviation;
-                                 if (Data.rrv) {
+                                 if (!Data.rrv)
-                                         deviation = dev16(pCurrent->y + (x + y * pParam->edged_width) * 32,
+                                         deviation = dev16(pCurrent->y + (x + y * iEdgedWidth) * 16, iEdgedWidth);
-                                                                                 pParam->edged_width)
+                                 else {
-                                                 + dev16(pCurrent->y + (x + y * pParam->edged_width) * 32 + 16,
+                                         deviation = dev16(pCurrent->y + (x + y * iEdgedWidth) * 32, iEdgedWidth)
-                                                                                 pParam->edged_width)
+                                                 + dev16(pCurrent->y + (x + y * iEdgedWidth) * 32 + 16, iEdgedWidth)
-                                                 + dev16(pCurrent->y + (x + y * pParam->edged_width) * 32 + 16 * pParam->edged_width,
+                                                 + dev16(pCurrent->y + (x + y * iEdgedWidth) * 32 + 16 * iEdgedWidth, iEdgedWidth)
-                                                                                 pParam->edged_width)
+                                                 + dev16(pCurrent->y + (x + y * iEdgedWidth) * 32 + 16 * (iEdgedWidth+1), iEdgedWidth);
-                                                 + dev16(pCurrent->y + (x + y * pParam->edged_width) * 32 + 16 * (pParam->edged_width+1),
+                                 }
-                                                                                 pParam->edged_width);
-                                 } else
-                                         deviation = dev16(pCurrent->y + (x + y * pParam->edged_width) * 16,
-                                                   pParam->edged_width);
                                          if (deviation < (pMB->sad16 - InterBias)) {
                                          if (++iIntra >= iLimit) return 1;
+                                         SkipMacroblockP(pMB, 0); //same thing
                                          pMB->mode = MODE_INTRA;
-                                         pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] =
-                                                         pMB->mvs[3] = zeroMV;
-                                         pMB->qmvs[0] = pMB->qmvs[1] = pMB->qmvs[2] =
-                                                         pMB->qmvs[3] = zeroMV;
-                                         pMB->sad16 = pMB->sad8[0] = pMB->sad8[1] = pMB->sad8[2] =
-                                                 pMB->sad8[3] = 0;
                                  }
                          }
                  }
          }
-         if (current->coding_type == S_VOP)      /* first GMC step only for S(GMC)-VOPs */
+         if (current->global_flags & XVID_GMC )  /* GMC only for S(GMC)-VOPs */
-                 current->GMC_MV = GlobalMotionEst( pMBs, pParam, current->fcode );
+         {
-         else
+                 current->warp = GlobalMotionEst( pMBs, pParam, current, reference, pRefH, pRefV, pRefHV);
-                 current->GMC_MV = zeroMV;
+         }
          return 0;
  }
- #define PMV_HALFPEL16 (PMV_HALFPELDIAMOND16|PMV_HALFPELREFINE16)
  static __inline int
  make_mask(const VECTOR * const pmv, const int i)
  {
-Line 901
+Line 983
          for (j = 0; j < i; j++) {
                  if (MVequal(pmv[i], pmv[j])) return 0; // same vector has been checked already
                  if (pmv[i].x == pmv[j].x) {
-                         if (pmv[i].y == pmv[j].y + iDiamondSize) { mask &= ~4; continue; }
+                         if (pmv[i].y == pmv[j].y + iDiamondSize) mask &= ~4;
-                         if (pmv[i].y == pmv[j].y - iDiamondSize) { mask &= ~8; continue; }
+                         else if (pmv[i].y == pmv[j].y - iDiamondSize) mask &= ~8;
                  } else
                          if (pmv[i].y == pmv[j].y) {
-                                 if (pmv[i].x == pmv[j].x + iDiamondSize) { mask &= ~1; continue; }
+                                 if (pmv[i].x == pmv[j].x + iDiamondSize) mask &= ~1;
-                                 if (pmv[i].x == pmv[j].x - iDiamondSize) { mask &= ~2; continue; }
+                                 else if (pmv[i].x == pmv[j].x - iDiamondSize) mask &= ~2;
                          }
          }
          return mask;
-Line 947
+Line 1029
          if (rrv) {
                  int i;
                  for (i = 0; i < 7; i++) {
-                         pmv[i].x = RRV_MV_SCALEUP(pmv[i].x); // halfzero->halfpel
+                         pmv[i].x = RRV_MV_SCALEUP(pmv[i].x);
                          pmv[i].y = RRV_MV_SCALEUP(pmv[i].y);
                  }
          }
-Line 977
+Line 1059
          get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 16,
                                                  pParam->width, pParam->height, Data->iFcode - Data->qpel, 0, Data->rrv);
-         get_pmvdata2(pMBs, pParam->mb_width, 0, x, y, 0, pmv, Data->temp);  //has to be changed to get_pmv(2)()
+         get_pmvdata2(pMBs, pParam->mb_width, 0, x, y, 0, pmv, Data->temp);
-         Data->temp[5] = Data->temp[7] = 256*4096; // to reset chroma-sad cache
+         Data->temp[5] = Data->temp[6] = 0; // chroma-sad cache
-         if (Data->rrv) i = 2; else i = 1;
+         i = Data->rrv ? 2 : 1;
          Data->Cur = pCur->y + (x + y * Data->iEdgedWidth) * 16*i;
          Data->CurV = pCur->v + (x + y * (Data->iEdgedWidth/2)) * 8*i;
          Data->CurU = pCur->u + (x + y * (Data->iEdgedWidth/2)) * 8*i;
-Line 1001
+Line 1083
          for(i = 0;  i < 5; i++)
                  Data->currentMV[i].x = Data->currentMV[i].y = 0;
-         if (pParam->m_quarterpel) Data->predMV = get_qpmv2(pMBs, pParam->mb_width, 0, x, y, 0);
+         if (Data->qpel) Data->predMV = get_qpmv2(pMBs, pParam->mb_width, 0, x, y, 0);
          else Data->predMV = pmv[0];
          i = d_mv_bits(0, 0, Data->predMV, Data->iFcode, 0, 0);
-         Data->iMinSAD[0] = pMB->sad16 + (Data->lambda16 * i * pMB->sad16)/1000;
+         Data->iMinSAD[0] = pMB->sad16 + ((Data->lambda16 * i * pMB->sad16)>>10);
-         Data->iMinSAD[1] = pMB->sad8[0] + (Data->lambda8 * i * (pMB->sad8[0]+NEIGH_8X8_BIAS))/100;
+         Data->iMinSAD[1] = pMB->sad8[0] + ((Data->lambda8 * i * (pMB->sad8[0]+NEIGH_8X8_BIAS)) >> 10);
          Data->iMinSAD[2] = pMB->sad8[1];
          Data->iMinSAD[3] = pMB->sad8[2];
          Data->iMinSAD[4] = pMB->sad8[3];
-         if ((x == 0) && (y == 0)) threshA = 512;
+         if (x | y) {
-         else {
                  threshA = Data->temp[0]; // that's when we keep this SAD atm
                  if (threshA < 512) threshA = 512;
-                 if (threshA > 1024) threshA = 1024; }
+                 else if (threshA > 1024) threshA = 1024;
+         } else threshA = 512;
          PreparePredictionsP(pmv, x, y, pParam->mb_width, pParam->mb_height,
                                          prevMBs + x + y * pParam->mb_width, Data->rrv);
-         if (Data->rrv) CheckCandidate = CheckCandidate32;
+         if (!Data->rrv) {
-         else if (inter4v || Data->chroma) CheckCandidate = CheckCandidate16;
+                 if (inter4v | Data->chroma) CheckCandidate = CheckCandidate16;
                  else CheckCandidate = CheckCandidate16no4v; //for extra speed
+         } else CheckCandidate = CheckCandidate32;
- /* main loop. checking all predictions */
+ /* main loop. checking all predictions (but first, which is 0,0 and has been checked in MotionEstimation())*/
          for (i = 1; i < 7; i++) {
                  if (!(mask = make_mask(pmv, i)) ) continue;
-                 (*CheckCandidate)(pmv[i].x, pmv[i].y, mask, &iDirection, Data);
+                 CheckCandidate(pmv[i].x, pmv[i].y, mask, &iDirection, Data);
                  if (Data->iMinSAD[0] <= threshA) break;
          }
          if ((Data->iMinSAD[0] <= threshA) ||
                          (MVequal(Data->currentMV[0], (prevMBs+x+y*pParam->mb_width)->mvs[0]) &&
-                         (Data->iMinSAD[0] < (prevMBs+x+y*pParam->mb_width)->sad16))) {
+                         (Data->iMinSAD[0] < (prevMBs+x+y*pParam->mb_width)->sad16)))
                  inter4v = 0;
-         } else {
+         else {
                  MainSearchFunc * MainSearchPtr;
                  if (MotionFlags & PMV_USESQUARES16) MainSearchPtr = SquareSearch;
-Line 1061
+Line 1144
                          if (!(MVequal(startMV, backupMV))) {
                                  bSAD = Data->iMinSAD[0]; Data->iMinSAD[0] = MV_MAX_ERROR;
-                                 (*CheckCandidate)(startMV.x, startMV.y, 255, &iDirection, Data);
+                                 CheckCandidate(startMV.x, startMV.y, 255, &iDirection, Data);
                                  (*MainSearchPtr)(startMV.x, startMV.y, Data, 255);
                                  if (bSAD < Data->iMinSAD[0]) {
                                          Data->currentMV[0] = backupMV;
-Line 1069
+Line 1152
                          }
                          backupMV = Data->currentMV[0];
-                         if (!MotionFlags & PMV_HALFPELREFINE16 || Data->rrv) startMV.x = startMV.y = 0;
+                         if (MotionFlags & PMV_HALFPELREFINE16 && !Data->rrv) startMV.x = startMV.y = 1;
-                         else startMV.x = startMV.y = 1;
+                         else startMV.x = startMV.y = 0;
                          if (!(MVequal(startMV, backupMV))) {
                                  bSAD = Data->iMinSAD[0]; Data->iMinSAD[0] = MV_MAX_ERROR;
-                                 (*CheckCandidate)(startMV.x, startMV.y, 255, &iDirection, Data);
+                                 CheckCandidate(startMV.x, startMV.y, 255, &iDirection, Data);
                                  (*MainSearchPtr)(startMV.x, startMV.y, Data, 255);
                                  if (bSAD < Data->iMinSAD[0]) {
                                          Data->currentMV[0] = backupMV;
-Line 1090
+Line 1173
                  Data->currentQMV[i].y = 2 * Data->currentMV[i].y;
          }
-         if((Data->qpel) && (MotionFlags & PMV_QUARTERPELREFINE16)) {
+         if (Data->qpel && MotionFlags & PMV_QUARTERPELREFINE16) {
                  Data->qpel_precision = 1;
                  get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 16,
                                  pParam->width, pParam->height, Data->iFcode, 1, 0);
-Line 1102
+Line 1184
          if (Data->iMinSAD[0] < (int32_t)iQuant * 30 ) inter4v = 0;
          if (inter4v) {
                  SearchData Data8;
-                 Data8.iFcode = Data->iFcode;
+                 memcpy(&Data8, Data, sizeof(SearchData)); //quick copy of common data
-                 Data8.lambda8 = Data->lambda8;
-                 Data8.iEdgedWidth = Data->iEdgedWidth;
-                 Data8.RefQ = Data->RefQ;
-                 Data8.qpel = Data->qpel;
-                 Data8.rrv = Data->rrv;
                  Search8(Data, 2*x, 2*y, MotionFlags, pParam, pMB, pMBs, 0, &Data8);
                  Search8(Data, 2*x + 1, 2*y, MotionFlags, pParam, pMB, pMBs, 1, &Data8);
                  Search8(Data, 2*x, 2*y + 1, MotionFlags, pParam, pMB, pMBs, 2, &Data8);
                  Search8(Data, 2*x + 1, 2*y + 1, MotionFlags, pParam, pMB, pMBs, 3, &Data8);
                  if (Data->chroma) {
-                         int sumx, sumy;
+                         int sumx = 0, sumy = 0;
+                         const int div = 1 + Data->qpel;
+                         const VECTOR * const mv = Data->qpel ? pMB->qmvs : pMB->mvs;
-                         if(pParam->m_quarterpel) {
+                         for (i = 0; i < 4; i++) {
-                                 sumx = pMB->qmvs[0].x/2 + pMB->qmvs[1].x/2 + pMB->qmvs[2].x/2 + pMB->qmvs[3].x/2;
+                                 sumx += mv[i].x / div;
-                                 sumy = pMB->qmvs[0].y/2 + pMB->qmvs[1].y/2 + pMB->qmvs[2].y/2 + pMB->qmvs[3].y/2;
+                                 sumy += mv[i].y / div;
-                         } else {
-                                 sumx = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
-                                 sumy = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
                          }
                          Data->iMinSAD[1] += ChromaSAD(  (sumx >> 3) + roundtab_76[sumx & 0xf],
-Line 1139
+Line 1216
                          Data->iMinSAD[3] + Data->iMinSAD[4] + IMV16X16 * (int32_t)iQuant )) {
  // INTER MODE
                  pMB->mode = MODE_INTER;
-                 pMB->mvs[0] = pMB->mvs[1]
+                 pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = Data->currentMV[0];
-                         = pMB->mvs[2] = pMB->mvs[3] = Data->currentMV[0];
+                 pMB->sad16 = pMB->sad8[0] = pMB->sad8[1] = pMB->sad8[2] = pMB->sad8[3] = Data->iMinSAD[0];
-                 pMB->sad16 = pMB->sad8[0] = pMB->sad8[1] =
+                 if(Data->qpel) {
-                         pMB->sad8[2] = pMB->sad8[3] =  Data->iMinSAD[0];
-                 if(pParam->m_quarterpel) {
                          pMB->qmvs[0] = pMB->qmvs[1]
                                  = pMB->qmvs[2] = pMB->qmvs[3] = Data->currentQMV[0];
                          pMB->pmvs[0].x = Data->currentQMV[0].x - Data->predMV.x;
-Line 1177
+Line 1251
          Data->currentMV = OldData->currentMV + 1 + block;
          Data->currentQMV = OldData->currentQMV + 1 + block;
-         if(pParam->m_quarterpel) {
+         if(Data->qpel) {
                  Data->predMV = get_qpmv2(pMBs, pParam->mb_width, 0, x/2, y/2, block);
                  if (block != 0) i = d_mv_bits(  Data->currentQMV->x, Data->currentQMV->y,
                                                                                  Data->predMV, Data->iFcode, 0, 0);
          } else {
                  Data->predMV = get_pmv2(pMBs, pParam->mb_width, 0, x/2, y/2, block);
-                 if (block != 0) {
                          if (block != 0) i = d_mv_bits(  Data->currentMV->x, Data->currentMV->y,
                                                                                          Data->predMV, Data->iFcode, 0, Data->rrv);
                  }
-         }
-         *(Data->iMinSAD) += (Data->lambda8 * i * (*Data->iMinSAD + NEIGH_8X8_BIAS))/100;
+         *(Data->iMinSAD) += (Data->lambda8 * i * (*Data->iMinSAD + NEIGH_8X8_BIAS))>>10;
-         if (MotionFlags & (PMV_EXTSEARCH8|PMV_HALFPELREFINE8)) {
+         if (MotionFlags & (PMV_EXTSEARCH8|PMV_HALFPELREFINE8|PMV_QUARTERPELREFINE8)) {
                  if (Data->rrv) i = 2; else i = 1;
-                 Data->Ref = OldData->Ref + i*8 * ((block&1) + pParam->edged_width*(block>>1));
+                 Data->Ref = OldData->Ref + i * 8 * ((block&1) + Data->iEdgedWidth*(block>>1));
-                 Data->RefH = OldData->RefH + i*8 * ((block&1) + pParam->edged_width*(block>>1));
+                 Data->RefH = OldData->RefH + i * 8 * ((block&1) + Data->iEdgedWidth*(block>>1));
-                 Data->RefV = OldData->RefV + i*8 * ((block&1) + pParam->edged_width*(block>>1));
+                 Data->RefV = OldData->RefV + i * 8 * ((block&1) + Data->iEdgedWidth*(block>>1));
-                 Data->RefHV = OldData->RefHV + i*8 * ((block&1) + pParam->edged_width*(block>>1));
+                 Data->RefHV = OldData->RefHV + i * 8 * ((block&1) + Data->iEdgedWidth*(block>>1));
-                 Data->Cur = OldData->Cur + i*8 * ((block&1) + pParam->edged_width*(block>>1));
+                 Data->Cur = OldData->Cur + i * 8 * ((block&1) + Data->iEdgedWidth*(block>>1));
                  Data->qpel_precision = 0;
                  get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 8,
-                                         pParam->width, pParam->height, OldData->iFcode - Data->qpel, 0, Data->rrv);
+                                         pParam->width, pParam->height, Data->iFcode - Data->qpel, 0, Data->rrv);
-                 if (Data->rrv) CheckCandidate = CheckCandidate16no4v;
+                 if (!Data->rrv) CheckCandidate = CheckCandidate8;
-                 else CheckCandidate = CheckCandidate8;
+                 else CheckCandidate = CheckCandidate16no4v;
                  if (MotionFlags & PMV_EXTSEARCH8) {
                          int32_t temp_sad = *(Data->iMinSAD); // store current MinSAD
-Line 1236
+Line 1307
                          }
                  }
-                 if(Data->qpel) {
+                 if (Data->qpel && MotionFlags & PMV_QUARTERPELREFINE8) {
-                         if((!(Data->currentQMV->x & 1)) && (!(Data->currentQMV->y & 1)) &&
-                                 (MotionFlags & PMV_QUARTERPELREFINE8)) {
                          Data->qpel_precision = 1;
                          get_range(&Data->min_dx, &Data->max_dx, &Data->min_dy, &Data->max_dy, x, y, 8,
-                                 pParam->width, pParam->height, OldData->iFcode, 1, 0);
+                                         pParam->width, pParam->height, Data->iFcode, 1, 0);
                          SubpelRefine(Data);
                          }
                  }
-         }
          if (Data->rrv) {
                          Data->currentMV->x = RRV_MV_SCALEDOWN(Data->currentMV->x);
-Line 1255
+Line 1323
          if(Data->qpel) {
                  pMB->pmvs[block].x = Data->currentQMV->x - Data->predMV.x;
                  pMB->pmvs[block].y = Data->currentQMV->y - Data->predMV.y;
-                 pMB->qmvs[block] = *(Data->currentQMV);
+                 pMB->qmvs[block] = *Data->currentQMV;
          } else {
                  pMB->pmvs[block].x = Data->currentMV->x - Data->predMV.x;
                  pMB->pmvs[block].y = Data->currentMV->y - Data->predMV.y;
          }
-         pMB->mvs[block] = *(Data->currentMV);
+         pMB->mvs[block] = *Data->currentMV;
-         pMB->sad8[block] =  4 * (*Data->iMinSAD);
+         pMB->sad8[block] = 4 * *Data->iMinSAD;
  }
- /* B-frames code starts here */
+ /* motion estimation for B-frames */
  static __inline VECTOR
  ChoosePred(const MACROBLOCK * const pMB, const uint32_t mode)
  {
  /* the stupidiest function ever */
-         if (mode == MODE_FORWARD) return pMB->mvs[0];
+         return (mode == MODE_FORWARD ? pMB->mvs[0] : pMB->b_mvs[0]);
-         else return pMB->b_mvs[0];
  }
  static void __inline
-Line 1305
+Line 1372
                  pmv[5].x = EVEN(pmv[5].x); pmv[5].y = EVEN(pmv[5].y);
          } else pmv[5].x = pmv[5].y = 0;
-         if ((x != 0)&&(y != 0)) {
+         if (x != 0 && y != 0) {
                  pmv[6] = ChoosePred(pMB-1-iWcount, mode_curr);
                  pmv[6].x = EVEN(pmv[6].x); pmv[6].y = EVEN(pmv[6].y);
          } else pmv[6].x = pmv[6].y = 0;
- // more?
  }
- /* search backward or forward, for b-frames */
+ /* search backward or forward */
  static void
- SearchBF(       const uint8_t * const pRef,
+ SearchBF(       const IMAGE * const pRef,
                          const uint8_t * const pRefH,
                          const uint8_t * const pRefV,
                          const uint8_t * const pRefHV,
-Line 1332
+Line 1397
                          SearchData * const Data)
  {
-         const int32_t iEdgedWidth = pParam->edged_width;
          int i, iDirection = 255, mask;
          VECTOR pmv[7];
          MainSearchFunc *MainSearchPtr;
          *Data->iMinSAD = MV_MAX_ERROR;
          Data->iFcode = iFcode;
          Data->qpel_precision = 0;
+         Data->temp[5] = Data->temp[6] = Data->temp[7] = 256*4096; // reset chroma-sad cache
-         Data->Ref = pRef + (x + y * iEdgedWidth) * 16;
+         Data->Ref = pRef->y + (x + y * Data->iEdgedWidth) * 16;
-         Data->RefH = pRefH + (x + y * iEdgedWidth) * 16;
+         Data->RefH = pRefH + (x + y * Data->iEdgedWidth) * 16;
-         Data->RefV = pRefV + (x + y * iEdgedWidth) * 16;
+         Data->RefV = pRefV + (x + y * Data->iEdgedWidth) * 16;
-         Data->RefHV = pRefHV + (x + y * iEdgedWidth) * 16;
+         Data->RefHV = pRefHV + (x + y * Data->iEdgedWidth) * 16;
+         Data->RefCU = pRef->u + (x + y * Data->iEdgedWidth/2) * 8;
+         Data->RefCV = pRef->v + (x + y * Data->iEdgedWidth/2) * 8;
          Data->predMV = *predMV;
-Line 1353
+Line 1419
          pmv[0] = Data->predMV;
          if (Data->qpel) { pmv[0].x /= 2; pmv[0].y /= 2; }
          PreparePredictionsBF(pmv, x, y, pParam->mb_width, pMB, mode_current);
          Data->currentMV->x = Data->currentMV->y = 0;
-Line 1364
+Line 1431
                  CheckCandidate16no4v(pmv[i].x, pmv[i].y, mask, &iDirection, Data);
          }
-         if (MotionFlags & PMV_USESQUARES16)
+         if (MotionFlags & PMV_USESQUARES16) MainSearchPtr = SquareSearch;
-                 MainSearchPtr = SquareSearch;
+         else if (MotionFlags & PMV_ADVANCEDDIAMOND16) MainSearchPtr = AdvDiamondSearch;
-         else if (MotionFlags & PMV_ADVANCEDDIAMOND16)
-                 MainSearchPtr = AdvDiamondSearch;
                  else MainSearchPtr = DiamondSearch;
          (*MainSearchPtr)(Data->currentMV->x, Data->currentMV->y, Data, iDirection);
-Line 1405
+Line 1470
                  if (mode_current == MODE_FORWARD) pMB->mvs[0] = *Data->currentMV;
                  else pMB->b_mvs[0] = *Data->currentMV;
          }
          if (mode_current == MODE_FORWARD)  *(Data->currentMV+2) = *Data->currentMV;
          else *(Data->currentMV+1) = *Data->currentMV; //we store currmv for interpolate search
  }
  static void
-Line 1419
+Line 1484
                                  const SearchData * const Data)
  {
          int dx = 0, dy = 0, b_dx = 0, b_dy = 0;
-         uint32_t sum;
+         int32_t sum;
          const int div = 1 + Data->qpel;
          int k;
-         const uint32_t quant = pMB->quant;
+         const uint32_t stride = Data->iEdgedWidth/2;
  //this is not full chroma compensation, only it's fullpel approximation. should work though
          for (k = 0; k < 4; k++) {
-Line 1437
+Line 1502
          b_dy = (b_dy >> 3) + roundtab_76[b_dy & 0xf];
          b_dx = (b_dx >> 3) + roundtab_76[b_dx & 0xf];
-         sum = sad8bi(pCur->u + 8*x + 8*y*(Data->iEdgedWidth/2),
+         sum = sad8bi(pCur->u + 8 * x + 8 * y * stride,
-                                         f_Ref->u + (y*8 + dy/2) * (Data->iEdgedWidth/2) + x*8 + dx/2,
+                                         f_Ref->u + (y*8 + dy/2) * stride + x*8 + dx/2,
-                                         b_Ref->u + (y*8 + b_dy/2) * (Data->iEdgedWidth/2) + x*8 + b_dx/2,
+                                         b_Ref->u + (y*8 + b_dy/2) * stride + x*8 + b_dx/2,
-                                         Data->iEdgedWidth/2);
+                                         stride);
-         if (sum >= 2 * MAX_CHROMA_SAD_FOR_SKIP * quant) return; //no skip
+         if (sum >= 2 * MAX_CHROMA_SAD_FOR_SKIP * pMB->quant) return; //no skip
-         sum += sad8bi(pCur->v + 8*x + 8*y*(Data->iEdgedWidth/2),
+         sum += sad8bi(pCur->v + 8*x + 8 * y * stride,
-                                         f_Ref->v + (y*8 + dy/2) * (Data->iEdgedWidth/2) + x*8 + dx/2,
+                                         f_Ref->v + (y*8 + dy/2) * stride + x*8 + dx/2,
-                                         b_Ref->v + (y*8 + b_dy/2) * (Data->iEdgedWidth/2) + x*8 + b_dx/2,
+                                         b_Ref->v + (y*8 + b_dy/2) * stride + x*8 + b_dx/2,
-                                         Data->iEdgedWidth/2);
+                                         stride);
-         if (sum < 2 * MAX_CHROMA_SAD_FOR_SKIP * quant) pMB->mode = MODE_DIRECT_NONE_MV; //skipped
+         if (sum < 2 * MAX_CHROMA_SAD_FOR_SKIP * pMB->quant) pMB->mode = MODE_DIRECT_NONE_MV; //skipped
  }
  static __inline uint32_t
  SearchDirect(const IMAGE * const f_Ref,
                                  const uint8_t * const f_RefH,
-Line 1475
+Line 1538
  {
          int32_t skip_sad;
-         int k;
+         int k = (x + Data->iEdgedWidth*y) * 16;
          MainSearchFunc *MainSearchPtr;
          *Data->iMinSAD = 256*4096;
+         Data->Ref = f_Ref->y + k;
+         Data->RefH = f_RefH + k;
+         Data->RefV = f_RefV + k;
+         Data->RefHV = f_RefHV + k;
+         Data->bRef = b_Ref->y + k;
+         Data->bRefH = b_RefH + k;
+         Data->bRefV = b_RefV + k;
+         Data->bRefHV = b_RefHV + k;
+         Data->RefCU = f_Ref->u + (x + (Data->iEdgedWidth/2) * y) * 8;
+         Data->RefCV = f_Ref->v + (x + (Data->iEdgedWidth/2) * y) * 8;
+         Data->b_RefCU = b_Ref->u + (x + (Data->iEdgedWidth/2) * y) * 8;
+         Data->b_RefCV = b_Ref->v + (x + (Data->iEdgedWidth/2) * y) * 8;
+         k = Data->qpel ? 4 : 2;
+         Data->max_dx = k * (pParam->width - x * 16);
+         Data->max_dy = k * (pParam->height - y * 16);
+         Data->min_dx = -k * (16 + x * 16);
+         Data->min_dy = -k * (16 + y * 16);
-         Data->Ref = f_Ref->y + (x + Data->iEdgedWidth*y) * 16;
+         Data->referencemv = Data->qpel ? b_mb->qmvs : b_mb->mvs;
-         Data->RefH = f_RefH + (x + Data->iEdgedWidth*y) * 16;
+         Data->qpel_precision = 0;
-         Data->RefV = f_RefV + (x + Data->iEdgedWidth*y) * 16;
-         Data->RefHV = f_RefHV + (x + Data->iEdgedWidth*y) * 16;
-         Data->bRef = b_Ref->y + (x + Data->iEdgedWidth*y) * 16;
-         Data->bRefH = b_RefH + (x + Data->iEdgedWidth*y) * 16;
-         Data->bRefV = b_RefV + (x + Data->iEdgedWidth*y) * 16;
-         Data->bRefHV = b_RefHV + (x + Data->iEdgedWidth*y) * 16;
-         Data->max_dx = 2 * pParam->width - 2 * (x) * 16;
-         Data->max_dy = 2 * pParam->height - 2 * (y) * 16;
-         Data->min_dx = -(2 * 16 + 2 * (x) * 16);
-         Data->min_dy = -(2 * 16 + 2 * (y) * 16);
-         if (Data->qpel) { //we measure in qpixels
-                 Data->max_dx *= 2;
-                 Data->max_dy *= 2;
-                 Data->min_dx *= 2;
-                 Data->min_dy *= 2;
-                 Data->referencemv = b_mb->qmvs;
-         } else Data->referencemv = b_mb->mvs;
-         Data->qpel_precision = 0; // it's a trick. it's 1 not 0, but we need 0 here
          for (k = 0; k < 4; k++) {
                  pMB->mvs[k].x = Data->directmvF[k].x = ((TRB * Data->referencemv[k].x) / TRD);
-Line 1509
+Line 1570
                  pMB->mvs[k].y = Data->directmvF[k].y = ((TRB * Data->referencemv[k].y) / TRD);
                  pMB->b_mvs[k].y = Data->directmvB[k].y = ((TRB - TRD) * Data->referencemv[k].y) / TRD;
-                 if ( ( pMB->b_mvs[k].x > Data->max_dx ) || ( pMB->b_mvs[k].x < Data->min_dx )
+                 if ( (pMB->b_mvs[k].x > Data->max_dx) | (pMB->b_mvs[k].x < Data->min_dx)
-                         || ( pMB->b_mvs[k].y > Data->max_dy ) || ( pMB->b_mvs[k].y < Data->min_dy )) {
+                         | (pMB->b_mvs[k].y > Data->max_dy) | (pMB->b_mvs[k].y < Data->min_dy) ) {
                          *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"
-Line 1526
+Line 1587
                  }
          }
-         if (b_mb->mode == MODE_INTER4V) CheckCandidate = CheckCandidateDirect;
+         CheckCandidate = b_mb->mode == MODE_INTER4V ? CheckCandidateDirect : CheckCandidateDirectno4v;
-         else CheckCandidate = CheckCandidateDirectno4v;
-         (*CheckCandidate)(0, 0, 255, &k, Data);
+         CheckCandidate(0, 0, 255, &k, Data);
  // initial (fast) skip decision
-         if (*Data->iMinSAD < pMB->quant * INITIAL_SKIP_THRESH*2) {
+         if (*Data->iMinSAD < pMB->quant * INITIAL_SKIP_THRESH * (2 + Data->chroma?1:0)) {
-                 SkipDecisionB(pCur, f_Ref, b_Ref, pMB, x, y, Data); //possible skip - checking chroma
+                 //possible skip
+                 if (Data->chroma) {
+                         pMB->mode = MODE_DIRECT_NONE_MV;
+                         return *Data->iMinSAD; // skip.
+                 } else {
+                         SkipDecisionB(pCur, f_Ref, b_Ref, pMB, x, y, Data);
                  if (pMB->mode == MODE_DIRECT_NONE_MV) return *Data->iMinSAD; // skip.
          }
+         }
          skip_sad = *Data->iMinSAD;
-Line 1552
+Line 1618
          *best_sad = *Data->iMinSAD;
-         if (b_mb->mode == MODE_INTER4V || Data->qpel) pMB->mode = MODE_DIRECT;
+         if (Data->qpel || b_mb->mode == MODE_INTER4V) pMB->mode = MODE_DIRECT;
          else pMB->mode = MODE_DIRECT_NO4V; //for faster compensation
          pMB->pmvs[3] = *Data->currentMV;
-Line 1584
+Line 1650
          return skip_sad;
  }
  static void
- SearchInterpolate(const uint8_t * const f_Ref,
+ SearchInterpolate(const IMAGE * const f_Ref,
                                  const uint8_t * const f_RefH,
                                  const uint8_t * const f_RefV,
                                  const uint8_t * const f_RefHV,
-                                 const uint8_t * const b_Ref,
+                                 const IMAGE * const b_Ref,
                                  const uint8_t * const b_RefH,
                                  const uint8_t * const b_RefV,
                                  const uint8_t * const b_RefHV,
-Line 1618
+Line 1683
          fData->iFcode = bData.bFcode = fcode; fData->bFcode = bData.iFcode = bcode;
          i = (x + y * fData->iEdgedWidth) * 16;
-         bData.bRef = fData->Ref = f_Ref + i;
+         bData.bRef = fData->Ref = f_Ref->y + i;
          bData.bRefH = fData->RefH = f_RefH + i;
          bData.bRefV = fData->RefV = f_RefV + i;
          bData.bRefHV = fData->RefHV = f_RefHV + i;
-         bData.Ref = fData->bRef = b_Ref + i;
+         bData.Ref = fData->bRef = b_Ref->y + i;
          bData.RefH = fData->bRefH = b_RefH + i;
          bData.RefV = fData->bRefV = b_RefV + i;
          bData.RefHV = fData->bRefHV = b_RefHV + i;
+         bData.b_RefCU = fData->RefCU = f_Ref->u + (x + (fData->iEdgedWidth/2) * y) * 8;
+         bData.b_RefCV = fData->RefCV = f_Ref->v + (x + (fData->iEdgedWidth/2) * y) * 8;
+         bData.RefCU = fData->b_RefCU = b_Ref->u + (x + (fData->iEdgedWidth/2) * y) * 8;
+         bData.RefCV = fData->b_RefCV = b_Ref->v + (x + (fData->iEdgedWidth/2) * y) * 8;
          bData.bpredMV = fData->predMV = *f_predMV;
          fData->bpredMV = bData.predMV = *b_predMV;
-Line 1646
+Line 1716
          CheckCandidateInt(fData->currentMV[0].x, fData->currentMV[0].y, 255, &iDirection, fData);
- //diamond. I wish we could use normal mainsearch functions (square, advdiamond)
+ //diamond
          do {
                  iDirection = 255;
                  // forward MV moves
-Line 1668
+Line 1737
          } while (!(iDirection));
+ //qpel refinement
          if (fData->qpel) {
                  if (*fData->iMinSAD > *best_sad + 500) return;
                  CheckCandidate = CheckCandidateInt;
-Line 1729
+Line 1799
          int32_t best_sad;
          uint32_t skip_sad;
          int f_count = 0, b_count = 0, i_count = 0, d_count = 0, n_count = 0;
-         static const VECTOR zeroMV={0,0};
          const MACROBLOCK * const b_mbs = b_reference->mbs;
          VECTOR f_predMV, b_predMV;      /* there is no prediction for direct mode*/
-Line 1743
+Line 1812
          int32_t iMinSAD;
          VECTOR currentMV[3];
          VECTOR currentQMV[3];
+         int32_t temp[8];
          memset(&Data, 0, sizeof(SearchData));
          Data.iEdgedWidth = pParam->edged_width;
          Data.currentMV = currentMV; Data.currentQMV = currentQMV;
          Data.iMinSAD = &iMinSAD;
          Data.lambda16 = lambda_vec16[frame->quant];
-         Data.chroma = frame->quant;
          Data.qpel = pParam->m_quarterpel;
          Data.rounding = 0;
+         Data.chroma = frame->motion_flags & PMV_CHROMA8;
+         Data.temp = temp;
          Data.RefQ = f_refV->u; // a good place, also used in MC (for similar purpose)
          // note: i==horizontal, j==vertical
-Line 1770
+Line 1841
                                  }
                          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;
                          pMB->quant = frame->quant;
  /* direct search comes first, because it (1) checks for SKIP-mode
-Line 1788
+Line 1861
                          if (pMB->mode == MODE_DIRECT_NONE_MV) { n_count++; continue; }
                          // forward search
-                         SearchBF(f_ref->y, f_refH->y, f_refV->y, f_refHV->y,
+                         SearchBF(f_ref, f_refH->y, f_refV->y, f_refHV->y,
                                                  &frame->image, i, j,
                                                  frame->motion_flags,
                                                  frame->fcode, pParam,
-Line 1796
+Line 1869
                                                  MODE_FORWARD, &Data);
                          // backward search
-                         SearchBF(b_ref->y, b_refH->y, b_refV->y, b_refHV->y,
+                         SearchBF(b_ref, b_refH->y, b_refV->y, b_refHV->y,
                                                  &frame->image, i, j,
                                                  frame->motion_flags,
                                                  frame->bcode, pParam,
-Line 1804
+Line 1877
                                                  MODE_BACKWARD, &Data);
                          // interpolate search comes last, because it uses data from forward and backward as prediction
-                         SearchInterpolate(f_ref->y, f_refH->y, f_refV->y, f_refHV->y,
+                         SearchInterpolate(f_ref, f_refH->y, f_refV->y, f_refHV->y,
-                                                 b_ref->y, b_refH->y, b_refV->y, b_refHV->y,
+                                                 b_ref, b_refH->y, b_refV->y, b_refHV->y,
                                                  &frame->image,
                                                  i, j,
                                                  frame->fcode, frame->bcode,
-Line 1823
+Line 1896
                          switch (pMB->mode) {
                                  case MODE_FORWARD:
                                          f_count++;
-                                         if (Data.qpel) f_predMV = pMB->qmvs[0];
+                                         f_predMV = Data.qpel ? pMB->qmvs[0] : pMB->mvs[0];
-                                         else f_predMV = pMB->mvs[0];
                                          break;
                                  case MODE_BACKWARD:
                                          b_count++;
-                                         if (Data.qpel) b_predMV = pMB->b_qmvs[0];
+                                         b_predMV = Data.qpel ? pMB->b_qmvs[0] : pMB->b_mvs[0];
-                                         else b_predMV = pMB->b_mvs[0];
                                          break;
                                  case MODE_INTERPOLATE:
                                          i_count++;
-                                         if (Data.qpel) {
+                                         f_predMV = Data.qpel ? pMB->qmvs[0] : pMB->mvs[0];
-                                                 f_predMV = pMB->qmvs[0];
+                                         b_predMV = Data.qpel ? pMB->b_qmvs[0] : pMB->b_mvs[0];
-                                                 b_predMV = pMB->b_qmvs[0];
-                                         } else {
-                                                 f_predMV = pMB->mvs[0];
-                                                 b_predMV = pMB->b_mvs[0];
-                                         }
                                          break;
                                  case MODE_DIRECT:
                                  case MODE_DIRECT_NO4V:
-Line 1901
+Line 1967
                          DiamondSearch(Data->currentMV->x, Data->currentMV->y, Data, i);
                  for (i = 0; i < 4; i++) {
-                         MACROBLOCK * MB = &pMBs[x + (i&1) + (y+(i>>1) * pParam->mb_width)];
+                         MACROBLOCK * MB = &pMBs[x + (i&1) + (y+(i>>1)) * pParam->mb_width];
                          MB->mvs[0] = MB->mvs[1] = MB->mvs[2] = MB->mvs[3] = Data->currentMV[i];
                          MB->mode = MODE_INTER;
                          MB->sad16 = Data->iMinSAD[i+1];
-Line 1913
+Line 1979
  #define INTRA_THRESH    1500
  #define INTER_THRESH    1400
  int
  MEanalysis(     const IMAGE * const pRef,
                          FRAMEINFO * const Current,
                          MBParam * const pParam,
                          int maxIntra, //maximum number if non-I frames
                          int intraCount, //number of non-I frames after last I frame; 0 if we force P/B frame
-                         int bCount) // number if B frames in a row
+                         int bCount) // number of B frames in a row
  {
          uint32_t x, y, intra = 0;
          int sSAD = 0;
          MACROBLOCK * const pMBs = Current->mbs;
          const IMAGE * const pCurrent = &Current->image;
          int IntraThresh = INTRA_THRESH, InterThresh = INTER_THRESH;
-         const VECTOR zeroMV = {0,0};
          int32_t iMinSAD[5], temp[5];
          VECTOR currentMV[5];
-Line 1940
+Line 2004
          Data.temp = temp;
          CheckCandidate = CheckCandidate32I;
-         if (intraCount < 10) // we're right after an I frame
+         if (intraCount != 0 && intraCount < 10) // we're right after an I frame
                  IntraThresh += 4 * (intraCount - 10) * (intraCount - 10);
          else
                  if ( 5*(maxIntra - intraCount) < maxIntra) // we're close to maximum. 2 sec when max is 10 sec
-Line 1961
+Line 2025
                          for (i = 0; i < 4; i++) {
                                  int dev;
-                                 MACROBLOCK *pMB = &pMBs[x+(i&1) + y+(i>>1) * pParam->mb_width];
+                                 MACROBLOCK *pMB = &pMBs[x+(i&1) + (y+(i>>1)) * pParam->mb_width];
                                  if (pMB->sad16 > IntraThresh) {
                                          dev = dev16(pCurrent->y + (x + (i&1) + (y + (i>>1))* pParam->edged_width) * 16,
                                                                            pParam->edged_width);
-Line 1975
+Line 2039
                  }
          }
          sSAD /= (pParam->mb_height-2)*(pParam->mb_width-2);
-         if (sSAD > IntraThresh + INTRA_BIAS ) return I_VOP;
+ //      if (sSAD > IntraThresh + INTRA_BIAS) return I_VOP;
          if (sSAD > InterThresh ) return P_VOP;
          emms();
          return B_VOP;
  }
- static void
- CheckGMC(int x, int y, const int dir, int * iDirection,
+ static WARPPOINTS
-                 const MACROBLOCK * const pMBs, uint32_t * bestcount, VECTOR * GMC,
+ GlobalMotionEst(const MACROBLOCK * const pMBs,
-                 const MBParam * const pParam)
+                                 const MBParam * const pParam,
+                                 const FRAMEINFO * const current,
+                                 const FRAMEINFO * const reference,
+                                 const IMAGE * const pRefH,
+                                 const IMAGE * const pRefV,
+                                 const IMAGE * const pRefHV      )
  {
-         uint32_t mx, my, a, count = 0;
-         for (my = 1; my < pParam->mb_height-1; my++)
+         const int deltax=8;             // upper bound for difference between a MV and it's neighbour MVs
-                 for (mx = 1; mx < pParam->mb_width-1; mx++) {
+         const int deltay=8;
-                         VECTOR mv;
+         const int grad=512;             // lower bound for deviation in MB
-                         const MACROBLOCK *pMB = &pMBs[mx + my * pParam->mb_width];
-                         if (pMB->mode == MODE_INTRA || pMB->mode == MODE_NOT_CODED) continue;
+         WARPPOINTS gmc;
-                         mv = pMB->mvs[0];
-                         a = ABS(mv.x - x) + ABS(mv.y - y);
+         uint32_t mx, my;
-                         if (a < 6) count += 6 - a;
-                 }
-         if (count > *bestcount) {
+         int MBh = pParam->mb_height;
-                 *bestcount = count;
+         int MBw = pParam->mb_width;
-                 *iDirection = dir;
-                 GMC->x = x; GMC->y = y;
+         int *MBmask= calloc(MBh*MBw,sizeof(int));
+         double DtimesF[4] = { 0.,0., 0., 0. };
+         double sol[4] = { 0., 0., 0., 0. };
+         double a,b,c,n,denom;
+         double meanx,meany;
+         int num,oldnum;
+         if (!MBmask) { fprintf(stderr,"Mem error\n"); return gmc;}
+ // filter mask of all blocks
+         for (my = 1; my < MBh-1; my++)
+         for (mx = 1; mx < MBw-1; mx++)
+         {
+                 const int mbnum = mx + my * MBw;
+                 const MACROBLOCK *pMB = &pMBs[mbnum];
+                 const VECTOR mv = pMB->mvs[0];
+                 if (pMB->mode == MODE_INTRA || pMB->mode == MODE_NOT_CODED)
+                         continue;
+                 if ( ( (ABS(mv.x -   (pMB-1)->mvs[0].x) < deltax) && (ABS(mv.y -   (pMB-1)->mvs[0].y) < deltay) )
+                 &&   ( (ABS(mv.x -   (pMB+1)->mvs[0].x) < deltax) && (ABS(mv.y -   (pMB+1)->mvs[0].y) < deltay) )
+                 &&   ( (ABS(mv.x - (pMB-MBw)->mvs[0].x) < deltax) && (ABS(mv.y - (pMB-MBw)->mvs[0].y) < deltay) )
+                 &&   ( (ABS(mv.x - (pMB+MBw)->mvs[0].x) < deltax) && (ABS(mv.y - (pMB+MBw)->mvs[0].y) < deltay) ) )
+                         MBmask[mbnum]=1;
          }
+         for (my = 1; my < MBh-1; my++)
+         for (mx = 1; mx < MBw-1; mx++)
+         {
+                 const uint8_t *const pCur = current->image.y + 16*my*pParam->edged_width + 16*mx;
+                 const int mbnum = mx + my * MBw;
+                 if (!MBmask[mbnum])
+                         continue;
+                 if (sad16 ( pCur, pCur+1 , pParam->edged_width, 65536) <= grad )
+                         MBmask[mbnum] = 0;
+                 if (sad16 ( pCur, pCur+pParam->edged_width, pParam->edged_width, 65536) <= grad )
+                         MBmask[mbnum] = 0;
  }
+         emms();
- static VECTOR
+         do {            /* until convergence */
- GlobalMotionEst(const MACROBLOCK * const pMBs, const MBParam * const pParam, const uint32_t iFcode)
- {
-         uint32_t count, bestcount = 0;
+         a = b = c = n = 0;
-         int x, y;
+         DtimesF[0] = DtimesF[1] = DtimesF[2] = DtimesF[3] = 0.;
-         VECTOR gmc = {0,0};
+         for (my = 0; my < MBh; my++)
-         int step, min_x, max_x, min_y, max_y;
+                 for (mx = 0; mx < MBw; mx++)
-         uint32_t mx, my;
+                 {
-         int iDirection, bDirection;
+                         const int mbnum = mx + my * MBw;
+                         const MACROBLOCK *pMB = &pMBs[mbnum];
+                         const VECTOR mv = pMB->mvs[0];
-         min_x = min_y = -32<<iFcode;
+                         if (!MBmask[mbnum])
-         max_x = max_y = 32<<iFcode;
+                                 continue;
- //step1: let's find a rough camera panning
+                         n++;
-         for (step = 32; step >= 2; step /= 2) {
+                         a += 16*mx+8;
-                 bestcount = 0;
+                         b += 16*my+8;
-                 for (y = min_y; y <= max_y; y += step)
+                         c += (16*mx+8)*(16*mx+8)+(16*my+8)*(16*my+8);
-                         for (x = min_x ; x <= max_x; x += step) {
-                                 count = 0;
+                         DtimesF[0] += (double)mv.x;
-                                 //for all macroblocks
+                         DtimesF[1] += (double)mv.x*(16*mx+8) + (double)mv.y*(16*my+8);
-                                 for (my = 1; my < pParam->mb_height-1; my++)
+                         DtimesF[2] += (double)mv.x*(16*my+8) - (double)mv.y*(16*mx+8);
-                                         for (mx = 1; mx < pParam->mb_width-1; mx++) {
+                         DtimesF[3] += (double)mv.y;
-                                                 const MACROBLOCK *pMB = &pMBs[mx + my * pParam->mb_width];
+                 }
-                                                 VECTOR mv;
+         denom = a*a+b*b-c*n;
+ /* Solve the system:     sol = (D'*E*D)^{-1} D'*E*F   */
+ /* D'*E*F has been calculated in the same loop as matrix */
+         sol[0] = -c*DtimesF[0] + a*DtimesF[1] + b*DtimesF[2];
+         sol[1] =  a*DtimesF[0] - n*DtimesF[1]                + b*DtimesF[3];
+         sol[2] =  b*DtimesF[0]                - n*DtimesF[2] - a*DtimesF[3];
+         sol[3] =                 b*DtimesF[1] - a*DtimesF[2] - c*DtimesF[3];
+         sol[0] /= denom;
+         sol[1] /= denom;
+         sol[2] /= denom;
+         sol[3] /= denom;
+         meanx = meany = 0.;
+         oldnum = 0;
+         for (my = 0; my < MBh; my++)
+                 for (mx = 0; mx < MBw; mx++)
+                 {
+                         const int mbnum = mx + my * MBw;
+                         const MACROBLOCK *pMB = &pMBs[mbnum];
+                         const VECTOR mv = pMB->mvs[0];
-                                                 if (pMB->mode == MODE_INTRA || pMB->mode == MODE_NOT_CODED)
+                         if (!MBmask[mbnum])
                                                          continue;
-                                                 mv = pMB->mvs[0];
+                         oldnum++;
-                                                 if ( ABS(mv.x - x) <= step && ABS(mv.y - y) <= step )   /* GMC translation is always halfpel-res */
+                         meanx += ABS(( sol[0] + (16*mx+8)*sol[1] + (16*my+8)*sol[2] ) - mv.x );
-                                                         count++;
+                         meany += ABS(( sol[3] - (16*mx+8)*sol[2] + (16*my+8)*sol[1] ) - mv.y );
                                          }
-                                 if (count >= bestcount) { bestcount = count; gmc.x = x; gmc.y = y; }
-                         }
+         if (4*meanx > oldnum)   /* better fit than 0.25 is useless */
-                 min_x = gmc.x - step;
+                 meanx /= oldnum;
-                 max_x = gmc.x + step;
+         else
-                 min_y = gmc.y - step;
+                 meanx = 0.25;
-                 max_y = gmc.y + step;
+         if (4*meany > oldnum)
+                 meany /= oldnum;
+         else
+                 meany = 0.25;
+ /*      fprintf(stderr,"sol = (%8.5f, %8.5f, %8.5f, %8.5f)\n",sol[0],sol[1],sol[2],sol[3]);
+         fprintf(stderr,"meanx = %8.5f  meany = %8.5f   %d\n",meanx,meany, oldnum);
+ */
+         num = 0;
+         for (my = 0; my < MBh; my++)
+                 for (mx = 0; mx < MBw; mx++)
+                 {
+                         const int mbnum = mx + my * MBw;
+                         const MACROBLOCK *pMB = &pMBs[mbnum];
+                         const VECTOR mv = pMB->mvs[0];
+                         if (!MBmask[mbnum])
+                                 continue;
+                         if  ( ( ABS(( sol[0] + (16*mx+8)*sol[1] + (16*my+8)*sol[2] ) - mv.x ) > meanx )
+                            || ( ABS(( sol[3] - (16*mx+8)*sol[2] + (16*my+8)*sol[1] ) - mv.y ) > meany ) )
+                                 MBmask[mbnum]=0;
+                         else
+                                 num++;
          }
-         if (bestcount < (pParam->mb_height-2)*(pParam->mb_width-2)/10)
+         } while ( (oldnum != num) && (num>=4) );
-                 gmc.x = gmc.y = 0; //no camara pan, no GMC
- // step2: let's refine camera panning using gradiend-descent approach
+         if (num < 4)
- // TODO: more warping points may be evaluated here (like in interpolate mode search - two vectors in one diamond)
+         {
-         bestcount = 0;
+                 gmc.duv[0].x= gmc.duv[0].y= gmc.duv[1].x= gmc.duv[1].y= gmc.duv[2].x= gmc.duv[2].y=0;
-         CheckGMC(gmc.x, gmc.y, 255, &iDirection, pMBs, &bestcount, &gmc, pParam);
+         } else {
-         do {
-                 x = gmc.x; y = gmc.y;
-                 bDirection = iDirection; iDirection = 0;
-                 if (bDirection & 1) CheckGMC(x - 1, y, 1+4+8, &iDirection, pMBs, &bestcount, &gmc, pParam);
-                 if (bDirection & 2) CheckGMC(x + 1, y, 2+4+8, &iDirection, pMBs, &bestcount, &gmc, pParam);
-                 if (bDirection & 4) CheckGMC(x, y - 1, 1+2+4, &iDirection, pMBs, &bestcount, &gmc, pParam);
-                 if (bDirection & 8) CheckGMC(x, y + 1, 1+2+8, &iDirection, pMBs, &bestcount, &gmc, pParam);
-         } while (iDirection);
+                 gmc.duv[0].x=(int)(sol[0]+0.5);
+                 gmc.duv[0].y=(int)(sol[3]+0.5);
-         if (pParam->m_quarterpel) {
+                 gmc.duv[1].x=(int)(sol[1]*pParam->width+0.5);
-                 gmc.x *= 2;
+                 gmc.duv[1].y=(int)(-sol[2]*pParam->width+0.5);
-                 gmc.y *= 2;     /* we store the halfpel value as pseudo-qpel to make comparison easier */
+                 gmc.duv[2].x=0;
+                 gmc.duv[2].y=0;
          }
+ //      fprintf(stderr,"wp1 = ( %4d, %4d)  wp2 = ( %4d, %4d) \n", gmc.duv[0].x, gmc.duv[0].y, gmc.duv[1].x, gmc.duv[1].y);
+         free(MBmask);
          return gmc;
  }

 Legend:



Removed from v.767
 


changed lines


 
Added in v.819
 Legend:



Removed from v.767
 


changed lines


 
Added in v.819
-Removed from v.767
+Added in v.819

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