📄 rdopt.c
字号:
}
}
}
}
/*!
*************************************************************************************
* \brief
* Sets Coefficients and reconstruction for an 8x8 block
*************************************************************************************
*/
void SetCoeffAndReconstruction8x8 (Macroblock* currMB)
{
int block, k, j, i;
int cur_ref[2];
//============= MIXED TRANSFORM SIZES FOR 8x8 PARTITION ==============
//--------------------------------------------------------------------
int l;
int bframe = img->type==B_SLICE;
if (currMB->luma_transform_size_8x8_flag)
{
//============= set mode and ref. frames ==============
for(i = 0;i<4;i++)
{
currMB->b8mode[i] = tr8x8.part8x8mode[i];
currMB->b8pdir[i] = tr8x8.part8x8pdir[i];
}
if (bframe)
{
for (j = 0;j<4;j++)
for (i = 0;i<4;i++)
{
k = 2*(j >> 1)+(i >> 1);
l = 2*(j & 0x01)+(i & 0x01);
enc_picture->ref_idx[LIST_0][img->block_y+j][img->block_x+i] = ((currMB->b8pdir[k] & 0x01) == 0) ? tr8x8.part8x8fwref[k] : - 1;
enc_picture->ref_idx[LIST_1][img->block_y+j][img->block_x+i] = (currMB->b8pdir[k] > 0) ? tr8x8.part8x8bwref[k] : - 1;
}
}
else
{
for (j = 0;j<4;j++)
for (i = 0;i<4;i++)
{
k = 2*(j >> 1)+(i >> 1);
l = 2*(j & 0x01)+(i & 0x01);
enc_picture->ref_idx[LIST_0][img->block_y+j][img->block_x+i] = tr8x8.part8x8fwref[k];
}
}
for (j = img->block_y;j<img->block_y + BLOCK_MULTIPLE;j++)
{
for (i = img->block_x;i<img->block_x + BLOCK_MULTIPLE;i++)
{
cur_ref[LIST_0] = (int) enc_picture->ref_idx[LIST_0][j][i];
enc_picture->ref_pic_id [LIST_0][j][i] =(cur_ref[LIST_0]>=0
? enc_picture->ref_pic_num[LIST_0 + currMB->list_offset][cur_ref[LIST_0]]
: -1);
}
}
if (bframe)
{
for (j = img->block_y; j < img->block_y + BLOCK_MULTIPLE; j++)
{
for (i = img->block_x;i<img->block_x + BLOCK_MULTIPLE;i++)
{
cur_ref[LIST_1] = (int) enc_picture->ref_idx[LIST_1][j][i];
enc_picture->ref_pic_id [LIST_1][j][i] = (cur_ref[LIST_1]>=0
? enc_picture->ref_pic_num[LIST_1 + currMB->list_offset][cur_ref[LIST_1]]
: -1);
}
}
}
//====== set the mv's for 8x8 partition with transform size 8x8 ======
//save the mv data for 4x4 transform
StoreMV8x8(1);
//set new mv data for 8x8 transform
RestoreMV8x8(0);
//============= get pre-calculated data ==============
//restore coefficients from 8x8 transform
for (block = 0; block<4; block++)
{
for (k = 0; k<4; k++)
for (j = 0; j<2; j++)
memcpy (img->cofAC[block][k][j],cofAC_8x8ts[block][k][j], 65 * sizeof(int));
}
//restore reconstruction
if (cnt_nonz8_8x8ts <= _LUMA_8x8_COEFF_COST_ &&
((img->qp + img->bitdepth_luma_qp_scale)!=0 || img->lossless_qpprime_flag==0) &&
(img->type!=SP_SLICE))// modif ES added last condition (we probably never go there so is the next modification useful ? check)
{
currMB->cbp = 0;
currMB->cbp_blk = 0;
if(!img->residue_transform_flag) // Residue Color Transform
{
for (j = 0; j < MB_BLOCK_SIZE; j++)
{
memcpy(&enc_picture->imgY[img->pix_y+j][img->pix_x], tr8x8.mpr8x8[j], MB_BLOCK_SIZE * sizeof(imgpel));
if(img->type==SP_SLICE &&(!si_frame_indicator && !sp2_frame_indicator ))
memcpy(&lrec[img->pix_y+j][img->pix_x],tr8x8.lrec[j], MB_BLOCK_SIZE * sizeof(int));
}
}
else
{
for (j = 0; j < MB_BLOCK_SIZE; j++)
memset(rec_resG[j], 0, MB_BLOCK_SIZE * sizeof(int));
}
}
else
{
currMB->cbp = cbp8_8x8ts;
currMB->cbp_blk = cbp_blk8_8x8ts;
if(!img->residue_transform_flag) // Residue Color Transform
{
for (j = 0; j < MB_BLOCK_SIZE; j++)
{
memcpy (&enc_picture->imgY[img->pix_y+j][img->pix_x],tr8x8.rec_mbY8x8[j], MB_BLOCK_SIZE * sizeof(imgpel));
if(img->type==SP_SLICE &&(!si_frame_indicator && !sp2_frame_indicator))
memcpy (&lrec[img->pix_y+j][img->pix_x],tr8x8.lrec[j], MB_BLOCK_SIZE * sizeof(int));
}
}
else
{
for (j = 0; j < MB_BLOCK_SIZE; j++)
memcpy (rec_resG[j], tr8x8.rec_resG_8x8[j], MB_BLOCK_SIZE * sizeof(int));
}
}
if(img->residue_transform_flag) // Residue Color Transform
{
// Residue Color Transform
for (j = 0; j < MB_BLOCK_SIZE; j++)
for (i = 0; i < MB_BLOCK_SIZE; i++)
{
mprRGB[0][j][i] = tr8x8.mprRGB_8x8[0][j][i];
mprRGB[1][j][i] = tr8x8.mprRGB_8x8[1][j][i];
mprRGB[2][j][i] = tr8x8.mprRGB_8x8[2][j][i];
resTrans_R[j][i] = tr8x8.resTrans_R_8x8[j][i];
resTrans_B[j][i] = tr8x8.resTrans_B_8x8[j][i];
}
}
}
else
{
//============= get pre-calculated data ==============
//---------------------------------------------------
//--- restore coefficients ---
for (block = 0; block<4+img->num_blk8x8_uv; block++)
{
for (k = 0; k<4; k++)
for (j = 0; j<2; j++)
memcpy (img->cofAC[block][k][j],cofAC8x8[block][k][j], 65 * sizeof(int));
}
if (cnt_nonz_8x8<=5 && img->type!=SP_SLICE &&
((img->qp + img->bitdepth_luma_qp_scale)!=0 || img->lossless_qpprime_flag==0))
{
currMB->cbp = 0;
currMB->cbp_blk = 0;
if(!img->residue_transform_flag) // Residue Color Transform
{
for (j = 0; j < MB_BLOCK_SIZE; j++)
{
memcpy (&enc_picture->imgY[img->pix_y+j][img->pix_x],tr4x4.mpr8x8[j], MB_BLOCK_SIZE * sizeof(imgpel));
if(img->type ==SP_SLICE &&(!si_frame_indicator && !sp2_frame_indicator))
memcpy (&lrec[img->pix_y+j][img->pix_x],tr4x4.lrec[j], MB_BLOCK_SIZE * sizeof(int)); // restore coeff. SP frame
}
}
else
{
for (j = 0; j < MB_BLOCK_SIZE; j++)
memset(rec_resG[j], 0, MB_BLOCK_SIZE * sizeof(int));
}
}
else
{
currMB->cbp = cbp8x8;
currMB->cbp_blk = cbp_blk8x8;
if(!img->residue_transform_flag) // Residue Color Transform
{
for (j = 0; j < MB_BLOCK_SIZE; j++)
{
memcpy (&enc_picture->imgY[img->pix_y+j][img->pix_x],tr4x4.rec_mbY8x8[j], MB_BLOCK_SIZE * sizeof(imgpel));
if(img->type==SP_SLICE &&(!si_frame_indicator && !sp2_frame_indicator))
memcpy (&lrec[img->pix_y+j][img->pix_x],tr4x4.lrec[j], MB_BLOCK_SIZE * sizeof(int));
}
}
else
{
for (j = 0; j < MB_BLOCK_SIZE; j++)
memcpy(rec_resG[j], tr4x4.rec_resG_8x8[j], MB_BLOCK_SIZE * sizeof(int));
}
}
if(img->residue_transform_flag) // Residue Color Transform
{
// Residue Color Transform
for (j = 0; j < MB_BLOCK_SIZE; j++)
{
memcpy(mprRGB[0][j], tr4x4.mprRGB_8x8[0][j], MB_BLOCK_SIZE * sizeof(int));
memcpy(mprRGB[1][j], tr4x4.mprRGB_8x8[1][j], MB_BLOCK_SIZE * sizeof(int));
memcpy(mprRGB[2][j], tr4x4.mprRGB_8x8[2][j], MB_BLOCK_SIZE * sizeof(int));
memcpy(resTrans_R[j], tr4x4.resTrans_R_8x8[j], MB_BLOCK_SIZE * sizeof(int));
memcpy(resTrans_B[j], tr4x4.resTrans_B_8x8[j], MB_BLOCK_SIZE * sizeof(int));
}
}
}
}
/*!
*************************************************************************************
* \brief
* Sets motion vectors for a macroblock
*************************************************************************************
*/
void SetMotionVectorsMB (Macroblock* currMB, int bframe)
{
int i, j, k, l, m, mode8, pdir8, ref, by, bx;
short ******all_mv = img->all_mv;
short ******pred_mv = img->pred_mv;
int bw_ref;
int jdiv, jmod;
if (!bframe)
{
for (j = 0; j<4; j++)
{
jmod = j & 0x01;
jdiv = j >> 1;
by = img->block_y+j;
for (i = 0; i<4; i++)
{
mode8 = currMB->b8mode[k=2*jdiv+(i>>1)];
l = 2*jmod + (i & 0x01);
bx = img->block_x+i;
pdir8 = currMB->b8pdir[k];
ref = enc_picture->ref_idx[LIST_0][by][bx];
if (pdir8>=0)
{
enc_picture->mv[LIST_0][by][bx][0] = all_mv [j][i][LIST_0][ ref][mode8][0];
enc_picture->mv[LIST_0][by][bx][1] = all_mv [j][i][LIST_0][ ref][mode8][1];
}
else
{
enc_picture->mv[LIST_0][by][bx][0] = 0;
enc_picture->mv[LIST_0][by][bx][1] = 0;
}
}
}
}
else
{
for (j = 0; j<4; j++)
{
jmod = j & 0x01;
jdiv = j >> 1;
by = img->block_y+j;
for (i = 0; i<4; i++)
{
mode8 = currMB->b8mode[k=2*jdiv+(i>>1)];
l = 2*jmod + (i & 0x01);
bx = img->block_x+i;
pdir8 = currMB->b8pdir[k];
ref = enc_picture->ref_idx[LIST_0][by][bx];
bw_ref = enc_picture->ref_idx[LIST_1][by][bx];
if (currMB->bi_pred_me && (pdir8 == 2) && currMB->mb_type==1)
{
all_mv = currMB->bi_pred_me == 1 ? img->bipred_mv1 : img->bipred_mv2;
ref = 0;
bw_ref = 0;
}
if (pdir8==-1) // intra
{
enc_picture->mv[LIST_0][by][bx][0] = 0;
enc_picture->mv[LIST_0][by][bx][1] = 0;
enc_picture->mv[LIST_1][by][bx][0] = 0;
enc_picture->mv[LIST_1][by][bx][1] = 0;
}
else if (pdir8==0) // list 0
{
enc_picture->mv[LIST_0][by][bx][0] = all_mv [j][i][LIST_0][ ref][mode8][0];
enc_picture->mv[LIST_0][by][bx][1] = all_mv [j][i][LIST_0][ ref][mode8][1];
enc_picture->mv[LIST_1][by][bx][0] = 0;
enc_picture->mv[LIST_1][by][bx][1] = 0;
enc_picture->ref_idx[LIST_1][by][bx] = -1;
}
else if (pdir8==1) // list 1
{
enc_picture->mv[LIST_0][by][bx][0] = 0;
enc_picture->mv[LIST_0][by][bx][1] = 0;
enc_picture->ref_idx[LIST_0][by][bx] = -1;
enc_picture->mv[LIST_1][by][bx][0] = all_mv [j][i][LIST_1][bw_ref][mode8][0];
enc_picture->mv[LIST_1][by][bx][1] = all_mv [j][i][LIST_1][bw_ref][mode8][1];
}
else if (pdir8==2) // bipredictive
{
enc_picture->mv[LIST_0][by][bx][0] = all_mv [j][i][LIST_0][ ref][mode8][0];
enc_picture->mv[LIST_0][by][bx][1] = all_mv [j][i][LIST_0][ ref][mode8][1];
enc_picture->mv[LIST_1][by][bx][0] = all_mv [j][i][LIST_1][bw_ref][mode8][0];
enc_picture->mv[LIST_1][by][bx][1] = all_mv [j][i][LIST_1][bw_ref][mode8][1];
}
else
{
error("invalid direction mode", 255);
}
}
}
}
// copy all the motion vectors into rdopt structure
// Can simplify this by copying the MV's of the best mode (TBD)
if(img->MbaffFrameFlag)
{
for(i = 0;i<4;i++)
{
for(j = 0;j<4;j++)
{
for (k = 0;k<2;k++)
{
for(l = 0;l<img->max_num_references;l++)
{
for(m = 0;m<9;m++)
{
rdopt->all_mv [j][i][k][l][m][0] = all_mv [j][i][k][l][m][0];
rdopt->pred_mv[j][i][k][l][m][0] = pred_mv[j][i][k][l][m][0];
rdopt->all_mv [j][i][k][l][m][1] = all_mv [j][i][k][l][m][1];
rdopt->pred_mv[j][i][k][l][m][1] = pred_mv[j][i][k][l][m][1];
}
}
}
}
}
}
}
/*!
*************************************************************************************
* \brief
* R-D Cost for a macroblock
*************************************************************************************
*/
int RDCost_for_macroblocks (double lambda, // <-- lagrange multiplier
int mode, // <-- modus (0-COPY/DIRECT, 1-16x16, 2-16x8, 3-8x16, 4-8x8(+), 5-Intra4x4, 6-Intra16x16)
double* min_rdcost, // <-> minimum rate-distortion cost
double* min_rate, // --> bitrate of mode which has minimum rate-distortion cost.
int i16mode )
{
int i, j, k; //, k, ****ip4;
int j1, j2;
int rate = 0, coeff_rate = 0;
int64 distortion = 0;
double rdcost;
int prev_mb_nr = FmoGetPreviousMBNr(img->current_mb_nr);
Macroblock *currMB = &img->mb_data[img->current_mb_nr];
Macroblock *prevMB = (prev_mb_nr >= 0) ? &img->mb_data[prev_mb_nr] : NULL;
int bframe = (img->type==B_SLICE);
int tmp_cc;
int use_of_cc = (img->type!=I_SLICE && input->symbol_mode!=CABAC);
int cc_rate, dummy;
int cr_cbp = 0, uv;
//=====
//===== SET REFERENCE FRAMES AND BLOCK MODES
//=====
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -