balkian/gists
1
0
Fork 0
mirror of https://github.com/balkian/gists.git synced 2024-11-21 17:22:29 +00:00
Code Issues Projects Releases Wiki Activity

Update example

Browse Source
This commit is contained in:
J. Fernando Sánchez 2018-12-09 16:34:59 +01:00
parent c9c55ac60b
commit 1eef636a58
2 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
coalition.py
View File

@ -1,34 +1,39 @@
from soil.agents import FSM, state, default_state from soil.agents import FSM, state, default_state
from random import randint from random import randint
MAX_WEALTH = 10e6 MAX_WEALTH = 2000
class CoalitionAgent(FSM): class CoalitionAgent(FSM):
defaults = { defaults = {
'wealth': -1, 'wealth': -1,
'wealth_threshold': 1000, 'wealth_threshold': 1000,
} }
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(CoalitionAgent, self).__init__(*args, **kwargs) super(CoalitionAgent, self).__init__(*args, **kwargs)
if self['wealth'] == -1: if self['wealth'] == -1:
self['wealth'] = randint(0, self.env.get('max_wealth', MAX_WEALTH)) self['wealth'] = int(randint(0, self.env.get('max_wealth', MAX_WEALTH)))
@default_state @default_state
@state @state
def looking_for_coalitions(self): def looking_for_coalitions(self):
for agent in self.get_agents(): for agent in self.get_agents(state_id=self.looking_for_coalitions.id):
if agent['wealth'] > self['wealth_threshold']: if agent['wealth'] > self['wealth_threshold']:
self.join_coalition(agent) self.join_coalition(agent)
friends = list(friend.id for friend in self.get_neighboring_agents())
self.info('List of friends: {}'.format(friends))
return self.idle return self.idle
@state @state
def idle(self): def idle(self):
# Do nothing # Do nothing
pass pass
def join_coalition(self, other): def join_coalition(self, other):
# Add your methods here, like adding edges between users... # Adill.detect.badobjectsdd your methods here, like adding edges between users...
# You'll probably want to check if you've already joined the user first # You'll probably want to check if you've already joined the user first
self.info('Joining {}'.format(other.id))
self.env.add_edge(self, other) self.env.add_edge(self, other)

3
coalition.yml
View File

@ -1,5 +1,6 @@
---
name: CoalitionExample name: CoalitionExample
agent_type: coalition.CoalitionAgent agent_type: coalition.CoalitionAgent
network_params: network_params:
generator: empty_graph generator: empty_graph
n: 100 n: 100