commonMain.it.unibo.tuprolog.solve.problog.ProblogSolverFactory.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of solve-problog-jvm Show documentation
State-machine-based implementation of ProbLog
There is a newer version: 1.0.4
package it.unibo.tuprolog.solve.problog

import it.unibo.tuprolog.solve.MutableSolver
import it.unibo.tuprolog.solve.Solver
import it.unibo.tuprolog.solve.SolverFactory
import it.unibo.tuprolog.solve.channel.InputChannel
import it.unibo.tuprolog.solve.channel.InputStore
import it.unibo.tuprolog.solve.channel.OutputChannel
import it.unibo.tuprolog.solve.channel.OutputStore
import it.unibo.tuprolog.solve.classic.ClassicSolverFactory
import it.unibo.tuprolog.solve.exception.Warning
import it.unibo.tuprolog.solve.flags.FlagStore
import it.unibo.tuprolog.solve.flags.TrackVariables
import it.unibo.tuprolog.solve.flags.invoke
import it.unibo.tuprolog.solve.library.Library
import it.unibo.tuprolog.solve.library.Runtime
import it.unibo.tuprolog.solve.problog.lib.ProblogLib
import it.unibo.tuprolog.solve.problog.lib.knowledge.ProblogTheory
import it.unibo.tuprolog.theory.Theory
import it.unibo.tuprolog.unify.Unificator
import kotlin.js.JsExport

@Suppress("NON_EXPORTABLE_TYPE")
@JsExport
object ProblogSolverFactory : SolverFactory {

    override val defaultBuiltins: Library
        get() = ProblogLib.DefaultBuiltins

    override val defaultFlags: FlagStore
        get() = ensureVariablesTracking(super.defaultFlags)

    private fun ensureVariablesTracking(flags: FlagStore): FlagStore =
        flags + TrackVariables { ON }

    override fun solverOf(
        unificator: Unificator,
        libraries: Runtime,
        flags: FlagStore,
        staticKb: Theory,
        dynamicKb: Theory,
        inputs: InputStore,
        outputs: OutputStore
    ): Solver =
        ProblogSolver(
            ClassicSolverFactory.solverOf(
                unificator,
                fixLibraries(libraries),
                ensureVariablesTracking(flags),
                ProblogTheory.of(unificator, staticKb),
                ProblogTheory.of(unificator, dynamicKb),
                inputs,
                outputs
            )
        )

    private fun fixLibraries(libraries: Runtime): Runtime {
        return if (ProblogLib.alias !in libraries) {
            libraries.plus(ProblogLib.MinimalBuiltins)
        } else {
            libraries
        }
    }

    override fun solverOf(
        unificator: Unificator,
        libraries: Runtime,
        flags: FlagStore,
        staticKb: Theory,
        dynamicKb: Theory,
        stdIn: InputChannel,
        stdOut: OutputChannel,
        stdErr: OutputChannel,
        warnings: OutputChannel
    ): Solver =
        ProblogSolver(
            ClassicSolverFactory.solverOf(
                unificator,
                fixLibraries(libraries),
                ensureVariablesTracking(flags),
                ProblogTheory.of(unificator, staticKb),
                ProblogTheory.of(unificator, dynamicKb),
                stdIn,
                stdOut,
                stdErr,
                warnings
            )
        )

    override fun mutableSolverOf(
        unificator: Unificator,
        libraries: Runtime,
        flags: FlagStore,
        staticKb: Theory,
        dynamicKb: Theory,
        stdIn: InputChannel,
        stdOut: OutputChannel,
        stdErr: OutputChannel,
        warnings: OutputChannel
    ): MutableSolver =
        MutableProblogSolver(
            ClassicSolverFactory.mutableSolverOf(
                unificator,
                fixLibraries(libraries),
                ensureVariablesTracking(flags),
                ProblogTheory.of(unificator, staticKb),
                ProblogTheory.of(unificator, dynamicKb),
                stdIn,
                stdOut,
                stdErr,
                warnings
            )
        )

    override fun mutableSolverOf(
        unificator: Unificator,
        libraries: Runtime,
        flags: FlagStore,
        staticKb: Theory,
        dynamicKb: Theory,
        inputs: InputStore,
        outputs: OutputStore
    ): MutableSolver =
        MutableProblogSolver(
            ClassicSolverFactory.mutableSolverOf(
                unificator,
                fixLibraries(libraries),
                ensureVariablesTracking(flags),
                ProblogTheory.of(unificator, staticKb),
                ProblogTheory.of(unificator, dynamicKb),
                inputs,
                outputs
            )
        )
}