trading Sugarscape: Add solara visualization

Author: rht

examples/sugarscape_g1mt/Readme.md

@@ -82,6 +82,7 @@ Then open your browser to [http://127.0.0.1:8521/](http://127.0.0.1:8521/) and p
 * ``sugarscape_g1mt/sugar_map.txt``: Provides sugar and spice landscape in raster type format.
 * ``server.py``: Sets up and launches and interactive visualization server.
 * ``run.py``: Runs Server, Single Run or Batch Run  with data collection and basic analysis.
+* `app.py`: Runs a visualization server via Solara (`solara run app.py`).
 
 ## Additional Resources
 

examples/sugarscape_g1mt/app.py

@@ -0,0 +1,63 @@
+import numpy as np
+import solara
+from matplotlib.figure import Figure
+from mesa_models.experimental import JupyterViz
+from sugarscape_g1mt.model import SugarscapeG1mt
+from sugarscape_g1mt.resource_agents import Sugar
+from sugarscape_g1mt.trader_agents import Trader
+
+
+def space_drawer(viz):
+    def portray(g):
+        layers = {
+            "sugar": [[np.nan for j in range(g.height)] for i in range(g.width)],
+            "spice": [[np.nan for j in range(g.height)] for i in range(g.width)],
+            "trader": {"x": [], "y": [], "c": "tab:red", "marker": "o", "s": 10},
+        }
+
+        for i in range(g.width):
+            for j in range(g.height):
+                content = g._grid[i][j]
+                for agent in content:
+                    if isinstance(agent, Trader):
+                        layers["trader"]["x"].append(i)
+                        layers["trader"]["y"].append(j)
+                    else:
+                        # Don't visualize resource with value <= 1.
+                        value = agent.amount if agent.amount > 1 else np.nan
+                        if isinstance(agent, Sugar):
+                            layers["sugar"][i][j] = value
+                        else:
+                            layers["spice"][i][j] = value
+        return layers
+
+    fig = Figure()
+    ax = fig.subplots()
+    out = portray(viz.model.grid)
+    # Sugar
+    # Important note: imshow by default draws from upper left. You have to
+    # always explicitly specify origin="lower".
+    im = ax.imshow(out["sugar"], cmap="spring", origin="lower")
+    fig.colorbar(im, orientation="vertical")
+    # Spice
+    ax.imshow(out["spice"], cmap="winter", origin="lower")
+    # Trader
+    ax.scatter(**out["trader"])
+    ax.set_axis_off()
+    solara.FigureMatplotlib(fig, dependencies=[viz.model, viz.df])
+
+
+model_params = {
+    "width": 50,
+    "height": 50,
+}
+
+page = JupyterViz(
+    SugarscapeG1mt,
+    model_params,
+    measures=["Trader", "Price"],
+    name="Sugarscape {G1, M, T}",
+    space_drawer=space_drawer,
+    play_interval=1500,
+)
+page  # noqa