| /* |
| * Copyright (c) 2016, Alliance for Open Media. All rights reserved. |
| * |
| * This source code is subject to the terms of the BSD 2 Clause License and |
| * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
| * was not distributed with this source code in the LICENSE file, you can |
| * obtain it at www.aomedia.org/license/software. If the Alliance for Open |
| * Media Patent License 1.0 was not distributed with this source code in the |
| * PATENTS file, you can obtain it at www.aomedia.org/license/patent. |
| */ |
| |
| #include <assert.h> |
| #include <math.h> |
| |
| #include "config/aom_config.h" |
| #include "config/aom_dsp_rtcd.h" |
| |
| #include "aom_dsp/psnr.h" |
| #include "aom_scale/yv12config.h" |
| |
| #if CONFIG_INTERNAL_STATS |
| #define STATIC |
| #else |
| #define STATIC static |
| #endif // CONFIG_INTERNAL_STATS |
| |
| STATIC double aom_sse_to_psnr(double samples, double peak, double sse) { |
| if (sse > 0.0) { |
| const double psnr = 10.0 * log10(samples * peak * peak / sse); |
| return psnr > MAX_PSNR ? MAX_PSNR : psnr; |
| } else { |
| return MAX_PSNR; |
| } |
| } |
| |
| #undef STATIC |
| |
| static int64_t encoder_sse(const uint8_t *a, int a_stride, const uint8_t *b, |
| int b_stride, int w, int h) { |
| int i, j; |
| int64_t sse = 0; |
| |
| for (i = 0; i < h; i++) { |
| for (j = 0; j < w; j++) { |
| const int diff = a[j] - b[j]; |
| sse += diff * diff; |
| } |
| |
| a += a_stride; |
| b += b_stride; |
| } |
| return sse; |
| } |
| |
| #if CONFIG_AV1_HIGHBITDEPTH |
| static int64_t encoder_highbd_sse(const uint8_t *a8, int a_stride, |
| const uint8_t *b8, int b_stride, int w, |
| int h) { |
| const uint16_t *a = CONVERT_TO_SHORTPTR(a8); |
| const uint16_t *b = CONVERT_TO_SHORTPTR(b8); |
| int64_t sse = 0; |
| for (int i = 0; i < h; ++i) { |
| for (int j = 0; j < w; ++j) { |
| const int diff = a[j] - b[j]; |
| sse += diff * diff; |
| } |
| a += a_stride; |
| b += b_stride; |
| } |
| return sse; |
| } |
| |
| #endif // CONFIG_AV1_HIGHBITDEPTH |
| |
| static int64_t get_sse(const uint8_t *a, int a_stride, const uint8_t *b, |
| int b_stride, int width, int height) { |
| const int dw = width % 16; |
| const int dh = height % 16; |
| int64_t total_sse = 0; |
| int x, y; |
| |
| if (dw > 0) { |
| total_sse += encoder_sse(&a[width - dw], a_stride, &b[width - dw], b_stride, |
| dw, height); |
| } |
| |
| if (dh > 0) { |
| total_sse += |
| encoder_sse(&a[(height - dh) * a_stride], a_stride, |
| &b[(height - dh) * b_stride], b_stride, width - dw, dh); |
| } |
| |
| for (y = 0; y < height / 16; ++y) { |
| const uint8_t *pa = a; |
| const uint8_t *pb = b; |
| for (x = 0; x < width / 16; ++x) { |
| total_sse += aom_sse(pa, a_stride, pb, b_stride, 16, 16); |
| |
| pa += 16; |
| pb += 16; |
| } |
| |
| a += 16 * a_stride; |
| b += 16 * b_stride; |
| } |
| |
| return total_sse; |
| } |
| |
| #if CONFIG_AV1_HIGHBITDEPTH |
| static int64_t highbd_get_sse_shift(const uint8_t *a8, int a_stride, |
| const uint8_t *b8, int b_stride, int width, |
| int height, unsigned int input_shift) { |
| const uint16_t *a = CONVERT_TO_SHORTPTR(a8); |
| const uint16_t *b = CONVERT_TO_SHORTPTR(b8); |
| int64_t total_sse = 0; |
| int x, y; |
| for (y = 0; y < height; ++y) { |
| for (x = 0; x < width; ++x) { |
| int64_t diff; |
| diff = (a[x] >> input_shift) - (b[x] >> input_shift); |
| total_sse += diff * diff; |
| } |
| a += a_stride; |
| b += b_stride; |
| } |
| return total_sse; |
| } |
| |
| static int64_t highbd_get_sse(const uint8_t *a, int a_stride, const uint8_t *b, |
| int b_stride, int width, int height) { |
| int64_t total_sse = 0; |
| int x, y; |
| const int dw = width % 16; |
| const int dh = height % 16; |
| |
| if (dw > 0) { |
| total_sse += encoder_highbd_sse(&a[width - dw], a_stride, &b[width - dw], |
| b_stride, dw, height); |
| } |
| if (dh > 0) { |
| total_sse += encoder_highbd_sse(&a[(height - dh) * a_stride], a_stride, |
| &b[(height - dh) * b_stride], b_stride, |
| width - dw, dh); |
| } |
| |
| for (y = 0; y < height / 16; ++y) { |
| const uint8_t *pa = a; |
| const uint8_t *pb = b; |
| for (x = 0; x < width / 16; ++x) { |
| total_sse += aom_highbd_sse(pa, a_stride, pb, b_stride, 16, 16); |
| pa += 16; |
| pb += 16; |
| } |
| a += 16 * a_stride; |
| b += 16 * b_stride; |
| } |
| return total_sse; |
| } |
| #endif // CONFIG_AV1_HIGHBITDEPTH |
| |
| uint64_t aom_get_y_var(const YV12_BUFFER_CONFIG *a, int hstart, int width, |
| int vstart, int height) { |
| return aom_var_2d_u8(a->y_buffer + vstart * a->y_stride + hstart, a->y_stride, |
| width, height) / |
| (width * height); |
| } |
| |
| uint64_t aom_get_u_var(const YV12_BUFFER_CONFIG *a, int hstart, int width, |
| int vstart, int height) { |
| return aom_var_2d_u8(a->u_buffer + vstart * a->uv_stride + hstart, |
| a->uv_stride, width, height) / |
| (width * height); |
| } |
| |
| uint64_t aom_get_v_var(const YV12_BUFFER_CONFIG *a, int hstart, int width, |
| int vstart, int height) { |
| return aom_var_2d_u8(a->v_buffer + vstart * a->uv_stride + hstart, |
| a->uv_stride, width, height) / |
| (width * height); |
| } |
| |
| int64_t aom_get_y_sse_part(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b, int hstart, int width, |
| int vstart, int height) { |
| return get_sse(a->y_buffer + vstart * a->y_stride + hstart, a->y_stride, |
| b->y_buffer + vstart * b->y_stride + hstart, b->y_stride, |
| width, height); |
| } |
| |
| int64_t aom_get_y_sse(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b) { |
| assert(a->y_crop_width == b->y_crop_width); |
| assert(a->y_crop_height == b->y_crop_height); |
| |
| return get_sse(a->y_buffer, a->y_stride, b->y_buffer, b->y_stride, |
| a->y_crop_width, a->y_crop_height); |
| } |
| |
| int64_t aom_get_u_sse_part(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b, int hstart, int width, |
| int vstart, int height) { |
| return get_sse(a->u_buffer + vstart * a->uv_stride + hstart, a->uv_stride, |
| b->u_buffer + vstart * b->uv_stride + hstart, b->uv_stride, |
| width, height); |
| } |
| |
| int64_t aom_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 aom_get_v_sse_part(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b, int hstart, int width, |
| int vstart, int height) { |
| return get_sse(a->v_buffer + vstart * a->uv_stride + hstart, a->uv_stride, |
| b->v_buffer + vstart * b->uv_stride + hstart, b->uv_stride, |
| width, height); |
| } |
| |
| int64_t aom_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_AV1_HIGHBITDEPTH |
| uint64_t aom_highbd_get_y_var(const YV12_BUFFER_CONFIG *a, int hstart, |
| int width, int vstart, int height) { |
| return aom_var_2d_u16(a->y_buffer + vstart * a->y_stride + hstart, |
| a->y_stride, width, height) / |
| (width * height); |
| } |
| |
| uint64_t aom_highbd_get_u_var(const YV12_BUFFER_CONFIG *a, int hstart, |
| int width, int vstart, int height) { |
| return aom_var_2d_u16(a->u_buffer + vstart * a->uv_stride + hstart, |
| a->uv_stride, width, height) / |
| (width * height); |
| } |
| |
| uint64_t aom_highbd_get_v_var(const YV12_BUFFER_CONFIG *a, int hstart, |
| int width, int vstart, int height) { |
| return aom_var_2d_u16(a->v_buffer + vstart * a->uv_stride + hstart, |
| a->uv_stride, width, height) / |
| (width * height); |
| } |
| |
| int64_t aom_highbd_get_y_sse_part(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b, int hstart, |
| int width, int vstart, int height) { |
| return highbd_get_sse( |
| a->y_buffer + vstart * a->y_stride + hstart, a->y_stride, |
| b->y_buffer + vstart * b->y_stride + hstart, b->y_stride, width, height); |
| } |
| |
| int64_t aom_highbd_get_y_sse(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b) { |
| assert(a->y_crop_width == b->y_crop_width); |
| assert(a->y_crop_height == b->y_crop_height); |
| assert((a->flags & YV12_FLAG_HIGHBITDEPTH) != 0); |
| assert((b->flags & YV12_FLAG_HIGHBITDEPTH) != 0); |
| |
| 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 aom_highbd_get_u_sse_part(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b, int hstart, |
| int width, int vstart, int height) { |
| return highbd_get_sse(a->u_buffer + vstart * a->uv_stride + hstart, |
| a->uv_stride, |
| b->u_buffer + vstart * b->uv_stride + hstart, |
| b->uv_stride, width, height); |
| } |
| |
| int64_t aom_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 aom_highbd_get_v_sse_part(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b, int hstart, |
| int width, int vstart, int height) { |
| return highbd_get_sse(a->v_buffer + vstart * a->uv_stride + hstart, |
| a->uv_stride, |
| b->v_buffer + vstart * b->uv_stride + hstart, |
| b->uv_stride, width, height); |
| } |
| |
| int64_t aom_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_AV1_HIGHBITDEPTH |
| |
| int64_t aom_get_sse_plane(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b, int plane, int highbd) { |
| #if CONFIG_AV1_HIGHBITDEPTH |
| if (highbd) { |
| switch (plane) { |
| case 0: return aom_highbd_get_y_sse(a, b); |
| case 1: return aom_highbd_get_u_sse(a, b); |
| case 2: return aom_highbd_get_v_sse(a, b); |
| default: assert(plane >= 0 && plane <= 2); return 0; |
| } |
| } else { |
| switch (plane) { |
| case 0: return aom_get_y_sse(a, b); |
| case 1: return aom_get_u_sse(a, b); |
| case 2: return aom_get_v_sse(a, b); |
| default: assert(plane >= 0 && plane <= 2); return 0; |
| } |
| } |
| #else |
| (void)highbd; |
| switch (plane) { |
| case 0: return aom_get_y_sse(a, b); |
| case 1: return aom_get_u_sse(a, b); |
| case 2: return aom_get_v_sse(a, b); |
| default: assert(plane >= 0 && plane <= 2); return 0; |
| } |
| #endif |
| } |
| |
| #if CONFIG_AV1_HIGHBITDEPTH |
| void aom_calc_highbd_psnr(const YV12_BUFFER_CONFIG *a, |
| const YV12_BUFFER_CONFIG *b, PSNR_STATS *psnr, |
| uint32_t bit_depth, uint32_t in_bit_depth) { |
| assert(a->y_crop_width == b->y_crop_width); |
| assert(a->y_crop_height == b->y_crop_height); |
| assert(a->uv_crop_width == b->uv_crop_width); |
| assert(a->uv_crop_height == b->uv_crop_height); |
| const int widths[3] = { a->y_crop_width, a->uv_crop_width, a->uv_crop_width }; |
| const int heights[3] = { a->y_crop_height, a->uv_crop_height, |
| a->uv_crop_height }; |
| const int a_strides[3] = { a->y_stride, a->uv_stride, a->uv_stride }; |
| const int b_strides[3] = { b->y_stride, b->uv_stride, b->uv_stride }; |
| int i; |
| uint64_t total_sse = 0; |
| uint32_t total_samples = 0; |
| #if CONFIG_LIBVMAF_PSNR_PEAK |
| double peak = (double)(255 << (in_bit_depth - 8)); |
| #else |
| double peak = (double)((1 << in_bit_depth) - 1); |
| #endif // CONFIG_LIBVMAF_PSNR_PEAK |
| const unsigned int input_shift = bit_depth - in_bit_depth; |
| |
| for (i = 0; i < 3; ++i) { |
| const int w = widths[i]; |
| const int h = heights[i]; |
| const uint32_t samples = w * h; |
| uint64_t sse; |
| if (a->flags & YV12_FLAG_HIGHBITDEPTH) { |
| if (input_shift) { |
| sse = highbd_get_sse_shift(a->buffers[i], a_strides[i], b->buffers[i], |
| b_strides[i], w, h, input_shift); |
| } else { |
| sse = highbd_get_sse(a->buffers[i], a_strides[i], b->buffers[i], |
| b_strides[i], w, h); |
| } |
| } else { |
| sse = get_sse(a->buffers[i], a_strides[i], b->buffers[i], b_strides[i], w, |
| h); |
| } |
| psnr->sse[1 + i] = sse; |
| psnr->samples[1 + i] = samples; |
| psnr->psnr[1 + i] = aom_sse_to_psnr(samples, peak, (double)sse); |
| |
| total_sse += sse; |
| total_samples += samples; |
| } |
| |
| psnr->sse[0] = total_sse; |
| psnr->samples[0] = total_samples; |
| psnr->psnr[0] = |
| aom_sse_to_psnr((double)total_samples, peak, (double)total_sse); |
| |
| // Compute PSNR based on stream bit depth |
| if ((a->flags & YV12_FLAG_HIGHBITDEPTH) && (in_bit_depth < bit_depth)) { |
| #if CONFIG_LIBVMAF_PSNR_PEAK |
| peak = (double)(255 << (bit_depth - 8)); |
| #else |
| peak = (double)((1 << bit_depth) - 1); |
| #endif // CONFIG_LIBVMAF_PSNR_PEAK |
| total_sse = 0; |
| total_samples = 0; |
| for (i = 0; i < 3; ++i) { |
| const int w = widths[i]; |
| const int h = heights[i]; |
| const uint32_t samples = w * h; |
| uint64_t sse; |
| sse = highbd_get_sse(a->buffers[i], a_strides[i], b->buffers[i], |
| b_strides[i], w, h); |
| psnr->sse_hbd[1 + i] = sse; |
| psnr->samples_hbd[1 + i] = samples; |
| psnr->psnr_hbd[1 + i] = aom_sse_to_psnr(samples, peak, (double)sse); |
| total_sse += sse; |
| total_samples += samples; |
| } |
| |
| psnr->sse_hbd[0] = total_sse; |
| psnr->samples_hbd[0] = total_samples; |
| psnr->psnr_hbd[0] = |
| aom_sse_to_psnr((double)total_samples, peak, (double)total_sse); |
| } |
| } |
| #endif |
| |
| void aom_calc_psnr(const YV12_BUFFER_CONFIG *a, const YV12_BUFFER_CONFIG *b, |
| PSNR_STATS *psnr) { |
| assert(a->y_crop_width == b->y_crop_width); |
| assert(a->y_crop_height == b->y_crop_height); |
| assert(a->uv_crop_width == b->uv_crop_width); |
| assert(a->uv_crop_height == b->uv_crop_height); |
| static const double peak = 255.0; |
| const int widths[3] = { a->y_crop_width, a->uv_crop_width, a->uv_crop_width }; |
| const int heights[3] = { a->y_crop_height, a->uv_crop_height, |
| a->uv_crop_height }; |
| const int a_strides[3] = { a->y_stride, a->uv_stride, a->uv_stride }; |
| const int b_strides[3] = { b->y_stride, b->uv_stride, b->uv_stride }; |
| int i; |
| uint64_t total_sse = 0; |
| uint32_t total_samples = 0; |
| |
| for (i = 0; i < 3; ++i) { |
| const int w = widths[i]; |
| const int h = heights[i]; |
| const uint32_t samples = w * h; |
| const uint64_t sse = |
| get_sse(a->buffers[i], a_strides[i], b->buffers[i], b_strides[i], w, h); |
| psnr->sse[1 + i] = sse; |
| psnr->samples[1 + i] = samples; |
| psnr->psnr[1 + i] = aom_sse_to_psnr(samples, peak, (double)sse); |
| |
| total_sse += sse; |
| total_samples += samples; |
| } |
| |
| psnr->sse[0] = total_sse; |
| psnr->samples[0] = total_samples; |
| psnr->psnr[0] = |
| aom_sse_to_psnr((double)total_samples, peak, (double)total_sse); |
| } |