AgentsAverage/SimpleAgent.py at master · Liana2707/AgentsAverage · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
from random import randint

from spade.agent import Agent

from SimpleBehaviour import (
    SimpleBehaviour,
    STATE_ONE, STATE_TWO, STATE_THREE, STATE_FOUR, STATE_FIVE,
    StateOne, StateTwo, StateThree, StateFour,  StateFive
)

COUNT_AGENTS = 0
connections = [
    [1, 2, 3, 7],
    [0, 6],
    [0, 4, 5],
    [0, 4, 7, 8],
    [2, 3, 5, 8],
    [2, 6, 4],
    [1, 5, 9],
    [0, 3],
    [4, 3, 9],
    [8]
]


class SimpleAgent(Agent):
    number_of_agents = 10
    finish = False

    def __init__(self, jid: str, password: str):
        global COUNT_AGENTS
        self.id = COUNT_AGENTS
        COUNT_AGENTS += 1
        self.neighbours = connections[self.id]
        self.number = randint(0, 100)
        self.info = {jid: self.number}

        super().__init__(jid, password)

    def set_number_of_agents(self, number_of_agents, new_neighbours):
        self.number_of_agents = number_of_agents
        self.neighbours += new_neighbours

    def finished(self):
        return self.finish

    async def setup(self):
        behaviour = SimpleBehaviour()

        behaviour.add_state(name=STATE_ONE, state=StateOne())
        behaviour.add_state(name=STATE_TWO, state=StateTwo(), initial=True)
        behaviour.add_state(name=STATE_THREE, state=StateThree())
        behaviour.add_state(name=STATE_FOUR, state=StateFour())
        behaviour.add_state(name=STATE_FIVE, state=StateFive())

        behaviour.add_transition(source=STATE_ONE, dest=STATE_TWO)
        behaviour.add_transition(source=STATE_ONE, dest=STATE_FIVE)

        behaviour.add_transition(source=STATE_TWO, dest=STATE_THREE)
        behaviour.add_transition(source=STATE_TWO, dest=STATE_FIVE)

        behaviour.add_transition(source=STATE_THREE, dest=STATE_FOUR)
        behaviour.add_transition(source=STATE_THREE, dest=STATE_FIVE)

        behaviour.add_transition(source=STATE_FOUR, dest=STATE_ONE)
        behaviour.add_transition(source=STATE_FOUR, dest=STATE_TWO)
        behaviour.add_transition(source=STATE_FOUR, dest=STATE_FIVE)

        self.add_behaviour(behaviour)