src/HOL/Tools/SMT/smt_builtin.ML
author boehmes
Mon Dec 20 22:02:57 2010 +0100 (2010-12-20)
changeset 41328 6792a5c92a58
parent 41281 679118e35378
child 41336 0ea5b9c7d233
permissions -rw-r--r--
avoid ML structure aliases (especially single-letter abbreviations)
boehmes@40277
     1
(*  Title:      HOL/Tools/SMT/smt_builtin.ML
boehmes@40277
     2
    Author:     Sascha Boehme, TU Muenchen
boehmes@40277
     3
boehmes@41059
     4
Tables of types and terms directly supported by SMT solvers.
boehmes@40277
     5
*)
boehmes@40277
     6
boehmes@40277
     7
signature SMT_BUILTIN =
boehmes@40277
     8
sig
boehmes@41059
     9
  (*built-in types*)
boehmes@41124
    10
  val add_builtin_typ: SMT_Utils.class ->
boehmes@41072
    11
    typ * (typ -> string option) * (typ -> int -> string option) ->
boehmes@41072
    12
    Context.generic -> Context.generic
boehmes@41072
    13
  val add_builtin_typ_ext: typ * (typ -> bool) -> Context.generic ->
boehmes@41072
    14
    Context.generic
boehmes@41281
    15
  val dest_builtin_typ: Proof.context -> typ -> string option
boehmes@41059
    16
  val is_builtin_typ_ext: Proof.context -> typ -> bool
boehmes@41059
    17
boehmes@41059
    18
  (*built-in numbers*)
boehmes@41281
    19
  val dest_builtin_num: Proof.context -> term -> (string * typ) option
boehmes@41059
    20
  val is_builtin_num: Proof.context -> term -> bool
boehmes@41059
    21
  val is_builtin_num_ext: Proof.context -> term -> bool
boehmes@41059
    22
boehmes@41059
    23
  (*built-in functions*)
boehmes@41059
    24
  type 'a bfun = Proof.context -> typ -> term list -> 'a
boehmes@41281
    25
  type bfunr = string * int * term list * (term list -> term)
boehmes@41124
    26
  val add_builtin_fun: SMT_Utils.class ->
boehmes@41281
    27
    (string * typ) * bfunr option bfun -> Context.generic -> Context.generic
boehmes@41124
    28
  val add_builtin_fun': SMT_Utils.class -> term * string -> Context.generic ->
boehmes@41072
    29
    Context.generic
boehmes@41072
    30
  val add_builtin_fun_ext: (string * typ) * bool bfun -> Context.generic ->
boehmes@41072
    31
    Context.generic
boehmes@41072
    32
  val add_builtin_fun_ext': string * typ -> Context.generic -> Context.generic
boehmes@41072
    33
  val add_builtin_fun_ext'': string -> Context.generic -> Context.generic
boehmes@41281
    34
  val dest_builtin_fun: Proof.context -> string * typ -> term list ->
boehmes@41281
    35
    bfunr option
boehmes@41281
    36
  val dest_builtin_eq: Proof.context -> term -> term -> bfunr option
boehmes@41281
    37
  val dest_builtin_pred: Proof.context -> string * typ -> term list ->
boehmes@41281
    38
    bfunr option
boehmes@41281
    39
  val dest_builtin_conn: Proof.context -> string * typ -> term list ->
boehmes@41281
    40
    bfunr option
boehmes@41281
    41
  val dest_builtin: Proof.context -> string * typ -> term list -> bfunr option
boehmes@41059
    42
  val is_builtin_fun: Proof.context -> string * typ -> term list -> bool
boehmes@41126
    43
  val is_builtin_fun_ext: Proof.context -> string * typ -> term list -> bool
boehmes@41059
    44
  val is_builtin_ext: Proof.context -> string * typ -> term list -> bool
boehmes@40277
    45
end
boehmes@40277
    46
boehmes@40277
    47
structure SMT_Builtin: SMT_BUILTIN =
boehmes@40277
    48
struct
boehmes@40277
    49
boehmes@41059
    50
boehmes@41059
    51
(* built-in tables *)
boehmes@41059
    52
boehmes@41059
    53
datatype ('a, 'b) kind = Ext of 'a | Int of 'b
boehmes@41059
    54
boehmes@41328
    55
type ('a, 'b) ttab = ((typ * ('a, 'b) kind) Ord_List.T) SMT_Utils.dict 
blanchet@40677
    56
boehmes@41059
    57
fun typ_ord ((T, _), (U, _)) =
boehmes@41059
    58
  let
boehmes@41059
    59
    fun tord (TVar _, Type _) = GREATER
boehmes@41059
    60
      | tord (Type _, TVar _) = LESS
boehmes@41059
    61
      | tord (Type (n, Ts), Type (m, Us)) =
boehmes@41059
    62
          if n = m then list_ord tord (Ts, Us)
boehmes@41059
    63
          else Term_Ord.typ_ord (T, U)
boehmes@41059
    64
      | tord TU = Term_Ord.typ_ord TU
boehmes@41059
    65
  in tord (T, U) end
boehmes@40277
    66
boehmes@41059
    67
fun insert_ttab cs T f =
boehmes@41328
    68
  SMT_Utils.dict_map_default (cs, [])
boehmes@41059
    69
    (Ord_List.insert typ_ord (perhaps (try Logic.varifyT_global) T, f))
boehmes@41059
    70
boehmes@41059
    71
fun merge_ttab ttabp =
boehmes@41328
    72
  SMT_Utils.dict_merge (uncurry (Ord_List.union typ_ord) o swap) ttabp
boehmes@40277
    73
boehmes@41059
    74
fun lookup_ttab ctxt ttab T =
boehmes@41124
    75
  let fun match (U, _) = Sign.typ_instance (ProofContext.theory_of ctxt) (T, U)
boehmes@41124
    76
  in
boehmes@41328
    77
    get_first (find_first match)
boehmes@41328
    78
      (SMT_Utils.dict_lookup ttab (SMT_Config.solver_class_of ctxt))
boehmes@41124
    79
  end
boehmes@41059
    80
boehmes@41059
    81
type ('a, 'b) btab = ('a, 'b) ttab Symtab.table
boehmes@41059
    82
boehmes@41059
    83
fun insert_btab cs n T f =
boehmes@41059
    84
  Symtab.map_default (n, []) (insert_ttab cs T f)
boehmes@41059
    85
boehmes@41059
    86
fun merge_btab btabp = Symtab.join (K merge_ttab) btabp
boehmes@40277
    87
boehmes@41059
    88
fun lookup_btab ctxt btab (n, T) =
boehmes@41059
    89
  (case Symtab.lookup btab n of
boehmes@41059
    90
    NONE => NONE
boehmes@41059
    91
  | SOME ttab => lookup_ttab ctxt ttab T)
boehmes@40277
    92
boehmes@41059
    93
boehmes@41059
    94
(* built-in types *)
boehmes@40277
    95
boehmes@41072
    96
structure Builtin_Types = Generic_Data
boehmes@41059
    97
(
boehmes@41059
    98
  type T =
boehmes@41059
    99
    (typ -> bool, (typ -> string option) * (typ -> int -> string option)) ttab
boehmes@41072
   100
  val empty = []
boehmes@41059
   101
  val extend = I
boehmes@41059
   102
  val merge = merge_ttab
boehmes@41059
   103
)
boehmes@41059
   104
boehmes@41059
   105
fun add_builtin_typ cs (T, f, g) =
boehmes@41059
   106
  Builtin_Types.map (insert_ttab cs T (Int (f, g)))
boehmes@41059
   107
boehmes@41059
   108
