mirror of
https://github.com/Z3Prover/z3
synced 2026-02-22 16:27:37 +00:00
Fix formatting of z3.go using gofmt
Co-authored-by: NikolajBjorner <3085284+NikolajBjorner@users.noreply.github.com>
This commit is contained in:
parent
d43c556a80
commit
d56ae1a669
1 changed files with 320 additions and 320 deletions
640
src/api/go/z3.go
640
src/api/go/z3.go
|
|
@ -54,69 +54,69 @@ package z3
|
||||||
*/
|
*/
|
||||||
import "C"
|
import "C"
|
||||||
import (
|
import (
|
||||||
"runtime"
|
"runtime"
|
||||||
"unsafe"
|
"unsafe"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Config represents a Z3 configuration object.
|
// Config represents a Z3 configuration object.
|
||||||
type Config struct {
|
type Config struct {
|
||||||
ptr C.Z3_config
|
ptr C.Z3_config
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewConfig creates a new Z3 configuration.
|
// NewConfig creates a new Z3 configuration.
|
||||||
func NewConfig() *Config {
|
func NewConfig() *Config {
|
||||||
cfg := &Config{ptr: C.Z3_mk_config()}
|
cfg := &Config{ptr: C.Z3_mk_config()}
|
||||||
runtime.SetFinalizer(cfg, func(c *Config) {
|
runtime.SetFinalizer(cfg, func(c *Config) {
|
||||||
C.Z3_del_config(c.ptr)
|
C.Z3_del_config(c.ptr)
|
||||||
})
|
})
|
||||||
return cfg
|
return cfg
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetParamValue sets a configuration parameter.
|
// SetParamValue sets a configuration parameter.
|
||||||
func (c *Config) SetParamValue(paramID, paramValue string) {
|
func (c *Config) SetParamValue(paramID, paramValue string) {
|
||||||
cParamID := C.CString(paramID)
|
cParamID := C.CString(paramID)
|
||||||
cParamValue := C.CString(paramValue)
|
cParamValue := C.CString(paramValue)
|
||||||
defer C.free(unsafe.Pointer(cParamID))
|
defer C.free(unsafe.Pointer(cParamID))
|
||||||
defer C.free(unsafe.Pointer(cParamValue))
|
defer C.free(unsafe.Pointer(cParamValue))
|
||||||
C.Z3_set_param_value(c.ptr, cParamID, cParamValue)
|
C.Z3_set_param_value(c.ptr, cParamID, cParamValue)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Context represents a Z3 logical context.
|
// Context represents a Z3 logical context.
|
||||||
type Context struct {
|
type Context struct {
|
||||||
ptr C.Z3_context
|
ptr C.Z3_context
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewContext creates a new Z3 context with default configuration.
|
// NewContext creates a new Z3 context with default configuration.
|
||||||
func NewContext() *Context {
|
func NewContext() *Context {
|
||||||
ctx := &Context{ptr: C.Z3_mk_context_rc(C.Z3_mk_config())}
|
ctx := &Context{ptr: C.Z3_mk_context_rc(C.Z3_mk_config())}
|
||||||
runtime.SetFinalizer(ctx, func(c *Context) {
|
runtime.SetFinalizer(ctx, func(c *Context) {
|
||||||
C.Z3_del_context(c.ptr)
|
C.Z3_del_context(c.ptr)
|
||||||
})
|
})
|
||||||
return ctx
|
return ctx
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewContextWithConfig creates a new Z3 context with the given configuration.
|
// NewContextWithConfig creates a new Z3 context with the given configuration.
|
||||||
func NewContextWithConfig(cfg *Config) *Context {
|
func NewContextWithConfig(cfg *Config) *Context {
|
||||||
ctx := &Context{ptr: C.Z3_mk_context_rc(cfg.ptr)}
|
ctx := &Context{ptr: C.Z3_mk_context_rc(cfg.ptr)}
|
||||||
runtime.SetFinalizer(ctx, func(c *Context) {
|
runtime.SetFinalizer(ctx, func(c *Context) {
|
||||||
C.Z3_del_context(c.ptr)
|
C.Z3_del_context(c.ptr)
|
||||||
})
|
})
|
||||||
return ctx
|
return ctx
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetParam sets a global or context parameter.
|
// SetParam sets a global or context parameter.
|
||||||
func (c *Context) SetParam(key, value string) {
|
func (c *Context) SetParam(key, value string) {
|
||||||
cKey := C.CString(key)
|
cKey := C.CString(key)
|
||||||
cValue := C.CString(value)
|
cValue := C.CString(value)
|
||||||
defer C.free(unsafe.Pointer(cKey))
|
defer C.free(unsafe.Pointer(cKey))
|
||||||
defer C.free(unsafe.Pointer(cValue))
|
defer C.free(unsafe.Pointer(cValue))
|
||||||
C.Z3_update_param_value(c.ptr, cKey, cValue)
|
C.Z3_update_param_value(c.ptr, cKey, cValue)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Symbol represents a Z3 symbol.
|
// Symbol represents a Z3 symbol.
|
||||||
type Symbol struct {
|
type Symbol struct {
|
||||||
ctx *Context
|
ctx *Context
|
||||||
ptr C.Z3_symbol
|
ptr C.Z3_symbol
|
||||||
}
|
}
|
||||||
|
|
||||||
// newSymbol creates a new Symbol.
|
// newSymbol creates a new Symbol.
|
||||||
|
|
@ -126,136 +126,136 @@ func newSymbol(ctx *Context, ptr C.Z3_symbol) *Symbol {
|
||||||
|
|
||||||
// MkIntSymbol creates an integer symbol.
|
// MkIntSymbol creates an integer symbol.
|
||||||
func (c *Context) MkIntSymbol(i int) *Symbol {
|
func (c *Context) MkIntSymbol(i int) *Symbol {
|
||||||
return &Symbol{
|
return &Symbol{
|
||||||
ctx: c,
|
ctx: c,
|
||||||
ptr: C.Z3_mk_int_symbol(c.ptr, C.int(i)),
|
ptr: C.Z3_mk_int_symbol(c.ptr, C.int(i)),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkStringSymbol creates a string symbol.
|
// MkStringSymbol creates a string symbol.
|
||||||
func (c *Context) MkStringSymbol(s string) *Symbol {
|
func (c *Context) MkStringSymbol(s string) *Symbol {
|
||||||
cStr := C.CString(s)
|
cStr := C.CString(s)
|
||||||
defer C.free(unsafe.Pointer(cStr))
|
defer C.free(unsafe.Pointer(cStr))
|
||||||
return &Symbol{
|
return &Symbol{
|
||||||
ctx: c,
|
ctx: c,
|
||||||
ptr: C.Z3_mk_string_symbol(c.ptr, cStr),
|
ptr: C.Z3_mk_string_symbol(c.ptr, cStr),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// String returns the string representation of the symbol.
|
// String returns the string representation of the symbol.
|
||||||
func (s *Symbol) String() string {
|
func (s *Symbol) String() string {
|
||||||
kind := C.Z3_get_symbol_kind(s.ctx.ptr, s.ptr)
|
kind := C.Z3_get_symbol_kind(s.ctx.ptr, s.ptr)
|
||||||
if kind == C.Z3_INT_SYMBOL {
|
if kind == C.Z3_INT_SYMBOL {
|
||||||
return string(rune(C.Z3_get_symbol_int(s.ctx.ptr, s.ptr)))
|
return string(rune(C.Z3_get_symbol_int(s.ctx.ptr, s.ptr)))
|
||||||
}
|
}
|
||||||
return C.GoString(C.Z3_get_symbol_string(s.ctx.ptr, s.ptr))
|
return C.GoString(C.Z3_get_symbol_string(s.ctx.ptr, s.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// AST represents a Z3 abstract syntax tree node.
|
// AST represents a Z3 abstract syntax tree node.
|
||||||
type AST struct {
|
type AST struct {
|
||||||
ctx *Context
|
ctx *Context
|
||||||
ptr C.Z3_ast
|
ptr C.Z3_ast
|
||||||
}
|
}
|
||||||
|
|
||||||
// incRef increments the reference count of the AST.
|
// incRef increments the reference count of the AST.
|
||||||
func (a *AST) incRef() {
|
func (a *AST) incRef() {
|
||||||
C.Z3_inc_ref(a.ctx.ptr, a.ptr)
|
C.Z3_inc_ref(a.ctx.ptr, a.ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// decRef decrements the reference count of the AST.
|
// decRef decrements the reference count of the AST.
|
||||||
func (a *AST) decRef() {
|
func (a *AST) decRef() {
|
||||||
C.Z3_dec_ref(a.ctx.ptr, a.ptr)
|
C.Z3_dec_ref(a.ctx.ptr, a.ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// String returns the string representation of the AST.
|
// String returns the string representation of the AST.
|
||||||
func (a *AST) String() string {
|
func (a *AST) String() string {
|
||||||
return C.GoString(C.Z3_ast_to_string(a.ctx.ptr, a.ptr))
|
return C.GoString(C.Z3_ast_to_string(a.ctx.ptr, a.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Hash returns the hash code of the AST.
|
// Hash returns the hash code of the AST.
|
||||||
func (a *AST) Hash() uint32 {
|
func (a *AST) Hash() uint32 {
|
||||||
return uint32(C.Z3_get_ast_hash(a.ctx.ptr, a.ptr))
|
return uint32(C.Z3_get_ast_hash(a.ctx.ptr, a.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Equal checks if two ASTs are equal.
|
// Equal checks if two ASTs are equal.
|
||||||
func (a *AST) Equal(other *AST) bool {
|
func (a *AST) Equal(other *AST) bool {
|
||||||
if a.ctx != other.ctx {
|
if a.ctx != other.ctx {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
return bool(C.Z3_is_eq_ast(a.ctx.ptr, a.ptr, other.ptr))
|
return bool(C.Z3_is_eq_ast(a.ctx.ptr, a.ptr, other.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Sort represents a Z3 sort (type).
|
// Sort represents a Z3 sort (type).
|
||||||
type Sort struct {
|
type Sort struct {
|
||||||
ctx *Context
|
ctx *Context
|
||||||
ptr C.Z3_sort
|
ptr C.Z3_sort
|
||||||
}
|
}
|
||||||
|
|
||||||
// newSort creates a new Sort and manages its reference count.
|
// newSort creates a new Sort and manages its reference count.
|
||||||
func newSort(ctx *Context, ptr C.Z3_sort) *Sort {
|
func newSort(ctx *Context, ptr C.Z3_sort) *Sort {
|
||||||
sort := &Sort{ctx: ctx, ptr: ptr}
|
sort := &Sort{ctx: ctx, ptr: ptr}
|
||||||
C.Z3_inc_ref(ctx.ptr, C.Z3_sort_to_ast(ctx.ptr, ptr))
|
C.Z3_inc_ref(ctx.ptr, C.Z3_sort_to_ast(ctx.ptr, ptr))
|
||||||
runtime.SetFinalizer(sort, func(s *Sort) {
|
runtime.SetFinalizer(sort, func(s *Sort) {
|
||||||
C.Z3_dec_ref(s.ctx.ptr, C.Z3_sort_to_ast(s.ctx.ptr, s.ptr))
|
C.Z3_dec_ref(s.ctx.ptr, C.Z3_sort_to_ast(s.ctx.ptr, s.ptr))
|
||||||
})
|
})
|
||||||
return sort
|
return sort
|
||||||
}
|
}
|
||||||
|
|
||||||
// String returns the string representation of the sort.
|
// String returns the string representation of the sort.
|
||||||
func (s *Sort) String() string {
|
func (s *Sort) String() string {
|
||||||
return C.GoString(C.Z3_sort_to_string(s.ctx.ptr, s.ptr))
|
return C.GoString(C.Z3_sort_to_string(s.ctx.ptr, s.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Equal checks if two sorts are equal.
|
// Equal checks if two sorts are equal.
|
||||||
func (s *Sort) Equal(other *Sort) bool {
|
func (s *Sort) Equal(other *Sort) bool {
|
||||||
if s.ctx != other.ctx {
|
if s.ctx != other.ctx {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
return bool(C.Z3_is_eq_sort(s.ctx.ptr, s.ptr, other.ptr))
|
return bool(C.Z3_is_eq_sort(s.ctx.ptr, s.ptr, other.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkBoolSort creates the Boolean sort.
|
// MkBoolSort creates the Boolean sort.
|
||||||
func (c *Context) MkBoolSort() *Sort {
|
func (c *Context) MkBoolSort() *Sort {
|
||||||
return newSort(c, C.Z3_mk_bool_sort(c.ptr))
|
return newSort(c, C.Z3_mk_bool_sort(c.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkBvSort creates a bit-vector sort of the given size.
|
// MkBvSort creates a bit-vector sort of the given size.
|
||||||
func (c *Context) MkBvSort(sz uint) *Sort {
|
func (c *Context) MkBvSort(sz uint) *Sort {
|
||||||
return newSort(c, C.Z3_mk_bv_sort(c.ptr, C.uint(sz)))
|
return newSort(c, C.Z3_mk_bv_sort(c.ptr, C.uint(sz)))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Expr represents a Z3 expression.
|
// Expr represents a Z3 expression.
|
||||||
type Expr struct {
|
type Expr struct {
|
||||||
ctx *Context
|
ctx *Context
|
||||||
ptr C.Z3_ast
|
ptr C.Z3_ast
|
||||||
}
|
}
|
||||||
|
|
||||||
// newExpr creates a new Expr and manages its reference count.
|
// newExpr creates a new Expr and manages its reference count.
|
||||||
func newExpr(ctx *Context, ptr C.Z3_ast) *Expr {
|
func newExpr(ctx *Context, ptr C.Z3_ast) *Expr {
|
||||||
expr := &Expr{ctx: ctx, ptr: ptr}
|
expr := &Expr{ctx: ctx, ptr: ptr}
|
||||||
C.Z3_inc_ref(ctx.ptr, ptr)
|
C.Z3_inc_ref(ctx.ptr, ptr)
|
||||||
runtime.SetFinalizer(expr, func(e *Expr) {
|
runtime.SetFinalizer(expr, func(e *Expr) {
|
||||||
C.Z3_dec_ref(e.ctx.ptr, e.ptr)
|
C.Z3_dec_ref(e.ctx.ptr, e.ptr)
|
||||||
})
|
})
|
||||||
return expr
|
return expr
|
||||||
}
|
}
|
||||||
|
|
||||||
// String returns the string representation of the expression.
|
// String returns the string representation of the expression.
|
||||||
func (e *Expr) String() string {
|
func (e *Expr) String() string {
|
||||||
return C.GoString(C.Z3_ast_to_string(e.ctx.ptr, e.ptr))
|
return C.GoString(C.Z3_ast_to_string(e.ctx.ptr, e.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Equal checks if two expressions are equal.
|
// Equal checks if two expressions are equal.
|
||||||
func (e *Expr) Equal(other *Expr) bool {
|
func (e *Expr) Equal(other *Expr) bool {
|
||||||
if e.ctx != other.ctx {
|
if e.ctx != other.ctx {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
return bool(C.Z3_is_eq_ast(e.ctx.ptr, e.ptr, other.ptr))
|
return bool(C.Z3_is_eq_ast(e.ctx.ptr, e.ptr, other.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetSort returns the sort of the expression.
|
// GetSort returns the sort of the expression.
|
||||||
func (e *Expr) GetSort() *Sort {
|
func (e *Expr) GetSort() *Sort {
|
||||||
return newSort(e.ctx, C.Z3_get_sort(e.ctx.ptr, e.ptr))
|
return newSort(e.ctx, C.Z3_get_sort(e.ctx.ptr, e.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Pattern represents a Z3 pattern for quantifier instantiation.
|
// Pattern represents a Z3 pattern for quantifier instantiation.
|
||||||
|
|
@ -309,179 +309,179 @@ func newParamDescrs(ctx *Context, ptr C.Z3_param_descrs) *ParamDescrs {
|
||||||
|
|
||||||
// MkTrue creates the Boolean constant true.
|
// MkTrue creates the Boolean constant true.
|
||||||
func (c *Context) MkTrue() *Expr {
|
func (c *Context) MkTrue() *Expr {
|
||||||
return newExpr(c, C.Z3_mk_true(c.ptr))
|
return newExpr(c, C.Z3_mk_true(c.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkFalse creates the Boolean constant false.
|
// MkFalse creates the Boolean constant false.
|
||||||
func (c *Context) MkFalse() *Expr {
|
func (c *Context) MkFalse() *Expr {
|
||||||
return newExpr(c, C.Z3_mk_false(c.ptr))
|
return newExpr(c, C.Z3_mk_false(c.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkBool creates a Boolean constant.
|
// MkBool creates a Boolean constant.
|
||||||
func (c *Context) MkBool(value bool) *Expr {
|
func (c *Context) MkBool(value bool) *Expr {
|
||||||
if value {
|
if value {
|
||||||
return c.MkTrue()
|
return c.MkTrue()
|
||||||
}
|
}
|
||||||
return c.MkFalse()
|
return c.MkFalse()
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkNumeral creates a numeral from a string.
|
// MkNumeral creates a numeral from a string.
|
||||||
func (c *Context) MkNumeral(numeral string, sort *Sort) *Expr {
|
func (c *Context) MkNumeral(numeral string, sort *Sort) *Expr {
|
||||||
cStr := C.CString(numeral)
|
cStr := C.CString(numeral)
|
||||||
defer C.free(unsafe.Pointer(cStr))
|
defer C.free(unsafe.Pointer(cStr))
|
||||||
return newExpr(c, C.Z3_mk_numeral(c.ptr, cStr, sort.ptr))
|
return newExpr(c, C.Z3_mk_numeral(c.ptr, cStr, sort.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkConst creates a constant (variable) with the given name and sort.
|
// MkConst creates a constant (variable) with the given name and sort.
|
||||||
func (c *Context) MkConst(name *Symbol, sort *Sort) *Expr {
|
func (c *Context) MkConst(name *Symbol, sort *Sort) *Expr {
|
||||||
return newExpr(c, C.Z3_mk_const(c.ptr, name.ptr, sort.ptr))
|
return newExpr(c, C.Z3_mk_const(c.ptr, name.ptr, sort.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkBoolConst creates a Boolean constant (variable) with the given name.
|
// MkBoolConst creates a Boolean constant (variable) with the given name.
|
||||||
func (c *Context) MkBoolConst(name string) *Expr {
|
func (c *Context) MkBoolConst(name string) *Expr {
|
||||||
sym := c.MkStringSymbol(name)
|
sym := c.MkStringSymbol(name)
|
||||||
return c.MkConst(sym, c.MkBoolSort())
|
return c.MkConst(sym, c.MkBoolSort())
|
||||||
}
|
}
|
||||||
|
|
||||||
// Boolean operations
|
// Boolean operations
|
||||||
|
|
||||||
// MkAnd creates a conjunction.
|
// MkAnd creates a conjunction.
|
||||||
func (c *Context) MkAnd(exprs ...*Expr) *Expr {
|
func (c *Context) MkAnd(exprs ...*Expr) *Expr {
|
||||||
if len(exprs) == 0 {
|
if len(exprs) == 0 {
|
||||||
return c.MkTrue()
|
return c.MkTrue()
|
||||||
}
|
}
|
||||||
if len(exprs) == 1 {
|
if len(exprs) == 1 {
|
||||||
return exprs[0]
|
return exprs[0]
|
||||||
}
|
}
|
||||||
cExprs := make([]C.Z3_ast, len(exprs))
|
cExprs := make([]C.Z3_ast, len(exprs))
|
||||||
for i, e := range exprs {
|
for i, e := range exprs {
|
||||||
cExprs[i] = e.ptr
|
cExprs[i] = e.ptr
|
||||||
}
|
}
|
||||||
return newExpr(c, C.Z3_mk_and(c.ptr, C.uint(len(exprs)), &cExprs[0]))
|
return newExpr(c, C.Z3_mk_and(c.ptr, C.uint(len(exprs)), &cExprs[0]))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkOr creates a disjunction.
|
// MkOr creates a disjunction.
|
||||||
func (c *Context) MkOr(exprs ...*Expr) *Expr {
|
func (c *Context) MkOr(exprs ...*Expr) *Expr {
|
||||||
if len(exprs) == 0 {
|
if len(exprs) == 0 {
|
||||||
return c.MkFalse()
|
return c.MkFalse()
|
||||||
}
|
}
|
||||||
if len(exprs) == 1 {
|
if len(exprs) == 1 {
|
||||||
return exprs[0]
|
return exprs[0]
|
||||||
}
|
}
|
||||||
cExprs := make([]C.Z3_ast, len(exprs))
|
cExprs := make([]C.Z3_ast, len(exprs))
|
||||||
for i, e := range exprs {
|
for i, e := range exprs {
|
||||||
cExprs[i] = e.ptr
|
cExprs[i] = e.ptr
|
||||||
}
|
}
|
||||||
return newExpr(c, C.Z3_mk_or(c.ptr, C.uint(len(exprs)), &cExprs[0]))
|
return newExpr(c, C.Z3_mk_or(c.ptr, C.uint(len(exprs)), &cExprs[0]))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkNot creates a negation.
|
// MkNot creates a negation.
|
||||||
func (c *Context) MkNot(expr *Expr) *Expr {
|
func (c *Context) MkNot(expr *Expr) *Expr {
|
||||||
return newExpr(c, C.Z3_mk_not(c.ptr, expr.ptr))
|
return newExpr(c, C.Z3_mk_not(c.ptr, expr.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkImplies creates an implication.
|
// MkImplies creates an implication.
|
||||||
func (c *Context) MkImplies(lhs, rhs *Expr) *Expr {
|
func (c *Context) MkImplies(lhs, rhs *Expr) *Expr {
|
||||||
return newExpr(c, C.Z3_mk_implies(c.ptr, lhs.ptr, rhs.ptr))
|
return newExpr(c, C.Z3_mk_implies(c.ptr, lhs.ptr, rhs.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkIff creates a bi-implication (if and only if).
|
// MkIff creates a bi-implication (if and only if).
|
||||||
func (c *Context) MkIff(lhs, rhs *Expr) *Expr {
|
func (c *Context) MkIff(lhs, rhs *Expr) *Expr {
|
||||||
return newExpr(c, C.Z3_mk_iff(c.ptr, lhs.ptr, rhs.ptr))
|
return newExpr(c, C.Z3_mk_iff(c.ptr, lhs.ptr, rhs.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkXor creates exclusive or.
|
// MkXor creates exclusive or.
|
||||||
func (c *Context) MkXor(lhs, rhs *Expr) *Expr {
|
func (c *Context) MkXor(lhs, rhs *Expr) *Expr {
|
||||||
return newExpr(c, C.Z3_mk_xor(c.ptr, lhs.ptr, rhs.ptr))
|
return newExpr(c, C.Z3_mk_xor(c.ptr, lhs.ptr, rhs.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Comparison operations
|
// Comparison operations
|
||||||
|
|
||||||
// MkEq creates an equality.
|
// MkEq creates an equality.
|
||||||
func (c *Context) MkEq(lhs, rhs *Expr) *Expr {
|
func (c *Context) MkEq(lhs, rhs *Expr) *Expr {
|
||||||
return newExpr(c, C.Z3_mk_eq(c.ptr, lhs.ptr, rhs.ptr))
|
return newExpr(c, C.Z3_mk_eq(c.ptr, lhs.ptr, rhs.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkDistinct creates a distinct constraint.
|
// MkDistinct creates a distinct constraint.
|
||||||
func (c *Context) MkDistinct(exprs ...*Expr) *Expr {
|
func (c *Context) MkDistinct(exprs ...*Expr) *Expr {
|
||||||
if len(exprs) <= 1 {
|
if len(exprs) <= 1 {
|
||||||
return c.MkTrue()
|
return c.MkTrue()
|
||||||
}
|
}
|
||||||
cExprs := make([]C.Z3_ast, len(exprs))
|
cExprs := make([]C.Z3_ast, len(exprs))
|
||||||
for i, e := range exprs {
|
for i, e := range exprs {
|
||||||
cExprs[i] = e.ptr
|
cExprs[i] = e.ptr
|
||||||
}
|
}
|
||||||
return newExpr(c, C.Z3_mk_distinct(c.ptr, C.uint(len(exprs)), &cExprs[0]))
|
return newExpr(c, C.Z3_mk_distinct(c.ptr, C.uint(len(exprs)), &cExprs[0]))
|
||||||
}
|
}
|
||||||
|
|
||||||
// FuncDecl represents a function declaration.
|
// FuncDecl represents a function declaration.
|
||||||
type FuncDecl struct {
|
type FuncDecl struct {
|
||||||
ctx *Context
|
ctx *Context
|
||||||
ptr C.Z3_func_decl
|
ptr C.Z3_func_decl
|
||||||
}
|
}
|
||||||
|
|
||||||
// newFuncDecl creates a new FuncDecl and manages its reference count.
|
// newFuncDecl creates a new FuncDecl and manages its reference count.
|
||||||
func newFuncDecl(ctx *Context, ptr C.Z3_func_decl) *FuncDecl {
|
func newFuncDecl(ctx *Context, ptr C.Z3_func_decl) *FuncDecl {
|
||||||
fd := &FuncDecl{ctx: ctx, ptr: ptr}
|
fd := &FuncDecl{ctx: ctx, ptr: ptr}
|
||||||
C.Z3_inc_ref(ctx.ptr, C.Z3_func_decl_to_ast(ctx.ptr, ptr))
|
C.Z3_inc_ref(ctx.ptr, C.Z3_func_decl_to_ast(ctx.ptr, ptr))
|
||||||
runtime.SetFinalizer(fd, func(f *FuncDecl) {
|
runtime.SetFinalizer(fd, func(f *FuncDecl) {
|
||||||
C.Z3_dec_ref(f.ctx.ptr, C.Z3_func_decl_to_ast(f.ctx.ptr, f.ptr))
|
C.Z3_dec_ref(f.ctx.ptr, C.Z3_func_decl_to_ast(f.ctx.ptr, f.ptr))
|
||||||
})
|
})
|
||||||
return fd
|
return fd
|
||||||
}
|
}
|
||||||
|
|
||||||
// String returns the string representation of the function declaration.
|
// String returns the string representation of the function declaration.
|
||||||
func (f *FuncDecl) String() string {
|
func (f *FuncDecl) String() string {
|
||||||
return C.GoString(C.Z3_func_decl_to_string(f.ctx.ptr, f.ptr))
|
return C.GoString(C.Z3_func_decl_to_string(f.ctx.ptr, f.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetName returns the name of the function declaration.
|
// GetName returns the name of the function declaration.
|
||||||
func (f *FuncDecl) GetName() *Symbol {
|
func (f *FuncDecl) GetName() *Symbol {
|
||||||
return &Symbol{
|
return &Symbol{
|
||||||
ctx: f.ctx,
|
ctx: f.ctx,
|
||||||
ptr: C.Z3_get_decl_name(f.ctx.ptr, f.ptr),
|
ptr: C.Z3_get_decl_name(f.ctx.ptr, f.ptr),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetArity returns the arity (number of parameters) of the function.
|
// GetArity returns the arity (number of parameters) of the function.
|
||||||
func (f *FuncDecl) GetArity() int {
|
func (f *FuncDecl) GetArity() int {
|
||||||
return int(C.Z3_get_arity(f.ctx.ptr, f.ptr))
|
return int(C.Z3_get_arity(f.ctx.ptr, f.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetDomain returns the sort of the i-th parameter.
|
// GetDomain returns the sort of the i-th parameter.
|
||||||
func (f *FuncDecl) GetDomain(i int) *Sort {
|
func (f *FuncDecl) GetDomain(i int) *Sort {
|
||||||
return newSort(f.ctx, C.Z3_get_domain(f.ctx.ptr, f.ptr, C.uint(i)))
|
return newSort(f.ctx, C.Z3_get_domain(f.ctx.ptr, f.ptr, C.uint(i)))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetRange returns the sort of the return value.
|
// GetRange returns the sort of the return value.
|
||||||
func (f *FuncDecl) GetRange() *Sort {
|
func (f *FuncDecl) GetRange() *Sort {
|
||||||
return newSort(f.ctx, C.Z3_get_range(f.ctx.ptr, f.ptr))
|
return newSort(f.ctx, C.Z3_get_range(f.ctx.ptr, f.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkFuncDecl creates a function declaration.
|
// MkFuncDecl creates a function declaration.
|
||||||
func (c *Context) MkFuncDecl(name *Symbol, domain []*Sort, range_ *Sort) *FuncDecl {
|
func (c *Context) MkFuncDecl(name *Symbol, domain []*Sort, range_ *Sort) *FuncDecl {
|
||||||
cDomain := make([]C.Z3_sort, len(domain))
|
cDomain := make([]C.Z3_sort, len(domain))
|
||||||
for i, s := range domain {
|
for i, s := range domain {
|
||||||
cDomain[i] = s.ptr
|
cDomain[i] = s.ptr
|
||||||
}
|
}
|
||||||
var domainPtr *C.Z3_sort
|
var domainPtr *C.Z3_sort
|
||||||
if len(domain) > 0 {
|
if len(domain) > 0 {
|
||||||
domainPtr = &cDomain[0]
|
domainPtr = &cDomain[0]
|
||||||
}
|
}
|
||||||
return newFuncDecl(c, C.Z3_mk_func_decl(c.ptr, name.ptr, C.uint(len(domain)), domainPtr, range_.ptr))
|
return newFuncDecl(c, C.Z3_mk_func_decl(c.ptr, name.ptr, C.uint(len(domain)), domainPtr, range_.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkApp creates a function application.
|
// MkApp creates a function application.
|
||||||
func (c *Context) MkApp(decl *FuncDecl, args ...*Expr) *Expr {
|
func (c *Context) MkApp(decl *FuncDecl, args ...*Expr) *Expr {
|
||||||
cArgs := make([]C.Z3_ast, len(args))
|
cArgs := make([]C.Z3_ast, len(args))
|
||||||
for i, a := range args {
|
for i, a := range args {
|
||||||
cArgs[i] = a.ptr
|
cArgs[i] = a.ptr
|
||||||
}
|
}
|
||||||
var argsPtr *C.Z3_ast
|
var argsPtr *C.Z3_ast
|
||||||
if len(args) > 0 {
|
if len(args) > 0 {
|
||||||
argsPtr = &cArgs[0]
|
argsPtr = &cArgs[0]
|
||||||
}
|
}
|
||||||
return newExpr(c, C.Z3_mk_app(c.ptr, decl.ptr, C.uint(len(args)), argsPtr))
|
return newExpr(c, C.Z3_mk_app(c.ptr, decl.ptr, C.uint(len(args)), argsPtr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Quantifier operations
|
// Quantifier operations
|
||||||
|
|
@ -516,276 +516,276 @@ func (c *Context) MkExists(bound []*Expr, body *Expr) *Expr {
|
||||||
|
|
||||||
// Simplify simplifies an expression.
|
// Simplify simplifies an expression.
|
||||||
func (e *Expr) Simplify() *Expr {
|
func (e *Expr) Simplify() *Expr {
|
||||||
return newExpr(e.ctx, C.Z3_simplify(e.ctx.ptr, e.ptr))
|
return newExpr(e.ctx, C.Z3_simplify(e.ctx.ptr, e.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkTypeVariable creates a type variable sort for use in polymorphic functions and datatypes
|
// MkTypeVariable creates a type variable sort for use in polymorphic functions and datatypes
|
||||||
func (c *Context) MkTypeVariable(name *Symbol) *Sort {
|
func (c *Context) MkTypeVariable(name *Symbol) *Sort {
|
||||||
return newSort(c, C.Z3_mk_type_variable(c.ptr, name.ptr))
|
return newSort(c, C.Z3_mk_type_variable(c.ptr, name.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Quantifier represents a quantified formula (forall or exists)
|
// Quantifier represents a quantified formula (forall or exists)
|
||||||
type Quantifier struct {
|
type Quantifier struct {
|
||||||
ctx *Context
|
ctx *Context
|
||||||
ptr C.Z3_ast
|
ptr C.Z3_ast
|
||||||
}
|
}
|
||||||
|
|
||||||
// newQuantifier creates a new Quantifier with proper memory management
|
// newQuantifier creates a new Quantifier with proper memory management
|
||||||
func newQuantifier(ctx *Context, ptr C.Z3_ast) *Quantifier {
|
func newQuantifier(ctx *Context, ptr C.Z3_ast) *Quantifier {
|
||||||
q := &Quantifier{ctx: ctx, ptr: ptr}
|
q := &Quantifier{ctx: ctx, ptr: ptr}
|
||||||
C.Z3_inc_ref(ctx.ptr, ptr)
|
C.Z3_inc_ref(ctx.ptr, ptr)
|
||||||
runtime.SetFinalizer(q, func(qf *Quantifier) {
|
runtime.SetFinalizer(q, func(qf *Quantifier) {
|
||||||
C.Z3_dec_ref(qf.ctx.ptr, qf.ptr)
|
C.Z3_dec_ref(qf.ctx.ptr, qf.ptr)
|
||||||
})
|
})
|
||||||
return q
|
return q
|
||||||
}
|
}
|
||||||
|
|
||||||
// AsExpr converts a Quantifier to an Expr
|
// AsExpr converts a Quantifier to an Expr
|
||||||
func (q *Quantifier) AsExpr() *Expr {
|
func (q *Quantifier) AsExpr() *Expr {
|
||||||
return newExpr(q.ctx, q.ptr)
|
return newExpr(q.ctx, q.ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// IsUniversal returns true if this is a universal quantifier (forall)
|
// IsUniversal returns true if this is a universal quantifier (forall)
|
||||||
func (q *Quantifier) IsUniversal() bool {
|
func (q *Quantifier) IsUniversal() bool {
|
||||||
return bool(C.Z3_is_quantifier_forall(q.ctx.ptr, q.ptr))
|
return bool(C.Z3_is_quantifier_forall(q.ctx.ptr, q.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// IsExistential returns true if this is an existential quantifier (exists)
|
// IsExistential returns true if this is an existential quantifier (exists)
|
||||||
func (q *Quantifier) IsExistential() bool {
|
func (q *Quantifier) IsExistential() bool {
|
||||||
return bool(C.Z3_is_quantifier_exists(q.ctx.ptr, q.ptr))
|
return bool(C.Z3_is_quantifier_exists(q.ctx.ptr, q.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetWeight returns the weight of the quantifier
|
// GetWeight returns the weight of the quantifier
|
||||||
func (q *Quantifier) GetWeight() int {
|
func (q *Quantifier) GetWeight() int {
|
||||||
return int(C.Z3_get_quantifier_weight(q.ctx.ptr, q.ptr))
|
return int(C.Z3_get_quantifier_weight(q.ctx.ptr, q.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetNumPatterns returns the number of patterns
|
// GetNumPatterns returns the number of patterns
|
||||||
func (q *Quantifier) GetNumPatterns() int {
|
func (q *Quantifier) GetNumPatterns() int {
|
||||||
return int(C.Z3_get_quantifier_num_patterns(q.ctx.ptr, q.ptr))
|
return int(C.Z3_get_quantifier_num_patterns(q.ctx.ptr, q.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetPattern returns the pattern at the given index
|
// GetPattern returns the pattern at the given index
|
||||||
func (q *Quantifier) GetPattern(idx int) *Pattern {
|
func (q *Quantifier) GetPattern(idx int) *Pattern {
|
||||||
ptr := C.Z3_get_quantifier_pattern_ast(q.ctx.ptr, q.ptr, C.uint(idx))
|
ptr := C.Z3_get_quantifier_pattern_ast(q.ctx.ptr, q.ptr, C.uint(idx))
|
||||||
return newPattern(q.ctx, ptr)
|
return newPattern(q.ctx, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetNumNoPatterns returns the number of no-patterns
|
// GetNumNoPatterns returns the number of no-patterns
|
||||||
func (q *Quantifier) GetNumNoPatterns() int {
|
func (q *Quantifier) GetNumNoPatterns() int {
|
||||||
return int(C.Z3_get_quantifier_num_no_patterns(q.ctx.ptr, q.ptr))
|
return int(C.Z3_get_quantifier_num_no_patterns(q.ctx.ptr, q.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetNoPattern returns the no-pattern at the given index
|
// GetNoPattern returns the no-pattern at the given index
|
||||||
func (q *Quantifier) GetNoPattern(idx int) *Pattern {
|
func (q *Quantifier) GetNoPattern(idx int) *Pattern {
|
||||||
ptr := C.Z3_get_quantifier_no_pattern_ast(q.ctx.ptr, q.ptr, C.uint(idx))
|
ptr := C.Z3_get_quantifier_no_pattern_ast(q.ctx.ptr, q.ptr, C.uint(idx))
|
||||||
return newPattern(q.ctx, (C.Z3_pattern)(unsafe.Pointer(ptr)))
|
return newPattern(q.ctx, (C.Z3_pattern)(unsafe.Pointer(ptr)))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetNumBound returns the number of bound variables
|
// GetNumBound returns the number of bound variables
|
||||||
func (q *Quantifier) GetNumBound() int {
|
func (q *Quantifier) GetNumBound() int {
|
||||||
return int(C.Z3_get_quantifier_num_bound(q.ctx.ptr, q.ptr))
|
return int(C.Z3_get_quantifier_num_bound(q.ctx.ptr, q.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetBoundName returns the name of the bound variable at the given index
|
// GetBoundName returns the name of the bound variable at the given index
|
||||||
func (q *Quantifier) GetBoundName(idx int) *Symbol {
|
func (q *Quantifier) GetBoundName(idx int) *Symbol {
|
||||||
ptr := C.Z3_get_quantifier_bound_name(q.ctx.ptr, q.ptr, C.uint(idx))
|
ptr := C.Z3_get_quantifier_bound_name(q.ctx.ptr, q.ptr, C.uint(idx))
|
||||||
return newSymbol(q.ctx, ptr)
|
return newSymbol(q.ctx, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetBoundSort returns the sort of the bound variable at the given index
|
// GetBoundSort returns the sort of the bound variable at the given index
|
||||||
func (q *Quantifier) GetBoundSort(idx int) *Sort {
|
func (q *Quantifier) GetBoundSort(idx int) *Sort {
|
||||||
ptr := C.Z3_get_quantifier_bound_sort(q.ctx.ptr, q.ptr, C.uint(idx))
|
ptr := C.Z3_get_quantifier_bound_sort(q.ctx.ptr, q.ptr, C.uint(idx))
|
||||||
return newSort(q.ctx, ptr)
|
return newSort(q.ctx, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetBody returns the body of the quantifier
|
// GetBody returns the body of the quantifier
|
||||||
func (q *Quantifier) GetBody() *Expr {
|
func (q *Quantifier) GetBody() *Expr {
|
||||||
ptr := C.Z3_get_quantifier_body(q.ctx.ptr, q.ptr)
|
ptr := C.Z3_get_quantifier_body(q.ctx.ptr, q.ptr)
|
||||||
return newExpr(q.ctx, ptr)
|
return newExpr(q.ctx, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// String returns the string representation of the quantifier
|
// String returns the string representation of the quantifier
|
||||||
func (q *Quantifier) String() string {
|
func (q *Quantifier) String() string {
|
||||||
return q.AsExpr().String()
|
return q.AsExpr().String()
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkQuantifier creates a quantifier with patterns
|
// MkQuantifier creates a quantifier with patterns
|
||||||
func (c *Context) MkQuantifier(isForall bool, weight int, sorts []*Sort, names []*Symbol, body *Expr, patterns []*Pattern) *Quantifier {
|
func (c *Context) MkQuantifier(isForall bool, weight int, sorts []*Sort, names []*Symbol, body *Expr, patterns []*Pattern) *Quantifier {
|
||||||
var forallInt C.bool
|
var forallInt C.bool
|
||||||
if isForall {
|
if isForall {
|
||||||
forallInt = true
|
forallInt = true
|
||||||
} else {
|
} else {
|
||||||
forallInt = false
|
forallInt = false
|
||||||
}
|
}
|
||||||
|
|
||||||
numBound := len(sorts)
|
numBound := len(sorts)
|
||||||
if numBound != len(names) {
|
if numBound != len(names) {
|
||||||
panic("Number of sorts must match number of names")
|
panic("Number of sorts must match number of names")
|
||||||
}
|
}
|
||||||
|
|
||||||
var cSorts []C.Z3_sort
|
var cSorts []C.Z3_sort
|
||||||
var cNames []C.Z3_symbol
|
var cNames []C.Z3_symbol
|
||||||
if numBound > 0 {
|
if numBound > 0 {
|
||||||
cSorts = make([]C.Z3_sort, numBound)
|
cSorts = make([]C.Z3_sort, numBound)
|
||||||
cNames = make([]C.Z3_symbol, numBound)
|
cNames = make([]C.Z3_symbol, numBound)
|
||||||
for i := 0; i < numBound; i++ {
|
for i := 0; i < numBound; i++ {
|
||||||
cSorts[i] = sorts[i].ptr
|
cSorts[i] = sorts[i].ptr
|
||||||
cNames[i] = names[i].ptr
|
cNames[i] = names[i].ptr
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var cPatterns []C.Z3_pattern
|
var cPatterns []C.Z3_pattern
|
||||||
var patternsPtr *C.Z3_pattern
|
var patternsPtr *C.Z3_pattern
|
||||||
numPatterns := len(patterns)
|
numPatterns := len(patterns)
|
||||||
if numPatterns > 0 {
|
if numPatterns > 0 {
|
||||||
cPatterns = make([]C.Z3_pattern, numPatterns)
|
cPatterns = make([]C.Z3_pattern, numPatterns)
|
||||||
for i := 0; i < numPatterns; i++ {
|
for i := 0; i < numPatterns; i++ {
|
||||||
cPatterns[i] = patterns[i].ptr
|
cPatterns[i] = patterns[i].ptr
|
||||||
}
|
}
|
||||||
patternsPtr = &cPatterns[0]
|
patternsPtr = &cPatterns[0]
|
||||||
}
|
}
|
||||||
|
|
||||||
var sortsPtr *C.Z3_sort
|
var sortsPtr *C.Z3_sort
|
||||||
var namesPtr *C.Z3_symbol
|
var namesPtr *C.Z3_symbol
|
||||||
if numBound > 0 {
|
if numBound > 0 {
|
||||||
sortsPtr = &cSorts[0]
|
sortsPtr = &cSorts[0]
|
||||||
namesPtr = &cNames[0]
|
namesPtr = &cNames[0]
|
||||||
}
|
}
|
||||||
|
|
||||||
ptr := C.Z3_mk_quantifier(c.ptr, forallInt, C.uint(weight), C.uint(numPatterns), patternsPtr,
|
ptr := C.Z3_mk_quantifier(c.ptr, forallInt, C.uint(weight), C.uint(numPatterns), patternsPtr,
|
||||||
C.uint(numBound), sortsPtr, namesPtr, body.ptr)
|
C.uint(numBound), sortsPtr, namesPtr, body.ptr)
|
||||||
return newQuantifier(c, ptr)
|
return newQuantifier(c, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkQuantifierConst creates a quantifier using constant bound variables
|
// MkQuantifierConst creates a quantifier using constant bound variables
|
||||||
func (c *Context) MkQuantifierConst(isForall bool, weight int, bound []*Expr, body *Expr, patterns []*Pattern) *Quantifier {
|
func (c *Context) MkQuantifierConst(isForall bool, weight int, bound []*Expr, body *Expr, patterns []*Pattern) *Quantifier {
|
||||||
var forallInt C.bool
|
var forallInt C.bool
|
||||||
if isForall {
|
if isForall {
|
||||||
forallInt = true
|
forallInt = true
|
||||||
} else {
|
} else {
|
||||||
forallInt = false
|
forallInt = false
|
||||||
}
|
}
|
||||||
|
|
||||||
numBound := len(bound)
|
numBound := len(bound)
|
||||||
var cBound []C.Z3_app
|
var cBound []C.Z3_app
|
||||||
var boundPtr *C.Z3_app
|
var boundPtr *C.Z3_app
|
||||||
if numBound > 0 {
|
if numBound > 0 {
|
||||||
cBound = make([]C.Z3_app, numBound)
|
cBound = make([]C.Z3_app, numBound)
|
||||||
for i := 0; i < numBound; i++ {
|
for i := 0; i < numBound; i++ {
|
||||||
cBound[i] = (C.Z3_app)(unsafe.Pointer(bound[i].ptr))
|
cBound[i] = (C.Z3_app)(unsafe.Pointer(bound[i].ptr))
|
||||||
}
|
}
|
||||||
boundPtr = &cBound[0]
|
boundPtr = &cBound[0]
|
||||||
}
|
}
|
||||||
|
|
||||||
var cPatterns []C.Z3_pattern
|
var cPatterns []C.Z3_pattern
|
||||||
var patternsPtr *C.Z3_pattern
|
var patternsPtr *C.Z3_pattern
|
||||||
numPatterns := len(patterns)
|
numPatterns := len(patterns)
|
||||||
if numPatterns > 0 {
|
if numPatterns > 0 {
|
||||||
cPatterns = make([]C.Z3_pattern, numPatterns)
|
cPatterns = make([]C.Z3_pattern, numPatterns)
|
||||||
for i := 0; i < numPatterns; i++ {
|
for i := 0; i < numPatterns; i++ {
|
||||||
cPatterns[i] = patterns[i].ptr
|
cPatterns[i] = patterns[i].ptr
|
||||||
}
|
}
|
||||||
patternsPtr = &cPatterns[0]
|
patternsPtr = &cPatterns[0]
|
||||||
}
|
}
|
||||||
|
|
||||||
ptr := C.Z3_mk_quantifier_const(c.ptr, forallInt, C.uint(weight), C.uint(numBound), boundPtr,
|
ptr := C.Z3_mk_quantifier_const(c.ptr, forallInt, C.uint(weight), C.uint(numBound), boundPtr,
|
||||||
C.uint(numPatterns), patternsPtr, body.ptr)
|
C.uint(numPatterns), patternsPtr, body.ptr)
|
||||||
return newQuantifier(c, ptr)
|
return newQuantifier(c, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Lambda represents a lambda expression
|
// Lambda represents a lambda expression
|
||||||
type Lambda struct {
|
type Lambda struct {
|
||||||
ctx *Context
|
ctx *Context
|
||||||
ptr C.Z3_ast
|
ptr C.Z3_ast
|
||||||
}
|
}
|
||||||
|
|
||||||
// newLambda creates a new Lambda with proper memory management
|
// newLambda creates a new Lambda with proper memory management
|
||||||
func newLambda(ctx *Context, ptr C.Z3_ast) *Lambda {
|
func newLambda(ctx *Context, ptr C.Z3_ast) *Lambda {
|
||||||
l := &Lambda{ctx: ctx, ptr: ptr}
|
l := &Lambda{ctx: ctx, ptr: ptr}
|
||||||
C.Z3_inc_ref(ctx.ptr, ptr)
|
C.Z3_inc_ref(ctx.ptr, ptr)
|
||||||
runtime.SetFinalizer(l, func(lam *Lambda) {
|
runtime.SetFinalizer(l, func(lam *Lambda) {
|
||||||
C.Z3_dec_ref(lam.ctx.ptr, lam.ptr)
|
C.Z3_dec_ref(lam.ctx.ptr, lam.ptr)
|
||||||
})
|
})
|
||||||
return l
|
return l
|
||||||
}
|
}
|
||||||
|
|
||||||
// AsExpr converts a Lambda to an Expr
|
// AsExpr converts a Lambda to an Expr
|
||||||
func (l *Lambda) AsExpr() *Expr {
|
func (l *Lambda) AsExpr() *Expr {
|
||||||
return newExpr(l.ctx, l.ptr)
|
return newExpr(l.ctx, l.ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetNumBound returns the number of bound variables
|
// GetNumBound returns the number of bound variables
|
||||||
func (l *Lambda) GetNumBound() int {
|
func (l *Lambda) GetNumBound() int {
|
||||||
return int(C.Z3_get_quantifier_num_bound(l.ctx.ptr, l.ptr))
|
return int(C.Z3_get_quantifier_num_bound(l.ctx.ptr, l.ptr))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetBoundName returns the name of the bound variable at the given index
|
// GetBoundName returns the name of the bound variable at the given index
|
||||||
func (l *Lambda) GetBoundName(idx int) *Symbol {
|
func (l *Lambda) GetBoundName(idx int) *Symbol {
|
||||||
ptr := C.Z3_get_quantifier_bound_name(l.ctx.ptr, l.ptr, C.uint(idx))
|
ptr := C.Z3_get_quantifier_bound_name(l.ctx.ptr, l.ptr, C.uint(idx))
|
||||||
return newSymbol(l.ctx, ptr)
|
return newSymbol(l.ctx, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetBoundSort returns the sort of the bound variable at the given index
|
// GetBoundSort returns the sort of the bound variable at the given index
|
||||||
func (l *Lambda) GetBoundSort(idx int) *Sort {
|
func (l *Lambda) GetBoundSort(idx int) *Sort {
|
||||||
ptr := C.Z3_get_quantifier_bound_sort(l.ctx.ptr, l.ptr, C.uint(idx))
|
ptr := C.Z3_get_quantifier_bound_sort(l.ctx.ptr, l.ptr, C.uint(idx))
|
||||||
return newSort(l.ctx, ptr)
|
return newSort(l.ctx, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetBody returns the body of the lambda expression
|
// GetBody returns the body of the lambda expression
|
||||||
func (l *Lambda) GetBody() *Expr {
|
func (l *Lambda) GetBody() *Expr {
|
||||||
ptr := C.Z3_get_quantifier_body(l.ctx.ptr, l.ptr)
|
ptr := C.Z3_get_quantifier_body(l.ctx.ptr, l.ptr)
|
||||||
return newExpr(l.ctx, ptr)
|
return newExpr(l.ctx, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// String returns the string representation of the lambda
|
// String returns the string representation of the lambda
|
||||||
func (l *Lambda) String() string {
|
func (l *Lambda) String() string {
|
||||||
return l.AsExpr().String()
|
return l.AsExpr().String()
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkLambda creates a lambda expression with sorts and names
|
// MkLambda creates a lambda expression with sorts and names
|
||||||
func (c *Context) MkLambda(sorts []*Sort, names []*Symbol, body *Expr) *Lambda {
|
func (c *Context) MkLambda(sorts []*Sort, names []*Symbol, body *Expr) *Lambda {
|
||||||
numBound := len(sorts)
|
numBound := len(sorts)
|
||||||
if numBound != len(names) {
|
if numBound != len(names) {
|
||||||
panic("Number of sorts must match number of names")
|
panic("Number of sorts must match number of names")
|
||||||
}
|
}
|
||||||
|
|
||||||
var cSorts []C.Z3_sort
|
var cSorts []C.Z3_sort
|
||||||
var cNames []C.Z3_symbol
|
var cNames []C.Z3_symbol
|
||||||
var sortsPtr *C.Z3_sort
|
var sortsPtr *C.Z3_sort
|
||||||
var namesPtr *C.Z3_symbol
|
var namesPtr *C.Z3_symbol
|
||||||
|
|
||||||
if numBound > 0 {
|
if numBound > 0 {
|
||||||
cSorts = make([]C.Z3_sort, numBound)
|
cSorts = make([]C.Z3_sort, numBound)
|
||||||
cNames = make([]C.Z3_symbol, numBound)
|
cNames = make([]C.Z3_symbol, numBound)
|
||||||
for i := 0; i < numBound; i++ {
|
for i := 0; i < numBound; i++ {
|
||||||
cSorts[i] = sorts[i].ptr
|
cSorts[i] = sorts[i].ptr
|
||||||
cNames[i] = names[i].ptr
|
cNames[i] = names[i].ptr
|
||||||
}
|
}
|
||||||
sortsPtr = &cSorts[0]
|
sortsPtr = &cSorts[0]
|
||||||
namesPtr = &cNames[0]
|
namesPtr = &cNames[0]
|
||||||
}
|
}
|
||||||
|
|
||||||
ptr := C.Z3_mk_lambda(c.ptr, C.uint(numBound), sortsPtr, namesPtr, body.ptr)
|
ptr := C.Z3_mk_lambda(c.ptr, C.uint(numBound), sortsPtr, namesPtr, body.ptr)
|
||||||
return newLambda(c, ptr)
|
return newLambda(c, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// MkLambdaConst creates a lambda expression using constant bound variables
|
// MkLambdaConst creates a lambda expression using constant bound variables
|
||||||
func (c *Context) MkLambdaConst(bound []*Expr, body *Expr) *Lambda {
|
func (c *Context) MkLambdaConst(bound []*Expr, body *Expr) *Lambda {
|
||||||
numBound := len(bound)
|
numBound := len(bound)
|
||||||
var cBound []C.Z3_app
|
var cBound []C.Z3_app
|
||||||
var boundPtr *C.Z3_app
|
var boundPtr *C.Z3_app
|
||||||
|
|
||||||
if numBound > 0 {
|
if numBound > 0 {
|
||||||
cBound = make([]C.Z3_app, numBound)
|
cBound = make([]C.Z3_app, numBound)
|
||||||
for i := 0; i < numBound; i++ {
|
for i := 0; i < numBound; i++ {
|
||||||
cBound[i] = (C.Z3_app)(unsafe.Pointer(bound[i].ptr))
|
cBound[i] = (C.Z3_app)(unsafe.Pointer(bound[i].ptr))
|
||||||
}
|
}
|
||||||
boundPtr = &cBound[0]
|
boundPtr = &cBound[0]
|
||||||
}
|
}
|
||||||
|
|
||||||
ptr := C.Z3_mk_lambda_const(c.ptr, C.uint(numBound), boundPtr, body.ptr)
|
ptr := C.Z3_mk_lambda_const(c.ptr, C.uint(numBound), boundPtr, body.ptr)
|
||||||
return newLambda(c, ptr)
|
return newLambda(c, ptr)
|
||||||
}
|
}
|
||||||
|
|
||||||
// astVectorToExprs converts a Z3_ast_vector to a slice of Expr.
|
// astVectorToExprs converts a Z3_ast_vector to a slice of Expr.
|
||||||
|
|
@ -793,14 +793,14 @@ return newLambda(c, ptr)
|
||||||
// incrementing it on entry and decrementing it on exit.
|
// incrementing it on entry and decrementing it on exit.
|
||||||
// The individual AST elements are already reference counted by newExpr.
|
// The individual AST elements are already reference counted by newExpr.
|
||||||
func astVectorToExprs(ctx *Context, vec C.Z3_ast_vector) []*Expr {
|
func astVectorToExprs(ctx *Context, vec C.Z3_ast_vector) []*Expr {
|
||||||
// Increment reference count for the vector since we're using it
|
// Increment reference count for the vector since we're using it
|
||||||
C.Z3_ast_vector_inc_ref(ctx.ptr, vec)
|
C.Z3_ast_vector_inc_ref(ctx.ptr, vec)
|
||||||
defer C.Z3_ast_vector_dec_ref(ctx.ptr, vec)
|
defer C.Z3_ast_vector_dec_ref(ctx.ptr, vec)
|
||||||
|
|
||||||
size := uint(C.Z3_ast_vector_size(ctx.ptr, vec))
|
size := uint(C.Z3_ast_vector_size(ctx.ptr, vec))
|
||||||
result := make([]*Expr, size)
|
result := make([]*Expr, size)
|
||||||
for i := uint(0); i < size; i++ {
|
for i := uint(0); i < size; i++ {
|
||||||
result[i] = newExpr(ctx, C.Z3_ast_vector_get(ctx.ptr, vec, C.uint(i)))
|
result[i] = newExpr(ctx, C.Z3_ast_vector_get(ctx.ptr, vec, C.uint(i)))
|
||||||
}
|
}
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue