add num_max_agents, cleanup

This commit is contained in:
Alexander Bocken 2023-05-18 12:46:48 +02:00
parent a8cb347e4a
commit 30661542b5
Signed by: Alexander
GPG Key ID: 1D237BE83F9B05E8
4 changed files with 16 additions and 32 deletions

View File

@ -88,6 +88,7 @@ class RandomWalkerAnt(Agent):
for neighbor in self.front_neighbors: for neighbor in self.front_neighbors:
if self.model.grid.is_food(neighbor): if self.model.grid.is_food(neighbor):
self.drop_pheromone = "B" self.drop_pheromone = "B"
self.look_for_pheromone = "A"
self.sensitivity = self.sensitivity_0 self.sensitivity = self.sensitivity_0
self.prev_pos = neighbor self.prev_pos = neighbor
@ -100,13 +101,12 @@ class RandomWalkerAnt(Agent):
self.drop_pheromone = "A" self.drop_pheromone = "A"
self.sensitivity = self.sensitivity_0 self.sensitivity = self.sensitivity_0
#TODO: Do we flip the ant here or reset prev pos?
# For now, flip ant just like at food
self.prev_pos = neighbor self.prev_pos = neighbor
self._next_pos = self.pos self._next_pos = self.pos
# recruit new ants # recruit new ants
for agent_id in self.model.get_unique_ids(self.model.num_new_recruits): for agent_id in self.model.get_unique_ids(self.model.num_new_recruits):
if self.model.schedule.get_agent_count() < self.model.num_max_agents:
agent = RandomWalkerAnt(unique_id=agent_id, model=self.model, look_for_pheromone="B", drop_pheromone="A") agent = RandomWalkerAnt(unique_id=agent_id, model=self.model, look_for_pheromone="B", drop_pheromone="A")
agent._next_pos = self.pos agent._next_pos = self.pos
self.model.schedule.add(agent) self.model.schedule.add(agent)
@ -176,7 +176,9 @@ class RandomWalkerAnt(Agent):
assert(self.prev_pos is not None) assert(self.prev_pos is not None)
all_neighbors = self.neighbors() all_neighbors = self.neighbors()
neighbors_at_the_back = self.neighbors(pos=self.prev_pos, include_center=True) neighbors_at_the_back = self.neighbors(pos=self.prev_pos, include_center=True)
return list(filter(lambda i: i not in neighbors_at_the_back, all_neighbors)) front_neighbors = list(filter(lambda i: i not in neighbors_at_the_back, all_neighbors))
assert(len(front_neighbors) == 3) # not sure whether always the case, used for debugging
return front_neighbors
@property @property
def front_neighbor(self): def front_neighbor(self):

21
main.py
View File

@ -11,6 +11,7 @@ from agent import RandomWalkerAnt
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from mesa.space import Coordinate from mesa.space import Coordinate
from mesa.datacollection import DataCollector
def main(): def main():
check_pheromone_exponential_decay() check_pheromone_exponential_decay()
@ -22,9 +23,6 @@ def check_pheromone_exponential_decay():
Check whether wanted exponential decay of pheromones on grid is done correctly Check whether wanted exponential decay of pheromones on grid is done correctly
shows plot of pheromone placed on grid vs. equivalent exponential decay function shows plot of pheromone placed on grid vs. equivalent exponential decay function
""" """
from mesa.datacollection import DataCollector
width = 21 width = 21
height = width height = width
num_initial_roamers = 0 num_initial_roamers = 0
@ -46,9 +44,6 @@ def check_pheromone_exponential_decay():
model.run_model() model.run_model()
a_test = model.datacollector.get_model_vars_dataframe()["pheromone_a"] a_test = model.datacollector.get_model_vars_dataframe()["pheromone_a"]
import matplotlib.pyplot as plt
import numpy as np
plt.figure() plt.figure()
xx = np.linspace(0,1000, 10000) xx = np.linspace(0,1000, 10000)
yy = a_test[0]*np.exp(-model.decay_rates["A"]*xx) yy = a_test[0]*np.exp(-model.decay_rates["A"]*xx)
@ -66,8 +61,6 @@ def check_ant_sensitivity_linear_decay():
shows plot of ant sensitivity placed on grid vs. equivalent linear decay function shows plot of ant sensitivity placed on grid vs. equivalent linear decay function
not food sources are on the grid for this run to not reset sensitivities not food sources are on the grid for this run to not reset sensitivities
""" """
from mesa.datacollection import DataCollector
width = 50 width = 50
height = width height = width
num_initial_roamers = 1 num_initial_roamers = 1
@ -91,9 +84,6 @@ def check_ant_sensitivity_linear_decay():
model.run_model() model.run_model()
a_test = model.datacollector.get_agent_vars_dataframe().reset_index()["sensitivity"] a_test = model.datacollector.get_agent_vars_dataframe().reset_index()["sensitivity"]
import matplotlib.pyplot as plt
import numpy as np
plt.figure() plt.figure()
xx = np.linspace(0,1000, 10000) xx = np.linspace(0,1000, 10000)
yy = a_test[0] - start*xx yy = a_test[0] - start*xx
@ -109,9 +99,6 @@ def check_ant_pheromone_exponential_decay():
Check whether wanted exponential decay of pheromone drop rate for ants is correctly modeled Check whether wanted exponential decay of pheromone drop rate for ants is correctly modeled
shows plot of pheromone placed on grid vs. equivalent exponential decay function shows plot of pheromone placed on grid vs. equivalent exponential decay function
""" """
from mesa.datacollection import DataCollector
width = 50 width = 50
height = width height = width
num_initial_roamers = 1 num_initial_roamers = 1
@ -134,10 +121,6 @@ def check_ant_pheromone_exponential_decay():
model.run_model() model.run_model()
a_test = model.datacollector.get_agent_vars_dataframe().reset_index()["pheromone_drop_rate"] a_test = model.datacollector.get_agent_vars_dataframe().reset_index()["pheromone_drop_rate"]
import matplotlib.pyplot as plt
import numpy as np
plt.figure() plt.figure()
xx = np.linspace(0,1000, 10000) xx = np.linspace(0,1000, 10000)
yy = a_test[0]*np.exp(-model.schedule.agents[0].betas["A"]*xx) yy = a_test[0]*np.exp(-model.schedule.agents[0].betas["A"]*xx)
@ -148,8 +131,6 @@ def check_ant_pheromone_exponential_decay():
plt.show() plt.show()
if __name__ == "__main__": if __name__ == "__main__":
main() main()

View File

@ -39,6 +39,7 @@ class ActiveWalkerModel(Model):
} }
for agent_id in self.get_unique_ids(num_initial_roamers): for agent_id in self.get_unique_ids(num_initial_roamers):
if self.schedule.get_agent_count() < self.num_max_agents:
agent = RandomWalkerAnt(unique_id=agent_id, model=self, look_for_pheromone="A", drop_pheromone="A") agent = RandomWalkerAnt(unique_id=agent_id, model=self, look_for_pheromone="A", drop_pheromone="A")
self.schedule.add(agent) self.schedule.add(agent)
self.grid.place_agent(agent, pos=nest_position) self.grid.place_agent(agent, pos=nest_position)

View File

@ -23,9 +23,9 @@ def setup(params=None):
if params is None: if params is None:
params = { params = {
"width": 50, "height": 50, "width": 50, "height": 50,
"num_max_agents" : 100, "num_max_agents" : 1000,
"nest_position" : (25,25), "nest_position" : (25,25),
"num_initial_roamers" : 5, "num_initial_roamers" : 20,
} }