src/Pure/Isar/named_target.ML
author wenzelm
Sun Oct 30 22:35:18 2011 +0100 (2011-10-30 ago)
changeset 45311 ca9e66c523a2
parent 45310 adaf2184b79d
child 45352 0b4038361a3a
permissions -rw-r--r--
even more uniform Local_Theory.declaration for locales (cf. 57def0b39696, aa35859c8741);
     1 (*  Title:      Pure/Isar/named_target.ML
     2     Author:     Makarius
     3     Author:     Florian Haftmann, TU Muenchen
     4 
     5 Targets for theory, locale and class.
     6 *)
     7 
     8 signature NAMED_TARGET =
     9 sig
    10   val init: (local_theory -> local_theory) -> string -> theory -> local_theory
    11   val theory_init: theory -> local_theory
    12   val reinit: local_theory -> local_theory -> local_theory
    13   val context_cmd: xstring -> theory -> local_theory
    14   val peek: local_theory -> {target: string, is_locale: bool, is_class: bool} option
    15 end;
    16 
    17 structure Named_Target: NAMED_TARGET =
    18 struct
    19 
    20 (* context data *)
    21 
    22 datatype target =
    23   Target of {target: string, is_locale: bool, is_class: bool,
    24     before_exit: local_theory -> local_theory};
    25 
    26 fun make_target target is_locale is_class before_exit =
    27   Target {target = target, is_locale = is_locale, is_class = is_class,
    28     before_exit = before_exit};
    29 
    30 fun named_target _ "" before_exit = make_target "" false false before_exit
    31   | named_target thy locale before_exit =
    32       if Locale.defined thy locale
    33       then make_target locale true (Class.is_class thy locale) before_exit
    34       else error ("No such locale: " ^ quote locale);
    35 
    36 structure Data = Proof_Data
    37 (
    38   type T = target option;
    39   fun init _ = NONE;
    40 );
    41 
    42 val peek =
    43   Data.get #> Option.map (fn Target {target, is_locale, is_class, ...} =>
    44     {target = target, is_locale = is_locale, is_class = is_class});
    45 
    46 
    47 (* generic declarations *)
    48 
    49 fun locale_declaration locale syntax decl lthy =
    50   let
    51     val add = if syntax then Locale.add_syntax_declaration else Locale.add_declaration;
    52     val locale_decl = Morphism.transform (Local_Theory.target_morphism lthy) decl;
    53   in
    54     lthy
    55     |> Local_Theory.target (add locale locale_decl)
    56   end;
    57 
    58 fun target_declaration (Target {target, ...}) {syntax, pervasive} decl =
    59   if target = "" then Generic_Target.theory_declaration decl
    60   else
    61     locale_declaration target syntax decl
    62     #> pervasive ? Generic_Target.theory_declaration decl
    63     #> not pervasive ? Context.proof_map (Morphism.form decl);
    64 
    65 
    66 (* consts in locales *)
    67 
    68 fun locale_const (Target {target, is_class, ...}) (prmode as (mode, _)) ((b, mx), rhs) phi =
    69   let
    70     val b' = Morphism.binding phi b;
    71     val rhs' = Morphism.term phi rhs;
    72     val arg = (b', Term.close_schematic_term rhs');
    73     val same_shape = Term.aconv_untyped (rhs, rhs');
    74     (* FIXME workaround based on educated guess *)
    75     val prefix' = Binding.prefix_of b';
    76     val is_canonical_class = is_class andalso
    77       (Binding.eq_name (b, b')
    78         andalso not (null prefix')
    79         andalso List.last prefix' = (Class.class_prefix target, false)
    80       orelse same_shape);
    81   in
    82     not is_canonical_class ?
    83       (Context.mapping_result
    84         (Sign.add_abbrev Print_Mode.internal arg)
    85         (Proof_Context.add_abbrev Print_Mode.internal arg)
    86       #-> (fn (lhs' as Const (d, _), _) =>
    87           same_shape ?
    88             (Context.mapping
    89               (Sign.revert_abbrev mode d) (Proof_Context.revert_abbrev mode d) #>
    90              Morphism.form (Proof_Context.target_notation true prmode [(lhs', mx)]))))
    91   end;
    92 
    93 fun locale_const_declaration (ta as Target {target, ...}) prmode arg =
    94   locale_declaration target true (locale_const ta prmode arg);
    95 
    96 
    97 (* define *)
    98 
    99 fun locale_foundation ta (((b, U), mx), (b_def, rhs)) (type_params, term_params) =
   100   Generic_Target.theory_foundation (((b, U), NoSyn), (b_def, rhs)) (type_params, term_params)
   101   #-> (fn (lhs, def) => locale_const_declaration ta Syntax.mode_default ((b, mx), lhs)
   102     #> pair (lhs, def))
   103 
   104 fun class_foundation (ta as Target {target, ...})
   105     (((b, U), mx), (b_def, rhs)) (type_params, term_params) =
   106   Generic_Target.theory_foundation (((b, U), NoSyn), (b_def, rhs)) (type_params, term_params)
   107   #-> (fn (lhs, def) => locale_const_declaration ta Syntax.mode_default ((b, NoSyn), lhs)
   108     #> Class.const target ((b, mx), (type_params, lhs))
   109     #> pair (lhs, def))
   110 
   111 fun target_foundation (ta as Target {is_locale, is_class, ...}) =
   112   if is_class then class_foundation ta
   113   else if is_locale then locale_foundation ta
   114   else Generic_Target.theory_foundation;
   115 
   116 
   117 (* notes *)
   118 
   119 fun locale_notes locale kind global_facts local_facts lthy =
   120   let
   121     val global_facts' = Attrib.map_facts (K I) global_facts;
   122     val local_facts' = Element.facts_map
   123       (Element.transform_ctxt (Local_Theory.target_morphism lthy)) local_facts;
   124   in
   125     lthy
   126     |> Local_Theory.background_theory (Global_Theory.note_thmss kind global_facts' #> snd)
   127     |> Local_Theory.target (Locale.add_thmss locale kind local_facts')
   128   end
   129 
   130 fun target_notes (Target {target, is_locale, ...}) =
   131   if is_locale then locale_notes target
   132   else fn kind => fn global_facts => fn _ => Generic_Target.theory_notes kind global_facts;
   133 
   134 
   135 (* abbrev *)
   136 
   137 fun locale_abbrev ta prmode ((b, mx), t) xs =
   138   Local_Theory.background_theory_result
   139     (Sign.add_abbrev Print_Mode.internal (b, t)) #->
   140       (fn (lhs, _) => locale_const_declaration ta prmode
   141         ((b, mx), Term.list_comb (Logic.unvarify_global lhs, xs)));
   142 
   143 fun target_abbrev (ta as Target {target, is_locale, is_class, ...})
   144     prmode (b, mx) (global_rhs, t') xs lthy =
   145   if is_locale then
   146     lthy
   147     |> locale_abbrev ta prmode ((b, if is_class then NoSyn else mx), global_rhs) xs
   148     |> is_class ? Class.abbrev target prmode ((b, mx), t')
   149   else
   150     lthy
   151     |> Generic_Target.theory_abbrev prmode ((b, mx), global_rhs);
   152 
   153 
   154 (* pretty *)
   155 
   156 fun pretty (Target {target, is_locale, is_class, ...}) ctxt =
   157   let
   158     val thy = Proof_Context.theory_of ctxt;
   159     val target_name =
   160       [Pretty.command (if is_class then "class" else "locale"),
   161         Pretty.str (" " ^ Locale.extern thy target)];
   162     val fixes = map (fn (x, T) => (Binding.name x, SOME T, NoSyn))
   163       (#1 (Proof_Context.inferred_fixes ctxt));
   164     val assumes = map (fn A => (Attrib.empty_binding, [(Thm.term_of A, [])]))
   165       (Assumption.all_assms_of ctxt);
   166     val elems =
   167       (if null fixes then [] else [Element.Fixes fixes]) @
   168       (if null assumes then [] else [Element.Assumes assumes]);
   169     val body_elems =
   170       if not is_locale then []
   171       else if null elems then [Pretty.block target_name]
   172       else [Pretty.block (Pretty.fbreaks (Pretty.block (target_name @ [Pretty.str " ="]) ::
   173         map (Pretty.chunks o Element.pretty_ctxt ctxt) elems))];
   174   in
   175     Pretty.block [Pretty.command "theory", Pretty.brk 1,
   176       Pretty.str (Context.theory_name (Proof_Context.theory_of ctxt))] :: body_elems
   177   end;
   178 
   179 
   180 (* init *)
   181 
   182 fun init_context (Target {target, is_locale, is_class, ...}) =
   183   if not is_locale then Proof_Context.init_global
   184   else if not is_class then Locale.init target
   185   else Class.init target;
   186 
   187 fun init before_exit target thy =
   188   let
   189     val ta = named_target thy target before_exit;
   190   in
   191     thy
   192     |> init_context ta
   193     |> Data.put (SOME ta)
   194     |> Local_Theory.init NONE (Long_Name.base_name target)
   195        {define = Generic_Target.define (target_foundation ta),
   196         notes = Generic_Target.notes (target_notes ta),
   197         abbrev = Generic_Target.abbrev (target_abbrev ta),
   198         declaration = target_declaration ta,
   199         pretty = pretty ta,
   200         exit = Local_Theory.target_of o before_exit}
   201   end;
   202 
   203 val theory_init = init I "";
   204 
   205 fun reinit lthy =
   206   (case Data.get lthy of
   207     SOME (Target {target, before_exit, ...}) => init before_exit target o Local_Theory.exit_global
   208   | NONE => error "Not in a named target");
   209 
   210 fun context_cmd "-" thy = init I "" thy
   211   | context_cmd target thy = init I (Locale.intern thy target) thy;
   212 
   213 end;