mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-02-05 10:17:30 -05:00
306ccd4927
-BEGIN VERIFY SCRIPT- ./contrib/devtools/copyright_header.py update ./ -END VERIFY SCRIPT- Commits of previous years: - 2021:f47dda2c58- 2020:fa0074e2d8- 2019:aaaaad6ac9
82 lines
1.9 KiB
C++
82 lines
1.9 KiB
C++
// Copyright (c) 2015-2022 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_BENCH_H
|
|
#define BITCOIN_BENCH_BENCH_H
|
|
|
|
#include <fs.h>
|
|
#include <util/macros.h>
|
|
|
|
#include <chrono>
|
|
#include <functional>
|
|
#include <map>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <bench/nanobench.h>
|
|
|
|
/*
|
|
* Usage:
|
|
|
|
static void NameOfYourBenchmarkFunction(benchmark::Bench& bench)
|
|
{
|
|
...do any setup needed...
|
|
|
|
bench.run([&] {
|
|
...do stuff you want to time; refer to src/bench/nanobench.h
|
|
for more information and the options that can be passed here...
|
|
});
|
|
|
|
...do any cleanup needed...
|
|
}
|
|
|
|
BENCHMARK(NameOfYourBenchmarkFunction);
|
|
|
|
*/
|
|
|
|
namespace benchmark {
|
|
|
|
using ankerl::nanobench::Bench;
|
|
|
|
typedef std::function<void(Bench&)> BenchFunction;
|
|
|
|
enum PriorityLevel : uint8_t
|
|
{
|
|
LOW = 1 << 0,
|
|
HIGH = 1 << 2,
|
|
};
|
|
|
|
// List priority labels, comma-separated and sorted by increasing priority
|
|
std::string ListPriorities();
|
|
uint8_t StringToPriority(const std::string& str);
|
|
|
|
struct Args {
|
|
bool is_list_only;
|
|
bool sanity_check;
|
|
std::chrono::milliseconds min_time;
|
|
std::vector<double> asymptote;
|
|
fs::path output_csv;
|
|
fs::path output_json;
|
|
std::string regex_filter;
|
|
uint8_t priority;
|
|
};
|
|
|
|
class BenchRunner
|
|
{
|
|
// maps from "name" -> (function, priority_level)
|
|
typedef std::map<std::string, std::pair<BenchFunction, PriorityLevel>> BenchmarkMap;
|
|
static BenchmarkMap& benchmarks();
|
|
|
|
public:
|
|
BenchRunner(std::string name, BenchFunction func, PriorityLevel level);
|
|
|
|
static void RunAll(const Args& args);
|
|
};
|
|
} // namespace benchmark
|
|
|
|
// BENCHMARK(foo) expands to: benchmark::BenchRunner bench_11foo("foo", foo, priority_level);
|
|
#define BENCHMARK(n, priority_level) \
|
|
benchmark::BenchRunner PASTE2(bench_, PASTE2(__LINE__, n))(STRINGIZE(n), n, priority_level);
|
|
|
|
#endif // BITCOIN_BENCH_BENCH_H
|