📄 estimation_rd_based.c
字号:
}
for (i = 0; i < 6; i++) {
Data->lambda[i] = (LAMBDA*pMB->lambda[i])>>LAMBDA_EXP;
}
min_rd = findRD_inter(Data, x, y, pParam, MotionFlags);
cbp = *Data->cbp;
if (coding_type == S_VOP) {
int gmc_rd;
*Data->iMinSAD = min_rd += BITS_MULT*1; /* mcsel */
gmc_rd = findRD_gmc(Data, vGMC, x, y);
if (gmc_rd < min_rd) {
mcsel = 1;
*Data->iMinSAD = min_rd = gmc_rd;
mode = MODE_INTER;
cbp = *Data->cbp;
}
}
if (inter4v) {
int v4_rd;
v4_rd = findRD_inter4v(Data, pMB, pMBs, x, y, pParam, MotionFlags, backup);
if (v4_rd < min_rd) {
Data->iMinSAD[0] = min_rd = v4_rd;
mode = MODE_INTER4V;
cbp = *Data->cbp;
}
}
/* there is no way for INTRA to take less than 24 bits - go to findRD_intra() for calculations */
if (min_rd > 24*BITS_MULT) {
intra_rd = findRD_intra(Data, pMB, x, y, pParam->mb_width);
if (intra_rd < min_rd) {
*Data->iMinSAD = min_rd = intra_rd;
mode = MODE_INTRA;
cbp = *Data->cbp;
}
}
pMB->sad16 = pMB->sad8[0] = pMB->sad8[1] = pMB->sad8[2] = pMB->sad8[3] = 0;
pMB->cbp = cbp;
if (mode == MODE_INTER && mcsel == 0) {
pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = Data->currentMV[0];
if(Data->qpel) {
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;
pMB->pmvs[0].y = Data->currentQMV[0].y - Data->predMV.y;
} else {
pMB->pmvs[0].x = Data->currentMV[0].x - Data->predMV.x;
pMB->pmvs[0].y = Data->currentMV[0].y - Data->predMV.y;
}
} else if (mode == MODE_INTER ) { /* but mcsel == 1 */
pMB->mcsel = 1;
if (Data->qpel) {
pMB->qmvs[0] = pMB->qmvs[1] = pMB->qmvs[2] = pMB->qmvs[3] = pMB->amv;
pMB->mvs[0].x = pMB->mvs[1].x = pMB->mvs[2].x = pMB->mvs[3].x = pMB->amv.x/2;
pMB->mvs[0].y = pMB->mvs[1].y = pMB->mvs[2].y = pMB->mvs[3].y = pMB->amv.y/2;
} else
pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = pMB->amv;
} else
if (mode == MODE_INTER4V) ; /* anything here? */
else /* INTRA, NOT_CODED */
ZeroMacroblockP(pMB, 0);
pMB->mode = mode;
}
void
xvid_me_ModeDecision_Fast(SearchData * const Data,
MACROBLOCK * const pMB,
const MACROBLOCK * const pMBs,
const int x, const int y,
const MBParam * const pParam,
const uint32_t MotionFlags,
const uint32_t VopFlags,
const uint32_t VolFlags,
const IMAGE * const pCurrent,
const IMAGE * const pRef,
const IMAGE * const vGMC,
const int coding_type)
{
int mode = MODE_INTER;
int mcsel = 0;
int inter4v = (VopFlags & XVID_VOP_INTER4V) && (pMB->dquant == 0);
const uint32_t iQuant = pMB->quant;
const int skip_possible = (coding_type == P_VOP) && (pMB->dquant == 0);
int sad;
int min_rd = -1, intra_rd, i, cbp = 63;
VECTOR backup[5], *v;
int sad_backup[5];
int InterBias = MV16_INTER_BIAS;
int thresh = 0;
int top = 0, top_right = 0, left = 0;
Data->scan_table = VopFlags & XVID_VOP_ALTERNATESCAN ?
scan_tables[2] : scan_tables[0];
pMB->mcsel = 0;
Data->iQuant = iQuant;
Data->quant_sq = iQuant*iQuant;
for (i = 0; i < 6; i++) {
Data->lambda[i] = (LAMBDA*pMB->lambda[i])>>LAMBDA_EXP;
}
/* INTER <-> INTER4V decision */
if ((Data->iMinSAD[0] + 75 < Data->iMinSAD[1] +
Data->iMinSAD[2] + Data->iMinSAD[3] + Data->iMinSAD[4])) { /* normal, fast, SAD-based mode decision */
if (inter4v == 0 || Data->iMinSAD[0] < Data->iMinSAD[1] + Data->iMinSAD[2] +
Data->iMinSAD[3] + Data->iMinSAD[4] + IMV16X16 * (int32_t)iQuant) {
mode = MODE_INTER;
sad = Data->iMinSAD[0];
} else {
mode = MODE_INTER4V;
sad = Data->iMinSAD[1] + Data->iMinSAD[2] +
Data->iMinSAD[3] + Data->iMinSAD[4] + IMV16X16 * (int32_t)iQuant;
Data->iMinSAD[0] = sad;
}
/* final skip decision, a.k.a. "the vector you found, really that good?" */
if (skip_possible && (pMB->sad16 < (int)iQuant * MAX_SAD00_FOR_SKIP))
if ( (100*sad)/(pMB->sad16+1) > FINAL_SKIP_THRESH)
if (Data->chroma || xvid_me_SkipDecisionP(pCurrent, pRef, x, y, Data->iEdgedWidth/2, iQuant)) {
mode = MODE_NOT_CODED;
sad = 0; /* Compiler warning */
goto early_out;
}
/* mcsel */
if (coding_type == S_VOP) {
int32_t iSAD = sad16(Data->Cur,
vGMC->y + 16*y*Data->iEdgedWidth + 16*x, Data->iEdgedWidth, 65536);
if (Data->chroma) {
iSAD += sad8(Data->CurU, vGMC->u + 8*y*(Data->iEdgedWidth/2) + 8*x, Data->iEdgedWidth/2);
iSAD += sad8(Data->CurV, vGMC->v + 8*y*(Data->iEdgedWidth/2) + 8*x, Data->iEdgedWidth/2);
}
if (iSAD <= sad) { /* mode decision GMC */
mode = MODE_INTER;
mcsel = 1;
sad = iSAD;
}
}
} else { /* Rate-Distortion INTER<->INTER4V */
Data->iQuant = iQuant;
v = Data->qpel ? Data->currentQMV : Data->currentMV;
/* final skip decision, a.k.a. "the vector you found, really that good?" */
if (skip_possible && (pMB->sad16 < (int)iQuant * MAX_SAD00_FOR_SKIP))
if ( (100*Data->iMinSAD[0])/(pMB->sad16+1) > FINAL_SKIP_THRESH)
if (Data->chroma || xvid_me_SkipDecisionP(pCurrent, pRef, x, y, Data->iEdgedWidth/2, iQuant)) {
mode = MODE_NOT_CODED;
sad = 0; /* Compiler warning */
goto early_out;
}
for (i = 0; i < 5; i++) {
sad_backup[i] = Data->iMinSAD[i];
Data->iMinSAD[i] = 256*4096;
backup[i] = v[i];
}
min_rd = findRD_inter(Data, x, y, pParam, MotionFlags);
cbp = *Data->cbp;
sad = sad_backup[0];
if (coding_type == S_VOP) {
int gmc_rd;
*Data->iMinSAD = min_rd += BITS_MULT*1; /* mcsel */
gmc_rd = findRD_gmc(Data, vGMC, x, y);
if (gmc_rd < min_rd) {
mcsel = 1;
*Data->iMinSAD = min_rd = gmc_rd;
mode = MODE_INTER;
cbp = *Data->cbp;
sad = sad16(Data->Cur,
vGMC->y + 16*y*Data->iEdgedWidth + 16*x, Data->iEdgedWidth, 65536);
if (Data->chroma) {
sad += sad8(Data->CurU, vGMC->u + 8*y*(Data->iEdgedWidth/2) + 8*x, Data->iEdgedWidth/2);
sad += sad8(Data->CurV, vGMC->v + 8*y*(Data->iEdgedWidth/2) + 8*x, Data->iEdgedWidth/2);
}
}
}
if (inter4v) {
int v4_rd;
v4_rd = findRD_inter4v(Data, pMB, pMBs, x, y, pParam, MotionFlags, backup);
if (v4_rd < min_rd) {
Data->iMinSAD[0] = min_rd = v4_rd;
mode = MODE_INTER4V;
cbp = *Data->cbp;
sad = sad_backup[1] + sad_backup[2] +
sad_backup[3] + sad_backup[4] + IMV16X16 * (int32_t)iQuant;
}
}
}
left = top = top_right = -1;
thresh = 0;
if((x > 0) && (y > 0) && (x < (int32_t) pParam->mb_width)) {
left = (&pMBs[(x-1) + y * pParam->mb_width])->sad16; /* left */
top = (&pMBs[x + (y-1) * pParam->mb_width])->sad16; /* top */
top_right = (&pMBs[(x+1) + (y-1) * pParam->mb_width])->sad16; /* top right */
if(((&pMBs[(x-1) + y * pParam->mb_width])->mode != MODE_INTRA) &&
((&pMBs[x + (y-1) * pParam->mb_width])->mode != MODE_INTRA) &&
((&pMBs[(x+1) + (y-1) * pParam->mb_width])->mode != MODE_INTRA)) {
thresh = MAX(MAX(top, left), top_right);
}
else
thresh = MIN(MIN(top, left), top_right);
}
/* INTRA <-> INTER decision */
if (sad < thresh) { /* normal, fast, SAD-based mode decision */
/* intra decision */
if (iQuant > 8) InterBias += 100 * (iQuant - 8); /* to make high quants work */
if (y != 0)
if ((pMB - pParam->mb_width)->mode == MODE_INTRA ) InterBias -= 80;
if (x != 0)
if ((pMB - 1)->mode == MODE_INTRA ) InterBias -= 80;
if (Data->chroma) InterBias += 50; /* dev8(chroma) ??? <-- yes, we need dev8 (no big difference though) */
if (InterBias < sad) {
int32_t deviation = dev16(Data->Cur, Data->iEdgedWidth);
if (deviation < (sad - InterBias)) mode = MODE_INTRA;
}
pMB->cbp = 63;
} else { /* Rate-Distortion INTRA<->INTER */
if(min_rd < 0) {
Data->iQuant = iQuant;
v = Data->qpel ? Data->currentQMV : Data->currentMV;
for (i = 0; i < 5; i++) {
Data->iMinSAD[i] = 256*4096;
backup[i] = v[i];
}
if(mode == MODE_INTER) {
min_rd = findRD_inter(Data, x, y, pParam, MotionFlags);
cbp = *Data->cbp;
if (coding_type == S_VOP) {
int gmc_rd;
*Data->iMinSAD = min_rd += BITS_MULT*1; /* mcsel */
gmc_rd = findRD_gmc(Data, vGMC, x, y);
if (gmc_rd < min_rd) {
mcsel = 1;
*Data->iMinSAD = min_rd = gmc_rd;
mode = MODE_INTER;
cbp = *Data->cbp;
}
}
}
if(mode == MODE_INTER4V) {
int v4_rd;
v4_rd = findRD_inter4v(Data, pMB, pMBs, x, y, pParam, MotionFlags, backup);
if (v4_rd < min_rd) {
Data->iMinSAD[0] = min_rd = v4_rd;
mode = MODE_INTER4V;
cbp = *Data->cbp;
}
}
}
intra_rd = findRD_intra(Data, pMB, x, y, pParam->mb_width);
if (intra_rd < min_rd) {
*Data->iMinSAD = min_rd = intra_rd;
mode = MODE_INTRA;
}
pMB->cbp = cbp;
}
early_out:
pMB->sad16 = pMB->sad8[0] = pMB->sad8[1] = pMB->sad8[2] = pMB->sad8[3] = sad;
if (mode == MODE_INTER && mcsel == 0) {
pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = Data->currentMV[0];
if(Data->qpel) {
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;
pMB->pmvs[0].y = Data->currentQMV[0].y - Data->predMV.y;
} else {
pMB->pmvs[0].x = Data->currentMV[0].x - Data->predMV.x;
pMB->pmvs[0].y = Data->currentMV[0].y - Data->predMV.y;
}
} else if (mode == MODE_INTER ) { /* but mcsel == 1 */
pMB->mcsel = 1;
if (Data->qpel) {
pMB->qmvs[0] = pMB->qmvs[1] = pMB->qmvs[2] = pMB->qmvs[3] = pMB->amv;
pMB->mvs[0].x = pMB->mvs[1].x = pMB->mvs[2].x = pMB->mvs[3].x = pMB->amv.x/2;
pMB->mvs[0].y = pMB->mvs[1].y = pMB->mvs[2].y = pMB->mvs[3].y = pMB->amv.y/2;
} else
pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = pMB->amv;
} else
if (mode == MODE_INTER4V) ; /* anything here? */
else /* INTRA, NOT_CODED */
ZeroMacroblockP(pMB, 0);
pMB->mode = mode;
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -