src/Pure/variable.ML
author wenzelm
Tue Nov 06 22:50:34 2007 +0100 (2007-11-06)
changeset 25316 17c183417f93
parent 25051 71cd45fdf332
child 25325 0659c05cc107
permissions -rw-r--r--
added is_const/declare_const for local scope of fixes/consts;
wenzelm@19899
     1
(*  Title:      Pure/variable.ML
wenzelm@19899
     2
    ID:         $Id$
wenzelm@19899
     3
    Author:     Makarius
wenzelm@19899
     4
wenzelm@19899
     5
Fixed type/term variables and polymorphic term abbreviations.
wenzelm@19899
     6
*)
wenzelm@19899
     7
wenzelm@19899
     8
signature VARIABLE =
wenzelm@19899
     9
sig
wenzelm@20303
    10
  val is_body: Proof.context -> bool
wenzelm@20303
    11
  val set_body: bool -> Proof.context -> Proof.context
wenzelm@20303
    12
  val restore_body: Proof.context -> Proof.context -> Proof.context
wenzelm@20303
    13
  val names_of: Proof.context -> Name.context
wenzelm@20303
    14
  val fixes_of: Proof.context -> (string * string) list
wenzelm@20303
    15
  val binds_of: Proof.context -> (typ * term) Vartab.table
wenzelm@24765
    16
  val maxidx_of: Proof.context -> int
wenzelm@20303
    17
  val constraints_of: Proof.context -> typ Vartab.table * sort Vartab.table
wenzelm@20303
    18
  val is_declared: Proof.context -> string -> bool
wenzelm@20303
    19
  val is_fixed: Proof.context -> string -> bool
wenzelm@20303
    20
  val newly_fixed: Proof.context -> Proof.context -> string -> bool
wenzelm@21571
    21
  val add_fixed: Proof.context -> term -> (string * typ) list -> (string * typ) list
wenzelm@20303
    22
  val default_type: Proof.context -> string -> typ option
wenzelm@20303
    23
  val def_type: Proof.context -> bool -> indexname -> typ option
wenzelm@20303
    24
  val def_sort: Proof.context -> indexname -> sort option
wenzelm@20303
    25
  val declare_constraints: term -> Proof.context -> Proof.context
wenzelm@20303
    26
  val declare_internal: term -> Proof.context -> Proof.context
wenzelm@20303
    27
  val declare_term: term -> Proof.context -> Proof.context
wenzelm@20303
    28
  val declare_prf: Proofterm.proof -> Proof.context -> Proof.context
wenzelm@20303
    29
  val declare_thm: thm -> Proof.context -> Proof.context
wenzelm@20303
    30
  val thm_context: thm -> Proof.context
wenzelm@20303
    31
  val variant_frees: Proof.context -> term list -> (string * 'a) list -> (string * 'a) list
wenzelm@20303
    32
  val add_binds: (indexname * term option) list -> Proof.context -> Proof.context
wenzelm@20303
    33
  val expand_binds: Proof.context -> term -> term
wenzelm@25316
    34
  val is_const: Proof.context -> string -> bool
wenzelm@25316
    35
  val declare_const: string -> Proof.context -> Proof.context
wenzelm@20303
    36
  val add_fixes: string list -> Proof.context -> string list * Proof.context
wenzelm@20303
    37
  val add_fixes_direct: string list -> Proof.context -> Proof.context
wenzelm@21369
    38
  val auto_fixes: term -> Proof.context -> Proof.context
wenzelm@20797
    39
  val variant_fixes: string list -> Proof.context -> string list * Proof.context
wenzelm@20303
    40
  val invent_types: sort list -> Proof.context -> (string * sort) list * Proof.context
wenzelm@20303
    41
  val export_terms: Proof.context -> Proof.context -> term list -> term list
wenzelm@20303
    42
  val exportT_terms: Proof.context -> Proof.context -> term list -> term list
wenzelm@20303
    43
  val exportT: Proof.context -> Proof.context -> thm list -> thm list
wenzelm@20303
    44
  val export_prf: Proof.context -> Proof.context -> Proofterm.proof -> Proofterm.proof
wenzelm@20303
    45
  val export: Proof.context -> Proof.context -> thm list -> thm list
wenzelm@21522
    46
  val export_morphism: Proof.context -> Proof.context -> morphism
wenzelm@20303
    47
  val importT_inst: term list -> Proof.context -> ((indexname * sort) * typ) list * Proof.context
wenzelm@20303
    48
  val import_inst: bool -> term list -> Proof.context ->
wenzelm@20303
    49
    (((indexname * sort) * typ) list * ((indexname * typ) * term) list) * Proof.context
wenzelm@20303
    50
  val importT_terms: term list -> Proof.context -> term list * Proof.context
