From 6880dfc96ba344d7eab533257ad80efaee027609 Mon Sep 17 00:00:00 2001
From: Zeno Rogue <zeno@attnam.com>
Date: Mon, 10 Aug 2020 18:41:07 +0200
Subject: [PATCH] rogueviz:: random walk:: improvements

---
 rogueviz/random-walk.cpp | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/rogueviz/random-walk.cpp b/rogueviz/random-walk.cpp
index a299dbee..26b2338f 100644
--- a/rogueviz/random-walk.cpp
+++ b/rogueviz/random-walk.cpp
@@ -67,7 +67,7 @@ bool advance_walkers(int delta) {
       continue;
     while(walkers[i].simulated < total_time) {
       walkers[i].simulated++;
-      if(rand() % branch_each == 0) {
+      if(branch_each && rand() % branch_each == 0) {
         walkers.push_back(walkers[i]);
         walkers.back().col = hrandpos() | 0x808080FF;
         walkers[i].col = hrandpos() | 0x808080FF;
@@ -176,7 +176,9 @@ void rw_slide(vector<tour::slide>& v, string title, string desc, reaction_t t) {
     setCanvas(mode, '0');
 
     if(mode == pmStart) {
+      tour::slide_backup(mapeditor::drawplayer, false);
       stop_game();
+      t();
       start_game();
       do_walkers = true;
       }
@@ -184,6 +186,12 @@ void rw_slide(vector<tour::slide>& v, string title, string desc, reaction_t t) {
     if(mode == pmStop) {
       do_walkers = false;
       }
+
+    if(mode == pmKey) {
+      drawn.clear();
+      walkers.clear();
+      total_time = 0;
+      }
     }}
     );
   }
@@ -198,11 +206,13 @@ auto msc =
 + addHook(hooks_clearmemory, 40, [] () {
     drawn.clear();
     walkers.clear();
+    total_time = 0;
     })
 + addHook(rvtour::hooks_build_rvtour, 180, [] (vector<tour::slide>& v) {
   v.push_back(tour::slide{
     cap+"random walk visualization", 10, tour::LEGAL::NONE | tour::QUICKSKIP,
     "Here we see random walk in various geometries.\n"
+    "Press '5' to reset.\n"
     ,
     [] (tour::presmode mode) {}
     });