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