radar_agent.py

import asyncio
from spade.behaviour import CyclicBehaviour
from spade.message import Message
from stationary_agent import StationaryAgent
import util


class ScanBehaviour(CyclicBehaviour):
    def __init__(self, name, timeline, hq_agent, locations):
        super().__init__()
        self.name = name
        self.timeline = timeline
        self.hq_agent = hq_agent
        self.locations = locations

        self.detected = False
        self.object = None
        self.object_id = None
        self.type = None
        self.at_x = None
        self.at_y = None

    async def on_start(self):
        util.mas_print_info("[RADAR] ({}) start scanning...".format(self.name))

    async def run(self):
        data = self.timeline.next()
        if not data:
            return

        if data['type'] == 'enemy':
            self.detected = True
            self.object = data['object']
            self.object_id = data['id']
            self.type = data['type']
            self.at_x = data['x']
            self.at_y = data['y']
        else:
            self.detected = False

        if self.detected:
            if not self.hq_agent.presence.is_available():
                await self.send_message('missile1')
                await self.send_message('missile2')
                util.mas_print_info("[RADAR] ({}) HQ not available, directly sent messages to missiles".format(self.name))
            elif util.circle_contains(self.locations['hq']['x'], self.locations['hq']['y'], 10, self.at_x, self.at_y):   # To HQ directly order missile to fire
                await self.send_message('missile1')
                await self.send_message('missile2')
                util.mas_print_info("[RADAR] ({}) directly sent messages to missiles".format(self.name))
            else:
                await self.send_message("hq")
                util.mas_print_info("[RADAR] ({}) sent message to HQ".format(self.name))

        await asyncio.sleep(util.step_delay)

    async def send_message(self, username):
        to = username + '@xmpp.test'
        msg = Message(to=to)  # Instantiate the message
        msg.set_metadata("performative", "inform")  # Set the "inform" FIPA performative
        msg.body = '{0}|{1}|{2}|{3}|{4}'.format(self.object, str(self.object_id), self.type, str(self.at_x),
                                                str(self.at_y))  # Set the message content

        util.mas_print_data('msg|{}|{}|{}'.format(self.name, username, msg.body))
        await self.send(msg)


class RadarAgent(StationaryAgent):

    def __init__(self, aid, timeline, hq_agent, locations):
        super().__init__(aid)

        self.timeline = timeline
        self.hq_agent = hq_agent
        self.locations = locations

    async def setup(self):
        util.mas_print_info("[RADAR] ({}) starting...".format(self.aid))
        b = ScanBehaviour(self.aid, self.timeline, self.hq_agent, self.locations)
        self.add_behaviour(b)