📄 macroblock.c
字号:
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
#ifdef USE_HP_FILTER//BROUND
if(input->UseHPFilter != 0)
{
img->mpr[j][i] = clip1a(((wbp_weight[0][fw_ref_idx][bw_ref_idx][0] * *fpred++ +
wbp_weight[1][fw_ref_idx][bw_ref_idx][0] * *bpred++ +
(wp_offset[0][fw_ref_idx][0] + wp_offset[1][bw_ref_idx][0] )*2*wp_luma_round +
2*wp_luma_round*(1-img->bipred_rounding_control)) >> (luma_log_weight_denom + 1))
);
}
else
{
img->mpr[j][i] = clip1a(((wbp_weight[0][fw_ref_idx][bw_ref_idx][0] * *fpred++ +
wbp_weight[1][fw_ref_idx][bw_ref_idx][0] * *bpred++ +
2*wp_luma_round) >> (luma_log_weight_denom + 1)) +
((wp_offset[0][fw_ref_idx][0] + wp_offset[1][bw_ref_idx][0] + 1)>>1));
}
#else
img->mpr[j][i] = clip1a(((wbp_weight[0][fw_ref_idx][bw_ref_idx][0] * *fpred++ +
wbp_weight[1][fw_ref_idx][bw_ref_idx][0] * *bpred++ +
2*wp_luma_round) >> (luma_log_weight_denom + 1)) +
((wp_offset[0][fw_ref_idx][0] + wp_offset[1][bw_ref_idx][0] + 1)>>1));
#endif
}
else if (p_dir==0)
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
img->mpr[j][i] = clip1a(((wp_weight[0][fw_ref_idx][0] * *fpred++ + wp_luma_round) >> luma_log_weight_denom) +
wp_offset[0][fw_ref_idx][0] );
}
else // p_dir==1
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
img->mpr[j][i] = clip1a(((wp_weight[1][bw_ref_idx][0] * *bpred++ + wp_luma_round) >> luma_log_weight_denom) +
wp_offset[1][bw_ref_idx][0] );
}
}
else
{
if (p_dir==2)
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
#ifdef USE_HP_FILTER//BROUND
if(input->UseHPFilter != 0)
{
img->mpr[j][i] = (*fpred++ + *bpred++ + 1 - img->bipred_rounding_control) >> 1;
}
else
{
img->mpr[j][i] = (*fpred++ + *bpred++ + 1) >> 1;
}
#else
img->mpr[j][i] = (*fpred++ + *bpred++ + 1) >> 1;
#endif
}
else if (p_dir==0)
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
img->mpr[j][i] = *fpred++;
}
else // p_dir==1
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
img->mpr[j][i] = *bpred++;
}
}
}
/*!
************************************************************************
* \brief
* Predict one 4x4 Luma block
************************************************************************
*/
void LumaPrediction4x4Bi ( int block_x, //!< relative horizontal block coordinate of 4x4 block
int block_y, //!< relative vertical block coordinate of 4x4 block
int p_dir, //!< prediction direction (0=list0, 1=list1, 2=bidir)
int fw_mode, //!< list0 prediction mode (1-7, 0=DIRECT if bw_mode=0)
int bw_mode, //!< list1 prediction mode (1-7, 0=DIRECT if fw_mode=0)
short fw_ref_idx, //!< reference frame for list0 prediction (-1: Intra4x4 pred. with fw_mode)
short bw_ref_idx, //!< reference frame for list1 prediction
int list //!< current list for prediction.
)
{
static int fw_pred[16];
static int bw_pred[16];
int i, j;
int block_x4 = block_x+4;
int block_y4 = block_y+4;
int pic_opix_x = img->opix_x + block_x;
int pic_opix_y = img->opix_y + block_y;
int bx = block_x >> 2;
int by = block_y >> 2;
int* fpred = fw_pred;
int* bpred = bw_pred;
Macroblock* currMB = &img->mb_data[img->current_mb_nr];
int apply_weights = ( (active_pps->weighted_pred_flag && (img->type == P_SLICE || img->type == SP_SLICE)) ||
(active_pps->weighted_bipred_idc && (img->type == B_SLICE)));
short ****mv_array = list ? img->bipred_mv1[by][bx] : img->bipred_mv2[by][bx];
if ((p_dir==0)||(p_dir==2))
{
OneComponentLumaPrediction4x4 (fw_pred, pic_opix_x, pic_opix_y, mv_array[LIST_0][fw_ref_idx][fw_mode], fw_ref_idx, listX[0+currMB->list_offset]);
}
if ((p_dir==1)||(p_dir==2))
{
OneComponentLumaPrediction4x4 (bw_pred, pic_opix_x, pic_opix_y, mv_array[LIST_1][bw_ref_idx][bw_mode], bw_ref_idx, listX[1+currMB->list_offset]);
}
if (apply_weights)
{
if (p_dir==2)
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
#ifdef USE_HP_FILTER//BROUND
if(input->UseHPFilter != 0)
{
img->mpr[j][i] = clip1a(((wbp_weight[0][fw_ref_idx][bw_ref_idx][0] * *fpred++ +
wbp_weight[1][fw_ref_idx][bw_ref_idx][0] * *bpred++ +
(wp_offset[0][fw_ref_idx][0] + wp_offset[1][bw_ref_idx][0])*2*wp_luma_round+
2*wp_luma_round*(1-img->bipred_rounding_control)) >> (luma_log_weight_denom + 1))
);
}
else
{
img->mpr[j][i] = clip1a(((wbp_weight[0][fw_ref_idx][bw_ref_idx][0] * *fpred++ +
wbp_weight[1][fw_ref_idx][bw_ref_idx][0] * *bpred++ +
2*wp_luma_round) >> (luma_log_weight_denom + 1)) +
((wp_offset[0][fw_ref_idx][0] + wp_offset[1][bw_ref_idx][0] + 1)>>1));
}
#else
img->mpr[j][i] = clip1a(((wbp_weight[0][fw_ref_idx][bw_ref_idx][0] * *fpred++ +
wbp_weight[1][fw_ref_idx][bw_ref_idx][0] * *bpred++ +
2*wp_luma_round) >> (luma_log_weight_denom + 1)) +
((wp_offset[0][fw_ref_idx][0] + wp_offset[1][bw_ref_idx][0] + 1)>>1));
#endif
}
else if (p_dir==0)
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
img->mpr[j][i] = clip1a(((wp_weight[0][fw_ref_idx][0] * *fpred++ + wp_luma_round) >> luma_log_weight_denom) +
wp_offset[0][fw_ref_idx][0] );
}
else // p_dir==1
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
img->mpr[j][i] = clip1a(((wp_weight[1][bw_ref_idx][0] * *bpred++ + wp_luma_round) >> luma_log_weight_denom)
+ wp_offset[1][bw_ref_idx][0] );
}
}
else
{
if (p_dir==2)
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
#ifdef USE_HP_FILTER//BROUND
if(input->UseHPFilter != 0)
{
img->mpr[j][i] = (*fpred++ + *bpred++ + 1 - img->bipred_rounding_control) / 2;
}
else
{
img->mpr[j][i] = (*fpred++ + *bpred++ + 1) / 2;
}
#else
img->mpr[j][i] = (*fpred++ + *bpred++ + 1) / 2;
#endif
}
else if (p_dir==0)
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
img->mpr[j][i] = *fpred++;
}
else // p_dir==1
{
for (j=block_y; j<block_y4; j++)
for (i=block_x; i<block_x4; i++)
img->mpr[j][i] = *bpred++;
}
}
}
/*!
************************************************************************
* \brief
* Residual Coding of an 8x8 Luma block (not for intra)
*
* \return
* coefficient cost
************************************************************************
*/
int LumaResidualCoding8x8 ( int *cbp, //!< Output: cbp (updated according to processed 8x8 luminance block)
int64 *cbp_blk, //!< Output: block cbp (updated according to processed 8x8 luminance block)
int block8x8, //!< block number of 8x8 block
short p_dir, //!< prediction direction
int fw_mode, //!< list0 prediction mode (1-7, 0=DIRECT)
int bw_mode, //!< list1 prediction mode (1-7, 0=DIRECT)
short fw_refframe, //!< reference picture for list0 prediction
short bw_refframe //!< reference picture for list0 prediction
)
{
int block_y, block_x, pic_pix_y, pic_pix_x, i, j, nonzero = 0, cbp_blk_mask;
int coeff_cost = 0;
int mb_y = (block8x8 >> 1) << 3;
int mb_x = (block8x8 & 0x01) << 3;
int pix_y;
int cbp_mask = 1 << block8x8;
int bxx, byy; // indexing curr_blk
int scrFlag = 0; // 0=noSCR, 1=strongSCR, 2=jmSCR
int skipped = (fw_mode == 0 && bw_mode == 0 && (img->type != B_SLICE));
Macroblock* currMB = &img->mb_data[img->current_mb_nr];
//set transform size
int need_8x8_transform = currMB->luma_transform_size_8x8_flag;
// Residue Color Transform
int residue_R, residue_G, residue_B, temp;
#ifdef ADAPTIVE_FD_SD_CODING
for (j=0;j<4;j++)
{
memset(img->cofAC[block8x8][j][0],0,65 * sizeof(int));
memset(img->cofAC[block8x8][j][1],0,65 * sizeof(int));
}
#endif
if (img->type==B_SLICE)
scrFlag = 1;
//===== loop over 4x4 blocks =====
for (byy=0, block_y=mb_y; block_y<mb_y+8; byy+=4, block_y+=4)
{
pic_pix_y = img->opix_y + block_y;
for (bxx=0, block_x=mb_x; block_x<mb_x+8; bxx+=4, block_x+=4)
{
pic_pix_x = img->opix_x + block_x;
cbp_blk_mask = (block_x>>2) + block_y;
// Residue Color Transform
if(img->residue_transform_flag)
{
ChromaPrediction4x4 (0, block_x, block_y, p_dir, fw_mode, bw_mode, fw_refframe, bw_refframe);
for (j=0; j<4; j++)
for (i=0; i<4; i++)
mprRGB[1][j+block_y][i+block_x] = img->mpr[j+block_y][i+block_x];
ChromaPrediction4x4 (1, block_x, block_y, p_dir, fw_mode, bw_mode, fw_refframe, bw_refframe);
for (j=0; j<4; j++)
for (i=0; i<4; i++)
mprRGB[2][j+block_y][i+block_x] = img->mpr[j+block_y][i+block_x];
}
//===== prediction of 4x4 block =====
LumaPrediction4x4 (block_x, block_y, p_dir, fw_mode, bw_mode, fw_refframe, bw_refframe);
// Residue Color Transform
if(img->residue_transform_flag)
{
for (j=0; j<4; j++)
for (i=0; i<4; i++)
mprRGB[0][j+block_y][i+block_x] = img->mpr[j+block_y][i+block_x];
}
//===== get displaced frame difference ======
if(!img->residue_transform_flag)
{
if(!need_8x8_transform)
{
//===== get displaced frame difference ======
for (j=0; j<4; j++)
{
pix_y = pic_pix_y + j;
for (i=0; i<4; i++)
{
img->m7[j][i] = imgY_org[pix_y][pic_pix_x + i] - img->mpr[j+block_y][i+block_x];
#ifdef ADAPTIVE_FD_SD_CODING
currMB->prederror [j+block_y][i+block_x] = img->m7 [j][i];
currMB->prediction[j+block_y][i+block_x] = img->mpr[j+block_y][i+block_x];
#endif
}
}
//===== DCT, Quantization, inverse Quantization, IDCT, Reconstruction =====
if ( (img->NoResidueDirect != 1 && !skipped ) ||
((img->qp + img->bitdepth_luma_qp_scale)==0 && img->lossless_qpprime_flag==1) )
{
//===== DCT, Quantization, inverse Quantization, IDCT, Reconstruction =====
if (img->type!=SP_SLICE)
#ifdef USE_INTRA_MDDT
nonzero = dct_luma (block_x, block_y, &coeff_cost, 0, -1);
#else
nonzero = dct_luma (block_x, block_y, &coeff_cost, 0);
#endif
else if(!si_frame_indicator && !sp2_frame_indicator)
nonzero = dct_luma_sp(block_x, block_y, &coeff_cost);// SP frame encoding
else
nonzero = dct_luma_sp2(block_x, block_y, &coeff_cost);//switching SP/SI encoding
if (nonzero)
{
(*cbp_blk) |= (int64)1 << cbp_blk_mask; // one bit for every 4x4 block
(*cbp) |= cbp_mask; // one bit for the 4x4 blocks of an 8x8 block
}
}
}
else
{
for (j=0; j<4; j++)
{
pix_y = pic_pix_y + j;
for (i=0; i<4; i++)
{
img->m7[j+byy][i+bxx] = imgY_org[pix_y][pic_pix_x+i] - img->mpr[j+block_y][i+block_x];
#ifdef ADAPTIVE_FD_SD_CODING
currMB->prederror [j+block_y][i+block_x] = img->m7 [j+byy][i+bxx];
currMB->prediction[j+block_y][i+block_x] = img->mpr[j+block_y][i+block_x];
#endif
}
}
}
}
else
{
/* Forward Residue Transform */
for (j=0; j<4; j++)
for (i=0; i<4; i++)
{
residue_B = imgUV_org[0][pic_pix_y+j][pic_pix_x+i] - mprRGB[1][j+block_y][i+block_x];
residue_G = imgY_org[pic_pix_y+j][pic_pix_x+i] - mprRGB[0][j+block_y][i+block_x];
residue_R = imgUV_org[1][pic_pix_y+j][pic_pix_x+i] - mprRGB[2][j+block_y][i+block_x];
resTrans_R[j+block_y][i+block_x] = residue_R-residue_B;
temp = residue_B+(resTrans_R[j+block_y][i+block_x]>>1);
resTrans_B[j+block_y][i+block_x] = residue_G-temp;
resTrans_G[j+block_y][i+block_x] = temp+(resTrans_B[j+block_y][i+block_x]>>1);
if(!need_8x8_transform)
img->m7[j][i] = resTrans_G[j+block_y][i+block_x];
else
img->m7[j+byy][i+bxx] = resTrans_G[j+block_y][i+block_x];
}
// Residue Color Transform
//===== DCT, Quantization, inverse Quantization, IDCT, Reconstruction =====
if (img->NoResidueDirect != 1 && !skipped && !need_8x8_transform )
{
//===== DCT, Quantization, inverse Quantization, IDCT, Reconstruction =====
#ifdef USE_INTRA_MDDT
if (img->type!=SP_SLICE) nonzero = dct_luma (block_x, block_y, &coeff_cost, 0, -1);
#else
if (img->type!=SP_SLICE) nonzero = dct_luma (block_x, block_y, &coeff_cost, 0);
#endif
else if(!si_frame_indicator && !sp2_frame_indicator)
nonzero = dct_luma_sp(block_x, block_y, &coeff_cost);//SP frame encoding
else
nonzero = dct_luma_sp2(block_x, block_y, &coeff_cost);//switching SP of SI encoding
for (j=0; j<4; j++)
for (i=0; i<4; i++)
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -