src/Pure/Isar/locale.ML
author ballarin
Mon May 24 10:48:32 2010 +0200 (2010-05-24)
changeset 37105 e464f84f3680
parent 37104 3877a6c45d57
child 37133 1d048c6940c8
permissions -rw-r--r--
Store registrations in efficient data structure.
haftmann@29361
     1
(*  Title:      Pure/Isar/locale.ML
haftmann@29361
     2
    Author:     Clemens Ballarin, TU Muenchen
haftmann@29361
     3
wenzelm@30725
     4
Locales -- managed Isar proof contexts, based on Pure predicates.
haftmann@29361
     5
haftmann@29361
     6
Draws basic ideas from Florian Kammueller's original version of
haftmann@29361
     7
locales, but uses the richer infrastructure of Isar instead of the raw
haftmann@29361
     8
meta-logic.  Furthermore, structured import of contexts (with merge
haftmann@29361
     9
and rename operations) are provided, as well as type-inference of the
haftmann@29361
    10
signature parts, and predicate definitions of the specification text.
haftmann@29361
    11
haftmann@29361
    12
Interpretation enables the reuse of theorems of locales in other
haftmann@29361
    13
contexts, namely those defined by theories, structured proofs and
haftmann@29361
    14
locales themselves.
haftmann@29361
    15
haftmann@29361
    16
See also:
haftmann@29361
    17
haftmann@29361
    18
[1] Clemens Ballarin. Locales and Locale Expressions in Isabelle/Isar.
haftmann@29361
    19
    In Stefano Berardi et al., Types for Proofs and Programs: International
haftmann@29361
    20
    Workshop, TYPES 2003, Torino, Italy, LNCS 3085, pages 34-50, 2004.
haftmann@29361
    21
[2] Clemens Ballarin. Interpretation of Locales in Isabelle: Managing
haftmann@29361
    22
    Dependencies between Locales. Technical Report TUM-I0607, Technische
haftmann@29361
    23
    Universitaet Muenchen, 2006.
haftmann@29361
    24
[3] Clemens Ballarin. Interpretation of Locales in Isabelle: Theories and
haftmann@29361
    25
    Proof Contexts. In J.M. Borwein and W.M. Farmer, MKM 2006, LNAI 4108,
haftmann@29361
    26
    pages 31-43, 2006.
haftmann@29361
    27
*)
haftmann@29361
    28
haftmann@29361
    29
signature LOCALE =
haftmann@29361
    30
sig
haftmann@29576
    31
  (* Locale specification *)
wenzelm@30344
    32
  val register_locale: binding ->
wenzelm@30755
    33
    (string * sort) list * ((string * typ) * mixfix) list ->
haftmann@29361
    34
    term option * term list ->
haftmann@29441
    35
    thm option * thm option -> thm list ->
wenzelm@35798
    36
    declaration list ->
wenzelm@30725
    37
    (string * (Attrib.binding * (thm list * Attrib.src list) list) list) list ->
wenzelm@30725
    38
    (string * morphism) list -> theory -> theory
haftmann@29361
    39
  val intern: theory -> xstring -> string
haftmann@29361
    40
  val extern: theory -> string -> xstring
haftmann@29392
    41
  val defined: theory -> string -> bool
wenzelm@30755
    42
  val params_of: theory -> string -> ((string * typ) * mixfix) list
haftmann@29441
    43
  val intros_of: theory -> string -> thm option * thm option
haftmann@29441
    44
  val axioms_of: theory -> string -> thm list
haftmann@29544
    45
  val instance_of: theory -> string -> morphism -> term list
haftmann@29361
    46
  val specification_of: theory -> string -> term option * term list
haftmann@29361
    47
haftmann@29361
    48
  (* Storing results *)
haftmann@29361
    49
  val add_thmss: string -> string -> (Attrib.binding * (thm list * Attrib.src list) list) list ->
haftmann@29361
    50
    Proof.context -> Proof.context
haftmann@29361
    51
  val add_declaration: string -> declaration -> Proof.context -> Proof.context
wenzelm@35798
    52
  val add_syntax_declaration: string -> declaration -> Proof.context -> Proof.context
haftmann@29361
    53
haftmann@29361
    54
  (* Activation *)
wenzelm@30764
    55
  val activate_declarations: string * morphism -> Proof.context -> Proof.context
wenzelm@30764
    56
  val activate_facts: string * morphism -> Context.generic -> Context.generic
haftmann@29361
    57
  val init: string -> theory -> Proof.context
haftmann@29361
    58
haftmann@29361
    59
  (* Reasoning about locales *)
wenzelm@30725
    60
  val get_witnesses: Proof.context -> thm list
wenzelm@30725
    61
  val get_intros: Proof.context -> thm list
wenzelm@30725
    62
  val get_unfolds: Proof.context -> thm list
wenzelm@30725
    63
  val witness_add: attribute
wenzelm@30725
    64
  val intro_add: attribute
