src/HOL/Tools/inductive_realizer.ML
author wenzelm
Thu Jul 05 20:01:26 2007 +0200 (2007-07-05)
changeset 23590 ad95084a5c63
parent 23577 c5b93c69afd3
child 24157 409cd6eaa7ea
permissions -rw-r--r--
renamed ObjectLogic.atomize_tac to ObjectLogic.atomize_prems_tac;
berghofe@13710
     1
(*  Title:      HOL/Tools/inductive_realizer.ML
berghofe@13710
     2
    ID:         $Id$
berghofe@13710
     3
    Author:     Stefan Berghofer, TU Muenchen
berghofe@13710
     4
berghofe@13710
     5
Porgram extraction from proofs involving inductive predicates:
berghofe@13710
     6
Realizers for induction and elimination rules
berghofe@13710
     7
*)
berghofe@13710
     8
berghofe@13710
     9
signature INDUCTIVE_REALIZER =
berghofe@13710
    10
sig
berghofe@13710
    11
  val add_ind_realizers: string -> string list -> theory -> theory
wenzelm@18708
    12
  val setup: theory -> theory
berghofe@13710
    13
end;
berghofe@13710
    14
berghofe@13710
    15
structure InductiveRealizer : INDUCTIVE_REALIZER =
berghofe@13710
    16
struct
berghofe@13710
    17
berghofe@22271
    18
(* FIXME: LocalTheory.note should return theorems with proper names! *)
berghofe@22606
    19
fun name_of_thm thm =
berghofe@22606
    20
  (case Symtab.dest (Proofterm.thms_of_proof' (proof_of thm) Symtab.empty) of
berghofe@22606
    21
     [(name, _)] => name
berghofe@22606
    22
   | _ => error ("name_of_thm: bad proof of theorem\n" ^ string_of_thm thm));
berghofe@22271
    23
berghofe@13710
    24
val all_simps = map (symmetric o mk_meta_eq) (thms "HOL.all_simps");
berghofe@13710
    25
berghofe@13710
    26
fun prf_of thm =
wenzelm@22596
    27
  let val {thy, prop, der = (_, prf), ...} = rep_thm thm
wenzelm@22596
    28
  in Reconstruct.expand_proof thy [("", NONE)] (Reconstruct.reconstruct_proof thy prop prf) end; (* FIXME *)
berghofe@13710
    29
berghofe@13710
    30
fun forall_intr_prf (t, prf) =
berghofe@13710
    31
  let val (a, T) = (case t of Var ((a, _), T) => (a, T) | Free p => p)
skalberg@15531
    32
  in Abst (a, SOME T, Proofterm.prf_abstract_over t prf) end;
berghofe@13710
    33
berghofe@22271
    34
fun forall_intr_term (t, u) =
berghofe@22271
    35
  let val (a, T) = (case t of Var ((a, _), T) => (a, T) | Free p => p)
berghofe@22271
    36
  in all T $ Abs (a, T, abstract_over (t, u)) end;
berghofe@22271
    37
berghofe@13710
    38
fun subsets [] = [[]]
berghofe@13710
    39
  | subsets (x::xs) =
berghofe@13710
    40
      let val ys = subsets xs
berghofe@13710
    41
      in ys @ map (cons x) ys end;
berghofe@13710
    42
berghofe@22271
    43
val pred_of = fst o dest_Const o head_of;
berghofe@13710
    44
berghofe@22271
    45
fun strip_all' used names (Const ("all", _) $ Abs (s, T, t)) =
berghofe@22271
    46
      let val (s', names') = (case names of
berghofe@22271
    47
          [] => (Name.variant used s, [])
berghofe@22271
    48
        | name :: names' => (name, names'))
berghofe@22271
    49
      in strip_all' (s'::used) names' (subst_bound (Free (s', T), t)) end
berghofe@22271
    50
  | strip_all' used names ((t as Const ("==>", _) $ P) $ Q) =
berghofe@22271
    51
      t $ strip_all' used names Q
berghofe@22271
    52
  | strip_all' _ _ t = t;
berghofe@22271
    53
berghofe@22271
    54
fun strip_all t = strip_all' (add_term_free_names (t, [])) [] t;
berghofe@22271
    55
berghofe@22271
    56
fun strip_one name (Const ("all", _) $ Abs (s, T, Const ("==>", _) $ P $ Q)) =
berghofe@22271
    57
      (subst_bound (Free (name, T), P), subst_bound (Free (name, T), Q))
berghofe@22271
    58
  | strip_one _ (Const ("==>", _) $ P $ Q) = (P, Q);
berghofe@13710
    59
skalberg@15574
    60
fun relevant_vars prop = foldr (fn
berghofe@13710
    61
      (Var ((a, i), T), vs) => (case strip_type T of
berghofe@22271
    62
        (_, Type (s, _)) => if s mem ["bool"] then (a, T) :: vs else vs
berghofe@13710
    63
      | _ => vs)
skalberg@15574
    64
    | (_, vs) => vs) [] (term_vars prop);
berghofe@13710
    65
berghofe@22271
    66
fun dt_of_intrs thy vs nparms intrs =
berghofe@13710
    67
  let
