src/Pure/pattern.ML
author berghofe
Fri May 31 18:47:11 2002 +0200 (2002-05-31 ago)
changeset 13195 98975cc13d28
parent 12980 8f717cbd4e44
child 13642 a3d97348ceb6
permissions -rw-r--r--
Changes to rewrite_term:
- now uses skeletons to speed up rewriting
- added interface for rewriting procedures
wenzelm@12784
     1
(*  Title:      Pure/pattern.ML
clasohm@0
     2
    ID:         $Id$
wenzelm@12784
     3
    Author:     Tobias Nipkow, Christine Heinzelmann, and Stefan Berghofer
wenzelm@12784
     4
    License:    GPL (GNU GENERAL PUBLIC LICENSE)
clasohm@0
     5
clasohm@0
     6
Unification of Higher-Order Patterns.
clasohm@0
     7
clasohm@0
     8
See also:
clasohm@0
     9
Tobias Nipkow. Functional Unification of Higher-Order Patterns.
clasohm@0
    10
In Proceedings of the 8th IEEE Symposium Logic in Computer Science, 1993.
nipkow@678
    11
nipkow@678
    12
TODO: optimize red by special-casing it
clasohm@0
    13
*)
clasohm@0
    14
paulson@2751
    15
infix aeconv;
paulson@2751
    16
clasohm@0
    17
signature PATTERN =
paulson@1501
    18
  sig
clasohm@0
    19
  type type_sig
clasohm@0
    20
  type sg
clasohm@0
    21
  type env
paulson@2751
    22
  val aeconv            : term * term -> bool
paulson@2725
    23
  val eta_contract      : term -> term
wenzelm@12781
    24
  val beta_eta_contract : term -> term
paulson@2725
    25
  val eta_contract_atom : term -> term
paulson@2725
    26
  val match             : type_sig -> term * term
paulson@2725
    27
                          -> (indexname*typ)list * (indexname*term)list
nipkow@4820
    28
  val first_order_match : type_sig -> term * term
nipkow@4820
    29
                          -> (indexname*typ)list * (indexname*term)list
paulson@2725
    30
  val matches           : type_sig -> term * term -> bool
nipkow@4667
    31
  val matches_subterm   : type_sig -> term * term -> bool
paulson@2725
    32
  val unify             : sg * env * (term * term)list -> env
nipkow@4820
    33
  val first_order       : term -> bool
nipkow@4820
    34
  val pattern           : term -> bool
berghofe@13195
    35
  val rewrite_term      : type_sig -> (term * term) list -> (term -> term option) list
berghofe@13195
    36
                          -> term -> term
clasohm@0
    37
  exception Unif
clasohm@0
    38
  exception MATCH
clasohm@0
    39
  exception Pattern
paulson@1501
    40
  end;
clasohm@0
    41
paulson@1501
    42
structure Pattern : PATTERN =
clasohm@0
    43
struct
clasohm@0
    44
clasohm@0
    45
type type_sig = Type.type_sig
clasohm@0
    46
type sg = Sign.sg
clasohm@0
    47
type env = Envir.env
clasohm@0
    48
clasohm@0
    49
exception Unif;
clasohm@0
    50
exception Pattern;
clasohm@0
    51
wenzelm@12784
    52
fun occurs(F,t,env) =
clasohm@0
    53
    let fun occ(Var(G,_))   = (case Envir.lookup(env,G) of
clasohm@0
    54
                                 Some(t) => occ t
clasohm@0
    55
                               | None    => F=G)
clasohm@0
    56
          | occ(t1$t2)      = occ t1 orelse occ t2
clasohm@0
    57
          | occ(Abs(_,_,t)) = occ t
clasohm@0
    58
          | occ _           = false
clasohm@0
    59
    in occ t end;
clasohm@0
    60
clasohm@0
    61
clasohm@0
    62
fun mapbnd f =
clasohm@0
    63
    let fun mpb d (Bound(i))     = if i < d then Bound(i) else Bound(f(i-d)+d)
clasohm@0
    64
          | mpb d (Abs(s,T,t))   = Abs(s,T,mpb(d+1) t)
nipkow@678
    65
          | mpb d ((u1 $ u2))    = (mpb d u1)$(mpb d u2)
nipkow@678
    66
          | mpb _ atom           = atom
clasohm@0
    67
    in mpb 0 end;
clasohm@0
    68
clasohm@0
    69
fun idx [] j     = ~10000
clasohm@0
    70
  | idx(i::is) j = if i=j then length is else idx is j;
clasohm@0
    71
clasohm@0
    72
fun at xs i = nth_elem (i,xs);
clasohm@0
    73
clasohm@0
    74
fun mkabs (binders,is,t)  =
clasohm@0
    75
    let fun mk(i::is) = let val (x,T) = nth_elem(i,binders)
wenzelm@12784
    76
                        in Abs(x,T,mk is) end
clasohm@0
    77
          | mk []     = t
clasohm@0
    78
    in mk is end;
clasohm@0
    79
clasohm@0
    80
val incr = mapbnd (fn i => i+1);
clasohm@0
    81
clasohm@0
    82
fun ints_of []             = []
wenzelm@12784
    83
  | ints_of (Bound i ::bs) =
clasohm@0
    84
      let val is = ints_of bs
berghofe@1576
    85
      in if i mem_int is then raise Pattern else i::is end
clasohm@0
    86
  | ints_of _              = raise Pattern;
clasohm@0
    87
berghofe@12232
    88
fun ints_of' env ts = ints_of (map (Envir.head_norm env) ts);
berghofe@12232
    89
clasohm@0
    90
clasohm@0
    91
fun app (s,(i::is)) = app (s$Bound(i),is)
clasohm@0
    92
  | app (s,[])      = s;
clasohm@0
    93
clasohm@0
    94
fun red (Abs(_,_,s)) (i::is) js = red s is (i::js)
nipkow@678
    95
  | red t            []      [] = t
nipkow@678
    96
  | red t            is      jn = app (mapbnd (at jn) t,is);
nipkow@678
    97
clasohm@0
    98
clasohm@0
    99
(* split_type ([T1,....,Tn]---> T,n,[]) = ([Tn,...,T1],T) *)
clasohm@0
   100
fun split_type (T,0,Ts)                    = (Ts,T)
clasohm@0
   101
  | split_type (Type ("fun",[T1,T2]),n,Ts) = split_type (T2,n-1,T1::Ts)
nipkow@678
   102
  | split_type _                           = error("split_type");
clasohm@0
   103
clasohm@0
   104
fun type_of_G (T,n,is) =
clasohm@0
   105
  let val (Ts,U) = split_type(T,n,[]) in map(at Ts)is ---> U end;
clasohm@0
   106
clasohm@0
   107
fun mkhnf (binders,is,G,js) = mkabs (binders, is, app(G,js));
clasohm@0
   108
clasohm@0
   109
fun mknewhnf(env,binders,is,F as (a,_),T,js) =
clasohm@0
   110
  let val (env',G) = Envir.genvar a (env,type_of_G(T,length is,js))
clasohm@0
   111
  in Envir.update((F,mkhnf(binders,is,G,js)),env') end;
clasohm@0
   112
clasohm@0
   113
clasohm@0
   114
(* mk_proj_list(is) = [ |is| - k | 1 <= k <= |is| and is[k] >= 0 ] *)
clasohm@0
   115
fun mk_proj_list is =
clasohm@0
   116
    let fun mk(i::is,j) = if i >= 0 then j :: mk(is,j-1) else mk(is,j-1)
clasohm@0
   117
          | mk([],_)    = []
clasohm@0
   118
    in mk(is,length is - 1) end;
clasohm@0
   119
clasohm@0
   120
fun proj(s,env,binders,is) =
clasohm@0
   121
    let fun trans d i = if i<d then i else (idx is (i-d))+d;
berghofe@12232
   122
        fun pr(s,env,d,binders) = (case Envir.head_norm env s of
clasohm@0
   123
              Abs(a,T,t) => let val (t',env') = pr(t,env,d+1,((a,T)::binders))
clasohm@0
   124
                            in (Abs(a,T,t'),env') end
clasohm@0
   125
            | t => (case strip_comb t of
clasohm@0
   126
                (c as Const _,ts) =>
clasohm@0
   127
                         let val (ts',env') = prs(ts,env,d,binders)
clasohm@0
   128
                         in (list_comb(c,ts'),env') end
clasohm@0
   129
                 | (f as Free _,ts) =>
clasohm@0
   130
                         let val (ts',env') = prs(ts,env,d,binders)
clasohm@0
   131
                         in (list_comb(f,ts'),env') end
clasohm@0
   132
                 | (Bound(i),ts) =>
clasohm@0
   133
                         let val j = trans d i
clasohm@0
   134
                         in if j < 0 then raise Unif
clasohm@0
   135
                            else let val (ts',env') = prs(ts,env,d,binders)
clasohm@0
   136
                                 in (list_comb(Bound j,ts'),env') end
clasohm@0
   137
                         end
clasohm@0
   138
                 | (Var(F as (a,_),Fty),ts) =>
berghofe@12232
   139
                      let val js = ints_of' env ts;
clasohm@0
   140
                          val js' = map (trans d) js;
clasohm@0
   141
                          val ks = mk_proj_list js';
clasohm@0
   142
                          val ls = filter (fn i => i >= 0) js'
clasohm@0
   143
                          val Hty = type_of_G(Fty,length js,ks)
clasohm@0
   144
                          val (env',H) = Envir.genvar a (env,Hty)
clasohm@0
   145
                          val env'' =
clasohm@0
   146
                                Envir.update((F,mkhnf(binders,js,H,ks)),env')
clasohm@0
   147
                      in (app(H,ls),env'') end
clasohm@0
   148
                 | _  => raise Pattern))
clasohm@0
   149
        and prs(s::ss,env,d,binders) =
clasohm@0
   150
              let val (s',env1) = pr(s,env,d,binders)
clasohm@0
   151
                  val (ss',env2) = prs(ss,env1,d,binders)
clasohm@0
   152
              in (s'::ss',env2) end
clasohm@0
   153
          | prs([],env,_,_) = ([],env)
clasohm@0
   154
   in if downto0(is,length binders - 1) then (s,env)
clasohm@0
   155
      else pr(s,env,0,binders)
clasohm@0
   156
   end;
clasohm@0
   157
clasohm@0
   158
clasohm@0
   159
(* mk_ff_list(is,js) = [ length(is) - k | 1 <= k <= |is| and is[k] = js[k] ] *)
wenzelm@12784
   160
fun mk_ff_list(is,js) =
wenzelm@12784
   161
    let fun mk([],[],_)        = []
clasohm@0
   162
          | mk(i::is,j::js, k) = if i=j then k :: mk(is,js,k-1)
clasohm@0
   163
                                        else mk(is,js,k-1)
nipkow@678
   164
          | mk _               = error"mk_ff_list"
clasohm@0
   165
    in mk(is,js,length is-1) end;
clasohm@0
   166
clasohm@0
   167
fun flexflex1(env,binders,F,Fty,is,js) =
clasohm@0
   168
  if is=js then env
clasohm@0
   169
  else let val ks = mk_ff_list(is,js)
clasohm@0
   170
       in mknewhnf(env,binders,is,F,Fty,ks) end;
clasohm@0
   171
clasohm@0
   172
fun flexflex2(env,binders,F,Fty,is,G,Gty,js) =
clasohm@0
   173
  let fun ff(F,Fty,is,G as (a,_),Gty,js) =
berghofe@1576
   174
            if js subset_int is
clasohm@0
   175
            then let val t= mkabs(binders,is,app(Var(G,Gty),map (idx is) js))
clasohm@0
   176
                 in Envir.update((F,t),env) end
berghofe@1576
   177
            else let val ks = is inter_int js
clasohm@0
   178
                     val Hty = type_of_G(Fty,length is,map (idx is) ks)
clasohm@0
   179
                     val (env',H) = Envir.genvar a (env,Hty)
clasohm@0
   180
                     fun lam(is) = mkabs(binders,is,app(H,map (idx is) ks));
clasohm@0
   181
                 in Envir.update((G,lam js), Envir.update((F,lam is),env'))
clasohm@0
   182
                 end;
clasohm@0
   183
  in if xless(G,F) then ff(F,Fty,is,G,Gty,js) else ff(G,Gty,js,F,Fty,is) end
clasohm@0
   184
clasohm@0
   185
val tsgr = ref(Type.tsig0);
clasohm@0
   186
clasohm@0
   187
fun unify_types(T,U, env as Envir.Envir{asol,iTs,maxidx}) =
clasohm@0
   188
  if T=U then env
wenzelm@12527
   189
  else let val (iTs',maxidx') = Type.unify (!tsgr) (iTs, maxidx) (U, T)
nipkow@1435
   190
       in Envir.Envir{asol=asol,maxidx=maxidx',iTs=iTs'} end
clasohm@0
   191
       handle Type.TUNIFY => raise Unif;
clasohm@0
   192
berghofe@12232
   193
fun unif binders (env,(s,t)) = case (Envir.head_norm env s, Envir.head_norm env t) of
clasohm@0
   194
      (Abs(ns,Ts,ts),Abs(nt,Tt,tt)) =>
clasohm@0
   195
         let val name = if ns = "" then nt else ns
clasohm@0
   196
         in unif ((name,Ts)::binders) (env,(ts,tt)) end
clasohm@0
   197
    | (Abs(ns,Ts,ts),t) => unif ((ns,Ts)::binders) (env,(ts,(incr t)$Bound(0)))
clasohm@0
   198
    | (t,Abs(nt,Tt,tt)) => unif ((nt,Tt)::binders) (env,((incr t)$Bound(0),tt))
clasohm@0
   199
    | p => cases(binders,env,p)
clasohm@0
   200
clasohm@0
   201
and cases(binders,env,(s,t)) = case (strip_comb s,strip_comb t) of
wenzelm@12784
   202
       ((Var(F,Fty),ss),(Var(G,Gty),ts)) =>
berghofe@12232
   203
         if F = G then flexflex1(env,binders,F,Fty,ints_of' env ss,ints_of' env ts)
berghofe@12232
   204
                  else flexflex2(env,binders,F,Fty,ints_of' env ss,G,Gty,ints_of' env ts)
berghofe@12232
   205
      | ((Var(F,_),ss),_)             => flexrigid(env,binders,F,ints_of' env ss,t)
berghofe@12232
   206
      | (_,(Var(F,_),ts))             => flexrigid(env,binders,F,ints_of' env ts,s)
clasohm@0
   207
      | ((Const c,ss),(Const d,ts))   => rigidrigid(env,binders,c,d,ss,ts)
clasohm@0
   208
      | ((Free(f),ss),(Free(g),ts))   => rigidrigid(env,binders,f,g,ss,ts)
wenzelm@12784
   209
      | ((Bound(i),ss),(Bound(j),ts)) => rigidrigidB (env,binders,i,j,ss,ts)
clasohm@0
   210
      | ((Abs(_),_),_)                => raise Pattern
clasohm@0
   211
      | (_,(Abs(_),_))                => raise Pattern
clasohm@0
   212
      | _                             => raise Unif
clasohm@0
   213
clasohm@0
   214
and rigidrigid (env,binders,(a,Ta),(b,Tb),ss,ts) =
clasohm@0
   215
      if a<>b then raise Unif
clasohm@0
   216
      else foldl (unif binders) (unify_types(Ta,Tb,env), ss~~ts)
clasohm@0
   217
clasohm@0
   218
and rigidrigidB (env,binders,i,j,ss,ts) =
clasohm@0
   219
     if i <> j then raise Unif else foldl (unif binders) (env ,ss~~ts)
clasohm@0
   220
clasohm@0
   221
and flexrigid (env,binders,F,is,t) =
clasohm@0
   222
      if occurs(F,t,env) then raise Unif
clasohm@0
   223
      else let val (u,env') = proj(t,env,binders,is)
clasohm@0
   224
           in Envir.update((F,mkabs(binders,is,u)),env') end;
clasohm@0
   225
clasohm@0
   226
fun unify(sg,env,tus) = (tsgr := #tsig(Sign.rep_sg sg);
clasohm@0
   227
                         foldl (unif []) (env,tus));
clasohm@0
   228
clasohm@0
   229
paulson@2725
   230
(*Eta-contract a term (fully)*)
nipkow@2792
   231
wenzelm@12797
   232
fun eta_contract t =
wenzelm@12797
   233
  let
wenzelm@12797
   234
    exception SAME;
wenzelm@12797
   235
    fun eta (Abs (a, T, body)) = 
wenzelm@12797
   236
      ((case eta body of
wenzelm@12797
   237
          body' as (f $ Bound 0) => 
wenzelm@12797
   238
	    if loose_bvar1 (f, 0) then Abs(a, T, body')
wenzelm@12797
   239
	    else incr_boundvars ~1 f
wenzelm@12797
   240
        | body' => Abs (a, T, body')) handle SAME =>
wenzelm@12797
   241
       (case body of
wenzelm@12797
   242
          (f $ Bound 0) => 
wenzelm@12797
   243
	    if loose_bvar1 (f, 0) then raise SAME
wenzelm@12797
   244
	    else incr_boundvars ~1 f
wenzelm@12797
   245
        | _ => raise SAME))
wenzelm@12797
   246
      | eta (f $ t) =
wenzelm@12797
   247
          (let val f' = eta f
wenzelm@12797
   248
           in f' $ etah t end handle SAME => f $ eta t)
wenzelm@12797
   249
      | eta _ = raise SAME
wenzelm@12797
   250
    and etah t = (eta t handle SAME => t)
wenzelm@12797
   251
  in etah t end;
clasohm@0
   252
wenzelm@12781
   253
val beta_eta_contract = eta_contract o Envir.beta_norm;
wenzelm@12781
   254
nipkow@6539
   255
(*Eta-contract a term from outside: just enough to reduce it to an atom
nipkow@6539
   256
DOESN'T QUITE WORK!
nipkow@6539
   257
*)
wenzelm@12784
   258
fun eta_contract_atom (t0 as Abs(a, T, body)) =
paulson@2725
   259
      (case  eta_contract2 body  of
wenzelm@12784
   260
        body' as (f $ Bound 0) =>
wenzelm@12784
   261
            if loose_bvar1(f,0) then Abs(a,T,body')
wenzelm@12784
   262
            else eta_contract_atom (incr_boundvars ~1 f)
paulson@2725
   263
      | _ => t0)
paulson@2725
   264
  | eta_contract_atom t = t
paulson@2725
   265
and eta_contract2 (f$t) = f $ eta_contract_atom t
paulson@2725
   266
  | eta_contract2 t     = eta_contract_atom t;
paulson@2725
   267
paulson@2725
   268
nipkow@678
   269
(*Tests whether 2 terms are alpha/eta-convertible and have same type.
nipkow@678
   270
  Note that Consts and Vars may have more than one type.*)
paulson@2751
   271
fun t aeconv u = aconv_aux (eta_contract_atom t, eta_contract_atom u)
paulson@2751
   272
and aconv_aux (Const(a,T), Const(b,U)) = a=b  andalso  T=U
paulson@2751
   273
  | aconv_aux (Free(a,T),  Free(b,U))  = a=b  andalso  T=U
paulson@2751
   274
  | aconv_aux (Var(v,T),   Var(w,U))   = eq_ix(v,w) andalso  T=U
wenzelm@12784
   275
  | aconv_aux (Bound i,    Bound j)    = i=j
paulson@2751
   276
  | aconv_aux (Abs(_,T,t), Abs(_,U,u)) = (t aeconv u)  andalso  T=U
wenzelm@12784
   277
  | aconv_aux (f$t,        g$u)        = (f aeconv g)  andalso (t aeconv u)
paulson@2751
   278
  | aconv_aux _ =  false;
nipkow@678
   279
nipkow@678
   280
nipkow@4820
   281
(*** Matching ***)
nipkow@4820
   282
nipkow@4820
   283
exception MATCH;
nipkow@4820
   284
nipkow@4820
   285
fun typ_match tsig args = (Type.typ_match tsig args)
nipkow@4820
   286
                          handle Type.TYPE_MATCH => raise MATCH;
nipkow@4820
   287
nipkow@4820
   288
(*First-order matching;
nipkow@4820
   289
  fomatch tsig (pattern, object) returns a (tyvar,typ)list and (var,term)list.
nipkow@4820
   290
  The pattern and object may have variables in common.
nipkow@4820
   291
  Instantiation does not affect the object, so matching ?a with ?a+1 works.
nipkow@4820
   292
  Object is eta-contracted on the fly (by eta-expanding the pattern).
nipkow@4820
   293
  Precondition: the pattern is already eta-contracted!
nipkow@4820
   294
  Note: types are matched on the fly *)
nipkow@4820
   295
fun fomatch tsig =
nipkow@4820
   296
  let
nipkow@4820
   297
    fun mtch (instsp as (tyinsts,insts)) = fn
nipkow@4820
   298
        (Var(ixn,T), t)  =>
wenzelm@12784
   299
          if loose_bvar(t,0) then raise MATCH
wenzelm@12784
   300
          else (case assoc_string_int(insts,ixn) of
wenzelm@12784
   301
                  None => (typ_match tsig (tyinsts, (T, fastype_of t)),
wenzelm@12784
   302
                           (ixn,t)::insts)
wenzelm@12784
   303
                | Some u => if t aeconv u then instsp else raise MATCH)
nipkow@4820
   304
      | (Free (a,T), Free (b,U)) =>
wenzelm@12784
   305
          if a=b then (typ_match tsig (tyinsts,(T,U)), insts) else raise MATCH
nipkow@4820
   306
      | (Const (a,T), Const (b,U))  =>
wenzelm@12784
   307
          if a=b then (typ_match tsig (tyinsts,(T,U)), insts) else raise MATCH
nipkow@4820
   308
      | (Bound i, Bound j)  =>  if  i=j  then  instsp  else raise MATCH
nipkow@4820
   309
      | (Abs(_,T,t), Abs(_,U,u))  =>
wenzelm@12784
   310
          mtch (typ_match tsig (tyinsts,(T,U)),insts) (t,u)
nipkow@4820
   311
      | (f$t, g$u) => mtch (mtch instsp (f,g)) (t, u)
nipkow@4820
   312
      | (t, Abs(_,U,u))  =>  mtch instsp ((incr t)$(Bound 0), u)
nipkow@4820
   313
      | _ => raise MATCH
nipkow@4820
   314
  in mtch end;
nipkow@4820
   315
berghofe@8406
   316
fun first_order_match tsig = apfst Vartab.dest o fomatch tsig (Vartab.empty, []);
berghofe@8406
   317
nipkow@4820
   318
(* Matching of higher-order patterns *)
nipkow@4820
   319
nipkow@4820
   320
fun match_bind(itms,binders,ixn,is,t) =
nipkow@4820
   321
  let val js = loose_bnos t
nipkow@4820
   322
  in if null is
nipkow@4820
   323
     then if null js then (ixn,t)::itms else raise MATCH
nipkow@4820
   324
     else if js subset_int is
nipkow@4820
   325
          then let val t' = if downto0(is,length binders - 1) then t
nipkow@4820
   326
                            else mapbnd (idx is) t
nipkow@4820
   327
               in (ixn, mkabs(binders,is,t')) :: itms end
nipkow@4820
   328
          else raise MATCH
nipkow@4820
   329
  end;
nipkow@4820
   330
nipkow@678
   331
fun match tsg (po as (pat,obj)) =
nipkow@678
   332
let
nipkow@4820
   333
  (* Pre: pat and obj have same type *)
nipkow@4820
   334
  fun mtch binders (env as (iTs,itms),(pat,obj)) =
nipkow@4820
   335
    case pat of
nipkow@4820
   336
      Abs(ns,Ts,ts) =>
nipkow@4820
   337
        (case obj of
nipkow@4820
   338
           Abs(nt,Tt,tt) => mtch ((nt,Tt)::binders) (env,(ts,tt))
berghofe@8406
   339
         | _ => let val Tt = typ_subst_TVars_Vartab iTs Ts
nipkow@4820
   340
                in mtch((ns,Tt)::binders)(env,(ts,(incr obj)$Bound(0))) end)
nipkow@4820
   341
    | _ => (case obj of
nipkow@4820
   342
              Abs(nt,Tt,tt) =>
nipkow@4820
   343
                mtch((nt,Tt)::binders)(env,((incr pat)$Bound(0),tt))
nipkow@4820
   344
            | _ => cases(binders,env,pat,obj))
nipkow@678
   345
nipkow@4820
   346
  and cases(binders,env as (iTs,itms),pat,obj) =
nipkow@4820
   347
    let val (ph,pargs) = strip_comb pat
nipkow@4820
   348
        fun rigrig1(iTs,oargs) =
nipkow@4820
   349
              foldl (mtch binders) ((iTs,itms), pargs~~oargs)
nipkow@4820
   350
        fun rigrig2((a,Ta),(b,Tb),oargs) =
nipkow@4820
   351
              if a<> b then raise MATCH
nipkow@4820
   352
              else rigrig1(typ_match tsg (iTs,(Ta,Tb)), oargs)
nipkow@4820
   353
    in case ph of
nipkow@4820
   354
         Var(ixn,_) =>
nipkow@4820
   355
           let val is = ints_of pargs
nipkow@4820
   356
           in case assoc_string_int(itms,ixn) of
nipkow@4820
   357
                None => (iTs,match_bind(itms,binders,ixn,is,obj))
nipkow@4820
   358
              | Some u => if obj aeconv (red u is []) then env
nipkow@4820
   359
                          else raise MATCH
nipkow@4820
   360
           end
nipkow@4820
   361
       | _ =>
nipkow@4820
   362
           let val (oh,oargs) = strip_comb obj
nipkow@4820
   363
           in case (ph,oh) of
nipkow@4820
   364
                (Const c,Const d) => rigrig2(c,d,oargs)
nipkow@4820
   365
              | (Free f,Free g)   => rigrig2(f,g,oargs)
nipkow@4820
   366
              | (Bound i,Bound j) => if i<>j then raise MATCH
nipkow@4820
   367
                                     else rigrig1(iTs,oargs)
nipkow@4820
   368
              | (Abs _, _)        => raise Pattern
nipkow@4820
   369
              | (_, Abs _)        => raise Pattern
nipkow@4820
   370
              | _                 => raise MATCH
nipkow@4820
   371
           end
nipkow@4820
   372
    end;
nipkow@678
   373
nipkow@4820
   374
  val pT = fastype_of pat
nipkow@4820
   375
  and oT = fastype_of obj
berghofe@8406
   376
  val iTs = typ_match tsg (Vartab.empty, (pT,oT))
nipkow@4820
   377
  val insts2 = (iTs,[])
nipkow@678
   378
berghofe@8406
   379
in apfst Vartab.dest (mtch [] (insts2, po)
berghofe@8406
   380
   handle Pattern => fomatch tsg insts2 po)
nipkow@678
   381
end;
clasohm@0
   382
clasohm@0
   383
(*Predicate: does the pattern match the object?*)
nipkow@678
   384
fun matches tsig po = (match tsig po; true) handle MATCH => false;
clasohm@0
   385
nipkow@4667
   386
(* Does pat match a subterm of obj? *)
nipkow@4667
   387
fun matches_subterm tsig (pat,obj) =
nipkow@4667
   388
  let fun msub(bounds,obj) = matches tsig (pat,obj) orelse
nipkow@4667
   389
            case obj of
nipkow@4667
   390
              Abs(x,T,t) => let val y = variant bounds x
nipkow@4667
   391
                                val f = Free(":" ^ y,T)
nipkow@4667
   392
                            in msub(x::bounds,subst_bound(f,t)) end
nipkow@4667
   393
            | s$t => msub(bounds,s) orelse msub(bounds,t)
nipkow@4667
   394
            | _ => false
nipkow@4667
   395
  in msub([],obj) end;
nipkow@4667
   396
nipkow@4820
   397
fun first_order(Abs(_,_,t)) = first_order t
nipkow@4820
   398
  | first_order(t $ u) = first_order t andalso first_order u andalso
nipkow@4820
   399
                         not(is_Var t)
nipkow@4820
   400
  | first_order _ = true;
nipkow@4820
   401
nipkow@4820
   402
fun pattern(Abs(_,_,t)) = pattern t
nipkow@4820
   403
  | pattern(t) = let val (head,args) = strip_comb t
nipkow@4820
   404
                 in if is_Var head
nipkow@4820
   405
                    then let val _ = ints_of args in true end
nipkow@4820
   406
                         handle Pattern => false
nipkow@4820
   407
                    else forall pattern args
nipkow@4820
   408
                 end;
nipkow@4820
   409
wenzelm@12784
   410
wenzelm@12784
   411
(* rewriting -- simple but fast *)
wenzelm@12784
   412
berghofe@13195
   413
fun rewrite_term tsig rules procs tm =
wenzelm@12784
   414
  let
berghofe@13195
   415
    val skel0 = Bound 0;
berghofe@13195
   416
wenzelm@12797
   417
    val rhs_names =
wenzelm@12797
   418
      foldr (fn ((_, rhs), names) => add_term_free_names (rhs, names)) (rules, []);
berghofe@13195
   419
wenzelm@12797
   420
    fun variant_absfree (x, T, t) =
wenzelm@12797
   421
      let
wenzelm@12797
   422
        val x' = variant (add_term_free_names (t, rhs_names)) x;
wenzelm@12797
   423
        val t' = subst_bound (Free (x', T), t);
wenzelm@12797
   424
      in (fn u => Abs (x, T, abstract_over (Free (x', T), u)), t') end;
wenzelm@12797
   425
wenzelm@12784
   426
    fun match_rew tm (tm1, tm2) =
berghofe@13195
   427
      let val rtm = if_none (Term.rename_abs tm1 tm tm2) tm2
berghofe@13195
   428
      in Some (subst_vars (match tsig (tm1, tm)) rtm, rtm)
berghofe@13195
   429
        handle MATCH => None
berghofe@13195
   430
      end;
wenzelm@12784
   431
berghofe@13195
   432
    fun rew (Abs (_, _, body) $ t) = Some (subst_bound (t, body), skel0)
berghofe@13195
   433
      | rew tm = (case get_first (match_rew tm) rules of
berghofe@13195
   434
          None => apsome (rpair skel0) (get_first (fn p => p tm) procs)
berghofe@13195
   435
        | x => x);
berghofe@13195
   436
berghofe@13195
   437
    fun rew1 (Var _) _ = None
berghofe@13195
   438
      | rew1 skel tm = (case rew2 skel tm of
wenzelm@12817
   439
          Some tm1 => (case rew tm1 of
berghofe@13195
   440
              Some (tm2, skel') => Some (if_none (rew1 skel' tm2) tm2)
wenzelm@12784
   441
            | None => Some tm1)
wenzelm@12817
   442
        | None => (case rew tm of
berghofe@13195
   443
              Some (tm1, skel') => Some (if_none (rew1 skel' tm1) tm1)
wenzelm@12784
   444
            | None => None))
wenzelm@12784
   445
berghofe@13195
   446
    and rew2 skel (tm1 $ tm2) = (case tm1 of
wenzelm@12784
   447
            Abs (_, _, body) =>
wenzelm@12784
   448
              let val tm' = subst_bound (tm2, body)
berghofe@13195
   449
              in Some (if_none (rew2 skel0 tm') tm') end
berghofe@13195
   450
          | _ => 
berghofe@13195
   451
            let val (skel1, skel2) = (case skel of
berghofe@13195
   452
                skel1 $ skel2 => (skel1, skel2)
berghofe@13195
   453
              | _ => (skel0, skel0))
berghofe@13195
   454
            in case rew1 skel1 tm1 of
berghofe@13195
   455
                Some tm1' => (case rew1 skel2 tm2 of
berghofe@13195
   456
                    Some tm2' => Some (tm1' $ tm2')
berghofe@13195
   457
                  | None => Some (tm1' $ tm2))
berghofe@13195
   458
              | None => (case rew1 skel2 tm2 of
berghofe@13195
   459
                    Some tm2' => Some (tm1 $ tm2')
berghofe@13195
   460
                  | None => None)
berghofe@13195
   461
            end)
berghofe@13195
   462
      | rew2 skel (Abs (x, T, tm)) =
berghofe@13195
   463
          let
berghofe@13195
   464
            val (abs, tm') = variant_absfree (x, T, tm);
berghofe@13195
   465
            val skel' = (case skel of Abs (_, _, skel') => skel' | _ => skel0)
berghofe@13195
   466
          in case rew1 skel' tm' of
wenzelm@12797
   467
              Some tm'' => Some (abs tm'')
berghofe@13195
   468
            | None => None
wenzelm@12797
   469
          end
berghofe@13195
   470
      | rew2 _ _ = None
wenzelm@12784
   471
berghofe@13195
   472
  in if_none (rew1 skel0 tm) tm end;
wenzelm@12784
   473
clasohm@0
   474
end;