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