| Index: third_party/libwebp/enc/filter.c
|
| diff --git a/third_party/libwebp/enc/filter.c b/third_party/libwebp/enc/filter.c
|
| index 04b36108aac35229116acb4e2dae3497449c0cd5..7fb78a39492bfdd08acfff9affa9cd4f48892bbe 100644
|
| --- a/third_party/libwebp/enc/filter.c
|
| +++ b/third_party/libwebp/enc/filter.c
|
| @@ -1,4 +1,4 @@
|
| -// Copyright 2011 Google Inc.
|
| +// Copyright 2011 Google Inc. All Rights Reserved.
|
| //
|
| // This code is licensed under the same terms as WebM:
|
| // Software License Agreement: http://www.webmproject.org/license/software/
|
| @@ -9,8 +9,7 @@
|
| //
|
| // Author: somnath@google.com (Somnath Banerjee)
|
|
|
| -#include <math.h>
|
| -#include "vp8enci.h"
|
| +#include "./vp8enci.h"
|
|
|
| #if defined(__cplusplus) || defined(c_plusplus)
|
| extern "C" {
|
| @@ -49,7 +48,7 @@ static void InitTables(void) {
|
| // Edge filtering functions
|
|
|
| // 4 pixels in, 2 pixels out
|
| -static inline void do_filter2(uint8_t* p, int step) {
|
| +static WEBP_INLINE void do_filter2(uint8_t* p, int step) {
|
| const int p1 = p[-2*step], p0 = p[-step], q0 = p[0], q1 = p[step];
|
| const int a = 3 * (q0 - p0) + sclip1[1020 + p1 - q1];
|
| const int a1 = sclip2[112 + ((a + 4) >> 3)];
|
| @@ -59,7 +58,7 @@ static inline void do_filter2(uint8_t* p, int step) {
|
| }
|
|
|
| // 4 pixels in, 4 pixels out
|
| -static inline void do_filter4(uint8_t* p, int step) {
|
| +static WEBP_INLINE void do_filter4(uint8_t* p, int step) {
|
| const int p1 = p[-2*step], p0 = p[-step], q0 = p[0], q1 = p[step];
|
| const int a = 3 * (q0 - p0);
|
| const int a1 = sclip2[112 + ((a + 4) >> 3)];
|
| @@ -72,17 +71,18 @@ static inline void do_filter4(uint8_t* p, int step) {
|
| }
|
|
|
| // high edge-variance
|
| -static inline int hev(const uint8_t* p, int step, int thresh) {
|
| +static WEBP_INLINE int hev(const uint8_t* p, int step, int thresh) {
|
| const int p1 = p[-2*step], p0 = p[-step], q0 = p[0], q1 = p[step];
|
| return (abs0[255 + p1 - p0] > thresh) || (abs0[255 + q1 - q0] > thresh);
|
| }
|
|
|
| -static inline int needs_filter(const uint8_t* p, int step, int thresh) {
|
| +static WEBP_INLINE int needs_filter(const uint8_t* p, int step, int thresh) {
|
| const int p1 = p[-2*step], p0 = p[-step], q0 = p[0], q1 = p[step];
|
| return (2 * abs0[255 + p0 - q0] + abs1[255 + p1 - q1]) <= thresh;
|
| }
|
|
|
| -static inline int needs_filter2(const uint8_t* p, int step, int t, int it) {
|
| +static WEBP_INLINE int needs_filter2(const uint8_t* p,
|
| + int step, int t, int it) {
|
| const int p3 = p[-4*step], p2 = p[-3*step], p1 = p[-2*step], p0 = p[-step];
|
| const int q0 = p[0], q1 = p[step], q2 = p[2*step], q3 = p[3*step];
|
| if ((2 * abs0[255 + p0 - q0] + abs1[255 + p1 - q1]) > t)
|
| @@ -132,8 +132,9 @@ static void SimpleHFilter16i(uint8_t* p, int stride, int thresh) {
|
| //------------------------------------------------------------------------------
|
| // Complex In-loop filtering (Paragraph 15.3)
|
|
|
| -static inline void FilterLoop24(uint8_t* p, int hstride, int vstride, int size,
|
| - int thresh, int ithresh, int hev_thresh) {
|
| +static WEBP_INLINE void FilterLoop24(uint8_t* p,
|
| + int hstride, int vstride, int size,
|
| + int thresh, int ithresh, int hev_thresh) {
|
| while (size-- > 0) {
|
| if (needs_filter2(p, hstride, thresh, ithresh)) {
|
| if (hev(p, hstride, hev_thresh)) {
|
| @@ -233,14 +234,21 @@ static void DoFilter(const VP8EncIterator* const it, int level) {
|
| // SSIM metric
|
|
|
| enum { KERNEL = 3 };
|
| -typedef struct {
|
| - double w, xm, ym, xxm, xym, yym;
|
| -} SSIMStats;
|
| -
|
| -static void Accumulate(const uint8_t* src1, int stride1,
|
| - const uint8_t* src2, int stride2,
|
| - int xo, int yo, int W, int H,
|
| - SSIMStats* const stats) {
|
| +static const double kMinValue = 1.e-10; // minimal threshold
|
| +
|
| +void VP8SSIMAddStats(const DistoStats* const src, DistoStats* const dst) {
|
| + dst->w += src->w;
|
| + dst->xm += src->xm;
|
| + dst->ym += src->ym;
|
| + dst->xxm += src->xxm;
|
| + dst->xym += src->xym;
|
| + dst->yym += src->yym;
|
| +}
|
| +
|
| +static void VP8SSIMAccumulate(const uint8_t* src1, int stride1,
|
| + const uint8_t* src2, int stride2,
|
| + int xo, int yo, int W, int H,
|
| + DistoStats* const stats) {
|
| const int ymin = (yo - KERNEL < 0) ? 0 : yo - KERNEL;
|
| const int ymax = (yo + KERNEL > H - 1) ? H - 1 : yo + KERNEL;
|
| const int xmin = (xo - KERNEL < 0) ? 0 : xo - KERNEL;
|
| @@ -262,7 +270,7 @@ static void Accumulate(const uint8_t* src1, int stride1,
|
| }
|
| }
|
|
|
| -static double GetSSIM(const SSIMStats* const stats) {
|
| +double VP8SSIMGet(const DistoStats* const stats) {
|
| const double xmxm = stats->xm * stats->xm;
|
| const double ymym = stats->ym * stats->ym;
|
| const double xmym = stats->xm * stats->ym;
|
| @@ -280,26 +288,49 @@ static double GetSSIM(const SSIMStats* const stats) {
|
| C2 = 58.5225 * w2;
|
| fnum = (2 * xmym + C1) * (2 * sxy + C2);
|
| fden = (xmxm + ymym + C1) * (sxx + syy + C2);
|
| - return (fden != 0) ? fnum / fden : 0.;
|
| + return (fden != 0.) ? fnum / fden : kMinValue;
|
| +}
|
| +
|
| +double VP8SSIMGetSquaredError(const DistoStats* const s) {
|
| + if (s->w > 0.) {
|
| + const double iw2 = 1. / (s->w * s->w);
|
| + const double sxx = s->xxm * s->w - s->xm * s->xm;
|
| + const double syy = s->yym * s->w - s->ym * s->ym;
|
| + const double sxy = s->xym * s->w - s->xm * s->ym;
|
| + const double SSE = iw2 * (sxx + syy - 2. * sxy);
|
| + if (SSE > kMinValue) return SSE;
|
| + }
|
| + return kMinValue;
|
| +}
|
| +
|
| +void VP8SSIMAccumulatePlane(const uint8_t* src1, int stride1,
|
| + const uint8_t* src2, int stride2,
|
| + int W, int H, DistoStats* const stats) {
|
| + int x, y;
|
| + for (y = 0; y < H; ++y) {
|
| + for (x = 0; x < W; ++x) {
|
| + VP8SSIMAccumulate(src1, stride1, src2, stride2, x, y, W, H, stats);
|
| + }
|
| + }
|
| }
|
|
|
| static double GetMBSSIM(const uint8_t* yuv1, const uint8_t* yuv2) {
|
| int x, y;
|
| - SSIMStats s = { .0, .0, .0, .0, .0, .0 };
|
| + DistoStats s = { .0, .0, .0, .0, .0, .0 };
|
|
|
| // compute SSIM in a 10 x 10 window
|
| for (x = 3; x < 13; x++) {
|
| for (y = 3; y < 13; y++) {
|
| - Accumulate(yuv1 + Y_OFF, BPS, yuv2 + Y_OFF, BPS, x, y, 16, 16, &s);
|
| + VP8SSIMAccumulate(yuv1 + Y_OFF, BPS, yuv2 + Y_OFF, BPS, x, y, 16, 16, &s);
|
| }
|
| }
|
| for (x = 1; x < 7; x++) {
|
| for (y = 1; y < 7; y++) {
|
| - Accumulate(yuv1 + U_OFF, BPS, yuv2 + U_OFF, BPS, x, y, 8, 8, &s);
|
| - Accumulate(yuv1 + V_OFF, BPS, yuv2 + V_OFF, BPS, x, y, 8, 8, &s);
|
| + VP8SSIMAccumulate(yuv1 + U_OFF, BPS, yuv2 + U_OFF, BPS, x, y, 8, 8, &s);
|
| + VP8SSIMAccumulate(yuv1 + V_OFF, BPS, yuv2 + V_OFF, BPS, x, y, 8, 8, &s);
|
| }
|
| }
|
| - return GetSSIM(&s);
|
| + return VP8SSIMGet(&s);
|
| }
|
|
|
| //------------------------------------------------------------------------------
|
|
|