Bitcoin Core  0.20.99
P2P Digital Currency
bench.h
Go to the documentation of this file.
1 /**********************************************************************
2  * Copyright (c) 2014 Pieter Wuille *
3  * Distributed under the MIT software license, see the accompanying *
4  * file COPYING or http://www.opensource.org/licenses/mit-license.php.*
5  **********************************************************************/
6 
7 #ifndef SECP256K1_BENCH_H
8 #define SECP256K1_BENCH_H
9 
10 #include <stdint.h>
11 #include <stdio.h>
12 #include <string.h>
13 #include "sys/time.h"
14 
15 static int64_t gettime_i64(void) {
16  struct timeval tv;
17  gettimeofday(&tv, NULL);
18  return (int64_t)tv.tv_usec + (int64_t)tv.tv_sec * 1000000LL;
19 }
20 
21 #define FP_EXP (6)
22 #define FP_MULT (1000000LL)
23 
24 /* Format fixed point number. */
25 void print_number(const int64_t x) {
26  int64_t x_abs, y;
27  int c, i, rounding;
28  size_t ptr;
29  char buffer[30];
30 
31  if (x == INT64_MIN) {
32  /* Prevent UB. */
33  printf("ERR");
34  return;
35  }
36  x_abs = x < 0 ? -x : x;
37 
38  /* Determine how many decimals we want to show (more than FP_EXP makes no
39  * sense). */
40  y = x_abs;
41  c = 0;
42  while (y > 0LL && y < 100LL * FP_MULT && c < FP_EXP) {
43  y *= 10LL;
44  c++;
45  }
46 
47  /* Round to 'c' decimals. */
48  y = x_abs;
49  rounding = 0;
50  for (i = c; i < FP_EXP; ++i) {
51  rounding = (y % 10) >= 5;
52  y /= 10;
53  }
54  y += rounding;
55 
56  /* Format and print the number. */
57  ptr = sizeof(buffer) - 1;
58  buffer[ptr] = 0;
59  if (c != 0) {
60  for (i = 0; i < c; ++i) {
61  buffer[--ptr] = '0' + (y % 10);
62  y /= 10;
63  }
64  buffer[--ptr] = '.';
65  }
66  do {
67  buffer[--ptr] = '0' + (y % 10);
68  y /= 10;
69  } while (y != 0);
70  if (x < 0) {
71  buffer[--ptr] = '-';
72  }
73  printf("%s", &buffer[ptr]);
74 }
75 
76 void run_benchmark(char *name, void (*benchmark)(void*, int), void (*setup)(void*), void (*teardown)(void*, int), void* data, int count, int iter) {
77  int i;
78  int64_t min = INT64_MAX;
79  int64_t sum = 0;
80  int64_t max = 0;
81  for (i = 0; i < count; i++) {
82  int64_t begin, total;
83  if (setup != NULL) {
84  setup(data);
85  }
86  begin = gettime_i64();
87  benchmark(data, iter);
88  total = gettime_i64() - begin;
89  if (teardown != NULL) {
90  teardown(data, iter);
91  }
92  if (total < min) {
93  min = total;
94  }
95  if (total > max) {
96  max = total;
97  }
98  sum += total;
99  }
100  printf("%s: min ", name);
101  print_number(min * FP_MULT / iter);
102  printf("us / avg ");
103  print_number(((sum * FP_MULT) / count) / iter);
104  printf("us / max ");
105  print_number(max * FP_MULT / iter);
106  printf("us\n");
107 }
108 
109 int have_flag(int argc, char** argv, char *flag) {
110  char** argm = argv + argc;
111  argv++;
112  if (argv == argm) {
113  return 1;
114  }
115  while (argv != NULL && argv != argm) {
116  if (strcmp(*argv, flag) == 0) {
117  return 1;
118  }
119  argv++;
120  }
121  return 0;
122 }
123 
124 int get_iters(int default_iters) {
125  char* env = getenv("SECP256K1_BENCH_ITERS");
126  if (env) {
127  return strtol(env, NULL, 0);
128  } else {
129  return default_iters;
130  }
131 }
132 
133 #endif /* SECP256K1_BENCH_H */
void run_benchmark(char *name, void(*benchmark)(void *, int), void(*setup)(void *), void(*teardown)(void *, int), void *data, int count, int iter)
Definition: bench.h:76
volatile double sum
Definition: examples.cpp:22
void print_number(const int64_t x)
Definition: bench.h:25
const char * name
Definition: rest.cpp:41
static int64_t gettime_i64(void)
Definition: bench.h:15
#define FP_MULT
Definition: bench.h:22
int get_iters(int default_iters)
Definition: bench.h:124
static int count
Definition: tests.c:35
#define FP_EXP
Definition: bench.h:21
void printf(const char *fmt, const Args &... args)
Format list of arguments to std::cout, according to the given format string.
Definition: tinyformat.h:1079
int have_flag(int argc, char **argv, char *flag)
Definition: bench.h:109