aboutsummaryrefslogtreecommitdiff
path: root/src/bench/bench.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/bench/bench.h')
-rw-r--r--src/bench/bench.h71
1 files changed, 71 insertions, 0 deletions
diff --git a/src/bench/bench.h b/src/bench/bench.h
new file mode 100644
index 0000000000..bf591a2be6
--- /dev/null
+++ b/src/bench/bench.h
@@ -0,0 +1,71 @@
+// Copyright (c) 2015 The Bitcoin Core developers
+// Distributed under the MIT software license, see the accompanying
+// file COPYING or http://www.opensource.org/licenses/mit-license.php.
+#ifndef BITCOIN_BENCH_H
+#define BITCOIN_BENCH_H
+
+// Simple micro-benchmarking framework; API mostly matches a subset of the Google Benchmark
+// framework (see https://github.com/google/benchmark)
+// Wny not use the Google Benchmark framework? Because adding Yet Another Dependency
+// (that uses cmake as its build system and has lots of features we don't need) isn't
+// worth it.
+
+/*
+ * Usage:
+
+static void CODE_TO_TIME(benchmark::State& state)
+{
+ ... do any setup needed...
+ while (state.KeepRunning()) {
+ ... do stuff you want to time...
+ }
+ ... do any cleanup needed...
+}
+
+BENCHMARK(CODE_TO_TIME);
+
+ */
+
+
+#include <boost/function.hpp>
+#include <boost/preprocessor/cat.hpp>
+#include <boost/preprocessor/stringize.hpp>
+#include <map>
+#include <string>
+
+namespace benchmark {
+
+ class State {
+ std::string name;
+ double maxElapsed;
+ double beginTime;
+ double lastTime, minTime, maxTime;
+ int64_t count;
+ int64_t timeCheckCount;
+ public:
+ State(std::string _name, double _maxElapsed) : name(_name), maxElapsed(_maxElapsed), count(0) {
+ minTime = std::numeric_limits<double>::max();
+ maxTime = std::numeric_limits<double>::min();
+ timeCheckCount = 1;
+ }
+ bool KeepRunning();
+ };
+
+ typedef boost::function<void(State&)> BenchFunction;
+
+ class BenchRunner
+ {
+ static std::map<std::string, BenchFunction> benchmarks;
+
+ public:
+ BenchRunner(std::string name, BenchFunction func);
+
+ static void RunAll(double elapsedTimeForOne=1.0);
+ };
+}
+
+// BENCHMARK(foo) expands to: benchmark::BenchRunner bench_11foo("foo", foo);
+#define BENCHMARK(n) \
+ benchmark::BenchRunner BOOST_PP_CAT(bench_, BOOST_PP_CAT(__LINE__, n))(BOOST_PP_STRINGIZE(n), n);
+
+#endif // BITCOIN_BENCH_H