📄 image.c
字号:
for (i = 0; i < ie2 + 4; i++)
{
for (j = 0; j < je2 + 3; j += 2)
{
if (i % 2 == 0)
{
/* '|' */
PutPel_14 (out4Y, j - IMG_PAD_SIZE * 4 + 1,
i - IMG_PAD_SIZE * 4,
(pel_t) (max
(0,
min (255,
(int) (FastPelY_14
(out4Y, j - IMG_PAD_SIZE * 4,
i - IMG_PAD_SIZE * 4) +
FastPelY_14 (out4Y,
min (je2 + 2,
j + 2) -
IMG_PAD_SIZE *
4,
i -
IMG_PAD_SIZE *
4)+1) / 2))));
}
else if ((j % 4 == 0 && i % 4 == 1) || (j % 4 == 2 && i % 4 == 3))
{
/* '/' */
PutPel_14 (out4Y, j - IMG_PAD_SIZE * 4 + 1,
i - IMG_PAD_SIZE * 4,
(pel_t) (max
(0,
min (255,
(int) (FastPelY_14
(out4Y, j - IMG_PAD_SIZE * 4,
min (ie2 + 2,
i + 1) -
IMG_PAD_SIZE * 4) +
FastPelY_14 (out4Y,
min (je2 + 2,
j + 2) -
IMG_PAD_SIZE *
4,
i -
IMG_PAD_SIZE *
4 - 1) + 1) / 2))));
}
else
{
/* '\' */
PutPel_14 (out4Y, j - IMG_PAD_SIZE * 4 + 1,
i - IMG_PAD_SIZE * 4,
(pel_t) (max
(0,
min (255,
(int) (FastPelY_14
(out4Y, j - IMG_PAD_SIZE * 4,
i - IMG_PAD_SIZE * 4 - 1) +
FastPelY_14 (out4Y,
min (je2 + 2,
j + 2) -
IMG_PAD_SIZE *
4, min (ie2 + 2,
i + 1) -
IMG_PAD_SIZE *
4) + 1) / 2))));
}
}
}
/* Chroma: */
for (j = 0; j < img->height_cr; j++)
{
memcpy (outU[j], imgU[j], img->width_cr); // just copy 1/1 pix, interpolate "online"
memcpy (outV[j], imgV[j], img->width_cr);
}
// Generate 1/1th pel representation (used for integer pel MV search)
GenerateFullPelRepresentation (out4Y, ref11, img->width, img->height);
}
/*!
************************************************************************
* \brief
* Find SNR for all three components
************************************************************************
*/
static void find_snr ()
{
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[imgY_org[j][i] - 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] - imgUV[0][j][i]];
diff_v += img->quad[imgUV_org[1][j][i] - 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] - 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] - imgUV[0][j][i])];
diff_v += img->quad[abs (imgUV_org[1][j][i] - 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 rotate_buffer ()
{
Frame *f;
f = fb->picbuf_short[1];
fb->picbuf_short[1] = fb->picbuf_short[0];
fb->picbuf_short[0] = f;
mref[0] = fb->picbuf_short[0]->mref;
mcef[0] = fb->picbuf_short[0]->mcef;
mref[1] = fb->picbuf_short[1]->mref;
mcef[1] = fb->picbuf_short[1]->mcef;
if (input->WeightedPrediction || input->WeightedBiprediction)
{
mref_w[0] = fb->picbuf_short[0]->mref_w;
mref_w[1] = fb->picbuf_short[1]->mref_w;
}
}
static void store_field_MV (int frame_number)
{
int i, j;
if (img->type != B_IMG) //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_IMG) //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)
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -