iai bench

Cargo.lock

@@ -248,6 +248,12 @@ dependencies = [
  "libc",
 ]
 
+[[package]]
+name = "iai"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "71a816c97c42258aa5834d07590b718b4c9a598944cd39a52dc25b351185d678"
+
 [[package]]
 name = "instant"
 version = "0.1.9"
@@ -658,6 +664,7 @@ dependencies = [
  "enumflags2",
  "err-derive",
  "fnv",
+ "iai",
  "jemallocator",
  "main_error",
  "no-panic",

Cargo.toml

@@ -40,9 +40,14 @@ fnv = "1.0"
 [dev-dependencies]
 pretty_assertions = "0.6"
 test-case = "1.0.0"
+iai = "0.1"
 
 [profile.release]
 lto = true
 
 [profile.dev]
 opt-level = 2
+
+[[bench]]
+name = "iai"
+harness = false

benches/allmessages.rs

@@ -37,7 +37,7 @@ impl MessageHandler for AllMessages {
 
 fn bench_all(input_file: &str, b: &mut Bencher) {
     let file = fs::read(input_file).expect("Unable to read file");
-    let demo = Demo::new(file);
+    let demo = Demo::new(&file);
     let stream = demo.get_stream();
     b.iter(|| {
         let _ = test::black_box(

benches/bench.rs

@@ -9,7 +9,7 @@ use tf_demo_parser::{Demo, DemoParser, Stream};
 
 fn bench_file(input_file: &str, b: &mut Bencher) {
     let file = fs::read(input_file).expect("Unable to read file");
-    let demo = Demo::new(file);
+    let demo = Demo::new(&file);
     let stream: Stream = demo.get_stream();
 
     b.iter(|| {

benches/iai.rs

@@ -0,0 +1,36 @@
+use iai::black_box;
+use std::fs;
+use tf_demo_parser::demo::message::Message;
+use tf_demo_parser::demo::parser::MessageHandler;
+use tf_demo_parser::{Demo, DemoParser, MessageType, ParserState};
+
+struct AllMessages;
+
+impl MessageHandler for AllMessages {
+    type Output = bool;
+
+    fn does_handle(_message_type: MessageType) -> bool {
+        true
+    }
+
+    fn handle_message(&mut self, message: &Message, _tick: u32) {
+        black_box(message);
+    }
+
+    fn into_output(self, _state: &ParserState) -> Self::Output {
+        black_box(true)
+    }
+}
+
+fn bench_all() {
+    let file = fs::read("data/gully.dem").expect("Unable to read file");
+    let demo = Demo::new(&file);
+    let stream = demo.get_stream();
+    black_box(
+        DemoParser::new_with_analyser(stream.clone(), AllMessages)
+            .parse()
+            .unwrap(),
+    );
+}
+
+iai::main!(bench_all);

benches/sendprop.rs

@@ -41,7 +41,7 @@ impl MessageHandler for SendPropAnalyser {
 
 fn flatten_bench(input_file: &str, b: &mut Bencher) {
     let file = fs::read(input_file).expect("Unable to read file");
-    let demo = Demo::new(file);
+    let demo = Demo::new(&file);
     let stream = demo.get_stream();
     let (_, send_tables) = DemoParser::new_with_analyser(stream.clone(), SendPropAnalyser)
         .parse()