src/HOL/Tools/SMT/smt_builtin.ML
author blanchet
Mon Sep 30 16:07:56 2013 +0200 (2013-09-30)
changeset 53998 b352d3d4a58a
parent 46042 ab32a87ba01a
child 53999 ba9254f3111b
permissions -rw-r--r--
just one data slot (record) per program unit
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@41354
    30
  val add_builtin_fun_ext: (string * typ) * term list bfun ->
boehmes@41354
    31
    Context.generic -> 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@41354
    42
  val dest_builtin_ext: Proof.context -> string * typ -> term list ->
boehmes@41354
    43
    term list option
boehmes@41059
    44
  val is_builtin_fun: Proof.context -> string * typ -> term list -> bool
boehmes@41126
    45
  val is_builtin_fun_ext: Proof.context -> string * typ -> term list -> bool
boehmes@40277
    46
end
boehmes@40277
    47
boehmes@40277
    48
structure SMT_Builtin: SMT_BUILTIN =
boehmes@40277
    49
struct
boehmes@40277
    50
boehmes@41059
    51
boehmes@41059
    52
(* built-in tables *)
boehmes@41059
    53
boehmes@41059
    54
datatype ('a, 'b) kind = Ext of 'a | Int of 'b
boehmes@41059
    55
boehmes@41328
    56
type ('a, 'b) ttab = ((typ * ('a, 'b) kind) Ord_List.T) SMT_Utils.dict 
blanchet@40677
    57
boehmes@41059
    58
fun typ_ord ((T, _), (U, _)) =
boehmes@41059
    59
  let
boehmes@41059
    60
    fun tord (TVar _, Type _) = GREATER
boehmes@41059
    61
      | tord (Type _, TVar _) = LESS
boehmes@41059
    62
      | tord (Type (n, Ts), Type (m, Us)) =
boehmes@41059
    63
          if n = m then list_ord tord (Ts, Us)
boehmes@41059
    64
          else Term_Ord.typ_ord (T, U)
boehmes@41059
    65
      | tord TU = Term_Ord.typ_ord TU
boehmes@41059
    66
  in tord (T, U) end
boehmes@40277
    67
boehmes@41059
    68
fun insert_ttab cs T f =
boehmes@41328
    69
  SMT_Utils.dict_map_default (cs, [])
boehmes@41059
    70
    (Ord_List.insert typ_ord (perhaps (try Logic.varifyT_global) T, f))
boehmes@41059
    71
boehmes@41059
    72
fun merge_ttab ttabp =
wenzelm@41473
    73
  SMT_Utils.dict_merge (Ord_List.merge typ_ord) ttabp
boehmes@40277
    74
boehmes@41059
    75
fun lookup_ttab ctxt ttab T =
wenzelm@42361
    76
  let fun match (U, _) = Sign.typ_instance (Proof_Context.theory_of ctxt) (T, U)
boehmes@41124
    77
  in
boehmes@41328
    78
    get_first (find_first match)
boehmes@41328
    79
      (SMT_Utils.dict_lookup ttab (SMT_Config.solver_class_of ctxt))
boehmes@41124
    80
  end
boehmes@41059
    81
boehmes@41059
    82
type ('a, 'b) btab = ('a, 'b) ttab Symtab.table
boehmes@41059
    83
boehmes@41059
    84
fun insert_btab cs n T f =
boehmes@41059
    85
  Symtab.map_default (n, []) (insert_ttab cs T f)
boehmes@41059
    86
boehmes@41059
    87
fun merge_btab btabp = Symtab.join (K merge_ttab) btabp
boehmes@40277
    88
boehmes@41059
    89
fun lookup_btab ctxt btab (n, T) =
boehmes@41059
    90
  (case Symtab.lookup btab n of
boehmes@41059
    91
    NONE => NONE
boehmes@41059
    92
  | SOME ttab => lookup_ttab ctxt ttab T)
boehmes@40277
    93
blanchet@53998
    94
type 'a bfun = Proof.context -> typ -> term list -> 'a
blanchet@53998
    95
blanchet@53998
    96
type bfunr = string * int * term list * (term list -> term)
blanchet@53998
    97
blanchet@53998
    98
structure Builtins = Generic_Data
blanchet@53998
    99
(
blanchet@53998
   100
  type T =
blanchet@53998
   101
    (typ -> bool, (typ -> string option) * (typ -> int -> string option)) ttab *
blanchet@53998
   102
    (term list bfun, bfunr option bfun) btab
blanchet@53998
   103
  val empty = ([], Symtab.empty)
blanchet@53998
   104
  val extend = I
blanchet@53998
   105
  fun merge ((t1, b1), (t2, b2)) = (merge_ttab (t1, t2), merge_btab (b1, b2))
blanchet@53998
   106
)
boehmes@41059
   107
boehmes@41059
   108
(* built-in types *)
boehmes@40277
   109
boehmes@41059
   110
fun add_builtin_typ cs (T, f, g) =
blanchet@53998
   111
  Builtins.map (apfst (insert_ttab cs T (Int (f, g))))
boehmes@41059
   112
boehmes@41059
   113
fun add_builtin_typ_ext (T, f) =
blanchet@53998
   114
  Builtins.map (apfst (insert_ttab SMT_Utils.basicC T (Ext f)))
boehmes@40277
   115
boehmes@41059
   116
fun lookup_builtin_typ ctxt =
blanchet@53998
   117
  lookup_ttab ctxt (fst (Builtins.get (Context.Proof ctxt)))
boehmes@40277
   118
boehmes@41281
   119
fun dest_builtin_typ ctxt T =
boehmes@41059
   120
  (case lookup_builtin_typ ctxt T of
boehmes@41059
   121
    SOME (_, Int (f, _)) => f T
boehmes@41059
   122
  | _ => NONE) 
boehmes@40277
   123
boehmes@41059
   124
fun is_builtin_typ_ext ctxt T =
boehmes@41059
   125
  (case lookup_builtin_typ ctxt T of
boehmes@41059
   126
    SOME (_, Int (f, _)) => is_some (f T)
boehmes@41059
   127
  | SOME (_, Ext f) => f T
boehmes@40277
   128
  | NONE => false)
boehmes@40277
   129
boehmes@41059
   130
boehmes@41059
   131
(* built-in numbers *)
boehmes@41059
   132
boehmes@41281
   133
fun dest_builtin_num ctxt t =
boehmes@41059
   134
  (case try HOLogic.dest_number t of
boehmes@41059
   135
    NONE => NONE
boehmes@41059
   136
  | SOME (T, i) =>
boehmes@41059
   137
      (case lookup_builtin_typ ctxt T of
boehmes@41127
   138
        SOME (_, Int (_, g)) => g T i |> Option.map (rpair T)
boehmes@41059
   139
      | _ => NONE))
boehmes@41059
   140
boehmes@41281
   141
val is_builtin_num = is_some oo dest_builtin_num
boehmes@41059
   142
boehmes@41059
   143
fun is_builtin_num_ext ctxt t =
boehmes@41059
   144
  (case try HOLogic.dest_number t of
boehmes@41059
   145
    NONE => false
boehmes@41059
   146
  | SOME (T, _) => is_builtin_typ_ext ctxt T)
boehmes@41059
   147
boehmes@41059
   148
boehmes@41059
   149
(* built-in functions *)
boehmes@41059
   150
boehmes@41059
   151
fun add_builtin_fun cs ((n, T), f) =
blanchet@53998
   152
  Builtins.map (apsnd (insert_btab cs n T (Int f)))
boehmes@40277
   153
boehmes@41059
   154
fun add_builtin_fun' cs (t, n) =
boehmes@41127
   155
  let
boehmes@41281
   156
    val c as (m, T) = Term.dest_Const t
boehmes@41281
   157
    fun app U ts = Term.list_comb (Const (m, U), ts)
boehmes@41281
   158
    fun bfun _ U ts = SOME (n, length (Term.binder_types T), ts, app U)
boehmes@41281
   159
  in add_builtin_fun cs (c, bfun) end
boehmes@41059
   160
boehmes@41059
   161
fun add_builtin_fun_ext ((n, T), f) =
blanchet@53998
   162
  Builtins.map (apsnd (insert_btab SMT_Utils.basicC n T (Ext f)))
boehmes@41059
   163
boehmes@41354
   164
fun add_builtin_fun_ext' c = add_builtin_fun_ext (c, fn _ => fn _ => I)
boehmes@41059
   165
boehmes@41072
   166
fun add_builtin_fun_ext'' n context =
boehmes@41072
   167
  let val thy = Context.theory_of context
boehmes@41072
   168
  in add_builtin_fun_ext' (n, Sign.the_const_type thy n) context end
boehmes@41059
   169
boehmes@41059
   170
fun lookup_builtin_fun ctxt =
blanchet@53998
   171
  lookup_btab ctxt (snd (Builtins.get (Context.Proof ctxt)))
boehmes@41059
   172
boehmes@41281
   173
fun dest_builtin_fun ctxt (c as (_, T)) ts =
boehmes@41059
   174
  (case lookup_builtin_fun ctxt c of
boehmes@41059
   175
    SOME (_, Int f) => f ctxt T ts
boehmes@41059
   176
  | _ => NONE)
boehmes@41059
   177
boehmes@41281
   178
fun dest_builtin_eq ctxt t u =
boehmes@41281
   179
  let
boehmes@41281
   180
    val aT = TFree (Name.aT, @{sort type})
boehmes@41281
   181
    val c = (@{const_name HOL.eq}, aT --> aT --> @{typ bool})
boehmes@41281
   182
    fun mk ts = Term.list_comb (HOLogic.eq_const (Term.fastype_of (hd ts)), ts)
boehmes@41281
   183
  in
boehmes@41281
   184
    dest_builtin_fun ctxt c []
boehmes@41281
   185
    |> Option.map (fn (n, i, _, _) => (n, i, [t, u], mk))
boehmes@41281
   186
  end
boehmes@41281
   187
boehmes@41281
   188
fun special_builtin_fun pred ctxt (c as (_, T)) ts =
boehmes@41281
   189
  if pred (Term.body_type T, Term.binder_types T) then
boehmes@41281
   190
    dest_builtin_fun ctxt c ts
boehmes@41281
   191
  else NONE
boehmes@41281
   192
boehmes@41281
   193
fun dest_builtin_pred ctxt = special_builtin_fun (equal @{typ bool} o fst) ctxt
boehmes@41281
   194
boehmes@41281
   195
fun dest_builtin_conn ctxt =
boehmes@41281
   196
  special_builtin_fun (forall (equal @{typ bool}) o (op ::)) ctxt
boehmes@41281
   197
boehmes@41281
   198
fun dest_builtin ctxt c ts =
boehmes@41281
   199
  let val t =Term.list_comb (Const c, ts)
boehmes@41281
   200
  in
boehmes@41281
   201
    (case dest_builtin_num ctxt t of
boehmes@41281
   202
      SOME (n, _) => SOME (n, 0, [], K t)
boehmes@41281
   203
    | NONE => dest_builtin_fun ctxt c ts)
boehmes@41281
   204
  end
boehmes@41281
   205
boehmes@41354
   206
fun dest_builtin_fun_ext ctxt (c as (_, T)) ts =    
boehmes@41354
   207
  (case lookup_builtin_fun ctxt c of
boehmes@41354
   208
    SOME (_, Int f) => f ctxt T ts |> Option.map (fn (_, _, us, _) => us)
boehmes@41354
   209
  | SOME (_, Ext f) => SOME (f ctxt T ts)
boehmes@41354
   210
  | NONE => NONE)
boehmes@41354
   211
boehmes@41354
   212
fun dest_builtin_ext ctxt c ts =
boehmes@41354
   213
  if is_builtin_num_ext ctxt (Term.list_comb (Const c, ts)) then SOME []
boehmes@41354
   214
  else dest_builtin_fun_ext ctxt c ts
boehmes@41354
   215
boehmes@41281
   216
fun is_builtin_fun ctxt c ts = is_some (dest_builtin_fun ctxt c ts)
boehmes@41059
   217
boehmes@41354
   218
fun is_builtin_fun_ext ctxt c ts = is_some (dest_builtin_fun_ext ctxt c ts)
boehmes@41059
   219
boehmes@40277
   220
end