src/Tools/Compute_Oracle/am_sml.ML
author obua
Mon, 09 Jul 2007 17:36:25 +0200
changeset 23663 84b5c89b8b49
child 24134 6e69e0031f34
permissions -rw-r--r--
new version of computing oracle
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
23663
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
     1
(*  Title:      Pure/Tools/am_sml.ML
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
     2
    ID:         $Id$
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
     3
    Author:     Steven Obua
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
     4
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
     5
    ToDO: "parameterless rewrite cannot be used in pattern": In a lot of cases it CAN be used, and these cases should be handled properly; 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
     6
          right now, all cases throw an exception.
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
     7
 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
     8
*)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
     9
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    10
signature AM_SML = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    11
sig
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    12
  include ABSTRACT_MACHINE
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    13
  val save_result : (string * term) -> unit
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    14
  val set_compiled_rewriter : (term -> term) -> unit				       
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    15
  val list_nth : 'a list * int -> 'a
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    16
end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    17
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    18
structure AM_SML : AM_SML = struct
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    19
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    20
open AbstractMachine;
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    21
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    22
type program = string * string * (int Inttab.table) * (int Inttab.table) * (term Inttab.table) * (term -> term)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    23
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    24
val saved_result = ref (NONE:(string*term)option)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    25
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    26
fun save_result r = (saved_result := SOME r)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    27
fun clear_result () = (saved_result := NONE)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    28
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    29
val list_nth = List.nth
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    30
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    31
(*fun list_nth (l,n) = (writeln (makestring ("list_nth", (length l,n))); List.nth (l,n))*)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    32
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    33
val compiled_rewriter = ref (NONE:(term -> term)Option.option)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    34
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    35
fun set_compiled_rewriter r = (compiled_rewriter := SOME r)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    36
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    37
fun importable (Var _) = false
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    38
  | importable (Const _) = true			   
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    39
  | importable (App (a, b)) = importable a andalso importable b
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    40
  | importable (Abs _) = false
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    41
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    42
(*Returns true iff at most 0 .. (free-1) occur unbound. therefore
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    43
  check_freevars 0 t iff t is closed*)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    44
fun check_freevars free (Var x) = x < free
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    45
  | check_freevars free (Const c) = true
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    46
  | check_freevars free (App (u, v)) = check_freevars free u andalso check_freevars free v
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    47
  | check_freevars free (Abs m) = check_freevars (free+1) m
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    48
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    49
fun count_patternvars PVar = 1
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    50
  | count_patternvars (PConst (_, ps)) =
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    51
      List.foldl (fn (p, count) => (count_patternvars p)+count) 0 ps
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    52
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    53
fun update_arity arity code a = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    54
    (case Inttab.lookup arity code of
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    55
	 NONE => Inttab.update_new (code, a) arity
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    56
       | SOME a' => if a > a' then Inttab.update (code, a) arity else arity)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    57
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    58
(* We have to find out the maximal arity of each constant *)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    59
fun collect_pattern_arity PVar arity = arity
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    60
  | collect_pattern_arity (PConst (c, args)) arity = fold collect_pattern_arity args (update_arity arity c (length args))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    61
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    62
(* We also need to find out the maximal toplevel arity of each function constant *)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    63
fun collect_pattern_toplevel_arity PVar arity = raise Compile "internal error: collect_pattern_toplevel_arity"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    64
  | collect_pattern_toplevel_arity (PConst (c, args)) arity = update_arity arity c (length args)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    65
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    66
local
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    67
fun collect applevel (Var _) arity = arity
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    68
  | collect applevel (Const c) arity = update_arity arity c applevel
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    69
  | collect applevel (Abs m) arity = collect 0 m arity
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    70
  | collect applevel (App (a,b)) arity = collect 0 b (collect (applevel + 1) a arity)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    71
in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    72
fun collect_term_arity t arity = collect 0 t arity
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    73
end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    74
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    75
fun collect_guard_arity (Guard (a,b)) arity  = collect_term_arity b (collect_term_arity a arity)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    76
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    77
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    78
fun rep n x = if n < 0 then raise Compile "internal error: rep" else if n = 0 then [] else x::(rep (n-1) x)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    79
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    80
fun beta (Const c) = Const c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    81
  | beta (Var i) = Var i
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    82
  | beta (App (Abs m, b)) = beta (unlift 0 (subst 0 m (lift 0 b)))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    83
  | beta (App (a, b)) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    84
    (case beta a of
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    85
	 Abs m => beta (App (Abs m, b))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    86
       | a => App (a, beta b))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    87
  | beta (Abs m) = Abs (beta m)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    88
