rogueviz:: added heatx

rogueviz/heatx.cpp

@@ -0,0 +1,145 @@
+#include "rogueviz.h"
+#include <unordered_map>
+
+// heat equation simulation
+// https://twitter.com/ZenoRogue/status/1208409387733307392
+
+// run with e.g.
+// -geo 1 -canvas 0 -smart 1 -smartlimit 999999 -heatx
+// -tes tessellations/sample/marjorie-rice.tes heat_scale=0.02 -canvas 0 -smart 1 -smartlimit 999999 -heatx
+
+namespace hr {
+
+namespace heatx {
+
+const int NOT_STARTED = 999999;
+const int OFF = 999998;
+
+int last_steps = NOT_STARTED;
+
+std::unordered_map<cell*, double> m1, m2, m3;
+
+ld delta = 0.01;
+
+int mode = 1;
+
+int qsteps = 2000;
+
+ld frac_per_frame = .001;
+
+ld frac;
+
+ld scale = 0.05;
+
+void advance_heat_wave() {
+
+  if(euclid && GDIM == 2)
+    pconf.scale = scale / max(frac, .15);
+
+  int steps = mode == 2 ? (frac * qsteps) : (frac * frac * qsteps);
+
+  if(steps != last_steps) {
+    celllister cl(cwt.at, 400, 20000, nullptr);
+    if(steps < last_steps) {
+      last_steps = 0;
+      m1.clear();
+      m2.clear();
+      for(cell *c: cl.lst) m1[c] = 0;
+      m2 = m1;
+      m1[cwt.at] = 1;
+      }
+    while(last_steps < steps) {
+      switch(mode) {
+        case 0:
+          // heat: average of adjacent
+          for(cell *c: cl.lst) {
+            ld v = m1[c];
+            forCellEx(c2, c) {
+              if(m1.count(c2)) v += m1[c2]; else v += m1[c];
+              }
+            v /= (1 + c->type);
+            m2[c] = v;
+            }
+          swap(m1, m2);
+          break;
+        case 1:
+          // heat: transfer to adjacent
+          for(auto& p: m2) p.second = 0;
+          for(cell *c: cl.lst) {
+            ld v = m1[c] / (1 + c->type);
+            m2[c] += v;
+            forCellEx(c2, c) {
+              if(m1.count(c2)) m2[c2] += v; else m2[c] += v;
+              }
+            }
+          swap(m1, m2);
+          break;
+        case 2:
+          // wave
+          for(cell *c: cl.lst) {
+            m3[c] = 0;
+            forCellEx(c2, c) {
+              if(m1.count(c2)) m3[c] += (m1[c2] - m1[c]);
+              }
+            }
+          for(cell *c: cl.lst) {
+            m1[c] += m2[c] * delta + m3[c] * delta * delta / 2;
+            m2[c] += m3[c] * delta;
+            }
+          break;
+        }
+      last_steps++;
+      }
+    ld maxv = 0;
+    for(auto p: m1) maxv = max(maxv, abs(p.second));
+    for(cell *c: cl.lst) {
+      ld x = m1[c] / maxv;
+      if(mode == 2) {
+        if(x < 0) c->landparam = gradient(0x001010, 0x1010FF, -1, x, 0);
+        else c->landparam = gradient(0x1010FF, 0xFFFFFF, 0, x, 1);
+        }
+      else {
+        if(x < 1/2.) c->landparam = gradient(0x001010, 0xFF1010, 0, x, 1/2.);
+        else c->landparam = gradient(0xFF1010, 0xFFFF10, 1/2., x, 1.);
+        if(x > .2 && x < .3) c->landparam |= 0x4040;
+        }
+      }
+    }
+  
+  // return false;
+  }
+
+void enable() {
+  using rogueviz::rv_hook;
+  rv_hook(hooks_frame, 100, advance_heat_wave);
+  rv_hook(anims::hooks_anim, 100, [] {
+     if(inHighQual) {
+       frac = std::fmod(ticks, anims::period) * 1. / anims::period;
+       }
+     else {
+       frac += frac_per_frame;
+       if(frac > 1) frac--;
+       }
+     });
+  rv_hook(shot::hooks_take, 100, [] { 
+     advance_heat_wave(); calcparam(); models::configure(); 
+     });
+  rv_hook(hooks_drawcell, 100, [] (cell *c, const shiftmatrix& V) {
+    if(WDIM == 3) 
+      queuepoly(face_the_player(V), cgi.shRing, darkena(c->landparam_color, 0, 0xFF));
+    return false;
+    });
+  rogueviz::cleanup.push_back([] { m1.clear(); m2.clear(); m3.clear(); last_steps = OFF; });
+  last_steps = NOT_STARTED; frac = 0;
+  }
+
+auto heathook = arg::add3("-heatx", enable)
+  + addHook(hooks_configfile, 100, [] {
+    param_f(delta, "heat_delta");
+    param_i(qsteps, "heat_qsteps");
+    param_f(frac_per_frame, "heat_pf");
+    param_f(scale, "heat_scale");
+    });
+
+}
+}