examples/boid: Update viz

EwoutH · EwoutH · commit 1648fe08c808 · 2024-10-15T16:48:51.000+02:00
Update the visualisation with sliders sliders for coloured boids.
diff --git a/examples/basic/boid_flockers/app.py b/examples/basic/boid_flockers/app.py
@@ -1,26 +1,57 @@
 from boid_flockers.model import BoidFlockers
 from mesa.visualization import SolaraViz, make_space_matplotlib
-
+from mesa.visualization import Slider
 
 def boid_draw(agent):
-    return {"color": "tab:red"}
+    if not agent.neighbors:  # Only for the first Frame
+        neighbors = len(agent.model.space.get_neighbors(agent.pos, agent.vision, False))
+    else:
+        neighbors = len(agent.neighbors)
 
+    if neighbors <= 1:
+        return {"color": "red", "size": 20}
+    elif neighbors >= 2:
+        return {"color": "green", "size": 20}
 
 model_params = {
-    "population": 100,
+    "population": Slider(
+        label="Number of boids",
+        value=100,
+        min=10,
+        max=200,
+        step=10,
+    ),
     "width": 100,
     "height": 100,
-    "speed": 5,
-    "vision": 10,
-    "separation": 2,
+    "speed": Slider(
+        label="Speed of Boids",
+        value=5,
+        min=1,
+        max=20,
+        step=1,
+    ),
+    "vision": Slider(
+        label="Vision of Bird (radius)",
+        value=10,
+        min=1,
+        max=50,
+        step=1,
+    ),
+    "separation": Slider(
+        label="Minimum Separation",
+        value=2,
+        min=1,
+        max=20,
+        step=1,
+    ),
 }
 
-model = BoidFlockers(100, 100, 100, 5, 10, 2)
+model = BoidFlockers()
 
 page = SolaraViz(
     model,
     [make_space_matplotlib(agent_portrayal=boid_draw)],
     model_params=model_params,
-    name="BoidFlockers",
+    name="Boid Flocking Model",
 )
 page  # noqa
