41 |
#include "motion_inlines.h" |
#include "motion_inlines.h" |
42 |
|
|
43 |
static const int xvid_me_lambda_vec8[32] = |
static const int xvid_me_lambda_vec8[32] = |
44 |
{ 0 ,(int)(1.00235 * NEIGH_TEND_8X8 + 0.5), |
{ 0 ,(int)(1.0 * NEIGH_TEND_8X8 + 0.5), |
45 |
(int)(1.15582*NEIGH_TEND_8X8 + 0.5), (int)(1.31976*NEIGH_TEND_8X8 + 0.5), |
(int)(2.0*NEIGH_TEND_8X8 + 0.5), (int)(3.0*NEIGH_TEND_8X8 + 0.5), |
46 |
(int)(1.49591*NEIGH_TEND_8X8 + 0.5), (int)(1.68601*NEIGH_TEND_8X8 + 0.5), |
(int)(4.0*NEIGH_TEND_8X8 + 0.5), (int)(5.0*NEIGH_TEND_8X8 + 0.5), |
47 |
(int)(1.89187*NEIGH_TEND_8X8 + 0.5), (int)(2.11542*NEIGH_TEND_8X8 + 0.5), |
(int)(6.0*NEIGH_TEND_8X8 + 0.5), (int)(7.0*NEIGH_TEND_8X8 + 0.5), |
48 |
(int)(2.35878*NEIGH_TEND_8X8 + 0.5), (int)(2.62429*NEIGH_TEND_8X8 + 0.5), |
(int)(8.0*NEIGH_TEND_8X8 + 0.5), (int)(9.0*NEIGH_TEND_8X8 + 0.5), |
49 |
(int)(2.91455*NEIGH_TEND_8X8 + 0.5), (int)(3.23253*NEIGH_TEND_8X8 + 0.5), |
(int)(10.0*NEIGH_TEND_8X8 + 0.5), (int)(11.0*NEIGH_TEND_8X8 + 0.5), |
50 |
(int)(3.58158*NEIGH_TEND_8X8 + 0.5), (int)(3.96555*NEIGH_TEND_8X8 + 0.5), |
(int)(12.0*NEIGH_TEND_8X8 + 0.5), (int)(13.0*NEIGH_TEND_8X8 + 0.5), |
51 |
(int)(4.38887*NEIGH_TEND_8X8 + 0.5), (int)(4.85673*NEIGH_TEND_8X8 + 0.5), |
(int)(14.0*NEIGH_TEND_8X8 + 0.5), (int)(15.0*NEIGH_TEND_8X8 + 0.5), |
52 |
(int)(5.37519*NEIGH_TEND_8X8 + 0.5), (int)(5.95144*NEIGH_TEND_8X8 + 0.5), |
(int)(16.0*NEIGH_TEND_8X8 + 0.5), (int)(17.0*NEIGH_TEND_8X8 + 0.5), |
53 |
(int)(6.59408*NEIGH_TEND_8X8 + 0.5), (int)(7.31349*NEIGH_TEND_8X8 + 0.5), |
(int)(18.0*NEIGH_TEND_8X8 + 0.5), (int)(19.0*NEIGH_TEND_8X8 + 0.5), |
54 |
(int)(8.12242*NEIGH_TEND_8X8 + 0.5), (int)(9.03669*NEIGH_TEND_8X8 + 0.5), |
(int)(20.0*NEIGH_TEND_8X8 + 0.5), (int)(21.0*NEIGH_TEND_8X8 + 0.5), |
55 |
(int)(10.0763*NEIGH_TEND_8X8 + 0.5), (int)(11.2669*NEIGH_TEND_8X8 + 0.5), |
(int)(22.0*NEIGH_TEND_8X8 + 0.5), (int)(23.0*NEIGH_TEND_8X8 + 0.5), |
56 |
(int)(12.6426*NEIGH_TEND_8X8 + 0.5), (int)(14.2493*NEIGH_TEND_8X8 + 0.5), |
(int)(24.0*NEIGH_TEND_8X8 + 0.5), (int)(25.0*NEIGH_TEND_8X8 + 0.5), |
57 |
(int)(16.1512*NEIGH_TEND_8X8 + 0.5), (int)(18.442*NEIGH_TEND_8X8 + 0.5), |
(int)(26.0*NEIGH_TEND_8X8 + 0.5), (int)(27.0*NEIGH_TEND_8X8 + 0.5), |
58 |
(int)(21.2656*NEIGH_TEND_8X8 + 0.5), (int)(24.8580*NEIGH_TEND_8X8 + 0.5), |
(int)(28.0*NEIGH_TEND_8X8 + 0.5), (int)(29.0*NEIGH_TEND_8X8 + 0.5), |
59 |
(int)(29.6436*NEIGH_TEND_8X8 + 0.5), (int)(36.4949*NEIGH_TEND_8X8 + 0.5) |
(int)(30.0*NEIGH_TEND_8X8 + 0.5), (int)(31.0*NEIGH_TEND_8X8 + 0.5) |
60 |
}; |
}; |
61 |
|
|
62 |
static void |
static void |
82 |
sad = sad16v(data->Cur, Reference, data->iEdgedWidth, data->temp); |
sad = sad16v(data->Cur, Reference, data->iEdgedWidth, data->temp); |
83 |
t = d_mv_bits(x, y, data->predMV, data->iFcode, data->qpel^data->qpel_precision); |
t = d_mv_bits(x, y, data->predMV, data->iFcode, data->qpel^data->qpel_precision); |
84 |
|
|
85 |
sad += (data->lambda16 * t * sad)>>10; |
sad += (data->lambda16 * t); |
86 |
data->temp[0] += (data->lambda8 * t * (data->temp[0] + NEIGH_8X8_BIAS))>>10; |
data->temp[0] += (data->lambda8 * t); |
87 |
|
|
88 |
if (data->chroma) { |
if (data->chroma) { |
89 |
if (sad >= data->iMinSAD[0]) goto no16; |
if (sad >= data->iMinSAD[0]) goto no16; |
129 |
sad = sad8(data->Cur, Reference, data->iEdgedWidth); |
sad = sad8(data->Cur, Reference, data->iEdgedWidth); |
130 |
t = d_mv_bits(x, y, data->predMV, data->iFcode, data->qpel^data->qpel_precision); |
t = d_mv_bits(x, y, data->predMV, data->iFcode, data->qpel^data->qpel_precision); |
131 |
|
|
132 |
sad += (data->lambda8 * t * (sad+NEIGH_8X8_BIAS))>>10; |
sad += (data->lambda8 * t); |
133 |
|
|
134 |
if (sad < *(data->iMinSAD)) { |
if (sad < *(data->iMinSAD)) { |
135 |
*(data->iMinSAD) = sad; |
*(data->iMinSAD) = sad; |
423 |
Data->predMV, Data->iFcode, 0); |
Data->predMV, Data->iFcode, 0); |
424 |
} |
} |
425 |
|
|
426 |
*(Data->iMinSAD) += (Data->lambda8 * i * (*Data->iMinSAD + NEIGH_8X8_BIAS))>>10; |
*(Data->iMinSAD) += (Data->lambda8 * i); |
427 |
|
|
428 |
if (MotionFlags & (XVID_ME_EXTSEARCH8|XVID_ME_HALFPELREFINE8|XVID_ME_QUARTERPELREFINE8)) { |
if (MotionFlags & (XVID_ME_EXTSEARCH8|XVID_ME_HALFPELREFINE8|XVID_ME_QUARTERPELREFINE8)) { |
429 |
|
|
557 |
else Data->predMV = pmv[0]; |
else Data->predMV = pmv[0]; |
558 |
|
|
559 |
i = d_mv_bits(0, 0, Data->predMV, Data->iFcode, 0); |
i = d_mv_bits(0, 0, Data->predMV, Data->iFcode, 0); |
560 |
Data->iMinSAD[0] = pMB->sad16 + ((Data->lambda16 * i * pMB->sad16)>>10); |
Data->iMinSAD[0] = pMB->sad16 + (Data->lambda16 * i); |
561 |
Data->iMinSAD[1] = pMB->sad8[0] + ((Data->lambda8 * i * (pMB->sad8[0]+NEIGH_8X8_BIAS)) >> 10); |
Data->iMinSAD[1] = pMB->sad8[0] + (Data->lambda8 * i); |
562 |
Data->iMinSAD[2] = pMB->sad8[1]; |
Data->iMinSAD[2] = pMB->sad8[1]; |
563 |
Data->iMinSAD[3] = pMB->sad8[2]; |
Data->iMinSAD[3] = pMB->sad8[2]; |
564 |
Data->iMinSAD[4] = pMB->sad8[3]; |
Data->iMinSAD[4] = pMB->sad8[3]; |
751 |
return Flags; |
return Flags; |
752 |
} |
} |
753 |
|
|
754 |
|
static __inline void |
755 |
|
motionStatsPVOP(int * const MVmax, int * const mvCount, int * const mvSum, |
756 |
|
const MACROBLOCK * const pMB, const int qpel) |
757 |
|
{ |
758 |
|
const VECTOR * const mv = qpel ? pMB->qmvs : pMB->mvs; |
759 |
|
int i; |
760 |
|
int max = *MVmax; |
761 |
|
|
762 |
|
switch (pMB->mode) { |
763 |
|
case MODE_INTER4V: |
764 |
|
*mvCount += 3; |
765 |
|
for(i = 3; i; i--) { |
766 |
|
if (mv[i].x > max) max = mv[i].x; |
767 |
|
else if (-mv[i].x - 1 > max) max = -mv[i].x - 1; |
768 |
|
*mvSum += mv[i].x * mv[i].x; |
769 |
|
if (mv[i].y > max) max = mv[i].y; |
770 |
|
else if (-mv[i].y - 1 > max) max = -mv[i].y - 1; |
771 |
|
*mvSum += mv[i].y * mv[i].y; |
772 |
|
} |
773 |
|
case MODE_INTER: |
774 |
|
(*mvCount)++; |
775 |
|
*mvSum += mv[0].x * mv[0].x; |
776 |
|
*mvSum += mv[0].y * mv[0].y; |
777 |
|
if (mv[0].x > max) max = mv[0].x; |
778 |
|
else if (-mv[0].x - 1 > max) max = -mv[0].x - 1; |
779 |
|
if (mv[0].y > max) max = mv[0].y; |
780 |
|
else if (-mv[0].y - 1 > max) max = -mv[0].y - 1; |
781 |
|
*MVmax = max; |
782 |
|
default: |
783 |
|
break; |
784 |
|
} |
785 |
|
} |
786 |
|
|
787 |
bool |
bool |
788 |
MotionEstimation(MBParam * const pParam, |
MotionEstimation(MBParam * const pParam, |
789 |
FRAMEINFO * const current, |
FRAMEINFO * const current, |
803 |
const uint32_t iEdgedWidth = pParam->edged_width; |
const uint32_t iEdgedWidth = pParam->edged_width; |
804 |
const uint32_t MotionFlags = MakeGoodMotionFlags(current->motion_flags, current->vop_flags, current->vol_flags); |
const uint32_t MotionFlags = MakeGoodMotionFlags(current->motion_flags, current->vop_flags, current->vol_flags); |
805 |
int stat_thresh = 0; |
int stat_thresh = 0; |
806 |
|
int MVmax = 0, mvSum = 0, mvCount = 0; |
807 |
|
|
808 |
uint32_t x, y; |
uint32_t x, y; |
809 |
int32_t sad00; |
int32_t sad00; |
858 |
} |
} |
859 |
|
|
860 |
if(MotionFlags & XVID_ME_DETECT_STATIC_MOTION) { |
if(MotionFlags & XVID_ME_DETECT_STATIC_MOTION) { |
861 |
|
VECTOR *cmpMV; |
862 |
|
VECTOR staticMV = { 0, 0 }; |
863 |
|
|
864 |
|
if (current->coding_type == S_VOP) |
865 |
|
cmpMV = &pMB->amv; |
866 |
|
else |
867 |
|
cmpMV = &staticMV; |
868 |
|
|
869 |
if(x > 0 && y > 0 && x < pParam->mb_width) { |
if(x > 0 && y > 0 && x < pParam->mb_width) { |
870 |
if(MVequal((&pMBs[(x-1) + y * pParam->mb_width])->mvs[0], zeroMV) && |
if(MVequal((&pMBs[(x-1) + y * pParam->mb_width])->mvs[0], *cmpMV) && |
871 |
MVequal((&pMBs[x + (y-1) * pParam->mb_width])->mvs[0], zeroMV) && |
MVequal((&pMBs[x + (y-1) * pParam->mb_width])->mvs[0], *cmpMV) && |
872 |
MVequal((&pMBs[(x+1) + (y-1) * pParam->mb_width])->mvs[0], zeroMV) && |
MVequal((&pMBs[(x+1) + (y-1) * pParam->mb_width])->mvs[0], *cmpMV) && |
873 |
MVequal(prevMB->mvs[0], zeroMV)) { |
MVequal(prevMB->mvs[0], *cmpMV)) { |
874 |
stat_thresh = MAX((&pMBs[(x-1) + y * pParam->mb_width])->sad16, |
stat_thresh = MAX((&pMBs[(x-1) + y * pParam->mb_width])->sad16, |
875 |
MAX((&pMBs[x + (y-1) * pParam->mb_width])->sad16, |
MAX((&pMBs[x + (y-1) * pParam->mb_width])->sad16, |
876 |
MAX((&pMBs[(x+1) + (y-1) * pParam->mb_width])->sad16, |
MAX((&pMBs[(x+1) + (y-1) * pParam->mb_width])->sad16, |
884 |
} |
} |
885 |
} |
} |
886 |
|
|
887 |
/* favorize (0,0) vector for cartoons */ |
/* favorize (0,0) or global vector for cartoons */ |
888 |
if ((current->vop_flags & XVID_VOP_CARTOON) && |
if (current->vop_flags & XVID_VOP_CARTOON) { |
889 |
((sad00 < pMB->quant * 4 * skip_thresh) || (sad00 < stat_thresh))) { |
if (current->coding_type == S_VOP) { |
890 |
|
int32_t iSAD = sad16(pCurrent->y + (x + y * iEdgedWidth) * 16, |
891 |
|
pGMC->y + 16*y*iEdgedWidth + 16*x, iEdgedWidth, 65536); |
892 |
|
|
893 |
|
if (Data.chroma) { |
894 |
|
iSAD += sad8(pCurrent->u + x*8 + y*(iEdgedWidth/2)*8, pGMC->u + 8*y*(iEdgedWidth/2) + 8*x, iEdgedWidth/2); |
895 |
|
iSAD += sad8(pCurrent->v + (x + y*(iEdgedWidth/2))*8, pGMC->v + 8*y*(iEdgedWidth/2) + 8*x, iEdgedWidth/2); |
896 |
|
} |
897 |
|
|
898 |
|
if (iSAD <= stat_thresh) { /* mode decision GMC */ |
899 |
|
pMB->mode = MODE_INTER; |
900 |
|
pMB->sad16 = pMB->sad8[0] = pMB->sad8[1] = pMB->sad8[2] = pMB->sad8[3] = iSAD; |
901 |
|
pMB->mcsel = 1; |
902 |
|
if (Data.qpel) { |
903 |
|
pMB->qmvs[0] = pMB->qmvs[1] = pMB->qmvs[2] = pMB->qmvs[3] = pMB->amv; |
904 |
|
pMB->mvs[0].x = pMB->mvs[1].x = pMB->mvs[2].x = pMB->mvs[3].x = pMB->amv.x/2; |
905 |
|
pMB->mvs[0].y = pMB->mvs[1].y = pMB->mvs[2].y = pMB->mvs[3].y = pMB->amv.y/2; |
906 |
|
} else |
907 |
|
pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = pMB->amv; |
908 |
|
|
909 |
|
continue; |
910 |
|
} |
911 |
|
} |
912 |
|
else if (sad00 < stat_thresh) { |
913 |
ZeroMacroblockP(pMB, sad00); |
ZeroMacroblockP(pMB, sad00); |
914 |
|
pMB->cbp = 0x3f; |
915 |
continue; |
continue; |
916 |
} |
} |
917 |
|
} |
918 |
|
|
919 |
SearchP(pRef, pRefH->y, pRefV->y, pRefHV->y, pCurrent, x, |
SearchP(pRef, pRefH->y, pRefV->y, pRefHV->y, pCurrent, x, |
920 |
y, MotionFlags, current->vop_flags, |
y, MotionFlags, current->vop_flags, |
935 |
pCurrent, pRef, pGMC, current->coding_type, sad00); |
pCurrent, pRef, pGMC, current->coding_type, sad00); |
936 |
|
|
937 |
|
|
938 |
|
motionStatsPVOP(&MVmax, &mvCount, &mvSum, pMB, Data.qpel); |
939 |
} |
} |
940 |
} |
} |
|
return 0; |
|
|
} |
|
941 |
|
|
942 |
|
current->fcode = getMinFcode(MVmax); |
943 |
|
current->sStat.iMvSum = mvSum; |
944 |
|
current->sStat.iMvCount = mvCount; |
945 |
|
|
946 |
|
return 0; |
947 |
|
} |