X-Git-Url: http://matita.cs.unibo.it/gitweb/?a=blobdiff_plain;f=helm%2Fsoftware%2Fcomponents%2Fextlib%2FhExtlib.ml;h=4a228b232736e916f4960048dfd9d15f0146e8eb;hb=2b837ca9e298eb44eee95d9ca0e331c577785dcb;hp=0d7254c8bd48cd34b8374e11f6c224d3620213a8;hpb=4b5527a6ccbe73ada47f903292959ed3dde3b5c6;p=helm.git diff --git a/helm/software/components/extlib/hExtlib.ml b/helm/software/components/extlib/hExtlib.ml index 0d7254c8b..4a228b232 100644 --- a/helm/software/components/extlib/hExtlib.ml +++ b/helm/software/components/extlib/hExtlib.ml @@ -27,7 +27,7 @@ (** PROFILING *) -let profiling_enabled = ref true ;; (* ComponentsConf.profiling *) +let profiling_enabled = ref false ;; (* ComponentsConf.profiling *) let something_profiled = ref false @@ -140,6 +140,69 @@ let flatten_map f l = List.flatten (List.map f l) ;; +let list_mapi f l = + let rec aux k = function + | [] -> [] + | h::tl -> f h k :: aux (k+1) tl + in + aux 0 l +;; + +let list_index p = + let rec aux n = + function + [] -> None + | he::_ when p he -> Some (n,he) + | _::tl -> aux (n + 1) tl + in + aux 0 +;; + +let rec list_iter_default2 f l1 def l2 = + match l1,l2 with + | [], _ -> () + | a::ta, b::tb -> f a b; list_iter_default2 f ta def tb + | a::ta, [] -> f a def; list_iter_default2 f ta def [] +;; + +let rec list_forall_default3 f l1 l2 def l3 = + match l1,l2,l3 with + | [], [], _ -> true + | [], _, _ + | _, [], _ -> raise (Invalid_argument "list_forall_default3") + | a::ta, b::tb, c::tc -> f a b c && list_forall_default3 f ta tb def tc + | a::ta, b::tb, [] -> f a b def && list_forall_default3 f ta tb def [] +;; + +exception FailureAt of int;; + +let list_forall_default3_var f l1 l2 def l3 = + let rec aux f l1 l2 def l3 i = + match l1,l2,l3 with + | [], [], _ -> true + | [], _, _ + | _, [], _ -> raise (Invalid_argument "list_forall_default3") + | a::ta, b::tb, c::tc -> + if f a b c then aux f ta tb def tc (i+1) + else raise (FailureAt i) + | a::ta, b::tb, [] -> + if f a b def then aux f ta tb def [] (i+1) + else raise (FailureAt i) + in aux f l1 l2 def l3 0 +;; + +let sharing_map f l = + let unchanged = ref true in + let rec aux b = function + | [] as t -> unchanged := b; t + | he::tl -> + let he1 = f he in + he1 :: aux (b && he1 == he) tl + in + let l1 = aux true l in + if !unchanged then l else l1 +;; + let rec list_uniq ?(eq=(=)) = function | [] -> [] | h::[] -> [h] @@ -185,21 +248,30 @@ let list_concat ?(sep = []) = in aux [] +let list_iter_sep ~sep f = + let rec aux = + function + | [] -> () + | [ last ] -> f last + | hd :: tl -> f hd; sep (); aux tl + in + aux + let rec list_findopt f l = - let rec aux = function + let rec aux k = function | [] -> None | x::tl -> - (match f x with - | None -> aux tl + (match f x k with + | None -> aux (succ k) tl | Some _ as rc -> rc) in - aux l + aux 0 l -let split_nth n l = +let split_nth msg n l = let rec aux acc n l = match n, l with | 0, _ -> List.rev acc, l - | n, [] -> raise (Failure "HExtlib.split_nth") + | n, [] -> raise (Failure ("HExtlib.split_nth: " ^ msg)) | n, hd :: tl -> aux (hd :: acc) (n - 1) tl in aux [] n l @@ -207,7 +279,13 @@ let list_last l = let l = List.rev l in try List.hd l with exn -> raise (Failure "HExtlib.list_last") ;; - + +let rec list_assoc_all a = function + | [] -> [] + | (x, y) :: tl when x = a -> y :: list_assoc_all a tl + | _ :: tl -> list_assoc_all a tl +;; + (** {2 File predicates} *) let is_dir fname = @@ -411,7 +489,7 @@ let loc_of_floc floc = Stdpp.first_pos floc, Stdpp.last_pos floc;; let floc_of_loc (loc_begin, loc_end) = Stdpp.make_loc (loc_begin, loc_end) -let dummy_floc = floc_of_loc (-1, -1) +let dummy_floc = floc_of_loc (0, 0) let raise_localized_exception ~offset floc exn = let x, y = loc_of_floc floc in @@ -479,3 +557,25 @@ let chop_prefix prefix s = let touch s = try close_out(open_out s) with Sys_error _ -> () ;; + +let rec mk_list x = function + | 0 -> [] + | n -> x :: mk_list x (n-1) +;; + +let list_seq start stop = + if start > stop then [] else + let rec aux pos = + if pos = stop then [] + else pos :: (aux (pos+1)) + in + aux start +;; + +let rec list_skip n l = + match n,l with + | 0,_ -> l + | n,_::l -> list_skip (n-1) l + | _, [] -> assert false +;; +