📄 b_frame.c
字号:
// *************************************************************************************
// *************************************************************************************
// B_frame.c : B picture coding
//
// Main contributors (see contributors.h for copyright, address and affiliation details)
//
// Byeong-Moon Jeon <jeonbm@lge.com>
// Yoon-Seong Soh <yunsung@lge.com>
// Thomas Stockhammer <stockhammer@ei.tum.de>
// Detlev Marpe <marpe@hhi.de>
// Guido Heising <heising@hhi.de>
// Thomas Wedi <wedi@tnt.uni-hannover.de>
// *************************************************************************************
// *************************************************************************************
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <memory.h>
#include "global.h"
#include "elements.h"
#include "b_frame.h"
#include "refbuf.h"
#ifdef _ADAPT_LAST_GROUP_
extern int *last_P_no;
#endif
/************************************************************************
*
* Name : SetLoopfilterStrength_B()
*
* Description: Set the filter strength for a macroblock of a B-frame
*
************************************************************************/
void SetLoopfilterStrength_B()
{
int i,j;
int ii,jj;
int i3,j3,mvDiffX,mvDiffY;
if (img->imod == INTRA_MB_OLD || img->imod == INTRA_MB_NEW)
{
for (i=0;i<4;i++)
{
ii=img->block_x+i;
i3=ii/2;
for (j=0;j<4;j++)
{
jj=img->block_y+j;
j3=jj/2;
loopb[ii+1][jj+1]=3;
loopb[ii ][jj+1]=max(loopb[ii ][jj+1],2);
loopb[ii+1][jj ]=max(loopb[ii+1][jj ],2);
loopb[ii+2][jj+1]=max(loopb[ii+2][jj+1],2);
loopb[ii+1][jj+2]=max(loopb[ii+1][jj+2],2);
loopc[i3+1][j3+1]=2;
loopc[i3 ][j3+1]=max(loopc[i3 ][j3+1],1);
loopc[i3+1][j3 ]=max(loopc[i3+1][j3 ],1);
loopc[i3+2][j3+1]=max(loopc[i3+2][j3+1],1);
loopc[i3+1][j3+2]=max(loopc[i3+1][j3+2],1);
}
}
}
if (img->imod==B_Forward || img->imod==B_Bidirect)
{
for (i=0;i<4;i++)
{
ii=img->block_x+i;
i3=ii/2;
for (j=0;j<4;j++)
{
jj=img->block_y+j;
j3=jj/2;
mvDiffX = tmp_fwMV[0][jj][ii+4] - tmp_fwMV[0][jj][ii-1+4];
mvDiffY = tmp_fwMV[1][jj][ii+4] - tmp_fwMV[1][jj][ii-1+4];
if((mvDiffX*mvDiffX >= 16 || mvDiffY*mvDiffY >= 16) && ii > 0)
{
loopb[ii ][jj+1]=max(loopb[ii ][jj+1],1);
loopb[ii+1][jj+1]=max(loopb[ii+1][jj+1],1);
loopc[i3 ][j3+1]=max(loopc[i3 ][j3+1],1);
loopc[i3+1][j3+1]=max(loopc[i3+1][j3+1],1);
}
if (jj >0)
{
mvDiffX = tmp_fwMV[0][jj][ii+4] - tmp_fwMV[0][jj-1][ii+4];
mvDiffY = tmp_fwMV[1][jj][ii+4] - tmp_fwMV[1][jj-1][ii+4];
if((mvDiffX*mvDiffX >= 16 || mvDiffY*mvDiffY >= 16))
{
loopb[ii+1][jj ]=max(loopb[ii+1][jj ],1);
loopb[ii+1][jj+1]=max(loopb[ii+1][jj+1],1);
loopc[i3+1][j3 ]=max(loopc[i3+1][j3 ],1);
loopc[i3+1][j3+1]=max(loopc[i3+1][j3+1],1);
}
}
}
}
}
if(img->imod==B_Backward || img->imod==B_Bidirect)
{
for (i=0;i<4;i++)
{
ii=img->block_x+i;
i3=ii/2;
for (j=0;j<4;j++)
{
jj=img->block_y+j;
j3=jj/2;
mvDiffX = tmp_bwMV[0][jj][ii+4] - tmp_bwMV[0][jj][ii-1+4];
mvDiffY = tmp_bwMV[1][jj][ii+4] - tmp_bwMV[1][jj][ii-1+4];
if((mvDiffX*mvDiffX >= 16 || mvDiffY*mvDiffY >= 16) && ii > 0)
{
loopb[ii ][jj+1]=max(loopb[ii ][jj+1],1);
loopb[ii+1][jj+1]=max(loopb[ii+1][jj+1],1);
loopc[i3 ][j3+1]=max(loopc[i3 ][j3+1],1);
loopc[i3+1][j3+1]=max(loopc[i3+1][j3+1],1);
}
if (jj > 0)
{
mvDiffX = tmp_bwMV[0][jj][ii+4] - tmp_bwMV[0][jj-1][ii+4];
mvDiffY = tmp_bwMV[1][jj][ii+4] - tmp_bwMV[1][jj-1][ii+4];
if((mvDiffX*mvDiffX >= 16 || mvDiffY*mvDiffY >= 16) && jj > 0)
{
loopb[ii+1][jj ]=max(loopb[ii+1][jj ],1);
loopb[ii+1][jj+1]=max(loopb[ii+1][jj+1],1);
loopc[i3+1][j3 ]=max(loopc[i3+1][j3 ],1);
loopc[i3+1][j3+1]=max(loopc[i3+1][j3+1],1);
}
}
}
}
}
// LG : loopb, loopc at Direct mode
if(img->imod==B_Direct)
{
for (i=0;i<4;i++)
{
ii=img->block_x+i;
i3=ii/2;
for (j=0;j<4;j++)
{
jj=img->block_y+j;
j3=jj/2;
mvDiffX = dfMV[0][jj][ii+4] - dfMV[0][jj][ii-1+4];
mvDiffY = dfMV[1][jj][ii+4] - dfMV[1][jj][ii-1+4];
if((mvDiffX*mvDiffX >= 16 || mvDiffY*mvDiffY >= 16) && ii > 0)
{
loopb[ii ][jj+1]=max(loopb[ii ][jj+1],1);
loopb[ii+1][jj+1]=max(loopb[ii+1][jj+1],1);
loopc[i3 ][j3+1]=max(loopc[i3 ][j3+1],1);
loopc[i3+1][j3+1]=max(loopc[i3+1][j3+1],1);
}
if (jj > 0)
{
mvDiffX = dfMV[0][jj][ii+4] - dfMV[0][jj-1][ii+4];
mvDiffY = dfMV[1][jj][ii+4] - dfMV[1][jj-1][ii+4];
if((mvDiffX*mvDiffX >= 16 || mvDiffY*mvDiffY >= 16) && jj > 0)
{
loopb[ii+1][jj ]=max(loopb[ii+1][jj ],1);
loopb[ii+1][jj+1]=max(loopb[ii+1][jj+1],1);
loopc[i3+1][j3 ]=max(loopc[i3+1][j3 ],1);
loopc[i3+1][j3+1]=max(loopc[i3+1][j3+1],1);
}
}
}
}
for (i=0;i<4;i++)
{
ii=img->block_x+i;
i3=ii/2;
for (j=0;j<4;j++)
{
jj=img->block_y+j;
j3=jj/2;
mvDiffX = dbMV[0][jj][ii+4] - dbMV[0][jj][ii-1+4];
mvDiffY = dbMV[1][jj][ii+4] - dbMV[1][jj][ii-1+4];
if((mvDiffX*mvDiffX >= 16 || mvDiffY*mvDiffY >= 16) && ii > 0)
{
loopb[ii ][jj+1]=max(loopb[ii ][jj+1],1);
loopb[ii+1][jj+1]=max(loopb[ii+1][jj+1],1);
loopc[i3 ][j3+1]=max(loopc[i3 ][j3+1],1);
loopc[i3+1][j3+1]=max(loopc[i3+1][j3+1],1);
}
if (jj > 0)
{
mvDiffX = dbMV[0][jj][ii+4] - dbMV[0][jj-1][ii+4];
mvDiffY = dbMV[1][jj][ii+4] - dbMV[1][jj-1][ii+4];
if((mvDiffX*mvDiffX >= 16 || mvDiffY*mvDiffY >= 16) && jj > 0)
{
loopb[ii+1][jj ]=max(loopb[ii+1][jj ],1);
loopb[ii+1][jj+1]=max(loopb[ii+1][jj+1],1);
loopc[i3+1][j3 ]=max(loopc[i3+1][j3 ],1);
loopc[i3+1][j3+1]=max(loopc[i3+1][j3+1],1);
}
}
}
}
}
}
/************************************************************************
*
* Name : SetRefFrameInfo_B()
*
* Description: Set reference frame information in global arrays
* depending on mode decision. Used for motion vector prediction.
*
************************************************************************/
void SetRefFrameInfo_B()
{
int i,j;
const int fw_predframe_no = img->mb_data[img->current_mb_nr].ref_frame;
if(img->imod==B_Direct)
{
for (j = 0; j < 4;j++)
{
for (i = 0; i < 4;i++)
{
fw_refFrArr[img->block_y+j][img->block_x+i] =
bw_refFrArr[img->block_y+j][img->block_x+i] = -1;
}
}
}
else
if (img->imod == B_Forward)
{
for (j = 0;j < 4;j++)
{
for (i = 0;i < 4;i++)
{
fw_refFrArr[img->block_y+j][img->block_x+i] = fw_predframe_no;
bw_refFrArr[img->block_y+j][img->block_x+i] = -1;
}
}
}
else
if(img->imod == B_Backward)
{
for (j = 0;j < 4;j++)
{
for (i = 0;i < 4;i++)
{
fw_refFrArr[img->block_y+j][img->block_x+i] = -1;
bw_refFrArr[img->block_y+j][img->block_x+i] = 0;
}
}
}
else
if(img->imod == B_Bidirect)
{
for (j = 0;j < 4;j++)
{
for (i = 0;i < 4;i++)
{
fw_refFrArr[img->block_y+j][img->block_x+i] = fw_predframe_no;
bw_refFrArr[img->block_y+j][img->block_x+i] = 0;
}
}
}
else /* 4x4-, 16x16-intra */
{
for (j = 0;j < 4;j++)
{
for (i = 0;i < 4;i++)
{
fw_refFrArr[img->block_y+j][img->block_x+i] =
bw_refFrArr[img->block_y+j][img->block_x+i] = -1;
}
}
}
}
/************************************************************************
*
* Name : LumaResidualCoding_B()
*
* Description: Performs DCT, quantization, run/level pre-coding and IDCT
* for the MC-compensated MB residue of a B-frame;
* current cbp (for LUMA only) is affected
*
************************************************************************/
void LumaResidualCoding_B()
{
int cbp_mask, sum_cnt_nonz, coeff_cost, nonzero;
int mb_y, mb_x, block_y, block_x, i, j, pic_pix_y, pic_pix_x, pic_block_x, pic_block_y;
int ii4, jj4, iii4, jjj4, i2, j2, fw_pred, bw_pred, ref_inx, df_pred, db_pred;
Macroblock *currMB = &img->mb_data[img->current_mb_nr];
switch(img->imod)
{
case B_Forward :
currMB->cbp=0;
sum_cnt_nonz=0;
for (mb_y=0; mb_y < MB_BLOCK_SIZE; mb_y += BLOCK_SIZE*2)
{
for (mb_x=0; mb_x < MB_BLOCK_SIZE; mb_x += BLOCK_SIZE*2)
{
cbp_mask=(int)pow(2,(mb_x/8+mb_y/4));
coeff_cost=0;
for (block_y=mb_y; block_y < mb_y+BLOCK_SIZE*2; block_y += BLOCK_SIZE)
{
pic_pix_y=img->pix_y+block_y;
pic_block_y=pic_pix_y/BLOCK_SIZE;
for (block_x=mb_x; block_x < mb_x+BLOCK_SIZE*2; block_x += BLOCK_SIZE)
{
pic_pix_x=img->pix_x+block_x;
pic_block_x=pic_pix_x/BLOCK_SIZE;
img->ipredmode[pic_block_x+1][pic_block_y+1]=0;
if(input->mv_res)
{
ii4=(img->pix_x+block_x)*8+tmp_fwMV[0][pic_block_y][pic_block_x+4];
jj4=(img->pix_y+block_y)*8+tmp_fwMV[1][pic_block_y][pic_block_x+4];
for (j=0;j<4;j++) {
j2=j*8;
for (i=0;i<4;i++) {
i2=i*8;
#ifdef UMV
img->mpr[i+block_x][j+block_y]=UMVPelY_18 (mref[img->fw_multframe_no], jj4+j2, ii4+i2); // refbuf
#else
img->mpr[i+block_x][j+block_y]=FastPelY_18 (mref[img->fw_multframe_no], jj4+j2, ii4+i2); // refbuf
#endif
//img->mpr[i+block_x][j+block_y]=get_eigthpix_pel(ii4+i2,jj4+j2,img->fw_multframe_no);
}
}
}
else
{
ii4=(img->pix_x+block_x)*4+tmp_fwMV[0][pic_block_y][pic_block_x+4];
jj4=(img->pix_y+block_y)*4+tmp_fwMV[1][pic_block_y][pic_block_x+4];
for (j=0;j<4;j++) {
j2=j*4;
for (i=0;i<4;i++) {
i2=i*4;
#ifdef UMV
img->mpr[i+block_x][j+block_y]=UMVPelY_14 (mref[img->fw_multframe_no], jj4+j2, ii4+i2); // refbuf
#else
img->mpr[i+block_x][j+block_y]=FastPelY_14 (mref[img->fw_multframe_no], jj4+j2, ii4+i2); // refbuf
#endif
}
}
}
for (j=0; j < BLOCK_SIZE; j++)
{
for (i=0; i < BLOCK_SIZE; i++)
{
img->m7[i][j]=
imgY_org[img->pix_y+block_y+j][img->pix_x+block_x+i]-img->mpr[i+block_x][j+block_y];
}
}
nonzero=dct_luma(block_x,block_y,&coeff_cost);
if (nonzero)
{
currMB->cbp |= cbp_mask;
}
} // block_x
} // block_y
if (coeff_cost > 3) {
sum_cnt_nonz += coeff_cost;
}
else /*discard */
{
currMB->cbp &= (63-cbp_mask);
for (i=mb_x; i < mb_x+BLOCK_SIZE*2; i++)
{
for (j=mb_y; j < mb_y+BLOCK_SIZE*2; j++)
{
imgY[img->pix_y+j][img->pix_x+i]=img->mpr[i][j];
}
}
}
} // mb_x
} // mb_y
if (sum_cnt_nonz <= 5 )
{
currMB->cbp &= 48; /* mask bit 4 and 5 */
for (i=0; i < MB_BLOCK_SIZE; i++)
{
for (j=0; j < MB_BLOCK_SIZE; j++)
{
imgY[img->pix_y+j][img->pix_x+i]=img->mpr[i][j];
}
}
}
break;
case B_Backward :
currMB->cbp=0;
sum_cnt_nonz=0;
for (mb_y=0; mb_y < MB_BLOCK_SIZE; mb_y += BLOCK_SIZE*2)
{
for (mb_x=0; mb_x < MB_BLOCK_SIZE; mb_x += BLOCK_SIZE*2)
{
cbp_mask=(int)pow(2,(mb_x/8+mb_y/4));
coeff_cost=0;
for (block_y=mb_y; block_y < mb_y+BLOCK_SIZE*2; block_y += BLOCK_SIZE)
{
pic_pix_y=img->pix_y+block_y;
pic_block_y=pic_pix_y/BLOCK_SIZE;
for (block_x=mb_x; block_x < mb_x+BLOCK_SIZE*2; block_x += BLOCK_SIZE)
{
pic_pix_x=img->pix_x+block_x;
pic_block_x=pic_pix_x/BLOCK_SIZE;
img->ipredmode[pic_block_x+1][pic_block_y+1]=0;
if(input->mv_res)
{
iii4=(img->pix_x+block_x)*8+tmp_bwMV[0][pic_block_y][pic_block_x+4];
jjj4=(img->pix_y+block_y)*8+tmp_bwMV[1][pic_block_y][pic_block_x+4];
for (j=0;j<4;j++) {
j2=j*8;
for (i=0;i<4;i++) {
i2=i*8;
#ifdef UMV
img->mpr[i+block_x][j+block_y]=UMVPelY_18 (mref_P, jjj4+j2, iii4+i2); // refbuf
#else
img->mpr[i+block_x][j+block_y]=FastPelY_18 (mref_P, jjj4+j2, iii4+i2); // refbuf
#endif
//img->mpr[i+block_x][j+block_y]=get_eigthpix_pel_P(iii4+i2,jjj4+j2);
//img->mpr[i+block_x][j+block_y]=mref_P[jjj4+j2][iii4+i2];
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -