src/HOL/Tools/refute.ML
author webertj
Thu Jun 17 22:01:23 2004 +0200 (2004-06-17)
changeset 14965 7155b319eafa
parent 14818 ad83019a66a4
child 14984 edbc81e60809
permissions -rw-r--r--
new SAT solver interface
webertj@14350
     1
(*  Title:      HOL/Tools/refute.ML
webertj@14350
     2
    ID:         $Id$
webertj@14350
     3
    Author:     Tjark Weber
webertj@14350
     4
    Copyright   2003-2004
webertj@14350
     5
webertj@14965
     6
Finite model generation for HOL formulas, using a SAT solver.
webertj@14350
     7
*)
webertj@14350
     8
webertj@14965
     9
(* TODO: case, recursion, size for IDTs are not supported yet *)
webertj@14456
    10
webertj@14456
    11
(* ------------------------------------------------------------------------- *)
webertj@14456
    12
(* Declares the 'REFUTE' signature as well as a structure 'Refute'.          *)
webertj@14456
    13
(* Documentation is available in the Isabelle/Isar theory 'HOL/Refute.thy'.  *)
webertj@14350
    14
(* ------------------------------------------------------------------------- *)
webertj@14350
    15
webertj@14350
    16
signature REFUTE =
webertj@14350
    17
sig
webertj@14350
    18
webertj@14456
    19
	exception REFUTE of string * string
webertj@14456
    20
webertj@14456
    21
(* ------------------------------------------------------------------------- *)
webertj@14807
    22
(* Model/interpretation related code (translation HOL -> propositional logic *)
webertj@14456
    23
(* ------------------------------------------------------------------------- *)
webertj@14456
    24
webertj@14807
    25
	type params
webertj@14456
    26
	type interpretation
webertj@14456
    27
	type model
webertj@14456
    28
	type arguments
webertj@14456
    29
webertj@14807
    30
	exception CANNOT_INTERPRET of Term.term
webertj@14807
    31
	exception MAXVARS_EXCEEDED
webertj@14456
    32
webertj@14807
    33
	val add_interpreter : string -> (theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) option) -> theory -> theory
webertj@14807
    34
	val add_printer     : string -> (theory -> model -> Term.term -> interpretation -> (int -> bool) -> Term.term option) -> theory -> theory
webertj@14456
    35
webertj@14807
    36
	val interpret : theory -> model -> arguments -> Term.term -> (interpretation * model * arguments)  (* exception CANNOT_INTERPRET *)
webertj@14807
    37
webertj@14807
    38
	val print       : theory -> model -> Term.term -> interpretation -> (int -> bool) -> Term.term
webertj@14456
    39
	val print_model : theory -> model -> (int -> bool) -> string
webertj@14456
    40
webertj@14456
    41
(* ------------------------------------------------------------------------- *)
webertj@14456
    42
(* Interface                                                                 *)
webertj@14456
    43
(* ------------------------------------------------------------------------- *)
webertj@14456
    44
webertj@14456
    45
	val set_default_param  : (string * string) -> theory -> theory
webertj@14456
    46
	val get_default_param  : theory -> string -> string option
webertj@14456
    47
	val get_default_params : theory -> (string * string) list
webertj@14807
    48
	val actual_params      : theory -> (string * string) list -> params
webertj@14456
    49
webertj@14807
    50
	val find_model : theory -> params -> Term.term -> bool -> unit
webertj@14456
    51
webertj@14456
    52
	val satisfy_term   : theory -> (string * string) list -> Term.term -> unit  (* tries to find a model for a formula *)
webertj@14456
    53
	val refute_term    : theory -> (string * string) list -> Term.term -> unit  (* tries to find a model that refutes a formula *)
webertj@14456
    54
	val refute_subgoal : theory -> (string * string) list -> Thm.thm -> int -> unit
webertj@14456
    55
webertj@14456
    56
	val setup : (theory -> theory) list
webertj@14456
    57
end;
webertj@14456
    58
webertj@14456
    59
structure Refute : REFUTE =
webertj@14456
    60
struct
webertj@14456
    61
webertj@14965
    62
	(* FIXME compatibility -- should avoid std_output altogether *)
wenzelm@14818
    63
	val std_output = Output.std_output o Output.output;
wenzelm@14818
    64
webertj@14456
    65
	open PropLogic;
webertj@14456
    66
webertj@14350
    67
	(* We use 'REFUTE' only for internal error conditions that should    *)
webertj@14350
    68
	(* never occur in the first place (i.e. errors caused by bugs in our *)
webertj@14350
    69
	(* code).  Otherwise (e.g. to indicate invalid input data) we use    *)
webertj@14350
    70
	(* 'error'.                                                          *)
webertj@14456
    71
	exception REFUTE of string * string;  (* ("in function", "cause") *)
webertj@14350
    72
webertj@14807
    73
	exception CANNOT_INTERPRET of Term.term;
webertj@14350
    74
webertj@14807
    75
	(* should be raised by an interpreter when more variables would be *)
webertj@14807
    76
	(* required than allowed by 'maxvars'                              *)
webertj@14807
    77
	exception MAXVARS_EXCEEDED;
webertj@14350
    78
webertj@14350
    79
(* ------------------------------------------------------------------------- *)
webertj@14350
    80
(* TREES                                                                     *)
webertj@14350
    81
(* ------------------------------------------------------------------------- *)
webertj@14350
    82
webertj@14350
    83
(* ------------------------------------------------------------------------- *)
webertj@14350
    84
(* tree: implements an arbitrarily (but finitely) branching tree as a list   *)
webertj@14350
    85
(*       of (lists of ...) elements                                          *)
webertj@14350
    86
(* ------------------------------------------------------------------------- *)
webertj@14350
    87
webertj@14350
    88
	datatype 'a tree =
webertj@14350
    89
		  Leaf of 'a
