src/Pure/Isar/locale.ML
author ballarin
Fri Sep 02 09:50:58 2005 +0200 (2005-09-02)
changeset 17228 19b460b39dad
parent 17221 6cd180204582
child 17257 0ab67cb765da
permissions -rw-r--r--
print_locale omits facts by default
wenzelm@12014
     1
(*  Title:      Pure/Isar/locale.ML
wenzelm@11896
     2
    ID:         $Id$
ballarin@15206
     3
    Author:     Clemens Ballarin, TU Muenchen; Markus Wenzel, LMU/TU Muenchen
wenzelm@11896
     4
wenzelm@12058
     5
Locales -- Isar proof contexts as meta-level predicates, with local
wenzelm@12529
     6
syntax and implicit structures.
wenzelm@12529
     7
ballarin@14215
     8
Draws some basic ideas from Florian Kammueller's original version of
wenzelm@12529
     9
locales, but uses the richer infrastructure of Isar instead of the raw
wenzelm@12529
    10
meta-logic.  Furthermore, we provide structured import of contexts
ballarin@14215
    11
(with merge and rename operations), as well as type-inference of the
wenzelm@13375
    12
signature parts, and predicate definitions of the specification text.
ballarin@14446
    13
ballarin@14446
    14
See also:
ballarin@14446
    15
ballarin@14446
    16
[1] Clemens Ballarin. Locales and Locale Expressions in Isabelle/Isar.
ballarin@14446
    17
    In Stefano Berardi et al., Types for Proofs and Programs: International
ballarin@15099
    18
    Workshop, TYPES 2003, Torino, Italy, LNCS 3085, pages 34-50, 2004.
wenzelm@11896
    19
*)
wenzelm@11896
    20
ballarin@16169
    21
(* TODO:
ballarin@16169
    22
- beta-eta normalisation of interpretation parameters
ballarin@16169
    23
- dangling type frees in locales
ballarin@16620
    24
- test subsumption of interpretations when merging theories
ballarin@17138
    25
- var vs. fixes in locale to be interpreted (interpretation in locale)
ballarin@17138
    26
  (implicit locale expressions generated by multiple registrations)
ballarin@17138
    27
- current finish_global adds unwanted lemmas to theory/locale
ballarin@16169
    28
*)
ballarin@16169
    29
wenzelm@11896
    30
signature LOCALE =
wenzelm@11896
    31
sig
wenzelm@12046
    32
  type context
