src/Pure/Isar/locale.ML
author wenzelm
Thu Jun 13 17:22:10 2002 +0200 (2002-06-13)
changeset 13211 aabdb4b33625
parent 12958 99f5c4a37b29
child 13297 e4ae0732e2be
permissions -rw-r--r--
BUG FIX in inst_thm: use current context instead of that of thm!!!
wenzelm@12014
     1
(*  Title:      Pure/Isar/locale.ML
wenzelm@11896
     2
    ID:         $Id$
wenzelm@12958
     3
    Author:     Markus Wenzel, LMU/TU München
wenzelm@11896
     4
    License:    GPL (GNU GENERAL PUBLIC LICENSE)
wenzelm@11896
     5
wenzelm@12058
     6
Locales -- Isar proof contexts as meta-level predicates, with local
wenzelm@12529
     7
syntax and implicit structures.
wenzelm@12529
     8
wenzelm@12529
     9
Draws some basic ideas from Florian Kammüller's original version of
wenzelm@12529
    10
locales, but uses the richer infrastructure of Isar instead of the raw
wenzelm@12529
    11
meta-logic.  Furthermore, we provide structured import of contexts
wenzelm@12529
    12
(with merge and rename operations), as well as type-inference of the
wenzelm@12529
    13
signature parts.
wenzelm@11896
    14
*)
wenzelm@11896
    15
wenzelm@11896
    16
signature LOCALE =
wenzelm@11896
    17
sig
wenzelm@12046
    18
  type context
wenzelm@12046
    19
  datatype ('typ, 'term, 'fact, 'att) elem =
