time.hpp

Go to the documentation of this file.

// -------------------------------------------------------//
//
// SHAMROCK code for hydrodynamics
// Copyright (c) 2021-2026 Timothée David--Cléris <tim.shamrock@proton.me>
// SPDX-License-Identifier: CeCILL Free Software License Agreement v2.1
// Shamrock is licensed under the CeCILL 2.1 License, see LICENSE for more information
//
// -------------------------------------------------------//
 
#pragma once

 
#include "shambase/Timer.hpp"
#include "shambase/aliases_float.hpp"
#include <functional>
 
namespace shambase {

    class FunctionTimer {
        f64 acc = 0;
        u32 run_count = 0;
 
        public:
        template<class Func>
        void time_func(Func &&f) {
            Timer t;
            t.start();
            f();
            t.stop();
            acc += t.elapsed_sec();
            run_count += 1;
        }

        inline f64 func_time_sec() { return acc / run_count; }
    };

    template<class Func>
    inline f64 timeit(Func &&f, u32 relaunch = 1) {
 
        FunctionTimer t;
 
        for (u32 i = 0; i < relaunch; i++) {
            t.time_func([&]() {
                f();
            });
        }
 
        return t.func_time_sec();
    }

    template<class Func>
    inline f64 timeitfor(Func &&f, f64 max_duration = 1) {
 
        FunctionTimer t;
        Timer tdur;
        tdur.start();
        do {
            t.time_func([&]() {
                f();
            });
            tdur.stop();
        } while (tdur.elapsed_sec() < max_duration);
 
        return t.func_time_sec();
    }

    struct BenchmarkResult {
        std::vector<f64> counts;

        std::vector<f64> times;
    };

    inline BenchmarkResult benchmark_pow_len(
        std::function<f64(u32)> func, u32 start, u32 end, f64 pow_exp) {
        BenchmarkResult res;
        for (f64 i = start; i < end; i *= pow_exp) {
            res.counts.push_back(i);
            res.times.push_back(func(u32(i)));
        }
 
        return res;
    }
 
} // namespace shambase