Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(233)

Side by Side Diff: source/libvpx/vp9/encoder/vp9_ssim.c

Issue 11974002: libvpx: Pull from upstream (Closed) Base URL: svn://chrome-svn/chrome/trunk/deps/third_party/libvpx/
Patch Set: Created 7 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 /* 1 /*
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source 5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 11
12 #include "vp9/encoder/vp9_onyx_int.h" 12 #include "vp9/encoder/vp9_onyx_int.h"
13 13
14 void vp9_ssim_parms_16x16_c(unsigned char *s, int sp, unsigned char *r, 14 void vp9_ssim_parms_16x16_c(uint8_t *s, int sp, uint8_t *r,
15 int rp, unsigned long *sum_s, unsigned long *sum_r, 15 int rp, unsigned long *sum_s, unsigned long *sum_r,
16 unsigned long *sum_sq_s, unsigned long *sum_sq_r, 16 unsigned long *sum_sq_s, unsigned long *sum_sq_r,
17 unsigned long *sum_sxr) { 17 unsigned long *sum_sxr) {
18 int i, j; 18 int i, j;
19 for (i = 0; i < 16; i++, s += sp, r += rp) { 19 for (i = 0; i < 16; i++, s += sp, r += rp) {
20 for (j = 0; j < 16; j++) { 20 for (j = 0; j < 16; j++) {
21 *sum_s += s[j]; 21 *sum_s += s[j];
22 *sum_r += r[j]; 22 *sum_r += r[j];
23 *sum_sq_s += s[j] * s[j]; 23 *sum_sq_s += s[j] * s[j];
24 *sum_sq_r += r[j] * r[j]; 24 *sum_sq_r += r[j] * r[j];
25 *sum_sxr += s[j] * r[j]; 25 *sum_sxr += s[j] * r[j];
26 } 26 }
27 } 27 }
28 } 28 }
29 void vp9_ssim_parms_8x8_c(unsigned char *s, int sp, unsigned char *r, int rp, 29 void vp9_ssim_parms_8x8_c(uint8_t *s, int sp, uint8_t *r, int rp,
30 unsigned long *sum_s, unsigned long *sum_r, 30 unsigned long *sum_s, unsigned long *sum_r,
31 unsigned long *sum_sq_s, unsigned long *sum_sq_r, 31 unsigned long *sum_sq_s, unsigned long *sum_sq_r,
32 unsigned long *sum_sxr) { 32 unsigned long *sum_sxr) {
33 int i, j; 33 int i, j;
34 for (i = 0; i < 8; i++, s += sp, r += rp) { 34 for (i = 0; i < 8; i++, s += sp, r += rp) {
35 for (j = 0; j < 8; j++) { 35 for (j = 0; j < 8; j++) {
36 *sum_s += s[j]; 36 *sum_s += s[j];
37 *sum_r += r[j]; 37 *sum_r += r[j];
38 *sum_sq_s += s[j] * s[j]; 38 *sum_sq_s += s[j] * s[j];
39 *sum_sq_r += r[j] * r[j]; 39 *sum_sq_r += r[j] * r[j];
(...skipping 18 matching lines...) Expand all
58 ssim_n = (2 * sum_s * sum_r + c1) * ((int64_t) 2 * count * sum_sxr - 58 ssim_n = (2 * sum_s * sum_r + c1) * ((int64_t) 2 * count * sum_sxr -
59 (int64_t) 2 * sum_s * sum_r + c2); 59 (int64_t) 2 * sum_s * sum_r + c2);
60 60
61 ssim_d = (sum_s * sum_s + sum_r * sum_r + c1) * 61 ssim_d = (sum_s * sum_s + sum_r * sum_r + c1) *
62 ((int64_t)count * sum_sq_s - (int64_t)sum_s * sum_s + 62 ((int64_t)count * sum_sq_s - (int64_t)sum_s * sum_s +
63 (int64_t)count * sum_sq_r - (int64_t) sum_r * sum_r + c2); 63 (int64_t)count * sum_sq_r - (int64_t) sum_r * sum_r + c2);
64 64
65 return ssim_n * 1.0 / ssim_d; 65 return ssim_n * 1.0 / ssim_d;
66 } 66 }
67 67
68 static double ssim_16x16(unsigned char *s, int sp, unsigned char *r, int rp) { 68 static double ssim_16x16(uint8_t *s, int sp, uint8_t *r, int rp) {
69 unsigned long sum_s = 0, sum_r = 0, sum_sq_s = 0, sum_sq_r = 0, sum_sxr = 0; 69 unsigned long sum_s = 0, sum_r = 0, sum_sq_s = 0, sum_sq_r = 0, sum_sxr = 0;
70 vp9_ssim_parms_16x16(s, sp, r, rp, &sum_s, &sum_r, &sum_sq_s, &sum_sq_r, 70 vp9_ssim_parms_16x16(s, sp, r, rp, &sum_s, &sum_r, &sum_sq_s, &sum_sq_r,
71 &sum_sxr); 71 &sum_sxr);
72 return similarity(sum_s, sum_r, sum_sq_s, sum_sq_r, sum_sxr, 256); 72 return similarity(sum_s, sum_r, sum_sq_s, sum_sq_r, sum_sxr, 256);
73 } 73 }
74 static double ssim_8x8(unsigned char *s, int sp, unsigned char *r, int rp) { 74 static double ssim_8x8(uint8_t *s, int sp, uint8_t *r, int rp) {
75 unsigned long sum_s = 0, sum_r = 0, sum_sq_s = 0, sum_sq_r = 0, sum_sxr = 0; 75 unsigned long sum_s = 0, sum_r = 0, sum_sq_s = 0, sum_sq_r = 0, sum_sxr = 0;
76 vp9_ssim_parms_8x8(s, sp, r, rp, &sum_s, &sum_r, &sum_sq_s, &sum_sq_r, 76 vp9_ssim_parms_8x8(s, sp, r, rp, &sum_s, &sum_r, &sum_sq_s, &sum_sq_r,
77 &sum_sxr); 77 &sum_sxr);
78 return similarity(sum_s, sum_r, sum_sq_s, sum_sq_r, sum_sxr, 64); 78 return similarity(sum_s, sum_r, sum_sq_s, sum_sq_r, sum_sxr, 64);
79 } 79 }
80 80
81 // We are using a 8x8 moving window with starting location of each 8x8 window 81 // We are using a 8x8 moving window with starting location of each 8x8 window
82 // on the 4x4 pixel grid. Such arrangement allows the windows to overlap 82 // on the 4x4 pixel grid. Such arrangement allows the windows to overlap
83 // block boundaries to penalize blocking artifacts. 83 // block boundaries to penalize blocking artifacts.
84 double vp9_ssim2(unsigned char *img1, unsigned char *img2, int stride_img1, 84 double vp9_ssim2(uint8_t *img1, uint8_t *img2, int stride_img1,
85 int stride_img2, int width, int height) { 85 int stride_img2, int width, int height) {
86 int i, j; 86 int i, j;
87 int samples = 0; 87 int samples = 0;
88 double ssim_total = 0; 88 double ssim_total = 0;
89 89
90 // sample point start with each 4x4 location 90 // sample point start with each 4x4 location
91 for (i = 0; i < height - 8; i += 4, img1 += stride_img1 * 4, img2 += stride_im g2 * 4) { 91 for (i = 0; i < height - 8; i += 4, img1 += stride_img1 * 4, img2 += stride_im g2 * 4) {
92 for (j = 0; j < width - 8; j += 4) { 92 for (j = 0; j < width - 8; j += 4) {
93 double v = ssim_8x8(img1 + j, stride_img1, img2 + j, stride_img2); 93 double v = ssim_8x8(img1 + j, stride_img1, img2 + j, stride_img2);
94 ssim_total += v; 94 ssim_total += v;
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after
138 c = vp9_ssim2(source->v_buffer, dest->v_buffer, 138 c = vp9_ssim2(source->v_buffer, dest->v_buffer,
139 source->uv_stride, dest->uv_stride, source->uv_width, 139 source->uv_stride, dest->uv_stride, source->uv_width,
140 source->uv_height); 140 source->uv_height);
141 *ssim_y = a; 141 *ssim_y = a;
142 *ssim_u = b; 142 *ssim_u = b;
143 *ssim_v = c; 143 *ssim_v = c;
144 ssim_all = (a * 4 + b + c) / 6; 144 ssim_all = (a * 4 + b + c) / 6;
145 145
146 return ssim_all; 146 return ssim_all;
147 } 147 }
OLDNEW
« no previous file with comments | « source/libvpx/vp9/encoder/vp9_segmentation.c ('k') | source/libvpx/vp9/encoder/vp9_temporal_filter.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698