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 |
|
} |