NativeSolver.cs

Go to the documentation of this file.

/*++
Copyright (c) 2012 Microsoft Corporation
 
Module Name:
 
    NativeSolver.cs
 
Abstract:
 
    Z3 Managed API: Native Solver
 
Author:
 
    Christoph Wintersteiger (cwinter) 2012-03-22
    Nikolaj Bjorner (nbjorner) 2022-03-01
 
Notes:
    
--*/
 
using System;
using System.Diagnostics;
using System.Collections.Generic;
using System.Linq;
 
namespace Microsoft.Z3
{
 
    using Z3_ast = System.IntPtr;
    using Z3_context = System.IntPtr;
    using Z3_func_decl = System.IntPtr;
    using Z3_params = System.IntPtr;
    using Z3_solver = System.IntPtr;
    using Z3_sort = System.IntPtr;
    using Z3_symbol = System.IntPtr;
 
    public class NativeSolver : IDisposable
    {
        public string Help => Native.Z3_solver_get_help(nCtx, z3solver);
 
        private void SetParam(Action<Z3_params> setter)
        {
            Z3_params p = Native.Z3_mk_params(nCtx);
            Native.Z3_params_inc_ref(nCtx, p);
            setter(p);
            Native.Z3_solver_set_params(nCtx, z3solver, p);
            Native.Z3_params_dec_ref(nCtx, p);
        }
 
        public void Set(string name, bool value)
        {
            SetParam((Z3_params p) => Native.Z3_params_set_bool(nCtx, p, Native.Z3_mk_string_symbol(nCtx, name), (byte)(value ? 1 : 0)));
        }
 
        public void Set(string name, uint value)
        {
            SetParam((Z3_params p) => Native.Z3_params_set_uint(nCtx, p, Native.Z3_mk_string_symbol(nCtx, name), value));
        }
 
        public void Set(string name, double value)
        {
            SetParam((Z3_params p) => Native.Z3_params_set_double(nCtx, p, Native.Z3_mk_string_symbol(nCtx, name), value));
        }
 
        public void Set(string name, string value)
        {
            var value_sym = Native.Z3_mk_string_symbol(nCtx, value);
            SetParam((Z3_params p) => Native.Z3_params_set_symbol(nCtx, p, Native.Z3_mk_string_symbol(nCtx, name), value_sym));
        }
 
#if false
            public void Set(string name, Symbol value) { Parameters = Context.MkParams().Add(name, value); }
        public void Set(Symbol name, bool value) { Parameters = Context.MkParams().Add(name, value); }
        public void Set(Symbol name, uint value) { Parameters = Context.MkParams().Add(name, value); }
        public void Set(Symbol name, double value) { Parameters = Context.MkParams().Add(name, value); }
        public void Set(Symbol name, string value) { Parameters = Context.MkParams().Add(name, value); }
        public void Set(Symbol name, Symbol value) { Parameters = Context.MkParams().Add(name, value); }
 
        public ParamDescrs ParameterDescriptions
        {
            get { return new ParamDescrs(Context, Native.Z3_solver_get_param_descrs(nCtx, NativeObject)); }
        }
#endif
 
        public uint NumScopes => Native.Z3_solver_get_num_scopes(nCtx, z3solver);
 
        public void Push() => Native.Z3_solver_push(nCtx, z3solver);
 
        public void Pop(uint n = 1) => Native.Z3_solver_pop(nCtx, z3solver, n);
 
        public void Reset() => Native.Z3_solver_reset(nCtx, z3solver);
 
        public void Assert(params Z3_ast[] constraints)
        {
            Debug.Assert(constraints != null);
            Debug.Assert(constraints.All(c => c != IntPtr.Zero));
 
            foreach (Z3_ast a in constraints)
            {
                Native.Z3_solver_assert(nCtx, z3solver, a);
            }
        }
 
        public void Add(params Z3_ast[] constraints) => Assert(constraints);
 
        public void Add(IEnumerable<Z3_ast> constraints) => Assert(constraints.ToArray());
 
        public void AssertInjective(Z3_func_decl f)
        {
            uint arity = Native.Z3_get_arity(nCtx, f);
            Z3_sort range = Native.Z3_get_range(nCtx, f);
            Z3_ast[] vars = new Z3_ast[arity];
            Z3_sort[] sorts = new Z3_sort[arity];
            Z3_symbol[] names = new Z3_symbol[arity];
            for (uint i = 0; i < arity; ++i)
            {
                Z3_sort domain = Native.Z3_get_domain(nCtx, f, i);
                vars[i] = ntvContext.MkBound(arity - i - 1, domain);
                sorts[i] = domain;
                names[i] = Native.Z3_mk_int_symbol(nCtx, (int)i);
            }
            Z3_ast app_f = IntPtr.Zero; // Context.MkApp(f, vars);
            for (uint i = 0; i < arity; ++i)
            {
                Z3_sort domain = Native.Z3_get_domain(nCtx, f, i);
                Z3_func_decl proj = ntvContext.MkFreshFuncDecl("inv", new Z3_sort[] { range }, domain);
                Z3_ast body = ntvContext.MkEq(vars[i], ntvContext.MkApp(proj, app_f));
                Z3_ast q = ntvContext.MkForall(names, sorts, body);
                Assert(q);
            }
        }
 
        public void AssertAndTrack(Z3_ast[] constraints, Z3_ast[] ps)
        {
            Debug.Assert(constraints != null);
            Debug.Assert(constraints.All(c => c != IntPtr.Zero));
            Debug.Assert(ps.All(c => c != IntPtr.Zero));
            if (constraints.Length != ps.Length)
                throw new Z3Exception("Argument size mismatch");
 
            for (int i = 0; i < constraints.Length; i++)
                Native.Z3_solver_assert_and_track(nCtx, z3solver, constraints[i], ps[i]);
        }
 
