github-repos/benchmark
1
0
Fork 0
mirror of https://github.com/google/benchmark.git synced 2025-01-30 05:40:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

add list test internals

Browse Source
This commit is contained in:
Eric Fiselier 2015-03-30 23:32:37 -04:00
parent 805145847b
commit 12f1c5f471
1 changed files with 29 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
src/benchmark.cc
View File

@ -37,6 +37,10 @@
#include "sysinfo.h"
#include "walltime.h"
DEFINE_bool(benchmark_list_tests, false,
"Print a list of benchmarks. This option overrides all other "
"options.");
DEFINE_string(benchmark_filter, ".",
"A regular expression that specifies the set of benchmarks "
"to execute. If this flag is empty, no benchmarks are run. "
@ -370,7 +374,7 @@ bool BenchmarkFamilies::FindBenchmarks(
MutexLock l(mutex_);
for (BenchmarkImp* family : families_) {
// Family was deleted or benchmark doesn't match
if (family == nullptr || !re.Match(family->name_)) continue;
if (family == nullptr) continue;
if (family->arg_count_ == -1) {
family->arg_count_ = 0;
@ -414,7 +418,8 @@ bool BenchmarkFamilies::FindBenchmarks(
instance.name += StringPrintF("/threads:%d", instance.threads);
}
benchmarks->push_back(instance);
if (re.Match(instance.name))
benchmarks->push_back(instance);
}
}
}
@ -766,19 +771,29 @@ void State::SetLabel(const char* label) {
}
namespace internal {
namespace {
void PrintBenchmarkList() {
std::vector<Benchmark::Instance> benchmarks;
auto families = BenchmarkFamilies::GetInstance();
if (!families->FindBenchmarks(".", &benchmarks)) return;
for (const internal::Benchmark::Instance& benchmark : benchmarks)
std::cout << benchmark.name << "\n";
}
void RunMatchingBenchmarks(const std::string& spec,
BenchmarkReporter* reporter) {
CHECK(reporter != nullptr);
if (spec.empty()) return;
std::vector<benchmark::internal::Benchmark::Instance> benchmarks;
auto families = benchmark::internal::BenchmarkFamilies::GetInstance();
std::vector<Benchmark::Instance> benchmarks;
auto families = BenchmarkFamilies::GetInstance();
if (!families->FindBenchmarks(spec, &benchmarks)) return;
// Determine the width of the name field using a minimum width of 10.
size_t name_field_width = 10;
for (const internal::Benchmark::Instance& benchmark : benchmarks) {
for (const Benchmark::Instance& benchmark : benchmarks) {
name_field_width =
std::max<size_t>(name_field_width, benchmark.name.size());
}
@ -814,6 +829,7 @@ std::unique_ptr<BenchmarkReporter> GetDefaultReporter() {
}
}
} // end namespace
} // end namespace internal
void RunSpecifiedBenchmarks() {
@ -821,6 +837,10 @@ void RunSpecifiedBenchmarks() {
}
void RunSpecifiedBenchmarks(BenchmarkReporter* reporter) {
if (FLAGS_benchmark_list_tests) {
internal::PrintBenchmarkList();
return;
}
std::string spec = FLAGS_benchmark_filter;
if (spec.empty() || spec == "all")
spec = "."; // Regexp that matches all benchmarks
@ -839,7 +859,8 @@ namespace internal {
void PrintUsageAndExit() {
fprintf(stdout,
"benchmark"
" [--benchmark_filter=<regex>]\n"
" [--benchmark_list_tests={true|false}]\n"
" [--benchmark_filter=<regex>]\n"
" [--benchmark_min_time=<min_time>]\n"
" [--benchmark_repetitions=<num_repetitions>]\n"
" [--benchmark_format=<tabular|json|csv>]\n"
@ -852,6 +873,8 @@ void ParseCommandLineFlags(int* argc, const char** argv) {
using namespace benchmark;
for (int i = 1; i < *argc; ++i) {
if (
ParseBoolFlag(argv[i], "benchmark_list_tests",
&FLAGS_benchmark_list_tests) ||
ParseStringFlag(argv[i], "benchmark_filter",
&FLAGS_benchmark_filter) ||
ParseDoubleFlag(argv[i], "benchmark_min_time",