wenzelm@20303
    51
  val import_terms: bool -> term list -> Proof.context -> term list * Proof.context
wenzelm@22601
    52
  val importT_thms: thm list -> Proof.context -> (ctyp list * thm list) * Proof.context
wenzelm@20303
    53
  val import_prf: bool -> Proofterm.proof -> Proof.context -> Proofterm.proof * Proof.context
wenzelm@22568
    54
  val import_thms: bool -> thm list -> Proof.context ->
wenzelm@20303
    55
    ((ctyp list * cterm list) * thm list) * Proof.context
wenzelm@21287
    56
  val tradeT: (Proof.context -> thm list -> thm list) -> Proof.context -> thm list -> thm list
wenzelm@21287
    57
  val trade: (Proof.context -> thm list -> thm list) -> Proof.context -> thm list -> thm list
wenzelm@20303
    58
  val focus: cterm -> Proof.context -> (cterm list * cterm) * Proof.context
wenzelm@20303
    59
  val focus_subgoal: int -> thm -> Proof.context -> (cterm list * cterm) * Proof.context
wenzelm@20303
    60
  val warn_extra_tfrees: Proof.context -> Proof.context -> unit
wenzelm@24694
    61
  val polymorphic_types: Proof.context -> term list -> (indexname * sort) list * term list
wenzelm@20303
    62
  val polymorphic: Proof.context -> term list -> term list
wenzelm@19899
    63
end;
wenzelm@19899
    64
wenzelm@19899
    65
structure Variable: VARIABLE =
wenzelm@19899
    66
struct
wenzelm@19899
    67
wenzelm@19899
    68
(** local context data **)
wenzelm@19899
    69
wenzelm@19899
    70
datatype data = Data of
wenzelm@20102
    71
 {is_body: bool,                        (*inner body mode*)
wenzelm@20162
    72
  names: Name.context,                  (*type/term variable names*)
wenzelm@25316
    73
  scope: bool Symtab.table,             (*local scope of fixes/consts*)
wenzelm@20162
    74
  fixes: (string * string) list,        (*term fixes -- extern/intern*)
wenzelm@20102
    75
  binds: (typ * term) Vartab.table,     (*term bindings*)
wenzelm@20162
    76
  type_occs: string list Symtab.table,  (*type variables -- possibly within term variables*)
wenzelm@24765
    77
  maxidx: int,                          (*maximum var index*)
wenzelm@20162
    78
  constraints:
wenzelm@20102
    79
    typ Vartab.table *                  (*type constraints*)
wenzelm@20162
    80
    sort Vartab.table};                 (*default sorts*)
wenzelm@19899
    81
wenzelm@25316
    82
fun make_data (is_body, names, scope, fixes, binds, type_occs, maxidx, constraints) =
wenzelm@25316
    83
  Data {is_body = is_body, names = names, scope = scope, fixes = fixes, binds = binds,
wenzelm@24765
    84
    type_occs = type_occs, maxidx = maxidx, constraints = constraints};
wenzelm@19899
    85
wenzelm@19899
    86
structure Data = ProofDataFun
wenzelm@19899
    87
(
wenzelm@19899
    88
  type T = data;
wenzelm@19899
    89
  fun init thy =
wenzelm@25316
    90
    make_data (false, Name.context, Symtab.empty, [], Vartab.empty, Symtab.empty,
wenzelm@24765
    91
      ~1, (Vartab.empty, Vartab.empty));
wenzelm@19899
    92
);
wenzelm@19899
    93
wenzelm@19899
    94
fun map_data f =
wenzelm@25316
    95
  Data.map (fn Data {is_body, names, scope, fixes, binds, type_occs, maxidx, constraints} =>
wenzelm@25316
    96
    make_data (f (is_body, names, scope, fixes, binds, type_occs, maxidx, constraints)));
wenzelm@25316
    97
wenzelm@25316
    98
fun map_names f =
wenzelm@25316
    99
  map_data (fn (is_body, names, scope, fixes, binds, type_occs, maxidx, constraints) =>
wenzelm@25316
   100
    (is_body, f names, scope, fixes, binds, type_occs, maxidx, constraints));
wenzelm@19899
   101
wenzelm@25316
   102
fun map_scope f =
wenzelm@25316
   103
  map_data (fn (is_body, names, scope, fixes, binds, type_occs, maxidx, constraints) =>
wenzelm@25316
   104
    (is_body, names, f scope, fixes, binds, type_occs, maxidx, constraints));
wenzelm@20162
   105
wenzelm@25316
   106
fun map_fixes f =
wenzelm@25316
   107
  map_data (fn (is_body, names, scope, fixes, binds, type_occs, maxidx, constraints) =>
wenzelm@25316
   108
    (is_body, names, scope, f fixes, binds, type_occs, maxidx, constraints));