        public void AssertAndTrack(Z3_ast constraint, Z3_ast p)
        {
            Debug.Assert(constraint != null);
            Debug.Assert(p != null);
 
            Native.Z3_solver_assert_and_track(nCtx, z3solver, constraint, p);
        }
 
        public void FromFile(string file)
            => Native.Z3_solver_from_file(nCtx, z3solver, file);
 
        public void FromString(string str)
            => Native.Z3_solver_from_string(nCtx, z3solver, str);
 
        public uint NumAssertions
            => (uint)ntvContext.ToArray(Native.Z3_solver_get_assertions(nCtx, z3solver)).Length;
 
        public Z3_ast[] Assertions
            => ntvContext.ToArray(Native.Z3_solver_get_assertions(nCtx, z3solver));
 
        public Z3_ast[] Units
            => ntvContext.ToArray(Native.Z3_solver_get_units(nCtx, z3solver));
 
        public Z3_ast[] NonUnits
            => ntvContext.ToArray(Native.Z3_solver_get_non_units(nCtx, z3solver));
 
        public Z3_ast[] Trail
            => ntvContext.ToArray(Native.Z3_solver_get_trail(nCtx, z3solver));
 
        public Z3_ast CongruenceRoot(Z3_ast t)
        {
            Debug.Assert(t != IntPtr.Zero);
            return Native.Z3_solver_congruence_root(nCtx, z3solver, t);
        }
 
        public Z3_ast CongruenceNext(Z3_ast t)
        {
            Debug.Assert(t != IntPtr.Zero);
            return Native.Z3_solver_congruence_next(nCtx, z3solver, t);
        }
 
        public Z3_ast CongruenceExplain(Z3_ast a, Z3_ast b)
        {
            Debug.Assert(a != IntPtr.Zero);
            Debug.Assert(b != IntPtr.Zero);
            return Native.Z3_solver_congruence_explain(nCtx, z3solver, a, b);
        }
 
        public Status Check(params Z3_ast[] assumptions)
        {
            Z3_lbool r;
            if (assumptions == null || assumptions.Length == 0)
                r = (Z3_lbool)Native.Z3_solver_check(nCtx, z3solver);
            else
                r = (Z3_lbool)Native.Z3_solver_check_assumptions(nCtx, z3solver, (uint)assumptions.Length, assumptions);
            return lboolToStatus(r);
        }
 
        public Status Check(IEnumerable<Z3_ast> assumptions)
        {
            Z3_lbool r;
            Z3_ast[] asms = assumptions.ToArray();
            if (asms.Length == 0)
                r = (Z3_lbool)Native.Z3_solver_check(nCtx, z3solver);
            else
                r = (Z3_lbool)Native.Z3_solver_check_assumptions(nCtx, z3solver, (uint)asms.Length, asms);
            return lboolToStatus(r);
        }
 
        public NativeModel Model
        {
            get
            {
                IntPtr x = Native.Z3_solver_get_model(nCtx, z3solver);
                return x == IntPtr.Zero
                    ? null
                    : new NativeModel(ntvContext, x);
            }
        }
 
        public Z3_ast Proof
            => Native.Z3_solver_get_proof(nCtx, z3solver);
 
        public Z3_ast[] UnsatCore
            => ntvContext.ToArray(Native.Z3_solver_get_unsat_core(nCtx, z3solver));
 
        public string ReasonUnknown
            => Native.Z3_solver_get_reason_unknown(nCtx, z3solver);
 
        public NativeSolver Translate(NativeContext ctx)
        {
            Debug.Assert(ctx != null);
            return new NativeSolver(ctx, Native.Z3_solver_translate(nCtx, z3solver, ctx.nCtx));
        }
 
        public void ImportModelConverter(NativeSolver src)
        {
            Debug.Assert(src != null);
 
            Native.Z3_solver_import_model_converter(nCtx, src.z3solver, z3solver);
        }
 
        public Statistics.Entry[] Statistics
        {
            get
            {
                var stats = Native.Z3_solver_get_statistics(nCtx, z3solver);
                return ntvContext.GetStatistics(stats);
            }
        }
 
        public override string ToString()
        {
            return Native.Z3_solver_to_string(nCtx, z3solver);
        }
 
        #region Internal
        readonly NativeContext ntvContext;
        Z3_solver z3solver;
        Z3_context nCtx => ntvContext.nCtx;
 
        internal NativeSolver(NativeContext nativeCtx, Z3_solver z3solver)
        {
            Debug.Assert(nativeCtx != null);
            Debug.Assert(z3solver != IntPtr.Zero);
 
            this.ntvContext = nativeCtx;
            this.z3solver = z3solver;
 
            Native.Z3_solver_inc_ref(nCtx, z3solver);
        }
 
        ~NativeSolver()
        {
            Dispose();
        }
 
        public void Dispose()
        {
            if (z3solver != IntPtr.Zero)
            {
                Native.Z3_solver_dec_ref(nCtx, z3solver);
                z3solver = IntPtr.Zero;
            }
            GC.SuppressFinalize(this);
        }
 
 
        private Status lboolToStatus(Z3_lbool r)
        {
            switch (r)
            {
                case Z3_lbool.Z3_L_TRUE: return Status.SATISFIABLE;
                case Z3_lbool.Z3_L_FALSE: return Status.UNSATISFIABLE;
                default: return Status.UNKNOWN;
            }
        }
 
        #endregion
    }
}