berghofe@13710
    68
    val iTs = term_tvars (prop_of (hd intrs));
berghofe@13710
    69
    val Tvs = map TVar iTs;
berghofe@22271
    70
    val (Const (s, _), ts) = strip_comb (HOLogic.dest_Trueprop
berghofe@22271
    71
      (Logic.strip_imp_concl (prop_of (hd intrs))));
berghofe@22271
    72
    val params = map dest_Var (Library.take (nparms, ts));
berghofe@13710
    73
    val tname = space_implode "_" (Sign.base_name s ^ "T" :: vs);
berghofe@22271
    74
    fun constr_of_intr intr = (Sign.base_name (name_of_thm intr),
wenzelm@19806
    75
      map (Logic.unvarifyT o snd) (rev (Term.add_vars (prop_of intr) []) \\ params) @
berghofe@13710
    76
        filter_out (equal Extraction.nullT) (map
wenzelm@19806
    77
          (Logic.unvarifyT o Extraction.etype_of thy vs []) (prems_of intr)),
berghofe@13710
    78
            NoSyn);
berghofe@13710
    79
  in (map (fn a => "'" ^ a) vs @ map (fst o fst) iTs, tname, NoSyn,
berghofe@13710
    80
    map constr_of_intr intrs)
berghofe@13710
    81
  end;
berghofe@13710
    82
berghofe@13710
    83
fun mk_rlz T = Const ("realizes", [T, HOLogic.boolT] ---> HOLogic.boolT);
berghofe@13710
    84
berghofe@22271
    85
(** turn "P" into "%r x. realizes r (P x)" **)
berghofe@13710
    86
berghofe@13710
    87
fun gen_rvar vs (t as Var ((a, 0), T)) =
berghofe@22271
    88
      if body_type T <> HOLogic.boolT then t else
berghofe@22271
    89
        let
berghofe@22271
    90
          val U = TVar (("'" ^ a, 0), HOLogic.typeS)
berghofe@22271
    91
          val Ts = binder_types T;
berghofe@22271
    92
          val i = length Ts;
berghofe@22271
    93
          val xs = map (pair "x") Ts;
berghofe@22271
    94
          val u = list_comb (t, map Bound (i - 1 downto 0))
berghofe@22271
    95
        in 
berghofe@22271
    96
          if a mem vs then
berghofe@22271
    97
            list_abs (("r", U) :: xs, mk_rlz U $ Bound i $ u)
berghofe@22271
    98
          else list_abs (xs, mk_rlz Extraction.nullT $ Extraction.nullt $ u)
berghofe@22271
    99
        end
berghofe@13710
   100
  | gen_rvar _ t = t;
berghofe@13710
   101
berghofe@22271
   102
fun mk_realizes_eqn n vs nparms intrs =
berghofe@13710
   103
  let
berghofe@22271
   104
    val concl = HOLogic.dest_Trueprop (concl_of (hd intrs));
berghofe@22271
   105
    val iTs = term_tvars concl;
berghofe@13710
   106
    val Tvs = map TVar iTs;
berghofe@22271
   107
    val (h as Const (s, T), us) = strip_comb concl;
berghofe@22271
   108
    val params = List.take (us, nparms);
berghofe@22271
   109
    val elTs = List.drop (binder_types T, nparms);
berghofe@22271
   110
    val predT = elTs ---> HOLogic.boolT;
berghofe@22271
   111
    val used = map (fst o fst o dest_Var) params;
berghofe@22271
   112
    val xs = map (Var o apfst (rpair 0))
berghofe@22271
   113
      (Name.variant_list used (replicate (length elTs) "x") ~~ elTs);
berghofe@13710
   114
    val rT = if n then Extraction.nullT
berghofe@13710
   115
      else Type (space_implode "_" (s ^ "T" :: vs),
berghofe@13710
   116
        map (fn a => TVar (("'" ^ a, 0), HOLogic.typeS)) vs @ Tvs);
berghofe@13710
   117
    val r = if n then Extraction.nullt else Var ((Sign.base_name s, 0), rT);
berghofe@22271
   118
    val S = list_comb (h, params @ xs);
berghofe@13710
   119
    val rvs = relevant_vars S;
berghofe@13710
   120
    val vs' = map fst rvs \\ vs;
berghofe@13710
   121
    val rname = space_implode "_" (s ^ "R" :: vs);
berghofe@13710
   122
berghofe@13710
   123
    fun mk_Tprem n v =
haftmann@17485
   124
      let val T = (the o AList.lookup (op =) rvs) v
berghofe@13710
   125
      in (Const ("typeof", T --> Type ("Type", [])) $ Var ((v, 0), T),
berghofe@13710
   126
        Extraction.mk_typ (if n then Extraction.nullT
berghofe@13710
   127
          else TVar (("'" ^ v, 0), HOLogic.typeS)))
berghofe@13710
   128
      end;
berghofe@13710
   129
berghofe@13710
   130
    val prems = map (mk_Tprem true) vs' @ map (mk_Tprem false) vs;
berghofe@22271
   131
    val ts = map (gen_rvar vs) params;
berghofe@13710
   132
    val argTs = map fastype_of ts;
berghofe@13710
   133
berghofe@22271
   134
  in ((prems, (Const ("typeof", HOLogic.boolT --> Type ("Type", [])) $ S,
berghofe@13710
   135
       Extraction.mk_typ rT)),
berghofe@22271
   136
    (prems, (mk_rlz rT $ r $ S,
berghofe@22271
   137
       if n then list_comb (Const (rname, argTs ---> predT), ts @ xs)
berghofe@22271
   138
       else list_comb (Const (rname, argTs @ [rT] ---> predT), ts @ [r] @ xs))))
berghofe@13710
   139
  end;
berghofe@13710
   140
berghofe@22271
   141
fun fun_of_prem thy rsets vs params rule ivs intr =
berghofe@13710
   142
  let
berghofe@22271
   143
    val ctxt = ProofContext.init thy
berghofe@22271
   144
    val args = map (Free o apfst fst o dest_Var) ivs;
berghofe@13710
   145
    val args' = map (Free o apfst fst)
wenzelm@16861
   146
      (Term.add_vars (prop_of intr) [] \\ params);
berghofe@13710
   147
    val rule' = strip_all rule;
berghofe@13710
   148
    val conclT = Extraction.etype_of thy vs [] (Logic.strip_imp_concl rule');
berghofe@13710
   149
    val used = map (fst o dest_Free) args;
berghofe@13710
   150
berghofe@13710
   151
    fun is_rec t = not (null (term_consts t inter rsets));
berghofe@13710
   152
berghofe@13710
   153
    fun is_meta (Const ("all", _) $ Abs (s, _, P)) = is_meta P
berghofe@13710
   154
      | is_meta (Const ("==>", _) $ _ $ Q) = is_meta Q
berghofe@22271
   155
      | is_meta (Const ("Trueprop", _) $ t) = (case head_of t of
berghofe@22271
   156
          Const (s, _) => can (InductivePackage.the_inductive ctxt) s
berghofe@22271
   157
        | _ => true)
berghofe@13710
   158
      | is_meta _ = false;
berghofe@13710
   159
berghofe@13710
   160
    fun fun_of ts rts args used (prem :: prems) =
berghofe@13710
   161
          let
berghofe@13710
   162
            val T = Extraction.etype_of thy vs [] prem;
wenzelm@20071
   163
            val [x, r] = Name.variant_list used ["x", "r"]
berghofe@13710
   164
          in if T = Extraction.nullT
berghofe@13710
   165
            then fun_of ts rts args used prems
berghofe@13710
   166
            else if is_rec prem then
berghofe@13710
   167
              if is_meta prem then
berghofe@13710
   168
                let
berghofe@13710
   169
                  val prem' :: prems' = prems;
berghofe@13710
   170
                  val U = Extraction.etype_of thy vs [] prem';
berghofe@13710
   171
                in if U = Extraction.nullT
berghofe@13710
   172
                  then fun_of (Free (x, T) :: ts)
berghofe@13710
   173
                    (Free (r, binder_types T ---> HOLogic.unitT) :: rts)
berghofe@13710
   174
                    (Free (x, T) :: args) (x :: r :: used) prems'
berghofe@13710
   175
                  else fun_of (Free (x, T) :: ts) (Free (r, U) :: rts)
berghofe@13710
   176
                    (Free (r, U) :: Free (x, T) :: args) (x :: r :: used) prems'
berghofe@13710
   177
                end
berghofe@13710
   178
              else (case strip_type T of
berghofe@13710
   179
                  (Ts, Type ("*", [T1, T2])) =>
berghofe@13710
   180
                    let
berghofe@13710
   181
                      val fx = Free (x, Ts ---> T1);
berghofe@13710
   182
                      val fr = Free (r, Ts ---> T2);
berghofe@13710
   183
                      val bs = map Bound (length Ts - 1 downto 0);
berghofe@13710
   184
                      val t = list_abs (map (pair "z") Ts,
berghofe@13710
   185
                        HOLogic.mk_prod (list_comb (fx, bs), list_comb (fr, bs)))
berghofe@13710
   186
                    in fun_of (fx :: ts) (fr :: rts) (t::args)
berghofe@13710
   187
                      (x :: r :: used) prems
berghofe@13710
   188
                    end
berghofe@13710
   189
                | (Ts, U) => fun_of (Free (x, T) :: ts)
berghofe@13710
   190
                    (Free (r, binder_types T ---> HOLogic.unitT) :: rts)
berghofe@13710
   191
                    (Free (x, T) :: args) (x :: r :: used) prems)
berghofe@13710
   192
            else fun_of (Free (x, T) :: ts) rts (Free (x, T) :: args)
berghofe@13710
   193
              (x :: used) prems
berghofe@13710
   194
          end
berghofe@13710
   195
      | fun_of ts rts args used [] =
berghofe@13710
   196
          let val xs = rev (rts @ ts)
berghofe@13710
   197
          in if conclT = Extraction.nullT
berghofe@13710
   198
            then list_abs_free (map dest_Free xs, HOLogic.unit)
berghofe@13710
   199
            else list_abs_free (map dest_Free xs, list_comb
berghofe@22271
   200
              (Free ("r" ^ Sign.base_name (name_of_thm intr),
berghofe@13710
   201
                map fastype_of (rev args) ---> conclT), rev args))
berghofe@13710
   202
          end
berghofe@13710
   203
berghofe@13921
   204
  in fun_of args' [] (rev args) used (Logic.strip_imp_prems rule') end;
berghofe@13710
   205
berghofe@13710
   206
fun indrule_realizer thy induct raw_induct rsets params vs rec_names rss intrs dummies =
berghofe@13710
   207
  let
berghofe@13710
   208
    val concls = HOLogic.dest_conj (HOLogic.dest_Trueprop (concl_of raw_induct));
skalberg@15570
   209
    val premss = List.mapPartial (fn (s, rs) => if s mem rsets then
berghofe@22271
   210
      SOME (rs, map (fn (_, r) => List.nth (prems_of raw_induct,
skalberg@15570
   211
        find_index_eq (prop_of r) (map prop_of intrs))) rs) else NONE) rss;
berghofe@22271
   212
    val fs = maps (fn ((intrs, prems), dummy) =>
berghofe@13710
   213
      let
berghofe@22271
   214
        val fs = map (fn (rule, (ivs, intr)) =>
berghofe@22271
   215
          fun_of_prem thy rsets vs params rule ivs intr) (prems ~~ intrs)
berghofe@22271
   216
      in if dummy then Const ("arbitrary",
berghofe@13710
   217
          HOLogic.unitT --> body_type (fastype_of (hd fs))) :: fs
berghofe@22271
   218
        else fs
berghofe@22271
   219
      end) (premss ~~ dummies);
wenzelm@16861
   220
    val frees = fold Term.add_frees fs [];
berghofe@13710
   221
    val Ts = map fastype_of fs;
berghofe@22271
   222
    fun name_of_fn intr = "r" ^ Sign.base_name (name_of_thm intr)
berghofe@22271
   223
  in
berghofe@22271
   224
    fst (fold_map (fn concl => fn names =>
berghofe@13710
   225
      let val T = Extraction.etype_of thy vs [] concl
berghofe@22271
   226
      in if T = Extraction.nullT then (Extraction.nullt, names) else
berghofe@22271
   227
        let
berghofe@22271
   228
          val Type ("fun", [U, _]) = T;
berghofe@22271
   229
          val a :: names' = names
berghofe@22271
   230
        in (list_abs_free (("x", U) :: List.mapPartial (fn intr =>
berghofe@22271
   231
          Option.map (pair (name_of_fn intr))
berghofe@22271
   232
            (AList.lookup (op =) frees (name_of_fn intr))) intrs,
berghofe@22271
   233
          list_comb (Const (a, Ts ---> T), fs) $ Free ("x", U)), names')
berghofe@22271
   234
        end
berghofe@22271
   235
      end) concls rec_names)
berghofe@13710
   236
  end;
berghofe@13710
   237
berghofe@13710
   238
fun add_dummy name dname (x as (_, (vs, s, mfx, cs))) =
wenzelm@23577
   239
  if (name: string) = s then (true, (vs, s, mfx, (dname, [HOLogic.unitT], NoSyn) :: cs))
berghofe@13710
   240
  else x;
berghofe@13710
   241
haftmann@18314
   242
fun add_dummies f [] _ thy =
haftmann@18314
   243
      (([], NONE), thy)
haftmann@18314
   244
  | add_dummies f dts used thy =
haftmann@18314
   245
      thy
haftmann@18314
   246
      |> f (map snd dts)
haftmann@18314
   247
      |-> (fn dtinfo => pair ((map fst dts), SOME dtinfo))
haftmann@18314
   248
    handle DatatypeAux.Datatype_Empty name' =>
berghofe@13710
   249
      let
berghofe@14888
   250
        val name = Sign.base_name name';
wenzelm@20071
   251
        val dname = Name.variant used "Dummy"
haftmann@18314
   252
      in
haftmann@18314
   253
        thy
haftmann@18314
   254
        |> add_dummies f (map (add_dummy name dname) dts) (dname :: used)
berghofe@14888
   255
      end;
berghofe@13710
   256
berghofe@22271
   257
fun mk_realizer thy vs (name, rule, rrule, rlz, rt) =
berghofe@13710
   258
  let
berghofe@13725
   259
    val rvs = map fst (relevant_vars (prop_of rule));
wenzelm@16861
   260
    val xs = rev (Term.add_vars (prop_of rule) []);
berghofe@13725
   261
    val vs1 = map Var (filter_out (fn ((a, _), _) => a mem rvs) xs);
wenzelm@16861
   262
    val rlzvs = rev (Term.add_vars (prop_of rrule) []);
haftmann@17485
   263
    val vs2 = map (fn (ixn, _) => Var (ixn, (the o AList.lookup (op =) rlzvs) ixn)) xs;
berghofe@22271
   264
    val rs = map Var (subtract (op = o pairself fst) xs rlzvs);
berghofe@22271
   265
    val rlz' = foldr forall_intr_term (prop_of rrule) (vs2 @ rs);
berghofe@22271
   266
    val rlz'' = foldr forall_intr_term rlz vs2
berghofe@22271
   267
  in (name, (vs,
berghofe@13710
   268
    if rt = Extraction.nullt then rt else
skalberg@15574
   269
      foldr (uncurry lambda) rt vs1,
berghofe@22271
   270
    ProofRewriteRules.un_hhf_proof rlz' rlz''
berghofe@22271
   271
      (foldr forall_intr_prf (prf_of rrule) (vs2 @ rs))))
berghofe@13710
   272
  end;
berghofe@13710
   273
berghofe@13710
   274
fun add_ind_realizer rsets intrs induct raw_induct elims (thy, vs) =
berghofe@13710
   275
  let
berghofe@22271
   276
    val qualifier = NameSpace.qualifier (name_of_thm induct);
berghofe@22271
   277
    val inducts = PureThy.get_thms thy (Name
berghofe@22271
   278
      (NameSpace.qualified qualifier "inducts"));
berghofe@13710
   279
    val iTs = term_tvars (prop_of (hd intrs));
berghofe@13710
   280
    val ar = length vs + length iTs;
berghofe@22790
   281
    val params = InductivePackage.params_of raw_induct;
berghofe@22790
   282
    val arities = InductivePackage.arities_of raw_induct;
berghofe@22271
   283
    val nparms = length params;
berghofe@13710
   284
    val params' = map dest_Var params;
berghofe@22790
   285
    val rss = InductivePackage.partition_rules raw_induct intrs;
berghofe@22271
   286
    val rss' = map (fn (((s, rs), (_, arity)), elim) =>
berghofe@22790
   287
      (s, (InductivePackage.infer_intro_vars elim arity rs ~~ rs)))
berghofe@22790
   288
        (rss ~~ arities ~~ elims);
wenzelm@21858
   289
    val (prfx, _) = split_last (NameSpace.explode (fst (hd rss)));
berghofe@13710
   290
    val tnames = map (fn s => space_implode "_" (s ^ "T" :: vs)) rsets;
wenzelm@16123
   291
berghofe@13710
   292
    val thy1 = thy |>
berghofe@13710
   293
      Theory.root_path |>
wenzelm@21858
   294
      Theory.add_path (NameSpace.implode prfx);
berghofe@13710
   295
    val (ty_eqs, rlz_eqs) = split_list
berghofe@22271
   296
      (map (fn (s, rs) => mk_realizes_eqn (not (s mem rsets)) vs nparms rs) rss);
berghofe@13710
   297
berghofe@13710
   298
    val thy1' = thy1 |>
berghofe@13710
   299
      Theory.copy |>
berghofe@13710
   300
      Theory.add_types (map (fn s => (Sign.base_name s, ar, NoSyn)) tnames) |>
wenzelm@19510
   301
      fold (fn s => AxClass.axiomatize_arity_i
wenzelm@19510
   302
        (s, replicate ar HOLogic.typeS, HOLogic.typeS)) tnames |>
berghofe@13710
   303
        Extraction.add_typeof_eqns_i ty_eqs;
skalberg@15570
   304
    val dts = List.mapPartial (fn (s, rs) => if s mem rsets then
berghofe@22271
   305
      SOME (dt_of_intrs thy1' vs nparms rs) else NONE) rss;
berghofe@13710
   306
berghofe@13710
   307
    (** datatype representing computational content of inductive set **)
berghofe@13710
   308
haftmann@18314
   309
    val ((dummies, dt_info), thy2) =
haftmann@18008
   310
      thy1
haftmann@18314
   311
      |> add_dummies
haftmann@18314
   312
           (DatatypePackage.add_datatype_i false false (map #2 dts))
haftmann@18314
   313
           (map (pair false) dts) []
haftmann@18314
   314
      ||> Extraction.add_typeof_eqns_i ty_eqs
haftmann@18314
   315
      ||> Extraction.add_realizes_eqns_i rlz_eqs;
haftmann@18314
   316
    fun get f = (these oo Option.map) f;
wenzelm@19046
   317
    val rec_names = distinct (op =) (map (fst o dest_Const o head_of o fst o
berghofe@13710
   318
      HOLogic.dest_eq o HOLogic.dest_Trueprop o prop_of) (get #rec_thms dt_info));
berghofe@13710
   319
    val (_, constrss) = foldl_map (fn ((recs, dummies), (s, rs)) =>
berghofe@13710
   320
      if s mem rsets then
berghofe@13710
   321
        let
berghofe@13710
   322
          val (d :: dummies') = dummies;
wenzelm@19473
   323
          val (recs1, recs2) = chop (length rs) (if d then tl recs else recs)
berghofe@13710
   324
        in ((recs2, dummies'), map (head_of o hd o rev o snd o strip_comb o
berghofe@13710
   325
          fst o HOLogic.dest_eq o HOLogic.dest_Trueprop o prop_of) recs1)
berghofe@13710
   326
        end
berghofe@13710
   327
      else ((recs, dummies), replicate (length rs) Extraction.nullt))
berghofe@13710
   328
        ((get #rec_thms dt_info, dummies), rss);
wenzelm@18929
   329
    val rintrs = map (fn (intr, c) => Envir.eta_contract
berghofe@13710
   330
      (Extraction.realizes_of thy2 vs
berghofe@22271
   331
        (if c = Extraction.nullt then c else list_comb (c, map Var (rev
berghofe@22271
   332
          (Term.add_vars (prop_of intr) []) \\ params'))) (prop_of intr)))
berghofe@22271
   333
            (maps snd rss ~~ List.concat constrss);
berghofe@22271
   334
    val (rlzpreds, rlzpreds') = split_list
berghofe@22271
   335
      (distinct (op = o pairself (#1 o #1)) (map (fn rintr =>
berghofe@22271
   336
        let
berghofe@22271
   337
          val Const (s, T) = head_of (HOLogic.dest_Trueprop
berghofe@22271
   338
            (Logic.strip_assums_concl rintr));
berghofe@22271
   339
          val s' = Sign.base_name s;
berghofe@22271
   340
          val T' = Logic.unvarifyT T
berghofe@22271
   341
        in ((s', SOME T', NoSyn),
berghofe@22271
   342
          (Const (s, T'), Free (s', T')))
berghofe@22271
   343
        end) rintrs));
berghofe@22271
   344
    val rlzparams = map (fn Var ((s, _), T) => (s, SOME (Logic.unvarifyT T)))
berghofe@22271
   345
      (List.take (snd (strip_comb
berghofe@22271
   346
        (HOLogic.dest_Trueprop (Logic.strip_assums_concl (hd rintrs)))), nparms));
berghofe@13710
   347
berghofe@13710
   348
    (** realizability predicate **)
berghofe@13710
   349
berghofe@22271
   350
    val (ind_info, thy3') = thy2 |>
berghofe@22606
   351
      InductivePackage.add_inductive_global false "" false false false
berghofe@22271
   352
        rlzpreds rlzparams (map (fn (rintr, intr) =>
berghofe@22271
   353
          ((Sign.base_name (name_of_thm intr), []),
berghofe@22271
   354
           subst_atomic rlzpreds' (Logic.unvarify rintr)))
berghofe@22271
   355
             (rintrs ~~ maps snd rss)) [] ||>
berghofe@13710
   356
      Theory.absolute_path;
berghofe@13710
   357
    val thy3 = PureThy.hide_thms false
berghofe@22271
   358
      (map name_of_thm (#intrs ind_info)) thy3';
berghofe@13710
   359
berghofe@13710
   360
    (** realizer for induction rule **)
berghofe@13710
   361
berghofe@22271
   362
    val Ps = List.mapPartial (fn _ $ M $ P => if pred_of M mem rsets then
skalberg@15531
   363
      SOME (fst (fst (dest_Var (head_of P)))) else NONE)
berghofe@13710
   364
        (HOLogic.dest_conj (HOLogic.dest_Trueprop (concl_of raw_induct)));
berghofe@13710
   365
berghofe@13710
   366
    fun add_ind_realizer (thy, Ps) =
berghofe@13710
   367
      let
berghofe@22271
   368
        val rs = indrule_realizer thy induct raw_induct rsets params'
berghofe@22271
   369
          (vs @ Ps) rec_names rss' intrs dummies;
berghofe@22271
   370
        val rlzs = map (fn (r, ind) => Extraction.realizes_of thy (vs @ Ps) r
berghofe@22271
   371
          (prop_of ind)) (rs ~~ inducts);
berghofe@22271
   372
        val used = foldr add_term_free_names [] rlzs;
berghofe@22271
   373
        val rnames = Name.variant_list used (replicate (length inducts) "r");
berghofe@22271
   374
        val rnames' = Name.variant_list
berghofe@22271
   375
          (used @ rnames) (replicate (length intrs) "s");
berghofe@22271
   376
        val rlzs' as (prems, _, _) :: _ = map (fn (rlz, name) =>
berghofe@22271
   377
          let
berghofe@22271
   378
            val (P, Q) = strip_one name (Logic.unvarify rlz);
berghofe@22271
   379
            val Q' = strip_all' [] rnames' Q
berghofe@22271
   380
          in
berghofe@22271
   381
            (Logic.strip_imp_prems Q', P, Logic.strip_imp_concl Q')
berghofe@22271
   382
          end) (rlzs ~~ rnames);
berghofe@22271
   383
        val concl = HOLogic.mk_Trueprop (foldr1 HOLogic.mk_conj (map
berghofe@22271
   384
          (fn (_, _ $ P, _ $ Q) => HOLogic.mk_imp (P, Q)) rlzs'));
berghofe@13710
   385
        val rews = map mk_meta_eq
berghofe@13710
   386
          (fst_conv :: snd_conv :: get #rec_thms dt_info);
berghofe@22271
   387
        val thm = Goal.prove_global thy [] prems concl (fn prems => EVERY
berghofe@22271
   388
          [rtac (#raw_induct ind_info) 1,
berghofe@13710
   389
           rewrite_goals_tac rews,
berghofe@13710
   390
           REPEAT ((resolve_tac prems THEN_ALL_NEW EVERY'
wenzelm@23590
   391
             [K (rewrite_goals_tac rews), ObjectLogic.atomize_prems_tac,
berghofe@13710
   392
              DEPTH_SOLVE_1 o FIRST' [atac, etac allE, etac impE]]) 1)]);
haftmann@18358
   393
        val (thm', thy') = PureThy.store_thm ((space_implode "_"
berghofe@22271
   394
          (NameSpace.qualified qualifier "induct" :: vs @ Ps @
berghofe@22271
   395
             ["correctness"]), thm), []) thy;
berghofe@22271
   396
        val thms = map (fn th => zero_var_indexes (rotate_prems ~1 (th RS mp)))
berghofe@22271
   397
          (DatatypeAux.split_conj_thm thm');
berghofe@22271
   398
        val ([thms'], thy'') = PureThy.add_thmss
berghofe@22271
   399
          [((space_implode "_"
berghofe@22271
   400
             (NameSpace.qualified qualifier "inducts" :: vs @ Ps @
berghofe@22271
   401
               ["correctness"]), thms), [])] thy';
berghofe@22271
   402
        val realizers = inducts ~~ thms' ~~ rlzs ~~ rs;
berghofe@13710
   403
      in
berghofe@13710
   404
        Extraction.add_realizers_i
berghofe@22271
   405
          (map (fn (((ind, corr), rlz), r) =>
berghofe@22271
   406
              mk_realizer thy' (vs @ Ps) (Thm.get_name ind, ind, corr, rlz, r))
berghofe@22271
   407
            realizers @ (case realizers of
berghofe@22271
   408
             [(((ind, corr), rlz), r)] =>
berghofe@22271
   409
               [mk_realizer thy' (vs @ Ps) (NameSpace.qualified qualifier "induct",
berghofe@22271
   410
                  ind, corr, rlz, r)]
berghofe@22271
   411
           | _ => [])) thy''
berghofe@13710
   412
      end;
berghofe@13710
   413
berghofe@13710
   414
    (** realizer for elimination rules **)
berghofe@13710
   415
berghofe@13710
   416
    val case_names = map (fst o dest_Const o head_of o fst o HOLogic.dest_eq o
berghofe@13710
   417
      HOLogic.dest_Trueprop o prop_of o hd) (get #case_thms dt_info);
berghofe@13710
   418
berghofe@13921
   419
    fun add_elim_realizer Ps
berghofe@13921
   420
      (((((elim, elimR), intrs), case_thms), case_name), dummy) thy =
berghofe@13710
   421
      let
berghofe@13710
   422
        val (prem :: prems) = prems_of elim;
berghofe@22271
   423
        fun reorder1 (p, (_, intr)) =
skalberg@15570
   424
          Library.foldl (fn (t, ((s, _), T)) => all T $ lambda (Free (s, T)) t)
wenzelm@16861
   425
            (strip_all p, Term.add_vars (prop_of intr) [] \\ params');
berghofe@22271
   426
        fun reorder2 ((ivs, intr), i) =
berghofe@22271
   427
          let val fs = Term.add_vars (prop_of intr) [] \\ params'
skalberg@15570
   428
          in Library.foldl (fn (t, x) => lambda (Var x) t)
berghofe@22271
   429
            (list_comb (Bound (i + length ivs), ivs), fs)
berghofe@13921
   430
          end;
berghofe@13921
   431
        val p = Logic.list_implies
berghofe@13921
   432
          (map reorder1 (prems ~~ intrs) @ [prem], concl_of elim);
berghofe@13710
   433
        val T' = Extraction.etype_of thy (vs @ Ps) [] p;
berghofe@13710
   434
        val T = if dummy then (HOLogic.unitT --> body_type T') --> T' else T';
berghofe@13921
   435
        val Ts = map (Extraction.etype_of thy (vs @ Ps) []) (prems_of elim);
berghofe@13710
   436
        val r = if null Ps then Extraction.nullt
berghofe@13710
   437
          else list_abs (map (pair "x") Ts, list_comb (Const (case_name, T),
berghofe@13710
   438
            (if dummy then
berghofe@13710
   439
               [Abs ("x", HOLogic.unitT, Const ("arbitrary", body_type T))]
berghofe@13710
   440
             else []) @
berghofe@13921
   441
            map reorder2 (intrs ~~ (length prems - 1 downto 0)) @
berghofe@13921
   442
            [Bound (length prems)]));
berghofe@22271
   443
        val rlz = Extraction.realizes_of thy (vs @ Ps) r (prop_of elim);
berghofe@22271
   444
        val rlz' = strip_all (Logic.unvarify rlz);
berghofe@13710
   445
        val rews = map mk_meta_eq case_thms;
berghofe@22271
   446
        val thm = Goal.prove_global thy []
berghofe@22271
   447
          (Logic.strip_imp_prems rlz') (Logic.strip_imp_concl rlz') (fn prems => EVERY
berghofe@13710
   448
          [cut_facts_tac [hd prems] 1,
berghofe@13710
   449
           etac elimR 1,
berghofe@22271
   450
           ALLGOALS (asm_simp_tac HOL_basic_ss),
berghofe@13710
   451
           rewrite_goals_tac rews,
wenzelm@23590
   452
           REPEAT ((resolve_tac prems THEN_ALL_NEW (ObjectLogic.atomize_prems_tac THEN'
berghofe@13710
   453
             DEPTH_SOLVE_1 o FIRST' [atac, etac allE, etac impE])) 1)]);
haftmann@18358
   454
        val (thm', thy') = PureThy.store_thm ((space_implode "_"
berghofe@22271
   455
          (name_of_thm elim :: vs @ Ps @ ["correctness"]), thm), []) thy
berghofe@13710
   456
      in
berghofe@13710
   457
        Extraction.add_realizers_i
berghofe@22271
   458
          [mk_realizer thy' (vs @ Ps) (name_of_thm elim, elim, thm', rlz, r)] thy'
berghofe@13710
   459
      end;
berghofe@13710
   460
berghofe@13710
   461
    (** add realizers to theory **)
berghofe@13710
   462
skalberg@15570
   463
    val thy4 = Library.foldl add_ind_realizer (thy3, subsets Ps);
berghofe@13710
   464
    val thy5 = Extraction.add_realizers_i
berghofe@22271
   465
      (map (mk_realizer thy4 vs) (map (fn (((rule, rrule), rlz), c) =>
berghofe@22271
   466
         (name_of_thm rule, rule, rrule, rlz,
berghofe@22271
   467
            list_comb (c, map Var (rev (Term.add_vars (prop_of rule) []) \\ params'))))
berghofe@22271
   468
              (List.concat (map snd rss) ~~ #intrs ind_info ~~ rintrs ~~
berghofe@22271
   469
                 List.concat constrss))) thy4;
berghofe@22271
   470
    val elimps = List.mapPartial (fn ((s, intrs), p) =>
berghofe@22271
   471
      if s mem rsets then SOME (p, intrs) else NONE)
berghofe@22271
   472
        (rss' ~~ (elims ~~ #elims ind_info));
skalberg@15570
   473
    val thy6 = Library.foldl (fn (thy, p as (((((elim, _), _), _), _), _)) => thy |>
berghofe@13710
   474
      add_elim_realizer [] p |> add_elim_realizer [fst (fst (dest_Var
berghofe@13710
   475
        (HOLogic.dest_Trueprop (concl_of elim))))] p) (thy5,
berghofe@13710
   476
           elimps ~~ get #case_thms dt_info ~~ case_names ~~ dummies)
berghofe@13710
   477
wenzelm@16123
   478
  in Theory.restore_naming thy thy6 end;
berghofe@13710
   479
berghofe@13710
   480
fun add_ind_realizers name rsets thy =
berghofe@13710
   481
  let
berghofe@13710
   482
    val (_, {intrs, induct, raw_induct, elims, ...}) =
berghofe@22271
   483
      InductivePackage.the_inductive (ProofContext.init thy) name;
berghofe@13710
   484
    val vss = sort (int_ord o pairself length)
berghofe@22271
   485
      (subsets (map fst (relevant_vars (concl_of (hd intrs)))))
berghofe@13710
   486
  in
skalberg@15570
   487
    Library.foldl (add_ind_realizer rsets intrs induct raw_induct elims) (thy, vss)
berghofe@13710
   488
  end
berghofe@13710
   489
wenzelm@20897
   490
fun rlz_attrib arg = Thm.declaration_attribute (fn thm => Context.mapping
berghofe@13710
   491
  let
berghofe@13710
   492
    fun err () = error "ind_realizer: bad rule";
berghofe@13710
   493
    val sets =
berghofe@13710
   494
      (case HOLogic.dest_conj (HOLogic.dest_Trueprop (concl_of thm)) of
berghofe@22271
   495
           [_] => [pred_of (HOLogic.dest_Trueprop (hd (prems_of thm)))]
berghofe@22271
   496
         | xs => map (pred_of o fst o HOLogic.dest_imp) xs)
skalberg@15570
   497
         handle TERM _ => err () | Empty => err ();
berghofe@13710
   498
  in 
wenzelm@18728
   499
    add_ind_realizers (hd sets)
wenzelm@18728
   500
      (case arg of
skalberg@15531
   501
        NONE => sets | SOME NONE => []
wenzelm@15703
   502
      | SOME (SOME sets') => sets \\ sets')
wenzelm@20897
   503
  end I);
berghofe@13710
   504
wenzelm@18728
   505
val ind_realizer = Attrib.syntax
wenzelm@15703
   506
 ((Scan.option (Scan.lift (Args.$$$ "irrelevant") |--
wenzelm@15703
   507
    Scan.option (Scan.lift (Args.colon) |--
wenzelm@18728
   508
      Scan.repeat1 Args.const))) >> rlz_attrib);
berghofe@13710
   509
wenzelm@18708
   510
val setup =
wenzelm@18728
   511
  Attrib.add_attributes [("ind_realizer", ind_realizer, "add realizers for inductive set")];
berghofe@13710
   512
berghofe@13710
   513
end;
wenzelm@15706
   514