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