📄 h.264
字号:
return coeff_cost;
}
/*!
************************************************************************
* \brief
* Set mode parameters and reference frames for an 8x8 block
************************************************************************
*/
void
SetModesAndRefframe (int b8, int* fw_mode, int* bw_mode, int* fw_ref, int* bw_ref)
{
Macroblock* currMB = &img->mb_data[img->current_mb_nr];
int j = 2*(b8/2);
int i = 2*(b8%2);
int** frefarr = refFrArr; // For MB level field/frame coding
int** fw_refarr = fw_refFrArr; // For MB level field/frame coding
int** bw_refarr = bw_refFrArr; // For MB level field/frame coding
int block_y = img->block_y; // For MB level field/frame coding
if(input->InterlaceCodingOption >= MB_CODING && mb_adaptive && img->field_mode)
{
block_y = img->field_block_y;
if(img->top_field)
{
frefarr = refFrArr_top;
fw_refarr = fw_refFrArr_top;
bw_refarr = bw_refFrArr_top;
}
else
{
frefarr = refFrArr_bot;
fw_refarr = fw_refFrArr_bot;
bw_refarr = bw_refFrArr_bot;
}
}
*fw_mode = *bw_mode = *fw_ref = *bw_ref = -1;
if (img->type!=B_SLICE)
{
*fw_ref = frefarr[block_y+j][img->block_x+i];
*bw_ref = 0;
*bw_mode = 0;
*fw_mode = currMB->b8mode[b8];
}
else
{
if (currMB->b8pdir[b8]==-1)
{
*fw_ref = -1;
*bw_ref = -1;
*fw_mode = 0;
*bw_mode = 0;
}
else if (currMB->b8pdir[b8]==0)
{
*fw_ref = fw_refarr[block_y+j][img->block_x+i];
*bw_ref = 0;
*fw_mode = currMB->b8mode[b8];
*bw_mode = 0;
}
else if (currMB->b8pdir[b8]==1)
{
*fw_ref = 0;
*bw_ref = bw_refarr[block_y+j][img->block_x+i];
*fw_mode = 0;
*bw_mode = currMB->b8mode[b8];
}
else
{
*fw_ref = fw_refarr[block_y+j][img->block_x+i];
*bw_ref = bw_refarr[block_y+j][img->block_x+i];
*fw_mode = currMB->b8mode[b8];
*bw_mode = currMB->b8mode[b8];
if (currMB->b8mode[b8]==0) // direct
{
if (img->type==B_SLICE && img->nal_reference_idc>0)
{
*fw_ref = 0;
*bw_ref = 0;
}
else if (img->type==B_SLICE)
{
*fw_ref = max(0,frefarr[block_y+j][img->block_x+i]);
*bw_ref = 0;
}
else
{
*fw_ref = max(0,frefarr[block_y+j][img->block_x+i]);
*bw_ref = 0;
}
}
}
}
}
/*!
************************************************************************
* \brief
* Residual Coding of a Luma macroblock (not for intra)
************************************************************************
*/
void
LumaResidualCoding ()
{
int i,j,block8x8,b8_x,b8_y;
int fw_mode, bw_mode, refframe;
int sum_cnt_nonz;
Macroblock *currMB = &img->mb_data[img->current_mb_nr];
currMB->cbp = 0 ;
currMB->cbp_blk = 0 ;
sum_cnt_nonz = 0 ;
for (block8x8=0; block8x8<4; block8x8++)
{
int bw_ref;
SetModesAndRefframe (block8x8, &fw_mode, &bw_mode, &refframe, &bw_ref);
sum_cnt_nonz += LumaResidualCoding8x8 (&(currMB->cbp), &(currMB->cbp_blk), block8x8,
fw_mode, bw_mode, refframe, bw_ref);
}
if (sum_cnt_nonz <= 5 )
{
currMB->cbp &= 0xfffff0 ;
currMB->cbp_blk &= 0xff0000 ;
for (i=0; i < MB_BLOCK_SIZE; i++)
{
for (j=0; j < MB_BLOCK_SIZE; j++)
{
enc_picture->imgY[img->pix_y+j][img->pix_x+i]=img->mpr[i][j];
}
}
if (img->type==SP_SLICE)
{
for(block8x8=0;block8x8<4;block8x8++)
{
b8_x=(block8x8&1)<<3;
b8_y=(block8x8&2)<<2;
for (i=0;i<8;i+=4)
for (j=0;j<8;j+=4)
copyblock_sp(b8_x+i,b8_y+j);
}
}
}
}
/*!
************************************************************************
* \brief
* Predict one component of a chroma 4x4 block
************************************************************************
*/
void
OneComponentChromaPrediction4x4 (int* mpred, // --> array to store prediction values
int pix_c_x, // <-- horizontal pixel coordinate of 4x4 block
int pix_c_y, // <-- vertical pixel coordinate of 4x4 block
int***** mv, // <-- motion vector array
int ref, // <-- reference frame parameter (0.../ -1: backward)
int blocktype, // <-- block type
int uv) // <-- chroma component
{
int i, j, ii, jj, ii0, jj0, ii1, jj1, if0, if1, jf0, jf1;
int incr;
int* mvb;
int refframe = (ref<0 ? 0 : ref);
pel_t** refimage;
int je = pix_c_y + 4;
int ie = pix_c_x + 4;
int f1 = 8 , f2=f1-1, f3=f1*f1, f4=f3>>1;
int s1 = 3;
int img_pic_c_y = img->pix_c_y;
int scale = 1;
int pred_dir = (ref<0 ? 1 : 0);
int ref_idx;
int list_offset;
StorablePicture **list;
int curr_mb_field = ((img->MbaffFrameFlag)&&(img->mb_data[img->current_mb_nr].mb_field));
// find out the correct list offsets
if (curr_mb_field)
{
if(img->current_mb_nr%2)
list_offset = 4; // top field mb
else
list_offset = 2; // bottom field mb
//max_y_cr = img->height_cr/2-1;
}
else
{
list_offset = 0; // no mb aff or frame mb
//max_y_cr = img->height_cr-1;
}
list = listX[0+list_offset+ pred_dir];
incr = (ref==-1 ? (!img->fld_type&&enc_picture!=enc_frame_picture): direct_mode ? (!img->fld_type&&enc_picture!=enc_frame_picture) : (enc_picture!=enc_frame_picture)) ;
if(input->InterlaceCodingOption >= MB_CODING && mb_adaptive && img->field_mode)
{
incr = (ref == -1 ? (img->top_field&&img->field_mode):(direct_mode ? (img->top_field&&img->field_mode):(img->field_mode)));
scale = 2;
img_pic_c_y = img->field_pix_c_y;
}
//refimage = img->type==B_SLICE? mcef [ref+1+incr][uv] : mcef [ref][uv];
refimage = (ref<0 ? listX[LIST_1][0]->imgUV[uv]: listX[LIST_0][ref]->imgUV[uv]);
for (j=pix_c_y; j<je; j++)
for (i=pix_c_x; i<ie; i++)
{
ref_idx = enc_picture->ref_idx[LIST_0+pred_dir][pix_c_x>>1][pix_c_y>>1];
mvb = mv [(i-img->pix_c_x)>>1][(j-img_pic_c_y)>>1][refframe][blocktype];
ii = (i<<s1) + mvb[0];
jj = (j<<s1) + mvb[1];
//SW
/*
if (!((img->MbaffFrameFlag)&&(img->mb_data[img->current_mb_nr].mb_field)))
j1=(img->pix_c_y+jj+joff)*f1+mv_array[if1][jf][1];
else
{
if (img->current_mb_nr%2 == 0)
j1=(img->pix_c_y)/2*f1 + (jj+joff)*f1+mv_array[if1][jf][1];
else
j1=(img->pix_c_y-8)/2*f1 + (jj+joff)*f1 +mv_array[if1][jf][1];
}
*/
if (ref_idx!=-1)
jj += list[ref_idx]->chroma_vector_adjustment;
else
{
if (img->type!= I_SLICE)
jj += list[0]->chroma_vector_adjustment;
}
ii0 = max (0, min (img->width_cr -1, ii>>s1 ));
jj0 = max (0, min (img->height_cr/scale-1, jj>>s1 )); // For MB level field/frame -- scale chroma height by 2
ii1 = max (0, min (img->width_cr -1, (ii+f2)>>s1));
jj1 = max (0, min (img->height_cr/scale-1, (jj+f2)>>s1));
if1 = (ii&f2); if0 = f1-if1;
jf1 = (jj&f2); jf0 = f1-jf1;
*mpred++ = (if0 * jf0 * refimage[jj0][ii0] +
if1 * jf0 * refimage[jj0][ii1] +
if0 * jf1 * refimage[jj1][ii0] +
if1 * jf1 * refimage[jj1][ii1] + f4) / f3;
}
}
/*!
************************************************************************
* \brief
* Predict an intra chroma 4x4 block
************************************************************************
*/
void IntraChromaPrediction4x4 (int uv, // <-- colour component
int block_x, // <-- relative horizontal block coordinate of 4x4 block
int block_y) // <-- relative vertical block coordinate of 4x4 block
{
int mode = img->mb_data[img->current_mb_nr].c_ipred_mode;
int i, j;
//===== prediction =====
for (j=block_y; j<block_y+4; j++)
for (i=block_x; i<block_x+4; i++)
{
img->mpr[i][j] = img->mprr_c[uv][mode][i][j];
}
}
/*!
************************************************************************
* \brief
* Predict one chroma 4x4 block
************************************************************************
*/
void
ChromaPrediction4x4 (int uv, // <-- colour component
int block_x, // <-- relative horizontal block coordinate of 4x4 block
int block_y, // <-- relative vertical block coordinate of 4x4 block
int fw_mode, // <-- forward prediction mode (1-7, 0=DIRECT if bw_mode=0)
int bw_mode, // <-- backward prediction mode (1-7, 0=DIRECT if fw_mode=0)
int fw_ref_frame, // <-- reference frame for forward prediction (if (<0) -> intra prediction)
int bw_ref_frame) // <-- reference frame for backward 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_pix_x = img->pix_c_x + block_x;
int pic_pix_y = img->pix_c_y + block_y;
int* fpred = fw_pred;
int* bpred = bw_pred;
int direct = (fw_mode == 0 && bw_mode == 0 && (img->type == B_SLICE));
//int skipped = (fw_mode == 0 && bw_mode == 0 && (img->type != B_SLICE && img->type != BS_IMG));
int skipped = (fw_mode == 0 && bw_mode == 0 && (img->type != B_SLICE));
int***** fmv_array = img->all_mv;
int***** bmv_array = img->all_bmv;
int fw_ref_idx, bw_ref_idx;
int apply_weights = ( (input->WeightedPrediction && (img->type == P_SLICE||img->type == SP_SLICE)) ||
(input->WeightedBiprediction && (img->type == B_SLICE)));
// (input->WeightedBiprediction && (img->type == B_SLICE || img->type == BS_IMG)));
if(input->InterlaceCodingOption >= MB_CODING && mb_adaptive && img->field_mode)
{
pic_pix_y = img->field_pix_c_y + block_y;
if(img->top_field)
{
fmv_array = img->all_mv_top;
bmv_array = img->all_bmv_top;
}
else
{
fmv_array = img->all_mv_bot;
bmv_array = img->all_bmv_bot;
}
}
direct_mode = direct && input->direct_type==0;
//===== INTRA PREDICTION =====
if (fw_ref_frame < 0)
{
IntraChromaPrediction4x4 (uv, block_x, block_y);
return;
}
if (input->direct_type && direct)
{
fw_ref_frame = fwdir_refFrArr[pic_pix_y/2][pic_pix_x/2];
bw_ref_frame = bwdir_refFrArr[pic_pix_y/2][pic_pix_x/2];
}
// if (img->type == BS_IMG)
if (img->type == B_SLICE && img->nal_reference_idc>0)
{
fw_ref_idx = fw_ref_frame;
bw_ref_idx = (bw_ref_frame < 2) ? 1-bw_ref_frame : bw_ref_frame;
}
else
{
fw_ref_idx = fw_ref_frame;
bw_ref_idx = bw_ref_frame;
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -