src/Pure/Isar/local_theory.ML
author wenzelm
Wed, 25 Jan 2006 00:21:39 +0100
changeset 18781 ea3b5e22eab5
parent 18767 2f064e6bea7e
child 18805 18863b33c831
permissions -rw-r--r--
added constant definition; tuned interfaces; tuned;

(*  Title:      Pure/Isar/local_theory.ML
    ID:         $Id$
    Author:     Makarius

Local theory operations, with optional target locale.
*)

signature LOCAL_THEORY =
sig
  val locale_of: Proof.context -> string option
  val params_of: Proof.context -> (string * typ) list
  val hyps_of: Proof.context -> term list
  val standard: Proof.context -> thm -> thm
  val pretty_consts: Proof.context -> (string * typ) list -> Pretty.T
  val theory: (theory -> theory) -> Proof.context -> Proof.context
  val theory_result: (theory -> 'a * theory) -> Proof.context -> 'a * Proof.context
  val init: xstring option -> theory -> Proof.context
  val init_i: string option -> theory -> Proof.context
  val exit: Proof.context -> theory
  val consts: ((string * typ) * mixfix) list -> Proof.context -> term list * Proof.context
  val axioms: ((string * Attrib.src list) * term list) list -> Proof.context ->
    (bstring * thm list) list * Proof.context
  val notes: string -> ((bstring * Attrib.src list) * (thm list * Attrib.src list) list) list ->
    Proof.context -> (bstring * thm list) list * Proof.context
  val note: (bstring * Attrib.src list) * thm list -> Proof.context ->
    (bstring * thm list) * Proof.context
  val def: (string * mixfix) * ((string * Attrib.src list) * term) ->
    Proof.context -> (term * (bstring * thm)) * Proof.context
  val def': (Proof.context -> term -> thm -> thm) ->
    (string * mixfix) * ((string * Attrib.src list) * term) ->
    Proof.context -> (term * (bstring * thm)) * Proof.context
end;

structure LocalTheory: LOCAL_THEORY =
struct


(** local context data **)

structure Data = ProofDataFun
(
  val name = "Isar/local_theory";
  type T =
   {locale: string option,
    params: (string * typ) list,
    hyps: term list,
    standard: Proof.context -> thm -> thm,
    exit: theory -> theory};
  fun init _ = {locale = NONE, params = [], hyps = [], standard = K Drule.standard, exit = I};
  fun print _ _ = ();
);

val _ = Context.add_setup Data.init;

val locale_of = #locale o Data.get;
val params_of = #params o Data.get;
val hyps_of = #hyps o Data.get;
fun standard ctxt = #standard (Data.get ctxt) ctxt;


(* pretty_consts *)

local

fun pretty_var ctxt (x, T) =
  Pretty.block [Pretty.str x, Pretty.str " ::", Pretty.brk 1,
    Pretty.quote (ProofContext.pretty_typ ctxt T)];

fun pretty_vars ctxt kind vs = Pretty.big_list kind (map (pretty_var ctxt) vs);

in

fun pretty_consts ctxt cs =
  (case params_of ctxt of
    [] => pretty_vars ctxt "constants" cs
  | ps => Pretty.chunks [pretty_vars ctxt "parameters" ps, pretty_vars ctxt "constants" cs]);

end;



(** theory **)

fun theory f ctxt =
  ProofContext.transfer (f (ProofContext.theory_of ctxt)) ctxt;

fun theory_result f ctxt =
  let val (res, thy') = f (ProofContext.theory_of ctxt)
  in (res, ProofContext.transfer thy' ctxt) end;

fun init_i NONE thy = ProofContext.init thy
  | init_i (SOME loc) thy =
      thy
      |> Locale.init loc
      |> ProofContext.inferred_fixes
      |> (fn (params, ctxt) => Data.put
          {locale = SOME loc,
           params = params,
           hyps = ProofContext.assms_of ctxt,
           standard = fn inner => ProofContext.export_standard inner ctxt,
           exit = Sign.restore_naming thy} ctxt)
      |> theory (Sign.add_path (Sign.base_name loc) #> Sign.no_base_names);

fun init target thy = init_i (Option.map (Locale.intern thy) target) thy;

fun exit ctxt = #exit (Data.get ctxt) (ProofContext.theory_of ctxt);



(** local theory **)

(* consts *)

fun consts decls ctxt =
  let
    val thy = ProofContext.theory_of ctxt;
    val params = params_of ctxt;
    val ps = map Free params;
    val Ps = map snd params;
  in
    ctxt
    |> theory (Sign.add_consts_i (decls |> map (fn ((c, T), mx) => (c, Ps ---> T, mx))))
    |> pair (decls |> map (fn ((c, T), _) =>
      Term.list_comb (Const (Sign.full_name thy c, Ps ---> T), ps)))
  end;


(* fact definition *)

fun notes kind facts ctxt =
  (case locale_of ctxt of
    NONE => ctxt |> theory_result
      (PureThy.note_thmss_i (Drule.kind kind)
        (Attrib.map_facts (Attrib.attribute_i (ProofContext.theory_of ctxt)) facts))
  | SOME loc => ctxt |> theory_result (Locale.note_thmss_i kind loc facts #> apsnd #1));

fun note (a, ths) ctxt =
  ctxt |> notes Drule.theoremK [(a, [(ths, [])])] |>> hd;


(* axioms *)

local

fun add_axiom hyps (name, prop) thy =
  let
    val name' = if name = "" then "axiom_" ^ string_of_int (serial ()) else name;
    val frees = rev ([] |> Term.add_frees prop |> fold Term.add_frees hyps);
    val prop' = Term.list_all_free (frees, Logic.list_implies (hyps, prop));
    val thy' = thy |> Theory.add_axioms_i [(name', prop')];

    val cert = Thm.cterm_of thy';
    fun all_polymorphic (x, T) th =
      let val var = cert (Var ((x, 0), #1 (Logic.dest_all (Thm.prop_of th))))
      in ((var, cert (Free (x, T))), Thm.forall_elim var th) end;
    fun implies_polymorphic hyp th = Thm.assume (cert hyp) COMP th;
    val th =
      Thm.get_axiom_i thy' (Sign.full_name thy' name')
      |> fold_map all_polymorphic frees |-> Drule.cterm_instantiate
      |> fold implies_polymorphic hyps
  in (th, thy') end;

in

fun axioms specs ctxt =
  let
    fun name_list name [x] = [(name, x)]
      | name_list name xs = PureThy.name_multi name xs;
    val fixes_ctxt = fold (fold ProofContext.fix_frees o snd) specs ctxt;
  in
    ctxt |> fold_map (fn ((name, atts), props) =>
      theory_result (fold_map (add_axiom (hyps_of ctxt)) (name_list name props))
      #-> (fn ths => note ((name, atts), map (standard fixes_ctxt) ths))) specs
  end;

end;


(* constant definition *)

local

fun add_def (name, prop) thy =
  let
    val thy' = thy |> Theory.add_defs_i false [(name, prop)];
    val th = Thm.get_axiom_i thy' (Sign.full_name thy' name);
    val cert = Thm.cterm_of thy';
    val subst = map2 (fn var => fn free => (cert (Var var), cert (Free free)))
      (Term.add_vars (Thm.prop_of th) []) (Term.add_frees prop []);
  in (Drule.cterm_instantiate subst th, thy') end;

in

fun def' finish (var, spec) ctxt =
  let
    val (x, mx) = var;
    val ((name, atts), rhs) = spec;
    val name' = if name = "" then Thm.def_name x else name;
  in
    ctxt
    |> consts [((x, Term.fastype_of rhs), mx)]
    |-> (fn [lhs] =>
      theory_result (add_def (name', Logic.mk_equals (lhs, rhs)))
      #-> (fn th => fn ctxt' => note ((name', atts), [finish ctxt' lhs th]) ctxt')
      #> apfst (fn (b, [th]) => (lhs, (b, th))))
  end;

end;

fun def (var, spec) =
  def' (fn ctxt => fn _ => standard (ProofContext.fix_frees (snd spec) ctxt)) (var, spec);

end;