wenzelm@12058
    20
    Fixes of (string * 'typ option * mixfix option) list |
wenzelm@12046
    21
    Assumes of ((string * 'att list) * ('term * ('term list * 'term list)) list) list |
wenzelm@12046
    22
    Defines of ((string * 'att list) * ('term * 'term list)) list |
wenzelm@12273
    23
    Notes of ((string * 'att list) * ('fact * 'att list) list) list
wenzelm@12273
    24
  datatype expr =
wenzelm@12273
    25
    Locale of string |
wenzelm@12273
    26
    Rename of expr * string option list |
wenzelm@12273
    27
    Merge of expr list
wenzelm@12273
    28
  val empty: expr
wenzelm@12273
    29
  datatype ('typ, 'term, 'fact, 'att) elem_expr =
wenzelm@12273
    30
    Elem of ('typ, 'term, 'fact, 'att) elem | Expr of expr
wenzelm@12046
    31
  type 'att element
wenzelm@12046
    32
  type 'att element_i
wenzelm@12046
    33
  type locale
wenzelm@12046
    34
  val intern: Sign.sg -> xstring -> string
wenzelm@12014
    35
  val cond_extern: Sign.sg -> string -> xstring
wenzelm@12502
    36
  val the_locale: theory -> string -> locale
wenzelm@12273
    37
  val attribute: ('att -> context attribute) -> ('typ, 'term, 'thm, 'att) elem_expr
wenzelm@12273
    38
    -> ('typ, 'term, 'thm, context attribute) elem_expr
wenzelm@12834
    39
  val locale_facts: theory -> xstring -> thm list
wenzelm@12834
    40
  val locale_facts_i: theory -> string -> thm list
wenzelm@12529
    41
  val read_context_statement: xstring option -> context attribute element list ->
wenzelm@12529
    42
    (string * (string list * string list)) list list -> context ->
wenzelm@12529
    43
    string option * context * context * (term * (term list * term list)) list list
wenzelm@12529
    44
  val cert_context_statement: string option -> context attribute element_i list ->
wenzelm@12529
    45
    (term * (term list * term list)) list list -> context ->
wenzelm@12529
    46
    string option * context * context * (term * (term list * term list)) list list
wenzelm@12758
    47
  val print_locales: theory -> unit
wenzelm@12758
    48
  val print_locale: theory -> expr -> context attribute element list -> unit
wenzelm@12273
    49
  val add_locale: bstring -> expr -> context attribute element list -> theory -> theory
wenzelm@12273
    50
  val add_locale_i: bstring -> expr -> context attribute element_i list -> theory -> theory
wenzelm@12958
    51
  val smart_have_thmss: string -> (string * 'a) Library.option ->
wenzelm@12958
    52
    ((bstring * theory attribute list) * (thm list * theory attribute list) list) list ->
wenzelm@12958
    53
    theory -> theory * (bstring * thm list) list
wenzelm@12711
    54
  val have_thmss: string -> xstring ->
wenzelm@12711
    55
    ((bstring * context attribute list) * (xstring * context attribute list) list) list ->
wenzelm@12711
    56
    theory -> theory * (bstring * thm list) list
wenzelm@12711
    57
  val have_thmss_i: string -> string ->
wenzelm@12711
    58
    ((bstring * context attribute list) * (thm list * context attribute list) list) list ->
wenzelm@12711
    59
    theory -> theory * (bstring * thm list) list
wenzelm@12958
    60
  val add_thmss: string -> ((string * thm list) * context attribute list) list ->
wenzelm@12958
    61
    context * theory -> (context * theory) * thm list list
wenzelm@11896
    62
  val setup: (theory -> theory) list
wenzelm@11896
    63
end;
wenzelm@12839
    64
wenzelm@12289
    65
structure Locale: LOCALE =
wenzelm@11896
    66
struct
wenzelm@11896
    67
wenzelm@12273
    68
(** locale elements and expressions **)
wenzelm@11896
    69
wenzelm@12014
    70
type context = ProofContext.context;
wenzelm@11896
    71
wenzelm@12046
    72
datatype ('typ, 'term, 'fact, 'att) elem =
wenzelm@12058
    73
  Fixes of (string * 'typ option * mixfix option) list |
wenzelm@12046
    74
  Assumes of ((string * 'att list) * ('term * ('term list * 'term list)) list) list |
wenzelm@12046
    75
  Defines of ((string * 'att list) * ('term * 'term list)) list |
wenzelm@12273
    76
  Notes of ((string * 'att list) * ('fact * 'att list) list) list;
wenzelm@12273
    77
wenzelm@12839
    78
datatype fact_kind = Assume | Define | Note;
wenzelm@12839
    79
wenzelm@12273
    80
datatype expr =
wenzelm@12273
    81
  Locale of string |
wenzelm@12273
    82
  Rename of expr * string option list |
wenzelm@12273
    83
  Merge of expr list;
wenzelm@11896
    84
wenzelm@12273
    85
val empty = Merge [];
wenzelm@12273
    86
wenzelm@12273
    87
datatype ('typ, 'term, 'fact, 'att) elem_expr =
wenzelm@12273
    88
  Elem of ('typ, 'term, 'fact, 'att) elem | Expr of expr;
wenzelm@12273
    89
wenzelm@12273
    90
type 'att element = (string, string, string, 'att) elem_expr;
wenzelm@12273
    91
type 'att element_i = (typ, term, thm list, 'att) elem_expr;
wenzelm@12070
    92
wenzelm@12070
    93
type locale =
wenzelm@12289
    94
 {import: expr,                                                         (*dynamic import*)
wenzelm@12289
    95
  elems: ((typ, term, thm list, context attribute) elem * stamp) list,  (*static content*)
wenzelm@12839
    96
  text: ((string * typ) list * term list) * ((string * typ) list * (term * term) list),
wenzelm@12839
    97
    (*local predicate specification and definitions*)
wenzelm@12839
    98
  params: (string * typ option) list * string list};                    (*all vs. local params*)
wenzelm@12063
    99
wenzelm@12839
   100
fun make_locale import elems text params =
wenzelm@12839
   101
 {import = import, elems = elems, text = text, params = params}: locale;
wenzelm@12063
   102
wenzelm@11896
   103
wenzelm@11896
   104
wenzelm@11896
   105
(** theory data **)
wenzelm@11896
   106
wenzelm@11896
   107
structure LocalesArgs =
wenzelm@11896
   108
struct
wenzelm@12014
   109
  val name = "Isar/locales";
wenzelm@12063
   110
  type T = NameSpace.T * locale Symtab.table;
wenzelm@11896
   111
wenzelm@12063
   112
  val empty = (NameSpace.empty, Symtab.empty);
wenzelm@12063
   113
  val copy = I;
wenzelm@12118
   114
  val prep_ext = I;
wenzelm@12289
   115
wenzelm@12289
   116
  (*joining of locale elements: only facts may be added later!*)
wenzelm@12839
   117
  fun join ({import, elems, text, params}: locale, {elems = elems', ...}: locale) =
wenzelm@12839
   118
    Some (make_locale import (gen_merge_lists eq_snd elems elems') text params);
wenzelm@12273
   119
  fun merge ((space1, locs1), (space2, locs2)) =
wenzelm@12289
   120
    (NameSpace.merge (space1, space2), Symtab.join join (locs1, locs2));
wenzelm@12289
   121
wenzelm@12273
   122
  fun print _ (space, locs) =
wenzelm@12273
   123
    Pretty.strs ("locales:" :: map (NameSpace.cond_extern space o #1) (Symtab.dest locs))
wenzelm@12014
   124
    |> Pretty.writeln;
wenzelm@11896
   125
end;
wenzelm@11896
   126
wenzelm@11896
   127
structure LocalesData = TheoryDataFun(LocalesArgs);
wenzelm@11896
   128
val print_locales = LocalesData.print;
wenzelm@11896
   129
wenzelm@12289
   130
val intern = NameSpace.intern o #1 o LocalesData.get_sg;
wenzelm@12289
   131
val cond_extern = NameSpace.cond_extern o #1 o LocalesData.get_sg;
wenzelm@12289
   132
wenzelm@12277
   133
wenzelm@12277
   134
(* access locales *)
wenzelm@12277
   135
wenzelm@12063
   136
fun declare_locale name =
wenzelm@12063
   137
  LocalesData.map (apfst (fn space => (NameSpace.extend (space, [name]))));
wenzelm@11896
   138
wenzelm@12273
   139
fun put_locale name loc = LocalesData.map (apsnd (fn locs => Symtab.update ((name, loc), locs)));
wenzelm@12063
   140
fun get_locale thy name = Symtab.lookup (#2 (LocalesData.get thy), name);
wenzelm@11896
   141
wenzelm@12014
   142
fun the_locale thy name =
wenzelm@12014
   143
  (case get_locale thy name of
wenzelm@12014
   144
    Some loc => loc
wenzelm@12014
   145
  | None => error ("Unknown locale " ^ quote name));
wenzelm@11896
   146
wenzelm@12046
   147
wenzelm@12277
   148
(* diagnostics *)
wenzelm@12273
   149
wenzelm@12277
   150
fun err_in_locale ctxt msg ids =
wenzelm@12277
   151
  let
wenzelm@12529
   152
    val sign = ProofContext.sign_of ctxt;
wenzelm@12529
   153
    fun prt_id (name, parms) =
wenzelm@12529
   154
      [Pretty.block (Pretty.breaks (map Pretty.str (cond_extern sign name :: parms)))];
wenzelm@12289
   155
    val prt_ids = flat (separate [Pretty.str " +", Pretty.brk 1] (map prt_id ids));
wenzelm@12502
   156
    val err_msg =
wenzelm@12529
   157
      if forall (equal "" o #1) ids then msg
wenzelm@12502
   158
      else msg ^ "\n" ^ Pretty.string_of (Pretty.block
wenzelm@12502
   159
        (Pretty.str "The error(s) above occurred in locale:" :: Pretty.brk 1 :: prt_ids));
wenzelm@12502
   160
  in raise ProofContext.CONTEXT (err_msg, ctxt) end;
wenzelm@12063
   161
wenzelm@12277
   162
wenzelm@12277
   163
wenzelm@12529
   164
(** primitives **)
wenzelm@12046
   165
wenzelm@12277
   166
(* renaming *)
wenzelm@12263
   167
wenzelm@12263
   168
fun rename ren x = if_none (assoc_string (ren, x)) x;
wenzelm@12263
   169
wenzelm@12263
   170
fun rename_term ren (Free (x, T)) = Free (rename ren x, T)
wenzelm@12263
   171
  | rename_term ren (t $ u) = rename_term ren t $ rename_term ren u
wenzelm@12263
   172
  | rename_term ren (Abs (x, T, t)) = Abs (x, T, rename_term ren t)
wenzelm@12263
   173
  | rename_term _ a = a;
wenzelm@12263
   174
wenzelm@12263
   175
fun rename_thm ren th =
wenzelm@12263
   176
  let
wenzelm@12263
   177
    val {sign, hyps, prop, maxidx, ...} = Thm.rep_thm th;
wenzelm@12263
   178
    val cert = Thm.cterm_of sign;
wenzelm@12502
   179
    val (xs, Ts) = Library.split_list (foldl Term.add_frees ([], prop :: hyps));
wenzelm@12263
   180
    val xs' = map (rename ren) xs;
wenzelm@12263
   181
    fun cert_frees names = map (cert o Free) (names ~~ Ts);
wenzelm@12263
   182
    fun cert_vars names = map (cert o Var o apfst (rpair (maxidx + 1))) (names ~~ Ts);
wenzelm@12263
   183
  in
wenzelm@12263
   184
    if xs = xs' then th
wenzelm@12263
   185
    else
wenzelm@12263
   186
      th
wenzelm@12263
   187
      |> Drule.implies_intr_list (map cert hyps)
wenzelm@12263
   188
      |> Drule.forall_intr_list (cert_frees xs)
wenzelm@12263
   189
      |> Drule.forall_elim_list (cert_vars xs)
wenzelm@12263
   190
      |> Thm.instantiate ([], cert_vars xs ~~ cert_frees xs')
wenzelm@12263
   191
      |> (fn th' => Drule.implies_elim_list th' (map (Thm.assume o cert o rename_term ren) hyps))
wenzelm@12263
   192
  end;
wenzelm@12263
   193
wenzelm@12529
   194
fun rename_elem ren (Fixes fixes) = Fixes (fixes |> map (fn (x, T, mx) =>
wenzelm@12529
   195
      let val x' = rename ren x in
wenzelm@12529
   196
        if x = x' then (x, T, mx)
wenzelm@12529
   197
        else (x', T, if mx = None then mx else Some Syntax.NoSyn)    (*drop syntax*)
wenzelm@12529
   198
      end))
wenzelm@12263
   199
  | rename_elem ren (Assumes asms) = Assumes (map (apsnd (map (fn (t, (ps, qs)) =>
wenzelm@12263
   200
      (rename_term ren t, (map (rename_term ren) ps, map (rename_term ren) qs))))) asms)
wenzelm@12263
   201
  | rename_elem ren (Defines defs) = Defines (map (apsnd (fn (t, ps) =>
wenzelm@12263
   202
      (rename_term ren t, map (rename_term ren) ps))) defs)
wenzelm@12273
   203
  | rename_elem ren (Notes facts) = Notes (map (apsnd (map (apfst (map (rename_thm ren))))) facts);
wenzelm@12263
   204
wenzelm@12839
   205
fun rename_text ren ((xs, body), (ys, defs)) =
wenzelm@12839
   206
  ((map (apfst (rename ren)) xs, map (rename_term ren) body),
wenzelm@12839
   207
   (map (apfst (rename ren)) ys, map (pairself (rename_term ren)) defs));
wenzelm@12839
   208
wenzelm@12529
   209
fun rename_facts prfx elem =
wenzelm@12307
   210
  let
wenzelm@12323
   211
    fun qualify (arg as ((name, atts), x)) =
wenzelm@12323
   212
      if name = "" then arg
wenzelm@12323
   213
      else ((NameSpace.pack (filter_out (equal "") (prfx @ [name])), atts), x);
wenzelm@12307
   214
  in
wenzelm@12307
   215
    (case elem of
wenzelm@12307
   216
      Fixes fixes => Fixes fixes
wenzelm@12307
   217
    | Assumes asms => Assumes (map qualify asms)
wenzelm@12307
   218
    | Defines defs => Defines (map qualify defs)
wenzelm@12307
   219
    | Notes facts => Notes (map qualify facts))
wenzelm@12307
   220
  end;
wenzelm@12307
   221
wenzelm@12263
   222
wenzelm@12502
   223
(* type instantiation *)
wenzelm@12502
   224
wenzelm@12502
   225
fun inst_type [] T = T
wenzelm@12502
   226
  | inst_type env T = Term.map_type_tfree (fn v => if_none (assoc (env, v)) (TFree v)) T;
wenzelm@12502
   227
wenzelm@12502
   228
fun inst_term [] t = t
wenzelm@12502
   229
  | inst_term env t = Term.map_term_types (inst_type env) t;
wenzelm@12502
   230
wenzelm@13211
   231
fun inst_thm _ [] th = th
wenzelm@13211
   232
  | inst_thm ctxt env th =
wenzelm@12502
   233
      let
wenzelm@13211
   234
        val sign = ProofContext.sign_of ctxt;
wenzelm@12575
   235
        val cert = Thm.cterm_of sign;
wenzelm@12575
   236
        val certT = Thm.ctyp_of sign;
wenzelm@13211
   237
        val {hyps, prop, maxidx, ...} = Thm.rep_thm th;
wenzelm@12579
   238
        val tfrees = foldr Term.add_term_tfree_names (prop :: hyps, []);
wenzelm@12579
   239
        val env' = filter (fn ((a, _), _) => a mem_string tfrees) env;
wenzelm@12502
   240
      in
wenzelm@12502
   241
        if null env' then th
wenzelm@12502
   242
        else
wenzelm@12502
   243
          th
wenzelm@12502
   244
          |> Drule.implies_intr_list (map cert hyps)
wenzelm@12575
   245
          |> Drule.tvars_intr_list (map (#1 o #1) env')
wenzelm@12502
   246
          |> (fn (th', al) => th' |>
wenzelm@12502
   247
            Thm.instantiate ((map (fn ((a, _), T) => (the (assoc (al, a)), certT T)) env'), []))
wenzelm@12502
   248
          |> (fn th'' => Drule.implies_elim_list th''
wenzelm@12502
   249
              (map (Thm.assume o cert o inst_term env') hyps))
wenzelm@12502
   250
      end;
wenzelm@12502
   251
wenzelm@13211
   252
fun inst_elem _ env (Fixes fixes) =
wenzelm@12502
   253
      Fixes (map (fn (x, T, mx) => (x, apsome (inst_type env) T, mx)) fixes)
wenzelm@13211
   254
  | inst_elem _ env (Assumes asms) = Assumes (map (apsnd (map (fn (t, (ps, qs)) =>
wenzelm@12502
   255
      (inst_term env t, (map (inst_term env) ps, map (inst_term env) qs))))) asms)
wenzelm@13211
   256
  | inst_elem _ env (Defines defs) = Defines (map (apsnd (fn (t, ps) =>
wenzelm@12502
   257
      (inst_term env t, map (inst_term env) ps))) defs)
wenzelm@13211
   258
  | inst_elem ctxt env (Notes facts) =
wenzelm@13211
   259
      Notes (map (apsnd (map (apfst (map (inst_thm ctxt env))))) facts);
wenzelm@12502
   260
wenzelm@12839
   261
fun inst_text env ((xs, body), (ys, defs)) =
wenzelm@12839
   262
 ((map (apsnd (inst_type env)) xs, map (inst_term env) body),
wenzelm@12839
   263
  (map (apsnd (inst_type env)) ys, map (pairself (inst_term env)) defs));
wenzelm@12839
   264
wenzelm@12502
   265
wenzelm@12529
   266
wenzelm@12529
   267
(** structured contexts: rename + merge + implicit type instantiation **)
wenzelm@12529
   268
wenzelm@12529
   269
(* parameter types *)
wenzelm@12529
   270
wenzelm@12529
   271
fun frozen_tvars ctxt Ts =
wenzelm@12529
   272
  let
wenzelm@12529
   273
    val tvars = rev (foldl Term.add_tvarsT ([], Ts));
wenzelm@12529
   274
    val tfrees = map TFree
wenzelm@12529
   275
      (Term.invent_type_names (ProofContext.used_types ctxt) (length tvars) ~~ map #2 tvars);
wenzelm@12529
   276
  in map #1 tvars ~~ tfrees end;
wenzelm@12529
   277
wenzelm@12529
   278
fun unify_frozen ctxt maxidx Ts Us =
wenzelm@12529
   279
  let
wenzelm@12529
   280
    val tsig = Sign.tsig_of (ProofContext.sign_of ctxt);
wenzelm@12546
   281
    fun unify (env, (Some T, Some U)) = (Type.unify tsig env (U, T)
wenzelm@12546
   282
          handle Type.TUNIFY => raise TYPE ("unify_frozen: failed to unify types", [U, T], []))
wenzelm@12529
   283
      | unify (env, _) = env;
wenzelm@12529
   284
    fun paramify (i, None) = (i, None)
wenzelm@12529
   285
      | paramify (i, Some T) = apsnd Some (Type.paramify_dummies (i, T));
wenzelm@12529
   286
wenzelm@12529
   287
    val (maxidx', Ts') = foldl_map paramify (maxidx, Ts);
wenzelm@12727
   288
    val (maxidx'', Us') = foldl_map paramify (maxidx', Us);
wenzelm@12529
   289
    val (unifier, _) = foldl unify ((Vartab.empty, maxidx''), Ts' ~~ Us');
wenzelm@12529
   290
    val Vs = map (apsome (Envir.norm_type unifier)) Us';
wenzelm@12529
   291
    val unifier' = Vartab.extend (unifier, frozen_tvars ctxt (mapfilter I Vs));
wenzelm@12532
   292
  in map (apsome (Envir.norm_type unifier')) Vs end;
wenzelm@12529
   293
wenzelm@12730
   294
fun params_of elemss = gen_distinct eq_fst (flat (map (snd o fst) elemss));
wenzelm@12529
   295
fun param_types ps = mapfilter (fn (_, None) => None | (x, Some T) => Some (x, T)) ps;
wenzelm@12529
   296
wenzelm@12529
   297
wenzelm@12529
   298
(* flatten expressions *)
wenzelm@11896
   299
wenzelm@12510
   300
local
wenzelm@12502
   301
wenzelm@12529
   302
fun unique_parms ctxt elemss =
wenzelm@12529
   303
  let
wenzelm@12529
   304
    val param_decls =
wenzelm@12529
   305
      flat (map (fn ((name, (ps, qs)), _) => map (rpair (name, ps)) qs) elemss)
wenzelm@12529
   306
      |> Symtab.make_multi |> Symtab.dest;
wenzelm@12529
   307
  in
wenzelm@12529
   308
    (case find_first (fn (_, ids) => length ids > 1) param_decls of
wenzelm@12529
   309
      Some (q, ids) => err_in_locale ctxt ("Multiple declaration of parameter " ^ quote q)
wenzelm@12529
   310
          (map (apsnd (map fst)) ids)
wenzelm@12529
   311
    | None => map (apfst (apsnd #1)) elemss)
wenzelm@12529
   312
  end;
wenzelm@12529
   313
wenzelm@12529
   314
fun unify_parms ctxt fixed_parms raw_parmss =
wenzelm@12502
   315
  let
wenzelm@12502
   316
    val tsig = Sign.tsig_of (ProofContext.sign_of ctxt);
wenzelm@12502
   317
    val maxidx = length raw_parmss;
wenzelm@12502
   318
    val idx_parmss = (0 upto maxidx - 1) ~~ raw_parmss;
wenzelm@12502
   319
wenzelm@12502
   320
    fun varify i = Term.map_type_tfree (fn (a, S) => TVar ((a, i), S));
wenzelm@12529
   321
    fun varify_parms (i, ps) = map (apsnd (varify i)) (param_types ps);
wenzelm@12529
   322
    val parms = fixed_parms @ flat (map varify_parms idx_parmss);
wenzelm@12502
   323
wenzelm@12529
   324
    fun unify T ((env, maxidx), U) = Type.unify tsig (env, maxidx) (U, T)
wenzelm@12529
   325
      handle Type.TUNIFY => raise TYPE ("unify_parms: failed to unify types", [U, T], []);
wenzelm@12502
   326
    fun unify_list (envir, T :: Us) = foldl (unify T) (envir, Us)
wenzelm@12502
   327
      | unify_list (envir, []) = envir;
wenzelm@12502
   328
    val (unifier, _) = foldl unify_list
wenzelm@12502
   329
      ((Vartab.empty, maxidx), map #2 (Symtab.dest (Symtab.make_multi parms)));
wenzelm@12502
   330
wenzelm@12502
   331
    val parms' = map (apsnd (Envir.norm_type unifier)) (gen_distinct eq_fst parms);
wenzelm@12502
   332
    val unifier' = Vartab.extend (unifier, frozen_tvars ctxt (map #2 parms'));
wenzelm@12502
   333
wenzelm@12502
   334
    fun inst_parms (i, ps) =
wenzelm@12502
   335
      foldr Term.add_typ_tfrees (mapfilter snd ps, [])
wenzelm@12502
   336
      |> mapfilter (fn (a, S) =>
wenzelm@12502
   337
          let val T = Envir.norm_type unifier' (TVar ((a, i), S))
wenzelm@12502
   338
          in if T = TFree (a, S) then None else Some ((a, S), T) end);
wenzelm@12502
   339
  in map inst_parms idx_parmss end;
wenzelm@12502
   340
wenzelm@12529
   341
in
wenzelm@12502
   342
wenzelm@12529
   343
fun unify_elemss _ _ [] = []
wenzelm@12529
   344
  | unify_elemss _ [] [elems] = [elems]
wenzelm@12529
   345
  | unify_elemss ctxt fixed_parms elemss =
wenzelm@12502
   346
      let
wenzelm@12529
   347
        val envs = unify_parms ctxt fixed_parms (map (#2 o #1) elemss);
wenzelm@12839
   348
        fun inst (((name, ps), (elems, text)), env) =
wenzelm@12839
   349
          ((name, map (apsnd (apsome (inst_type env))) ps),
wenzelm@13211
   350
            (map (inst_elem ctxt env) elems, inst_text env text));
wenzelm@12839
   351
      in map inst (elemss ~~ envs) end;
wenzelm@12502
   352
wenzelm@12575
   353
fun flatten_expr ctxt (prev_idents, expr) =
wenzelm@12014
   354
  let
wenzelm@12014
   355
    val thy = ProofContext.theory_of ctxt;
wenzelm@12263
   356
wenzelm@12289
   357
    fun renaming (Some x :: xs) (y :: ys) = (y, x) :: renaming xs ys
wenzelm@12289
   358
      | renaming (None :: xs) (y :: ys) = renaming xs ys
wenzelm@12273
   359
      | renaming [] _ = []
wenzelm@12289
   360
      | renaming xs [] = raise ERROR_MESSAGE ("Too many arguments in renaming: " ^
wenzelm@12289
   361
          commas (map (fn None => "_" | Some x => quote x) xs));
wenzelm@12289
   362
wenzelm@12289
   363
    fun rename_parms ren (name, ps) =
wenzelm@12289
   364
      let val ps' = map (rename ren) ps in
wenzelm@12289
   365
        (case duplicates ps' of [] => (name, ps')
wenzelm@12289
   366
        | dups => err_in_locale ctxt ("Duplicate parameters: " ^ commas_quote dups) [(name, ps')])
wenzelm@12289
   367
      end;
wenzelm@12263
   368
wenzelm@12273
   369
    fun identify ((ids, parms), Locale name) =
wenzelm@12289
   370
          let
wenzelm@12289
   371
            val {import, params, ...} = the_locale thy name;
wenzelm@12289
   372
            val ps = map #1 (#1 params);
wenzelm@12289
   373
          in
wenzelm@12273
   374
            if (name, ps) mem ids then (ids, parms)
wenzelm@12277
   375
            else
wenzelm@12289
   376
              let val (ids', parms') = identify ((ids, parms), import);  (*acyclic dependencies!*)
wenzelm@12289
   377
              in (ids' @ [(name, ps)], merge_lists parms' ps) end
wenzelm@12273
   378
          end
wenzelm@12273
   379
      | identify ((ids, parms), Rename (e, xs)) =
wenzelm@12273
   380
          let
wenzelm@12273
   381
            val (ids', parms') = identify (([], []), e);
wenzelm@12839
   382
            val ren = renaming xs parms'
wenzelm@12839
   383
              handle ERROR_MESSAGE msg => err_in_locale ctxt msg ids';
wenzelm@12289
   384
            val ids'' = distinct (map (rename_parms ren) ids');
wenzelm@12289
   385
            val parms'' = distinct (flat (map #2 ids''));
wenzelm@12289
   386
          in (merge_lists ids ids'', merge_lists parms parms'') end
wenzelm@12273
   387
      | identify (arg, Merge es) = foldl identify (arg, es);
wenzelm@12014
   388
wenzelm@12307
   389
    fun eval (name, xs) =
wenzelm@12273
   390
      let
wenzelm@12839
   391
        val {params = (ps, qs), elems, text, ...} = the_locale thy name;
wenzelm@12307
   392
        val ren = filter_out (op =) (map #1 ps ~~ xs);
wenzelm@12839
   393
        val (params', elems', text') =
wenzelm@12839
   394
          if null ren then ((ps, qs), map #1 elems, text)
wenzelm@12502
   395
          else ((map (apfst (rename ren)) ps, map (rename ren) qs),
wenzelm@12839
   396
            map (rename_elem ren o #1) elems, rename_text ren text);
wenzelm@12529
   397
        val elems'' = map (rename_facts [NameSpace.base name, space_implode "_" xs]) elems';
wenzelm@12839
   398
      in ((name, params'), (elems'', text')) end;
wenzelm@12307
   399
wenzelm@12575
   400
    val idents = gen_rems (op =) (#1 (identify (([], []), expr)), prev_idents);
wenzelm@12575
   401
    val raw_elemss = unique_parms ctxt (map eval idents);
wenzelm@12529
   402
    val elemss = unify_elemss ctxt [] raw_elemss;
wenzelm@12575
   403
  in (prev_idents @ idents, elemss) end;
wenzelm@12046
   404
wenzelm@12510
   405
end;
wenzelm@12510
   406
wenzelm@12070
   407
wenzelm@12529
   408
(* activate elements *)
wenzelm@12273
   409
wenzelm@12510
   410
local
wenzelm@12510
   411
wenzelm@12839
   412
fun activate_elem (ctxt, Fixes fixes) =
wenzelm@12839
   413
      (ctxt |> ProofContext.add_fixes fixes, [])
wenzelm@12834
   414
  | activate_elem (ctxt, Assumes asms) =
wenzelm@12834
   415
      ctxt |> ProofContext.fix_frees (flat (map (map #1 o #2) asms))
wenzelm@12839
   416
      |> ProofContext.assume_i ProofContext.export_assume asms
wenzelm@12839
   417
      |> apsnd (map (pair Assume))
wenzelm@12834
   418
  | activate_elem (ctxt, Defines defs) =
wenzelm@12834
   419
      ctxt |> ProofContext.assume_i ProofContext.export_def
wenzelm@12834
   420
        (map (fn ((name, atts), (t, ps)) =>
wenzelm@12834
   421
          let val (c, t') = ProofContext.cert_def ctxt t
wenzelm@12834
   422
          in ((if name = "" then Thm.def_name c else name, atts), [(t', (ps, []))]) end) defs)
wenzelm@12839
   423
      |> apsnd (map (pair Define))
wenzelm@12839
   424
  | activate_elem (ctxt, Notes facts) =
wenzelm@12839
   425
      ctxt |> ProofContext.have_thmss_i facts
wenzelm@12839
   426
      |> apsnd (map (pair Note));
wenzelm@12263
   427
wenzelm@12834
   428
fun activate_elems ((name, ps), elems) = ProofContext.qualified_result (fn ctxt =>
wenzelm@12834
   429
  foldl_map activate_elem (ctxt, elems) handle ProofContext.CONTEXT (msg, ctxt) =>
wenzelm@12529
   430
    err_in_locale ctxt msg [(name, map fst ps)]);
wenzelm@12502
   431
wenzelm@12834
   432
fun activate_elemss prep_facts = foldl_map (fn (ctxt, ((name, ps), raw_elems)) =>
wenzelm@12834
   433
  let
wenzelm@12834
   434
    val elems = map (prep_facts ctxt) raw_elems;
wenzelm@12834
   435
    val res = ((name, ps), elems);
wenzelm@12834
   436
    val (ctxt', facts) = apsnd flat (activate_elems res ctxt);
wenzelm@12834
   437
  in (ctxt', (res, facts)) end);
wenzelm@12834
   438
wenzelm@12546
   439
in
wenzelm@12546
   440
wenzelm@12834
   441
fun activate_facts prep_facts ctxt_elemss =
wenzelm@12834
   442
  let val (ctxt', (elemss', factss)) = apsnd split_list (activate_elemss prep_facts ctxt_elemss)
wenzelm@12834
   443
  in (ctxt', (elemss', flat factss)) end;
wenzelm@12546
   444
wenzelm@12958
   445
fun apply_facts name (ctxt, facts) =
wenzelm@12958
   446
  let val (ctxt', (_, facts')) = activate_facts (K I) (ctxt, [((name, []), [Notes facts])])
wenzelm@12958
   447
  in (ctxt', map (#2 o #2) facts') end;
wenzelm@12958
   448
wenzelm@12510
   449
end;
wenzelm@12510
   450
wenzelm@12307
   451
wenzelm@12529
   452
wenzelm@12529
   453
(** prepare context elements **)
wenzelm@12529
   454
wenzelm@12529
   455
(* expressions *)
wenzelm@12529
   456
wenzelm@12529
   457
fun intern_expr sg (Locale xname) = Locale (intern sg xname)
wenzelm@12529
   458
  | intern_expr sg (Merge exprs) = Merge (map (intern_expr sg) exprs)
wenzelm@12529
   459
  | intern_expr sg (Rename (expr, xs)) = Rename (intern_expr sg expr, xs);
wenzelm@12529
   460
wenzelm@12529
   461
wenzelm@12546
   462
(* attributes *)
wenzelm@12546
   463
wenzelm@12546
   464
local fun read_att attrib (x, srcs) = (x, map attrib srcs) in
wenzelm@12546
   465
wenzelm@12546
   466
fun attribute _ (Elem (Fixes fixes)) = Elem (Fixes fixes)
wenzelm@12546
   467
  | attribute attrib (Elem (Assumes asms)) = Elem (Assumes (map (apfst (read_att attrib)) asms))
wenzelm@12546
   468
  | attribute attrib (Elem (Defines defs)) = Elem (Defines (map (apfst (read_att attrib)) defs))
wenzelm@12546
   469
  | attribute attrib (Elem (Notes facts)) =
wenzelm@12546
   470
      Elem (Notes (map (apfst (read_att attrib) o apsnd (map (read_att attrib))) facts))
wenzelm@12546
   471
  | attribute _ (Expr expr) = Expr expr;
wenzelm@12546
   472
wenzelm@12546
   473
end;
wenzelm@12546
   474
wenzelm@12546
   475
wenzelm@12529
   476
(* parameters *)
wenzelm@12502
   477
wenzelm@12502
   478
local
wenzelm@12502
   479
wenzelm@12529
   480
fun prep_fixes prep_vars ctxt fixes =
wenzelm@12529
   481
  let val vars = snd (foldl_map prep_vars (ctxt, map (fn (x, T, _) => ([x], T)) fixes))
wenzelm@12529
   482
  in map (fn (([x'], T'), (_, _, mx)) => (x', T', mx)) (vars ~~ fixes) end;
wenzelm@12529
   483
wenzelm@12529
   484
in
wenzelm@12529
   485
wenzelm@12529
   486
fun read_fixes x = prep_fixes ProofContext.read_vars x;
wenzelm@12529
   487
fun cert_fixes x = prep_fixes ProofContext.cert_vars x;
wenzelm@12529
   488
wenzelm@12529
   489
end;
wenzelm@12529
   490
wenzelm@12529
   491
wenzelm@12529
   492
(* propositions and bindings *)
wenzelm@12529
   493
wenzelm@12529
   494
datatype ('a, 'b) int_ext = Int of 'a | Ext of 'b;
wenzelm@12529
   495
wenzelm@12529
   496
local
wenzelm@12529
   497
wenzelm@12839
   498
local
wenzelm@12839
   499
wenzelm@12727
   500
fun declare_int_elem (ctxt, Fixes fixes) =
wenzelm@12575
   501
      (ctxt |> ProofContext.add_fixes (map (fn (x, T, mx) =>
wenzelm@12727
   502
        (x, apsome (Term.map_type_tfree (Type.param 0)) T, mx)) fixes), [])
wenzelm@12727
   503
  | declare_int_elem (ctxt, _) = (ctxt, []);
wenzelm@12529
   504
wenzelm@12529
   505
fun declare_ext_elem prep_fixes (ctxt, Fixes fixes) =
wenzelm@12575
   506
      (ctxt |> ProofContext.add_fixes (prep_fixes ctxt fixes), [])
wenzelm@12529
   507
  | declare_ext_elem _ (ctxt, Assumes asms) = (ctxt, map #2 asms)
wenzelm@12529
   508
  | declare_ext_elem _ (ctxt, Defines defs) = (ctxt, map (fn (_, (t, ps)) => [(t, (ps, []))]) defs)
wenzelm@12529
   509
  | declare_ext_elem _ (ctxt, Notes facts) = (ctxt, []);
wenzelm@12529
   510
wenzelm@12727
   511
fun declare_elems prep_fixes (ctxt, ((name, ps), elems)) =
wenzelm@12529
   512
  let val (ctxt', propps) =
wenzelm@12529
   513
    (case elems of
wenzelm@12839
   514
      Int (es, _) => foldl_map declare_int_elem (ctxt, es)
wenzelm@12546
   515
    | Ext e => foldl_map (declare_ext_elem prep_fixes) (ctxt, [e]))
wenzelm@12529
   516
    handle ProofContext.CONTEXT (msg, ctxt) => err_in_locale ctxt msg [(name, map fst ps)]
wenzelm@12727
   517
  in (ctxt', propps) end;
wenzelm@12727
   518
wenzelm@12839
   519
in
wenzelm@12839
   520
wenzelm@12727
   521
fun declare_elemss prep_fixes fixed_params raw_elemss ctxt =
wenzelm@12727
   522
  let
wenzelm@12727
   523
    val int_elemss =
wenzelm@12727
   524
      raw_elemss
wenzelm@12727
   525
      |> mapfilter (fn (id, Int es) => Some (id, es) | _ => None)
wenzelm@12727
   526
      |> unify_elemss ctxt fixed_params;
wenzelm@12727
   527
    val (_, raw_elemss') =
wenzelm@12727
   528
      foldl_map (fn ((_, es) :: elemss, (id, Int _)) => (elemss, (id, Int es)) | x => x)
wenzelm@12727
   529
        (int_elemss, raw_elemss);
wenzelm@12727
   530
  in foldl_map (declare_elems prep_fixes) (ctxt, raw_elemss') end;
wenzelm@12529
   531
wenzelm@12839
   532
end;
wenzelm@12529
   533
wenzelm@12839
   534
val empty_text = (([], []), ([], []));
wenzelm@12839
   535
wenzelm@12839
   536
fun merge_text (((xs1, spec1), (ys1, env1)), ((xs2, spec2), (ys2, env2))) =
wenzelm@12839
   537
  ((gen_merge_lists eq_fst xs1 xs2, spec1 @ spec2), (ys1 @ ys2, env1 @ env2));
wenzelm@12839
   538
wenzelm@12839
   539
local
wenzelm@12839
   540
wenzelm@12839
   541
val norm_term = Envir.beta_norm oo Term.subst_atomic;
wenzelm@12839
   542
wenzelm@12839
   543
fun abstract_def eq =    (*assumes well-formedness according to ProofContext.cert_def*)
wenzelm@12839
   544
  let
wenzelm@12839
   545
    val body = Term.strip_all_body eq;
wenzelm@12839
   546
    val vars = map Free (Term.rename_wrt_term body (Term.strip_all_vars eq));
wenzelm@12839
   547
    val (lhs, rhs) = Logic.dest_equals (Term.subst_bounds (vars, body));
wenzelm@12839
   548
    val (f, xs) = Term.strip_comb lhs;
wenzelm@12839
   549
  in (Term.dest_Free f, Term.list_abs_free (map Term.dest_Free xs, rhs)) end;
wenzelm@12502
   550
wenzelm@12839
   551
fun bind_def ctxt (name, ps) ((all_text, text), eq) =
wenzelm@12839
   552
  let
wenzelm@12839
   553
    val ((all_xs, _), (all_ys, all_env)) = all_text;
wenzelm@12839
   554
    val (y, b) = abstract_def eq;
wenzelm@12839
   555
    val b' = norm_term all_env b;
wenzelm@12839
   556
    val txt = ((Term.add_frees ([], b'), []), ([y], [(Free y, b')]));
wenzelm@12839
   557
    fun err msg = err_in_locale ctxt (msg ^ ": " ^ quote (#1 y)) [(name, map fst ps)];
wenzelm@12839
   558
  in
wenzelm@12839
   559
    conditional (y mem all_xs) (fn () => err "Attempt to define previously specified variable");
wenzelm@12839
   560
    conditional (y mem all_ys) (fn () => err "Attempt to redefine variable");
wenzelm@12839
   561
    (merge_text (all_text, txt), merge_text (text, txt))
wenzelm@12839
   562
  end;
wenzelm@12575
   563
wenzelm@12839
   564
fun eval_text _ _ (all_text, Fixes _) = (all_text, empty_text)
wenzelm@12839
   565
  | eval_text _ _ (all_text, Assumes asms) =
wenzelm@12839
   566
      let
wenzelm@12839
   567
        val ts = map (norm_term (#2 (#2 all_text))) (flat (map (map #1 o #2) asms));
wenzelm@12839
   568
        val txt = ((foldl Term.add_frees ([], ts), ts), ([], []));
wenzelm@12839
   569
      in (merge_text (all_text, txt), txt) end
wenzelm@12839
   570
  | eval_text ctxt id (all_text, Defines defs) =
wenzelm@12839
   571
      foldl (bind_def ctxt id) ((all_text, empty_text), map (#1 o #2) defs)
wenzelm@12839
   572
  | eval_text _ _ (all_text, Notes _) = (all_text, empty_text);
wenzelm@12839
   573
wenzelm@12502
   574
wenzelm@12839
   575
fun finish_ext_elem parms _ (Fixes fixes, _) = Fixes (map (fn (x, _, mx) =>
wenzelm@12727
   576
      (x, assoc_string (parms, x), mx)) fixes)
wenzelm@12839
   577
  | finish_ext_elem _ close (Assumes asms, propp) =
wenzelm@12839
   578
      close (Assumes (map #1 asms ~~ propp))
wenzelm@12839
   579
  | finish_ext_elem _ close (Defines defs, propp) =
wenzelm@12727
   580
      close (Defines (map #1 defs ~~ map (fn [(t, (ps, []))] => (t, ps)) propp))
wenzelm@12839
   581
  | finish_ext_elem _ _ (Notes facts, _) = Notes facts;
wenzelm@12839
   582
wenzelm@12839
   583
fun finish_parms parms ((name, ps), elems) =
wenzelm@12839
   584
  ((name, map (fn (x, _) => (x, assoc (parms, x))) ps), elems);
wenzelm@12839
   585
wenzelm@12839
   586
fun finish_elems ctxt parms _ ((all_text, int_text, ext_text), ((id, Int e), _)) =
wenzelm@12839
   587
      let val [(_, (es, txt))] = unify_elemss ctxt parms [(id, e)]
wenzelm@12839
   588
      in ((merge_text (all_text, txt), merge_text (int_text, txt), ext_text), (id, map Int es)) end
wenzelm@12839
   589
  | finish_elems ctxt parms close ((all_text, int_text, ext_text), ((id, Ext e), [propp])) =
wenzelm@12839
   590
      let
wenzelm@12839
   591
        val e' = finish_ext_elem parms close (e, propp);
wenzelm@12839
   592
        val (all_text', txt) = eval_text ctxt id (all_text, e');
wenzelm@12839
   593
      in ((all_text', int_text, merge_text (ext_text, txt)), (id, [Ext e'])) end;
wenzelm@12839
   594
wenzelm@12839
   595
in
wenzelm@12510
   596
wenzelm@12839
   597
fun finish_elemss ctxt parms close =
wenzelm@12839
   598
  foldl_map (apsnd (finish_parms parms) o finish_elems ctxt parms close);
wenzelm@12839
   599
wenzelm@12839
   600
end;
wenzelm@12839
   601
wenzelm@12839
   602
fun closeup ctxt elem =
wenzelm@12502
   603
  let
wenzelm@12839
   604
    fun close_frees t =
wenzelm@12839
   605
      let val frees = rev (filter_out (ProofContext.is_fixed ctxt o #1) (Term.add_frees ([], t)))
wenzelm@12839
   606
      in Term.list_all_free (frees, t) end;
wenzelm@12510
   607
wenzelm@12839
   608
    fun no_binds ps =
wenzelm@12839
   609
      if null ps then ps
wenzelm@12839
   610
      else raise ProofContext.CONTEXT ("Illegal term bindings in locale element", ctxt);
wenzelm@12839
   611
  in
wenzelm@12839
   612
    (case elem of
wenzelm@12839
   613
      Assumes asms => Assumes (asms |> map (fn (a, propps) =>
wenzelm@12839
   614
        (a, map (fn (t, (ps, qs)) => (close_frees t, (no_binds ps, no_binds qs))) propps)))
wenzelm@12839
   615
    | Defines defs => Defines (defs |> map (fn (a, (t, ps)) =>
wenzelm@12839
   616
        (a, (close_frees (#2 (ProofContext.cert_def ctxt t)), no_binds ps))))
wenzelm@12839
   617
    | e => e)
wenzelm@12839
   618
  end;
wenzelm@12277
   619
wenzelm@12529
   620
fun prep_elemss prep_fixes prepp do_close context fixed_params raw_elemss raw_concl =
wenzelm@12529
   621
  let
wenzelm@12727
   622
    val (raw_ctxt, raw_proppss) = declare_elemss prep_fixes fixed_params raw_elemss context;
wenzelm@12529
   623
    val raw_propps = map flat raw_proppss;
wenzelm@12529
   624
    val raw_propp = flat raw_propps;
wenzelm@12529
   625
    val (ctxt, all_propp) =
wenzelm@12529
   626
      prepp (ProofContext.declare_terms (map Free fixed_params) raw_ctxt, raw_concl @ raw_propp);
wenzelm@12529
   627
    val ctxt = ProofContext.declare_terms (flat (map (map fst) all_propp)) ctxt;
wenzelm@12502
   628
wenzelm@12529
   629
    val all_propp' = map2 (op ~~)
wenzelm@12529
   630
      (#1 (#2 (ProofContext.bind_propp_schematic_i (ctxt, all_propp))), map (map snd) all_propp);
wenzelm@12529
   631
    val n = length raw_concl;
wenzelm@12529
   632
    val concl = take (n, all_propp');
wenzelm@12529
   633
    val propp = drop (n, all_propp');
wenzelm@12529
   634
    val propps = unflat raw_propps propp;
wenzelm@12839
   635
    val proppss = map (uncurry unflat) (raw_proppss ~~ propps);
wenzelm@12502
   636
wenzelm@12529
   637
    val xs = map #1 (params_of raw_elemss);
wenzelm@12727
   638
    val typing = unify_frozen ctxt 0
wenzelm@12529
   639
      (map (ProofContext.default_type raw_ctxt) xs)
wenzelm@12529
   640
      (map (ProofContext.default_type ctxt) xs);
wenzelm@12529
   641
    val parms = param_types (xs ~~ typing);
wenzelm@12273
   642
wenzelm@12529
   643
    val close = if do_close then closeup ctxt else I;
wenzelm@12839
   644
    val ((_, int_text, ext_text), elemss) = finish_elemss ctxt parms close
wenzelm@12839
   645
      ((empty_text, empty_text, empty_text), raw_elemss ~~ proppss);
wenzelm@12839
   646
  in (parms, elemss, (merge_text (int_text, ext_text), ext_text), concl) end;
wenzelm@12502
   647
wenzelm@12502
   648
in
wenzelm@12502
   649
wenzelm@12529
   650
fun read_elemss x = prep_elemss read_fixes ProofContext.read_propp_schematic x;
wenzelm@12529
   651
fun cert_elemss x = prep_elemss cert_fixes ProofContext.cert_propp_schematic x;
wenzelm@12529
   652
wenzelm@12529
   653
end;
wenzelm@12529
   654
wenzelm@12529
   655
wenzelm@12529
   656
(* facts *)
wenzelm@12529
   657
wenzelm@12529
   658
local
wenzelm@12529
   659
wenzelm@12529
   660
fun prep_name ctxt (name, atts) =
wenzelm@12529
   661
  if NameSpace.is_qualified name then
wenzelm@12529
   662
    raise ProofContext.CONTEXT ("Illegal qualified name: " ^ quote name, ctxt)
wenzelm@12529
   663
  else (name, atts);
wenzelm@12529
   664
wenzelm@12529
   665
fun prep_facts _ _ (Int elem) = elem
wenzelm@12529
   666
  | prep_facts _ _ (Ext (Fixes fixes)) = Fixes fixes
wenzelm@12529
   667
  | prep_facts _ ctxt (Ext (Assumes asms)) = Assumes (map (apfst (prep_name ctxt)) asms)
wenzelm@12529
   668
  | prep_facts _ ctxt (Ext (Defines defs)) = Defines (map (apfst (prep_name ctxt)) defs)
wenzelm@12529
   669
  | prep_facts get ctxt (Ext (Notes facts)) = Notes (facts |> map (fn (a, bs) =>
wenzelm@12529
   670
      (prep_name ctxt a, map (apfst (get ctxt)) bs)));
wenzelm@12529
   671
wenzelm@12529
   672
in
wenzelm@12529
   673
wenzelm@12529
   674
fun get_facts x = prep_facts ProofContext.get_thms x;
wenzelm@12529
   675
fun get_facts_i x = prep_facts (K I) x;
wenzelm@12529
   676
wenzelm@12529
   677
end;
wenzelm@12529
   678
wenzelm@12529
   679
wenzelm@12546
   680
(* full context statements: import + elements + conclusion *)
wenzelm@12529
   681
wenzelm@12529
   682
local
wenzelm@12529
   683
wenzelm@12529
   684
fun prep_context_statement prep_expr prep_elemss prep_facts
wenzelm@12529
   685
    close fixed_params import elements raw_concl context =
wenzelm@12529
   686
  let
wenzelm@12529
   687
    val sign = ProofContext.sign_of context;
wenzelm@12575
   688
    fun flatten (ids, Elem (Fixes fixes)) =
wenzelm@12575
   689
          (ids, [(("", map (rpair None o #1) fixes), Ext (Fixes fixes))])
wenzelm@12575
   690
      | flatten (ids, Elem elem) = (ids, [(("", []), Ext elem)])
wenzelm@12575
   691
      | flatten (ids, Expr expr) =
wenzelm@12575
   692
          let val (ids', elemss) = flatten_expr context (ids, prep_expr sign expr)
wenzelm@12575
   693
          in (ids', map (apsnd Int) elemss) end
wenzelm@12529
   694
wenzelm@12575
   695
    val (import_ids, raw_import_elemss) = flatten ([], Expr import);
wenzelm@12575
   696
    val raw_elemss = flat (#2 ((foldl_map flatten (import_ids, elements))));
wenzelm@12839
   697
    val (parms, all_elemss, texts, concl) =
wenzelm@12529
   698
      prep_elemss close context fixed_params (raw_import_elemss @ raw_elemss) raw_concl;
wenzelm@12529
   699
wenzelm@12529
   700
    val n = length raw_import_elemss;
wenzelm@12839
   701
    val (import_ctxt, (import_elemss, import_facts)) =
wenzelm@12839
   702
      activate_facts prep_facts (context, take (n, all_elemss));
wenzelm@12839
   703
    val (ctxt, (elemss, facts)) =
wenzelm@12839
   704
      activate_facts prep_facts (import_ctxt, drop (n, all_elemss));
wenzelm@12834
   705
  in
wenzelm@12834
   706
    ((((import_ctxt, (import_elemss, import_facts)),
wenzelm@12839
   707
      (ctxt, (elemss, facts))), texts), concl)
wenzelm@12834
   708
  end;
wenzelm@12529
   709
wenzelm@12529
   710
val gen_context = prep_context_statement intern_expr read_elemss get_facts;
wenzelm@12529
   711
val gen_context_i = prep_context_statement (K I) cert_elemss get_facts_i;
wenzelm@12529
   712
wenzelm@12834
   713
fun gen_facts prep_locale thy name =
wenzelm@12834
   714
  let val ((((_, (_, facts)), _), _), _) = thy |> ProofContext.init
wenzelm@12834
   715
    |> gen_context_i false [] (Locale (prep_locale (Theory.sign_of thy) name)) [] [];
wenzelm@12839
   716
  in flat (map (#2 o #2) facts) end;
wenzelm@12834
   717
wenzelm@12529
   718
fun gen_statement prep_locale prep_ctxt raw_locale elems concl ctxt =
wenzelm@12529
   719
  let
wenzelm@12529
   720
    val thy = ProofContext.theory_of ctxt;
wenzelm@12529
   721
    val locale = apsome (prep_locale (Theory.sign_of thy)) raw_locale;
wenzelm@12529
   722
    val (fixed_params, import) =
wenzelm@12529
   723
      (case locale of None => ([], empty)
wenzelm@12529
   724
      | Some name => (param_types (#1 (#params (the_locale thy name))), Locale name));
wenzelm@12730
   725
    val ((((locale_ctxt, _), (elems_ctxt, _)), _), concl') =
wenzelm@12529
   726
      prep_ctxt false fixed_params import elems concl ctxt;
wenzelm@12529
   727
  in (locale, locale_ctxt, elems_ctxt, concl') end;
wenzelm@12529
   728
wenzelm@12529
   729
in
wenzelm@12529
   730
wenzelm@12529
   731
fun read_context x y z = #1 (gen_context true [] x y [] z);
wenzelm@12529
   732
fun cert_context x y z = #1 (gen_context_i true [] x y [] z);
wenzelm@12834
   733
val locale_facts = gen_facts intern;
wenzelm@12834
   734
val locale_facts_i = gen_facts (K I);
wenzelm@12529
   735
val read_context_statement = gen_statement intern gen_context;
wenzelm@12529
   736
val cert_context_statement = gen_statement (K I) gen_context_i;
wenzelm@12502
   737
wenzelm@12502
   738
end;
wenzelm@11896
   739
wenzelm@11896
   740
wenzelm@11896
   741
wenzelm@12070
   742
(** print locale **)
wenzelm@12070
   743
wenzelm@12758
   744
fun print_locale thy import body =
wenzelm@12070
   745
  let
wenzelm@12070
   746
    val sg = Theory.sign_of thy;
wenzelm@12289
   747
    val thy_ctxt = ProofContext.init thy;
wenzelm@12839
   748
    val (((_, (import_elemss, _)), (ctxt, (elemss, _))), (((pred_xs, pred_ts), _), _)) =
wenzelm@12758
   749
      read_context import body thy_ctxt;
wenzelm@12758
   750
    val all_elems = flat (map #2 (import_elemss @ elemss));
wenzelm@12070
   751
wenzelm@12307
   752
    val prt_typ = Pretty.quote o ProofContext.pretty_typ ctxt;
wenzelm@12307
   753
    val prt_term = Pretty.quote o ProofContext.pretty_term ctxt;
wenzelm@12307
   754
    val prt_thm = Pretty.quote o ProofContext.pretty_thm ctxt;
wenzelm@12070
   755
wenzelm@12070
   756
    fun prt_syn syn =
wenzelm@12070
   757
      let val s = (case syn of None => "(structure)" | Some mx => Syntax.string_of_mixfix mx)
wenzelm@12575
   758
      in if s = "" then [] else [Pretty.brk 2, Pretty.str s] end;
wenzelm@12070
   759
    fun prt_fix (x, Some T, syn) = Pretty.block (Pretty.str (x ^ " ::") :: Pretty.brk 1 ::
wenzelm@12070
   760
          prt_typ T :: Pretty.brk 1 :: prt_syn syn)
wenzelm@12070
   761
      | prt_fix (x, None, syn) = Pretty.block (Pretty.str x :: Pretty.brk 1 :: prt_syn syn);
wenzelm@12070
   762
wenzelm@12307
   763
    fun prt_name "" = [Pretty.brk 1]
wenzelm@12307
   764
      | prt_name name = [Pretty.str (ProofContext.cond_extern ctxt name ^ ":"), Pretty.brk 1];
wenzelm@12307
   765
    fun prt_asm ((a, _), ts) = Pretty.block (prt_name a @ Pretty.breaks (map (prt_term o fst) ts));
wenzelm@12307
   766
    fun prt_def ((a, _), (t, _)) = Pretty.block (prt_name a @ [prt_term t]);
wenzelm@12307
   767
    fun prt_fact ((a, _), ths) = Pretty.block
wenzelm@12307
   768
      (prt_name a @ Pretty.breaks (map prt_thm (flat (map fst ths))));
wenzelm@12070
   769
wenzelm@12289
   770
    fun items _ [] = []
wenzelm@12289
   771
      | items prfx (x :: xs) = Pretty.block [Pretty.str prfx, Pretty.brk 1, x] :: items "  and" xs;
wenzelm@12289
   772
    fun prt_elem (Fixes fixes) = items "fixes" (map prt_fix fixes)
wenzelm@12289
   773
      | prt_elem (Assumes asms) = items "assumes" (map prt_asm asms)
wenzelm@12289
   774
      | prt_elem (Defines defs) = items "defines" (map prt_def defs)
wenzelm@12289
   775
      | prt_elem (Notes facts) = items "notes" (map prt_fact facts);
wenzelm@12730
   776
wenzelm@12730
   777
    val prt_pred =
wenzelm@12730
   778
      if null pred_ts then Pretty.str ""
wenzelm@12730
   779
      else
wenzelm@12730
   780
        Library.foldr1 Logic.mk_conjunction pred_ts
wenzelm@12806
   781
        |> ObjectLogic.atomize_term sg
wenzelm@12730
   782
        |> curry Term.list_abs_free pred_xs
wenzelm@12730
   783
        |> prt_term;
wenzelm@12277
   784
  in
wenzelm@12758
   785
    [Pretty.big_list "context elements:" (map (Pretty.chunks o prt_elem) all_elems),
wenzelm@12730
   786
      Pretty.big_list "predicate text:" [prt_pred]] |> Pretty.chunks |> Pretty.writeln
wenzelm@12277
   787
  end;
wenzelm@12070
   788
wenzelm@12070
   789
wenzelm@12070
   790
wenzelm@11896
   791
(** define locales **)
wenzelm@11896
   792
wenzelm@12063
   793
(* add_locale(_i) *)
wenzelm@11896
   794
wenzelm@12502
   795
local
wenzelm@12502
   796
wenzelm@12834
   797
fun gen_add_locale prep_ctxt prep_expr bname raw_import raw_body thy =
wenzelm@12063
   798
  let
wenzelm@12063
   799
    val sign = Theory.sign_of thy;
wenzelm@11896
   800
    val name = Sign.full_name sign bname;
wenzelm@12502
   801
    val _ = conditional (is_some (get_locale thy name)) (fn () =>
wenzelm@12502
   802
      error ("Duplicate definition of locale " ^ quote name));
wenzelm@11896
   803
wenzelm@12273
   804
    val thy_ctxt = ProofContext.init thy;
wenzelm@12834
   805
    val (((import_ctxt, (import_elemss, _)), (body_ctxt, (body_elemss, _))),
wenzelm@12839
   806
        (int_ext_text, ext_text)) = prep_ctxt raw_import raw_body thy_ctxt;
wenzelm@12839
   807
wenzelm@12839
   808
    val import_parms = params_of import_elemss;
wenzelm@12839
   809
    val body_parms = params_of body_elemss;
wenzelm@12839
   810
    val all_parms = import_parms @ body_parms;
wenzelm@12839
   811
wenzelm@12839
   812
    (* FIXME *)
wenzelm@12839
   813
    val ((_, spec), defs) = int_ext_text;
wenzelm@12839
   814
    val ((xs, _), _) = int_ext_text;
wenzelm@12839
   815
    val xs' = all_parms |> mapfilter (fn (p, _) =>
wenzelm@12839
   816
      (case assoc_string (xs, p) of None => None | Some T => Some (p, T)));
wenzelm@12839
   817
wenzelm@12730
   818
    val import = prep_expr sign raw_import;
wenzelm@12510
   819
    val elems = flat (map snd body_elemss);
wenzelm@12063
   820
  in
wenzelm@12063
   821
    thy
wenzelm@12063
   822
    |> declare_locale name
wenzelm@12289
   823
    |> put_locale name (make_locale import (map (fn e => (e, stamp ())) elems)
wenzelm@12839
   824
        ((xs', spec), defs) (all_parms, map fst body_parms))
wenzelm@12063
   825
  end;
wenzelm@12063
   826
wenzelm@12502
   827
in
wenzelm@12502
   828
wenzelm@12529
   829
val add_locale = gen_add_locale read_context intern_expr;
wenzelm@12502
   830
val add_locale_i = gen_add_locale cert_context (K I);
wenzelm@12502
   831
wenzelm@12502
   832
end;
wenzelm@12063
   833
wenzelm@11896
   834
wenzelm@12730
   835
(* store results *)
wenzelm@11896
   836
wenzelm@12706
   837
local
wenzelm@12706
   838
wenzelm@12702
   839
fun hide_bound_names names thy =
wenzelm@12702
   840
  thy |> PureThy.hide_thms false
wenzelm@12702
   841
    (map (Sign.full_name (Theory.sign_of thy)) (filter_out (equal "") names));
wenzelm@12702
   842
wenzelm@12958
   843
in
wenzelm@12958
   844
wenzelm@12706
   845
fun have_thmss_qualified kind loc args thy =
wenzelm@12706
   846
  thy
wenzelm@12706
   847
  |> Theory.add_path (Sign.base_name loc)
wenzelm@12711
   848
  |> PureThy.have_thmss_i (Drule.kind kind) args
wenzelm@12706
   849
  |>> hide_bound_names (map (#1 o #1) args)
wenzelm@12706
   850
  |>> Theory.parent_path;
wenzelm@12706
   851
wenzelm@12958
   852
fun smart_have_thmss kind None = PureThy.have_thmss_i (Drule.kind kind)
wenzelm@12958
   853
  | smart_have_thmss kind (Some (loc, _)) = have_thmss_qualified kind loc;
wenzelm@12958
   854
wenzelm@12958
   855
end;
wenzelm@12958
   856
wenzelm@12958
   857
local
wenzelm@12958
   858
wenzelm@12958
   859
fun put_facts loc args thy =
wenzelm@12958
   860
  let
wenzelm@12958
   861
    val {import, elems, text, params} = the_locale thy loc;
wenzelm@12958
   862
    val note = Notes (map (fn ((a, more_atts), th_atts) =>
wenzelm@12958
   863
      ((a, more_atts), map (apfst (map (curry Thm.name_thm a))) th_atts)) args);
wenzelm@12958
   864
  in thy |> put_locale loc (make_locale import (elems @ [(note, stamp ())]) text params) end;
wenzelm@12958
   865
wenzelm@12706
   866
fun gen_have_thmss prep_locale get_thms kind raw_loc raw_args thy =
wenzelm@12706
   867
  let
wenzelm@12706
   868
    val thy_ctxt = ProofContext.init thy;
wenzelm@12706
   869
    val loc = prep_locale (Theory.sign_of thy) raw_loc;
wenzelm@12730
   870
    val loc_ctxt = #1 (#1 (#1 (cert_context (Locale loc) [] thy_ctxt)));
wenzelm@12706
   871
    val args = map (apsnd (map (apfst (get_thms loc_ctxt)))) raw_args;
wenzelm@12706
   872
    val export = Drule.local_standard o ProofContext.export_single loc_ctxt thy_ctxt;
wenzelm@12711
   873
    val results = map (map export o #2) (#2 (ProofContext.have_thmss_i args loc_ctxt));
wenzelm@12706
   874
    val args' = map (rpair [] o #1 o #1) args ~~ map (single o Thm.no_attributes) results;
wenzelm@12706
   875
  in
wenzelm@12706
   876
    thy
wenzelm@12706
   877
    |> put_facts loc args
wenzelm@12706
   878
    |> have_thmss_qualified kind loc args'
wenzelm@12706
   879
  end;
wenzelm@12706
   880
wenzelm@12706
   881
in
wenzelm@12706
   882
wenzelm@12711
   883
val have_thmss = gen_have_thmss intern ProofContext.get_thms;
wenzelm@12711
   884
val have_thmss_i = gen_have_thmss (K I) (K I);
wenzelm@12711
   885
wenzelm@12958
   886
fun add_thmss loc args (ctxt, thy) =
wenzelm@12958
   887
  let
wenzelm@12958
   888
    val args' = map (fn ((a, ths), atts) => ((a, atts), [(ths, [])])) args;
wenzelm@12958
   889
    val (ctxt', facts') = apply_facts loc (ctxt, args');
wenzelm@12958
   890
  in ((ctxt', thy |> put_facts loc args'), facts') end;
wenzelm@12702
   891
wenzelm@12706
   892
end;
wenzelm@12063
   893
wenzelm@11896
   894
wenzelm@12730
   895
wenzelm@11896
   896
(** locale theory setup **)
wenzelm@12063
   897
wenzelm@11896
   898
val setup =
wenzelm@11896
   899
 [LocalesData.init];
wenzelm@11896
   900
wenzelm@11896
   901
end;
wenzelm@13211
   902