module S = Iter
open Iter.Infix

[@@@ocaml.warning "-5"]

let small = [ 10; 20; 50; 100; 500 ]
let medium = small @ [ 1000; 10_000; 100_000 ]
let big = medium @ [ 500_000; 1_000_000; 2_000_000 ]
let bench_fold n = 0 -- n |> S.fold ( + ) 0 |> ignore

let bench_flatmap n =
  0 -- n |> S.flat_map (fun i -> i -- (i + 5)) |> fun _ -> ()

let bench_product n = S.product (0 -- n) (0 -- n) (fun _ -> ())

let _ =
  List.iter
    (fun (name, bench, sizes) ->
      Format.printf "-------------------------------------------------------@.";
      Format.printf "bench %s@." name;
      List.iter
        (fun n ->
          let name = name ^ " on " ^ string_of_int n in
          let res = Benchmark.throughput1 2 ~name bench n in
          Benchmark.tabulate res)
        sizes)
    [
      "fold", bench_fold, big;
      "flatmap", bench_flatmap, medium;
      "product", bench_product, small;
    ];
  ()