48 lines
1.2 KiB
C++
48 lines
1.2 KiB
C++
#include "fces/controller.hpp"
|
|
#include "fces/population.hpp"
|
|
#include <benchmark/benchmark.h>
|
|
|
|
using namespace fces;
|
|
|
|
static void BM_ControllerDecideUpdate(benchmark::State &state) {
|
|
FuzzyController ctrl;
|
|
std::vector<std::vector<float>> stats(state.range(0),
|
|
{0.1f, 0.2f, 0.3f, 0.4f, 0.5f});
|
|
|
|
for (auto _ : state) {
|
|
auto actions = ctrl.decide_update(stats, 0.0f, 0.5f, 0.0f, 0.1f, 0.0f, 0.0f,
|
|
1.0f, 0.0f);
|
|
benchmark::DoNotOptimize(actions);
|
|
}
|
|
}
|
|
BENCHMARK(BM_ControllerDecideUpdate)->Arg(10)->Arg(50)->Arg(200);
|
|
|
|
static void BM_Evolve(benchmark::State &state) {
|
|
Population pop(state.range(0));
|
|
|
|
for (auto _ : state) {
|
|
pop.evolve(2.0f, -0.01f, 0.5f);
|
|
}
|
|
}
|
|
BENCHMARK(BM_Evolve)->Arg(50)->Arg(100)->Arg(200);
|
|
|
|
static void BM_Mutation(benchmark::State &state) {
|
|
FuzzyController ctrl;
|
|
|
|
for (auto _ : state) {
|
|
auto child = ctrl.mutate(2.0f, 1.0f);
|
|
benchmark::DoNotOptimize(child);
|
|
}
|
|
}
|
|
BENCHMARK(BM_Mutation);
|
|
|
|
static void BM_Crossover(benchmark::State &state) {
|
|
FuzzyController a, b;
|
|
|
|
for (auto _ : state) {
|
|
auto child = a.crossover(b);
|
|
benchmark::DoNotOptimize(child);
|
|
}
|
|
}
|
|
BENCHMARK(BM_Crossover);
|