/* SPDX-License-Identifier: GPL-2.0 */
/* Copyright (c) 2026 Meta Platforms, Inc. and affiliates. */

#ifndef __BENCH_BPF_TIMING_H__
#define __BENCH_BPF_TIMING_H__

#include <stdbool.h>
#include <linux/types.h>
#include "bench.h"

#ifndef BENCH_NR_SAMPLES
#define BENCH_NR_SAMPLES        4096
#endif
#ifndef BENCH_NR_CPUS
#define BENCH_NR_CPUS           256
#endif

typedef void (*bpf_bench_run_fn)(void *ctx);

struct bpf_bench_timing {
        __u64 (*samples)[BENCH_NR_SAMPLES];     /* skel->bss->timing_samples */
        __u32 *idx;                             /* skel->bss->timing_idx */
        volatile __u32 *timing_enabled;         /* &skel->bss->timing_enabled */
        volatile __u32 *batch_iters_bss;        /* &skel->bss->batch_iters */
        __u32 batch_iters;
        __u32 target_samples;
        __u32 nr_cpus;
        int warmup_ticks;
        bool done;
        bool machine_readable;
};

#define BENCH_TIMING_INIT(t, skel, iters) do {                          \
        (t)->samples = (skel)->bss->timing_samples;                     \
        (t)->idx = (skel)->bss->timing_idx;                             \
        (t)->timing_enabled = &(skel)->bss->timing_enabled;             \
        (t)->batch_iters_bss = &(skel)->bss->batch_iters;               \
        (t)->batch_iters = (iters);                                     \
        (t)->target_samples = 200;                                      \
        (t)->nr_cpus = env.nr_cpus;                                     \
        (t)->warmup_ticks = 0;                                          \
        (t)->done = false;                                              \
        (t)->machine_readable = false;                                  \
} while (0)

void bpf_bench_timing_measure(struct bpf_bench_timing *t, struct bench_res *res);
void bpf_bench_timing_report(struct bpf_bench_timing *t, const char *name, const char *desc);
void bpf_bench_calibrate(struct bpf_bench_timing *t, bpf_bench_run_fn run_fn, void *ctx);

#endif /* __BENCH_BPF_TIMING_H__ */