wenzelm@30725
    65
  val unfold_add: attribute
haftmann@29361
    66
  val intro_locales_tac: bool -> Proof.context -> thm list -> tactic
haftmann@29361
    67
haftmann@31988
    68
  (* Registrations and dependencies *)
ballarin@32845
    69
  val add_registration: string * morphism -> (morphism * bool) option ->
ballarin@32845
    70
    morphism -> theory -> theory
ballarin@32845
    71
  val amend_registration: string * morphism -> morphism * bool ->
ballarin@32845
    72
    morphism -> theory -> theory
haftmann@32074
    73
  val add_dependency: string -> string * morphism -> morphism -> theory -> theory
haftmann@29361
    74
haftmann@29361
    75
  (* Diagnostic *)
haftmann@29361
    76
  val print_locales: theory -> unit
wenzelm@30344
    77
  val print_locale: theory -> bool -> xstring -> unit
ballarin@32804
    78
  val print_registrations: theory -> string -> unit
haftmann@29361
    79
end;
haftmann@29361
    80
haftmann@29361
    81
structure Locale: LOCALE =
haftmann@29361
    82
struct
haftmann@29361
    83
haftmann@29361
    84
datatype ctxt = datatype Element.ctxt;
haftmann@29361
    85
haftmann@29392
    86
haftmann@29392
    87
(*** Theory data ***)
haftmann@29361
    88
haftmann@29361
    89
datatype locale = Loc of {
haftmann@29392
    90
  (** static part **)
wenzelm@30755
    91
  parameters: (string * sort) list * ((string * typ) * mixfix) list,
haftmann@29361
    92
    (* type and term parameters *)
haftmann@29361
    93
  spec: term option * term list,
haftmann@29361
    94
    (* assumptions (as a single predicate expression) and defines *)
haftmann@29441
    95
  intros: thm option * thm option,
haftmann@29441
    96
  axioms: thm list,
haftmann@29392
    97
  (** dynamic part **)
ballarin@36096
    98
  syntax_decls: (declaration * serial) list,
wenzelm@35798
    99
    (* syntax declarations *)
ballarin@36096
   100
  notes: ((string * (Attrib.binding * (thm list * Attrib.src list) list) list) * serial) list,
haftmann@29361
   101
    (* theorem declarations *)
ballarin@36651
   102
  dependencies: ((string * (morphism * morphism)) * serial) list
haftmann@29361
   103
    (* locale dependencies (sublocale relation) *)
haftmann@29392
   104
};
haftmann@29361
   105
wenzelm@35798
   106
fun mk_locale ((parameters, spec, intros, axioms), ((syntax_decls, notes), dependencies)) =
haftmann@29441
   107
  Loc {parameters = parameters, intros = intros, axioms = axioms, spec = spec,
wenzelm@35798
   108
    syntax_decls = syntax_decls, notes = notes, dependencies = dependencies};
wenzelm@35798
   109
wenzelm@35798
   110
fun map_locale f (Loc {parameters, spec, intros, axioms, syntax_decls, notes, dependencies}) =
wenzelm@35798
   111
  mk_locale (f ((parameters, spec, intros, axioms), ((syntax_decls, notes), dependencies)));
wenzelm@30754
   112
wenzelm@35798
   113
