src/Tools/eqsubst.ML
author wenzelm
Fri Mar 13 21:25:15 2009 +0100 (2009-03-13)
changeset 30513 1796b8ea88aa
parent 30510 4120fc59dd85
child 31301 952d2d0c4446
permissions -rw-r--r--
eliminated type Args.T;
pervasive types 'a parser and 'a context_parser;
wenzelm@30160
     1
(*  Title:      Tools/eqsubst.ML
wenzelm@29269
     2
    Author:     Lucas Dixon, University of Edinburgh
paulson@15481
     3
wenzelm@18598
     4
A proof method to perform a substiution using an equation.
wenzelm@18598
     5
*)
dixon@15538
     6
wenzelm@18591
     7
signature EQSUBST =
paulson@15481
     8
sig
wenzelm@29269
     9
  (* a type abbreviation for match information *)
dixon@19871
    10
  type match =
dixon@19871
    11
       ((indexname * (sort * typ)) list (* type instantiations *)
dixon@19871
    12
        * (indexname * (typ * term)) list) (* term instantiations *)
dixon@19871
    13
       * (string * typ) list (* fake named type abs env *)
dixon@19871
    14
       * (string * typ) list (* type abs env *)
dixon@19871
    15
       * term (* outer term *)
dixon@19871
    16
dixon@19871
    17
  type searchinfo =
dixon@19871
    18
       theory
dixon@19871
    19
       * int (* maxidx *)
dixon@19871
    20
       * Zipper.T (* focusterm to search under *)
dixon@19871
    21
dixon@19871
    22
    exception eqsubst_occL_exp of
dixon@19871
    23
       string * int list * Thm.thm list * int * Thm.thm
dixon@19871
    24
    
dixon@19871
    25
    (* low level substitution functions *)
dixon@19871
    26
    val apply_subst_in_asm :
dixon@19871
    27
       int ->
dixon@19871
    28
       Thm.thm ->
dixon@19871
    29
       Thm.thm ->
dixon@19871
    30
       (Thm.cterm list * int * 'a * Thm.thm) * match -> Thm.thm Seq.seq
dixon@19871
    31
    val apply_subst_in_concl :
dixon@19871
    32
       int ->
dixon@19871
    33
       Thm.thm ->
dixon@19871
    34
       Thm.cterm list * Thm.thm ->
dixon@19871
    35
       Thm.thm -> match -> Thm.thm Seq.seq
dixon@19871
    36
dixon@19871
    37
    (* matching/unification within zippers *)
dixon@19871
    38
    val clean_match_z :
dixon@19871
    39
       Context.theory -> Term.term -> Zipper.T -> match option
dixon@19871
    40
    val clean_unify_z :
dixon@19871
    41
       Context.theory -> int -> Term.term -> Zipper.T -> match Seq.seq
dixon@19871
    42
dixon@19871
    43
    (* skipping things in seq seq's *)
dixon@19871
    44
dixon@19871
    45
   (* skipping non-empty sub-sequences but when we reach the end
dixon@19871
    46
      of the seq, remembering how much we have left to skip. *)
dixon@19871
    47
    datatype 'a skipseq = SkipMore of int
dixon@19871
    48
      | SkipSeq of 'a Seq.seq Seq.seq;
dixon@19871
    49
dixon@19871
    50
    val skip_first_asm_occs_search :
dixon@19871
    51
       ('a -> 'b -> 'c Seq.seq Seq.seq) ->
dixon@19871
    52
       'a -> int -> 'b -> 'c skipseq
dixon@19871
    53
    val skip_first_occs_search :
dixon@19871
    54
       int -> ('a -> 'b -> 'c Seq.seq Seq.seq) -> 'a -> 'b -> 'c Seq.seq
dixon@19871
    55
    val skipto_skipseq : int -> 'a Seq.seq Seq.seq -> 'a skipseq
dixon@19871
    56
dixon@19871
    57
    (* tactics *)
dixon@19871
    58
    val eqsubst_asm_tac :
dixon@19871
    59
       Proof.context ->
dixon@19871
    60
       int list -> Thm.thm list -> int -> Thm.thm -> Thm.thm Seq.seq
dixon@19871
    61
    val eqsubst_asm_tac' :
dixon@19871
    62
       Proof.context ->
dixon@19871
    63
       (searchinfo -> int -> Term.term -> match skipseq) ->
dixon@19871
    64
       int -> Thm.thm -> int -> Thm.thm -> Thm.thm Seq.seq
dixon@19871
    65
    val eqsubst_tac :
dixon@19871
    66
       Proof.context ->
dixon@22727
    67
       int list -> (* list of occurences to rewrite, use [0] for any *)
dixon@22727
    68
       Thm.thm list -> int -> Thm.thm -> Thm.thm Seq.seq
dixon@19871
    69
    val eqsubst_tac' :
dixon@22727
    70
       Proof.context -> (* proof context *)
dixon@22727
    71
       (searchinfo -> Term.term -> match Seq.seq) (* search function *)
dixon@22727
    72
       -> Thm.thm (* equation theorem to rewrite with *)
dixon@22727
    73
       -> int (* subgoal number in goal theorem *)
dixon@22727
    74
       -> Thm.thm (* goal theorem *)
dixon@22727
    75
       -> Thm.thm Seq.seq (* rewritten goal theorem *)
dixon@19871
    76
dixon@19871
    77
dixon@19871
    78
    val fakefree_badbounds :
dixon@19871
    79
       (string * Term.typ) list ->
dixon@19871
    80
       Term.term ->
dixon@19871
    81
       (string * Term.typ) list * (string * Term.typ) list * Term.term
dixon@19871
    82
dixon@19871
    83
    val mk_foo_match :
dixon@19871
    84
       (Term.term -> Term.term) ->
dixon@19871
    85
       ('a * Term.typ) list -> Term.term -> Term.term
dixon@19871
    86
dixon@19871
    87
    (* preparing substitution *)
dixon@19871
    88
    val prep_meta_eq : Proof.context -> Thm.thm -> Thm.thm list
dixon@19871
    89
    val prep_concl_subst :
dixon@19871
    90
       int -> Thm.thm -> (Thm.cterm list * Thm.thm) * searchinfo
dixon@19871
    91
    val prep_subst_in_asm :
dixon@19871
    92
       int -> Thm.thm -> int ->
dixon@19871
    93
       (Thm.cterm list * int * int * Thm.thm) * searchinfo
dixon@19871
    94
    val prep_subst_in_asms :
dixon@19871
    95
       int -> Thm.thm ->
dixon@19871
    96
       ((Thm.cterm list * int * int * Thm.thm) * searchinfo) list
dixon@19871
    97
    val prep_zipper_match :
dixon@19871
    98
       Zipper.T -> Term.term * ((string * Term.typ) list * (string * Term.typ) list * Term.term)
dixon@19871
    99
dixon@19871
   100
    (* search for substitutions *)
dixon@19871
   101
    val valid_match_start : Zipper.T -> bool
dixon@19871
   102
    val search_lr_all : Zipper.T -> Zipper.T Seq.seq
dixon@19871
   103
    val search_lr_valid : (Zipper.T -> bool) -> Zipper.T -> Zipper.T Seq.seq
dixon@19871
   104
    val searchf_lr_unify_all :
dixon@19871
   105
       searchinfo -> Term.term -> match Seq.seq Seq.seq
dixon@19871
   106
    val searchf_lr_unify_valid :
dixon@19871
   107
       searchinfo -> Term.term -> match Seq.seq Seq.seq
narboux@23064
   108
    val searchf_bt_unify_valid :
narboux@23064
   109
       searchinfo -> Term.term -> match Seq.seq Seq.seq
dixon@19871
   110
dixon@19871
   111
    (* syntax tools *)
wenzelm@30513
   112
    val ith_syntax : int list parser
wenzelm@30513
   113
    val options_syntax : bool parser
dixon@19871
   114
dixon@19871
   115
    (* Isar level hooks *)
wenzelm@20289
   116
    val eqsubst_asm_meth : Proof.context -> int list -> Thm.thm list -> Proof.method
wenzelm@20289
   117
    val eqsubst_meth : Proof.context -> int list -> Thm.thm list -> Proof.method
wenzelm@20289
   118
    val subst_meth : Method.src -> Proof.context -> Proof.method
dixon@19871
   119
    val setup : theory -> theory
dixon@19871
   120
paulson@15481
   121
end;
paulson@15481
   122
dixon@19835
   123
structure EqSubst
dixon@19871
   124
: EQSUBST
dixon@19835
   125
= struct
dixon@16004
   126
dixon@19835
   127
structure Z = Zipper;
dixon@19835
   128
dixon@19835
   129
(* changes object "=" to meta "==" which prepares a given rewrite rule *)
wenzelm@18598
   130
fun prep_meta_eq ctxt =
wenzelm@30318
   131
  Simplifier.mksimps (Simplifier.local_simpset_of ctxt) #> map Drule.zero_var_indexes;
wenzelm@18598
   132
paulson@15481
   133
dixon@15915
   134
  (* a type abriviation for match information *)
wenzelm@16978
   135
  type match =
wenzelm@16978
   136
       ((indexname * (sort * typ)) list (* type instantiations *)
wenzelm@16978
   137
        * (indexname * (typ * term)) list) (* term instantiations *)
wenzelm@16978
   138
       * (string * typ) list (* fake named type abs env *)
wenzelm@16978
   139
       * (string * typ) list (* type abs env *)
wenzelm@16978
   140
       * term (* outer term *)
dixon@15550
   141
wenzelm@16978
   142
  type searchinfo =
wenzelm@18598
   143
       theory
dixon@16004
   144
       * int (* maxidx *)
dixon@19835
   145
       * Zipper.T (* focusterm to search under *)
dixon@19835
   146
dixon@19835
   147
dixon@19835
   148
(* skipping non-empty sub-sequences but when we reach the end
dixon@19835
   149
   of the seq, remembering how much we have left to skip. *)
dixon@19835
   150
datatype 'a skipseq = SkipMore of int
dixon@19835
   151
  | SkipSeq of 'a Seq.seq Seq.seq;
dixon@19835
   152
(* given a seqseq, skip the first m non-empty seq's, note deficit *)
dixon@19835
   153
fun skipto_skipseq m s = 
dixon@19835
   154
    let 
dixon@19835
   155
      fun skip_occs n sq = 
dixon@19835
   156
          case Seq.pull sq of 
dixon@19835
   157
            NONE => SkipMore n
dixon@19835
   158
          | SOME (h,t) => 
dixon@19835
   159
            (case Seq.pull h of NONE => skip_occs n t
dixon@19835
   160
             | SOME _ => if n <= 1 then SkipSeq (Seq.cons h t)
dixon@19835
   161
                         else skip_occs (n - 1) t)
dixon@19835
   162
    in (skip_occs m s) end;
dixon@19835
   163
dixon@19835
   164
(* note: outerterm is the taget with the match replaced by a bound 
dixon@19835
   165
         variable : ie: "P lhs" beocmes "%x. P x" 
dixon@19835
   166
         insts is the types of instantiations of vars in lhs
dixon@19835
   167
         and typinsts is the type instantiations of types in the lhs
dixon@19835
   168
         Note: Final rule is the rule lifted into the ontext of the 
dixon@19835
   169
         taget thm. *)
dixon@19835
   170
fun mk_foo_match mkuptermfunc Ts t = 
dixon@19835
   171
    let 
dixon@19835
   172
      val ty = Term.type_of t
dixon@19835
   173
      val bigtype = (rev (map snd Ts)) ---> ty
dixon@19835
   174
      fun mk_foo 0 t = t
dixon@19835
   175
        | mk_foo i t = mk_foo (i - 1) (t $ (Bound (i - 1)))
dixon@19835
   176
      val num_of_bnds = (length Ts)
dixon@19835
   177
      (* foo_term = "fooabs y0 ... yn" where y's are local bounds *)
dixon@19835
   178
      val foo_term = mk_foo num_of_bnds (Bound num_of_bnds)
dixon@19835
   179
    in Abs("fooabs", bigtype, mkuptermfunc foo_term) end;
dixon@19835
   180
dixon@19835
   181
(* T is outer bound vars, n is number of locally bound vars *)
dixon@19835
   182
(* THINK: is order of Ts correct...? or reversed? *)
dixon@19835
   183
fun fakefree_badbounds Ts t = 
dixon@19835
   184
    let val (FakeTs,Ts,newnames) = 
dixon@19835
   185
            List.foldr (fn ((n,ty),(FakeTs,Ts,usednames)) => 
wenzelm@20071
   186
                           let val newname = Name.variant usednames n
dixon@19835
   187
                           in ((RWTools.mk_fake_bound_name newname,ty)::FakeTs,
dixon@19835
   188
                               (newname,ty)::Ts, 
dixon@19835
   189
                               newname::usednames) end)
dixon@19835
   190
                       ([],[],[])
dixon@19835
   191
                       Ts
dixon@19835
   192
    in (FakeTs, Ts, Term.subst_bounds (map Free FakeTs, t)) end;
dixon@19835
   193
dixon@19835
   194
(* before matching we need to fake the bound vars that are missing an
dixon@19835
   195
abstraction. In this function we additionally construct the
dixon@19835
   196
abstraction environment, and an outer context term (with the focus
dixon@19835
   197
abstracted out) for use in rewriting with RWInst.rw *)
dixon@19835
   198
fun prep_zipper_match z = 
dixon@19835
   199
    let 
dixon@19835
   200
      val t = Z.trm z  
dixon@19835
   201
      val c = Z.ctxt z
dixon@19835
   202
      val Ts = Z.C.nty_ctxt c
dixon@19835
   203
      val (FakeTs', Ts', t') = fakefree_badbounds Ts t
dixon@19835
   204
      val absterm = mk_foo_match (Z.C.apply c) Ts' t'
dixon@19835
   205
    in
dixon@19835
   206
      (t', (FakeTs', Ts', absterm))
dixon@19835
   207
    end;
dixon@19835
   208
dixon@19835
   209
(* Matching and Unification with exception handled *)
dixon@19835
   210
fun clean_match thy (a as (pat, t)) =
dixon@19835
   211
  let val (tyenv, tenv) = Pattern.match thy a (Vartab.empty, Vartab.empty)
dixon@19835
   212
  in SOME (Vartab.dest tyenv, Vartab.dest tenv)
dixon@19835
   213
  end handle Pattern.MATCH => NONE;
dixon@27033
   214
dixon@19835
   215
(* given theory, max var index, pat, tgt; returns Seq of instantiations *)
dixon@27033
   216
fun clean_unify thry ix (a as (pat, tgt)) =
dixon@19835
   217
    let
dixon@19835
   218
      (* type info will be re-derived, maybe this can be cached
dixon@19835
   219
         for efficiency? *)
dixon@19835
   220
      val pat_ty = Term.type_of pat;
dixon@19835
   221
      val tgt_ty = Term.type_of tgt;
dixon@19835
   222
      (* is it OK to ignore the type instantiation info?
dixon@19835
   223
         or should I be using it? *)
dixon@19835
   224
      val typs_unify =
wenzelm@29269
   225
          SOME (Sign.typ_unify thry (pat_ty, tgt_ty) (Vartab.empty, ix))
dixon@19835
   226
            handle Type.TUNIFY => NONE;
dixon@19835
   227
    in
dixon@19835
   228
      case typs_unify of
dixon@19835
   229
        SOME (typinsttab, ix2) =>
dixon@19835
   230
        let
dixon@19835
   231
      (* is it right to throw away the flexes?
dixon@19835
   232
         or should I be using them somehow? *)
dixon@19835
   233
          fun mk_insts env =
dixon@19835
   234
            (Vartab.dest (Envir.type_env env),
dixon@19835
   235
             Envir.alist_of env);
dixon@19835
   236
          val initenv = Envir.Envir {asol = Vartab.empty,
dixon@19835
   237
                                     iTs = typinsttab, maxidx = ix2};
dixon@27033
   238
          val useq = Unify.smash_unifiers thry [a] initenv
dixon@19835
   239
	            handle UnequalLengths => Seq.empty
dixon@19835
   240
		               | Term.TERM _ => Seq.empty;
dixon@19835
   241
          fun clean_unify' useq () =
dixon@19835
   242
              (case (Seq.pull useq) of
dixon@19835
   243
                 NONE => NONE
dixon@19835
   244
               | SOME (h,t) => SOME (mk_insts h, Seq.make (clean_unify' t)))
dixon@27033
   245
	            handle UnequalLengths => NONE
dixon@27033
   246
                   | Term.TERM _ => NONE
dixon@19835
   247
        in
dixon@19835
   248
          (Seq.make (clean_unify' useq))
dixon@19835
   249
        end
dixon@19835
   250
      | NONE => Seq.empty
dixon@19835
   251
    end;
dixon@19835
   252
dixon@19835
   253
(* Matching and Unification for zippers *)
dixon@19835
   254
(* Note: Ts is a modified version of the original names of the outer
dixon@19835
   255
bound variables. New names have been introduced to make sure they are
dixon@19835
   256
unique w.r.t all names in the term and each other. usednames' is
dixon@19835
   257
oldnames + new names. *)
dixon@19835
   258
fun clean_match_z thy pat z = 
dixon@19835
   259
    let val (t, (FakeTs,Ts,absterm)) = prep_zipper_match z in
dixon@19835
   260
      case clean_match thy (pat, t) of 
dixon@19835
   261
        NONE => NONE 
dixon@19835
   262
      | SOME insts => SOME (insts, FakeTs, Ts, absterm) end;
dixon@19835
   263
(* ix = max var index *)
dixon@19835
   264
fun clean_unify_z sgn ix pat z = 
dixon@19835
   265
    let val (t, (FakeTs, Ts,absterm)) = prep_zipper_match z in
dixon@19835
   266
    Seq.map (fn insts => (insts, FakeTs, Ts, absterm)) 
dixon@19835
   267
            (clean_unify sgn ix (t, pat)) end;
dixon@19835
   268
dixon@15550
   269
dixon@15538
   270
(* FOR DEBUGGING...
dixon@15538
   271
type trace_subst_errT = int (* subgoal *)
wenzelm@16978
   272
        * thm (* thm with all goals *)
dixon@15538
   273
        * (Thm.cterm list (* certified free var placeholders for vars *)
wenzelm@16978
   274
           * thm)  (* trivial thm of goal concl *)
dixon@15538
   275
            (* possible matches/unifiers *)
wenzelm@16978
   276
        * thm (* rule *)
wenzelm@16978
   277
        * (((indexname * typ) list (* type instantiations *)
wenzelm@16978
   278
              * (indexname * term) list ) (* term instantiations *)
wenzelm@16978
   279
             * (string * typ) list (* Type abs env *)
wenzelm@16978
   280
             * term) (* outer term *);
dixon@15538
   281
dixon@15538
   282
val trace_subst_err = (ref NONE : trace_subst_errT option ref);
dixon@15538
   283
val trace_subst_search = ref false;
dixon@15538
   284
exception trace_subst_exp of trace_subst_errT;
dixon@19835
   285
*)
dixon@19835
   286
dixon@19835
   287
dixon@19835
   288
fun bot_left_leaf_of (l $ r) = bot_left_leaf_of l
dixon@19835
   289
  | bot_left_leaf_of (Abs(s,ty,t)) = bot_left_leaf_of t
dixon@19835
   290
  | bot_left_leaf_of x = x;
dixon@15538
   291
dixon@19975
   292
(* Avoid considering replacing terms which have a var at the head as
dixon@19975
   293
   they always succeed trivially, and uninterestingly. *)
dixon@19835
   294
fun valid_match_start z =
dixon@19835
   295
    (case bot_left_leaf_of (Z.trm z) of 
dixon@19975
   296
      Var _ => false 
dixon@19975
   297
      | _ => true);
dixon@19975
   298
dixon@15814
   299
(* search from top, left to right, then down *)
dixon@19871
   300
val search_lr_all = ZipperSearch.all_bl_ur;
paulson@15481
   301
dixon@15814
   302
(* search from top, left to right, then down *)
dixon@19871
   303
fun search_lr_valid validf =
dixon@19835
   304
    let 
dixon@19835
   305
      fun sf_valid_td_lr z = 
dixon@19835
   306
          let val here = if validf z then [Z.Here z] else [] in
dixon@19835
   307
            case Z.trm z 
dixon@19871
   308
             of _ $ _ => [Z.LookIn (Z.move_down_left z)] 
dixon@19871
   309
                         @ here 
dixon@19871
   310
                         @ [Z.LookIn (Z.move_down_right z)]
dixon@19835
   311
              | Abs _ => here @ [Z.LookIn (Z.move_down_abs z)]
dixon@19835
   312
              | _ => here
dixon@19835
   313
          end;
dixon@19835
   314
    in Z.lzy_search sf_valid_td_lr end;
dixon@15814
   315
narboux@23064
   316
(* search from bottom to top, left to right *)
narboux@23064
   317
narboux@23064
   318
fun search_bt_valid validf =
narboux@23064
   319
    let 
narboux@23064
   320
      fun sf_valid_td_lr z = 
narboux@23064
   321
          let val here = if validf z then [Z.Here z] else [] in
narboux@23064
   322
            case Z.trm z 
narboux@23064
   323
             of _ $ _ => [Z.LookIn (Z.move_down_left z), 
narboux@23064
   324
                          Z.LookIn (Z.move_down_right z)] @ here
narboux@23064
   325
              | Abs _ => [Z.LookIn (Z.move_down_abs z)] @ here
narboux@23064
   326
              | _ => here
narboux@23064
   327
          end;
narboux@23064
   328
    in Z.lzy_search sf_valid_td_lr end;
narboux@23064
   329
narboux@23064
   330
fun searchf_unify_gen f (sgn, maxidx, z) lhs =
narboux@23064
   331
    Seq.map (clean_unify_z sgn maxidx lhs) 
narboux@23064
   332
            (Z.limit_apply f z);
narboux@23064
   333
dixon@15814
   334
(* search all unifications *)
narboux@23064
   335
val searchf_lr_unify_all =
narboux@23064
   336
    searchf_unify_gen search_lr_all;
paulson@15481
   337
dixon@15814
   338
(* search only for 'valid' unifiers (non abs subterms and non vars) *)
narboux@23064
   339
val searchf_lr_unify_valid = 
narboux@23064
   340
    searchf_unify_gen (search_lr_valid valid_match_start);
dixon@15929
   341
narboux@23064
   342
val searchf_bt_unify_valid =
narboux@23064
   343
    searchf_unify_gen (search_bt_valid valid_match_start);
dixon@15814
   344
dixon@15538
   345
(* apply a substitution in the conclusion of the theorem th *)
dixon@15538
   346
(* cfvs are certified free var placeholders for goal params *)
dixon@15538
   347
(* conclthm is a theorem of for just the conclusion *)
dixon@15538
   348
(* m is instantiation/match information *)
dixon@15538
   349
(* rule is the equation for substitution *)
wenzelm@16978
   350
fun apply_subst_in_concl i th (cfvs, conclthm) rule m =
dixon@15538
   351
    (RWInst.rw m rule conclthm)
dixon@15855
   352
      |> IsaND.unfix_frees cfvs
dixon@15915
   353
      |> RWInst.beta_eta_contract
dixon@15538
   354
      |> (fn r => Tactic.rtac r i th);
paulson@15481
   355
paulson@15481
   356
(* substitute within the conclusion of goal i of gth, using a meta
dixon@15538
   357
equation rule. Note that we assume rule has var indicies zero'd *)
wenzelm@16978
   358
fun prep_concl_subst i gth =
wenzelm@16978
   359
    let
paulson@15481
   360
      val th = Thm.incr_indexes 1 gth;
paulson@15481
   361
      val tgt_term = Thm.prop_of th;
paulson@15481
   362
wenzelm@22578
   363
      val sgn = Thm.theory_of_thm th;
paulson@15481
   364
      val ctermify = Thm.cterm_of sgn;
paulson@15481
   365
      val trivify = Thm.trivial o ctermify;
paulson@15481
   366
paulson@15481
   367
      val (fixedbody, fvs) = IsaND.fix_alls_term i tgt_term;
paulson@15481
   368
      val cfvs = rev (map ctermify fvs);
paulson@15481
   369
dixon@15538
   370
      val conclterm = Logic.strip_imp_concl fixedbody;
dixon@15538
   371
      val conclthm = trivify conclterm;
dixon@27033
   372
      val maxidx = Thm.maxidx_of th;
dixon@19835
   373
      val ft = ((Z.move_down_right (* ==> *)
dixon@19835
   374
                 o Z.move_down_left (* Trueprop *)
dixon@19835
   375
                 o Z.mktop
dixon@16004
   376
                 o Thm.prop_of) conclthm)
paulson@15481
   377
    in
dixon@16004
   378
      ((cfvs, conclthm), (sgn, maxidx, ft))
paulson@15481
   379
    end;
paulson@15481
   380
paulson@15481
   381
(* substitute using an object or meta level equality *)
wenzelm@18598
   382
fun eqsubst_tac' ctxt searchf instepthm i th =
wenzelm@16978
   383
    let
dixon@16004
   384
      val (cvfsconclthm, searchinfo) = prep_concl_subst i th;
wenzelm@18598
   385
      val stepthms = Seq.of_list (prep_meta_eq ctxt instepthm);
dixon@15538
   386
      fun rewrite_with_thm r =
dixon@15538
   387
          let val (lhs,_) = Logic.dest_equals (Thm.concl_of r);
wenzelm@18598
   388
          in searchf searchinfo lhs
wenzelm@18598
   389
             |> Seq.maps (apply_subst_in_concl i th cvfsconclthm r) end;
wenzelm@18598
   390
    in stepthms |> Seq.maps rewrite_with_thm end;
dixon@15538
   391
dixon@15538
   392
wenzelm@19047
   393
(* distinct subgoals *)
dixon@15959
   394
fun distinct_subgoals th =
wenzelm@19047
   395
  the_default th (SINGLE distinct_subgoals_tac th);
dixon@15538
   396
wenzelm@19047
   397
(* General substitution of multiple occurances using one of
dixon@15936
   398
   the given theorems*)
dixon@19835
   399
dixon@19835
   400
wenzelm@16978
   401
exception eqsubst_occL_exp of
wenzelm@16978
   402
          string * (int list) * (thm list) * int * thm;
wenzelm@16978
   403
fun skip_first_occs_search occ srchf sinfo lhs =
dixon@19835
   404
    case (skipto_skipseq occ (srchf sinfo lhs)) of
dixon@19835
   405
      SkipMore _ => Seq.empty
dixon@19835
   406
    | SkipSeq ss => Seq.flat ss;
dixon@16004
   407
dixon@22727
   408
(* The occL is a list of integers indicating which occurence
dixon@22727
   409
w.r.t. the search order, to rewrite. Backtracking will also find later
dixon@22727
   410
occurences, but all earlier ones are skipped. Thus you can use [0] to
dixon@22727
   411
just find all rewrites. *)
dixon@22727
   412
wenzelm@18598
   413
fun eqsubst_tac ctxt occL thms i th =
dixon@15936
   414
    let val nprems = Thm.nprems_of th in
dixon@15936
   415
      if nprems < i then Seq.empty else
wenzelm@16978
   416
      let val thmseq = (Seq.of_list thms)
wenzelm@16978
   417
        fun apply_occ occ th =
wenzelm@18598
   418
            thmseq |> Seq.maps
dixon@19835
   419
                    (fn r => eqsubst_tac' 
dixon@19835
   420
                               ctxt 
dixon@19835
   421
                               (skip_first_occs_search
dixon@19871
   422
                                  occ searchf_lr_unify_valid) r
dixon@15936
   423
                                 (i + ((Thm.nprems_of th) - nprems))
dixon@15936
   424
                                 th);
wenzelm@16978
   425
        val sortedoccL =
dixon@16004
   426
            Library.sort (Library.rev_order o Library.int_ord) occL;
dixon@15936
   427
      in
dixon@16004
   428
        Seq.map distinct_subgoals (Seq.EVERY (map apply_occ sortedoccL) th)
dixon@15936
   429
      end
dixon@15959
   430
    end
dixon@15959
   431
    handle THM _ => raise eqsubst_occL_exp ("THM",occL,thms,i,th);
dixon@15959
   432
paulson@15481
   433
paulson@15481
   434
(* inthms are the given arguments in Isar, and treated as eqstep with
paulson@15481
   435
   the first one, then the second etc *)
wenzelm@18598
   436
fun eqsubst_meth ctxt occL inthms =
wenzelm@30510
   437
    SIMPLE_METHOD' (eqsubst_tac ctxt occL inthms);
paulson@15481
   438
dixon@16004
   439
(* apply a substitution inside assumption j, keeps asm in the same place *)
wenzelm@16978
   440
fun apply_subst_in_asm i th rule ((cfvs, j, ngoalprems, pth),m) =
wenzelm@16978
   441
    let
dixon@16004
   442
      val th2 = Thm.rotate_rule (j - 1) i th; (* put premice first *)
wenzelm@16978
   443
      val preelimrule =
dixon@16004
   444
          (RWInst.rw m rule pth)
wenzelm@21708
   445
            |> (Seq.hd o prune_params_tac)
dixon@16004
   446
            |> Thm.permute_prems 0 ~1 (* put old asm first *)
dixon@16004
   447
            |> IsaND.unfix_frees cfvs (* unfix any global params *)
dixon@16004
   448
            |> RWInst.beta_eta_contract; (* normal form *)
wenzelm@16978
   449
  (*    val elimrule =
dixon@16004
   450
          preelimrule
dixon@16004
   451
            |> Tactic.make_elim (* make into elim rule *)
dixon@16004
   452
            |> Thm.lift_rule (th2, i); (* lift into context *)
dixon@16007
   453
   *)
dixon@16004
   454
    in
dixon@16004
   455
      (* ~j because new asm starts at back, thus we subtract 1 *)
dixon@16007
   456
      Seq.map (Thm.rotate_rule (~j) ((Thm.nprems_of rule) + i))
dixon@16007
   457
      (Tactic.dtac preelimrule i th2)
dixon@16007
   458
wenzelm@16978
   459
      (* (Thm.bicompose
dixon@16004
   460
                 false (* use unification *)
dixon@16004
   461
                 (true, (* elim resolution *)
dixon@16007
   462
                  elimrule, (2 + (Thm.nprems_of rule)) - ngoalprems)
dixon@16007
   463
                 i th2) *)
dixon@16004
   464
    end;
paulson@15481
   465
paulson@15481
   466
dixon@15538
   467
(* prepare to substitute within the j'th premise of subgoal i of gth,
dixon@15538
   468
using a meta-level equation. Note that we assume rule has var indicies
dixon@15538
   469
zero'd. Note that we also assume that premt is the j'th premice of
dixon@15538
   470
subgoal i of gth. Note the repetition of work done for each
dixon@15538
   471
assumption, i.e. this can be made more efficient for search over
dixon@15538
   472
multiple assumptions.  *)
wenzelm@16978
   473
fun prep_subst_in_asm i gth j =
wenzelm@16978
   474
    let
paulson@15481
   475
      val th = Thm.incr_indexes 1 gth;
paulson@15481
   476
      val tgt_term = Thm.prop_of th;
paulson@15481
   477
wenzelm@22578
   478
      val sgn = Thm.theory_of_thm th;
paulson@15481
   479
      val ctermify = Thm.cterm_of sgn;
paulson@15481
   480
      val trivify = Thm.trivial o ctermify;
paulson@15481
   481
paulson@15481
   482
      val (fixedbody, fvs) = IsaND.fix_alls_term i tgt_term;
paulson@15481
   483
      val cfvs = rev (map ctermify fvs);
paulson@15481
   484
haftmann@18011
   485
      val asmt = nth (Logic.strip_imp_prems fixedbody) (j - 1);
dixon@15538
   486
      val asm_nprems = length (Logic.strip_imp_prems asmt);
dixon@15538
   487
dixon@15538
   488
      val pth = trivify asmt;
dixon@27033
   489
      val maxidx = Thm.maxidx_of th;
dixon@15538
   490
dixon@19835
   491
      val ft = ((Z.move_down_right (* trueprop *)
dixon@19835
   492
                 o Z.mktop
dixon@16004
   493
                 o Thm.prop_of) pth)
dixon@16004
   494
    in ((cfvs, j, asm_nprems, pth), (sgn, maxidx, ft)) end;
paulson@15481
   495
dixon@15538
   496
(* prepare subst in every possible assumption *)
wenzelm@16978
   497
fun prep_subst_in_asms i gth =
dixon@16004
   498
    map (prep_subst_in_asm i gth)
dixon@19835
   499
        ((fn l => Library.upto (1, length l))
dixon@16004
   500
           (Logic.prems_of_goal (Thm.prop_of gth) i));
dixon@15538
   501
dixon@15538
   502
dixon@15538
   503
(* substitute in an assumption using an object or meta level equality *)
wenzelm@18598
   504
fun eqsubst_asm_tac' ctxt searchf skipocc instepthm i th =
wenzelm@16978
   505
    let
dixon@16004
   506
      val asmpreps = prep_subst_in_asms i th;
wenzelm@18598
   507
      val stepthms = Seq.of_list (prep_meta_eq ctxt instepthm);
dixon@16004
   508
      fun rewrite_with_thm r =
dixon@16004
   509
          let val (lhs,_) = Logic.dest_equals (Thm.concl_of r)
dixon@16004
   510
            fun occ_search occ [] = Seq.empty
dixon@16004
   511
              | occ_search occ ((asminfo, searchinfo)::moreasms) =
wenzelm@16978
   512
                (case searchf searchinfo occ lhs of
dixon@19835
   513
                   SkipMore i => occ_search i moreasms
dixon@19835
   514
                 | SkipSeq ss =>
wenzelm@19861
   515
                   Seq.append (Seq.map (Library.pair asminfo) (Seq.flat ss))
wenzelm@19861
   516
                               (occ_search 1 moreasms))
dixon@16004
   517
                              (* find later substs also *)
wenzelm@16978
   518
          in
wenzelm@18598
   519
            occ_search skipocc asmpreps |> Seq.maps (apply_subst_in_asm i th r)
dixon@16004
   520
          end;
wenzelm@18598
   521
    in stepthms |> Seq.maps rewrite_with_thm end;
dixon@15538
   522
dixon@16004
   523
wenzelm@16978
   524
fun skip_first_asm_occs_search searchf sinfo occ lhs =
dixon@19835
   525
    skipto_skipseq occ (searchf sinfo lhs);
dixon@16004
   526
wenzelm@18598
   527
fun eqsubst_asm_tac ctxt occL thms i th =
wenzelm@16978
   528
    let val nprems = Thm.nprems_of th
dixon@15538
   529
    in
dixon@16004
   530
      if nprems < i then Seq.empty else
wenzelm@16978
   531
      let val thmseq = (Seq.of_list thms)
wenzelm@16978
   532
        fun apply_occ occK th =
wenzelm@18598
   533
            thmseq |> Seq.maps
wenzelm@16978
   534
                    (fn r =>
wenzelm@18598
   535
                        eqsubst_asm_tac' ctxt (skip_first_asm_occs_search
dixon@19871
   536
                                            searchf_lr_unify_valid) occK r
dixon@16004
   537
                                         (i + ((Thm.nprems_of th) - nprems))
dixon@16004
   538
                                         th);
wenzelm@16978
   539
        val sortedoccs =
dixon@16004
   540
            Library.sort (Library.rev_order o Library.int_ord) occL
dixon@16004
   541
      in
dixon@16004
   542
        Seq.map distinct_subgoals
dixon@16004
   543
                (Seq.EVERY (map apply_occ sortedoccs) th)
dixon@16004
   544
      end
dixon@16004
   545
    end
dixon@16004
   546
    handle THM _ => raise eqsubst_occL_exp ("THM",occL,thms,i,th);
paulson@15481
   547
paulson@15481
   548
(* inthms are the given arguments in Isar, and treated as eqstep with
paulson@15481
   549
   the first one, then the second etc *)
wenzelm@18598
   550
fun eqsubst_asm_meth ctxt occL inthms =
wenzelm@30510
   551
    SIMPLE_METHOD' (eqsubst_asm_tac ctxt occL inthms);
paulson@15481
   552
paulson@15481
   553
(* syntax for options, given "(asm)" will give back true, without
paulson@15481
   554
   gives back false *)
paulson@15481
   555
val options_syntax =
paulson@15481
   556
    (Args.parens (Args.$$$ "asm") >> (K true)) ||
paulson@15481
   557
     (Scan.succeed false);
dixon@15936
   558
dixon@15929
   559
val ith_syntax =
wenzelm@27809
   560
    Scan.optional (Args.parens (Scan.repeat OuterParse.nat)) [0];
paulson@15481
   561
wenzelm@18598
   562
(* combination method that takes a flag (true indicates that subst
wenzelm@18598
   563
should be done to an assumption, false = apply to the conclusion of
wenzelm@18598
   564
the goal) as well as the theorems to use *)
wenzelm@18598
   565
fun subst_meth src =
wenzelm@18988
   566
  Method.syntax ((Scan.lift options_syntax) -- (Scan.lift ith_syntax) -- Attrib.thms) src
haftmann@21879
   567
  #> (fn (((asmflag, occL), inthms), ctxt) =>
wenzelm@18598
   568
    (if asmflag then eqsubst_asm_meth else eqsubst_meth) ctxt occL inthms);
paulson@15481
   569
wenzelm@18598
   570
wenzelm@16978
   571
val setup =
wenzelm@18833
   572
  Method.add_method ("subst", subst_meth, "single-step substitution");
paulson@15481
   573
wenzelm@16978
   574
end;