📄 block.c
字号:
if (!active_pps->constrained_intra_pred_flag)
{
up_avail = up.available;
left_avail = left[1].available;
left_up_avail = left[0].available;
}
else
{
up_avail = up.available ? img->intra_block[up.mb_addr] : 0;
for (i=1, left_avail=1; i<17;i++)
left_avail &= left[i].available ? img->intra_block[left[i].mb_addr]: 0;
left_up_avail = left[0].available ? img->intra_block[left[0].mb_addr]: 0;
}
switch (predmode)
{
case VERT_PRED_16: // vertical prediction from block above
if (!up_avail)
error ("invalid 16x16 intra pred Mode VERT_PRED_16",500);
for(j=0;j<MB_BLOCK_SIZE;j++)
for(i=0;i<MB_BLOCK_SIZE;i++)
img->mpr[i][j]=imgY[up.pos_y][up.pos_x+i];// store predicted 16x16 block
break;
case HOR_PRED_16: // horisontal prediction from left block
if (!left_avail)
error ("invalid 16x16 intra pred Mode VERT_PRED_16",500);
for(j=0;j<MB_BLOCK_SIZE;j++)
for(i=0;i<MB_BLOCK_SIZE;i++)
img->mpr[i][j]=imgY[left[j+1].pos_y][left[j+1].pos_x]; // store predicted 16x16 block
break;
case DC_PRED_16: // DC prediction
s1=s2=0;
for (i=0; i < MB_BLOCK_SIZE; i++)
{
if (up_avail)
s1 += imgY[up.pos_y][up.pos_x+i]; // sum hor pix
if (left_avail)
s2 += imgY[left[i+1].pos_y][left[i+1].pos_x]; // sum vert pix
}
if (up_avail && left_avail)
s0=(s1+s2+16)>>5; // no edge
if (!up_avail && left_avail)
s0=(s2+8)>>4; // upper edge
if (up_avail && !left_avail)
s0=(s1+8)>>4; // left edge
if (!up_avail && !left_avail)
s0=img->dc_pred_value; // top left corner, nothing to predict from
for(i=0;i<MB_BLOCK_SIZE;i++)
for(j=0;j<MB_BLOCK_SIZE;j++)
{
img->mpr[i][j]=s0;
}
break;
case PLANE_16:// 16 bit integer plan pred
if (!up_avail || !left_up_avail || !left_avail)
error ("invalid 16x16 intra pred Mode PLANE_16",500);
ih=0;
iv=0;
for (i=1;i<9;i++)
{
if (i<8)
ih += i*(imgY[up.pos_y][up.pos_x+7+i] - imgY[up.pos_y][up.pos_x+7-i]);
else
ih += i*(imgY[up.pos_y][up.pos_x+7+i] - imgY[left[0].pos_y][left[0].pos_x]);
iv += i*(imgY[left[8+i].pos_y][left[8+i].pos_x] - imgY[left[8-i].pos_y][left[8-i].pos_x]);
}
ib=(5*ih+32)>>6;
ic=(5*iv+32)>>6;
iaa=16*(imgY[up.pos_y][up.pos_x+15]+imgY[left[16].pos_y][left[16].pos_x]);
for (j=0;j< MB_BLOCK_SIZE;j++)
{
for (i=0;i< MB_BLOCK_SIZE;i++)
{
img->mpr[i][j]=max(0,min((iaa+(i-7)*ib +(j-7)*ic + 16)>>5, img->max_imgpel_value));
}
}// store plane prediction
break;
default:
{ // indication of fault in bitstream,exit
printf("illegal 16x16 intra prediction mode input: %d\n",predmode);
return SEARCH_SYNC;
}
}
return DECODING_OK;
}
void intrapred_chroma(struct img_par *img, int uv)
{
int i,j, ii, jj, ioff, joff;
imgpel ***imgUV = dec_picture->imgUV;
int js[4][4];
int pred;
int ih, iv, ib, ic, iaa;
int b8, b4;
int yuv = dec_picture->chroma_format_idc - 1;
int blk_x, blk_y;
int block_pos[3][4][4]= //[yuv][b8][b4]
{
{ {0, 1, 2, 3},{0, 0, 0, 0},{0, 0, 0, 0},{0, 0, 0, 0}},
{ {0, 1, 2, 3},{2, 3, 2, 3},{0, 0, 0, 0},{0, 0, 0, 0}},
{ {0, 1, 2, 3},{1, 1, 3, 3},{2, 3, 2, 3},{3, 3, 3, 3}}
};
int s0, s1, s2, s3;
int mb_nr=img->current_mb_nr;
Macroblock *currMB = &img->mb_data[img->current_mb_nr];
PixelPos up; //!< pixel position p(0,-1)
PixelPos left[17]; //!< pixel positions p(-1, -1..16)
int up_avail, left_avail[2], left_up_avail;
int cr_MB_x = img->mb_cr_size_x;
int cr_MB_y = img->mb_cr_size_y;
for (i=0;i<cr_MB_y+1;i++)
{
getNeighbour(mb_nr, -1, i-1, 0, &left[i]);
}
getNeighbour(mb_nr, 0, -1, 0, &up);
if (!active_pps->constrained_intra_pred_flag)
{
up_avail = up.available;
left_avail[0] = left_avail[1] = left[1].available;
left_up_avail = left[0].available;
}
else
{
up_avail = up.available ? img->intra_block[up.mb_addr] : 0;
for (i=0, left_avail[0]=1; i<cr_MB_y/2;i++)
left_avail[0] &= left[i+1].available ? img->intra_block[left[i+1].mb_addr]: 0;
for (i=cr_MB_y/2, left_avail[1]=1; i<cr_MB_y;i++)
left_avail[1] &= left[i+1].available ? img->intra_block[left[i+1].mb_addr]: 0;
left_up_avail = left[0].available ? img->intra_block[left[0].mb_addr]: 0;
}
if (currMB->c_ipred_mode == DC_PRED_8)
{
// DC prediction
for(b8=0; b8<img->num_blk8x8_uv/2;b8++)
{
for (b4=0; b4<4; b4++)
{
blk_y = subblk_offset_y[yuv][b8][b4] + 1;
blk_x = subblk_offset_x[yuv][b8][b4];
s0=s1=s2=s3=0;
js[b8][b4]=img->dc_pred_value;
//===== get prediction value =====
switch (block_pos[yuv][b8][b4])
{
case 0: //===== TOP LEFT =====
if (up_avail) for (i=blk_x;i<(blk_x+4);i++) s0 += imgUV[uv][up.pos_y][up.pos_x + i];
if (left_avail[0]) for (i=blk_y;i<(blk_y+4);i++) s2 += imgUV[uv][left[i].pos_y][left[i].pos_x];
if (up_avail && left_avail[0]) js[b8][b4] = (s0+s2+4) >> 3;
else if (up_avail) js[b8][b4] = (s0 +2) >> 2;
else if (left_avail[0]) js[b8][b4] = (s2 +2) >> 2;
break;
case 1: //===== TOP RIGHT =====
if (up_avail) for (i=blk_x;i<(blk_x+4);i++) s1 += imgUV[uv][up.pos_y][up.pos_x + i];
else if (left_avail[0]) for (i=blk_y;i<(blk_y+4);i++) s2 += imgUV[uv][left[i].pos_y][left[i].pos_x];
if (up_avail) js[b8][b4] = (s1 +2) >> 2;
else if (left_avail[0]) js[b8][b4] = (s2 +2) >> 2;
break;
case 2: //===== BOTTOM LEFT =====
if (left_avail[1]) for (i=blk_y;i<(blk_y+4);i++) s3 += imgUV[uv][left[i].pos_y][left[i].pos_x];
else if (up_avail) for (i=blk_x;i<(blk_x+4);i++) s0 += imgUV[uv][up.pos_y][up.pos_x + i];
if (left_avail[1]) js[b8][b4] = (s3 +2) >> 2;
else if (up_avail) js[b8][b4] = (s0 +2) >> 2;
break;
case 3: //===== BOTTOM RIGHT =====
if (up_avail) for (i=blk_x;i<(blk_x+4);i++) s1 += imgUV[uv][up.pos_y][up.pos_x + i];
if (left_avail[1]) for (i=blk_y;i<(blk_y+4);i++) s3 += imgUV[uv][left[i].pos_y][left[i].pos_x];
if (up_avail && left_avail[1]) js[b8][b4] = (s1+s3+4) >> 3;
else if (up_avail) js[b8][b4] = (s1 +2) >> 2;
else if (left_avail[1]) js[b8][b4] = (s3 +2) >> 2;
break;
}
}
}
}
if (PLANE_8 == currMB->c_ipred_mode)
{
// plane prediction
if (!left_up_avail || !left_avail[0] || !left_avail[1] || !up_avail)
error("unexpected PLANE_8 chroma intra prediction mode",-1);
ih = cr_MB_x/2*(imgUV[uv][up.pos_y][up.pos_x+cr_MB_x-1] - imgUV[uv][left[0].pos_y][left[0].pos_x]);
for (i=0;i<cr_MB_x/2-1;i++)
ih += (i+1)*(imgUV[uv][up.pos_y][up.pos_x+cr_MB_x/2 +i] -
imgUV[uv][up.pos_y][up.pos_x+cr_MB_x/2-2-i]);
iv = cr_MB_y/2*(imgUV[uv][left[cr_MB_y].pos_y][left[cr_MB_y].pos_x] - imgUV[uv][left[0].pos_y][left[0].pos_x]);
for (i=0;i<cr_MB_y/2-1;i++)
iv += (i+1)*(imgUV[uv][left[cr_MB_y/2+1+i].pos_y][left[cr_MB_y/2+1+i].pos_x] -
imgUV[uv][left[cr_MB_y/2-1-i].pos_y][left[cr_MB_y/2-1-i].pos_x]);
ib= ((cr_MB_x == 8?17:5)*ih+2*cr_MB_x)>>(cr_MB_x == 8?5:6);
ic= ((cr_MB_y == 8?17:5)*iv+2*cr_MB_y)>>(cr_MB_y == 8?5:6);
iaa=16*(imgUV[uv][left[cr_MB_y].pos_y][left[cr_MB_y].pos_x] +
imgUV[uv][up.pos_y][up.pos_x+cr_MB_x-1]);
for (j=0; j<cr_MB_y; j++)
for (i=0; i<cr_MB_x; i++)
img->mpr[i][j]=max(0,min(img->max_imgpel_value_uv,(iaa+(i-cr_MB_x/2+1)*ib+(j-cr_MB_y/2+1)*ic+16)>>5));
}
else
{
switch (currMB->c_ipred_mode)
{
case DC_PRED_8:
for (b8=0;b8<img->num_blk8x8_uv/2;b8++)
{
for (b4=0;b4<4;b4++)
{
joff = subblk_offset_y[yuv][b8][b4];
ioff = subblk_offset_x[yuv][b8][b4];
for (ii=0; ii<BLOCK_SIZE; ii++)
for (jj=0; jj<BLOCK_SIZE; jj++)
{
img->mpr[ii+ioff][jj+joff]=js[b8][b4];
}
}
}
break;
case HOR_PRED_8:
if (!left_avail[0] || !left_avail[1])
error("unexpected HOR_PRED_8 chroma intra prediction mode",-1);
for (j=0;j<2;j++)
{
joff=j*cr_MB_y/2;
for(i=0;i<2;i++)
{
ioff=i*cr_MB_x/2;
for (jj=0; jj<cr_MB_y/2; jj++)
{
pred = imgUV[uv][left[1+jj+joff].pos_y][left[1+jj+joff].pos_x];
for (ii=0; ii<cr_MB_x/2; ii++)
img->mpr[ii+ioff][jj+joff]=pred;
}
}
}
break;
case VERT_PRED_8:
if (!up_avail)
error("unexpected VERT_PRED_8 chroma intra prediction mode",-1);
for (j=0;j<2;j++)
{
joff=j*cr_MB_y/2;
for(i=0;i<2;i++)
{
ioff=i*cr_MB_x/2;
for (ii=0; ii<cr_MB_x/2; ii++)
{
pred = imgUV[uv][up.pos_y][up.pos_x+ii+ioff];
for (jj=0; jj<cr_MB_y/2; jj++)
img->mpr[ii+ioff][jj+joff]=pred;
}
}
}
break;
default:
error("illegal chroma intra prediction mode", 600);
break;
}
}
}
/*!
***********************************************************************
* \brief
* Inverse 4x4 transformation, transforms cof to m7
***********************************************************************
*/
void itrans(struct img_par *img, //!< image parameters
int ioff, //!< index to 4x4 block
int joff, //!<
int i0, //!<
int j0,
int chroma)
{
int i,j,i1,j1;
int m5[4];
int m6[4];
Boolean lossless_qpprime = ((img->qp + img->bitdepth_luma_qp_scale)==0 && img->lossless_qpprime_flag==1);
// Residue Color Transform
int residue_transform_flag = img->residue_transform_flag;
// horizontal
for (j=0;j<BLOCK_SIZE && !lossless_qpprime;j++)
{
for (i=0;i<BLOCK_SIZE;i++)
{
m5[i]=img->cof[i0][j0][i][j];
}
m6[0]=(m5[0]+m5[2]);
m6[1]=(m5[0]-m5[2]);
m6[2]=(m5[1]>>1)-m5[3];
m6[3]=m5[1]+(m5[3]>>1);
for (i=0;i<2;i++)
{
i1=3-i;
img->m7[i][j]=m6[i]+m6[i1];
img->m7[i1][j]=m6[i]-m6[i1];
}
}
// vertical
for (i=0;i<BLOCK_SIZE && !lossless_qpprime;i++)
{
for (j=0;j<BLOCK_SIZE;j++)
m5[j]=img->m7[i][j];
m6[0]=(m5[0]+m5[2]);
m6[1]=(m5[0]-m5[2]);
m6[2]=(m5[1]>>1)-m5[3];
m6[3]=m5[1]+(m5[3]>>1);
for (j=0;j<2;j++)
{
j1=3-j;
// Residue Color Transform
if(!residue_transform_flag)
{
if(!chroma)
{
img->m7[i][j] =max(0,min(img->max_imgpel_value,(m6[j]+m6[j1]+((long)img->mpr[i+ioff][j+joff] <<DQ_BITS)+DQ_ROUND)>>DQ_BITS));
img->m7[i][j1]=max(0,min(img->max_imgpel_value,(m6[j]-m6[j1]+((long)img->mpr[i+ioff][j1+joff]<<DQ_BITS)+DQ_ROUND)>>DQ_BITS));
} else
{
img->m7[i][j] =max(0,min(img->max_imgpel_value_uv,(m6[j]+m6[j1]+((long)img->mpr[i+ioff][j+joff] <<DQ_BITS)+DQ_ROUND)>>DQ_BITS));
img->m7[i][j1]=max(0,min(img->max_imgpel_value_uv,(m6[j]-m6[j1]+((long)img->mpr[i+ioff][j1+joff]<<DQ_BITS)+DQ_ROUND)>>DQ_BITS));
}
}
else{
img->m7[i][j] =(m6[j]+m6[j1]+DQ_ROUND)>>DQ_BITS;
img->m7[i][j1]=(m6[j]-m6[j1]+DQ_ROUND)>>DQ_BITS;
}
}
}
// Residue Color Transform
if(!residue_transform_flag)
{
for (i=0;i<BLOCK_SIZE && lossless_qpprime;i++)
for (j=0;j<BLOCK_SIZE;j++)
if(!chroma)
img->m7[i][j] = max(0,min(img->max_imgpel_value,img->cof[i0][j0][i][j]+(long)img->mpr[i+ioff][j+joff]));
else
img->m7[i][j] = max(0,min(img->max_imgpel_value_uv,img->cof[i0][j0][i][j]+(long)img->mpr[i+ioff][j+joff]));
}
else
{
for (i=0;i<BLOCK_SIZE && lossless_qpprime;i++)
for (j=0;j<BLOCK_SIZE;j++)
img->m7[i][j] = img->cof[i0][j0][i][j];
}
}
/*!
************************************************************************
* \brief
* For mapping the q-matrix to the active id and calculate quantisation values
*
* \param pps
* Picture parameter set
* \param sps
* Sequence parameter set
*
************************************************************************
*/
void AssignQuantParam(pic_parameter_set_rbsp_t* pps, seq_parameter_set_rbsp_t* sps)
{
int i;
if(!pps->pic_scaling_matrix_present_flag && !sps->seq_scaling_matrix_present_flag)
{
for(i=0; i<8; i++)
qmatrix[i] = (i<6) ? quant_org:quant8_org;
}
else
{
if(sps->seq_scaling_matrix_present_flag) // check sps first
{
for(i=0; i<8; i++)
{
if(i<6)
{
if(!sps->seq_scaling_list_present_flag[i]) // fall-back rule A
{
if((i==0) || (i==3))
qmatrix[i] = (i==0) ? quant_intra_default:quant_inter_default;
else
qmatrix[i] = qmatrix[i-1];
}
else
{
if(sps->UseDefaultScalingMatrix4x4Flag[i])
qmatrix[i] = (i<3) ? quant_intra_default:quant_inter_default;
else
qmatrix[i] = sps->ScalingList4x4[i];
}
}
else
{
if(!sps->seq_scaling_list_present_flag[i] || sps->UseDefaultScalingMatrix8x8Flag[i-6]) // fall-back rule A
qmatrix[i] = (i==6) ? quant8_intra_default:quant8_inter_default;
else
qmatrix[i] = sps->ScalingList8x8[i-6];
}
}
}
if(pps->pic_scaling_matrix_present_flag) // then check pps
{
for(i=0; i<8; i++)
{
if(i<6)
{
if(!pps->pic_scaling_list_present_flag[i]) // fall-back rule B
{
if((i==0) || (i==3))
{
if(!sps->seq_scaling_matrix_present_flag)
qmatrix[i] = (i==0) ? quant_intra_default:quant_inter_default;
}
else
qmatrix[i] = qmatrix[i-1];
}
else
{
if(pps->UseDefaultScalingMatrix4x4Flag[i])
qmatrix[i] = (i<3) ? quant_intra_default:quant_inter_default;
else
qmatrix[i] = pps->ScalingList4x4[i];
}
}
else
{
if(!pps->pic_scaling_list_present_flag[i]) // fall-back rule B
{
if(!sps->seq_scaling_matrix_present_flag)
qmatrix[i] = (i==6) ? quant8_intra_default:quant8_inter_default;
}
else if(pps->UseDefaultScalingMatrix8x8Flag[i-6])
qmatrix[i] = (i==6) ? quant8_intra_default:quant8_inter_default;
else
qmatrix[i] = pps->ScalingList8x8[i-6];
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -