Viz: Refactor Matplotlib plotting

mesa/examples/advanced/wolf_sheep/app.py

@@ -7,33 +7,28 @@
     make_space_matplotlib,
 )
 
-WOLF_COLOR = "#000000"
-SHEEP_COLOR = "#648FFF"
-
 
 def wolf_sheep_portrayal(agent):
     if agent is None:
         return
 
     portrayal = {
         "size": 25,
-        "shape": "s",  # square marker
     }
 
     if isinstance(agent, Wolf):
-        portrayal["color"] = WOLF_COLOR
-        portrayal["Layer"] = 3
+        portrayal["color"] = "tab:orange"
+        portrayal["marker"] = "o"
     elif isinstance(agent, Sheep):
-        portrayal["color"] = SHEEP_COLOR
-        portrayal["Layer"] = 2
+        portrayal["color"] = "tab:blue"
+        portrayal["marker"] = "o"
     elif isinstance(agent, GrassPatch):
         if agent.fully_grown:
-            portrayal["color"] = "#00FF00"
+            portrayal["color"] = "tab:green"
         else:
-            portrayal["color"] = "#84e184"
-        # portrayal["shape"] = "rect"
-        # portrayal["Filled"] = "true"
-        portrayal["Layer"] = 1
+            portrayal["color"] = "tab:brown"
+        portrayal["marker"] = "s"
+        portrayal["size"] = 75
 
     return portrayal
 
@@ -62,10 +57,20 @@
 }
 
 
-space_component = make_space_matplotlib(wolf_sheep_portrayal)
-lineplot_component = make_plot_measure(["Wolves", "Sheep", "Grass"])
+def post_process(ax):
+    ax.set_aspect("equal")
+    ax.set_xticks([])
+    ax.set_yticks([])
+
+
+space_component = make_space_matplotlib(
+    wolf_sheep_portrayal, draw_grid=False, post_process=post_process
+)
+lineplot_component = make_plot_measure(
+    {"Wolves": "tab:orange", "Sheep": "tab:blue", "Grass": "tab:green"}
+)
 
-model = WolfSheep()
+model = WolfSheep(grass=True)
 
 
 page = SolaraViz(

mesa/examples/basic/conways_game_of_life/app.py

@@ -1,3 +1,11 @@
+import os.path
+import sys
+
+sys.path.insert(
+    0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../../"))
+)
+
+
 from mesa.examples.basic.conways_game_of_life.model import ConwaysGameOfLife
 from mesa.visualization import (
     SolaraViz,
@@ -6,7 +14,7 @@
 
 
 def agent_portrayal(agent):
-    return {"color": "white" if agent.state == 0 else "black"}
+    return {"c": "white" if agent.state == 0 else "black", "marker": "s"}
 
 
 model_params = {