diff options
| author | letouzey | 2007-07-11 21:48:04 +0000 |
|---|---|---|
| committer | letouzey | 2007-07-11 21:48:04 +0000 |
| commit | 5f2b3fd5c17c29ffc734eef05bdb22b44d015edf (patch) | |
| tree | 9b03e7a9800355412e364e0528c5214b622d4888 /lib/util.ml | |
| parent | 2ed747a81ed14d91112b9b3360c6e5ab4ff897eb (diff) | |
Slight cleanup of refl_omega.ml : in particular it uses now list
utilities from Util. Some additions in Util, and simplifications
in various files.
git-svn-id: svn+ssh://scm.gforge.inria.fr/svn/coq/trunk@9969 85f007b7-540e-0410-9357-904b9bb8a0f7
Diffstat (limited to 'lib/util.ml')
| -rw-r--r-- | lib/util.ml | 39 |
1 files changed, 32 insertions, 7 deletions
diff --git a/lib/util.ml b/lib/util.ml index 590d649931..7e59d1474c 100644 --- a/lib/util.ml +++ b/lib/util.ml @@ -216,6 +216,8 @@ let list_index x = in index_x 1 +let list_index0 x l = list_index x l - 1 + let list_unique_index x = let rec index_x n = function | y::l -> @@ -329,6 +331,18 @@ let rec list_distinct l = | [] -> true in loop l +let rec list_merge_uniq cmp l1 l2 = + match l1, l2 with + | [], l2 -> l2 + | l1, [] -> l1 + | h1 :: t1, h2 :: t2 -> + let c = cmp h1 h2 in + if c = 0 + then h1 :: list_merge_uniq cmp t1 t2 + else if c <= 0 + then h1 :: list_merge_uniq cmp t1 l2 + else h2 :: list_merge_uniq cmp l1 t2 + let rec list_duplicates = function | [] -> [] | x::l -> @@ -389,6 +403,9 @@ let rec list_skipn n l = match n,l with | _, [] -> failwith "list_fromn" | n, _::l -> list_skipn (pred n) l +let rec list_addn n x l = + if n = 0 then l else x :: (list_addn (pred n) x l) + let list_prefix_of prefl l = let rec prefrec = function | (h1::t1, h2::t2) -> h1 = h2 && prefrec (t1,t2) @@ -410,6 +427,7 @@ let list_drop_prefix p l = | None -> l let list_map_append f l = List.flatten (List.map f l) +let list_join_map = list_map_append (* Alias *) let list_map_append2 f l1 l2 = List.flatten (List.map2 f l1 l2) @@ -420,8 +438,6 @@ let list_share_tails l1 l2 = in shr_rev [] (List.rev l1, List.rev l2) -let list_join_map f l = List.flatten (List.map f l) - let rec list_fold_map f e = function | [] -> (e,[]) | h::t -> @@ -445,13 +461,22 @@ let list_fold_map' f l e = let list_map_assoc f = List.map (fun (x,a) -> (x,f a)) +(* A generic cartesian product: for any operator (**), + [list_cartesian (**) [x1;x2] [y1;y2] = [x1**y1; x1**y2; x2**y1; x2**y1]], + and so on if there are more elements in the lists. *) + +let rec list_cartesian op l1 l2 = + list_map_append (fun x -> List.map (op x) l2) l1 + +(* [list_cartesians] is an n-ary cartesian product: it iterates + [list_cartesian] over a list of lists. *) + +let list_cartesians op init ll = + List.fold_right (list_cartesian op) ll [init] + (* list_combinations [[a;b];[c;d]] gives [[a;c];[a;d];[b;c];[b;d]] *) -let rec list_combinations = function - | [] -> [[]] - | l::ll -> - let res = list_combinations ll in - list_map_append (fun x -> List.map (fun l -> x::l) res) l +let list_combinations l = list_cartesians (fun x l -> x::l) [] l (* Arrays *) |
