| 
23150
 | 
     1  | 
(*  Title:      HOL/Tools/TFL/utils.ML
  | 
| 
 | 
     2  | 
    ID:         $Id$
  | 
| 
 | 
     3  | 
    Author:     Konrad Slind, Cambridge University Computer Laboratory
  | 
| 
 | 
     4  | 
    Copyright   1997  University of Cambridge
  | 
| 
 | 
     5  | 
  | 
| 
 | 
     6  | 
Basic utilities.
  | 
| 
 | 
     7  | 
*)
  | 
| 
 | 
     8  | 
  | 
| 
 | 
     9  | 
signature UTILS =
  | 
| 
 | 
    10  | 
sig
  | 
| 
 | 
    11  | 
  exception ERR of {module: string, func: string, mesg: string}
 | 
| 
 | 
    12  | 
  val C: ('a -> 'b -> 'c) -> 'b -> 'a -> 'c
 | 
| 
 | 
    13  | 
  val end_itlist: ('a -> 'a -> 'a) -> 'a list -> 'a
 | 
| 
 | 
    14  | 
  val itlist2: ('a -> 'b -> 'c -> 'c) -> 'a list -> 'b list -> 'c -> 'c
 | 
| 
 | 
    15  | 
  val pluck: ('a -> bool) -> 'a list -> 'a * 'a list
 | 
| 
 | 
    16  | 
  val zip3: 'a list -> 'b list -> 'c list -> ('a*'b*'c) list
 | 
| 
 | 
    17  | 
  val take: ('a -> 'b) -> int * 'a list -> 'b list
 | 
| 
 | 
    18  | 
end;
  | 
| 
 | 
    19  | 
  | 
| 
 | 
    20  | 
structure Utils: UTILS =
  | 
| 
 | 
    21  | 
struct
  | 
| 
 | 
    22  | 
  | 
| 
 | 
    23  | 
(*standard exception for TFL*)
  | 
| 
 | 
    24  | 
exception ERR of {module: string, func: string, mesg: string};
 | 
| 
 | 
    25  | 
  | 
| 
 | 
    26  | 
fun UTILS_ERR func mesg = ERR {module = "Utils", func = func, mesg = mesg};
 | 
| 
 | 
    27  | 
  | 
| 
 | 
    28  | 
  | 
| 
 | 
    29  | 
fun C f x y = f y x
  | 
| 
 | 
    30  | 
  | 
| 
 | 
    31  | 
fun end_itlist f [] = raise (UTILS_ERR "end_itlist" "list too short")
  | 
| 
 | 
    32  | 
  | end_itlist f [x] = x 
  | 
| 
 | 
    33  | 
  | end_itlist f (x :: xs) = f x (end_itlist f xs);
  | 
| 
 | 
    34  | 
  | 
| 
 | 
    35  | 
fun itlist2 f L1 L2 base_value =
  | 
| 
 | 
    36  | 
 let fun it ([],[]) = base_value
  | 
| 
 | 
    37  | 
       | it ((a::rst1),(b::rst2)) = f a b (it (rst1,rst2))
  | 
| 
 | 
    38  | 
       | it _ = raise UTILS_ERR "itlist2" "different length lists"
  | 
| 
 | 
    39  | 
 in  it (L1,L2)
  | 
| 
 | 
    40  | 
 end;
  | 
| 
 | 
    41  | 
  | 
| 
 | 
    42  | 
fun pluck p  =
  | 
| 
 | 
    43  | 
  let fun remv ([],_) = raise UTILS_ERR "pluck" "item not found"
  | 
| 
 | 
    44  | 
        | remv (h::t, A) = if p h then (h, rev A @ t) else remv (t,h::A)
  | 
| 
 | 
    45  | 
  in fn L => remv(L,[])
  | 
| 
 | 
    46  | 
  end;
  | 
| 
 | 
    47  | 
  | 
| 
 | 
    48  | 
fun take f =
  | 
| 
 | 
    49  | 
  let fun grab(0,L) = []
  | 
| 
 | 
    50  | 
        | grab(n, x::rst) = f x::grab(n-1,rst)
  | 
| 
 | 
    51  | 
  in grab
  | 
| 
 | 
    52  | 
  end;
  | 
| 
 | 
    53  | 
  | 
| 
 | 
    54  | 
fun zip3 [][][] = []
  | 
| 
 | 
    55  | 
  | zip3 (x::l1) (y::l2) (z::l3) = (x,y,z)::zip3 l1 l2 l3
  | 
| 
 | 
    56  | 
  | zip3 _ _ _ = raise UTILS_ERR "zip3" "different lengths";
  | 
| 
 | 
    57  | 
  | 
| 
 | 
    58  | 
  | 
| 
 | 
    59  | 
end;
  |