From 0ad73a2cffd44654559536391c860d74ba6d5b21 Mon Sep 17 00:00:00 2001
From: Simon Cruanes <simon.cruanes.2007@m4x.org>
Date: Thu, 13 Nov 2014 13:06:34 +0100
Subject: [PATCH] more bechmarks on lists

---
 benchs/benchs.ml | 69 +++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 68 insertions(+), 1 deletion(-)

diff --git a/benchs/benchs.ml b/benchs/benchs.ml
index 63d26ef9..022e3004 100644
--- a/benchs/benchs.ml
+++ b/benchs/benchs.ml
@@ -6,6 +6,8 @@ let draw_line () =
 
 module L = struct
 
+  (* FLAT MAP *)
+
   let f_ x =
     if x mod 10 = 0 then []
     else if x mod 5 = 1 then [x;x+1]
@@ -22,17 +24,82 @@ module L = struct
       ] in
     Benchmark.tabulate res
 
+  (* APPEND *)
+
+  let append_ f (l1, l2, l3) =
+    ignore (f (f l1 l2) l3)
+
+  let bench_append ?(time=2) n =
+    draw_line ();
+    Printf.printf "append for %d elements\n" n;
+    let l1 = CCList.(1 -- n) in
+    let l2 = CCList.(n+1 -- 2*n) in
+    let l3 = CCList.(2*n+1 -- 3*n) in
+    let arg = l1, l2, l3 in
+    let res = Benchmark.throughputN time
+      [ "CCList.append", append_ CCList.append, arg
+      ; "List.append", append_ List.append, arg
+      ] in
+    Benchmark.tabulate res
+
+  (* FLATTEN *)
+
+  let bench_flatten ?(time=2) n =
+    draw_line ();
+    Printf.printf "flatten for %d elements\n" n;
+    let l = CCList.Idx.mapi (fun i x -> CCList.(x -- (x+ min i 100))) CCList.(1 -- n) in
+    let res = Benchmark.throughputN time
+      [ "CCList.flatten", CCList.flatten, l
+      ; "List.flatten", List.flatten, l
+      ; "fold_right append", (fun l -> List.fold_right List.append l []), l
+      ; "CCList.(fold_right append)", (fun l->CCList.fold_right CCList.append l []), l
+      ] in
+    Benchmark.tabulate res
+
+
+  (* MAIN *)
+
   let run() =
+    bench_flatten 100;
+    bench_flatten 10_000;
+    bench_flatten ~time:4 100_000;
     bench_flatmap 100;
     bench_flatmap 10_000;
     bench_flatmap ~time:4 100_000;
+    bench_append 100;
+    bench_append 10_000;
+    bench_append ~time:4 100_000;
     ()
 end
 
 (* TODO *)
 
+let tbl_ =
+  [ "list", L.run
+  ]
+
+let bench_all () =
+  List.iter (fun (name, run) ->
+    draw_line ();
+    Printf.printf "run tests for %s...\n" name;
+    run()
+  ) tbl_
+
+let which_ = ref ("all", bench_all)
+let set_which s =
+  if s = "all" then which_ := s, bench_all
+  else try
+    let run = List.assoc s tbl_ in
+    which_ := s, run
+  with Not_found ->
+    failwith ("unknown test " ^ s)
+let options = []
+
 let () =
-  L.run ();
+  Arg.parse options set_which "benchs [which]";
+  let name, run = !which_ in
+  Printf.printf "run test %s\n" name;
+  run ();
   ()