benchmark_runner.h

// Copyright 2015 Google Inc. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
#ifndef BENCHMARK_RUNNER_H_
#define BENCHMARK_RUNNER_H_
 
#include <memory>
#include <thread>
#include <vector>
 
#include "benchmark_api_internal.h"
#include "perf_counters.h"
#include "thread_manager.h"
 
namespace benchmark {
 
namespace internal {
 
extern MemoryManager* memory_manager;
extern ProfilerManager* profiler_manager;
 
struct RunResults {
  std::vector<BenchmarkReporter::Run> non_aggregates;
  std::vector<BenchmarkReporter::Run> aggregates_only;
 
  bool display_report_aggregates_only = false;
  bool file_report_aggregates_only = false;
};
 
struct BENCHMARK_EXPORT BenchTimeType {
  enum { UNSPECIFIED, ITERS, TIME } tag;
  union {
    IterationCount iters;
    double time;
  };
};
 
BENCHMARK_EXPORT
BenchTimeType ParseBenchMinTime(const std::string& value);
 
class BenchmarkRunner {
 public:
  BenchmarkRunner(const benchmark::internal::BenchmarkInstance& b_,
                  benchmark::internal::PerfCountersMeasurement* pcm_,
                  BenchmarkReporter::PerFamilyRunReports* reports_for_family);
 
  int GetNumRepeats() const { return repeats; }
 
  bool HasRepeatsRemaining() const {
    return GetNumRepeats() != num_repetitions_done;
  }
 
  void DoOneRepetition();
 
  RunResults&& GetResults();
 
  BenchmarkReporter::PerFamilyRunReports* GetReportsForFamily() const {
    return reports_for_family;
  }
 
  double GetMinTime() const { return min_time; }
 
  bool HasExplicitIters() const { return has_explicit_iteration_count; }
 
  IterationCount GetIters() const { return iters; }
 
 private:
  RunResults run_results;
 
  const benchmark::internal::BenchmarkInstance& b;
  BenchmarkReporter::PerFamilyRunReports* reports_for_family;
 
  BenchTimeType parsed_benchtime_flag;
  const double min_time;
  const double min_warmup_time;
  bool warmup_done;
  const int repeats;
  const bool has_explicit_iteration_count;
 
  int num_repetitions_done = 0;
 
  std::unique_ptr<ThreadRunnerBase> thread_runner;
 
  IterationCount iters;  // preserved between repetitions!
  // So only the first repetition has to find/calculate it,
  // the other repetitions will just use that precomputed iteration count.
 
  PerfCountersMeasurement* const perf_counters_measurement_ptr = nullptr;
 
  struct IterationResults {
    internal::ThreadManager::Result results;
    IterationCount iters;
    double seconds;
  };
  IterationResults DoNIterations();
 
  MemoryManager::Result RunMemoryManager(IterationCount memory_iterations);
 
  void RunProfilerManager(IterationCount profile_iterations);
 
  IterationCount PredictNumItersNeeded(const IterationResults& i) const;
 
  bool ShouldReportIterationResults(const IterationResults& i) const;
 
  double GetMinTimeToApply() const;
 
  void FinishWarmUp(const IterationCount& i);
 
  void RunWarmUp();
};
 
}  // namespace internal
 
}  // end namespace benchmark
 
#endif  // BENCHMARK_RUNNER_H_