From: Matthias Goergens <matthias.goergens@citrix.com>
Date: Tue, 2 Feb 2010 16:00:16 +0000 (+0000)
Subject: Adds take and tails to our extended list module
X-Git-Url: http://xenbits.xensource.com/gitweb?a=commitdiff_plain;h=dacca59f516e8aa94b88745e95455fd636fd1cbc;p=xcp%2Fxen-api-libs.git

Adds take and tails to our extended list module

`take' gives the first n elements of a list (or less if the list is
shorter).  `tails' gives a list of all suffixes of a list.

Signed-off-by: Matthias Goergens <matthias.goergens@citrix.com>
---

diff --git a/stdext/listext.ml b/stdext/listext.ml
index e1f7532..29ad0ee 100644
--- a/stdext/listext.ml
+++ b/stdext/listext.ml
@@ -11,6 +11,7 @@
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU Lesser General Public License for more details.
  *)
+open Fun
 module List = struct include List
 
 (** Turn a list into a set *)
@@ -174,4 +175,21 @@ let assoc_default k l d =
 (* Like the Lisp cons *)
 let cons a b = a :: b
 
+(* Could use fold_left to get the same value, but that would necessarily go through the whole list everytime, instead of the first n items, only. *)
+(* ToDo: This is complicated enough to warrant a test. *)
+(* Is it wise to fail silently on negative values?  (They are treated as zero, here.)
+   Pro: Would mask fewer bugs.
+   Con: Less robust.
+*)
+let take n list =
+    let rec helper i acc list =
+	if i <= 0 || list = []
+	then acc
+	else helper (i-1)  (List.hd list :: acc) (List.tl list)
+    in List.rev $ helper n [] list
+
+(* Thanks to sharing we only use linear space. (Roughly double the space needed for the spine of the original list) *)
+let rec tails = function
+    | [] -> [[]]
+    | (_::xs) as l -> l :: tails xs
 end
diff --git a/stdext/listext.mli b/stdext/listext.mli
index b1097b6..c8b4ced 100644
--- a/stdext/listext.mli
+++ b/stdext/listext.mli
@@ -172,4 +172,9 @@ module List :
     (* Like Lisp cons*)
     val cons : 'a -> 'a list -> 'a list
 
+    (* take n list: Return the first n elements of list (or less if list is shorter).*)
+    val take : int -> 'a list -> 'a list
+	
+    val tails : 'a list -> ('a list) list
+
   end