Fix: missing check to avoid stepping on vars of non positive arity

[fireball-separation.git] / ocaml / num.ml

diff --git a/ocaml/num.ml b/ocaml/num.ml
index 57cb19d068dee18d7df1cc070bc99cf448513e38..71f493545df93edce3e4a9b844b838b3b54e2c66 100644 (file)
--- a/ocaml/num.ml
+++ b/ocaml/num.ml
@@ -99,19 +99,18 @@ let free_vars = (List.map fst) ++ free_vars';;
 module ToScott =
 struct
 
-let rec scott_of_nf =
- function
-  | `N n -> Scott.mk_n n
-  | `Var(v,_) -> Pure.V v
-  | `Match(t,_,liftno,bs,args) ->
-      let bs = List.map (fun (n,t) -> n, scott_of_nf (lift liftno (t :> nf))) !bs in
-      let t = scott_of_nf (t :> nf) in
-      let m = Scott.mk_match t bs in
-      List.fold_left (fun acc t -> Pure.A(acc,scott_of_nf t)) m (args :> nf list)
-  | `I((v,_), args) -> Listx.fold_left (fun acc t -> Pure.A(acc,scott_of_nf t)) (Pure.V v) (args :> nf Listx.listx)
-  | `Lam(b,f) -> Pure.L (scott_of_nf f)
-  | `Bottom -> let f x = Pure.A (x,x) in f (Pure.L (f (Pure.V 0)))
-  | `Pacman -> let f x = Pure.A (x,x) in f (Pure.L (Pure.L (f (Pure.V 0))))
+let rec scott_of_nf = function
+ | `N n -> Scott.mk_n n
+ | `Var(v,_) -> Pure.V v
+ | `Match(t,_,liftno,bs,args) ->
+ let bs = List.map (fun (n,t) -> n, scott_of_nf (lift liftno (t :> nf))) !bs in
+     let t = scott_of_nf (t :> nf) in
+     let m = Scott.mk_match t bs in
+     List.fold_left (fun acc t -> Pure.A(acc,scott_of_nf t)) m (args :> nf list)
+ | `I((v,_), args) -> Listx.fold_left (fun acc t -> Pure.A(acc,scott_of_nf t)) (Pure.V v) (args :> nf Listx.listx)
+ | `Lam(_,t) -> Pure.L (scott_of_nf t)
+ | `Bottom -> Pure.B
+ | `Pacman -> let f x = Pure.A (x,x) in f (Pure.L (Pure.L (f (Pure.V 0))))
 end
 
 
@@ -127,7 +126,7 @@ let rec string_of_term l  =
   | `Lam(_,`Bottom) -> "BOMB"
   | `Lam _ as t -> "(" ^ string_of_term_no_pars_lam l t ^ ")"
   | `Match(t,(v,ar),bs_lift,bs,args) ->
-     "[match_"^ string_of_var v ^ (if debug_display_arities then ":"^ string_of_int ar else "") ^"_ " ^ string_of_term_no_pars l (t :> nf) ^
+     "["^ string_of_var v ^ (if debug_display_arities then ":"^ string_of_int ar else "") ^",match " ^ string_of_term_no_pars l (t :> nf) ^
      " with " ^ String.concat " | " (List.map (fun (n,t) -> string_of_int n ^ " => " ^ string_of_term l (lift bs_lift (t :> nf))) !bs) ^ "] " ^
      String.concat " " (List.map (string_of_term l) (args :> nf list)) ^ ")"
   | `Bottom -> "BOT"
@@ -370,6 +369,6 @@ let max_arity_tms n =
   | `Match(u,_,_,bs,args) -> max (max (aux l (u :> nf)) (aux_tms l (args :> nf list))) (aux_tms l (List.map snd !bs))
   | `N _ | `Bottom | `Pacman -> None
  and aux_tms l =
-  List.fold_left (fun acc t -> Pervasives.max acc (aux l t)) None in
+  List.fold_left (fun acc t -> max acc (aux l t)) None in
  fun tms -> aux_tms 0 (tms :> nf list)
 ;;