src/HOL/Tools/Lifting/lifting_setup.ML

--- a/src/HOL/Tools/Lifting/lifting_setup.ML	Wed Apr 18 23:13:11 2012 +0200
+++ b/src/HOL/Tools/Lifting/lifting_setup.ML	Wed Apr 18 23:57:44 2012 +0200
@@ -8,11 +8,11 @@
 sig
   exception SETUP_LIFTING_INFR of string
 
-  val setup_lifting_infr: thm -> local_theory -> local_theory
+  val setup_lifting_infr: bool -> thm -> local_theory -> local_theory
 
-  val setup_by_quotient: thm -> thm option -> local_theory -> local_theory
+  val setup_by_quotient: bool -> thm -> thm option -> local_theory -> local_theory
 
-  val setup_by_typedef_thm: thm -> local_theory -> local_theory
+  val setup_by_typedef_thm: bool -> thm -> local_theory -> local_theory
 end;
 
 structure Lifting_Setup: LIFTING_SETUP =
@@ -38,8 +38,8 @@
     (def_thm, lthy'')
   end
 
-fun define_abs_type quot_thm lthy =
-  if Lifting_Def.can_generate_code_cert quot_thm then
+fun define_abs_type no_code quot_thm lthy =
+  if not no_code andalso Lifting_Def.can_generate_code_cert quot_thm then
     let
       val abs_type_thm = quot_thm RS @{thm Quotient_abs_rep}
       val add_abstype_attribute = 
@@ -78,7 +78,7 @@
                                                 @ (map Pretty.string_of errs)))
   end
 
-fun setup_lifting_infr quot_thm lthy =
+fun setup_lifting_infr no_code quot_thm lthy =
   let
     val _ = quot_thm_sanity_check lthy quot_thm
     val (_, qtyp) = Lifting_Term.quot_thm_rty_qty quot_thm
@@ -89,10 +89,10 @@
     lthy
       |> Local_Theory.declaration {syntax = false, pervasive = true}
         (fn phi => Lifting_Info.update_quotients qty_full_name (quot_info phi))
-      |> define_abs_type quot_thm
+      |> define_abs_type no_code quot_thm
   end
 
-fun setup_by_quotient quot_thm maybe_reflp_thm lthy =
+fun setup_by_quotient no_code quot_thm maybe_reflp_thm lthy =
   let
     val transfer_attr = Attrib.internal (K Transfer.transfer_add)
     val (_, qty) = Lifting_Term.quot_thm_rty_qty quot_thm
@@ -119,10 +119,10 @@
         [quot_thm RS @{thm Quotient_right_total}])
       |> (snd oo Local_Theory.note) ((qualify "rel_eq_transfer", [transfer_attr]), 
         [quot_thm RS @{thm Quotient_rel_eq_transfer}])
-      |> setup_lifting_infr quot_thm
+      |> setup_lifting_infr no_code quot_thm
   end
 
-fun setup_by_typedef_thm typedef_thm lthy =
+fun setup_by_typedef_thm no_code typedef_thm lthy =
   let
     val transfer_attr = Attrib.internal (K Transfer.transfer_add)
     val (_ $ rep_fun $ _ $ typedef_set) = (HOLogic.dest_Trueprop o prop_of) typedef_thm
@@ -169,22 +169,53 @@
         [[quot_thm] MRSL @{thm Quotient_right_unique}])
       |> (snd oo Local_Theory.note) ((qualify "right_total", [transfer_attr]), 
         [[quot_thm] MRSL @{thm Quotient_right_total}])
-      |> setup_lifting_infr quot_thm
+      |> setup_lifting_infr no_code quot_thm
   end
 
-fun setup_lifting_cmd xthm lthy =
+fun setup_lifting_cmd no_code xthm opt_reflp_xthm lthy =
   let 
     val input_thm = singleton (Attrib.eval_thms lthy) xthm
     val input_term = (HOLogic.dest_Trueprop o prop_of) input_thm
+      handle TERM _ => error "Unsupported type of a theorem. Only Quotient or type_definition are supported."
+
+    fun sanity_check_reflp_thm reflp_thm = 
+      let
+        val reflp_tm = (HOLogic.dest_Trueprop o prop_of) reflp_thm
+          handle TERM _ => error "Invalid form of the reflexivity theorem. Use \"reflp R\"."
+      in
+        case reflp_tm of
+          Const (@{const_name reflp}, _) $ _ => ()
+          | _ => error "Invalid form of the reflexivity theorem. Use \"reflp R\"."
+      end
+
+    fun setup_quotient () = 
+      case opt_reflp_xthm of
+        SOME reflp_xthm => 
+          let
+            val reflp_thm = singleton (Attrib.eval_thms lthy) reflp_xthm
+            val _ = sanity_check_reflp_thm reflp_thm
+          in
+            setup_by_quotient no_code input_thm (SOME reflp_thm) lthy
+          end
+        | NONE => setup_by_quotient no_code input_thm NONE lthy
+
+    fun setup_typedef () = 
+      case opt_reflp_xthm of
+        SOME _ => error "The reflexivity theorem cannot be specified if the type_definition theorem is used."
+        | NONE => setup_by_typedef_thm no_code input_thm lthy
   in
     case input_term of
-      (Const (@{const_name Quotient}, _) $ _ $ _ $ _ $ _) => setup_lifting_infr input_thm lthy
-      | (Const (@{const_name type_definition}, _) $ _ $ _ $ _) => setup_by_typedef_thm input_thm lthy
+      (Const (@{const_name Quotient}, _) $ _ $ _ $ _ $ _) => setup_quotient ()
+      | (Const (@{const_name type_definition}, _) $ _ $ _ $ _) => setup_typedef ()
       | _ => error "Unsupported type of a theorem. Only Quotient or type_definition are supported."
   end
 
+val opt_no_code =
+  Scan.optional (@{keyword "("} |-- Parse.!!! ((Parse.reserved "no_code" >> K true) --| @{keyword ")"})) false
+
 val _ = 
   Outer_Syntax.local_theory @{command_spec "setup_lifting"}
     "Setup lifting infrastructure" 
-      (Parse_Spec.xthm >> (fn xthm => setup_lifting_cmd xthm))
+      (opt_no_code -- Parse_Spec.xthm -- Scan.option Parse_Spec.xthm >> 
+        (fn ((no_code, xthm), opt_reflp_xthm) => setup_lifting_cmd no_code xthm opt_reflp_xthm))
 end;