Hawk/Dove with Multiple Risk-Attitudes

%23%20%2F%2F%2F%20script%0A%23%20requires-python%20%3D%20%22%3E%3D3.11%22%0A%23%20dependencies%20%3D%20%5B%0A%23%20%20%20%22altair%3E%3D5.5.0%2C%3C6.0%22%2C%0A%23%20%20%20%22pandas%22%2C%0A%23%20%20%20%22marimo%3E%3D0.23.11%22%2C%0A%23%20%5D%0A%23%20%2F%2F%2F%0A%23%20mesa%20and%20simulatingrisk%20are%20installed%20in%20the%20init%20cell%20below%20because%20mesa's%0A%23%20dependency%20on%20tornado%20(a%20C-extension%20package)%20must%20be%20loaded%20from%20Pyodide's%0A%23%20built-ins%20before%20micropip%20attempts%20to%20resolve%20it%20from%20PyPI.%0A%0Aimport%20marimo%0A%0A__generated_with%20%3D%20%220.23.11%22%0Aapp%20%3D%20marimo.App(%0A%20%20%20%20width%3D%22full%22%2C%0A%20%20%20%20app_title%3D%22Hawk%2FDove%20with%20Multiple%20Risk-Attitudes%22%2C%0A%20%20%20%20html_head_file%3D%22docs%2Fdocs_head.html%22%2C%0A)%0A%0A%0A%40app.cell%0Aasync%20def%20_()%3A%0A%20%20%20%20import%20sys%0A%0A%20%20%20%20import%20marimo%20as%20mo%0A%0A%20%20%20%20if%20sys.platform%20%3D%3D%20%22emscripten%22%3A%0A%20%20%20%20%20%20%20%20%23%20Running%20in%20Pyodide%2FWASM.%0A%20%20%20%20%20%20%20%20%23%20mesa%202.1.5's%20__init__.py%20does%0A%20%20%20%20%20%20%20%20%23%20%60from%20mesa_viz_tornado.ModularVisualization%20import%20*%60%0A%20%20%20%20%20%20%20%20%23%20unconditionally.%20We%20never%20use%20that%20visualization%20layer%20(we%20draw%20with%20altair)%2C%0A%20%20%20%20%20%20%20%20%23%20so%20stub%20it%20out%20before%20importing%20mesa%20to%20avoid%20the%20ModuleNotFoundError.%0A%20%20%20%20%20%20%20%20import%20types%0A%0A%20%20%20%20%20%20%20%20_tornado_stub%20%3D%20types.ModuleType(%22mesa_viz_tornado%22)%0A%20%20%20%20%20%20%20%20sys.modules%5B%22mesa_viz_tornado%22%5D%20%3D%20_tornado_stub%0A%20%20%20%20%20%20%20%20for%20submodule%20in%20%5B%0A%20%20%20%20%20%20%20%20%20%20%20%20%22ModularVisualization%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%22modules%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%22UserParam%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%22TextVisualization%22%2C%0A%20%20%20%20%20%20%20%20%5D%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20import_name%20%3D%20f%22mesa_viz_tornado.%7Bsubmodule%7D%22%0A%20%20%20%20%20%20%20%20%20%20%20%20setattr(_tornado_stub%2C%20submodule%2C%20types.ModuleType(import_name))%0A%20%20%20%20%20%20%20%20%20%20%20%20sys.modules%5Bimport_name%5D%20%3D%20getattr(_tornado_stub%2C%20submodule)%0A%0A%20%20%20%20%20%20%20%20import%20micropip%0A%0A%20%20%20%20%20%20%20%20await%20micropip.install(%22mesa%3D%3D2.1.5%22%2C%20deps%3DFalse)%0A%20%20%20%20%20%20%20%20await%20micropip.install(%5B%22networkx%22%2C%20%22numpy%22%2C%20%22tqdm%22%5D)%0A%20%20%20%20%20%20%20%20%23%20uncomment%20and%20enable%20for%20local%20development%20involving%20app%20ui%0A%20%20%20%20%20%20%20%20%23%20await%20micropip.install(%0A%20%20%20%20%20%20%20%20%23%20%20%20%20%20str(%0A%20%20%20%20%20%20%20%20%23%20%20%20%20%20%20%20%20%20mo.notebook_location()%0A%20%20%20%20%20%20%20%20%23%20%20%20%20%20%20%20%20%20%2F%20%22public%22%0A%20%20%20%20%20%20%20%20%23%20%20%20%20%20%20%20%20%20%2F%20%22simulatingrisk-1.2.0-py3-none-any.whl%22%0A%20%20%20%20%20%20%20%20%23%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20%23%20)%0A%20%20%20%20%20%20%20%20await%20micropip.install(%22simulatingrisk%3E%3D1.2.0%22%2C%20deps%3DFalse)%0A%0A%20%20%20%20from%20simulatingrisk.hawkdove.ui%20import%20(%0A%20%20%20%20%20%20%20%20agent_portrayal%2C%0A%20%20%20%20%20%20%20%20draw_hawkdove_agent_space%2C%0A%20%20%20%20)%0A%20%20%20%20from%20simulatingrisk.hawkdovemulti.model%20import%20HawkDoveMultipleRiskModel%0A%20%20%20%20from%20simulatingrisk.hawkdovemulti.ui%20import%20ui_controls%0A%20%20%20%20from%20simulatingrisk.hawkdovemulti.viz%20import%20(%0A%20%20%20%20%20%20%20%20plot_agents_by_risk%2C%0A%20%20%20%20%20%20%20%20plot_hawks_by_risk%2C%0A%20%20%20%20%20%20%20%20plot_risklevel_changes%2C%0A%20%20%20%20%20%20%20%20plot_wealth_by_risklevel%2C%0A%20%20%20%20)%0A%20%20%20%20from%20simulatingrisk.ui_common%20import%20init_control_buttons%2C%20init_refresh%0A%20%20%20%20from%20simulatingrisk.doc_utils%20import%20docs_header%0A%0A%20%20%20%20return%20(%0A%20%20%20%20%20%20%20%20HawkDoveMultipleRiskModel%2C%0A%20%20%20%20%20%20%20%20agent_portrayal%2C%0A%20%20%20%20%20%20%20%20docs_header%2C%0A%20%20%20%20%20%20%20%20draw_hawkdove_agent_space%2C%0A%20%20%20%20%20%20%20%20init_control_buttons%2C%0A%20%20%20%20%20%20%20%20init_refresh%2C%0A%20%20%20%20%20%20%20%20mo%2C%0A%20%20%20%20%20%20%20%20plot_agents_by_risk%2C%0A%20%20%20%20%20%20%20%20plot_hawks_by_risk%2C%0A%20%20%20%20%20%20%20%20plot_risklevel_changes%2C%0A%20%20%20%20%20%20%20%20plot_wealth_by_risklevel%2C%0A%20%20%20%20%20%20%20%20ui_controls%2C%0A%20%20%20%20)%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20control_buttons%2C%0A%20%20%20%20docs_header%2C%0A%20%20%20%20mo%2C%0A%20%20%20%20refresh%2C%0A%20%20%20%20simulation_display%2C%0A%20%20%20%20simulation_status%2C%0A%20%20%20%20ui_controls%2C%0A)%3A%0A%20%20%20%20mo.hstack(%0A%20%20%20%20%20%20%20%20%5B%0A%20%20%20%20%20%20%20%20%20%20%20%20mo.vstack(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%5B%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20docs_header()%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20mo.md(%22---%22)%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20*ui_controls.values()%2C%20%20%23%20model%20parameters%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20mo.hstack(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20control_buttons%2C%20gap%3D1%2C%20align%3D%22start%22%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20)%2C%20%20%23%20simulation%20run%2Fstep%2Fstop%20controls%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20refresh%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20simulation_status%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%23%20make%20sure%20users%20know%20to%20reset%20to%20run%20with%20new%20parameters%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20mo.md(%22---%5Cn%E2%93%98%20Parameter%20changes%20take%20effect%20on%20**Reset**.%22)%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%5D%0A%20%20%20%20%20%20%20%20%20%20%20%20)%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20simulation_display%2C%0A%20%20%20%20%20%20%20%20%5D%2C%0A%20%20%20%20%20%20%20%20%23%20widths%3D%5B0%2C%201%5D%2C%20%20%23%20minimizing%20first%20doesn't%20work%20for%20all%20components%0A%20%20%20%20%20%20%20widths%3D%5B1%2C%204%5D%2C%0A%20%20%20%20%20%20%20%20align%3D%22start%22%2C%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(mo)%3A%0A%20%20%20%20%23%20Mutable%20simulation%20state%20%E2%80%94%20step_count%20increments%20on%20each%20model%20step%2C%0A%20%20%20%20%23%20which%20causes%20the%20chart%20cell%20to%20re-run%20and%20redraw.%0A%20%20%20%20model_state%2C%20set_model_state%20%3D%20mo.state(None)%0A%20%20%20%20step_count%2C%20set_step_count%20%3D%20mo.state(0)%0A%20%20%20%20%23%20display_step%20only%20updates%20every%20N%20steps%20for%20large%20grids%2C%20throttling%20chart%20re-renders.%0A%20%20%20%20%23%20The%20status%20bar%20uses%20step_count%20for%20accuracy%3B%20charts%20use%20display_step.%0A%20%20%20%20display_step%2C%20set_display_step%20%3D%20mo.state(0)%0A%20%20%20%20is_running%2C%20set_is_running%20%3D%20mo.state(False)%0A%20%20%20%20%23%20Remember%20the%20user's%20selected%20auto-refresh%20interval%20across%20pause%2Fresume%20so%0A%20%20%20%20%23%20we%20can%20re-arm%20the%20refresh%20widget%20at%20the%20same%20speed%20when%20play%20is%20clicked.%0A%20%20%20%20refresh_interval%2C%20set_refresh_interval%20%3D%20mo.state(%220.5s%22)%0A%20%20%20%20return%20(%0A%20%20%20%20%20%20%20%20display_step%2C%0A%20%20%20%20%20%20%20%20is_running%2C%0A%20%20%20%20%20%20%20%20model_state%2C%0A%20%20%20%20%20%20%20%20refresh_interval%2C%0A%20%20%20%20%20%20%20%20set_display_step%2C%0A%20%20%20%20%20%20%20%20set_is_running%2C%0A%20%20%20%20%20%20%20%20set_model_state%2C%0A%20%20%20%20%20%20%20%20set_refresh_interval%2C%0A%20%20%20%20%20%20%20%20set_step_count%2C%0A%20%20%20%20%20%20%20%20step_count%2C%0A%20%20%20%20)%0A%0A%0A%40app.cell%0Adef%20_(init_control_buttons)%3A%0A%20%20%20%20%23%20assign%20separately%20so%20we%20can%20watch%20the%20values%20of%20each%20button%0A%20%20%20%20run_btn%2C%20pause_btn%2C%20step_btn%2C%20reset_btn%20%3D%20init_control_buttons()%0A%20%20%20%20%23%20combine%20for%20display%20in%20the%20UI%0A%20%20%20%20control_buttons%20%3D%20run_btn%2C%20pause_btn%2C%20step_btn%2C%20reset_btn%0A%20%20%20%20return%20control_buttons%2C%20pause_btn%2C%20reset_btn%2C%20run_btn%2C%20step_btn%0A%0A%0A%40app.cell%0Adef%20_(init_refresh%2C%20is_running%2C%20refresh_interval%2C%20set_refresh_interval)%3A%0A%20%20%20%20refresh%20%3D%20init_refresh(%0A%20%20%20%20%20%20%20%20is_running%3Dis_running()%2C%0A%20%20%20%20%20%20%20%20refresh_interval%3Drefresh_interval()%2C%0A%20%20%20%20%20%20%20%20set_refresh_interval%3Dset_refresh_interval%2C%0A%20%20%20%20)%0A%20%20%20%20return%20(refresh%2C)%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20HawkDoveMultipleRiskModel%2C%0A%20%20%20%20mo%2C%0A%20%20%20%20pause_btn%2C%0A%20%20%20%20reset_btn%2C%0A%20%20%20%20run_btn%2C%0A%20%20%20%20set_display_step%2C%0A%20%20%20%20set_is_running%2C%0A%20%20%20%20set_model_state%2C%0A%20%20%20%20set_step_count%2C%0A%20%20%20%20ui_controls%2C%0A)%3A%0A%20%20%20%20%23%20Handle%20run%20%2F%20pause%20%2F%20reset%20button%20clicks.%0A%20%20%20%20mo.stop(not%20any(%5Brun_btn.value%2C%20pause_btn.value%2C%20reset_btn.value%5D))%0A%0A%20%20%20%20if%20run_btn.value%3A%0A%20%20%20%20%20%20%20%20set_is_running(True)%0A%20%20%20%20elif%20pause_btn.value%3A%0A%20%20%20%20%20%20%20%20set_is_running(False)%0A%20%20%20%20elif%20reset_btn.value%3A%0A%20%20%20%20%20%20%20%20_m%20%3D%20HawkDoveMultipleRiskModel(**%7Bk%3A%20v.value%20for%20k%2C%20v%20in%20ui_controls.items()%7D)%0A%20%20%20%20%20%20%20%20set_model_state(_m)%0A%20%20%20%20%20%20%20%20set_step_count(0)%0A%20%20%20%20%20%20%20%20set_display_step(0)%0A%20%20%20%20%20%20%20%20set_is_running(False)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20HawkDoveMultipleRiskModel%2C%0A%20%20%20%20is_running%2C%0A%20%20%20%20mo%2C%0A%20%20%20%20model_state%2C%0A%20%20%20%20refresh%2C%0A%20%20%20%20set_display_step%2C%0A%20%20%20%20set_is_running%2C%0A%20%20%20%20set_model_state%2C%0A%20%20%20%20set_step_count%2C%0A%20%20%20%20step_btn%2C%0A%20%20%20%20ui_controls%2C%0A)%3A%0A%20%20%20%20%23%20Advance%20the%20simulation.%20Triggered%20on%20every%20refresh%20tick%20and%20on%20step_btn%20click.%0A%20%20%20%20%23%20mo.stop%20exits%20when%20paused%20and%20the%20step%20button%20wasn't%20just%20pressed.%0A%20%20%20%20refresh%0A%20%20%20%20mo.stop(not%20(is_running()%20or%20step_btn.value))%0A%0A%20%20%20%20_model%20%3D%20model_state()%0A%20%20%20%20if%20_model%20is%20None%3A%0A%20%20%20%20%20%20%20%20_model%20%3D%20HawkDoveMultipleRiskModel(%0A%20%20%20%20%20%20%20%20%20%20%20%20**%7Bk%3A%20v.value%20for%20k%2C%20v%20in%20ui_controls.items()%7D%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20set_model_state(_model)%0A%0A%20%20%20%20if%20_model.running%3A%0A%20%20%20%20%20%20%20%20_model.step()%0A%20%20%20%20%20%20%20%20set_step_count(_model.schedule.steps)%0A%20%20%20%20%20%20%20%20%23%20throttle%20chart%20re-renders%20for%20large%20grids%3A%20only%20update%20display%20every%20N%20steps%0A%20%20%20%20%20%20%20%20%23%20always%20redraw%20on%20manual%20step%20so%20the%20user%20sees%20the%20result%20immediately%0A%20%20%20%20%20%20%20%20_n%20%3D%20max(1%2C%20_model.grid.width%20*%20_model.grid.height%20%2F%2F%20500)%0A%20%20%20%20%20%20%20%20if%20step_btn.value%20or%20_model.schedule.steps%20%25%20_n%20%3D%3D%200%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20set_display_step(_model.schedule.steps)%0A%20%20%20%20else%3A%0A%20%20%20%20%20%20%20%20set_is_running(False)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(is_running%2C%20mo%2C%20model_state%2C%20step_count)%3A%0A%20%20%20%20_model%20%3D%20model_state()%0A%20%20%20%20_step%20%3D%20step_count()%0A%20%20%20%20_run_label%20%3D%20%22%E2%96%B6%20Running%22%20if%20is_running()%20else%20%22%E2%8F%B8%20Paused%22%0A%20%20%20%20_sim_status%20%3D%20_model.status%20if%20_model%20else%20%22not%20started%22%0A%20%20%20%20_n%20%3D%20max(1%2C%20_model.grid.width%20*%20_model.grid.height%20%2F%2F%20500)%20if%20_model%20else%201%0A%20%20%20%20_refresh_note%20%3D%20f%22%5Cn%5Cn_Charts%20refresh%20every%20%7B_n%7D%20steps._%22%20if%20_n%20%3E%201%20else%20%22%22%0A%20%20%20%20simulation_status%20%3D%20mo.md(%0A%20%20%20%20%20%20%20%20f%22**Step%3A**%20%7B_step%7D%20%7C%20**Simulation%3A**%20%7B_run_label%7D%20%7C%20**Status%3A**%20%7B_sim_status%7D%7B_refresh_note%7D%22%0A%20%20%20%20)%0A%20%20%20%20return%20(simulation_status%2C)%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20agent_portrayal%2C%0A%20%20%20%20display_step%2C%0A%20%20%20%20draw_hawkdove_agent_space%2C%0A%20%20%20%20mo%2C%0A%20%20%20%20model_state%2C%0A%20%20%20%20plot_agents_by_risk%2C%0A%20%20%20%20plot_hawks_by_risk%2C%0A%20%20%20%20plot_risklevel_changes%2C%0A%20%20%20%20plot_wealth_by_risklevel%2C%0A)%3A%0A%20%20%20%20%23%20Charts%20%E2%80%94%20depends%20on%20display_step%2C%20which%20is%20throttled%20for%20large%20grids%20to%20reduce%20re-renders.%0A%20%20%20%20_step%20%3D%20display_step()%0A%20%20%20%20_model%20%3D%20model_state()%0A%0A%20%20%20%20%23%20smallest%20the%20agent%20grid%20should%20be%20displayed%0A%20%20%20%20_min_grid_dimension%20%3D%20275%0A%20%20%20%20%23%20width%20padding%20for%20legends%0A%20%20%20%20_grid_width_padding%20%3D%2075%0A%20%20%20%20_min_grid_height%20%3D%20325%0A%20%20%20%20_grid_space_per_agent%20%3D%2018%0A%0A%20%20%20%20if%20_model%20is%20None%3A%0A%20%20%20%20%20%20%20%20simulation_display%20%3D%20mo.md(%0A%20%20%20%20%20%20%20%20%20%20%20%20%22_Click%20**%E2%96%B6%20Run**%20or%20**%E2%8F%AD%20Step**%20to%20start%20the%20simulation._%22%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20else%3A%0A%20%20%20%20%20%20%20%20_grid%20%3D%20draw_hawkdove_agent_space(_model%2C%20agent_portrayal)%0A%20%20%20%20%20%20%20%20%23%20set%20grid%20chart%20based%20on%20grid%20size%20(grid%20size%20%3D%20%23%20agents%20in%20each%20dimension)%0A%20%20%20%20%20%20%20%20%23%20NOTE%3A%20set%20from%20model%2C%20not%20ui%2C%20since%20display%20must%20match%20current%20model%0A%20%20%20%20%20%20%20%20%23%20(changed%20ui%20values%20only%20applied%20on%20reset%2Frestart)%0A%20%20%20%20%20%20%20%20_grid_dimension%20%3D%20max(%0A%20%20%20%20%20%20%20%20%20%20%20%20_grid_space_per_agent%20*%20_model.grid.width%2C%20_min_grid_dimension%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20_grid_height%20%3D%20max(_min_grid_height%2C%20_grid_dimension)%0A%0A%20%20%20%20%20%20%20%20_grid%20%3D%20_grid.properties(%0A%20%20%20%20%20%20%20%20%20%20%20%20width%3D_grid_dimension%20%2B%20_grid_width_padding%2C%20height%3D_grid_height%0A%20%20%20%20%20%20%20%20)%0A%0A%20%20%20%20%20%20%20%20_agent_df%20%3D%20(%0A%20%20%20%20%20%20%20%20%20%20%20%20_model.datacollector.get_agent_vars_dataframe().reset_index().dropna()%0A%20%20%20%20%20%20%20%20)%0A%0A%20%20%20%20%20%20%20%20_charts%20%3D%20%5B_grid%5D%0A%0A%20%20%20%20%20%20%20%20if%20not%20_agent_df.empty%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20_charts.append(plot_agents_by_risk(_model))%0A%20%20%20%20%20%20%20%20%20%20%20%20_charts.append(plot_hawks_by_risk(_model))%0A%20%20%20%20%20%20%20%20%20%20%20%20_charts.append(plot_wealth_by_risklevel(_model))%0A%20%20%20%20%20%20%20%20%20%20%20%20_charts.append(plot_risklevel_changes(_model))%0A%0A%20%20%20%20%20%20%20%20%23%20if%20grid%20is%20close%20to%20chart%20size%2C%20just%20flow%20all%20the%20charts%20together%20and%20wrap%0A%20%20%20%20%20%20%20%20if%20_grid_dimension%20%3C%20350%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20simulation_display%20%3D%20mo.hstack(_charts%2C%20gap%3D0%2C%20wrap%3DTrue)%0A%20%20%20%20%20%20%20%20else%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20simulation_display%20%3D%20mo.hstack(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%5B_grid%2C%20mo.hstack(_charts%5B1%3A%5D%2C%20gap%3D0%2C%20wrap%3DTrue)%5D%2C%20gap%3D0%0A%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20return%20(simulation_display%2C)%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A