it.unibo.jakta.agents.bdi.environment.impl.EnvironmentImpl.kt Maven / Gradle / Ivy

Go to download
package it.unibo.jakta.agents.bdi.environment.impl

import it.unibo.jakta.agents.bdi.Agent
import it.unibo.jakta.agents.bdi.AgentID
import it.unibo.jakta.agents.bdi.messages.Message
import it.unibo.jakta.agents.bdi.actions.ExternalAction
import it.unibo.jakta.agents.bdi.beliefs.BeliefBase
import it.unibo.jakta.agents.bdi.environment.Environment
import it.unibo.jakta.agents.bdi.messages.MessageQueue
import it.unibo.jakta.agents.bdi.perception.Perception

open class EnvironmentImpl(
    override val externalActions: Map,
    override val agentIDs: Map = emptyMap(),
    override val messageBoxes: Map = mapOf(),
    override var perception: Perception,
    override val data: Map = emptyMap(),
) : Environment {
    override fun getNextMessage(agentName: String): Message? = messageBoxes[agentIDs[agentName]]?.lastOrNull()

    override fun popMessage(agentName: String): Environment {
        val message = getNextMessage(agentName)
        return if (message != null) {
            copy(
                messageBoxes = messageBoxes + mapOf(
                    agentIDs[agentName]!! to messageBoxes[agentIDs[agentName]]!!.minus(message)
                )
            )
        } else this
    }

    override fun submitMessage(agentName: String, message: Message) =
        if (messageBoxes.contains(agentIDs[agentName])) {
            copy(
                messageBoxes = messageBoxes + mapOf(
                    agentIDs[agentName]!! to messageBoxes[agentIDs[agentName]]!!.plus(message)
                )
            )
        } else this

    override fun broadcastMessage(message: Message): Environment = copy(
        messageBoxes = messageBoxes.entries.associate { it.key to it.value + message },
    )

    override fun addAgent(agent: Agent): Environment =
        if (!agentIDs.contains(agent.name)) {
            copy(
                agentIDs = agentIDs + mapOf(agent.name to agent.agentID),
                messageBoxes = messageBoxes + mapOf(agent.agentID to emptyList()),
            )
        } else this

    override fun removeAgent(agentName: String): Environment =
        if (agentIDs.contains(agentName)) {
            copy(
                messageBoxes = messageBoxes - agentIDs[agentName]!!,
                agentIDs = agentIDs - agentName,
            )
        } else this

    override fun percept(): BeliefBase = perception.percept()

    override fun addData(key: String, value: Any): Environment = copy(data = data + Pair(key, value))

    override fun removeData(key: String): Environment = copy(data = data - key)

    override fun updateData(newData: Map): Environment = copy(data = newData)
    override fun copy(
        agentIDs: Map,
        externalActions: Map,
        messageBoxes: Map,
        perception: Perception,
        data: Map
    ): Environment = EnvironmentImpl(
        externalActions,
        agentIDs,
        messageBoxes,
        perception,
        data
    )

    override fun toString(): String = """
        Environment(
           class=${this.javaClass}
           actions=${externalActions.values},
           agents=${agentIDs.values}, 
           messages=${messageBoxes.keys},
           perception=$perception
        )
    """.trimIndent()
}