ReCo.jl/src/benchmark.jl

using Statistics: mean
using Dates: now
using BenchmarkTools: @benchmark
using JSON3: JSON3

using ReCo

function run_benchmarks(
    dir::String="";
    n_particles::Int64=1000,
    v::Float64=20.0,
    duration::Float64=2.0,
    n_bundle_snapshots::Int64=0,
    comment="",
)
    if length(dir) == 0
        dir = init_sim(; n_particles=n_particles, v=v, parent_dir="benchmark")
    end

    benchmark = @benchmark run_sim(
        $dir; duration=$duration, n_bundle_snapshots=$n_bundle_snapshots
    )

    display(benchmark)

    open("exports/benchmark.txt", "a+") do f
        JSON3.pretty(
            f,
            Dict(
                "n_particles" => n_particles,
                "v" => v,
                "duration" => duration,
                "n_bundle_snapshots" => n_bundle_snapshots,
                "comment" => comment,
                "datetime" => now(),
                "mean_time/ns" => mean(benchmark.times),
                "allocs" => benchmark.allocs,
                "memory" => benchmark.memory,
            ),
        )
        write(f, "\n")
    end

    return dir
end