src/HOL/Tools/res_hol_clause.ML
author immler@in.tum.de
Wed Feb 25 10:02:10 2009 +0100 (2009-02-25 ago)
changeset 30150 4d5a98cebb24
parent 30149 6b7ad52c5770
child 30151 629f3a92863e
permissions -rw-r--r--
removed local ref const_needs_hBOOL;
renamed some const_min_arity parameters to cma
wenzelm@24311
     1
(* ID: $Id$
mengj@17998
     2
   Author: Jia Meng, NICTA
mengj@17998
     3
wenzelm@24311
     4
FOL clauses translated from HOL formulae.
mengj@17998
     5
*)
mengj@17998
     6
wenzelm@24311
     7
signature RES_HOL_CLAUSE =
wenzelm@24311
     8
sig
wenzelm@24311
     9
  val ext: thm
wenzelm@24311
    10
  val comb_I: thm
wenzelm@24311
    11
  val comb_K: thm
wenzelm@24311
    12
  val comb_B: thm
wenzelm@24311
    13
  val comb_C: thm
wenzelm@24311
    14
  val comb_S: thm
paulson@24385
    15
  datatype type_level = T_FULL | T_CONST
wenzelm@24311
    16
  val typ_level: type_level ref
wenzelm@24311
    17
  val minimize_applies: bool ref
wenzelm@24311
    18
  type axiom_name = string
wenzelm@24311
    19
  type polarity = bool
wenzelm@24311
    20
  type clause_id = int
wenzelm@24311
    21
  datatype combterm =
wenzelm@24311
    22
      CombConst of string * ResClause.fol_type * ResClause.fol_type list (*Const and Free*)
wenzelm@24311
    23
    | CombVar of string * ResClause.fol_type
wenzelm@24311
    24
    | CombApp of combterm * combterm
wenzelm@24311
    25
  datatype literal = Literal of polarity * combterm
wenzelm@24311
    26
  val strip_comb: combterm -> combterm * combterm list
paulson@24940
    27
  val literals_of_term: theory -> term -> literal list * typ list
wenzelm@28835
    28
  exception TOO_TRIVIAL
wenzelm@24323
    29
  val tptp_write_file: theory -> bool -> thm list -> string ->
wenzelm@24311
    30
    (thm * (axiom_name * clause_id)) list * ResClause.classrelClause list *
wenzelm@24311
    31
      ResClause.arityClause list -> string list -> axiom_name list
wenzelm@24323
    32
  val dfg_write_file: theory -> bool -> thm list -> string ->
wenzelm@24311
    33
    (thm * (axiom_name * clause_id)) list * ResClause.classrelClause list *
wenzelm@24311
    34
      ResClause.arityClause list -> string list -> axiom_name list
wenzelm@24311
    35
end
mengj@17998
    36
wenzelm@24311
    37
structure ResHolClause: RES_HOL_CLAUSE =
mengj@17998
    38
struct
mengj@17998
    39
paulson@22078
    40
structure RC = ResClause;
paulson@22078
    41
mengj@20791
    42
(* theorems for combinators and function extensionality *)
mengj@20791
    43
val ext = thm "HOL.ext";
wenzelm@21254
    44
val comb_I = thm "ATP_Linkup.COMBI_def";
wenzelm@21254
    45
val comb_K = thm "ATP_Linkup.COMBK_def";
wenzelm@21254
    46
val comb_B = thm "ATP_Linkup.COMBB_def";
wenzelm@21254
    47
val comb_C = thm "ATP_Linkup.COMBC_def";
wenzelm@21254
    48
val comb_S = thm "ATP_Linkup.COMBS_def";
wenzelm@21254
    49
val fequal_imp_equal = thm "ATP_Linkup.fequal_imp_equal";
wenzelm@21254
    50
val equal_imp_fequal = thm "ATP_Linkup.equal_imp_fequal";
mengj@20791
    51
paulson@22825
    52
paulson@22825
    53
(*The different translations of types*)
paulson@24385
    54
datatype type_level = T_FULL | T_CONST;
paulson@22825
    55
paulson@22825
    56
val typ_level = ref T_CONST;
paulson@22825
    57
paulson@22064
    58
(*If true, each function will be directly applied to as many arguments as possible, avoiding
paulson@24385
    59
  use of the "apply" operator. Use of hBOOL is also minimized.*)
paulson@22078
    60
val minimize_applies = ref true;
paulson@22064
    61
immler@30149
    62
fun min_arity_of const_min_arity c = getOpt (Symtab.lookup const_min_arity c, 0);
paulson@22064
    63
paulson@22825
    64
(*True if the constant ever appears outside of the top-level position in literals.
paulson@22825
    65
  If false, the constant always receives all of its arguments and is used as a predicate.*)
immler@30150
    66
fun needs_hBOOL const_needs_hBOOL c = not (!minimize_applies) orelse
immler@30150
    67
                    getOpt (Symtab.lookup const_needs_hBOOL c, false);
paulson@22064
    68
mengj@19198
    69
mengj@17998
    70
(******************************************************)
mengj@17998
    71
(* data types for typed combinator expressions        *)
mengj@17998
    72
(******************************************************)
mengj@17998
    73
mengj@17998
    74
type axiom_name = string;
mengj@17998
    75
type polarity = bool;
mengj@17998
    76
type clause_id = int;
mengj@17998
    77
paulson@23386
    78
datatype combterm = CombConst of string * RC.fol_type * RC.fol_type list (*Const and Free*)
wenzelm@24311
    79
                  | CombVar of string * RC.fol_type
wenzelm@24311
    80
                  | CombApp of combterm * combterm
wenzelm@24311
    81
mengj@17998
    82
datatype literal = Literal of polarity * combterm;
mengj@17998
    83
wenzelm@24311
    84
datatype clause =
wenzelm@24311
    85
         Clause of {clause_id: clause_id,
wenzelm@24311
    86
                    axiom_name: axiom_name,
wenzelm@24311
    87
                    th: thm,
wenzelm@24311
    88
                    kind: RC.kind,
wenzelm@24311
    89
                    literals: literal list,
paulson@24940
    90
                    ctypes_sorts: typ list};
mengj@17998
    91
mengj@17998
    92
mengj@17998
    93
(*********************************************************************)
mengj@17998
    94
(* convert a clause with type Term.term to a clause with type clause *)
mengj@17998
    95
(*********************************************************************)
mengj@17998
    96
paulson@21561
    97
fun isFalse (Literal(pol, CombConst(c,_,_))) =
paulson@20360
    98
      (pol andalso c = "c_False") orelse
paulson@20360
    99
      (not pol andalso c = "c_True")
mengj@17998
   100
  | isFalse _ = false;
mengj@17998
   101
paulson@21561
   102
fun isTrue (Literal (pol, CombConst(c,_,_))) =
mengj@17998
   103
      (pol andalso c = "c_True") orelse
mengj@17998
   104
      (not pol andalso c = "c_False")
mengj@17998
   105
  | isTrue _ = false;
wenzelm@24311
   106
wenzelm@24311
   107
fun isTaut (Clause {literals,...}) = exists isTrue literals;
mengj@17998
   108
mengj@18440
   109
fun type_of (Type (a, Ts)) =
paulson@21561
   110
      let val (folTypes,ts) = types_of Ts
paulson@22078
   111
      in  (RC.Comp(RC.make_fixed_type_const a, folTypes), ts)  end
mengj@18440
   112
  | type_of (tp as (TFree(a,s))) =
paulson@24940
   113
      (RC.AtomF (RC.make_fixed_type_var a), [tp])
mengj@18440
   114
  | type_of (tp as (TVar(v,s))) =
paulson@24940
   115
      (RC.AtomV (RC.make_schematic_type_var v), [tp])
mengj@18440
   116
and types_of Ts =
paulson@22078
   117
      let val (folTyps,ts) = ListPair.unzip (map type_of Ts)
paulson@22078
   118
      in  (folTyps, RC.union_all ts)  end;
mengj@17998
   119
mengj@17998
   120
(* same as above, but no gathering of sort information *)
wenzelm@24311
   121
fun simp_type_of (Type (a, Ts)) =
paulson@22078
   122
      RC.Comp(RC.make_fixed_type_const a, map simp_type_of Ts)
paulson@22078
   123
  | simp_type_of (TFree (a,s)) = RC.AtomF(RC.make_fixed_type_var a)
paulson@22078
   124
  | simp_type_of (TVar (v,s)) = RC.AtomV(RC.make_schematic_type_var v);
mengj@18440
   125
mengj@18356
   126
wenzelm@24323
   127
fun const_type_of thy (c,t) =
paulson@22078
   128
      let val (tp,ts) = type_of t
wenzelm@24323
   129
      in  (tp, ts, map simp_type_of (Sign.const_typargs thy (c,t))) end;
mengj@18356
   130
mengj@17998
   131
(* convert a Term.term (with combinators) into a combterm, also accummulate sort info *)
wenzelm@24323
   132
fun combterm_of thy (Const(c,t)) =
wenzelm@24323
   133
      let val (tp,ts,tvar_list) = const_type_of thy (c,t)
wenzelm@24311
   134
          val c' = CombConst(RC.make_fixed_const c, tp, tvar_list)
paulson@22078
   135
      in  (c',ts)  end
wenzelm@24323
   136
  | combterm_of thy (Free(v,t)) =
paulson@21513
   137
      let val (tp,ts) = type_of t
paulson@25243
   138
          val v' = CombConst(RC.make_fixed_var v, tp, [])
paulson@22078
   139
      in  (v',ts)  end
wenzelm@24323
   140
  | combterm_of thy (Var(v,t)) =
paulson@21513
   141
      let val (tp,ts) = type_of t
wenzelm@24311
   142
          val v' = CombVar(RC.make_schematic_var v,tp)
paulson@22078
   143
      in  (v',ts)  end
paulson@24385
   144
  | combterm_of thy (P $ Q) =
wenzelm@24323
   145
      let val (P',tsP) = combterm_of thy P
wenzelm@24323
   146
          val (Q',tsQ) = combterm_of thy Q
paulson@24385
   147
      in  (CombApp(P',Q'), tsP union tsQ)  end
paulson@24385
   148
  | combterm_of thy (t as Abs _) = raise RC.CLAUSE("HOL CLAUSE",t);
mengj@17998
   149
wenzelm@24323
   150
fun predicate_of thy ((Const("Not",_) $ P), polarity) = predicate_of thy (P, not polarity)
paulson@25008
   151
  | predicate_of thy (t,polarity) = (combterm_of thy (Envir.eta_contract t), polarity);
mengj@17998
   152
wenzelm@24323
   153
fun literals_of_term1 thy args (Const("Trueprop",_) $ P) = literals_of_term1 thy args P
wenzelm@24323
   154
  | literals_of_term1 thy args (Const("op |",_) $ P $ Q) =
wenzelm@24323
   155
      literals_of_term1 thy (literals_of_term1 thy args P) Q
wenzelm@24323
   156
  | literals_of_term1 thy (lits,ts) P =
wenzelm@24323
   157
      let val ((pred,ts'),pol) = predicate_of thy (P,true)
paulson@21509
   158
      in
wenzelm@24311
   159
          (Literal(pol,pred)::lits, ts union ts')
paulson@21509
   160
      end;
mengj@17998
   161
wenzelm@24323
   162
fun literals_of_term thy P = literals_of_term1 thy ([],[]) P;
mengj@17998
   163
wenzelm@28835
   164
(* Problem too trivial for resolution (empty clause) *)
wenzelm@28835
   165
exception TOO_TRIVIAL;
wenzelm@28835
   166
mengj@17998
   167
(* making axiom and conjecture clauses *)
wenzelm@24323
   168
fun make_clause thy (clause_id,axiom_name,kind,th) =
paulson@24385
   169
    let val (lits,ctypes_sorts) = literals_of_term thy (prop_of th)
mengj@17998
   170
    in
wenzelm@24311
   171
        if forall isFalse lits
wenzelm@28835
   172
        then raise TOO_TRIVIAL
wenzelm@24311
   173
        else
wenzelm@24311
   174
            Clause {clause_id = clause_id, axiom_name = axiom_name, th = th, kind = kind,
paulson@24937
   175
                    literals = lits, ctypes_sorts = ctypes_sorts}
mengj@17998
   176
    end;
mengj@17998
   177
mengj@20016
   178
wenzelm@24323
   179
fun add_axiom_clause thy ((th,(name,id)), pairs) =
wenzelm@24323
   180
  let val cls = make_clause thy (id, name, RC.Axiom, th)
paulson@21573
   181
  in
paulson@21573
   182
      if isTaut cls then pairs else (name,cls)::pairs
paulson@21573
   183
  end;
mengj@19354
   184
wenzelm@24323
   185
fun make_axiom_clauses thy = foldl (add_axiom_clause thy) [];
mengj@19354
   186
wenzelm@24323
   187
fun make_conjecture_clauses_aux _ _ [] = []
wenzelm@24323
   188
  | make_conjecture_clauses_aux thy n (th::ths) =
wenzelm@24323
   189
      make_clause thy (n,"conjecture", RC.Conjecture, th) ::
wenzelm@24323
   190
      make_conjecture_clauses_aux thy (n+1) ths;
mengj@17998
   191
wenzelm@24323
   192
fun make_conjecture_clauses thy = make_conjecture_clauses_aux thy 0;
mengj@17998
   193
mengj@17998
   194
mengj@17998
   195
(**********************************************************************)
mengj@17998
   196
(* convert clause into ATP specific formats:                          *)
mengj@17998
   197
(* TPTP used by Vampire and E                                         *)
mengj@19720
   198
(* DFG used by SPASS                                                  *)
mengj@17998
   199
(**********************************************************************)
mengj@17998
   200
paulson@22078
   201
(*Result of a function type; no need to check that the argument type matches.*)
paulson@22078
   202
fun result_type (RC.Comp ("tc_fun", [_, tp2])) = tp2
wenzelm@27187
   203
  | result_type _ = error "result_type"
paulson@22078
   204
paulson@21513
   205
fun type_of_combterm (CombConst(c,tp,_)) = tp
paulson@21513
   206
  | type_of_combterm (CombVar(v,tp)) = tp
paulson@22078
   207
  | type_of_combterm (CombApp(t1,t2)) = result_type (type_of_combterm t1);
mengj@17998
   208
paulson@22064
   209
(*gets the head of a combinator application, along with the list of arguments*)
paulson@22064
   210
fun strip_comb u =
paulson@22078
   211
    let fun stripc (CombApp(t,u), ts) = stripc (t, u::ts)
paulson@22064
   212
        |   stripc  x =  x
paulson@22064
   213
    in  stripc(u,[])  end;
paulson@22064
   214
paulson@22851
   215
val type_wrapper = "ti";
paulson@22851
   216
immler@30150
   217
fun head_needs_hBOOL const_needs_hBOOL (CombConst(c,_,_)) = needs_hBOOL const_needs_hBOOL c
immler@30150
   218
  | head_needs_hBOOL const_needs_hBOOL _ = true;
paulson@22851
   219
wenzelm@24311
   220
fun wrap_type (s, tp) =
paulson@22851
   221
  if !typ_level=T_FULL then
paulson@22851
   222
      type_wrapper ^ RC.paren_pack [s, RC.string_of_fol_type tp]
paulson@22851
   223
  else s;
wenzelm@24311
   224
paulson@22851
   225
fun apply ss = "hAPP" ^ RC.paren_pack ss;
paulson@22851
   226
paulson@22064
   227
fun rev_apply (v, []) = v
paulson@22851
   228
  | rev_apply (v, arg::args) = apply [rev_apply (v, args), arg];
paulson@22064
   229
paulson@22064
   230
fun string_apply (v, args) = rev_apply (v, rev args);
paulson@22064
   231
paulson@22064
   232
(*Apply an operator to the argument strings, using either the "apply" operator or
paulson@22064
   233
  direct function application.*)
immler@30150
   234
fun string_of_applic cma (CombConst(c,tp,tvars), args) =
paulson@22064
   235
      let val c = if c = "equal" then "c_fequal" else c
immler@30150
   236
          val nargs = min_arity_of cma c
paulson@22851
   237
          val args1 = List.take(args, nargs)
wenzelm@27187
   238
            handle Subscript => error ("string_of_applic: " ^ c ^ " has arity " ^
wenzelm@27187
   239
                                         Int.toString nargs ^ " but is applied to " ^
wenzelm@27187
   240
                                         space_implode ", " args)
paulson@22064
   241
          val args2 = List.drop(args, nargs)
paulson@22851
   242
          val targs = if !typ_level = T_CONST then map RC.string_of_fol_type tvars
paulson@22851
   243
                      else []
paulson@21513
   244
      in
wenzelm@24311
   245
          string_apply (c ^ RC.paren_pack (args1@targs), args2)
paulson@21513
   246
      end
immler@30150
   247
  | string_of_applic cma (CombVar(v,tp), args) = string_apply (v, args)
immler@30149
   248
  | string_of_applic _ _ = error "string_of_applic";
paulson@22064
   249
immler@30150
   250
fun wrap_type_if cnh (head, s, tp) = if head_needs_hBOOL cnh head then wrap_type (s, tp) else s;
wenzelm@24311
   251
immler@30150
   252
fun string_of_combterm cma cnh t =
paulson@22064
   253
  let val (head, args) = strip_comb t
immler@30150
   254
  in  wrap_type_if cnh (head,
immler@30150
   255
                    string_of_applic cma (head, map (string_of_combterm cma cnh) args),
paulson@22851
   256
                    type_of_combterm t)
paulson@22851
   257
  end;
mengj@18356
   258
paulson@22064
   259
(*Boolean-valued terms are here converted to literals.*)
immler@30150
   260
fun boolify cma cnh t = "hBOOL" ^ RC.paren_pack [string_of_combterm cma cnh t];
paulson@22064
   261
immler@30150
   262
fun string_of_predicate cma cnh t =
paulson@22064
   263
  case t of
paulson@22078
   264
      (CombApp(CombApp(CombConst("equal",_,_), t1), t2)) =>
wenzelm@24311
   265
          (*DFG only: new TPTP prefers infix equality*)
immler@30150
   266
          ("equal" ^ RC.paren_pack [string_of_combterm cma cnh t1, string_of_combterm cma cnh t2])
wenzelm@24311
   267
    | _ =>
paulson@22064
   268
          case #1 (strip_comb t) of
immler@30150
   269
              CombConst(c,_,_) => if needs_hBOOL cnh c then boolify cma cnh t else string_of_combterm cma cnh t
immler@30150
   270
            | _ => boolify cma cnh t;
mengj@18356
   271
wenzelm@24311
   272
fun string_of_clausename (cls_id,ax_name) =
paulson@22078
   273
    RC.clause_prefix ^ RC.ascii_of ax_name ^ "_" ^ Int.toString cls_id;
mengj@17998
   274
wenzelm@24311
   275
fun string_of_type_clsname (cls_id,ax_name,idx) =
mengj@17998
   276
    string_of_clausename (cls_id,ax_name) ^ "_tcs" ^ (Int.toString idx);
mengj@17998
   277
mengj@17998
   278
paulson@21561
   279
(*** tptp format ***)
mengj@19720
   280
immler@30150
   281
fun tptp_of_equality cma cnh pol (t1,t2) =
paulson@21513
   282
  let val eqop = if pol then " = " else " != "
immler@30150
   283
  in  string_of_combterm cma cnh t1 ^ eqop ^ string_of_combterm cma cnh t2  end;
paulson@21513
   284
immler@30150
   285
fun tptp_literal cma cnh (Literal(pol, CombApp(CombApp(CombConst("equal",_,_), t1), t2))) =
immler@30150
   286
      tptp_of_equality cma cnh pol (t1,t2)
immler@30150
   287
  | tptp_literal cma cnh (Literal(pol,pred)) =
immler@30150
   288
      RC.tptp_sign pol (string_of_predicate cma cnh pred);
wenzelm@24311
   289
paulson@22064
   290
(*Given a clause, returns its literals paired with a list of literals concerning TFrees;
paulson@22064
   291
  the latter should only occur in conjecture clauses.*)
immler@30150
   292
fun tptp_type_lits cma cnh pos (Clause{literals, ctypes_sorts, ...}) =
immler@30150
   293
      (map (tptp_literal cma cnh) literals, 
paulson@24937
   294
       map (RC.tptp_of_typeLit pos) (RC.add_typs ctypes_sorts));
wenzelm@24311
   295
immler@30150
   296
fun clause2tptp cma cnh (cls as Clause{axiom_name,clause_id,kind,ctypes_sorts,...}) =
immler@30150
   297
  let val (lits,tylits) = tptp_type_lits cma cnh (kind = RC.Conjecture) cls
paulson@24937
   298
  in
paulson@24937
   299
      (RC.gen_tptp_cls(clause_id,axiom_name,kind,lits,tylits), tylits)
paulson@24937
   300
  end;
mengj@17998
   301
mengj@17998
   302
paulson@21561
   303
(*** dfg format ***)
paulson@21561
   304
immler@30150
   305
fun dfg_literal cma cnh (Literal(pol,pred)) = RC.dfg_sign pol (string_of_predicate cma cnh pred);
mengj@19720
   306
immler@30150
   307
fun dfg_type_lits cma cnh pos (Clause{literals, ctypes_sorts, ...}) =
immler@30150
   308
      (map (dfg_literal cma cnh) literals, 
paulson@24937
   309
       map (RC.dfg_of_typeLit pos) (RC.add_typs ctypes_sorts));
mengj@19720
   310
paulson@22078
   311
fun get_uvars (CombConst _) vars = vars
paulson@22078
   312
  | get_uvars (CombVar(v,_)) vars = (v::vars)
paulson@22078
   313
  | get_uvars (CombApp(P,Q)) vars = get_uvars P (get_uvars Q vars);
mengj@19720
   314
mengj@19720
   315
fun get_uvars_l (Literal(_,c)) = get_uvars c [];
mengj@19720
   316
paulson@22078
   317
fun dfg_vars (Clause {literals,...}) = RC.union_all (map get_uvars_l literals);
wenzelm@24311
   318
immler@30150
   319
fun clause2dfg cma cnh (cls as Clause{axiom_name,clause_id,kind,ctypes_sorts,...}) =
immler@30150
   320
  let val (lits,tylits) = dfg_type_lits cma cnh (kind = RC.Conjecture) cls
paulson@24937
   321
      val vars = dfg_vars cls
paulson@24937
   322
      val tvars = RC.get_tvar_strs ctypes_sorts
paulson@24937
   323
  in
paulson@24937
   324
      (RC.gen_dfg_cls(clause_id, axiom_name, kind, lits, tylits, tvars@vars), tylits)
paulson@24937
   325
  end;
paulson@24937
   326
mengj@19720
   327
paulson@22064
   328
(** For DFG format: accumulate function and predicate declarations **)
mengj@19720
   329
paulson@22078
   330
fun addtypes tvars tab = foldl RC.add_foltype_funcs tab tvars;
mengj@19720
   331
immler@30150
   332
fun add_decls cma cnh (CombConst(c,tp,tvars), (funcs,preds)) =
paulson@22064
   333
      if c = "equal" then (addtypes tvars funcs, preds)
paulson@21561
   334
      else
immler@30150
   335
	let val arity = min_arity_of cma c
paulson@24385
   336
	    val ntys = if !typ_level = T_CONST then length tvars else 0
paulson@24385
   337
	    val addit = Symtab.update(c, arity+ntys)
paulson@24385
   338
	in
immler@30150
   339
	    if needs_hBOOL cnh c then (addtypes tvars (addit funcs), preds)
paulson@24385
   340
	    else (addtypes tvars funcs, addit preds)
paulson@24385
   341
	end
immler@30150
   342
  | add_decls _ _ (CombVar(_,ctp), (funcs,preds)) =
paulson@22078
   343
      (RC.add_foltype_funcs (ctp,funcs), preds)
immler@30150
   344
  | add_decls cma cnh (CombApp(P,Q),decls) = add_decls cma cnh (P,add_decls cma cnh (Q,decls));
mengj@19720
   345
immler@30150
   346
fun add_literal_decls cma cnh (Literal(_,c), decls) = add_decls cma cnh (c,decls);
mengj@19720
   347
immler@30150
   348
fun add_clause_decls cma cnh (Clause {literals, ...}, decls) =
immler@30150
   349
    foldl (add_literal_decls cma cnh) decls literals
wenzelm@27187
   350
    handle Symtab.DUP a => error ("function " ^ a ^ " has multiple arities")
mengj@19720
   351
immler@30150
   352
fun decls_of_clauses cma cnh clauses arity_clauses =
paulson@24385
   353
  let val init_functab = Symtab.update (type_wrapper,2) (Symtab.update ("hAPP",2) RC.init_functab)
paulson@22064
   354
      val init_predtab = Symtab.update ("hBOOL",1) Symtab.empty
immler@30150
   355
      val (functab,predtab) = (foldl (add_clause_decls cma cnh) (init_functab, init_predtab) clauses)
paulson@22064
   356
  in
wenzelm@24311
   357
      (Symtab.dest (foldl RC.add_arityClause_funcs functab arity_clauses),
paulson@22064
   358
       Symtab.dest predtab)
paulson@22064
   359
  end;
mengj@19720
   360
paulson@21398
   361
fun add_clause_preds (Clause {ctypes_sorts, ...}, preds) =
paulson@22078
   362
  foldl RC.add_type_sort_preds preds ctypes_sorts
wenzelm@27187
   363
  handle Symtab.DUP a => error ("predicate " ^ a ^ " has multiple arities")
paulson@21398
   364
paulson@21398
   365
(*Higher-order clauses have only the predicates hBOOL and type classes.*)
wenzelm@24311
   366
fun preds_of_clauses clauses clsrel_clauses arity_clauses =
mengj@19720
   367
    Symtab.dest
wenzelm@24311
   368
        (foldl RC.add_classrelClause_preds
wenzelm@24311
   369
               (foldl RC.add_arityClause_preds
wenzelm@24311
   370
                      (foldl add_clause_preds Symtab.empty clauses)
wenzelm@24311
   371
                      arity_clauses)
wenzelm@24311
   372
               clsrel_clauses)
mengj@19720
   373
mengj@18440
   374
mengj@18440
   375
(**********************************************************************)
mengj@19198
   376
(* write clauses to files                                             *)
mengj@19198
   377
(**********************************************************************)
mengj@19198
   378
paulson@21573
   379
val init_counters =
paulson@21573
   380
    Symtab.make [("c_COMBI", 0), ("c_COMBK", 0),
wenzelm@24311
   381
                 ("c_COMBB", 0), ("c_COMBC", 0),
paulson@24943
   382
                 ("c_COMBS", 0)];
wenzelm@24311
   383
wenzelm@24311
   384
fun count_combterm (CombConst(c,tp,_), ct) =
paulson@21573
   385
     (case Symtab.lookup ct c of NONE => ct  (*no counter*)
paulson@21573
   386
                               | SOME n => Symtab.update (c,n+1) ct)
paulson@21573
   387
  | count_combterm (CombVar(v,tp), ct) = ct
paulson@22078
   388
  | count_combterm (CombApp(t1,t2), ct) = count_combterm(t1, count_combterm(t2, ct));
paulson@21573
   389
paulson@21573
   390
fun count_literal (Literal(_,t), ct) = count_combterm(t,ct);
paulson@21573
   391
paulson@21573
   392
fun count_clause (Clause{literals,...}, ct) = foldl count_literal ct literals;
paulson@21573
   393
wenzelm@24311
   394
fun count_user_clause user_lemmas (Clause{axiom_name,literals,...}, ct) =
paulson@21573
   395
  if axiom_name mem_string user_lemmas then foldl count_literal ct literals
paulson@21573
   396
  else ct;
paulson@21573
   397
wenzelm@27178
   398
fun cnf_helper_thms thy =
wenzelm@27178
   399
  ResAxioms.cnf_rules_pairs thy o map ResAxioms.pairname
mengj@20644
   400
wenzelm@24323
   401
fun get_helper_clauses thy isFO (conjectures, axclauses, user_lemmas) =
paulson@23386
   402
  if isFO then []  (*first-order*)
paulson@23386
   403
  else
paulson@22064
   404
    let val ct0 = foldl count_clause init_counters conjectures
paulson@22064
   405
        val ct = foldl (count_user_clause user_lemmas) ct0 axclauses
paulson@22064
   406
        fun needed c = valOf (Symtab.lookup ct c) > 0
wenzelm@24311
   407
        val IK = if needed "c_COMBI" orelse needed "c_COMBK"
wenzelm@27178
   408
                 then (Output.debug (fn () => "Include combinator I K"); cnf_helper_thms thy [comb_I,comb_K])
wenzelm@24311
   409
                 else []
wenzelm@24311
   410
        val BC = if needed "c_COMBB" orelse needed "c_COMBC"
wenzelm@27178
   411
                 then (Output.debug (fn () => "Include combinator B C"); cnf_helper_thms thy [comb_B,comb_C])
paulson@21135
   412
                 else []
wenzelm@24311
   413
        val S = if needed "c_COMBS"
wenzelm@27178
   414
                then (Output.debug (fn () => "Include combinator S"); cnf_helper_thms thy [comb_S])
wenzelm@24311
   415
                else []
wenzelm@27178
   416
        val other = cnf_helper_thms thy [ext,fequal_imp_equal,equal_imp_fequal]
mengj@20791
   417
    in
paulson@24943
   418
        map #2 (make_axiom_clauses thy (other @ IK @ BC @ S))
paulson@23386
   419
    end;
mengj@20791
   420
paulson@22064
   421
(*Find the minimal arity of each function mentioned in the term. Also, note which uses
paulson@22064
   422
  are not at top level, to see if hBOOL is needed.*)
immler@30150
   423
fun count_constants_term toplev t (const_min_arity, const_needs_hBOOL) =
paulson@22064
   424
  let val (head, args) = strip_comb t
paulson@22064
   425
      val n = length args
immler@30150
   426
      val (const_min_arity, const_needs_hBOOL) = fold (count_constants_term false) args (const_min_arity, const_needs_hBOOL)
paulson@22064
   427
  in
paulson@22064
   428
      case head of
wenzelm@24311
   429
          CombConst (a,_,_) => (*predicate or function version of "equal"?*)
wenzelm@24311
   430
            let val a = if a="equal" andalso not toplev then "c_fequal" else a
immler@30149
   431
            val const_min_arity = Symtab.map_default (a,n) (curry Int.min n) const_min_arity
wenzelm@24311
   432
            in
immler@30150
   433
              if toplev then (const_min_arity, const_needs_hBOOL)
immler@30150
   434
              else (const_min_arity, Symtab.update (a,true) (const_needs_hBOOL))
wenzelm@24311
   435
            end
immler@30150
   436
        | ts => (const_min_arity, const_needs_hBOOL)
paulson@22064
   437
  end;
paulson@22064
   438
paulson@22064
   439
(*A literal is a top-level term*)
immler@30150
   440
fun count_constants_lit (Literal (_,t)) (const_min_arity, const_needs_hBOOL) =
immler@30150
   441
  count_constants_term true t (const_min_arity, const_needs_hBOOL);
paulson@22064
   442
immler@30150
   443
fun count_constants_clause (Clause{literals,...}) (const_min_arity, const_needs_hBOOL) =
immler@30150
   444
  fold count_constants_lit literals (const_min_arity, const_needs_hBOOL);
paulson@22064
   445
immler@30150
   446
fun display_arity const_needs_hBOOL (c,n) =
wenzelm@24311
   447
  Output.debug (fn () => "Constant: " ^ c ^ " arity:\t" ^ Int.toString n ^
immler@30150
   448
                (if needs_hBOOL const_needs_hBOOL c then " needs hBOOL" else ""));
paulson@22064
   449
wenzelm@24311
   450
fun count_constants (conjectures, axclauses, helper_clauses) =
paulson@24385
   451
  if !minimize_applies then
immler@30150
   452
     let val (const_min_arity, const_needs_hBOOL) =
immler@30150
   453
          fold count_constants_clause conjectures (Symtab.empty, Symtab.empty)
immler@30149
   454
       |> fold count_constants_clause axclauses
immler@30149
   455
       |> fold count_constants_clause helper_clauses
immler@30150
   456
     val _ = List.app (display_arity const_needs_hBOOL) (Symtab.dest (const_min_arity))
immler@30150
   457
     in (const_min_arity, const_needs_hBOOL) end
immler@30150
   458
  else (Symtab.empty, Symtab.empty);
paulson@22064
   459
mengj@20791
   460
(* tptp format *)
wenzelm@24311
   461
mengj@19198
   462
(* write TPTP format to a single file *)
wenzelm@24323
   463
fun tptp_write_file thy isFO thms filename (ax_tuples,classrel_clauses,arity_clauses) user_lemmas =
paulson@23386
   464
    let val _ = Output.debug (fn () => ("Preparing to write the TPTP file " ^ filename))
paulson@23386
   465
        val _ = RC.dfg_format := false
wenzelm@24323
   466
        val conjectures = make_conjecture_clauses thy thms
wenzelm@24323
   467
        val (clnames,axclauses) = ListPair.unzip (make_axiom_clauses thy ax_tuples)
wenzelm@24323
   468
        val helper_clauses = get_helper_clauses thy isFO (conjectures, axclauses, user_lemmas)
immler@30150
   469
        val (const_min_arity, const_needs_hBOOL) = count_constants (conjectures, axclauses, helper_clauses);
immler@30150
   470
        val (tptp_clss,tfree_litss) = ListPair.unzip (map (clause2tptp const_min_arity const_needs_hBOOL) conjectures)
wenzelm@24311
   471
        val tfree_clss = map RC.tptp_tfree_clause (foldl (op union_string) [] tfree_litss)
paulson@22064
   472
        val out = TextIO.openOut filename
mengj@19198
   473
    in
immler@30150
   474
        List.app (curry TextIO.output out o #1 o (clause2tptp const_min_arity const_needs_hBOOL)) axclauses;
wenzelm@24311
   475
        RC.writeln_strs out tfree_clss;
wenzelm@24311
   476
        RC.writeln_strs out tptp_clss;
wenzelm@24311
   477
        List.app (curry TextIO.output out o RC.tptp_classrelClause) classrel_clauses;
wenzelm@24311
   478
        List.app (curry TextIO.output out o RC.tptp_arity_clause) arity_clauses;
immler@30150
   479
        List.app (curry TextIO.output out o #1 o (clause2tptp const_min_arity const_needs_hBOOL)) helper_clauses;
wenzelm@24311
   480
        TextIO.closeOut out;
wenzelm@24311
   481
        clnames
mengj@19198
   482
    end;
mengj@19198
   483
mengj@19720
   484
mengj@19720
   485
(* dfg format *)
mengj@19720
   486
wenzelm@24323
   487
fun dfg_write_file thy isFO thms filename (ax_tuples,classrel_clauses,arity_clauses) user_lemmas =
paulson@23386
   488
    let val _ = Output.debug (fn () => ("Preparing to write the DFG file " ^ filename))
paulson@23386
   489
        val _ = RC.dfg_format := true
wenzelm@24323
   490
        val conjectures = make_conjecture_clauses thy thms
wenzelm@24323
   491
        val (clnames,axclauses) = ListPair.unzip (make_axiom_clauses thy ax_tuples)
wenzelm@24323
   492
        val helper_clauses = get_helper_clauses thy isFO (conjectures, axclauses, user_lemmas)
immler@30150
   493
        val (const_min_arity, const_needs_hBOOL) = count_constants (conjectures, axclauses, helper_clauses);
immler@30150
   494
        val (dfg_clss, tfree_litss) = ListPair.unzip (map (clause2dfg const_min_arity const_needs_hBOOL) conjectures)
wenzelm@24311
   495
        and probname = Path.implode (Path.base (Path.explode filename))
immler@30150
   496
        val axstrs = map (#1 o (clause2dfg const_min_arity const_needs_hBOOL)) axclauses
wenzelm@24311
   497
        val tfree_clss = map RC.dfg_tfree_clause (RC.union_all tfree_litss)
wenzelm@24311
   498
        val out = TextIO.openOut filename
immler@30150
   499
        val helper_clauses_strs = map (#1 o (clause2dfg const_min_arity const_needs_hBOOL)) helper_clauses
immler@30150
   500
        val (funcs,cl_preds) = decls_of_clauses const_min_arity const_needs_hBOOL (helper_clauses @ conjectures @ axclauses) arity_clauses
wenzelm@24311
   501
        and ty_preds = preds_of_clauses axclauses classrel_clauses arity_clauses
mengj@19720
   502
    in
wenzelm@24311
   503
        TextIO.output (out, RC.string_of_start probname);
wenzelm@24311
   504
        TextIO.output (out, RC.string_of_descrip probname);
wenzelm@24311
   505
        TextIO.output (out, RC.string_of_symbols
wenzelm@24311
   506
                              (RC.string_of_funcs funcs)
wenzelm@24311
   507
                              (RC.string_of_preds (cl_preds @ ty_preds)));
wenzelm@24311
   508
        TextIO.output (out, "list_of_clauses(axioms,cnf).\n");
wenzelm@24311
   509
        RC.writeln_strs out axstrs;
wenzelm@24311
   510
        List.app (curry TextIO.output out o RC.dfg_classrelClause) classrel_clauses;
wenzelm@24311
   511
        List.app (curry TextIO.output out o RC.dfg_arity_clause) arity_clauses;
wenzelm@24311
   512
        RC.writeln_strs out helper_clauses_strs;
wenzelm@24311
   513
        TextIO.output (out, "end_of_list.\n\nlist_of_clauses(conjectures,cnf).\n");
wenzelm@24311
   514
        RC.writeln_strs out tfree_clss;
wenzelm@24311
   515
        RC.writeln_strs out dfg_clss;
wenzelm@24311
   516
        TextIO.output (out, "end_of_list.\n\n");
wenzelm@24311
   517
        (*VarWeight=3 helps the HO problems, probably by counteracting the presence of hAPP*)
wenzelm@24311
   518
        TextIO.output (out, "list_of_settings(SPASS).\n{*\nset_flag(VarWeight,3).\n*}\nend_of_list.\n\n");
wenzelm@24311
   519
        TextIO.output (out, "end_problem.\n");
wenzelm@24311
   520
        TextIO.closeOut out;
wenzelm@24311
   521
        clnames
mengj@19720
   522
    end;
mengj@19720
   523
wenzelm@21254
   524
end