Blame src/enc/picture_psnr_enc.c

Packit 9c6abc
// Copyright 2014 Google Inc. All Rights Reserved.
Packit 9c6abc
//
Packit 9c6abc
// Use of this source code is governed by a BSD-style license
Packit 9c6abc
// that can be found in the COPYING file in the root of the source
Packit 9c6abc
// tree. An additional intellectual property rights grant can be found
Packit 9c6abc
// in the file PATENTS. All contributing project authors may
Packit 9c6abc
// be found in the AUTHORS file in the root of the source tree.
Packit 9c6abc
// -----------------------------------------------------------------------------
Packit 9c6abc
//
Packit 9c6abc
// WebPPicture tools for measuring distortion
Packit 9c6abc
//
Packit 9c6abc
// Author: Skal (pascal.massimino@gmail.com)
Packit 9c6abc
Packit 9c6abc
#include "src/webp/encode.h"
Packit 9c6abc
Packit 9c6abc
#if !(defined(WEBP_DISABLE_STATS) || defined(WEBP_REDUCE_SIZE))
Packit 9c6abc
Packit 9c6abc
#include <math.h>
Packit 9c6abc
#include <stdlib.h>
Packit 9c6abc
Packit 9c6abc
#include "src/dsp/dsp.h"
Packit 9c6abc
#include "src/enc/vp8i_enc.h"
Packit 9c6abc
#include "src/utils/utils.h"
Packit 9c6abc
Packit 9c6abc
typedef double (*AccumulateFunc)(const uint8_t* src, int src_stride,
Packit 9c6abc
                                 const uint8_t* ref, int ref_stride,
Packit 9c6abc
                                 int w, int h);
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// local-min distortion
Packit 9c6abc
//
Packit 9c6abc
// For every pixel in the *reference* picture, we search for the local best
Packit 9c6abc
// match in the compressed image. This is not a symmetrical measure.
Packit 9c6abc
Packit 9c6abc
#define RADIUS 2  // search radius. Shouldn't be too large.
Packit 9c6abc
Packit 9c6abc
static double AccumulateLSIM(const uint8_t* src, int src_stride,
Packit 9c6abc
                             const uint8_t* ref, int ref_stride,
Packit 9c6abc
                             int w, int h) {
Packit 9c6abc
  int x, y;
Packit 9c6abc
  double total_sse = 0.;
Packit 9c6abc
  for (y = 0; y < h; ++y) {
Packit 9c6abc
    const int y_0 = (y - RADIUS < 0) ? 0 : y - RADIUS;
Packit 9c6abc
    const int y_1 = (y + RADIUS + 1 >= h) ? h : y + RADIUS + 1;
Packit 9c6abc
    for (x = 0; x < w; ++x) {
Packit 9c6abc
      const int x_0 = (x - RADIUS < 0) ? 0 : x - RADIUS;
Packit 9c6abc
      const int x_1 = (x + RADIUS + 1 >= w) ? w : x + RADIUS + 1;
Packit 9c6abc
      double best_sse = 255. * 255.;
Packit 9c6abc
      const double value = (double)ref[y * ref_stride + x];
Packit 9c6abc
      int i, j;
Packit 9c6abc
      for (j = y_0; j < y_1; ++j) {
Packit 9c6abc
        const uint8_t* const s = src + j * src_stride;
Packit 9c6abc
        for (i = x_0; i < x_1; ++i) {
Packit 9c6abc
          const double diff = s[i] - value;
Packit 9c6abc
          const double sse = diff * diff;
Packit 9c6abc
          if (sse < best_sse) best_sse = sse;
Packit 9c6abc
        }
Packit 9c6abc
      }
Packit 9c6abc
      total_sse += best_sse;
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
  return total_sse;
Packit 9c6abc
}
Packit 9c6abc
#undef RADIUS
Packit 9c6abc
Packit 9c6abc
static double AccumulateSSE(const uint8_t* src, int src_stride,
Packit 9c6abc
                            const uint8_t* ref, int ref_stride,
Packit 9c6abc
                            int w, int h) {
Packit 9c6abc
  int y;
Packit 9c6abc
  double total_sse = 0.;
Packit 9c6abc
  for (y = 0; y < h; ++y) {
Packit 9c6abc
    total_sse += VP8AccumulateSSE(src, ref, w);
Packit 9c6abc
    src += src_stride;
Packit 9c6abc
    ref += ref_stride;
Packit 9c6abc
  }
Packit 9c6abc
  return total_sse;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
Packit 9c6abc
static double AccumulateSSIM(const uint8_t* src, int src_stride,
Packit 9c6abc
                             const uint8_t* ref, int ref_stride,
Packit 9c6abc
                             int w, int h) {
Packit 9c6abc
  const int w0 = (w < VP8_SSIM_KERNEL) ? w : VP8_SSIM_KERNEL;
Packit 9c6abc
  const int w1 = w - VP8_SSIM_KERNEL - 1;
Packit 9c6abc
  const int h0 = (h < VP8_SSIM_KERNEL) ? h : VP8_SSIM_KERNEL;
Packit 9c6abc
  const int h1 = h - VP8_SSIM_KERNEL - 1;
Packit 9c6abc
  int x, y;
Packit 9c6abc
  double sum = 0.;
Packit 9c6abc
  for (y = 0; y < h0; ++y) {
Packit 9c6abc
    for (x = 0; x < w; ++x) {
Packit 9c6abc
      sum += VP8SSIMGetClipped(src, src_stride, ref, ref_stride, x, y, w, h);
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
  for (; y < h1; ++y) {
Packit 9c6abc
    for (x = 0; x < w0; ++x) {
Packit 9c6abc
      sum += VP8SSIMGetClipped(src, src_stride, ref, ref_stride, x, y, w, h);
Packit 9c6abc
    }
Packit 9c6abc
    for (; x < w1; ++x) {
Packit 9c6abc
      const int off1 = x - VP8_SSIM_KERNEL + (y - VP8_SSIM_KERNEL) * src_stride;
Packit 9c6abc
      const int off2 = x - VP8_SSIM_KERNEL + (y - VP8_SSIM_KERNEL) * ref_stride;
Packit 9c6abc
      sum += VP8SSIMGet(src + off1, src_stride, ref + off2, ref_stride);
Packit 9c6abc
    }
Packit 9c6abc
    for (; x < w; ++x) {
Packit 9c6abc
      sum += VP8SSIMGetClipped(src, src_stride, ref, ref_stride, x, y, w, h);
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
  for (; y < h; ++y) {
Packit 9c6abc
    for (x = 0; x < w; ++x) {
Packit 9c6abc
      sum += VP8SSIMGetClipped(src, src_stride, ref, ref_stride, x, y, w, h);
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
  return sum;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
// Distortion
Packit 9c6abc
Packit 9c6abc
// Max value returned in case of exact similarity.
Packit 9c6abc
static const double kMinDistortion_dB = 99.;
Packit 9c6abc
Packit 9c6abc
static double GetPSNR(double v, double size) {
Packit 9c6abc
  return (v > 0. && size > 0.) ? -4.3429448 * log(v / (size * 255 * 255.))
Packit 9c6abc
                               : kMinDistortion_dB;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
static double GetLogSSIM(double v, double size) {
Packit 9c6abc
  v = (size > 0.) ? v / size : 1.;
Packit 9c6abc
  return (v < 1.) ? -10.0 * log10(1. - v) : kMinDistortion_dB;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
int WebPPlaneDistortion(const uint8_t* src, size_t src_stride,
Packit 9c6abc
                        const uint8_t* ref, size_t ref_stride,
Packit 9c6abc
                        int width, int height, size_t x_step,
Packit 9c6abc
                        int type, float* distortion, float* result) {
Packit 9c6abc
  uint8_t* allocated = NULL;
Packit 9c6abc
  const AccumulateFunc metric = (type == 0) ? AccumulateSSE :
Packit 9c6abc
                                (type == 1) ? AccumulateSSIM :
Packit 9c6abc
                                              AccumulateLSIM;
Packit 9c6abc
  if (src == NULL || ref == NULL ||
Packit 9c6abc
      src_stride < x_step * width || ref_stride < x_step * width ||
Packit 9c6abc
      result == NULL || distortion == NULL) {
Packit 9c6abc
    return 0;
Packit 9c6abc
  }
Packit 9c6abc
Packit 9c6abc
  VP8SSIMDspInit();
Packit 9c6abc
  if (x_step != 1) {   // extract a packed plane if needed
Packit 9c6abc
    int x, y;
Packit 9c6abc
    uint8_t* tmp1;
Packit 9c6abc
    uint8_t* tmp2;
Packit 9c6abc
    allocated =
Packit 9c6abc
        (uint8_t*)WebPSafeMalloc(2ULL * width * height, sizeof(*allocated));
Packit 9c6abc
    if (allocated == NULL) return 0;
Packit 9c6abc
    tmp1 = allocated;
Packit 9c6abc
    tmp2 = tmp1 + (size_t)width * height;
Packit 9c6abc
    for (y = 0; y < height; ++y) {
Packit 9c6abc
      for (x = 0; x < width; ++x) {
Packit 9c6abc
        tmp1[x + y * width] = src[x * x_step + y * src_stride];
Packit 9c6abc
        tmp2[x + y * width] = ref[x * x_step + y * ref_stride];
Packit 9c6abc
      }
Packit 9c6abc
    }
Packit 9c6abc
    src = tmp1;
Packit 9c6abc
    ref = tmp2;
Packit 9c6abc
  }
Packit 9c6abc
  *distortion = (float)metric(src, width, ref, width, width, height);
Packit 9c6abc
  WebPSafeFree(allocated);
Packit 9c6abc
Packit 9c6abc
  *result = (type == 1) ? (float)GetLogSSIM(*distortion, (double)width * height)
Packit 9c6abc
                        : (float)GetPSNR(*distortion, (double)width * height);
Packit 9c6abc
  return 1;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
#ifdef WORDS_BIGENDIAN
Packit 9c6abc
#define BLUE_OFFSET 3   // uint32_t 0x000000ff is 0x00,00,00,ff in memory
Packit 9c6abc
#else
Packit 9c6abc
#define BLUE_OFFSET 0   // uint32_t 0x000000ff is 0xff,00,00,00 in memory
Packit 9c6abc
#endif
Packit 9c6abc
Packit 9c6abc
int WebPPictureDistortion(const WebPPicture* src, const WebPPicture* ref,
Packit 9c6abc
                          int type, float results[5]) {
Packit 9c6abc
  int w, h, c;
Packit 9c6abc
  int ok = 0;
Packit 9c6abc
  WebPPicture p0, p1;
Packit 9c6abc
  double total_size = 0., total_distortion = 0.;
Packit 9c6abc
  if (src == NULL || ref == NULL ||
Packit 9c6abc
      src->width != ref->width || src->height != ref->height ||
Packit 9c6abc
      results == NULL) {
Packit 9c6abc
    return 0;
Packit 9c6abc
  }
Packit 9c6abc
Packit 9c6abc
  VP8SSIMDspInit();
Packit 9c6abc
  if (!WebPPictureInit(&p0) || !WebPPictureInit(&p1)) return 0;
Packit 9c6abc
  w = src->width;
Packit 9c6abc
  h = src->height;
Packit 9c6abc
  if (!WebPPictureView(src, 0, 0, w, h, &p0)) goto Error;
Packit 9c6abc
  if (!WebPPictureView(ref, 0, 0, w, h, &p1)) goto Error;
Packit 9c6abc
Packit 9c6abc
  // We always measure distortion in ARGB space.
Packit 9c6abc
  if (p0.use_argb == 0 && !WebPPictureYUVAToARGB(&p0)) goto Error;
Packit 9c6abc
  if (p1.use_argb == 0 && !WebPPictureYUVAToARGB(&p1)) goto Error;
Packit 9c6abc
  for (c = 0; c < 4; ++c) {
Packit 9c6abc
    float distortion;
Packit 9c6abc
    const size_t stride0 = 4 * (size_t)p0.argb_stride;
Packit 9c6abc
    const size_t stride1 = 4 * (size_t)p1.argb_stride;
Packit 9c6abc
    // results are reported as BGRA
Packit 9c6abc
    const int offset = c ^ BLUE_OFFSET;
Packit 9c6abc
    if (!WebPPlaneDistortion((const uint8_t*)p0.argb + offset, stride0,
Packit 9c6abc
                             (const uint8_t*)p1.argb + offset, stride1,
Packit 9c6abc
                             w, h, 4, type, &distortion, results + c)) {
Packit 9c6abc
      goto Error;
Packit 9c6abc
    }
Packit 9c6abc
    total_distortion += distortion;
Packit 9c6abc
    total_size += w * h;
Packit 9c6abc
  }
Packit 9c6abc
Packit 9c6abc
  results[4] = (type == 1) ? (float)GetLogSSIM(total_distortion, total_size)
Packit 9c6abc
                           : (float)GetPSNR(total_distortion, total_size);
Packit 9c6abc
  ok = 1;
Packit 9c6abc
Packit 9c6abc
 Error:
Packit 9c6abc
  WebPPictureFree(&p0;;
Packit 9c6abc
  WebPPictureFree(&p1;;
Packit 9c6abc
  return ok;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
#undef BLUE_OFFSET
Packit 9c6abc
Packit 9c6abc
#else  // defined(WEBP_DISABLE_STATS)
Packit 9c6abc
int WebPPlaneDistortion(const uint8_t* src, size_t src_stride,
Packit 9c6abc
                        const uint8_t* ref, size_t ref_stride,
Packit 9c6abc
                        int width, int height, size_t x_step,
Packit 9c6abc
                        int type, float* distortion, float* result) {
Packit 9c6abc
  (void)src;
Packit 9c6abc
  (void)src_stride;
Packit 9c6abc
  (void)ref;
Packit 9c6abc
  (void)ref_stride;
Packit 9c6abc
  (void)width;
Packit 9c6abc
  (void)height;
Packit 9c6abc
  (void)x_step;
Packit 9c6abc
  (void)type;
Packit 9c6abc
  if (distortion == NULL || result == NULL) return 0;
Packit 9c6abc
  *distortion = 0.f;
Packit 9c6abc
  *result = 0.f;
Packit 9c6abc
  return 1;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
int WebPPictureDistortion(const WebPPicture* src, const WebPPicture* ref,
Packit 9c6abc
                          int type, float results[5]) {
Packit 9c6abc
  int i;
Packit 9c6abc
  (void)src;
Packit 9c6abc
  (void)ref;
Packit 9c6abc
  (void)type;
Packit 9c6abc
  if (results == NULL) return 0;
Packit 9c6abc
  for (i = 0; i < 5; ++i) results[i] = 0.f;
Packit 9c6abc
  return 1;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
#endif  // !defined(WEBP_DISABLE_STATS)