src/HOL/Tools/Sledgehammer/sledgehammer_proof_reconstruct.ML
author blanchet
Wed Apr 28 15:34:55 2010 +0200 (2010-04-28)
changeset 36491 6769f8eacaac
parent 36488 32c92af68ec9
child 36492 60532b9bcd1c
permissions -rw-r--r--
unskolemize formulas in proof reconstruction + detect newer SPASS versions to avoid truncating identifiers if not necessary (truncating confuses proof reconstruction)
blanchet@35826
     1
(*  Title:      HOL/Tools/Sledgehammer/sledgehammer_proof_reconstruct.ML
wenzelm@33310
     2
    Author:     Lawrence C Paulson and Claire Quigley, Cambridge University Computer Laboratory
blanchet@36392
     3
    Author:     Jasmin Blanchette, TU Muenchen
paulson@21978
     4
wenzelm@33310
     5
Transfer of proofs from external provers.
wenzelm@33310
     6
*)
wenzelm@33310
     7
blanchet@35826
     8
signature SLEDGEHAMMER_PROOF_RECONSTRUCT =
paulson@24425
     9
sig
blanchet@36281
    10
  type minimize_command = string list -> string
blanchet@36393
    11
  type name_pool = Sledgehammer_FOL_Clause.name_pool
blanchet@36281
    12
paulson@25492
    13
  val chained_hint: string
paulson@24425
    14
  val invert_const: string -> string
paulson@24425
    15
  val invert_type_const: string -> string
wenzelm@33243
    16
  val num_typargs: theory -> string -> int
paulson@24425
    17
  val make_tvar: string -> typ
paulson@24425
    18
  val strip_prefix: string -> string -> string option
blanchet@36063
    19
  val metis_line: int -> int -> string list -> string
blanchet@36223
    20
  val metis_proof_text:
blanchet@36287
    21
    minimize_command * string * string vector * thm * int
blanchet@36281
    22
    -> string * string list
blanchet@36223
    23
  val isar_proof_text:
blanchet@36488
    24
    name_pool option * bool * int * Proof.context * int list list
blanchet@36287
    25
    -> minimize_command * string * string vector * thm * int
blanchet@36287
    26
    -> string * string list
blanchet@36223
    27
  val proof_text:
blanchet@36488
    28
    bool -> name_pool option * bool * int * Proof.context * int list list
blanchet@36287
    29
    -> minimize_command * string * string vector * thm * int
blanchet@36287
    30
    -> string * string list
paulson@24425
    31
end;
paulson@21978
    32
blanchet@35826
    33
structure Sledgehammer_Proof_Reconstruct : SLEDGEHAMMER_PROOF_RECONSTRUCT =
paulson@21978
    34
struct
paulson@21978
    35
blanchet@36478
    36
open Sledgehammer_Util
blanchet@35865
    37
open Sledgehammer_FOL_Clause
blanchet@35865
    38
open Sledgehammer_Fact_Preprocessor
paulson@21978
    39
blanchet@36281
    40
type minimize_command = string list -> string
blanchet@36281
    41
blanchet@36291
    42
fun is_ident_char c = Char.isAlphaNum c orelse c = #"_"
blanchet@36392
    43
fun is_head_digit s = Char.isDigit (String.sub (s, 0))
blanchet@36291
    44
blanchet@36402
    45
fun is_axiom_clause_number thm_names num = num <= Vector.length thm_names
blanchet@36291
    46
blanchet@36393
    47
fun ugly_name NONE s = s
blanchet@36393
    48
  | ugly_name (SOME the_pool) s =
blanchet@36393
    49
    case Symtab.lookup (snd the_pool) s of
blanchet@36393
    50
      SOME s' => s'
blanchet@36393
    51
    | NONE => s
blanchet@36393
    52
blanchet@36491
    53
(* Hack: Could return false positives (e.g., a user happens to declare a
blanchet@36491
    54
   constant called "SomeTheory.sko_means_shoe_in_swedish". *)
blanchet@36491
    55
val is_skolem_const_name = String.isPrefix skolem_prefix o Long_Name.base_name
blanchet@36491
    56
blanchet@36491
    57
fun smart_lambda v t =
blanchet@36491
    58
    Abs (case v of
blanchet@36491
    59
           Const (s, _) => if is_skolem_const_name s then "g"
blanchet@36491
    60
                           else Long_Name.base_name s
blanchet@36491
    61
         | Var ((s, _), _) => s
blanchet@36491
    62
         | Free (s, _) => s
blanchet@36491
    63
         | _ => "", fastype_of v, abstract_over (v, t))
blanchet@36491
    64
blanchet@36491
    65
fun forall_of v t = HOLogic.all_const (fastype_of v) $ smart_lambda v t
blanchet@36491
    66
fun exists_of v t = HOLogic.exists_const (fastype_of v) $ smart_lambda v t
blanchet@36491
    67
blanchet@36491
    68
datatype ('a, 'b, 'c, 'd, 'e) raw_step =
blanchet@36491
    69
  Definition of 'a * 'b * 'c |
blanchet@36491
    70
  Inference of 'a * 'd * 'e list
blanchet@36491
    71
paulson@21978
    72
(**** PARSING OF TSTP FORMAT ****)
paulson@21978
    73
blanchet@36291
    74
(* Syntax trees, either term list or formulae *)
blanchet@36486
    75
datatype node = IntLeaf of int | StrNode of string * node list
paulson@21978
    76
blanchet@36486
    77
fun atom x = StrNode (x, [])
paulson@21978
    78
blanchet@36486
    79
fun scons (x, y) = StrNode ("cons", [x, y])
blanchet@36392
    80
val slist_of = List.foldl scons (atom "nil")
paulson@21978
    81
paulson@21978
    82
(*Strings enclosed in single quotes, e.g. filenames*)
blanchet@36392
    83
val parse_quoted = $$ "'" |-- Scan.repeat (~$$ "'") --| $$ "'" >> implode;
paulson@21978
    84
paulson@21978
    85
(*Integer constants, typically proof line numbers*)
blanchet@36392
    86
val parse_integer = Scan.many1 is_head_digit >> (the o Int.fromString o implode)
paulson@21978
    87
blanchet@36369
    88
(* needed for SPASS's output format *)
blanchet@36393
    89
fun repair_bool_literal "true" = "c_True"
blanchet@36393
    90
  | repair_bool_literal "false" = "c_False"
blanchet@36393
    91
fun repair_name pool "equal" = "c_equal"
blanchet@36393
    92
  | repair_name pool s = ugly_name pool s
blanchet@36392
    93
(* Generalized first-order terms, which include file names, numbers, etc. *)
blanchet@36393
    94
(* The "x" argument is not strictly necessary, but without it Poly/ML loops
blanchet@36393
    95
   forever at compile time. *)
blanchet@36393
    96
fun parse_term pool x =
blanchet@36486
    97
     (parse_quoted >> atom
blanchet@36486
    98
   || parse_integer >> IntLeaf
blanchet@36393
    99
   || $$ "$" |-- Symbol.scan_id >> (atom o repair_bool_literal)
blanchet@36393
   100
   || (Symbol.scan_id >> repair_name pool)
blanchet@36486
   101
      -- Scan.optional ($$ "(" |-- parse_terms pool --| $$ ")") [] >> StrNode
blanchet@36393
   102
   || $$ "(" |-- parse_term pool --| $$ ")"
blanchet@36393
   103
   || $$ "[" |-- Scan.optional (parse_terms pool) [] --| $$ "]" >> slist_of) x
blanchet@36393
   104
and parse_terms pool x =
blanchet@36393
   105
  (parse_term pool ::: Scan.repeat ($$ "," |-- parse_term pool)) x
paulson@21978
   106
blanchet@36486
   107
fun negate_node u = StrNode ("c_Not", [u])
blanchet@36486
   108
fun equate_nodes u1 u2 = StrNode ("c_equal", [u1, u2])
paulson@21978
   109
blanchet@36392
   110
(* Apply equal or not-equal to a term. *)
blanchet@36486
   111
fun repair_predicate_term (u, NONE) = u
blanchet@36486
   112
  | repair_predicate_term (u1, SOME (NONE, u2)) = equate_nodes u1 u2
blanchet@36486
   113
  | repair_predicate_term (u1, SOME (SOME _, u2)) =
blanchet@36486
   114
    negate_node (equate_nodes u1 u2)
blanchet@36393
   115
fun parse_predicate_term pool =
blanchet@36393
   116
  parse_term pool -- Scan.option (Scan.option ($$ "!") --| $$ "="
blanchet@36393
   117
                                  -- parse_term pool)
blanchet@36393
   118
  >> repair_predicate_term
blanchet@36486
   119
(* Literals can involve "~", "=", and "!=". *)
blanchet@36393
   120
fun parse_literal pool x =
blanchet@36486
   121
  ($$ "~" |-- parse_literal pool >> negate_node || parse_predicate_term pool) x
paulson@21978
   122
blanchet@36393
   123
fun parse_literals pool =
blanchet@36393
   124
  parse_literal pool ::: Scan.repeat ($$ "|" |-- parse_literal pool)
paulson@21978
   125
blanchet@36486
   126
(* Clause: a list of literals separated by disjunction operators ("|"). *)
blanchet@36393
   127
fun parse_clause pool =
blanchet@36393
   128
  $$ "(" |-- parse_literals pool --| $$ ")" || Scan.single (parse_literal pool)
blanchet@36291
   129
blanchet@36486
   130
fun ints_of_node (IntLeaf n) = cons n
blanchet@36486
   131
  | ints_of_node (StrNode (_, us)) = fold ints_of_node us
blanchet@36392
   132
val parse_tstp_annotations =
blanchet@36393
   133
  Scan.optional ($$ "," |-- parse_term NONE
blanchet@36393
   134
                   --| Scan.option ($$ "," |-- parse_terms NONE)
blanchet@36486
   135
                 >> (fn source => ints_of_node source [])) []
blanchet@36486
   136
blanchet@36486
   137
fun parse_definition pool =
blanchet@36486
   138
  $$ "(" |-- parse_literal NONE --| Scan.this_string "<=>"
blanchet@36486
   139
  -- parse_clause pool --| $$ ")"
blanchet@36291
   140
blanchet@36486
   141
(* Syntax: cnf(<num>, <formula_role>, <cnf_formula> <annotations>).
blanchet@36486
   142
   The <num> could be an identifier, but we assume integers. *)
blanchet@36486
   143
fun finish_tstp_definition_line (num, (u, us)) = Definition (num, u, us)
blanchet@36486
   144
fun finish_tstp_inference_line ((num, us), deps) = Inference (num, us, deps)
blanchet@36393
   145
fun parse_tstp_line pool =
blanchet@36486
   146
     ((Scan.this_string "fof" -- $$ "(") |-- parse_integer --| $$ ","
blanchet@36486
   147
       --| Scan.this_string "definition" --| $$ "," -- parse_definition pool
blanchet@36486
   148
       --| parse_tstp_annotations --| $$ ")" --| $$ "."
blanchet@36486
   149
      >> finish_tstp_definition_line)
blanchet@36486
   150
  || ((Scan.this_string "cnf" -- $$ "(") |-- parse_integer --| $$ ","
blanchet@36486
   151
       --| Symbol.scan_id --| $$ "," -- parse_clause pool
blanchet@36486
   152
       -- parse_tstp_annotations --| $$ ")" --| $$ "."
blanchet@36486
   153
      >> finish_tstp_inference_line)
blanchet@36291
   154
blanchet@36291
   155
(**** PARSING OF SPASS OUTPUT ****)
blanchet@36291
   156
blanchet@36392
   157
(* SPASS returns clause references of the form "x.y". We ignore "y", whose role
blanchet@36392
   158
   is not clear anyway. *)
blanchet@36392
   159
val parse_dot_name = parse_integer --| $$ "." --| parse_integer
paulson@21978
   160
blanchet@36392
   161
val parse_spass_annotations =
blanchet@36392
   162
  Scan.optional ($$ ":" |-- Scan.repeat (parse_dot_name
blanchet@36392
   163
                                         --| Scan.option ($$ ","))) []
blanchet@36291
   164
blanchet@36392
   165
(* It is not clear why some literals are followed by sequences of stars. We
blanchet@36392
   166
   ignore them. *)
blanchet@36393
   167
fun parse_starred_predicate_term pool =
blanchet@36393
   168
  parse_predicate_term pool --| Scan.repeat ($$ "*" || $$ " ")
blanchet@36291
   169
blanchet@36393
   170
fun parse_horn_clause pool =
blanchet@36393
   171
  Scan.repeat (parse_starred_predicate_term pool) --| $$ "-" --| $$ ">"
blanchet@36393
   172
  -- Scan.repeat (parse_starred_predicate_term pool)
blanchet@36392
   173
  >> (fn ([], []) => [atom "c_False"]
blanchet@36486
   174
       | (clauses1, clauses2) => map negate_node clauses1 @ clauses2)
paulson@21978
   175
blanchet@36486
   176
(* Syntax: <num>[0:<inference><annotations>] ||
blanchet@36393
   177
           <cnf_formulas> -> <cnf_formulas>. *)
blanchet@36486
   178
fun finish_spass_line ((num, deps), us) = Inference (num, us, deps)
blanchet@36402
   179
fun parse_spass_line pool =
blanchet@36392
   180
  parse_integer --| $$ "[" --| $$ "0" --| $$ ":" --| Symbol.scan_id
blanchet@36392
   181
  -- parse_spass_annotations --| $$ "]" --| $$ "|" --| $$ "|"
blanchet@36393
   182
  -- parse_horn_clause pool --| $$ "."
blanchet@36486
   183
  >> finish_spass_line
blanchet@36291
   184
blanchet@36402
   185
fun parse_line pool = fst o (parse_tstp_line pool || parse_spass_line pool)
paulson@21978
   186
paulson@21978
   187
(**** INTERPRETATION OF TSTP SYNTAX TREES ****)
paulson@21978
   188
blanchet@36486
   189
exception NODE of node
paulson@21978
   190
paulson@21978
   191
(*If string s has the prefix s1, return the result of deleting it.*)
wenzelm@23139
   192
fun strip_prefix s1 s =
immler@31038
   193
  if String.isPrefix s1 s
blanchet@35865
   194
  then SOME (undo_ascii_of (String.extract (s, size s1, NONE)))
paulson@21978
   195
  else NONE;
paulson@21978
   196
paulson@21978
   197
(*Invert the table of translations between Isabelle and ATPs*)
paulson@21978
   198
val type_const_trans_table_inv =
blanchet@35865
   199
      Symtab.make (map swap (Symtab.dest type_const_trans_table));
paulson@21978
   200
paulson@21978
   201
fun invert_type_const c =
paulson@21978
   202
    case Symtab.lookup type_const_trans_table_inv c of
paulson@21978
   203
        SOME c' => c'
paulson@21978
   204
      | NONE => c;
paulson@21978
   205
blanchet@36285
   206
fun make_tvar s = TVar (("'" ^ s, 0), HOLogic.typeS);
blanchet@36285
   207
fun make_tparam s = TypeInfer.param 0 (s, HOLogic.typeS)
paulson@21978
   208
fun make_var (b,T) = Var((b,0),T);
paulson@21978
   209
paulson@21978
   210
(*Type variables are given the basic sort, HOL.type. Some will later be constrained
paulson@21978
   211
  by information from type literals, or by type inference.*)
blanchet@36486
   212
fun type_of_node (u as IntLeaf _) = raise NODE u
blanchet@36486
   213
  | type_of_node (u as StrNode (a, us)) =
blanchet@36486
   214
    let val Ts = map type_of_node us in
blanchet@36486
   215
      case strip_prefix tconst_prefix a of
blanchet@36486
   216
        SOME b => Type (invert_type_const b, Ts)
blanchet@36486
   217
      | NONE =>
blanchet@36486
   218
        if not (null us) then
blanchet@36486
   219
          raise NODE u  (*only tconsts have type arguments*)
blanchet@36486
   220
        else case strip_prefix tfree_prefix a of
blanchet@36486
   221
          SOME b => TFree ("'" ^ b, HOLogic.typeS)
blanchet@36486
   222
        | NONE =>
blanchet@36486
   223
          case strip_prefix tvar_prefix a of
blanchet@36486
   224
            SOME b => make_tvar b
blanchet@36486
   225
          | NONE => make_tparam a  (* Variable from the ATP, say "X1" *)
blanchet@36486
   226
    end
paulson@21978
   227
paulson@21978
   228
(*Invert the table of translations between Isabelle and ATPs*)
paulson@21978
   229
val const_trans_table_inv =
blanchet@36402
   230
  Symtab.update ("fequal", @{const_name "op ="})
blanchet@36402
   231
                (Symtab.make (map swap (Symtab.dest const_trans_table)))
paulson@21978
   232
blanchet@36402
   233
fun invert_const c = c |> Symtab.lookup const_trans_table_inv |> the_default c
paulson@21978
   234
paulson@21978
   235
(*The number of type arguments of a constant, zero if it's monomorphic*)
paulson@21978
   236
fun num_typargs thy s = length (Sign.const_typargs thy (s, Sign.the_const_type thy s));
paulson@21978
   237
paulson@21978
   238
(*Generates a constant, given its type arguments*)
paulson@21978
   239
fun const_of thy (a,Ts) = Const(a, Sign.const_instance thy (a,Ts));
paulson@21978
   240
blanchet@36486
   241
fun fix_atp_variable_name s =
blanchet@36486
   242
  let
blanchet@36486
   243
    fun subscript_name s n = s ^ nat_subscript n
blanchet@36486
   244
    val s = String.map Char.toLower s
blanchet@36486
   245
  in
blanchet@36486
   246
    case space_explode "_" s of
blanchet@36486
   247
      [_] => (case take_suffix Char.isDigit (String.explode s) of
blanchet@36486
   248
                (cs1 as _ :: _, cs2 as _ :: _) =>
blanchet@36486
   249
                subscript_name (String.implode cs1)
blanchet@36486
   250
                               (the (Int.fromString (String.implode cs2)))
blanchet@36486
   251
              | (_, _) => s)
blanchet@36486
   252
    | [s1, s2] => (case Int.fromString s2 of
blanchet@36486
   253
                     SOME n => subscript_name s1 n
blanchet@36486
   254
                   | NONE => s)
blanchet@36486
   255
    | _ => s
blanchet@36486
   256
  end
blanchet@36486
   257
paulson@21978
   258
(*First-order translation. No types are known for variables. HOLogic.typeT should allow
paulson@21978
   259
  them to be inferred.*)
blanchet@36486
   260
fun term_of_node args thy u =
blanchet@36486
   261
  case u of
blanchet@36486
   262
    IntLeaf _ => raise NODE u
blanchet@36486
   263
  | StrNode ("hBOOL", [u]) => term_of_node [] thy u  (* ignore hBOOL *)
blanchet@36486
   264
  | StrNode ("hAPP", [u1, u2]) => term_of_node (u2 :: args) thy u1
blanchet@36486
   265
  | StrNode (a, us) =>
blanchet@36486
   266
    case strip_prefix const_prefix a of
blanchet@36486
   267
      SOME "equal" =>
blanchet@36486
   268
      list_comb (Const (@{const_name "op ="}, HOLogic.typeT),
blanchet@36486
   269
                 map (term_of_node [] thy) us)
blanchet@36486
   270
    | SOME b =>
blanchet@36486
   271
      let
blanchet@36486
   272
        val c = invert_const b
blanchet@36486
   273
        val nterms = length us - num_typargs thy c
blanchet@36486
   274
        val ts = map (term_of_node [] thy) (take nterms us @ args)
blanchet@36486
   275
        (*Extra args from hAPP come AFTER any arguments given directly to the
blanchet@36486
   276
          constant.*)
blanchet@36486
   277
        val Ts = map type_of_node (drop nterms us)
blanchet@36486
   278
      in list_comb(const_of thy (c, Ts), ts) end
blanchet@36486
   279
    | NONE => (*a variable, not a constant*)
blanchet@36486
   280
      let
blanchet@36486
   281
        val opr =
blanchet@36486
   282
          (* a Free variable is typically a Skolem function *)
blanchet@36486
   283
          case strip_prefix fixed_var_prefix a of
blanchet@36486
   284
            SOME b => Free (b, HOLogic.typeT)
blanchet@36486
   285
          | NONE =>
blanchet@36486
   286
            case strip_prefix schematic_var_prefix a of
blanchet@36486
   287
              SOME b => make_var (b, HOLogic.typeT)
blanchet@36486
   288
            | NONE =>
blanchet@36486
   289
              (* Variable from the ATP, say "X1" *)
blanchet@36486
   290
              make_var (fix_atp_variable_name a, HOLogic.typeT)
blanchet@36486
   291
      in list_comb (opr, map (term_of_node [] thy) (us @ args)) end
paulson@21978
   292
blanchet@36392
   293
(* Type class literal applied to a type. Returns triple of polarity, class,
blanchet@36392
   294
   type. *)
blanchet@36486
   295
fun constraint_of_node pos (StrNode ("c_Not", [u])) =
blanchet@36486
   296
    constraint_of_node (not pos) u
blanchet@36486
   297
  | constraint_of_node pos u = case u of
blanchet@36486
   298
        IntLeaf _ => raise NODE u
blanchet@36486
   299
      | StrNode (a, us) =>
blanchet@36486
   300
            (case (strip_prefix class_prefix a, map type_of_node us) of
blanchet@36486
   301
                 (SOME b, [T]) => (pos, b, T)
blanchet@36486
   302
               | _ => raise NODE u)
paulson@21978
   303
paulson@21978
   304
(** Accumulate type constraints in a clause: negative type literals **)
paulson@21978
   305
blanchet@36485
   306
fun add_var (key, z)  = Vartab.map_default (key, []) (cons z)
paulson@21978
   307
blanchet@36485
   308
fun add_constraint ((false, cl, TFree(a,_)), vt) = add_var ((a,~1),cl) vt
blanchet@36485
   309
  | add_constraint ((false, cl, TVar(ix,_)), vt) = add_var (ix,cl) vt
paulson@21978
   310
  | add_constraint (_, vt) = vt;
paulson@21978
   311
blanchet@36491
   312
fun is_positive_literal (@{const Not} $ _) = false
blanchet@36402
   313
  | is_positive_literal t = true
blanchet@36402
   314
blanchet@36485
   315
fun negate_term thy (Const (@{const_name All}, T) $ Abs (s, T', t')) =
blanchet@36402
   316
    Const (@{const_name Ex}, T) $ Abs (s, T', negate_term thy t')
blanchet@36402
   317
  | negate_term thy (Const (@{const_name Ex}, T) $ Abs (s, T', t')) =
blanchet@36402
   318
    Const (@{const_name All}, T) $ Abs (s, T', negate_term thy t')
blanchet@36402
   319
  | negate_term thy (@{const "op -->"} $ t1 $ t2) =
blanchet@36402
   320
    @{const "op &"} $ t1 $ negate_term thy t2
blanchet@36402
   321
  | negate_term thy (@{const "op &"} $ t1 $ t2) =
blanchet@36402
   322
    @{const "op |"} $ negate_term thy t1 $ negate_term thy t2
blanchet@36402
   323
  | negate_term thy (@{const "op |"} $ t1 $ t2) =
blanchet@36402
   324
    @{const "op &"} $ negate_term thy t1 $ negate_term thy t2
blanchet@36486
   325
  | negate_term _ (@{const Not} $ t) = t
blanchet@36486
   326
  | negate_term _ t = @{const Not} $ t
blanchet@36402
   327
blanchet@36402
   328
fun clause_for_literals _ [] = HOLogic.false_const
blanchet@36402
   329
  | clause_for_literals _ [lit] = lit
blanchet@36402
   330
  | clause_for_literals thy lits =
blanchet@36402
   331
    case List.partition is_positive_literal lits of
blanchet@36402
   332
      (pos_lits as _ :: _, neg_lits as _ :: _) =>
blanchet@36402
   333
      @{const "op -->"}
blanchet@36402
   334
          $ foldr1 HOLogic.mk_conj (map (negate_term thy) neg_lits)
blanchet@36402
   335
          $ foldr1 HOLogic.mk_disj pos_lits
blanchet@36402
   336
    | _ => foldr1 HOLogic.mk_disj lits
blanchet@36402
   337
blanchet@36402
   338
(* Final treatment of the list of "real" literals from a clause.
blanchet@36402
   339
   No "real" literals means only type information. *)
blanchet@36402
   340
fun finish_clause _ [] = HOLogic.true_const
blanchet@36402
   341
  | finish_clause thy lits =
blanchet@36402
   342
    lits |> filter_out (curry (op =) HOLogic.false_const) |> rev
blanchet@36402
   343
         |> clause_for_literals thy
paulson@22491
   344
paulson@21978
   345
(*Accumulate sort constraints in vt, with "real" literals in lits.*)
blanchet@36486
   346
fun lits_of_nodes thy (vt, lits) [] = (vt, finish_clause thy lits)
blanchet@36486
   347
  | lits_of_nodes thy (vt, lits) (u :: us) =
blanchet@36486
   348
    lits_of_nodes thy (add_constraint (constraint_of_node true u, vt), lits) us
blanchet@36486
   349
    handle NODE _ => lits_of_nodes thy (vt, term_of_node [] thy u :: lits) us
paulson@21978
   350
paulson@21978
   351
(*Update TVars/TFrees with detected sort constraints.*)
blanchet@36393
   352
fun repair_sorts vt =
paulson@21978
   353
  let fun tysubst (Type (a, Ts)) = Type (a, map tysubst Ts)
wenzelm@33035
   354
        | tysubst (TVar (xi, s)) = TVar (xi, the_default s (Vartab.lookup vt xi))
wenzelm@33035
   355
        | tysubst (TFree (x, s)) = TFree (x, the_default s (Vartab.lookup vt (x, ~1)))
paulson@21978
   356
      fun tmsubst (Const (a, T)) = Const (a, tysubst T)
paulson@21978
   357
        | tmsubst (Free (a, T)) = Free (a, tysubst T)
paulson@21978
   358
        | tmsubst (Var (xi, T)) = Var (xi, tysubst T)
paulson@21978
   359
        | tmsubst (t as Bound _) = t
paulson@21978
   360
        | tmsubst (Abs (a, T, t)) = Abs (a, tysubst T, tmsubst t)
blanchet@36486
   361
        | tmsubst (t1 $ t2) = tmsubst t1 $ tmsubst t2
blanchet@36285
   362
  in not (Vartab.is_empty vt) ? tmsubst end;
paulson@21978
   363
blanchet@36486
   364
(* Interpret a list of syntax trees as a clause, given by "real" literals and
blanchet@36486
   365
   sort constraints. "vt" holds the initial sort constraints, from the
blanchet@36486
   366
   conjecture clauses. *)
blanchet@36486
   367
fun clause_of_nodes ctxt vt us =
blanchet@36486
   368
  let val (vt, dt) = lits_of_nodes (ProofContext.theory_of ctxt) (vt, []) us in
blanchet@36486
   369
    dt |> repair_sorts vt
blanchet@36291
   370
  end
blanchet@36486
   371
fun check_clause ctxt =
blanchet@36486
   372
  TypeInfer.constrain HOLogic.boolT
blanchet@36486
   373
  #> Syntax.check_term (ProofContext.set_mode ProofContext.mode_schematic ctxt)
blanchet@36486
   374
fun checked_clause_of_nodes ctxt = check_clause ctxt oo clause_of_nodes ctxt
paulson@21978
   375
blanchet@36486
   376
(** Global sort constraints on TFrees (from tfree_tcs) are positive unit
blanchet@36486
   377
    clauses. **)
paulson@21978
   378
blanchet@36486
   379
fun add_tfree_constraint (true, cl, TFree (a, _)) = add_var ((a, ~1), cl)
blanchet@36486
   380
  | add_tfree_constraint _ = I
paulson@21978
   381
fun tfree_constraints_of_clauses vt [] = vt
blanchet@36486
   382
  | tfree_constraints_of_clauses vt ([lit] :: uss) =
blanchet@36486
   383
    (tfree_constraints_of_clauses (add_tfree_constraint
blanchet@36486
   384
                                          (constraint_of_node true lit) vt) uss
blanchet@36486
   385
     handle NODE _ => (* Not a positive type constraint? Ignore the literal. *)
blanchet@36486
   386
     tfree_constraints_of_clauses vt uss)
blanchet@36486
   387
  | tfree_constraints_of_clauses vt (_ :: uss) =
blanchet@36486
   388
    tfree_constraints_of_clauses vt uss
paulson@21978
   389
paulson@21978
   390
paulson@21978
   391
(**** Translation of TSTP files to Isar Proofs ****)
paulson@21978
   392
blanchet@36486
   393
fun unvarify_term (Var ((s, 0), T)) = Free (s, T)
blanchet@36486
   394
  | unvarify_term t = raise TERM ("unvarify_term: non-Var", [t])
paulson@21978
   395
blanchet@36486
   396
fun clauses_in_lines (Definition (_, u, us)) = u :: us
blanchet@36486
   397
  | clauses_in_lines (Inference (_, us, _)) = us
paulson@21978
   398
blanchet@36486
   399
fun decode_line vt (Definition (num, u, us)) ctxt =
blanchet@36486
   400
    let
blanchet@36486
   401
      val cl1 = clause_of_nodes ctxt vt [u]
blanchet@36486
   402
      val vars = snd (strip_comb cl1)
blanchet@36486
   403
      val frees = map unvarify_term vars
blanchet@36486
   404
      val unvarify_args = subst_atomic (vars ~~ frees)
blanchet@36486
   405
      val cl2 = clause_of_nodes ctxt vt us
blanchet@36486
   406
      val (cl1, cl2) =
blanchet@36486
   407
        HOLogic.eq_const HOLogic.typeT $ cl1 $ cl2
blanchet@36486
   408
        |> unvarify_args |> check_clause ctxt |> HOLogic.dest_eq
blanchet@36486
   409
    in
blanchet@36486
   410
      (Definition (num, cl1, cl2),
blanchet@36486
   411
       fold Variable.declare_term (maps OldTerm.term_frees [cl1, cl2]) ctxt)
blanchet@36486
   412
    end
blanchet@36486
   413
  | decode_line vt (Inference (num, us, deps)) ctxt =
blanchet@36486
   414
    let val cl = us |> clause_of_nodes ctxt vt |> check_clause ctxt in
blanchet@36486
   415
      (Inference (num, cl, deps),
blanchet@36486
   416
       fold Variable.declare_term (OldTerm.term_frees cl) ctxt)
blanchet@36486
   417
    end
blanchet@36486
   418
fun decode_lines ctxt lines =
blanchet@36486
   419
  let
blanchet@36486
   420
    val vt = tfree_constraints_of_clauses Vartab.empty
blanchet@36486
   421
                                          (map clauses_in_lines lines)
blanchet@36486
   422
  in #1 (fold_map (decode_line vt) lines ctxt) end
paulson@21978
   423
blanchet@36486
   424
fun aint_inference _ (Definition _) = true
blanchet@36486
   425
  | aint_inference t (Inference (_, t', _)) = not (t aconv t')
blanchet@36486
   426
blanchet@36486
   427
(* No "real" literals means only type information (tfree_tcs, clsrel, or
blanchet@36486
   428
   clsarity). *)
blanchet@36486
   429
val is_only_type_information = curry (op aconv) HOLogic.true_const
blanchet@36486
   430
blanchet@36486
   431
fun replace_one_dep (old, new) dep = if dep = old then new else [dep]
blanchet@36486
   432
fun replace_deps_in_line _ (line as Definition _) = line
blanchet@36486
   433
  | replace_deps_in_line p (Inference (num, t, deps)) =
blanchet@36486
   434
    Inference (num, t, fold (union (op =) o replace_one_dep p) deps [])
paulson@21978
   435
paulson@22491
   436
(*Discard axioms; consolidate adjacent lines that prove the same clause, since they differ
paulson@22491
   437
  only in type information.*)
blanchet@36486
   438
fun add_line _ (line as Definition _) lines = line :: lines
blanchet@36486
   439
  | add_line thm_names (Inference (num, t, [])) lines =
blanchet@36486
   440
    (* No dependencies: axiom or conjecture clause *)
blanchet@36486
   441
    if is_axiom_clause_number thm_names num then
blanchet@36486
   442
      (* Axioms are not proof lines. *)
blanchet@36486
   443
      if is_only_type_information t then
blanchet@36486
   444
        map (replace_deps_in_line (num, [])) lines
blanchet@36486
   445
      (* Is there a repetition? If so, replace later line by earlier one. *)
blanchet@36486
   446
      else case take_prefix (aint_inference t) lines of
blanchet@36486
   447
        (_, []) => lines (*no repetition of proof line*)
blanchet@36486
   448
      | (pre, Inference (num', _, _) :: post) =>
blanchet@36486
   449
        pre @ map (replace_deps_in_line (num', [num])) post
blanchet@36486
   450
    else
blanchet@36486
   451
      Inference (num, t, []) :: lines
blanchet@36486
   452
  | add_line _ (Inference (num, t, deps)) lines =
blanchet@36486
   453
    (* Type information will be deleted later; skip repetition test. *)
blanchet@36486
   454
    if is_only_type_information t then
blanchet@36486
   455
      Inference (num, t, deps) :: lines
blanchet@36486
   456
    (* Is there a repetition? If so, replace later line by earlier one. *)
blanchet@36486
   457
    else case take_prefix (aint_inference t) lines of
blanchet@36486
   458
      (* FIXME: Doesn't this code risk conflating proofs involving different
blanchet@36486
   459
         types?? *)
blanchet@36486
   460
       (_, []) => Inference (num, t, deps) :: lines
blanchet@36486
   461
     | (pre, Inference (num', t', _) :: post) =>
blanchet@36486
   462
       Inference (num, t', deps) ::
blanchet@36486
   463
       pre @ map (replace_deps_in_line (num', [num])) post
paulson@22044
   464
blanchet@36486
   465
(* Recursively delete empty lines (type information) from the proof. *)
blanchet@36486
   466
fun add_nontrivial_line (Inference (num, t, [])) lines =
blanchet@36486
   467
    if is_only_type_information t then delete_dep num lines
blanchet@36486
   468
    else Inference (num, t, []) :: lines
blanchet@36486
   469
  | add_nontrivial_line line lines = line :: lines
blanchet@36395
   470
and delete_dep num lines =
blanchet@36486
   471
  fold_rev add_nontrivial_line (map (replace_deps_in_line (num, [])) lines) []
blanchet@36486
   472
blanchet@36486
   473
fun is_bad_free (Free (a, _)) = String.isPrefix skolem_prefix a
blanchet@36486
   474
  | is_bad_free _ = false
paulson@22470
   475
blanchet@36486
   476
fun add_desired_line _ _ (line as Definition _) (j, lines) = (j, line :: lines)
blanchet@36486
   477
  | add_desired_line ctxt _ (Inference (num, t, [])) (j, lines) =
blanchet@36486
   478
    (j, Inference (num, t, []) :: lines)  (* conjecture clauses must be kept *)
blanchet@36486
   479
  | add_desired_line ctxt shrink_factor (Inference (num, t, deps)) (j, lines) =
blanchet@36402
   480
    (j + 1,
blanchet@36486
   481
     if is_only_type_information t orelse
blanchet@36486
   482
        not (null (Term.add_tvars t [])) orelse
blanchet@36486
   483
        exists_subterm is_bad_free t orelse
blanchet@36474
   484
        (length deps < 2 orelse j mod shrink_factor <> 0) then
blanchet@36486
   485
       map (replace_deps_in_line (num, deps)) lines  (* delete line *)
blanchet@36402
   486
     else
blanchet@36486
   487
       Inference (num, t, deps) :: lines)
paulson@21978
   488
blanchet@36402
   489
(** EXTRACTING LEMMAS **)
paulson@21979
   490
blanchet@36223
   491
(* A list consisting of the first number in each line is returned.
blanchet@36395
   492
   TSTP: Interesting lines have the form "cnf(108, axiom, ...)", where the
blanchet@36223
   493
   number (108) is extracted.
blanchet@36395
   494
   SPASS: Lines have the form "108[0:Inp] ...", where the first number (108) is
blanchet@36223
   495
   extracted. *)
blanchet@36402
   496
fun extract_clause_numbers_in_atp_proof atp_proof =
blanchet@35865
   497
  let
blanchet@36395
   498
    val tokens_of = String.tokens (not o is_ident_char)
blanchet@36402
   499
    fun extract_num ("cnf" :: num :: "axiom" :: _) = Int.fromString num
blanchet@36402
   500
      | extract_num ("cnf" :: num :: "negated_conjecture" :: _) =
blanchet@36402
   501
        Int.fromString num
blanchet@36395
   502
      | extract_num (num :: "0" :: "Inp" :: _) = Int.fromString num
blanchet@36395
   503
      | extract_num _ = NONE
blanchet@36402
   504
  in atp_proof |> split_lines |> map_filter (extract_num o tokens_of) end
wenzelm@33310
   505
  
blanchet@36395
   506
(* Used to label theorems chained into the Sledgehammer call (or rather
blanchet@36395
   507
   goal?) *)
blanchet@36395
   508
val chained_hint = "sledgehammer_chained"
blanchet@35865
   509
blanchet@36063
   510
fun apply_command _ 1 = "by "
blanchet@36063
   511
  | apply_command 1 _ = "apply "
blanchet@36063
   512
  | apply_command i _ = "prefer " ^ string_of_int i ^ " apply "
blanchet@36063
   513
fun metis_command i n [] =
blanchet@36063
   514
    apply_command i n ^ "metis"
blanchet@36063
   515
  | metis_command i n xs =
blanchet@36063
   516
    apply_command i n ^ "(metis " ^ space_implode " " xs ^ ")"
blanchet@36063
   517
fun metis_line i n xs =
blanchet@36063
   518
  "Try this command: " ^
blanchet@36063
   519
  Markup.markup Markup.sendback (metis_command i n xs) ^ ".\n" 
blanchet@36281
   520
fun minimize_line _ [] = ""
blanchet@36281
   521
  | minimize_line minimize_command facts =
blanchet@36281
   522
    case minimize_command facts of
blanchet@36281
   523
      "" => ""
blanchet@36281
   524
    | command =>
blanchet@36065
   525
      "To minimize the number of lemmas, try this command: " ^
blanchet@36281
   526
      Markup.markup Markup.sendback command ^ ".\n"
immler@31840
   527
blanchet@36402
   528
fun metis_proof_text (minimize_command, atp_proof, thm_names, goal, i) =
blanchet@36063
   529
  let
blanchet@36231
   530
    val lemmas =
blanchet@36402
   531
      atp_proof |> extract_clause_numbers_in_atp_proof
blanchet@36402
   532
                |> filter (is_axiom_clause_number thm_names)
blanchet@36402
   533
                |> map (fn i => Vector.sub (thm_names, i - 1))
blanchet@36402
   534
                |> filter_out (fn s => s = "??.unknown" orelse s = chained_hint)
blanchet@36402
   535
                |> sort_distinct string_ord
blanchet@36063
   536
    val n = Logic.count_prems (prop_of goal)
blanchet@36395
   537
  in (metis_line i n lemmas ^ minimize_line minimize_command lemmas, lemmas) end
immler@31037
   538
blanchet@36486
   539
val is_valid_line =
blanchet@36486
   540
  String.isPrefix "fof(" orf String.isPrefix "cnf(" orf String.isSubstring "||"
blanchet@36402
   541
blanchet@36486
   542
(** Isar proof construction and manipulation **)
blanchet@36486
   543
blanchet@36486
   544
fun merge_fact_sets (ls1, ss1) (ls2, ss2) =
blanchet@36486
   545
  (union (op =) ls1 ls2, union (op =) ss1 ss2)
blanchet@36402
   546
blanchet@36402
   547
type label = string * int
blanchet@36402
   548
type facts = label list * string list
blanchet@36402
   549
blanchet@36402
   550
datatype qualifier = Show | Then | Moreover | Ultimately
blanchet@36291
   551
blanchet@36402
   552
datatype step =
blanchet@36478
   553
  Fix of (string * typ) list |
blanchet@36486
   554
  Let of term * term |
blanchet@36402
   555
  Assume of label * term |
blanchet@36402
   556
  Have of qualifier list * label * term * byline
blanchet@36402
   557
and byline =
blanchet@36402
   558
  Facts of facts |
blanchet@36402
   559
  CaseSplit of step list list * facts
blanchet@36402
   560
blanchet@36402
   561
val raw_prefix = "X"
blanchet@36402
   562
val assum_prefix = "A"
blanchet@36402
   563
val fact_prefix = "F"
blanchet@36402
   564
blanchet@36475
   565
fun add_fact_from_dep thm_names num =
blanchet@36475
   566
  if is_axiom_clause_number thm_names num then
blanchet@36480
   567
    apsnd (insert (op =) (Vector.sub (thm_names, num - 1)))
blanchet@36475
   568
  else
blanchet@36480
   569
    apfst (insert (op =) (raw_prefix, num))
blanchet@36402
   570
blanchet@36491
   571
fun forall_vars t = fold_rev forall_of (map Var (Term.add_vars t [])) t
blanchet@36491
   572
blanchet@36486
   573
fun step_for_line _ _ (Definition (num, t1, t2)) = Let (t1, t2)
blanchet@36486
   574
  | step_for_line _ _ (Inference (num, t, [])) = Assume ((raw_prefix, num), t)
blanchet@36486
   575
  | step_for_line thm_names j (Inference (num, t, deps)) =
blanchet@36486
   576
    Have (if j = 1 then [Show] else [], (raw_prefix, num),
blanchet@36491
   577
          forall_vars t,
blanchet@36475
   578
          Facts (fold (add_fact_from_dep thm_names) deps ([], [])))
blanchet@36291
   579
blanchet@36291
   580
fun strip_spaces_in_list [] = ""
blanchet@36402
   581
  | strip_spaces_in_list [c1] = if Char.isSpace c1 then "" else str c1
blanchet@36402
   582
  | strip_spaces_in_list [c1, c2] =
blanchet@36402
   583
    strip_spaces_in_list [c1] ^ strip_spaces_in_list [c2]
blanchet@36291
   584
  | strip_spaces_in_list (c1 :: c2 :: c3 :: cs) =
blanchet@36291
   585
    if Char.isSpace c1 then
blanchet@36291
   586
      strip_spaces_in_list (c2 :: c3 :: cs)
blanchet@36291
   587
    else if Char.isSpace c2 then
blanchet@36291
   588
      if Char.isSpace c3 then
blanchet@36291
   589
        strip_spaces_in_list (c1 :: c3 :: cs)
blanchet@36291
   590
      else
blanchet@36402
   591
        str c1 ^ (if forall is_ident_char [c1, c3] then " " else "") ^
blanchet@36291
   592
        strip_spaces_in_list (c3 :: cs)
blanchet@36291
   593
    else
blanchet@36291
   594
      str c1 ^ strip_spaces_in_list (c2 :: c3 :: cs)
blanchet@36291
   595
val strip_spaces = strip_spaces_in_list o String.explode
blanchet@36291
   596
blanchet@36474
   597
fun proof_from_atp_proof pool ctxt shrink_factor atp_proof thm_names frees =
blanchet@36402
   598
  let
blanchet@36486
   599
    val lines =
blanchet@36486
   600
      atp_proof
blanchet@36486
   601
      |> split_lines |> map strip_spaces |> filter is_valid_line
blanchet@36486
   602
      |> map (parse_line pool o explode)
blanchet@36486
   603
      |> decode_lines ctxt
blanchet@36486
   604
      |> rpair [] |-> fold_rev (add_line thm_names)
blanchet@36486
   605
      |> rpair [] |-> fold_rev add_nontrivial_line
blanchet@36486
   606
      |> rpair (0, []) |-> fold_rev (add_desired_line ctxt shrink_factor)
blanchet@36486
   607
      |> snd
blanchet@36402
   608
  in
blanchet@36402
   609
    (if null frees then [] else [Fix frees]) @
blanchet@36486
   610
    map2 (step_for_line thm_names) (length lines downto 1) lines
blanchet@36402
   611
  end
blanchet@36402
   612
blanchet@36402
   613
val indent_size = 2
blanchet@36402
   614
val no_label = ("", ~1)
blanchet@36402
   615
blanchet@36402
   616
fun no_show qs = not (member (op =) qs Show)
blanchet@36402
   617
blanchet@36402
   618
(* When redirecting proofs, we keep information about the labels seen so far in
blanchet@36402
   619
   the "backpatches" data structure. The first component indicates which facts
blanchet@36402
   620
   should be associated with forthcoming proof steps. The second component is a
blanchet@36402
   621
   pair ("keep_ls", "drop_ls"), where "keep_ls" are the labels to keep and
blanchet@36402
   622
   "drop_ls" are those that should be dropped in a case split. *)
blanchet@36402
   623
type backpatches = (label * facts) list * (label list * label list)
blanchet@36402
   624
blanchet@36402
   625
fun using_of_step (Have (_, _, _, by)) =
blanchet@36402
   626
    (case by of
blanchet@36402
   627
       Facts (ls, _) => ls
blanchet@36402
   628
     | CaseSplit (proofs, (ls, _)) => fold (union (op =) o using_of) proofs ls)
blanchet@36402
   629
  | using_of_step _ = []
blanchet@36402
   630
and using_of proof = fold (union (op =) o using_of_step) proof []
blanchet@36402
   631
blanchet@36402
   632
fun new_labels_of_step (Fix _) = []
blanchet@36486
   633
  | new_labels_of_step (Let _) = []
blanchet@36402
   634
  | new_labels_of_step (Assume (l, _)) = [l]
blanchet@36402
   635
  | new_labels_of_step (Have (_, l, _, _)) = [l]
blanchet@36402
   636
val new_labels_of = maps new_labels_of_step
blanchet@36402
   637
blanchet@36402
   638
val join_proofs =
blanchet@36402
   639
  let
blanchet@36402
   640
    fun aux _ [] = NONE
blanchet@36402
   641
      | aux proof_tail (proofs as (proof1 :: _)) =
blanchet@36402
   642
        if exists null proofs then
blanchet@36402
   643
          NONE
blanchet@36402
   644
        else if forall (curry (op =) (hd proof1) o hd) (tl proofs) then
blanchet@36402
   645
          aux (hd proof1 :: proof_tail) (map tl proofs)
blanchet@36402
   646
        else case hd proof1 of
blanchet@36402
   647
          Have ([], l, t, by) =>
blanchet@36402
   648
          if forall (fn Have ([], l', t', _) :: _ => (l, t) = (l', t')
blanchet@36402
   649
                      | _ => false) (tl proofs) andalso
blanchet@36402
   650
             not (exists (member (op =) (maps new_labels_of proofs))
blanchet@36402
   651
                         (using_of proof_tail)) then
blanchet@36402
   652
            SOME (l, t, map rev proofs, proof_tail)
blanchet@36402
   653
          else
blanchet@36402
   654
            NONE
blanchet@36402
   655
        | _ => NONE
blanchet@36402
   656
  in aux [] o map rev end
blanchet@36402
   657
blanchet@36402
   658
fun case_split_qualifiers proofs =
blanchet@36402
   659
  case length proofs of
blanchet@36402
   660
    0 => []
blanchet@36402
   661
  | 1 => [Then]
blanchet@36402
   662
  | _ => [Ultimately]
blanchet@36402
   663
blanchet@36402
   664
val index_in_shape = find_index o exists o curry (op =)
blanchet@36402
   665
blanchet@36491
   666
fun redirect_proof thy conjecture_shape hyp_ts concl_t proof =
wenzelm@33310
   667
  let
blanchet@36402
   668
    val concl_ls = map (pair raw_prefix) (List.last conjecture_shape)
blanchet@36402
   669
    fun find_hyp (_, j) = nth hyp_ts (index_in_shape j conjecture_shape)
blanchet@36402
   670
    fun first_pass ([], contra) = ([], contra)
blanchet@36491
   671
      | first_pass ((step as Fix _) :: proof, contra) =
blanchet@36491
   672
        first_pass (proof, contra) |>> cons step
blanchet@36491
   673
      | first_pass ((step as Let _) :: proof, contra) =
blanchet@36491
   674
        first_pass (proof, contra) |>> cons step
blanchet@36491
   675
      | first_pass ((step as Assume (l, t)) :: proof, contra) =
blanchet@36402
   676
        if member (op =) concl_ls l then
blanchet@36491
   677
          first_pass (proof, contra ||> cons step)
blanchet@36402
   678
        else
blanchet@36402
   679
          first_pass (proof, contra) |>> cons (Assume (l, find_hyp l))
blanchet@36491
   680
      | first_pass ((step as Have (qs, l, t, Facts (ls, ss))) :: proof,
blanchet@36491
   681
                    contra) =
blanchet@36402
   682
        if exists (member (op =) (fst contra)) ls then
blanchet@36491
   683
          first_pass (proof, contra |>> cons l ||> cons step)
blanchet@36402
   684
        else
blanchet@36491
   685
          first_pass (proof, contra) |>> cons step
blanchet@36402
   686
      | first_pass _ = raise Fail "malformed proof"
blanchet@36402
   687
    val (proof_top, (contra_ls, contra_proof)) =
blanchet@36402
   688
      first_pass (proof, (concl_ls, []))
blanchet@36402
   689
    val backpatch_label = the_default ([], []) oo AList.lookup (op =) o fst
blanchet@36402
   690
    fun backpatch_labels patches ls =
blanchet@36402
   691
      fold merge_fact_sets (map (backpatch_label patches) ls) ([], [])
blanchet@36402
   692
    fun second_pass end_qs ([], assums, patches) =
blanchet@36402
   693
        ([Have (end_qs, no_label,
blanchet@36402
   694
                if length assums < length concl_ls then
blanchet@36491
   695
                  clause_for_literals thy (map (negate_term thy o fst) assums)
blanchet@36402
   696
                else
blanchet@36402
   697
                  concl_t,
blanchet@36402
   698
                Facts (backpatch_labels patches (map snd assums)))], patches)
blanchet@36402
   699
      | second_pass end_qs (Assume (l, t) :: proof, assums, patches) =
blanchet@36402
   700
        second_pass end_qs (proof, (t, l) :: assums, patches)
blanchet@36402
   701
      | second_pass end_qs (Have (qs, l, t, Facts (ls, ss)) :: proof, assums,
blanchet@36402
   702
                            patches) =
blanchet@36402
   703
        if member (op =) (snd (snd patches)) l andalso
blanchet@36402
   704
           not (AList.defined (op =) (fst patches) l) then
blanchet@36402
   705
          second_pass end_qs (proof, assums, patches ||> apsnd (append ls))
blanchet@36402
   706
        else
blanchet@36402
   707
          (case List.partition (member (op =) contra_ls) ls of
blanchet@36402
   708
             ([contra_l], co_ls) =>
blanchet@36402
   709
             if no_show qs then
blanchet@36402
   710
               second_pass end_qs
blanchet@36402
   711
                           (proof, assums,
blanchet@36402
   712
                            patches |>> cons (contra_l, (l :: co_ls, ss)))
blanchet@36402
   713
               |>> cons (if member (op =) (fst (snd patches)) l then
blanchet@36491
   714
                           Assume (l, negate_term thy t)
blanchet@36402
   715
                         else
blanchet@36491
   716
                           Have (qs, l, negate_term thy t,
blanchet@36402
   717
                                 Facts (backpatch_label patches l)))
blanchet@36402
   718
             else
blanchet@36402
   719
               second_pass end_qs (proof, assums,
blanchet@36402
   720
                                   patches |>> cons (contra_l, (co_ls, ss)))
blanchet@36402
   721
           | (contra_ls as _ :: _, co_ls) =>
blanchet@36402
   722
             let
blanchet@36402
   723
               val proofs =
blanchet@36402
   724
                 map_filter
blanchet@36402
   725
                     (fn l =>
blanchet@36402
   726
                         if member (op =) concl_ls l then
blanchet@36402
   727
                           NONE
blanchet@36402
   728
                         else
blanchet@36402
   729
                           let
blanchet@36402
   730
                             val drop_ls = filter (curry (op <>) l) contra_ls
blanchet@36402
   731
                           in
blanchet@36402
   732
                             second_pass []
blanchet@36402
   733
                                 (proof, assums,
blanchet@36402
   734
                                  patches ||> apfst (insert (op =) l)
blanchet@36402
   735
                                          ||> apsnd (union (op =) drop_ls))
blanchet@36402
   736
                             |> fst |> SOME
blanchet@36402
   737
                           end) contra_ls
blanchet@36402
   738
               val facts = (co_ls, [])
blanchet@36402
   739
             in
blanchet@36402
   740
               (case join_proofs proofs of
blanchet@36402
   741
                  SOME (l, t, proofs, proof_tail) =>
blanchet@36402
   742
                  Have (case_split_qualifiers proofs @
blanchet@36402
   743
                        (if null proof_tail then end_qs else []), l, t,
blanchet@36402
   744
                        CaseSplit (proofs, facts)) :: proof_tail
blanchet@36402
   745
                | NONE =>
blanchet@36402
   746
                  [Have (case_split_qualifiers proofs @ end_qs, no_label,
blanchet@36402
   747
                         concl_t, CaseSplit (proofs, facts))],
blanchet@36402
   748
                patches)
blanchet@36402
   749
             end
blanchet@36402
   750
           | _ => raise Fail "malformed proof")
blanchet@36402
   751
       | second_pass _ _ = raise Fail "malformed proof"
blanchet@36486
   752
    val proof_bottom =
blanchet@36486
   753
      second_pass [Show] (contra_proof, [], ([], ([], []))) |> fst
blanchet@36402
   754
  in proof_top @ proof_bottom end
blanchet@36402
   755
blanchet@36402
   756
val kill_duplicate_assumptions_in_proof =
blanchet@36402
   757
  let
blanchet@36402
   758
    fun relabel_facts subst =
blanchet@36402
   759
      apfst (map (fn l => AList.lookup (op =) subst l |> the_default l))
blanchet@36491
   760
    fun do_step (step as Assume (l, t)) (proof, subst, assums) =
blanchet@36402
   761
        (case AList.lookup (op aconv) assums t of
blanchet@36402
   762
           SOME l' => (proof, (l', l) :: subst, assums)
blanchet@36491
   763
         | NONE => (step :: proof, subst, (t, l) :: assums))
blanchet@36402
   764
      | do_step (Have (qs, l, t, by)) (proof, subst, assums) =
blanchet@36402
   765
        (Have (qs, l, t,
blanchet@36402
   766
               case by of
blanchet@36402
   767
                 Facts facts => Facts (relabel_facts subst facts)
blanchet@36402
   768
               | CaseSplit (proofs, facts) =>
blanchet@36402
   769
                 CaseSplit (map do_proof proofs, relabel_facts subst facts)) ::
blanchet@36402
   770
         proof, subst, assums)
blanchet@36491
   771
      | do_step step (proof, subst, assums) = (step :: proof, subst, assums)
blanchet@36402
   772
    and do_proof proof = fold do_step proof ([], [], []) |> #1 |> rev
blanchet@36402
   773
  in do_proof end
blanchet@36402
   774
blanchet@36491
   775
fun unskolemize_term t =
blanchet@36491
   776
  fold exists_of (Term.add_consts t []
blanchet@36491
   777
                  |> filter (is_skolem_const_name o fst) |> map Const) t
blanchet@36491
   778
blanchet@36491
   779
fun unskolemize_step (Have (qs, l, t, by)) =
blanchet@36491
   780
    Have (qs, l, unskolemize_term t, by)
blanchet@36491
   781
  | unskolemize_step step = step
blanchet@36491
   782
blanchet@36402
   783
val then_chain_proof =
blanchet@36402
   784
  let
blanchet@36402
   785
    fun aux _ [] = []
blanchet@36491
   786
      | aux _ ((step as Assume (l, _)) :: proof) = step :: aux l proof
blanchet@36402
   787
      | aux l' (Have (qs, l, t, by) :: proof) =
blanchet@36402
   788
        (case by of
blanchet@36402
   789
           Facts (ls, ss) =>
blanchet@36402
   790
           Have (if member (op =) ls l' then
blanchet@36402
   791
                   (Then :: qs, l, t,
blanchet@36402
   792
                    Facts (filter_out (curry (op =) l') ls, ss))
blanchet@36402
   793
                 else
blanchet@36402
   794
                   (qs, l, t, Facts (ls, ss)))
blanchet@36402
   795
         | CaseSplit (proofs, facts) =>
blanchet@36402
   796
           Have (qs, l, t, CaseSplit (map (aux no_label) proofs, facts))) ::
blanchet@36402
   797
        aux l proof
blanchet@36491
   798
      | aux _ (step :: proof) = step :: aux no_label proof
blanchet@36402
   799
  in aux no_label end
blanchet@36402
   800
blanchet@36402
   801
fun kill_useless_labels_in_proof proof =
blanchet@36402
   802
  let
blanchet@36402
   803
    val used_ls = using_of proof
blanchet@36402
   804
    fun do_label l = if member (op =) used_ls l then l else no_label
blanchet@36486
   805
    fun kill (Assume (l, t)) = Assume (do_label l, t)
blanchet@36402
   806
      | kill (Have (qs, l, t, by)) =
blanchet@36402
   807
        Have (qs, do_label l, t,
blanchet@36402
   808
              case by of
blanchet@36402
   809
                CaseSplit (proofs, facts) =>
blanchet@36402
   810
                CaseSplit (map (map kill) proofs, facts)
blanchet@36402
   811
              | _ => by)
blanchet@36491
   812
      | kill step = step
blanchet@36402
   813
  in map kill proof end
blanchet@36402
   814
blanchet@36402
   815
fun prefix_for_depth n = replicate_string (n + 1)
blanchet@36402
   816
blanchet@36402
   817
val relabel_proof =
blanchet@36402
   818
  let
blanchet@36402
   819
    fun aux _ _ _ [] = []
blanchet@36402
   820
      | aux subst depth (next_assum, next_fact) (Assume (l, t) :: proof) =
blanchet@36402
   821
        if l = no_label then
blanchet@36402
   822
          Assume (l, t) :: aux subst depth (next_assum, next_fact) proof
blanchet@36402
   823
        else
blanchet@36402
   824
          let val l' = (prefix_for_depth depth assum_prefix, next_assum) in
blanchet@36402
   825
            Assume (l', t) ::
blanchet@36402
   826
            aux ((l, l') :: subst) depth (next_assum + 1, next_fact) proof
blanchet@36402
   827
          end
blanchet@36402
   828
      | aux subst depth (next_assum, next_fact) (Have (qs, l, t, by) :: proof) =
blanchet@36402
   829
        let
blanchet@36402
   830
          val (l', subst, next_fact) =
blanchet@36402
   831
            if l = no_label then
blanchet@36402
   832
              (l, subst, next_fact)
blanchet@36402
   833
            else
blanchet@36402
   834
              let
blanchet@36402
   835
                val l' = (prefix_for_depth depth fact_prefix, next_fact)
blanchet@36402
   836
              in (l', (l, l') :: subst, next_fact + 1) end
blanchet@36486
   837
          val relabel_facts = apfst (map_filter (AList.lookup (op =) subst))
blanchet@36402
   838
          val by =
blanchet@36402
   839
            case by of
blanchet@36402
   840
              Facts facts => Facts (relabel_facts facts)
blanchet@36402
   841
            | CaseSplit (proofs, facts) =>
blanchet@36402
   842
              CaseSplit (map (aux subst (depth + 1) (1, 1)) proofs,
blanchet@36402
   843
                         relabel_facts facts)
blanchet@36402
   844
        in
blanchet@36402
   845
          Have (qs, l', t, by) ::
blanchet@36402
   846
          aux subst depth (next_assum, next_fact) proof
blanchet@36402
   847
        end
blanchet@36491
   848
      | aux subst depth nextp (step :: proof) =
blanchet@36491
   849
        step :: aux subst depth nextp proof
blanchet@36402
   850
  in aux [] 0 (1, 1) end
blanchet@36402
   851
blanchet@36488
   852
fun string_for_proof ctxt i n =
blanchet@36402
   853
  let
blanchet@36478
   854
    fun fix_print_mode f =
blanchet@36478
   855
      PrintMode.setmp (filter (curry (op =) Symbol.xsymbolsN)
blanchet@36478
   856
                      (print_mode_value ())) f
blanchet@36402
   857
    fun do_indent ind = replicate_string (ind * indent_size) " "
blanchet@36478
   858
    fun do_free (s, T) =
blanchet@36478
   859
      maybe_quote s ^ " :: " ^
blanchet@36478
   860
      maybe_quote (fix_print_mode (Syntax.string_of_typ ctxt) T)
blanchet@36402
   861
    fun do_raw_label (s, j) = s ^ string_of_int j
blanchet@36402
   862
    fun do_label l = if l = no_label then "" else do_raw_label l ^ ": "
blanchet@36402
   863
    fun do_have qs =
blanchet@36402
   864
      (if member (op =) qs Moreover then "moreover " else "") ^
blanchet@36402
   865
      (if member (op =) qs Ultimately then "ultimately " else "") ^
blanchet@36402
   866
      (if member (op =) qs Then then
blanchet@36402
   867
         if member (op =) qs Show then "thus" else "hence"
blanchet@36402
   868
       else
blanchet@36402
   869
         if member (op =) qs Show then "show" else "have")
blanchet@36478
   870
    val do_term = maybe_quote o fix_print_mode (Syntax.string_of_term ctxt)
blanchet@36402
   871
    fun do_using [] = ""
blanchet@36402
   872
      | do_using ls = "using " ^ (space_implode " " (map do_raw_label ls)) ^ " "
blanchet@36486
   873
    fun do_by_facts [] = "by metis"
blanchet@36486
   874
      | do_by_facts ss = "by (metis " ^ space_implode " " ss ^ ")"
blanchet@36486
   875
    fun do_facts (ls, ss) = do_using ls ^ do_by_facts ss
blanchet@36478
   876
    and do_step ind (Fix xs) =
blanchet@36478
   877
        do_indent ind ^ "fix " ^ space_implode " and " (map do_free xs) ^ "\n"
blanchet@36486
   878
      | do_step ind (Let (t1, t2)) =
blanchet@36486
   879
        do_indent ind ^ "let " ^ do_term t1 ^ " = " ^ do_term t2 ^ "\n"
blanchet@36402
   880
      | do_step ind (Assume (l, t)) =
blanchet@36402
   881
        do_indent ind ^ "assume " ^ do_label l ^ do_term t ^ "\n"
blanchet@36402
   882
      | do_step ind (Have (qs, l, t, Facts facts)) =
blanchet@36402
   883
        do_indent ind ^ do_have qs ^ " " ^
blanchet@36479
   884
        do_label l ^ do_term t ^ " " ^ do_facts facts ^ "\n"
blanchet@36402
   885
      | do_step ind (Have (qs, l, t, CaseSplit (proofs, facts))) =
blanchet@36402
   886
        space_implode (do_indent ind ^ "moreover\n")
blanchet@36402
   887
                      (map (do_block ind) proofs) ^
blanchet@36479
   888
        do_indent ind ^ do_have qs ^ " " ^ do_label l ^ do_term t ^ " " ^
blanchet@36478
   889
        do_facts facts ^ "\n"
blanchet@36402
   890
    and do_steps prefix suffix ind steps =
blanchet@36402
   891
      let val s = implode (map (do_step ind) steps) in
blanchet@36402
   892
        replicate_string (ind * indent_size - size prefix) " " ^ prefix ^
blanchet@36402
   893
        String.extract (s, ind * indent_size,
blanchet@36402
   894
                        SOME (size s - ind * indent_size - 1)) ^
blanchet@36402
   895
        suffix ^ "\n"
blanchet@36402
   896
      end
blanchet@36402
   897
    and do_block ind proof = do_steps "{ " " }" (ind + 1) proof
blanchet@36480
   898
    (* One-step proofs are pointless; better use the Metis one-liner. *)
blanchet@36480
   899
    and do_proof [_] = ""
blanchet@36480
   900
      | do_proof proof =
blanchet@36480
   901
        (if i <> 1 then "prefer " ^ string_of_int i ^ "\n" else "") ^
blanchet@36480
   902
        do_indent 0 ^ "proof -\n" ^
blanchet@36480
   903
        do_steps "" "" 1 proof ^
blanchet@36480
   904
        do_indent 0 ^ (if n <> 1 then "next" else "qed") ^ "\n"
blanchet@36488
   905
  in do_proof end
blanchet@36402
   906
blanchet@36488
   907
fun isar_proof_text (pool, debug, shrink_factor, ctxt, conjecture_shape)
blanchet@36402
   908
                    (minimize_command, atp_proof, thm_names, goal, i) =
blanchet@36402
   909
  let
blanchet@36402
   910
    val thy = ProofContext.theory_of ctxt
blanchet@36402
   911
    val (frees, hyp_ts, concl_t) = strip_subgoal goal i
blanchet@36402
   912
    val n = Logic.count_prems (prop_of goal)
blanchet@36223
   913
    val (one_line_proof, lemma_names) =
blanchet@36402
   914
      metis_proof_text (minimize_command, atp_proof, thm_names, goal, i)
blanchet@36283
   915
    fun isar_proof_for () =
blanchet@36474
   916
      case proof_from_atp_proof pool ctxt shrink_factor atp_proof thm_names
blanchet@36474
   917
                                frees
blanchet@36491
   918
           |> redirect_proof thy conjecture_shape hyp_ts concl_t
blanchet@36402
   919
           |> kill_duplicate_assumptions_in_proof
blanchet@36491
   920
           |> map unskolemize_step
blanchet@36402
   921
           |> then_chain_proof
blanchet@36402
   922
           |> kill_useless_labels_in_proof
blanchet@36402
   923
           |> relabel_proof
blanchet@36488
   924
           |> string_for_proof ctxt i n of
blanchet@36283
   925
        "" => ""
blanchet@36402
   926
      | proof => "\nStructured proof:\n" ^ Markup.markup Markup.sendback proof
blanchet@35868
   927
    val isar_proof =
blanchet@36402
   928
      if debug then
blanchet@36283
   929
        isar_proof_for ()
blanchet@36283
   930
      else
blanchet@36283
   931
        try isar_proof_for ()
blanchet@36287
   932
        |> the_default "Warning: The Isar proof construction failed.\n"
blanchet@36283
   933
  in (one_line_proof ^ isar_proof, lemma_names) end
paulson@21978
   934
blanchet@36422
   935
fun proof_text isar_proof isar_params =
blanchet@36422
   936
  if isar_proof then isar_proof_text isar_params else metis_proof_text
blanchet@36223
   937
immler@31038
   938
end;