wenzelm@15703
    33
  datatype ('typ, 'term, 'fact) elem =
wenzelm@12058
    34
    Fixes of (string * 'typ option * mixfix option) list |
ballarin@16169
    35
    Constrains of (string * 'typ) list |
wenzelm@15703
    36
    Assumes of ((string * Attrib.src list) * ('term * ('term list * 'term list)) list) list |
wenzelm@15703
    37
    Defines of ((string * Attrib.src list) * ('term * 'term list)) list |
wenzelm@15703
    38
    Notes of ((string * Attrib.src list) * ('fact * Attrib.src list) list) list
wenzelm@12273
    39
  datatype expr =
wenzelm@12273
    40
    Locale of string |
ballarin@16102
    41
    Rename of expr * (string * mixfix option) option list |
wenzelm@12273
    42
    Merge of expr list
wenzelm@12273
    43
  val empty: expr
ballarin@15206
    44
  datatype 'a elem_expr = Elem of 'a | Expr of expr
ballarin@15206
    45
ballarin@15206
    46
  (* Abstract interface to locales *)
wenzelm@15703
    47
  type element
wenzelm@15703
    48
  type element_i
wenzelm@12046
    49
  type locale
wenzelm@16458
    50
  val intern: theory -> xstring -> string
wenzelm@16458
    51
  val extern: theory -> string -> xstring
wenzelm@12502
    52
  val the_locale: theory -> string -> locale
wenzelm@15703
    53
  val intern_attrib_elem: theory ->
wenzelm@15703
    54
    ('typ, 'term, 'fact) elem -> ('typ, 'term, 'fact) elem
wenzelm@15703
    55
  val intern_attrib_elem_expr: theory ->
wenzelm@15703
    56
    ('typ, 'term, 'fact) elem elem_expr -> ('typ, 'term, 'fact) elem elem_expr
ballarin@15206
    57
ballarin@15596
    58
  (* Processing of locale statements *)
wenzelm@15703
    59
  val read_context_statement: xstring option -> element elem_expr list ->
wenzelm@12529
    60
    (string * (string list * string list)) list list -> context ->
ballarin@15206
    61
    string option * (cterm list * cterm list) * context * context * 
ballarin@15206
    62
      (term * (term list * term list)) list list
wenzelm@15703
    63
  val cert_context_statement: string option -> element_i elem_expr list ->
wenzelm@12529
    64
    (term * (term list * term list)) list list -> context ->
ballarin@15206
    65
    string option * (cterm list * cterm list) * context * context *
ballarin@15206
    66
      (term * (term list * term list)) list list
ballarin@15596
    67
ballarin@15596
    68
  (* Diagnostic functions *)
wenzelm@12758
    69
  val print_locales: theory -> unit
ballarin@17228
    70
  val print_locale: theory -> bool -> expr -> element list -> unit
ballarin@17138
    71
  val print_global_registrations: bool -> string -> theory -> unit
ballarin@17138
    72
  val print_local_registrations': bool -> string -> context -> unit
ballarin@17138
    73
  val print_local_registrations: bool -> string -> context -> unit
ballarin@15596
    74
ballarin@15696
    75
  (* Storing results *)
wenzelm@17109
    76
  val add_locale_context: bool -> bstring -> expr -> element list -> theory
haftmann@17142
    77
    -> (element_i list * ProofContext.context) * theory
wenzelm@17109
    78
  val add_locale_context_i: bool -> bstring -> expr -> element_i list -> theory
haftmann@17142
    79
    -> (element_i list * ProofContext.context) * theory
wenzelm@15703
    80
  val add_locale: bool -> bstring -> expr -> element list -> theory -> theory
wenzelm@15703
    81
  val add_locale_i: bool -> bstring -> expr -> element_i list -> theory -> theory
wenzelm@15703
    82
  val smart_note_thmss: string -> string option ->
wenzelm@12958
    83
    ((bstring * theory attribute list) * (thm list * theory attribute list) list) list ->
wenzelm@12958
    84
    theory -> theory * (bstring * thm list) list
wenzelm@15703
    85
  val note_thmss: string -> xstring ->
wenzelm@15703
    86
    ((bstring * Attrib.src list) * (thmref * Attrib.src list) list) list ->
wenzelm@17109
    87
    theory -> (theory * ProofContext.context) * (bstring * thm list) list
wenzelm@15703
    88
  val note_thmss_i: string -> string ->
wenzelm@15703
    89
    ((bstring * Attrib.src list) * (thm list * Attrib.src list) list) list ->
wenzelm@17109
    90
    theory -> (theory * ProofContext.context) * (bstring * thm list) list
wenzelm@15703
    91
  val add_thmss: string -> string -> ((string * thm list) * Attrib.src list) list ->
wenzelm@13375
    92
    theory * context -> (theory * context) * (string * thm list) list
ballarin@15596
    93
ballarin@15624
    94
  (* Locale interpretation *)
ballarin@15624
    95
  val prep_global_registration:
wenzelm@15703
    96
    string * Attrib.src list -> expr -> string option list -> theory ->
ballarin@17033
    97
    ((string * term list) * term list) list * (thm list -> theory -> theory)
ballarin@17033
    98
  val prep_local_registration:
ballarin@17033
    99
    string * Attrib.src list -> expr -> string option list -> context ->
ballarin@17033
   100
    ((string * term list) * term list) list * (thm list -> context -> context)
ballarin@16736
   101
  val prep_registration_in_locale:
ballarin@17000
   102
    string -> expr -> theory ->
ballarin@17033
   103
    ((string * string list) * term list) list * (thm list -> theory -> theory)
wenzelm@11896
   104
end;
wenzelm@12839
   105
wenzelm@12289
   106
structure Locale: LOCALE =
wenzelm@11896
   107
struct
wenzelm@11896
   108
wenzelm@12273
   109
(** locale elements and expressions **)
wenzelm@11896
   110
wenzelm@12014
   111
type context = ProofContext.context;
wenzelm@11896
   112
wenzelm@15703
   113
datatype ('typ, 'term, 'fact) elem =
wenzelm@12058
   114
  Fixes of (string * 'typ option * mixfix option) list |
ballarin@16169
   115
  Constrains of (string * 'typ) list |
wenzelm@15703
   116
  Assumes of ((string * Attrib.src list) * ('term * ('term list * 'term list)) list) list |
wenzelm@15703
   117
  Defines of ((string * Attrib.src list) * ('term * 'term list)) list |
wenzelm@15703
   118
  Notes of ((string * Attrib.src list) * ('fact * Attrib.src list) list) list;
wenzelm@12273
   119
wenzelm@12273
   120
datatype expr =
wenzelm@12273
   121
  Locale of string |
ballarin@16102
   122
  Rename of expr * (string * mixfix option) option list |
wenzelm@12273
   123
  Merge of expr list;
wenzelm@11896
   124
wenzelm@12273
   125
val empty = Merge [];
wenzelm@12273
   126
ballarin@15206
   127
datatype 'a elem_expr =
ballarin@15206
   128
  Elem of 'a | Expr of expr;
wenzelm@12273
   129
wenzelm@15703
   130
type element = (string, string, thmref) elem;
wenzelm@15703
   131
type element_i = (typ, term, thm list) elem;
wenzelm@12070
   132
wenzelm@12070
   133
type locale =
ballarin@15206
   134
 {predicate: cterm list * thm list,
ballarin@16736
   135
    (* CB: For locales with "(open)" this entry is ([], []).
ballarin@15206
   136
       For new-style locales, which declare predicates, if the locale declares
ballarin@15206
   137
       no predicates, this is also ([], []).
ballarin@15206
   138
       If the locale declares predicates, the record field is
ballarin@15206
   139
       ([statement], axioms), where statement is the locale predicate applied
ballarin@15206
   140
       to the (assumed) locale parameters.  Axioms contains the projections
ballarin@15206
   141
       from the locale predicate to the normalised assumptions of the locale
ballarin@15206
   142
       (cf. [1], normalisation of locale expressions.)
ballarin@15206
   143
    *)
ballarin@15206
   144
  import: expr,                                       (*dynamic import*)
wenzelm@15703
   145
  elems: (element_i * stamp) list,                    (*static content*)
ballarin@16736
   146
  params: ((string * typ) * mixfix option) list * string list,
ballarin@16102
   147
                                                      (*all/local params*)
ballarin@16736
   148
  regs: ((string * string list) * thm list) list}
ballarin@16736
   149
    (* Registrations: indentifiers and witness theorems of locales interpreted
ballarin@16736
   150
       in the locale.
ballarin@16736
   151
    *)
wenzelm@12063
   152
wenzelm@11896
   153
wenzelm@15703
   154
(* CB: an internal (Int) locale element was either imported or included,
wenzelm@15703
   155
   an external (Ext) element appears directly in the locale text. *)
wenzelm@15703
   156
wenzelm@15703
   157
datatype ('a, 'b) int_ext = Int of 'a | Ext of 'b;
wenzelm@15703
   158
wenzelm@15703
   159
wenzelm@15703
   160
ballarin@15837
   161
(** term and type instantiation, using symbol tables **)
ballarin@16620
   162
(** functions for term instantiation beta-reduce the result
ballarin@16620
   163
    unless the instantiation table is empty (inst_tab_term)
ballarin@16620
   164
    or the instantiation has no effect (inst_tab_thm) **)
ballarin@15837
   165
ballarin@15837
   166
(* instantiate TFrees *)
ballarin@15837
   167
ballarin@15837
   168
fun tinst_tab_type tinst T = if Symtab.is_empty tinst
ballarin@15837
   169
      then T
ballarin@15837
   170
      else Term.map_type_tfree
wenzelm@17221
   171
        (fn (v as (x, _)) => getOpt (Symtab.curried_lookup tinst x, (TFree v))) T;
ballarin@15837
   172
ballarin@15837
   173
fun tinst_tab_term tinst t = if Symtab.is_empty tinst
ballarin@15837
   174
      then t
ballarin@15837
   175
      else Term.map_term_types (tinst_tab_type tinst) t;
ballarin@15837
   176
wenzelm@16458
   177
fun tinst_tab_thm thy tinst thm = if Symtab.is_empty tinst
ballarin@15837
   178
      then thm
ballarin@15837
   179
      else let
wenzelm@16458
   180
          val cert = Thm.cterm_of thy;
wenzelm@16458
   181
          val certT = Thm.ctyp_of thy;
ballarin@15837
   182
          val {hyps, prop, ...} = Thm.rep_thm thm;
ballarin@15837
   183
          val tfrees = foldr Term.add_term_tfree_names [] (prop :: hyps);
ballarin@15837
   184
          val tinst' = Symtab.dest tinst |>
ballarin@15837
   185
                List.filter (fn (a, _) => a mem_string tfrees);
ballarin@15837
   186
        in
ballarin@15837
   187
          if null tinst' then thm
ballarin@15837
   188
          else thm |> Drule.implies_intr_list (map cert hyps)
ballarin@15837
   189
            |> Drule.tvars_intr_list (map #1 tinst')
ballarin@15837
   190
            |> (fn (th, al) => th |> Thm.instantiate
ballarin@15837
   191
                ((map (fn (a, T) => (certT (TVar (valOf (assoc (al, a)))), certT T)) tinst'),
ballarin@15837
   192
                  []))
ballarin@15837
   193
            |> (fn th => Drule.implies_elim_list th
ballarin@15837
   194
                 (map (Thm.assume o cert o tinst_tab_term tinst) hyps))
ballarin@15837
   195
        end;
ballarin@15837
   196
ballarin@15837
   197
(* instantiate TFrees and Frees *)
ballarin@15837
   198
ballarin@15837
   199
fun inst_tab_term (inst, tinst) = if Symtab.is_empty inst
ballarin@15837
   200
      then tinst_tab_term tinst
ballarin@15837
   201
      else (* instantiate terms and types simultaneously *)
ballarin@15837
   202
        let
ballarin@15837
   203
          fun instf (Const (x, T)) = Const (x, tinst_tab_type tinst T)
wenzelm@17221
   204
            | instf (Free (x, T)) = (case Symtab.curried_lookup inst x of
ballarin@15837
   205
                 NONE => Free (x, tinst_tab_type tinst T)
ballarin@15837
   206
               | SOME t => t)
ballarin@15837
   207
            | instf (Var (xi, T)) = Var (xi, tinst_tab_type tinst T)
ballarin@15837
   208
            | instf (b as Bound _) = b
ballarin@15837
   209
            | instf (Abs (x, T, t)) = Abs (x, tinst_tab_type tinst T, instf t)
ballarin@15837
   210
            | instf (s $ t) = instf s $ instf t
ballarin@16620
   211
        in Envir.beta_norm o instf end;
ballarin@15837
   212
wenzelm@16458
   213
fun inst_tab_thm thy (inst, tinst) thm = if Symtab.is_empty inst
wenzelm@16458
   214
      then tinst_tab_thm thy tinst thm
ballarin@15837
   215
      else let
wenzelm@16458
   216
          val cert = Thm.cterm_of thy;
wenzelm@16458
   217
          val certT = Thm.ctyp_of thy;
ballarin@15837
   218
          val {hyps, prop, ...} = Thm.rep_thm thm;
ballarin@15837
   219
          (* type instantiations *)
ballarin@15837
   220
          val tfrees = foldr Term.add_term_tfree_names [] (prop :: hyps);
ballarin@15837
   221
          val tinst' = Symtab.dest tinst |>
ballarin@15837
   222
                List.filter (fn (a, _) => a mem_string tfrees);
ballarin@15837
   223
          (* term instantiations;
ballarin@15837
   224
             note: lhss are type instantiated, because
ballarin@15837
   225
                   type insts will be done first*)
ballarin@15837
   226
          val frees = foldr Term.add_term_frees [] (prop :: hyps);
ballarin@15837
   227
          val inst' = Symtab.dest inst |>
ballarin@15837
   228
                List.mapPartial (fn (a, t) =>
ballarin@15837
   229
                  get_first (fn (Free (x, T)) => 
ballarin@15837
   230
                    if a = x then SOME (Free (x, tinst_tab_type tinst T), t)
ballarin@15837
   231
                    else NONE) frees);
ballarin@15837
   232
        in
wenzelm@16458
   233
          if null tinst' andalso null inst' then tinst_tab_thm thy tinst thm
ballarin@15837
   234
          else thm |> Drule.implies_intr_list (map cert hyps)
ballarin@15837
   235
            |> Drule.tvars_intr_list (map #1 tinst')
ballarin@15837
   236
            |> (fn (th, al) => th |> Thm.instantiate
ballarin@15837
   237
                ((map (fn (a, T) => (certT (TVar (valOf (assoc (al, a)))), certT T)) tinst'),
ballarin@15837
   238
                  []))
ballarin@15837
   239
            |> Drule.forall_intr_list (map (cert o #1) inst')
ballarin@15837
   240
            |> Drule.forall_elim_list (map (cert o #2) inst') 
ballarin@16620
   241
            |> Drule.fconv_rule (Thm.beta_conversion true)
ballarin@15837
   242
            |> (fn th => Drule.implies_elim_list th
ballarin@15837
   243
                 (map (Thm.assume o cert o inst_tab_term (inst, tinst)) hyps))
ballarin@15837
   244
        end;
ballarin@15837
   245
ballarin@15837
   246
ballarin@17138
   247
fun inst_tab_att thy (inst as (_, tinst)) =
ballarin@17138
   248
    Args.map_values I (tinst_tab_type tinst)
ballarin@17138
   249
    (inst_tab_term inst) (inst_tab_thm thy inst);
ballarin@17138
   250
ballarin@17138
   251
fun inst_tab_atts thy inst = map (inst_tab_att thy inst);
ballarin@17138
   252
ballarin@17138
   253
ballarin@16736
   254
(** management of registrations in theories and proof contexts **)
wenzelm@11896
   255
ballarin@15837
   256
structure Registrations :
ballarin@15837
   257
  sig
ballarin@15837
   258
    type T
ballarin@15837
   259
    val empty: T
ballarin@15837
   260
    val join: T * T -> T
ballarin@15837
   261
    val dest: T -> (term list * ((string * Attrib.src list) * thm list)) list
wenzelm@16458
   262
    val lookup: theory -> T * term list ->
ballarin@15837
   263
      ((string * Attrib.src list) * thm list) option
wenzelm@16458
   264
    val insert: theory -> term list * (string * Attrib.src list) -> T ->
ballarin@15837
   265
      T * (term list * ((string * Attrib.src list) * thm list)) list
ballarin@15837
   266
    val add_witness: term list -> thm -> T -> T
ballarin@15837
   267
  end =
ballarin@15837
   268
struct
ballarin@15837
   269
  (* a registration consists of theorems instantiating locale assumptions
ballarin@15837
   270
     and prefix and attributes, indexed by parameter instantiation *)
ballarin@15837
   271
  type T = ((string * Attrib.src list) * thm list) Termtab.table;
ballarin@15837
   272
ballarin@15837
   273
  val empty = Termtab.empty;
ballarin@15837
   274
ballarin@15837
   275
  (* term list represented as single term, for simultaneous matching *)
ballarin@15837
   276
  fun termify ts =
ballarin@15837
   277
    Library.foldl (op $) (Const ("", map fastype_of ts ---> propT), ts);
ballarin@15837
   278
  fun untermify t =
ballarin@15837
   279
    let fun ut (Const _) ts = ts
ballarin@15837
   280
          | ut (s $ t) ts = ut s (t::ts)
ballarin@15837
   281
    in ut t [] end;
ballarin@15837
   282
ballarin@16620
   283
  (* joining of registrations: prefix and attributes of left theory,
ballarin@15837
   284
     thms are equal, no attempt to subsumption testing *)
wenzelm@16458
   285
  fun join (r1, r2) = Termtab.join (fn _ => fn (reg, _) => SOME reg) (r1, r2);
ballarin@15837
   286
ballarin@15837
   287
  fun dest regs = map (apfst untermify) (Termtab.dest regs);
ballarin@15837
   288
ballarin@15837
   289
  (* registrations that subsume t *)
wenzelm@17203
   290
  fun subsumers thy t regs =
wenzelm@17203
   291
    List.filter (fn (t', _) => Pattern.matches thy (t', t)) (Termtab.dest regs);
ballarin@15837
   292
ballarin@15837
   293
  (* look up registration, pick one that subsumes the query *)
ballarin@15837
   294
  fun lookup sign (regs, ts) =
ballarin@15837
   295
    let
ballarin@15837
   296
      val t = termify ts;
wenzelm@17203
   297
      val subs = subsumers sign t regs;
ballarin@15837
   298
    in (case subs of
ballarin@15837
   299
        [] => NONE
ballarin@15837
   300
      | ((t', (attn, thms)) :: _) => let
wenzelm@17203
   301
            val (tinst, inst) = Pattern.match sign (t', t);
ballarin@15837
   302
            (* thms contain Frees, not Vars *)
ballarin@15837
   303
            val tinst' = tinst |> Vartab.dest
ballarin@15837
   304
                 |> map (fn ((x, 0), (_, T)) => (x, Type.unvarifyT T))
ballarin@15837
   305
                 |> Symtab.make;
ballarin@15837
   306
            val inst' = inst |> Vartab.dest
ballarin@15837
   307
                 |> map (fn ((x, 0), (_, t)) => (x, Logic.unvarify t))
ballarin@15837
   308
                 |> Symtab.make;
ballarin@15837
   309
          in
ballarin@15837
   310
            SOME (attn, map (inst_tab_thm sign (inst', tinst')) thms)
ballarin@15837
   311
          end)
ballarin@15837
   312
    end;
ballarin@15837
   313
ballarin@15837
   314
  (* add registration if not subsumed by ones already present,
ballarin@15837
   315
     additionally returns registrations that are strictly subsumed *)
ballarin@15837
   316
  fun insert sign (ts, attn) regs =
ballarin@15837
   317
    let
ballarin@15837
   318
      val t = termify ts;
wenzelm@17203
   319
      val subs = subsumers sign t regs ;
ballarin@15837
   320
    in (case subs of
ballarin@15837
   321
        [] => let
ballarin@15837
   322
                val sups =
wenzelm@17203
   323
                  List.filter (fn (t', _) => Pattern.matches sign (t, t')) (Termtab.dest regs);
ballarin@15837
   324
                val sups' = map (apfst untermify) sups
wenzelm@17221
   325
              in (Termtab.curried_update (t, (attn, [])) regs, sups') end
ballarin@15837
   326
      | _ => (regs, []))
ballarin@15837
   327
    end;
ballarin@15837
   328
ballarin@15837
   329
  (* add witness theorem to registration,
ballarin@16169
   330
     only if instantiation is exact, otherwise exception Option raised *)
ballarin@15837
   331
  fun add_witness ts thm regs =
ballarin@15837
   332
    let
ballarin@15837
   333
      val t = termify ts;
wenzelm@17221
   334
      val (x, thms) = valOf (Termtab.curried_lookup regs t);
ballarin@15837
   335
    in
wenzelm@17221
   336
      Termtab.curried_update (t, (x, thm::thms)) regs
ballarin@15837
   337
    end;
ballarin@15837
   338
end;
ballarin@15837
   339
ballarin@16736
   340
ballarin@15837
   341
(** theory data **)
ballarin@15596
   342
wenzelm@16458
   343
structure GlobalLocalesData = TheoryDataFun
wenzelm@16458
   344
(struct
wenzelm@12014
   345
  val name = "Isar/locales";
ballarin@15837
   346
  type T = NameSpace.T * locale Symtab.table * Registrations.T Symtab.table;
ballarin@15596
   347
    (* 1st entry: locale namespace,
ballarin@15596
   348
       2nd entry: locales of the theory,
ballarin@15837
   349
       3rd entry: registrations, indexed by locale name *)
wenzelm@11896
   350
ballarin@15596
   351
  val empty = (NameSpace.empty, Symtab.empty, Symtab.empty);
wenzelm@12063
   352
  val copy = I;
wenzelm@16458
   353
  val extend = I;
wenzelm@12289
   354
ballarin@16736
   355
  fun join_locs _ ({predicate, import, elems, params, regs}: locale,
ballarin@16736
   356
      {elems = elems', regs = regs', ...}: locale) =
ballarin@15596
   357
    SOME {predicate = predicate, import = import,
ballarin@15596
   358
      elems = gen_merge_lists eq_snd elems elems',
ballarin@16736
   359
      params = params,
ballarin@16736
   360
      regs = merge_alists regs regs'};
wenzelm@16458
   361
  fun merge _ ((space1, locs1, regs1), (space2, locs2, regs2)) =
ballarin@15596
   362
    (NameSpace.merge (space1, space2), Symtab.join join_locs (locs1, locs2),
wenzelm@16458
   363
     Symtab.join (K (SOME o Registrations.join)) (regs1, regs2));
wenzelm@12289
   364
ballarin@15596
   365
  fun print _ (space, locs, _) =
wenzelm@16346
   366
    Pretty.strs ("locales:" :: map #1 (NameSpace.extern_table (space, locs)))
wenzelm@12014
   367
    |> Pretty.writeln;
wenzelm@16458
   368
end);
wenzelm@11896
   369
wenzelm@15801
   370
val _ = Context.add_setup [GlobalLocalesData.init];
wenzelm@15801
   371
wenzelm@15801
   372
ballarin@15624
   373
ballarin@15624
   374
(** context data **)
wenzelm@11896
   375
wenzelm@16458
   376
structure LocalLocalesData = ProofDataFun
wenzelm@16458
   377
(struct
ballarin@15624
   378
  val name = "Isar/locales";
ballarin@15837
   379
  type T = Registrations.T Symtab.table;
ballarin@15837
   380
    (* registrations, indexed by locale name *)
ballarin@15624
   381
  fun init _ = Symtab.empty;
ballarin@15624
   382
  fun print _ _ = ();
wenzelm@16458
   383
end);
ballarin@15624
   384
wenzelm@15801
   385
val _ = Context.add_setup [LocalLocalesData.init];
wenzelm@12289
   386
wenzelm@12277
   387
wenzelm@12277
   388
(* access locales *)
wenzelm@12277
   389
ballarin@15624
   390
val print_locales = GlobalLocalesData.print;
ballarin@15624
   391
wenzelm@16458
   392
val intern = NameSpace.intern o #1 o GlobalLocalesData.get;
wenzelm@16458
   393
val extern = NameSpace.extern o #1 o GlobalLocalesData.get;
ballarin@15624
   394
wenzelm@16144
   395
fun declare_locale name thy =
wenzelm@16144
   396
  thy |> GlobalLocalesData.map (fn (space, locs, regs) =>
wenzelm@16458
   397
    (Sign.declare_name thy name space, locs, regs));
wenzelm@11896
   398
ballarin@15596
   399
fun put_locale name loc =
ballarin@15624
   400
  GlobalLocalesData.map (fn (space, locs, regs) =>
wenzelm@17221
   401
    (space, Symtab.curried_update (name, loc) locs, regs));
ballarin@15596
   402
wenzelm@17221
   403
fun get_locale thy name = Symtab.curried_lookup (#2 (GlobalLocalesData.get thy)) name;
wenzelm@11896
   404
wenzelm@12014
   405
fun the_locale thy name =
wenzelm@12014
   406
  (case get_locale thy name of
skalberg@15531
   407
    SOME loc => loc
skalberg@15531
   408
  | NONE => error ("Unknown locale " ^ quote name));
wenzelm@11896
   409
wenzelm@12046
   410
ballarin@15596
   411
(* access registrations *)
ballarin@15596
   412
ballarin@15696
   413
(* Ids of global registrations are varified,
ballarin@15696
   414
   Ids of local registrations are not.
ballarin@15696
   415
   Thms of registrations are never varified. *)
ballarin@15696
   416
ballarin@15624
   417
(* retrieve registration from theory or context *)
ballarin@15624
   418
ballarin@15696
   419
fun gen_get_registrations get thy_ctxt name =
wenzelm@17221
   420
  case Symtab.curried_lookup (get thy_ctxt) name of
ballarin@15696
   421
      NONE => []
ballarin@15837
   422
    | SOME reg => Registrations.dest reg;
ballarin@15696
   423
ballarin@15696
   424
val get_global_registrations =
ballarin@15696
   425
     gen_get_registrations (#3 o GlobalLocalesData.get);
ballarin@15696
   426
val get_local_registrations =
ballarin@15696
   427
     gen_get_registrations LocalLocalesData.get;
ballarin@15696
   428
wenzelm@16458
   429
fun gen_get_registration get thy_of thy_ctxt (name, ps) =
wenzelm@17221
   430
  case Symtab.curried_lookup (get thy_ctxt) name of
ballarin@15624
   431
      NONE => NONE
wenzelm@16458
   432
    | SOME reg => Registrations.lookup (thy_of thy_ctxt) (reg, ps);
ballarin@15624
   433
ballarin@15624
   434
val get_global_registration =
wenzelm@16458
   435
     gen_get_registration (#3 o GlobalLocalesData.get) I;
ballarin@15624
   436
val get_local_registration =
wenzelm@16458
   437
     gen_get_registration LocalLocalesData.get ProofContext.theory_of;
ballarin@15596
   438
ballarin@15624
   439
val test_global_registration = isSome oo get_global_registration;
ballarin@15624
   440
val test_local_registration = isSome oo get_local_registration;
ballarin@15624
   441
fun smart_test_registration ctxt id =
ballarin@15624
   442
  let
ballarin@15624
   443
    val thy = ProofContext.theory_of ctxt;
ballarin@15624
   444
  in
ballarin@15624
   445
    test_global_registration thy id orelse test_local_registration ctxt id
ballarin@15624
   446
  end;
ballarin@15624
   447
ballarin@15624
   448
ballarin@15837
   449
(* add registration to theory or context, ignored if subsumed *)
ballarin@15624
   450
wenzelm@16458
   451
fun gen_put_registration map_data thy_of (name, ps) attn thy_ctxt =
ballarin@15837
   452
  map_data (fn regs =>
ballarin@15837
   453
    let
wenzelm@16458
   454
      val thy = thy_of thy_ctxt;
wenzelm@17221
   455
      val reg = getOpt (Symtab.curried_lookup regs name, Registrations.empty);
wenzelm@16458
   456
      val (reg', sups) = Registrations.insert thy (ps, attn) reg;
ballarin@15837
   457
      val _ = if not (null sups) then warning
ballarin@15837
   458
                ("Subsumed interpretation(s) of locale " ^
wenzelm@16458
   459
                 quote (extern thy name) ^
ballarin@15837
   460
                 "\nby interpretation(s) with the following prefix(es):\n" ^
ballarin@15837
   461
                  commas_quote (map (fn (_, ((s, _), _)) => s) sups))
ballarin@15837
   462
              else ();
wenzelm@17221
   463
    in Symtab.curried_update (name, reg') regs end) thy_ctxt;
ballarin@15624
   464
ballarin@15624
   465
val put_global_registration =
ballarin@15624
   466
     gen_put_registration (fn f =>
wenzelm@16458
   467
       GlobalLocalesData.map (fn (space, locs, regs) => (space, locs, f regs))) I;
ballarin@15837
   468
val put_local_registration =
wenzelm@16458
   469
     gen_put_registration LocalLocalesData.map ProofContext.theory_of;
ballarin@15596
   470
ballarin@17000
   471
fun put_registration_in_locale name id thy =
ballarin@17000
   472
    let
ballarin@17000
   473
      val {predicate, import, elems, params, regs} = the_locale thy name;
ballarin@17000
   474
    in
ballarin@17000
   475
      put_locale name {predicate = predicate, import = import,
ballarin@17000
   476
        elems = elems, params = params, regs = regs @ [(id, [])]} thy
ballarin@17000
   477
    end;
ballarin@17000
   478
ballarin@15624
   479
ballarin@15624
   480
(* add witness theorem to registration in theory or context,
ballarin@15596
   481
   ignored if registration not present *)
ballarin@15596
   482
wenzelm@17221
   483
fun gen_add_witness map_regs (name, ps) thm =
wenzelm@17221
   484
  map_regs (Symtab.map_entry name (Registrations.add_witness ps thm));
ballarin@15596
   485
ballarin@15624
   486
val add_global_witness =
ballarin@15624
   487
     gen_add_witness (fn f =>
ballarin@15624
   488
       GlobalLocalesData.map (fn (space, locs, regs) =>
ballarin@15624
   489
         (space, locs, f regs)));
ballarin@15624
   490
val add_local_witness = gen_add_witness LocalLocalesData.map;
ballarin@15596
   491
ballarin@17000
   492
fun add_witness_in_locale name id thm thy =
ballarin@17000
   493
    let
ballarin@17000
   494
      val {predicate, import, elems, params, regs} = the_locale thy name;
ballarin@17000
   495
      fun add (id', thms) =
ballarin@17000
   496
          if id = id' then (id', thm :: thms) else (id', thms);
ballarin@17000
   497
    in
ballarin@17000
   498
      put_locale name {predicate = predicate, import = import,
ballarin@17000
   499
	elems = elems, params = params, regs = map add regs} thy
ballarin@17000
   500
    end;
ballarin@15596
   501
ballarin@14215
   502
(* import hierarchy
ballarin@14215
   503
   implementation could be more efficient, eg. by maintaining a database
ballarin@14215
   504
   of dependencies *)
ballarin@14215
   505
ballarin@14215
   506
fun imports thy (upper, lower) =
ballarin@14215
   507
  let
ballarin@14215
   508
    fun imps (Locale name) low = (name = low) orelse
ballarin@14215
   509
      (case get_locale thy name of
skalberg@15531
   510
           NONE => false
skalberg@15531
   511
         | SOME {import, ...} => imps import low)
ballarin@14215
   512
      | imps (Rename (expr, _)) low = imps expr low
ballarin@14215
   513
      | imps (Merge es) low = exists (fn e => imps e low) es;
ballarin@14215
   514
  in
wenzelm@16458
   515
    imps (Locale (intern thy upper)) (intern thy lower)
ballarin@14215
   516
  end;
ballarin@14215
   517
ballarin@14215
   518
ballarin@15624
   519
(* printing of registrations *)
ballarin@15596
   520
ballarin@17138
   521
fun gen_print_registrations get_regs mk_ctxt msg show_wits loc thy_ctxt =
ballarin@15596
   522
  let
wenzelm@15703
   523
    val ctxt = mk_ctxt thy_ctxt;
wenzelm@15703
   524
    val thy = ProofContext.theory_of ctxt;
wenzelm@15703
   525
wenzelm@15703
   526
    val prt_term = Pretty.quote o ProofContext.pretty_term ctxt;
ballarin@17096
   527
    fun prt_inst ts =
ballarin@17096
   528
        Pretty.enclose "(" ")" (Pretty.breaks (map prt_term ts));
ballarin@17096
   529
    fun prt_attn (prfx, atts) =
ballarin@17096
   530
        Pretty.breaks (Pretty.str prfx :: Args.pretty_attribs ctxt atts);
ballarin@17096
   531
    val prt_thm = Pretty.quote o ProofContext.pretty_thm ctxt;
ballarin@17096
   532
    fun prt_thms thms =
ballarin@17096
   533
        Pretty.enclose "[" "]" (Pretty.breaks (map prt_thm thms));
ballarin@17096
   534
    fun prt_reg (ts, (("", []), thms)) =
ballarin@17138
   535
        if show_wits
ballarin@17096
   536
          then Pretty.block [prt_inst ts, Pretty.fbrk, prt_thms thms]
ballarin@17096
   537
          else prt_inst ts
ballarin@17096
   538
      | prt_reg (ts, (attn, thms)) =
ballarin@17138
   539
        if show_wits
ballarin@17096
   540
          then Pretty.block ((prt_attn attn @
ballarin@17096
   541
            [Pretty.str ":", Pretty.brk 1, prt_inst ts, Pretty.fbrk,
ballarin@17096
   542
              prt_thms thms]))
ballarin@17096
   543
          else Pretty.block ((prt_attn attn @
ballarin@17096
   544
            [Pretty.str ":", Pretty.brk 1, prt_inst ts]));
wenzelm@15703
   545
wenzelm@16458
   546
    val loc_int = intern thy loc;
ballarin@15624
   547
    val regs = get_regs thy_ctxt;
wenzelm@17221
   548
    val loc_regs = Symtab.curried_lookup regs loc_int;
ballarin@15596
   549
  in
ballarin@15596
   550
    (case loc_regs of
ballarin@15624
   551
        NONE => Pretty.str ("No interpretations in " ^ msg ^ ".")
ballarin@15763
   552
      | SOME r => let
ballarin@15837
   553
            val r' = Registrations.dest r;
ballarin@15763
   554
            val r'' = Library.sort_wrt (fn (_, ((prfx, _), _)) => prfx) r';
ballarin@15763
   555
          in Pretty.big_list ("Interpretations in " ^ msg ^ ":")
ballarin@17096
   556
            (map prt_reg r'')
ballarin@15763
   557
          end)
ballarin@15596
   558
    |> Pretty.writeln
ballarin@15596
   559
  end;
ballarin@15596
   560
ballarin@15624
   561
val print_global_registrations =
ballarin@15624
   562
     gen_print_registrations (#3 o GlobalLocalesData.get)
wenzelm@15703
   563
       ProofContext.init "theory";
ballarin@15624
   564
val print_local_registrations' =
ballarin@15624
   565
     gen_print_registrations LocalLocalesData.get
wenzelm@15703
   566
       I "context";
ballarin@17138
   567
fun print_local_registrations show_wits loc ctxt =
ballarin@17138
   568
  (print_global_registrations show_wits loc (ProofContext.theory_of ctxt);
ballarin@17138
   569
   print_local_registrations' show_wits loc ctxt);
ballarin@15624
   570
ballarin@15596
   571
wenzelm@12277
   572
(* diagnostics *)
wenzelm@12273
   573
wenzelm@12277
   574
fun err_in_locale ctxt msg ids =
wenzelm@12277
   575
  let
wenzelm@16458
   576
    val thy = ProofContext.theory_of ctxt;
wenzelm@12529
   577
    fun prt_id (name, parms) =
wenzelm@16458
   578
      [Pretty.block (Pretty.breaks (map Pretty.str (extern thy name :: parms)))];
skalberg@15570
   579
    val prt_ids = List.concat (separate [Pretty.str " +", Pretty.brk 1] (map prt_id ids));
wenzelm@12502
   580
    val err_msg =
wenzelm@12529
   581
      if forall (equal "" o #1) ids then msg
wenzelm@12502
   582
      else msg ^ "\n" ^ Pretty.string_of (Pretty.block
wenzelm@12502
   583
        (Pretty.str "The error(s) above occurred in locale:" :: Pretty.brk 1 :: prt_ids));
wenzelm@12502
   584
  in raise ProofContext.CONTEXT (err_msg, ctxt) end;
wenzelm@12063
   585
ballarin@15206
   586
(* Version for identifiers with axioms *)
ballarin@15206
   587
ballarin@15206
   588
fun err_in_locale' ctxt msg ids' = err_in_locale ctxt msg (map fst ids');
wenzelm@12277
   589
wenzelm@12277
   590
wenzelm@12529
   591
(** primitives **)
wenzelm@12046
   592
wenzelm@15703
   593
(* map elements *)
wenzelm@15703
   594
wenzelm@15703
   595
fun map_elem {name, var, typ, term, fact, attrib} =
wenzelm@15703
   596
  fn Fixes fixes => Fixes (fixes |> map (fn (x, T, mx) =>
wenzelm@15703
   597
       let val (x', mx') = var (x, mx) in (x', Option.map typ T, mx') end))
ballarin@16169
   598
   | Constrains csts => Constrains (csts |> map (fn (x, T) =>
ballarin@16169
   599
       let val (x', _) = var (x, SOME Syntax.NoSyn) in (x', typ T) end))
wenzelm@15703
   600
   | Assumes asms => Assumes (asms |> map (fn ((a, atts), propps) =>
wenzelm@15703
   601
      ((name a, map attrib atts), propps |> map (fn (t, (ps, qs)) =>
wenzelm@15703
   602
        (term t, (map term ps, map term qs))))))
wenzelm@15703
   603
   | Defines defs => Defines (defs |> map (fn ((a, atts), (t, ps)) =>
wenzelm@15703
   604
      ((name a, map attrib atts), (term t, map term ps))))
wenzelm@15703
   605
   | Notes facts => Notes (facts |> map (fn ((a, atts), bs) =>
wenzelm@15703
   606
      ((name a, map attrib atts), bs |> map (fn (ths, btts) => (fact ths, map attrib btts)))));
wenzelm@15703
   607
wenzelm@15703
   608
fun map_values typ term thm = map_elem
wenzelm@15703
   609
  {name = I, var = I, typ = typ, term = term, fact = map thm,
wenzelm@15703
   610
    attrib = Args.map_values I typ term thm};
wenzelm@15703
   611
wenzelm@15703
   612
wenzelm@15703
   613
(* map attributes *)
wenzelm@15703
   614
wenzelm@15703
   615
fun map_attrib_elem f = map_elem {name = I, var = I, typ = I, term = I, fact = I, attrib = f};
wenzelm@15703
   616
wenzelm@16458
   617
fun intern_attrib_elem thy = map_attrib_elem (Attrib.intern_src thy);
wenzelm@15703
   618
wenzelm@15703
   619
fun intern_attrib_elem_expr thy (Elem elem) = Elem (intern_attrib_elem thy elem)
wenzelm@15703
   620
  | intern_attrib_elem_expr _ (Expr expr) = Expr expr;
wenzelm@15703
   621
wenzelm@15703
   622
wenzelm@12277
   623
(* renaming *)
wenzelm@12263
   624
ballarin@16102
   625
(* ren maps names to (new) names and syntax; represented as association list *)
wenzelm@12263
   626
wenzelm@15703
   627
fun rename_var ren (x, mx) =
ballarin@16102
   628
  case assoc_string (ren, x) of
ballarin@16102
   629
      NONE => (x, mx)
ballarin@16102
   630
    | SOME (x', NONE) =>
ballarin@16102
   631
        (x', if mx = NONE then mx else SOME Syntax.NoSyn)     (*drop syntax*)
ballarin@16102
   632
    | SOME (x', SOME mx') =>
ballarin@16102
   633
        if mx = NONE then raise ERROR_MESSAGE
ballarin@16102
   634
          ("Attempt to change syntax of structure parameter " ^ quote x)
ballarin@16102
   635
        else (x', SOME mx');                                (*change syntax*)
ballarin@16102
   636
ballarin@16102
   637
fun rename ren x =
ballarin@16102
   638
  case assoc_string (ren, x) of
ballarin@16102
   639
      NONE => x
ballarin@16102
   640
    | SOME (x', _) => x';                                   (*ignore syntax*)
wenzelm@15703
   641
wenzelm@12263
   642
fun rename_term ren (Free (x, T)) = Free (rename ren x, T)
wenzelm@12263
   643
  | rename_term ren (t $ u) = rename_term ren t $ rename_term ren u
wenzelm@12263
   644
  | rename_term ren (Abs (x, T, t)) = Abs (x, T, rename_term ren t)
wenzelm@12263
   645
  | rename_term _ a = a;
wenzelm@12263
   646
wenzelm@12263
   647
fun rename_thm ren th =
wenzelm@12263
   648
  let
wenzelm@16458
   649
    val {thy, hyps, prop, maxidx, ...} = Thm.rep_thm th;
wenzelm@16458
   650
    val cert = Thm.cterm_of thy;
wenzelm@16861
   651
    val (xs, Ts) = Library.split_list (fold Term.add_frees (prop :: hyps) []);
wenzelm@12263
   652
    val xs' = map (rename ren) xs;
wenzelm@12263
   653
    fun cert_frees names = map (cert o Free) (names ~~ Ts);
wenzelm@12263
   654
    fun cert_vars names = map (cert o Var o apfst (rpair (maxidx + 1))) (names ~~ Ts);
wenzelm@12263
   655
  in
wenzelm@12263
   656
    if xs = xs' then th
wenzelm@12263
   657
    else
wenzelm@12263
   658
      th
wenzelm@12263
   659
      |> Drule.implies_intr_list (map cert hyps)
wenzelm@12263
   660
      |> Drule.forall_intr_list (cert_frees xs)
wenzelm@12263
   661
      |> Drule.forall_elim_list (cert_vars xs)
wenzelm@12263
   662
      |> Thm.instantiate ([], cert_vars xs ~~ cert_frees xs')
wenzelm@12263
   663
      |> (fn th' => Drule.implies_elim_list th' (map (Thm.assume o cert o rename_term ren) hyps))
wenzelm@12263
   664
  end;
wenzelm@12263
   665
wenzelm@15703
   666
fun rename_elem ren =
wenzelm@15703
   667
  map_values I (rename_term ren) (rename_thm ren) o
wenzelm@15703
   668
  map_elem {name = I, typ = I, term = I, fact = I, attrib = I, var = rename_var ren};
wenzelm@12263
   669
wenzelm@16144
   670
fun rename_facts prfx =
wenzelm@16144
   671
  map_elem {var = I, typ = I, term = I, fact = I, attrib = I, name = NameSpace.qualified prfx};
wenzelm@12307
   672
wenzelm@12263
   673
wenzelm@12502
   674
(* type instantiation *)
wenzelm@12502
   675
wenzelm@12502
   676
fun inst_type [] T = T
skalberg@15570
   677
  | inst_type env T = Term.map_type_tfree (fn v => getOpt (assoc (env, v), TFree v)) T;
wenzelm@12502
   678
wenzelm@12502
   679
fun inst_term [] t = t
wenzelm@12502
   680
  | inst_term env t = Term.map_term_types (inst_type env) t;
wenzelm@12502
   681
wenzelm@13211
   682
fun inst_thm _ [] th = th
wenzelm@13211
   683
  | inst_thm ctxt env th =
wenzelm@12502
   684
      let
wenzelm@16458
   685
        val thy = ProofContext.theory_of ctxt;
wenzelm@16458
   686
        val cert = Thm.cterm_of thy;
wenzelm@16458
   687
        val certT = Thm.ctyp_of thy;
wenzelm@13211
   688
        val {hyps, prop, maxidx, ...} = Thm.rep_thm th;
skalberg@15574
   689
        val tfrees = foldr Term.add_term_tfree_names [] (prop :: hyps);
skalberg@15570
   690
        val env' = List.filter (fn ((a, _), _) => a mem_string tfrees) env;
wenzelm@12502
   691
      in
wenzelm@12502
   692
        if null env' then th
wenzelm@12502
   693
        else
wenzelm@12502
   694
          th
wenzelm@12502
   695
          |> Drule.implies_intr_list (map cert hyps)
wenzelm@12575
   696
          |> Drule.tvars_intr_list (map (#1 o #1) env')
wenzelm@12502
   697
          |> (fn (th', al) => th' |>
berghofe@15798
   698
            Thm.instantiate ((map (fn ((a, _), T) =>
berghofe@15798
   699
              (certT (TVar (valOf (assoc (al, a)))), certT T)) env'), []))
wenzelm@12502
   700
          |> (fn th'' => Drule.implies_elim_list th''
wenzelm@12502
   701
              (map (Thm.assume o cert o inst_term env') hyps))
wenzelm@12502
   702
      end;
wenzelm@12502
   703
wenzelm@15703
   704
fun inst_elem ctxt env =
wenzelm@15703
   705
  map_values (inst_type env) (inst_term env) (inst_thm ctxt env);
wenzelm@12502
   706
wenzelm@12502
   707
ballarin@15696
   708
(* term and type instantiation, variant using symbol tables *)
ballarin@15696
   709
ballarin@15696
   710
(* instantiate TFrees *)
ballarin@15696
   711
wenzelm@16458
   712
fun tinst_tab_elem thy tinst =
wenzelm@16458
   713
  map_values (tinst_tab_type tinst) (tinst_tab_term tinst) (tinst_tab_thm thy tinst);
wenzelm@15749
   714
ballarin@15696
   715
(* instantiate TFrees and Frees *)
ballarin@15696
   716
wenzelm@16458
   717
fun inst_tab_elem thy (inst as (_, tinst)) =
wenzelm@16458
   718
  map_values (tinst_tab_type tinst) (inst_tab_term inst) (inst_tab_thm thy inst);
ballarin@15696
   719
wenzelm@16458
   720
fun inst_tab_elems thy inst ((n, ps), elems) =
wenzelm@16458
   721
      ((n, map (inst_tab_term inst) ps), map (inst_tab_elem thy inst) elems);
ballarin@15696
   722
wenzelm@12529
   723
wenzelm@12529
   724
(** structured contexts: rename + merge + implicit type instantiation **)
wenzelm@12529
   725
wenzelm@12529
   726
(* parameter types *)
wenzelm@12529
   727
ballarin@14508
   728
(* CB: frozen_tvars has the following type:
berghofe@15798
   729
  ProofContext.context -> Term.typ list -> (Term.indexname * (sort * Term.typ)) list *)
ballarin@14508
   730
wenzelm@12529
   731
fun frozen_tvars ctxt Ts =
wenzelm@12529
   732
  let
wenzelm@16861
   733
    val tvars = rev (fold Term.add_tvarsT Ts []);
wenzelm@12529
   734
    val tfrees = map TFree
wenzelm@14695
   735
      (Term.invent_names (ProofContext.used_types ctxt) "'a" (length tvars) ~~ map #2 tvars);
berghofe@15798
   736
  in map (fn ((x, S), y) => (x, (S, y))) (tvars ~~ tfrees) end;
wenzelm@12529
   737
wenzelm@12529
   738
fun unify_frozen ctxt maxidx Ts Us =
wenzelm@12529
   739
  let
skalberg@15531
   740
    fun paramify (i, NONE) = (i, NONE)
skalberg@15531
   741
      | paramify (i, SOME T) = apsnd SOME (TypeInfer.paramify_dummies (i, T));
wenzelm@12529
   742
wenzelm@12529
   743
    val (maxidx', Ts') = foldl_map paramify (maxidx, Ts);
wenzelm@12727
   744
    val (maxidx'', Us') = foldl_map paramify (maxidx', Us);
wenzelm@16947
   745
    val thy = ProofContext.theory_of ctxt;
ballarin@14215
   746
wenzelm@16947
   747
    fun unify (env, (SOME T, SOME U)) = (Sign.typ_unify thy (U, T) env
ballarin@14215
   748
          handle Type.TUNIFY =>
ballarin@14215
   749
            raise TYPE ("unify_frozen: failed to unify types", [U, T], []))
ballarin@14215
   750
      | unify (env, _) = env;
skalberg@15570
   751
    val (unifier, _) = Library.foldl unify ((Vartab.empty, maxidx''), Ts' ~~ Us');
skalberg@15570
   752
    val Vs = map (Option.map (Envir.norm_type unifier)) Us';
skalberg@15570
   753
    val unifier' = Vartab.extend (unifier, frozen_tvars ctxt (List.mapPartial I Vs));
skalberg@15570
   754
  in map (Option.map (Envir.norm_type unifier')) Vs end;
wenzelm@12529
   755
skalberg@15570
   756
fun params_of elemss = gen_distinct eq_fst (List.concat (map (snd o fst) elemss));
skalberg@15570
   757
fun params_of' elemss = gen_distinct eq_fst (List.concat (map (snd o fst o fst) elemss));
ballarin@16102
   758
fun params_syn_of syn elemss =
ballarin@16102
   759
  gen_distinct eq_fst (List.concat (map (snd o fst) elemss)) |>
wenzelm@17221
   760
    map (apfst (fn x => (x, valOf (Symtab.curried_lookup syn x))));
ballarin@16102
   761
ballarin@14508
   762
ballarin@14508
   763
(* CB: param_types has the following type:
skalberg@15531
   764
  ('a * 'b option) list -> ('a * 'b) list *)
skalberg@15570
   765
fun param_types ps = List.mapPartial (fn (_, NONE) => NONE | (x, SOME T) => SOME (x, T)) ps;
wenzelm@12529
   766
wenzelm@12529
   767
ballarin@16102
   768
fun merge_syntax ctxt ids ss = Symtab.merge (op =) ss
ballarin@16102
   769
  handle Symtab.DUPS xs => err_in_locale ctxt
ballarin@16105
   770
    ("Conflicting syntax for parameter(s): " ^ commas_quote xs) (map fst ids);
ballarin@16102
   771
ballarin@16102
   772
ballarin@17000
   773
(* Distinction of assumed vs. derived identifiers.
ballarin@17000
   774
   The former may have axioms relating assumptions of the context to
ballarin@17000
   775
   assumptions of the specification fragment (for locales with
ballarin@17000
   776
   predicates).  The latter have witness theorems relating assumptions of the
ballarin@17000
   777
   specification fragment to assumptions of other (assumed) specification
ballarin@17000
   778
   fragments. *)
ballarin@17000
   779
ballarin@17000
   780
datatype 'a mode = Assumed of 'a | Derived of 'a;
ballarin@17000
   781
ballarin@17000
   782
fun map_mode f (Assumed x) = Assumed (f x)
ballarin@17000
   783
  | map_mode f (Derived x) = Derived (f x);
ballarin@17000
   784
wenzelm@12529
   785
(* flatten expressions *)
wenzelm@11896
   786
wenzelm@12510
   787
local
wenzelm@12502
   788
ballarin@15206
   789
(* CB: OUTDATED unique_parms has the following type:
ballarin@14508
   790
     'a ->
ballarin@14508
   791
     (('b * (('c * 'd) list * Symtab.key list)) * 'e) list ->
ballarin@14508
   792
     (('b * ('c * 'd) list) * 'e) list  *)
ballarin@14508
   793
wenzelm@12529
   794
fun unique_parms ctxt elemss =
wenzelm@12529
   795
  let
wenzelm@12529
   796
    val param_decls =
skalberg@15570
   797
      List.concat (map (fn (((name, (ps, qs)), _), _) => map (rpair (name, ps)) qs) elemss)
wenzelm@12529
   798
      |> Symtab.make_multi |> Symtab.dest;
wenzelm@12529
   799
  in
wenzelm@12529
   800
    (case find_first (fn (_, ids) => length ids > 1) param_decls of
skalberg@15531
   801
      SOME (q, ids) => err_in_locale ctxt ("Multiple declaration of parameter " ^ quote q)
wenzelm@12529
   802
          (map (apsnd (map fst)) ids)
skalberg@15531
   803
    | NONE => map (apfst (apfst (apsnd #1))) elemss)
wenzelm@12529
   804
  end;
wenzelm@12529
   805
ballarin@16102
   806
fun unify_parms ctxt (fixed_parms : (string * typ) list)
ballarin@16102
   807
  (raw_parmss : (string * typ option) list list) =
wenzelm@12502
   808
  let
wenzelm@16458
   809
    val thy = ProofContext.theory_of ctxt;
wenzelm@12502
   810
    val maxidx = length raw_parmss;
wenzelm@12502
   811
    val idx_parmss = (0 upto maxidx - 1) ~~ raw_parmss;
wenzelm@12502
   812
wenzelm@12502
   813
    fun varify i = Term.map_type_tfree (fn (a, S) => TVar ((a, i), S));
wenzelm@12529
   814
    fun varify_parms (i, ps) = map (apsnd (varify i)) (param_types ps);
skalberg@15570
   815
    val parms = fixed_parms @ List.concat (map varify_parms idx_parmss);
wenzelm@12502
   816
ballarin@15206
   817
    fun unify T ((env, maxidx), U) =
wenzelm@16947
   818
      Sign.typ_unify thy (U, T) (env, maxidx)
ballarin@15206
   819
      handle Type.TUNIFY =>
wenzelm@16458
   820
        let val prt = Sign.string_of_typ thy
ballarin@15206
   821
        in raise TYPE ("unify_parms: failed to unify types " ^
ballarin@15206
   822
          prt U ^ " and " ^ prt T, [U, T], [])
ballarin@15206
   823
        end
skalberg@15570
   824
    fun unify_list (envir, T :: Us) = Library.foldl (unify T) (envir, Us)
wenzelm@12502
   825
      | unify_list (envir, []) = envir;
skalberg@15570
   826
    val (unifier, _) = Library.foldl unify_list
wenzelm@12502
   827
      ((Vartab.empty, maxidx), map #2 (Symtab.dest (Symtab.make_multi parms)));
wenzelm@12502
   828
wenzelm@12502
   829
    val parms' = map (apsnd (Envir.norm_type unifier)) (gen_distinct eq_fst parms);
wenzelm@12502
   830
    val unifier' = Vartab.extend (unifier, frozen_tvars ctxt (map #2 parms'));
wenzelm@12502
   831
wenzelm@12502
   832
    fun inst_parms (i, ps) =
skalberg@15574
   833
      foldr Term.add_typ_tfrees [] (List.mapPartial snd ps)
skalberg@15570
   834
      |> List.mapPartial (fn (a, S) =>
wenzelm@12502
   835
          let val T = Envir.norm_type unifier' (TVar ((a, i), S))
skalberg@15531
   836
          in if T = TFree (a, S) then NONE else SOME ((a, S), T) end)
ballarin@16102
   837
  in map inst_parms idx_parmss end : ((string * Term.sort) * Term.typ) list list;
wenzelm@12502
   838
wenzelm@12529
   839
in
wenzelm@12502
   840
wenzelm@12529
   841
fun unify_elemss _ _ [] = []
wenzelm@12529
   842
  | unify_elemss _ [] [elems] = [elems]
wenzelm@12529
   843
  | unify_elemss ctxt fixed_parms elemss =
wenzelm@12502
   844
      let
ballarin@15206
   845
        val envs = unify_parms ctxt fixed_parms (map (#2 o #1 o #1) elemss);
ballarin@17000
   846
        fun inst ((((name, ps), mode), elems), env) =
skalberg@15570
   847
          (((name, map (apsnd (Option.map (inst_type env))) ps), 
ballarin@17000
   848
              map_mode (map (inst_thm ctxt env)) mode),
ballarin@17000
   849
            map (inst_elem ctxt env) elems);
wenzelm@12839
   850
      in map inst (elemss ~~ envs) end;
wenzelm@12502
   851
ballarin@17000
   852
(* like unify_elemss, but does not touch mode, additional
ballarin@16736
   853
   parameter c_parms for enforcing further constraints (eg. syntax) *)
ballarin@16102
   854
ballarin@16102
   855
fun unify_elemss' _ _ [] [] = []
ballarin@16102
   856
  | unify_elemss' _ [] [elems] [] = [elems]
ballarin@16102
   857
  | unify_elemss' ctxt fixed_parms elemss c_parms =
ballarin@16102
   858
      let
ballarin@16102
   859
        val envs = unify_parms ctxt fixed_parms (map (#2 o #1 o #1) elemss @ map single c_parms);
ballarin@17033
   860
        fun inst ((((name, ps), (ps', mode)), elems), env) =
ballarin@17033
   861
          (((name, map (apsnd (Option.map (inst_type env))) ps),
ballarin@17033
   862
              (ps', mode)),
ballarin@16102
   863
           map (inst_elem ctxt env) elems);
ballarin@16102
   864
      in map inst (elemss ~~ (Library.take (length elemss, envs))) end;
ballarin@16102
   865
ballarin@17000
   866
ballarin@15596
   867
(* flatten_expr:
ballarin@15596
   868
   Extend list of identifiers by those new in locale expression expr.
ballarin@15596
   869
   Compute corresponding list of lists of locale elements (one entry per
ballarin@15596
   870
   identifier).
ballarin@15596
   871
ballarin@15596
   872
   Identifiers represent locale fragments and are in an extended form:
ballarin@15596
   873
     ((name, ps), (ax_ps, axs))
ballarin@15596
   874
   (name, ps) is the locale name with all its parameters.
ballarin@15596
   875
   (ax_ps, axs) is the locale axioms with its parameters;
ballarin@15596
   876
     axs are always taken from the top level of the locale hierarchy,
ballarin@15596
   877
     hence axioms may contain additional parameters from later fragments:
ballarin@15596
   878
     ps subset of ax_ps.  axs is either singleton or empty.
ballarin@15596
   879
ballarin@15596
   880
   Elements are enriched by identifier-like information:
ballarin@15596
   881
     (((name, ax_ps), axs), elems)
ballarin@15596
   882
   The parameters in ax_ps are the axiom parameters, but enriched by type
ballarin@15596
   883
   info: now each entry is a pair of string and typ option.  Axioms are
ballarin@15596
   884
   type-instantiated.
ballarin@15596
   885
ballarin@15596
   886
*)
ballarin@15596
   887
ballarin@16102
   888
fun flatten_expr ctxt ((prev_idents, prev_syntax), expr) =
wenzelm@12014
   889
  let
wenzelm@12014
   890
    val thy = ProofContext.theory_of ctxt;
ballarin@15596
   891
    (* thy used for retrieval of locale info,
ballarin@15596
   892
       ctxt for error messages, parameter unification and instantiation
ballarin@15596
   893
       of axioms *)
wenzelm@12263
   894
skalberg@15531
   895
    fun renaming (SOME x :: xs) (y :: ys) = (y, x) :: renaming xs ys
skalberg@15531
   896
      | renaming (NONE :: xs) (y :: ys) = renaming xs ys
wenzelm@12273
   897
      | renaming [] _ = []
wenzelm@12289
   898
      | renaming xs [] = raise ERROR_MESSAGE ("Too many arguments in renaming: " ^
ballarin@16102
   899
          commas (map (fn NONE => "_" | SOME x => quote (fst x)) xs));
wenzelm@12289
   900
ballarin@17000
   901
    fun rename_parms top ren ((name, ps), (parms, mode)) =
wenzelm@12289
   902
      let val ps' = map (rename ren) ps in
ballarin@17000
   903
        (case duplicates ps' of
ballarin@17096
   904
          [] => ((name, ps'),
ballarin@17096
   905
                 if top then (map (rename ren) parms,
ballarin@17096
   906
                   map_mode (map (rename_thm ren)) mode)
ballarin@17096
   907
                 else (parms, mode))
wenzelm@12289
   908
        | dups => err_in_locale ctxt ("Duplicate parameters: " ^ commas_quote dups) [(name, ps')])
wenzelm@12289
   909
      end;
wenzelm@12263
   910
ballarin@17000
   911
    (* add registrations of (name, ps), recursively;
ballarin@17000
   912
       adjust hyps of witness theorems *)
ballarin@17000
   913
ballarin@17000
   914
    fun add_regs (name, ps) (ths, ids) =
ballarin@17000
   915
        let
ballarin@17000
   916
          val {params, regs, ...} = the_locale thy name;
ballarin@17096
   917
          val ps' = map #1 (#1 params);
ballarin@17096
   918
          val ren = map #1 ps' ~~ map (fn (x, _) => (x, NONE)) ps;
ballarin@17000
   919
            (* dummy syntax, since required by rename *)
ballarin@17096
   920
          val ps'' = map (fn ((p, _), (_, T)) => (p, T)) (ps ~~ ps');
ballarin@17096
   921
          val [env] = unify_parms ctxt ps [map (apsnd SOME) ps''];
ballarin@17096
   922
            (* propagate parameter types, to keep them consistent *)
ballarin@17000
   923
          val regs' = map (fn ((name, ps), ths) =>
ballarin@17000
   924
              ((name, map (rename ren) ps), ths)) regs;
ballarin@17000
   925
          val new_regs = gen_rems eq_fst (regs', ids);
ballarin@17000
   926
          val new_ids = map fst new_regs;
ballarin@17096
   927
          val new_idTs = map (apsnd (map (fn p => (p, valOf (assoc (ps, p)))))) new_ids;
ballarin@17096
   928
ballarin@17000
   929
          val new_ths = map (fn (_, ths') =>
ballarin@17096
   930
              map (Drule.satisfy_hyps ths o rename_thm ren o inst_thm ctxt env) ths') new_regs;
ballarin@17000
   931
          val new_ids' = map (fn (id, ths) =>
ballarin@17000
   932
              (id, ([], Derived ths))) (new_ids ~~ new_ths);
ballarin@17000
   933
        in
ballarin@17096
   934
          fold add_regs new_idTs (ths @ List.concat new_ths, ids @ new_ids')
ballarin@17000
   935
        end;
ballarin@17000
   936
ballarin@17000
   937
    (* distribute top-level axioms over assumed ids *)
ballarin@17000
   938
ballarin@17000
   939
    fun axiomify all_ps ((name, parms), (_, Assumed _)) axioms =
ballarin@17000
   940
        let
ballarin@17000
   941
          val {elems, ...} = the_locale thy name;
ballarin@17000
   942
          val ts = List.concat (map
ballarin@17000
   943
            (fn (Assumes asms, _) => List.concat (map (map #1 o #2) asms)
ballarin@17000
   944
              | _ => [])
ballarin@17000
   945
            elems);
ballarin@17000
   946
          val (axs1, axs2) = splitAt (length ts, axioms);
ballarin@17000
   947
        in (((name, parms), (all_ps, Assumed axs1)), axs2) end
ballarin@17000
   948
      | axiomify all_ps (id, (_, Derived ths)) axioms =
ballarin@17000
   949
          ((id, (all_ps, Derived ths)), axioms);
ballarin@17000
   950
ballarin@17096
   951
    (* identifiers of an expression *)
ballarin@17096
   952
ballarin@15206
   953
    fun identify top (Locale name) =
ballarin@15596
   954
    (* CB: ids_ax is a list of tuples of the form ((name, ps), axs),
ballarin@15206
   955
       where name is a locale name, ps a list of parameter names and axs
ballarin@15206
   956
       a list of axioms relating to the identifier, axs is empty unless
ballarin@15206
   957
       identify at top level (top = true);
ballarin@14215
   958
       parms is accumulated list of parameters *)
wenzelm@12289
   959
          let
ballarin@15206
   960
            val {predicate = (_, axioms), import, params, ...} =
ballarin@15206
   961
              the_locale thy name;
ballarin@16102
   962
            val ps = map (#1 o #1) (#1 params);
ballarin@17096
   963
            val (ids', parms', _) = identify false import;
ballarin@15206
   964
                (* acyclic import dependencies *)
ballarin@17000
   965
            val ids'' = ids' @ [((name, ps), ([], Assumed []))];
ballarin@17096
   966
            val (_, ids''') = add_regs (name, map #1 (#1 params)) ([], ids'');
ballarin@17000
   967
ballarin@17000
   968
            val ids_ax = if top then fst
ballarin@17000
   969
                 (fold_map (axiomify ps) ids''' axioms)
ballarin@17000
   970
              else ids''';
ballarin@16102
   971
            val syn = Symtab.make (map (apfst fst) (#1 params));
ballarin@17096
   972
            in (ids_ax, merge_lists parms' ps, syn) end
ballarin@15206
   973
      | identify top (Rename (e, xs)) =
wenzelm@12273
   974
          let
ballarin@17096
   975
            val (ids', parms', syn') = identify top e;
wenzelm@12839
   976
            val ren = renaming xs parms'
ballarin@15206
   977
              handle ERROR_MESSAGE msg => err_in_locale' ctxt msg ids';
ballarin@17096
   978
ballarin@15206
   979
            val ids'' = gen_distinct eq_fst (map (rename_parms top ren) ids');
skalberg@15570
   980
            val parms'' = distinct (List.concat (map (#2 o #1) ids''));
ballarin@16102
   981
            val syn'' = syn' |> Symtab.dest |> map (rename_var ren) |>
ballarin@16102
   982
                  Symtab.make;
ballarin@16102
   983
            (* check for conflicting syntax? *)
ballarin@17096
   984
          in (ids'', parms'', syn'') end
ballarin@15206
   985
      | identify top (Merge es) =
ballarin@17096
   986
          fold (fn e => fn (ids, parms, syn) =>
ballarin@17000
   987
                   let
ballarin@17096
   988
                     val (ids', parms', syn') = identify top e
ballarin@17000
   989
                   in
ballarin@17000
   990
                     (merge_alists ids ids',
ballarin@17000
   991
                      merge_lists parms parms',
ballarin@17096
   992
                      merge_syntax ctxt ids' (syn, syn'))
ballarin@17000
   993
                   end)
ballarin@17096
   994
            es ([], [], Symtab.empty);
ballarin@17000
   995
wenzelm@12014
   996
ballarin@15206
   997
    (* CB: enrich identifiers by parameter types and 
ballarin@16102
   998
       the corresponding elements (with renamed parameters),
ballarin@16102
   999
       also takes care of parameter syntax *)
ballarin@15206
  1000
ballarin@16102
  1001
    fun eval syn ((name, xs), axs) =
wenzelm@12273
  1002
      let
wenzelm@13308
  1003
        val {params = (ps, qs), elems, ...} = the_locale thy name;
ballarin@16620
  1004
        val ps' = map (apsnd SOME o #1) ps;
ballarin@16102
  1005
        val ren = map #1 ps' ~~
wenzelm@17221
  1006
              map (fn x => (x, valOf (Symtab.curried_lookup syn x))) xs;
wenzelm@13308
  1007
        val (params', elems') =
ballarin@16102
  1008
          if null ren then ((ps', qs), map #1 elems)
ballarin@16102
  1009
          else ((map (apfst (rename ren)) ps', map (rename ren) qs),
wenzelm@13308
  1010
            map (rename_elem ren o #1) elems);
wenzelm@13375
  1011
        val elems'' = map (rename_facts (space_implode "_" xs)) elems';
ballarin@15206
  1012
      in (((name, params'), axs), elems'') end;
wenzelm@12307
  1013
ballarin@16102
  1014
    (* type constraint for renamed parameter with syntax *)
ballarin@16102
  1015
    fun type_syntax NONE = NONE
ballarin@16102
  1016
      | type_syntax (SOME mx) = let
ballarin@16102
  1017
            val Ts = map (fn x => TFree (x, [])) (Term.invent_names [] "'mxa"
ballarin@16102
  1018
              (Syntax.mixfix_args mx + 1))
ballarin@16102
  1019
          in Ts |> Library.split_last |> op ---> |> SOME end;
ballarin@16102
  1020
ballarin@16102
  1021
    (* compute identifiers and syntax, merge with previous ones *)
ballarin@17096
  1022
    val (ids, _, syn) = identify true expr;
ballarin@16102
  1023
    val idents = gen_rems eq_fst (ids, prev_idents);
ballarin@16102
  1024
    val syntax = merge_syntax ctxt ids (syn, prev_syntax);
ballarin@15206
  1025
    (* add types to params, check for unique params and unify them *)
ballarin@16102
  1026
    val raw_elemss = unique_parms ctxt (map (eval syntax) idents);
ballarin@16102
  1027
    val elemss = unify_elemss' ctxt [] raw_elemss
ballarin@16102
  1028
         (map (apsnd type_syntax) (Symtab.dest syntax));
ballarin@15206
  1029
    (* replace params in ids by params from axioms,
ballarin@17033
  1030
       adjust types in mode *)
ballarin@15206
  1031
    val all_params' = params_of' elemss;
ballarin@15206
  1032
    val all_params = param_types all_params';
ballarin@17000
  1033
    val elemss' = map (fn (((name, _), (ps, mode)), elems) =>
ballarin@17000
  1034
         (((name, map (fn p => (p, assoc (all_params, p))) ps), mode), elems))
ballarin@15206
  1035
         elemss;
ballarin@17000
  1036
    fun inst_th th = let
ballarin@15206
  1037
         val {hyps, prop, ...} = Thm.rep_thm th;
wenzelm@16861
  1038
         val ps = map (apsnd SOME) (fold Term.add_frees (prop :: hyps) []);
ballarin@15206
  1039
         val [env] = unify_parms ctxt all_params [ps];
ballarin@15206
  1040
         val th' = inst_thm ctxt env th;
ballarin@15206
  1041
       in th' end;
ballarin@17000
  1042
    val final_elemss = map (fn ((id, mode), elems) =>
ballarin@17000
  1043
         ((id, map_mode (map inst_th) mode), elems)) elemss';
ballarin@17000
  1044
ballarin@16102
  1045
  in ((prev_idents @ idents, syntax), final_elemss) end;
wenzelm@12046
  1046
wenzelm@12510
  1047
end;
wenzelm@12510
  1048
wenzelm@12070
  1049
wenzelm@12529
  1050
(* activate elements *)
wenzelm@12273
  1051
wenzelm@12510
  1052
local
wenzelm@12510
  1053
wenzelm@13399
  1054
fun export_axioms axs _ hyps th =
wenzelm@13399
  1055
  th |> Drule.satisfy_hyps axs
ballarin@15206
  1056
     (* CB: replace meta-hyps, using axs, by a single meta-hyp. *)
wenzelm@13399
  1057
  |> Drule.implies_intr_list (Library.drop (length axs, hyps))
ballarin@15206
  1058
     (* CB: turn remaining hyps into assumptions. *)
ballarin@15206
  1059
  |> Seq.single
wenzelm@12263
  1060
ballarin@17000
  1061
ballarin@17000
  1062
(* NB: derived ids contain only facts at this stage *)
ballarin@17000
  1063
ballarin@17000
  1064
fun activate_elem _ ((ctxt, mode), Fixes fixes) =
ballarin@17000
  1065
      ((ctxt |> ProofContext.add_fixes fixes, mode), [])
ballarin@17000
  1066
  | activate_elem _ ((ctxt, mode), Constrains _) =
ballarin@17000
  1067
      ((ctxt, mode), [])
ballarin@17000
  1068
  | activate_elem _ ((ctxt, Assumed axs), Assumes asms) =
wenzelm@13399
  1069
      let
wenzelm@17109
  1070
        val asms' = Attrib.map_specs (Attrib.context_attribute_i ctxt) asms;
wenzelm@15703
  1071
        val ts = List.concat (map (map #1 o #2) asms');
wenzelm@15703
  1072
        val (ps, qs) = splitAt (length ts, axs);
wenzelm@13420
  1073
        val (ctxt', _) =
wenzelm@13399
  1074
          ctxt |> ProofContext.fix_frees ts
wenzelm@15703
  1075
          |> ProofContext.assume_i (export_axioms ps) asms';
ballarin@17000
  1076
      in ((ctxt', Assumed qs), []) end
ballarin@17000
  1077
  | activate_elem _ ((ctxt, Derived ths), Assumes asms) =
ballarin@17000
  1078
      ((ctxt, Derived ths), [])
ballarin@17000
  1079
  | activate_elem _ ((ctxt, Assumed axs), Defines defs) =
ballarin@15596
  1080
      let
wenzelm@17109
  1081
        val defs' = Attrib.map_specs (Attrib.context_attribute_i ctxt) defs;
ballarin@15596
  1082
        val (ctxt', _) =
wenzelm@13399
  1083
        ctxt |> ProofContext.assume_i ProofContext.export_def
wenzelm@15703
  1084
          (defs' |> map (fn ((name, atts), (t, ps)) =>
wenzelm@13399
  1085
            let val (c, t') = ProofContext.cert_def ctxt t
wenzelm@13399
  1086
            in ((if name = "" then Thm.def_name c else name, atts), [(t', (ps, []))]) end))
ballarin@17000
  1087
      in ((ctxt', Assumed axs), []) end
ballarin@17000
  1088
  | activate_elem _ ((ctxt, Derived ths), Defines defs) =
ballarin@17000
  1089
      ((ctxt, Derived ths), [])
ballarin@17000
  1090
  | activate_elem is_ext ((ctxt, mode), Notes facts) =
ballarin@15596
  1091
      let
wenzelm@17109
  1092
        val facts' = Attrib.map_facts (Attrib.context_attribute_i ctxt) facts;
wenzelm@15703
  1093
        val (ctxt', res) = ctxt |> ProofContext.note_thmss_i facts';
ballarin@17000
  1094
      in ((ctxt', mode), if is_ext then res else []) end;
wenzelm@12502
  1095
ballarin@17000
  1096
fun activate_elems (((name, ps), mode), elems) ctxt =
ballarin@17033
  1097
  let
ballarin@17033
  1098
    val ((ctxt', _), res) =
ballarin@17033
  1099
        foldl_map (activate_elem (name = "")) ((ProofContext.qualified_names ctxt, mode), elems)
wenzelm@13399
  1100
      handle ProofContext.CONTEXT (msg, ctxt) => err_in_locale ctxt msg [(name, map fst ps)]
ballarin@15696
  1101
    val ctxt'' = if name = "" then ctxt'
ballarin@15696
  1102
          else let
ballarin@15696
  1103
              val ps' = map (fn (n, SOME T) => Free (n, T)) ps;
ballarin@15696
  1104
              val ctxt'' = put_local_registration (name, ps') ("", []) ctxt'
ballarin@17000
  1105
            in case mode of
ballarin@17000
  1106
                   Assumed axs => fold (fn ax => fn ctxt =>
ballarin@17000
  1107
                      add_local_witness (name, ps')
ballarin@17000
  1108
                       (Thm.assume (Thm.cprop_of ax)) ctxt) axs ctxt''
ballarin@17000
  1109
                 | Derived ths => fold (fn th => fn ctxt =>
ballarin@17000
  1110
                     add_local_witness (name, ps') th ctxt) ths ctxt''
ballarin@15696
  1111
            end
wenzelm@16144
  1112
  in (ProofContext.restore_naming ctxt ctxt'', res) end;
wenzelm@13399
  1113
ballarin@17000
  1114
fun activate_elemss prep_facts =
ballarin@17000
  1115
    fold_map (fn (((name, ps), mode), raw_elems) => fn ctxt =>
ballarin@17000
  1116
      let
ballarin@17000
  1117
        val elems = map (prep_facts ctxt) raw_elems;
ballarin@17000
  1118
        val (ctxt', res) = apsnd List.concat
ballarin@17000
  1119
            (activate_elems (((name, ps), mode), elems) ctxt);
ballarin@17000
  1120
        val elems' = map (map_attrib_elem Args.closure) elems;
ballarin@17000
  1121
      in ((((name, ps), elems'), res), ctxt') end);
wenzelm@12834
  1122
wenzelm@12546
  1123
in
wenzelm@12546
  1124
ballarin@15206
  1125
(* CB: activate_facts prep_facts (ctxt, elemss),
ballarin@15206
  1126
   where elemss is a list of pairs consisting of identifiers and
ballarin@15206
  1127
   context elements, extends ctxt by the context elements yielding
ballarin@15206
  1128
   ctxt' and returns (ctxt', (elemss', facts)).
ballarin@15206
  1129
   Identifiers in the argument are of the form ((name, ps), axs) and
ballarin@15206
  1130
   assumptions use the axioms in the identifiers to set up exporters
ballarin@15206
  1131
   in ctxt'.  elemss' does not contain identifiers and is obtained
ballarin@15206
  1132
   from elemss and the intermediate context with prep_facts.
wenzelm@15703
  1133
   If read_facts or cert_facts is used for prep_facts, these also remove
ballarin@14508
  1134
   the internal/external markers from elemss. *)
ballarin@14508
  1135
ballarin@17000
  1136
fun activate_facts prep_facts (ctxt, args) =
ballarin@17000
  1137
    let
ballarin@17000
  1138
      val (res, ctxt') = activate_elemss prep_facts args ctxt;
ballarin@17000
  1139
    in
ballarin@17000
  1140
      (ctxt', apsnd List.concat (split_list res))
ballarin@17000
  1141
    end;
wenzelm@12546
  1142
wenzelm@15703
  1143
fun activate_note prep_facts (ctxt, args) =
wenzelm@15703
  1144
  let
wenzelm@15703
  1145
    val (ctxt', ([(_, [Notes args'])], facts)) =
ballarin@17000
  1146
      activate_facts prep_facts (ctxt, [((("", []), Assumed []), [Ext (Notes args)])]);
wenzelm@15703
  1147
  in (ctxt', (args', facts)) end;
wenzelm@15703
  1148
wenzelm@12510
  1149
end;
wenzelm@12510
  1150
wenzelm@12307
  1151
ballarin@15696
  1152
(* register elements *)
ballarin@15696
  1153
ballarin@17000
  1154
(* not used
ballarin@15696
  1155
fun register_elems (((_, ps), (((name, ax_ps), axs), _)), ctxt) =
ballarin@15696
  1156
  if name = "" then ctxt
ballarin@15696
  1157
      else let val ps' = map (fn (n, SOME T) => Free (n, T)) ax_ps
ballarin@15696
  1158
          val ctxt' = put_local_registration (name, ps') ("", []) ctxt
ballarin@15696
  1159
        in foldl (fn (ax, ctxt) =>
ballarin@15696
  1160
          add_local_witness (name, ps') ax ctxt) ctxt' axs
ballarin@15696
  1161
        end;
ballarin@15696
  1162
ballarin@15696
  1163
fun register_elemss id_elemss ctxt = 
ballarin@15696
  1164
  foldl register_elems ctxt id_elemss;
ballarin@17000
  1165
*)
ballarin@15696
  1166
ballarin@15696
  1167
wenzelm@12529
  1168
(** prepare context elements **)
wenzelm@12529
  1169
wenzelm@12529
  1170
(* expressions *)
wenzelm@12529
  1171
wenzelm@16458
  1172
fun intern_expr thy (Locale xname) = Locale (intern thy xname)
wenzelm@16458
  1173
  | intern_expr thy (Merge exprs) = Merge (map (intern_expr thy) exprs)
wenzelm@16458
  1174
  | intern_expr thy (Rename (expr, xs)) = Rename (intern_expr thy expr, xs);
wenzelm@12529
  1175
wenzelm@12529
  1176
wenzelm@12529
  1177
(* parameters *)
wenzelm@12502
  1178
wenzelm@12502
  1179
local
wenzelm@12502
  1180
ballarin@16169
  1181
fun prep_parms prep_vars ctxt parms =
ballarin@16169
  1182
  let val vars = snd (foldl_map prep_vars (ctxt, map (fn (x, T) => ([x], T)) parms))
ballarin@16169
  1183
  in map (fn ([x'], T') => (x', T')) vars end;
wenzelm@12529
  1184
wenzelm@12529
  1185
in
wenzelm@12529
  1186
ballarin@16169
  1187
fun read_parms x = prep_parms ProofContext.read_vars x;
ballarin@16169
  1188
fun cert_parms x = prep_parms ProofContext.cert_vars x;
wenzelm@12529
  1189
wenzelm@12529
  1190
end;
wenzelm@12529
  1191
wenzelm@12529
  1192
wenzelm@12529
  1193
(* propositions and bindings *)
wenzelm@12529
  1194
ballarin@17000
  1195
(* flatten (ctxt, prep_expr) ((ids, syn), expr)
ballarin@17000
  1196
   normalises expr (which is either a locale
ballarin@14508
  1197
   expression or a single context element) wrt.
ballarin@14508
  1198
   to the list ids of already accumulated identifiers.
ballarin@16102
  1199
   It returns (ids', syn', elemss) where ids' is an extension of ids
ballarin@14508
  1200
   with identifiers generated for expr, and elemss is the list of
ballarin@16102
  1201
   context elements generated from expr.
ballarin@16102
  1202
   syn and syn' are symtabs mapping parameter names to their syntax.  syn'
ballarin@16102
  1203
   is an extension of syn.
ballarin@16102
  1204
   For details, see flatten_expr.
ballarin@16102
  1205
ballarin@15596
  1206
   Additionally, for a locale expression, the elems are grouped into a single
ballarin@15596
  1207
   Int; individual context elements are marked Ext.  In this case, the
ballarin@15596
  1208
   identifier-like information of the element is as follows:
ballarin@15596
  1209
   - for Fixes: (("", ps), []) where the ps have type info NONE
ballarin@15596
  1210
   - for other elements: (("", []), []).
ballarin@15206
  1211
   The implementation of activate_facts relies on identifier names being
ballarin@15206
  1212
   empty strings for external elements.
ballarin@15596
  1213
*)
ballarin@14508
  1214
ballarin@16102
  1215
fun flatten (ctxt, _) ((ids, syn), Elem (Fixes fixes)) = let
ballarin@17000
  1216
	val ids' = ids @ [(("", map #1 fixes), ([], Assumed []))]
ballarin@16102
  1217
      in
ballarin@16102
  1218
	((ids',
ballarin@16102
  1219
	 merge_syntax ctxt ids'
ballarin@16102
  1220
	   (syn, Symtab.make (map (fn fx => (#1 fx, #3 fx)) fixes))
ballarin@16102
  1221
	   handle Symtab.DUPS xs => err_in_locale ctxt
ballarin@16102
  1222
	     ("Conflicting syntax for parameters: " ^ commas_quote xs)
ballarin@16102
  1223
             (map #1 ids')),
ballarin@17000
  1224
	 [((("", map (rpair NONE o #1) fixes), Assumed []), Ext (Fixes fixes))])
ballarin@16102
  1225
      end
ballarin@16102
  1226
  | flatten _ ((ids, syn), Elem elem) =
ballarin@17000
  1227
      ((ids @ [(("", []), ([], Assumed []))], syn), [((("", []), Assumed []), Ext elem)])
ballarin@16102
  1228
  | flatten (ctxt, prep_expr) ((ids, syn), Expr expr) =
ballarin@16102
  1229
      apsnd (map (apsnd Int)) (flatten_expr ctxt ((ids, syn), prep_expr expr));
ballarin@14508
  1230
wenzelm@12529
  1231
local
wenzelm@12529
  1232
wenzelm@12839
  1233
local
wenzelm@12839
  1234
wenzelm@12727
  1235
fun declare_int_elem (ctxt, Fixes fixes) =
wenzelm@12575
  1236
      (ctxt |> ProofContext.add_fixes (map (fn (x, T, mx) =>
skalberg@15570
  1237
        (x, Option.map (Term.map_type_tfree (TypeInfer.param 0)) T, mx)) fixes), [])
wenzelm@12727
  1238
  | declare_int_elem (ctxt, _) = (ctxt, []);
wenzelm@12529
  1239
ballarin@16169
  1240
fun declare_ext_elem prep_parms (ctxt, Fixes fixes) =
ballarin@16169
  1241
      let
ballarin@16169
  1242
        val parms = map (fn (x, T, _) => (x, T)) fixes;
ballarin@16169
  1243
        val parms' = prep_parms ctxt parms;
ballarin@16169
  1244
        val fixes' = map (fn ((x, T), (_, _, mx)) => (x, T, mx)) (parms' ~~ fixes);
ballarin@16169
  1245
      in (ctxt |> ProofContext.add_fixes fixes', []) end
ballarin@16169
  1246
  | declare_ext_elem prep_parms (ctxt, Constrains csts) =
ballarin@16169
  1247
      let
ballarin@16169
  1248
        val parms = map (fn (x, T) => (x, SOME T)) csts;
ballarin@16169
  1249
        val parms' = prep_parms ctxt parms;
ballarin@16169
  1250
        val ts = map (fn (x, SOME T) => Free (x, T)) parms';
ballarin@16169
  1251
      in (Library.fold ProofContext.declare_term ts ctxt, []) end
wenzelm@12529
  1252
  | declare_ext_elem _ (ctxt, Assumes asms) = (ctxt, map #2 asms)
wenzelm@12529
  1253
  | declare_ext_elem _ (ctxt, Defines defs) = (ctxt, map (fn (_, (t, ps)) => [(t, (ps, []))]) defs)
wenzelm@12529
  1254
  | declare_ext_elem _ (ctxt, Notes facts) = (ctxt, []);
wenzelm@12529
  1255
ballarin@17000
  1256
fun declare_elems prep_parms (ctxt, (((name, ps), Assumed _), elems)) =
ballarin@17000
  1257
    let val (ctxt', propps) =
ballarin@17000
  1258
      (case elems of
ballarin@17000
  1259
        Int es => foldl_map declare_int_elem (ctxt, es)
ballarin@17000
  1260
      | Ext e => foldl_map (declare_ext_elem prep_parms) (ctxt, [e]))
ballarin@17000
  1261
      handle ProofContext.CONTEXT (msg, ctxt) =>
ballarin@17000
  1262
        err_in_locale ctxt msg [(name, map fst ps)]
ballarin@17000
  1263
    in (ctxt', propps) end
ballarin@17000
  1264
  | declare_elems _ (ctxt, ((_, Derived _), elems)) = (ctxt, []);
wenzelm@12727
  1265
wenzelm@12839
  1266
in
wenzelm@12839
  1267
ballarin@14215
  1268
(* CB: only called by prep_elemss. *)
ballarin@14215
  1269
ballarin@16169
  1270
fun declare_elemss prep_parms fixed_params raw_elemss ctxt =
wenzelm@12727
  1271
  let
ballarin@14215
  1272
    (* CB: fix of type bug of goal in target with context elements.
ballarin@14215
  1273
       Parameters new in context elements must receive types that are
ballarin@14215
  1274
       distinct from types of parameters in target (fixed_params).  *)
ballarin@14215
  1275
    val ctxt_with_fixed =
wenzelm@16028
  1276
      fold ProofContext.declare_term (map Free fixed_params) ctxt;
wenzelm@12727
  1277
    val int_elemss =
wenzelm@12727
  1278
      raw_elemss
skalberg@15570
  1279
      |> List.mapPartial (fn (id, Int es) => SOME (id, es) | _ => NONE)
ballarin@14215
  1280
      |> unify_elemss ctxt_with_fixed fixed_params;
wenzelm@12727
  1281
    val (_, raw_elemss') =
wenzelm@12727
  1282
      foldl_map (fn ((_, es) :: elemss, (id, Int _)) => (elemss, (id, Int es)) | x => x)
wenzelm@12727
  1283
        (int_elemss, raw_elemss);
ballarin@16169
  1284
  in foldl_map (declare_elems prep_parms) (ctxt, raw_elemss') end;
wenzelm@12529
  1285
wenzelm@12839
  1286
end;
wenzelm@12529
  1287
wenzelm@12839
  1288
local
wenzelm@12839
  1289
ballarin@15596
  1290
(* CB: normalise Assumes and Defines wrt. previous definitions *)
ballarin@14508
  1291
wenzelm@12839
  1292
val norm_term = Envir.beta_norm oo Term.subst_atomic;
wenzelm@12839
  1293
ballarin@17000
  1294
ballarin@15596
  1295
(* CB: following code (abstract_term, abstract_thm, bind_def)
ballarin@15596
  1296
   used in eval_text for Defines elements. *)
ballarin@15596
  1297
wenzelm@13336
  1298
fun abstract_term eq =    (*assumes well-formedness according to ProofContext.cert_def*)
wenzelm@12839
  1299
  let
wenzelm@12839
  1300
    val body = Term.strip_all_body eq;
wenzelm@12839
  1301
    val vars = map Free (Term.rename_wrt_term body (Term.strip_all_vars eq));
wenzelm@12839
  1302
    val (lhs, rhs) = Logic.dest_equals (Term.subst_bounds (vars, body));
wenzelm@12839
  1303
    val (f, xs) = Term.strip_comb lhs;
wenzelm@13336
  1304
    val eq' = Term.list_abs_free (map Term.dest_Free xs, rhs);
wenzelm@13336
  1305
  in (Term.dest_Free f, eq') end;
wenzelm@13336
  1306
wenzelm@16458
  1307
fun abstract_thm thy eq =
wenzelm@16458
  1308
  Thm.assume (Thm.cterm_of thy eq) |> Drule.gen_all |> Drule.abs_def;
wenzelm@12502
  1309
wenzelm@13336
  1310
fun bind_def ctxt (name, ps) ((xs, env, ths), eq) =
wenzelm@12839
  1311
  let
wenzelm@13336
  1312
    val ((y, T), b) = abstract_term eq;
wenzelm@13308
  1313
    val b' = norm_term env b;
wenzelm@16458
  1314
    val th = abstract_thm (ProofContext.theory_of ctxt) eq;
wenzelm@13308
  1315
    fun err msg = err_in_locale ctxt (msg ^ ": " ^ quote y) [(name, map fst ps)];
wenzelm@12839
  1316
  in
wenzelm@13308
  1317
    conditional (exists (equal y o #1) xs) (fn () =>
wenzelm@13308
  1318
      err "Attempt to define previously specified variable");
wenzelm@13308
  1319
    conditional (exists (fn (Free (y', _), _) => y = y' | _ => false) env) (fn () =>
wenzelm@13308
  1320
      err "Attempt to redefine variable");
wenzelm@16861
  1321
    (Term.add_frees b' xs, (Free (y, T), b') :: env, th :: ths)
wenzelm@12839
  1322
  end;
wenzelm@12575
  1323
ballarin@17000
  1324
ballarin@17000
  1325
(* CB: for finish_elems (Int and Ext),
ballarin@17000
  1326
   extracts specification, only of assumed elements *)
ballarin@15206
  1327
wenzelm@13308
  1328
fun eval_text _ _ _ (text, Fixes _) = text
ballarin@16169
  1329
  | eval_text _ _ _ (text, Constrains _) = text
ballarin@17000
  1330
  | eval_text _ (_, Assumed _) is_ext ((((exts, exts'), (ints, ints')), (xs, env, defs)), Assumes asms) =
wenzelm@13394
  1331
      let
skalberg@15570
  1332
        val ts = List.concat (map (map #1 o #2) asms);
wenzelm@13394
  1333
        val ts' = map (norm_term env) ts;
wenzelm@13394
  1334
        val spec' =
wenzelm@13394
  1335
          if is_ext then ((exts @ ts, exts' @ ts'), (ints, ints'))
wenzelm@13394
  1336
          else ((exts, exts'), (ints @ ts, ints' @ ts'));
wenzelm@16861
  1337
      in (spec', (fold Term.add_frees ts' xs, env, defs)) end
ballarin@17000
  1338
  | eval_text _ (_, Derived _) _ (text, Assumes _) = text
ballarin@17000
  1339
  | eval_text ctxt (id, Assumed _) _ ((spec, binds), Defines defs) =
skalberg@15570
  1340
      (spec, Library.foldl (bind_def ctxt id) (binds, map (#1 o #2) defs))
ballarin@17000
  1341
  | eval_text _ (_, Derived _) _ (text, Defines _) = text
wenzelm@13308
  1342
  | eval_text _ _ _ (text, Notes _) = text;
wenzelm@13308
  1343
ballarin@17000
  1344
ballarin@17000
  1345
(* for finish_elems (Int),
ballarin@17000
  1346
   remove redundant elements of derived identifiers,
ballarin@17000
  1347
   turn assumptions and definitions into facts,
ballarin@17000
  1348
   adjust hypotheses of facts using witness theorems *)
ballarin@17000
  1349
ballarin@17096
  1350
fun finish_derived _ wits _ (Notes facts) = (Notes facts)
ballarin@17096
  1351
      |> map_values I I (Drule.satisfy_hyps wits) |> SOME;
ballarin@17096
  1352
ballarin@17096
  1353
fun finish_derived _ _ (Assumed _) (Fixes fixes) = SOME (Fixes fixes)
ballarin@17096
  1354
  | finish_derived _ _ (Assumed _) (Constrains csts) = SOME (Constrains csts)
ballarin@17096
  1355
  | finish_derived _ _ (Assumed _) (Assumes asms) = SOME (Assumes asms)
ballarin@17096
  1356
  | finish_derived _ _ (Assumed _) (Defines defs) = SOME (Defines defs)
ballarin@17096
  1357
ballarin@17000
  1358
  | finish_derived _ _ (Derived _) (Fixes _) = NONE
ballarin@17000
  1359
  | finish_derived _ _ (Derived _) (Constrains _) = NONE
ballarin@17000
  1360
  | finish_derived sign wits (Derived _) (Assumes asms) = asms
ballarin@17096
  1361
      |> map (apsnd (map (fn (a, _) => ([Thm.assume (cterm_of sign a)], []))))
ballarin@17096
  1362
      |> Notes |> map_values I I (Drule.satisfy_hyps wits) |> SOME
ballarin@17000
  1363
  | finish_derived sign wits (Derived _) (Defines defs) = defs
ballarin@17096
  1364
      |> map (apsnd (fn (d, _) => [([Thm.assume (cterm_of sign d)], [])]))
ballarin@17096
  1365
      |> Notes |> map_values I I (Drule.satisfy_hyps wits) |> SOME
ballarin@17000
  1366
ballarin@17096
  1367
  | finish_derived _ wits _ (Notes facts) = (Notes facts)
ballarin@17096
  1368
      |> map_values I I (Drule.satisfy_hyps wits) |> SOME;
ballarin@17000
  1369
ballarin@15206
  1370
(* CB: for finish_elems (Ext) *)
ballarin@15206
  1371
wenzelm@13308
  1372
fun closeup _ false elem = elem
wenzelm@13308
  1373
  | closeup ctxt true elem =
wenzelm@12839
  1374
      let
wenzelm@13308
  1375
        fun close_frees t =
wenzelm@13308
  1376
          let val frees = rev (filter_out (ProofContext.is_fixed ctxt o #1)
wenzelm@16861
  1377
            (Term.add_frees t []))
wenzelm@13308
  1378
          in Term.list_all_free (frees, t) end;
wenzelm@13308
  1379
wenzelm@13308
  1380
        fun no_binds [] = []
wenzelm@13308
  1381
          | no_binds _ =
wenzelm@13308
  1382
              raise ProofContext.CONTEXT ("Illegal term bindings in locale element", ctxt);
wenzelm@13308
  1383
      in
wenzelm@13308
  1384
        (case elem of
wenzelm@13308
  1385
          Assumes asms => Assumes (asms |> map (fn (a, propps) =>
wenzelm@13308
  1386
            (a, map (fn (t, (ps, qs)) => (close_frees t, (no_binds ps, no_binds qs))) propps)))
wenzelm@13308
  1387
        | Defines defs => Defines (defs |> map (fn (a, (t, ps)) =>
wenzelm@13308
  1388
            (a, (close_frees (#2 (ProofContext.cert_def ctxt t)), no_binds ps))))
wenzelm@13308
  1389
        | e => e)
wenzelm@13308
  1390
      end;
wenzelm@12839
  1391
wenzelm@12502
  1392
wenzelm@12839
  1393
fun finish_ext_elem parms _ (Fixes fixes, _) = Fixes (map (fn (x, _, mx) =>
wenzelm@12727
  1394
      (x, assoc_string (parms, x), mx)) fixes)
ballarin@16169
  1395
  | finish_ext_elem parms _ (Constrains csts, _) =
ballarin@16169
  1396
      Constrains (map (fn (x, _) => (x, valOf (assoc_string (parms, x)))) csts)
wenzelm@12839
  1397
  | finish_ext_elem _ close (Assumes asms, propp) =
wenzelm@12839
  1398
      close (Assumes (map #1 asms ~~ propp))
wenzelm@12839
  1399
  | finish_ext_elem _ close (Defines defs, propp) =
wenzelm@12727
  1400
      close (Defines (map #1 defs ~~ map (fn [(t, (ps, []))] => (t, ps)) propp))
wenzelm@12839
  1401
  | finish_ext_elem _ _ (Notes facts, _) = Notes facts;
wenzelm@12839
  1402
ballarin@17000
  1403
ballarin@15206
  1404
(* CB: finish_parms introduces type info from parms to identifiers *)
skalberg@15531
  1405
(* CB: only needed for types that have been NONE so far???
ballarin@15206
  1406
   If so, which are these??? *)
ballarin@15206
  1407
ballarin@17000
  1408
fun finish_parms parms (((name, ps), mode), elems) =
ballarin@17000
  1409
  (((name, map (fn (x, _) => (x, assoc (parms, x))) ps), mode), elems);
wenzelm@12839
  1410
ballarin@17000
  1411
fun finish_elems ctxt parms _ ((text, wits), ((id, Int e), _)) =
wenzelm@12839
  1412
      let
ballarin@17000
  1413
        val [(id' as (_, mode), es)] = unify_elemss ctxt parms [(id, e)];
ballarin@17000
  1414
        val wits' = case mode of Assumed _ => wits | Derived ths => wits @ ths;
skalberg@15570
  1415
        val text' = Library.foldl (eval_text ctxt id' false) (text, es);
ballarin@17000
  1416
        val es' = List.mapPartial
ballarin@17000
  1417
              (finish_derived (ProofContext.theory_of ctxt) wits' mode) es;
ballarin@17000
  1418
      in ((text', wits'), (id', map Int es')) end
ballarin@17000
  1419
  | finish_elems ctxt parms do_close ((text, wits), ((id, Ext e), [propp])) =
wenzelm@13308
  1420
      let
wenzelm@13308
  1421
        val e' = finish_ext_elem parms (closeup ctxt do_close) (e, propp);
wenzelm@13375
  1422
        val text' = eval_text ctxt id true (text, e');
ballarin@17000
  1423
      in ((text', wits), (id, [Ext e'])) end
wenzelm@12839
  1424
wenzelm@12839
  1425
in
wenzelm@12510
  1426
ballarin@15206
  1427
(* CB: only called by prep_elemss *)
ballarin@15206
  1428
wenzelm@13375
  1429
fun finish_elemss ctxt parms do_close =
wenzelm@13375
  1430
  foldl_map (apsnd (finish_parms parms) o finish_elems ctxt parms do_close);
wenzelm@12839
  1431
wenzelm@12839
  1432
end;
wenzelm@12839
  1433
ballarin@16736
  1434
ballarin@16736
  1435
(* CB: type inference and consistency checks for locales.
ballarin@16736
  1436
ballarin@16736
  1437
   Works by building a context (through declare_elemss), extracting the
ballarin@16736
  1438
   required information and adjusting the context elements (finish_elemss).
ballarin@16736
  1439
   Can also universally close free vars in assms and defs.  This is only
ballarin@17000
  1440
   needed for Ext elements and controlled by parameter do_close.
ballarin@17000
  1441
ballarin@17000
  1442
   Only elements of assumed identifiers are considered.
ballarin@16736
  1443
*)
ballarin@15127
  1444
ballarin@16169
  1445
fun prep_elemss prep_parms prepp do_close context fixed_params raw_elemss raw_concl =
wenzelm@12529
  1446
  let
ballarin@15127
  1447
    (* CB: contexts computed in the course of this function are discarded.
ballarin@15127
  1448
       They are used for type inference and consistency checks only. *)
ballarin@15206
  1449
    (* CB: fixed_params are the parameters (with types) of the target locale,
ballarin@15206
  1450
       empty list if there is no target. *)
ballarin@14508
  1451
    (* CB: raw_elemss are list of pairs consisting of identifiers and
ballarin@14508
  1452
       context elements, the latter marked as internal or external. *)
ballarin@16169
  1453
    val (raw_ctxt, raw_proppss) = declare_elemss prep_parms fixed_params raw_elemss context;
ballarin@14508
  1454
    (* CB: raw_ctxt is context with additional fixed variables derived from
ballarin@14508
  1455
       the fixes elements in raw_elemss,
ballarin@14508
  1456
       raw_proppss contains assumptions and definitions from the
ballarin@15206
  1457
       external elements in raw_elemss. *)
skalberg@15570
  1458
    val raw_propps = map List.concat raw_proppss;
skalberg@15570
  1459
    val raw_propp = List.concat raw_propps;
ballarin@15206
  1460
wenzelm@16028
  1461
    (* CB: add type information from fixed_params to context (declare_term) *)
ballarin@15206
  1462
    (* CB: process patterns (conclusion and external elements only) *)
wenzelm@12529
  1463
    val (ctxt, all_propp) =
wenzelm@16028
  1464
      prepp (fold ProofContext.declare_term (map Free fixed_params) raw_ctxt, raw_concl @ raw_propp);
ballarin@15206
  1465
    (* CB: add type information from conclusion and external elements
ballarin@15206
  1466
       to context *)
wenzelm@16028
  1467
    val ctxt = fold ProofContext.declare_term (List.concat (map (map fst) all_propp)) ctxt;
wenzelm@12502
  1468
ballarin@15206
  1469
    (* CB: resolve schematic variables (patterns) in conclusion and external
ballarin@15206
  1470
       elements. *)
wenzelm@12529
  1471
    val all_propp' = map2 (op ~~)
wenzelm@12529
  1472
      (#1 (#2 (ProofContext.bind_propp_schematic_i (ctxt, all_propp))), map (map snd) all_propp);
nipkow@13629
  1473
    val (concl, propp) = splitAt(length raw_concl, all_propp');
wenzelm@12529
  1474
    val propps = unflat raw_propps propp;
wenzelm@12839
  1475
    val proppss = map (uncurry unflat) (raw_proppss ~~ propps);
wenzelm@12502
  1476
ballarin@15206
  1477
    (* CB: obtain all parameters from identifier part of raw_elemss *)
ballarin@15206
  1478
    val xs = map #1 (params_of' raw_elemss);
wenzelm@12727
  1479
    val typing = unify_frozen ctxt 0
wenzelm@12529
  1480
      (map (ProofContext.default_type raw_ctxt) xs)
wenzelm@12529
  1481
      (map (ProofContext.default_type ctxt) xs);
wenzelm@12529
  1482
    val parms = param_types (xs ~~ typing);
ballarin@14508
  1483
    (* CB: parms are the parameters from raw_elemss, with correct typing. *)
wenzelm@12273
  1484
ballarin@14508
  1485
    (* CB: extract information from assumes and defines elements
ballarin@16169
  1486
       (fixes, constrains and notes in raw_elemss don't have an effect on
ballarin@16169
  1487
       text and elemss), compute final form of context elements. *)
ballarin@17000
  1488
    val ((text, _), elemss) = finish_elemss ctxt parms do_close
ballarin@17000
  1489
      ((((([], []), ([], [])), ([], [], [])), []), raw_elemss ~~ proppss);
ballarin@14508
  1490
    (* CB: text has the following structure:
ballarin@14508
  1491
           (((exts, exts'), (ints, ints')), (xs, env, defs))
ballarin@14508
  1492
       where
ballarin@14508
  1493
         exts: external assumptions (terms in external assumes elements)
ballarin@14508
  1494
         exts': dito, normalised wrt. env
ballarin@14508
  1495
         ints: internal assumptions (terms in internal assumes elements)
ballarin@14508
  1496
         ints': dito, normalised wrt. env
ballarin@14508
  1497
         xs: the free variables in exts' and ints' and rhss of definitions,
ballarin@14508
  1498
           this includes parameters except defined parameters
ballarin@14508
  1499
         env: list of term pairs encoding substitutions, where the first term
ballarin@14508
  1500
           is a free variable; substitutions represent defines elements and
ballarin@14508
  1501
           the rhs is normalised wrt. the previous env
ballarin@14508
  1502
         defs: theorems representing the substitutions from defines elements
ballarin@14508
  1503
           (thms are normalised wrt. env).
ballarin@14508
  1504
       elemss is an updated version of raw_elemss:
ballarin@16169
  1505
         - type info added to Fixes and modified in Constrains
ballarin@14508
  1506
         - axiom and definition statement replaced by corresponding one
ballarin@14508
  1507
           from proppss in Assumes and Defines
ballarin@14508
  1508
         - Facts unchanged
ballarin@14508
  1509
       *)
wenzelm@13308
  1510
  in ((parms, elemss, concl), text) end;
wenzelm@12502
  1511
wenzelm@12502
  1512
in
wenzelm@12502
  1513
ballarin@16169
  1514
fun read_elemss x = prep_elemss read_parms ProofContext.read_propp_schematic x;
ballarin@16169
  1515
fun cert_elemss x = prep_elemss cert_parms ProofContext.cert_propp_schematic x;
wenzelm@12529
  1516
wenzelm@12529
  1517
end;
wenzelm@12529
  1518
wenzelm@12529
  1519
wenzelm@15703
  1520
(* facts and attributes *)
wenzelm@12529
  1521
wenzelm@12529
  1522
local
wenzelm@12529
  1523
wenzelm@15703
  1524
fun prep_name ctxt name =
ballarin@15206
  1525
  (* CB: reject qualified theorem names in locale declarations *)
wenzelm@12529
  1526
  if NameSpace.is_qualified name then
wenzelm@12529
  1527
    raise ProofContext.CONTEXT ("Illegal qualified name: " ^ quote name, ctxt)
wenzelm@15703
  1528
  else name;
wenzelm@12529
  1529
wenzelm@15703
  1530
fun prep_facts _ _ ctxt (Int elem) =
wenzelm@15703
  1531
      map_values I I (Thm.transfer (ProofContext.theory_of ctxt)) elem
wenzelm@15703
  1532
  | prep_facts get intern ctxt (Ext elem) = elem |> map_elem
wenzelm@15703
  1533
     {var = I, typ = I, term = I,
wenzelm@15703
  1534
      name = prep_name ctxt,
wenzelm@15703
  1535
      fact = get ctxt,
wenzelm@16458
  1536
      attrib = Args.assignable o intern (ProofContext.theory_of ctxt)};
wenzelm@12529
  1537
wenzelm@12529
  1538
in
wenzelm@12529
  1539
wenzelm@15703
  1540
fun read_facts x = prep_facts ProofContext.get_thms Attrib.intern_src x;
wenzelm@15703
  1541
fun cert_facts x = prep_facts (K I) (K I) x;
wenzelm@12529
  1542
wenzelm@12529
  1543
end;
wenzelm@12529
  1544
wenzelm@12529
  1545
wenzelm@12546
  1546
(* full context statements: import + elements + conclusion *)
wenzelm@12529
  1547
wenzelm@12529
  1548
local
wenzelm@12529
  1549
wenzelm@12529
  1550
fun prep_context_statement prep_expr prep_elemss prep_facts
ballarin@15206
  1551
    do_close fixed_params import elements raw_concl context =
wenzelm@12529
  1552
  let
wenzelm@16458
  1553
    val thy = ProofContext.theory_of context;
wenzelm@13375
  1554
wenzelm@16458
  1555
    val ((import_ids, import_syn), raw_import_elemss) =
wenzelm@16458
  1556
      flatten (context, prep_expr thy) (([], Symtab.empty), Expr import);
ballarin@14215
  1557
    (* CB: normalise "includes" among elements *)
wenzelm@16458
  1558
    val ((ids, syn), raw_elemsss) = foldl_map (flatten (context, prep_expr thy))
ballarin@16102
  1559
      ((import_ids, import_syn), elements);
ballarin@15696
  1560
ballarin@15696
  1561
    val raw_elemss = List.concat raw_elemsss;
ballarin@14508
  1562
    (* CB: raw_import_elemss @ raw_elemss is the normalised list of
ballarin@14508
  1563
       context elements obtained from import and elements. *)
wenzelm@13375
  1564
    val ((parms, all_elemss, concl), (spec, (_, _, defs))) = prep_elemss do_close
wenzelm@13336
  1565
      context fixed_params (raw_import_elemss @ raw_elemss) raw_concl;
ballarin@15696
  1566
    (* replace extended ids (for axioms) by ids *)
ballarin@17000
  1567
    val all_elemss' = map (fn (((_, ps), _), (((n, ps'), mode), elems)) =>
ballarin@17000
  1568
        (((n, map (fn p => (p, assoc (ps', p) |> valOf)) ps), mode), elems))
ballarin@15696
  1569
      (ids ~~ all_elemss);
ballarin@15696
  1570
ballarin@15206
  1571
    (* CB: all_elemss and parms contain the correct parameter types *)
ballarin@15696
  1572
    val (ps,qs) = splitAt(length raw_import_elemss, all_elemss')
ballarin@15206
  1573
    val (import_ctxt, (import_elemss, _)) =
ballarin@15206
  1574
      activate_facts prep_facts (context, ps);
ballarin@14215
  1575
ballarin@15206
  1576
    val (ctxt, (elemss, _)) =
ballarin@15206
  1577
      activate_facts prep_facts (import_ctxt, qs);
ballarin@15212
  1578
    val stmt = gen_distinct Term.aconv
ballarin@17000
  1579
       (List.concat (map (fn ((_, Assumed axs), _) =>
ballarin@17000
  1580
         List.concat (map (#hyps o Thm.rep_thm) axs)
ballarin@17000
  1581
                           | ((_, Derived _), _) => []) qs));
wenzelm@16458
  1582
    val cstmt = map (cterm_of thy) stmt;
wenzelm@12834
  1583
  in
ballarin@16102
  1584
    ((((import_ctxt, import_elemss), (ctxt, elemss, syn)), (parms, spec, defs)), (cstmt, concl))
wenzelm@12834
  1585
  end;
wenzelm@12529
  1586
wenzelm@15703
  1587
val gen_context = prep_context_statement intern_expr read_elemss read_facts;
wenzelm@15703
  1588
val gen_context_i = prep_context_statement (K I) cert_elemss cert_facts;
wenzelm@12529
  1589
wenzelm@12529
  1590
fun gen_statement prep_locale prep_ctxt raw_locale elems concl ctxt =
wenzelm@12529
  1591
  let
wenzelm@12529
  1592
    val thy = ProofContext.theory_of ctxt;
wenzelm@16458
  1593
    val locale = Option.map (prep_locale thy) raw_locale;
ballarin@15206
  1594
    val (target_stmt, fixed_params, import) =
skalberg@15531
  1595
      (case locale of NONE => ([], [], empty)
skalberg@15531
  1596
      | SOME name =>
ballarin@15206
  1597
          let val {predicate = (stmt, _), params = (ps, _), ...} =
ballarin@15206
  1598
            the_locale thy name
ballarin@16620
  1599
          in (stmt, map fst ps, Locale name) end);
ballarin@16102
  1600
    val ((((locale_ctxt, locale_elemss), (elems_ctxt, _, _)), _), (elems_stmt, concl')) =
ballarin@15206
  1601
      prep_ctxt false fixed_params import elems concl ctxt;
ballarin@15206
  1602
  in (locale, (target_stmt, elems_stmt), locale_ctxt, elems_ctxt, concl') end;
wenzelm@13399
  1603
wenzelm@12529
  1604
in
wenzelm@12529
  1605
ballarin@15127
  1606
(* CB: processing of locales for add_locale(_i) and print_locale *)
ballarin@15127
  1607
  (* CB: arguments are: x->import, y->body (elements), z->context *)
ballarin@15206
  1608
fun read_context x y z = #1 (gen_context true [] x (map Elem y) [] z);
ballarin@15206
  1609
fun cert_context x y z = #1 (gen_context_i true [] x (map Elem y) [] z);
ballarin@14215
  1610
ballarin@15127
  1611
(* CB: processing of locales for note_thmss(_i),
ballarin@15127
  1612
   Proof.multi_theorem(_i) and antiquotations with option "locale" *)
wenzelm@12529
  1613
val read_context_statement = gen_statement intern gen_context;
wenzelm@12529
  1614
val cert_context_statement = gen_statement (K I) gen_context_i;
wenzelm@12502
  1615
wenzelm@12502
  1616
end;
wenzelm@11896
  1617
wenzelm@11896
  1618
wenzelm@13336
  1619
(** define locales **)
wenzelm@13336
  1620
wenzelm@13336
  1621
(* print locale *)
wenzelm@12070
  1622
ballarin@17228
  1623
fun print_locale thy show_facts import body =
wenzelm@12070
  1624
  let
wenzelm@12289
  1625
    val thy_ctxt = ProofContext.init thy;
ballarin@16102
  1626
    val (((_, import_elemss), (ctxt, elemss, _)), _) = read_context import body thy_ctxt;
skalberg@15570
  1627
    val all_elems = List.concat (map #2 (import_elemss @ elemss));
wenzelm@12070
  1628
wenzelm@12307
  1629
    val prt_typ = Pretty.quote o ProofContext.pretty_typ ctxt;
wenzelm@12307
  1630
    val prt_term = Pretty.quote o ProofContext.pretty_term ctxt;
wenzelm@12307
  1631
    val prt_thm = Pretty.quote o ProofContext.pretty_thm ctxt;
wenzelm@15703
  1632
    val prt_atts = Args.pretty_attribs ctxt;
wenzelm@12070
  1633
wenzelm@12070
  1634
    fun prt_syn syn =
skalberg@15531
  1635
      let val s = (case syn of NONE => "(structure)" | SOME mx => Syntax.string_of_mixfix mx)
wenzelm@12575
  1636
      in if s = "" then [] else [Pretty.brk 2, Pretty.str s] end;
skalberg@15531
  1637
    fun prt_fix (x, SOME T, syn) = Pretty.block (Pretty.str (x ^ " ::") :: Pretty.brk 1 ::
wenzelm@12070
  1638
          prt_typ T :: Pretty.brk 1 :: prt_syn syn)
skalberg@15531
  1639
      | prt_fix (x, NONE, syn) = Pretty.block (Pretty.str x :: Pretty.brk 1 :: prt_syn syn);
ballarin@16169
  1640
    fun prt_cst (x, T) = Pretty.block [Pretty.str (x ^ " ::"), prt_typ T];
wenzelm@12070
  1641
wenzelm@16346
  1642
    fun prt_name name = Pretty.str (ProofContext.extern_thm ctxt name);
wenzelm@15703
  1643
    fun prt_name_atts (name, atts) =
wenzelm@15703
  1644
      if name = "" andalso null atts then []
wenzelm@15703
  1645
      else [Pretty.block (Pretty.breaks (prt_name name :: prt_atts atts @ [Pretty.str ":"]))];
wenzelm@15703
  1646
wenzelm@15703
  1647
    fun prt_asm (a, ts) =
wenzelm@15703
  1648
      Pretty.block (Pretty.breaks (prt_name_atts a @ map (prt_term o fst) ts));
wenzelm@15703
  1649
    fun prt_def (a, (t, _)) =
wenzelm@15703
  1650
      Pretty.block (Pretty.breaks (prt_name_atts a @ [prt_term t]));
wenzelm@15703
  1651
wenzelm@15703
  1652
    fun prt_fact (ths, []) = map prt_thm ths
wenzelm@15703
  1653
      | prt_fact (ths, atts) =
wenzelm@15703
  1654
          Pretty.enclose "(" ")" (Pretty.breaks (map prt_thm ths)) :: prt_atts atts;
wenzelm@15703
  1655
    fun prt_note (a, ths) =
wenzelm@15703
  1656
      Pretty.block (Pretty.breaks (List.concat (prt_name_atts a :: map prt_fact ths)));
wenzelm@12070
  1657
wenzelm@12289
  1658
    fun items _ [] = []
wenzelm@12289
  1659
      | items prfx (x :: xs) = Pretty.block [Pretty.str prfx, Pretty.brk 1, x] :: items "  and" xs;
wenzelm@12289
  1660
    fun prt_elem (Fixes fixes) = items "fixes" (map prt_fix fixes)
ballarin@16169
  1661
      | prt_elem (Constrains csts) = items "constrains" (map prt_cst csts)
wenzelm@12289
  1662
      | prt_elem (Assumes asms) = items "assumes" (map prt_asm asms)
wenzelm@12289
  1663
      | prt_elem (Defines defs) = items "defines" (map prt_def defs)
ballarin@17228
  1664
      | prt_elem (Notes facts) =
ballarin@17228
  1665
        if show_facts then items "notes" (map prt_note facts) else [];
wenzelm@12277
  1666
  in
wenzelm@13336
  1667
    Pretty.big_list "context elements:" (map (Pretty.chunks o prt_elem) all_elems)
wenzelm@13336
  1668
    |> Pretty.writeln
wenzelm@12277
  1669
  end;
wenzelm@12070
  1670
wenzelm@12070
  1671
wenzelm@12706
  1672
wenzelm@16144
  1673
(** store results **)
wenzelm@12702
  1674
wenzelm@16144
  1675
(* note_thmss_qualified *)
ballarin@15696
  1676
wenzelm@14564
  1677
fun note_thmss_qualified kind name args thy =
wenzelm@12706
  1678
  thy
wenzelm@13375
  1679
  |> Theory.add_path (Sign.base_name name)
wenzelm@16144
  1680
  |> Theory.no_base_names
wenzelm@14564
  1681
  |> PureThy.note_thmss_i (Drule.kind kind) args
wenzelm@16144
  1682
  |>> Theory.restore_naming thy;
wenzelm@16144
  1683
wenzelm@12706
  1684
ballarin@15696
  1685
(* accesses of interpreted theorems *)
ballarin@15696
  1686
wenzelm@16144
  1687
local
wenzelm@16144
  1688
wenzelm@16144
  1689
(*fully qualified name in theory is T.p.r.n where
wenzelm@16144
  1690
  T: theory name, p: interpretation prefix, r: renaming prefix, n: name*)
wenzelm@16144
  1691
fun global_accesses _ [] = []
wenzelm@16144
  1692
  | global_accesses "" [T, n] = [[T, n], [n]]
wenzelm@16144
  1693
  | global_accesses "" [T, r, n] = [[T, r, n], [T, n], [r, n], [n]]
wenzelm@16144
  1694
  | global_accesses _ [T, p, n] = [[T, p, n], [p, n]]
wenzelm@16144
  1695
  | global_accesses _ [T, p, r, n] = [[T, p, r, n], [T, p, n], [p, r, n], [p, n]]
wenzelm@16144
  1696
  | global_accesses _ names = error ("Bad name declaration " ^ quote (NameSpace.pack names));
ballarin@15696
  1697
wenzelm@16144
  1698
(*fully qualified name in context is p.r.n where
wenzelm@16144
  1699
  p: interpretation prefix, r: renaming prefix, n: name*)
wenzelm@16144
  1700
fun local_accesses _ [] = []
wenzelm@16144
  1701
  | local_accesses "" [n] = [[n]]
wenzelm@16144
  1702
  | local_accesses "" [r, n] = [[r, n], [n]]
wenzelm@16144
  1703
  | local_accesses _ [p, n] = [[p, n]]
wenzelm@16144
  1704
  | local_accesses _ [p, r, n] = [[p, r, n], [p, n]]
wenzelm@16144
  1705
  | local_accesses _ names = error ("Bad name declaration " ^ quote (NameSpace.pack names));
wenzelm@16144
  1706
wenzelm@16144
  1707
in
ballarin@15696
  1708
wenzelm@16144
  1709
fun global_note_accesses_i kind prfx args thy =
wenzelm@16144
  1710
  thy
wenzelm@16144
  1711
  |> Theory.qualified_names
wenzelm@16144
  1712
  |> Theory.custom_accesses (global_accesses prfx)
wenzelm@16144
  1713
  |> PureThy.note_thmss_i kind args
wenzelm@16144
  1714
  |>> Theory.restore_naming thy;
ballarin@15696
  1715
wenzelm@16144
  1716
fun local_note_accesses_i prfx args ctxt =
wenzelm@16144
  1717
  ctxt
wenzelm@16144
  1718
  |> ProofContext.qualified_names
wenzelm@16144
  1719
  |> ProofContext.custom_accesses (local_accesses prfx)
wenzelm@16144
  1720
  |> ProofContext.note_thmss_i args
wenzelm@16144
  1721
  |>> ProofContext.restore_naming ctxt;
wenzelm@16144
  1722
wenzelm@16144
  1723
end;
ballarin@15696
  1724
ballarin@15696
  1725
ballarin@17138
  1726
(* collect witness theorems for global registration;
ballarin@17138
  1727
   requires parameters and flattened list of (assumed!) identifiers
ballarin@17138
  1728
   instead of recomputing it from the target *)
ballarin@17138
  1729
ballarin@17138
  1730
fun collect_global_witnesses thy parms ids vts = let
ballarin@17138
  1731
    val ts = map Logic.unvarify vts;
ballarin@17138
  1732
    val (parms, parmTs) = split_list parms;
ballarin@17138
  1733
    val parmvTs = map Type.varifyT parmTs;
ballarin@17138
  1734
    val vtinst = fold (Sign.typ_match thy) (parmvTs ~~ map Term.fastype_of ts) Vartab.empty;
ballarin@17138
  1735
    val tinst = Vartab.dest vtinst |> map (fn ((x, 0), (_, T)) => (x, T))
ballarin@17138
  1736
        |> Symtab.make;            
ballarin@17138
  1737
    (* replace parameter names in ids by instantiations *)
ballarin@17138
  1738
    val vinst = Symtab.make (parms ~~ vts);
wenzelm@17221
  1739
    fun vinst_names ps = map (the o Symtab.curried_lookup vinst) ps;
ballarin@17138
  1740
    val inst = Symtab.make (parms ~~ ts);
ballarin@17138
  1741
    val ids' = map (apsnd vinst_names) ids;
ballarin@17138
  1742
    val wits = List.concat (map (snd o valOf o get_global_registration thy) ids');
ballarin@17138
  1743
  in ((inst, tinst), wits) end;
ballarin@17138
  1744
ballarin@17138
  1745
ballarin@15696
  1746
(* store instantiations of args for all registered interpretations
ballarin@15696
  1747
   of the theory *)
ballarin@15696
  1748
ballarin@15696
  1749
fun note_thmss_registrations kind target args thy =
ballarin@15596
  1750
  let
ballarin@17138
  1751
    val parms = the_locale thy target |> #params |> fst |> map fst;
wenzelm@16458
  1752
    val ids = flatten (ProofContext.init thy, intern_expr thy)
ballarin@17033
  1753
      (([], Symtab.empty), Expr (Locale target)) |> fst |> fst
ballarin@17033
  1754
      |> List.mapPartial (fn (id, (_, Assumed _)) => SOME id | _ => NONE)
ballarin@15696
  1755
ballarin@15696
  1756
    val regs = get_global_registrations thy target;
ballarin@15696
  1757
ballarin@15696
  1758
    (* add args to thy for all registrations *)
ballarin@15596
  1759
ballarin@15696
  1760
    fun activate (thy, (vts, ((prfx, atts2), _))) =
ballarin@15696
  1761
      let
ballarin@17138
  1762
        val ((inst, tinst), prems) = collect_global_witnesses thy parms ids vts;
ballarin@15696
  1763
        val args' = map (fn ((n, atts), [(ths, [])]) =>
ballarin@17033
  1764
            ((NameSpace.qualified prfx n,
ballarin@17138
  1765
              map (Attrib.global_attribute_i thy)
ballarin@17138
  1766
                  (inst_tab_atts thy (inst, tinst) atts @ atts2)),
ballarin@17138
  1767
             [(map (Drule.standard o Drule.satisfy_hyps prems o
ballarin@17138
  1768
                inst_tab_thm thy (inst, tinst)) ths, [])]))
ballarin@15696
  1769
          args;
wenzelm@16144
  1770
      in global_note_accesses_i (Drule.kind kind) prfx args' thy |> fst end;
ballarin@15696
  1771
  in Library.foldl activate (thy, regs) end;
ballarin@15596
  1772
ballarin@15596
  1773
skalberg@15531
  1774
fun smart_note_thmss kind NONE = PureThy.note_thmss_i (Drule.kind kind)
wenzelm@15703
  1775
  | smart_note_thmss kind (SOME loc) = note_thmss_qualified kind loc;
wenzelm@12958
  1776
wenzelm@12958
  1777
wenzelm@12958
  1778
local
wenzelm@12958
  1779
ballarin@15696
  1780
(* add facts to locale in theory *)
ballarin@15696
  1781
wenzelm@12958
  1782
fun put_facts loc args thy =
wenzelm@12958
  1783
  let
ballarin@16736
  1784
    val {predicate, import, elems, params, regs} = the_locale thy loc;
wenzelm@15703
  1785
    val note = Notes (map (fn ((a, atts), bs) =>
wenzelm@15703
  1786
      ((a, atts), map (apfst (map (curry Thm.name_thm a))) bs)) args);
wenzelm@15703
  1787
  in
wenzelm@15703
  1788
    thy |> put_locale loc {predicate = predicate, import = import,
ballarin@16736
  1789
      elems = elems @ [(note, stamp ())], params = params, regs = regs}
wenzelm@15703
  1790
  end;
wenzelm@12958
  1791
ballarin@15696
  1792
(* add theorem to locale and theory,
ballarin@15696
  1793
   base for theorems (in loc) and declare (in loc) *)
ballarin@15696
  1794
wenzelm@15703
  1795
fun gen_note_thmss prep_locale prep_facts kind raw_loc args thy =
wenzelm@12706
  1796
  let
wenzelm@12706
  1797
    val thy_ctxt = ProofContext.init thy;
wenzelm@16458
  1798
    val loc = prep_locale thy raw_loc;
wenzelm@15703
  1799
    val (_, (stmt, _), loc_ctxt, _, _) = cert_context_statement (SOME loc) [] [] thy_ctxt;
ballarin@15206
  1800
    val export = ProofContext.export_standard stmt loc_ctxt thy_ctxt;
wenzelm@15703
  1801
wenzelm@17109
  1802
    val (ctxt', (args', facts)) = activate_note prep_facts (loc_ctxt, args);
wenzelm@15703
  1803
    val facts' =
wenzelm@15703
  1804
      map (rpair [] o #1 o #1) args' ~~
wenzelm@15703
  1805
      map (single o Thm.no_attributes o map export o #2) facts;
wenzelm@17109
  1806
wenzelm@17109
  1807
    val (thy', result) =
wenzelm@17109
  1808
      thy
wenzelm@17109
  1809
      |> put_facts loc args'
wenzelm@17109
  1810
      |> note_thmss_registrations kind loc args'
wenzelm@17109
  1811
      |> note_thmss_qualified kind loc facts';
wenzelm@17109
  1812
  in ((thy', ctxt'), result) end;
wenzelm@12706
  1813
wenzelm@12706
  1814
in
wenzelm@12706
  1815
wenzelm@15703
  1816
val note_thmss = gen_note_thmss intern read_facts;
wenzelm@15703
  1817
val note_thmss_i = gen_note_thmss (K I) cert_facts;
wenzelm@12711
  1818
ballarin@15696
  1819
fun add_thmss kind loc args (thy, ctxt) =
wenzelm@12958
  1820
  let
wenzelm@15703
  1821
    val (ctxt', (args', facts)) = activate_note cert_facts
wenzelm@15703
  1822
      (ctxt, args |> map (fn ((a, ths), atts) => ((a, atts), [(ths, [])])));
wenzelm@15703
  1823
    val thy' =
wenzelm@15703
  1824
      thy
wenzelm@15703
  1825
      |> put_facts loc args'
wenzelm@15703
  1826
      |> note_thmss_registrations kind loc args';
wenzelm@15703
  1827
  in ((thy', ctxt'), facts) end;
wenzelm@12702
  1828
wenzelm@12706
  1829
end;
wenzelm@12063
  1830
wenzelm@11896
  1831
wenzelm@13336
  1832
(* predicate text *)
ballarin@15596
  1833
(* CB: generate locale predicates and delta predicates *)
wenzelm@13336
  1834
wenzelm@13375
  1835
local
wenzelm@13375
  1836
ballarin@15206
  1837
(* introN: name of theorems for introduction rules of locale and
ballarin@15206
  1838
     delta predicates;
ballarin@15206
  1839
   axiomsN: name of theorem set with destruct rules for locale predicates,
ballarin@15206
  1840
     also name suffix of delta predicates. *)
ballarin@15206
  1841
wenzelm@13375
  1842
val introN = "intro";
ballarin@15206
  1843
val axiomsN = "axioms";
wenzelm@13375
  1844
wenzelm@16458
  1845
fun atomize_spec thy ts =
wenzelm@13375
  1846
  let
skalberg@15574
  1847
    val t = foldr1 Logic.mk_conjunction ts;
wenzelm@16458
  1848
    val body = ObjectLogic.atomize_term thy t;
wenzelm@13375
  1849
    val bodyT = Term.fastype_of body;
wenzelm@13375
  1850
  in
wenzelm@16458
  1851
    if bodyT = propT then (t, propT, Thm.reflexive (Thm.cterm_of thy t))
wenzelm@16458
  1852
    else (body, bodyT, ObjectLogic.atomize_rule thy (Thm.cterm_of thy t))
wenzelm@13375
  1853
  end;
wenzelm@13375
  1854
wenzelm@13394
  1855
fun aprop_tr' n c = (c, fn args =>
wenzelm@13394
  1856
  if length args = n then Syntax.const "_aprop" $ Term.list_comb (Syntax.free c, args)
wenzelm@13394
  1857
  else raise Match);
wenzelm@13336
  1858
ballarin@15104
  1859
(* CB: define one predicate including its intro rule and axioms
ballarin@15104
  1860
   - bname: predicate name
ballarin@15104
  1861
   - parms: locale parameters
ballarin@15104
  1862
   - defs: thms representing substitutions from defines elements
ballarin@15104
  1863
   - ts: terms representing locale assumptions (not normalised wrt. defs)
ballarin@15104
  1864
   - norm_ts: terms representing locale assumptions (normalised wrt. defs)
ballarin@15104
  1865
   - thy: the theory
ballarin@15104
  1866
*)
ballarin@15104
  1867
wenzelm@13420
  1868
fun def_pred bname parms defs ts norm_ts thy =
wenzelm@13375
  1869
  let
wenzelm@16458
  1870
    val name = Sign.full_name thy bname;
wenzelm@13375
  1871
wenzelm@16458
  1872
    val (body, bodyT, body_eq) = atomize_spec thy norm_ts;
wenzelm@13394
  1873
    val env = Term.add_term_free_names (body, []);
skalberg@15570
  1874
    val xs = List.filter (fn (x, _) => x mem_string env) parms;
wenzelm@13394
  1875
    val Ts = map #2 xs;
skalberg@15574
  1876
    val extraTs = (Term.term_tfrees body \\ foldr Term.add_typ_tfrees [] Ts)
wenzelm@13394
  1877
      |> Library.sort_wrt #1 |> map TFree;
wenzelm@13399
  1878
    val predT = map Term.itselfT extraTs ---> Ts ---> bodyT;
wenzelm@13336
  1879
wenzelm@13394
  1880
    val args = map Logic.mk_type extraTs @ map Free xs;
wenzelm@13394
  1881
    val head = Term.list_comb (Const (name, predT), args);
wenzelm@16458
  1882
    val statement = ObjectLogic.assert_propT thy head;
wenzelm@13375
  1883
wenzelm@13375
  1884
    val (defs_thy, [pred_def]) =
wenzelm@13375
  1885
      thy
wenzelm@13394
  1886
      |> (if bodyT <> propT then I else
wenzelm@13394
  1887
        Theory.add_trfuns ([], [], map (aprop_tr' (length args)) (NameSpace.accesses' name), []))
wenzelm@13375
  1888
      |> Theory.add_consts_i [(bname, predT, Syntax.NoSyn)]
wenzelm@13375
  1889
      |> PureThy.add_defs_i false [((Thm.def_name bname, Logic.mk_equals (head, body)), [])];
wenzelm@13394
  1890
wenzelm@16458
  1891
    val cert = Thm.cterm_of defs_thy;
wenzelm@13375
  1892
wenzelm@16947
  1893
    val intro = Drule.standard (Tactic.prove defs_thy [] norm_ts statement (fn _ =>
wenzelm@13375
  1894
      Tactic.rewrite_goals_tac [pred_def] THEN
wenzelm@13375
  1895
      Tactic.compose_tac (false, body_eq RS Drule.equal_elim_rule1, 1) 1 THEN
wenzelm@16947
  1896
      Tactic.compose_tac (false, Drule.conj_intr_list (map (Thm.assume o cert) norm_ts), 0) 1));
wenzelm@13375
  1897
wenzelm@13375
  1898
    val conjuncts =
wenzelm@13399
  1899
      Drule.equal_elim_rule1 OF [Thm.symmetric body_eq,
wenzelm@13399
  1900
        Tactic.rewrite_rule [pred_def] (Thm.assume (cert statement))]
wenzelm@13375
  1901
      |> Drule.conj_elim_precise (length ts);
wenzelm@13394
  1902
    val axioms = (ts ~~ conjuncts) |> map (fn (t, ax) =>
wenzelm@16458
  1903
      Tactic.prove_plain defs_thy [] [] t (fn _ =>
wenzelm@13375
  1904
        Tactic.rewrite_goals_tac defs THEN
wenzelm@13375
  1905
        Tactic.compose_tac (false, ax, 0) 1));
wenzelm@13394
  1906
  in (defs_thy, (statement, intro, axioms)) end;
wenzelm@13375
  1907
ballarin@15104
  1908
(* CB: modify the locale elements:
ballarin@15623
  1909
   - assumes elements become notes elements,
ballarin@15104
  1910
   - notes elements are lifted
ballarin@15104
  1911
*)
ballarin@15104
  1912
wenzelm@15703
  1913
fun change_elem (axms, Assumes asms) =
wenzelm@13394
  1914
      apsnd Notes ((axms, asms) |> foldl_map (fn (axs, (a, spec)) =>
nipkow@13629
  1915
        let val (ps,qs) = splitAt(length spec, axs)
nipkow@13629
  1916
        in (qs, (a, [(ps, [])])) end))
wenzelm@15703
  1917
  | change_elem e = e;
wenzelm@13394
  1918
ballarin@15206
  1919
(* CB: changes only "new" elems, these have identifier ("", _). *)
ballarin@15206
  1920
wenzelm@13394
  1921
fun change_elemss axioms elemss = (axioms, elemss) |> foldl_map
wenzelm@13394
  1922
  (fn (axms, (id as ("", _), es)) =>
wenzelm@15703
  1923
    foldl_map change_elem (axms, map (map_values I I (Drule.satisfy_hyps axioms)) es)
wenzelm@15703
  1924
    |> apsnd (pair id)
wenzelm@13394
  1925
  | x => x) |> #2;
wenzelm@13394
  1926
wenzelm@13394
  1927
in
wenzelm@13375
  1928
ballarin@15104
  1929
(* CB: main predicate definition function *)
ballarin@15104
  1930
wenzelm@13394
  1931
fun define_preds bname (parms, ((exts, exts'), (ints, ints')), defs) elemss thy =
wenzelm@13394
  1932
  let
wenzelm@13394
  1933
    val (thy', (elemss', more_ts)) =
skalberg@15531
  1934
      if null exts then (thy, (elemss, []))
wenzelm@13394
  1935
      else
wenzelm@13394
  1936
        let
skalberg@15531
  1937
          val aname = if null ints then bname else bname ^ "_" ^ axiomsN;
wenzelm@13394
  1938
          val (def_thy, (statement, intro, axioms)) =
wenzelm@13394
  1939
            thy |> def_pred aname parms defs exts exts';
ballarin@16325
  1940
          val elemss' = change_elemss (map (Drule.zero_var_indexes o Drule.gen_all) axioms) elemss @
wenzelm@13420
  1941
            [(("", []), [Assumes [((bname ^ "_" ^ axiomsN, []), [(statement, ([], []))])]])];
wenzelm@13394
  1942
        in
wenzelm@14564
  1943
          def_thy |> note_thmss_qualified "" aname
ballarin@14254
  1944
            [((introN, []), [([intro], [])])]
wenzelm@13394
  1945
          |> #1 |> rpair (elemss', [statement])
wenzelm@13394
  1946
        end;
ballarin@15206
  1947
    val (thy'', predicate) =
skalberg@15531
  1948
      if null ints then (thy', ([], []))
wenzelm@13394
  1949
      else
wenzelm@13394
  1950
        let
wenzelm@13394
  1951
          val (def_thy, (statement, intro, axioms)) =
wenzelm@13394
  1952
            thy' |> def_pred bname parms defs (ints @ more_ts) (ints' @ more_ts);
wenzelm@16458
  1953
          val cstatement = Thm.cterm_of def_thy statement;
wenzelm@13394
  1954
        in
wenzelm@14564
  1955
          def_thy |> note_thmss_qualified "" bname
ballarin@14254
  1956
            [((introN, []), [([intro], [])]),
ballarin@14254
  1957
             ((axiomsN, []), [(map Drule.standard axioms, [])])]
wenzelm@13415
  1958
          |> #1 |> rpair ([cstatement], axioms)
wenzelm@13394
  1959
        end;
ballarin@15206
  1960
  in (thy'', (elemss', predicate)) end;
wenzelm@13375
  1961
wenzelm@13375
  1962
end;
wenzelm@13336
  1963
wenzelm@13336
  1964
wenzelm@13297
  1965
(* add_locale(_i) *)
wenzelm@13297
  1966
wenzelm@13297
  1967
local
wenzelm@13297
  1968
wenzelm@13394
  1969
fun gen_add_locale prep_ctxt prep_expr do_pred bname raw_import raw_body thy =
ballarin@15596
  1970
  (* CB: do_pred controls generation of predicates.
ballarin@15596
  1971
         true -> with, false -> without predicates. *)
wenzelm@13297
  1972
  let
wenzelm@16458
  1973
    val name = Sign.full_name thy bname;
skalberg@15570
  1974
    val _ = conditional (isSome (get_locale thy name)) (fn () =>
wenzelm@13297
  1975
      error ("Duplicate definition of locale " ^ quote name));
wenzelm@13297
  1976
wenzelm@13297
  1977
    val thy_ctxt = ProofContext.init thy;
ballarin@17228
  1978
    val (((import_ctxt, import_elemss), (body_ctxt, body_elemss, syn)),
ballarin@17228
  1979
      text as (parms, ((_, exts'), _), _)) =
wenzelm@13375
  1980
      prep_ctxt raw_import raw_body thy_ctxt;
wenzelm@13375
  1981
    val elemss = import_elemss @ body_elemss;
wenzelm@16458
  1982
    val import = prep_expr thy raw_import;
wenzelm@13297
  1983
ballarin@17228
  1984
    val extraTs = foldr Term.add_term_tfrees [] exts' \\
ballarin@17228
  1985
      foldr Term.add_typ_tfrees [] (map #2 parms);
ballarin@17228
  1986
    val _ = if null extraTs then ()
ballarin@17228
  1987
      else warning ("Encountered type variables in specification text that don't occur in the\n" ^ "locale parameters.");        
ballarin@17228
  1988
ballarin@15206
  1989
    val (pred_thy, (elemss', predicate as (predicate_statement, predicate_axioms))) =
wenzelm@13394
  1990
      if do_pred then thy |> define_preds bname text elemss
wenzelm@13415
  1991
      else (thy, (elemss, ([], [])));
wenzelm@13375
  1992
    val pred_ctxt = ProofContext.init pred_thy;
wenzelm@13420
  1993
ballarin@15206
  1994
    fun axiomify axioms elemss = 
ballarin@15206
  1995
      (axioms, elemss) |> foldl_map (fn (axs, (id, elems)) => let
skalberg@15570
  1996
                   val ts = List.concat (List.mapPartial (fn (Assumes asms) =>
skalberg@15570
  1997
                     SOME (List.concat (map (map #1 o #2) asms)) | _ => NONE) elems);
ballarin@15206
  1998
                   val (axs1, axs2) = splitAt (length ts, axs);
ballarin@17000
  1999
                 in (axs2, ((id, Assumed axs1), elems)) end)
ballarin@15206
  2000
        |> snd;
ballarin@15206
  2001
    val (ctxt, (_, facts)) = activate_facts (K I)
ballarin@15206
  2002
      (pred_ctxt, axiomify predicate_axioms elemss');
ballarin@15206
  2003
    val export = ProofContext.export_standard predicate_statement ctxt pred_ctxt;
wenzelm@13420
  2004
    val facts' = facts |> map (fn (a, ths) => ((a, []), [(map export ths, [])]));
haftmann@17142
  2005
    val elems' = List.concat (map #2 (List.filter (equal "" o #1 o #1) elemss'))
wenzelm@13297
  2006
  in
wenzelm@13375
  2007
    pred_thy
wenzelm@14564
  2008
    |> note_thmss_qualified "" name facts' |> #1
wenzelm@13297
  2009
    |> declare_locale name
wenzelm@16458
  2010
    |> put_locale name {predicate = predicate, import = import,
haftmann@17142
  2011
        elems = map (fn e => (e, stamp ())) elems',
ballarin@16102
  2012
        params = (params_of elemss' |>
wenzelm@17221
  2013
          map (fn (x, SOME T) => ((x, T), the (Symtab.curried_lookup syn x))), map #1 (params_of body_elemss)),
ballarin@16736
  2014
        regs = []}
haftmann@17142
  2015
    |> pair (elems', body_ctxt)
wenzelm@13297
  2016
  end;
wenzelm@13297
  2017
wenzelm@13297
  2018
in
wenzelm@13297
  2019
wenzelm@17109
  2020
val add_locale_context = gen_add_locale read_context intern_expr;
wenzelm@17109
  2021
val add_locale_context_i = gen_add_locale cert_context (K I);
haftmann@17142
  2022
fun add_locale b = #2 oooo add_locale_context b;
haftmann@17142
  2023
fun add_locale_i b = #2 oooo add_locale_context_i b;
wenzelm@13297
  2024
wenzelm@13297
  2025
end;
wenzelm@13297
  2026
wenzelm@15801
  2027
val _ = Context.add_setup
wenzelm@15801
  2028
 [add_locale_i true "var" empty [Fixes [(Syntax.internal "x", NONE, SOME Syntax.NoSyn)]],
wenzelm@15801
  2029
  add_locale_i true "struct" empty [Fixes [(Syntax.internal "S", NONE, NONE)]]];
wenzelm@15801
  2030
wenzelm@13297
  2031
wenzelm@12730
  2032
ballarin@15598
  2033
(** Interpretation commands **)
ballarin@15596
  2034
ballarin@15596
  2035
local
ballarin@15596
  2036
ballarin@15596
  2037
(* extract proof obligations (assms and defs) from elements *)
ballarin@15596
  2038
ballarin@17138
  2039
fun extract_asms_elem (Fixes _) ts = ts
ballarin@17138
  2040
  | extract_asms_elem (Constrains _) ts = ts
ballarin@17138
  2041
  | extract_asms_elem (Assumes asms) ts =
ballarin@15596
  2042
      ts @ List.concat (map (fn (_, ams) => map (fn (t, _) => t) ams) asms)
ballarin@17138
  2043
  | extract_asms_elem (Defines defs) ts =
ballarin@16169
  2044
      ts @ map (fn (_, (def, _)) => def) defs
ballarin@17138
  2045
  | extract_asms_elem (Notes _) ts = ts;
ballarin@15596
  2046
ballarin@17000
  2047
fun extract_asms_elems ((id, Assumed _), elems) =
ballarin@17138
  2048
      (id, fold extract_asms_elem elems [])
ballarin@17138
  2049
  | extract_asms_elems ((id, Derived _), _) = (id, []);
ballarin@15596
  2050
ballarin@17138
  2051
fun extract_asms_elemss elemss = map extract_asms_elems elemss;
ballarin@15596
  2052
ballarin@15624
  2053
(* activate instantiated facts in theory or context *)
ballarin@15596
  2054
ballarin@17138
  2055
fun gen_activate_facts_elemss get_reg note attrib std put_reg add_wit
ballarin@17033
  2056
        attn all_elemss new_elemss propss result thy_ctxt =
ballarin@17033
  2057
    let
ballarin@17033
  2058
      fun activate_elem disch (prfx, atts) (Notes facts) thy_ctxt =
ballarin@17033
  2059
          let
ballarin@17033
  2060
            val facts' = facts
ballarin@17033
  2061
              (* discharge hyps in attributes *)
wenzelm@17109
  2062
              |> Attrib.map_facts (attrib thy_ctxt o Args.map_values I I I disch)
ballarin@17033
  2063
              (* insert interpretation attributes *)
ballarin@17033
  2064
              |> map (apfst (apsnd (fn a => a @ map (attrib thy_ctxt) atts)))
ballarin@17033
  2065
              (* discharge hyps *)
ballarin@17033
  2066
              |> map (apsnd (map (apfst (map disch))))
ballarin@17033
  2067
              (* prefix names *)
ballarin@17033
  2068
              |> map (apfst (apfst (NameSpace.qualified prfx)))
ballarin@17138
  2069
          in fst (note prfx facts' thy_ctxt) end
ballarin@17033
  2070
        | activate_elem _ _ _ thy_ctxt = thy_ctxt;
ballarin@17033
  2071
ballarin@17138
  2072
      fun activate_elems disch ((id, _), elems) thy_ctxt =
ballarin@17033
  2073
          let
ballarin@17033
  2074
            val ((prfx, atts2), _) = valOf (get_reg thy_ctxt id)
ballarin@17033
  2075
                handle Option => sys_error ("Unknown registration of " ^
ballarin@17033
  2076
                  quote (fst id) ^ " while activating facts.");
ballarin@17033
  2077
          in
ballarin@17033
  2078
            fold (activate_elem disch (prfx, atts2)) elems thy_ctxt
ballarin@17033
  2079
          end;
ballarin@15596
  2080
ballarin@17033
  2081
      val thmss = unflat (map snd propss) result;
ballarin@17033
  2082
ballarin@17033
  2083
      val thy_ctxt' = thy_ctxt
ballarin@17033
  2084
        (* add registrations *)
ballarin@17033
  2085
        |> fold (fn ((id, _), _) => put_reg id attn) new_elemss
ballarin@17033
  2086
        (* add witnesses of Assumed elements *)
ballarin@17033
  2087
        |> fold (fn (id, thms) => fold (add_wit id) thms)
ballarin@17033
  2088
           (map fst propss ~~ thmss);
ballarin@15596
  2089
ballarin@17033
  2090
      val prems = List.concat (List.mapPartial
ballarin@17033
  2091
            (fn ((id, Assumed _), _) => Option.map snd (get_reg thy_ctxt' id)
ballarin@17033
  2092
              | ((_, Derived _), _) => NONE) all_elemss);
ballarin@17033
  2093
      val disch = Drule.satisfy_hyps prems;
ballarin@17033
  2094
      val disch' = std o Drule.fconv_rule (Thm.beta_conversion true) o disch;
ballarin@17033
  2095
ballarin@17033
  2096
      val thy_ctxt'' = thy_ctxt'
ballarin@17033
  2097
        (* add witnesses of Derived elements *)
ballarin@17033
  2098
        |> fold (fn (id, thms) => fold (add_wit id o disch) thms)
ballarin@17033
  2099
           (List.mapPartial (fn ((_, Assumed _), _) => NONE
ballarin@17033
  2100
              | ((id, Derived thms), _) => SOME (id, thms)) all_elemss)
ballarin@17033
  2101
    in
ballarin@17033
  2102
      thy_ctxt''
ballarin@17033
  2103
        (* add facts to theory or context *)
ballarin@17033
  2104
        |> fold (activate_elems disch') new_elemss
ballarin@17033
  2105
    end;
ballarin@15596
  2106
ballarin@15624
  2107
val global_activate_facts_elemss = gen_activate_facts_elemss
ballarin@15696
  2108
      (fn thy => fn (name, ps) =>
ballarin@15696
  2109
        get_global_registration thy (name, map Logic.varify ps))
wenzelm@16144
  2110
      (global_note_accesses_i (Drule.kind ""))
ballarin@17033
  2111
      Attrib.global_attribute_i Drule.standard
ballarin@17033
  2112
      (fn (name, ps) => put_global_registration (name, map Logic.varify ps))
ballarin@17033
  2113
      (fn (n, ps) => fn thm =>
ballarin@17033
  2114
         add_global_witness (n, map Logic.varify ps) (Drule.freeze_all thm));
wenzelm@16144
  2115
ballarin@15624
  2116
val local_activate_facts_elemss = gen_activate_facts_elemss
ballarin@15696
  2117
      get_local_registration
wenzelm@16144
  2118
      local_note_accesses_i
ballarin@17033
  2119
      Attrib.context_attribute_i I
ballarin@17033
  2120
      put_local_registration
ballarin@17033
  2121
      add_local_witness;
ballarin@15624
  2122
ballarin@17033
  2123
fun gen_prep_registration mk_ctxt is_local read_terms test_reg activate
ballarin@15624
  2124
    attn expr insts thy_ctxt =
ballarin@15596
  2125
  let
ballarin@15624
  2126
    val ctxt = mk_ctxt thy_ctxt;
wenzelm@16458
  2127
    val thy = ProofContext.theory_of ctxt;
ballarin@15596
  2128
ballarin@15696
  2129
    val ctxt' = ctxt |> ProofContext.theory_of |> ProofContext.init;
wenzelm@16458
  2130
    val ((ids, _), raw_elemss) = flatten (ctxt', intern_expr thy)
ballarin@16102
  2131
          (([], Symtab.empty), Expr expr);
ballarin@17000
  2132
    val ((parms, all_elemss, _), (_, (_, defs, _))) =
ballarin@16736
  2133
          read_elemss false ctxt' [] raw_elemss [];
ballarin@15596
  2134
ballarin@15596
  2135
    (** compute instantiation **)
ballarin@15596
  2136
ballarin@15696
  2137
    (* user input *)
ballarin@15596
  2138
    val insts = if length parms < length insts
ballarin@15596
  2139
         then error "More arguments than parameters in instantiation."
ballarin@15596
  2140
         else insts @ replicate (length parms - length insts) NONE;
ballarin@15596
  2141
    val (ps, pTs) = split_list parms;
ballarin@15596
  2142
    val pvTs = map Type.varifyT pTs;
ballarin@15598
  2143
ballarin@15598
  2144
    (* instantiations given by user *)
ballarin@15596
  2145
    val given = List.mapPartial (fn (_, (NONE, _)) => NONE
ballarin@15596
  2146
         | (x, (SOME inst, T)) => SOME (x, (inst, T))) (ps ~~ (insts ~~ pvTs));
ballarin@15596
  2147
    val (given_ps, given_insts) = split_list given;
ballarin@15596
  2148
    val tvars = foldr Term.add_typ_tvars [] pvTs;
ballarin@15596
  2149
    val used = foldr Term.add_typ_varnames [] pvTs;
ballarin@15596
  2150
    fun sorts (a, i) = assoc (tvars, (a, i));
ballarin@15624
  2151
    val (vs, vinst) = read_terms thy_ctxt sorts used given_insts;
ballarin@15696
  2152
    val vars = foldl Term.add_term_tvar_ixns [] vs \\ map fst tvars;
wenzelm@16861
  2153
    val vars' = fold Term.add_term_varnames vs vars;
ballarin@15696
  2154
    val _ = if null vars' then ()
ballarin@15696
  2155
         else error ("Illegal schematic variable(s) in instantiation: " ^
ballarin@15696
  2156
           commas_quote (map Syntax.string_of_vname vars'));
ballarin@15598
  2157
    (* replace new types (which are TFrees) by ones with new names *)
ballarin@15598
  2158
    val new_Tnames = foldr Term.add_term_tfree_names [] vs;
ballarin@15598
  2159
    val new_Tnames' = Term.invent_names used "'a" (length new_Tnames);
ballarin@15696
  2160
    val renameT =
ballarin@15696
  2161
          if is_local then I
ballarin@15696
  2162
          else Type.unvarifyT o Term.map_type_tfree (fn (a, s) =>
ballarin@15696
  2163
            TFree (valOf (assoc (new_Tnames ~~ new_Tnames', a)), s));
ballarin@15696
  2164
    val rename =
ballarin@15696
  2165
          if is_local then I
ballarin@15696
  2166
          else Term.map_term_types renameT;
ballarin@15598
  2167
ballarin@15598
  2168
    val tinst = Symtab.make (map
ballarin@15696
  2169
                (fn ((x, 0), T) => (x, T |> renameT)
wenzelm@16850
  2170
                  | ((_, n), _) => sys_error "Internal error var in prep_registration") vinst);
ballarin@15696
  2171
    val inst = Symtab.make (given_ps ~~ map rename vs);
ballarin@15596
  2172
ballarin@15596
  2173
    (* defined params without user input *)
ballarin@15596
  2174
    val not_given = List.mapPartial (fn (x, (NONE, T)) => SOME (x, T)
ballarin@15596
  2175
         | (_, (SOME _, _)) => NONE) (ps ~~ (insts ~~ pTs));
ballarin@15596
  2176
    fun add_def ((inst, tinst), (p, pT)) =
ballarin@15596
  2177
      let
ballarin@15596
  2178
        val (t, T) = case find_first (fn (Free (a, _), _) => a = p) defs of
ballarin@15596
  2179
               NONE => error ("Instance missing for parameter " ^ quote p)
ballarin@15596
  2180
             | SOME (Free (_, T), t) => (t, T);
ballarin@15696
  2181
        val d = t |> inst_tab_term (inst, tinst) |> Envir.beta_norm;
wenzelm@17221
  2182
      in (Symtab.curried_update_new (p, d) inst, tinst) end;
ballarin@15596
  2183
    val (inst, tinst) = Library.foldl add_def ((inst, tinst), not_given);
ballarin@15696
  2184
    (* Note: inst and tinst contain no vars. *)
ballarin@15596
  2185
ballarin@15596
  2186
    (** compute proof obligations **)
ballarin@15596
  2187
ballarin@15598
  2188
    (* restore "small" ids *)
ballarin@17000
  2189
    val ids' = map (fn ((n, ps), (_, mode)) =>
ballarin@17000
  2190
          ((n, map (fn p => Free (p, valOf (assoc (parms, p)))) ps), mode)) ids;
ballarin@15596
  2191
    (* instantiate ids and elements *)
ballarin@15596
  2192
    val inst_elemss = map
ballarin@17033
  2193
          (fn ((id, _), ((_, mode), elems)) =>
ballarin@17000
  2194
             inst_tab_elems thy (inst, tinst) (id, map (fn Int e => e) elems)
ballarin@17033
  2195
               |> apfst (fn id => (id, map_mode (map (inst_tab_thm thy (inst, tinst))) mode))) 
ballarin@15596
  2196
          (ids' ~~ all_elemss);
ballarin@15596
  2197
ballarin@15624
  2198
    (* remove fragments already registered with theory or context *)
ballarin@17000
  2199
    val new_inst_elemss = List.filter (fn ((id, _), _) =>
ballarin@15624
  2200
          not (test_reg thy_ctxt id)) inst_elemss;
wenzelm@15703
  2201
    val new_ids = map #1 new_inst_elemss;
ballarin@15596
  2202
ballarin@15596
  2203
    val propss = extract_asms_elemss new_inst_elemss;
ballarin@15596
  2204
wenzelm@15703
  2205
    val bind_attrib = Attrib.crude_closure ctxt o Args.assignable;
wenzelm@16458
  2206
    val attn' = apsnd (map (bind_attrib o Attrib.intern_src thy)) attn;
ballarin@15596
  2207
ballarin@17033
  2208
  in (propss, activate attn' inst_elemss new_inst_elemss propss) end;
ballarin@15624
  2209
ballarin@15624
  2210
in
ballarin@15596
  2211
ballarin@15624
  2212
val prep_global_registration = gen_prep_registration
ballarin@15696
  2213
     ProofContext.init false
ballarin@15624
  2214
     (fn thy => fn sorts => fn used =>
wenzelm@16458
  2215
       Sign.read_def_terms (thy, K NONE, sorts) used true)
ballarin@15696
  2216
     (fn thy => fn (name, ps) =>
ballarin@15696
  2217
       test_global_registration thy (name, map Logic.varify ps))
ballarin@15624
  2218
     global_activate_facts_elemss;
ballarin@15624
  2219
ballarin@15624
  2220
val prep_local_registration = gen_prep_registration
ballarin@15696
  2221
     I true
ballarin@15624
  2222
     (fn ctxt => ProofContext.read_termTs ctxt (K false) (K NONE))
ballarin@15624
  2223
     smart_test_registration
ballarin@15624
  2224
     local_activate_facts_elemss;
ballarin@15596
  2225
ballarin@17000
  2226
fun prep_registration_in_locale target expr thy =
ballarin@17000
  2227
  (* target already in internal form *)
ballarin@17000
  2228
  let
ballarin@17000
  2229
    val ctxt = ProofContext.init thy;
ballarin@17138
  2230
    val ((raw_target_ids, target_syn), _) = flatten (ctxt, I)
ballarin@17000
  2231
        (([], Symtab.empty), Expr (Locale target));
ballarin@17000
  2232
    val fixed = the_locale thy target |> #params |> #1 |> map #1;
ballarin@17000
  2233
    val ((all_ids, syn), raw_elemss) = flatten (ctxt, intern_expr thy)
ballarin@17138
  2234
        ((raw_target_ids, target_syn), Expr expr);
ballarin@17138
  2235
    val (target_ids, ids) = splitAt (length raw_target_ids, all_ids);
ballarin@17138
  2236
    val ((parms, elemss, _), _) = read_elemss false ctxt fixed raw_elemss [];
ballarin@17000
  2237
ballarin@17000
  2238
    (** compute proof obligations **)
ballarin@17000
  2239
ballarin@17000
  2240
    (* restore "small" ids, with mode *)
ballarin@17000
  2241
    val ids' = map (apsnd snd) ids;
ballarin@17000
  2242
    (* remove Int markers *)
ballarin@17000
  2243
    val elemss' = map (fn (_, es) =>
ballarin@17000
  2244
        map (fn Int e => e) es) elemss
ballarin@17000
  2245
    (* extract assumptions and defs *)
ballarin@17138
  2246
    val ids_elemss = ids' ~~ elemss';
ballarin@17138
  2247
    val propss = extract_asms_elemss ids_elemss;
ballarin@17000
  2248
ballarin@17138
  2249
    (** activation function:
ballarin@17138
  2250
        - add registrations to the target locale
ballarin@17138
  2251
        - add induced registrations for all global registrations of
ballarin@17138
  2252
          the target, unless already present
ballarin@17138
  2253
        - add facts of induced registrations to theory **)
ballarin@17138
  2254
ballarin@17138
  2255
    val t_ids = List.mapPartial
ballarin@17138
  2256
        (fn (id, (_, Assumed _)) => SOME id | _ => NONE) target_ids;
ballarin@17138
  2257
ballarin@17033
  2258
    fun activate locale_results thy = let
ballarin@17138
  2259
        val ids_elemss_thmss = ids_elemss ~~
ballarin@17138
  2260
	    unflat (map snd propss) locale_results;
ballarin@17138
  2261
        val regs = get_global_registrations thy target;
ballarin@17138
  2262
ballarin@17138
  2263
        fun activate_id (((id, Assumed _), _), thms) thy =
ballarin@17033
  2264
            thy |> put_registration_in_locale target id
ballarin@17138
  2265
                |> fold (add_witness_in_locale target id) thms
ballarin@17138
  2266
          | activate_id _ thy = thy;
ballarin@17138
  2267
ballarin@17138
  2268
        fun activate_reg (vts, ((prfx, atts2), _)) thy = let
ballarin@17138
  2269
            val ((inst, tinst), wits) =
ballarin@17138
  2270
                collect_global_witnesses thy fixed t_ids vts;
ballarin@17138
  2271
            fun inst_parms ps = map (fn p =>
ballarin@17138
  2272
                  valOf (assoc (map #1 fixed ~~ vts, p))) ps;
ballarin@17138
  2273
            val disch = Drule.fconv_rule (Thm.beta_conversion true) o
ballarin@17138
  2274
                Drule.satisfy_hyps wits;
ballarin@17138
  2275
            val new_elemss = List.filter (fn (((name, ps), _), _) =>
ballarin@17138
  2276
                not (test_global_registration thy (name, inst_parms ps))) (ids_elemss);
ballarin@17138
  2277
            fun activate_assumed_id (((_, Derived _), _), _) thy = thy
ballarin@17138
  2278
              | activate_assumed_id ((((name, ps), Assumed _), _), thms) thy = let
ballarin@17138
  2279
                val ps' = inst_parms ps;
ballarin@17138
  2280
              in
ballarin@17138
  2281
                if test_global_registration thy (name, ps')
ballarin@17138
  2282
                then thy
ballarin@17138
  2283
                else thy
ballarin@17138
  2284
                  |> put_global_registration (name, ps') (prfx, atts2)
ballarin@17138
  2285
                  |> fold (fn thm => fn thy => add_global_witness (name, ps')
ballarin@17138
  2286
                       ((disch o inst_tab_thm thy (inst, tinst)) thm) thy) thms
ballarin@17138
  2287
              end;
ballarin@17138
  2288
ballarin@17138
  2289
            fun activate_derived_id ((_, Assumed _), _) thy = thy
ballarin@17138
  2290
              | activate_derived_id (((name, ps), Derived ths), _) thy = let
ballarin@17138
  2291
                val ps' = inst_parms ps;
ballarin@17138
  2292
              in
ballarin@17138
  2293
                if test_global_registration thy (name, ps')
ballarin@17138
  2294
                then thy
ballarin@17138
  2295
                else thy
ballarin@17138
  2296
                  |> put_global_registration (name, ps') (prfx, atts2)
ballarin@17138
  2297
                  |> fold (fn thm => fn thy => add_global_witness (name, ps')
ballarin@17138
  2298
                       ((disch o inst_tab_thm thy (inst, tinst) o Drule.satisfy_hyps locale_results) thm) thy) ths
ballarin@17138
  2299
              end;
ballarin@17138
  2300
ballarin@17138
  2301
            fun activate_elem (Notes facts) thy =
ballarin@17138
  2302
                let
ballarin@17138
  2303
                  val facts' = facts
ballarin@17138
  2304
                      |> Attrib.map_facts (Attrib.global_attribute_i thy o
ballarin@17138
  2305
                         Args.map_values I (tinst_tab_type tinst)
ballarin@17138
  2306
                           (inst_tab_term (inst, tinst))
ballarin@17138
  2307
                           (disch o inst_tab_thm thy (inst, tinst) o
ballarin@17138
  2308
                            Drule.satisfy_hyps locale_results))
ballarin@17138
  2309
                      |> map (apfst (apsnd (fn a => a @ map (Attrib.global_attribute thy) atts2)))
ballarin@17138
  2310
                      |> map (apsnd (map (apfst (map (disch o inst_tab_thm thy (inst, tinst) o Drule.satisfy_hyps locale_results)))))
ballarin@17138
  2311
                      |> map (apfst (apfst (NameSpace.qualified prfx)))
ballarin@17138
  2312
                in
ballarin@17138
  2313
                  fst (global_note_accesses_i (Drule.kind "") prfx facts' thy)
ballarin@17138
  2314
                end
ballarin@17138
  2315
              | activate_elem _ thy = thy;
ballarin@17138
  2316
ballarin@17138
  2317
            fun activate_elems (_, elems) thy = fold activate_elem elems thy;
ballarin@17138
  2318
ballarin@17138
  2319
          in thy |> fold activate_assumed_id ids_elemss_thmss
ballarin@17138
  2320
                 |> fold activate_derived_id ids_elemss
ballarin@17138
  2321
                 |> fold activate_elems new_elemss end;
ballarin@17033
  2322
      in
ballarin@17138
  2323
        thy |> fold activate_id ids_elemss_thmss
ballarin@17138
  2324
            |> fold activate_reg regs
ballarin@17033
  2325
      end;
ballarin@17000
  2326
ballarin@17033
  2327
  in (propss, activate) end;
ballarin@17000
  2328
ballarin@15596
  2329
end;  (* local *)
ballarin@15596
  2330
ballarin@15596
  2331
wenzelm@11896
  2332
end;