]> matita.cs.unibo.it Git - fireball-separation.git/blobdiff - ocaml/lambda4.ml
New representation of arities in variables
[fireball-separation.git] / ocaml / lambda4.ml
index 3e9f1ef81c9e27fb96f7693baeb71c9b3da7b887..f71f36005f220d09bdea65baf15aed2f181181bd 100644 (file)
@@ -3,7 +3,7 @@ open Util.Vars
 open Pure
 open Num
 
-let bomb = ref(`Var ~-1);;
+let bomb = ref(`Var(-1,-666));;
 
 type problem =
  { freshno: int
@@ -21,7 +21,7 @@ for all head
 
 (* let ($) f x = f x;; *)
 
-let subterms tms freshno =
+(* let subterms tms freshno =
  let apply_var =
   let no = ref freshno in
   function t -> incr no; mk_app t (`Var !no) in
@@ -42,7 +42,7 @@ let subterms tms freshno =
  in let tms' = (* Util.sort_uniq ~compare:eta_compare*) (Util.concat_map aux tms) in
  tms @ tms'
  (* List.map (fun (t, v) -> match t with `N _ -> t | _ -> mk_app t v) (List.combine tms (List.mapi (fun i _ -> `Var(freshno+i)) tms)) *)
-;;
+;; *)
 
 let all_terms p =
 (match p.div with None -> [] | Some t -> [(t :> i_n_var)])
@@ -77,39 +77,39 @@ let failwithProblem p reason =
  failwith reason
 ;;
 
-let make_fresh_var p =
+let make_fresh_var p arity =
  let freshno = p.freshno + 1 in
- {p with freshno}, `Var freshno
+ {p with freshno}, `Var(freshno,arity)
 ;;
 
-let make_fresh_vars p m =
Array.fold_left
+let make_fresh_vars p arities =
List.fold_right
   (* fold_left vs. fold_right hides/shows the bug in problem q7 *)
-  (fun (p, vars) _ -> let p, var = make_fresh_var p in p, var::vars)
+  (fun arity (p, vars) -> let p, var = make_fresh_var p arity in p, var::vars)
+  arities
   (p, [])
-  (Array.make m ())
 ;;
 
 let simple_expand_match ps =
   let rec aux level = function
   | #i_num_var as t -> aux_i_num_var level t
-  | `Lam(b,ar,t) -> `Lam(b,ar,aux (level+1) t)
+  | `Lam(b,t) -> `Lam(b,aux (level+1) t)
   and aux_i_num_var level = function
-  | `Match(u,ar,bs_lift,bs,args) as torig ->
+  | `Match(u,v,bs_lift,bs,args) as torig ->
     let u = aux_i_num_var level u in
     bs := List.map (fun (n, x) -> n, aux 0 x) !bs;
     (try
        (match u with
          | #i_n_var as u ->
             let i = index_of (lift (-level) u) (ps :> nf list) (* can raise Not_found *)
-            in let t = mk_match (`N i) ar bs_lift bs args in
+            in let t = mk_match (`N i) v bs_lift bs args in
             if t <> torig then
             aux level (t :> nf)
            else raise Not_found
          | _ -> raise Not_found)
       with Not_found ->
-       `Match(cast_to_i_num_var u,ar,bs_lift,bs,List.map (aux level) args))
-  | `I(k,args) -> `I(k,Listx.map (aux level) args)
+       `Match(cast_to_i_num_var u,v,bs_lift,bs,List.map (aux level) args))
+  | `I(v,args) -> `I(v,Listx.map (aux level) args)
   | `N _ | `Var _ as t -> t
 in aux_i_num_var 0;;
 
@@ -143,7 +143,7 @@ prerr_endline ("# INST0: " ^ string_of_var x ^ " := " ^ print ~l inst));*)
      let t = subst false x inst (t :> nf) in
 (*prerr_endline ("SUBSTITUTED t:" ^ print (t :> nf));*)
      let freshno,new_t,acc_new_ps =
-      expand_match (freshno,acc_ps@`Var(max_int/3)::todo_ps,acc_new_ps) t
+      expand_match (freshno,acc_ps@`Var(max_int/3,-666)::todo_ps,acc_new_ps) t
      in
       aux (freshno,acc_ps@[new_t],acc_new_ps) todo_ps
 
@@ -175,7 +175,7 @@ prerr_endline ("# INST0: " ^ string_of_var x ^ " := " ^ print ~l inst));*)
 
   and expand_match ((freshno,acc_ps,acc_new_ps) as acc) t =
    match t with
-   | `Match(u',ar,bs_lift,bs,args) ->
+   | `Match(u',orig,bs_lift,bs,args) ->
         let freshno,u,acc_new_ps = expand_match acc (u' :> nf) in
         let acc_new_ps,i =
          match u with
@@ -194,11 +194,11 @@ List.iter (fun x -> prerr_endline ("IN2: " ^ print (fst x :> nf))) super_simplif
           if List.exists (fun (j,_) -> i=j) !bs then
            freshno
           else
-           let freshno,v = freshno+1, `Var (freshno+1) in (* make_fresh_var freshno in *)
+           let freshno,v = freshno+1, `Var (freshno+1, snd orig - 1) in (* make_fresh_var freshno in *)
            bs := !bs @ [i, v] ;
            freshno in
 (*prerr_endlie ("t DA RIDURRE:" ^ print (`Match(`N i,arity,bs_lift,bs,args) :> nf) ^ " more_args=" ^ string_of_int more_args);*)
-         let t = mk_match (`N i) ar bs_lift bs args in
+         let t = mk_match (`N i) orig bs_lift bs args in
 (*prerr_endline ("NUOVO t:" ^ print (fst t :> nf) ^ " more_args=" ^ string_of_int (snd t));*)
           expand_match (freshno,acc_ps,acc_new_ps) t
    | `Lam _ -> raise ExpandedToLambda
@@ -243,10 +243,10 @@ let rec dangerous arities showstoppers =
       (match t with
           `N _ -> List.iter (dangerous arities showstoppers) args
         | `Match _ as t -> dangerous arities showstoppers t ; List.iter (dangerous arities showstoppers) args
-        | `Var x -> dangerous_inert arities showstoppers x args 2 (* 2 coming from Scott's encoding *)
-        | `I(x,args') -> dangerous_inert arities showstoppers x (Listx.to_list args' @ args) 2 (* 2 coming from Scott's encoding *)
+        | `Var(x,_) -> dangerous_inert arities showstoppers x args 2 (* 2 coming from Scott's encoding *)
+        | `I((x,_),args') -> dangerous_inert arities showstoppers x (Listx.to_list args' @ args) 2 (* 2 coming from Scott's encoding *)
       )
-  | `I(k,args) -> dangerous_inert arities showstoppers k (Listx.to_list args) 0
+  | `I((k,_),args) -> dangerous_inert arities showstoppers k (Listx.to_list args) 0
 
 and dangerous_inert arities showstoppers k args more_args =
  List.iter (dangerous arities showstoppers) args ;
@@ -268,10 +268,10 @@ let rec dangerous_conv arities showstoppers =
       (match t with
           `N _ -> concat_map (dangerous_conv arities showstoppers) args
         | `Match _ as t -> dangerous_conv arities showstoppers t @ concat_map (dangerous_conv arities showstoppers) args
-        | `Var x -> dangerous_inert_conv arities showstoppers x args 2 (* 2 coming from Scott's encoding *)
-        | `I(x,args') -> dangerous_inert_conv arities showstoppers x (Listx.to_list args' @ args) 2 (* 2 coming from Scott's encoding *)
+        | `Var(x,_) -> dangerous_inert_conv arities showstoppers x args 2 (* 2 coming from Scott's encoding *)
+        | `I((x,_),args') -> dangerous_inert_conv arities showstoppers x (Listx.to_list args' @ args) 2 (* 2 coming from Scott's encoding *)
       )
-  | `I(k,args) -> dangerous_inert_conv arities showstoppers k (Listx.to_list args) 0
+  | `I((k,_),args) -> dangerous_inert_conv arities showstoppers k (Listx.to_list args) 0
 
 and dangerous_inert_conv arities showstoppers k args more_args =
  concat_map (dangerous_conv arities showstoppers) args @
@@ -345,7 +345,7 @@ let critical_showstoppers p =
   let showstoppers_eat =
    let heads_and_arities =
     List.sort (fun (k,_) (h,_) -> compare k h)
-     (filter_map (function `Var k -> Some (k,0) | `I(k,args) -> Some (k,Listx.length args) | _ -> None ) p.ps) in
+     (filter_map (function `Var(k,_) -> Some (k,0) | `I((k,_),args) -> Some (k,Listx.length args) | _ -> None ) p.ps) in
    let rec multiple_arities =
     function
        []
@@ -397,8 +397,8 @@ prerr_endline ("# INST_IN_EAT: " ^ string_of_var x ^ " := " ^ print ~l inst));
    then p
    else
     let n = match div with `I(_,args) -> Listx.length args | `Var _ -> 0 in
-    let p, bomb' = make_fresh_var p in
-    (if !bomb <> `Var (-1) then
+    let p, bomb' = make_fresh_var p (-666) in
+    (if !bomb <> `Var (-1,-666) then
      failwithProblem p
       ("Bomb was duplicated! It was " ^ string_of_nf !bomb ^
        ", tried to change it to " ^ string_of_nf bomb'));
@@ -446,25 +446,28 @@ List.iter (fun l -> prerr_endline (String.concat " " (List.map string_of_var l))
   `Continue p
 
 let instantiate p x n =
- (if `Var x = !bomb
+ (if hd_of_i_var (cast_to_i_var !bomb) = x
    then failwithProblem p ("BOMB (" ^ string_of_nf !bomb ^ ") cannot be instantiated!"));
- let p,vars = make_fresh_vars p n in
- let p,zero = make_fresh_var p in
- let zero = Listx.Nil zero in
- let args = if n = 0 then zero else Listx.append zero (Listx.from_list vars) in
+ (* ACL FIXME compute arity of x correctly below! *)
+ let arity_of_x = -666 in
+ (* AC: FIXME compute arities correctly below! *)
+ let arities = Array.to_list (Array.make (n+1) 0) in
+ let p,vars = make_fresh_vars p arities in
+ (* let p,zero = make_fresh_var p in *)
+ (* let zero = Listx.Nil zero in *)
+ (* let args = if n = 0 then zero else Listx.append zero (Listx.from_list vars) in *)
+ let args = Listx.from_list (vars :> nf list) in
  let bs = ref [] in
- let arity1 = (*assert false; -666*) 0 in
- let arity2 = (*assert false; -666*) 0 in
- let inst = `Lam(false,arity1,`Match(`I(0,Listx.map (lift 1) args),arity2,1,bs,[])) in
+ let inst = `Lam(false,`Match(`I((0,0),Listx.map (lift 1) args),(x,arity_of_x),1,bs,[])) in
  let p = {p with deltas=bs::p.deltas} in
  subst_in_problem x inst p
 ;;
 
 let compute_special_k tms =
   let rec aux k (t: nf) = Pervasives.max k (match t with
-    | `Lam(b,_,t) -> aux (k + if b then 1 else 0) t
+    | `Lam(b,t) -> aux (k + if b then 1 else 0) t
     | `I(n, tms) -> Listx.max (Listx.map (aux 0) tms)
-    | `Match(t,_,liftno, bs, args) ->
+    | `Match(t, _, liftno, bs, args) ->
         List.fold_left max 0 (List.map (aux 0) ((t :> nf)::args@List.map snd !bs))
     | `N _ -> 0
     | `Var _ -> 0
@@ -590,7 +593,7 @@ let optimize_numerals p =
     | `N n -> `N (List.nth perm n)
     | `I _ -> assert false
     | `Var _ as t -> t
-    | `Lam(v,ar,t) -> `Lam(v, ar, aux t)
+    | `Lam(v,t) -> `Lam(v, aux t)
     | `Match(_,_,_,bs,_) as t -> (bs := List.map (fun (n,t) -> (List.nth perm n, t)) !bs); t
     in List.map (fun (n,t) -> (n,aux t))
   in
@@ -617,7 +620,7 @@ let env_of_sigma freshno sigma should_explode =
     e,Pure.lift (-n-1) (snd (List.find (fun (i,_) -> i = n) sigma)),[]
    with
     Not_found ->
-     if should_explode && `Var n = !bomb
+     if should_explode && n = hd_of_i_var (cast_to_i_var !bomb)
       then ([], (let f t = Pure.A(t,t) in f (Pure.L (f (Pure.V 0)))), [])
       else ([],Pure.V n,[]))::e
  in aux 0
@@ -657,7 +660,7 @@ let main problems =
   List.iter
    (fun (p,n,cmds) ->
     Console.print_hline();
-    bomb := `Var (-1);
+    bomb := `Var (-1,-666);
     let p_finale = aux p n cmds in
     let freshno,sigma = p_finale.freshno, p_finale.sigma in
     prerr_endline ("------- <DONE> ------\n ");
@@ -727,7 +730,7 @@ in
 
 (********************** problems *******************)
 
-let zero = `Var 0;;
+let zero = `Var(0,-1);;
 
 let append_zero =
  function
@@ -760,7 +763,7 @@ let magic_conv ~div ~conv ~nums cmds =
   let ps = List.map append_zero tms in (* crea lista applicando zeri o dummies *)
   let freshno = List.length var_names in
   let deltas =
-   let dummy = `Var (max_int / 2) in
+   let dummy = `Var (max_int / 2, -666) in
     [ ref (Array.to_list (Array.init (List.length ps) (fun i -> i, dummy))) ] in
 
   {freshno; div; conv; ps; sigma=[] ; deltas}, special_k, cmds