📄 image.c
字号:
/*!
************************************************************************
* \brief
* Upsample 4 times, store them in out4x. Color is simply copied
*
* \par Input:
* srcy, srcu, srcv, out4y, out4u, out4v
*
* \par Side Effects_
* Uses (writes) img4Y_tmp. This should be moved to a static variable
* in this module
************************************************************************/
void UnifiedOneForthPix (StorablePicture *s)
{
int is;
int i, j, j4;
int ie2, je2, jj, maxy;
imgpel **out4Y;
imgpel *ref11;
imgpel **imgY = s->imgY;
int img_width =s->size_x;
int img_height=s->size_y;
// don't upsample twice
if (s->imgY_ups || s->imgY_11)
return;
s->imgY_11 = malloc ((s->size_x * s->size_y) * sizeof (imgpel));
if (NULL == s->imgY_11)
no_mem_exit("alloc_storable_picture: s->imgY_11");
get_mem2Dnew (&(s->imgY_ups), (2*IMG_PAD_SIZE + s->size_y)*4, (2*IMG_PAD_SIZE + s->size_x)*4, sizeof(imgpel));
if (input->WeightedPrediction || input->WeightedBiprediction)
{
s->imgY_11_w = malloc ((s->size_x * s->size_y) * sizeof (imgpel));
get_mem2Dnew (&(s->imgY_ups_w), (2*IMG_PAD_SIZE + s->size_y)*4, (2*IMG_PAD_SIZE + s->size_x)*4, sizeof(imgpel));
}
out4Y = s->imgY_ups;
ref11 = s->imgY_11;
for (j = -IMG_PAD_SIZE; j < s->size_y + IMG_PAD_SIZE; j++)
{
for (i = -IMG_PAD_SIZE; i < s->size_x + IMG_PAD_SIZE; i++)
{
jj = max (0, min (s->size_y - 1, j));
is =
(ONE_FOURTH_TAP[0][0] *
(imgY[jj][max (0, min (s->size_x - 1, i))] +
imgY[jj][max (0, min (s->size_x - 1, i + 1))]) +
ONE_FOURTH_TAP[1][0] *
(imgY[jj][max (0, min (s->size_x - 1, i - 1))] +
imgY[jj][max (0, min (s->size_x - 1, i + 2))]) +
ONE_FOURTH_TAP[2][0] *
(imgY[jj][max (0, min (s->size_x - 1, i - 2))] +
imgY[jj][max (0, min (s->size_x - 1, i + 3))]));
img4Y_tmp[j + IMG_PAD_SIZE][(i + IMG_PAD_SIZE) * 2] = imgY[jj][max (0, min (s->size_x - 1, i))] * 1024; // 1/1 pix pos
img4Y_tmp[j + IMG_PAD_SIZE][(i + IMG_PAD_SIZE) * 2 + 1] = is * 32; // 1/2 pix pos
}
}
for (i = 0; i < (s->size_x + 2 * IMG_PAD_SIZE) * 2; i++)
{
for (j = 0; j < s->size_y + 2 * IMG_PAD_SIZE; j++)
{
j4 = j * 4;
maxy = s->size_y + 2 * IMG_PAD_SIZE - 1;
// change for TML4, use 6 TAP vertical filter
is =
(ONE_FOURTH_TAP[0][0] *
(img4Y_tmp[j][i] + img4Y_tmp[min (maxy, j + 1)][i]) +
ONE_FOURTH_TAP[1][0] * (img4Y_tmp[max (0, j - 1)][i] +
img4Y_tmp[min (maxy, j + 2)][i]) +
ONE_FOURTH_TAP[2][0] * (img4Y_tmp[max (0, j - 2)][i] +
img4Y_tmp[min (maxy, j + 3)][i])) / 32;
PutPel_14 (out4Y, (j - IMG_PAD_SIZE) * 4, (i - IMG_PAD_SIZE * 2) * 2, (pel_t) max (0, min(img->max_imgpel_value , (int) ((img4Y_tmp[j][i] + 512) / 1024)))); // 1/2 pix
PutPel_14 (out4Y, (j - IMG_PAD_SIZE) * 4 + 2, (i - IMG_PAD_SIZE * 2) * 2, (pel_t) max (0, min(img->max_imgpel_value, (int) ((is + 512) / 1024)))); // 1/2 pix
}
}
/* 1/4 pix */
/* luma */
ie2 = (s->size_x + 2 * IMG_PAD_SIZE - 1) * 4;
je2 = (s->size_y + 2 * IMG_PAD_SIZE - 1) * 4;
for (j = 0; j < je2 + 4; j += 2)
for (i = 0; i < ie2 + 3; i += 2)
{
/* '-' */
PutPel_14 (out4Y, j - IMG_PAD_SIZE * 4, i - IMG_PAD_SIZE * 4 + 1,
(pel_t) (max (0, min (img->max_imgpel_value,(int) (FastPelY_14 (out4Y, j - IMG_PAD_SIZE * 4,
i - IMG_PAD_SIZE * 4, img_height, img_width) + FastPelY_14 (out4Y,
j - IMG_PAD_SIZE * 4, min (ie2 + 2, i + 2) - IMG_PAD_SIZE * 4, img_height, img_width)+1) / 2))));
}
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 (img->max_imgpel_value, (int) (FastPelY_14 (out4Y, j - IMG_PAD_SIZE * 4,
i - IMG_PAD_SIZE * 4, img_height, img_width) + FastPelY_14 (out4Y,
min (je2 + 2, j + 2) - IMG_PAD_SIZE * 4, i - IMG_PAD_SIZE * 4, img_height, img_width)+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 (img->max_imgpel_value, (int) (FastPelY_14 (out4Y, j - IMG_PAD_SIZE * 4,
min (ie2 + 2, i + 1) - IMG_PAD_SIZE * 4, img_height, img_width) + FastPelY_14 (out4Y,
min (je2 + 2, j + 2) - IMG_PAD_SIZE * 4, i - IMG_PAD_SIZE * 4 - 1, img_height, img_width) + 1) / 2))));
}
else
{
/* '\' */
PutPel_14 (out4Y, j - IMG_PAD_SIZE * 4 + 1, i - IMG_PAD_SIZE * 4,
(pel_t) (max (0, min (img->max_imgpel_value, (int) (FastPelY_14 (out4Y, j - IMG_PAD_SIZE * 4,
i - IMG_PAD_SIZE * 4 - 1, img_height, img_width) + FastPelY_14 (out4Y,
min (je2 + 2, j + 2) - IMG_PAD_SIZE * 4,
min (ie2 + 2, i + 1) - IMG_PAD_SIZE * 4, img_height, img_width) + 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, s->size_x, s->size_y);
}
/*!
************************************************************************
* \brief
* Find SNR for all three components
************************************************************************
*/
static void find_snr ()
{
int i, j;
int64 diff_y, diff_u, diff_v;
int impix;
int impix_cr;
unsigned int max_pix_value_sqd = img->max_imgpel_value * img->max_imgpel_value;
unsigned int max_pix_value_sqd_uv = img->max_imgpel_value_uv * img->max_imgpel_value_uv; // WSKIM SAMSUNG AIT
// Calculate PSNR for Y, U and V.
// Luma.
impix = img->height * img->width;
impix_cr = img->height_cr * img->width_cr;
if (img->fld_flag != 0)
{
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
{
imgY_org = imgY_org_frm;
imgUV_org = imgUV_org_frm;
if(input->PicInterlace==ADAPTIVE_CODING)
{
enc_picture = enc_frame_picture;
}
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]];
}
}
}
#if ZEROSNR
if (diff_y == 0)
diff_y = 1;
if (diff_u == 0)
diff_u = 1;
if (diff_v == 0)
diff_v = 1;
#endif
// Collecting SNR statistics
if (diff_y != 0)
{
snr->snr_y = (float) (10 * log10 (max_pix_value_sqd * (double)((double) impix / diff_y))); // luma snr for current frame
snr->snr_u = (float) (10 * log10 (max_pix_value_sqd_uv * (double)((double) impix_cr / diff_u))); // u croma snr for current frame, 1/4 of luma samples
snr->snr_v = (float) (10 * log10 (max_pix_value_sqd_uv * (double)((double) impix_cr / diff_v))); // v croma snr for current frame, 1/4 of luma samples
}
if (img->number == 0)
{
snr->snr_y1 = snr->snr_y; // keep luma snr for first frame // S.SUN 06072004
snr->snr_u1 = snr->snr_u; // keep croma u snr for first frame
snr->snr_v1 = snr->snr_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;
int64 diff_y, diff_u, diff_v;
int impix;
// Calculate PSNR for Y, U and V.
// Luma.
impix = img->height * img->width;
if (img->structure!=FRAME)
{
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
{
imgY_org = imgY_org_frm;
imgUV_org = imgUV_org_frm;
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;
}
/*!
************************************************************************
* \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 bframe = (img->type == B_SLICE);
int mode;
int b8mode, b8pdir;
int list_offset = ((img->MbaffFrameFlag)&&(currMB->mb_field))? img->current_mb_nr%2 ? 4 : 2 : 0;
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->prev_qp=rdopt->prev_qp;
currMB->prev_delta_qp=rdopt->prev_delta_qp;
currMB->qp=rdopt->qp;
currMB->c_ipred_mode = rdopt->c_ipred_mode;
for (i = 0; i < 4+img->num_blk8x8_uv; i++)
for (j = 0; j < 4; j++)
for (k = 0; k < 2; k++)
for (l = 0; l < 65; 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++)
{
enc_picture->ref_idx[LIST_0][img->block_x + i][img->block_y + j] = rdopt->refar[LIST_0][j][i];
enc_picture->ref_pic_id [LIST_0][img->block_x+i][img->block_y+j] = enc_picture->ref_pic_num[LIST_0 + list_offset][enc_picture->ref_idx[LIST_0][img->block_x+i][img->block_y+j]];
if (bframe)
{
enc_picture->ref_idx[LIST_1][img->block_x + i][img->block_y + j] = rdopt->refar[LIST_1][j][i];
enc_picture->ref_pic_id [LIST_1][img->block_x+i][img->block_y+j] = enc_picture->ref_pic_num[LIST_1 + list_offset][enc_picture->ref_idx[LIST_1][img->block_x+i][img->block_y+j]];
}
}
//===== 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 < img->mb_cr_size_y; j++)
for (i = 0; i < img->mb_cr_size_x; i++)
{
enc_picture->imgUV[0][img->pix_c_y + j][img->pix_c_x + i] = rdopt->rec_mbU[j][i];
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -