From 25b0004822cbd34f38334bcd26a198c84b37744c Mon Sep 17 00:00:00 2001
From: Adrian Heine <mail@adrianheine.de>
Date: Thu, 22 Jul 2021 22:48:02 +0200
Subject: [PATCH] Add benchmark

---
 Cargo.toml           |  7 +++++++
 benches/benchmark.rs | 48 ++++++++++++++++++++++++++++++++++++++++++++
 src/main.rs          |  2 +-
 src/world.rs         | 18 +++++++++++++----
 4 files changed, 70 insertions(+), 5 deletions(-)
 create mode 100644 benches/benchmark.rs

diff --git a/Cargo.toml b/Cargo.toml
index 83dc44d..ca0c338 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -7,3 +7,10 @@ authors = ["Adrian Heine <mail@adrianheine.de>"]
 [dependencies]
 tui = "0.15"
 termion = "1.5"
+
+[dev-dependencies]
+criterion = "0.3"
+
+[[bench]]
+name = "benchmark"
+harness = false
diff --git a/benches/benchmark.rs b/benches/benchmark.rs
new file mode 100644
index 0000000..cc2ed77
--- /dev/null
+++ b/benches/benchmark.rs
@@ -0,0 +1,48 @@
+use criterion::{BenchmarkId, measurement::Measurement};
+use criterion::{criterion_group, criterion_main, Criterion, Throughput, BenchmarkGroup};
+use gntag::agent::{Agent, SimpleAgent};
+use gntag::world::ActualWorld;
+
+fn world<M: Measurement>(mut group: BenchmarkGroup<M>, validate: bool) {
+  let width: isize = 1000;
+  for spacing in (50..100).step_by(10) {
+    group.throughput(Throughput::Elements((width / spacing).pow(2) as u64));
+    group.bench_with_input(
+      BenchmarkId::from_parameter(spacing),
+      &spacing,
+      |b, &spacing| {
+        let mut agents: Vec<(_, Box<dyn Agent>)> = vec![];
+        for x in (0..width).step_by(spacing as usize) {
+          for y in (0..width).step_by(spacing as usize) {
+            agents.push(((x, y).into(), Box::new(SimpleAgent)));
+          }
+        }
+        let mut world = ActualWorld::new((width, width).into(), agents, validate);
+        b.iter(|| world.do_step());
+      },
+    );
+  }
+  group.finish();
+}
+
+fn world_validating(c: &mut Criterion) {
+  let group = c.benchmark_group("world_validating");
+  world(group, true)
+}
+
+fn world_nonvalidating(c: &mut Criterion) {
+  let group = c.benchmark_group("world_nonvalidating");
+  world(group, false)
+}
+
+criterion_group!(benches, world_validating, world_nonvalidating);
+criterion_main!(benches);
+
+// 5000/1000 ->     9,993
+//    5/01 ->      23,252
+//   50/10 ->      14,000
+//  100/20 ->      14,000
+//  100/10 ->      96,000
+//  100/05 ->     917,767
+//  200/10 ->     916,172
+// 1000/10 -> 443,482,526
diff --git a/src/main.rs b/src/main.rs
index 518c107..8cb1afe 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -42,7 +42,7 @@ fn run_simulation(view: &Arc<Mutex<Option<TerminalView<impl Backend>>>>) {
       agents.push(((x, y).into(), Box::new(SimpleAgent)));
     }
   }
-  let mut world = ActualWorld::new((width, height).into(), agents);
+  let mut world = ActualWorld::new((width, height).into(), agents, true);
 
   let mut gen = 0;
   loop {
diff --git a/src/world.rs b/src/world.rs
index 59f46d5..9eb2c49 100644
--- a/src/world.rs
+++ b/src/world.rs
@@ -5,6 +5,7 @@ pub struct ActualWorld {
   size: Position,
   pub agents: HashMap<AgentId, Box<dyn Agent>>,
   pub state: WorldState,
+  validating: bool,
 }
 
 pub struct WorldState {
@@ -16,7 +17,11 @@ pub struct WorldState {
 impl ActualWorld {
   /// Agents receive incrementing ids starting with 0
   /// The last agent is 'It'
-  pub fn new(size: Position, input_agents: Vec<(Position, Box<dyn Agent>)>) -> Self {
+  pub fn new(
+    size: Position,
+    input_agents: Vec<(Position, Box<dyn Agent>)>,
+    validating: bool,
+  ) -> Self {
     let agent_count = input_agents.len();
     assert!(agent_count > 0);
     let mut id = 0;
@@ -36,6 +41,7 @@ impl ActualWorld {
       size,
       agents,
       state,
+      validating,
     }
   }
 
@@ -77,8 +83,9 @@ impl ActualWorld {
       tagged_by: self.state.tagged_by,
     };
     for (id, mv) in moves {
-      self.check_move(id, &mv);
-      //println!("{} {:?}", id, mv);
+      if self.validating {
+        self.check_move(id, &mv);
+      }
       let mut new_pos = None;
       match mv {
         Move::Noop => {}
@@ -135,7 +142,7 @@ mod test {
   #[test]
   #[should_panic]
   fn empty() {
-    ActualWorld::new(Position { x: 0, y: 0 }, vec![]);
+    ActualWorld::new(Position { x: 0, y: 0 }, vec![], true);
   }
 
   #[test]
@@ -143,6 +150,7 @@ mod test {
     let mut world = ActualWorld::new(
       Position { x: 0, y: 0 },
       vec![(Position { x: 0, y: 0 }, Box::new(NullAgent))],
+      true,
     );
     world.do_step();
     assert_eq!(world.state.tagged, 0);
@@ -163,6 +171,7 @@ mod test {
           Box::new(ScriptedAgent::new(vec![Move::TryTag(0)])),
         ),
       ],
+      true,
     );
     world.do_step();
     assert_eq!(world.state.tagged, 0);
@@ -178,6 +187,7 @@ mod test {
         Position { x: 0, y: 0 },
         Box::new(ScriptedAgent::new(vec![Move::TryMove((-1, -1).into())])),
       )],
+      true,
     );
     world.do_step();
   }