estimation_rd_based.c

这是一个压缩解压包,用C语言进行编程的,里面有详细的源代码.

	}

	for (i = 0; i < 6; i++) {
		/* HVS models, anyone ? */
		Data->lambda[i] = LAMBDA;
	}

	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++) {
		/* HVS models, anyone ? */
		Data->lambda[i] = LAMBDA;
	}

	/* 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;
}