📄 image.c
字号:
diff_u += img->quad[imgUV_org[0][j][i] - enc_picture->imgUV[0][j][i]];
diff_v += img->quad[imgUV_org[1][j][i] - enc_picture->imgUV[1][j][i]];
}
}
// Collecting SNR statistics
if (diff_y != 0)
{
snr->snr_y = (float) (10 * log10 (65025 * (float) impix / (float) diff_y)); // luma snr for current frame
snr->snr_u = (float) (10 * log10 (65025 * (float) impix / (float) (4 * diff_u))); // u croma snr for current frame, 1/4 of luma samples
snr->snr_v = (float) (10 * log10 (65025 * (float) impix / (float) (4 * diff_v))); // v croma snr for current frame, 1/4 of luma samples
}
if (img->number == 0)
{
snr->snr_y1 = (float) (10 * log10 (65025 * (float) impix / (float) diff_y)); // keep luma snr for first frame
snr->snr_u1 = (float) (10 * log10 (65025 * (float) impix / (float) (4 * diff_u))); // keep croma u snr for first frame
snr->snr_v1 = (float) (10 * log10 (65025 * (float) impix / (float) (4 * diff_v))); // keep croma v snr for first frame
snr->snr_ya = snr->snr_y1;
snr->snr_ua = snr->snr_u1;
snr->snr_va = snr->snr_v1;
}
// B pictures
else
{
snr->snr_ya = (float) (snr->snr_ya * (img->number + Bframe_ctr) + snr->snr_y) / (img->number + Bframe_ctr + 1); // average snr lume for all frames inc. first
snr->snr_ua = (float) (snr->snr_ua * (img->number + Bframe_ctr) + snr->snr_u) / (img->number + Bframe_ctr + 1); // average snr u croma for all frames inc. first
snr->snr_va = (float) (snr->snr_va * (img->number + Bframe_ctr) + snr->snr_v) / (img->number + Bframe_ctr + 1); // average snr v croma for all frames inc. first
}
}
/*!
************************************************************************
* \brief
* Find distortion for all three components
************************************************************************
*/
static void find_distortion ()
{
int i, j;
int diff_y, diff_u, diff_v;
int impix;
// Calculate PSNR for Y, U and V.
// Luma.
impix = img->height * img->width;
diff_y = 0;
for (i = 0; i < img->width; ++i)
{
for (j = 0; j < img->height; ++j)
{
diff_y += img->quad[abs (imgY_org[j][i] - enc_picture->imgY[j][i])];
}
}
// Chroma.
diff_u = 0;
diff_v = 0;
for (i = 0; i < img->width_cr; i++)
{
for (j = 0; j < img->height_cr; j++)
{
diff_u += img->quad[abs (imgUV_org[0][j][i] - enc_picture->imgUV[0][j][i])];
diff_v += img->quad[abs (imgUV_org[1][j][i] - enc_picture->imgUV[1][j][i])];
}
}
// Calculate real PSNR at find_snr_avg()
snr->snr_y = (float) diff_y;
snr->snr_u = (float) diff_u;
snr->snr_v = (float) diff_v;
}
static void store_field_MV (int frame_number)
{
int i, j;
if (img->type != B_SLICE) //all I- and P-frames
{
if (img->fld_flag)
{
for (i = 0; i < img->width / 4 + 4; i++)
{
for (j = 0; j < img->height / 8; j++)
{
tmp_mv_frm[0][2 * j][i] = tmp_mv_frm[0][2 * j + 1][i] =
tmp_mv_top[0][j][i];
tmp_mv_frm[0][2 * j][i] = tmp_mv_frm[0][2 * j + 1][i] = tmp_mv_top[0][j][i]; // ??
tmp_mv_frm[1][2 * j][i] = tmp_mv_frm[1][2 * j + 1][i] =
tmp_mv_top[1][j][i] * 2;
tmp_mv_frm[1][2 * j][i] = tmp_mv_frm[1][2 * j + 1][i] = tmp_mv_top[1][j][i] * 2; // ??
if (input->direct_type
&& (input->successive_Bframe != 0
|| input->StoredBPictures > 0)
&& (i < img->width / 4))
{
moving_block_frm[2 * j + 1][i] =
moving_block_frm[2 * j][i] =
((refFrArr_top[j][i] != 0)
|| (refFrArr_bot[j][i] != 0)
|| (abs (tmp_mv_top[0][j][i + 4]) >> 1)
|| (abs (tmp_mv_top[1][j][i + 4]) >> 1)
|| (abs (tmp_mv_bot[0][j][i + 4]) >> 1)
|| (abs (tmp_mv_bot[1][j][i + 4]) >> 1));
moving_block_top[j][i] = ((refFrArr_top[j][i] != 0)
||
(abs
(tmp_mv_top[0][j][i + 4])
>> 1)
||
(abs
(tmp_mv_top[1][j][i + 4])
>> 1));
moving_block_bot[j][i] = ((refFrArr_bot[j][i] != 0)
||
(abs
(tmp_mv_bot[0][j][i + 4])
>> 1)
||
(abs
(tmp_mv_bot[1][j][i + 4])
>> 1));
}
if ((i % 2 == 0) && (j % 2 == 0) && (i < img->width / 4))
{
if (refFrArr_top[j][i] == -1)
{
refFrArr_frm[2 * j][i] =
refFrArr_frm[2 * j + 1][i] = -1;
refFrArr_frm[2 * (j + 1)][i] =
refFrArr_frm[2 * (j + 1) + 1][i] = -1;
refFrArr_frm[2 * j][i + 1] =
refFrArr_frm[2 * j + 1][i + 1] = -1;
refFrArr_frm[2 * (j + 1)][i + 1] =
refFrArr_frm[2 * (j + 1) + 1][i + 1] = -1;
}
else
{
refFrArr_frm[2 * j][i] =
refFrArr_frm[2 * j + 1][i] =
(int) (refFrArr_top[j][i] / 2);
refFrArr_frm[2 * (j + 1)][i] =
refFrArr_frm[2 * (j + 1) + 1][i] =
(int) (refFrArr_top[j][i] / 2);
refFrArr_frm[2 * j][i + 1] =
refFrArr_frm[2 * j + 1][i + 1] =
(int) (refFrArr_top[j][i] / 2);
refFrArr_frm[2 * (j + 1)][i + 1] =
refFrArr_frm[2 * (j + 1) + 1][i + 1] =
(int) (refFrArr_top[j][i] / 2);
}
}
}
}
}
else
{
for (i = 0; i < img->width / 4 + 4; i++)
{
for (j = 0; j < img->height / 8; j++)
{
tmp_mv_top[0][j][i] = tmp_mv_bot[0][j][i] =
(int) (tmp_mv_frm[0][2 * j][i]);
tmp_mv_top[1][j][i] = tmp_mv_bot[1][j][i] =
(int) ((tmp_mv_frm[1][2 * j][i]) / 2);
if (input->direct_type
&& (input->successive_Bframe != 0
|| input->StoredBPictures > 0)
&& i < img->width / 4)
{
moving_block_top[j][i] = moving_block_bot[j][i] =
((refFrArr_frm[2 * j][i] != 0)
|| (refFrArr_frm[2 * j + 1][i] != 0)
|| (abs (tmp_mv_frm[0][2 * j][i + 4]) >> 1)
|| (abs (tmp_mv_frm[1][2 * j][i + 4]) >> 1)
|| (abs (tmp_mv_frm[0][2 * j + 1][i + 4]) >> 1)
|| (abs (tmp_mv_frm[1][2 * j + 1][i + 4]) >> 1));
moving_block_frm[2 * j][i] =
((refFrArr_frm[2 * j][i] != 0)
|| (abs (tmp_mv_frm[0][2 * j][i + 4]) >> 1)
|| (abs (tmp_mv_frm[1][2 * j][i + 4]) >> 1));
moving_block_frm[2 * j + 1][i] =
((refFrArr_frm[2 * j + 1][i] != 0)
|| (abs (tmp_mv_frm[0][2 * j + 1][i + 4]) >> 1)
|| (abs (tmp_mv_frm[1][2 * j + 1][i + 4]) >> 1));
}
if ((i % 2 == 0) && (j % 2 == 0) && (i < img->width / 4))
{
if (refFrArr_frm[2 * j][i] == -1)
{
refFrArr_top[j][i] = refFrArr_bot[j][i] = -1;
refFrArr_top[j + 1][i] = refFrArr_bot[j + 1][i] =
-1;
refFrArr_top[j][i + 1] = refFrArr_bot[j][i + 1] =
-1;
refFrArr_top[j + 1][i + 1] =
refFrArr_bot[j + 1][i + 1] = -1;
}
else
{
refFrArr_top[j][i] = refFrArr_bot[j][i] =
refFrArr_frm[2 * j][i] * 2;
refFrArr_top[j + 1][i] = refFrArr_bot[j + 1][i] =
refFrArr_frm[2 * j][i] * 2;
refFrArr_top[j][i + 1] = refFrArr_bot[j][i + 1] =
refFrArr_frm[2 * j][i] * 2;
refFrArr_top[j + 1][i + 1] =
refFrArr_bot[j + 1][i + 1] =
refFrArr_frm[2 * j][i] * 2;
}
}
}
}
}
}
}
static void store_direct_moving_flag (int frame_number)
{
int i, j;
if (img->type != B_SLICE) //all I- and P-frames
for (i = 0; i < img->width / 4; i++)
for (j = 0; j < img->height / 8; j++)
if (input->direct_type
&& (input->successive_Bframe != 0
|| input->StoredBPictures > 0))
{
moving_block_frm[2 * j][i] = ((refFrArr_frm[2 * j][i] != 0)
||
(abs (tmp_mv_frm[0][2 * j][i + 4])
>> 1)
||
(abs (tmp_mv_frm[1][2 * j][i + 4])
>> 1));
moving_block_frm[2 * j + 1][i] =
((refFrArr_frm[2 * j + 1][i] != 0)
|| (abs (tmp_mv_frm[0][2 * j + 1][i + 4]) >> 1)
|| (abs (tmp_mv_frm[1][2 * j + 1][i + 4]) >> 1));
}
}
/*!
************************************************************************
* \brief
* Just a placebo
************************************************************************
*/
Boolean dummy_slice_too_big (int bits_slice)
{
return FALSE;
}
/*!
***************************************************************************
// For MB level field/frame coding
***************************************************************************
*/
void copy_rdopt_data (int bot_block)
{
int mb_nr = img->current_mb_nr;
Macroblock *currMB = &img->mb_data[mb_nr];
int i, j, k, l;
int **frefar =
((img->type == B_SLICE || img->type == BS_IMG) ? fw_refFrArr : refFrArr);
int **brefar = bw_refFrArr;
int **frefar_fld;
int **brefar_fld = (bot_block ? bw_refFrArr_bot : bw_refFrArr_top);
int bframe = (img->type == B_SLICE || img->type == BS_IMG);
int mode;
int offset_x, offset_y;
int field_y;
int block_y = (bot_block ? (img->block_y - 4) / 2 : (img->block_y / 2));
frefar_fld = ((img->type == B_SLICE || img->type == BS_IMG) ?
(bot_block ? fw_refFrArr_bot : fw_refFrArr_top) :
(bot_block ? refFrArr_bot : refFrArr_top));
mode = rdopt->mode;
currMB->mb_type = rdopt->mb_type; // copy mb_type
currMB->cbp = rdopt->cbp; // copy cbp
currMB->cbp_blk = rdopt->cbp_blk; // copy cbp_blk
img->i16offset = rdopt->i16offset;
currMB->c_ipred_mode = rdopt->c_ipred_mode;
if (img->type != B_SLICE && img->type != BS_IMG)
field_mb[img->mb_y][img->mb_x] = MBPairIsField;
if (img->field_mode)
{
if (img->top_field && (img->type != B_SLICE && img->type != BS_IMG)
&& mode == 0)
TopFieldIsSkipped = 1; //set top field MB skipped to 1 for skipped top field MBs
else if (img->top_field && (img->type == B_SLICE || img->type == BS_IMG)
&& mode == 0)
TopFieldIsSkipped = currMB->cbp ? 0 : 1; // for direct mode check if cbp is zero (skipped) or not
else if (img->top_field) // don't change it for bottom field in case TopFieldIsSkipped=1
TopFieldIsSkipped = 0;
/* if(!bot_block && TopFieldIsSkipped)
fprintf(stderr,"Skipping Top Field %d\n", img->current_mb_nr);
else if(bot_block && mode == 0 && !TopFieldIsSkipped)
fprintf(stderr,"Skipping Bottom Field %d\n", img->current_mb_nr);
else if(bot_block && TopFieldIsSkipped && mode == 0)
fprintf(stderr,"Skipping Field Pair %d\n", img->current_mb_nr);
*/
}
else
{
if (!bot_block && (img->type != B_SLICE && img->type != BS_IMG)
&& mode == 0)
TopFrameIsSkipped = 1; //set top frame MB skipped to 1 for skipped top field MBs
else if (!bot_block && (img->type == B_SLICE || img->type == BS_IMG)
&& mode == 0)
TopFrameIsSkipped = currMB->cbp ? 0 : 1;
else if (!bot_block)
TopFrameIsSkipped = 0;
/* if(!bot_block && TopFrameIsSkipped)
fprintf(stderr,"Skipping Top Frame %d\n", img->current_mb_nr);
else if(bot_block && mode == 0 && !TopFrameIsSkipped)
fprintf(stderr,"Skipping Bottom Frame %d\n", img->current_mb_nr);
else if(bot_block && TopFrameIsSkipped && mode == 0)
fprintf(stderr,"Skipping Frame Pair %d\n", img->current_mb_nr);
*/
}
for (i = 0; i < 6; i++)
for (j = 0; j < 4; j++)
for (k = 0; k < 2; k++)
for (l = 0; l < 18; l++)
img->cofAC[i][j][k][l] = rdopt->cofAC[i][j][k][l];
for (i = 0; i < 3; i++)
for (k = 0; k < 2; k++)
for (l = 0; l < 18; l++)
img->cofDC[i][k][l] = rdopt->cofDC[i][k][l];
for (j = 0; j < 4; j++)
for (i = 0; i < 4; i++)
{
frefar[img->block_y + j][img->block_x + i] = rdopt->frefar[j][i]; // copy refFrArr
if (bframe)
brefar[img->block_y + j][img->block_x + i] = rdopt->brefar[j][i]; // copy bw_refFrArr
}
if (!MBPairIsField)
{
//===== reconstruction values =====
for (j = 0; j < 16; j++)
for (i = 0; i < 16; i++)
{
enc_picture->imgY[img->pix_y + j][img->pix_x + i] = rdopt->rec_mbY[j][i];
}
for (j = 0; j < 8; j++)
for (i = 0; i < 8; i++)
{
enc_picture->imgUV[0][img->pix_c_y + j][img->pix_c_x + i] = rdopt->rec_mbU[j][i];
enc_picture->imgUV[1][img->pix_c_y + j][img->pix_c_x + i] = rdopt->rec_mbV[j][i];
}
}
else
{
if (!bot_⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -