Alexander/ants
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dynamic normalization for pheromones

Browse Source
This commit is contained in:
Alexander Bocken 2023-05-11 17:57:39 +02:00
parent e45d33bf8d
commit ecb9cc26e9
Signed by: Alexander
GPG Key ID: 1D237BE83F9B05E8
1 changed files with 36 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
server.py
View File

@ -39,14 +39,16 @@ def setup(params=None):
canvas_width = 500
canvas_height = 500
def __init__(self, portrayal_method, grid_width, grid_height, canvas_width=500, canvas_height=500,):
def __init__(self, portrayal_method, grid_width, grid_height, canvas_width=500, canvas_height=500, norm_method=None):
super().__init__(portrayal_method, grid_width, grid_height, canvas_width, canvas_height)
self.norm_method = norm_method
def render(self, model):
grid_state = defaultdict(list)
norm = self.norm_method(model)
for x in range(model.grid.width):
for y in range(model.grid.height):
portrayal = self.portrayal_method(model, (x, y))
portrayal = self.portrayal_method(model, (x, y), norm)
if portrayal:
portrayal["x"] = x
portrayal["y"] = y
@ -63,13 +65,13 @@ def setup(params=None):
return max(int(255 - level * 255 / normalization), 0)
def portray_ant_density(model, pos):
def portray_ant_density(model, pos, norm):
if model.grid.is_nest(pos):
col = "red"
elif model.grid.is_food(pos):
col = "green"
else:
col = get_color(level=len(model.grid[pos]), normalization=5)
col = get_color(level=len(model.grid[pos]), normalization=norm)
col = f"rgb({col}, {col}, {col})"
@ -83,9 +85,12 @@ def setup(params=None):
"Color": col,
}
def portray_pheromone_density(model, pos):
col_a = get_color(level=model.grid.fields["A"][pos], normalization=3)
col_b = get_color(level=model.grid.fields["B"][pos], normalization=3)
def get_max_grid_val(model, key):
return np.max(model.grid.fields[key])
def portray_pheromone_density(model, pos, norm):
col_a = get_color(level=model.grid.fields["A"][pos], normalization=norm)
col_b = get_color(level=model.grid.fields["B"][pos], normalization=norm)
return {
"Shape": "hex",
"r": 1,
@ -101,10 +106,31 @@ def setup(params=None):
width = params['width']
height = params['height']
pixel_ratio = 10
grid_ants = CanvasHexGridMultiAgents(portray_ant_density, width, height, width*pixel_ratio, height*pixel_ratio)
grid_pheromones = CanvasHexGridMultiAgents(portray_pheromone_density, width, height, width*pixel_ratio, height*pixel_ratio)
grid_ants = CanvasHexGridMultiAgents(portray_ant_density,
width, height, width*pixel_ratio, height*pixel_ratio,
norm_method=lambda m: 5)
def norm_ants(model):
return 5
def norm_pheromones(model):
max_a = np.max(model.grid.fields["A"])
max_b = np.max(model.grid.fields["B"])
return np.ceil(np.max([max_a, max_b, 20]) + 1e-4).astype(int)
grid_pheromones = CanvasHexGridMultiAgents(portray_pheromone_density,
width, height, width*pixel_ratio, height*pixel_ratio,
norm_method=norm_pheromones
)
test_text = TextElement()
return ModularServer(ActiveWalkerModel, [lambda m: "<h3>Ant density</h3><h5>Nest: Red, Food: Green</h5>", grid_ants, lambda m: "<h3>Pheromone Density</h3><h5>Pheromone A: Cyan, Pheromone B: Pink</h5>", grid_pheromones],
return ModularServer(ActiveWalkerModel,
[lambda m: "<h3>Ant density</h3><h5>Nest: Red, Food: Green</h5>",
grid_ants,
lambda m: f"<h5>Normalization Value: {norm_ants(m)}</h5>",
lambda m: "<h3>Pheromone Density</h3><h5>Pheromone A: Cyan, Pheromone B: Pink</h5>",
grid_pheromones,
lambda m: f"<h5>Normalization Value: {norm_pheromones(m)}</h5>"
],
"Active Random Walker Ants", params)
if __name__ == "__main__":