webertj@14350
    90
		| Node of ('a tree) list;
webertj@14350
    91
webertj@14350
    92
	(* ('a -> 'b) -> 'a tree -> 'b tree *)
webertj@14350
    93
webertj@14350
    94
	fun tree_map f tr =
webertj@14350
    95
		case tr of
webertj@14350
    96
		  Leaf x  => Leaf (f x)
webertj@14350
    97
		| Node xs => Node (map (tree_map f) xs);
webertj@14350
    98
webertj@14350
    99
	(* ('a * 'b -> 'a) -> 'a * ('b tree) -> 'a *)
webertj@14350
   100
webertj@14350
   101
	fun tree_foldl f =
webertj@14350
   102
	let
webertj@14350
   103
		fun itl (e, Leaf x)  = f(e,x)
webertj@14350
   104
		  | itl (e, Node xs) = foldl (tree_foldl f) (e,xs)
webertj@14350
   105
	in
webertj@14350
   106
		itl
webertj@14350
   107
	end;
webertj@14350
   108
webertj@14350
   109
	(* 'a tree * 'b tree -> ('a * 'b) tree *)
webertj@14350
   110
webertj@14350
   111
	fun tree_pair (t1,t2) =
webertj@14350
   112
		case t1 of
webertj@14350
   113
		  Leaf x =>
webertj@14350
   114
			(case t2 of
webertj@14350
   115
				  Leaf y => Leaf (x,y)
webertj@14350
   116
				| Node _ => raise REFUTE ("tree_pair", "trees are of different height (second tree is higher)"))
webertj@14350
   117
		| Node xs =>
webertj@14350
   118
			(case t2 of
webertj@14807
   119
				  (* '~~' will raise an exception if the number of branches in   *)
webertj@14807
   120
				  (* both trees is different at the current node                 *)
webertj@14350
   121
				  Node ys => Node (map tree_pair (xs ~~ ys))
webertj@14350
   122
				| Leaf _  => raise REFUTE ("tree_pair", "trees are of different height (first tree is higher)"));
webertj@14350
   123
webertj@14456
   124
webertj@14350
   125
(* ------------------------------------------------------------------------- *)
webertj@14807
   126
(* params: parameters that control the translation into a propositional      *)
webertj@14807
   127
(*         formula/model generation                                          *)
webertj@14807
   128
(*                                                                           *)
webertj@14807
   129
(* The following parameters are supported (and required (!), except for      *)
webertj@14807
   130
(* "sizes"):                                                                 *)
webertj@14807
   131
(*                                                                           *)
webertj@14807
   132
(* Name          Type    Description                                         *)
webertj@14807
   133
(*                                                                           *)
webertj@14807
   134
(* "sizes"       (string * int) list                                         *)
webertj@14807
   135
(*                       Size of ground types (e.g. 'a=2), or depth of IDTs. *)
webertj@14807
   136
(* "minsize"     int     If >0, minimal size of each ground type/IDT depth.  *)
webertj@14807
   137
(* "maxsize"     int     If >0, maximal size of each ground type/IDT depth.  *)
webertj@14807
   138
(* "maxvars"     int     If >0, use at most 'maxvars' Boolean variables      *)
webertj@14807
   139
(*                       when transforming the term into a propositional     *)
webertj@14807
   140
(*                       formula.                                            *)
webertj@14807
   141
(* "maxtime"     int     If >0, terminate after at most 'maxtime' seconds.   *)
webertj@14807
   142
(* "satsolver"   string  SAT solver to be used.                              *)
webertj@14807
   143
(* ------------------------------------------------------------------------- *)
webertj@14807
   144
webertj@14807
   145
	type params =
webertj@14807
   146
		{
webertj@14807
   147
			sizes    : (string * int) list,
webertj@14807
   148
			minsize  : int,
webertj@14807
   149
			maxsize  : int,
webertj@14807
   150
			maxvars  : int,
webertj@14807
   151
			maxtime  : int,
webertj@14807
   152
			satsolver: string
webertj@14807
   153
		};
webertj@14807
   154
webertj@14807
   155
(* ------------------------------------------------------------------------- *)
webertj@14456
   156
(* interpretation: a term's interpretation is given by a variable of type    *)
webertj@14456
   157
(*                 'interpretation'                                          *)
webertj@14350
   158
(* ------------------------------------------------------------------------- *)
webertj@14350
   159
webertj@14456
   160
	type interpretation =
webertj@14456
   161
		prop_formula list tree;
webertj@14350
   162
webertj@14350
   163
(* ------------------------------------------------------------------------- *)
webertj@14456
   164
(* model: a model specifies the size of types and the interpretation of      *)
webertj@14456
   165
(*        terms                                                              *)
webertj@14350
   166
(* ------------------------------------------------------------------------- *)
webertj@14350
   167
webertj@14456
   168
	type model =
webertj@14456
   169
		(Term.typ * int) list * (Term.term * interpretation) list;
webertj@14350
   170
webertj@14456
   171
(* ------------------------------------------------------------------------- *)
webertj@14456
   172
(* arguments: additional arguments required during interpretation of terms   *)
webertj@14456
   173
(* ------------------------------------------------------------------------- *)
webertj@14807
   174
webertj@14456
   175
	type arguments =
webertj@14807
   176
		{
webertj@14807
   177
			(* just passed unchanged from 'params' *)
webertj@14807
   178
			maxvars   : int,
webertj@14807
   179
			(* these may change during the translation *)
webertj@14807
   180
			next_idx  : int,
webertj@14807
   181
			bounds    : interpretation list,
webertj@14807
   182
			wellformed: prop_formula
webertj@14807
   183
		};
webertj@14456
   184
webertj@14350
   185
webertj@14456
   186
	structure RefuteDataArgs =
webertj@14456
   187
	struct
webertj@14456
   188
		val name = "HOL/refute";
webertj@14456
   189
		type T =
webertj@14456
   190
			{interpreters: (string * (theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) option)) list,
webertj@14807
   191
			 printers: (string * (theory -> model -> Term.term -> interpretation -> (int -> bool) -> Term.term option)) list,
webertj@14456
   192
			 parameters: string Symtab.table};
webertj@14456
   193
		val empty = {interpreters = [], printers = [], parameters = Symtab.empty};
webertj@14456
   194
		val copy = I;
webertj@14456
   195
		val prep_ext = I;
webertj@14456
   196
		fun merge
webertj@14456
   197
			({interpreters = in1, printers = pr1, parameters = pa1},
webertj@14456
   198
			 {interpreters = in2, printers = pr2, parameters = pa2}) =
webertj@14456
   199
			{interpreters = rev (merge_alists (rev in1) (rev in2)),
webertj@14456
   200
			 printers = rev (merge_alists (rev pr1) (rev pr2)),
webertj@14456
   201
			 parameters = Symtab.merge (op=) (pa1, pa2)};
webertj@14456
   202
		fun print sg {interpreters, printers, parameters} =
webertj@14456
   203
			Pretty.writeln (Pretty.chunks
webertj@14456
   204
				[Pretty.strs ("default parameters:" :: flat (map (fn (name,value) => [name, "=", value]) (Symtab.dest parameters))),
webertj@14456
   205
				 Pretty.strs ("interpreters:" :: map fst interpreters),
webertj@14456
   206
				 Pretty.strs ("printers:" :: map fst printers)]);
webertj@14456
   207
	end;
webertj@14456
   208
webertj@14456
   209
	structure RefuteData = TheoryDataFun(RefuteDataArgs);
webertj@14456
   210
webertj@14350
   211
webertj@14350
   212
(* ------------------------------------------------------------------------- *)
webertj@14456
   213
(* interpret: tries to interpret the term 't' using a suitable interpreter;  *)
webertj@14456
   214
(*            returns the interpretation and a (possibly extended) model     *)
webertj@14456
   215
(*            that keeps track of the interpretation of subterms             *)
webertj@14807
   216
(* Note: exception 'CANNOT_INTERPRET t' is raised if the term cannot be      *)
webertj@14456
   217
(*       interpreted by any interpreter                                      *)
webertj@14350
   218
(* ------------------------------------------------------------------------- *)
webertj@14350
   219
webertj@14807
   220
	(* theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) *)
webertj@14456
   221
webertj@14456
   222
	fun interpret thy model args t =
webertj@14456
   223
		(case get_first (fn (_, f) => f thy model args t) (#interpreters (RefuteData.get thy)) of
webertj@14807
   224
		  None   => raise (CANNOT_INTERPRET t)
webertj@14456
   225
		| Some x => x);
webertj@14456
   226
webertj@14456
   227
(* ------------------------------------------------------------------------- *)
webertj@14807
   228
(* print: tries to convert the constant denoted by the term 't' into a term  *)
webertj@14807
   229
(*        using a suitable printer                                           *)
webertj@14456
   230
(* ------------------------------------------------------------------------- *)
webertj@14350
   231
webertj@14807
   232
	(* theory -> model -> Term.term -> interpretation -> (int -> bool) -> Term.term *)
webertj@14456
   233
webertj@14456
   234
	fun print thy model t intr assignment =
webertj@14456
   235
		(case get_first (fn (_, f) => f thy model t intr assignment) (#printers (RefuteData.get thy)) of
webertj@14807
   236
		  None   => Const ("<<no printer available>>", fastype_of t)
webertj@14807
   237
		| Some x => x);
webertj@14456
   238
webertj@14456
   239
(* ------------------------------------------------------------------------- *)
webertj@14456
   240
(* print_model: turns the model into a string, using a fixed interpretation  *)
webertj@14807
   241
(*              (given by an assignment for Boolean variables) and suitable  *)
webertj@14456
   242
(*              printers                                                     *)
webertj@14456
   243
(* ------------------------------------------------------------------------- *)
webertj@14456
   244
webertj@14807
   245
	(* theory -> model -> (int -> bool) -> string *)
webertj@14807
   246
webertj@14456
   247
	fun print_model thy model assignment =
webertj@14456
   248
	let
webertj@14456
   249
		val (typs, terms) = model
webertj@14807
   250
		val typs_msg =
webertj@14807
   251
			if null typs then
webertj@14807
   252
				"empty universe (no type variables in term)\n"
webertj@14807
   253
			else
webertj@14807
   254
				"Size of types: " ^ commas (map (fn (T,i) => Sign.string_of_typ (sign_of thy) T ^ ": " ^ string_of_int i) typs) ^ "\n"
webertj@14807
   255
		val show_consts_msg =
webertj@14807
   256
			if not (!show_consts) andalso Library.exists (is_Const o fst) terms then
webertj@14807
   257
				"set \"show_consts\" to show the interpretation of constants\n"
webertj@14807
   258
			else
webertj@14807
   259
				""
webertj@14807
   260
		val terms_msg =
webertj@14807
   261
			if null terms then
webertj@14807
   262
				"empty interpretation (no free variables in term)\n"
webertj@14807
   263
			else
webertj@14807
   264
				space_implode "\n" (mapfilter (fn (t,intr) =>
webertj@14807
   265
					(* print constants only if 'show_consts' is true *)
webertj@14807
   266
					if (!show_consts) orelse not (is_Const t) then
webertj@14807
   267
						Some (Sign.string_of_term (sign_of thy) t ^ ": " ^ Sign.string_of_term (sign_of thy) (print thy model t intr assignment))
webertj@14807
   268
					else
webertj@14807
   269
						None) terms) ^ "\n"
webertj@14456
   270
	in
webertj@14807
   271
		typs_msg ^ show_consts_msg ^ terms_msg
webertj@14456
   272
	end;
webertj@14456
   273
webertj@14456
   274
webertj@14456
   275
(* ------------------------------------------------------------------------- *)
webertj@14456
   276
(* PARAMETER MANAGEMENT                                                      *)
webertj@14456
   277
(* ------------------------------------------------------------------------- *)
webertj@14456
   278
webertj@14456
   279
	(* string -> (theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) option) -> theory -> theory *)
webertj@14456
   280
webertj@14456
   281
	fun add_interpreter name f thy =
webertj@14456
   282
	let
webertj@14456
   283
		val {interpreters, printers, parameters} = RefuteData.get thy
webertj@14456
   284
	in
webertj@14456
   285
		case assoc (interpreters, name) of
webertj@14456
   286
		  None   => RefuteData.put {interpreters = (name, f) :: interpreters, printers = printers, parameters = parameters} thy
webertj@14456
   287
		| Some _ => error ("Interpreter " ^ name ^ " already declared")
webertj@14456
   288
	end;
webertj@14456
   289
webertj@14807
   290
	(* string -> (theory -> model -> Term.term -> interpretation -> (int -> bool) -> Term.term option) -> theory -> theory *)
webertj@14456
   291
webertj@14456
   292
	fun add_printer name f thy =
webertj@14456
   293
	let
webertj@14456
   294
		val {interpreters, printers, parameters} = RefuteData.get thy
webertj@14456
   295
	in
webertj@14456
   296
		case assoc (printers, name) of
webertj@14456
   297
		  None   => RefuteData.put {interpreters = interpreters, printers = (name, f) :: printers, parameters = parameters} thy
webertj@14456
   298
		| Some _ => error ("Printer " ^ name ^ " already declared")
webertj@14456
   299
	end;
webertj@14456
   300
webertj@14456
   301
(* ------------------------------------------------------------------------- *)
webertj@14456
   302
(* set_default_param: stores the '(name, value)' pair in RefuteData's        *)
webertj@14456
   303
(*                    parameter table                                        *)
webertj@14456
   304
(* ------------------------------------------------------------------------- *)
webertj@14456
   305
webertj@14456
   306
	(* (string * string) -> theory -> theory *)
webertj@14456
   307
webertj@14456
   308
	fun set_default_param (name, value) thy =
webertj@14456
   309
	let
webertj@14456
   310
		val {interpreters, printers, parameters} = RefuteData.get thy
webertj@14456
   311
	in
webertj@14456
   312
		case Symtab.lookup (parameters, name) of
webertj@14456
   313
		  None   => RefuteData.put
webertj@14456
   314
			{interpreters = interpreters, printers = printers, parameters = Symtab.extend (parameters, [(name, value)])} thy
webertj@14456
   315
		| Some _ => RefuteData.put
webertj@14456
   316
			{interpreters = interpreters, printers = printers, parameters = Symtab.update ((name, value), parameters)} thy
webertj@14350
   317
	end;
webertj@14350
   318
webertj@14350
   319
(* ------------------------------------------------------------------------- *)
webertj@14456
   320
(* get_default_param: retrieves the value associated with 'name' from        *)
webertj@14456
   321
(*                    RefuteData's parameter table                           *)
webertj@14456
   322
(* ------------------------------------------------------------------------- *)
webertj@14456
   323
webertj@14456
   324
	(* theory -> string -> string option *)
webertj@14456
   325
webertj@14456
   326
	fun get_default_param thy name = Symtab.lookup ((#parameters o RefuteData.get) thy, name);
webertj@14456
   327
webertj@14456
   328
(* ------------------------------------------------------------------------- *)
webertj@14456
   329
(* get_default_params: returns a list of all '(name, value)' pairs that are  *)
webertj@14456
   330
(*                     stored in RefuteData's parameter table                *)
webertj@14456
   331
(* ------------------------------------------------------------------------- *)
webertj@14456
   332
webertj@14456
   333
	(* theory -> (string * string) list *)
webertj@14456
   334
webertj@14456
   335
	fun get_default_params thy = (Symtab.dest o #parameters o RefuteData.get) thy;
webertj@14456
   336
webertj@14456
   337
(* ------------------------------------------------------------------------- *)
webertj@14456
   338
(* actual_params: takes a (possibly empty) list 'params' of parameters that  *)
webertj@14456
   339
(*      override the default parameters currently specified in 'thy', and    *)
webertj@14807
   340
(*      returns a record that can be passed to 'find_model'.                 *)
webertj@14456
   341
(* ------------------------------------------------------------------------- *)
webertj@14456
   342
webertj@14807
   343
	(* theory -> (string * string) list -> params *)
webertj@14456
   344
webertj@14807
   345
	fun actual_params thy override =
webertj@14456
   346
	let
webertj@14456
   347
		(* (string * string) list * string -> int *)
webertj@14456
   348
		fun read_int (parms, name) =
webertj@14456
   349
			case assoc_string (parms, name) of
webertj@14456
   350
			  Some s => (case Int.fromString s of
webertj@14456
   351
				  SOME i => i
webertj@14456
   352
				| NONE   => error ("parameter " ^ quote name ^ " (value is " ^ quote s ^ ") must be an integer value"))
webertj@14456
   353
			| None   => error ("parameter " ^ quote name ^ " must be assigned a value")
webertj@14456
   354
		(* (string * string) list * string -> string *)
webertj@14456
   355
		fun read_string (parms, name) =
webertj@14456
   356
			case assoc_string (parms, name) of
webertj@14456
   357
			  Some s => s
webertj@14456
   358
			| None   => error ("parameter " ^ quote name ^ " must be assigned a value")
webertj@14456
   359
		(* (string * string) list *)
webertj@14807
   360
		val allparams = override @ (get_default_params thy)  (* 'override' first, defaults last *)
webertj@14456
   361
		(* int *)
webertj@14456
   362
		val minsize   = read_int (allparams, "minsize")
webertj@14456
   363
		val maxsize   = read_int (allparams, "maxsize")
webertj@14456
   364
		val maxvars   = read_int (allparams, "maxvars")
webertj@14807
   365
      val maxtime   = read_int (allparams, "maxtime")
webertj@14456
   366
		(* string *)
webertj@14456
   367
		val satsolver = read_string (allparams, "satsolver")
webertj@14807
   368
		(* all remaining parameters of the form "string=int" are collected in  *)
webertj@14807
   369
		(* 'sizes'                                                             *)
webertj@14807
   370
		(* TODO: it is currently not possible to specify a size for a type     *)
webertj@14807
   371
		(*       whose name is one of the other parameters (e.g. 'maxvars')    *)
webertj@14807
   372
		(* (string * int) list *)
webertj@14807
   373
		val sizes     = mapfilter
webertj@14807
   374
			(fn (name,value) => (case Int.fromString value of SOME i => Some (name, i) | NONE => None))
webertj@14807
   375
			(filter (fn (name,_) => name<>"minsize" andalso name<>"maxsize" andalso name<>"maxvars" andalso name<>"maxtime" andalso name<>"satsolver")
webertj@14807
   376
				allparams)
webertj@14456
   377
	in
webertj@14807
   378
		{sizes=sizes, minsize=minsize, maxsize=maxsize, maxvars=maxvars, maxtime=maxtime, satsolver=satsolver}
webertj@14807
   379
	end;
webertj@14807
   380
webertj@14807
   381
webertj@14807
   382
(* ------------------------------------------------------------------------- *)
webertj@14807
   383
(* TRANSLATION HOL -> PROPOSITIONAL LOGIC, BOOLEAN ASSIGNMENT -> MODEL       *)
webertj@14807
   384
(* ------------------------------------------------------------------------- *)
webertj@14807
   385
webertj@14807
   386
(* ------------------------------------------------------------------------- *)
webertj@14807
   387
(* collect_axioms: collects (monomorphic, universally quantified versions    *)
webertj@14807
   388
(*                 of) all HOL axioms that are relevant w.r.t 't'            *)
webertj@14807
   389
(* ------------------------------------------------------------------------- *)
webertj@14807
   390
webertj@14807
   391
	(* TODO: to make the collection of axioms more easily extensible, this    *)
webertj@14807
   392
	(*       function could be based on user-supplied "axiom collectors",     *)
webertj@14807
   393
	(*       similar to 'interpret'/interpreters or 'print'/printers          *)
webertj@14807
   394
webertj@14807
   395
	(* theory -> Term.term -> Term.term list *)
webertj@14807
   396
webertj@14807
   397
	(* Which axioms are "relevant" for a particular term/type goes hand in    *)
webertj@14807
   398
	(* hand with the interpretation of that term/type by its interpreter (see *)
webertj@14807
   399
	(* way below): if the interpretation respects an axiom anyway, the axiom  *)
webertj@14807
   400
	(* does not need to be added as a constraint here.                        *)
webertj@14807
   401
webertj@14807
   402
	(* When an axiom is added as relevant, further axioms may need to be      *)
webertj@14807
   403
	(* added as well (e.g. when a constant is defined in terms of other       *)
webertj@14807
   404
	(* constants).  To avoid infinite recursion (which should not happen for  *)
webertj@14807
   405
	(* constants anyway, but it could happen for "typedef"-related axioms,    *)
webertj@14807
   406
	(* since they contain the type again), we use an accumulator 'axs' and    *)
webertj@14807
   407
	(* add a relevant axiom only if it is not in 'axs' yet.                   *)
webertj@14807
   408
webertj@14807
   409
	fun collect_axioms thy t =
webertj@14807
   410
	let
webertj@14807
   411
		val _ = std_output "Adding axioms..."
webertj@14807
   412
		(* (string * Term.term) list *)
webertj@14807
   413
		val axioms = flat (map (Symtab.dest o #axioms o Theory.rep_theory) (thy :: Theory.ancestors_of thy))
webertj@14807
   414
		(* given a constant 's' of type 'T', which is a subterm of 't', where  *)
webertj@14807
   415
		(* 't' has a (possibly) more general type, the schematic type          *)
webertj@14807
   416
		(* variables in 't' are instantiated to match the type 'T'             *)
webertj@14807
   417
		(* (string * Term.typ) * Term.term -> Term.term *)
webertj@14807
   418
		fun specialize_type ((s, T), t) =
webertj@14807
   419
		let
webertj@14807
   420
			fun find_typeSubs (Const (s', T')) =
webertj@14807
   421
				(if s=s' then
webertj@14807
   422
					Some (Type.typ_match (Sign.tsig_of (sign_of thy)) (Vartab.empty, (T', T)))
webertj@14807
   423
				else
webertj@14807
   424
					None
webertj@14807
   425
				handle Type.TYPE_MATCH => None)
webertj@14807
   426
			  | find_typeSubs (Free _)           = None
webertj@14807
   427
			  | find_typeSubs (Var _)            = None
webertj@14807
   428
			  | find_typeSubs (Bound _)          = None
webertj@14807
   429
			  | find_typeSubs (Abs (_, _, body)) = find_typeSubs body
webertj@14807
   430
			  | find_typeSubs (t1 $ t2)          = (case find_typeSubs t1 of Some x => Some x | None => find_typeSubs t2)
webertj@14807
   431
			val typeSubs = (case find_typeSubs t of
webertj@14807
   432
				  Some x => x
webertj@14807
   433
				| None   => raise REFUTE ("collect_axioms", "no type instantiation found for " ^ quote s ^ " in " ^ Sign.string_of_term (sign_of thy) t))
webertj@14807
   434
		in
webertj@14807
   435
			map_term_types
webertj@14807
   436
				(map_type_tvar
webertj@14807
   437
					(fn (v,_) =>
webertj@14807
   438
						case Vartab.lookup (typeSubs, v) of
webertj@14807
   439
						  None =>
webertj@14807
   440
							(* schematic type variable not instantiated *)
webertj@14807
   441
							raise REFUTE ("collect_axioms", "term " ^ Sign.string_of_term (sign_of thy) t ^ " still has a polymorphic type (after instantiating type of " ^ quote s ^ ")")
webertj@14807
   442
						| Some typ =>
webertj@14807
   443
							typ))
webertj@14807
   444
					t
webertj@14807
   445
		end
webertj@14807
   446
		(* Term.term list * Term.typ -> Term.term list *)
webertj@14807
   447
		fun collect_type_axioms (axs, T) =
webertj@14807
   448
			case T of
webertj@14807
   449
			(* simple types *)
webertj@14807
   450
			  Type ("prop", [])      => axs
webertj@14807
   451
			| Type ("fun", [T1, T2]) => collect_type_axioms (collect_type_axioms (axs, T1), T2)
webertj@14807
   452
			| Type ("set", [T1])     => collect_type_axioms (axs, T1)
webertj@14807
   453
			| Type (s, Ts)           =>
webertj@14807
   454
				let
webertj@14807
   455
					(* look up the definition of a type, as created by "typedef" *)
webertj@14807
   456
					(* (string * Term.term) list -> (string * Term.term) option *)
webertj@14807
   457
					fun get_typedefn [] =
webertj@14807
   458
						None
webertj@14807
   459
					  | get_typedefn ((axname,ax)::axms) =
webertj@14807
   460
						(let
webertj@14807
   461
							(* Term.term -> Term.typ option *)
webertj@14807
   462
							fun type_of_type_definition (Const (s', T')) =
webertj@14807
   463
								if s'="Typedef.type_definition" then
webertj@14807
   464
									Some T'
webertj@14807
   465
								else
webertj@14807
   466
									None
webertj@14807
   467
							  | type_of_type_definition (Free _)           = None
webertj@14807
   468
							  | type_of_type_definition (Var _)            = None
webertj@14807
   469
							  | type_of_type_definition (Bound _)          = None
webertj@14807
   470
							  | type_of_type_definition (Abs (_, _, body)) = type_of_type_definition body
webertj@14807
   471
							  | type_of_type_definition (t1 $ t2)          = (case type_of_type_definition t1 of Some x => Some x | None => type_of_type_definition t2)
webertj@14807
   472
						in
webertj@14807
   473
							case type_of_type_definition ax of
webertj@14807
   474
							  Some T' =>
webertj@14807
   475
								let
webertj@14807
   476
									val T''      = (domain_type o domain_type) T'
webertj@14807
   477
									val typeSubs = Type.typ_match (Sign.tsig_of (sign_of thy)) (Vartab.empty, (T'', T))
webertj@14807
   478
									val unvar_ax = map_term_types
webertj@14807
   479
										(map_type_tvar
webertj@14807
   480
											(fn (v,_) =>
webertj@14807
   481
												case Vartab.lookup (typeSubs, v) of
webertj@14807
   482
												  None =>
webertj@14807
   483
													(* schematic type variable not instantiated *)
webertj@14807
   484
													raise ERROR
webertj@14807
   485
												| Some typ =>
webertj@14807
   486
													typ))
webertj@14807
   487
										ax
webertj@14807
   488
								in
webertj@14807
   489
									Some (axname, unvar_ax)
webertj@14807
   490
								end
webertj@14807
   491
							| None =>
webertj@14807
   492
								get_typedefn axms
webertj@14807
   493
						end
webertj@14807
   494
						handle ERROR           => get_typedefn axms
webertj@14807
   495
						     | MATCH           => get_typedefn axms
webertj@14807
   496
						     | Type.TYPE_MATCH => get_typedefn axms)
webertj@14807
   497
				in
webertj@14807
   498
					case DatatypePackage.datatype_info thy s of
webertj@14807
   499
					  Some info =>  (* inductive datatype *)
webertj@14807
   500
							(* only collect relevant type axioms for the argument types *)
webertj@14807
   501
							foldl collect_type_axioms (axs, Ts)
webertj@14807
   502
					| None =>
webertj@14807
   503
						(case get_typedefn axioms of
webertj@14807
   504
						  Some (axname, ax) => 
webertj@14807
   505
							if mem_term (ax, axs) then
webertj@14807
   506
								(* collect relevant type axioms for the argument types *)
webertj@14807
   507
								foldl collect_type_axioms (axs, Ts)
webertj@14807
   508
							else
webertj@14807
   509
								(std_output (" " ^ axname);
webertj@14807
   510
								collect_term_axioms (ax :: axs, ax))
webertj@14807
   511
						| None =>
webertj@14807
   512
							(* at least collect relevant type axioms for the argument types *)
webertj@14807
   513
							foldl collect_type_axioms (axs, Ts))
webertj@14807
   514
				end
webertj@14807
   515
			(* TODO: include sort axioms *)
webertj@14807
   516
			| TFree (_, sorts)       => ((*if not (null sorts) then std_output " *ignoring sorts*" else ();*) axs)
webertj@14807
   517
			| TVar  (_, sorts)       => ((*if not (null sorts) then std_output " *ignoring sorts*" else ();*) axs)
webertj@14807
   518
		(* Term.term list * Term.term -> Term.term list *)
webertj@14807
   519
		and collect_term_axioms (axs, t) =
webertj@14807
   520
			case t of
webertj@14807
   521
			(* Pure *)
webertj@14807
   522
			  Const ("all", _)                => axs
webertj@14807
   523
			| Const ("==", _)                 => axs
webertj@14807
   524
			| Const ("==>", _)                => axs
webertj@14807
   525
			(* HOL *)
webertj@14807
   526
			| Const ("Trueprop", _)           => axs
webertj@14807
   527
			| Const ("Not", _)                => axs
webertj@14807
   528
			| Const ("True", _)               => axs  (* redundant, since 'True' is also an IDT constructor *)
webertj@14807
   529
			| Const ("False", _)              => axs  (* redundant, since 'False' is also an IDT constructor *)
webertj@14807
   530
			| Const ("arbitrary", T)          => collect_type_axioms (axs, T)
webertj@14807
   531
			| Const ("The", T)                =>
webertj@14807
   532
				let
webertj@14807
   533
					val ax = specialize_type (("The", T), (the o assoc) (axioms, "HOL.the_eq_trivial"))
webertj@14807
   534
				in
webertj@14807
   535
					if mem_term (ax, axs) then
webertj@14807
   536
						collect_type_axioms (axs, T)
webertj@14807
   537
					else
webertj@14807
   538
						(std_output " HOL.the_eq_trivial";
webertj@14807
   539
						collect_term_axioms (ax :: axs, ax))
webertj@14807
   540
				end
webertj@14807
   541
			| Const ("Hilbert_Choice.Eps", T) =>
webertj@14807
   542
				let
webertj@14807
   543
					val ax = specialize_type (("Hilbert_Choice.Eps", T), (the o assoc) (axioms, "Hilbert_Choice.someI"))
webertj@14807
   544
				in
webertj@14807
   545
					if mem_term (ax, axs) then
webertj@14807
   546
						collect_type_axioms (axs, T)
webertj@14807
   547
					else
webertj@14807
   548
						(std_output " Hilbert_Choice.someI";
webertj@14807
   549
						collect_term_axioms (ax :: axs, ax))
webertj@14807
   550
				end
webertj@14807
   551
			| Const ("All", _) $ t1           => collect_term_axioms (axs, t1)
webertj@14807
   552
			| Const ("Ex", _) $ t1            => collect_term_axioms (axs, t1)
webertj@14807
   553
			| Const ("op =", T)               => collect_type_axioms (axs, T)
webertj@14807
   554
			| Const ("op &", _)               => axs
webertj@14807
   555
			| Const ("op |", _)               => axs
webertj@14807
   556
			| Const ("op -->", _)             => axs
webertj@14807
   557
			(* sets *)
webertj@14807
   558
			| Const ("Collect", T)            => collect_type_axioms (axs, T)
webertj@14807
   559
			| Const ("op :", T)               => collect_type_axioms (axs, T)
webertj@14807
   560
			(* other optimizations *)
webertj@14807
   561
			| Const ("Finite_Set.card", T)    => collect_type_axioms (axs, T)
webertj@14807
   562
			(* simply-typed lambda calculus *)
webertj@14807
   563
			| Const (s, T)                    =>
webertj@14807
   564
				let
webertj@14807
   565
					(* look up the definition of a constant, as created by "constdefs" *)
webertj@14807
   566
					(* string -> Term.typ -> (string * Term.term) list -> (string * Term.term) option *)
webertj@14807
   567
					fun get_defn [] =
webertj@14807
   568
						None
webertj@14807
   569
					  | get_defn ((axname,ax)::axms) =
webertj@14807
   570
						(let
webertj@14807
   571
							val (lhs, _) = Logic.dest_equals ax  (* equations only *)
webertj@14807
   572
							val c        = head_of lhs
webertj@14807
   573
							val (s', T') = dest_Const c
webertj@14807
   574
						in
webertj@14807
   575
							if s=s' then
webertj@14807
   576
								let
webertj@14807
   577
									val typeSubs = Type.typ_match (Sign.tsig_of (sign_of thy)) (Vartab.empty, (T', T))
webertj@14807
   578
									val unvar_ax = map_term_types
webertj@14807
   579
										(map_type_tvar
webertj@14807
   580
											(fn (v,_) =>
webertj@14807
   581
												case Vartab.lookup (typeSubs, v) of
webertj@14807
   582
												  None =>
webertj@14807
   583
													(* schematic type variable not instantiated *)
webertj@14807
   584
													raise ERROR
webertj@14807
   585
												| Some typ =>
webertj@14807
   586
													typ))
webertj@14807
   587
										ax
webertj@14807
   588
								in
webertj@14807
   589
									Some (axname, unvar_ax)
webertj@14807
   590
								end
webertj@14807
   591
							else
webertj@14807
   592
								get_defn axms
webertj@14807
   593
						end
webertj@14807
   594
						handle ERROR           => get_defn axms
webertj@14807
   595
						     | TERM _          => get_defn axms
webertj@14807
   596
						     | Type.TYPE_MATCH => get_defn axms)
webertj@14807
   597
						(* unit -> bool *)
webertj@14807
   598
						fun is_IDT_constructor () =
webertj@14807
   599
							(case body_type T of
webertj@14807
   600
							  Type (s', _) =>
webertj@14807
   601
								(case DatatypePackage.constrs_of thy s' of
webertj@14807
   602
								  Some constrs =>
webertj@14807
   603
									Library.exists (fn c =>
webertj@14807
   604
										(case c of
webertj@14807
   605
										  Const (cname, ctype) =>
webertj@14810
   606
											cname = s andalso Type.typ_instance (Sign.tsig_of (sign_of thy)) (T, ctype)
webertj@14807
   607
										| _ =>
webertj@14807
   608
											raise REFUTE ("collect_axioms", "IDT constructor is not a constant")))
webertj@14807
   609
										constrs
webertj@14807
   610
								| None =>
webertj@14807
   611
									false)
webertj@14807
   612
							| _  =>
webertj@14807
   613
								false)
webertj@14807
   614
						(* unit -> bool *)
webertj@14807
   615
						fun is_IDT_recursor () =
webertj@14807
   616
							(* the type of a recursion operator: [T1,...,Tn,IDT]--->TResult (where *)
webertj@14807
   617
							(* the T1,...,Tn depend on the types of the datatype's constructors)   *)
webertj@14807
   618
							((case last_elem (binder_types T) of
webertj@14807
   619
							  Type (s', _) =>
webertj@14807
   620
								(case DatatypePackage.datatype_info thy s' of
webertj@14807
   621
								  Some info =>
webertj@14807
   622
									(* TODO: I'm not quite sute if comparing the names is sufficient, or if *)
webertj@14807
   623
									(*       we should also check the type                                  *)
webertj@14807
   624
									s mem (#rec_names info)
webertj@14807
   625
								| None =>  (* not an inductive datatype *)
webertj@14807
   626
									false)
webertj@14807
   627
							| _ =>  (* a (free or schematic) type variable *)
webertj@14807
   628
								false)
webertj@14807
   629
							handle LIST "last_elem" => false)  (* not even a function type *)
webertj@14807
   630
				in
webertj@14807
   631
					if is_IDT_constructor () orelse is_IDT_recursor () then
webertj@14807
   632
						(* only collect relevant type axioms *)
webertj@14807
   633
						collect_type_axioms (axs, T)
webertj@14807
   634
					else
webertj@14807
   635
						(case get_defn axioms of
webertj@14807
   636
						  Some (axname, ax) => 
webertj@14807
   637
							if mem_term (ax, axs) then
webertj@14807
   638
								(* collect relevant type axioms *)
webertj@14807
   639
								collect_type_axioms (axs, T)
webertj@14807
   640
							else
webertj@14807
   641
								(std_output (" " ^ axname);
webertj@14807
   642
								collect_term_axioms (ax :: axs, ax))
webertj@14807
   643
						| None =>
webertj@14807
   644
							(* collect relevant type axioms *)
webertj@14807
   645
							collect_type_axioms (axs, T))
webertj@14807
   646
				end
webertj@14807
   647
			| Free (_, T)                     => collect_type_axioms (axs, T)
webertj@14807
   648
			| Var (_, T)                      => collect_type_axioms (axs, T)
webertj@14807
   649
			| Bound i                         => axs
webertj@14807
   650
			| Abs (_, T, body)                => collect_term_axioms (collect_type_axioms (axs, T), body)
webertj@14807
   651
			| t1 $ t2                         => collect_term_axioms (collect_term_axioms (axs, t1), t2)
webertj@14807
   652
		(* universal closure over schematic variables *)
webertj@14807
   653
		(* Term.term -> Term.term *)
webertj@14807
   654
		fun close_form t =
webertj@14807
   655
		let
webertj@14807
   656
			(* (Term.indexname * Term.typ) list *)
webertj@14807
   657
			val vars = sort_wrt (fst o fst) (map dest_Var (term_vars t))
webertj@14807
   658
		in
webertj@14807
   659
			foldl
webertj@14807
   660
				(fn (t', ((x,i),T)) => (Term.all T) $ Abs (x, T, abstract_over (Var((x,i),T), t')))
webertj@14807
   661
				(t, vars)
webertj@14807
   662
		end
webertj@14807
   663
		(* Term.term list *)
webertj@14807
   664
		val result = map close_form (collect_term_axioms ([], t))
webertj@14807
   665
		val _ = writeln " ...done."
webertj@14807
   666
	in
webertj@14807
   667
		result
webertj@14456
   668
	end;
webertj@14456
   669
webertj@14456
   670
(* ------------------------------------------------------------------------- *)
webertj@14807
   671
(* ground_types: collects all ground types in a term (including argument     *)
webertj@14807
   672
(*               types of other types), suppressing duplicates.  Does not    *)
webertj@14807
   673
(*               return function types, set types, non-recursive IDTs, or    *)
webertj@14807
   674
(*               'propT'.  For IDTs, also the argument types of constructors *)
webertj@14807
   675
(*               are considered.                                             *)
webertj@14807
   676
(* ------------------------------------------------------------------------- *)
webertj@14807
   677
webertj@14807
   678
	(* theory -> Term.term -> Term.typ list *)
webertj@14807
   679
webertj@14807
   680
	fun ground_types thy t =
webertj@14807
   681
	let
webertj@14807
   682
		(* Term.typ * Term.typ list -> Term.typ list *)
webertj@14807
   683
		fun collect_types (T, acc) =
webertj@14807
   684
			if T mem acc then
webertj@14807
   685
				acc  (* prevent infinite recursion (for IDTs) *)
webertj@14807
   686
			else
webertj@14807
   687
				(case T of
webertj@14807
   688
				  Type ("fun", [T1, T2]) => collect_types (T1, collect_types (T2, acc))
webertj@14807
   689
				| Type ("prop", [])      => acc
webertj@14807
   690
				| Type ("set", [T1])     => collect_types (T1, acc)
webertj@14807
   691
				| Type (s, Ts)           =>
webertj@14807
   692
					(case DatatypePackage.datatype_info thy s of
webertj@14807
   693
					  Some info =>  (* inductive datatype *)
webertj@14807
   694
						let
webertj@14807
   695
							val index               = #index info
webertj@14807
   696
							val descr               = #descr info
webertj@14807
   697
							val (_, dtyps, constrs) = (the o assoc) (descr, index)
webertj@14807
   698
							val typ_assoc           = dtyps ~~ Ts
webertj@14807
   699
							(* sanity check: every element in 'dtyps' must be a 'DtTFree' *)
webertj@14807
   700
							val _ = (if Library.exists (fn d =>
webertj@14807
   701
									case d of DatatypeAux.DtTFree _ => false | _ => true) dtyps
webertj@14807
   702
								then
webertj@14807
   703
									raise REFUTE ("ground_types", "datatype argument (for type " ^ Sign.string_of_typ (sign_of thy) (Type (s, Ts)) ^ ") is not a variable")
webertj@14807
   704
								else
webertj@14807
   705
									())
webertj@14807
   706
							(* DatatypeAux.descr -> (DatatypeAux.dtyp * Term.typ) list -> DatatypeAux.dtyp -> Term.typ *)
webertj@14807
   707
							fun typ_of_dtyp descr typ_assoc (DatatypeAux.DtTFree a) =
webertj@14807
   708
								(* replace a 'DtTFree' variable by the associated type *)
webertj@14807
   709
								(the o assoc) (typ_assoc, DatatypeAux.DtTFree a)
webertj@14807
   710
							  | typ_of_dtyp descr typ_assoc (DatatypeAux.DtRec i) =
webertj@14807
   711
								let
webertj@14807
   712
									val (s, ds, _) = (the o assoc) (descr, i)
webertj@14807
   713
								in
webertj@14807
   714
									Type (s, map (typ_of_dtyp descr typ_assoc) ds)
webertj@14807
   715
								end
webertj@14807
   716
							  | typ_of_dtyp descr typ_assoc (DatatypeAux.DtType (s, ds)) =
webertj@14807
   717
								Type (s, map (typ_of_dtyp descr typ_assoc) ds)
webertj@14807
   718
							(* if the current type is a recursive IDT (i.e. a depth is required), add it to 'acc' *)
webertj@14807
   719
							val acc' = (if Library.exists (fn (_, ds) => Library.exists DatatypeAux.is_rec_type ds) constrs then
webertj@14807
   720
									T ins acc
webertj@14807
   721
								else
webertj@14807
   722
									acc)
webertj@14807
   723
							(* collect argument types *)
webertj@14807
   724
							val acc_args = foldr collect_types (Ts, acc')
webertj@14807
   725
							(* collect constructor types *)
webertj@14807
   726
							val acc_constrs = foldr collect_types (flat (map (fn (_, ds) => map (typ_of_dtyp descr typ_assoc) ds) constrs), acc_args)
webertj@14807
   727
						in
webertj@14807
   728
							acc_constrs
webertj@14807
   729
						end
webertj@14807
   730
					| None =>  (* not an inductive datatype, e.g. defined via "typedef" or "typedecl" *)
webertj@14807
   731
						T ins (foldr collect_types (Ts, acc)))
webertj@14807
   732
				| TFree _                => T ins acc
webertj@14807
   733
				| TVar _                 => T ins acc)
webertj@14807
   734
	in
webertj@14807
   735
		it_term_types collect_types (t, [])
webertj@14807
   736
	end;
webertj@14807
   737
webertj@14807
   738
(* ------------------------------------------------------------------------- *)
webertj@14807
   739
(* string_of_typ: (rather naive) conversion from types to strings, used to   *)
webertj@14807
   740
(*                look up the size of a type in 'sizes'.  Parameterized      *)
webertj@14807
   741
(*                types with different parameters (e.g. "'a list" vs. "bool  *)
webertj@14807
   742
(*                list") are identified.                                     *)
webertj@14807
   743
(* ------------------------------------------------------------------------- *)
webertj@14807
   744
webertj@14807
   745
	(* Term.typ -> string *)
webertj@14807
   746
webertj@14807
   747
	fun string_of_typ (Type (s, _))     = s
webertj@14807
   748
	  | string_of_typ (TFree (s, _))    = s
webertj@14807
   749
	  | string_of_typ (TVar ((s,_), _)) = s;
webertj@14807
   750
webertj@14807
   751
(* ------------------------------------------------------------------------- *)
webertj@14807
   752
(* first_universe: returns the "first" (i.e. smallest) universe by assigning *)
webertj@14807
   753
(*                 'minsize' to every type for which no size is specified in *)
webertj@14807
   754
(*                 'sizes'                                                   *)
webertj@14807
   755
(* ------------------------------------------------------------------------- *)
webertj@14807
   756
webertj@14807
   757
	(* Term.typ list -> (string * int) list -> int -> (Term.typ * int) list *)
webertj@14807
   758
webertj@14807
   759
	fun first_universe xs sizes minsize =
webertj@14807
   760
	let
webertj@14807
   761
		fun size_of_typ T =
webertj@14807
   762
			case assoc (sizes, string_of_typ T) of
webertj@14807
   763
			  Some n => n
webertj@14807
   764
			| None   => minsize
webertj@14807
   765
	in
webertj@14807
   766
		map (fn T => (T, size_of_typ T)) xs
webertj@14807
   767
	end;
webertj@14807
   768
webertj@14807
   769
(* ------------------------------------------------------------------------- *)
webertj@14807
   770
(* next_universe: enumerates all universes (i.e. assignments of sizes to     *)
webertj@14807
   771
(*                types), where the minimal size of a type is given by       *)
webertj@14807
   772
(*                'minsize', the maximal size is given by 'maxsize', and a   *)
webertj@14807
   773
(*                type may have a fixed size given in 'sizes'                *)
webertj@14456
   774
(* ------------------------------------------------------------------------- *)
webertj@14456
   775
webertj@14807
   776
	(* (Term.typ * int) list -> (string * int) list -> int -> int -> (Term.typ * int) list option *)
webertj@14456
   777
webertj@14807
   778
	fun next_universe xs sizes minsize maxsize =
webertj@14456
   779
	let
webertj@14807
   780
		(* int -> int list -> int list option *)
webertj@14807
   781
		fun add1 _ [] =
webertj@14807
   782
			None  (* overflow *)
webertj@14807
   783
		  | add1 max (x::xs) =
webertj@14807
   784
		 	if x<max orelse max<0 then
webertj@14807
   785
				Some ((x+1)::xs)  (* add 1 to the head *)
webertj@14807
   786
			else
webertj@14807
   787
				apsome (fn xs' => 0 :: xs') (add1 max xs)  (* carry-over *)
webertj@14807
   788
		(* int -> int list * int list -> int list option *)
webertj@14807
   789
		fun shift _ (_, []) =
webertj@14807
   790
			None
webertj@14807
   791
		  | shift max (zeros, x::xs) =
webertj@14807
   792
			if x=0 then
webertj@14807
   793
				shift max (0::zeros, xs)
webertj@14456
   794
			else
webertj@14807
   795
				apsome (fn xs' => (x-1) :: (zeros @ xs')) (add1 max xs)
webertj@14807
   796
		(* creates the "first" list of length 'len', where the sum of all list *)
webertj@14807
   797
		(* elements is 'sum', and the length of the list is 'len'              *)
webertj@14807
   798
		(* int -> int -> int -> int list option *)
webertj@14807
   799
		fun make_first 0 sum _ =
webertj@14807
   800
			if sum=0 then
webertj@14807
   801
				Some []
webertj@14807
   802
			else
webertj@14807
   803
				None
webertj@14807
   804
		  | make_first len sum max =
webertj@14807
   805
			if sum<=max orelse max<0 then
webertj@14807
   806
				apsome (fn xs' => sum :: xs') (make_first (len-1) 0 max)
webertj@14807
   807
			else
webertj@14807
   808
				apsome (fn xs' => max :: xs') (make_first (len-1) (sum-max) max)
webertj@14807
   809
		(* enumerates all int lists with a fixed length, where 0<=x<='max' for *)
webertj@14807
   810
		(* all list elements x (unless 'max'<0)                                *)
webertj@14807
   811
		(* int -> int list -> int list option *)
webertj@14807
   812
		fun next max xs =
webertj@14807
   813
			(case shift max ([], xs) of
webertj@14807
   814
			  Some xs' =>
webertj@14807
   815
				Some xs'
webertj@14807
   816
			| None =>
webertj@14456
   817
				let
webertj@14807
   818
					val (len, sum) = foldl (fn ((l, s), x) => (l+1, s+x)) ((0, 0), xs)
webertj@14456
   819
				in
webertj@14807
   820
					make_first len (sum+1) max  (* increment 'sum' by 1 *)
webertj@14807
   821
				end)
webertj@14807
   822
		(* only consider those types for which the size is not fixed *)
webertj@14807
   823
		val mutables = filter (fn (T, _) => assoc (sizes, string_of_typ T) = None) xs
webertj@14807
   824
		(* subtract 'minsize' from every size (will be added again at the end) *)
webertj@14807
   825
		val diffs = map (fn (_, n) => n-minsize) mutables
webertj@14807
   826
	in
webertj@14807
   827
		case next (maxsize-minsize) diffs of
webertj@14807
   828
		  Some diffs' =>
webertj@14807
   829
			(* merge with those types for which the size is fixed *)
webertj@14807
   830
			Some (snd (foldl_map (fn (ds, (T, _)) =>
webertj@14807
   831
				case assoc (sizes, string_of_typ T) of
webertj@14807
   832
				  Some n => (ds, (T, n))                      (* return the fixed size *)
webertj@14807
   833
				| None   => (tl ds, (T, minsize + (hd ds))))  (* consume the head of 'ds', add 'minsize' *)
webertj@14807
   834
				(diffs', xs)))
webertj@14807
   835
		| None =>
webertj@14807
   836
			None
webertj@14807
   837
	end;
webertj@14807
   838
webertj@14807
   839
(* ------------------------------------------------------------------------- *)
webertj@14807
   840
(* toTrue: converts the interpretation of a Boolean value to a propositional *)
webertj@14807
   841
(*         formula that is true iff the interpretation denotes "true"        *)
webertj@14807
   842
(* ------------------------------------------------------------------------- *)
webertj@14807
   843
webertj@14807
   844
	(* interpretation -> prop_formula *)
webertj@14807
   845
webertj@14807
   846
	fun toTrue (Leaf [fm,_]) = fm
webertj@14807
   847
	  | toTrue _             = raise REFUTE ("toTrue", "interpretation does not denote a Boolean value");
webertj@14807
   848
webertj@14807
   849
(* ------------------------------------------------------------------------- *)
webertj@14807
   850
(* toFalse: converts the interpretation of a Boolean value to a              *)
webertj@14807
   851
(*          propositional formula that is true iff the interpretation        *)
webertj@14807
   852
(*          denotes "false"                                                  *)
webertj@14807
   853
(* ------------------------------------------------------------------------- *)
webertj@14807
   854
webertj@14807
   855
	(* interpretation -> prop_formula *)
webertj@14807
   856
webertj@14807
   857
	fun toFalse (Leaf [_,fm]) = fm
webertj@14807
   858
	  | toFalse _             = raise REFUTE ("toFalse", "interpretation does not denote a Boolean value");
webertj@14807
   859
webertj@14807
   860
(* ------------------------------------------------------------------------- *)
webertj@14807
   861
(* find_model: repeatedly calls 'interpret' with appropriate parameters,     *)
webertj@14807
   862
(*             applies a SAT solver, and (in case a model is found) displays *)
webertj@14807
   863
(*             the model to the user by calling 'print_model'                *)
webertj@14807
   864
(* thy       : the current theory                                            *)
webertj@14807
   865
(* {...}     : parameters that control the translation/model generation      *)
webertj@14807
   866
(* t         : term to be translated into a propositional formula            *)
webertj@14807
   867
(* negate    : if true, find a model that makes 't' false (rather than true) *)
webertj@14807
   868
(* Note: exception 'TimeOut' is raised if the algorithm does not terminate   *)
webertj@14807
   869
(*       within 'maxtime' seconds (if 'maxtime' >0)                          *)
webertj@14807
   870
(* ------------------------------------------------------------------------- *)
webertj@14807
   871
webertj@14807
   872
	(* theory -> params -> Term.term -> bool -> unit *)
webertj@14807
   873
webertj@14807
   874
	fun find_model thy {sizes, minsize, maxsize, maxvars, maxtime, satsolver} t negate =
webertj@14807
   875
	let
webertj@14807
   876
		(* unit -> unit *)
webertj@14807
   877
		fun wrapper () =
webertj@14807
   878
		let
webertj@14807
   879
			(* Term.term list *)
webertj@14807
   880
			val axioms = collect_axioms thy t
webertj@14807
   881
			(* Term.typ list *)
webertj@14807
   882
			val types  = foldl (fn (acc, t') => acc union (ground_types thy t')) ([], t :: axioms)
webertj@14807
   883
			val _      = writeln ("Ground types: "
webertj@14807
   884
				^ (if null types then "none."
webertj@14807
   885
				   else commas (map (Sign.string_of_typ (sign_of thy)) types)))
webertj@14807
   886
			(* (Term.typ * int) list -> unit *)
webertj@14807
   887
			fun find_model_loop universe =
webertj@14807
   888
			(let
webertj@14807
   889
				val init_model             = (universe, [])
webertj@14807
   890
				val init_args              = {maxvars = maxvars, next_idx = 1, bounds = [], wellformed = True}
webertj@14807
   891
				val _                      = std_output ("Translating term (sizes: " ^ commas (map (fn (_, n) => string_of_int n) universe) ^ ") ...")
webertj@14807
   892
				(* translate 't' and all axioms *)
webertj@14807
   893
				val ((model, args), intrs) = foldl_map (fn ((m, a), t') =>
webertj@14807
   894
					let
webertj@14807
   895
						val (i, m', a') = interpret thy m a t'
webertj@14807
   896
					in
webertj@14807
   897
						((m', a'), i)
webertj@14807
   898
					end) ((init_model, init_args), t :: axioms)
webertj@14807
   899
				(* make 't' either true or false, and make all axioms true, and *)
webertj@14807
   900
				(* add the well-formedness side condition                       *)
webertj@14807
   901
				val fm_t  = (if negate then toFalse else toTrue) (hd intrs)
webertj@14807
   902
				val fm_ax = PropLogic.all (map toTrue (tl intrs))
webertj@14807
   903
				val fm    = PropLogic.all [#wellformed args, fm_ax, fm_t]
webertj@14456
   904
			in
webertj@14807
   905
				std_output " invoking SAT solver...";
webertj@14965
   906
				(case SatSolver.invoke_solver satsolver fm of
webertj@14965
   907
				  SatSolver.SATISFIABLE assignment =>
webertj@14965
   908
					writeln ("\n*** Model found: ***\n" ^ print_model thy model (fn i => case assignment i of Some b => b | None => true))
webertj@14965
   909
				| _ =>  (* SatSolver.UNSATISFIABLE, SatSolver.UNKNOWN *)
webertj@14807
   910
					(std_output " no model found.\n";
webertj@14807
   911
					case next_universe universe sizes minsize maxsize of
webertj@14807
   912
					  Some universe' => find_model_loop universe'
webertj@14965
   913
					| None           => writeln "Search terminated, no larger universe within the given limits."))
webertj@14965
   914
				handle SatSolver.NOT_CONFIGURED =>
webertj@14965
   915
					error ("SAT solver " ^ quote satsolver ^ " is not configured.")
webertj@14807
   916
			end handle MAXVARS_EXCEEDED =>
webertj@14807
   917
				writeln ("\nSearch terminated, number of Boolean variables (" ^ string_of_int maxvars ^ " allowed) exceeded.")
webertj@14807
   918
			| CANNOT_INTERPRET t' =>
webertj@14807
   919
				error ("Unable to interpret term " ^ Sign.string_of_term (sign_of thy) t'))
webertj@14807
   920
			in
webertj@14807
   921
				find_model_loop (first_universe types sizes minsize)
webertj@14456
   922
			end
webertj@14807
   923
		in
webertj@14807
   924
			(* some parameter sanity checks *)
webertj@14807
   925
			assert (minsize>=1) ("\"minsize\" is " ^ string_of_int minsize ^ ", must be at least 1");
webertj@14807
   926
			assert (maxsize>=1) ("\"maxsize\" is " ^ string_of_int maxsize ^ ", must be at least 1");
webertj@14807
   927
			assert (maxsize>=minsize) ("\"maxsize\" (=" ^ string_of_int maxsize ^ ") is less than \"minsize\" (=" ^ string_of_int minsize ^ ").");
webertj@14807
   928
			assert (maxvars>=0) ("\"maxvars\" is " ^ string_of_int maxvars ^ ", must be at least 0");
webertj@14807
   929
			assert (maxtime>=0) ("\"maxtime\" is " ^ string_of_int maxtime ^ ", must be at least 0");
webertj@14807
   930
			(* enter loop with/without time limit *)
webertj@14807
   931
			writeln ("Trying to find a model that " ^ (if negate then "refutes" else "satisfies") ^ ": "
webertj@14807
   932
				^ Sign.string_of_term (sign_of thy) t);
webertj@14807
   933
			if maxtime>0 then
webertj@14965
   934
				(TimeLimit.timeLimit (Time.fromSeconds (Int.toLarge maxtime))
webertj@14807
   935
					wrapper ()
webertj@14965
   936
				handle TimeLimit.TimeOut =>
webertj@14807
   937
					writeln ("\nSearch terminated, time limit ("
webertj@14965
   938
						^ string_of_int maxtime ^ (if maxtime=1 then " second" else " seconds")
webertj@14965
   939
						^ ") exceeded."))
webertj@14807
   940
			else
webertj@14807
   941
				wrapper ()
webertj@14807
   942
		end;
webertj@14456
   943
webertj@14456
   944
webertj@14456
   945
(* ------------------------------------------------------------------------- *)
webertj@14456
   946
(* INTERFACE, PART 2: FINDING A MODEL                                        *)
webertj@14350
   947
(* ------------------------------------------------------------------------- *)
webertj@14350
   948
webertj@14350
   949
(* ------------------------------------------------------------------------- *)
webertj@14456
   950
(* satisfy_term: calls 'find_model' to find a model that satisfies 't'       *)
webertj@14456
   951
(* params      : list of '(name, value)' pairs used to override default      *)
webertj@14456
   952
(*               parameters                                                  *)
webertj@14350
   953
(* ------------------------------------------------------------------------- *)
webertj@14350
   954
webertj@14456
   955
	(* theory -> (string * string) list -> Term.term -> unit *)
webertj@14350
   956
webertj@14456
   957
	fun satisfy_term thy params t =
webertj@14807
   958
		find_model thy (actual_params thy params) t false;
webertj@14350
   959
webertj@14350
   960
(* ------------------------------------------------------------------------- *)
webertj@14456
   961
(* refute_term: calls 'find_model' to find a model that refutes 't'          *)
webertj@14456
   962
(* params     : list of '(name, value)' pairs used to override default       *)
webertj@14456
   963
(*              parameters                                                   *)
webertj@14350
   964
(* ------------------------------------------------------------------------- *)
webertj@14350
   965
webertj@14456
   966
	(* theory -> (string * string) list -> Term.term -> unit *)
webertj@14350
   967
webertj@14456
   968
	fun refute_term thy params t =
webertj@14350
   969
	let
webertj@14807
   970
		(* disallow schematic type variables, since we cannot properly negate  *)
webertj@14807
   971
		(* terms containing them (their logical meaning is that there EXISTS a *)
webertj@14807
   972
		(* type s.t. ...; to refute such a formula, we would have to show that *)
webertj@14807
   973
		(* for ALL types, not ...)                                             *)
webertj@14456
   974
		val _ = assert (null (term_tvars t)) "Term to be refuted contains schematic type variables"
webertj@14456
   975
		(* existential closure over schematic variables *)
webertj@14456
   976
		(* (Term.indexname * Term.typ) list *)
webertj@14456
   977
		val vars = sort_wrt (fst o fst) (map dest_Var (term_vars t))
webertj@14456
   978
		(* Term.term *)
webertj@14456
   979
		val ex_closure = foldl
webertj@14456
   980
			(fn (t', ((x,i),T)) => (HOLogic.exists_const T) $ Abs (x, T, abstract_over (Var((x,i),T), t')))
webertj@14456
   981
			(t, vars)
webertj@14456
   982
		(* If 't' is of type 'propT' (rather than 'boolT'), applying  *)
webertj@14456
   983
		(* 'HOLogic.exists_const' is not type-correct.  However, this *)
webertj@14807
   984
		(* is not really a problem as long as 'find_model' still      *)
webertj@14456
   985
		(* interprets the resulting term correctly, without checking  *)
webertj@14456
   986
		(* its type.                                                  *)
webertj@14350
   987
	in
webertj@14807
   988
		find_model thy (actual_params thy params) ex_closure true
webertj@14350
   989
	end;
webertj@14350
   990
webertj@14350
   991
(* ------------------------------------------------------------------------- *)
webertj@14456
   992
(* refute_subgoal: calls 'refute_term' on a specific subgoal                 *)
webertj@14456
   993
(* params        : list of '(name, value)' pairs used to override default    *)
webertj@14456
   994
(*                 parameters                                                *)
webertj@14456
   995
(* subgoal       : 0-based index specifying the subgoal number               *)
webertj@14350
   996
(* ------------------------------------------------------------------------- *)
webertj@14350
   997
webertj@14456
   998
	(* theory -> (string * string) list -> Thm.thm -> int -> unit *)
webertj@14350
   999
webertj@14456
  1000
	fun refute_subgoal thy params thm subgoal =
webertj@14456
  1001
		refute_term thy params (nth_elem (subgoal, prems_of thm));
webertj@14350
  1002
webertj@14350
  1003
webertj@14350
  1004
(* ------------------------------------------------------------------------- *)
webertj@14456
  1005
(* INTERPRETERS                                                              *)
webertj@14350
  1006
(* ------------------------------------------------------------------------- *)
webertj@14350
  1007
webertj@14350
  1008
(* ------------------------------------------------------------------------- *)
webertj@14807
  1009
(* make_constants: returns all interpretations that have the same tree       *)
webertj@14807
  1010
(*                 structure as 'intr', but consist of unit vectors with     *)
webertj@14807
  1011
(*                 'True'/'False' only (no Boolean variables)                *)
webertj@14350
  1012
(* ------------------------------------------------------------------------- *)
webertj@14350
  1013
webertj@14807
  1014
	(* interpretation -> interpretation list *)
webertj@14350
  1015
webertj@14807
  1016
	fun make_constants intr =
webertj@14456
  1017
	let
webertj@14350
  1018
		(* returns a list with all unit vectors of length n *)
webertj@14456
  1019
		(* int -> interpretation list *)
webertj@14350
  1020
		fun unit_vectors n =
webertj@14350
  1021
		let
webertj@14350
  1022
			(* returns the k-th unit vector of length n *)
webertj@14456
  1023
			(* int * int -> interpretation *)
webertj@14350
  1024
			fun unit_vector (k,n) =
webertj@14350
  1025
				Leaf ((replicate (k-1) False) @ (True :: (replicate (n-k) False)))
webertj@14456
  1026
			(* int -> interpretation list -> interpretation list *)
webertj@14350
  1027
			fun unit_vectors_acc k vs =
webertj@14350
  1028
				if k>n then [] else (unit_vector (k,n))::(unit_vectors_acc (k+1) vs)
webertj@14350
  1029
		in
webertj@14350
  1030
			unit_vectors_acc 1 []
webertj@14350
  1031
		end
webertj@14350
  1032
		(* concatenates 'x' with every list in 'xss', returning a new list of lists *)
webertj@14350
  1033
		(* 'a -> 'a list list -> 'a list list *)
webertj@14350
  1034
		fun cons_list x xss =
webertj@14350
  1035
			map (fn xs => x::xs) xss
webertj@14350
  1036
		(* returns a list of lists, each one consisting of n (possibly identical) elements from 'xs' *)
webertj@14350
  1037
		(* int -> 'a list -> 'a list list *)
webertj@14350
  1038
		fun pick_all 1 xs =
webertj@14350
  1039
			map (fn x => [x]) xs
webertj@14350
  1040
		  | pick_all n xs =
webertj@14350
  1041
			let val rec_pick = pick_all (n-1) xs in
webertj@14350
  1042
				foldl (fn (acc,x) => (cons_list x rec_pick) @ acc) ([],xs)
webertj@14350
  1043
			end
webertj@14807
  1044
	in
webertj@14807
  1045
		case intr of
webertj@14807
  1046
		  Leaf xs => unit_vectors (length xs)
webertj@14807
  1047
		| Node xs => map (fn xs' => Node xs') (pick_all (length xs) (make_constants (hd xs)))
webertj@14807
  1048
	end;
webertj@14807
  1049
webertj@14807
  1050
(* ------------------------------------------------------------------------- *)
webertj@14807
  1051
(* size_of_type: returns the number of constants in a type (i.e. 'length     *)
webertj@14807
  1052
(*               (make_constants intr)', but implemented more efficiently)   *)
webertj@14807
  1053
(* ------------------------------------------------------------------------- *)
webertj@14807
  1054
webertj@14807
  1055
	(* interpretation -> int *)
webertj@14807
  1056
webertj@14807
  1057
	fun size_of_type intr =
webertj@14807
  1058
	let
webertj@14807
  1059
		(* power(a,b) computes a^b, for a>=0, b>=0 *)
webertj@14807
  1060
		(* int * int -> int *)
webertj@14807
  1061
		fun power (a,0) = 1
webertj@14807
  1062
		  | power (a,1) = a
webertj@14807
  1063
		  | power (a,b) = let val ab = power(a,b div 2) in ab * ab * power(a,b mod 2) end
webertj@14807
  1064
	in
webertj@14807
  1065
		case intr of
webertj@14807
  1066
		  Leaf xs => length xs
webertj@14807
  1067
		| Node xs => power (size_of_type (hd xs), length xs)
webertj@14807
  1068
	end;
webertj@14807
  1069
webertj@14807
  1070
(* ------------------------------------------------------------------------- *)
webertj@14807
  1071
(* TT/FF: interpretations that denote "true" or "false", respectively        *)
webertj@14807
  1072
(* ------------------------------------------------------------------------- *)
webertj@14807
  1073
webertj@14807
  1074
	(* interpretation *)
webertj@14807
  1075
webertj@14807
  1076
	val TT = Leaf [True, False];
webertj@14807
  1077
webertj@14807
  1078
	val FF = Leaf [False, True];
webertj@14807
  1079
webertj@14807
  1080
(* ------------------------------------------------------------------------- *)
webertj@14807
  1081
(* make_equality: returns an interpretation that denotes (extensional)       *)
webertj@14807
  1082
(*                equality of two interpretations                            *)
webertj@14807
  1083
(* ------------------------------------------------------------------------- *)
webertj@14807
  1084
webertj@14807
  1085
	(* We could in principle represent '=' on a type T by a particular        *)
webertj@14807
  1086
	(* interpretation.  However, the size of that interpretation is quadratic *)
webertj@14807
  1087
	(* in the size of T.  Therefore comparing the interpretations 'i1' and    *)
webertj@14807
  1088
	(* 'i2' directly is more efficient than constructing the interpretation   *)
webertj@14807
  1089
	(* for equality on T first, and "applying" this interpretation to 'i1'    *)
webertj@14807
  1090
	(* and 'i2' in the usual way (cf. 'interpretation_apply') then.           *)
webertj@14807
  1091
webertj@14807
  1092
	(* interpretation * interpretation -> interpretation *)
webertj@14807
  1093
webertj@14807
  1094
	fun make_equality (i1, i2) =
webertj@14807
  1095
	let
webertj@14807
  1096
		(* interpretation * interpretation -> prop_formula *)
webertj@14807
  1097
		fun equal (i1, i2) =
webertj@14807
  1098
			(case i1 of
webertj@14807
  1099
			  Leaf xs =>
webertj@14807
  1100
				(case i2 of
webertj@14807
  1101
				  Leaf ys => PropLogic.dot_product (xs, ys)
webertj@14807
  1102
				| Node _  => raise REFUTE ("make_equality", "second interpretation is higher"))
webertj@14807
  1103
			| Node xs =>
webertj@14807
  1104
				(case i2 of
webertj@14807
  1105
				  Leaf _  => raise REFUTE ("make_equality", "first interpretation is higher")
webertj@14807
  1106
				| Node ys => PropLogic.all (map equal (xs ~~ ys))))
webertj@14807
  1107
		(* interpretation * interpretation -> prop_formula *)
webertj@14807
  1108
		fun not_equal (i1, i2) =
webertj@14807
  1109
			(case i1 of
webertj@14807
  1110
			  Leaf xs =>
webertj@14807
  1111
				(case i2 of
webertj@14807
  1112
				  Leaf ys => PropLogic.all ((PropLogic.exists xs) :: (PropLogic.exists ys) ::
webertj@14807
  1113
					(map (fn (x,y) => SOr (SNot x, SNot y)) (xs ~~ ys)))  (* defined and not equal *)
webertj@14807
  1114
				| Node _  => raise REFUTE ("make_equality", "second interpretation is higher"))
webertj@14807
  1115
			| Node xs =>
webertj@14807
  1116
				(case i2 of
webertj@14807
  1117
				  Leaf _  => raise REFUTE ("make_equality", "first interpretation is higher")
webertj@14807
  1118
				| Node ys => PropLogic.exists (map not_equal (xs ~~ ys))))
webertj@14350
  1119
	in
webertj@14807
  1120
		(* a value may be undefined; therefore 'not_equal' is not just the     *)
webertj@14807
  1121
		(* negation of 'equal':                                                *)
webertj@14807
  1122
		(* - two interpretations are 'equal' iff they are both defined and     *)
webertj@14807
  1123
		(*   denote the same value                                             *)
webertj@14807
  1124
		(* - two interpretations are 'not_equal' iff they are both defined at  *)
webertj@14807
  1125
		(*   least partially, and a defined part denotes different values      *)
webertj@14807
  1126
		(* - an undefined interpretation is neither 'equal' nor 'not_equal' to *)
webertj@14807
  1127
		(*   another value                                                     *)
webertj@14807
  1128
		Leaf [equal (i1, i2), not_equal (i1, i2)]
webertj@14807
  1129
	end;
webertj@14807
  1130
webertj@14807
  1131
webertj@14807
  1132
	(* theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) option *)
webertj@14807
  1133
webertj@14807
  1134
	(* simply typed lambda calculus: Isabelle's basic term syntax, with type  *)
webertj@14807
  1135
	(* variables, function types, and propT                                   *)
webertj@14807
  1136
webertj@14807
  1137
	fun stlc_interpreter thy model args t =
webertj@14807
  1138
	let
webertj@14807
  1139
		val (typs, terms)                           = model
webertj@14807
  1140
		val {maxvars, next_idx, bounds, wellformed} = args
webertj@14807
  1141
		(* Term.typ -> (interpretation * model * arguments) option *)
webertj@14807
  1142
		fun interpret_groundterm T =
webertj@14807
  1143
		let
webertj@14807
  1144
			(* unit -> (interpretation * model * arguments) option *)
webertj@14807
  1145
			fun interpret_groundtype () =
webertj@14807
  1146
			let
webertj@14807
  1147
				val size = (if T = Term.propT then 2 else (the o assoc) (typs, T))  (* the model MUST specify a size for ground types *)
webertj@14807
  1148
				val next = (if size=2 then next_idx+1 else next_idx+size)  (* optimization for types with size 2 *)
webertj@14807
  1149
				val _    = (if next-1>maxvars andalso maxvars>0 then raise MAXVARS_EXCEEDED else ())  (* check if 'maxvars' is large enough *)
webertj@14807
  1150
				(* prop_formula list *)
webertj@14807
  1151
				val fms  = (if size=2 then [BoolVar next_idx, Not (BoolVar next_idx)]
webertj@14807
  1152
					else (map BoolVar (next_idx upto (next_idx+size-1))))
webertj@14807
  1153
				(* interpretation *)
webertj@14807
  1154
				val intr = Leaf fms
webertj@14807
  1155
				(* prop_formula list -> prop_formula *)
webertj@14807
  1156
				fun one_of_two_false []      = True
webertj@14807
  1157
				  | one_of_two_false (x::xs) = SAnd (PropLogic.all (map (fn x' => SOr (SNot x, SNot x')) xs), one_of_two_false xs)
webertj@14807
  1158
				(* prop_formula list -> prop_formula *)
webertj@14807
  1159
				fun exactly_one_true xs = SAnd (PropLogic.exists xs, one_of_two_false xs)
webertj@14807
  1160
				(* prop_formula *)
webertj@14807
  1161
				val wf   = (if size=2 then True else exactly_one_true fms)
webertj@14807
  1162
			in
webertj@14807
  1163
				(* extend the model, increase 'next_idx', add well-formedness condition *)
webertj@14807
  1164
				Some (intr, (typs, (t, intr)::terms), {maxvars = maxvars, next_idx = next, bounds = bounds, wellformed = SAnd (wellformed, wf)})
webertj@14807
  1165
			end
webertj@14807
  1166
		in
webertj@14807
  1167
			case T of
webertj@14807
  1168
			  Type ("fun", [T1, T2]) =>
webertj@14807
  1169
				let
webertj@14807
  1170
					(* we create 'size_of_type (interpret (... T1))' different copies *)
webertj@14807
  1171
					(* of the interpretation for 'T2', which are then combined into a *)
webertj@14807
  1172
					(* single new interpretation                                      *)
webertj@14807
  1173
					val (i1, _, _) =
webertj@14807
  1174
						(interpret thy model {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", T1))
webertj@14807
  1175
						handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1176
					(* make fresh copies, with different variable indices *)
webertj@14807
  1177
					(* 'idx': next variable index                         *)
webertj@14807
  1178
					(* 'n'  : number of copies                            *)
webertj@14807
  1179
					(* int -> int -> (int * interpretation list * prop_formula *)
webertj@14807
  1180
					fun make_copies idx 0 =
webertj@14807
  1181
						(idx, [], True)
webertj@14807
  1182
					  | make_copies idx n =
webertj@14807
  1183
						let
webertj@14807
  1184
							val (copy, _, new_args) =
webertj@14807
  1185
								(interpret thy (typs, []) {maxvars = maxvars, next_idx = idx, bounds = [], wellformed = True} (Free ("dummy", T2))
webertj@14807
  1186
								handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1187
							val (idx', copies, wf') = make_copies (#next_idx new_args) (n-1)
webertj@14807
  1188
						in
webertj@14807
  1189
							(idx', copy :: copies, SAnd (#wellformed new_args, wf'))
webertj@14807
  1190
						end
webertj@14807
  1191
					val (next, copies, wf) = make_copies next_idx (size_of_type i1)
webertj@14807
  1192
					(* combine copies into a single interpretation *)
webertj@14807
  1193
					val intr = Node copies
webertj@14807
  1194
				in
webertj@14807
  1195
					(* extend the model, increase 'next_idx', add well-formedness condition *)
webertj@14807
  1196
					Some (intr, (typs, (t, intr)::terms), {maxvars = maxvars, next_idx = next, bounds = bounds, wellformed = SAnd (wellformed, wf)})
webertj@14807
  1197
				end
webertj@14807
  1198
			| Type _  => interpret_groundtype ()
webertj@14807
  1199
			| TFree _ => interpret_groundtype ()
webertj@14807
  1200
			| TVar  _ => interpret_groundtype ()
webertj@14807
  1201
		end
webertj@14807
  1202
	in
webertj@14807
  1203
		case assoc (terms, t) of
webertj@14807
  1204
		  Some intr =>
webertj@14807
  1205
			(* return an existing interpretation *)
webertj@14807
  1206
			Some (intr, model, args)
webertj@14807
  1207
		| None =>
webertj@14807
  1208
			(case t of
webertj@14807
  1209
			  Const (_, T)     =>
webertj@14807
  1210
				interpret_groundterm T
webertj@14807
  1211
			| Free (_, T)      =>
webertj@14807
  1212
				interpret_groundterm T
webertj@14807
  1213
			| Var (_, T)       =>
webertj@14807
  1214
				interpret_groundterm T
webertj@14807
  1215
			| Bound i          =>
webertj@14807
  1216
				Some (nth_elem (i, #bounds args), model, args)
webertj@14807
  1217
			| Abs (x, T, body) =>
webertj@14807
  1218
				let
webertj@14807
  1219
					(* create all constants of type 'T' *)
webertj@14807
  1220
					val (i, _, _) =
webertj@14807
  1221
						(interpret thy model {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", T))
webertj@14807
  1222
						handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1223
					val constants = make_constants i
webertj@14807
  1224
					(* interpret the 'body' separately for each constant *)
webertj@14807
  1225
					val ((model', args'), bodies) = foldl_map
webertj@14807
  1226
						(fn ((m,a), c) =>
webertj@14807
  1227
							let
webertj@14807
  1228
								(* add 'c' to 'bounds' *)
webertj@14807
  1229
								val (i', m', a') = interpret thy m {maxvars = #maxvars a, next_idx = #next_idx a, bounds = (c :: #bounds a), wellformed = #wellformed a} body
webertj@14807
  1230
							in
webertj@14807
  1231
								(* keep the new model m' and 'next_idx' and 'wellformed', but use old 'bounds' *)
webertj@14807
  1232
								((m', {maxvars = maxvars, next_idx = #next_idx a', bounds = bounds, wellformed = #wellformed a'}), i')
webertj@14807
  1233
							end)
webertj@14807
  1234
						((model, args), constants)
webertj@14807
  1235
				in
webertj@14807
  1236
					Some (Node bodies, model', args')
webertj@14807
  1237
				end
webertj@14807
  1238
			| t1 $ t2          =>
webertj@14807
  1239
				let
webertj@14807
  1240
					(* auxiliary functions *)
webertj@14807
  1241
					(* interpretation * interpretation -> interpretation *)
webertj@14807
  1242
					fun interpretation_disjunction (tr1,tr2) =
webertj@14807
  1243
						tree_map (fn (xs,ys) => map (fn (x,y) => SOr(x,y)) (xs ~~ ys)) (tree_pair (tr1,tr2))
webertj@14807
  1244
					(* prop_formula * interpretation -> interpretation *)
webertj@14807
  1245
					fun prop_formula_times_interpretation (fm,tr) =
webertj@14807
  1246
						tree_map (map (fn x => SAnd (fm,x))) tr
webertj@14807
  1247
					(* prop_formula list * interpretation list -> interpretation *)
webertj@14807
  1248
					fun prop_formula_list_dot_product_interpretation_list ([fm],[tr]) =
webertj@14807
  1249
						prop_formula_times_interpretation (fm,tr)
webertj@14807
  1250
					  | prop_formula_list_dot_product_interpretation_list (fm::fms,tr::trees) =
webertj@14807
  1251
						interpretation_disjunction (prop_formula_times_interpretation (fm,tr), prop_formula_list_dot_product_interpretation_list (fms,trees))
webertj@14807
  1252
					  | prop_formula_list_dot_product_interpretation_list (_,_) =
webertj@14807
  1253
						raise REFUTE ("stlc_interpreter", "empty list (in dot product)")
webertj@14807
  1254
					(* concatenates 'x' with every list in 'xss', returning a new list of lists *)
webertj@14807
  1255
					(* 'a -> 'a list list -> 'a list list *)
webertj@14807
  1256
					fun cons_list x xss =
webertj@14807
  1257
						map (fn xs => x::xs) xss
webertj@14807
  1258
					(* returns a list of lists, each one consisting of one element from each element of 'xss' *)
webertj@14807
  1259
					(* 'a list list -> 'a list list *)
webertj@14807
  1260
					fun pick_all [xs] =
webertj@14807
  1261
						map (fn x => [x]) xs
webertj@14807
  1262
					  | pick_all (xs::xss) =
webertj@14807
  1263
						let val rec_pick = pick_all xss in
webertj@14807
  1264
							foldl (fn (acc,x) => (cons_list x rec_pick) @ acc) ([],xs)
webertj@14807
  1265
						end
webertj@14807
  1266
					  | pick_all _ =
webertj@14807
  1267
						raise REFUTE ("stlc_interpreter", "empty list (in pick_all)")
webertj@14807
  1268
					(* interpretation -> prop_formula list *)
webertj@14807
  1269
					fun interpretation_to_prop_formula_list (Leaf xs) =
webertj@14807
  1270
						xs
webertj@14807
  1271
					  | interpretation_to_prop_formula_list (Node trees) =
webertj@14807
  1272
						map PropLogic.all (pick_all (map interpretation_to_prop_formula_list trees))
webertj@14807
  1273
					(* interpretation * interpretation -> interpretation *)
webertj@14807
  1274
					fun interpretation_apply (tr1,tr2) =
webertj@14807
  1275
						(case tr1 of
webertj@14807
  1276
						  Leaf _ =>
webertj@14807
  1277
							raise REFUTE ("stlc_interpreter", "first interpretation is a leaf")
webertj@14807
  1278
						| Node xs =>
webertj@14807
  1279
							prop_formula_list_dot_product_interpretation_list (interpretation_to_prop_formula_list tr2, xs))
webertj@14807
  1280
					(* interpret 't1' and 't2' separately *)
webertj@14807
  1281
					val (intr1, model1, args1) = interpret thy model args t1
webertj@14807
  1282
					val (intr2, model2, args2) = interpret thy model1 args1 t2
webertj@14807
  1283
				in
webertj@14807
  1284
					Some (interpretation_apply (intr1,intr2), model2, args2)
webertj@14807
  1285
				end)
webertj@14807
  1286
	end;
webertj@14807
  1287
webertj@14807
  1288
	(* theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) option *)
webertj@14807
  1289
webertj@14807
  1290
	fun Pure_interpreter thy model args t =
webertj@14456
  1291
		case t of
webertj@14807
  1292
		  Const ("all", _) $ t1 =>  (* in the meta-logic, 'all' MUST be followed by an argument term *)
webertj@14807
  1293
			let
webertj@14807
  1294
				val (i, m, a) = interpret thy model args t1
webertj@14807
  1295
			in
webertj@14807
  1296
				case i of
webertj@14807
  1297
				  Node xs =>
webertj@14807
  1298
					let
webertj@14807
  1299
						val fmTrue  = PropLogic.all (map toTrue xs)
webertj@14807
  1300
						val fmFalse = PropLogic.exists (map toFalse xs)
webertj@14807
  1301
					in
webertj@14807
  1302
						Some (Leaf [fmTrue, fmFalse], m, a)
webertj@14807
  1303
					end
webertj@14807
  1304
				| _ =>
webertj@14807
  1305
					raise REFUTE ("Pure_interpreter", "\"all\" is not followed by a function")
webertj@14807
  1306
			end
webertj@14807
  1307
		| Const ("==", _) $ t1 $ t2 =>
webertj@14807
  1308
			let
webertj@14807
  1309
				val (i1, m1, a1) = interpret thy model args t1
webertj@14807
  1310
				val (i2, m2, a2) = interpret thy m1 a1 t2
webertj@14807
  1311
			in
webertj@14807
  1312
				Some (make_equality (i1, i2), m2, a2)
webertj@14807
  1313
			end
webertj@14807
  1314
		| Const ("==>", _) =>  (* simpler than translating 'Const ("==>", _) $ t1 $ t2' *)
webertj@14807
  1315
			Some (Node [Node [TT, FF], Node [TT, TT]], model, args)
webertj@14807
  1316
		| _ => None;
webertj@14807
  1317
webertj@14807
  1318
	(* theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) option *)
webertj@14807
  1319
webertj@14807
  1320
	fun HOLogic_interpreter thy model args t =
webertj@14807
  1321
	let
webertj@14807
  1322
		(* Term.term -> int -> Term.term *)
webertj@14807
  1323
		fun eta_expand t i =
webertj@14807
  1324
		let
webertj@14807
  1325
			val Ts = binder_types (fastype_of t)
webertj@14807
  1326
		in
webertj@14807
  1327
			foldr (fn (T, t) => Abs ("<eta_expand>", T, t))
webertj@14807
  1328
				(take (i, Ts), list_comb (t, map Bound (i-1 downto 0)))
webertj@14807
  1329
		end
webertj@14807
  1330
	in
webertj@14807
  1331
	(* ------------------------------------------------------------------------- *)
webertj@14807
  1332
	(* Providing interpretations directly is more efficient than unfolding the   *)
webertj@14807
  1333
	(* logical constants.  IN HOL however, logical constants can themselves be   *)
webertj@14807
  1334
	(* arguments.  "All" and "Ex" are then translated just like any other        *)
webertj@14807
  1335
	(* constant, with the relevant axiom being added by 'collect_axioms'.        *)
webertj@14807
  1336
	(* ------------------------------------------------------------------------- *)
webertj@14807
  1337
		case t of
webertj@14807
  1338
		  Const ("Trueprop", _) =>
webertj@14807
  1339
			Some (Node [TT, FF], model, args)
webertj@14807
  1340
		| Const ("Not", _) =>
webertj@14807
  1341
			Some (Node [FF, TT], model, args)
webertj@14807
  1342
		| Const ("True", _) =>  (* redundant, since 'True' is also an IDT constructor *)
webertj@14807
  1343
			Some (TT, model, args)
webertj@14807
  1344
		| Const ("False", _) =>  (* redundant, since 'False' is also an IDT constructor *)
webertj@14807
  1345
			Some (FF, model, args)
webertj@14807
  1346
		| Const ("All", _) $ t1 =>
webertj@14350
  1347
			let
webertj@14807
  1348
				val (i, m, a) = interpret thy model args t1
webertj@14807
  1349
			in
webertj@14807
  1350
				case i of
webertj@14807
  1351
				  Node xs =>
webertj@14807
  1352
					let
webertj@14807
  1353
						val fmTrue  = PropLogic.all (map toTrue xs)
webertj@14807
  1354
						val fmFalse = PropLogic.exists (map toFalse xs)
webertj@14807
  1355
					in
webertj@14807
  1356
						Some (Leaf [fmTrue, fmFalse], m, a)
webertj@14807
  1357
					end
webertj@14807
  1358
				| _ =>
webertj@14807
  1359
					raise REFUTE ("HOLogic_interpreter", "\"All\" is not followed by a function")
webertj@14807
  1360
			end
webertj@14807
  1361
		| Const ("Ex", _) $ t1 =>
webertj@14807
  1362
			let
webertj@14807
  1363
				val (i, m, a) = interpret thy model args t1
webertj@14807
  1364
			in
webertj@14807
  1365
				case i of
webertj@14807
  1366
				  Node xs =>
webertj@14807
  1367
					let
webertj@14807
  1368
						val fmTrue  = PropLogic.exists (map toTrue xs)
webertj@14807
  1369
						val fmFalse = PropLogic.all (map toFalse xs)
webertj@14807
  1370
					in
webertj@14807
  1371
						Some (Leaf [fmTrue, fmFalse], m, a)
webertj@14807
  1372
					end
webertj@14807
  1373
				| _ =>
webertj@14807
  1374
					raise REFUTE ("HOLogic_interpreter", "\"Ex\" is not followed by a function")
webertj@14807
  1375
			end
webertj@14807
  1376
		| Const ("op =", _) $ t1 $ t2 =>
webertj@14807
  1377
			let
webertj@14807
  1378
				val (i1, m1, a1) = interpret thy model args t1
webertj@14807
  1379
				val (i2, m2, a2) = interpret thy m1 a1 t2
webertj@14807
  1380
			in
webertj@14807
  1381
				Some (make_equality (i1, i2), m2, a2)
webertj@14807
  1382
			end
webertj@14807
  1383
		| Const ("op =", _) $ t1 =>
webertj@14807
  1384
			(Some (interpret thy model args (eta_expand t 1)) handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1385
		| Const ("op =", _) =>
webertj@14807
  1386
			(Some (interpret thy model args (eta_expand t 2)) handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1387
		| Const ("op &", _) =>
webertj@14807
  1388
			Some (Node [Node [TT, FF], Node [FF, FF]], model, args)
webertj@14807
  1389
		| Const ("op |", _) =>
webertj@14807
  1390
			Some (Node [Node [TT, TT], Node [TT, FF]], model, args)
webertj@14807
  1391
		| Const ("op -->", _) =>
webertj@14807
  1392
			Some (Node [Node [TT, FF], Node [TT, TT]], model, args)
webertj@14807
  1393
		| _ => None
webertj@14807
  1394
	end;
webertj@14807
  1395
webertj@14807
  1396
	(* theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) option *)
webertj@14807
  1397
webertj@14807
  1398
	fun set_interpreter thy model args t =
webertj@14807
  1399
	(* "T set" is isomorphic to "T --> bool" *)
webertj@14807
  1400
	let
webertj@14807
  1401
		val (typs, terms) = model
webertj@14807
  1402
		(* Term.term -> int -> Term.term *)
webertj@14807
  1403
		fun eta_expand t i =
webertj@14807
  1404
		let
webertj@14807
  1405
			val Ts = binder_types (fastype_of t)
webertj@14807
  1406
		in
webertj@14807
  1407
			foldr (fn (T, t) => Abs ("<eta_expand>", T, t))
webertj@14807
  1408
				(take (i, Ts), list_comb (t, map Bound (i-1 downto 0)))
webertj@14807
  1409
		end
webertj@14807
  1410
	in
webertj@14807
  1411
		case assoc (terms, t) of
webertj@14807
  1412
		  Some intr =>
webertj@14807
  1413
			(* return an existing interpretation *)
webertj@14807
  1414
			Some (intr, model, args)
webertj@14807
  1415
		| None =>
webertj@14807
  1416
			(case t of
webertj@14807
  1417
			  Free (x, Type ("set", [T])) =>
webertj@14807
  1418
				(let
webertj@14807
  1419
					val (intr, _, args') = interpret thy (typs, []) args (Free (x, T --> HOLogic.boolT))
webertj@14807
  1420
				in
webertj@14807
  1421
					Some (intr, (typs, (t, intr)::terms), args')
webertj@14807
  1422
				end handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1423
			| Var ((x,i), Type ("set", [T])) =>
webertj@14807
  1424
				(let
webertj@14807
  1425
					val (intr, _, args') = interpret thy (typs, []) args (Var ((x,i), T --> HOLogic.boolT))
webertj@14807
  1426
				in
webertj@14807
  1427
					Some (intr, (typs, (t, intr)::terms), args')
webertj@14807
  1428
				end handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1429
			| Const (s, Type ("set", [T])) =>
webertj@14807
  1430
				(let
webertj@14807
  1431
					val (intr, _, args') = interpret thy (typs, []) args (Const (s, T --> HOLogic.boolT))
webertj@14807
  1432
				in
webertj@14807
  1433
					Some (intr, (typs, (t, intr)::terms), args')
webertj@14807
  1434
				end handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1435
			(* 'Collect' == identity *)
webertj@14807
  1436
			| Const ("Collect", _) $ t1 =>
webertj@14807
  1437
				Some (interpret thy model args t1)
webertj@14807
  1438
			| Const ("Collect", _) =>
webertj@14807
  1439
				(Some (interpret thy model args (eta_expand t 1)) handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1440
			(* 'op :' == application *)
webertj@14807
  1441
			| Const ("op :", _) $ t1 $ t2 =>
webertj@14807
  1442
				Some (interpret thy model args (t2 $ t1))
webertj@14807
  1443
			| Const ("op :", _) $ t1 =>
webertj@14807
  1444
				(Some (interpret thy model args (eta_expand t 1)) handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1445
			| Const ("op :", _) =>
webertj@14807
  1446
				(Some (interpret thy model args (eta_expand t 2)) handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1447
			| _ => None)
webertj@14807
  1448
	end;
webertj@14807
  1449
webertj@14807
  1450
	(* theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) option *)
webertj@14807
  1451
webertj@14807
  1452
	fun IDT_interpreter thy model args t =
webertj@14807
  1453
	let
webertj@14807
  1454
		val (typs, terms) = model
webertj@14807
  1455
		(* DatatypeAux.descr -> (DatatypeAux.dtyp * Term.typ) list -> DatatypeAux.dtyp -> Term.typ *)
webertj@14807
  1456
		fun typ_of_dtyp descr typ_assoc (DatatypeAux.DtTFree a) =
webertj@14807
  1457
			(* replace a 'DtTFree' variable by the associated type *)
webertj@14807
  1458
			(the o assoc) (typ_assoc, DatatypeAux.DtTFree a)
webertj@14807
  1459
		  | typ_of_dtyp descr typ_assoc (DatatypeAux.DtRec i) =
webertj@14807
  1460
			let
webertj@14807
  1461
				val (s, ds, _) = (the o assoc) (descr, i)
webertj@14807
  1462
			in
webertj@14807
  1463
				Type (s, map (typ_of_dtyp descr typ_assoc) ds)
webertj@14807
  1464
			end
webertj@14807
  1465
		  | typ_of_dtyp descr typ_assoc (DatatypeAux.DtType (s, ds)) =
webertj@14807
  1466
			Type (s, map (typ_of_dtyp descr typ_assoc) ds)
webertj@14807
  1467
		(* int list -> int *)
webertj@14807
  1468
		fun sum xs = foldl op+ (0, xs)
webertj@14807
  1469
		(* int list -> int *)
webertj@14807
  1470
		fun product xs = foldl op* (1, xs)
webertj@14807
  1471
		(* the size of an IDT is the sum (over its constructors) of the        *)
webertj@14807
  1472
		(* product (over their arguments) of the size of the argument type     *)
webertj@14807
  1473
		(* (Term.typ * int) list -> DatatypeAux.descr -> (DatatypeAux.dtyp * Term.typ) list -> (string * DatatypeAux.dtyp list) list -> int *)
webertj@14807
  1474
		fun size_of_dtyp typs descr typ_assoc constrs =
webertj@14807
  1475
			sum (map (fn (_, ds) =>
webertj@14807
  1476
				product (map (fn d =>
webertj@14807
  1477
					let
webertj@14807
  1478
						val T         = typ_of_dtyp descr typ_assoc d
webertj@14807
  1479
						val (i, _, _) =
webertj@14807
  1480
							(interpret thy (typs, []) {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", T))
webertj@14807
  1481
							handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1482
					in
webertj@14807
  1483
						size_of_type i
webertj@14807
  1484
					end) ds)) constrs)
webertj@14807
  1485
		(* Term.typ -> (interpretation * model * arguments) option *)
webertj@14807
  1486
		fun interpret_variable (Type (s, Ts)) =
webertj@14807
  1487
			(case DatatypePackage.datatype_info thy s of
webertj@14807
  1488
			  Some info =>  (* inductive datatype *)
webertj@14807
  1489
				let
webertj@14807
  1490
					val (typs, terms) = model
webertj@14807
  1491
					(* int option -- only recursive IDTs have an associated depth *)
webertj@14807
  1492
					val depth         = assoc (typs, Type (s, Ts))
webertj@14807
  1493
				in
webertj@14807
  1494
					if depth = (Some 0) then  (* termination condition to avoid infinite recursion *)
webertj@14807
  1495
						(* return a leaf of size 0 *)
webertj@14807
  1496
						Some (Leaf [], model, args)
webertj@14807
  1497
					else
webertj@14807
  1498
						let
webertj@14807
  1499
							val index               = #index info
webertj@14807
  1500
							val descr               = #descr info
webertj@14807
  1501
							val (_, dtyps, constrs) = (the o assoc) (descr, index)
webertj@14807
  1502
							val typ_assoc           = dtyps ~~ Ts
webertj@14807
  1503
							(* sanity check: every element in 'dtyps' must be a 'DtTFree' *)
webertj@14807
  1504
							val _ = (if Library.exists (fn d =>
webertj@14807
  1505
									case d of DatatypeAux.DtTFree _ => false | _ => true) dtyps
webertj@14807
  1506
								then
webertj@14807
  1507
									raise REFUTE ("IDT_interpreter", "datatype argument (for type " ^ Sign.string_of_typ (sign_of thy) (Type (s, Ts)) ^ ") is not a variable")
webertj@14807
  1508
								else
webertj@14807
  1509
									())
webertj@14807
  1510
							(* if the model specifies a depth for the current type, decrement it to avoid infinite recursion *)
webertj@14807
  1511
							val typs'    = (case depth of None => typs | Some n => overwrite (typs, (Type (s, Ts), n-1)))
webertj@14807
  1512
							(* recursively compute the size of the datatype *)
webertj@14807
  1513
							val size     = size_of_dtyp typs' descr typ_assoc constrs
webertj@14807
  1514
							val next_idx = #next_idx args
webertj@14807
  1515
							val next     = (if size=2 then next_idx+1 else next_idx+size)  (* optimization for types with size 2 *)
webertj@14807
  1516
							val _        = (if next-1>(#maxvars args) andalso (#maxvars args)>0 then raise MAXVARS_EXCEEDED else ())  (* check if 'maxvars' is large enough *)
webertj@14807
  1517
							(* prop_formula list *)
webertj@14807
  1518
							val fms      = (if size=2 then [BoolVar next_idx, Not (BoolVar next_idx)]
webertj@14807
  1519
								else (map BoolVar (next_idx upto (next_idx+size-1))))
webertj@14807
  1520
							(* interpretation *)
webertj@14807
  1521
							val intr     = Leaf fms
webertj@14807
  1522
							(* prop_formula list -> prop_formula *)
webertj@14807
  1523
							fun one_of_two_false []      = True
webertj@14807
  1524
							  | one_of_two_false (x::xs) = SAnd (PropLogic.all (map (fn x' => SOr (SNot x, SNot x')) xs), one_of_two_false xs)
webertj@14807
  1525
							(* prop_formula list -> prop_formula *)
webertj@14807
  1526
							fun exactly_one_true xs = SAnd (PropLogic.exists xs, one_of_two_false xs)
webertj@14807
  1527
							(* prop_formula *)
webertj@14807
  1528
							val wf       = (if size=2 then True else exactly_one_true fms)
webertj@14807
  1529
						in
webertj@14807
  1530
							(* extend the model, increase 'next_idx', add well-formedness condition *)
webertj@14807
  1531
							Some (intr, (typs, (t, intr)::terms), {maxvars = #maxvars args, next_idx = next, bounds = #bounds args, wellformed = SAnd (#wellformed args, wf)})
webertj@14807
  1532
						end
webertj@14807
  1533
				end
webertj@14807
  1534
			| None =>  (* not an inductive datatype *)
webertj@14807
  1535
				None)
webertj@14807
  1536
		  | interpret_variable _ =  (* a (free or schematic) type variable *)
webertj@14807
  1537
			None
webertj@14807
  1538
	in
webertj@14807
  1539
		case assoc (terms, t) of
webertj@14807
  1540
		  Some intr =>
webertj@14807
  1541
			(* return an existing interpretation *)
webertj@14807
  1542
			Some (intr, model, args)
webertj@14807
  1543
		| None =>
webertj@14807
  1544
			(case t of
webertj@14807
  1545
			  Free (_, T)  => interpret_variable T
webertj@14807
  1546
			| Var (_, T)   => interpret_variable T
webertj@14807
  1547
			| Const (s, T) =>
webertj@14807
  1548
				(* TODO: case, recursion, size *)
webertj@14807
  1549
				let
webertj@14807
  1550
					(* unit -> (interpretation * model * arguments) option *)
webertj@14807
  1551
					fun interpret_constructor () =
webertj@14807
  1552
						(case body_type T of
webertj@14807
  1553
						  Type (s', Ts') =>
webertj@14807
  1554
							(case DatatypePackage.datatype_info thy s' of
webertj@14807
  1555
							  Some info =>  (* body type is an inductive datatype *)
webertj@14807
  1556
								let
webertj@14807
  1557
									val index               = #index info
webertj@14807
  1558
									val descr               = #descr info
webertj@14807
  1559
									val (_, dtyps, constrs) = (the o assoc) (descr, index)
webertj@14807
  1560
									val typ_assoc           = dtyps ~~ Ts'
webertj@14807
  1561
									(* sanity check: every element in 'dtyps' must be a 'DtTFree' *)
webertj@14807
  1562
									val _ = (if Library.exists (fn d =>
webertj@14807
  1563
											case d of DatatypeAux.DtTFree _ => false | _ => true) dtyps
webertj@14807
  1564
										then
webertj@14807
  1565
											raise REFUTE ("IDT_interpreter", "datatype argument (for type " ^ Sign.string_of_typ (sign_of thy) (Type (s', Ts')) ^ ") is not a variable")
webertj@14807
  1566
										else
webertj@14807
  1567
											())
webertj@14807
  1568
									(* split the constructors into those occuring before/after 'Const (s, T)' *)
webertj@14807
  1569
									val (constrs1, constrs2) = take_prefix (fn (cname, ctypes) =>
webertj@14810
  1570
										not (cname = s andalso Type.typ_instance (Sign.tsig_of (sign_of thy)) (T,
webertj@14807
  1571
											map (typ_of_dtyp descr typ_assoc) ctypes ---> Type (s', Ts')))) constrs
webertj@14807
  1572
								in
webertj@14807
  1573
									case constrs2 of
webertj@14807
  1574
									  [] =>
webertj@14807
  1575
										(* 'Const (s, T)' is not a constructor of this datatype *)
webertj@14807
  1576
										None
webertj@14807
  1577
									| c::cs =>
webertj@14807
  1578
										let
webertj@14807
  1579
											(* int option -- only recursive IDTs have an associated depth *)
webertj@14807
  1580
											val depth = assoc (typs, Type (s', Ts'))
webertj@14807
  1581
											val typs' = (case depth of None => typs | Some n => overwrite (typs, (Type (s', Ts'), n-1)))
webertj@14807
  1582
											(* constructors before 'Const (s, T)' generate elements of the datatype *)
webertj@14807
  1583
											val offset  = size_of_dtyp typs' descr typ_assoc constrs1
webertj@14807
  1584
											(* 'Const (s, T)' and constructors after it generate elements of the datatype *)
webertj@14807
  1585
											val total   = offset + (size_of_dtyp typs' descr typ_assoc constrs2)
webertj@14807
  1586
											(* create an interpretation that corresponds to the constructor 'Const (s, T)' *)
webertj@14807
  1587
											(* by recursion over its argument types                                        *)
webertj@14807
  1588
											(* DatatypeAux.dtyp list -> interpretation *)
webertj@14807
  1589
											fun make_partial [] =
webertj@14807
  1590
												(* all entries of the leaf are 'False' *)
webertj@14807
  1591
												Leaf (replicate total False)
webertj@14807
  1592
											  | make_partial (d::ds) =
webertj@14807
  1593
												let
webertj@14807
  1594
													(* compute the "new" size of the type 'd' *)
webertj@14807
  1595
													val T         = typ_of_dtyp descr typ_assoc d
webertj@14807
  1596
													val (i, _, _) =
webertj@14807
  1597
														(interpret thy (typs, []) {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", T))
webertj@14807
  1598
														handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1599
												in
webertj@14807
  1600
													(* all entries of the whole subtree are 'False' *)
webertj@14807
  1601
													Node (replicate (size_of_type i) (make_partial ds))
webertj@14807
  1602
												end
webertj@14807
  1603
											(* int * DatatypeAux.dtyp list -> int * interpretation *)
webertj@14807
  1604
											fun make_constr (offset, []) =
webertj@14807
  1605
												if offset<total then
webertj@14807
  1606
													(offset+1, Leaf ((replicate offset False) @ True :: (replicate (total-offset-1) False)))
webertj@14807
  1607
												else
webertj@14807
  1608
													raise REFUTE ("IDT_interpreter", "internal error: offset >= total")
webertj@14807
  1609
											  | make_constr (offset, d::ds) =
webertj@14807
  1610
												let
webertj@14807
  1611
													(* compute the "new" and "old" size of the type 'd' *)
webertj@14807
  1612
													val T         = typ_of_dtyp descr typ_assoc d
webertj@14807
  1613
													val (i, _, _) =
webertj@14807
  1614
														(interpret thy (typs, []) {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", T))
webertj@14807
  1615
														handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1616
													val (i', _, _) =
webertj@14807
  1617
														(interpret thy (typs', []) {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", T))
webertj@14807
  1618
														handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1619
													val size  = size_of_type i
webertj@14807
  1620
													val size' = size_of_type i'
webertj@14807
  1621
													val _ = if size<size' then
webertj@14807
  1622
															raise REFUTE ("IDT_interpreter", "internal error: new size < old size")
webertj@14807
  1623
														else
webertj@14807
  1624
															()
webertj@14807
  1625
													val (new_offset, intrs) = foldl_map make_constr (offset, replicate size' ds)
webertj@14807
  1626
												in
webertj@14807
  1627
													(* the first size' elements of the type actually yield a result *)
webertj@14807
  1628
													(* element, while the remaining size-size' elements don't       *)
webertj@14807
  1629
													(new_offset, Node (intrs @ (replicate (size-size') (make_partial ds))))
webertj@14807
  1630
												end
webertj@14807
  1631
										in
webertj@14807
  1632
											Some ((snd o make_constr) (offset, snd c), model, args)
webertj@14807
  1633
										end
webertj@14807
  1634
								end
webertj@14807
  1635
							| None =>  (* body type is not an inductive datatype *)
webertj@14807
  1636
								None)
webertj@14807
  1637
						| _ =>  (* body type is a (free or schematic) type variable *)
webertj@14807
  1638
							None)
webertj@14807
  1639
				in
webertj@14807
  1640
					case interpret_constructor () of
webertj@14807
  1641
					  Some x => Some x
webertj@14807
  1642
					| None   => interpret_variable T
webertj@14807
  1643
				end
webertj@14807
  1644
			| _ => None)
webertj@14807
  1645
	end;
webertj@14807
  1646
webertj@14807
  1647
	(* theory -> model -> arguments -> Term.term -> (interpretation * model * arguments) option *)
webertj@14807
  1648
webertj@14807
  1649
	(* only an optimization: 'card' could in principle be interpreted with    *)
webertj@14807
  1650
	(* interpreters available already (using its definition), but the code    *)
webertj@14807
  1651
	(* below is much more efficient                                           *)
webertj@14807
  1652
webertj@14807
  1653
	fun Finite_Set_card_interpreter thy model args t =
webertj@14807
  1654
		case t of
webertj@14807
  1655
		  Const ("Finite_Set.card", Type ("fun", [Type ("set", [T]), Type ("nat", [])])) =>
webertj@14807
  1656
			let
webertj@14807
  1657
				val (i_nat, _, _) =
webertj@14807
  1658
					(interpret thy model {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", Type ("nat", [])))
webertj@14807
  1659
						handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1660
				val size_nat      = size_of_type i_nat
webertj@14807
  1661
				val (i_set, _, _) =
webertj@14807
  1662
					(interpret thy model {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", Type ("set", [T])))
webertj@14807
  1663
						handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1664
				val constants     = make_constants i_set
webertj@14807
  1665
				(* interpretation -> int *)
webertj@14807
  1666
				fun number_of_elements (Node xs) =
webertj@14807
  1667
					foldl (fn (n, x) =>
webertj@14807
  1668
						if x=TT then n+1 else if x=FF then n else raise REFUTE ("Finite_Set_card_interpreter", "interpretation for set type does not yield a Boolean")) (0, xs)
webertj@14807
  1669
				  | number_of_elements (Leaf _) =
webertj@14807
  1670
					raise REFUTE ("Finite_Set_card_interpreter", "interpretation for set type is a leaf")
webertj@14807
  1671
				(* takes an interpretation for a set and returns an interpretation for a 'nat' *)
webertj@14807
  1672
				(* interpretation -> interpretation *)
webertj@14807
  1673
				fun card i =
webertj@14807
  1674
					let
webertj@14807
  1675
						val n = number_of_elements i
webertj@14807
  1676
					in
webertj@14807
  1677
						if n<size_nat then
webertj@14807
  1678
							Leaf ((replicate n False) @ True :: (replicate (size_nat-n-1) False))
webertj@14456
  1679
						else
webertj@14807
  1680
							Leaf (replicate size_nat False)
webertj@14807
  1681
					end
webertj@14350
  1682
			in
webertj@14807
  1683
				Some (Node (map card constants), model, args)
webertj@14350
  1684
			end
webertj@14807
  1685
		| _ =>
webertj@14807
  1686
			None;
webertj@14807
  1687
webertj@14807
  1688
webertj@14807
  1689
(* ------------------------------------------------------------------------- *)
webertj@14807
  1690
(* PRINTERS                                                                  *)
webertj@14807
  1691
(* ------------------------------------------------------------------------- *)
webertj@14807
  1692
webertj@14807
  1693
	(* theory -> model -> Term.term -> interpretation -> (int -> bool) -> Term.term option *)
webertj@14807
  1694
webertj@14807
  1695
	fun stlc_printer thy model t intr assignment =
webertj@14807
  1696
	let
webertj@14807
  1697
		(* Term.term -> Term.typ option *)
webertj@14807
  1698
		fun typeof (Free (_, T))  = Some T
webertj@14807
  1699
		  | typeof (Var (_, T))   = Some T
webertj@14807
  1700
		  | typeof (Const (_, T)) = Some T
webertj@14807
  1701
		  | typeof _              = None
webertj@14807
  1702
		(* string -> string *)
webertj@14807
  1703
		fun strip_leading_quote s =
webertj@14807
  1704
			(implode o (fn ss => case ss of [] => [] | x::xs => if x="'" then xs else ss) o explode) s
webertj@14807
  1705
		(* Term.typ -> string *)
webertj@14807
  1706
		fun string_of_typ (Type (s, _))     = s
webertj@14807
  1707
		  | string_of_typ (TFree (x, _))    = strip_leading_quote x
webertj@14807
  1708
		  | string_of_typ (TVar ((x,i), _)) = strip_leading_quote x ^ string_of_int i
webertj@14807
  1709
		(* interpretation -> int *)
webertj@14807
  1710
		fun index_from_interpretation (Leaf xs) =
webertj@14350
  1711
			let
webertj@14807
  1712
				val idx = find_index (PropLogic.eval assignment) xs
webertj@14350
  1713
			in
webertj@14807
  1714
				if idx<0 then
webertj@14807
  1715
					raise REFUTE ("stlc_printer", "illegal interpretation: no value assigned (SAT solver unsound?)")
webertj@14807
  1716
				else
webertj@14807
  1717
					idx
webertj@14350
  1718
			end
webertj@14807
  1719
		  | index_from_interpretation _ =
webertj@14807
  1720
			raise REFUTE ("stlc_printer", "interpretation for ground type is not a leaf")
webertj@14807
  1721
	in
webertj@14807
  1722
		case typeof t of
webertj@14807
  1723
		  Some T =>
webertj@14807
  1724
			(case T of
webertj@14807
  1725
			  Type ("fun", [T1, T2]) =>
webertj@14807
  1726
				(let
webertj@14807
  1727
					(* create all constants of type 'T1' *)
webertj@14807
  1728
					val (i, _, _) = interpret thy model {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", T1))
webertj@14807
  1729
					val constants = make_constants i
webertj@14807
  1730
					(* interpretation list *)
webertj@14807
  1731
					val results = (case intr of
webertj@14807
  1732
						  Node xs => xs
webertj@14807
  1733
						| _       => raise REFUTE ("stlc_printer", "interpretation for function type is a leaf"))
webertj@14807
  1734
					(* Term.term list *)
webertj@14807
  1735
					val pairs = map (fn (arg, result) =>
webertj@14807
  1736
						HOLogic.mk_prod
webertj@14807
  1737
							(print thy model (Free ("dummy", T1)) arg assignment,
webertj@14807
  1738
							 print thy model (Free ("dummy", T2)) result assignment))
webertj@14807
  1739
						(constants ~~ results)
webertj@14807
  1740
					(* Term.typ *)
webertj@14807
  1741
					val HOLogic_prodT = HOLogic.mk_prodT (T1, T2)
webertj@14807
  1742
					val HOLogic_setT  = HOLogic.mk_setT HOLogic_prodT
webertj@14807
  1743
					(* Term.term *)
webertj@14807
  1744
					val HOLogic_empty_set = Const ("{}", HOLogic_setT)
webertj@14807
  1745
					val HOLogic_insert    = Const ("insert", HOLogic_prodT --> HOLogic_setT --> HOLogic_setT)
webertj@14807
  1746
				in
webertj@14807
  1747
					Some (foldr (fn (pair, acc) => HOLogic_insert $ pair $ acc) (pairs, HOLogic_empty_set))
webertj@14807
  1748
				end handle CANNOT_INTERPRET _ => None)
webertj@14807
  1749
			| Type ("prop", [])      =>
webertj@14807
  1750
				(case index_from_interpretation intr of
webertj@14807
  1751
				  0 => Some (HOLogic.mk_Trueprop HOLogic.true_const)
webertj@14807
  1752
				| 1 => Some (HOLogic.mk_Trueprop HOLogic.false_const)
webertj@14807
  1753
				| _ => raise REFUTE ("stlc_interpreter", "illegal interpretation for a propositional value"))
webertj@14807
  1754
			| Type _  => Some (Const (string_of_typ T ^ string_of_int (index_from_interpretation intr), T))
webertj@14807
  1755
			| TFree _ => Some (Const (string_of_typ T ^ string_of_int (index_from_interpretation intr), T))
webertj@14807
  1756
			| TVar _  => Some (Const (string_of_typ T ^ string_of_int (index_from_interpretation intr), T)))
webertj@14807
  1757
		| None =>
webertj@14807
  1758
			None
webertj@14350
  1759
	end;
webertj@14350
  1760
webertj@14456
  1761
	(* theory -> model -> Term.term -> interpretation -> (int -> bool) -> string option *)
webertj@14350
  1762
webertj@14807
  1763
	fun set_printer thy model t intr assignment =
webertj@14807
  1764
	let
webertj@14807
  1765
		(* Term.term -> Term.typ option *)
webertj@14807
  1766
		fun typeof (Free (_, T))  = Some T
webertj@14807
  1767
		  | typeof (Var (_, T))   = Some T
webertj@14807
  1768
		  | typeof (Const (_, T)) = Some T
webertj@14807
  1769
		  | typeof _              = None
webertj@14807
  1770
	in
webertj@14807
  1771
		case typeof t of
webertj@14807
  1772
		  Some (Type ("set", [T])) =>
webertj@14807
  1773
			(let
webertj@14807
  1774
				(* create all constants of type 'T' *)
webertj@14807
  1775
				val (i, _, _) = interpret thy model {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", T))
webertj@14807
  1776
				val constants = make_constants i
webertj@14807
  1777
				(* interpretation list *)
webertj@14807
  1778
				val results = (case intr of
webertj@14807
  1779
					  Node xs => xs
webertj@14807
  1780
					| _       => raise REFUTE ("set_printer", "interpretation for set type is a leaf"))
webertj@14807
  1781
				(* Term.term list *)
webertj@14807
  1782
				val elements = mapfilter (fn (arg, result) =>
webertj@14807
  1783
					case result of
webertj@14807
  1784
					  Leaf [fmTrue, fmFalse] =>
webertj@14807
  1785
						if PropLogic.eval assignment fmTrue then
webertj@14807
  1786
							Some (print thy model (Free ("dummy", T)) arg assignment)
webertj@14807
  1787
						else if PropLogic.eval assignment fmFalse then
webertj@14807
  1788
							None
webertj@14807
  1789
						else
webertj@14807
  1790
							raise REFUTE ("set_printer", "illegal interpretation: no value assigned (SAT solver unsound?)")
webertj@14807
  1791
					| _ =>
webertj@14807
  1792
						raise REFUTE ("set_printer", "illegal interpretation for a Boolean value"))
webertj@14807
  1793
					(constants ~~ results)
webertj@14807
  1794
				(* Term.typ *)
webertj@14807
  1795
				val HOLogic_setT  = HOLogic.mk_setT T
webertj@14807
  1796
				(* Term.term *)
webertj@14807
  1797
				val HOLogic_empty_set = Const ("{}", HOLogic_setT)
webertj@14807
  1798
				val HOLogic_insert    = Const ("insert", T --> HOLogic_setT --> HOLogic_setT)
webertj@14807
  1799
			in
webertj@14807
  1800
				Some (foldl (fn (acc, elem) => HOLogic_insert $ elem $ acc) (HOLogic_empty_set, elements))
webertj@14807
  1801
			end handle CANNOT_INTERPRET _ => None)
webertj@14456
  1802
		| _ =>
webertj@14807
  1803
			None
webertj@14807
  1804
	end;
webertj@14350
  1805
webertj@14807
  1806
	(* theory -> model -> Term.term -> interpretation -> (int -> bool) -> Term.term option *)
webertj@14350
  1807
webertj@14807
  1808
	fun IDT_printer thy model t intr assignment =
webertj@14350
  1809
	let
webertj@14807
  1810
		(* Term.term -> Term.typ option *)
webertj@14807
  1811
		fun typeof (Free (_, T))  = Some T
webertj@14807
  1812
		  | typeof (Var (_, T))   = Some T
webertj@14807
  1813
		  | typeof (Const (_, T)) = Some T
webertj@14807
  1814
		  | typeof _              = None
webertj@14350
  1815
	in
webertj@14807
  1816
		case typeof t of
webertj@14807
  1817
		  Some (Type (s, Ts)) =>
webertj@14807
  1818
			(case DatatypePackage.datatype_info thy s of
webertj@14807
  1819
			  Some info =>  (* inductive datatype *)
webertj@14807
  1820
				let
webertj@14807
  1821
					val (typs, _)           = model
webertj@14807
  1822
					val index               = #index info
webertj@14807
  1823
					val descr               = #descr info
webertj@14807
  1824
					val (_, dtyps, constrs) = (the o assoc) (descr, index)
webertj@14807
  1825
					val typ_assoc           = dtyps ~~ Ts
webertj@14807
  1826
					(* sanity check: every element in 'dtyps' must be a 'DtTFree' *)
webertj@14807
  1827
					val _ = (if Library.exists (fn d =>
webertj@14807
  1828
							case d of DatatypeAux.DtTFree _ => false | _ => true) dtyps
webertj@14807
  1829
						then
webertj@14807
  1830
							raise REFUTE ("IDT_printer", "datatype argument (for type " ^ Sign.string_of_typ (sign_of thy) (Type (s, Ts)) ^ ") is not a variable")
webertj@14807
  1831
						else
webertj@14807
  1832
							())
webertj@14807
  1833
					(* the index of the element in the datatype *)
webertj@14807
  1834
					val element = (case intr of
webertj@14807
  1835
						  Leaf xs => find_index (PropLogic.eval assignment) xs
webertj@14807
  1836
						| Node _  => raise REFUTE ("IDT_printer", "interpretation is not a leaf"))
webertj@14807
  1837
					val _ = (if element<0 then raise REFUTE ("IDT_printer", "invalid interpretation (no value assigned)") else ())
webertj@14807
  1838
					(* int option -- only recursive IDTs have an associated depth *)
webertj@14807
  1839
					val depth = assoc (typs, Type (s, Ts))
webertj@14807
  1840
					val typs' = (case depth of None => typs | Some n => overwrite (typs, (Type (s, Ts), n-1)))
webertj@14807
  1841
					(* DatatypeAux.descr -> (DatatypeAux.dtyp * Term.typ) list -> DatatypeAux.dtyp -> Term.typ *)
webertj@14807
  1842
					fun typ_of_dtyp descr typ_assoc (DatatypeAux.DtTFree a) =
webertj@14807
  1843
						(* replace a 'DtTFree' variable by the associated type *)
webertj@14807
  1844
						(the o assoc) (typ_assoc, DatatypeAux.DtTFree a)
webertj@14807
  1845
					  | typ_of_dtyp descr typ_assoc (DatatypeAux.DtRec i) =
webertj@14807
  1846
						let
webertj@14807
  1847
							val (s, ds, _) = (the o assoc) (descr, i)
webertj@14807
  1848
						in
webertj@14807
  1849
							Type (s, map (typ_of_dtyp descr typ_assoc) ds)
webertj@14807
  1850
						end
webertj@14807
  1851
					  | typ_of_dtyp descr typ_assoc (DatatypeAux.DtType (s, ds)) =
webertj@14807
  1852
						Type (s, map (typ_of_dtyp descr typ_assoc) ds)
webertj@14807
  1853
					(* int list -> int *)
webertj@14807
  1854
					fun sum xs = foldl op+ (0, xs)
webertj@14807
  1855
					(* int list -> int *)
webertj@14807
  1856
					fun product xs = foldl op* (1, xs)
webertj@14807
  1857
					(* the size of an IDT is the sum (over its constructors) of the        *)
webertj@14807
  1858
					(* product (over their arguments) of the size of the argument type     *)
webertj@14807
  1859
					(* (Term.typ * int) list -> DatatypeAux.descr -> (DatatypeAux.dtyp * Term.typ) list -> (string * DatatypeAux.dtyp list) list -> int *)
webertj@14807
  1860
					fun size_of_dtyp typs descr typ_assoc xs =
webertj@14807
  1861
						sum (map (fn (_, ds) =>
webertj@14807
  1862
							product (map (fn d =>
webertj@14807
  1863
								let
webertj@14807
  1864
									val T         = typ_of_dtyp descr typ_assoc d
webertj@14807
  1865
									val (i, _, _) =
webertj@14807
  1866
										(interpret thy (typs, []) {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", T))
webertj@14807
  1867
										handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14456
  1868
					in
webertj@14807
  1869
						size_of_type i
webertj@14807
  1870
					end) ds)) xs)
webertj@14807
  1871
					(* int -> DatatypeAux.dtyp list -> Term.term list *)
webertj@14807
  1872
					fun make_args n [] =
webertj@14807
  1873
						if n<>0 then
webertj@14807
  1874
							raise REFUTE ("IDT_printer", "error computing the element: remainder is not 0")
webertj@14456
  1875
						else
webertj@14807
  1876
							[]
webertj@14807
  1877
					  | make_args n (d::ds) =
webertj@14350
  1878
						let
webertj@14807
  1879
							val dT        = typ_of_dtyp descr typ_assoc d
webertj@14807
  1880
							val (i, _, _) =
webertj@14807
  1881
								(interpret thy (typs', []) {maxvars=0, next_idx=1, bounds=[], wellformed=True} (Free ("dummy", dT))
webertj@14807
  1882
								handle CANNOT_INTERPRET _ => raise CANNOT_INTERPRET t)
webertj@14807
  1883
							val size      = size_of_type i
webertj@14807
  1884
							val consts    = make_constants i  (* we only need the (n mod size)-th element of *)
webertj@14807
  1885
								(* this list, so there might be a more efficient implementation that does not *)
webertj@14807
  1886
								(* generate all constants                                                     *)
webertj@14807
  1887
						in
webertj@14807
  1888
							(print thy (typs', []) (Free ("dummy", dT)) (nth_elem (n mod size, consts)) assignment)::(make_args (n div size) ds)
webertj@14807
  1889
						end
webertj@14807
  1890
					(* int -> (string * DatatypeAux.dtyp list) list -> Term.term *)
webertj@14807
  1891
					fun make_term _ [] =
webertj@14807
  1892
						raise REFUTE ("IDT_printer", "invalid interpretation (value too large - not enough constructors)")
webertj@14807
  1893
					  | make_term n (c::cs) =
webertj@14807
  1894
						let
webertj@14807
  1895
							val c_size = size_of_dtyp typs' descr typ_assoc [c]
webertj@14807
  1896
						in
webertj@14807
  1897
							if n<c_size then
webertj@14456
  1898
								let
webertj@14807
  1899
									val (cname, cargs) = c
webertj@14807
  1900
									val c_term = Const (cname, (map (typ_of_dtyp descr typ_assoc) cargs) ---> Type (s, Ts))
webertj@14456
  1901
								in
webertj@14807
  1902
									foldl op$ (c_term, rev (make_args n (rev cargs)))
webertj@14456
  1903
								end
webertj@14807
  1904
							else
webertj@14807
  1905
								make_term (n-c_size) cs
webertj@14350
  1906
						end
webertj@14807
  1907
				in
webertj@14807
  1908
					Some (make_term element constrs)
webertj@14807
  1909
				end
webertj@14807
  1910
			| None =>  (* not an inductive datatype *)
webertj@14807
  1911
				None)
webertj@14807
  1912
		| _ =>  (* a (free or schematic) type variable *)
webertj@14456
  1913
			None
webertj@14350
  1914
	end;
webertj@14350
  1915
webertj@14350
  1916
webertj@14350
  1917
(* ------------------------------------------------------------------------- *)
webertj@14456
  1918
(* use 'setup Refute.setup' in an Isabelle theory to initialize the 'Refute' *)
webertj@14456
  1919
(* structure                                                                 *)
webertj@14350
  1920
(* ------------------------------------------------------------------------- *)
webertj@14350
  1921
webertj@14350
  1922
(* ------------------------------------------------------------------------- *)
webertj@14456
  1923
(* Note: the interpreters and printers are used in reverse order; however,   *)
webertj@14456
  1924
(*       an interpreter that can handle non-atomic terms ends up being       *)
webertj@14807
  1925
(*       applied before the 'stlc_interpreter' breaks the term apart into    *)
webertj@14807
  1926
(*       subterms that are then passed to other interpreters!                *)
webertj@14350
  1927
(* ------------------------------------------------------------------------- *)
webertj@14350
  1928
webertj@14456
  1929
	(* (theory -> theory) list *)
webertj@14350
  1930
webertj@14456
  1931
	val setup =
webertj@14456
  1932
		[RefuteData.init,
webertj@14807
  1933
		 add_interpreter "stlc"            stlc_interpreter,
webertj@14807
  1934
		 add_interpreter "Pure"            Pure_interpreter,
webertj@14807
  1935
		 add_interpreter "HOLogic"         HOLogic_interpreter,
webertj@14807
  1936
		 add_interpreter "set"             set_interpreter,
webertj@14807
  1937
		 add_interpreter "IDT"             IDT_interpreter,
webertj@14807
  1938
		 add_interpreter "Finite_Set.card" Finite_Set_card_interpreter,
webertj@14807
  1939
		 add_printer "stlc" stlc_printer,
webertj@14807
  1940
		 add_printer "set"  set_printer,
webertj@14807
  1941
		 add_printer "IDT"  IDT_printer];
webertj@14350
  1942
webertj@14350
  1943
end