it.unibo.jakta.agents.bdi.dsl.plans.PlanScope.kt Maven / Gradle / Ivy

Go to download
package it.unibo.jakta.agents.bdi.dsl.plans

import it.unibo.jakta.agents.bdi.Prolog2Jakta
import it.unibo.jakta.agents.bdi.beliefs.Belief
import it.unibo.jakta.agents.bdi.events.AchievementGoalTrigger
import it.unibo.jakta.agents.bdi.events.BeliefBaseRevision
import it.unibo.jakta.agents.bdi.events.TestGoalTrigger
import it.unibo.jakta.agents.bdi.events.Trigger
import it.unibo.jakta.agents.bdi.goals.EmptyGoal
import it.unibo.jakta.agents.bdi.goals.Goal
import it.unibo.jakta.agents.bdi.plans.Plan
import it.unibo.tuprolog.core.Scope
import it.unibo.tuprolog.core.Struct
import it.unibo.tuprolog.core.Truth
import kotlin.reflect.KClass

data class PlanScope(
    private val scope: Scope,
    private val trigger: Struct,
    private val triggerType: KClass,
) {
    private var guard: Struct = Truth.TRUE
    private var goals: List = mutableListOf()
    var failure: Boolean = false

    infix fun onlyIf(guards: GuardScope.() -> Struct): PlanScope {
        guard = GuardScope(scope).let(guards)
        guard = guard.accept(Prolog2Jakta).castToStruct()
        return this
    }

    infix fun then(body: BodyScope.() -> Unit): PlanScope {
        goals = BodyScope(scope).also(body).build()
        return this
    }

    fun build(): Plan {
        if (failure) {
            return when (triggerType) {
                BeliefBaseRevision::class -> Plan.ofBeliefBaseRemoval(
                    Belief.from(trigger),
                    goals.ifEmpty { listOf(EmptyGoal()) },
                    guard,
                )
                TestGoalTrigger::class -> Plan.ofTestGoalFailure(
                    trigger,
                    goals.ifEmpty { listOf(EmptyGoal()) },
                    guard,
                )
                AchievementGoalTrigger::class -> Plan.ofAchievementGoalFailure(
                    trigger,
                    goals.ifEmpty { listOf(EmptyGoal()) },
                    guard,
                )
                else -> throw IllegalArgumentException("Unknown trigger type: $triggerType")
            }
        } else {
            return when (triggerType) {
                BeliefBaseRevision::class -> Plan.ofBeliefBaseAddition(
                    Belief.from(trigger),
                    goals.ifEmpty { listOf(EmptyGoal()) },
                    guard,
                )
                TestGoalTrigger::class -> Plan.ofTestGoalInvocation(
                    trigger,
                    goals.ifEmpty { listOf(EmptyGoal()) },
                    guard,
                )
                AchievementGoalTrigger::class -> Plan.ofAchievementGoalInvocation(
                    trigger,
                    goals.ifEmpty { listOf(EmptyGoal()) },
                    guard,
                )
                else -> throw IllegalArgumentException("Unknown trigger type: $triggerType")
            }
        }
    }
}