wenzelm@19899
   109
wenzelm@25316
   110
fun map_binds f =
wenzelm@25316
   111
  map_data (fn (is_body, names, scope, fixes, binds, type_occs, maxidx, constraints) =>
wenzelm@25316
   112
    (is_body, names, scope, fixes, f binds, type_occs, maxidx, constraints));
wenzelm@24765
   113
wenzelm@25316
   114
fun map_type_occs f =
wenzelm@25316
   115
  map_data (fn (is_body, names, scope, fixes, binds, type_occs, maxidx, constraints) =>
wenzelm@25316
   116
    (is_body, names, scope, fixes, binds, f type_occs, maxidx, constraints));
wenzelm@19899
   117
wenzelm@25316
   118
fun map_maxidx f =
wenzelm@25316
   119
  map_data (fn (is_body, names, scope, fixes, binds, type_occs, maxidx, constraints) =>
wenzelm@25316
   120
    (is_body, names, scope, fixes, binds, type_occs, f maxidx, constraints));
wenzelm@20102
   121
wenzelm@25316
   122
fun map_constraints f =
wenzelm@25316
   123
  map_data (fn (is_body, names, scope, fixes, binds, type_occs, maxidx, constraints) =>
wenzelm@25316
   124
    (is_body, names, scope, fixes, binds, type_occs, maxidx, f constraints));
wenzelm@19899
   125
wenzelm@19899
   126
fun rep_data ctxt = Data.get ctxt |> (fn Data args => args);
wenzelm@19899
   127
wenzelm@19899
   128
val is_body = #is_body o rep_data;
wenzelm@24765
   129
wenzelm@25316
   130
fun set_body b = map_data (fn (_, names, scope, fixes, binds, type_occs, maxidx, constraints) =>
wenzelm@25316
   131
  (b, names, scope, fixes, binds, type_occs, maxidx, constraints));
wenzelm@24765
   132
wenzelm@19899
   133
fun restore_body ctxt = set_body (is_body ctxt);
wenzelm@19899
   134
wenzelm@20102
   135
val names_of = #names o rep_data;
wenzelm@20102
   136
val fixes_of = #fixes o rep_data;
wenzelm@19899
   137
val binds_of = #binds o rep_data;
wenzelm@20162
   138
val type_occs_of = #type_occs o rep_data;
wenzelm@24765
   139
val maxidx_of = #maxidx o rep_data;
wenzelm@20162
   140
val constraints_of = #constraints o rep_data;
wenzelm@19899
   141
wenzelm@20162
   142
val is_declared = Name.is_declared o names_of;
wenzelm@19899
   143
fun is_fixed ctxt x = exists (fn (_, y) => x = y) (fixes_of ctxt);
wenzelm@20149
   144
fun newly_fixed inner outer x = is_fixed inner x andalso not (is_fixed outer x);
wenzelm@19899
   145
wenzelm@21571
   146
fun add_fixed ctxt = Term.fold_aterms
wenzelm@21571
   147
  (fn Free (x, T) => if is_fixed ctxt x then insert (op =) (x, T) else I | _ => I);
wenzelm@21571
   148
wenzelm@19899
   149
wenzelm@19899
   150
wenzelm@19899
   151
(** declarations **)
wenzelm@19899
   152
wenzelm@19899
   153
(* default sorts and types *)
wenzelm@19899
   154
wenzelm@20162
   155
