main.io.ksmt.solver.z3.KZ3SMTLibParser.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ksmt-z3-core Show documentation

Kotlin API for various SMT solvers

There is a newer version: 0.5.26

package io.ksmt.solver.z3

import com.microsoft.z3.BoolExpr
import com.microsoft.z3.Context
import com.microsoft.z3.Z3Exception
import io.ksmt.KContext
import io.ksmt.expr.KExpr
import io.ksmt.parser.KSMTLibParseException
import io.ksmt.parser.KSMTLibParser
import io.ksmt.sort.KBoolSort
import java.nio.file.Path
import kotlin.io.path.absolutePathString

class KZ3SMTLibParser(private val ctx: KContext) : KSMTLibParser {
    override fun parse(smtLibFile: Path): List> = parse {
        parseSMTLIB2File(
            smtLibFile.absolutePathString(),
            emptyArray(),
            emptyArray(),
            emptyArray(),
            emptyArray()
        )
    }

    override fun parse(smtLibString: String): List> = parse {
        parseSMTLIB2String(
            smtLibString,
            emptyArray(),
            emptyArray(),
            emptyArray(),
            emptyArray()
        )
    }

    private fun parse(parser: Context.() -> Array) = try {
        KZ3Context(ctx).use {
            it.convertAssertions(it.nativeContext.parser().toList())
        }
    } catch (ex: Z3Exception) {
        throw KSMTLibParseException(ex)
    }

    private fun KZ3Context.convertAssertions(assertions: List): List> {
        val converter = KZ3ExprConverter(ctx, this)
        return with(converter) { assertions.map { nativeContext.unwrapAST(it).convertExpr() } }
    }

    companion object {
        init {
            // ensure z3 native library is loaded
            KZ3Solver(KContext()).close()
        }
    }
}