src/HOL/Tools/res_reconstruct.ML
author wenzelm
Wed Oct 21 00:36:12 2009 +0200 (2009-10-21)
changeset 33035 15eab423e573
parent 32994 ccc07fbbfefd
child 33039 5018f6a76b3f
permissions -rw-r--r--
standardized basic operations on type option;
wenzelm@29268
     1
(*  Author:     L C Paulson and Claire Quigley, Cambridge University Computer Laboratory *)
paulson@21978
     2
paulson@21978
     3
(***************************************************************************)
paulson@21979
     4
(*  Code to deal with the transfer of proofs from a prover process         *)
paulson@21978
     5
(***************************************************************************)
paulson@21978
     6
signature RES_RECONSTRUCT =
paulson@24425
     7
sig
paulson@25492
     8
  val chained_hint: string
wenzelm@28477
     9
paulson@24425
    10
  val fix_sorts: sort Vartab.table -> term -> term
paulson@24425
    11
  val invert_const: string -> string
paulson@24425
    12
  val invert_type_const: string -> string
paulson@24425
    13
  val num_typargs: Context.theory -> string -> int
paulson@24425
    14
  val make_tvar: string -> typ
paulson@24425
    15
  val strip_prefix: string -> string -> string option
paulson@25710
    16
  val setup: Context.theory -> Context.theory
wenzelm@28477
    17
  (* extracting lemma list*)
immler@30874
    18
  val find_failure: string -> string option
immler@31840
    19
  val lemma_list: bool -> string ->
boehmes@32451
    20
    string * string vector * (int * int) * Proof.context * Thm.thm * int -> string * string list
wenzelm@28477
    21
  (* structured proofs *)
immler@31840
    22
  val structured_proof: string ->
boehmes@32451
    23
    string * string vector * (int * int) * Proof.context * Thm.thm * int -> string * string list
paulson@24425
    24
end;
paulson@21978
    25
paulson@24425
    26
structure ResReconstruct : RES_RECONSTRUCT =
paulson@21978
    27
struct
paulson@21978
    28
paulson@21978
    29
val trace_path = Path.basic "atp_trace";
paulson@21978
    30
wenzelm@32955
    31
fun trace s =
wenzelm@32955
    32
  if ! ResAxioms.trace then File.append (File.tmp_path trace_path) s
wenzelm@32955
    33
  else ();
paulson@21978
    34
wenzelm@32091
    35
fun string_of_thm ctxt = PrintMode.setmp [] (Display.string_of_thm ctxt);
paulson@26333
    36
paulson@25710
    37
(*For generating structured proofs: keep every nth proof line*)
paulson@26333
    38
val (modulus, modulus_setup) = Attrib.config_int "sledgehammer_modulus" 1;
paulson@21978
    39
paulson@25710
    40
(*Indicates whether to include sort information in generated proofs*)
paulson@26333
    41
val (recon_sorts, recon_sorts_setup) = Attrib.config_bool "sledgehammer_sorts" true;
paulson@22491
    42
wenzelm@28477
    43
(*Indicated whether to generate full proofs or just lemma lists - now via setup of atps*)
wenzelm@28477
    44
(* val (full_proofs, full_proofs_setup) = Attrib.config_bool "sledgehammer_full" false; *)
paulson@26333
    45
wenzelm@28477
    46
val setup = modulus_setup #> recon_sorts_setup;
paulson@22044
    47
paulson@21978
    48
(**** PARSING OF TSTP FORMAT ****)
paulson@21978
    49
paulson@21978
    50
(*Syntax trees, either termlist or formulae*)
paulson@21978
    51
datatype stree = Int of int | Br of string * stree list;
paulson@21978
    52
paulson@21978
    53
fun atom x = Br(x,[]);
paulson@21978
    54
paulson@21978
    55
fun scons (x,y) = Br("cons", [x,y]);
wenzelm@30190
    56
val listof = List.foldl scons (atom "nil");
paulson@21978
    57
paulson@21978
    58
(*Strings enclosed in single quotes, e.g. filenames*)
paulson@21978
    59
val quoted = $$"'" |-- Scan.repeat (~$$"'") --| $$"'" >> implode;
paulson@21978
    60
paulson@21978
    61
(*Intended for $true and $false*)
paulson@21978
    62
fun tf s = "c_" ^ str (Char.toUpper (String.sub(s,0))) ^ String.extract(s,1,NONE);
paulson@21978
    63
val truefalse = $$"$" |-- Symbol.scan_id >> (atom o tf);
paulson@21978
    64
paulson@21978
    65
(*Integer constants, typically proof line numbers*)
paulson@21978
    66
fun is_digit s = Char.isDigit (String.sub(s,0));
wenzelm@33035
    67
val integer = Scan.many1 is_digit >> (the o Int.fromString o implode);
paulson@21978
    68
paulson@21978
    69
(*Generalized FO terms, which include filenames, numbers, etc.*)
wenzelm@25999
    70
fun termlist x = (term ::: Scan.repeat ($$"," |-- term)) x
paulson@21978
    71
and term x = (quoted >> atom || integer>>Int || truefalse ||
paulson@21978
    72
              Symbol.scan_id -- Scan.optional ($$"(" |-- termlist --| $$")") [] >> Br ||
paulson@21978
    73
              $$"(" |-- term --| $$")" ||
paulson@24547
    74
              $$"[" |-- Scan.optional termlist [] --| $$"]" >> listof) x;
paulson@21978
    75
paulson@21978
    76
fun negate t = Br("c_Not", [t]);
paulson@21978
    77
fun equate (t1,t2) = Br("c_equal", [t1,t2]);
paulson@21978
    78
paulson@21978
    79
(*Apply equal or not-equal to a term*)
paulson@21978
    80
fun syn_equal (t, NONE) = t
paulson@21978
    81
  | syn_equal (t1, SOME (NONE, t2)) = equate (t1,t2)
paulson@21978
    82
  | syn_equal (t1, SOME (SOME _, t2)) = negate (equate (t1,t2));
paulson@21978
    83
paulson@21978
    84
(*Literals can involve negation, = and !=.*)
paulson@24547
    85
fun literal x = ($$"~" |-- literal >> negate ||
paulson@24547
    86
                 (term -- Scan.option (Scan.option ($$"!") --| $$"=" -- term) >> syn_equal)) x;
paulson@21978
    87
wenzelm@25999
    88
val literals = literal ::: Scan.repeat ($$"|" |-- literal);
paulson@21978
    89
paulson@21978
    90
(*Clause: a list of literals separated by the disjunction sign*)
paulson@24547
    91
val clause = $$"(" |-- literals --| $$")" || Scan.single literal;
paulson@21978
    92
paulson@21978
    93
val annotations = $$"," |-- term -- Scan.option ($$"," |-- termlist);
paulson@21978
    94
wenzelm@25718
    95
(*<cnf_annotated> ::= cnf(<name>,<formula_role>,<cnf_formula><annotations>).
paulson@21978
    96
  The <name> could be an identifier, but we assume integers.*)
wenzelm@23139
    97
val tstp_line = (Scan.this_string "cnf" -- $$"(") |--
wenzelm@23139
    98
                integer --| $$"," -- Symbol.scan_id --| $$"," --
paulson@21978
    99
                clause -- Scan.option annotations --| $$ ")";
paulson@21978
   100
paulson@21978
   101
paulson@21978
   102
(**** INTERPRETATION OF TSTP SYNTAX TREES ****)
paulson@21978
   103
paulson@21978
   104
exception STREE of stree;
paulson@21978
   105
paulson@21978
   106
(*If string s has the prefix s1, return the result of deleting it.*)
wenzelm@23139
   107
fun strip_prefix s1 s =
immler@31038
   108
  if String.isPrefix s1 s
paulson@24182
   109
  then SOME (ResClause.undo_ascii_of (String.extract (s, size s1, NONE)))
paulson@21978
   110
  else NONE;
paulson@21978
   111
paulson@21978
   112
(*Invert the table of translations between Isabelle and ATPs*)
paulson@21978
   113
val type_const_trans_table_inv =
paulson@21978
   114
      Symtab.make (map swap (Symtab.dest ResClause.type_const_trans_table));
paulson@21978
   115
paulson@21978
   116
fun invert_type_const c =
paulson@21978
   117
    case Symtab.lookup type_const_trans_table_inv c of
paulson@21978
   118
        SOME c' => c'
paulson@21978
   119
      | NONE => c;
paulson@21978
   120
paulson@21978
   121
fun make_tvar b = TVar(("'" ^ b, 0), HOLogic.typeS);
paulson@21978
   122
fun make_var (b,T) = Var((b,0),T);
paulson@21978
   123
paulson@21978
   124
(*Type variables are given the basic sort, HOL.type. Some will later be constrained
paulson@21978
   125
  by information from type literals, or by type inference.*)
paulson@21978
   126
fun type_of_stree t =
paulson@21978
   127
  case t of
paulson@21978
   128
      Int _ => raise STREE t
wenzelm@23139
   129
    | Br (a,ts) =>
paulson@21978
   130
        let val Ts = map type_of_stree ts
wenzelm@23139
   131
        in
paulson@21978
   132
          case strip_prefix ResClause.tconst_prefix a of
paulson@21978
   133
              SOME b => Type(invert_type_const b, Ts)
wenzelm@23139
   134
            | NONE =>
paulson@21978
   135
                if not (null ts) then raise STREE t  (*only tconsts have type arguments*)
wenzelm@23139
   136
                else
paulson@21978
   137
                case strip_prefix ResClause.tfree_prefix a of
paulson@21978
   138
                    SOME b => TFree("'" ^ b, HOLogic.typeS)
wenzelm@23139
   139
                  | NONE =>
paulson@21978
   140
                case strip_prefix ResClause.tvar_prefix a of
paulson@21978
   141
                    SOME b => make_tvar b
paulson@21978
   142
                  | NONE => make_tvar a   (*Variable from the ATP, say X1*)
paulson@21978
   143
        end;
paulson@21978
   144
paulson@21978
   145
(*Invert the table of translations between Isabelle and ATPs*)
paulson@21978
   146
val const_trans_table_inv =
wenzelm@23139
   147
      Symtab.update ("fequal", "op =")
paulson@22731
   148
        (Symtab.make (map swap (Symtab.dest ResClause.const_trans_table)));
paulson@21978
   149
paulson@21978
   150
fun invert_const c =
paulson@21978
   151
    case Symtab.lookup const_trans_table_inv c of
paulson@21978
   152
        SOME c' => c'
paulson@21978
   153
      | NONE => c;
paulson@21978
   154
paulson@21978
   155
(*The number of type arguments of a constant, zero if it's monomorphic*)
paulson@21978
   156
fun num_typargs thy s = length (Sign.const_typargs thy (s, Sign.the_const_type thy s));
paulson@21978
   157
paulson@21978
   158
(*Generates a constant, given its type arguments*)
paulson@21978
   159
fun const_of thy (a,Ts) = Const(a, Sign.const_instance thy (a,Ts));
paulson@21978
   160
paulson@21978
   161
(*First-order translation. No types are known for variables. HOLogic.typeT should allow
paulson@21978
   162
  them to be inferred.*)
paulson@22428
   163
fun term_of_stree args thy t =
paulson@21978
   164
  case t of
paulson@21978
   165
      Int _ => raise STREE t
paulson@22428
   166
    | Br ("hBOOL",[t]) => term_of_stree [] thy t  (*ignore hBOOL*)
paulson@22428
   167
    | Br ("hAPP",[t,u]) => term_of_stree (u::args) thy t
wenzelm@23139
   168
    | Br (a,ts) =>
paulson@21978
   169
        case strip_prefix ResClause.const_prefix a of
wenzelm@23139
   170
            SOME "equal" =>
paulson@22731
   171
              list_comb(Const ("op =", HOLogic.typeT), List.map (term_of_stree [] thy) ts)
wenzelm@23139
   172
          | SOME b =>
paulson@21978
   173
              let val c = invert_const b
paulson@21978
   174
                  val nterms = length ts - num_typargs thy c
paulson@22428
   175
                  val us = List.map (term_of_stree [] thy) (List.take(ts,nterms) @ args)
paulson@22428
   176
                  (*Extra args from hAPP come AFTER any arguments given directly to the
paulson@22428
   177
                    constant.*)
paulson@21978
   178
                  val Ts = List.map type_of_stree (List.drop(ts,nterms))
paulson@21978
   179
              in  list_comb(const_of thy (c, Ts), us)  end
paulson@21978
   180
          | NONE => (*a variable, not a constant*)
paulson@21978
   181
              let val T = HOLogic.typeT
paulson@21978
   182
                  val opr = (*a Free variable is typically a Skolem function*)
paulson@21978
   183
                    case strip_prefix ResClause.fixed_var_prefix a of
paulson@21978
   184
                        SOME b => Free(b,T)
wenzelm@23139
   185
                      | NONE =>
paulson@21978
   186
                    case strip_prefix ResClause.schematic_var_prefix a of
paulson@21978
   187
                        SOME b => make_var (b,T)
paulson@21978
   188
                      | NONE => make_var (a,T)    (*Variable from the ATP, say X1*)
paulson@23519
   189
              in  list_comb (opr, List.map (term_of_stree [] thy) (ts@args))  end;
paulson@21978
   190
wenzelm@23139
   191
(*Type class literal applied to a type. Returns triple of polarity, class, type.*)
paulson@21978
   192
fun constraint_of_stree pol (Br("c_Not",[t])) = constraint_of_stree (not pol) t
paulson@21978
   193
  | constraint_of_stree pol t = case t of
paulson@21978
   194
        Int _ => raise STREE t
wenzelm@23139
   195
      | Br (a,ts) =>
paulson@21978
   196
            (case (strip_prefix ResClause.class_prefix a, map type_of_stree ts) of
paulson@21978
   197
                 (SOME b, [T]) => (pol, b, T)
paulson@21978
   198
               | _ => raise STREE t);
paulson@21978
   199
paulson@21978
   200
(** Accumulate type constraints in a clause: negative type literals **)
paulson@21978
   201
paulson@21978
   202
fun addix (key,z)  = Vartab.map_default (key,[]) (cons z);
paulson@21978
   203
paulson@21978
   204
fun add_constraint ((false, cl, TFree(a,_)), vt) = addix ((a,~1),cl) vt
paulson@21978
   205
  | add_constraint ((false, cl, TVar(ix,_)), vt) = addix (ix,cl) vt
paulson@21978
   206
  | add_constraint (_, vt) = vt;
paulson@21978
   207
paulson@21978
   208
(*False literals (which E includes in its proofs) are deleted*)
paulson@21978
   209
val nofalses = filter (not o equal HOLogic.false_const);
paulson@21978
   210
paulson@22491
   211
(*Final treatment of the list of "real" literals from a clause.*)
paulson@22491
   212
fun finish [] = HOLogic.true_const  (*No "real" literals means only type information*)
wenzelm@23139
   213
  | finish lits =
paulson@22491
   214
      case nofalses lits of
paulson@22491
   215
          [] => HOLogic.false_const  (*The empty clause, since we started with real literals*)
paulson@22491
   216
        | xs => foldr1 HOLogic.mk_disj (rev xs);
paulson@22491
   217
paulson@21978
   218
(*Accumulate sort constraints in vt, with "real" literals in lits.*)
wenzelm@32994
   219
fun lits_of_strees _ (vt, lits) [] = (vt, finish lits)
wenzelm@23139
   220
  | lits_of_strees ctxt (vt, lits) (t::ts) =
paulson@22012
   221
      lits_of_strees ctxt (add_constraint (constraint_of_stree true t, vt), lits) ts
wenzelm@23139
   222
      handle STREE _ =>
paulson@22428
   223
      lits_of_strees ctxt (vt, term_of_stree [] (ProofContext.theory_of ctxt) t :: lits) ts;
paulson@21978
   224
paulson@21978
   225
(*Update TVars/TFrees with detected sort constraints.*)
paulson@21978
   226
fun fix_sorts vt =
paulson@21978
   227
  let fun tysubst (Type (a, Ts)) = Type (a, map tysubst Ts)
wenzelm@33035
   228
        | tysubst (TVar (xi, s)) = TVar (xi, the_default s (Vartab.lookup vt xi))
wenzelm@33035
   229
        | tysubst (TFree (x, s)) = TFree (x, the_default s (Vartab.lookup vt (x, ~1)))
paulson@21978
   230
      fun tmsubst (Const (a, T)) = Const (a, tysubst T)
paulson@21978
   231
        | tmsubst (Free (a, T)) = Free (a, tysubst T)
paulson@21978
   232
        | tmsubst (Var (xi, T)) = Var (xi, tysubst T)
paulson@21978
   233
        | tmsubst (t as Bound _) = t
paulson@21978
   234
        | tmsubst (Abs (a, T, t)) = Abs (a, tysubst T, tmsubst t)
paulson@21978
   235
        | tmsubst (t $ u) = tmsubst t $ tmsubst u;
paulson@21978
   236
  in fn t => if Vartab.is_empty vt then t else tmsubst t end;
paulson@21978
   237
paulson@21978
   238
(*Interpret a list of syntax trees as a clause, given by "real" literals and sort constraints.
paulson@21978
   239
  vt0 holds the initial sort constraints, from the conjecture clauses.*)
paulson@23519
   240
fun clause_of_strees ctxt vt0 ts =
wenzelm@22728
   241
  let val (vt, dt) = lits_of_strees ctxt (vt0,[]) ts in
wenzelm@24680
   242
    singleton (Syntax.check_terms ctxt) (TypeInfer.constrain HOLogic.boolT (fix_sorts vt dt))
wenzelm@22728
   243
  end;
paulson@21978
   244
wenzelm@29268
   245
fun gen_all_vars t = fold_rev Logic.all (OldTerm.term_vars t) t;
paulson@21978
   246
paulson@21978
   247
fun ints_of_stree_aux (Int n, ns) = n::ns
wenzelm@30190
   248
  | ints_of_stree_aux (Br(_,ts), ns) = List.foldl ints_of_stree_aux ns ts;
paulson@21978
   249
paulson@21978
   250
fun ints_of_stree t = ints_of_stree_aux (t, []);
paulson@21978
   251
paulson@25086
   252
fun decode_tstp vt0 (name, role, ts, annots) ctxt =
paulson@21978
   253
  let val deps = case annots of NONE => [] | SOME (source,_) => ints_of_stree source
paulson@25086
   254
      val cl = clause_of_strees ctxt vt0 ts
wenzelm@29268
   255
  in  ((name, role, cl, deps), fold Variable.declare_term (OldTerm.term_frees cl) ctxt)  end;
paulson@21978
   256
paulson@21978
   257
fun dest_tstp ((((name, role), ts), annots), chs) =
paulson@21978
   258
  case chs of
paulson@21978
   259
          "."::_ => (name, role, ts, annots)
paulson@21978
   260
        | _ => error ("TSTP line not terminated by \".\": " ^ implode chs);
paulson@21978
   261
paulson@21978
   262
paulson@21978
   263
(** Global sort constraints on TFrees (from tfree_tcs) are positive unit clauses. **)
paulson@21978
   264
paulson@21978
   265
fun add_tfree_constraint ((true, cl, TFree(a,_)), vt) = addix ((a,~1),cl) vt
paulson@21978
   266
  | add_tfree_constraint (_, vt) = vt;
paulson@21978
   267
paulson@21978
   268
fun tfree_constraints_of_clauses vt [] = vt
wenzelm@23139
   269
  | tfree_constraints_of_clauses vt ([lit]::tss) =
paulson@21978
   270
      (tfree_constraints_of_clauses (add_tfree_constraint (constraint_of_stree true lit, vt)) tss
paulson@21978
   271
       handle STREE _ => (*not a positive type constraint: ignore*)
paulson@21978
   272
       tfree_constraints_of_clauses vt tss)
paulson@21978
   273
  | tfree_constraints_of_clauses vt (_::tss) = tfree_constraints_of_clauses vt tss;
paulson@21978
   274
paulson@21978
   275
paulson@21978
   276
(**** Translation of TSTP files to Isar Proofs ****)
paulson@21978
   277
paulson@22012
   278
fun decode_tstp_list ctxt tuples =
paulson@21978
   279
  let val vt0 = tfree_constraints_of_clauses Vartab.empty (map #3 tuples)
paulson@25086
   280
  in  #1 (fold_map (decode_tstp vt0) tuples ctxt) end;
paulson@21978
   281
paulson@23519
   282
(** Finding a matching assumption. The literals may be permuted, and variable names
immler@31038
   283
    may disagree. We have to try all combinations of literals (quadratic!) and
paulson@23519
   284
    match up the variable names consistently. **)
paulson@23519
   285
immler@31038
   286
fun strip_alls_aux n (Const("all",_)$Abs(a,T,t))  =
paulson@23519
   287
      strip_alls_aux (n+1) (subst_bound (Var ((a,n), T), t))
paulson@23519
   288
  | strip_alls_aux _ t  =  t;
paulson@23519
   289
paulson@23519
   290
val strip_alls = strip_alls_aux 0;
paulson@23519
   291
paulson@23519
   292
exception MATCH_LITERAL;
paulson@22012
   293
paulson@23519
   294
(*Ignore types: they are not to be trusted...*)
paulson@23519
   295
fun match_literal (t1$u1) (t2$u2) env =
paulson@23519
   296
      match_literal t1 t2 (match_literal u1 u2 env)
immler@31038
   297
  | match_literal (Abs (_,_,t1)) (Abs (_,_,t2)) env =
paulson@23519
   298
      match_literal t1 t2 env
immler@31038
   299
  | match_literal (Bound i1) (Bound i2) env =
paulson@23519
   300
      if i1=i2 then env else raise MATCH_LITERAL
immler@31038
   301
  | match_literal (Const(a1,_)) (Const(a2,_)) env =
paulson@23519
   302
      if a1=a2 then env else raise MATCH_LITERAL
immler@31038
   303
  | match_literal (Free(a1,_)) (Free(a2,_)) env =
paulson@23519
   304
      if a1=a2 then env else raise MATCH_LITERAL
paulson@23519
   305
  | match_literal (Var(ix1,_)) (Var(ix2,_)) env = insert (op =) (ix1,ix2) env
wenzelm@32994
   306
  | match_literal _ _ _ = raise MATCH_LITERAL;
paulson@23519
   307
paulson@23519
   308
(*Checking that all variable associations are unique. The list env contains no
paulson@23519
   309
  repetitions, but does it contain say (x,y) and (y,y)? *)
immler@31038
   310
fun good env =
paulson@23519
   311
  let val (xs,ys) = ListPair.unzip env
paulson@23519
   312
  in  not (has_duplicates (op=) xs orelse has_duplicates (op=) ys)  end;
paulson@23519
   313
paulson@23519
   314
(*Match one list of literals against another, ignoring types and the order of
paulson@23519
   315
  literals. Sorting is unreliable because we don't have types or variable names.*)
paulson@23519
   316
fun matches_aux _ [] [] = true
paulson@23519
   317
  | matches_aux env (lit::lits) ts =
paulson@23519
   318
      let fun match1 us [] = false
paulson@23519
   319
            | match1 us (t::ts) =
paulson@23519
   320
                let val env' = match_literal lit t env
immler@31038
   321
                in  (good env' andalso matches_aux env' lits (us@ts)) orelse
immler@31038
   322
                    match1 (t::us) ts
paulson@23519
   323
                end
paulson@23519
   324
                handle MATCH_LITERAL => match1 (t::us) ts
immler@31038
   325
      in  match1 [] ts  end;
paulson@23519
   326
paulson@23519
   327
(*Is this length test useful?*)
immler@31038
   328
fun matches (lits1,lits2) =
immler@31038
   329
  length lits1 = length lits2  andalso
paulson@23519
   330
  matches_aux [] (map Envir.eta_contract lits1) (map Envir.eta_contract lits2);
paulson@21999
   331
paulson@21999
   332
fun permuted_clause t =
paulson@24958
   333
  let val lits = HOLogic.disjuncts t
paulson@21999
   334
      fun perm [] = NONE
wenzelm@23139
   335
        | perm (ctm::ctms) =
paulson@24958
   336
            if matches (lits, HOLogic.disjuncts (HOLogic.dest_Trueprop (strip_alls ctm)))
paulson@23519
   337
            then SOME ctm else perm ctms
paulson@21999
   338
  in perm end;
paulson@21999
   339
wenzelm@32994
   340
fun have_or_show "show " _ = "show \""
paulson@22470
   341
  | have_or_show have lname = have ^ lname ^ ": \""
paulson@22470
   342
paulson@21999
   343
(*ctms is a list of conjecture clauses as yielded by Isabelle. Those returned by the
paulson@21999
   344
  ATP may have their literals reordered.*)
paulson@21999
   345
fun isar_lines ctxt ctms =
wenzelm@28572
   346
  let val string_of = PrintMode.setmp [] (fn term => Syntax.string_of_term ctxt term)
paulson@26333
   347
      val _ = trace ("\n\nisar_lines: start\n")
wenzelm@32994
   348
      fun doline _ (lname, t, []) =  (*No deps: it's a conjecture clause, with no proof.*)
paulson@21999
   349
           (case permuted_clause t ctms of
paulson@21999
   350
                SOME u => "assume " ^ lname ^ ": \"" ^ string_of u ^ "\"\n"
paulson@21999
   351
              | NONE => "assume? " ^ lname ^ ": \"" ^ string_of t ^ "\"\n")  (*no match!!*)
paulson@22470
   352
        | doline have (lname, t, deps) =
paulson@23519
   353
            have_or_show have lname ^ string_of (gen_all_vars (HOLogic.mk_Trueprop t)) ^
paulson@22372
   354
            "\"\n  by (metis " ^ space_implode " " deps ^ ")\n"
paulson@21978
   355
      fun dolines [(lname, t, deps)] = [doline "show " (lname, t, deps)]
paulson@21978
   356
        | dolines ((lname, t, deps)::lines) = doline "have " (lname, t, deps) :: dolines lines
wenzelm@32966
   357
  in setmp_CRITICAL show_sorts (Config.get ctxt recon_sorts) dolines end;
paulson@21978
   358
paulson@21978
   359
fun notequal t (_,t',_) = not (t aconv t');
paulson@21978
   360
paulson@22491
   361
(*No "real" literals means only type information*)
paulson@23519
   362
fun eq_types t = t aconv HOLogic.true_const;
paulson@21978
   363
paulson@22731
   364
fun replace_dep (old:int, new) dep = if dep=old then new else [dep];
paulson@21978
   365
wenzelm@23139
   366
fun replace_deps (old:int, new) (lno, t, deps) =
wenzelm@30190
   367
      (lno, t, List.foldl (op union_int) [] (map (replace_dep (old, new)) deps));
paulson@21978
   368
paulson@22491
   369
(*Discard axioms; consolidate adjacent lines that prove the same clause, since they differ
paulson@22491
   370
  only in type information.*)
paulson@21978
   371
fun add_prfline ((lno, "axiom", t, []), lines) =  (*axioms are not proof lines*)
paulson@22491
   372
      if eq_types t (*must be clsrel/clsarity: type information, so delete refs to it*)
wenzelm@23139
   373
      then map (replace_deps (lno, [])) lines
paulson@22470
   374
      else
paulson@22470
   375
       (case take_prefix (notequal t) lines of
paulson@22470
   376
           (_,[]) => lines                  (*no repetition of proof line*)
wenzelm@32994
   377
         | (pre, (lno', _, _) :: post) =>   (*repetition: replace later line by earlier one*)
paulson@22470
   378
             pre @ map (replace_deps (lno', [lno])) post)
wenzelm@32994
   379
  | add_prfline ((lno, _, t, []), lines) =  (*no deps: conjecture clause*)
paulson@22470
   380
      (lno, t, []) :: lines
wenzelm@32994
   381
  | add_prfline ((lno, _, t, deps), lines) =
paulson@22491
   382
      if eq_types t then (lno, t, deps) :: lines
paulson@22491
   383
      (*Type information will be deleted later; skip repetition test.*)
paulson@22491
   384
      else (*FIXME: Doesn't this code risk conflating proofs involving different types??*)
paulson@22044
   385
      case take_prefix (notequal t) lines of
paulson@22044
   386
         (_,[]) => (lno, t, deps) :: lines  (*no repetition of proof line*)
wenzelm@32994
   387
       | (pre, (lno', t', _) :: post) =>
paulson@22044
   388
           (lno, t', deps) ::               (*repetition: replace later line by earlier one*)
paulson@22044
   389
           (pre @ map (replace_deps (lno', [lno])) post);
paulson@22044
   390
paulson@22470
   391
(*Recursively delete empty lines (type information) from the proof.*)
paulson@22470
   392
fun add_nonnull_prfline ((lno, t, []), lines) = (*no dependencies, so a conjecture clause*)
paulson@22491
   393
     if eq_types t (*must be type information, tfree_tcs, clsrel, clsarity: delete refs to it*)
wenzelm@23139
   394
     then delete_dep lno lines
wenzelm@23139
   395
     else (lno, t, []) :: lines
paulson@22470
   396
  | add_nonnull_prfline ((lno, t, deps), lines) = (lno, t, deps) :: lines
wenzelm@30190
   397
and delete_dep lno lines = List.foldr add_nonnull_prfline [] (map (replace_deps (lno, [])) lines);
paulson@22470
   398
paulson@24937
   399
fun bad_free (Free (a,_)) = String.isPrefix "sko_" a
paulson@22731
   400
  | bad_free _ = false;
paulson@22731
   401
wenzelm@23139
   402
(*TVars are forbidden in goals. Also, we don't want lines with <2 dependencies.
paulson@22491
   403
  To further compress proofs, setting modulus:=n deletes every nth line, and nlines
paulson@22491
   404
  counts the number of proof lines processed so far.
paulson@22491
   405
  Deleted lines are replaced by their own dependencies. Note that the "add_nonnull_prfline"
paulson@22044
   406
  phase may delete some dependencies, hence this phase comes later.*)
paulson@25710
   407
fun add_wanted_prfline ctxt ((lno, t, []), (nlines, lines)) =
paulson@22491
   408
      (nlines, (lno, t, []) :: lines)   (*conjecture clauses must be kept*)
paulson@25710
   409
  | add_wanted_prfline ctxt ((lno, t, deps), (nlines, lines)) =
wenzelm@29272
   410
      if eq_types t orelse not (null (Term.add_tvars t [])) orelse
wenzelm@29268
   411
         exists_subterm bad_free t orelse
paulson@24937
   412
         (not (null lines) andalso   (*final line can't be deleted for these reasons*)
immler@31038
   413
          (length deps < 2 orelse nlines mod (Config.get ctxt modulus) <> 0))
paulson@22491
   414
      then (nlines+1, map (replace_deps (lno, deps)) lines) (*Delete line*)
paulson@22491
   415
      else (nlines+1, (lno, t, deps) :: lines);
paulson@21978
   416
paulson@21999
   417
(*Replace numeric proof lines by strings, either from thm_names or sequential line numbers*)
paulson@21978
   418
fun stringify_deps thm_names deps_map [] = []
paulson@21978
   419
  | stringify_deps thm_names deps_map ((lno, t, deps) :: lines) =
paulson@21978
   420
      if lno <= Vector.length thm_names  (*axiom*)
wenzelm@23139
   421
      then (Vector.sub(thm_names,lno-1), t, []) :: stringify_deps thm_names deps_map lines
paulson@21979
   422
      else let val lname = Int.toString (length deps_map)
wenzelm@23139
   423
               fun fix lno = if lno <= Vector.length thm_names
paulson@21978
   424
                             then SOME(Vector.sub(thm_names,lno-1))
paulson@21978
   425
                             else AList.lookup op= deps_map lno;
wenzelm@32952
   426
           in  (lname, t, map_filter fix (distinct (op=) deps)) ::
paulson@21978
   427
               stringify_deps thm_names ((lno,lname)::deps_map) lines
paulson@21978
   428
           end;
paulson@21978
   429
paulson@24547
   430
val proofstart = "proof (neg_clausify)\n";
paulson@21979
   431
paulson@21979
   432
fun isar_header [] = proofstart
paulson@21999
   433
  | isar_header ts = proofstart ^ "fix " ^ space_implode " " ts ^ "\n";
paulson@21979
   434
paulson@22012
   435
fun decode_tstp_file cnfs ctxt th sgno thm_names =
paulson@26333
   436
  let val _ = trace "\ndecode_tstp_file: start\n"
paulson@26333
   437
      val tuples = map (dest_tstp o tstp_line o explode) cnfs
paulson@26333
   438
      val _ = trace (Int.toString (length tuples) ^ " tuples extracted\n")
paulson@24552
   439
      val ctxt = ProofContext.set_mode ProofContext.mode_schematic ctxt
wenzelm@30190
   440
      val raw_lines = List.foldr add_prfline [] (decode_tstp_list ctxt tuples)
paulson@26333
   441
      val _ = trace (Int.toString (length raw_lines) ^ " raw_lines extracted\n")
wenzelm@30190
   442
      val nonnull_lines = List.foldr add_nonnull_prfline [] raw_lines
paulson@26333
   443
      val _ = trace (Int.toString (length nonnull_lines) ^ " nonnull_lines extracted\n")
wenzelm@30190
   444
      val (_,lines) = List.foldr (add_wanted_prfline ctxt) (0,[]) nonnull_lines
paulson@26333
   445
      val _ = trace (Int.toString (length lines) ^ " lines extracted\n")
wenzelm@32258
   446
      val (ccls,fixes) = ResAxioms.neg_conjecture_clauses ctxt th sgno
paulson@26333
   447
      val _ = trace (Int.toString (length ccls) ^ " conjecture clauses\n")
paulson@21999
   448
      val ccls = map forall_intr_vars ccls
wenzelm@32091
   449
      val _ =
wenzelm@32955
   450
        if ! ResAxioms.trace then app (fn th => trace ("\nccl: " ^ string_of_thm ctxt th)) ccls
wenzelm@32091
   451
        else ()
paulson@26333
   452
      val ilines = isar_lines ctxt (map prop_of ccls) (stringify_deps thm_names [] lines)
paulson@26333
   453
      val _ = trace "\ndecode_tstp_file: finishing\n"
wenzelm@23139
   454
  in
wenzelm@32952
   455
    isar_header (map #1 fixes) ^ implode ilines ^ "qed\n"
wenzelm@32258
   456
  end;
paulson@21978
   457
paulson@21978
   458
immler@31866
   459
  (*=== EXTRACTING PROOF-TEXT === *)
immler@31866
   460
immler@31866
   461
  val begin_proof_strings = ["# SZS output start CNFRefutation.",
immler@31866
   462
      "=========== Refutation ==========",
immler@31866
   463
  "Here is a proof"];
immler@31866
   464
  val end_proof_strings = ["# SZS output end CNFRefutation",
immler@31866
   465
      "======= End of refutation =======",
immler@31866
   466
  "Formulae used in the proof"];
immler@31866
   467
  fun get_proof_extract proof =
immler@31866
   468
    let
immler@31866
   469
    (*splits to_split by the first possible of a list of splitters*)
immler@31866
   470
    val (begin_string, end_string) =
immler@31866
   471
      (find_first (fn s => String.isSubstring s proof) begin_proof_strings,
immler@31866
   472
      find_first (fn s => String.isSubstring s proof) end_proof_strings)
immler@31866
   473
    in
immler@31866
   474
      if is_none begin_string orelse is_none end_string
immler@31866
   475
      then error "Could not extract proof (no substring indicating a proof)"
immler@31866
   476
      else proof |> first_field (the begin_string) |> the |> snd
immler@31866
   477
                 |> first_field (the end_string) |> the |> fst end;
immler@31866
   478
immler@31866
   479
(* ==== CHECK IF PROOF OF E OR VAMPIRE WAS SUCCESSFUL === *)
immler@31866
   480
immler@30874
   481
  val failure_strings_E = ["SZS status: Satisfiable","SZS status Satisfiable",
immler@30874
   482
    "SZS status: ResourceOut","SZS status ResourceOut","# Cannot determine problem status"];
wenzelm@28477
   483
  val failure_strings_vampire = ["Satisfiability detected", "Refutation not found", "CANNOT PROVE"];
immler@30874
   484
  val failure_strings_SPASS = ["SPASS beiseite: Completion found.",
immler@30874
   485
    "SPASS beiseite: Ran out of time.", "SPASS beiseite: Maximal number of loops exceeded."];
immler@30874
   486
  val failure_strings_remote = ["Remote-script could not extract proof"];
immler@30874
   487
  fun find_failure proof =
immler@29597
   488
    let val failures =
immler@31038
   489
      map_filter (fn s => if String.isSubstring s proof then SOME s else NONE)
immler@31866
   490
        (failure_strings_E @ failure_strings_vampire @ failure_strings_SPASS @ failure_strings_remote)
immler@31866
   491
    val correct = null failures andalso
immler@31866
   492
      exists (fn s => String.isSubstring s proof) begin_proof_strings andalso
immler@31866
   493
      exists (fn s => String.isSubstring s proof) end_proof_strings
immler@31866
   494
    in
immler@31866
   495
      if correct then NONE
immler@31866
   496
      else if null failures then SOME "Output of ATP not in proper format"
immler@31866
   497
      else SOME (hd failures) end;
immler@31866
   498
immler@31038
   499
  (* === EXTRACTING LEMMAS === *)
wenzelm@28477
   500
  (* lines have the form "cnf(108, axiom, ...",
wenzelm@28477
   501
  the number (108) has to be extracted)*)
immler@31840
   502
  fun get_step_nums false proofextract =
wenzelm@28477
   503
    let val toks = String.tokens (not o Char.isAlphaNum)
wenzelm@28477
   504
    fun inputno ("cnf"::ntok::"axiom"::_) = Int.fromString ntok
immler@31840
   505
      | inputno ("cnf"::ntok::"negated"::"conjecture"::_) = Int.fromString ntok
wenzelm@28477
   506
      | inputno _ = NONE
wenzelm@28477
   507
    val lines = split_lines proofextract
wenzelm@32952
   508
    in  map_filter (inputno o toks) lines  end
immler@31840
   509
  (*String contains multiple lines. We want those of the form
wenzelm@28477
   510
    "253[0:Inp] et cetera..."
wenzelm@28477
   511
    A list consisting of the first number in each line is returned. *)
immler@31840
   512
  |  get_step_nums true proofextract =
wenzelm@28477
   513
    let val toks = String.tokens (not o Char.isAlphaNum)
wenzelm@28477
   514
    fun inputno (ntok::"0"::"Inp"::_) = Int.fromString ntok
wenzelm@28477
   515
      | inputno _ = NONE
wenzelm@28477
   516
    val lines = split_lines proofextract
wenzelm@32952
   517
    in  map_filter (inputno o toks) lines  end
wenzelm@28477
   518
    
immler@31038
   519
  (*extracting lemmas from tstp-output between the lines from above*)
immler@31840
   520
  fun extract_lemmas get_step_nums (proof, thm_names, conj_count, _, _, _) =
wenzelm@28477
   521
    let
wenzelm@28477
   522
    (* get the names of axioms from their numbers*)
wenzelm@28477
   523
    fun get_axiom_names thm_names step_nums =
wenzelm@28477
   524
      let
immler@31840
   525
      val last_axiom = Vector.length thm_names
immler@31840
   526
      fun is_axiom n = n <= last_axiom
wenzelm@32565
   527
      fun is_conj n = n >= fst conj_count andalso n < fst conj_count + snd conj_count
wenzelm@28477
   528
      fun getname i = Vector.sub(thm_names, i-1)
immler@31038
   529
      in
immler@31840
   530
        (sort_distinct string_ord (filter (fn x => x <> "??.unknown")
immler@31840
   531
          (map getname (filter is_axiom step_nums))),
immler@31840
   532
        exists is_conj step_nums)
wenzelm@28477
   533
      end
wenzelm@28477
   534
    val proofextract = get_proof_extract proof
immler@31038
   535
    in
wenzelm@28477
   536
      get_axiom_names thm_names (get_step_nums proofextract)
wenzelm@28477
   537
    end;
immler@31410
   538
immler@31410
   539
  (*Used to label theorems chained into the sledgehammer call*)
immler@31410
   540
  val chained_hint = "CHAINED";
immler@31410
   541
  val nochained = filter_out (fn y => y = chained_hint)
wenzelm@28477
   542
    
immler@31038
   543
  (* metis-command *)
immler@31038
   544
  fun metis_line [] = "apply metis"
immler@31038
   545
    | metis_line xs = "apply (metis " ^ space_implode " " xs ^ ")"
immler@31038
   546
immler@31038
   547
  (* atp_minimize [atp=<prover>] <lemmas> *)
immler@31038
   548
  fun minimize_line _ [] = ""
immler@31038
   549
    | minimize_line name lemmas = "For minimizing the number of lemmas try this command:\n" ^
immler@31410
   550
          (Markup.markup Markup.sendback) ("atp_minimize [atp=" ^ name ^ "] " ^
immler@31410
   551
                                           space_implode " " (nochained lemmas))
immler@31037
   552
immler@31038
   553
  fun sendback_metis_nochained lemmas =
immler@31410
   554
    (Markup.markup Markup.sendback o metis_line) (nochained lemmas)
immler@31840
   555
immler@31840
   556
  fun lemma_list dfg name result =
immler@31840
   557
    let val (lemmas, used_conj) = extract_lemmas (get_step_nums dfg) result
boehmes@32451
   558
    in (sendback_metis_nochained lemmas ^ "\n" ^ minimize_line name lemmas ^
immler@31840
   559
      (if used_conj then ""
boehmes@32451
   560
       else "\nWarning: Goal is provable because context is inconsistent."),
boehmes@32469
   561
       nochained lemmas)
immler@31840
   562
    end;
immler@31037
   563
immler@31038
   564
  (* === Extracting structured Isar-proof === *)
immler@31840
   565
  fun structured_proof name (result as (proof, thm_names, conj_count, ctxt, goal, subgoalno)) =
immler@31038
   566
    let
immler@31038
   567
    (*Could use split_lines, but it can return blank lines...*)
immler@31038
   568
    val lines = String.tokens (equal #"\n");
immler@31038
   569
    val nospaces = String.translate (fn c => if Char.isSpace c then "" else str c)
immler@31038
   570
    val proofextract = get_proof_extract proof
immler@31038
   571
    val cnfs = filter (String.isPrefix "cnf(") (map nospaces (lines proofextract))
boehmes@32451
   572
    val (one_line_proof, lemma_names) = lemma_list false name result
immler@31038
   573
    val structured = if chained_hint mem_string (String.tokens (fn c => c = #" ") one_line_proof) then ""
immler@31038
   574
                else decode_tstp_file cnfs ctxt goal subgoalno thm_names
immler@31038
   575
    in
boehmes@32451
   576
    (one_line_proof ^ "\n\n" ^ (Markup.markup Markup.sendback) structured, lemma_names)
immler@31038
   577
  end
paulson@21978
   578
immler@31038
   579
end;