From 1644640075ddd8947a710a25e39073e740069eda Mon Sep 17 00:00:00 2001
From: Simon Cruanes <simon.cruanes.2007@m4x.org>
Date: Sun, 31 Mar 2013 22:23:26 +0200
Subject: [PATCH] LazyGraph.limit_depth implemented

---
 lazyGraph.ml  | 22 ++++++++++++++++++++++
 lazyGraph.mli |  5 +++++
 2 files changed, 27 insertions(+)

diff --git a/lazyGraph.ml b/lazyGraph.ml
index c54f86d6..eb4d25f5 100644
--- a/lazyGraph.ml
+++ b/lazyGraph.ml
@@ -458,6 +458,28 @@ let rev_path p =
   | (v,e,v')::p' -> rev ((v',e,v)::acc) p'
   in rev [] p
 
+(** [limit_depth g depth start] returns the same graph as [graph], but
+    keeping only nodes that are at distance at most [depth] from
+    some vertex in [start] (which must be finite). *)
+let limit_depth g depth start =
+  assert (depth >= 0);
+  (* compute set of vertices which are within the required distance *)
+  let set = mk_map ~eq:g.eq ~hash:g.hash in
+  let open Gen.Infix in
+  Full.bfs_full g start
+    |> Gen.takeWhile
+      (function
+        | Full.EnterVertex (id, _, _, path) -> List.length path <= depth
+        | _ -> true)
+    |> Gen.iter
+      (function
+        | Full.EnterVertex (id, _, _, _) -> set.map_add id ()
+        | _ -> ());
+  let force v =
+    if not (set.map_mem v) then Empty
+    else g.force v
+  in {g with force=force; }
+
 (** {2 Lazy transformations} *)
 
 let union ?(combine=fun x y -> x) g1 g2 =
diff --git a/lazyGraph.mli b/lazyGraph.mli
index 21ebada1..f52bb713 100644
--- a/lazyGraph.mli
+++ b/lazyGraph.mli
@@ -167,6 +167,11 @@ val is_dag_full : ('id, _, _) t -> 'id Gen.t -> bool
 val rev_path : ('id, 'e) path -> ('id, 'e) path
   (** Reverse the path *)
 
+val limit_depth : ('id, 'v, 'e) t -> int -> 'id Gen.t -> ('id, 'v, 'e) t
+  (** [limit_depth g depth start] returns the same graph as [graph], but
+      keeping only nodes that are at distance at most [depth] from
+      some vertex in [start] (which must be finite). *)
+
 (** {2 Lazy transformations} *)
 
 val union : ?combine:('v -> 'v -> 'v) ->