and subst x (Const c) t = Const c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    89
  | subst x (Var i) t = if i = x then t else Var i
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    90
  | subst x (App (a,b)) t = App (subst x a t, subst x b t)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    91
  | subst x (Abs m) t = Abs (subst (x+1) m (lift 0 t))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    92
and lift level (Const c) = Const c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    93
  | lift level (App (a,b)) = App (lift level a, lift level b)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    94
  | lift level (Var i) = if i < level then Var i else Var (i+1)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    95
  | lift level (Abs m) = Abs (lift (level + 1) m)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    96
and unlift level (Const c) = Const c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    97
  | unlift level (App (a, b)) = App (unlift level a, unlift level b)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    98
  | unlift level (Abs m) = Abs (unlift (level+1) m)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
    99
  | unlift level (Var i) = if i < level then Var i else Var (i-1)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   100
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   101
fun nlift level n (Var m) = if m < level then Var m else Var (m+n) 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   102
  | nlift level n (Const c) = Const c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   103
  | nlift level n (App (a,b)) = App (nlift level n a, nlift level n b)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   104
  | nlift level n (Abs b) = Abs (nlift (level+1) n b)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   105
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   106
fun subst_const (c, t) (Const c') = if c = c' then t else Const c'
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   107
  | subst_const _ (Var i) = Var i
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   108
  | subst_const ct (App (a, b)) = App (subst_const ct a, subst_const ct b)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   109
  | subst_const ct (Abs m) = Abs (subst_const ct m)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   110
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   111
(* Remove all rules that are just parameterless rewrites. This is necessary because SML does not allow functions with no parameters. *)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   112
fun inline_rules rules =
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   113
    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   114
	fun term_contains_const c (App (a, b)) = term_contains_const c a orelse term_contains_const c b
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   115
	  | term_contains_const c (Abs m) = term_contains_const c m
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   116
	  | term_contains_const c (Var i) = false
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   117
	  | term_contains_const c (Const c') = (c = c')
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   118
	fun find_rewrite [] = NONE
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   119
	  | find_rewrite ((prems, PConst (c, []), r) :: _) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   120
	    if check_freevars 0 r then 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   121
		if term_contains_const c r then 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   122
		    raise Compile "parameterless rewrite is caught in cycle"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   123
		else if not (null prems) then
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   124
		    raise Compile "parameterless rewrite may not be guarded"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   125
		else
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   126
		    SOME (c, r) 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   127
	    else raise Compile "unbound variable on right hand side or guards of rule"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   128
	  | find_rewrite (_ :: rules) = find_rewrite rules
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   129
	fun remove_rewrite (c,r) [] = []
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   130
	  | remove_rewrite (cr as (c,r)) ((rule as (prems', PConst (c', args), r'))::rules) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   131
	    (if c = c' then 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   132
		 if null args andalso r = r' andalso null (prems') then 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   133
		     remove_rewrite cr rules 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   134
		 else raise Compile "incompatible parameterless rewrites found"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   135
	     else
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   136
		 rule :: (remove_rewrite cr rules))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   137
	  | remove_rewrite cr (r::rs) = r::(remove_rewrite cr rs)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   138
	fun pattern_contains_const c (PConst (c', args)) = (c = c' orelse exists (pattern_contains_const c) args)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   139
	  | pattern_contains_const c (PVar) = false
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   140
	fun inline_rewrite (ct as (c, _)) (prems, p, r) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   141
	    if pattern_contains_const c p then 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   142
		raise Compile "parameterless rewrite cannot be used in pattern"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   143
	    else (map (fn (Guard (a,b)) => Guard (subst_const ct a, subst_const ct b)) prems, p, subst_const ct r)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   144
	fun inline inlined rules =
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   145
	    (case find_rewrite rules of 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   146
		 NONE => (Inttab.make inlined, rules)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   147
	       | SOME ct => 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   148
		 let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   149
		     val rules = map (inline_rewrite ct) (remove_rewrite ct rules)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   150
		     val inlined =  ct :: (map (fn (c', r) => (c', subst_const ct r)) inlined)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   151
		 in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   152
		     inline inlined rules
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   153
		 end)		
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   154
    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   155
	inline [] rules		
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   156
    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   157
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   158
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   159
(*
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   160
   Calculate the arity, the toplevel_arity, and adjust rules so that all toplevel pattern constants have maximal arity.
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   161
   Also beta reduce the adjusted right hand side of a rule.   
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   162
*)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   163
fun adjust_rules rules = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   164
    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   165
	val arity = fold (fn (prems, p, t) => fn arity => fold collect_guard_arity prems (collect_term_arity t (collect_pattern_arity p arity))) rules Inttab.empty
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   166
	val toplevel_arity = fold (fn (_, p, t) => fn arity => collect_pattern_toplevel_arity p arity) rules Inttab.empty
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   167
	fun arity_of c = the (Inttab.lookup arity c)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   168
	fun toplevel_arity_of c = the (Inttab.lookup toplevel_arity c)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   169
	fun adjust_pattern PVar = PVar
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   170
	  | adjust_pattern (C as PConst (c, args)) = if (length args <> arity_of c) then raise Compile ("Constant inside pattern must have maximal arity") else C
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   171
	fun adjust_rule (_, PVar, _) = raise Compile ("pattern may not be a variable")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   172
	  | adjust_rule (_, PConst (_, []), _) = raise Compile ("cannot deal with rewrites that take no parameters")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   173
	  | adjust_rule (rule as (prems, p as PConst (c, args),t)) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   174
	    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   175
		val patternvars_counted = count_patternvars p
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   176
		fun check_fv t = check_freevars patternvars_counted t
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   177
		val _ = if not (check_fv t) then raise Compile ("unbound variables on right hand side of rule") else () 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   178
		val _ = if not (forall (fn (Guard (a,b)) => check_fv a andalso check_fv b) prems) then raise Compile ("unbound variables in guards") else () 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   179
		val args = map adjust_pattern args	        
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   180
		val len = length args
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   181
		val arity = arity_of c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   182
		val lift = nlift 0
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   183
		fun adjust_tm n t = if n=0 then t else adjust_tm (n-1) (App (t, Var (n-1)))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   184
		fun adjust_term n t = adjust_tm n (lift n t)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   185
		fun adjust_guard n (Guard (a,b)) = Guard (adjust_term n a, adjust_term n b)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   186
	    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   187
		if len = arity then
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   188
		    rule
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   189
		else if arity >= len then  
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   190
		    (map (adjust_guard (arity-len)) prems, PConst (c, args @ (rep (arity-len) PVar)), adjust_term (arity-len) t)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   191
		else (raise Compile "internal error in adjust_rule")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   192
	    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   193
	fun beta_guard (Guard (a,b)) = Guard (beta a, beta b)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   194
	fun beta_rule (prems, p, t) = ((map beta_guard prems, p, beta t) handle Match => raise Compile "beta_rule")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   195
    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   196
	(arity, toplevel_arity, map (beta_rule o adjust_rule) rules)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   197
    end		    
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   198
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   199
fun print_term module arity_of toplevel_arity_of pattern_var_count pattern_lazy_var_count =
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   200
let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   201
    fun str x = string_of_int x
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   202
    fun protect_blank s = if exists_string Symbol.is_ascii_blank s then "(" ^ s ^")" else s
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   203
    val module_prefix = (case module of NONE => "" | SOME s => s^".")											  
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   204
    fun print_apps d f [] = f
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   205
      | print_apps d f (a::args) = print_apps d (module_prefix^"app "^(protect_blank f)^" "^(protect_blank (print_term d a))) args
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   206
    and print_call d (App (a, b)) args = print_call d a (b::args) 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   207
      | print_call d (Const c) args = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   208
	(case arity_of c of 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   209
	     NONE => print_apps d (module_prefix^"Const "^(str c)) args 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   210
	   | SOME 0 => module_prefix^"C"^(str c)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   211
	   | SOME a =>
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   212
	     let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   213
		 val len = length args
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   214
	     in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   215
		 if a <= len then 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   216
		     let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   217
			 val strict_a = (case toplevel_arity_of c of SOME sa => sa | NONE => a)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   218
			 val _ = if strict_a > a then raise Compile "strict" else ()
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   219
			 val s = module_prefix^"c"^(str c)^(concat (map (fn t => " "^(protect_blank (print_term d t))) (List.take (args, strict_a))))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   220
			 val s = s^(concat (map (fn t => " (fn () => "^print_term d t^")") (List.drop (List.take (args, a), strict_a))))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   221
		     in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   222
			 print_apps d s (List.drop (args, a))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   223
		     end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   224
		 else 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   225
		     let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   226
			 fun mk_apps n t = if n = 0 then t else mk_apps (n-1) (App (t, Var (n - 1)))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   227
			 fun mk_lambdas n t = if n = 0 then t else mk_lambdas (n-1) (Abs t)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   228
			 fun append_args [] t = t
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   229
			   | append_args (c::cs) t = append_args cs (App (t, c))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   230
		     in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   231
			 print_term d (mk_lambdas (a-len) (mk_apps (a-len) (nlift 0 (a-len) (append_args args (Const c)))))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   232
		     end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   233
	     end)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   234
      | print_call d t args = print_apps d (print_term d t) args
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   235
    and print_term d (Var x) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   236
	if x < d then 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   237
	    "b"^(str (d-x-1)) 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   238
	else 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   239
	    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   240
		val n = pattern_var_count - (x-d) - 1
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   241
		val x = "x"^(str n)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   242
	    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   243
		if n < pattern_var_count - pattern_lazy_var_count then 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   244
		    x
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   245
		else 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   246
		    "("^x^" ())"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   247
	    end								
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   248
      | print_term d (Abs c) = module_prefix^"Abs (fn b"^(str d)^" => "^(print_term (d + 1) c)^")"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   249
      | print_term d t = print_call d t []
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   250
in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   251
    print_term 0 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   252
end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   253
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   254
fun section n = if n = 0 then [] else (section (n-1))@[n-1]
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   255
			 			
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   256
fun print_rule gnum arity_of toplevel_arity_of (guards, p, t) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   257
    let	
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   258
	fun str x = Int.toString x		    
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   259
	fun print_pattern top n PVar = (n+1, "x"^(str n))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   260
	  | print_pattern top n (PConst (c, [])) = (n, (if top then "c" else "C")^(str c)^(if top andalso gnum > 0 then "_"^(str gnum) else ""))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   261
	  | print_pattern top n (PConst (c, args)) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   262
	    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   263
		val f = (if top then "c" else "C")^(str c)^(if top andalso gnum > 0 then "_"^(str gnum) else "")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   264
		val (n, s) = print_pattern_list 0 top (n, f) args
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   265
	    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   266
		(n, s)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   267
	    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   268
	and print_pattern_list' counter top (n,p) [] = if top then (n,p) else (n,p^")")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   269
	  | print_pattern_list' counter top (n, p) (t::ts) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   270
	    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   271
		val (n, t) = print_pattern false n t
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   272
	    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   273
		print_pattern_list' (counter + 1) top (n, if top then p^" (a"^(str counter)^" as ("^t^"))" else p^", "^t) ts
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   274
	    end	
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   275
	and print_pattern_list counter top (n, p) (t::ts) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   276
	    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   277
		val (n, t) = print_pattern false n t
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   278
	    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   279
		print_pattern_list' (counter + 1) top (n, if top then p^" (a"^(str counter)^" as ("^t^"))" else p^" ("^t) ts
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   280
	    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   281
	val c = (case p of PConst (c, _) => c | _ => raise Match)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   282
	val (n, pattern) = print_pattern true 0 p
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   283
	val lazy_vars = the (arity_of c) - the (toplevel_arity_of c)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   284
	fun print_tm tm = print_term NONE arity_of toplevel_arity_of n lazy_vars tm
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   285
        fun print_guard (Guard (a,b)) = "term_eq ("^(print_tm a)^") ("^(print_tm b)^")"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   286
	val else_branch = "c"^(str c)^"_"^(str (gnum+1))^(concat (map (fn i => " a"^(str i)) (section (the (arity_of c)))))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   287
	fun print_guards t [] = print_tm t
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   288
	  | print_guards t (g::gs) = "if ("^(print_guard g)^")"^(concat (map (fn g => " andalso ("^(print_guard g)^")") gs))^" then ("^(print_tm t)^") else "^else_branch
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   289
    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   290
	(if null guards then gnum else gnum+1, pattern^" = "^(print_guards t guards))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   291
    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   292
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   293
fun group_rules rules =
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   294
    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   295
	fun add_rule (r as (_, PConst (c,_), _)) groups =
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   296
	    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   297
		val rs = (case Inttab.lookup groups c of NONE => [] | SOME rs => rs)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   298
	    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   299
		Inttab.update (c, r::rs) groups
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   300
	    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   301
	  | add_rule _ _ = raise Compile "internal error group_rules"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   302
    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   303
	fold_rev add_rule rules Inttab.empty
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   304
    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   305
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   306
fun sml_prog name code rules = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   307
    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   308
	val buffer = ref ""
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   309
	fun write s = (buffer := (!buffer)^s)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   310
	fun writeln s = (write s; write "\n")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   311
	fun writelist [] = ()
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   312
	  | writelist (s::ss) = (writeln s; writelist ss)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   313
	fun str i = Int.toString i
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   314
	val (inlinetab, rules) = inline_rules rules
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   315
	val (arity, toplevel_arity, rules) = adjust_rules rules
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   316
	val rules = group_rules rules
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   317
	val constants = Inttab.keys arity
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   318
	fun arity_of c = Inttab.lookup arity c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   319
	fun toplevel_arity_of c = Inttab.lookup toplevel_arity c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   320
	fun rep_str s n = concat (rep n s)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   321
	fun indexed s n = s^(str n)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   322
        fun string_of_tuple [] = ""
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   323
	  | string_of_tuple (x::xs) = "("^x^(concat (map (fn s => ", "^s) xs))^")"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   324
        fun string_of_args [] = ""
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   325
	  | string_of_args (x::xs) = x^(concat (map (fn s => " "^s) xs))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   326
	fun default_case gnum c = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   327
	    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   328
		val leftargs = concat (map (indexed " x") (section (the (arity_of c))))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   329
		val rightargs = section (the (arity_of c))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   330
		val strict_args = (case toplevel_arity_of c of NONE => the (arity_of c) | SOME sa => sa)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   331
		val xs = map (fn n => if n < strict_args then "x"^(str n) else "x"^(str n)^"()") rightargs
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   332
		val right = (indexed "C" c)^" "^(string_of_tuple xs)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   333
		val debug_lazy = "(print x"^(string_of_int (strict_args - 1))^";"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   334
		val right = if strict_args < the (arity_of c) then debug_lazy^"raise AM_SML.Run \"unresolved lazy call: "^(string_of_int c)^"\")" else right		
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   335
	    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   336
		(indexed "c" c)^(if gnum > 0 then "_"^(str gnum) else "")^leftargs^" = "^right
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   337
	    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   338
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   339
	fun eval_rules c = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   340
	    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   341
		val arity = the (arity_of c)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   342
		val strict_arity = (case toplevel_arity_of c of NONE => arity | SOME sa => sa)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   343
		fun eval_rule n = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   344
		    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   345
			val sc = string_of_int c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   346
			val left = fold (fn i => fn s => "AbstractMachine.App ("^s^(indexed ", x" i)^")") (section n) ("AbstractMachine.Const "^sc)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   347
                        fun arg i = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   348
			    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   349
				val x = indexed "x" i
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   350
				val x = if i < n then "(eval bounds "^x^")" else x
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   351
				val x = if i < strict_arity then x else "(fn () => "^x^")"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   352
			    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   353
				x
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   354
			    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   355
			val right = "c"^sc^" "^(string_of_args (map arg (section arity)))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   356
			val right = fold_rev (fn i => fn s => "Abs (fn "^(indexed "x" i)^" => "^s^")") (List.drop (section arity, n)) right		
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   357
			val right = if arity > 0 then right else "C"^sc
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   358
		    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   359
			"  | eval bounds ("^left^") = "^right
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   360
		    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   361
	    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   362
		map eval_rule (rev (section (arity + 1)))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   363
	    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   364
        
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   365
	fun mk_constr_type_args n = if n > 0 then " of Term "^(rep_str " * Term" (n-1)) else ""
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   366
	val _ = writelist [        
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   367
		"structure "^name^" = struct",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   368
		"",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   369
		"datatype Term = Const of int | App of Term * Term | Abs of (Term -> Term)",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   370
		"         "^(concat (map (fn c => " | C"^(str c)^(mk_constr_type_args (the (arity_of c)))) constants)),
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   371
		""]
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   372
	fun make_constr c argprefix = "(C"^(str c)^" "^(string_of_tuple (map (fn i => argprefix^(str i)) (section (the (arity_of c)))))^")"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   373
	fun make_term_eq c = "  | term_eq "^(make_constr c "a")^" "^(make_constr c "b")^" = "^
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   374
                             (case the (arity_of c) of 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   375
				  0 => "true"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   376
				| n => 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   377
				  let 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   378
				      val eqs = map (fn i => "term_eq a"^(str i)^" b"^(str i)) (section n)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   379
				      val (eq, eqs) = (List.hd eqs, map (fn s => " andalso "^s) (List.tl eqs))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   380
				  in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   381
				      eq^(concat eqs)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   382
				  end)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   383
	val _ = writelist [
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   384
		"fun term_eq (Const c1) (Const c2) = (c1 = c2)",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   385
		"  | term_eq (App (a1,a2)) (App (b1,b2)) = term_eq a1 b1 andalso term_eq a2 b2"]
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   386
	val _ = writelist (map make_term_eq constants)		
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   387
	val _ = writelist [
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   388
		"  | term_eq _ _ = false",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   389
                "" 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   390
		] 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   391
	val _ = writelist [
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   392
		"fun app (Abs a) b = a b",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   393
		"  | app a b = App (a, b)",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   394
		""]	
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   395
	fun defcase gnum c = (case arity_of c of NONE => [] | SOME a => if a > 0 then [default_case gnum c] else [])
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   396
	fun writefundecl [] = () 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   397
	  | writefundecl (x::xs) = writelist ((("and "^x)::(map (fn s => "  | "^s) xs)))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   398
	fun list_group c = (case Inttab.lookup rules c of 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   399
				NONE => [defcase 0 c]
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   400
			      | SOME rs => 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   401
				let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   402
				    val rs = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   403
					fold
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   404
					    (fn r => 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   405
					     fn rs =>
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   406
						let 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   407
						    val (gnum, l, rs) = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   408
							(case rs of 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   409
							     [] => (0, [], []) 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   410
							   | (gnum, l)::rs => (gnum, l, rs))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   411
						    val (gnum', r) = print_rule gnum arity_of toplevel_arity_of r 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   412
						in 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   413
						    if gnum' = gnum then 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   414
							(gnum, r::l)::rs
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   415
						    else
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   416
							let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   417
							    val args = concat (map (fn i => " a"^(str i)) (section (the (arity_of c))))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   418
							    fun gnumc g = if g > 0 then "c"^(str c)^"_"^(str g)^args else "c"^(str c)^args
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   419
							    val s = gnumc (gnum) ^ " = " ^ gnumc (gnum') 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   420
							in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   421
							    (gnum', [])::(gnum, s::r::l)::rs
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   422
							end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   423
						end)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   424
					rs []
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   425
				    val rs = (case rs of [] => [(0,defcase 0 c)] | (gnum,l)::rs => (gnum, (defcase gnum c)@l)::rs)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   426
				in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   427
				    rev (map (fn z => rev (snd z)) rs)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   428
				end)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   429
	val _ = map (fn z => (map writefundecl z; writeln "")) (map list_group constants)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   430
	val _ = writelist [
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   431
		"fun convert (Const i) = AM_SML.Const i",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   432
		"  | convert (App (a, b)) = AM_SML.App (convert a, convert b)",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   433
                "  | convert (Abs _) = raise AM_SML.Run \"no abstraction in result allowed\""]	
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   434
	fun make_convert c = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   435
	    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   436
		val args = map (indexed "a") (section (the (arity_of c)))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   437
		val leftargs = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   438
		    case args of
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   439
			[] => ""
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   440
		      | (x::xs) => "("^x^(concat (map (fn s => ", "^s) xs))^")"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   441
		val args = map (indexed "convert a") (section (the (arity_of c)))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   442
		val right = fold (fn x => fn s => "AM_SML.App ("^s^", "^x^")") args ("AM_SML.Const "^(str c))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   443
	    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   444
		"  | convert (C"^(str c)^" "^leftargs^") = "^right
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   445
	    end 		
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   446
	val _ = writelist (map make_convert constants)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   447
	val _ = writelist [
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   448
		"",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   449
		"fun eval bounds (AbstractMachine.Abs m) = Abs (fn b => eval (b::bounds) m)",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   450
		"  | eval bounds (AbstractMachine.Var i) = AM_SML.list_nth (bounds, i)"]
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   451
	val _ = map (writelist o eval_rules) constants
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   452
	val _ = writelist [
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   453
                "  | eval bounds (AbstractMachine.App (a, b)) = app (eval bounds a) (eval bounds b)",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   454
                "  | eval bounds (AbstractMachine.Const c) = Const c"]                
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   455
	val _ = writelist [		
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   456
		"",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   457
		"fun export term = AM_SML.save_result (\""^code^"\", convert term)",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   458
		"",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   459
                "val _ = AM_SML.set_compiled_rewriter (fn t => convert (eval [] t))",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   460
                "",
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   461
		"end"]
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   462
    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   463
	(arity, toplevel_arity, inlinetab, !buffer)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   464
    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   465
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   466
val guid_counter = ref 0
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   467
fun get_guid () = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   468
    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   469
	val c = !guid_counter
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   470
	val _ = guid_counter := !guid_counter + 1
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   471
    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   472
	(LargeInt.toString (Time.toMicroseconds (Time.now ()))) ^ (string_of_int c)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   473
    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   474
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   475
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   476
fun writeTextFile name s = File.write (Path.explode name) s
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   477
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   478
fun use_source src = use_text "" Output.ml_output false src
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   479
    
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   480
fun compile eqs = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   481
    let
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   482
	val guid = get_guid ()
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   483
	val code = Real.toString (random ())
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   484
	val module = "AMSML_"^guid
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   485
	val (arity, toplevel_arity, inlinetab, source) = sml_prog module code eqs
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   486
(*	val _ = writeTextFile "Gencode.ML" source*)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   487
	val _ = compiled_rewriter := NONE
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   488
	val _ = use_source source
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   489
    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   490
	case !compiled_rewriter of 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   491
	    NONE => raise Compile "broken link to compiled function"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   492
	  | SOME f => (module, code, arity, toplevel_arity, inlinetab, f)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   493
    end
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   494
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   495
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   496
fun run' (module, code, arity, toplevel_arity, inlinetab, compiled_fun) t = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   497
    let	
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   498
	val _ = if check_freevars 0 t then () else raise Run ("can only compute closed terms")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   499
	fun inline (Const c) = (case Inttab.lookup inlinetab c of NONE => Const c | SOME t => t)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   500
	  | inline (Var i) = Var i
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   501
	  | inline (App (a, b)) = App (inline a, inline b)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   502
	  | inline (Abs m) = Abs (inline m)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   503
	val t = beta (inline t)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   504
	fun arity_of c = Inttab.lookup arity c		 	 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   505
	fun toplevel_arity_of c = Inttab.lookup toplevel_arity c
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   506
	val term = print_term NONE arity_of toplevel_arity_of 0 0 t 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   507
        val source = "local open "^module^" in val _ = export ("^term^") end"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   508
	val _ = writeTextFile "Gencode_call.ML" source
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   509
	val _ = clear_result ()
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   510
	val _ = use_source source
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   511
    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   512
	case !saved_result of 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   513
	    NONE => raise Run "broken link to compiled code"
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   514
	  | SOME (code', t) => (clear_result (); if code' = code then t else raise Run "link to compiled code was hijacked")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   515
    end		
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   516
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   517
fun run (module, code, arity, toplevel_arity, inlinetab, compiled_fun) t = 
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   518
    let	
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   519
	val _ = if check_freevars 0 t then () else raise Run ("can only compute closed terms")
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   520
	fun inline (Const c) = (case Inttab.lookup inlinetab c of NONE => Const c | SOME t => t)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   521
	  | inline (Var i) = Var i
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   522
	  | inline (App (a, b)) = App (inline a, inline b)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   523
	  | inline (Abs m) = Abs (inline m)
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   524
    in
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   525
	compiled_fun (beta (inline t))
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   526
    end	
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   527
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   528
fun discard p = ()
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   529
			 	  
84b5c89b8b49 new version of computing oracle
obua
parents:
diff changeset
   530
end