fun default_type ctxt x = Vartab.lookup (#1 (constraints_of ctxt)) (x, ~1);
wenzelm@19899
   156
wenzelm@19899
   157
fun def_type ctxt pattern xi =
wenzelm@20162
   158
  let val {binds, constraints = (types, _), ...} = rep_data ctxt in
wenzelm@19899
   159
    (case Vartab.lookup types xi of
wenzelm@19899
   160
      NONE =>
wenzelm@19899
   161
        if pattern then NONE
wenzelm@19899
   162
        else Vartab.lookup binds xi |> Option.map (TypeInfer.polymorphicT o #1)
wenzelm@19899
   163
    | some => some)
wenzelm@19899
   164
  end;
wenzelm@19899
   165
wenzelm@20162
   166
val def_sort = Vartab.lookup o #2 o constraints_of;
wenzelm@20162
   167
wenzelm@20162
   168
wenzelm@20162
   169
(* names *)
wenzelm@20162
   170
wenzelm@24765
   171
fun declare_type_names t =
wenzelm@24765
   172
  map_names (fold_types (fold_atyps (fn TFree (a, _) => Name.declare a | _ => I)) t) #>
wenzelm@24765
   173
  map_maxidx (fold_types Term.maxidx_typ t);
wenzelm@20162
   174
wenzelm@20162
   175
fun declare_names t =
wenzelm@20162
   176
  declare_type_names t #>
wenzelm@24765
   177
  map_names (fold_aterms (fn Free (x, _) => Name.declare x | _ => I) t) #>
wenzelm@24765
   178
  map_maxidx (Term.maxidx_term t);
wenzelm@20162
   179
wenzelm@20162
   180
wenzelm@20162
   181
(* type occurrences *)
wenzelm@20162
   182
wenzelm@24719
   183
fun decl_type_occsT T = fold_atyps (fn TFree (a, _) => Symtab.default (a, []) | _ => I) T;
wenzelm@24719
   184
wenzelm@22671
   185
val decl_type_occs = fold_term_types
wenzelm@20162
   186
  (fn Free (x, _) => fold_atyps (fn TFree (a, _) => Symtab.insert_list (op =) (a, x) | _ => I)
wenzelm@24719
   187
    | _ => decl_type_occsT);
wenzelm@19899
   188
wenzelm@24719
   189
val declare_type_occsT = map_type_occs o fold_types decl_type_occsT;
wenzelm@22671
   190
val declare_type_occs = map_type_occs o decl_type_occs;
wenzelm@22671
   191
wenzelm@19899
   192
wenzelm@20162
   193
(* constraints *)
wenzelm@19899
   194
wenzelm@21355
   195
fun constrain_tvar (xi, S) =
wenzelm@21355
   196
  if S = dummyS then Vartab.delete_safe xi else Vartab.update (xi, S);
wenzelm@21355
   197
wenzelm@20162
   198
fun declare_constraints t = map_constraints (fn (types, sorts) =>
wenzelm@20162
   199
  let
wenzelm@20162
   200
    val types' = fold_aterms
wenzelm@20162
   201
      (fn Free (x, T) => Vartab.update ((x, ~1), T)
wenzelm@20162
   202
        | Var v => Vartab.update v
wenzelm@20162
   203
        | _ => I) t types;
wenzelm@20162
   204
    val sorts' = fold_types (fold_atyps
wenzelm@21355
   205
      (fn TFree (x, S) => constrain_tvar ((x, ~1), S)
wenzelm@21355
   206
        | TVar v => constrain_tvar v
wenzelm@20162
   207
        | _ => I)) t sorts;
wenzelm@20162
   208
  in (types', sorts') end)
wenzelm@24719
   209
  #> declare_type_occsT t
wenzelm@22711
   210
  #> declare_type_names t;
wenzelm@19899
   211
wenzelm@20162
   212
wenzelm@20162
   213
(* common declarations *)
wenzelm@20162
   214
wenzelm@20162
   215
fun declare_internal t =
wenzelm@20162
   216
  declare_names t #>
wenzelm@20162
   217
  declare_type_occs t;
wenzelm@19911
   218
wenzelm@20162
   219
fun declare_term t =
wenzelm@20162
   220
  declare_internal t #>
wenzelm@20162
   221
  declare_constraints t;
wenzelm@19899
   222
wenzelm@20303
   223
val declare_prf = Proofterm.fold_proof_terms declare_internal (declare_internal o Logic.mk_type);
wenzelm@20303
   224
wenzelm@22691
   225
val declare_thm = Thm.fold_terms declare_internal;
wenzelm@21522
   226
fun thm_context th = declare_thm th (ProofContext.init (Thm.theory_of_thm th));
wenzelm@19899
   227
wenzelm@19899
   228
wenzelm@19899
   229
(* renaming term/type frees *)
wenzelm@19899
   230
wenzelm@20162
   231
fun variant_frees ctxt ts frees =
wenzelm@19899
   232
  let
wenzelm@20162
   233
    val names = names_of (fold declare_names ts ctxt);
wenzelm@20084
   234
    val xs = fst (Name.variants (map #1 frees) names);
wenzelm@20084
   235
  in xs ~~ map snd frees end;
wenzelm@19899
   236
wenzelm@19899
   237
wenzelm@19899
   238
wenzelm@20303
   239
(** term bindings **)
wenzelm@20303
   240
wenzelm@20303
   241
fun add_bind (xi, NONE) = map_binds (Vartab.delete_safe xi)
wenzelm@20303
   242
  | add_bind ((x, i), SOME t) =
wenzelm@20303
   243
      let
wenzelm@25051
   244
        val u = Term.close_schematic_term t;
wenzelm@25051
   245
        val U = Term.fastype_of u;
wenzelm@25051
   246
      in declare_term u #> map_binds (Vartab.update ((x, i), (U, u))) end;
wenzelm@20303
   247
wenzelm@20303
   248
val add_binds = fold add_bind;
wenzelm@20303
   249
wenzelm@20303
   250
fun expand_binds ctxt =
wenzelm@20303
   251
  let
wenzelm@20303
   252
    val binds = binds_of ctxt;
wenzelm@21799
   253
    val get = fn Var (xi, _) => Vartab.lookup binds xi | _ => NONE;
wenzelm@21799
   254
  in Envir.beta_norm o Envir.expand_term get end;
wenzelm@20303
   255
wenzelm@20303
   256
wenzelm@20303
   257
wenzelm@25316
   258
(** local scope **)
wenzelm@25316
   259
wenzelm@25316
   260
fun is_const ctxt x = the_default false (Symtab.lookup (#scope (rep_data ctxt)) x);
wenzelm@25316
   261
wenzelm@25316
   262
fun declare_fixed x = map_scope (Symtab.update (x, false));
wenzelm@25316
   263
fun declare_const c = map_scope (Symtab.update (c, true));
wenzelm@25316
   264
wenzelm@25316
   265
wenzelm@25316
   266
wenzelm@19911
   267
(** fixes **)
wenzelm@19911
   268
wenzelm@20084
   269
local
wenzelm@20084
   270
wenzelm@19911
   271
fun no_dups [] = ()
wenzelm@19911
   272
  | no_dups dups = error ("Duplicate fixed variable(s): " ^ commas_quote dups);
wenzelm@19911
   273
wenzelm@20102
   274
fun new_fixes names' xs xs' =
wenzelm@20102
   275
  map_names (K names') #>
wenzelm@25316
   276
  fold declare_fixed xs #>
wenzelm@20102
   277
  map_fixes (fn fixes => (rev (xs ~~ xs') @ fixes)) #>
wenzelm@20162
   278
  fold (declare_constraints o Syntax.free) xs' #>
wenzelm@20084
   279
  pair xs';
wenzelm@20084
   280
wenzelm@20084
   281
in
wenzelm@20084
   282
wenzelm@19911
   283
fun add_fixes xs ctxt =
wenzelm@19911
   284
  let
wenzelm@19911
   285
    val _ =
wenzelm@20084
   286
      (case filter (can Name.dest_skolem) xs of [] => ()
wenzelm@19911
   287
      | bads => error ("Illegal internal Skolem constant(s): " ^ commas_quote bads));
wenzelm@20084
   288
    val _ = no_dups (duplicates (op =) xs);
wenzelm@20084
   289
    val (ys, zs) = split_list (fixes_of ctxt);
wenzelm@20102
   290
    val names = names_of ctxt;
wenzelm@20084
   291
    val (xs', names') =
wenzelm@20149
   292
      if is_body ctxt then Name.variants xs names |>> map Name.skolem
wenzelm@20084
   293
      else (no_dups (xs inter_string ys); no_dups (xs inter_string zs);
wenzelm@20084
   294
        (xs, fold Name.declare xs names));
wenzelm@20102
   295
  in ctxt |> new_fixes names' xs xs' end;
wenzelm@19911
   296
wenzelm@20797
   297
fun variant_fixes raw_xs ctxt =
wenzelm@20084
   298
  let
wenzelm@20102
   299
    val names = names_of ctxt;
wenzelm@20149
   300
    val xs = map Name.clean raw_xs;
wenzelm@20149
   301
    val (xs', names') = Name.variants xs names |>> map Name.skolem;
wenzelm@20102
   302
  in ctxt |> new_fixes names' xs xs' end;
wenzelm@20084
   303
wenzelm@20084
   304
end;
wenzelm@19911
   305
wenzelm@20251
   306
wenzelm@20251
   307
fun add_fixes_direct xs ctxt = ctxt
wenzelm@20251
   308
  |> set_body false
wenzelm@20251
   309
  |> (snd o add_fixes xs)
wenzelm@20251
   310
  |> restore_body ctxt;
wenzelm@20251
   311
wenzelm@20251
   312
fun fix_frees t ctxt = ctxt
wenzelm@20251
   313
  |> add_fixes_direct
wenzelm@20251
   314
      (rev (fold_aterms (fn Free (x, _) =>
wenzelm@21369
   315
        if is_fixed ctxt x then I else insert (op =) x | _ => I) t []));
wenzelm@21369
   316
wenzelm@21369
   317
fun auto_fixes t ctxt =
wenzelm@21369
   318
  (if is_body ctxt then ctxt else fix_frees t ctxt)
wenzelm@20251
   319
  |> declare_term t;
wenzelm@20251
   320
wenzelm@19911
   321
fun invent_types Ss ctxt =
wenzelm@19911
   322
  let
wenzelm@24848
   323
    val tfrees = Name.invents (names_of ctxt) Name.aT (length Ss) ~~ Ss;
wenzelm@20162
   324
    val ctxt' = fold (declare_constraints o Logic.mk_type o TFree) tfrees ctxt;
wenzelm@19911
   325
  in (tfrees, ctxt') end;
wenzelm@19911
   326
wenzelm@19911
   327
wenzelm@19911
   328
wenzelm@22671
   329
(** export -- generalize type/term variables (beware of closure sizes) **)
wenzelm@19911
   330
wenzelm@19911
   331
fun export_inst inner outer =
wenzelm@19911
   332
  let
wenzelm@20162
   333
    val declared_outer = is_declared outer;
wenzelm@19911
   334
    val fixes_inner = fixes_of inner;
wenzelm@19911
   335
    val fixes_outer = fixes_of outer;
wenzelm@19911
   336
wenzelm@19911
   337
    val gen_fixes = map #2 (Library.take (length fixes_inner - length fixes_outer, fixes_inner));
wenzelm@20162
   338
    val still_fixed = not o member (op =) gen_fixes;
wenzelm@22671
   339
wenzelm@22671
   340
    val type_occs_inner = type_occs_of inner;
wenzelm@22671
   341
    fun gen_fixesT ts =
wenzelm@19911
   342
      Symtab.fold (fn (a, xs) =>
wenzelm@20162
   343
        if declared_outer a orelse exists still_fixed xs
wenzelm@22671
   344
        then I else cons a) (fold decl_type_occs ts type_occs_inner) [];
wenzelm@19911
   345
  in (gen_fixesT, gen_fixes) end;
wenzelm@19911
   346
wenzelm@19911
   347
fun exportT_inst inner outer = #1 (export_inst inner outer);
wenzelm@19911
   348
wenzelm@22671
   349
fun exportT_terms inner outer =
wenzelm@22671
   350
  let val mk_tfrees = exportT_inst inner outer in
wenzelm@22671
   351
    fn ts => ts |> map
wenzelm@22671
   352
      (TermSubst.generalize (mk_tfrees ts, [])
wenzelm@22671
   353
        (fold (Term.fold_types Term.maxidx_typ) ts ~1 + 1))
wenzelm@22671
   354
  end;
wenzelm@19911
   355
wenzelm@22671
   356
fun export_terms inner outer =
wenzelm@22671
   357
  let val (mk_tfrees, tfrees) = export_inst inner outer in
wenzelm@22671
   358
    fn ts => ts |> map
wenzelm@22671
   359
      (TermSubst.generalize (mk_tfrees ts, tfrees)
wenzelm@22671
   360
        (fold Term.maxidx_term ts ~1 + 1))
wenzelm@22671
   361
  end;
wenzelm@19911
   362
wenzelm@20303
   363
fun export_prf inner outer prf =
wenzelm@20303
   364
  let
wenzelm@22671
   365
    val (mk_tfrees, frees) = export_inst (declare_prf prf inner) outer;
wenzelm@22671
   366
    val tfrees = mk_tfrees [];
wenzelm@20303
   367
    val idx = Proofterm.maxidx_proof prf ~1 + 1;
wenzelm@22671
   368
    val gen_term = TermSubst.generalize_option (tfrees, frees) idx;
wenzelm@22671
   369
    val gen_typ = TermSubst.generalizeT_option tfrees idx;
wenzelm@20303
   370
  in Proofterm.map_proof_terms_option gen_term gen_typ prf end;
wenzelm@20303
   371
wenzelm@22671
   372
wenzelm@22671
   373
fun gen_export (mk_tfrees, frees) ths =
wenzelm@19911
   374
  let
wenzelm@22671
   375
    val tfrees = mk_tfrees (map Thm.full_prop_of ths);
wenzelm@22671
   376
    val maxidx = fold Thm.maxidx_thm ths ~1;
wenzelm@22671
   377
  in map (Thm.generalize (tfrees, frees) (maxidx + 1)) ths end;
wenzelm@19911
   378
wenzelm@22671
   379
fun exportT inner outer = gen_export (exportT_inst inner outer, []);
wenzelm@22671
   380
fun export inner outer = gen_export (export_inst inner outer);
wenzelm@19911
   381
wenzelm@21522
   382
fun export_morphism inner outer =
wenzelm@21522
   383
  let
wenzelm@21522
   384
    val fact = export inner outer;
wenzelm@21522
   385
    val term = singleton (export_terms inner outer);
wenzelm@21522
   386
    val typ = Logic.type_map term;
wenzelm@21522
   387
  in Morphism.morphism {name = I, var = I, typ = typ, term = term, fact = fact} end;
wenzelm@19911
   388
wenzelm@19911
   389
wenzelm@24765
   390
wenzelm@19911
   391
(** import -- fix schematic type/term variables **)
wenzelm@19911
   392
wenzelm@19911
   393
fun importT_inst ts ctxt =
wenzelm@19911
   394
  let
wenzelm@19911
   395
    val tvars = rev (fold Term.add_tvars ts []);
wenzelm@19911
   396
    val (tfrees, ctxt') = invent_types (map #2 tvars) ctxt;
wenzelm@19911
   397
  in (tvars ~~ map TFree tfrees, ctxt') end;
wenzelm@19911
   398
wenzelm@19911
   399
fun import_inst is_open ts ctxt =
wenzelm@19911
   400
  let
wenzelm@20198
   401
    val ren = if is_open then I else Name.internal;
wenzelm@19911
   402
    val (instT, ctxt') = importT_inst ts ctxt;
wenzelm@20509
   403
    val vars = map (apsnd (TermSubst.instantiateT instT)) (rev (fold Term.add_vars ts []));
wenzelm@20797
   404
    val (xs, ctxt'') = variant_fixes (map (ren o #1 o #1) vars) ctxt';
wenzelm@19911
   405
    val inst = vars ~~ map Free (xs ~~ map #2 vars);
wenzelm@19911
   406
  in ((instT, inst), ctxt'') end;
wenzelm@19911
   407
wenzelm@19911
   408
fun importT_terms ts ctxt =
wenzelm@19911
   409
  let val (instT, ctxt') = importT_inst ts ctxt
wenzelm@20509
   410
  in (map (TermSubst.instantiate (instT, [])) ts, ctxt') end;
wenzelm@19911
   411
wenzelm@19911
   412
fun import_terms is_open ts ctxt =
wenzelm@19911
   413
  let val (inst, ctxt') = import_inst is_open ts ctxt
wenzelm@20509
   414
  in (map (TermSubst.instantiate inst) ts, ctxt') end;
wenzelm@19911
   415
wenzelm@22601
   416
fun importT_thms ths ctxt =
wenzelm@19911
   417
  let
wenzelm@21522
   418
    val thy = ProofContext.theory_of ctxt;
wenzelm@19911
   419
    val certT = Thm.ctyp_of thy;
wenzelm@19911
   420
    val (instT, ctxt') = importT_inst (map Thm.full_prop_of ths) ctxt;
wenzelm@19911
   421
    val instT' = map (fn (v, T) => (certT (TVar v), certT T)) instT;
wenzelm@19911
   422
    val ths' = map (Thm.instantiate (instT', [])) ths;
wenzelm@20220
   423
  in ((map #2 instT', ths'), ctxt') end;
wenzelm@19911
   424
wenzelm@20303
   425
fun import_prf is_open prf ctxt =
wenzelm@20303
   426
  let
wenzelm@20303
   427
    val ts = rev (Proofterm.fold_proof_terms cons (cons o Logic.mk_type) prf []);
wenzelm@20303
   428
    val (insts, ctxt') = import_inst is_open ts ctxt;
wenzelm@20303
   429
  in (Proofterm.instantiate insts prf, ctxt') end;
wenzelm@20303
   430
wenzelm@22568
   431
fun import_thms is_open ths ctxt =
wenzelm@19911
   432
  let
wenzelm@21522
   433
    val thy = ProofContext.theory_of ctxt;
wenzelm@19911
   434
    val cert = Thm.cterm_of thy;
wenzelm@19911
   435
    val certT = Thm.ctyp_of thy;
wenzelm@19911
   436
    val ((instT, inst), ctxt') = import_inst is_open (map Thm.full_prop_of ths) ctxt;
wenzelm@19911
   437
    val instT' = map (fn (v, T) => (certT (TVar v), certT T)) instT;
wenzelm@19911
   438
    val inst' = map (fn (v, t) => (cert (Var v), cert t)) inst;
wenzelm@19911
   439
    val ths' = map (Thm.instantiate (instT', inst')) ths;
wenzelm@20220
   440
  in (((map #2 instT', map #2 inst'), ths'), ctxt') end;
wenzelm@19911
   441
wenzelm@19911
   442
wenzelm@19926
   443
(* import/export *)
wenzelm@19926
   444
wenzelm@21287
   445
fun gen_trade imp exp f ctxt ths =
wenzelm@20220
   446
  let val ((_, ths'), ctxt') = imp ths ctxt
wenzelm@21287
   447
  in exp ctxt' ctxt (f ctxt' ths') end;
wenzelm@19926
   448
wenzelm@22601
   449
val tradeT = gen_trade importT_thms exportT;
wenzelm@22568
   450
val trade = gen_trade (import_thms true) export;
wenzelm@19926
   451
wenzelm@19926
   452
wenzelm@20162
   453
(* focus on outermost parameters *)
wenzelm@20149
   454
wenzelm@20149
   455
fun forall_elim_prop t prop =
wenzelm@20579
   456
  Thm.beta_conversion false (Thm.capply (Thm.dest_arg prop) t)
wenzelm@20579
   457
  |> Thm.cprop_of |> Thm.dest_arg;
wenzelm@20149
   458
wenzelm@20149
   459
fun focus goal ctxt =
wenzelm@20149
   460
  let
wenzelm@20149
   461
    val cert = Thm.cterm_of (Thm.theory_of_cterm goal);
wenzelm@20149
   462
    val t = Thm.term_of goal;
wenzelm@20162
   463
    val ps = Term.variant_frees t (Term.strip_all_vars t);   (*as they are printed :-*)
wenzelm@20149
   464
    val (xs, Ts) = split_list ps;
wenzelm@20797
   465
    val (xs', ctxt') = variant_fixes xs ctxt;
wenzelm@20220
   466
    val ps' = ListPair.map (cert o Free) (xs', Ts);
wenzelm@20220
   467
    val goal' = fold forall_elim_prop ps' goal;
wenzelm@20149
   468
  in ((ps', goal'), ctxt') end;
wenzelm@20149
   469
wenzelm@20303
   470
fun focus_subgoal i st =
wenzelm@20303
   471
  let
wenzelm@22691
   472
    val all_vars = Thm.fold_terms Term.add_vars st [];
wenzelm@20303
   473
    val no_binds = map (fn (xi, _) => (xi, NONE)) all_vars;
wenzelm@20303
   474
  in
wenzelm@20303
   475
    add_binds no_binds #>
wenzelm@20303
   476
    fold (declare_constraints o Var) all_vars #>
wenzelm@20303
   477
    focus (Thm.cprem_of st i)
wenzelm@20303
   478
  end;
wenzelm@20303
   479
wenzelm@20303
   480
wenzelm@19911
   481
wenzelm@19911
   482
(** implicit polymorphism **)
wenzelm@19899
   483
wenzelm@19899
   484
(* warn_extra_tfrees *)
wenzelm@19899
   485
wenzelm@19899
   486
fun warn_extra_tfrees ctxt1 ctxt2 =
wenzelm@19899
   487
  let
wenzelm@19911
   488
    fun occs_typ a = Term.exists_subtype (fn TFree (b, _) => a = b | _ => false);
wenzelm@20162
   489
    fun occs_free a x =
wenzelm@20162
   490
      (case def_type ctxt1 false (x, ~1) of
wenzelm@20162
   491
        SOME T => if occs_typ a T then I else cons (a, x)
wenzelm@20162
   492
      | NONE => cons (a, x));
wenzelm@19899
   493
wenzelm@20162
   494
    val occs1 = type_occs_of ctxt1;
wenzelm@20162
   495
    val occs2 = type_occs_of ctxt2;
wenzelm@19911
   496
    val extras = Symtab.fold (fn (a, xs) =>
wenzelm@19911
   497
      if Symtab.defined occs1 a then I else fold (occs_free a) xs) occs2 [];
wenzelm@19899
   498
    val tfrees = map #1 extras |> sort_distinct string_ord;
wenzelm@19899
   499
    val frees = map #2 extras |> sort_distinct string_ord;
wenzelm@19899
   500
  in
wenzelm@19899
   501
    if null extras then ()
wenzelm@19899
   502
    else warning ("Introduced fixed type variable(s): " ^ commas tfrees ^ " in " ^
wenzelm@19899
   503
      space_implode " or " (map quote frees))
wenzelm@19899
   504
  end;
wenzelm@19899
   505
wenzelm@19899
   506
wenzelm@20149
   507
(* polymorphic terms *)
wenzelm@19899
   508
wenzelm@24694
   509
fun polymorphic_types ctxt ts =
wenzelm@20003
   510
  let
wenzelm@20003
   511
    val ctxt' = fold declare_term ts ctxt;
wenzelm@20162
   512
    val occs = type_occs_of ctxt;
wenzelm@20162
   513
    val occs' = type_occs_of ctxt';
wenzelm@20162
   514
    val types = Symtab.fold (fn (a, _) => if Symtab.defined occs a then I else cons a) occs' [];
wenzelm@24765
   515
    val idx = maxidx_of ctxt' + 1;
wenzelm@24694
   516
    val Ts' = (fold o fold_types o fold_atyps)
wenzelm@24694
   517
      (fn T as TFree _ =>
wenzelm@24694
   518
          (case TermSubst.generalizeT types idx T of TVar v => insert (op =) v | _ => I)
wenzelm@24694
   519
        | _ => I) ts [];
wenzelm@24694
   520
    val ts' = map (TermSubst.generalize (types, []) idx) ts;
wenzelm@24694
   521
  in (rev Ts', ts') end;
wenzelm@24694
   522
wenzelm@24694
   523
fun polymorphic ctxt ts = snd (polymorphic_types ctxt ts);
wenzelm@20003
   524
wenzelm@19899
   525
end;