src/Pure/Isar/class.ML
author haftmann
Mon Oct 08 22:03:21 2007 +0200 (2007-10-08)
changeset 24914 95cda5dd58d5
parent 24901 d3cbf79769b9
child 24920 2a45e400fdad
permissions -rw-r--r--
added proper subclass concept; improved class target
haftmann@24218
     1
(*  Title:      Pure/Isar/class.ML
haftmann@24218
     2
    ID:         $Id$
haftmann@24218
     3
    Author:     Florian Haftmann, TU Muenchen
haftmann@24218
     4
haftmann@24218
     5
Type classes derived from primitive axclasses and locales.
haftmann@24218
     6
*)
haftmann@24218
     7
haftmann@24218
     8
signature CLASS =
haftmann@24218
     9
sig
haftmann@24218
    10
  val fork_mixfix: bool -> string option -> mixfix -> mixfix * mixfix
haftmann@24218
    11
haftmann@24218
    12
  val axclass_cmd: bstring * xstring list
haftmann@24589
    13
    -> ((bstring * Attrib.src list) * string list) list
haftmann@24589
    14
    -> theory -> class * theory
haftmann@24589
    15
  val classrel_cmd: xstring * xstring -> theory -> Proof.state
haftmann@24914
    16
haftmann@24748
    17
  val class: bool -> bstring -> class list -> Element.context_i Locale.element list
haftmann@24218
    18
    -> string list -> theory -> string * Proof.context
haftmann@24748
    19
  val class_cmd: bool -> bstring -> xstring list -> Element.context Locale.element list
haftmann@24589
    20
    -> xstring list -> theory -> string * Proof.context
haftmann@24914
    21
  val init: class -> Proof.context -> Proof.context;
haftmann@24423
    22
  val add_const_in_class: string -> (string * term) * Syntax.mixfix
haftmann@24914
    23
    -> theory -> string * theory
haftmann@24914
    24
  val add_abbrev_in_class: string -> Syntax.mode -> (string * term) * mixfix
haftmann@24914
    25
    -> theory -> string * theory
haftmann@24914
    26
  val remove_constraint: sort -> string -> Proof.context -> Proof.context
haftmann@24589
    27
  val intro_classes_tac: thm list -> tactic
haftmann@24589
    28
  val default_intro_classes_tac: thm list -> tactic
haftmann@24589
    29
  val class_of_locale: theory -> string -> class option
haftmann@24914
    30
  val locale_of_class: theory -> class -> string
haftmann@24914
    31
  val local_syntax: theory -> class -> bool
haftmann@24589
    32
  val print_classes: theory -> unit
haftmann@24423
    33
haftmann@24589
    34
  val instance_arity: (theory -> theory) -> arity list -> theory -> Proof.state
haftmann@24589
    35
  val instance: arity list -> ((bstring * Attrib.src list) * term) list
haftmann@24423
    36
    -> (thm list -> theory -> theory)
haftmann@24218
    37
    -> theory -> Proof.state
haftmann@24589
    38
  val instance_cmd: (bstring * xstring list * xstring) list
haftmann@24589
    39
    -> ((bstring * Attrib.src list) * xstring) list
haftmann@24423
    40
    -> (thm list -> theory -> theory)
haftmann@24218
    41
    -> theory -> Proof.state
haftmann@24589
    42
  val prove_instance: tactic -> arity list
haftmann@24218
    43
    -> ((bstring * Attrib.src list) * term) list
haftmann@24423
    44
    -> theory -> thm list * theory
haftmann@24423
    45
  val unoverload: theory -> conv
haftmann@24423
    46
  val overload: theory -> conv
haftmann@24423
    47
  val unoverload_const: theory -> string * typ -> string
haftmann@24423
    48
  val inst_const: theory -> string * string -> string
haftmann@24423
    49
  val param_const: theory -> string -> (string * string) option
haftmann@24435
    50
  val params_of_sort: theory -> sort -> (string * (string * typ)) list
haftmann@24218
    51
end;
haftmann@24218
    52
haftmann@24218
    53
structure Class : CLASS =
haftmann@24218
    54
struct
haftmann@24218
    55
haftmann@24218
    56
(** auxiliary **)
haftmann@24218
    57
haftmann@24218
    58
fun fork_mixfix is_loc some_class mx =
haftmann@24218
    59
  let
haftmann@24218
    60
    val mx' = Syntax.unlocalize_mixfix mx;
