* commit 'dd33637c18629c3e554ebb146bbeb45c9745a5cf':
tiny_psnr: switch f32 handling to floating point
Conflicts:
tests/tiny_psnr.c
Merged-by: Michael Niedermayer <michaelni@gmx.at>
... | ... |
@@ -22,8 +22,12 @@ |
22 | 22 |
#include <stdlib.h> |
23 | 23 |
#include <string.h> |
24 | 24 |
#include <inttypes.h> |
25 |
+#include <math.h> |
|
26 |
+#include <float.h> |
|
27 |
+#include <limits.h> |
|
25 | 28 |
|
26 | 29 |
#include "libavutil/intfloat.h" |
30 |
+#include "libavutil/intreadwrite.h" |
|
27 | 31 |
|
28 | 32 |
#define FFMIN(a, b) ((a) > (b) ? (b) : (a)) |
29 | 33 |
#define F 100 |
... | ... |
@@ -121,17 +125,22 @@ static float get_f32l(uint8_t *p) |
121 | 121 |
return v.f; |
122 | 122 |
} |
123 | 123 |
|
124 |
+static double get_f64l(uint8_t *p) |
|
125 |
+{ |
|
126 |
+ return av_int2double(AV_RL64(p)); |
|
127 |
+} |
|
128 |
+ |
|
124 | 129 |
static int run_psnr(FILE *f[2], int len, int shift, int skip_bytes) |
125 | 130 |
{ |
126 | 131 |
int i, j; |
127 | 132 |
uint64_t sse = 0; |
128 |
- uint64_t dev; |
|
133 |
+ double sse_d = 0.0; |
|
129 | 134 |
uint8_t buf[2][SIZE]; |
130 |
- uint64_t psnr; |
|
131 | 135 |
int64_t max = (1LL << (8 * len)) - 1; |
132 | 136 |
int size0 = 0; |
133 | 137 |
int size1 = 0; |
134 |
- int maxdist = 0; |
|
138 |
+ uint64_t maxdist = 0; |
|
139 |
+ double maxdist_d = 0.0; |
|
135 | 140 |
int noseek; |
136 | 141 |
|
137 | 142 |
noseek = fseek(f[0], 0, SEEK_SET) || |
... | ... |
@@ -168,23 +177,42 @@ static int run_psnr(FILE *f[2], int len, int shift, int skip_bytes) |
168 | 168 |
int s1 = fread(buf[1], 1, SIZE, f[1]); |
169 | 169 |
|
170 | 170 |
for (j = 0; j < FFMIN(s0, s1); j += len) { |
171 |
- int64_t a = buf[0][j]; |
|
172 |
- int64_t b = buf[1][j]; |
|
173 |
- int dist; |
|
174 |
- if (len == 2) { |
|
175 |
- a = get_s16l(buf[0] + j); |
|
176 |
- b = get_s16l(buf[1] + j); |
|
177 |
- } else if (len == 4) { |
|
178 |
- a = get_f32l(buf[0] + j) * (1 << 24); |
|
179 |
- b = get_f32l(buf[1] + j) * (1 << 24); |
|
180 |
- } else { |
|
181 |
- a = buf[0][j]; |
|
182 |
- b = buf[1][j]; |
|
171 |
+ switch (len) { |
|
172 |
+ case 1: |
|
173 |
+ case 2: { |
|
174 |
+ int64_t a = buf[0][j]; |
|
175 |
+ int64_t b = buf[1][j]; |
|
176 |
+ int dist; |
|
177 |
+ if (len == 2) { |
|
178 |
+ a = get_s16l(buf[0] + j); |
|
179 |
+ b = get_s16l(buf[1] + j); |
|
180 |
+ } else { |
|
181 |
+ a = buf[0][j]; |
|
182 |
+ b = buf[1][j]; |
|
183 |
+ } |
|
184 |
+ sse += (a - b) * (a - b); |
|
185 |
+ dist = abs(a - b); |
|
186 |
+ if (dist > maxdist) |
|
187 |
+ maxdist = dist; |
|
188 |
+ break; |
|
189 |
+ } |
|
190 |
+ case 4: |
|
191 |
+ case 8: { |
|
192 |
+ double dist, a, b; |
|
193 |
+ if (len == 8) { |
|
194 |
+ a = get_f64l(buf[0] + j); |
|
195 |
+ b = get_f64l(buf[1] + j); |
|
196 |
+ } else { |
|
197 |
+ a = get_f32l(buf[0] + j); |
|
198 |
+ b = get_f32l(buf[1] + j); |
|
199 |
+ } |
|
200 |
+ dist = fabs(a - b); |
|
201 |
+ sse_d += (a - b) * (a - b); |
|
202 |
+ if (dist > maxdist_d) |
|
203 |
+ maxdist_d = dist; |
|
204 |
+ break; |
|
205 |
+ } |
|
183 | 206 |
} |
184 |
- sse += (a - b) * (a - b); |
|
185 |
- dist = abs(a - b); |
|
186 |
- if (dist > maxdist) |
|
187 |
- maxdist = dist; |
|
188 | 207 |
} |
189 | 208 |
size0 += s0; |
190 | 209 |
size1 += s1; |
... | ... |
@@ -195,18 +223,44 @@ static int run_psnr(FILE *f[2], int len, int shift, int skip_bytes) |
195 | 195 |
i = FFMIN(size0, size1) / len; |
196 | 196 |
if (!i) |
197 | 197 |
i = 1; |
198 |
- dev = int_sqrt(((sse / i) * F * F) + (((sse % i) * F * F) + i / 2) / i); |
|
199 |
- if (sse) |
|
200 |
- psnr = ((2 * log16(max << 16) + log16(i) - log16(sse)) * |
|
201 |
- 284619LL * F + (1LL << 31)) / (1LL << 32); |
|
202 |
- else |
|
203 |
- psnr = 1000 * F - 1; // floating point free infinity :) |
|
204 |
- |
|
205 |
- printf("stddev:%5d.%02d PSNR:%3d.%02d MAXDIFF:%5d bytes:%9d/%9d\n", |
|
206 |
- (int)(dev / F), (int)(dev % F), |
|
207 |
- (int)(psnr / F), (int)(psnr % F), |
|
208 |
- maxdist, size0, size1); |
|
209 |
- return psnr; |
|
198 |
+ switch (len) { |
|
199 |
+ case 1: |
|
200 |
+ case 2: { |
|
201 |
+ uint64_t psnr; |
|
202 |
+ uint64_t dev = int_sqrt(((sse / i) * F * F) + (((sse % i) * F * F) + i / 2) / i); |
|
203 |
+ if (sse) |
|
204 |
+ psnr = ((2 * log16(max << 16) + log16(i) - log16(sse)) * |
|
205 |
+ 284619LL * F + (1LL << 31)) / (1LL << 32); |
|
206 |
+ else |
|
207 |
+ psnr = 1000 * F - 1; // floating point free infinity :) |
|
208 |
+ |
|
209 |
+ printf("stddev:%5d.%02d PSNR:%3d.%02d MAXDIFF:%5"PRIu64" bytes:%9d/%9d\n", |
|
210 |
+ (int)(dev / F), (int)(dev % F), |
|
211 |
+ (int)(psnr / F), (int)(psnr % F), |
|
212 |
+ maxdist, size0, size1); |
|
213 |
+ return psnr; |
|
214 |
+ } |
|
215 |
+ case 4: |
|
216 |
+ case 8: { |
|
217 |
+ char psnr_str[64]; |
|
218 |
+ double psnr = INT_MAX; |
|
219 |
+ double dev = sqrt(sse_d / i); |
|
220 |
+ uint64_t scale = (len == 4) ? (1ULL << 24) : (1ULL << 32); |
|
221 |
+ |
|
222 |
+ if (sse_d) { |
|
223 |
+ psnr = 2 * log(DBL_MAX) - log(i / sse_d); |
|
224 |
+ snprintf(psnr_str, sizeof(psnr_str), "%5.02f", psnr); |
|
225 |
+ } else |
|
226 |
+ snprintf(psnr_str, sizeof(psnr_str), "inf"); |
|
227 |
+ |
|
228 |
+ maxdist = maxdist_d * scale; |
|
229 |
+ |
|
230 |
+ printf("stddev:%10.2f PSNR:%s MAXDIFF:%10"PRIu64" bytes:%9d/%9d\n", |
|
231 |
+ dev * scale, psnr_str, maxdist, size0, size1); |
|
232 |
+ return psnr; |
|
233 |
+ } |
|
234 |
+ } |
|
235 |
+ return -1; |
|
210 | 236 |
} |
211 | 237 |
|
212 | 238 |
int main(int argc, char *argv[]) |
... | ... |
@@ -227,6 +281,8 @@ int main(int argc, char *argv[]) |
227 | 227 |
len = 2; |
228 | 228 |
} else if (!strcmp(argv[3], "f32")) { |
229 | 229 |
len = 4; |
230 |
+ } else if (!strcmp(argv[3], "f64")) { |
|
231 |
+ len = 8; |
|
230 | 232 |
} else { |
231 | 233 |
char *end; |
232 | 234 |
len = strtol(argv[3], &end, 0); |