INCLUDE "utils.ml"
open Format
module M = Map.Make(struct type t = int let compare = compare end)
type info = { sat : StateSet.t;
todo : StateSet.t;
msg : string;
}
let info = Hashtbl.create 2017
let final = Hashtbl.create 2017
let max_round = ref 0
let buff = Buffer.create 20
let fmt = formatter_of_buffer buff
let trace ?(msg="") nid r t d =
if r > !max_round then max_round := r;
let m = try Hashtbl.find info nid with Not_found -> M.empty in
let () = pp_print_flush fmt () in
let _ = fprintf fmt
"node: %i %s todo: %a sat: %a _______________________ "
nid msg StateSet.print t StateSet.print d
in
let () = pp_print_flush fmt () in
let msg = Buffer.contents buff in
let () = Buffer.clear buff in
let old_inf = try M.find r m with Not_found -> [] in
let m' = M.add r ({ sat = d; todo = t; msg = msg }::old_inf) m in
Hashtbl.replace info nid m'
let finalize_node n r b =
Hashtbl.replace final n (b,r)
module K =
struct
type t = int * StateSet.t * StateSet.t
let hash (a,b,c) = HASHINT3(a, (b.StateSet.id :> int), (c.StateSet.id :> int))
let equal ((a1,b1,c1) as x) ((a2,b2,c2) as y) =
x == y || (a1 == a2 && b1 == b2 && c1 == c2)
end
module CTable = Hashtbl.Make (K)
let ctable = CTable.create 20
let rgb x =
let h = K.hash x in
let r = h land 0xff
and g = (h lsr 8) land 0xff
and b = (h lsr 16) land 0xff
in
r, g, b
let color ((a,b,c) as x) =
try
CTable.find ctable x
with
Not_found ->
let r,g,b = rgb x in
let s = "rgb(" ^ (string_of_int r) ^ ","
^ (string_of_int g) ^ ","
^ (string_of_int b) ^ ")"
in
CTable.add ctable x s;
s
let text_color x =
let r,g,b = rgb x in
let av = (r + g + b) / 3 in
if av > 128 then "rgb(0,0,0)"
else "rgb(255,255,255)"
let gen_trace (type s) = fun auto t tree ->
let module T = (val (t) : Tree.S with type t = s) in
let root = T.root tree in
let rec loop osvg ohtml node parent x y =
if node != T.nil then begin
let m =
try
Hashtbl.find info (T.preorder tree node)
with Not_found -> M.empty
in
let node_id = T.preorder tree node in
let marked, last_round = try Hashtbl.find final node_id with Not_found ->
Printf.eprintf ">>> %i\n%!" node_id; false, !max_round;
in
let scolor, tcolor =
let { sat ; todo; _ } =
match M.find last_round m with
[] -> { sat = StateSet.empty; todo= StateSet.empty; msg = "ERROR" }
| [ e ] -> e
| l -> List.hd (List.tl (List.rev l))
in
let c = (last_round, StateSet.union sat todo, StateSet.empty) in
color c, text_color c
in
let tag = QName.to_string (T.tag tree node) in
let lbox = (String.length tag + 2) * 10 in
let s_node = "node" ^ (string_of_int node_id) in
fprintf osvg
"\n%!"
s_node
s_node
x y
lbox
scolor
(if marked
then ";stroke-width:4"
else ";stroke-width:2;stroke-dasharray:2,2");
fprintf osvg "%s\n"
(x+10)
(y+15)
tcolor s_node tag;
fprintf ohtml "data['%s'] = new Array();\n" s_node;
M.iter
(fun i l ->
let msg = String.concat "" (List.rev_map (fun x -> x.msg) l) in
fprintf ohtml "data['%s'][%i] = '%s';\n" s_node i msg)
m;
let first = T.first_child tree node in
let maxw1, maxy1 = loop osvg ohtml first node x (y + 40) in
let next = T.next_sibling tree node in
let x_next = max (x+lbox) (maxw1+10) in
if node != root then begin
if node == T.first_child tree parent then
fprintf osvg "\n"
(x + lbox / 2) (y-20) (x + lbox / 2) (y);
if next != T.nil then
fprintf osvg "\n"
(x + lbox) (y+10) x_next (y+10);
end;
let maxw2, maxy2 = loop osvg ohtml next node x_next y in
maxw2, max maxy1 maxy2
end
else x, y
in
ignore (Sys.command "mkdir -p tests/trace");
let osvg_ = open_out "tests/trace/trace.svg" in
let ohtml_ = open_out "tests/trace/trace.html" in
let osvg = formatter_of_out_channel osvg_ in
let ohtml = formatter_of_out_channel ohtml_ in
fprintf ohtml "\
\
\
\
\
%a
\
\n
\n
\n%!";
pp_print_flush ohtml ();
close_out ohtml_;
close_in fi