haftmann@24901
    61
    val mx_global = if not is_loc orelse (is_some some_class andalso not (mx = mx'))
haftmann@24901
    62
      then mx' else NoSyn;
haftmann@24218
    63
    val mx_local = if is_loc then mx else NoSyn;
haftmann@24218
    64
  in (mx_global, mx_local) end;
haftmann@24218
    65
haftmann@24589
    66
fun prove_interpretation tac prfx_atts expr insts =
haftmann@24589
    67
  Locale.interpretation_i I prfx_atts expr insts
haftmann@24589
    68
  #> Proof.global_terminal_proof
haftmann@24589
    69
      (Method.Basic (K (Method.SIMPLE_METHOD tac), Position.none), NONE)
haftmann@24589
    70
  #> ProofContext.theory_of;
haftmann@24589
    71
haftmann@24589
    72
fun OF_LAST thm1 thm2 =
haftmann@24589
    73
  let
haftmann@24589
    74
    val n = (length o Logic.strip_imp_prems o prop_of) thm2;
haftmann@24589
    75
  in (thm1 RSN (n, thm2)) end;
haftmann@24589
    76
haftmann@24589
    77
fun strip_all_ofclass thy sort =
haftmann@24589
    78
  let
wenzelm@24847
    79
    val typ = TVar ((Name.aT, 0), sort);
haftmann@24589
    80
    fun prem_inclass t =
haftmann@24589
    81
      case Logic.strip_imp_prems t
haftmann@24589
    82
       of ofcls :: _ => try Logic.dest_inclass ofcls
haftmann@24589
    83
        | [] => NONE;
haftmann@24589
    84
    fun strip_ofclass class thm =
haftmann@24589
    85
      thm OF (fst o AxClass.of_sort thy (typ, [class])) AxClass.cache;
haftmann@24589
    86
    fun strip thm = case (prem_inclass o Thm.prop_of) thm
haftmann@24589
    87
     of SOME (_, class) => thm |> strip_ofclass class |> strip
haftmann@24589
    88
      | NONE => thm;
haftmann@24589
    89
  in strip end;
haftmann@24589
    90
haftmann@24589
    91
haftmann@24589
    92
(** axclass command **)
haftmann@24589
    93
haftmann@24218
    94
fun axclass_cmd (class, raw_superclasses) raw_specs thy =
haftmann@24218
    95
  let
haftmann@24218
    96
    val ctxt = ProofContext.init thy;
haftmann@24218
    97
    val superclasses = map (Sign.read_class thy) raw_superclasses;
haftmann@24589
    98
    val name_atts = map ((apsnd o map) (Attrib.attribute thy) o fst)
haftmann@24589
    99
      raw_specs;
haftmann@24589
   100
    val axiomss = ProofContext.read_propp (ctxt, map (map (rpair []) o snd)
haftmann@24589
   101
          raw_specs)
haftmann@24218
   102
      |> snd
haftmann@24218
   103
      |> (map o map) fst;
haftmann@24589
   104
  in
haftmann@24589
   105
    AxClass.define_class (class, superclasses) []
haftmann@24589
   106
      (name_atts ~~ axiomss) thy
haftmann@24589
   107
  end;
haftmann@24218
   108
haftmann@24218
   109
local
haftmann@24218
   110
haftmann@24218
   111
fun gen_instance mk_prop add_thm after_qed insts thy =
haftmann@24218
   112
  let
haftmann@24218
   113
    fun after_qed' results =
haftmann@24218
   114
      ProofContext.theory ((fold o fold) add_thm results #> after_qed);
haftmann@24218
   115
  in
haftmann@24218
   116
    thy
haftmann@24218
   117
    |> ProofContext.init
haftmann@24589
   118
    |> Proof.theorem_i NONE after_qed' ((map (fn t => [(t, [])])
haftmann@24589
   119
        o maps (mk_prop thy)) insts)
haftmann@24218
   120
  end;
haftmann@24218
   121
haftmann@24218
   122
in
haftmann@24218
   123
haftmann@24589
   124
val instance_arity =
haftmann@24218
   125
  gen_instance (Logic.mk_arities oo Sign.cert_arity) AxClass.add_arity;
haftmann@24589
   126
val classrel =
haftmann@24218
   127
  gen_instance (single oo (Logic.mk_classrel oo AxClass.cert_classrel))
haftmann@24218
   128
    AxClass.add_classrel I o single;
haftmann@24589
   129
val classrel_cmd =
haftmann@24589
   130
  gen_instance (single oo (Logic.mk_classrel oo AxClass.read_classrel))
haftmann@24589
   131
    AxClass.add_classrel I o single;
haftmann@24218
   132
haftmann@24218
   133
end; (*local*)
haftmann@24218
   134
haftmann@24218
   135
haftmann@24589
   136
(** explicit constants for overloaded definitions **)
haftmann@24304
   137
haftmann@24304
   138
structure InstData = TheoryDataFun
haftmann@24304
   139
(
haftmann@24423
   140
  type T = (string * thm) Symtab.table Symtab.table * (string * string) Symtab.table;
haftmann@24423
   141
    (*constant name ~> type constructor ~> (constant name, equation),
haftmann@24423
   142
        constant name ~> (constant name, type constructor)*)
haftmann@24423
   143
  val empty = (Symtab.empty, Symtab.empty);
haftmann@24304
   144
  val copy = I;
haftmann@24304
   145
  val extend = I;
haftmann@24423
   146
  fun merge _ ((taba1, tabb1), (taba2, tabb2)) =
haftmann@24423
   147
    (Symtab.join (K (Symtab.merge (K true))) (taba1, taba2),
haftmann@24423
   148
      Symtab.merge (K true) (tabb1, tabb2));
haftmann@24304
   149
);
haftmann@24304
   150
haftmann@24589
   151
fun inst_thms f thy = (Symtab.fold (Symtab.fold (cons o f o snd o snd) o snd) o fst)
haftmann@24589
   152
    (InstData.get thy) [];
haftmann@24589
   153
fun add_inst (c, tyco) inst = (InstData.map o apfst
haftmann@24589
   154
      o Symtab.map_default (c, Symtab.empty)) (Symtab.update_new (tyco, inst))
haftmann@24423
   155
  #> (InstData.map o apsnd) (Symtab.update_new (fst inst, (c, tyco)));
haftmann@24304
   156
haftmann@24589
   157
fun unoverload thy = MetaSimplifier.rewrite false (inst_thms I thy);
haftmann@24589
   158
fun overload thy = MetaSimplifier.rewrite false (inst_thms symmetric thy);
haftmann@24304
   159
haftmann@24304
   160
fun inst_const thy (c, tyco) =
haftmann@24423
   161
  (fst o the o Symtab.lookup ((the o Symtab.lookup (fst (InstData.get thy))) c)) tyco;
haftmann@24423
   162
fun unoverload_const thy (c_ty as (c, _)) =
haftmann@24423
   163
  case AxClass.class_of_param thy c
haftmann@24423
   164
   of SOME class => (case Sign.const_typargs thy c_ty
haftmann@24589
   165
       of [Type (tyco, _)] => (case Symtab.lookup
haftmann@24589
   166
           ((the o Symtab.lookup (fst (InstData.get thy))) c) tyco
haftmann@24423
   167
             of SOME (c, _) => c
haftmann@24423
   168
              | NONE => c)
haftmann@24423
   169
        | [_] => c)
haftmann@24423
   170
    | NONE => c;
haftmann@24423
   171
haftmann@24423
   172
val param_const = Symtab.lookup o snd o InstData.get;
haftmann@24304
   173
haftmann@24304
   174
fun add_inst_def (class, tyco) (c, ty) thy =
haftmann@24304
   175
  let
haftmann@24304
   176
    val tyco_base = NameSpace.base tyco;
haftmann@24304
   177
    val name_inst = NameSpace.base class ^ "_" ^ tyco_base ^ "_inst";
haftmann@24304
   178
    val c_inst_base = NameSpace.base c ^ "_" ^ tyco_base;
haftmann@24304
   179
  in
haftmann@24304
   180
    thy
haftmann@24304
   181
    |> Sign.sticky_prefix name_inst
haftmann@24435
   182
    |> PureThy.simple_def ("", [])
haftmann@24435
   183
         (((c_inst_base, ty, Syntax.NoSyn), []), Const (c, ty))
haftmann@24435
   184
    |-> (fn (c_inst, thm) => add_inst (c, tyco) (c_inst, symmetric thm))
haftmann@24435
   185
    |> Sign.restore_naming thy
haftmann@24304
   186
  end;
haftmann@24304
   187
haftmann@24304
   188
fun add_inst_def' (class, tyco) (c, ty) thy =
haftmann@24423
   189
  if case Symtab.lookup (fst (InstData.get thy)) c
haftmann@24304
   190
   of NONE => true
haftmann@24304
   191
    | SOME tab => is_none (Symtab.lookup tab tyco)
haftmann@24304
   192
  then add_inst_def (class, tyco) (c, Logic.unvarifyT ty) thy
haftmann@24304
   193
  else thy;
haftmann@24304
   194
haftmann@24304
   195
fun add_def ((class, tyco), ((name, prop), atts)) thy =
haftmann@24304
   196
  let
haftmann@24589
   197
    val ((lhs as Const (c, ty), args), rhs) =
haftmann@24589
   198
      (apfst Term.strip_comb o Logic.dest_equals) prop;
haftmann@24701
   199
    fun (*add_inst' def ([], (Const (c_inst, ty))) =
haftmann@24423
   200
          if forall (fn TFree _ => true | _ => false) (Sign.const_typargs thy (c_inst, ty))
haftmann@24304
   201
          then add_inst (c, tyco) (c_inst, def)
haftmann@24304
   202
          else add_inst_def (class, tyco) (c, ty)
haftmann@24701
   203
      |*) add_inst' _ t = add_inst_def (class, tyco) (c, ty);
haftmann@24304
   204
  in
haftmann@24304
   205
    thy
haftmann@24304
   206
    |> PureThy.add_defs_i true [((name, prop), map (Attrib.attribute thy) atts)]
haftmann@24304
   207
    |-> (fn [def] => add_inst' def (args, rhs) #> pair def)
haftmann@24304
   208
  end;
haftmann@24304
   209
haftmann@24304
   210
haftmann@24589
   211
(** instances with implicit parameter handling **)
haftmann@24218
   212
haftmann@24218
   213
local
haftmann@24218
   214
wenzelm@24707
   215
fun gen_read_def thy prep_att parse_prop ((raw_name, raw_atts), raw_t) =
haftmann@24218
   216
  let
wenzelm@24707
   217
    val ctxt = ProofContext.init thy;
wenzelm@24707
   218
    val t = parse_prop ctxt raw_t |> Syntax.check_prop ctxt;
haftmann@24218
   219
    val ((c, ty), _) = Sign.cert_def (Sign.pp thy) t;
haftmann@24218
   220
    val atts = map (prep_att thy) raw_atts;
haftmann@24218
   221
    val insts = Consts.typargs (Sign.consts_of thy) (c, ty);
haftmann@24218
   222
    val name = case raw_name
haftmann@24218
   223
     of "" => NONE
haftmann@24218
   224
      | _ => SOME raw_name;
haftmann@24218
   225
  in (c, (insts, ((name, t), atts))) end;
haftmann@24218
   226
wenzelm@24707
   227
fun read_def_cmd thy = gen_read_def thy Attrib.intern_src Syntax.parse_prop;
haftmann@24218
   228
fun read_def thy = gen_read_def thy (K I) (K I);
haftmann@24218
   229
haftmann@24589
   230
fun gen_instance prep_arity read_def do_proof raw_arities raw_defs after_qed theory =
haftmann@24218
   231
  let
haftmann@24218
   232
    val arities = map (prep_arity theory) raw_arities;
haftmann@24218
   233
    val _ = if null arities then error "at least one arity must be given" else ();
haftmann@24218
   234
    val _ = case (duplicates (op =) o map #1) arities
haftmann@24218
   235
     of [] => ()
haftmann@24218
   236
      | dupl_tycos => error ("type constructors occur more than once in arities: "
haftmann@24218
   237
          ^ (commas o map quote) dupl_tycos);
haftmann@24836
   238
    fun get_remove_constraint c thy =
haftmann@24836
   239
      let
haftmann@24836
   240
        val ty = Sign.the_const_constraint thy c;
haftmann@24836
   241
      in
haftmann@24836
   242
        thy
haftmann@24836
   243
        |> Sign.add_const_constraint (c, NONE)
haftmann@24836
   244
        |> pair (c, Logic.unvarifyT ty)
haftmann@24836
   245
      end;
haftmann@24218
   246
    fun get_consts_class tyco ty class =
haftmann@24218
   247
      let
haftmann@24218
   248
        val cs = (these o Option.map snd o try (AxClass.params_of_class theory)) class;
haftmann@24218
   249
        val subst_ty = map_type_tfree (K ty);
haftmann@24218
   250
      in
haftmann@24304
   251
        map (fn (c, ty) => (c, ((class, tyco), subst_ty ty))) cs
haftmann@24218
   252
      end;
haftmann@24218
   253
    fun get_consts_sort (tyco, asorts, sort) =
haftmann@24218
   254
      let
haftmann@24589
   255
        val ty = Type (tyco, map (fn (v, sort) => TVar ((v, 0), sort))
wenzelm@24847
   256
          (Name.names Name.context Name.aT asorts))
wenzelm@24731
   257
      in maps (get_consts_class tyco ty) (Sign.complete_sort theory sort) end;
haftmann@24218
   258
    val cs = maps get_consts_sort arities;
haftmann@24218
   259
    fun mk_typnorm thy (ty, ty_sc) =
haftmann@24218
   260
      case try (Sign.typ_match thy (Logic.varifyT ty_sc, ty)) Vartab.empty
haftmann@24218
   261
       of SOME env => SOME (Logic.varifyT #> Envir.typ_subst_TVars env #> Logic.unvarifyT)
haftmann@24218
   262
        | NONE => NONE;
haftmann@24218
   263
    fun read_defs defs cs thy_read =
haftmann@24218
   264
      let
haftmann@24218
   265
        fun read raw_def cs =
haftmann@24218
   266
          let
haftmann@24218
   267
            val (c, (inst, ((name_opt, t), atts))) = read_def thy_read raw_def;
haftmann@24218
   268
            val ty = Consts.instance (Sign.consts_of thy_read) (c, inst);
haftmann@24304
   269
            val ((class, tyco), ty') = case AList.lookup (op =) cs c
haftmann@24218
   270
             of NONE => error ("illegal definition for constant " ^ quote c)
haftmann@24218
   271
              | SOME class_ty => class_ty;
haftmann@24218
   272
            val name = case name_opt
haftmann@24218
   273
             of NONE => Thm.def_name (Logic.name_arity (tyco, [], c))
haftmann@24218
   274
              | SOME name => name;
haftmann@24218
   275
            val t' = case mk_typnorm thy_read (ty', ty)
haftmann@24218
   276
             of NONE => error ("illegal definition for constant " ^
haftmann@24218
   277
                  quote (c ^ "::" ^ setmp show_sorts true
haftmann@24218
   278
                    (Sign.string_of_typ thy_read) ty))
haftmann@24218
   279
              | SOME norm => map_types norm t
haftmann@24218
   280
          in (((class, tyco), ((name, t'), atts)), AList.delete (op =) c cs) end;
haftmann@24218
   281
      in fold_map read defs cs end;
haftmann@24304
   282
    val (defs, other_cs) = read_defs raw_defs cs
haftmann@24218
   283
      (fold Sign.primitive_arity arities (Theory.copy theory));
haftmann@24423
   284
    fun after_qed' cs defs =
wenzelm@24766
   285
      fold Sign.add_const_constraint (map (apsnd SOME) cs)
haftmann@24423
   286
      #> after_qed defs;
haftmann@24218
   287
  in
haftmann@24218
   288
    theory
haftmann@24748
   289
    |> fold_map get_remove_constraint (map fst cs |> distinct (op =))
haftmann@24304
   290
    ||>> fold_map add_def defs
haftmann@24304
   291
    ||> fold (fn (c, ((class, tyco), ty)) => add_inst_def' (class, tyco) (c, ty)) other_cs
haftmann@24423
   292
    |-> (fn (cs, defs) => do_proof (after_qed' cs defs) arities defs)
haftmann@24218
   293
  end;
haftmann@24218
   294
haftmann@24423
   295
fun tactic_proof tac f arities defs =
haftmann@24218
   296
  fold (fn arity => AxClass.prove_arity arity tac) arities
haftmann@24423
   297
  #> f
haftmann@24423
   298
  #> pair defs;
haftmann@24218
   299
haftmann@24218
   300
in
haftmann@24218
   301
haftmann@24589
   302
val instance = gen_instance Sign.cert_arity read_def
haftmann@24589
   303
  (fn f => fn arities => fn defs => instance_arity f arities);
haftmann@24589
   304
val instance_cmd = gen_instance Sign.read_arity read_def_cmd
haftmann@24589
   305
  (fn f => fn arities => fn defs => instance_arity f arities);
haftmann@24589
   306
fun prove_instance tac arities defs =
haftmann@24589
   307
  gen_instance Sign.cert_arity read_def
haftmann@24589
   308
    (tactic_proof tac) arities defs (K I);
haftmann@24218
   309
haftmann@24218
   310
end; (*local*)
haftmann@24218
   311
haftmann@24218
   312
haftmann@24218
   313
haftmann@24589
   314
(** class data **)
haftmann@24218
   315
haftmann@24218
   316
datatype class_data = ClassData of {
haftmann@24218
   317
  locale: string,
haftmann@24218
   318
  consts: (string * string) list
haftmann@24836
   319
    (*locale parameter ~> constant name*),
haftmann@24748
   320
  local_sort: sort,
haftmann@24589
   321
  inst: typ Symtab.table * term Symtab.table
haftmann@24589
   322
    (*canonical interpretation*),
haftmann@24657
   323
  intro: thm,
haftmann@24748
   324
  local_syntax: bool,
haftmann@24657
   325
  defs: thm list,
haftmann@24914
   326
  operations: (string * (term * int) option) list,
haftmann@24836
   327
    (*constant name ~> (locale term, instantiaton index of class typ)*)
haftmann@24914
   328
  constraints: (string * typ) list
haftmann@24657
   329
};
haftmann@24218
   330
haftmann@24657
   331
fun rep_class_data (ClassData d) = d;
haftmann@24914
   332
fun mk_class_data ((locale, consts, local_sort, inst, intro, local_syntax),
haftmann@24914
   333
    (defs, (operations, constraints))) =
haftmann@24914
   334
  ClassData { locale = locale, consts = consts, local_sort = local_sort, inst = inst,
haftmann@24914
   335
    intro = intro, local_syntax = local_syntax, defs = defs, operations = operations,
haftmann@24914
   336
    constraints = constraints };
haftmann@24914
   337
fun map_class_data f (ClassData { locale, consts, local_sort, inst, intro,
haftmann@24914
   338
    local_syntax, defs, operations, constraints }) =
haftmann@24914
   339
  mk_class_data (f ((locale, consts, local_sort, inst, intro, local_syntax),
haftmann@24914
   340
    (defs, (operations, constraints))));
haftmann@24914
   341
fun merge_class_data _ (ClassData { locale = locale, consts = consts,
haftmann@24914
   342
    local_sort = local_sort, inst = inst, intro = intro, local_syntax = local_syntax,
haftmann@24914
   343
    defs = defs1, operations = operations1, constraints = constraints1 },
haftmann@24748
   344
  ClassData { locale = _, consts = _, local_sort = _, inst = _, intro = _, local_syntax = _,
haftmann@24914
   345
    defs = defs2, operations = operations2, constraints = constraints2 }) =
haftmann@24748
   346
  mk_class_data ((locale, consts, local_sort, inst, intro, local_syntax),
haftmann@24914
   347
    (Thm.merge_thms (defs1, defs2),
haftmann@24914
   348
      (AList.merge (op =) (K true) (operations1, operations2),
haftmann@24914
   349
      AList.merge (op =) (K true) (constraints1, constraints2))));
haftmann@24218
   350
haftmann@24218
   351
fun merge_pair f1 f2 ((x1, y1), (x2, y2)) = (f1 (x1, x2), f2 (y1, y2));
haftmann@24218
   352
haftmann@24218
   353
structure ClassData = TheoryDataFun
haftmann@24218
   354
(
haftmann@24657
   355
  type T = class_data Graph.T * class Symtab.table
haftmann@24657
   356
    (*locale name ~> class name*);
haftmann@24218
   357
  val empty = (Graph.empty, Symtab.empty);
haftmann@24218
   358
  val copy = I;
haftmann@24218
   359
  val extend = I;
haftmann@24657
   360
  fun merge _ = merge_pair (Graph.join merge_class_data) (Symtab.merge (K true));
haftmann@24218
   361
);
haftmann@24218
   362
haftmann@24218
   363
haftmann@24218
   364
(* queries *)
haftmann@24218
   365
haftmann@24657
   366
val lookup_class_data = Option.map rep_class_data oo try o Graph.get_node
haftmann@24589
   367
  o fst o ClassData.get;
haftmann@24218
   368
fun class_of_locale thy = Symtab.lookup ((snd o ClassData.get) thy);
haftmann@24218
   369
haftmann@24589
   370
fun the_class_data thy class = case lookup_class_data thy class
haftmann@24589
   371
 of NONE => error ("undeclared class " ^ quote class)
haftmann@24589
   372
  | SOME data => data;
haftmann@24218
   373
haftmann@24914
   374
val locale_of_class = #locale oo the_class_data;
haftmann@24914
   375
haftmann@24218
   376
val ancestry = Graph.all_succs o fst o ClassData.get;
haftmann@24218
   377
haftmann@24435
   378
fun params_of_sort thy =
haftmann@24218
   379
  let
haftmann@24218
   380
    fun params class =
haftmann@24218
   381
      let
haftmann@24218
   382
        val const_typs = (#params o AxClass.get_definition thy) class;
haftmann@24657
   383
        val const_names = (#consts o the_class_data thy) class;
haftmann@24218
   384
      in
haftmann@24218
   385
        (map o apsnd) (fn c => (c, (the o AList.lookup (op =) const_typs) c)) const_names
haftmann@24218
   386
      end;
haftmann@24218
   387
  in maps params o ancestry thy end;
haftmann@24218
   388
haftmann@24657
   389
fun these_defs thy = maps (these o Option.map #defs o lookup_class_data thy) o ancestry thy;
haftmann@24218
   390
haftmann@24218
   391
fun these_intros thy =
haftmann@24657
   392
  Graph.fold (fn (_, (data, _)) => insert Thm.eq_thm ((#intro o rep_class_data) data))
haftmann@24218
   393
    ((fst o ClassData.get) thy) [];
haftmann@24218
   394
haftmann@24836
   395
fun these_operations thy =
haftmann@24836
   396
  maps (#operations o the_class_data thy) o ancestry thy;
haftmann@24657
   397
haftmann@24914
   398
fun these_constraints thy =
haftmann@24914
   399
  maps (#constraints o the_class_data thy) o ancestry thy;
haftmann@24914
   400
haftmann@24836
   401
fun local_operation thy = Option.join oo AList.lookup (op =) o these_operations thy;
haftmann@24657
   402
haftmann@24901
   403
fun sups_local_sort thy sort =
haftmann@24901
   404
  let
haftmann@24901
   405
    val sups = filter (is_some o lookup_class_data thy) sort
haftmann@24901
   406
      |> Sign.minimize_sort thy;
haftmann@24901
   407
    val local_sort = case sups
haftmann@24901
   408
     of sup :: _ => #local_sort (the_class_data thy sup)
haftmann@24901
   409
      | [] => sort;
haftmann@24901
   410
  in (sups, local_sort) end;
haftmann@24901
   411
haftmann@24836
   412
fun local_syntax thy = #local_syntax o the_class_data thy;
haftmann@24748
   413
haftmann@24218
   414
fun print_classes thy =
haftmann@24218
   415
  let
haftmann@24218
   416
    val algebra = Sign.classes_of thy;
haftmann@24218
   417
    val arities =
haftmann@24218
   418
      Symtab.empty
haftmann@24218
   419
      |> Symtab.fold (fn (tyco, arities) => fold (fn (class, _) =>
haftmann@24218
   420
           Symtab.map_default (class, []) (insert (op =) tyco)) arities)
haftmann@24218
   421
             ((#arities o Sorts.rep_algebra) algebra);
haftmann@24218
   422
    val the_arities = these o Symtab.lookup arities;
haftmann@24218
   423
    fun mk_arity class tyco =
haftmann@24218
   424
      let
haftmann@24218
   425
        val Ss = Sorts.mg_domain algebra tyco [class];
haftmann@24218
   426
      in Sign.pretty_arity thy (tyco, Ss, [class]) end;
haftmann@24218
   427
    fun mk_param (c, ty) = Pretty.str (Sign.extern_const thy c ^ " :: "
haftmann@24218
   428
      ^ setmp show_sorts false (Sign.string_of_typ thy o Type.strip_sorts) ty);
haftmann@24218
   429
    fun mk_entry class = (Pretty.block o Pretty.fbreaks o map_filter I) [
haftmann@24218
   430
      (SOME o Pretty.str) ("class " ^ class ^ ":"),
haftmann@24218
   431
      (SOME o Pretty.block) [Pretty.str "supersort: ",
wenzelm@24731
   432
        (Sign.pretty_sort thy o Sign.minimize_sort thy o Sign.super_classes thy) class],
haftmann@24657
   433
      Option.map (Pretty.str o prefix "locale: " o #locale) (lookup_class_data thy class),
haftmann@24218
   434
      ((fn [] => NONE | ps => (SOME o Pretty.block o Pretty.fbreaks) (Pretty.str "parameters:" :: ps)) o map mk_param
haftmann@24218
   435
        o these o Option.map #params o try (AxClass.get_definition thy)) class,
haftmann@24218
   436
      (SOME o Pretty.block o Pretty.breaks) [
haftmann@24218
   437
        Pretty.str "instances:",
haftmann@24218
   438
        Pretty.list "" "" (map (mk_arity class) (the_arities class))
haftmann@24218
   439
      ]
haftmann@24218
   440
    ]
haftmann@24218
   441
  in
haftmann@24589
   442
    (Pretty.writeln o Pretty.chunks o separate (Pretty.str "")
haftmann@24589
   443
      o map mk_entry o Sorts.all_classes) algebra
haftmann@24218
   444
  end;
haftmann@24218
   445
haftmann@24218
   446
haftmann@24218
   447
(* updaters *)
haftmann@24218
   448
haftmann@24748
   449
fun add_class_data ((class, superclasses), (locale, consts, local_sort, inst, intro, local_syntax)) =
haftmann@24218
   450
  ClassData.map (fn (gr, tab) => (
haftmann@24218
   451
    gr
haftmann@24914
   452
    |> Graph.new_node (class, mk_class_data ((locale, (map o pairself) fst consts,
haftmann@24914
   453
        local_sort, inst, intro, local_syntax),
haftmann@24914
   454
          ([], (map (apfst fst o apsnd (SOME o rpair 0 o Free) o swap) consts, map snd consts))))
haftmann@24218
   455
    |> fold (curry Graph.add_edge class) superclasses,
haftmann@24218
   456
    tab
haftmann@24218
   457
    |> Symtab.update (locale, class)
haftmann@24218
   458
  ));
haftmann@24218
   459
haftmann@24901
   460
fun register_const class (entry, def) =
haftmann@24657
   461
  (ClassData.map o apfst o Graph.map_node class o map_class_data o apsnd)
haftmann@24914
   462
    (fn (defs, (operations, constraints)) =>
haftmann@24914
   463
      (def :: defs, (apsnd (SOME o snd) entry :: operations, apsnd fst entry :: constraints)));
haftmann@24836
   464
haftmann@24914
   465
fun register_abbrev class (abbrev, ty) =
haftmann@24836
   466
  (ClassData.map o apfst o Graph.map_node class o map_class_data o apsnd o apsnd)
haftmann@24914
   467
    (fn (operations, constraints) => ((abbrev, NONE) :: operations, (abbrev, ty) :: constraints));
haftmann@24218
   468
haftmann@24589
   469
haftmann@24589
   470
(** rule calculation, tactics and methods **)
haftmann@24589
   471
haftmann@24589
   472
fun class_intro thy locale class sups =
haftmann@24589
   473
  let
haftmann@24589
   474
    fun class_elim class =
haftmann@24589
   475
      case (map Drule.unconstrainTs o #axioms o AxClass.get_definition thy) class
haftmann@24589
   476
       of [thm] => SOME thm
haftmann@24589
   477
        | [] => NONE;
haftmann@24589
   478
    val pred_intro = case Locale.intros thy locale
haftmann@24589
   479
     of ([ax_intro], [intro]) => intro |> OF_LAST ax_intro |> SOME
haftmann@24589
   480
      | ([intro], []) => SOME intro
haftmann@24589
   481
      | ([], [intro]) => SOME intro
haftmann@24589
   482
      | _ => NONE;
haftmann@24589
   483
    val pred_intro' = pred_intro
haftmann@24589
   484
      |> Option.map (fn intro => intro OF map_filter class_elim sups);
haftmann@24589
   485
    val class_intro = (#intro o AxClass.get_definition thy) class;
haftmann@24589
   486
    val raw_intro = case pred_intro'
haftmann@24589
   487
     of SOME pred_intro => class_intro |> OF_LAST pred_intro
haftmann@24589
   488
      | NONE => class_intro;
haftmann@24589
   489
    val sort = Sign.super_classes thy class;
wenzelm@24847
   490
    val typ = TVar ((Name.aT, 0), sort);
haftmann@24589
   491
    val defs = these_defs thy sups;
haftmann@24589
   492
  in
haftmann@24589
   493
    raw_intro
haftmann@24589
   494
    |> Drule.instantiate' [SOME (Thm.ctyp_of thy typ)] []
haftmann@24589
   495
    |> strip_all_ofclass thy sort
haftmann@24589
   496
    |> Thm.strip_shyps
haftmann@24589
   497
    |> MetaSimplifier.rewrite_rule defs
haftmann@24589
   498
    |> Drule.unconstrainTs
haftmann@24589
   499
  end;
haftmann@24589
   500
haftmann@24589
   501
fun class_interpretation class facts defs thy =
haftmann@24589
   502
  let
haftmann@24657
   503
    val { locale, inst, ... } = the_class_data thy class;
haftmann@24589
   504
    val tac = (ALLGOALS o ProofContext.fact_tac) facts;
haftmann@24589
   505
    val prfx = Logic.const_of_class (NameSpace.base class);
haftmann@24589
   506
  in
haftmann@24589
   507
    prove_interpretation tac ((false, prfx), []) (Locale.Locale locale)
haftmann@24589
   508
      (inst, defs) thy
haftmann@24589
   509
  end;
haftmann@24589
   510
haftmann@24589
   511
fun interpretation_in_rule thy (class1, class2) =
haftmann@24589
   512
  let
haftmann@24589
   513
    fun mk_axioms class =
haftmann@24589
   514
      let
haftmann@24657
   515
        val { locale, inst = (_, insttab), ... } = the_class_data thy class;
haftmann@24589
   516
      in
haftmann@24589
   517
        Locale.global_asms_of thy locale
haftmann@24589
   518
        |> maps snd
haftmann@24589
   519
        |> (map o map_aterms) (fn Free (s, _) => (the o Symtab.lookup insttab) s | t => t)
wenzelm@24847
   520
        |> (map o map_types o map_atyps) (fn TFree _ => TFree (Name.aT, [class1]) | T => T)
haftmann@24589
   521
        |> map (ObjectLogic.ensure_propT thy)
haftmann@24589
   522
      end;
haftmann@24589
   523
    val (prems, concls) = pairself mk_axioms (class1, class2);
haftmann@24589
   524
  in
haftmann@24589
   525
    Goal.prove_global thy [] prems (Logic.mk_conjunction_list concls)
haftmann@24589
   526
      (Locale.intro_locales_tac true (ProofContext.init thy))
haftmann@24589
   527
  end;
haftmann@24218
   528
haftmann@24218
   529
fun intro_classes_tac facts st =
haftmann@24218
   530
  let
haftmann@24218
   531
    val thy = Thm.theory_of_thm st;
haftmann@24218
   532
    val classes = Sign.all_classes thy;
haftmann@24218
   533
    val class_trivs = map (Thm.class_triv thy) classes;
haftmann@24218
   534
    val class_intros = these_intros thy;
haftmann@24218
   535
    fun add_axclass_intro class =
haftmann@24218
   536
      case try (AxClass.get_definition thy) class of SOME {intro, ...} => cons intro | _ => I;
haftmann@24218
   537
    val axclass_intros = fold add_axclass_intro classes [];
haftmann@24218
   538
  in
haftmann@24218
   539
    st
haftmann@24218
   540
    |> ((ALLGOALS (Method.insert_tac facts THEN'
haftmann@24218
   541
          REPEAT_ALL_NEW (resolve_tac (class_trivs @ class_intros @ axclass_intros))))
haftmann@24218
   542
            THEN Tactic.distinct_subgoals_tac)
haftmann@24218
   543
  end;
haftmann@24218
   544
haftmann@24218
   545
fun default_intro_classes_tac [] = intro_classes_tac []
haftmann@24218
   546
  | default_intro_classes_tac _ = Tactical.no_tac;    (*no error message!*)
haftmann@24218
   547
haftmann@24218
   548
fun default_tac rules ctxt facts =
haftmann@24218
   549
  HEADGOAL (Method.some_rule_tac rules ctxt facts) ORELSE
haftmann@24218
   550
    default_intro_classes_tac facts;
haftmann@24218
   551
haftmann@24218
   552
val _ = Context.add_setup (Method.add_methods
haftmann@24218
   553
 [("intro_classes", Method.no_args (Method.METHOD intro_classes_tac),
haftmann@24218
   554
    "back-chain introduction rules of classes"),
haftmann@24218
   555
  ("default", Method.thms_ctxt_args (Method.METHOD oo default_tac),
haftmann@24218
   556
    "apply some intro/elim rule")]);
haftmann@24218
   557
haftmann@24218
   558
haftmann@24589
   559
(** classes and class target **)
haftmann@24218
   560
haftmann@24901
   561
(* class context initialization *)
haftmann@24748
   562
haftmann@24914
   563
fun remove_constraint local_sort c ctxt =
haftmann@24914
   564
  let
haftmann@24914
   565
    val ty = ProofContext.the_const_constraint ctxt c;
haftmann@24914
   566
    val ty' = map_atyps (fn ty as TFree (v, _) => if v = Name.aT
haftmann@24914
   567
      then TFree (v, local_sort) else ty | ty => ty) ty;
haftmann@24914
   568
  in
haftmann@24914
   569
    ctxt
haftmann@24914
   570
    |> ProofContext.add_const_constraint (c, SOME ty')
haftmann@24914
   571
  end;
haftmann@24914
   572
haftmann@24914
   573
fun sort_term_check sups local_sort ts ctxt =
haftmann@24748
   574
  let
haftmann@24901
   575
    val thy = ProofContext.theory_of ctxt;
haftmann@24901
   576
    val local_operation = local_operation thy sups;
haftmann@24914
   577
    val constraints = these_constraints thy sups;
haftmann@24901
   578
    val consts = ProofContext.consts_of ctxt;
haftmann@24901
   579
    fun check_typ (c, ty) (t', idx) = case nth (Consts.typargs consts (c, ty)) idx
haftmann@24901
   580
     of TFree (v, _) => if v = Name.aT
haftmann@24901
   581
          then apfst (AList.update (op =) ((c, ty), t')) else I
haftmann@24901
   582
      | TVar (vi, _) => if TypeInfer.is_param vi
haftmann@24901
   583
          then apfst (AList.update (op =) ((c, ty), t'))
haftmann@24901
   584
            #> apsnd (insert (op =) vi) else I
haftmann@24901
   585
      | _ => I;
haftmann@24901
   586
    fun add_const (Const (c, ty)) = (case local_operation c
haftmann@24901
   587
         of SOME (t', idx) => check_typ (c, ty) (t', idx)
haftmann@24901
   588
          | NONE => I)
haftmann@24901
   589
      | add_const _ = I;
haftmann@24901
   590
    val (cs, typarams) = (fold o fold_aterms) add_const ts ([], []);
haftmann@24901
   591
    val ts' = map (map_aterms
haftmann@24901
   592
        (fn t as Const (c, ty) => the_default t (AList.lookup (op =) cs (c, ty)) | t => t)
haftmann@24901
   593
      #> (map_types o map_type_tvar) (fn var as (vi, _) => if member (op =) typarams vi
haftmann@24901
   594
           then TFree (Name.aT, local_sort) else TVar var)) ts;
haftmann@24901
   595
    val ctxt' = fold (ProofContext.add_const_constraint o apsnd SOME) constraints ctxt;
haftmann@24901
   596
  in (ts', ctxt') end;
haftmann@24748
   597
haftmann@24914
   598
fun init_class_ctxt sups local_sort ctxt =
haftmann@24914
   599
  let
haftmann@24914
   600
    val operations = these_operations (ProofContext.theory_of ctxt) sups;
haftmann@24914
   601
  in
haftmann@24914
   602
    ctxt
haftmann@24914
   603
    |> Variable.declare_term
haftmann@24914
   604
        (Logic.mk_type (TFree (Name.aT, local_sort)))
haftmann@24914
   605
    |> fold (remove_constraint local_sort o fst) operations
haftmann@24914
   606
    |> Context.proof_map (Syntax.add_term_check 50 "class"
haftmann@24914
   607
        (sort_term_check sups local_sort))
haftmann@24914
   608
  end;
haftmann@24901
   609
haftmann@24901
   610
fun init class ctxt =
haftmann@24914
   611
  init_class_ctxt [class]
haftmann@24914
   612
    ((#local_sort o the_class_data (ProofContext.theory_of ctxt)) class) ctxt;
haftmann@24914
   613
haftmann@24748
   614
haftmann@24589
   615
(* class definition *)
haftmann@24218
   616
haftmann@24218
   617
local
haftmann@24218
   618
wenzelm@24707
   619
fun read_param thy raw_t =  (* FIXME ProofContext.read_const (!?) *)
haftmann@24218
   620
  let
wenzelm@24707
   621
    val t = Syntax.read_term_global thy raw_t
haftmann@24218
   622
  in case try dest_Const t
haftmann@24218
   623
   of SOME (c, _) => c
haftmann@24218
   624
    | NONE => error ("Not a constant: " ^ Sign.string_of_term thy t)
haftmann@24218
   625
  end;
haftmann@24218
   626
haftmann@24748
   627
fun gen_class_spec prep_class prep_expr process_expr thy raw_supclasses raw_includes_elems =
haftmann@24218
   628
  let
haftmann@24748
   629
    val supclasses = map (prep_class thy) raw_supclasses;
haftmann@24901
   630
    val (sups, local_sort) = sups_local_sort thy supclasses;
haftmann@24748
   631
    val supsort = Sign.minimize_sort thy supclasses;
haftmann@24748
   632
    val suplocales = map (Locale.Locale o #locale o the_class_data thy) sups;
haftmann@24748
   633
    val (raw_elems, includes) = fold_rev (fn Locale.Elem e => apfst (cons e)
haftmann@24748
   634
      | Locale.Expr i => apsnd (cons (prep_expr thy i))) raw_includes_elems ([], []);
haftmann@24748
   635
    val supexpr = Locale.Merge suplocales;
haftmann@24748
   636
    val supparams = (map fst o Locale.parameters_of_expr thy) supexpr;
haftmann@24748
   637
    val supconsts = AList.make (the o AList.lookup (op =) (params_of_sort thy sups))
haftmann@24748
   638
      (map fst supparams);
haftmann@24748
   639
    val mergeexpr = Locale.Merge (suplocales @ includes);
haftmann@24748
   640
    val constrain = Element.Constrains ((map o apsnd o map_atyps)
wenzelm@24847
   641
      (fn TFree (_, sort) => TFree (Name.aT, sort)) supparams);
haftmann@24748
   642
  in
haftmann@24748
   643
    ProofContext.init thy
haftmann@24748
   644
    |> Locale.cert_expr supexpr [constrain]
haftmann@24748
   645
    |> snd
haftmann@24914
   646
    |> init_class_ctxt sups local_sort
haftmann@24748
   647
    |> process_expr Locale.empty raw_elems
haftmann@24748
   648
    |> fst
haftmann@24836
   649
    |> (fn elems => ((((sups, supconsts), (supsort, local_sort, mergeexpr)),
haftmann@24748
   650
          (*FIXME*) if null includes then constrain :: elems else elems)))
haftmann@24748
   651
  end;
haftmann@24748
   652
haftmann@24748
   653
val read_class_spec = gen_class_spec Sign.intern_class Locale.intern_expr Locale.read_expr;
haftmann@24748
   654
val check_class_spec = gen_class_spec (K I) (K I) Locale.cert_expr;
haftmann@24748
   655
haftmann@24748
   656
fun gen_class prep_spec prep_param local_syntax bname
haftmann@24748
   657
    raw_supclasses raw_includes_elems raw_other_consts thy =
haftmann@24748
   658
  let
haftmann@24836
   659
    val (((sups, supconsts), (supsort, local_sort, mergeexpr)), elems) =
haftmann@24748
   660
      prep_spec thy raw_supclasses raw_includes_elems;
haftmann@24748
   661
    val other_consts = map (prep_param thy) raw_other_consts;
haftmann@24589
   662
    fun mk_instT class = Symtab.empty
wenzelm@24847
   663
      |> Symtab.update (Name.aT, TFree (Name.aT, [class]));
haftmann@24589
   664
    fun mk_inst class param_names cs =
haftmann@24589
   665
      Symtab.empty
haftmann@24589
   666
      |> fold2 (fn v => fn (c, ty) => Symtab.update (v, Const
haftmann@24589
   667
           (c, Term.map_type_tfree (fn (v, _) => TFree (v, [class])) ty))) param_names cs;
haftmann@24218
   668
    fun extract_params thy name_locale =
haftmann@24218
   669
      let
haftmann@24218
   670
        val params = Locale.parameters_of thy name_locale;
haftmann@24748
   671
        val _ = if Sign.subsort thy (supsort, local_sort) then () else error
haftmann@24748
   672
          ("Sort " ^ Sign.string_of_sort thy local_sort
haftmann@24748
   673
            ^ " is less general than permitted least general sort "
haftmann@24748
   674
            ^ Sign.string_of_sort thy supsort);
haftmann@24218
   675
      in
haftmann@24901
   676
        (map fst params, params
wenzelm@24847
   677
        |> (map o apfst o apsnd o Term.map_type_tfree) (K (TFree (Name.aT, local_sort)))
haftmann@24901
   678
        |> (map o apsnd) (fork_mixfix true (SOME "") #> fst)
haftmann@24218
   679
        |> chop (length supconsts)
haftmann@24901
   680
        |> snd)
haftmann@24218
   681
      end;
haftmann@24218
   682
    fun extract_assumes name_locale params thy cs =
haftmann@24218
   683
      let
haftmann@24218
   684
        val consts = supconsts @ (map (fst o fst) params ~~ cs);
haftmann@24218
   685
        fun subst (Free (c, ty)) =
haftmann@24218
   686
              Const ((fst o the o AList.lookup (op =) consts) c, ty)
haftmann@24218
   687
          | subst t = t;
haftmann@24218
   688
        fun prep_asm ((name, atts), ts) =
haftmann@24218
   689
          ((NameSpace.base name, map (Attrib.attribute thy) atts),
haftmann@24589
   690
            (map o map_aterms) subst ts);
haftmann@24218
   691
      in
haftmann@24218
   692
        Locale.global_asms_of thy name_locale
haftmann@24218
   693
        |> map prep_asm
haftmann@24218
   694
      end;
haftmann@24218
   695
    fun note_intro name_axclass class_intro =
haftmann@24218
   696
      PureThy.note_thmss_qualified "" ((Logic.const_of_class o NameSpace.base) name_axclass)
haftmann@24218
   697
        [(("intro", []), [([class_intro], [])])]
haftmann@24218
   698
      #> snd
haftmann@24218
   699
  in
haftmann@24218
   700
    thy
haftmann@24748
   701
    |> Locale.add_locale_i (SOME "") bname mergeexpr elems
haftmann@24218
   702
    |-> (fn name_locale => ProofContext.theory_result (
haftmann@24218
   703
      `(fn thy => extract_params thy name_locale)
haftmann@24901
   704
      #-> (fn (globals, params) =>
haftmann@24589
   705
        AxClass.define_class_params (bname, supsort) params
haftmann@24589
   706
          (extract_assumes name_locale params) other_consts
haftmann@24589
   707
      #-> (fn (name_axclass, (consts, axioms)) =>
haftmann@24218
   708
        `(fn thy => class_intro thy name_locale name_axclass sups)
haftmann@24218
   709
      #-> (fn class_intro =>
haftmann@24218
   710
        add_class_data ((name_axclass, sups),
haftmann@24914
   711
          (name_locale, map fst params ~~ consts, local_sort,
haftmann@24657
   712
            (mk_instT name_axclass, mk_inst name_axclass (map fst globals)
haftmann@24748
   713
              (map snd supconsts @ consts)), class_intro, local_syntax))
haftmann@24218
   714
      #> note_intro name_axclass class_intro
haftmann@24589
   715
      #> class_interpretation name_axclass axioms []
haftmann@24218
   716
      #> pair name_axclass
haftmann@24218
   717
      )))))
haftmann@24218
   718
  end;
haftmann@24218
   719
haftmann@24218
   720
in
haftmann@24218
   721
haftmann@24748
   722
val class_cmd = gen_class read_class_spec read_param;
haftmann@24748
   723
val class = gen_class check_class_spec (K I);
haftmann@24218
   724
haftmann@24218
   725
end; (*local*)
haftmann@24218
   726
haftmann@24218
   727
haftmann@24589
   728
(* definition in class target *)
haftmann@24218
   729
haftmann@24218
   730
fun export_fixes thy class =
haftmann@24218
   731
  let
haftmann@24435
   732
    val consts = params_of_sort thy [class];
haftmann@24218
   733
    fun subst_aterm (t as Free (v, ty)) = (case AList.lookup (op =) consts v
haftmann@24218
   734
         of SOME (c, _) => Const (c, ty)
haftmann@24218
   735
          | NONE => t)
haftmann@24218
   736
      | subst_aterm t = t;
haftmann@24657
   737
  in Term.map_aterms subst_aterm end;
haftmann@24218
   738
haftmann@24901
   739
fun mk_operation_entry thy (c, rhs) =
haftmann@24901
   740
  let
haftmann@24914
   741
    val ty = fastype_of rhs;
haftmann@24914
   742
    val typargs = Sign.const_typargs thy (c, ty);
haftmann@24901
   743
    val typidx = find_index (fn TFree (w, _) => Name.aT = w | _ => false) typargs;
haftmann@24914
   744
  in (c, (ty, (rhs, typidx))) end;
haftmann@24901
   745
haftmann@24218
   746
fun add_const_in_class class ((c, rhs), syn) thy =
haftmann@24218
   747
  let
haftmann@24218
   748
    val prfx = (Logic.const_of_class o NameSpace.base) class;
haftmann@24657
   749
    fun mk_name c =
haftmann@24218
   750
      let
haftmann@24218
   751
        val n1 = Sign.full_name thy c;
haftmann@24218
   752
        val n2 = NameSpace.qualifier n1;
haftmann@24218
   753
        val n3 = NameSpace.base n1;
haftmann@24657
   754
      in NameSpace.implode [n2, prfx, n3] end;
haftmann@24657
   755
    val constrain_sort = curry (Sorts.inter_sort (Sign.classes_of thy)) [class];
haftmann@24748
   756
    val rhs' = export_fixes thy class rhs;
haftmann@24657
   757
    val subst_typ = Term.map_type_tfree (fn var as (w, sort) =>
wenzelm@24847
   758
      if w = Name.aT then TFree (w, constrain_sort sort) else TFree var);
haftmann@24218
   759
    val ty' = Term.fastype_of rhs';
haftmann@24657
   760
    val ty'' = subst_typ ty';
haftmann@24657
   761
    val c' = mk_name c;
haftmann@24657
   762
    val def = (c, Logic.mk_equals (Const (c', ty'), rhs'));
haftmann@24901
   763
    val (syn', _) = fork_mixfix true (SOME class) syn;
haftmann@24657
   764
    fun interpret def thy =
haftmann@24218
   765
      let
haftmann@24589
   766
        val def' = symmetric def;
haftmann@24218
   767
        val def_eq = Thm.prop_of def';
haftmann@24901
   768
        val entry = mk_operation_entry thy (c', rhs);
haftmann@24218
   769
      in
haftmann@24657
   770
        thy
haftmann@24657
   771
        |> class_interpretation class [def'] [def_eq]
haftmann@24901
   772
        |> register_const class (entry, def')
haftmann@24218
   773
      end;
haftmann@24218
   774
  in
haftmann@24218
   775
    thy
haftmann@24218
   776
    |> Sign.add_path prfx
wenzelm@24770
   777
    |> Sign.add_consts_authentic [] [(c, ty', syn')]
haftmann@24218
   778
    |> Sign.parent_path
haftmann@24218
   779
    |> Sign.sticky_prefix prfx
haftmann@24218
   780
    |> PureThy.add_defs_i false [(def, [])]
haftmann@24218
   781
    |-> (fn [def] => interpret def)
wenzelm@24766
   782
    |> Sign.add_const_constraint (c', SOME ty'')
haftmann@24218
   783
    |> Sign.restore_naming thy
haftmann@24914
   784
    |> pair c'
haftmann@24218
   785
  end;
haftmann@24218
   786
haftmann@24901
   787
haftmann@24901
   788
(* abbreviation in class target *)
haftmann@24901
   789
haftmann@24901
   790
fun add_abbrev_in_class class prmode ((c, rhs), syn) thy =
haftmann@24836
   791
  let
haftmann@24901
   792
    val prfx = (Logic.const_of_class o NameSpace.base) class;
haftmann@24901
   793
    fun mk_name c =
haftmann@24901
   794
      let
haftmann@24901
   795
        val n1 = Sign.full_name thy c;
haftmann@24901
   796
        val n2 = NameSpace.qualifier n1;
haftmann@24901
   797
        val n3 = NameSpace.base n1;
haftmann@24901
   798
      in NameSpace.implode [n2, prfx, prfx, n3] end;
haftmann@24901
   799
    val c' = mk_name c;
haftmann@24901
   800
    val rhs' = export_fixes thy class rhs;
haftmann@24901
   801
    val ty' = fastype_of rhs';
haftmann@24836
   802
  in
haftmann@24836
   803
    thy
haftmann@24901
   804
    |> Sign.add_notation prmode [(Const (c', ty'), syn)]
haftmann@24914
   805
    |> register_abbrev class (c', ty')
haftmann@24914
   806
    |> pair c'
haftmann@24836
   807
  end;
haftmann@24836
   808
haftmann@24218
   809
end;