fun merge_locale
wenzelm@35798
   114
 (Loc {parameters, spec, intros, axioms, syntax_decls, notes, dependencies},
wenzelm@35798
   115
  Loc {syntax_decls = syntax_decls', notes = notes', dependencies = dependencies', ...}) =
wenzelm@35798
   116
    mk_locale
haftmann@29441
   117
      ((parameters, spec, intros, axioms),
wenzelm@35798
   118
        ((merge (eq_snd op =) (syntax_decls, syntax_decls'),
haftmann@29441
   119
          merge (eq_snd op =) (notes, notes')),
haftmann@29441
   120
            merge (eq_snd op =) (dependencies, dependencies')));
haftmann@29361
   121
wenzelm@33522
   122
structure Locales = Theory_Data
haftmann@29361
   123
(
wenzelm@33095
   124
  type T = locale Name_Space.table;
wenzelm@33159
   125
  val empty : T = Name_Space.empty_table "locale";
haftmann@29361
   126
  val extend = I;
wenzelm@33522
   127
  val merge = Name_Space.join_tables (K merge_locale);
haftmann@29361
   128
);
haftmann@29361
   129
wenzelm@33095
   130
val intern = Name_Space.intern o #1 o Locales.get;
wenzelm@33095
   131
val extern = Name_Space.extern o #1 o Locales.get;
haftmann@29392
   132
wenzelm@30725
   133
val get_locale = Symtab.lookup o #2 o Locales.get;
wenzelm@30725
   134
val defined = Symtab.defined o #2 o Locales.get;
haftmann@29361
   135
wenzelm@30725
   136
fun the_locale thy name =
wenzelm@30725
   137
  (case get_locale thy name of
wenzelm@30725
   138
    SOME (Loc loc) => loc
wenzelm@30725
   139
  | NONE => error ("Unknown locale " ^ quote name));
haftmann@29361
   140
wenzelm@35798
   141
fun register_locale binding parameters spec intros axioms syntax_decls notes dependencies thy =
wenzelm@33095
   142
  thy |> Locales.map (Name_Space.define true (Sign.naming_of thy)
wenzelm@30725
   143
    (binding,
wenzelm@30725
   144
      mk_locale ((parameters, spec, intros, axioms),
ballarin@36096
   145
        ((map (fn decl => (decl, serial ())) syntax_decls, map (fn n => (n, serial ())) notes),
ballarin@36651
   146
          map (fn d => (d |> apsnd (rpair Morphism.identity), serial ())) dependencies))) #> snd);
ballarin@36651
   147
          (* FIXME *)
haftmann@29361
   148
haftmann@29392
   149
fun change_locale name =
wenzelm@30725
   150
  Locales.map o apsnd o Symtab.map_entry name o map_locale o apsnd;
haftmann@29361
   151
haftmann@29361
   152
fun print_locales thy =
wenzelm@33095
   153
  Pretty.strs ("locales:" :: map #1 (Name_Space.extern_table (Locales.get thy)))
haftmann@29392
   154
  |> Pretty.writeln;
haftmann@29361
   155
haftmann@29361
   156
haftmann@29361
   157
(*** Primitive operations ***)
haftmann@29361
   158
haftmann@29392
   159
fun params_of thy = snd o #parameters o the_locale thy;
haftmann@29361
   160
haftmann@29441
   161
fun intros_of thy = #intros o the_locale thy;
haftmann@29441
   162
haftmann@29441
   163
fun axioms_of thy = #axioms o the_locale thy;
haftmann@29441
   164
haftmann@29392
   165
fun instance_of thy name morph = params_of thy name |>
wenzelm@30755
   166
  map (Morphism.term morph o Free o #1);
haftmann@29361
   167
haftmann@29392
   168
fun specification_of thy = #spec o the_locale thy;
haftmann@29361
   169
haftmann@29544
   170
fun dependencies_of thy name = the_locale thy name |>
haftmann@29544
   171
  #dependencies |> map fst;
haftmann@29544
   172
haftmann@29361
   173
haftmann@29361
   174
(*** Activate context elements of locale ***)
haftmann@29361
   175
haftmann@29361
   176
(** Identifiers: activated locales in theory or proof context **)
haftmann@29361
   177
ballarin@37105
   178
(* subsumption *)
ballarin@37103
   179
fun ident_le thy ((n: string, ts), (m, ss)) = (m = n) andalso Pattern.matchess thy (ss, ts);
ballarin@37105
   180
  (* smaller term is more general *)
ballarin@37105
   181
ballarin@37105
   182
(* total order *)
ballarin@37105
   183
fun ident_ord ((n: string, ts), (m, ss)) =
ballarin@37105
   184
  case fast_string_ord (m, n) of
ballarin@37105
   185
      EQUAL => list_ord Term_Ord.fast_term_ord (ts, ss)
ballarin@37105
   186
    | ord => ord;
haftmann@29361
   187
haftmann@29361
   188
local
haftmann@29361
   189
wenzelm@30754
   190
datatype 'a delayed = Ready of 'a | ToDo of 'a delayed * 'a delayed;
haftmann@29361
   191
wenzelm@33519
   192
structure Identifiers = Generic_Data
haftmann@29361
   193
(
wenzelm@30754
   194
  type T = (string * term list) list delayed;
wenzelm@30754
   195
  val empty = Ready [];
haftmann@29361
   196
  val extend = I;
wenzelm@33519
   197
  val merge = ToDo;
haftmann@29361
   198
);
haftmann@29361
   199
ballarin@37103
   200
fun finish thy (ToDo (i1, i2)) = merge (ident_le thy) (finish thy i1, finish thy i2)
haftmann@29361
   201
  | finish _ (Ready ids) = ids;
haftmann@29361
   202
haftmann@29361
   203
val _ = Context.>> (Context.map_theory (Theory.at_begin (fn thy =>
wenzelm@30725
   204
  (case Identifiers.get (Context.Theory thy) of
wenzelm@30725
   205
    Ready _ => NONE
wenzelm@30725
   206
  | ids => SOME (Context.theory_map (Identifiers.put (Ready (finish thy ids))) thy)))));
haftmann@29361
   207
wenzelm@30764
   208
in
haftmann@29361
   209
wenzelm@30764
   210
val get_idents = (fn Ready ids => ids) o Identifiers.get;
wenzelm@30764
   211
val put_idents = Identifiers.put o Ready;
haftmann@29361
   212
haftmann@29361
   213
end;
haftmann@29361
   214
haftmann@29361
   215
haftmann@29361
   216
(** Resolve locale dependencies in a depth-first fashion **)
haftmann@29361
   217
haftmann@29361
   218
local
haftmann@29361
   219
haftmann@29361
   220
val roundup_bound = 120;
haftmann@29361
   221
ballarin@32803
   222
fun add thy depth export (name, morph) (deps, marked) =
haftmann@29361
   223
  if depth > roundup_bound
haftmann@29361
   224
  then error "Roundup bound exceeded (sublocale relation probably not terminating)."
haftmann@29361
   225
  else
haftmann@29361
   226
    let
wenzelm@30754
   227
      val dependencies = dependencies_of thy name;
ballarin@32803
   228
      val instance = instance_of thy name (morph $> export);
haftmann@29361
   229
    in
ballarin@37103
   230
      if member (ident_le thy) marked (name, instance)
haftmann@29361
   231
      then (deps, marked)
haftmann@29361
   232
      else
haftmann@29361
   233
        let
ballarin@36651
   234
          val dependencies' = map (fn (name, (morph', export')) => (name, morph' $> export' $> morph)) dependencies;
haftmann@29361
   235
          val marked' = (name, instance) :: marked;
ballarin@32803
   236
          val (deps', marked'') = fold_rev (add thy (depth + 1) export) dependencies' ([], marked');
haftmann@29361
   237
        in
haftmann@29361
   238
          ((name, morph) :: deps' @ deps, marked'')
haftmann@29361
   239
        end
haftmann@29361
   240
    end;
haftmann@29361
   241
haftmann@29361
   242
in
haftmann@29361
   243
ballarin@33541
   244
(* Note that while identifiers always have the external (exported) view, activate_dep
ballarin@32803
   245
  is presented with the internal view. *)
ballarin@32803
   246
ballarin@32803
   247
fun roundup thy activate_dep export (name, morph) (marked, input) =
haftmann@29361
   248
  let
haftmann@29361
   249
    (* Find all dependencies incuding new ones (which are dependencies enriching
haftmann@29361
   250
      existing registrations). *)
ballarin@32803
   251
    val (dependencies, marked') = add thy 0 export (name, morph) ([], []);
ballarin@32800
   252
    (* Filter out fragments from marked; these won't be activated. *)
haftmann@29361
   253
    val dependencies' = filter_out (fn (name, morph) =>
ballarin@37103
   254
      member (ident_le thy) marked (name, instance_of thy name (morph $> export))) dependencies;
haftmann@29361
   255
  in
ballarin@37103
   256
    (merge (ident_le thy) (marked, marked'), input |> fold_rev activate_dep dependencies')
haftmann@29361
   257
  end;
haftmann@29361
   258
haftmann@29361
   259
end;
haftmann@29361
   260
haftmann@29361
   261
haftmann@29361
   262
(* Declarations, facts and entire locale content *)
haftmann@29361
   263
wenzelm@35798
   264
fun activate_syntax_decls (name, morph) context =
haftmann@29361
   265
  let
wenzelm@30764
   266
    val thy = Context.theory_of context;
wenzelm@35798
   267
    val {syntax_decls, ...} = the_locale thy name;
haftmann@29361
   268
  in
wenzelm@30764
   269
    context
wenzelm@35798
   270
    |> fold_rev (fn (decl, _) => decl morph) syntax_decls
haftmann@29361
   271
  end;
haftmann@29361
   272
haftmann@29361
   273
fun activate_notes activ_elem transfer thy (name, morph) input =
haftmann@29361
   274
  let
haftmann@29392
   275
    val {notes, ...} = the_locale thy name;
haftmann@29361
   276
    fun activate ((kind, facts), _) input =
haftmann@29361
   277
      let
haftmann@29361
   278
        val facts' = facts |> Element.facts_map (Element.morph_ctxt (transfer input $> morph))
haftmann@29361
   279
      in activ_elem (Notes (kind, facts')) input end;
haftmann@29361
   280
  in
haftmann@29361
   281
    fold_rev activate notes input
haftmann@29361
   282
  end;
haftmann@29361
   283
haftmann@29361
   284
fun activate_all name thy activ_elem transfer (marked, input) =
haftmann@29361
   285
  let
wenzelm@30764
   286
    val {parameters = (_, params), spec = (asm, defs), ...} = the_locale thy name;
wenzelm@30764
   287
    val input' = input |>
wenzelm@30755
   288
      (not (null params) ?
wenzelm@30755
   289
        activ_elem (Fixes (map (fn ((x, T), mx) => (Binding.name x, SOME T, mx)) params))) |>
haftmann@29361
   290
      (* FIXME type parameters *)
wenzelm@30764
   291
      (case asm of SOME A => activ_elem (Assumes [(Attrib.empty_binding, [(A, [])])]) | _ => I) |>
haftmann@29361
   292
      (if not (null defs)
haftmann@29361
   293
        then activ_elem (Defines (map (fn def => (Attrib.empty_binding, (def, []))) defs))
wenzelm@30764
   294
        else I);
wenzelm@30773
   295
    val activate = activate_notes activ_elem transfer thy;
wenzelm@30764
   296
  in
ballarin@32803
   297
    roundup thy activate Morphism.identity (name, Morphism.identity) (marked, input')
haftmann@29361
   298
  end;
haftmann@29361
   299
haftmann@29361
   300
haftmann@29361
   301
(** Public activation functions **)
haftmann@29361
   302
wenzelm@30775
   303
fun activate_declarations dep = Context.proof_map (fn context =>
wenzelm@30764
   304
  let
wenzelm@30764
   305
    val thy = Context.theory_of context;
wenzelm@35798
   306
  in
wenzelm@35798
   307
    roundup thy activate_syntax_decls Morphism.identity dep (get_idents context, context)
wenzelm@35798
   308
    |-> put_idents
wenzelm@35798
   309
  end);
haftmann@29361
   310
wenzelm@30764
   311
fun activate_facts dep context =
wenzelm@30764
   312
  let
wenzelm@30764
   313
    val thy = Context.theory_of context;
wenzelm@30775
   314
    val activate = activate_notes Element.init (Element.transfer_morphism o Context.theory_of) thy;
ballarin@32803
   315
  in roundup thy activate Morphism.identity dep (get_idents context, context) |-> put_idents end;
haftmann@29361
   316
haftmann@29361
   317
fun init name thy =
wenzelm@30775
   318
  activate_all name thy Element.init (Element.transfer_morphism o Context.theory_of)
wenzelm@36610
   319
    ([], Context.Proof (ProofContext.init_global thy)) |-> put_idents |> Context.proof_of;
haftmann@29361
   320
wenzelm@30764
   321
fun print_locale thy show_facts raw_name =
haftmann@29361
   322
  let
wenzelm@30764
   323
    val name = intern thy raw_name;
wenzelm@30764
   324
    val ctxt = init name thy;
wenzelm@30764
   325
    fun cons_elem (elem as Notes _) = show_facts ? cons elem
wenzelm@30764
   326
      | cons_elem elem = cons elem;
wenzelm@30764
   327
    val elems =
wenzelm@30764
   328
      activate_all name thy cons_elem (K (Element.transfer_morphism thy)) ([], [])
wenzelm@30764
   329
      |> snd |> rev;
haftmann@29361
   330
  in
wenzelm@30764
   331
    Pretty.big_list "locale elements:" (map (Pretty.chunks o Element.pretty_ctxt ctxt) elems)
wenzelm@30764
   332
    |> Pretty.writeln
wenzelm@30764
   333
  end;
haftmann@29361
   334
haftmann@29361
   335
haftmann@29361
   336
(*** Registrations: interpretations in theories ***)
haftmann@29361
   337
ballarin@37105
   338
structure Idtab = Table(type key = string * term list val ord = ident_ord);
ballarin@37105
   339
wenzelm@33522
   340
structure Registrations = Theory_Data
haftmann@29361
   341
(
ballarin@37105
   342
  type T = ((morphism * morphism) * serial) Idtab.table *
ballarin@37105
   343
    (* registrations, indexed by locale name and instance;
ballarin@36090
   344
       serial points to mixin list *)
ballarin@37105
   345
    (((morphism * bool) * serial) list) Inttab.table;
ballarin@37105
   346
    (* table of mixin lists, per list mixins in reverse order of declaration;
ballarin@36090
   347
       lists indexed by registration serial,
ballarin@36090
   348
       entries for empty lists may be omitted *)
ballarin@37105
   349
  val empty = (Idtab.empty, Inttab.empty);
haftmann@29361
   350
  val extend = I;
wenzelm@33522
   351
  fun merge ((r1, m1), (r2, m2)) : T =
ballarin@37105
   352
    (Idtab.join (K (fst)) (r1, r2), (* pick left registration, FIXME? *)
ballarin@37105
   353
      Inttab.join (K (Library.merge (eq_snd op =))) (m1, m2));
haftmann@29361
   354
    (* FIXME consolidate with dependencies, consider one data slot only *)
haftmann@29361
   355
);
haftmann@29361
   356
ballarin@32801
   357
ballarin@32801
   358
(* Primitive operations *)
ballarin@32801
   359
ballarin@37104
   360
fun add_reg thy export (name, morph) =
ballarin@37105
   361
  Registrations.map (apfst (Idtab.insert (K false)
ballarin@37105
   362
    ((name, instance_of thy name (morph $> export)), ((morph, export), serial ()))));
haftmann@31988
   363
ballarin@36095
   364
fun add_mixin serial' mixin =
ballarin@36095
   365
  (* registration to be amended identified by its serial id *)
ballarin@37105
   366
  Registrations.map (apsnd (Inttab.map_default (serial', []) (cons (mixin, serial ()))));
ballarin@32801
   367
ballarin@32804
   368
fun get_mixins thy (name, morph) =
ballarin@32804
   369
  let
ballarin@32804
   370
    val (regs, mixins) = Registrations.get thy;
ballarin@32804
   371
  in
ballarin@37105
   372
    case Idtab.lookup regs (name, instance_of thy name morph) of
ballarin@32804
   373
      NONE => []
ballarin@37105
   374
    | SOME (_, serial) => the_default [] (Inttab.lookup mixins serial)
ballarin@32804
   375
  end;
ballarin@32804
   376
ballarin@32804
   377
fun collect_mixins thy (name, morph) =
ballarin@32804
   378
  roundup thy (fn dep => fn mixins =>
ballarin@32804
   379
    merge (eq_snd op =) (mixins, get_mixins thy dep)) Morphism.identity (name, morph) ([], [])
ballarin@37102
   380
  |> snd |> filter (snd o fst);  (* only inheritable mixins *) (* FIXME *)
ballarin@36095
   381
  (* FIXME refactor usage *)
ballarin@32804
   382
ballarin@36091
   383
fun compose_mixins mixins =
ballarin@36091
   384
  fold_rev Morphism.compose (map (fst o fst) mixins) Morphism.identity;
ballarin@36091
   385
ballarin@36091
   386
fun reg_morph mixins ((name, (base, export)), serial) =
ballarin@36091
   387
  let val mix = the_default [] (AList.lookup (op =) mixins serial) |> compose_mixins;
ballarin@36091
   388
  in (name, base $> mix $> export) end;
ballarin@36091
   389
ballarin@36091
   390
fun these_registrations thy name = Registrations.get thy
ballarin@37105
   391
  |>> Idtab.dest
ballarin@37105
   392
  |>> filter (curry (op =) name o fst o fst)
ballarin@36095
   393
  (* with inherited mixins *)
ballarin@37105
   394
  |-> (fn regs => fn _ => map (fn ((name, _), ((base, export) ,_)) =>
ballarin@36091
   395
    (name, base $> (collect_mixins thy (name, base $> export) |> compose_mixins) $> export)) regs);
ballarin@36091
   396
ballarin@37101
   397
fun all_registrations thy = Registrations.get thy (* FIXME clone *)
ballarin@37105
   398
  |>> Idtab.dest
ballarin@37101
   399
  (* with inherited mixins *)
ballarin@37105
   400
  |-> (fn regs => fn _ => map (fn ((name, _), ((base, export) ,_)) =>
ballarin@37101
   401
    (name, base $> (collect_mixins thy (name, base $> export) |> compose_mixins) $> export)) regs);
ballarin@32804
   402
ballarin@32804
   403
fun activate_notes' activ_elem transfer thy export (name, morph) input =
ballarin@32804
   404
  let
ballarin@32804
   405
    val {notes, ...} = the_locale thy name;
ballarin@32804
   406
    fun activate ((kind, facts), _) input =
ballarin@32804
   407
      let
ballarin@32804
   408
        val mixin = collect_mixins thy (name, morph $> export) |> compose_mixins;
ballarin@32804
   409
        val facts' = facts |> Element.facts_map (Element.morph_ctxt (transfer input $> morph $> mixin $> export))
ballarin@32804
   410
      in activ_elem (Notes (kind, facts')) input end;
ballarin@32804
   411
  in
ballarin@32804
   412
    fold_rev activate notes input
ballarin@32804
   413
  end;
ballarin@32804
   414
ballarin@32804
   415
fun activate_facts' export dep context =
ballarin@32804
   416
  let
ballarin@32804
   417
    val thy = Context.theory_of context;
ballarin@32804
   418
    val activate = activate_notes' Element.init (Element.transfer_morphism o Context.theory_of) thy export;
ballarin@32804
   419
  in roundup thy activate export dep (get_idents context, context) |-> put_idents end;
ballarin@32804
   420
ballarin@32804
   421
ballarin@32804
   422
(* Diagnostic *)
ballarin@32804
   423
ballarin@36652
   424
fun pretty_reg thy (name, morph) =
ballarin@32804
   425
  let
ballarin@32804
   426
    val name' = extern thy name;
wenzelm@36610
   427
    val ctxt = ProofContext.init_global thy;
ballarin@32804
   428
    fun prt_qual (qual, mand) = Pretty.str (qual ^ (if mand then "!" else "?"));
ballarin@32804
   429
    fun prt_quals qs = Pretty.separate "." (map prt_qual qs) |> Pretty.block;
ballarin@32804
   430
    val prt_term = Pretty.quote o Syntax.pretty_term ctxt;
ballarin@32804
   431
    fun prt_term' t = if !show_types
ballarin@32804
   432
      then Pretty.block [prt_term t, Pretty.brk 1, Pretty.str "::",
ballarin@32804
   433
        Pretty.brk 1, (Pretty.quote o Syntax.pretty_typ ctxt) (type_of t)]
ballarin@32804
   434
      else prt_term t;
ballarin@32804
   435
    fun prt_inst ts =
ballarin@32804
   436
      Pretty.block (Pretty.breaks (Pretty.str name' :: map prt_term' ts));
ballarin@36652
   437
ballarin@36652
   438
    val qs = Binding.name "x" |> Morphism.binding morph |> Binding.prefix_of;
ballarin@36652
   439
    val ts = instance_of thy name morph;
ballarin@32804
   440
  in
ballarin@36652
   441
    case qs of
ballarin@36652
   442
       [] => prt_inst ts
ballarin@36652
   443
     | qs => Pretty.block [prt_quals qs, Pretty.brk 1, Pretty.str ":",
ballarin@36652
   444
         Pretty.brk 1, prt_inst ts]
ballarin@32804
   445
  end;
ballarin@32801
   446
ballarin@36652
   447
fun print_registrations thy raw_name =
ballarin@36652
   448
  (case these_registrations thy (intern thy raw_name) of
ballarin@36652
   449
      [] => Pretty.str ("no interpretations")
ballarin@36652
   450
    | regs => Pretty.big_list "interpretations:" (map (pretty_reg thy) (rev regs)))
ballarin@36652
   451
  |> Pretty.writeln;
ballarin@36652
   452
ballarin@32801
   453
ballarin@32801
   454
(* Add and extend registrations *)
ballarin@32801
   455
ballarin@32804
   456
fun amend_registration (name, morph) mixin export thy =
ballarin@32804
   457
  let
ballarin@32804
   458
    val regs = Registrations.get thy |> fst;
ballarin@32804
   459
    val base = instance_of thy name (morph $> export);
ballarin@32804
   460
  in
ballarin@37105
   461
    case Idtab.lookup regs (name, base) of
ballarin@32804
   462
        NONE => error ("No interpretation of locale " ^
ballarin@32804
   463
          quote (extern thy name) ^ " and\nparameter instantiation " ^
ballarin@32804
   464
          space_implode " " (map (quote o Syntax.string_of_term_global thy) base) ^
ballarin@32804
   465
          " available")
ballarin@36095
   466
      | SOME (_, serial') => add_mixin serial' mixin thy
ballarin@32804
   467
  end;
ballarin@32804
   468
ballarin@32801
   469
(* Note that a registration that would be subsumed by an existing one will not be
ballarin@32801
   470
   generated, and it will not be possible to amend it. *)
ballarin@32801
   471
ballarin@32804
   472
fun add_registration (name, base_morph) mixin export thy =
ballarin@32801
   473
  let
ballarin@37101
   474
    val mix = case mixin of NONE => Morphism.identity
ballarin@37101
   475
          | SOME (mix, _) => mix;
ballarin@37102
   476
    val morph = base_morph $> mix;
ballarin@37102
   477
    val inst = instance_of thy name morph;
ballarin@32801
   478
  in
ballarin@37103
   479
    if member (ident_le thy) (get_idents (Context.Theory thy)) (name, inst)
ballarin@32801
   480
    then thy
ballarin@32801
   481
    else
ballarin@36095
   482
      (get_idents (Context.Theory thy), thy)
ballarin@36095
   483
      (* add new registrations with inherited mixins *)
ballarin@37104
   484
      |> roundup thy (add_reg thy export) export (name, morph)
ballarin@36095
   485
      |> snd
ballarin@36095
   486
      (* add mixin *)
ballarin@36095
   487
      |> (case mixin of NONE => I
ballarin@37102
   488
           | SOME mixin => amend_registration (name, morph) mixin export)
ballarin@36095
   489
      (* activate import hierarchy as far as not already active *)
ballarin@37102
   490
      |> Context.theory_map (activate_facts' export (name, morph))
ballarin@32801
   491
  end;
ballarin@32801
   492
haftmann@29361
   493
haftmann@31988
   494
(*** Dependencies ***)
haftmann@31988
   495
ballarin@37102
   496
fun add_dependency loc (name, morph) export thy =
ballarin@37102
   497
  let
ballarin@37102
   498
    val thy' = (change_locale loc o apsnd) (cons ((name, (morph, export)), serial ())) thy;
ballarin@37102
   499
    val (_, regs) = fold_rev (roundup thy' cons export)
ballarin@37102
   500
      (all_registrations thy') (get_idents (Context.Theory thy'), []);
ballarin@37102
   501
  in
ballarin@37102
   502
    thy'
ballarin@37102
   503
    |> fold_rev (fn dep => add_registration dep NONE export) regs
ballarin@37102
   504
  end;
haftmann@31988
   505
haftmann@31988
   506
haftmann@29361
   507
(*** Storing results ***)
haftmann@29361
   508
haftmann@29361
   509
(* Theorems *)
haftmann@29361
   510
haftmann@29361
   511
fun add_thmss loc kind args ctxt =
haftmann@29361
   512
  let
wenzelm@30777
   513
    val (Notes args', ctxt') = Element.activate_i (Notes (kind, args)) ctxt;
haftmann@29361
   514
    val ctxt'' = ctxt' |> ProofContext.theory (
ballarin@36088
   515
      (change_locale loc o apfst o apsnd) (cons (args', serial ()))
haftmann@29392
   516
        #>
haftmann@29361
   517
      (* Registrations *)
haftmann@31988
   518
      (fn thy => fold_rev (fn (_, morph) =>
haftmann@29361
   519
            let
haftmann@29361
   520
              val args'' = snd args' |> Element.facts_map (Element.morph_ctxt morph) |>
haftmann@29361
   521
                Attrib.map_facts (Attrib.attribute_i thy)
wenzelm@30438
   522
            in PureThy.note_thmss kind args'' #> snd end)
haftmann@31988
   523
        (these_registrations thy loc) thy))
haftmann@29361
   524
  in ctxt'' end;
haftmann@29361
   525
haftmann@29361
   526
haftmann@29361
   527
(* Declarations *)
haftmann@29361
   528
wenzelm@35798
   529
fun add_declaration loc decl =
wenzelm@33643
   530
  add_thmss loc ""
wenzelm@35798
   531
    [((Binding.conceal Binding.empty,
wenzelm@35798
   532
        [Attrib.internal (fn phi => Thm.declaration_attribute (K (decl phi)))]),
wenzelm@33278
   533
      [([Drule.dummy_thm], [])])];
haftmann@29361
   534
wenzelm@35798
   535
fun add_syntax_declaration loc decl =
ballarin@36096
   536
  ProofContext.theory ((change_locale loc o apfst o apfst) (cons (decl, serial ())))
wenzelm@35798
   537
  #> add_declaration loc decl;
haftmann@29361
   538
wenzelm@30725
   539
haftmann@29361
   540
(*** Reasoning about locales ***)
haftmann@29361
   541
wenzelm@30725
   542
(* Storage for witnesses, intro and unfold rules *)
haftmann@29361
   543
wenzelm@33519
   544
structure Thms = Generic_Data
wenzelm@30725
   545
(
wenzelm@30725
   546
  type T = thm list * thm list * thm list;
wenzelm@30725
   547
  val empty = ([], [], []);
wenzelm@30725
   548
  val extend = I;
wenzelm@33519
   549
  fun merge ((witnesses1, intros1, unfolds1), (witnesses2, intros2, unfolds2)) =
wenzelm@30725
   550
   (Thm.merge_thms (witnesses1, witnesses2),
wenzelm@30725
   551
    Thm.merge_thms (intros1, intros2),
wenzelm@30725
   552
    Thm.merge_thms (unfolds1, unfolds2));
wenzelm@30725
   553
);
haftmann@29361
   554
wenzelm@30725
   555
val get_witnesses = #1 o Thms.get o Context.Proof;
wenzelm@30725
   556
val get_intros = #2 o Thms.get o Context.Proof;
wenzelm@30725
   557
val get_unfolds = #3 o Thms.get o Context.Proof;
haftmann@29361
   558
wenzelm@30725
   559
val witness_add =
wenzelm@30725
   560
  Thm.declaration_attribute (fn th => Thms.map (fn (x, y, z) => (Thm.add_thm th x, y, z)));
wenzelm@30725
   561
val intro_add =
wenzelm@30725
   562
  Thm.declaration_attribute (fn th => Thms.map (fn (x, y, z) => (x, Thm.add_thm th y, z)));
wenzelm@30725
   563
val unfold_add =
wenzelm@30725
   564
  Thm.declaration_attribute (fn th => Thms.map (fn (x, y, z) => (x, y, Thm.add_thm th z)));
haftmann@29361
   565
wenzelm@30725
   566
wenzelm@30725
   567
(* Tactic *)
wenzelm@30725
   568
ballarin@36093
   569
fun gen_intro_locales_tac intros_tac eager ctxt =
ballarin@36093
   570
  intros_tac
wenzelm@30725
   571
    (get_witnesses ctxt @ get_intros ctxt @ (if eager then get_unfolds ctxt else []));
haftmann@29361
   572
ballarin@36093
   573
val intro_locales_tac = gen_intro_locales_tac Method.intros_tac;
ballarin@36093
   574
val try_intro_locales_tac= gen_intro_locales_tac Method.try_intros_tac;
ballarin@36093
   575
haftmann@29361
   576
val _ = Context.>> (Context.map_theory
ballarin@36093
   577
 (Method.setup (Binding.name "intro_locales") (Scan.succeed (METHOD o try_intro_locales_tac false))
wenzelm@30515
   578
    "back-chain introduction rules of locales without unfolding predicates" #>
ballarin@36093
   579
  Method.setup (Binding.name "unfold_locales") (Scan.succeed (METHOD o try_intro_locales_tac true))
wenzelm@30515
   580
    "back-chain all introduction rules of locales"));
haftmann@29361
   581
haftmann@29361
   582
end;