From b992794a77418729ba8f3b470eaa522fa8e0d7b4 Mon Sep 17 00:00:00 2001
From: Guillaume Bury <guillaume.bury@gmail.com>
Date: Mon, 17 Nov 2014 15:48:41 +0100
Subject: [PATCH] Added diff computing in bench_stats

---
 Makefile            |  2 +-
 util/bench_stats.ml | 81 +++++++++++++++++++++++++++++++++++++++++----
 2 files changed, 75 insertions(+), 8 deletions(-)

diff --git a/Makefile b/Makefile
index 96a311ca..ab950931 100644
--- a/Makefile
+++ b/Makefile
@@ -28,7 +28,7 @@ $(TEST): $(LIB)
 bench: $(TEST)
 	cd bench && $(MAKE)
 
-stats:
+stats: $(TEST)
 	./bench_stats.native `git rev-parse HEAD`
 
 log:
diff --git a/util/bench_stats.ml b/util/bench_stats.ml
index 20b5e2cc..6d10db05 100644
--- a/util/bench_stats.ml
+++ b/util/bench_stats.ml
@@ -7,6 +7,19 @@ Copyright 2014 Simon Cruanes
 exception Commit_not_found
 exception Commit_ambiguous
 
+(* Arguments parsing *)
+let usage = "Usage : ./bench_stats [options] commit1 [commit2]"
+
+let arg_commit = ref []
+let anon s = arg_commit := s :: !arg_commit
+
+let long_diff = ref false
+let args = Arg.align [
+    "-long", Arg.Set long_diff,
+    " Print a long diff instead of a short one";
+]
+
+(* Access functions *)
 let get_commit s =
     let d = Filename.concat (Filename.dirname Sys.argv.(0)) "bench" in
     match Parselog.complete (Filename.concat d "logs") s with
@@ -14,6 +27,8 @@ let get_commit s =
     | [] -> raise Commit_not_found
     | _ -> raise Commit_ambiguous
 
+
+(* Stats computing *)
 type commit_stats = {
     nb_sat : int;
     nb_unsat : int;
@@ -57,6 +72,45 @@ let get_stats h =
         avg_time = !avg;
     }
 
+(* Diff computing *)
+type diff = {
+    sat_limit : string list;
+    sat_nopb : string list;
+    unsat_limit : string list;
+    unsat_nopb : string list;
+    disagree : (Parselog.pb * Parselog.pb) list;
+}
+
+let empty_diff = {
+    sat_limit = [];
+    sat_nopb = [];
+    unsat_limit = [];
+    unsat_nopb = [];
+    disagree = [];
+}
+
+let diff h h' =
+    let aux f pb acc =
+        try
+            let pb' = Hashtbl.find h' f in
+            Parselog.(match pb.pb_st, pb'.pb_st with
+            | Sat, Timeout | Sat, Spaceout ->
+                    { acc with sat_limit = f :: acc.sat_limit }
+            | Unsat, Timeout | Unsat, Spaceout ->
+                    { acc with unsat_limit = f :: acc.unsat_limit }
+            | Sat, Unsat | Unsat, Sat ->
+                    { acc with disagree = (pb, pb') :: acc.disagree }
+            | _, _ -> acc
+            )
+        with Not_found ->
+            match Parselog.(pb.pb_st) with
+            | Parselog.Sat -> { acc with sat_nopb = f :: acc.sat_nopb }
+            | Parselog.Unsat -> { acc with unsat_nopb = f :: acc.unsat_nopb }
+            | _ -> acc
+    in
+    Hashtbl.fold aux h empty_diff
+
+(* Printing *)
 let print_stats s =
     let sha, h = get_commit s in
     let st = get_stats h in
@@ -64,15 +118,28 @@ let print_stats s =
     st.avg_time (st.nb_sat + st.nb_unsat)
     (st.nb_sat + st.nb_unsat + st.nb_timeout + st.nb_spaceout) st.nb_timeout st.nb_spaceout
 
-let print_diff _ _ = ()
+let print_diff_short s1 s2 =
+    let sha1, h1 = get_commit s1 in
+    let sha2, h2 = get_commit s2 in
+    let only1 = diff h1 h2 in
+    let only2 = diff h2 h1 in
+    Format.printf "1:%s -> 2:%s@\n" sha1 sha2;
+    Format.printf "Lost (only in 1) : %d (+ %d not in pb list)@\n"
+    (List.length only1.sat_limit + List.length only1.unsat_limit) (List.length only1.sat_nopb + List.length only1.unsat_nopb);
+    Format.printf "Won  (only in 2) : %d (+ %d not in pb list)@\n"
+    (List.length only2.sat_limit + List.length only2.unsat_limit) (List.length only2.sat_nopb + List.length only2.unsat_nopb);
+    match List.sort_uniq Pervasives.compare (List.rev_append only1.disagree only2.disagree) with
+    | [] -> ()
+    | l -> Format.printf "WARNING : %d incoherence@\n" (List.length l)
 
+(* Main function *)
 let main () =
-    if Array.length Sys.argv = 2 then
-        print_stats Sys.argv.(1)
-    else if Array.length Sys.argv = 3 then
-        print_diff Sys.argv.(1) Sys.argv.(2)
-    else
-        Format.fprintf Format.std_formatter "Don't know what to do !@."
+    Arg.parse args anon usage;
+    (* Warning : the 'anon' function reverses the order of the arguments *)
+    match !arg_commit with
+    | [c] -> print_stats c
+    | [c1; c2] -> print_diff_short c2 c1
+    | _ -> Arg.usage args usage
 ;;
 
 try