| OLD | NEW |
| 1 // Copyright 2014 Google Inc. All Rights Reserved. | 1 // Copyright 2014 Google Inc. All Rights Reserved. |
| 2 // | 2 // |
| 3 // Use of this source code is governed by a BSD-style license | 3 // Use of this source code is governed by a BSD-style license |
| 4 // that can be found in the COPYING file in the root of the source | 4 // that can be found in the COPYING file in the root of the source |
| 5 // tree. An additional intellectual property rights grant can be found | 5 // tree. An additional intellectual property rights grant can be found |
| 6 // in the file PATENTS. All contributing project authors may | 6 // in the file PATENTS. All contributing project authors may |
| 7 // be found in the AUTHORS file in the root of the source tree. | 7 // be found in the AUTHORS file in the root of the source tree. |
| 8 // ----------------------------------------------------------------------------- | 8 // ----------------------------------------------------------------------------- |
| 9 // | 9 // |
| 10 // WebPPicture tools for measuring distortion | 10 // WebPPicture tools for measuring distortion |
| 11 // | 11 // |
| 12 // Author: Skal (pascal.massimino@gmail.com) | 12 // Author: Skal (pascal.massimino@gmail.com) |
| 13 | 13 |
| 14 #include <math.h> | 14 #include <math.h> |
| 15 #include <stdlib.h> | 15 #include <stdlib.h> |
| 16 | 16 |
| 17 #include "./vp8enci.h" | 17 #include "./vp8enci.h" |
| 18 #include "../utils/utils.h" | 18 #include "../utils/utils.h" |
| 19 | 19 |
| 20 //------------------------------------------------------------------------------ | 20 //------------------------------------------------------------------------------ |
| 21 // local-min distortion | 21 // local-min distortion |
| 22 // | 22 // |
| 23 // For every pixel in the *reference* picture, we search for the local best | 23 // For every pixel in the *reference* picture, we search for the local best |
| 24 // match in the compressed image. This is not a symmetrical measure. | 24 // match in the compressed image. This is not a symmetrical measure. |
| 25 | 25 |
| 26 #define RADIUS 2 // search radius. Shouldn't be too large. | 26 #define RADIUS 2 // search radius. Shouldn't be too large. |
| 27 | 27 |
| 28 static void AccumulateLSIM(const uint8_t* src, int src_stride, | 28 static void AccumulateLSIM(const uint8_t* src, int src_stride, |
| 29 const uint8_t* ref, int ref_stride, | 29 const uint8_t* ref, int ref_stride, |
| 30 int w, int h, DistoStats* stats) { | 30 int w, int h, VP8DistoStats* stats) { |
| 31 int x, y; | 31 int x, y; |
| 32 double total_sse = 0.; | 32 double total_sse = 0.; |
| 33 for (y = 0; y < h; ++y) { | 33 for (y = 0; y < h; ++y) { |
| 34 const int y_0 = (y - RADIUS < 0) ? 0 : y - RADIUS; | 34 const int y_0 = (y - RADIUS < 0) ? 0 : y - RADIUS; |
| 35 const int y_1 = (y + RADIUS + 1 >= h) ? h : y + RADIUS + 1; | 35 const int y_1 = (y + RADIUS + 1 >= h) ? h : y + RADIUS + 1; |
| 36 for (x = 0; x < w; ++x) { | 36 for (x = 0; x < w; ++x) { |
| 37 const int x_0 = (x - RADIUS < 0) ? 0 : x - RADIUS; | 37 const int x_0 = (x - RADIUS < 0) ? 0 : x - RADIUS; |
| 38 const int x_1 = (x + RADIUS + 1 >= w) ? w : x + RADIUS + 1; | 38 const int x_1 = (x + RADIUS + 1 >= w) ? w : x + RADIUS + 1; |
| 39 double best_sse = 255. * 255.; | 39 double best_sse = 255. * 255.; |
| 40 const double value = (double)ref[y * ref_stride + x]; | 40 const double value = (double)ref[y * ref_stride + x]; |
| (...skipping 23 matching lines...) Expand all Loading... |
| 64 | 64 |
| 65 // Max value returned in case of exact similarity. | 65 // Max value returned in case of exact similarity. |
| 66 static const double kMinDistortion_dB = 99.; | 66 static const double kMinDistortion_dB = 99.; |
| 67 static float GetPSNR(const double v) { | 67 static float GetPSNR(const double v) { |
| 68 return (float)((v > 0.) ? -4.3429448 * log(v / (255 * 255.)) | 68 return (float)((v > 0.) ? -4.3429448 * log(v / (255 * 255.)) |
| 69 : kMinDistortion_dB); | 69 : kMinDistortion_dB); |
| 70 } | 70 } |
| 71 | 71 |
| 72 int WebPPictureDistortion(const WebPPicture* src, const WebPPicture* ref, | 72 int WebPPictureDistortion(const WebPPicture* src, const WebPPicture* ref, |
| 73 int type, float result[5]) { | 73 int type, float result[5]) { |
| 74 DistoStats stats[5]; | 74 VP8DistoStats stats[5]; |
| 75 int w, h; | 75 int w, h; |
| 76 | 76 |
| 77 memset(stats, 0, sizeof(stats)); | 77 memset(stats, 0, sizeof(stats)); |
| 78 | 78 |
| 79 VP8SSIMDspInit(); |
| 80 |
| 79 if (src == NULL || ref == NULL || | 81 if (src == NULL || ref == NULL || |
| 80 src->width != ref->width || src->height != ref->height || | 82 src->width != ref->width || src->height != ref->height || |
| 81 src->use_argb != ref->use_argb || result == NULL) { | 83 src->use_argb != ref->use_argb || result == NULL) { |
| 82 return 0; | 84 return 0; |
| 83 } | 85 } |
| 84 w = src->width; | 86 w = src->width; |
| 85 h = src->height; | 87 h = src->height; |
| 86 | 88 |
| 87 if (src->use_argb == 1) { | 89 if (src->use_argb == 1) { |
| 88 if (src->argb == NULL || ref->argb == NULL) { | 90 if (src->argb == NULL || ref->argb == NULL) { |
| (...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 166 result[c] = GetPSNR(v); | 168 result[c] = GetPSNR(v); |
| 167 } | 169 } |
| 168 // Accumulate forward | 170 // Accumulate forward |
| 169 if (c < 4) VP8SSIMAddStats(&stats[c], &stats[4]); | 171 if (c < 4) VP8SSIMAddStats(&stats[c], &stats[4]); |
| 170 } | 172 } |
| 171 } | 173 } |
| 172 return 1; | 174 return 1; |
| 173 } | 175 } |
| 174 | 176 |
| 175 //------------------------------------------------------------------------------ | 177 //------------------------------------------------------------------------------ |
| OLD | NEW |