from mlagents_envs.side_channel import SideChannel, IncomingMessage, OutgoingMessage
from mlagents_envs.exception import UnityCommunicationException
from mlagents_envs.base_env import AgentId
import uuid


class AgentParametersChannel(SideChannel):
    """
    This is the SideChannel for sending agent-specific parameters to Unity.
    You can send parameters to an environment with the command
    set_float_parameter.
    """

    def __init__(self) -> None:
        channel_id = uuid.UUID(("534c891e-810f-11ea-a9d0-822485860401"))
        super().__init__(channel_id)

    def on_message_received(self, msg: IncomingMessage) -> None:
        raise UnityCommunicationException(
            "The EnvironmentParametersChannel received a message from Unity, "
            + "this should not have happend."
        )

    def set_float_parameter(self, agent_id: AgentId, key: str, value: float) -> None:
        """
        Sets a float environment parameter in the Unity Environment.
        :param key: The string identifier of the parameter.
        :param value: The float value of the parameter.
        """
        msg = OutgoingMessage()
        msg.write_int32(agent_id)
        msg.write_string(key)
        msg.write_float32(value)
        super().queue_message_to_send(msg)