src/Pure/term.ML
author wenzelm
Tue Oct 04 19:01:37 2005 +0200 (2005-10-04)
changeset 17756 d4a35f82fbb4
parent 17642 e063c0403650
child 17777 05f5532a8289
permissions -rw-r--r--
minor tweaks for Poplog/ML;
wenzelm@9536
     1
(*  Title:      Pure/term.ML
clasohm@0
     2
    ID:         $Id$
wenzelm@9536
     3
    Author:     Lawrence C Paulson, Cambridge University Computer Laboratory
clasohm@0
     4
    Copyright   Cambridge University 1992
clasohm@1364
     5
wenzelm@4444
     6
Simply typed lambda-calculus: types, terms, and basic operations.
clasohm@0
     7
*)
clasohm@0
     8
clasohm@1364
     9
infix 9  $;
clasohm@1364
    10
infixr 5 -->;
wenzelm@4444
    11
infixr --->;
wenzelm@4444
    12
infix aconv;
clasohm@1364
    13
wenzelm@4444
    14
signature BASIC_TERM =
wenzelm@4444
    15
sig
wenzelm@4444
    16
  type indexname
wenzelm@4444
    17
  type class
wenzelm@4444
    18
  type sort
wenzelm@14829
    19
  type arity
wenzelm@4444
    20
  datatype typ =
wenzelm@4444
    21
    Type  of string * typ list |
wenzelm@4444
    22
    TFree of string * sort |
wenzelm@4444
    23
    TVar  of indexname * sort
wenzelm@16537
    24
  datatype term =
wenzelm@16537
    25
    Const of string * typ |
wenzelm@16537
    26
    Free of string * typ |
wenzelm@16537
    27
    Var of indexname * typ |
wenzelm@16537
    28
    Bound of int |
wenzelm@16537
    29
    Abs of string * typ * term |
wenzelm@17756
    30
    $ of term * term
wenzelm@16537
    31
  exception TYPE of string * typ list * term list
wenzelm@16537
    32
  exception TERM of string * term list
wenzelm@16710
    33
  val dummyT: typ
wenzelm@16710
    34
  val no_dummyT: typ -> typ
wenzelm@4444
    35
  val --> : typ * typ -> typ
wenzelm@4444
    36
  val ---> : typ list * typ -> typ
wenzelm@16710
    37
  val dest_Type: typ -> string * typ list
wenzelm@16710
    38
  val dest_TVar: typ -> indexname * sort
wenzelm@16710
    39
  val dest_TFree: typ -> string * sort
wenzelm@16710
    40
  val is_Bound: term -> bool
wenzelm@16710
    41
  val is_Const: term -> bool
wenzelm@16710
    42
  val is_Free: term -> bool
wenzelm@16710
    43
  val is_Var: term -> bool
wenzelm@4444
    44
  val is_TVar: typ -> bool
paulson@15573
    45
  val is_funtype: typ -> bool
wenzelm@16710
    46
  val dest_Const: term -> string * typ
wenzelm@16710
    47
  val dest_Free: term -> string * typ
wenzelm@16710
    48
  val dest_Var: term -> indexname * typ
wenzelm@4444
    49
  val domain_type: typ -> typ
wenzelm@4480
    50
  val range_type: typ -> typ
wenzelm@4444
    51
  val binder_types: typ -> typ list
wenzelm@4444
    52
  val body_type: typ -> typ
wenzelm@4444
    53
  val strip_type: typ -> typ list * typ
wenzelm@16710
    54
  val type_of1: typ list * term -> typ
wenzelm@4444
    55
  val type_of: term -> typ
wenzelm@16710
    56
  val fastype_of1: typ list * term -> typ
wenzelm@4444
    57
  val fastype_of: term -> typ
wenzelm@10806
    58
  val list_abs: (string * typ) list * term -> term
wenzelm@4444
    59
  val strip_abs_body: term -> term
wenzelm@4444
    60
  val strip_abs_vars: term -> (string * typ) list
wenzelm@4444
    61
  val strip_qnt_body: string -> term -> term
wenzelm@4444
    62
  val strip_qnt_vars: string -> term -> (string * typ) list
wenzelm@4444
    63
  val list_comb: term * term list -> term
wenzelm@4444
    64
  val strip_comb: term -> term * term list
wenzelm@4444
    65
  val head_of: term -> term
wenzelm@4444
    66
  val size_of_term: term -> int
wenzelm@4444
    67
  val map_type_tvar: (indexname * sort -> typ) -> typ -> typ
wenzelm@4444
    68
  val map_type_tfree: (string * sort -> typ) -> typ -> typ
wenzelm@4444
    69
  val map_term_types: (typ -> typ) -> term -> term
wenzelm@4444
    70
  val it_term_types: (typ * 'a -> 'a) -> term * 'a -> 'a
wenzelm@16943
    71
  val fold_atyps: (typ -> 'a -> 'a) -> typ -> 'a -> 'a
wenzelm@16943
    72
  val fold_aterms: (term -> 'a -> 'a) -> term -> 'a -> 'a
wenzelm@16943
    73
  val fold_term_types: (term -> typ -> 'a -> 'a) -> term -> 'a -> 'a
wenzelm@16943
    74
  val fold_types: (typ -> 'a -> 'a) -> term -> 'a -> 'a
wenzelm@16943
    75
  val add_term_names: term * string list -> string list
wenzelm@16943
    76
  val add_term_varnames: term -> indexname list -> indexname list
wenzelm@16943
    77
  val term_varnames: term -> indexname list
wenzelm@16710
    78
  val aconv: term * term -> bool
wenzelm@16710
    79
  val aconvs: term list * term list -> bool
wenzelm@16537
    80
  structure Vartab: TABLE
wenzelm@16537
    81
  structure Typtab: TABLE
wenzelm@16537
    82
  structure Termtab: TABLE
wenzelm@4444
    83
  val itselfT: typ -> typ
wenzelm@4444
    84
  val a_itselfT: typ
wenzelm@4444
    85
  val propT: typ
wenzelm@4444
    86
  val implies: term
wenzelm@4444
    87
  val all: typ -> term
wenzelm@4444
    88
  val equals: typ -> term
wenzelm@4444
    89
  val strip_all_body: term -> term
wenzelm@4444
    90
  val strip_all_vars: term -> (string * typ) list
wenzelm@4444
    91
  val incr_bv: int * int * term -> term
wenzelm@4444
    92
  val incr_boundvars: int -> term -> term
wenzelm@4444
    93
  val add_loose_bnos: term * int * int list -> int list
wenzelm@4444
    94
  val loose_bnos: term -> int list
wenzelm@4444
    95
  val loose_bvar: term * int -> bool
wenzelm@4444
    96
  val loose_bvar1: term * int -> bool
wenzelm@4444
    97
  val subst_bounds: term list * term -> term
wenzelm@4444
    98
  val subst_bound: term * term -> term
wenzelm@4444
    99
  val betapply: term * term -> term
wenzelm@4444
   100
  val eq_ix: indexname * indexname -> bool
wenzelm@4444
   101
  val ins_ix: indexname * indexname list -> indexname list
wenzelm@4444
   102
  val mem_ix: indexname * indexname list -> bool
wenzelm@4444
   103
  val mem_term: term * term list -> bool
wenzelm@4444
   104
  val subset_term: term list * term list -> bool
wenzelm@4444
   105
  val eq_set_term: term list * term list -> bool
wenzelm@4444
   106
  val ins_term: term * term list -> term list
wenzelm@4444
   107
  val union_term: term list * term list -> term list
paulson@5585
   108
  val inter_term: term list * term list -> term list
wenzelm@4444
   109
  val could_unify: term * term -> bool
wenzelm@4444
   110
  val subst_free: (term * term) list -> term -> term
wenzelm@4444
   111
  val xless: (string * int) * indexname -> bool
wenzelm@4444
   112
  val abstract_over: term * term -> term
wenzelm@11922
   113
  val lambda: term -> term -> term
wenzelm@4444
   114
  val absfree: string * typ * term -> term
wenzelm@4444
   115
  val list_abs_free: (string * typ) list * term -> term
wenzelm@4444
   116
  val list_all_free: (string * typ) list * term -> term
wenzelm@4444
   117
  val list_all: (string * typ) list * term -> term
wenzelm@16710
   118
  val subst_atomic: (term * term) list -> term -> term
wenzelm@16710
   119
  val typ_subst_atomic: (typ * typ) list -> typ -> typ
wenzelm@16710
   120
  val subst_atomic_types: (typ * typ) list -> term -> term
wenzelm@16710
   121
  val typ_subst_TVars: (indexname * typ) list -> typ -> typ
wenzelm@16710
   122
  val subst_TVars: (indexname * typ) list -> term -> term
wenzelm@16710
   123
  val subst_Vars: (indexname * term) list -> term -> term
wenzelm@16710
   124
  val subst_vars: (indexname * typ) list * (indexname * term) list -> term -> term
wenzelm@16710
   125
  val is_first_order: string list -> term -> bool
wenzelm@4444
   126
  val maxidx_of_typ: typ -> int
wenzelm@4444
   127
  val maxidx_of_typs: typ list -> int
wenzelm@4444
   128
  val maxidx_of_term: term -> int
wenzelm@4444
   129
  val variant: string list -> string -> string
wenzelm@4444
   130
  val variantlist: string list * string list -> string list
wenzelm@16537
   131
    (*note reversed order of args wrt. variant!*)
wenzelm@4444
   132
  val add_typ_classes: typ * class list -> class list
wenzelm@4444
   133
  val add_typ_tycons: typ * string list -> string list
wenzelm@4444
   134
  val add_term_classes: term * class list -> class list
wenzelm@16710
   135
  val add_term_tycons: term * string list -> string list
wenzelm@4444
   136
  val add_term_consts: term * string list -> string list
nipkow@13646
   137
  val term_consts: term -> string list
wenzelm@16943
   138
  val exists_subterm: (term -> bool) -> term -> bool
wenzelm@16710
   139
  val exists_Const: (string * typ -> bool) -> term -> bool
wenzelm@16710
   140
  val add_term_free_names: term * string list -> string list
wenzelm@16710
   141
  val add_typ_tvars: typ * (indexname * sort) list -> (indexname * sort) list
wenzelm@16710
   142
  val add_typ_tfree_names: typ * string list -> string list
wenzelm@16710
   143
  val add_typ_tfrees: typ * (string * sort) list -> (string * sort) list
wenzelm@16710
   144
  val add_typ_varnames: typ * string list -> string list
wenzelm@16710
   145
  val add_term_tvars: term * (indexname * sort) list -> (indexname * sort) list
wenzelm@16710
   146
  val add_term_tfrees: term * (string * sort) list -> (string * sort) list
wenzelm@16710
   147
  val add_term_tfree_names: term * string list -> string list
wenzelm@16710
   148
  val add_term_tvarnames: term * string list -> string list
wenzelm@16710
   149
  val typ_tfrees: typ -> (string * sort) list
wenzelm@16710
   150
  val typ_tvars: typ -> (indexname * sort) list
wenzelm@16710
   151
  val term_tfrees: term -> (string * sort) list
wenzelm@16710
   152
  val term_tvars: term -> (indexname * sort) list
wenzelm@16710
   153
  val add_typ_ixns: indexname list * typ -> indexname list
wenzelm@16710
   154
  val add_term_tvar_ixns: term * indexname list -> indexname list
wenzelm@16710
   155
  val add_term_vars: term * term list -> term list
wenzelm@16710
   156
  val term_vars: term -> term list
wenzelm@4444
   157
  val add_term_frees: term * term list -> term list
wenzelm@4444
   158
  val term_frees: term -> term list
wenzelm@16710
   159
  val variant_abs: string * typ * term -> string * term
wenzelm@16710
   160
  val rename_wrt_term: term -> (string * typ) list -> (string * typ) list
wenzelm@15986
   161
  val show_question_marks: bool ref
wenzelm@4444
   162
end;
clasohm@0
   163
wenzelm@4444
   164
signature TERM =
wenzelm@4444
   165
sig
wenzelm@4444
   166
  include BASIC_TERM
wenzelm@16710
   167
  val argument_type_of: term -> typ
wenzelm@16943
   168
  val add_tvarsT: typ -> (indexname * sort) list -> (indexname * sort) list
wenzelm@16943
   169
  val add_tvars: term -> (indexname * sort) list -> (indexname * sort) list
wenzelm@16943
   170
  val add_vars: term -> (indexname * typ) list -> (indexname * typ) list
wenzelm@16943
   171
  val add_tfreesT: typ -> (string * sort) list -> (string * sort) list
wenzelm@16943
   172
  val add_tfrees: term -> (string * sort) list -> (string * sort) list
wenzelm@16943
   173
  val add_frees: term -> (string * typ) list -> (string * typ) list
wenzelm@16678
   174
  val fast_indexname_ord: indexname * indexname -> order
wenzelm@16537
   175
  val indexname_ord: indexname * indexname -> order
wenzelm@16537
   176
  val sort_ord: sort * sort -> order
wenzelm@16537
   177
  val typ_ord: typ * typ -> order
wenzelm@16678
   178
  val fast_term_ord: term * term -> order
wenzelm@16537
   179
  val term_ord: term * term -> order
wenzelm@16537
   180
  val hd_ord: term * term -> order
wenzelm@16537
   181
  val termless: term * term -> bool
nipkow@16570
   182
  val term_lpo: (string -> int) -> term * term -> order
wenzelm@12981
   183
  val match_bvars: (term * term) * (string * string) list -> (string * string) list
wenzelm@12981
   184
  val rename_abs: term -> term -> term -> term option
wenzelm@16943
   185
  val eq_tvar: (indexname * sort) * (indexname * sort) -> bool
wenzelm@16882
   186
  val eq_var: (indexname * typ) * (indexname * typ) -> bool
wenzelm@16943
   187
  val tvar_ord: (indexname * sort) * (indexname * sort) -> order
wenzelm@16943
   188
  val var_ord: (indexname * typ) * (indexname * typ) -> order
wenzelm@16882
   189
  val instantiate: ((indexname * sort) * typ) list * ((indexname * typ) * term) list
wenzelm@16882
   190
    -> term -> term
wenzelm@16882
   191
  val instantiateT: ((indexname * sort) * typ) list -> typ -> typ
wenzelm@16710
   192
  val maxidx_typ: typ -> int -> int
wenzelm@16710
   193
  val maxidx_typs: typ list -> int -> int
wenzelm@16710
   194
  val maxidx_term: term -> int -> int
wenzelm@12499
   195
  val invent_names: string list -> string -> int -> string list
wenzelm@16338
   196
  val map_typ: (string -> string) -> (string -> string) -> typ -> typ
wenzelm@16338
   197
  val map_term: (string -> string) -> (string -> string) -> (string -> string) -> term -> term
wenzelm@16710
   198
  val dest_abs: string * typ * term -> string * term
wenzelm@16990
   199
  val bound: int -> string
wenzelm@16990
   200
  val is_bound: string -> bool
wenzelm@16943
   201
  val zero_var_indexesT: typ -> typ
wenzelm@16943
   202
  val zero_var_indexes: term -> term
wenzelm@16943
   203
  val zero_var_indexes_inst: term ->
wenzelm@16943
   204
    ((indexname * sort) * typ) list * ((indexname * typ) * term) list
wenzelm@9536
   205
  val dummy_patternN: string
wenzelm@9536
   206
  val no_dummy_patterns: term -> term
wenzelm@9536
   207
  val replace_dummy_patterns: int * term -> int * term
wenzelm@10552
   208
  val is_replaced_dummy_pattern: indexname -> bool
wenzelm@16035
   209
  val show_dummy_patterns: term -> term
wenzelm@13484
   210
  val adhoc_freeze_vars: term -> term * string list
wenzelm@14786
   211
  val string_of_vname: indexname -> string
wenzelm@14786
   212
  val string_of_vname': indexname -> string
wenzelm@4444
   213
end;
wenzelm@4444
   214
wenzelm@4444
   215
structure Term: TERM =
clasohm@1364
   216
struct
clasohm@0
   217
clasohm@0
   218
(*Indexnames can be quickly renamed by adding an offset to the integer part,
clasohm@0
   219
  for resolution.*)
wenzelm@16537
   220
type indexname = string * int;
clasohm@0
   221
wenzelm@4626
   222
(* Types are classified by sorts. *)
clasohm@0
   223
type class = string;
clasohm@0
   224
type sort  = class list;
wenzelm@14829
   225
type arity = string * sort list * sort;
clasohm@0
   226
clasohm@0
   227
(* The sorts attached to TFrees and TVars specify the sort of that variable *)
clasohm@0
   228
datatype typ = Type  of string * typ list
clasohm@0
   229
             | TFree of string * sort
wenzelm@9536
   230
             | TVar  of indexname * sort;
clasohm@0
   231
paulson@6033
   232
(*Terms.  Bound variables are indicated by depth number.
clasohm@0
   233
  Free variables, (scheme) variables and constants have names.
wenzelm@4626
   234
  An term is "closed" if every bound variable of level "lev"
wenzelm@13000
   235
  is enclosed by at least "lev" abstractions.
clasohm@0
   236
clasohm@0
   237
  It is possible to create meaningless terms containing loose bound vars
clasohm@0
   238
  or type mismatches.  But such terms are not allowed in rules. *)
clasohm@0
   239
wenzelm@13000
   240
datatype term =
clasohm@0
   241
    Const of string * typ
wenzelm@13000
   242
  | Free  of string * typ
clasohm@0
   243
  | Var   of indexname * typ
clasohm@0
   244
  | Bound of int
clasohm@0
   245
  | Abs   of string*typ*term
wenzelm@3965
   246
  | op $  of term*term;
clasohm@0
   247
wenzelm@16537
   248
(*Errors involving type mismatches*)
clasohm@0
   249
exception TYPE of string * typ list * term list;
clasohm@0
   250
wenzelm@16537
   251
(*Errors errors involving terms*)
clasohm@0
   252
exception TERM of string * term list;
clasohm@0
   253
clasohm@0
   254
(*Note variable naming conventions!
clasohm@0
   255
    a,b,c: string
clasohm@0
   256
    f,g,h: functions (including terms of function type)
clasohm@0
   257
    i,j,m,n: int
clasohm@0
   258
    t,u: term
clasohm@0
   259
    v,w: indexnames
clasohm@0
   260
    x,y: any
clasohm@0
   261
    A,B,C: term (denoting formulae)
clasohm@0
   262
    T,U: typ
clasohm@0
   263
*)
clasohm@0
   264
clasohm@0
   265
paulson@6033
   266
(** Types **)
paulson@6033
   267
wenzelm@16537
   268
(*dummy type for parsing and printing etc.*)
wenzelm@16537
   269
val dummyT = Type ("dummy", []);
wenzelm@16537
   270
wenzelm@16537
   271
fun no_dummyT typ =
wenzelm@16537
   272
  let
wenzelm@16537
   273
    fun check (T as Type ("dummy", _)) =
wenzelm@16537
   274
          raise TYPE ("Illegal occurrence of '_' dummy type", [T], [])
wenzelm@16537
   275
      | check (Type (_, Ts)) = List.app check Ts
wenzelm@16537
   276
      | check _ = ();
wenzelm@16537
   277
  in check typ; typ end;
wenzelm@16537
   278
paulson@6033
   279
fun S --> T = Type("fun",[S,T]);
paulson@6033
   280
paulson@6033
   281
(*handy for multiple args: [T1,...,Tn]--->T  gives  T1-->(T2--> ... -->T)*)
skalberg@15570
   282
val op ---> = Library.foldr (op -->);
paulson@6033
   283
paulson@6033
   284
fun dest_Type (Type x) = x
paulson@6033
   285
  | dest_Type T = raise TYPE ("dest_Type", [T], []);
dixon@15914
   286
fun dest_TVar (TVar x) = x
dixon@15914
   287
  | dest_TVar T = raise TYPE ("dest_TVar", [T], []);
dixon@15914
   288
fun dest_TFree (TFree x) = x
dixon@15914
   289
  | dest_TFree T = raise TYPE ("dest_TFree", [T], []);
paulson@6033
   290
wenzelm@16537
   291
clasohm@0
   292
(** Discriminators **)
clasohm@0
   293
nipkow@7318
   294
fun is_Bound (Bound _) = true
nipkow@7318
   295
  | is_Bound _         = false;
nipkow@7318
   296
clasohm@0
   297
fun is_Const (Const _) = true
clasohm@0
   298
  | is_Const _ = false;
clasohm@0
   299
clasohm@0
   300
fun is_Free (Free _) = true
clasohm@0
   301
  | is_Free _ = false;
clasohm@0
   302
clasohm@0
   303
fun is_Var (Var _) = true
clasohm@0
   304
  | is_Var _ = false;
clasohm@0
   305
clasohm@0
   306
fun is_TVar (TVar _) = true
clasohm@0
   307
  | is_TVar _ = false;
clasohm@0
   308
paulson@15573
   309
(*Differs from proofterm/is_fun in its treatment of TVar*)
paulson@15573
   310
fun is_funtype (Type("fun",[_,_])) = true
paulson@15573
   311
  | is_funtype _ = false;
paulson@15573
   312
wenzelm@16537
   313
clasohm@0
   314
(** Destructors **)
clasohm@0
   315
clasohm@0
   316
fun dest_Const (Const x) =  x
clasohm@0
   317
  | dest_Const t = raise TERM("dest_Const", [t]);
clasohm@0
   318
clasohm@0
   319
fun dest_Free (Free x) =  x
clasohm@0
   320
  | dest_Free t = raise TERM("dest_Free", [t]);
clasohm@0
   321
clasohm@0
   322
fun dest_Var (Var x) =  x
clasohm@0
   323
  | dest_Var t = raise TERM("dest_Var", [t]);
clasohm@0
   324
clasohm@0
   325
paulson@4464
   326
fun domain_type (Type("fun", [T,_])) = T
paulson@4464
   327
and range_type  (Type("fun", [_,T])) = T;
paulson@4064
   328
clasohm@0
   329
(* maps  [T1,...,Tn]--->T  to the list  [T1,T2,...,Tn]*)
clasohm@0
   330
fun binder_types (Type("fun",[S,T])) = S :: binder_types T
clasohm@0
   331
  | binder_types _   =  [];
clasohm@0
   332
clasohm@0
   333
(* maps  [T1,...,Tn]--->T  to T*)
clasohm@0
   334
fun body_type (Type("fun",[S,T])) = body_type T
clasohm@0
   335
  | body_type T   =  T;
clasohm@0
   336
clasohm@0
   337
(* maps  [T1,...,Tn]--->T  to   ([T1,T2,...,Tn], T)  *)
clasohm@0
   338
fun strip_type T : typ list * typ =
clasohm@0
   339
  (binder_types T, body_type T);
clasohm@0
   340
clasohm@0
   341
clasohm@0
   342
(*Compute the type of the term, checking that combinations are well-typed
clasohm@0
   343
  Ts = [T0,T1,...] holds types of bound variables 0, 1, ...*)
clasohm@0
   344
fun type_of1 (Ts, Const (_,T)) = T
clasohm@0
   345
  | type_of1 (Ts, Free  (_,T)) = T
skalberg@15570
   346
  | type_of1 (Ts, Bound i) = (List.nth (Ts,i)
skalberg@15570
   347
        handle Subscript => raise TYPE("type_of: bound variable", [], [Bound i]))
clasohm@0
   348
  | type_of1 (Ts, Var (_,T)) = T
clasohm@0
   349
  | type_of1 (Ts, Abs (_,T,body)) = T --> type_of1(T::Ts, body)
wenzelm@13000
   350
  | type_of1 (Ts, f$u) =
clasohm@0
   351
      let val U = type_of1(Ts,u)
clasohm@0
   352
          and T = type_of1(Ts,f)
clasohm@0
   353
      in case T of
wenzelm@9536
   354
            Type("fun",[T1,T2]) =>
wenzelm@9536
   355
              if T1=U then T2  else raise TYPE
wenzelm@9536
   356
                    ("type_of: type mismatch in application", [T1,U], [f$u])
wenzelm@13000
   357
          | _ => raise TYPE
wenzelm@9536
   358
                    ("type_of: function type is expected in application",
wenzelm@9536
   359
                     [T,U], [f$u])
clasohm@0
   360
      end;
clasohm@0
   361
clasohm@0
   362
fun type_of t : typ = type_of1 ([],t);
clasohm@0
   363
clasohm@0
   364
(*Determines the type of a term, with minimal checking*)
wenzelm@13000
   365
fun fastype_of1 (Ts, f$u) =
lcp@61
   366
    (case fastype_of1 (Ts,f) of
wenzelm@9536
   367
        Type("fun",[_,T]) => T
wenzelm@9536
   368
        | _ => raise TERM("fastype_of: expected function type", [f$u]))
lcp@61
   369
  | fastype_of1 (_, Const (_,T)) = T
lcp@61
   370
  | fastype_of1 (_, Free (_,T)) = T
skalberg@15570
   371
  | fastype_of1 (Ts, Bound i) = (List.nth(Ts,i)
skalberg@15570
   372
         handle Subscript => raise TERM("fastype_of: Bound", [Bound i]))
wenzelm@13000
   373
  | fastype_of1 (_, Var (_,T)) = T
lcp@61
   374
  | fastype_of1 (Ts, Abs (_,T,u)) = T --> fastype_of1 (T::Ts, u);
lcp@61
   375
lcp@61
   376
fun fastype_of t : typ = fastype_of1 ([],t);
clasohm@0
   377
wenzelm@16678
   378
(*Determine the argument type of a function*)
wenzelm@16678
   379
fun argument_type_of tm =
wenzelm@16678
   380
  let
wenzelm@16678
   381
    fun argT i (Type ("fun", [T, U])) = if i = 0 then T else argT (i - 1) U
wenzelm@16678
   382
      | argT _ T = raise TYPE ("argument_type_of", [T], []);
wenzelm@16678
   383
wenzelm@16678
   384
    fun arg 0 _ (Abs (_, T, _)) = T
wenzelm@16678
   385
      | arg i Ts (Abs (_, T, t)) = arg (i - 1) (T :: Ts) t
wenzelm@16678
   386
      | arg i Ts (t $ _) = arg (i + 1) Ts t
wenzelm@16678
   387
      | arg i Ts a = argT i (fastype_of1 (Ts, a));
wenzelm@16678
   388
  in arg 0 [] tm end;
wenzelm@16678
   389
clasohm@0
   390
skalberg@15570
   391
val list_abs = Library.foldr (fn ((x, T), t) => Abs (x, T, t));
wenzelm@10806
   392
clasohm@0
   393
(* maps  (x1,...,xn)t   to   t  *)
wenzelm@13000
   394
fun strip_abs_body (Abs(_,_,t))  =  strip_abs_body t
clasohm@0
   395
  | strip_abs_body u  =  u;
clasohm@0
   396
clasohm@0
   397
(* maps  (x1,...,xn)t   to   [x1, ..., xn]  *)
wenzelm@13000
   398
fun strip_abs_vars (Abs(a,T,t))  =  (a,T) :: strip_abs_vars t
clasohm@0
   399
  | strip_abs_vars u  =  [] : (string*typ) list;
clasohm@0
   400
clasohm@0
   401
clasohm@0
   402
fun strip_qnt_body qnt =
clasohm@0
   403
let fun strip(tm as Const(c,_)$Abs(_,_,t)) = if c=qnt then strip t else tm
clasohm@0
   404
      | strip t = t
clasohm@0
   405
in strip end;
clasohm@0
   406
clasohm@0
   407
fun strip_qnt_vars qnt =
clasohm@0
   408
let fun strip(Const(c,_)$Abs(a,T,t)) = if c=qnt then (a,T)::strip t else []
clasohm@0
   409
      | strip t  =  [] : (string*typ) list
clasohm@0
   410
in strip end;
clasohm@0
   411
clasohm@0
   412
clasohm@0
   413
(* maps   (f, [t1,...,tn])  to  f(t1,...,tn) *)
skalberg@15570
   414
val list_comb : term * term list -> term = Library.foldl (op $);
clasohm@0
   415
clasohm@0
   416
clasohm@0
   417
(* maps   f(t1,...,tn)  to  (f, [t1,...,tn]) ; naturally tail-recursive*)
wenzelm@13000
   418
fun strip_comb u : term * term list =
clasohm@0
   419
    let fun stripc (f$t, ts) = stripc (f, t::ts)
wenzelm@13000
   420
        |   stripc  x =  x
clasohm@0
   421
    in  stripc(u,[])  end;
clasohm@0
   422
clasohm@0
   423
clasohm@0
   424
(* maps   f(t1,...,tn)  to  f , which is never a combination *)
clasohm@0
   425
fun head_of (f$t) = head_of f
clasohm@0
   426
  | head_of u = u;
clasohm@0
   427
clasohm@0
   428
wenzelm@16599
   429
(*number of atoms and abstractions in a term*)
wenzelm@16599
   430
fun size_of_term tm =
wenzelm@16599
   431
  let
wenzelm@16678
   432
    fun add_size (t $ u, n) = add_size (t, add_size (u, n))
wenzelm@16678
   433
      | add_size (Abs (_ ,_, t), n) = add_size (t, n + 1)
wenzelm@16678
   434
      | add_size (_, n) = n + 1;
wenzelm@16678
   435
  in add_size (tm, 0) end;
clasohm@0
   436
wenzelm@16678
   437
fun map_type_tvar f =
wenzelm@16678
   438
  let
wenzelm@16678
   439
    fun map_aux (Type (a, Ts)) = Type (a, map map_aux Ts)
wenzelm@16678
   440
      | map_aux (TVar x) = f x
wenzelm@16678
   441
      | map_aux T = T;
wenzelm@16678
   442
  in map_aux end;
nipkow@949
   443
wenzelm@16678
   444
fun map_type_tfree f =
wenzelm@16678
   445
  let
wenzelm@16678
   446
    fun map_aux (Type (a, Ts)) = Type (a, map map_aux Ts)
wenzelm@16678
   447
      | map_aux (TFree x) = f x
wenzelm@16678
   448
      | map_aux T = T;
wenzelm@16678
   449
  in map_aux end;
nipkow@949
   450
clasohm@0
   451
fun map_term_types f =
wenzelm@16678
   452
  let
wenzelm@16678
   453
    fun map_aux (Const (a, T)) = Const (a, f T)
wenzelm@16678
   454
      | map_aux (Free (a, T)) = Free (a, f T)
wenzelm@16678
   455
      | map_aux (Var (v, T)) = Var (v, f T)
wenzelm@16678
   456
      | map_aux (t as Bound _)  = t
wenzelm@16678
   457
      | map_aux (Abs (a, T, t)) = Abs (a, f T, map_aux t)
wenzelm@16678
   458
      | map_aux (t $ u) = map_aux t $ map_aux u;
wenzelm@16678
   459
  in map_aux end;
clasohm@0
   460
clasohm@0
   461
(* iterate a function over all types in a term *)
clasohm@0
   462
fun it_term_types f =
clasohm@0
   463
let fun iter(Const(_,T), a) = f(T,a)
clasohm@0
   464
      | iter(Free(_,T), a) = f(T,a)
clasohm@0
   465
      | iter(Var(_,T), a) = f(T,a)
clasohm@0
   466
      | iter(Abs(_,T,t), a) = iter(t,f(T,a))
clasohm@0
   467
      | iter(f$u, a) = iter(f, iter(u, a))
clasohm@0
   468
      | iter(Bound _, a) = a
clasohm@0
   469
in iter end
clasohm@0
   470
clasohm@0
   471
wenzelm@16943
   472
(* fold types and terms *)
wenzelm@16943
   473
wenzelm@16943
   474
(*fold atoms of type*)
wenzelm@16943
   475
fun fold_atyps f (Type (_, Ts)) = fold (fold_atyps f) Ts
wenzelm@16943
   476
  | fold_atyps f T = f T;
wenzelm@16943
   477
wenzelm@16943
   478
(*fold atoms of term*)
wenzelm@16943
   479
fun fold_aterms f (t $ u) = fold_aterms f t #> fold_aterms f u
wenzelm@16943
   480
  | fold_aterms f (Abs (_, _, t)) = fold_aterms f t
wenzelm@16943
   481
  | fold_aterms f a = f a;
wenzelm@16943
   482
wenzelm@16943
   483
(*fold types of term*)
wenzelm@16943
   484
fun fold_term_types f (t as Const (_, T)) = f t T
wenzelm@16943
   485
  | fold_term_types f (t as Free (_, T)) = f t T
wenzelm@16943
   486
  | fold_term_types f (t as Var (_, T)) = f t T
wenzelm@16943
   487
  | fold_term_types f (Bound _) = I
wenzelm@16943
   488
  | fold_term_types f (t as Abs (_, T, b)) = f t T #> fold_term_types f b
wenzelm@16943
   489
  | fold_term_types f (t $ u) = fold_term_types f t #> fold_term_types f u;
wenzelm@16943
   490
wenzelm@16943
   491
fun fold_types f = fold_term_types (K f);
wenzelm@16943
   492
wenzelm@16943
   493
(*collect variables*)
wenzelm@16943
   494
val add_tvarsT = fold_atyps (fn TVar v => insert (op =) v | _ => I);
wenzelm@16943
   495
val add_tvars = fold_types add_tvarsT;
wenzelm@16943
   496
val add_vars = fold_aterms (fn Var v => insert (op =) v | _ => I);
wenzelm@16943
   497
val add_tfreesT = fold_atyps (fn TFree v => insert (op =) v | _ => I);
wenzelm@16943
   498
val add_tfrees = fold_types add_tfreesT;
wenzelm@16943
   499
val add_frees = fold_aterms (fn Free v => insert (op =) v | _ => I);
wenzelm@16943
   500
wenzelm@16943
   501
(*collect variable names*)
wenzelm@16943
   502
val add_term_varnames = fold_aterms (fn Var (xi, _) => insert (op =) xi | _ => I);
wenzelm@16943
   503
fun term_varnames t = add_term_varnames t [];
wenzelm@16943
   504
wenzelm@16943
   505
wenzelm@16678
   506
(** Comparing terms, types, sorts etc. **)
wenzelm@16537
   507
wenzelm@16678
   508
(* fast syntactic comparison *)
wenzelm@16678
   509
wenzelm@16678
   510
fun fast_indexname_ord ((x, i), (y, j)) =
wenzelm@16678
   511
  (case int_ord (i, j) of EQUAL => fast_string_ord (x, y) | ord => ord);
wenzelm@16537
   512
wenzelm@16599
   513
fun sort_ord SS =
wenzelm@16599
   514
  if pointer_eq SS then EQUAL
wenzelm@16990
   515
  else dict_ord fast_string_ord SS;
wenzelm@16678
   516
wenzelm@16678
   517
local
wenzelm@16537
   518
wenzelm@16678
   519
fun cons_nr (TVar _) = 0
wenzelm@16678
   520
  | cons_nr (TFree _) = 1
wenzelm@16678
   521
  | cons_nr (Type _) = 2;
wenzelm@16537
   522
wenzelm@16678
   523
in
wenzelm@16537
   524
wenzelm@16537
   525
fun typ_ord TU =
wenzelm@16537
   526
  if pointer_eq TU then EQUAL
wenzelm@16537
   527
  else
wenzelm@16537
   528
    (case TU of
wenzelm@16678
   529
      (Type (a, Ts), Type (b, Us)) =>
wenzelm@16990
   530
        (case fast_string_ord (a, b) of EQUAL => dict_ord typ_ord (Ts, Us) | ord => ord)
wenzelm@16678
   531
    | (TFree (a, S), TFree (b, S')) =>
wenzelm@16678
   532
        (case fast_string_ord (a, b) of EQUAL => sort_ord (S, S') | ord => ord)
wenzelm@16678
   533
    | (TVar (xi, S), TVar (yj, S')) =>
wenzelm@16678
   534
        (case fast_indexname_ord (xi, yj) of EQUAL => sort_ord (S, S') | ord => ord)
wenzelm@16678
   535
    | (T, U) => int_ord (cons_nr T, cons_nr U));
wenzelm@16678
   536
wenzelm@16678
   537
end;
wenzelm@16678
   538
wenzelm@16678
   539
local
wenzelm@16678
   540
wenzelm@16678
   541
fun cons_nr (Const _) = 0
wenzelm@16678
   542
  | cons_nr (Free _) = 1
wenzelm@16678
   543
  | cons_nr (Var _) = 2
wenzelm@16678
   544
  | cons_nr (Bound _) = 3
wenzelm@16678
   545
  | cons_nr (Abs _) = 4
wenzelm@16678
   546
  | cons_nr (_ $ _) = 5;
wenzelm@16678
   547
wenzelm@16678
   548
fun struct_ord (Abs (_, _, t), Abs (_, _, u)) = struct_ord (t, u)
wenzelm@16678
   549
  | struct_ord (t1 $ t2, u1 $ u2) =
wenzelm@16678
   550
      (case struct_ord (t1, u1) of EQUAL => struct_ord (t2, u2) | ord => ord)
wenzelm@16678
   551
  | struct_ord (t, u) = int_ord (cons_nr t, cons_nr u);
wenzelm@16678
   552
wenzelm@16678
   553
fun atoms_ord (Abs (_, _, t), Abs (_, _, u)) = atoms_ord (t, u)
wenzelm@16678
   554
  | atoms_ord (t1 $ t2, u1 $ u2) =
wenzelm@16678
   555
      (case atoms_ord (t1, u1) of EQUAL => atoms_ord (t2, u2) | ord => ord)
wenzelm@16678
   556
  | atoms_ord (Const (a, _), Const (b, _)) = fast_string_ord (a, b)
wenzelm@16678
   557
  | atoms_ord (Free (x, _), Free (y, _)) = fast_string_ord (x, y)
wenzelm@16678
   558
  | atoms_ord (Var (xi, _), Var (yj, _)) = fast_indexname_ord (xi, yj)
wenzelm@16678
   559
  | atoms_ord (Bound i, Bound j) = int_ord (i, j)
wenzelm@16678
   560
  | atoms_ord _ = sys_error "atoms_ord";
wenzelm@16678
   561
wenzelm@16678
   562
fun types_ord (Abs (_, T, t), Abs (_, U, u)) =
wenzelm@16678
   563
      (case typ_ord (T, U) of EQUAL => types_ord (t, u) | ord => ord)
wenzelm@16678
   564
  | types_ord (t1 $ t2, u1 $ u2) =
wenzelm@16678
   565
      (case types_ord (t1, u1) of EQUAL => types_ord (t2, u2) | ord => ord)
wenzelm@16678
   566
  | types_ord (Const (_, T), Const (_, U)) = typ_ord (T, U)
wenzelm@16678
   567
  | types_ord (Free (_, T), Free (_, U)) = typ_ord (T, U)
wenzelm@16678
   568
  | types_ord (Var (_, T), Var (_, U)) = typ_ord (T, U)
wenzelm@16678
   569
  | types_ord (Bound _, Bound _) = EQUAL
wenzelm@16678
   570
  | types_ord _ = sys_error "types_ord";
wenzelm@16678
   571
wenzelm@16678
   572
in
wenzelm@16678
   573
wenzelm@16678
   574
fun fast_term_ord tu =
wenzelm@16678
   575
  if pointer_eq tu then EQUAL
wenzelm@16678
   576
  else
wenzelm@16678
   577
    (case struct_ord tu of
wenzelm@16678
   578
      EQUAL => (case atoms_ord tu of EQUAL => types_ord tu | ord => ord)
wenzelm@16678
   579
    | ord => ord);
wenzelm@16678
   580
wenzelm@16678
   581
fun op aconv tu = (fast_term_ord tu = EQUAL);
wenzelm@16678
   582
fun aconvs ts_us = (list_ord fast_term_ord ts_us = EQUAL);
wenzelm@16678
   583
wenzelm@16678
   584
structure Vartab = TableFun(type key = indexname val ord = fast_indexname_ord);
wenzelm@16678
   585
structure Typtab = TableFun(type key = typ val ord = typ_ord);
wenzelm@16678
   586
structure Termtab = TableFun(type key = term val ord = fast_term_ord);
wenzelm@16678
   587
wenzelm@16678
   588
end;
wenzelm@16537
   589
wenzelm@16537
   590
wenzelm@16537
   591
(* term_ord *)
wenzelm@16537
   592
wenzelm@16537
   593
(*a linear well-founded AC-compatible ordering for terms:
wenzelm@16537
   594
  s < t <=> 1. size(s) < size(t) or
wenzelm@16537
   595
            2. size(s) = size(t) and s=f(...) and t=g(...) and f<g or
wenzelm@16537
   596
            3. size(s) = size(t) and s=f(s1..sn) and t=f(t1..tn) and
wenzelm@16537
   597
               (s1..sn) < (t1..tn) (lexicographically)*)
wenzelm@16678
   598
wenzelm@16678
   599
fun indexname_ord ((x, i), (y, j)) =
wenzelm@16678
   600
  (case int_ord (i, j) of EQUAL => string_ord (x, y) | ord => ord);
wenzelm@16678
   601
wenzelm@16667
   602
local
wenzelm@16667
   603
wenzelm@16667
   604
fun hd_depth (t $ _, n) = hd_depth (t, n + 1)
wenzelm@16667
   605
  | hd_depth p = p;
wenzelm@16537
   606
wenzelm@16537
   607
fun dest_hd (Const (a, T)) = (((a, 0), T), 0)
wenzelm@16537
   608
  | dest_hd (Free (a, T)) = (((a, 0), T), 1)
wenzelm@16537
   609
  | dest_hd (Var v) = (v, 2)
wenzelm@16537
   610
  | dest_hd (Bound i) = ((("", i), dummyT), 3)
wenzelm@16537
   611
  | dest_hd (Abs (_, T, _)) = ((("", 0), T), 4);
wenzelm@16537
   612
wenzelm@16667
   613
in
wenzelm@16667
   614
wenzelm@16537
   615
fun term_ord tu =
wenzelm@16537
   616
  if pointer_eq tu then EQUAL
wenzelm@16537
   617
  else
wenzelm@16537
   618
    (case tu of
wenzelm@16537
   619
      (Abs (_, T, t), Abs(_, U, u)) =>
wenzelm@16537
   620
        (case term_ord (t, u) of EQUAL => typ_ord (T, U) | ord => ord)
wenzelm@16667
   621
    | (t, u) =>
wenzelm@16537
   622
        (case int_ord (size_of_term t, size_of_term u) of
wenzelm@16537
   623
          EQUAL =>
wenzelm@16943
   624
            (case prod_ord hd_ord int_ord (hd_depth (t, 0), hd_depth (u, 0)) of
wenzelm@16943
   625
              EQUAL => args_ord (t, u) | ord => ord)
wenzelm@16537
   626
        | ord => ord))
wenzelm@16537
   627
and hd_ord (f, g) =
wenzelm@16537
   628
  prod_ord (prod_ord indexname_ord typ_ord) int_ord (dest_hd f, dest_hd g)
wenzelm@16667
   629
and args_ord (f $ t, g $ u) =
wenzelm@16667
   630
      (case args_ord (f, g) of EQUAL => term_ord (t, u) | ord => ord)
wenzelm@16667
   631
  | args_ord _ = EQUAL;
wenzelm@16537
   632
wenzelm@16537
   633
fun termless tu = (term_ord tu = LESS);
wenzelm@16537
   634
wenzelm@16667
   635
end;
wenzelm@16667
   636
wenzelm@16667
   637
wenzelm@16667
   638
(** Lexicographic path order on terms **)
wenzelm@16667
   639
wenzelm@16667
   640
(*
nipkow@16570
   641
  See Baader & Nipkow, Term rewriting, CUP 1998.
nipkow@16570
   642
  Without variables.  Const, Var, Bound, Free and Abs are treated all as
nipkow@16570
   643
  constants.
nipkow@16570
   644
nipkow@16570
   645
  f_ord maps strings to integers and serves two purposes:
nipkow@16570
   646
  - Predicate on constant symbols.  Those that are not recognised by f_ord
nipkow@16570
   647
    must be mapped to ~1.
nipkow@16570
   648
  - Order on the recognised symbols.  These must be mapped to distinct
nipkow@16570
   649
    integers >= 0.
nipkow@16570
   650
wenzelm@16667
   651
*)
nipkow@16570
   652
nipkow@16570
   653
local
wenzelm@16667
   654
fun dest_hd f_ord (Const (a, T)) =
nipkow@16570
   655
      let val ord = f_ord a in
nipkow@16570
   656
        if ord = ~1 then ((1, ((a, 0), T)), 0) else ((0, (("", ord), T)), 0)
nipkow@16570
   657
      end
nipkow@16570
   658
  | dest_hd _ (Free (a, T)) = ((1, ((a, 0), T)), 0)
nipkow@16570
   659
  | dest_hd _ (Var v) = ((1, v), 1)
nipkow@16570
   660
  | dest_hd _ (Bound i) = ((1, (("", i), dummyT)), 2)
nipkow@16570
   661
  | dest_hd _ (Abs (_, T, _)) = ((1, (("", 0), T)), 3);
nipkow@16570
   662
nipkow@16570
   663
fun term_lpo f_ord (s, t) =
nipkow@16570
   664
  let val (f, ss) = strip_comb s and (g, ts) = strip_comb t in
nipkow@16570
   665
    if forall (fn si => term_lpo f_ord (si, t) = LESS) ss
nipkow@16570
   666
    then case hd_ord f_ord (f, g) of
wenzelm@16667
   667
        GREATER =>
wenzelm@16667
   668
          if forall (fn ti => term_lpo f_ord (s, ti) = GREATER) ts
wenzelm@16667
   669
          then GREATER else LESS
nipkow@16570
   670
      | EQUAL =>
wenzelm@16667
   671
          if forall (fn ti => term_lpo f_ord (s, ti) = GREATER) ts
wenzelm@16667
   672
          then list_ord (term_lpo f_ord) (ss, ts)
wenzelm@16667
   673
          else LESS
nipkow@16570
   674
      | LESS => LESS
nipkow@16570
   675
    else GREATER
nipkow@16570
   676
  end
nipkow@16570
   677
and hd_ord f_ord (f, g) = case (f, g) of
nipkow@16570
   678
    (Abs (_, T, t), Abs (_, U, u)) =>
nipkow@16570
   679
      (case term_lpo f_ord (t, u) of EQUAL => typ_ord (T, U) | ord => ord)
nipkow@16570
   680
  | (_, _) => prod_ord (prod_ord int_ord
nipkow@16570
   681
                  (prod_ord indexname_ord typ_ord)) int_ord
nipkow@16570
   682
                (dest_hd f_ord f, dest_hd f_ord g)
nipkow@16570
   683
in
nipkow@16570
   684
val term_lpo = term_lpo
nipkow@16570
   685
end;
nipkow@16570
   686
wenzelm@16537
   687
clasohm@0
   688
(** Connectives of higher order logic **)
clasohm@0
   689
wenzelm@375
   690
fun itselfT ty = Type ("itself", [ty]);
wenzelm@14854
   691
val a_itselfT = itselfT (TFree ("'a", []));
wenzelm@375
   692
clasohm@0
   693
val propT : typ = Type("prop",[]);
clasohm@0
   694
clasohm@0
   695
val implies = Const("==>", propT-->propT-->propT);
clasohm@0
   696
clasohm@0
   697
fun all T = Const("all", (T-->propT)-->propT);
clasohm@0
   698
clasohm@0
   699
fun equals T = Const("==", T-->T-->propT);
clasohm@0
   700
clasohm@0
   701
(* maps  !!x1...xn. t   to   t  *)
wenzelm@13000
   702
fun strip_all_body (Const("all",_)$Abs(_,_,t))  =  strip_all_body t
clasohm@0
   703
  | strip_all_body t  =  t;
clasohm@0
   704
clasohm@0
   705
(* maps  !!x1...xn. t   to   [x1, ..., xn]  *)
clasohm@0
   706
fun strip_all_vars (Const("all",_)$Abs(a,T,t))  =
wenzelm@13000
   707
                (a,T) :: strip_all_vars t
clasohm@0
   708
  | strip_all_vars t  =  [] : (string*typ) list;
clasohm@0
   709
clasohm@0
   710
(*increments a term's non-local bound variables
clasohm@0
   711
  required when moving a term within abstractions
clasohm@0
   712
     inc is  increment for bound variables
clasohm@0
   713
     lev is  level at which a bound variable is considered 'loose'*)
wenzelm@13000
   714
fun incr_bv (inc, lev, u as Bound i) = if i>=lev then Bound(i+inc) else u
clasohm@0
   715
  | incr_bv (inc, lev, Abs(a,T,body)) =
wenzelm@9536
   716
        Abs(a, T, incr_bv(inc,lev+1,body))
wenzelm@13000
   717
  | incr_bv (inc, lev, f$t) =
clasohm@0
   718
      incr_bv(inc,lev,f) $ incr_bv(inc,lev,t)
clasohm@0
   719
  | incr_bv (inc, lev, u) = u;
clasohm@0
   720
clasohm@0
   721
fun incr_boundvars  0  t = t
clasohm@0
   722
  | incr_boundvars inc t = incr_bv(inc,0,t);
clasohm@0
   723
wenzelm@12981
   724
(*Scan a pair of terms; while they are similar,
wenzelm@12981
   725
  accumulate corresponding bound vars in "al"*)
wenzelm@12981
   726
fun match_bvs(Abs(x,_,s),Abs(y,_,t), al) =
wenzelm@12981
   727
      match_bvs(s, t, if x="" orelse y="" then al
wenzelm@12981
   728
                                          else (x,y)::al)
wenzelm@12981
   729
  | match_bvs(f$s, g$t, al) = match_bvs(f,g,match_bvs(s,t,al))
wenzelm@12981
   730
  | match_bvs(_,_,al) = al;
wenzelm@12981
   731
wenzelm@12981
   732
(* strip abstractions created by parameters *)
wenzelm@12981
   733
fun match_bvars((s,t),al) = match_bvs(strip_abs_body s, strip_abs_body t, al);
wenzelm@12981
   734
wenzelm@12981
   735
fun rename_abs pat obj t =
wenzelm@12981
   736
  let
wenzelm@12981
   737
    val ren = match_bvs (pat, obj, []);
wenzelm@12981
   738
    fun ren_abs (Abs (x, T, b)) =
haftmann@17271
   739
          Abs (if_none (AList.lookup (op =) ren x) x, T, ren_abs b)
wenzelm@12981
   740
      | ren_abs (f $ t) = ren_abs f $ ren_abs t
wenzelm@12981
   741
      | ren_abs t = t
skalberg@15531
   742
  in if null ren then NONE else SOME (ren_abs t) end;
clasohm@0
   743
clasohm@0
   744
(*Accumulate all 'loose' bound vars referring to level 'lev' or beyond.
clasohm@0
   745
   (Bound 0) is loose at level 0 *)
wenzelm@13000
   746
fun add_loose_bnos (Bound i, lev, js) =
wenzelm@9536
   747
        if i<lev then js  else  (i-lev) ins_int js
clasohm@0
   748
  | add_loose_bnos (Abs (_,_,t), lev, js) = add_loose_bnos (t, lev+1, js)
clasohm@0
   749
  | add_loose_bnos (f$t, lev, js) =
wenzelm@13000
   750
        add_loose_bnos (f, lev, add_loose_bnos (t, lev, js))
clasohm@0
   751
  | add_loose_bnos (_, _, js) = js;
clasohm@0
   752
clasohm@0
   753
fun loose_bnos t = add_loose_bnos (t, 0, []);
clasohm@0
   754
clasohm@0
   755
(* loose_bvar(t,k) iff t contains a 'loose' bound variable referring to
clasohm@0
   756
   level k or beyond. *)
clasohm@0
   757
fun loose_bvar(Bound i,k) = i >= k
clasohm@0
   758
  | loose_bvar(f$t, k) = loose_bvar(f,k) orelse loose_bvar(t,k)
clasohm@0
   759
  | loose_bvar(Abs(_,_,t),k) = loose_bvar(t,k+1)
clasohm@0
   760
  | loose_bvar _ = false;
clasohm@0
   761
nipkow@2792
   762
fun loose_bvar1(Bound i,k) = i = k
nipkow@2792
   763
  | loose_bvar1(f$t, k) = loose_bvar1(f,k) orelse loose_bvar1(t,k)
nipkow@2792
   764
  | loose_bvar1(Abs(_,_,t),k) = loose_bvar1(t,k+1)
nipkow@2792
   765
  | loose_bvar1 _ = false;
clasohm@0
   766
clasohm@0
   767
(*Substitute arguments for loose bound variables.
clasohm@0
   768
  Beta-reduction of arg(n-1)...arg0 into t replacing (Bound i) with (argi).
wenzelm@4626
   769
  Note that for ((%x y. c) a b), the bound vars in c are x=1 and y=0
wenzelm@9536
   770
        and the appropriate call is  subst_bounds([b,a], c) .
clasohm@0
   771
  Loose bound variables >=n are reduced by "n" to
clasohm@0
   772
     compensate for the disappearance of lambdas.
clasohm@0
   773
*)
wenzelm@13000
   774
fun subst_bounds (args: term list, t) : term =
clasohm@0
   775
  let val n = length args;
clasohm@0
   776
      fun subst (t as Bound i, lev) =
wenzelm@9536
   777
           (if i<lev then  t    (*var is locally bound*)
wenzelm@9536
   778
            else  incr_boundvars lev (List.nth(args, i-lev))
wenzelm@9536
   779
                    handle Subscript => Bound(i-n)  (*loose: change it*))
wenzelm@9536
   780
        | subst (Abs(a,T,body), lev) = Abs(a, T,  subst(body,lev+1))
wenzelm@9536
   781
        | subst (f$t, lev) =  subst(f,lev)  $  subst(t,lev)
wenzelm@9536
   782
        | subst (t,lev) = t
clasohm@0
   783
  in   case args of [] => t  | _ => subst (t,0)  end;
clasohm@0
   784
paulson@2192
   785
(*Special case: one argument*)
wenzelm@13000
   786
fun subst_bound (arg, t) : term =
paulson@2192
   787
  let fun subst (t as Bound i, lev) =
wenzelm@9536
   788
            if i<lev then  t    (*var is locally bound*)
wenzelm@9536
   789
            else  if i=lev then incr_boundvars lev arg
wenzelm@9536
   790
                           else Bound(i-1)  (*loose: change it*)
wenzelm@9536
   791
        | subst (Abs(a,T,body), lev) = Abs(a, T,  subst(body,lev+1))
wenzelm@9536
   792
        | subst (f$t, lev) =  subst(f,lev)  $  subst(t,lev)
wenzelm@9536
   793
        | subst (t,lev) = t
paulson@2192
   794
  in  subst (t,0)  end;
paulson@2192
   795
clasohm@0
   796
(*beta-reduce if possible, else form application*)
paulson@2192
   797
fun betapply (Abs(_,_,t), u) = subst_bound (u,t)
clasohm@0
   798
  | betapply (f,u) = f$u;
clasohm@0
   799
wenzelm@14786
   800
wenzelm@16882
   801
(** Specialized equality, membership, insertion etc. **)
paulson@2192
   802
wenzelm@16882
   803
(* indexnames *)
wenzelm@16882
   804
wenzelm@16724
   805
fun eq_ix ((x, i): indexname, (y, j)) = i = j andalso x = y;
paulson@2192
   806
wenzelm@2959
   807
fun mem_ix (_, []) = false
wenzelm@16882
   808
  | mem_ix (x, y :: ys) = eq_ix (x, y) orelse mem_ix (x, ys);
paulson@2192
   809
wenzelm@16882
   810
fun ins_ix (x, xs) = if mem_ix (x, xs) then xs else x :: xs;
paulson@2192
   811
clasohm@0
   812
wenzelm@16882
   813
(* variables *)
wenzelm@16882
   814
wenzelm@16943
   815
fun eq_tvar ((xi, S: sort), (xi', S')) = eq_ix (xi, xi') andalso S = S';
wenzelm@16943
   816
fun eq_var ((xi, T: typ), (xi', T')) = eq_ix (xi, xi') andalso T = T';
wenzelm@16943
   817
wenzelm@16943
   818
val tvar_ord = prod_ord indexname_ord sort_ord;
wenzelm@16943
   819
val var_ord = prod_ord indexname_ord typ_ord;
wenzelm@16882
   820
wenzelm@16882
   821
wenzelm@16882
   822
(* terms *)
paulson@2176
   823
paulson@2176
   824
fun mem_term (_, []) = false
wenzelm@16882
   825
  | mem_term (t, t'::ts) = t aconv t' orelse mem_term (t, ts);
paulson@2176
   826
paulson@2182
   827
fun subset_term ([], ys) = true
paulson@2182
   828
  | subset_term (x :: xs, ys) = mem_term (x, ys) andalso subset_term(xs, ys);
paulson@2182
   829
paulson@2182
   830
fun eq_set_term (xs, ys) =
paulson@2182
   831
    xs = ys orelse (subset_term (xs, ys) andalso subset_term (ys, xs));
paulson@2182
   832
paulson@2176
   833
fun ins_term(t,ts) = if mem_term(t,ts) then ts else t :: ts;
paulson@2176
   834
paulson@2176
   835
fun union_term (xs, []) = xs
paulson@2176
   836
  | union_term ([], ys) = ys
paulson@2176
   837
  | union_term ((x :: xs), ys) = union_term (xs, ins_term (x, ys));
paulson@2176
   838
paulson@5585
   839
fun inter_term ([], ys) = []
paulson@5585
   840
  | inter_term (x::xs, ys) =
paulson@5585
   841
      if mem_term (x,ys) then x :: inter_term(xs,ys) else inter_term(xs,ys);
paulson@5585
   842
wenzelm@16882
   843
wenzelm@13000
   844
(*A fast unification filter: true unless the two terms cannot be unified.
clasohm@0
   845
  Terms must be NORMAL.  Treats all Vars as distinct. *)
clasohm@0
   846
fun could_unify (t,u) =
clasohm@0
   847
  let fun matchrands (f$t, g$u) = could_unify(t,u) andalso  matchrands(f,g)
wenzelm@9536
   848
        | matchrands _ = true
clasohm@0
   849
  in case (head_of t , head_of u) of
wenzelm@9536
   850
        (_, Var _) => true
clasohm@0
   851
      | (Var _, _) => true
clasohm@0
   852
      | (Const(a,_), Const(b,_)) =>  a=b andalso matchrands(t,u)
clasohm@0
   853
      | (Free(a,_), Free(b,_)) =>  a=b andalso matchrands(t,u)
clasohm@0
   854
      | (Bound i, Bound j) =>  i=j andalso matchrands(t,u)
clasohm@0
   855
      | (Abs _, _) =>  true   (*because of possible eta equality*)
clasohm@0
   856
      | (_, Abs _) =>  true
clasohm@0
   857
      | _ => false
clasohm@0
   858
  end;
clasohm@0
   859
clasohm@0
   860
(*Substitute new for free occurrences of old in a term*)
clasohm@0
   861
fun subst_free [] = (fn t=>t)
clasohm@0
   862
  | subst_free pairs =
wenzelm@13000
   863
      let fun substf u =
haftmann@17314
   864
            case AList.lookup (op aconv) pairs u of
skalberg@15531
   865
                SOME u' => u'
skalberg@15531
   866
              | NONE => (case u of Abs(a,T,t) => Abs(a, T, substf t)
wenzelm@9536
   867
                                 | t$u' => substf t $ substf u'
wenzelm@9536
   868
                                 | _ => u)
clasohm@0
   869
      in  substf  end;
clasohm@0
   870
clasohm@0
   871
(*a total, irreflexive ordering on index names*)
clasohm@0
   872
fun xless ((a,i), (b,j): indexname) = i<j  orelse  (i=j andalso a<b);
clasohm@0
   873
clasohm@0
   874
wenzelm@13000
   875
(*Abstraction of the term "body" over its occurrences of v,
clasohm@0
   876
    which must contain no loose bound variables.
clasohm@0
   877
  The resulting term is ready to become the body of an Abs.*)
wenzelm@16882
   878
fun abstract_over (v, body) =
wenzelm@16882
   879
  let
wenzelm@16990
   880
    exception SAME;
wenzelm@16990
   881
    fun abs lev tm =
wenzelm@16990
   882
      if v aconv tm then Bound lev
wenzelm@16882
   883
      else
wenzelm@16990
   884
        (case tm of
wenzelm@16990
   885
          Abs (a, T, t) => Abs (a, T, abs (lev + 1) t)
wenzelm@16990
   886
        | t $ u => (abs lev t $ (abs lev u handle SAME => u) handle SAME => t $ abs lev u)
wenzelm@16990
   887
        | _ => raise SAME);
wenzelm@16990
   888
  in abs 0 body handle SAME => body end;
clasohm@0
   889
berghofe@13665
   890
fun lambda (v as Free (x, T)) t = Abs (x, T, abstract_over (v, t))
berghofe@13665
   891
  | lambda (v as Var ((x, _), T)) t = Abs (x, T, abstract_over (v, t))
berghofe@13665
   892
  | lambda v t = raise TERM ("lambda", [v, t]);
clasohm@0
   893
clasohm@0
   894
(*Form an abstraction over a free variable.*)
clasohm@0
   895
fun absfree (a,T,body) = Abs(a, T, abstract_over (Free(a,T), body));
clasohm@0
   896
clasohm@0
   897
(*Abstraction over a list of free variables*)
clasohm@0
   898
fun list_abs_free ([ ] ,     t) = t
wenzelm@13000
   899
  | list_abs_free ((a,T)::vars, t) =
clasohm@0
   900
      absfree(a, T, list_abs_free(vars,t));
clasohm@0
   901
clasohm@0
   902
(*Quantification over a list of free variables*)
clasohm@0
   903
fun list_all_free ([], t: term) = t
wenzelm@13000
   904
  | list_all_free ((a,T)::vars, t) =
clasohm@0
   905
        (all T) $ (absfree(a, T, list_all_free(vars,t)));
clasohm@0
   906
clasohm@0
   907
(*Quantification over a list of variables (already bound in body) *)
clasohm@0
   908
fun list_all ([], t) = t
wenzelm@13000
   909
  | list_all ((a,T)::vars, t) =
clasohm@0
   910
        (all T) $ (Abs(a, T, list_all(vars,t)));
clasohm@0
   911
wenzelm@16678
   912
(*Replace the ATOMIC term ti by ui;    inst = [(t1,u1), ..., (tn,un)].
clasohm@0
   913
  A simultaneous substitution:  [ (a,b), (b,a) ] swaps a and b.  *)
wenzelm@16678
   914
fun subst_atomic [] tm = tm
wenzelm@16678
   915
  | subst_atomic inst tm =
wenzelm@16678
   916
      let
wenzelm@16678
   917
        fun subst (Abs (a, T, body)) = Abs (a, T, subst body)
wenzelm@16678
   918
          | subst (t $ u) = subst t $ subst u
haftmann@17314
   919
          | subst t = if_none (AList.lookup (op aconv) inst t) t;
wenzelm@16678
   920
      in subst tm end;
clasohm@0
   921
wenzelm@16678
   922
(*Replace the ATOMIC type Ti by Ui;    inst = [(T1,U1), ..., (Tn,Un)].*)
wenzelm@16678
   923
fun typ_subst_atomic [] ty = ty
wenzelm@16678
   924
  | typ_subst_atomic inst ty =
wenzelm@16678
   925
      let
wenzelm@16678
   926
        fun subst (Type (a, Ts)) = Type (a, map subst Ts)
haftmann@17314
   927
          | subst T = if_none (AList.lookup (op = : typ * typ -> bool) inst T) T;
wenzelm@16678
   928
      in subst ty end;
berghofe@15797
   929
wenzelm@16678
   930
fun subst_atomic_types [] tm = tm
wenzelm@16678
   931
  | subst_atomic_types inst tm = map_term_types (typ_subst_atomic inst) tm;
wenzelm@16678
   932
wenzelm@16678
   933
fun typ_subst_TVars [] ty = ty
wenzelm@16678
   934
  | typ_subst_TVars inst ty =
wenzelm@16678
   935
      let
wenzelm@16678
   936
        fun subst (Type (a, Ts)) = Type (a, map subst Ts)
haftmann@17271
   937
          | subst (T as TVar (xi, _)) = if_none (AList.lookup (op =) inst xi) T
wenzelm@16678
   938
          | subst T = T;
wenzelm@16678
   939
      in subst ty end;
clasohm@0
   940
wenzelm@16678
   941
fun subst_TVars [] tm = tm
wenzelm@16678
   942
  | subst_TVars inst tm = map_term_types (typ_subst_TVars inst) tm;
clasohm@0
   943
wenzelm@16678
   944
(*see also Envir.norm_term*)
wenzelm@16678
   945
fun subst_Vars [] tm = tm
wenzelm@16678
   946
  | subst_Vars inst tm =
wenzelm@16678
   947
      let
haftmann@17271
   948
        fun subst (t as Var (xi, _)) = if_none (AList.lookup (op =) inst xi) t
wenzelm@16678
   949
          | subst (Abs (a, T, t)) = Abs (a, T, subst t)
wenzelm@16678
   950
          | subst (t $ u) = subst t $ subst u
wenzelm@16678
   951
          | subst t = t;
wenzelm@16678
   952
      in subst tm end;
clasohm@0
   953
wenzelm@16678
   954
(*see also Envir.norm_term*)
wenzelm@16678
   955
fun subst_vars ([], []) tm = tm
wenzelm@16678
   956
  | subst_vars ([], inst) tm = subst_Vars inst tm
wenzelm@16678
   957
  | subst_vars (instT, inst) tm =
wenzelm@16678
   958
      let
wenzelm@16678
   959
        fun subst (Const (a, T)) = Const (a, typ_subst_TVars instT T)
wenzelm@16678
   960
          | subst (Free (a, T)) = Free (a, typ_subst_TVars instT T)
wenzelm@16678
   961
          | subst (t as Var (xi, T)) =
haftmann@17271
   962
              (case AList.lookup (op =) inst xi of
wenzelm@16678
   963
                NONE => Var (xi, typ_subst_TVars instT T)
wenzelm@16678
   964
              | SOME t => t)
wenzelm@16678
   965
          | subst (t as Bound _) = t
wenzelm@16678
   966
          | subst (Abs (a, T, t)) = Abs (a, typ_subst_TVars instT T, subst t)
wenzelm@16678
   967
          | subst (t $ u) = subst t $ subst u;
wenzelm@16678
   968
      in subst tm end;
clasohm@0
   969
clasohm@0
   970
wenzelm@16943
   971
(* instantiation of schematic variables (types before terms) *)
wenzelm@16882
   972
wenzelm@16882
   973
local exception SAME in
wenzelm@16882
   974
wenzelm@16943
   975
fun instantiateT_same [] _ = raise SAME
wenzelm@16943
   976
  | instantiateT_same instT ty =
wenzelm@16882
   977
      let
wenzelm@16882
   978
        fun subst_typ (Type (a, Ts)) = Type (a, subst_typs Ts)
wenzelm@16882
   979
          | subst_typ (TVar v) =
haftmann@17314
   980
              (case AList.lookup eq_tvar instT v of
wenzelm@16882
   981
                SOME T => T
wenzelm@16882
   982
              | NONE => raise SAME)
wenzelm@16882
   983
          | subst_typ _ = raise SAME
wenzelm@16882
   984
        and subst_typs (T :: Ts) =
wenzelm@16882
   985
            (subst_typ T :: (subst_typs Ts handle SAME => Ts)
wenzelm@16882
   986
              handle SAME => T :: subst_typs Ts)
wenzelm@16882
   987
          | subst_typs [] = raise SAME;
wenzelm@16882
   988
      in subst_typ ty end;
wenzelm@16882
   989
wenzelm@16882
   990
fun instantiate ([], []) tm = tm
wenzelm@16882
   991
  | instantiate (instT, inst) tm =
wenzelm@16882
   992
      let
wenzelm@16943
   993
        val substT = instantiateT_same instT;
wenzelm@16943
   994
        fun subst (Const (c, T)) = Const (c, substT T)
wenzelm@16943
   995
          | subst (Free (x, T)) = Free (x, substT T)
wenzelm@16882
   996
          | subst (Var (xi, T)) =
wenzelm@16943
   997
              let val (T', same) = (substT T, false) handle SAME => (T, true) in
haftmann@17314
   998
                (case AList.lookup eq_var inst (xi, T') of
wenzelm@16882
   999
                   SOME t => t
wenzelm@16882
  1000
                 | NONE => if same then raise SAME else Var (xi, T'))
wenzelm@16882
  1001
              end
wenzelm@16882
  1002
          | subst (Bound _) = raise SAME
wenzelm@16882
  1003
          | subst (Abs (x, T, t)) =
wenzelm@16943
  1004
              (Abs (x, substT T, subst t handle SAME => t)
wenzelm@16882
  1005
                handle SAME => Abs (x, T, subst t))
wenzelm@16882
  1006
          | subst (t $ u) = (subst t $ (subst u handle SAME => u) handle SAME => t $ subst u);
wenzelm@16882
  1007
      in subst tm handle SAME => tm end;
wenzelm@16882
  1008
wenzelm@16882
  1009
fun instantiateT instT ty =
wenzelm@16943
  1010
  instantiateT_same instT ty handle SAME => ty;
wenzelm@16882
  1011
wenzelm@16882
  1012
end;
wenzelm@16882
  1013
wenzelm@16882
  1014
paulson@15573
  1015
(** Identifying first-order terms **)
paulson@15573
  1016
paulson@15573
  1017
(*Argument Ts is a reverse list of binder types, needed if term t contains Bound vars*)
paulson@15573
  1018
fun has_not_funtype Ts t = not (is_funtype (fastype_of1 (Ts,t)));
paulson@15573
  1019
wenzelm@16537
  1020
(*First order means in all terms of the form f(t1,...,tn) no argument has a
paulson@16589
  1021
  function type. The supplied quantifiers are excluded: their argument always
paulson@16589
  1022
  has a function type through a recursive call into its body.*)
wenzelm@16667
  1023
fun is_first_order quants =
paulson@16589
  1024
  let fun first_order1 Ts (Abs (_,T,body)) = first_order1 (T::Ts) body
wenzelm@16667
  1025
        | first_order1 Ts (Const(q,_) $ Abs(a,T,body)) =
wenzelm@16667
  1026
            q mem_string quants  andalso   (*it is a known quantifier*)
paulson@16589
  1027
            not (is_funtype T)   andalso first_order1 (T::Ts) body
wenzelm@16667
  1028
        | first_order1 Ts t =
wenzelm@16667
  1029
            case strip_comb t of
wenzelm@16667
  1030
                 (Var _, ts) => forall (first_order1 Ts andf has_not_funtype Ts) ts
wenzelm@16667
  1031
               | (Free _, ts) => forall (first_order1 Ts andf has_not_funtype Ts) ts
wenzelm@16667
  1032
               | (Const _, ts) => forall (first_order1 Ts andf has_not_funtype Ts) ts
wenzelm@16667
  1033
               | (Bound _, ts) => forall (first_order1 Ts andf has_not_funtype Ts) ts
wenzelm@16667
  1034
               | (Abs _, ts) => false (*not in beta-normal form*)
wenzelm@16667
  1035
               | _ => error "first_order: unexpected case"
paulson@16589
  1036
    in  first_order1 []  end;
paulson@15573
  1037
wenzelm@16710
  1038
wenzelm@16990
  1039
(* maximum index of typs and terms *)
clasohm@0
  1040
wenzelm@16710
  1041
fun maxidx_typ (TVar ((_, j), _)) i = Int.max (i, j)
wenzelm@16710
  1042
  | maxidx_typ (Type (_, Ts)) i = maxidx_typs Ts i
wenzelm@16710
  1043
  | maxidx_typ (TFree _) i = i
wenzelm@16710
  1044
and maxidx_typs [] i = i
wenzelm@16710
  1045
  | maxidx_typs (T :: Ts) i = maxidx_typs Ts (maxidx_typ T i);
clasohm@0
  1046
wenzelm@16710
  1047
fun maxidx_term (Var ((_, j), T)) i = maxidx_typ T (Int.max (i, j))
wenzelm@16710
  1048
  | maxidx_term (Const (_, T)) i = maxidx_typ T i
wenzelm@16710
  1049
  | maxidx_term (Free (_, T)) i = maxidx_typ T i
wenzelm@16710
  1050
  | maxidx_term (Bound _) i = i
wenzelm@16710
  1051
  | maxidx_term (Abs (_, T, t)) i = maxidx_term t (maxidx_typ T i)
wenzelm@16710
  1052
  | maxidx_term (t $ u) i = maxidx_term u (maxidx_term t i);
clasohm@0
  1053
wenzelm@16710
  1054
fun maxidx_of_typ T = maxidx_typ T ~1;
wenzelm@16710
  1055
fun maxidx_of_typs Ts = maxidx_typs Ts ~1;
wenzelm@16710
  1056
fun maxidx_of_term t = maxidx_term t ~1;
berghofe@13665
  1057
clasohm@0
  1058
clasohm@0
  1059
clasohm@0
  1060
(**** Syntax-related declarations ****)
clasohm@0
  1061
clasohm@0
  1062
(*** Printing ***)
clasohm@0
  1063
wenzelm@16943
  1064
(*Makes a variant of a name distinct from the names in 'used'.
wenzelm@14676
  1065
  First attaches the suffix and then increments this;
wenzelm@12306
  1066
  preserves a suffix of underscores "_". *)
wenzelm@16943
  1067
fun variant used name =
wenzelm@12306
  1068
  let
wenzelm@12306
  1069
    val (c, u) = pairself implode (Library.take_suffix (equal "_") (Symbol.explode name));
wenzelm@16943
  1070
    fun vary2 c = if ((c ^ u) mem_string used) then vary2 (Symbol.bump_string c) else c;
wenzelm@16943
  1071
    fun vary1 c = if ((c ^ u) mem_string used) then vary2 (Symbol.bump_init c) else c;
wenzelm@12306
  1072
  in vary1 (if c = "" then "u" else c) ^ u end;
clasohm@0
  1073
clasohm@0
  1074
(*Create variants of the list of names, with priority to the first ones*)
clasohm@0
  1075
fun variantlist ([], used) = []
wenzelm@13000
  1076
  | variantlist(b::bs, used) =
clasohm@0
  1077
      let val b' = variant used b
clasohm@0
  1078
      in  b' :: variantlist (bs, b'::used)  end;
clasohm@0
  1079
wenzelm@14695
  1080
(*Invent fresh names*)
wenzelm@14695
  1081
fun invent_names _ _ 0 = []
wenzelm@14695
  1082
  | invent_names used a n =
wenzelm@14695
  1083
      let val b = Symbol.bump_string a in
wenzelm@14695
  1084
        if a mem_string used then invent_names used b n
wenzelm@14695
  1085
        else a :: invent_names used b (n - 1)
wenzelm@14695
  1086
      end;
wenzelm@11353
  1087
wenzelm@16537
  1088
wenzelm@4017
  1089
(** Consts etc. **)
wenzelm@4017
  1090
skalberg@15574
  1091
fun add_typ_classes (Type (_, Ts), cs) = foldr add_typ_classes cs Ts
wenzelm@4017
  1092
  | add_typ_classes (TFree (_, S), cs) = S union cs
wenzelm@4017
  1093
  | add_typ_classes (TVar (_, S), cs) = S union cs;
wenzelm@4017
  1094
wenzelm@16294
  1095
fun add_typ_tycons (Type (c, Ts), cs) = foldr add_typ_tycons (c ins_string cs) Ts
wenzelm@4017
  1096
  | add_typ_tycons (_, cs) = cs;
wenzelm@4017
  1097
wenzelm@4017
  1098
val add_term_classes = it_term_types add_typ_classes;
wenzelm@4017
  1099
val add_term_tycons = it_term_types add_typ_tycons;
wenzelm@4017
  1100
wenzelm@9319
  1101
fun add_term_consts (Const (c, _), cs) = c ins_string cs
wenzelm@4017
  1102
  | add_term_consts (t $ u, cs) = add_term_consts (t, add_term_consts (u, cs))
wenzelm@4017
  1103
  | add_term_consts (Abs (_, _, t), cs) = add_term_consts (t, cs)
wenzelm@4017
  1104
  | add_term_consts (_, cs) = cs;
wenzelm@4017
  1105
nipkow@13646
  1106
fun term_consts t = add_term_consts(t,[]);
nipkow@13646
  1107
wenzelm@16943
  1108
fun exists_subterm P =
wenzelm@16943
  1109
  let
wenzelm@16943
  1110
    fun ex tm = P tm orelse
wenzelm@16943
  1111
      (case tm of
wenzelm@16943
  1112
        t $ u => ex t orelse ex u
wenzelm@16943
  1113
      | Abs (_, _, t) => ex t
wenzelm@16943
  1114
      | _ => false);
wenzelm@16943
  1115
  in ex end;
obua@16108
  1116
wenzelm@16943
  1117
fun exists_Const P = exists_subterm (fn Const c => P c | _ => false);
nipkow@4631
  1118
wenzelm@4017
  1119
(*map classes, tycons*)
wenzelm@4017
  1120
fun map_typ f g (Type (c, Ts)) = Type (g c, map (map_typ f g) Ts)
wenzelm@4017
  1121
  | map_typ f _ (TFree (x, S)) = TFree (x, map f S)
wenzelm@4017
  1122
  | map_typ f _ (TVar (xi, S)) = TVar (xi, map f S);
wenzelm@4017
  1123
wenzelm@4017
  1124
(*map classes, tycons, consts*)
wenzelm@4017
  1125
fun map_term f g h (Const (c, T)) = Const (h c, map_typ f g T)
wenzelm@4017
  1126
  | map_term f g _ (Free (x, T)) = Free (x, map_typ f g T)
wenzelm@4017
  1127
  | map_term f g _ (Var (xi, T)) = Var (xi, map_typ f g T)
wenzelm@4017
  1128
  | map_term _ _ _ (t as Bound _) = t
wenzelm@4017
  1129
  | map_term f g h (Abs (x, T, t)) = Abs (x, map_typ f g T, map_term f g h t)
wenzelm@4017
  1130
  | map_term f g h (t $ u) = map_term f g h t $ map_term f g h u;
wenzelm@4017
  1131
wenzelm@4017
  1132
clasohm@0
  1133
(** TFrees and TVars **)
clasohm@0
  1134
wenzelm@12802
  1135
(*Accumulates the names of Frees in the term, suppressing duplicates.*)
wenzelm@12802
  1136
fun add_term_free_names (Free(a,_), bs) = a ins_string bs
wenzelm@12802
  1137
  | add_term_free_names (f$u, bs) = add_term_free_names (f, add_term_free_names(u, bs))
wenzelm@12802
  1138
  | add_term_free_names (Abs(_,_,t), bs) = add_term_free_names(t,bs)
wenzelm@12802
  1139
  | add_term_free_names (_, bs) = bs;
wenzelm@12802
  1140
clasohm@0
  1141
(*Accumulates the names in the term, suppressing duplicates.
clasohm@0
  1142
  Includes Frees and Consts.  For choosing unambiguous bound var names.*)
wenzelm@10666
  1143
fun add_term_names (Const(a,_), bs) = NameSpace.base a ins_string bs
paulson@2176
  1144
  | add_term_names (Free(a,_), bs) = a ins_string bs
clasohm@0
  1145
  | add_term_names (f$u, bs) = add_term_names (f, add_term_names(u, bs))
clasohm@0
  1146
  | add_term_names (Abs(_,_,t), bs) = add_term_names(t,bs)
clasohm@0
  1147
  | add_term_names (_, bs) = bs;
clasohm@0
  1148
clasohm@0
  1149
(*Accumulates the TVars in a type, suppressing duplicates. *)
skalberg@15574
  1150
fun add_typ_tvars(Type(_,Ts),vs) = foldr add_typ_tvars vs Ts
clasohm@0
  1151
  | add_typ_tvars(TFree(_),vs) = vs
wenzelm@16294
  1152
  | add_typ_tvars(TVar(v),vs) = insert (op =) v vs;
clasohm@0
  1153
clasohm@0
  1154
(*Accumulates the TFrees in a type, suppressing duplicates. *)
skalberg@15574
  1155
fun add_typ_tfree_names(Type(_,Ts),fs) = foldr add_typ_tfree_names fs Ts
paulson@2176
  1156
  | add_typ_tfree_names(TFree(f,_),fs) = f ins_string fs
clasohm@0
  1157
  | add_typ_tfree_names(TVar(_),fs) = fs;
clasohm@0
  1158
skalberg@15574
  1159
fun add_typ_tfrees(Type(_,Ts),fs) = foldr add_typ_tfrees fs Ts
wenzelm@16294
  1160
  | add_typ_tfrees(TFree(f),fs) = insert (op =) f fs
clasohm@0
  1161
  | add_typ_tfrees(TVar(_),fs) = fs;
clasohm@0
  1162
skalberg@15574
  1163
fun add_typ_varnames(Type(_,Ts),nms) = foldr add_typ_varnames nms Ts
paulson@2176
  1164
  | add_typ_varnames(TFree(nm,_),nms) = nm ins_string nms
paulson@2176
  1165
  | add_typ_varnames(TVar((nm,_),_),nms) = nm ins_string nms;
nipkow@949
  1166
clasohm@0
  1167
(*Accumulates the TVars in a term, suppressing duplicates. *)
clasohm@0
  1168
val add_term_tvars = it_term_types add_typ_tvars;
clasohm@0
  1169
clasohm@0
  1170
(*Accumulates the TFrees in a term, suppressing duplicates. *)
clasohm@0
  1171
val add_term_tfrees = it_term_types add_typ_tfrees;
clasohm@0
  1172
val add_term_tfree_names = it_term_types add_typ_tfree_names;
clasohm@0
  1173
nipkow@949
  1174
val add_term_tvarnames = it_term_types add_typ_varnames;
nipkow@949
  1175
clasohm@0
  1176
(*Non-list versions*)
clasohm@0
  1177
fun typ_tfrees T = add_typ_tfrees(T,[]);
clasohm@0
  1178
fun typ_tvars T = add_typ_tvars(T,[]);
clasohm@0
  1179
fun term_tfrees t = add_term_tfrees(t,[]);
clasohm@0
  1180
fun term_tvars t = add_term_tvars(t,[]);
clasohm@0
  1181
nipkow@949
  1182
(*special code to enforce left-to-right collection of TVar-indexnames*)
nipkow@949
  1183
skalberg@15570
  1184
fun add_typ_ixns(ixns,Type(_,Ts)) = Library.foldl add_typ_ixns (ixns,Ts)
wenzelm@13000
  1185
  | add_typ_ixns(ixns,TVar(ixn,_)) = if mem_ix (ixn, ixns) then ixns
wenzelm@9536
  1186
                                     else ixns@[ixn]
nipkow@949
  1187
  | add_typ_ixns(ixns,TFree(_)) = ixns;
nipkow@949
  1188
nipkow@949
  1189
fun add_term_tvar_ixns(Const(_,T),ixns) = add_typ_ixns(ixns,T)
nipkow@949
  1190
  | add_term_tvar_ixns(Free(_,T),ixns) = add_typ_ixns(ixns,T)
nipkow@949
  1191
  | add_term_tvar_ixns(Var(_,T),ixns) = add_typ_ixns(ixns,T)
nipkow@949
  1192
  | add_term_tvar_ixns(Bound _,ixns) = ixns
nipkow@949
  1193
  | add_term_tvar_ixns(Abs(_,T,t),ixns) =
nipkow@949
  1194
      add_term_tvar_ixns(t,add_typ_ixns(ixns,T))
nipkow@949
  1195
  | add_term_tvar_ixns(f$t,ixns) =
nipkow@949
  1196
      add_term_tvar_ixns(t,add_term_tvar_ixns(f,ixns));
nipkow@949
  1197
wenzelm@16537
  1198
clasohm@0
  1199
(** Frees and Vars **)
clasohm@0
  1200
clasohm@0
  1201
(*Accumulates the Vars in the term, suppressing duplicates*)
clasohm@0
  1202
fun add_term_vars (t, vars: term list) = case t of
wenzelm@16990
  1203
    Var   _ => OrdList.insert term_ord t vars
clasohm@0
  1204
  | Abs (_,_,body) => add_term_vars(body,vars)
clasohm@0
  1205
  | f$t =>  add_term_vars (f, add_term_vars(t, vars))
clasohm@0
  1206
  | _ => vars;
clasohm@0
  1207
clasohm@0
  1208
fun term_vars t = add_term_vars(t,[]);
clasohm@0
  1209
clasohm@0
  1210
(*Accumulates the Frees in the term, suppressing duplicates*)
clasohm@0
  1211
fun add_term_frees (t, frees: term list) = case t of
wenzelm@16990
  1212
    Free   _ => OrdList.insert term_ord t frees
clasohm@0
  1213
  | Abs (_,_,body) => add_term_frees(body,frees)
clasohm@0
  1214
  | f$t =>  add_term_frees (f, add_term_frees(t, frees))
clasohm@0
  1215
  | _ => frees;
clasohm@0
  1216
clasohm@0
  1217
fun term_frees t = add_term_frees(t,[]);
clasohm@0
  1218
clasohm@0
  1219
(*Given an abstraction over P, replaces the bound variable by a Free variable
wenzelm@16678
  1220
  having a unique name*)
clasohm@0
  1221
fun variant_abs (a,T,P) =
clasohm@0
  1222
  let val b = variant (add_term_names(P,[])) a
paulson@2192
  1223
  in  (b,  subst_bound (Free(b,T), P))  end;
clasohm@0
  1224
wenzelm@16678
  1225
fun dest_abs (x, T, body) =
wenzelm@16678
  1226
  let
wenzelm@16678
  1227
    fun name_clash (Free (y, _)) = (x = y)
wenzelm@16678
  1228
      | name_clash (t $ u) = name_clash t orelse name_clash u
wenzelm@16678
  1229
      | name_clash (Abs (_, _, t)) = name_clash t
wenzelm@16678
  1230
      | name_clash _ = false;
wenzelm@16678
  1231
  in
wenzelm@16678
  1232
    if name_clash body then
wenzelm@16678
  1233
      dest_abs (variant [x] x, T, body)    (*potentially slow, but rarely happens*)
wenzelm@16678
  1234
    else (x, subst_bound (Free (x, T), body))
wenzelm@16678
  1235
  end;
wenzelm@16678
  1236
wenzelm@16990
  1237
(*names for numbered variables --
wenzelm@16990
  1238
  preserves order wrt. int_ord vs. string_ord, avoids allocating new strings*)
wenzelm@16990
  1239
local
wenzelm@16990
  1240
  val small_int = Vector.tabulate (1000, fn i =>
wenzelm@16990
  1241
    let val leading = if i < 10 then "00" else if i < 100 then "0" else ""
wenzelm@16990
  1242
    in ":" ^ leading ^ string_of_int i end);
wenzelm@16990
  1243
in
wenzelm@16990
  1244
  fun bound n =
wenzelm@16990
  1245
    if n < 1000 then Vector.sub (small_int, n)
wenzelm@16990
  1246
    else ":" ^ bound (n div 1000) ^ Vector.sub (small_int, n mod 1000);
wenzelm@16990
  1247
end;
wenzelm@16990
  1248
wenzelm@16990
  1249
val is_bound = String.isPrefix ":";
wenzelm@16943
  1250
clasohm@0
  1251
(* renames and reverses the strings in vars away from names *)
clasohm@0
  1252
fun rename_aTs names vars : (string*typ)list =
clasohm@0
  1253
  let fun rename_aT (vars,(a,T)) =
wenzelm@9536
  1254
                (variant (map #1 vars @ names) a, T) :: vars
skalberg@15570
  1255
  in Library.foldl rename_aT ([],vars) end;
clasohm@0
  1256
clasohm@0
  1257
fun rename_wrt_term t = rename_aTs (add_term_names(t,[]));
clasohm@1364
  1258
paulson@1417
  1259
wenzelm@16943
  1260
(* zero var indexes *)
wenzelm@4286
  1261
wenzelm@16943
  1262
fun zero_var_inst vars =
wenzelm@16943
  1263
  fold (fn v as ((x, i), X) => fn (used, inst) =>
wenzelm@16943
  1264
    let
wenzelm@17642
  1265
      val x' = variant used (if is_bound x then "u" else x);
wenzelm@16943
  1266
      val used' = x' :: used;
wenzelm@16943
  1267
    in if x = x' andalso i = 0 then (used', inst) else (used', (v, ((x', 0), X)) :: inst) end)
wenzelm@16943
  1268
  vars ([], []) |> #2;
wenzelm@4286
  1269
wenzelm@16943
  1270
fun zero_var_indexesT ty =
wenzelm@16943
  1271
  instantiateT (map (apsnd TVar) (zero_var_inst (sort tvar_ord (add_tvarsT ty [])))) ty;
haftmann@16790
  1272
wenzelm@16943
  1273
fun zero_var_indexes_inst tm =
wenzelm@16943
  1274
  let
wenzelm@16943
  1275
    val instT = map (apsnd TVar) (zero_var_inst (sort tvar_ord (fold_types add_tvarsT tm [])));
wenzelm@16943
  1276
    val inst =
wenzelm@16943
  1277
      add_vars tm [] |> map (apsnd (instantiateT instT))
wenzelm@16943
  1278
      |> sort var_ord |> zero_var_inst |> map (apsnd Var);
wenzelm@16943
  1279
  in (instT, inst) end;
wenzelm@12499
  1280
wenzelm@16943
  1281
fun zero_var_indexes tm = instantiate (zero_var_indexes_inst tm) tm;
wenzelm@4286
  1282
paulson@1417
  1283
wenzelm@9536
  1284
(* dummy patterns *)
wenzelm@9536
  1285
wenzelm@9536
  1286
val dummy_patternN = "dummy_pattern";
wenzelm@9536
  1287
wenzelm@9536
  1288
fun is_dummy_pattern (Const ("dummy_pattern", _)) = true
wenzelm@9536
  1289
  | is_dummy_pattern _ = false;
wenzelm@9536
  1290
wenzelm@9536
  1291
fun no_dummy_patterns tm =
haftmann@16787
  1292
  if not (fold_aterms (fn t => fn b => b orelse is_dummy_pattern t) tm false) then tm
wenzelm@9536
  1293
  else raise TERM ("Illegal occurrence of '_' dummy pattern", [tm]);
wenzelm@9536
  1294
wenzelm@11903
  1295
fun replace_dummy Ts (i, Const ("dummy_pattern", T)) =
wenzelm@11903
  1296
      (i + 1, list_comb (Var (("_dummy_", i), Ts ---> T), map Bound (0 upto length Ts - 1)))
wenzelm@11903
  1297
  | replace_dummy Ts (i, Abs (x, T, t)) =
wenzelm@11903
  1298
      let val (i', t') = replace_dummy (T :: Ts) (i, t)
wenzelm@11903
  1299
      in (i', Abs (x, T, t')) end
wenzelm@11903
  1300
  | replace_dummy Ts (i, t $ u) =
wenzelm@11903
  1301
      let val (i', t') = replace_dummy Ts (i, t); val (i'', u') = replace_dummy Ts (i', u)
wenzelm@11903
  1302
      in (i'', t' $ u') end
wenzelm@11903
  1303
  | replace_dummy _ (i, a) = (i, a);
wenzelm@11903
  1304
wenzelm@11903
  1305
val replace_dummy_patterns = replace_dummy [];
wenzelm@9536
  1306
wenzelm@10552
  1307
fun is_replaced_dummy_pattern ("_dummy_", _) = true
wenzelm@10552
  1308
  | is_replaced_dummy_pattern _ = false;
wenzelm@9536
  1309
wenzelm@16035
  1310
fun show_dummy_patterns (Var (("_dummy_", _), T)) = Const ("dummy_pattern", T)
wenzelm@16035
  1311
  | show_dummy_patterns (t $ u) = show_dummy_patterns t $ show_dummy_patterns u
wenzelm@16035
  1312
  | show_dummy_patterns (Abs (x, T, t)) = Abs (x, T, show_dummy_patterns t)
wenzelm@16035
  1313
  | show_dummy_patterns a = a;
wenzelm@16035
  1314
wenzelm@13484
  1315
wenzelm@13484
  1316
(* adhoc freezing *)
wenzelm@13484
  1317
wenzelm@13484
  1318
fun adhoc_freeze_vars tm =
wenzelm@13484
  1319
  let
wenzelm@13484
  1320
    fun mk_inst (var as Var ((a, i), T)) =
wenzelm@13484
  1321
      let val x = a ^ Library.gensym "_" ^ string_of_int i
wenzelm@13484
  1322
      in ((var,  Free(x, T)), x) end;
wenzelm@13484
  1323
    val (insts, xs) = split_list (map mk_inst (term_vars tm));
wenzelm@13484
  1324
  in (subst_atomic insts tm, xs) end;
wenzelm@13484
  1325
wenzelm@13484
  1326
wenzelm@14786
  1327
(* string_of_vname *)
wenzelm@14786
  1328
wenzelm@15986
  1329
val show_question_marks = ref true;
berghofe@15472
  1330
wenzelm@14786
  1331
fun string_of_vname (x, i) =
wenzelm@14786
  1332
  let
wenzelm@15986
  1333
    val question_mark = if ! show_question_marks then "?" else "";
wenzelm@15986
  1334
    val idx = string_of_int i;
wenzelm@15986
  1335
    val dot =
wenzelm@15986
  1336
      (case rev (Symbol.explode x) of
wenzelm@15986
  1337
        _ :: "\\<^isub>" :: _ => false
wenzelm@15986
  1338
      | _ :: "\\<^isup>" :: _ => false
wenzelm@15986
  1339
      | c :: _ => Symbol.is_digit c
wenzelm@15986
  1340
      | _ => true);
wenzelm@14786
  1341
  in
wenzelm@15986
  1342
    if dot then question_mark ^ x ^ "." ^ idx
wenzelm@15986
  1343
    else if i <> 0 then question_mark ^ x ^ idx
wenzelm@15986
  1344
    else question_mark ^ x
wenzelm@14786
  1345
  end;
wenzelm@14786
  1346
wenzelm@14786
  1347
fun string_of_vname' (x, ~1) = x
wenzelm@14786
  1348
  | string_of_vname' xi = string_of_vname xi;
wenzelm@14786
  1349
clasohm@1364
  1350
end;
clasohm@1364
  1351
wenzelm@4444
  1352
structure BasicTerm: BASIC_TERM = Term;
wenzelm@4444
  1353
open BasicTerm;