fun add_builtin_typ_ext (T, f) =
boehmes@41328
   109
  Builtin_Types.map (insert_ttab SMT_Utils.basicC T (Ext f))
boehmes@40277
   110
boehmes@41059
   111
fun lookup_builtin_typ ctxt =
boehmes@41072
   112
  lookup_ttab ctxt (Builtin_Types.get (Context.Proof ctxt))
boehmes@40277
   113
boehmes@41281
   114
fun dest_builtin_typ ctxt T =
boehmes@41059
   115
  (case lookup_builtin_typ ctxt T of
boehmes@41059
   116
    SOME (_, Int (f, _)) => f T
boehmes@41059
   117
  | _ => NONE) 
boehmes@40277
   118
boehmes@41059
   119
fun is_builtin_typ_ext ctxt T =
boehmes@41059
   120
  (case lookup_builtin_typ ctxt T of
boehmes@41059
   121
    SOME (_, Int (f, _)) => is_some (f T)
boehmes@41059
   122
  | SOME (_, Ext f) => f T
boehmes@40277
   123
  | NONE => false)
boehmes@40277
   124
boehmes@41059
   125
boehmes@41059
   126
(* built-in numbers *)
boehmes@41059
   127
boehmes@41281
   128
fun dest_builtin_num ctxt t =
boehmes@41059
   129
  (case try HOLogic.dest_number t of
boehmes@41059
   130
    NONE => NONE
boehmes@41059
   131
  | SOME (T, i) =>
boehmes@41059
   132
      (case lookup_builtin_typ ctxt T of
boehmes@41127
   133
        SOME (_, Int (_, g)) => g T i |> Option.map (rpair T)
boehmes@41059
   134
      | _ => NONE))
boehmes@41059
   135
boehmes@41281
   136
val is_builtin_num = is_some oo dest_builtin_num
boehmes@41059
   137
boehmes@41059
   138
fun is_builtin_num_ext ctxt t =
boehmes@41059
   139
  (case try HOLogic.dest_number t of
boehmes@41059
   140
    NONE => false
boehmes@41059
   141
  | SOME (T, _) => is_builtin_typ_ext ctxt T)
boehmes@41059
   142
boehmes@41059
   143
boehmes@41059
   144
(* built-in functions *)
boehmes@41059
   145
boehmes@41059
   146
type 'a bfun = Proof.context -> typ -> term list -> 'a
boehmes@41059
   147
boehmes@41281
   148
type bfunr = string * int * term list * (term list -> term)
boehmes@41281
   149
boehmes@41072
   150
structure Builtin_Funcs = Generic_Data
boehmes@41059
   151
(
boehmes@41281
   152
  type T = (bool bfun, bfunr option bfun) btab
boehmes@41126
   153
  val empty = Symtab.empty
boehmes@41059
   154
  val extend = I
boehmes@41059
   155
  val merge = merge_btab
boehmes@41059
   156
)
boehmes@41059
   157
boehmes@41059
   158
fun add_builtin_fun cs ((n, T), f) =
boehmes@41059
   159
  Builtin_Funcs.map (insert_btab cs n T (Int f))
boehmes@40277
   160
boehmes@41059
   161
fun add_builtin_fun' cs (t, n) =
boehmes@41127
   162
  let
boehmes@41281
   163
    val c as (m, T) = Term.dest_Const t
boehmes@41281
   164
    fun app U ts = Term.list_comb (Const (m, U), ts)
boehmes@41281
   165
    fun bfun _ U ts = SOME (n, length (Term.binder_types T), ts, app U)
boehmes@41281
   166
  in add_builtin_fun cs (c, bfun) end
boehmes@41059
   167
boehmes@41059
   168
fun add_builtin_fun_ext ((n, T), f) =
boehmes@41328
   169
  Builtin_Funcs.map (insert_btab SMT_Utils.basicC n T (Ext f))
boehmes@41059
   170
boehmes@41126
   171
fun add_builtin_fun_ext' c =
boehmes@41126
   172
  add_builtin_fun_ext (c, fn _ => fn _ => fn _ => true)
boehmes@41059
   173
boehmes@41072
   174
fun add_builtin_fun_ext'' n context =
boehmes@41072
   175
  let val thy = Context.theory_of context
boehmes@41072
   176
  in add_builtin_fun_ext' (n, Sign.the_const_type thy n) context end
boehmes@41059
   177
boehmes@41059
   178
fun lookup_builtin_fun ctxt =
boehmes@41072
   179
  lookup_btab ctxt (Builtin_Funcs.get (Context.Proof ctxt))
boehmes@41059
   180
boehmes@41281
   181
fun dest_builtin_fun ctxt (c as (_, T)) ts =
boehmes@41059
   182
  (case lookup_builtin_fun ctxt c of
boehmes@41059
   183
    SOME (_, Int f) => f ctxt T ts
boehmes@41059
   184
  | _ => NONE)
boehmes@41059
   185
boehmes@41281
   186
fun dest_builtin_eq ctxt t u =
boehmes@41281
   187
  let
boehmes@41281
   188
    val aT = TFree (Name.aT, @{sort type})
boehmes@41281
   189
    val c = (@{const_name HOL.eq}, aT --> aT --> @{typ bool})
boehmes@41281
   190
    fun mk ts = Term.list_comb (HOLogic.eq_const (Term.fastype_of (hd ts)), ts)
boehmes@41281
   191
  in
boehmes@41281
   192
    dest_builtin_fun ctxt c []
boehmes@41281
   193
    |> Option.map (fn (n, i, _, _) => (n, i, [t, u], mk))
boehmes@41281
   194
  end
boehmes@41281
   195
boehmes@41281
   196
fun special_builtin_fun pred ctxt (c as (_, T)) ts =
boehmes@41281
   197
  if pred (Term.body_type T, Term.binder_types T) then
boehmes@41281
   198
    dest_builtin_fun ctxt c ts
boehmes@41281
   199
  else NONE
boehmes@41281
   200
boehmes@41281
   201
fun dest_builtin_pred ctxt = special_builtin_fun (equal @{typ bool} o fst) ctxt
boehmes@41281
   202
boehmes@41281
   203
fun dest_builtin_conn ctxt =
boehmes@41281
   204
  special_builtin_fun (forall (equal @{typ bool}) o (op ::)) ctxt
boehmes@41281
   205
boehmes@41281
   206
fun dest_builtin ctxt c ts =
boehmes@41281
   207
  let val t =Term.list_comb (Const c, ts)
boehmes@41281
   208
  in
boehmes@41281
   209
    (case dest_builtin_num ctxt t of
boehmes@41281
   210
      SOME (n, _) => SOME (n, 0, [], K t)
boehmes@41281
   211
    | NONE => dest_builtin_fun ctxt c ts)
boehmes@41281
   212
  end
boehmes@41281
   213
boehmes@41281
   214
fun is_builtin_fun ctxt c ts = is_some (dest_builtin_fun ctxt c ts)
boehmes@41059
   215
boehmes@41059
   216
fun is_builtin_fun_ext ctxt (c as (_, T)) ts =
boehmes@41059
   217
  (case lookup_builtin_fun ctxt c of
boehmes@41059
   218
    SOME (_, Int f) => is_some (f ctxt T ts)
boehmes@41059
   219
  | SOME (_, Ext f) => f ctxt T ts
boehmes@41059
   220
  | NONE => false)
boehmes@41059
   221
boehmes@41280
   222
fun is_builtin_ext ctxt c ts =
boehmes@41280
   223
  is_builtin_num_ext ctxt (Term.list_comb (Const c, ts)) orelse 
boehmes@41280
   224
  is_builtin_fun_ext ctxt c ts
boehmes@40277
   225
boehmes@40277
   226
end