simon/ocaml-containers
1
0
Fork 0
mirror of https://github.com/c-cube/ocaml-containers.git synced 2025-12-06 19:25:28 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

bugfix in CCGraph.Dot.pp

Browse source
This commit is contained in:
Simon Cruanes 2015-06-10 15:11:09 +02:00
parent 02088a6dd7
commit 572f45560b
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/data/CCGraph.ml
View file

@ -375,19 +375,21 @@ module Dot = struct
| `Label l -> Format.fprintf out "label=\"%s\"" l
| `Other (name, value) -> Format.fprintf out "%s=\"%s\"" name value
(* map from vertices to integers *)
and get_id =
and get_node =
let count = ref 0 in
fun v ->
try (tbl.find v).id
try tbl.find v
with Not_found ->
let n = !count in
let node = {id= !count; explored=false} in
incr count;
tbl.add v {explored=false; id=n};
n
tbl.add v node;
node
and vertex_explored v =
try (tbl.find v).explored
with Not_found -> false
in
let set_explored v = (get_node v).explored <- true
and get_id v = (get_node v).id in
(* the unique name of a vertex *)
let pp_vertex out v = Format.fprintf out "vertex_%d" (get_id v) in
(* print preamble *)
@ -395,7 +397,7 @@ module Dot = struct
(* traverse *)
let tags = {
get_tag=vertex_explored;
set_tag=(fun v -> ignore (get_id v)); (* allocate new ID *)
set_tag=set_explored; (* allocate new ID *)
} in
let events = Traverse.Event.dfs_tag ~tags ~graph seq in
Seq.iter
@ -445,6 +447,6 @@ let divisors_graph = {
divisors (j+1) i yield
)
in
divisors 2 i
divisors 1 i
);
}