📄 image.c
字号:
impix = img->height * img->width;
//SW paff
// if (input->InterlaceCodingOption == FIELD_CODING)
if (input->InterlaceCodingOption != FRAME_CODING)
{
diff_y = 0;
for (i = 0; i < img->width; ++i)
{
for (j = 0; j < img->height; ++j)
{
diff_y += img->quad[imgY_org[j][i] - imgY_com[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[imgUV_org[0][j][i] - imgUV_com[0][j][i]];
diff_v += img->quad[imgUV_org[1][j][i] - imgUV_com[1][j][i]];
}
}
}else
{
diff_y = 0;
for (i = 0; i < img->width; ++i)
{
for (j = 0; j < img->height; ++j)
{
diff_y += img->quad[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[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;
//sw paff
//if (input->InterlaceCodingOption == FIELD_CODING)
if (input->InterlaceCodingOption != FRAME_CODING)
{
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] - imgY_com[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] - imgUV_com[0][j][i])];
diff_v += img->quad[abs (imgUV_org[1][j][i] - imgUV_com[1][j][i])];
}
}
}else
{
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) ? 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);
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) ?
(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)
field_mb[img->mb_y][img->mb_x] = MBPairIsField;
if (img->field_mode)
{
if (img->top_field && (img->type != B_SLICE)
&& 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)
&& mode == 0)⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -