sidekick/doc/guide.md
2021-10-21 20:42:12 -04:00

13 KiB

Sidekick

This is a guide to using Sidekick and to implementing custom SMT solvers.

Quickstart

You will need opam and a relatively recent version of OCaml. I'd recommend OCaml 4.11, even though versions as old as 4.04 are supported. Using merlin or ocaml-lsp in your editor is strongly recommended, to help exploring the API and to get fast feedback on mistakes.

The API documentation can be found online.

Clone this repository:

$ git clone https://github.com/c-cube/sidekick.git
$ cd sidekick

Then create a local switch (much simpler in the long run). This will take a little while as it compiles OCaml.

$ opam sw create . 4.11.0

Let's test that sidekick is installed. In an ocaml toplevel (just run ocaml), type the first three lines (without the leading #, which indicates the OCaml prompt):

# #use "topfind";;
# #require "sidekick-base";;
# #show Sidekick_base;;
module Sidekick_base :
  sig
    module Base_types = Sidekick_base__.Base_types
...
  end

This should print the interface of the Sidekick_base library.

For real work it's better to create a new project (e.g. using dune init executable <some name>), have it depend on sidekick, and write your own SMT solver in this project by compiling it to native code. Here we use the toplevel because it is interactive, and because it helps me write this guide.

The basics

First, let us reiterate, as in the README that there are several libraries in this project. We're going to use these libraries:

  • sidekick: definition of algorithms, data-structures, solvers, and decision procedures for SMT solvers. This is almost entirely functorized, which means that the component are parametrized as much as possible so as to be agnostic to how precisely terms, formulas, clauses, etc. are implemented.

    One can use this library directly, but it means providing an implementation to each functor, which can be sometimes intricate.

  • sidekick-base: a fixed definition of terms, types, statements, etc. that is compatible with sidekick. This gives you a starting point to manipulate logic formulas and use SMT on them.

    The sublibrary sidekick-base.solver provides an instance of the main Solver, along with a few theories. Let us peek into it now:

# #require "sidekick-base.solver";;
# #show Sidekick_base_solver;;
module Sidekick_base_solver :
  sig
    module Solver_arg : sig ... end
    module Solver : sig ... end
    module Th_data : sig ... end
    module Th_bool : sig ... end
    module Th_lra : sig ... end
    val th_bool : Solver.theory
    val th_data : Solver.theory
    val th_lra : Solver.theory
  end

Let's bring more all these things into scope, and install some printers for legibility:

# open Sidekick_base;;
# open Sidekick_base_solver;;
# #install_printer Term.pp;;
# #install_printer Lit.pp;;
# #install_printer Ty.pp;;
# #install_printer Fun.pp;;
# #install_printer Model.pp;;
# #install_printer Solver.Model.pp;;

First steps in solving

To solve a formula, we need to first create some terms and a solver.

Manipulating terms

We're going to need a term store. All terms in sidekick live in a store, which is necessary for hashconsing (and could be used for more data-oriented representation of terms in alternative implementations.)

# let tstore = Term.create ();;
val tstore : Term.store = <abstr>

# Term.store_size tstore;;
- : int = 2

Interesting, there are already two terms that are predefined. Let's peek at them:

# let all_terms_init =
    Term.store_iter tstore |> Iter.to_list |> List.sort Term.compare;;
val all_terms_init : Term.t list = [true; false]

# Term.true_ tstore;;
- : Term.t = true

# (* check it's the same term *)
  Term.(equal (true_ tstore) (List.hd all_terms_init));;
- : bool = true

# Term.(equal (false_ tstore) (List.hd all_terms_init));;
- : bool = false

Cool. Similarly, we need to manipulate types.

Sidekick_base the type store is merely unit because we just use a global hashconsing state for types. In general we'd need to carry around a type store as well. The only predefined type is bool, the type of booleans:

# Ty.bool ();;
- : Ty.t = Bool

Now we can define new terms and constants. Let's try to define a few boolean constants named "p", "q", "r":

# let p = Term.const_undefined tstore (ID.make "p") @@ Ty.bool();;
val p : Term.t = p
# let q = Term.const_undefined tstore (ID.make "q") @@ Ty.bool();;
val q : Term.t = q
# let r = Term.const_undefined tstore (ID.make "r") @@ Ty.bool();;
val r : Term.t = r

# Term.ty p;;
- : Ty.t = Bool

# Term.equal p q;;
- : bool = false

# Term.view p;;
- : Term.t Term.view = Sidekick_base.Term.App_fun (p/3, [||])

# Term.store_iter tstore |> Iter.to_list |> List.sort Term.compare;;
- : Term.t list = [true; false; p; q; r]

We can now build formulas from these.

# let p_eq_q = Term.eq tstore p q;;
val p_eq_q : Term.t = (= p q)

# let p_imp_r = Form.imply tstore p r;;
val p_imp_r : Term.t = (=> p r)

Using a solver.

We can create a solver by passing Solver.create a term store and a type store (which in our case is simply () : unit). A list of theories can be added initially, or later using Solver.add_theory.

# let solver = Solver.create ~theories:[th_bool] ~proof:(Proof.empty) tstore () ();;
val solver : Solver.t = <abstr>

# Solver.add_theory;;
- : Solver.t -> Solver.theory -> unit = <fun>

Alright, let's do some solving now ⚙️. We're going to assert several formulas and check satisfiability in between each.

We start with p = q.

# p_eq_q;;
- : Term.t = (= p q)
# Solver.assert_term solver p_eq_q;;
- : unit = ()
# Solver.solve ~assumptions:[] solver;;
- : Solver.res =
Sidekick_base_solver.Solver.Sat
 (model
  (true := true)
  (false := false)
  (p := true)
  (q := true)
  (_tseitin_equiv_0 := true))

It is satisfiable, and we got a model where "p" and "q" are both false. We also get an internal term _tseitin_equiv_0 in the model, which is produced by the theory of boolean when it encoded the equivalence.

We can also ask Sidekick to check satisfiability under assumptions, meaning we temporarily add some hypotheses to the solver and check whether the assertions and hypotheses are satisfiable together.

# Solver.solve solver
    ~assumptions:[Solver.mk_lit_t solver p;
                  Solver.mk_lit_t solver q ~sign:false];;
- : Solver.res =
Sidekick_base_solver.Solver.Unsat
 {Sidekick_base_solver.Solver.unsat_core = <fun>; unsat_proof_step = <fun>}

Here it's unsat, because we asserted "p = q", and then assumed "p" to be true and "q" to be false. Deconstructing the result further, we could obtain an unsat core (a subset of the assumptions sufficient to explain the unsatisfiability), or a proof (to check the unsatisfiability result somehow).

Note that this doesn't affect satisfiability without assumptions:

# Solver.solve ~assumptions:[] solver;;
- : Solver.res =
Sidekick_base_solver.Solver.Sat
 (model
  (true := true)
  (false := false)
  (p := true)
  (q := true)
  (_tseitin_equiv_0 := true))

We can therefore add more formulas and see where it leads us.

# p_imp_r;;
- : Term.t = (=> p r)
# Solver.assert_term solver p_imp_r;;
- : unit = ()
# Solver.solve ~assumptions:[] solver;;
- : Solver.res =
Sidekick_base_solver.Solver.Sat
 (model
  (true := true)
  (false := false)
  (p := true)
  (q := true)
  (r := true)
  (_tseitin_equiv_0 := true)
  (_tseitin_implies_1 := true))

Still satisfiable, but now we see r in the model, too. And now:

# let q_imp_not_r = Form.imply tstore q (Form.not_ tstore r);;
val q_imp_not_r : Term.t = (=> q (not r))
# Solver.assert_term solver q_imp_not_r;;
- : unit = ()

# Solver.assert_term solver p;;
- : unit = ()

# Solver.solve ~assumptions:[] solver;;
- : Solver.res =
Sidekick_base_solver.Solver.Unsat
 {Sidekick_base_solver.Solver.unsat_core = <fun>; unsat_proof_step = <fun>}

This time we got unsat and there is no way of undoing it. It comes from the fact that p=q, but p and q imply contradictory formulas (r and ¬r), so when we force p to be true, q is true too and the contradiction is triggered.

A bit of arithmetic

We can solve linear real arithmetic problems as well. Let's create a new solver and add the theory of reals to it.

# let solver = Solver.create ~theories:[th_bool; th_lra] ~proof:(Proof.empty) tstore () ();;
val solver : Solver.t = <abstr>

Create a few arithmetic constants.

# let real = Ty.real ();;
val real : Ty.t = Real
# let a = Term.const_undefined tstore (ID.make "a") real;;
val a : Term.t = a
# let b = Term.const_undefined tstore (ID.make "b") real;;
val b : Term.t = b

# Term.ty a;;
- : Ty.t = Real

# let a_leq_b = Term.LRA.(leq tstore (var tstore a) (var tstore b));;
val a_leq_b : Term.t = (<= a b)

We can play with assertions now:

# Solver.assert_term solver a_leq_b;;
- : unit = ()
# Solver.solve ~assumptions:[] solver;;
- : Solver.res =
Sidekick_base_solver.Solver.Sat
 (model
  (true := true)
  (false := false)
  (_sk_lra__le0 := _sk_lra__le0)
  ((_sk_lra__le0 <= 0) := true))


# let a_geq_1 = Term.LRA.(geq tstore a (const tstore (Q.of_int 1)));;
val a_geq_1 : Term.t = (>= a 1)
# let b_leq_half = Term.LRA.(leq tstore b (const tstore (Q.of_string "1/2")));;
val b_leq_half : Term.t = (<= b 1/2)

# let res = Solver.solve solver
    ~assumptions:[Solver.mk_lit_t solver p;
                  Solver.mk_lit_t solver a_geq_1;
                  Solver.mk_lit_t solver b_leq_half];;
val res : Solver.res =
  Sidekick_base_solver.Solver.Unsat
   {Sidekick_base_solver.Solver.unsat_core = <fun>; unsat_proof_step = <fun>}

# match res with Solver.Unsat {unsat_core=us; _} -> us() |> Iter.to_list | _ -> assert false;;
- : Proof.lit list = [(a >= 1); (b <= 1/2)]

This just showed that a=1, b=1/2, a>=b is unsatisfiable. The junk assumption p was not used during the proof and therefore doesn't appear in the unsat core we extract from res.

Functions and congruence closure

We can define function symbols, not just constants. Let's also define u, an uninterpreted type.

# let u = Ty.atomic_uninterpreted (ID.make "u");;
val u : Ty.t = u/12

# let u1 = Term.const_undefined tstore (ID.make "u1") u;;
val u1 : Term.t = u1
# let u2 = Term.const_undefined tstore (ID.make "u2") u;;
val u2 : Term.t = u2
# let u3 = Term.const_undefined tstore (ID.make "u3") u;;
val u3 : Term.t = u3

# let f1 = Fun.mk_undef' (ID.make "f1") [u] u;;
val f1 : Fun.t = f1/16
# Fun.view f1;;
- : Fun.view =
Sidekick_base.Fun.Fun_undef
 {Sidekick_base.Base_types.fun_ty_args = [u/12]; fun_ty_ret = u/12}

# let f1_u1 = Term.app_fun_l tstore f1 [u1];;
val f1_u1 : Term.t = (f1 u1)

# Term.ty f1_u1;;
- : Ty.t = u/12
# Term.view f1_u1;;
- : Term.t Term.view = Sidekick_base.Term.App_fun (f1/16, [|u1|])

Anyway, Sidekick knows how to reason about functions.

# let solver = Solver.create ~theories:[] ~proof:(Proof.empty) tstore () ();;
val solver : Solver.t = <abstr>

# (* helper *)
  let appf1 x = Term.app_fun_l tstore f1 x;;
val appf1 : Term.t list -> Term.t = <fun>

# Solver.assert_term solver (Term.eq tstore u2 (appf1 [u1]));;
- : unit = ()
# Solver.assert_term solver (Term.eq tstore u3 (appf1 [u2]));;
- : unit = ()
# Solver.assert_term solver (Term.eq tstore u1 (appf1 [appf1 [u1]]));;
- : unit = ()
# Solver.assert_term solver
  (Term.eq tstore u1 (appf1 [appf1 [appf1 [u1]]]));;
- : unit = ()

# Solver.solve solver
    ~assumptions:[Solver.mk_lit_t solver ~sign:false (Term.eq tstore u1 (appf1[u1]))];;
- : Solver.res =
Sidekick_base_solver.Solver.Unsat
 {Sidekick_base_solver.Solver.unsat_core = <fun>; unsat_proof_step = <fun>}

# Solver.solve solver
    ~assumptions:[Solver.mk_lit_t solver ~sign:false (Term.eq tstore u2 u3)];;
- : Solver.res =
Sidekick_base_solver.Solver.Unsat
 {Sidekick_base_solver.Solver.unsat_core = <fun>; unsat_proof_step = <fun>}

Assuming: f1(u1)=u2, f1(u2)=u3, f1^2(u1)=u1, f1^3(u1)=u1, we proved that f1(u1)=u1, then that u2=u3 because both are equal to u1.

Extending sidekick

The most important function here is Solver.add_theory (or the ~theories argument to Solver.create). It means one can just implement a new theory in the same vein as the existing ones and add it to the solver. Note that the theory doesn't need to be functorized, it could directly work on the term representation you use (or Sidekick_base.Term if you reuse that).

The API for theories is found here. In addition to using terms, types, etc. it has access to a Solver_internal.t object, which contains the internal API of the SMT solver. The internal solver allows a theory to do many things, including:

  • propagate literals
  • raise conflicts (add a lemma which contradicts the current candidate model)
  • create new literals
  • hook into the central Congruence Closure
  • register preprocessing functions (typically, to transform some constructs into clauses)
  • register simplification functions

TODO: extending terms

TODO: basic custom theory (enums?)

`