-
Notifications
You must be signed in to change notification settings - Fork 1
/
benchmark.cpp
92 lines (67 loc) · 1.82 KB
/
benchmark.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include <benchmark/benchmark.h>
#include "fastimagehash.h"
#include <fcntl.h>
char *filepath;
uchar tmp[512];
void *load_test_file(size_t *buf_len) {
FILE *file = fopen(filepath, "rb");
fseek(file, 0, SEEK_END);
*buf_len = ftell(file);
fclose(file);
void *buf = malloc(*buf_len);
file = fopen(filepath, "rb");
fread(buf, *buf_len, 1, file);
return buf;
}
static void BM_phash(benchmark::State &state) {
size_t size;
void *buf = load_test_file(&size);
for (auto _ : state) {
phash_mem(buf, size, tmp, state.range(), 4);
}
free(buf);
}
static void BM_whash(benchmark::State &state) {
size_t size;
void *buf = load_test_file(&size);
for (auto _ : state) {
whash_mem(buf, size, tmp, state.range(), 0, 0, "haar");
}
free(buf);
}
static void BM_dhash(benchmark::State &state) {
size_t size;
void *buf = load_test_file(&size);
for (auto _ : state) {
dhash_mem(buf, size, tmp, state.range());
}
free(buf);
}
static void BM_ahash(benchmark::State &state) {
size_t size;
void *buf = load_test_file(&size);
for (auto _ : state) {
ahash_mem(buf, size, tmp, state.range());
}
free(buf);
}
static void BM_mhash(benchmark::State &state) {
size_t size;
void *buf = load_test_file(&size);
for (auto _ : state) {
mhash_mem(buf, size, tmp, state.range());
}
free(buf);
}
BENCHMARK(BM_phash)->ArgName("size")->Arg(8);
BENCHMARK(BM_whash)->ArgName("size")->Arg(8);
BENCHMARK(BM_dhash)->ArgName("size")->Arg(8);
BENCHMARK(BM_ahash)->ArgName("size")->Arg(8);
BENCHMARK(BM_mhash)->ArgName("size")->Arg(8);
int main(int argc, char **argv) {
filepath = argv[1];
argv[1] = argv[0];
argc -= 1;
::benchmark::Initialize(&argc, argv + 1);
::benchmark::RunSpecifiedBenchmarks();
}