diff --git a/vp10/encoder/encoder.c b/vp10/encoder/encoder.c index 639fb3f42279fbbc1559a0bfbb9292c1f654d2ce..efd17f112e1b01acc50dd2a5fd70f93889d2f4b2 100644 --- a/vp10/encoder/encoder.c +++ b/vp10/encoder/encoder.c @@ -3446,35 +3446,18 @@ static void loopfilter_frame(VP10_COMP *cpi, VP10_COMMON *cm) { // encoded #if CLPF_FILTER_ALL_PLANES vpx_yv12_copy_frame(cm->frame_to_show, &cpi->last_frame_uf); - before = - get_sse(cpi->Source->y_buffer, cpi->Source->y_stride, - cm->frame_to_show->y_buffer, cm->frame_to_show->y_stride, - cpi->Source->y_crop_width, cpi->Source->y_crop_height) + - get_sse(cpi->Source->u_buffer, cpi->Source->uv_stride, - cm->frame_to_show->u_buffer, cm->frame_to_show->uv_stride, - cpi->Source->uv_crop_width, cpi->Source->uv_crop_height) + - get_sse(cpi->Source->v_buffer, cpi->Source->uv_stride, - cm->frame_to_show->v_buffer, cm->frame_to_show->uv_stride, - cpi->Source->uv_crop_width, cpi->Source->uv_crop_height); + before = vpx_get_y_sse(cpi->Source, cm->frame_to_show) + + vpx_get_u_sse(cpi->Source, cm->frame_to_show) + + vpx_get_v_sse(cpi->Source, cm->frame_to_show); vp10_clpf_frame(cm->frame_to_show, cm, xd); - after = get_sse(cpi->Source->y_buffer, cpi->Source->y_stride, - cm->frame_to_show->y_buffer, cm->frame_to_show->y_stride, - cpi->Source->y_crop_width, cpi->Source->y_crop_height) + - get_sse(cpi->Source->u_buffer, cpi->Source->uv_stride, - cm->frame_to_show->u_buffer, cm->frame_to_show->uv_stride, - cpi->Source->uv_crop_width, cpi->Source->uv_crop_height) + - get_sse(cpi->Source->v_buffer, cpi->Source->uv_stride, - cm->frame_to_show->v_buffer, cm->frame_to_show->uv_stride, - cpi->Source->uv_crop_width, cpi->Source->uv_crop_height); + after = vpx_get_y_sse(cpi->Source, cm->frame_to_show) + + vpx_get_u_sse(cpi->Source, cm->frame_to_show) + + vpx_get_v_sse(cpi->Source, cm->frame_to_show); #else vpx_yv12_copy_y(cm->frame_to_show, &cpi->last_frame_uf); - before = get_sse(cpi->Source->y_buffer, cpi->Source->y_stride, - cm->frame_to_show->y_buffer, cm->frame_to_show->y_stride, - cpi->Source->y_crop_width, cpi->Source->y_crop_height); + before = vpx_get_y_sse(cpi->Source, cm->frame_to_show); vp10_clpf_frame(cm->frame_to_show, cm, xd); - after = get_sse(cpi->Source->y_buffer, cpi->Source->y_stride, - cm->frame_to_show->y_buffer, cm->frame_to_show->y_stride, - cpi->Source->y_crop_width, cpi->Source->y_crop_height); + after = vpx_get_y_sse(cpi->Source, cm->frame_to_show); #endif if (before < after) { // No improvement, restore original diff --git a/vpx_dsp/psnr.c b/vpx_dsp/psnr.c index 47afd4388abfba8d05e9295c1ee6ffd9d6219688..65997c70461b6b754fd250f2a5a0ee0bb0367bd6 100644 --- a/vpx_dsp/psnr.c +++ b/vpx_dsp/psnr.c @@ -184,6 +184,24 @@ int64_t vpx_get_y_sse(const YV12_BUFFER_CONFIG *a, a->y_crop_width, a->y_crop_height); } +int64_t vpx_get_u_sse(const YV12_BUFFER_CONFIG *a, + const YV12_BUFFER_CONFIG *b) { + assert(a->uv_crop_width == b->uv_crop_width); + assert(a->uv_crop_height == b->uv_crop_height); + + return get_sse(a->u_buffer, a->uv_stride, b->u_buffer, b->uv_stride, + a->uv_crop_width, a->uv_crop_height); +} + +int64_t vpx_get_v_sse(const YV12_BUFFER_CONFIG *a, + const YV12_BUFFER_CONFIG *b) { + assert(a->uv_crop_width == b->uv_crop_width); + assert(a->uv_crop_height == b->uv_crop_height); + + return get_sse(a->v_buffer, a->uv_stride, b->v_buffer, b->uv_stride, + a->uv_crop_width, a->uv_crop_height); +} + #if CONFIG_VP9_HIGHBITDEPTH int64_t vpx_highbd_get_y_sse(const YV12_BUFFER_CONFIG *a, const YV12_BUFFER_CONFIG *b) { @@ -195,6 +213,28 @@ int64_t vpx_highbd_get_y_sse(const YV12_BUFFER_CONFIG *a, return highbd_get_sse(a->y_buffer, a->y_stride, b->y_buffer, b->y_stride, a->y_crop_width, a->y_crop_height); } + +int64_t vpx_highbd_get_u_sse(const YV12_BUFFER_CONFIG *a, + const YV12_BUFFER_CONFIG *b) { + assert(a->uv_crop_width == b->uv_crop_width); + assert(a->uv_crop_height == b->uv_crop_height); + assert((a->flags & YV12_FLAG_HIGHBITDEPTH) != 0); + assert((b->flags & YV12_FLAG_HIGHBITDEPTH) != 0); + + return highbd_get_sse(a->u_buffer, a->uv_stride, b->u_buffer, b->uv_stride, + a->uv_crop_width, a->uv_crop_height); +} + +int64_t vpx_highbd_get_v_sse(const YV12_BUFFER_CONFIG *a, + const YV12_BUFFER_CONFIG *b) { + assert(a->uv_crop_width == b->uv_crop_width); + assert(a->uv_crop_height == b->uv_crop_height); + assert((a->flags & YV12_FLAG_HIGHBITDEPTH) != 0); + assert((b->flags & YV12_FLAG_HIGHBITDEPTH) != 0); + + return highbd_get_sse(a->v_buffer, a->uv_stride, b->v_buffer, b->uv_stride, + a->uv_crop_width, a->uv_crop_height); +} #endif // CONFIG_VP9_HIGHBITDEPTH #if CONFIG_VP9_HIGHBITDEPTH diff --git a/vpx_dsp/psnr.h b/vpx_dsp/psnr.h index f321131d0b9f740b9d1b85396c5828ff57d8252b..fe977827b845767333b6991ea8b6daa13f403bb8 100644 --- a/vpx_dsp/psnr.h +++ b/vpx_dsp/psnr.h @@ -37,9 +37,15 @@ typedef struct { */ double vpx_sse_to_psnr(double samples, double peak, double sse); int64_t vpx_get_y_sse(const YV12_BUFFER_CONFIG *a, const YV12_BUFFER_CONFIG *b); +int64_t vpx_get_u_sse(const YV12_BUFFER_CONFIG *a, const YV12_BUFFER_CONFIG *b); +int64_t vpx_get_v_sse(const YV12_BUFFER_CONFIG *a, const YV12_BUFFER_CONFIG *b); #if CONFIG_VP9_HIGHBITDEPTH int64_t vpx_highbd_get_y_sse(const YV12_BUFFER_CONFIG *a, const YV12_BUFFER_CONFIG *b); +int64_t vpx_highbd_get_u_sse(const YV12_BUFFER_CONFIG *a, + const YV12_BUFFER_CONFIG *b); +int64_t vpx_highbd_get_v_sse(const YV12_BUFFER_CONFIG *a, + const YV12_BUFFER_CONFIG *b); void vpx_calc_highbd_psnr(const YV12_BUFFER_CONFIG *a, const YV12_BUFFER_CONFIG *b, PSNR_STATS *psnr, unsigned int bit_depth, unsigned int in_bit_depth);