src/Pure/Isar/rule_cases.ML
author wenzelm
Tue Dec 05 22:14:42 2006 +0100 (2006-12-05)
changeset 21658 5e31241e1e3c
parent 21646 c07b5b0e8492
child 21708 45e7491bea47
permissions -rw-r--r--
Attrib.internal: morphism;
wenzelm@8364
     1
(*  Title:      Pure/Isar/rule_cases.ML
wenzelm@8364
     2
    ID:         $Id$
wenzelm@8364
     3
    Author:     Markus Wenzel, TU Muenchen
wenzelm@8364
     4
wenzelm@18229
     5
Annotations and local contexts of rules.
wenzelm@8364
     6
*)
wenzelm@8364
     7
wenzelm@18188
     8
infix 1 THEN_ALL_NEW_CASES;
wenzelm@18188
     9
wenzelm@18229
    10
signature BASIC_RULE_CASES =
wenzelm@18229
    11
sig
wenzelm@18229
    12
  type cases
wenzelm@18229
    13
  type cases_tactic
wenzelm@18229
    14
  val CASES: cases -> tactic -> cases_tactic
wenzelm@18229
    15
  val NO_CASES: tactic -> cases_tactic
wenzelm@18229
    16
  val SUBGOAL_CASES: ((term * int) -> cases_tactic) -> int -> cases_tactic
wenzelm@18229
    17
  val THEN_ALL_NEW_CASES: (int -> cases_tactic) * (int -> tactic) -> int -> cases_tactic
wenzelm@18229
    18
end
wenzelm@18229
    19
wenzelm@8364
    20
signature RULE_CASES =
wenzelm@8364
    21
sig
wenzelm@18229
    22
  include BASIC_RULE_CASES
wenzelm@18608
    23
  datatype T = Case of
wenzelm@18608
    24
   {fixes: (string * typ) list,
wenzelm@18608
    25
    assumes: (string * term list) list,
wenzelm@18608
    26
    binds: (indexname * term option) list,
wenzelm@18608
    27
    cases: (string * T) list}
wenzelm@18608
    28
  val strip_params: term -> (string * typ) list
wenzelm@18608
    29
  val make_common: bool -> theory * term -> (string * string list) list -> cases
wenzelm@18608
    30
  val make_nested: bool -> term -> theory * term -> (string * string list) list -> cases
wenzelm@18608
    31
  val make_simple: bool -> theory * term -> string -> cases
wenzelm@18608
    32
  val apply: term list -> T -> T
wenzelm@18237
    33
  val consume: thm list -> thm list -> ('a * int) * thm ->
wenzelm@18237
    34
    (('a * (int * thm list)) * thm) Seq.seq
wenzelm@18477
    35
  val add_consumes: int -> thm -> thm
wenzelm@18728
    36
  val consumes: int -> attribute
wenzelm@18728
    37
  val consumes_default: int -> attribute
wenzelm@8364
    38
  val name: string list -> thm -> thm
wenzelm@18728
    39
  val case_names: string list -> attribute
wenzelm@18728
    40
  val case_conclusion: string * string list -> attribute
wenzelm@12761
    41
  val save: thm -> thm -> thm
wenzelm@18237
    42
  val get: thm -> (string * string list) list * int
wenzelm@8427
    43
  val rename_params: string list list -> thm -> thm
wenzelm@18728
    44
  val params: string list list -> attribute
wenzelm@20289
    45
  val mutual_rule: Proof.context -> thm list -> (int list * thm) option
wenzelm@20289
    46
  val strict_mutual_rule: Proof.context -> thm list -> int list * thm
wenzelm@8364
    47
end;
wenzelm@8364
    48
wenzelm@8364
    49
structure RuleCases: RULE_CASES =
wenzelm@8364
    50
struct
wenzelm@8364
    51
wenzelm@18608
    52
(** cases **)
wenzelm@18608
    53
wenzelm@18608
    54
datatype T = Case of
wenzelm@18608
    55
 {fixes: (string * typ) list,
wenzelm@18608
    56
  assumes: (string * term list) list,
wenzelm@18608
    57
  binds: (indexname * term option) list,
wenzelm@18608
    58
  cases: (string * T) list};
wenzelm@18608
    59
wenzelm@18608
    60
type cases = (string * T option) list;
wenzelm@18608
    61
wenzelm@18608
    62
val case_conclN = "case";
wenzelm@18608
    63
val case_hypsN = "hyps";
wenzelm@18608
    64
val case_premsN = "prems";
wenzelm@18608
    65
wenzelm@20087
    66
val strip_params = map (apfst (perhaps (try Name.dest_skolem))) o Logic.strip_params;
wenzelm@18608
    67
wenzelm@18608
    68
local
wenzelm@18608
    69
wenzelm@18608
    70
fun abs xs t = Term.list_abs (xs, t);
wenzelm@18608
    71
fun app us t = Term.betapplys (t, us);
wenzelm@18608
    72
wenzelm@18608
    73
fun dest_binops cs tm =
wenzelm@18608
    74
  let
wenzelm@18608
    75
    val n = length cs;
wenzelm@18608
    76
    fun dest 0 _ = []
wenzelm@18608
    77
      | dest 1 t = [t]
wenzelm@18608
    78
      | dest k (_ $ t $ u) = t :: dest (k - 1) u
wenzelm@18608
    79
      | dest _ _ = raise TERM ("Expected " ^ string_of_int n ^ " binop arguments", [tm]);
wenzelm@18608
    80
  in cs ~~ dest n tm end;
wenzelm@18608
    81
wenzelm@18608
    82
fun extract_fixes NONE prop = (strip_params prop, [])
wenzelm@18608
    83
  | extract_fixes (SOME outline) prop =
wenzelm@19012
    84
      chop (length (Logic.strip_params outline)) (strip_params prop);
wenzelm@18608
    85
wenzelm@18608
    86
fun extract_assumes _ NONE prop = ([("", Logic.strip_assums_hyp prop)], [])
wenzelm@18608
    87
  | extract_assumes qual (SOME outline) prop =
wenzelm@18608
    88
      let val (hyps, prems) =
wenzelm@19012
    89
        chop (length (Logic.strip_assums_hyp outline)) (Logic.strip_assums_hyp prop)
wenzelm@18608
    90
      in ([(qual case_hypsN, hyps)], [(qual case_premsN, prems)]) end;
wenzelm@18608
    91
wenzelm@18608
    92
fun extract_case is_open thy (case_outline, raw_prop) name concls =
wenzelm@18608
    93
  let
wenzelm@20087
    94
    val rename = if is_open then I else (apfst Name.internal);
wenzelm@18608
    95
wenzelm@18608
    96
    val props = Logic.dest_conjunctions (Drule.norm_hhf thy raw_prop);
wenzelm@18608
    97
    val len = length props;
wenzelm@18608
    98
    val nested = is_some case_outline andalso len > 1;
wenzelm@18608
    99
wenzelm@18608
   100
    fun extract prop =
wenzelm@18608
   101
      let
wenzelm@18608
   102
        val (fixes1, fixes2) = extract_fixes case_outline prop
wenzelm@18608
   103
          |> apfst (map rename);
wenzelm@18608
   104
        val abs_fixes = abs (fixes1 @ fixes2);
wenzelm@18608
   105
        fun abs_fixes1 t =
wenzelm@18608
   106
          if not nested then abs_fixes t
wenzelm@18608
   107
          else abs fixes1 (app (map (Term.dummy_pattern o #2) fixes2) (abs fixes2 t));
wenzelm@18608
   108
wenzelm@18608
   109
        val (assumes1, assumes2) = extract_assumes (NameSpace.qualified name) case_outline prop
wenzelm@19482
   110
          |> pairself (map (apsnd (maps Logic.dest_conjunctions)));
wenzelm@18608
   111
wenzelm@18608
   112
        val concl = ObjectLogic.drop_judgment thy (Logic.strip_assums_concl prop);
wenzelm@18608
   113
        val binds =
wenzelm@18608
   114
          (case_conclN, concl) :: dest_binops concls concl
wenzelm@18608
   115
          |> map (fn (x, t) => ((x, 0), SOME (abs_fixes t)));
wenzelm@18608
   116
      in
wenzelm@18608
   117
       ((fixes1, map (apsnd (map abs_fixes1)) assumes1),
wenzelm@18608
   118
        ((fixes2, map (apsnd (map abs_fixes)) assumes2), binds))
wenzelm@18608
   119
      end;
wenzelm@18608
   120
wenzelm@18608
   121
    val cases = map extract props;
wenzelm@18608
   122
wenzelm@18608
   123
    fun common_case ((fixes1, assumes1), ((fixes2, assumes2), binds)) =
wenzelm@18608
   124
      Case {fixes = fixes1 @ fixes2, assumes = assumes1 @ assumes2, binds = binds, cases = []};
wenzelm@18608
   125
    fun inner_case (_, ((fixes2, assumes2), binds)) =
wenzelm@18608
   126
      Case {fixes = fixes2, assumes = assumes2, binds = binds, cases = []};
wenzelm@18608
   127
    fun nested_case ((fixes1, assumes1), _) =
wenzelm@18608
   128
      Case {fixes = fixes1, assumes = assumes1, binds = [],
wenzelm@18608
   129
        cases = map string_of_int (1 upto len) ~~ map inner_case cases};
wenzelm@18608
   130
  in
wenzelm@18608
   131
    if len = 0 then NONE
wenzelm@18608
   132
    else if len = 1 then SOME (common_case (hd cases))
wenzelm@19046
   133
    else if is_none case_outline orelse length (distinct (op =) (map fst cases)) > 1 then NONE
wenzelm@18608
   134
    else SOME (nested_case (hd cases))
wenzelm@18608
   135
  end;
wenzelm@18608
   136
wenzelm@18608
   137
fun make is_open rule_struct (thy, prop) cases =
wenzelm@18608
   138
  let
wenzelm@18608
   139
    val n = length cases;
wenzelm@21576
   140
    val nprems = Logic.count_prems prop;
wenzelm@18608
   141
    fun add_case (name, concls) (cs, i) =
wenzelm@18608
   142
      ((case try (fn () =>
wenzelm@18608
   143
          (Option.map (curry Logic.nth_prem i) rule_struct, Logic.nth_prem (i, prop))) () of
wenzelm@18608
   144
        NONE => (name, NONE)
wenzelm@18608
   145
      | SOME p => (name, extract_case is_open thy p name concls)) :: cs, i - 1);
wenzelm@18608
   146
  in fold_rev add_case (Library.drop (n - nprems, cases)) ([], n) |> #1 end;
wenzelm@18608
   147
wenzelm@18608
   148
in
wenzelm@18608
   149
wenzelm@18608
   150
fun make_common is_open = make is_open NONE;
wenzelm@18608
   151
fun make_nested is_open rule_struct = make is_open (SOME rule_struct);
wenzelm@18608
   152
fun make_simple is_open (thy, prop) name = [(name, extract_case is_open thy (NONE, prop) "" [])];
wenzelm@18608
   153
wenzelm@18608
   154
fun apply args =
wenzelm@18608
   155
  let
wenzelm@18608
   156
    fun appl (Case {fixes, assumes, binds, cases}) =
wenzelm@18608
   157
      let
wenzelm@18608
   158
        val assumes' = map (apsnd (map (app args))) assumes;
wenzelm@18608
   159
        val binds' = map (apsnd (Option.map (app args))) binds;
wenzelm@18608
   160
        val cases' = map (apsnd appl) cases;
wenzelm@18608
   161
      in Case {fixes = fixes, assumes = assumes', binds = binds', cases = cases'} end;
wenzelm@18608
   162
  in appl end;
wenzelm@18608
   163
wenzelm@18608
   164
end;
wenzelm@18608
   165
wenzelm@18608
   166
wenzelm@18608
   167
wenzelm@18229
   168
(** tactics with cases **)
wenzelm@18229
   169
wenzelm@18229
   170
type cases_tactic = thm -> (cases * thm) Seq.seq;
wenzelm@18229
   171
wenzelm@18229
   172
fun CASES cases tac st = Seq.map (pair cases) (tac st);
wenzelm@18229
   173
fun NO_CASES tac = CASES [] tac;
wenzelm@18229
   174
wenzelm@18229
   175
fun SUBGOAL_CASES tac i st =
wenzelm@18229
   176
  (case try Logic.nth_prem (i, Thm.prop_of st) of
wenzelm@18229
   177
    SOME goal => tac (goal, i) st
wenzelm@18229
   178
  | NONE => Seq.empty);
wenzelm@18229
   179
wenzelm@18229
   180
fun (tac1 THEN_ALL_NEW_CASES tac2) i st =
wenzelm@18229
   181
  st |> tac1 i |> Seq.maps (fn (cases, st') =>
wenzelm@18237
   182
    CASES cases (Seq.INTERVAL tac2 i (i + nprems_of st' - nprems_of st)) st');
wenzelm@18229
   183
wenzelm@18229
   184
wenzelm@18229
   185
wenzelm@18229
   186
(** consume facts **)
wenzelm@18229
   187
wenzelm@18477
   188
local
wenzelm@18477
   189
wenzelm@18477
   190
fun unfold_prems n defs th =
wenzelm@18477
   191
  if null defs then th
wenzelm@18477
   192
  else Drule.fconv_rule (Drule.goals_conv (fn i => i <= n) (Tactic.rewrite true defs)) th;
wenzelm@18477
   193
wenzelm@18477
   194
fun unfold_prems_concls defs th =
wenzelm@18477
   195
  if null defs orelse not (can Logic.dest_conjunction (Thm.concl_of th)) then th
wenzelm@18477
   196
  else
wenzelm@18477
   197
    Drule.fconv_rule
wenzelm@19423
   198
      (Drule.concl_conv ~1 (Conjunction.conv ~1
wenzelm@18477
   199
        (K (Drule.prems_conv ~1 (K (Tactic.rewrite true defs)))))) th;
wenzelm@18477
   200
wenzelm@18477
   201
in
wenzelm@18477
   202
wenzelm@18477
   203
fun consume defs facts ((xx, n), th) =
wenzelm@18237
   204
  let val m = Int.min (length facts, n) in
wenzelm@18477
   205
    th
wenzelm@18477
   206
    |> unfold_prems n defs
wenzelm@18477
   207
    |> unfold_prems_concls defs
wenzelm@18237
   208
    |> Drule.multi_resolve (Library.take (m, facts))
wenzelm@18477
   209
    |> Seq.map (pair (xx, (n - m, Library.drop (m, facts))))
wenzelm@18237
   210
  end;
wenzelm@10811
   211
wenzelm@18477
   212
end;
wenzelm@18477
   213
wenzelm@10811
   214
val consumes_tagN = "consumes";
wenzelm@10811
   215
wenzelm@18229
   216
fun lookup_consumes th =
wenzelm@18229
   217
  let fun err () = raise THM ("Malformed 'consumes' tag of theorem", 0, [th]) in
wenzelm@21646
   218
    (case AList.lookup (op =) (Thm.get_tags th) (consumes_tagN) of
skalberg@15531
   219
      NONE => NONE
skalberg@15531
   220
    | SOME [s] => (case Syntax.read_nat s of SOME n => SOME n | _ => err ())
wenzelm@10530
   221
    | _ => err ())
wenzelm@10530
   222
  end;
wenzelm@10530
   223
wenzelm@18477
   224
fun get_consumes th = the_default 0 (lookup_consumes th);
wenzelm@18477
   225
skalberg@15531
   226
fun put_consumes NONE th = th
skalberg@15531
   227
  | put_consumes (SOME n) th = th
wenzelm@18799
   228
      |> PureThy.untag_rule consumes_tagN
wenzelm@18909
   229
      |> PureThy.tag_rule
wenzelm@18909
   230
        (consumes_tagN, [Library.string_of_int (if n < 0 then Thm.nprems_of th + n else n)]);
wenzelm@10530
   231
wenzelm@18477
   232
fun add_consumes k th = put_consumes (SOME (k + get_consumes th)) th;
wenzelm@18477
   233
wenzelm@12761
   234
val save_consumes = put_consumes o lookup_consumes;
wenzelm@12761
   235
wenzelm@18728
   236
fun consumes n x = Thm.rule_attribute (K (put_consumes (SOME n))) x;
wenzelm@21395
   237
wenzelm@16148
   238
fun consumes_default n x =
wenzelm@21395
   239
  if is_some (lookup_consumes (#2 x)) then x else consumes n x;
wenzelm@8364
   240
wenzelm@8364
   241
wenzelm@18229
   242
wenzelm@18229
   243
(** case names **)
wenzelm@18229
   244
wenzelm@18237
   245
val case_names_tagN = "case_names";
wenzelm@8364
   246
wenzelm@18229
   247
fun add_case_names NONE = I
wenzelm@18229
   248
  | add_case_names (SOME names) =
wenzelm@18799
   249
      PureThy.untag_rule case_names_tagN
wenzelm@18799
   250
      #> PureThy.tag_rule (case_names_tagN, names);
wenzelm@8364
   251
wenzelm@21646
   252
fun lookup_case_names th = AList.lookup (op =) (Thm.get_tags th) case_names_tagN;
wenzelm@10530
   253
wenzelm@18229
   254
val save_case_names = add_case_names o lookup_case_names;
wenzelm@18229
   255
val name = add_case_names o SOME;
wenzelm@18728
   256
fun case_names ss = Thm.rule_attribute (K (name ss));
wenzelm@8364
   257
wenzelm@10530
   258
wenzelm@18229
   259
wenzelm@18229
   260
(** case conclusions **)
wenzelm@18229
   261
wenzelm@18229
   262
val case_concl_tagN = "case_conclusion";
wenzelm@18229
   263
wenzelm@18229
   264
fun is_case_concl name ((a, b :: _): tag) = (a = case_concl_tagN andalso b = name)
wenzelm@18229
   265
  | is_case_concl _ _ = false;
wenzelm@18229
   266
wenzelm@21646
   267
fun add_case_concl (name, cs) = Thm.map_tags (fn tags =>
wenzelm@18229
   268
  filter_out (is_case_concl name) tags @ [(case_concl_tagN, name :: cs)]);
wenzelm@18229
   269
wenzelm@18229
   270
fun get_case_concls th name =
wenzelm@21646
   271
  (case find_first (is_case_concl name) (Thm.get_tags th) of
wenzelm@18237
   272
    SOME (_, _ :: cs) => cs
wenzelm@18237
   273
  | _ => []);
wenzelm@18229
   274
wenzelm@18229
   275
fun save_case_concls th =
wenzelm@21646
   276
  let val concls = Thm.get_tags th |> map_filter
wenzelm@18237
   277
    (fn (a, b :: cs) =>
wenzelm@18237
   278
      if a = case_concl_tagN then SOME (b, cs) else NONE
wenzelm@18229
   279
    | _ => NONE)
wenzelm@18229
   280
  in fold add_case_concl concls end;
wenzelm@18229
   281
wenzelm@18728
   282
fun case_conclusion concl = Thm.rule_attribute (fn _ => add_case_concl concl);
wenzelm@18229
   283
wenzelm@18229
   284
wenzelm@18229
   285
wenzelm@18229
   286
(** case declarations **)
wenzelm@18229
   287
wenzelm@10530
   288
(* access hints *)
wenzelm@10530
   289
wenzelm@18229
   290
fun save th = save_consumes th #> save_case_names th #> save_case_concls th;
wenzelm@12761
   291
wenzelm@18229
   292
fun get th =
wenzelm@12761
   293
  let
wenzelm@18477
   294
    val n = get_consumes th;
wenzelm@18229
   295
    val cases =
wenzelm@18229
   296
      (case lookup_case_names th of
wenzelm@18229
   297
        NONE => map (rpair [] o Library.string_of_int) (1 upto (Thm.nprems_of th - n))
wenzelm@18229
   298
      | SOME names => map (fn name => (name, get_case_concls th name)) names);
wenzelm@18229
   299
  in (cases, n) end;
wenzelm@10530
   300
wenzelm@8364
   301
wenzelm@8427
   302
(* params *)
wenzelm@8364
   303
wenzelm@18229
   304
fun rename_params xss th =
wenzelm@18229
   305
  th
wenzelm@18229
   306
  |> fold_index (fn (i, xs) => Thm.rename_params_rule (xs, i + 1)) xss
wenzelm@18229
   307
  |> save th;
wenzelm@8364
   308
wenzelm@18728
   309
fun params xss = Thm.rule_attribute (K (rename_params xss));
wenzelm@8364
   310
wenzelm@18477
   311
wenzelm@18477
   312
wenzelm@18477
   313
(** mutual_rule **)
wenzelm@18477
   314
wenzelm@18477
   315
local
wenzelm@18477
   316
wenzelm@18477
   317
fun equal_cterms ts us =
wenzelm@18477
   318
  list_ord (Term.fast_term_ord o pairself Thm.term_of) (ts, us) = EQUAL;
wenzelm@18477
   319
wenzelm@19917
   320
fun prep_rule n th =
wenzelm@18477
   321
  let
wenzelm@19917
   322
    val th' = Thm.permute_prems 0 n th;
wenzelm@18477
   323
    val prems = Library.take (Thm.nprems_of th' - n, Drule.cprems_of th');
wenzelm@18477
   324
    val th'' = Drule.implies_elim_list th' (map Thm.assume prems);
wenzelm@18477
   325
  in (prems, (n, th'')) end;
wenzelm@18477
   326
wenzelm@18477
   327
in
wenzelm@18477
   328
wenzelm@19917
   329
fun mutual_rule _ [] = NONE
wenzelm@19917
   330
  | mutual_rule _ [th] = SOME ([0], th)
wenzelm@19917
   331
  | mutual_rule ctxt (ths as th :: _) =
wenzelm@18477
   332
      let
wenzelm@20218
   333
        val ((_, ths'), ctxt') = Variable.import true ths ctxt;
wenzelm@19917
   334
        val rules as (prems, _) :: _ = map (prep_rule (get_consumes th)) ths';
wenzelm@19917
   335
        val (ns, rls) = split_list (map #2 rules);
wenzelm@18477
   336
      in
wenzelm@18477
   337
        if not (forall (equal_cterms prems o #1) rules) then NONE
wenzelm@18477
   338
        else
wenzelm@18477
   339
          SOME (ns,
wenzelm@19917
   340
            rls
wenzelm@19423
   341
            |> foldr1 (uncurry Conjunction.intr)
wenzelm@18477
   342
            |> Drule.implies_intr_list prems
wenzelm@19917
   343
            |> singleton (Variable.export ctxt' ctxt)
wenzelm@19917
   344
            |> save th
wenzelm@18477
   345
            |> put_consumes (SOME 0))
wenzelm@18477
   346
      end;
wenzelm@18477
   347
wenzelm@18477
   348
end;
wenzelm@18477
   349
wenzelm@19917
   350
fun strict_mutual_rule ctxt ths =
wenzelm@19917
   351
  (case mutual_rule ctxt ths of
wenzelm@18581
   352
    NONE => error "Failed to join given rules into one mutual rule"
wenzelm@18581
   353
  | SOME res => res);
wenzelm@18581
   354
wenzelm@8364
   355
end;
wenzelm@17113
   356
wenzelm@18229
   357
structure BasicRuleCases: BASIC_RULE_CASES = RuleCases;
wenzelm@18229
   358
open BasicRuleCases;