src/Tools/interpretation_with_defs.ML

 Interpretation accompanied with mixin definitions.  EXPERIMENTAL.
 *)
 
 signature INTERPRETATION_WITH_DEFS =
 sig
-  val interpretation: Expression.expression_i ->
+  val permanent_interpretation: Expression.expression_i ->
     (Attrib.binding * ((binding * mixfix) * term)) list -> (Attrib.binding * term) list ->
-    theory -> Proof.state
-  val interpretation_cmd: Expression.expression ->
+    local_theory -> Proof.state
+  val permanent_interpretation_cmd: Expression.expression ->
     (Attrib.binding * ((binding * mixfix) * string)) list -> (Attrib.binding * string) list ->
-    theory -> Proof.state
+    local_theory -> Proof.state
 end;
 
 structure Interpretation_With_Defs : INTERPRETATION_WITH_DEFS =
 struct
 
-fun note_eqns_register deps witss def_eqns attrss eqns export export' =
+local
+
+(* reading *)
+
+fun prep_interpretation prep_expr prep_term prep_prop prep_attr expression raw_defs raw_eqns initial_ctxt =
   let
-    fun meta_rewrite context =
-      map (Local_Defs.meta_rewrite_rule (Context.proof_of context) #> Drule.abs_def) o
-        maps snd;
+    (*reading*)
+    val ((_, deps, _), proto_deps_ctxt) = prep_expr expression initial_ctxt;
+    val deps_ctxt = fold Locale.activate_declarations deps proto_deps_ctxt;
+    val prep = Syntax.check_terms deps_ctxt #> Variable.export_terms deps_ctxt deps_ctxt;
+    val rhss = (prep o map (prep_term deps_ctxt o snd o snd)) raw_defs;
+    val eqns = (prep o map (prep_prop deps_ctxt o snd)) raw_eqns;
+
+    (*defining*)
+    val pre_defs = map2 (fn ((name, atts), ((b, mx), _)) => fn rhs =>
+      ((b, mx), ((Thm.def_binding_optional b name, atts), rhs))) raw_defs rhss;
+    val (defs, defs_ctxt) = fold_map Local_Theory.define pre_defs initial_ctxt;
+    val def_eqns = map (Thm.symmetric o Morphism.thm (Local_Theory.standard_morphism defs_ctxt initial_ctxt) o snd o snd) defs;
+    val base_ctxt = if null def_eqns then defs_ctxt else Local_Theory.restore defs_ctxt;
+      (*the "if" prevents restoring a proof context which is no local theory*)
+
+    (*setting up*)
+    val ((propss, deps, export), expr_ctxt) = prep_expr expression base_ctxt;
+      (*FIXME: only re-prepare if defs are given*)
+    val attrss = map (apsnd (map (prep_attr (Proof_Context.theory_of expr_ctxt))) o fst) raw_eqns;
+    val goal_ctxt = fold Variable.auto_fixes eqns expr_ctxt;
+    val export' = Variable.export_morphism goal_ctxt expr_ctxt;
+  in (((propss, deps, export, export'), (def_eqns, eqns, attrss)), goal_ctxt) end;
+
+val cert_interpretation = prep_interpretation Expression.cert_goal_expression (K I) (K I) (K I);
+val read_interpretation = prep_interpretation Expression.read_goal_expression Syntax.parse_term Syntax.parse_prop Attrib.intern_src;
+
+
+(* generic interpretation machinery *)
+
+fun meta_rewrite eqns ctxt = (map (Local_Defs.meta_rewrite_rule ctxt #> Drule.abs_def) (maps snd eqns), ctxt);
+
+fun note_eqns_register note activate deps witss def_eqns eqns attrss export export' ctxt =
+  let
+    val facts = (Attrib.empty_binding, [(map (Morphism.thm (export' $> export)) def_eqns, [])])
+      :: map2 (fn attrs => fn eqn => (attrs, [([Morphism.thm (export' $> export) eqn], [])])) attrss eqns;
+    val (eqns', ctxt') = ctxt
+      |> note Thm.lemmaK facts
+      |-> meta_rewrite;
+    val dep_morphs = map2 (fn (dep, morph) => fn wits =>
+      (dep, morph $> Element.satisfy_morphism (map (Element.transform_witness export') wits))) deps witss;
+    fun activate' dep_morph ctxt = activate dep_morph
+      (Option.map (rpair true) (Element.eq_morphism (Proof_Context.theory_of ctxt) eqns')) export ctxt;
   in
-    Attrib.generic_notes Thm.lemmaK
-      (attrss ~~ map (fn eqn => [([Morphism.thm (export' $> export) eqn], [])]) eqns)
-    #-> (fn facts => `(fn context => meta_rewrite context facts))
-    #-> (fn eqns => fold (fn ((dep, morph), wits) =>
-      fn context =>
-        Locale.add_registration
-          (dep, morph $> Element.satisfy_morphism (map (Element.transform_witness export') wits))
-          (Element.eq_morphism (Context.theory_of context) (def_eqns @ eqns) |>
-            Option.map (rpair true))
-          export context) (deps ~~ witss))
+    ctxt'
+    |> fold activate' dep_morphs
   end;
 
-local
+fun generic_interpretation prep_interpretation setup_proof note add_registration
+    expression raw_defs raw_eqns initial_ctxt = 
+  let
+    val (((propss, deps, export, export'), (def_eqns, eqns, attrss)), goal_ctxt) = 
+      prep_interpretation expression raw_defs raw_eqns initial_ctxt;
+    fun after_qed witss eqns =
+      note_eqns_register note add_registration deps witss def_eqns eqns attrss export export';
+  in setup_proof after_qed propss eqns goal_ctxt end;
 
-fun gen_interpretation prep_expr prep_decl parse_term parse_prop prep_attr
-    expression raw_defs raw_eqns theory =
-  let
-    val (_, (_, defs_ctxt)) =
-      prep_decl expression I [] (Proof_Context.init_global theory);
 
-    val rhss = map (parse_term defs_ctxt o snd o snd) raw_defs
-      |> Syntax.check_terms defs_ctxt;
-    val defs = map2 (fn ((name, atts), ((b, mx), _)) => fn rhs =>
-      ((b, mx), ((Thm.def_binding_optional b name, atts), rhs))) raw_defs rhss;
+(* interpretation with permanent registration *)
 
-    val (def_eqns, theory') = theory
-      |> Named_Target.theory_init
-      |> fold_map (Local_Theory.define) defs
-      |>> map (Thm.symmetric o snd o snd)
-      |> Local_Theory.exit_result_global (map o Morphism.thm);
+fun subscribe lthy =
+  if Named_Target.is_theory lthy
+  then Generic_Target.theory_registration
+  else Generic_Target.locale_dependency (Named_Target.the_name lthy);
 
-    val ((propss, deps, export), expr_ctxt) = theory'
-      |> Proof_Context.init_global
-      |> prep_expr expression;
-
-    val eqns = map (parse_prop expr_ctxt o snd) raw_eqns
-      |> Syntax.check_terms expr_ctxt;
-    val attrss = map ((apsnd o map) (prep_attr theory) o fst) raw_eqns;
-    val goal_ctxt = fold Variable.auto_fixes eqns expr_ctxt;
-    val export' = Variable.export_morphism goal_ctxt expr_ctxt;
-
-    fun after_qed witss eqns =
-      (Proof_Context.background_theory o Context.theory_map)
-        (note_eqns_register deps witss def_eqns attrss eqns export export');
-
-  in Element.witness_proof_eqs after_qed propss eqns goal_ctxt end;
+fun gen_permanent_interpretation prep_interpretation expression raw_defs raw_eqns lthy =
+  generic_interpretation prep_interpretation Element.witness_proof_eqs Local_Theory.notes_kind (subscribe lthy)
+    expression raw_defs raw_eqns lthy
 
 in
 
-fun interpretation x = gen_interpretation Expression.cert_goal_expression
-  Expression.cert_declaration (K I) (K I) (K I) x;
-fun interpretation_cmd x = gen_interpretation Expression.read_goal_expression
-  Expression.read_declaration Syntax.parse_term Syntax.parse_prop Attrib.intern_src x;
+fun permanent_interpretation x = gen_permanent_interpretation cert_interpretation x;
+fun permanent_interpretation_cmd x = gen_permanent_interpretation read_interpretation x;
 
 end;
 
 val _ =
-  Outer_Syntax.command @{command_spec "interpretation"}
-    "prove interpretation of locale expression in theory"
+  Outer_Syntax.local_theory_to_proof @{command_spec "permanent_interpretation"}
+    "prove interpretation of locale expression into named theory"
     (Parse.!!! (Parse_Spec.locale_expression true) --
       Scan.optional (@{keyword "defines"} |-- Parse.and_list1 (Parse_Spec.opt_thm_name ":"
         -- ((Parse.binding -- Parse.opt_mixfix') --| @{keyword "is"} -- Parse.term))) [] --
       Scan.optional (Parse.where_ |-- Parse.and_list1 (Parse_Spec.opt_thm_name ":" -- Parse.prop)) []
-      >> (fn ((expr, defs), equations) => Toplevel.print o
-          Toplevel.theory_to_proof (interpretation_cmd expr defs equations)));
+      >> (fn ((expr, defs), eqns) => permanent_interpretation_cmd expr defs eqns));
 
 end;